Ejemplo n.º 1
0
        public async Task <AppConnectionDescriptor> AuthenticateAsync(ITransportConnection connection)
        {
            Log.Trace("Accepting new connection {0}", connection.Id);
            var channel = await connection.IncomingChannels.ReadAsync().ConfigureAwait(false);

            var frame = await channel.In.ReadAsync().ConfigureAwait(false);

            using (var payload = frame.Payload)
                using (var connectRequest = _serializer.DeserializeConnectRequest(payload))
                {
                    if (!_registryService.IsApplicationDefined(connectRequest.ApplicationId))
                    {
                        throw new BrokerException($"Connection rejected because application id is unknown to broker: {connectRequest.ApplicationId}");
                    }
                    if (!_connectionTracker.IsAppInstanceRegistered(connectRequest.ApplicationInstanceId))
                    {
                        Log.Debug("Connection with unknown application instance id: "
                                  + $"ApplicationInstanceId={connectRequest.ApplicationInstanceId}, ApplicationId={connectRequest.ApplicationId}");
                        _connectionTracker.ReportConnectionError(new AppConnectionDescriptor(
                                                                     UniqueId.Empty,
                                                                     connectRequest.ApplicationId,
                                                                     connectRequest.ApplicationInstanceId,
                                                                     connection.TransportType));

                        if (_features.HasFlag(BrokerFeatures.CheckAppInstanceId))
                        {
                            throw new BrokerException("Connection rejected because application instance id is unknown to broker: "
                                                      + $"ApplicationInstanceId={connectRequest.ApplicationInstanceId}, ApplicationId={connectRequest.ApplicationId}");
                        }
                    }
                    using (var connectResponse = _messageFactory.CreateConnectResponse(connection.Id))
                    {
                        var serializedResponse = _serializer.Serialize(connectResponse);
                        try
                        {
                            Log.Trace("Sending connect response ({0} bytes): {1}", connectResponse, serializedResponse.Count);
                            await channel.Out.WriteAsync(new TransportMessageFrame(serializedResponse)).ConfigureAwait(false);
                        }
                        catch
                        {
                            serializedResponse.Dispose();
                            throw;
                        }
                        channel.Out.TryComplete();
                        await channel.Completion.ConfigureAwait(false);

                        return(new AppConnectionDescriptor(
                                   connectResponse.ConnectionId,
                                   connectRequest.ApplicationId,
                                   connectRequest.ApplicationInstanceId,
                                   connection.TransportType));
                    }
                }
        }
Ejemplo n.º 2
0
        public async Task <IAppConnection> HandleAsync(ITransportConnection connection)
        {
            Log.Trace("Accepting new connection {0}", connection.Id);
            var channel = await connection.IncomingChannels.ReadAsync().ConfigureAwait(false);

            var frame = await channel.In.ReadAsync().ConfigureAwait(false);

            using (var payload = frame.Payload)
                using (var connectRequest = _serializer.DeserializeConnectRequest(payload))
                {
                    if (!_registryService.IsApplicationDefined(connectRequest.ApplicationId))
                    {
                        throw new BrokerException($"Connection rejected because application id is unknown to broker: {connectRequest.ApplicationId}");
                    }
                    using (var connectResponse = _messageFactory.CreateConnectResponse(connection.Id))
                    {
                        var serializedResponse = _serializer.Serialize(connectResponse);
                        try
                        {
                            Log.Trace("Sending connect response ({0} bytes): {1}", connectResponse, serializedResponse.Count);
                            await channel.Out.WriteAsync(new TransportMessageFrame(serializedResponse)).ConfigureAwait(false);
                        }
                        catch
                        {
                            serializedResponse.Dispose();
                            throw;
                        }
                        channel.Out.TryComplete();
                        await channel.Completion.ConfigureAwait(false);

                        var info =
                            new AppConnectionDescriptor(
                                connectResponse.ConnectionId,
                                connectRequest.ApplicationId,
                                connectRequest.ApplicationInstanceId);
                        var clientConnection = _connectionTracker.AcceptConnection(connection, info);
                        Log.Info("New connection accepted: {0}", clientConnection);
                        return(clientConnection);
                    }
                }
        }