Example #1
0
 public async Task StartReceive()
 {
     quitFlag = false;
     while (!quitFlag)
     {
         List <byte[]> datas;
         bool          succeed;
         if (_dataChannel)
         {
             succeed = _rtsp.TryDequeData(out datas, 10);
         }
         else
         {
             succeed = _rtsp.TryDequeControl(out datas, 1);
         }
         if (succeed)
         {
             foreach (byte[] data in datas)
             {
                 if (!_dataChannel)
                 {
                     WSRtspResponse response = new WSRtspResponse();
                     response.Seq        = Seq;
                     response.RtspBuffer = data;
                     await _ws.SendAsync(response.ToArray(), _dataChannel?WebSocketMessageType.Binary : WebSocketMessageType.Text, true, CancellationToken.None);
                 }
                 else
                 {
                     await _ws.SendAsync(data, _dataChannel?WebSocketMessageType.Binary : WebSocketMessageType.Text, true, CancellationToken.None);
                 }
             }
         }
         else
         {
             if (_dataChannel)
             {
                 await Task.Delay(1);
             }
             else
             {
                 await Task.Delay(10);
             }
         }
     }
 }
Example #2
0
        public override async Task ReceiveAsync(WebSocket socket, WebSocketReceiveResult result, byte[] buffer)
        {
            if (buffer == null || buffer.Length == 0)
            {
                return;
            }
            var           socketId = WebSocketConnectionManager.GetId(socket);
            WSRtspContext wsrtsp   = null;

            if (ws_rtsps.ContainsKey(socketId))
            {
                wsrtsp = ws_rtsps[socketId];
            }
            if (result.MessageType == WebSocketMessageType.Text)
            {
                string package = Encoding.UTF8.GetString(buffer);
                string command = getWSPCommand(package);
                string seq     = getByKey(package, "seq");
                if (command == "INIT")//建立新链接.
                {
                    string host = getByKey(package, "host");
                    string port = getByKey(package, "port");
                    if (port == null)
                    {
                        port = "554";
                    }
                    if (host != null)
                    {
                        try
                        {
                            RtspProxy rtsp      = new RtspProxy();
                            bool      connected = rtsp.Connect(host, Convert.ToInt32(port));
                            if (connected)
                            {
                                rtsp.Start();
                                wsrtsp = new WSRtspContext(socket, rtsp, false);
                                wsrtsp.ControlWebSocketId = socketId;
                                ws_rtsps.TryAdd(socketId, wsrtsp);
                                wsrtsp.Seq = seq;
                                //返回握手.
                                WSRtspResponse response = new WSRtspResponse();
                                response.Seq       = seq;
                                response.Shakehand = true;
                                response.Channel   = socketId;
                                await socket.SendAsync(response.ToArray(), WebSocketMessageType.Text, true, CancellationToken.None);

                                //启动接受rtsp控制报文,发送给ws.
                                await wsrtsp.StartReceive();
                            }
                            else
                            {
                                await removeSocket(wsrtsp);

                                return;
                            }
                        }
                        catch (Exception ex)
                        {
                            //Logger.Error(String.Format("connect to rtsp {0} Error:{1}", package, ex.Message));
                            return;
                        }
                    }
                }
                else if (command == "JOIN")//建立数据通道.
                {
                    string channel = getByKey(package, "channel");
                    if (channel != null && ws_rtsps.ContainsKey(channel))
                    {
                        WSRtspContext controlwsrtsp = ws_rtsps[channel];
                        controlwsrtsp.DataWebSocketId = socketId;
                        WSRtspContext datawsrtsp = new WSRtspContext(socket, controlwsrtsp.Rtsp, true);
                        datawsrtsp.ControlWebSocketId = controlwsrtsp.ControlWebSocketId;
                        datawsrtsp.DataWebSocketId    = socketId;
                        ws_rtsps.TryAdd(socketId, datawsrtsp);
                        //返回握手.
                        WSRtspResponse response = new WSRtspResponse();
                        response.Seq = seq;
                        await socket.SendAsync(response.ToArray(), WebSocketMessageType.Text, true, CancellationToken.None);

                        //启动接受rtsp数据报文,发送给ws.
                        await datawsrtsp.StartReceive();
                    }
                    return;
                }//WSP/1.1 JOIN channel: 127.0.0.1 - 2 18467 seq: 3
                else
                {
                    wsrtsp.Seq = seq;
                    try
                    {
                        await wsrtsp.Send(getRtspBuffer(package));
                    }
                    catch
                    {
                        await removeSocket(wsrtsp);
                    }
                }
            }
            else if (wsrtsp != null)
            {
                try
                {
                    await wsrtsp.Send(buffer);
                }
                catch
                {
                    await removeSocket(wsrtsp);
                }
            }//WebSocketMessageType.Data
        }