Exemplo n.º 1
0
        protected void NegotiateChannels(Connection.Tcp connection)
        {
            Message.Base msg;

            if (connection.Receive(out msg))
            {
                if (msg.IsType <Message.Negotiation.Start>())
                {
                    connection.Parameters.guid = Guid.NewGuid();

                    SaveMonitor(connection);

                    // Send a new Guid for these connections and the number of associated channels
                    Message.Negotiation.New n = new Message.Negotiation.New
                    {
                        guid       = connection.Parameters.guid,
                        nbChannels = (ushort)Math.Min(channels != null ? channels.Count : 0, ushort.MaxValue)
                    };

                    connection.Send(n);

                    if (n.nbChannels <= 0)
                    {
                        Drop(connection, "No channels configured.");
                    }

                    for (ushort i = 0; i < n.nbChannels; i++)
                    {
                        ServerChannel channel = channels[i];

                        Message.Negotiation.Parameters param = new Message.Negotiation.Parameters
                        {
                            guid          = n.guid,
                            channel       = i,
                            type          = channel.type,
                            heartbeat     = channel.parameters.Heartbeat,
                            autoReconnect = !channel.parameters.disableAutoReconnect
                        };

                        connection.Send(param);
                    }
                }
                else if (msg.IsType <Message.Negotiation.Channel.TCP>())
                {
                    Message.Negotiation.Channel.TCP tcp = (Message.Negotiation.Channel.TCP)msg;

                    connection.SetConfig(tcp.guid, tcp.channel, channels[tcp.channel].parameters.Heartbeat);

                    SaveChannel(connection);
                }
                else
                {
                    Drop(connection, "Unsupported negotiation command '{0}'.", msg.GetType());
                }
            }
            else
            {
                Drop(connection, "Expected to receive some negotiation command.");
            }
        }
Exemplo n.º 2
0
        protected void NegotiateChannels(Connection.Tcp connection)
        {
            // Check if we must negotiate other channel or just open the current one
            if (connection.Remote == Guid.Empty)
            {
                connection.Send(new Message.Negotiation.Start());

                Message.Base msg;

                if (connection.Receive(out msg) && msg.IsType <Message.Negotiation.New>())
                {
                    Message.Negotiation.New n = (Message.Negotiation.New)msg;

                    connection.Parameters.guid = n.guid;

                    ushort nb_channels = n.nbChannels;

                    if (nb_channels > 0)
                    {
                        List <Message.Negotiation.Parameters> parameters = new List <Message.Negotiation.Parameters>(nb_channels);

                        for (ushort i = 0; i < nb_channels; i++)
                        {
                            if (connection.Receive(out msg) && msg.IsType <Message.Negotiation.Parameters>())
                            {
                                Message.Negotiation.Parameters param = (Message.Negotiation.Parameters)msg;

                                parameters.Add(param);
                            }
                            else
                            {
                                Drop(connection, "Expected to receive channel parameterts for channel {0}.", i);
                            }
                        }

                        SaveMonitor(connection);

                        foreach (Message.Negotiation.Parameters param in parameters)
                        {
                            if (!Ready(param.guid, param.channel))
                            {
                                switch (param.type)
                                {
                                case Channel.Type.TCP:
                                    ConnectTcp(param);
                                    break;

                                case Channel.Type.UDP:
                                    ConnectUdp(connection, param);
                                    break;
                                }
                            }
                        }

                        state = State.RUNNING;
                    }
                    else
                    {
                        Drop(connection, "No channels configured.");
                    }
                }
                else
                {
                    Drop(connection, "Expected to receive the new connection negotiation parameters.");
                }
            }
            else
            {
                Message.Negotiation.Channel.TCP tcp = new Message.Negotiation.Channel.TCP
                {
                    guid    = connection.Remote,
                    channel = connection.Channel
                };

                connection.Send(tcp);

                SaveChannel(connection);
            }
        }