public override async Task OnDisconnected(WebSocket socket) { var socketId = WebSocketConnectionManager.GetId(socket); if (ws_rtsps.ContainsKey(socketId)) { WSRtspContext wsrtsp; ws_rtsps.TryRemove(socketId, out wsrtsp); wsrtsp.StopReceive(); if (wsrtsp.Rtsp != null) { wsrtsp.Rtsp.Close(); } }//关闭rtsp. await base.OnDisconnected(socket); }
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 }