public void ProcessMessage(UdpReflectorMessage message, IPEndPoint source, Socket localSocket) { IPEndPoint localEP = localSocket.LocalEndPoint as IPEndPoint; if (message.Type == UdpReflectorMessageType.LEAVE) { // a background thread will eventually remove the multicast group, if necessary lock (clientRegTable) { clientRegTable.Remove(source); } return; } else if (message.Type == UdpReflectorMessageType.JOIN) { IPEndPoint multicastEP = message.MulticastEP; JoinMulticastGroup(multicastEP); subscribedMulticastGroups[multicastEP.Address] = multicastEP.Address; ClientEntry entry = new ClientEntry(source, multicastEP, DateTime.Now); lock (clientRegTable) { clientRegTable.Add(source, entry); } // Multi-reflector cascading join, if required. We send an identical join request on the // same port to the parent's IP address if (ParentReflector != null) { IPEndPoint parentEP = new IPEndPoint(ParentReflector, localEP.Port); UdpReflectorMessage joinMessage = new UdpReflectorMessage(UdpReflectorMessageType.JOIN, multicastEP); BufferChunk chunk = joinMessage.ToBufferChunk(); localSocket.SendTo(chunk.Buffer, chunk.Index, chunk.Length, SocketFlags.None, parentEP); // Need to add the parent to our local dispatch table (to simulate a remote join) ClientEntry parentEntry = new ClientEntry(parentEP, multicastEP, DateTime.Now); lock (clientRegTable) { clientRegTable.Add(parentEP, parentEntry); } } } else if (message.Type == UdpReflectorMessageType.PING) { // acknowledge that we're alive... UdpReflectorMessage pingReply = new UdpReflectorMessage(UdpReflectorMessageType.PING_REPLY); BufferChunk bufferChunk = pingReply.ToBufferChunk(); localSocket.SendTo(bufferChunk.Buffer, bufferChunk.Index, bufferChunk.Length, SocketFlags.None, source); } else { // unknown or unhandled message type... } }
private void CheckForReflector() { // send a ping UdpReflectorMessage ping = new UdpReflectorMessage(UdpReflectorMessageType.PING); BufferChunk buffer = ping.ToBufferChunk(); reflectorSocket.SendTo(buffer.Buffer, buffer.Index, buffer.Length, SocketFlags.None, ReflectorAddress); // wait for response byte [] byteBuffer = new byte[512]; int count = reflectorSocket.Receive(byteBuffer); UdpReflectorMessage pingReply = new UdpReflectorMessage(byteBuffer, count); if (pingReply.Type == UdpReflectorMessageType.PING_REPLY) { this.ReportProgress(100); this.lastSuccessfulReading = DateTime.Now; } // wait a bit... Thread.Sleep(1500); }
/// <summary> /// Forwards the traffic from unicast to unicast/multicast for the given trafficType /// </summary> public void Start() { int size = 0; TrafficType traffTypes; lock (ThreadTypeData) { if (idxThreadTypeData < ThreadTypeData.Length) { traffTypes = ThreadTypeData[idxThreadTypeData]; Thread.CurrentThread.Name = "Reflector_UCtoUCMC_" + traffTypes.ToString(); idxThreadTypeData++; } else { throw new Exception(Strings.CreatedThreadsExceedThreadTypesDefined); } } #region Assigning appropriate sockets to "(mc/uc)(Ref/Srv)Sock" // The Ref(erence) socket variables are assigned to the socket protocol that this thread is not listening on // but may use for inter-protocol communication. For example if mcSrvSock is an IPv4 socket, mcRefSock would be // an IPv6 socket and vice versa. Socket mcRefSock = null; Socket ucRefSock = null; Socket ucSrvSock = null; int ucPort = 0; EndPoint remoteEP = null; switch (traffTypes) { case TrafficType.IPv4RTCP: ucPort = ReflectorMgr.ReflectorUnicastRTPListenPort + 1; remoteEP = new IPEndPoint(IPAddress.Any, 0); mcRefSock = ReflectorMgr.Sockets.SockMCv6RTCP; ucRefSock = ReflectorMgr.Sockets.SockUCv6RTCP; ucSrvSock = ReflectorMgr.Sockets.SockUCv4RTCP; break; case TrafficType.IPv6RTCP: ucPort = ReflectorMgr.ReflectorUnicastRTPListenPort + 1; remoteEP = new IPEndPoint(IPAddress.IPv6Any, 0); mcRefSock = ReflectorMgr.Sockets.SockMCv4RTCP; ucRefSock = ReflectorMgr.Sockets.SockUCv4RTCP; ucSrvSock = ReflectorMgr.Sockets.SockUCv6RTCP; break; case TrafficType.IPv4RTP: ucPort = ReflectorMgr.ReflectorUnicastRTPListenPort; remoteEP = new IPEndPoint(IPAddress.Any, 0); mcRefSock = ReflectorMgr.Sockets.SockMCv6RTP; ucRefSock = ReflectorMgr.Sockets.SockUCv6RTP; ucSrvSock = ReflectorMgr.Sockets.SockUCv4RTP; break; case TrafficType.IPv6RTP: ucPort = ReflectorMgr.ReflectorUnicastRTPListenPort; remoteEP = new IPEndPoint(IPAddress.IPv6Any, 0); mcRefSock = ReflectorMgr.Sockets.SockMCv4RTP; ucRefSock = ReflectorMgr.Sockets.SockUCv4RTP; ucSrvSock = ReflectorMgr.Sockets.SockUCv6RTP; break; default: Debug.Assert(false); throw new ArgumentException(Strings.InvalidTrafficTypeCombination); } #endregion IPEndPoint groupEP = null; byte [] buf = new byte[1500]; IList <IPEndPoint> members = new List <IPEndPoint>(); while (true) { try { //EndPoint ep = null; size = ucSrvSock.ReceiveFrom(buf, ref remoteEP); // First, check whether this is a control message (JOIN or LEAVE) if (size <= 50) { try { UdpReflectorMessage message = new UdpReflectorMessage(buf, size); registrar.ProcessMessage(message, remoteEP as IPEndPoint, ucSrvSock); continue; // read next message } catch (InvalidUdpReflectorMessage) { // fall through } } if ((traffTypes & TrafficType.RTP) == TrafficType.RTP) { ReflectorMgr.PC[ReflectorPC.ID.UnicastPacketsReceived]++; } ClientEntry entry = registrar.GetEntry(remoteEP as IPEndPoint); if (entry != null) { registrar.MarkAsActive(entry); // Make sure this node isn't also sending over multicast... if (registrar.IsIPAddressUsingMulticast(entry.ClientEP.Address)) { eventLog.WriteEntry("Warning: receving both unicast and multicast from: " + entry.ClientEP.Address, EventLogEntryType.Warning, (int)ReflectorEventLog.ID.UCtoUCMCException); continue; // read next message without propogating further... } // lookup the (first) group which this client is a member of that group. groupEP = entry.GroupEP; // Find the other members of the group registrar.MemberLookup(members, groupEP); // Send the data to the Multicast side if (SendMulticast) { if (groupEP.AddressFamily == ucSrvSock.AddressFamily) { ucSrvSock.SendTo(buf, 0, size, SocketFlags.None, groupEP); } else if ((mcRefSock != null) && (groupEP.AddressFamily == ucRefSock.AddressFamily)) { ucRefSock.SendTo(buf, 0, size, SocketFlags.None, groupEP); } } // Send the data to all unicast client members except the sender. for (int i = 0; i < members.Count; i++) { if (!remoteEP.Equals(members[i])) { if (members[i].AddressFamily == ucSrvSock.AddressFamily) { ucSrvSock.SendTo(buf, 0, size, SocketFlags.None, members[i]); } else if ((ucRefSock != null) && (members[i].AddressFamily == ucRefSock.AddressFamily)) { ucRefSock.SendTo(buf, 0, size, SocketFlags.None, members[i]); } } } if ((traffTypes & TrafficType.RTP) == TrafficType.RTP) { ReflectorMgr.PC[ReflectorPC.ID.UCtoUCPacketsSent] += members.Count - 1; } } } // On stopping the service, avoid the AbortException written in the event viewer catch (ThreadAbortException) {} catch (Exception e) // Connection reset by peer! this happens occasionally when a UC client leaves. { eventLog.WriteEntry(string.Format(CultureInfo.CurrentCulture, Strings.UCtoUCMCForwarderException, traffTypes, remoteEP, e.ToString()), EventLogEntryType.Warning, (int)ReflectorEventLog.ID.UCtoUCMCException); } } }