private void ReadFrameAsync() { try { //frame parameters int signalType; BinaryID channelName = new BinaryID(new byte[20]); int dataLength; byte[] dataBuffer = new byte[BUFFER_SIZE]; while (true) { #region Read frame from base stream //read frame signal signalType = _baseStream.ReadByte(); if (signalType == -1) { return; //End of stream } //read channel name OffsetStream.StreamRead(_baseStream, channelName.ID, 0, 20); //read data length OffsetStream.StreamRead(_baseStream, dataBuffer, 0, 2); dataLength = BitConverter.ToUInt16(dataBuffer, 0); //read data if (dataLength > 0) { OffsetStream.StreamRead(_baseStream, dataBuffer, 0, dataLength); } #endregion switch ((SignalType)signalType) { case SignalType.NOOP: break; case SignalType.ConnectChannelBitChatNetwork: #region ConnectChannelBitChatNetwork lock (_bitChatNetworkChannels) { if (_bitChatNetworkChannels.ContainsKey(channelName)) { WriteFrame(SignalType.DisconnectChannelBitChatNetwork, channelName, null, 0, 0); } else { ChannelStream channel = new ChannelStream(this, channelName.Clone(), ChannelType.BitChatNetwork); _bitChatNetworkChannels.Add(channel.ChannelName, channel); BitChatNetworkChannelRequest.BeginInvoke(this, channel.ChannelName, channel, null, null); } } //check if tcp relay is hosted for the channel. reply back tcp relay peers list if available try { List <IPEndPoint> peerEPs = TcpRelayService.GetPeerEPs(channelName, this); if ((peerEPs != null) && (peerEPs.Count > 0)) { using (MemoryStream mS = new MemoryStream(128)) { mS.WriteByte(Convert.ToByte(peerEPs.Count)); foreach (IPEndPoint peerEP in peerEPs) { IPEndPointParser.WriteTo(peerEP, mS); } byte[] data = mS.ToArray(); WriteFrame(SignalType.TcpRelayResponsePeerList, channelName, data, 0, data.Length); } } } catch { } #endregion break; case SignalType.DataChannelBitChatNetwork: #region DataChannelBitChatNetwork try { ChannelStream channel = null; lock (_bitChatNetworkChannels) { channel = _bitChatNetworkChannels[channelName]; } channel.WriteBuffer(dataBuffer, 0, dataLength, _channelWriteTimeout); } catch { } #endregion break; case SignalType.DisconnectChannelBitChatNetwork: #region DisconnectChannelBitChatNetwork try { ChannelStream channel; lock (_bitChatNetworkChannels) { channel = _bitChatNetworkChannels[channelName]; _bitChatNetworkChannels.Remove(channelName); } channel.SetDisconnected(); channel.Dispose(); } catch { } #endregion break; case SignalType.ConnectChannelProxyTunnel: #region ConnectChannelProxyTunnel { ChannelStream remoteChannel1 = null; lock (_proxyChannels) { if (_proxyChannels.ContainsKey(channelName)) { WriteFrame(SignalType.DisconnectChannelProxy, channelName, null, 0, 0); } else { //add first stream into list remoteChannel1 = new ChannelStream(this, channelName.Clone(), ChannelType.ProxyTunnel); _proxyChannels.Add(remoteChannel1.ChannelName, remoteChannel1); } } if (remoteChannel1 != null) { IPEndPoint tunnelToRemotePeerEP = ConvertChannelNameToEp(channelName); //get remote peer ep Connection remotePeerConnection = _connectionManager.GetExistingConnection(tunnelToRemotePeerEP); //get remote channel service if (remotePeerConnection == null) { remoteChannel1.Dispose(); } else { try { //get remote proxy connection channel stream ChannelStream remoteChannel2 = remotePeerConnection.RequestProxyConnection(_remotePeerEP); //join current and remote stream Joint joint = new Joint(remoteChannel1, remoteChannel2); joint.Disposed += joint_Disposed; lock (_proxyTunnelJointList) { _proxyTunnelJointList.Add(joint); } joint.Start(); } catch { remoteChannel1.Dispose(); } } } } #endregion break; case SignalType.ConnectChannelProxyConnection: #region ConnectChannelProxyConnection lock (_proxyChannels) { if (_proxyChannels.ContainsKey(channelName)) { WriteFrame(SignalType.DisconnectChannelProxy, channelName, null, 0, 0); } else { //add proxy channel stream into list ChannelStream channel = new ChannelStream(this, channelName.Clone(), ChannelType.ProxyTunnel); _proxyChannels.Add(channel.ChannelName, channel); //pass channel as connection async ThreadPool.QueueUserWorkItem(AcceptProxyConnectionAsync, channel); } } #endregion break; case SignalType.DataChannelProxy: #region DataChannelProxy try { ChannelStream channel = null; lock (_proxyChannels) { channel = _proxyChannels[channelName]; } channel.WriteBuffer(dataBuffer, 0, dataLength, _channelWriteTimeout); } catch { } #endregion break; case SignalType.DisconnectChannelProxy: #region DisconnectChannelProxy try { ChannelStream channel; lock (_proxyChannels) { channel = _proxyChannels[channelName]; _proxyChannels.Remove(channelName); } channel.SetDisconnected(); channel.Dispose(); } catch { } #endregion break; case SignalType.PeerStatusQuery: #region PeerStatusQuery try { if (_connectionManager.IsPeerConnectionAvailable(ConvertChannelNameToEp(channelName))) { WriteFrame(SignalType.PeerStatusAvailable, channelName, null, 0, 0); } } catch { } #endregion break; case SignalType.PeerStatusAvailable: #region PeerStatusAvailable try { lock (_peerStatusLockList) { object lockObject = _peerStatusLockList[channelName]; lock (lockObject) { Monitor.Pulse(lockObject); } } } catch { } #endregion break; case SignalType.StartTcpRelay: #region StartTcpRelay { BinaryID[] networkIDs; Uri[] trackerURIs; using (MemoryStream mS = new MemoryStream(dataBuffer, 0, dataLength, false)) { //read network id list networkIDs = new BinaryID[mS.ReadByte()]; byte[] XORnetworkID = new byte[20]; for (int i = 0; i < networkIDs.Length; i++) { mS.Read(XORnetworkID, 0, 20); byte[] networkID = new byte[20]; for (int j = 0; j < 20; j++) { networkID[j] = (byte)(channelName.ID[j] ^ XORnetworkID[j]); } networkIDs[i] = new BinaryID(networkID); } //read tracker uri list trackerURIs = new Uri[mS.ReadByte()]; byte[] data = new byte[255]; for (int i = 0; i < trackerURIs.Length; i++) { int length = mS.ReadByte(); mS.Read(data, 0, length); trackerURIs[i] = new Uri(Encoding.UTF8.GetString(data, 0, length)); } } lock (_tcpRelays) { foreach (BinaryID networkID in networkIDs) { if (!_tcpRelays.ContainsKey(networkID)) { TcpRelayService relay = TcpRelayService.StartTcpRelay(networkID, this, _connectionManager.LocalPort, _connectionManager.DhtClient, trackerURIs); _tcpRelays.Add(networkID, relay); } } } WriteFrame(SignalType.TcpRelayResponseSuccess, channelName, null, 0, 0); } #endregion break; case SignalType.StopTcpRelay: #region StopTcpRelay { BinaryID[] networkIDs; using (MemoryStream mS = new MemoryStream(dataBuffer, 0, dataLength, false)) { //read network id list networkIDs = new BinaryID[mS.ReadByte()]; byte[] XORnetworkID = new byte[20]; for (int i = 0; i < networkIDs.Length; i++) { mS.Read(XORnetworkID, 0, 20); byte[] networkID = new byte[20]; for (int j = 0; j < 20; j++) { networkID[j] = (byte)(channelName.ID[j] ^ XORnetworkID[j]); } networkIDs[i] = new BinaryID(networkID); } } lock (_tcpRelays) { foreach (BinaryID networkID in networkIDs) { if (_tcpRelays.ContainsKey(networkID)) { _tcpRelays[networkID].StopTcpRelay(this); _tcpRelays.Remove(networkID); } } } WriteFrame(SignalType.TcpRelayResponseSuccess, channelName, null, 0, 0); } #endregion break; case SignalType.TcpRelayResponseSuccess: #region TcpRelayResponseSuccess try { lock (_tcpRelayRequestLockList) { object lockObject = _tcpRelayRequestLockList[channelName]; lock (lockObject) { Monitor.Pulse(lockObject); } } } catch { } #endregion break; case SignalType.TcpRelayResponsePeerList: #region TcpRelayResponsePeerList using (MemoryStream mS = new MemoryStream(dataBuffer, 0, dataLength, false)) { int count = mS.ReadByte(); List <IPEndPoint> peerEPs = new List <IPEndPoint>(count); for (int i = 0; i < count; i++) { peerEPs.Add(IPEndPointParser.Parse(mS)); } TcpRelayPeersAvailable.BeginInvoke(this, channelName.Clone(), peerEPs, null, null); } #endregion break; case SignalType.DhtPacketData: #region DhtPacketData byte[] response = _connectionManager.DhtClient.ProcessPacket(dataBuffer, 0, dataLength, _remotePeerEP.Address); if (response != null) { WriteFrame(SignalType.DhtPacketData, BinaryID.GenerateRandomID160(), response, 0, response.Length); } #endregion break; case SignalType.BitChatNetworkInvitation: #region ChannelInvitationBitChatNetwork if (_connectionManager.Profile.AllowInboundInvitations) { BitChatNetworkInvitation.BeginInvoke(channelName.Clone(), _remotePeerEP, Encoding.UTF8.GetString(dataBuffer, 0, dataLength), null, null); } #endregion break; default: throw new IOException("Invalid frame signal type."); } } } catch { } finally { Dispose(); } }
private void ReadDataAsync() { try { int channelDataLength; int frameSignal; byte[] channelNameBuffer = new byte[20]; BinaryID channelName = new BinaryID(channelNameBuffer); ChannelType channelType; byte[] buffer = new byte[65536]; while (true) { //read frame signal frameSignal = _baseStream.ReadByte(); //read channel name OffsetStream.StreamRead(_baseStream, channelNameBuffer, 0, 20); switch (frameSignal) { case SIGNAL_DATA: { //read channel type channelType = (ChannelType)_baseStream.ReadByte(); OffsetStream.StreamRead(_baseStream, buffer, 0, 2); channelDataLength = BitConverter.ToUInt16(buffer, 0) + 1; OffsetStream.StreamRead(_baseStream, buffer, 0, channelDataLength); //switch frame ChannelStream channel = null; try { switch (channelType) { case ChannelType.BitChatNetwork: lock (_bitChatNetworkChannels) { channel = _bitChatNetworkChannels[channelName]; } channel.WriteBuffer(buffer, 0, channelDataLength, _channelWriteTimeout); break; case ChannelType.ProxyTunnel: lock (_proxyTunnelChannels) { channel = _proxyTunnelChannels[channelName]; } channel.WriteBuffer(buffer, 0, channelDataLength, _channelWriteTimeout); break; case ChannelType.VirtualConnection: lock (_virtualConnectionChannels) { channel = _virtualConnectionChannels[channelName]; } channel.WriteBuffer(buffer, 0, channelDataLength, _channelWriteTimeout); break; } } catch { if (channel != null) { channel.Dispose(); } } } break; case SIGNAL_CONNECT_BIT_CHAT_NETWORK: { ChannelStream channel = null; try { channel = new ChannelStream(this, channelName, ChannelType.BitChatNetwork); lock (_bitChatNetworkChannels) { _bitChatNetworkChannels.Add(channelName, channel); } Debug.Write("Connection.ReadDataAsync", "SIGNAL_CONNECT_BIT_CHAT_NETWORK; channel: " + channelName.ToString()); _requestHandler.BeginInvoke(this, channelName, ChannelType.BitChatNetwork, channel, null, null); } catch { if (channel != null) { channel.Dispose(); } } } break; case SIGNAL_DISCONNECT_BIT_CHAT_NETWORK: try { lock (_bitChatNetworkChannels) { _bitChatNetworkChannels[channelName].Dispose(); } Debug.Write("Connection.ReadDataAsync", "SIGNAL_DISCONNECT_BIT_CHAT_NETWORK; channel: " + channelName.ToString()); } catch { } break; case SIGNAL_PEER_STATUS: try { if (_connectionManager.IsPeerConnectionAvailable(ConvertToIP(channelName.ID))) { WriteSignalFrame(channelName, SIGNAL_PEER_STATUS_AVAILABLE); } Debug.Write("Connection.ReadDataAsync", "SIGNAL_PEER_STATUS; peerIP: " + ConvertToIP(channelName.ID).ToString()); } catch { } break; case SIGNAL_PEER_STATUS_AVAILABLE: try { lock (_peerStatusLockList) { object lockObject = _peerStatusLockList[channelName]; lock (lockObject) { Monitor.Pulse(lockObject); } } Debug.Write("Connection.ReadDataAsync", "SIGNAL_PEER_STATUS_AVAILABLE; peerIP: " + ConvertToIP(channelName.ID).ToString()); } catch { } break; case SIGNAL_CONNECT_PROXY_TUNNEL: { ChannelStream remoteChannel1 = null; Stream remoteChannel2 = null; try { //get remote peer ep IPEndPoint tunnelToPeerEP = ConvertToIP(channelName.ID); //add first stream into list remoteChannel1 = new ChannelStream(this, channelName, ChannelType.ProxyTunnel); lock (_proxyTunnelChannels) { _proxyTunnelChannels.Add(channelName, remoteChannel1); } //get remote channel service Connection remotePeerConnection = _connectionManager.GetExistingConnection(tunnelToPeerEP); //get remote stream for virtual connection remoteChannel2 = remotePeerConnection.RequestVirtualConnectionChannel(_remotePeerEP); //join current and remote stream Joint joint = new Joint(remoteChannel1, remoteChannel2); joint.Disposed += joint_Disposed; lock (_tunnelJointList) { _tunnelJointList.Add(joint); } joint.Start(); Debug.Write("Connection.ReadDataAsync", "SIGNAL_CONNECT_PROXY_TUNNEL; tunnel to peerIP: " + tunnelToPeerEP.ToString()); } catch { if (remoteChannel1 != null) { remoteChannel1.Dispose(); } if (remoteChannel2 != null) { remoteChannel2.Dispose(); } } } break; case SIGNAL_DISCONNECT_PROXY_TUNNEL: try { lock (_proxyTunnelChannels) { _proxyTunnelChannels[channelName].Dispose(); } Debug.Write("Connection.ReadDataAsync", "SIGNAL_DISCONNECT_PROXY_TUNNEL; channel: " + channelName.ToString()); } catch { } break; case SIGNAL_CONNECT_VIRTUAL_CONNECTION: { ChannelStream channel = null; try { //add current stream into list channel = new ChannelStream(this, channelName, ChannelType.VirtualConnection); lock (_virtualConnectionChannels) { _virtualConnectionChannels.Add(channelName, channel); } IPEndPoint virtualRemotePeerEP = ConvertToIP(channelName.ID); Debug.Write("Connection.ReadDataAsync", "SIGNAL_CONNECT_VIRTUAL_CONNECTION; tunnel from peerIP: " + virtualRemotePeerEP.ToString()); //pass channel as virtual connection async ThreadPool.QueueUserWorkItem(AcceptVirtualConnectionAsync, new object[] { channel, virtualRemotePeerEP }); } catch { if (channel != null) { channel.Dispose(); } } } break; case SIGNAL_DISCONNECT_VIRTUAL_CONNECTION: try { lock (_virtualConnectionChannels) { _virtualConnectionChannels[channelName].Dispose(); } Debug.Write("Connection.ReadDataAsync", "SIGNAL_DISCONNECT_VIRTUAL_CONNECTION; channel: " + channelName.ToString()); } catch { } break; default: throw new IOException("Invalid ChannelManager frame type."); } } } catch { } finally { Dispose(); } }