Esempio n. 1
0
        public RtmpServer(
            SerializationContext context,
            X509Certificate2 cert                      = null,
            ObjectEncoding object_encoding             = ObjectEncoding.Amf0,
            ParameterAuthCallback publishParameterAuth = null,
            ParameterAuthCallback playParameterAuth    = null,
            string bindIp         = "0.0.0.0",
            int bindRtmpPort      = 1935,
            int bindWebsocketPort = -1
            )
        {
            this.context   = context;
            objectEncoding = object_encoding;
            if (bindWebsocketPort != -1)
            {
                var server = new WebSocketServer("ws://" + bindIp.ToString() + ":" + bindWebsocketPort.ToString());
                if (cert != null)
                {
                    this.cert                  = cert;
                    server.Certificate         = cert;
                    server.EnabledSslProtocols = SslProtocols.Default;
                }
                server.ListenerSocket.NoDelay = true;
                server.Start(socket =>
                {
                    socket.OnOpen = () =>
                    {
                        var path = socket.ConnectionInfo.Path.Split('/');
                        if (path.Length != 3)
                        {
                            socket.Close();
                        }
                        ushort client_id       = GetNewClientId();
                        IStreamConnect connect = new WebsocketConnect(socket, context, object_encoding);
                        lock (connects)
                        {
                            connects.Add(client_id, new StreamConnectState()
                            {
                                Connect    = connect,
                                LastPing   = DateTime.UtcNow,
                                ReaderTask = null,
                                WriterTask = null
                            });
                        }
                        try
                        {
                            SendMetadata(path[1], path[2], connect, flvHeader: true);

                            ConnectToClient(path[1], path[2], client_id, ChannelType.Audio);
                            ConnectToClient(path[1], path[2], client_id, ChannelType.Video);
                        }
                        catch { Close(client_id); }
                    };
                    socket.OnPing = b => socket.SendPong(b);
                });
            }

            if (publishParameterAuth != null)
            {
                this.publishParameterAuth = publishParameterAuth;
            }
            if (playParameterAuth != null)
            {
                this.playParameterAuth = playParameterAuth;
            }
            ioThread = new Thread(() =>
            {
                while (true)
                {
                    foreach (var current in connects)
                    {
                        StreamConnectState state = current.Value;
                        ushort client_id         = current.Key;
                        IStreamConnect connect   = state.Connect;
                        if (connect.IsDisconnected)
                        {
                            Close(client_id);
                            continue;
                        }
                        try
                        {
                            if (state.WriterTask == null || state.WriterTask.IsCompleted)
                            {
                                state.WriterTask = connect.WriteOnceAsync();
                            }
                            if (state.WriterTask.IsCanceled || state.WriterTask.IsFaulted)
                            {
                                throw state.WriterTask.Exception;
                            }
                            if (state.LastPing == null || DateTime.UtcNow - state.LastPing >= new TimeSpan(0, 0, PingPeriod))
                            {
                                connect.PingAsync(PingTimeout);
                                state.LastPing = DateTime.UtcNow;
                            }


                            if (state.ReaderTask == null || state.ReaderTask.IsCompleted)
                            {
                                state.ReaderTask = connect.ReadOnceAsync();
                            }
                            if (state.ReaderTask.IsCanceled || state.ReaderTask.IsFaulted)
                            {
                                throw state.ReaderTask.Exception;
                            }
                        }
                        catch
                        {
                            Close(client_id);
                            continue;
                        }
                    }
                    var prepare_add_length = prepare_to_add.Count;
                    if (prepare_add_length != 0)
                    {
                        for (int i = 0; i < prepare_add_length; i++)
                        {
                            var current = prepare_to_add[0];
                            connects.Add(current.Key, current.Value);
                            prepare_to_add.RemoveAt(0);
                        }
                    }


                    var prepare_remove_length = prepare_to_remove.Count;
                    if (prepare_remove_length != 0)
                    {
                        for (int i = 0; i < prepare_remove_length; i++)
                        {
                            var current = prepare_to_remove[0];
                            connects.Remove(current);
                            prepare_to_remove.RemoveAt(0);
                        }
                    }
                }
            })
            {
                IsBackground = true
            };

            ioThread.Start();

            listener         = new Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp);
            listener.NoDelay = true;
            IPEndPoint localEndPoint = new IPEndPoint(IPAddress.Parse(bindIp), bindRtmpPort);

            listener.Bind(localEndPoint);
            listener.Listen(10);
        }
Esempio n. 2
0
        /// <summary>
        /// 启动连接状态检查线程
        /// 执行rtmp的读写异步任务
        /// </summary>
        void ClientWorkHandler()
        {
            _clientWorkThread = new Thread(() =>
            {
                while (true)
                {
                    try
                    {
                        if (ClientSessions.Count() < 1)
                        {
                            Thread.Sleep(1);
                            continue;
                        }

                        ClientSessionDictionary nclientSessions = null;

                        lock (_locker)
                        {
                            nclientSessions = ClientSessions.Clone() as ClientSessionDictionary;
                        }

                        Parallel.ForEach(nclientSessions, current =>
                        {
                            ClientSession state    = current.Value;
                            ushort client_id       = current.Key;
                            IStreamConnect connect = state.Connect;

                            try
                            {
                                if (connect.IsDisconnected)
                                {
                                    DisconnectSession(client_id);
                                }
                                else
                                {
                                    if (state.WriterTask == null)
                                    {
                                        state.WriterTask = connect.WriteOnceAsync();
                                    }
                                    else
                                    {
                                        if (state.WriterTask.IsCompleted)
                                        {
                                            state.WriterTask = connect.WriteOnceAsync();
                                        }
                                        if (state.WriterTask.IsCanceled || state.WriterTask.IsFaulted)
                                        {
                                            this.DisconnectSession(current.Key);
                                            //throw state.WriterTask.Exception;
                                        }
                                        if (state.LastPing == null || DateTime.UtcNow - state.LastPing >= new TimeSpan(0, 0, _pingPeriod))
                                        {
                                            connect.PingAsync(_pingTimeout);
                                            state.LastPing = DateTime.UtcNow;
                                        }

                                        if (state.ReaderTask == null || state.ReaderTask.IsCompleted)
                                        {
                                            state.ReaderTask = connect.ReadOnceAsync();
                                        }

                                        if (state.ReaderTask.IsCanceled || state.ReaderTask.IsFaulted)
                                        {
                                            this.DisconnectSession(current.Key);
                                            //throw state.ReaderTask.Exception;
                                        }
                                    }
                                }
                            }
                            catch
                            {
                                DisconnectSession(client_id);
                            }
                        });
                    }
                    catch (Exception ex)
                    {
                        FleckLog.Error("ConnectStateCheckUp.Thread.Error", ex);
                    }
                }
            })
            {
                IsBackground = true
            };
            _clientWorkThread.Start();
        }