Initial

mcast/mcastrcv.cpp

@@ -0,0 +1,185 @@
+#include <common/windows.hpp>
+#include <common/array.hpp>
+#include <common/systime.hpp>
+#include <socket/socket.hpp>
+#include <socket/sockaddr.hpp>
+#include <socket/ipmcast.hpp>
+#include <socket/wsadata.hpp>
+#include <iostream.h>
+
+void outDebug(const String &string);
+void mcastsnd(int destPort,String destIP,Array<BYTE> &message,int frequency,int ttl=1);
+void mcastrcv(int destPort,String rcvIPAddress,int ttl=1);
+
+void main(int argc,char **argv)
+{
+  String strOp;
+	String strIPAddress;
+	String strPort;
+	String strData;
+	String strFrequency="1000";
+	Array<BYTE> message;
+
+  outDebug("MCAST 1.00 - Sean Kessler.");
+  if(1==argc)
+	{
+	  outDebug("USAGE: mcast send|receive address:port {frequency (ms)} {stringdata}");
+	  outDebug("(ie) mcast send 234.5.6.7:4567 1000 hello");
+		return;
+	}
+	strOp=argv[1];
+	strOp.lower();
+	strIPAddress=String(argv[2]).betweenString(0,':');
+	strPort=String(argv[2]).betweenString(':',0);
+	if(5==argc)strData=argv[4];
+	else strData="hello world";
+	if(argc>=4)strFrequency=argv[3];
+	message.size(strData.length()+1);
+	::memset(&message[0],0,message.size());
+	::strcpy((char*)&message[0],strData.str());
+	outDebug(String("mode=")+strOp);
+	outDebug(String("address=")+strIPAddress);
+	outDebug(String("port=")+strPort);
+	if(strOp=="send")mcastsnd(strPort.toInt(),strIPAddress,message,strFrequency.toInt());
+	else if(strOp=="receive")mcastrcv(strPort.toInt(),strIPAddress);
+	else outDebug("USAGE: mcast send|receive address:port");
+}
+
+void mcastrcv(int destPort,String rcvIPAddress,int ttl)
+{
+  WSASystem wsaSystem;
+	SocketAddress socketAddressLocal;
+	SocketAddress socketAddressTo;
+	SocketAddress socketAddressFrom;
+	IPMReq ipMCast;
+  Socket socket;
+
+  outDebug("Socket create.");
+	if(!socket.create(Socket::DomainInternet,Socket::SocketDGram,Socket::ProtocolNone))
+	{
+    outDebug("Socket creation failed.");
+		return;
+	}
+	if(!socket.reuseAddress(true))
+	{
+	  outDebug("Error setting option REUSEADDR.");
+		socket.destroy();
+		return;
+	}
+	outDebug("Socket created.");
+	socketAddressLocal.family(AF_INET);
+	socketAddressLocal.internetAddress(InternetAddress());
+	socketAddressLocal.port(htons(destPort));
+	outDebug(String("Socket bind to ")+socketAddressLocal.toString());
+	if(!socket.bind(socketAddressLocal))
+	{
+	  outDebug("Bind failed.");
+		socket.destroy();
+		return;
+	}
+	outDebug("Socket is bound.");
+	if(!socket.setMulticastTTL(ttl))
+	{
+	  outDebug("[Error setting option SET_MULTICAST_TTL.]");
+		socket.destroy();
+		return;
+	}
+	ipMCast.multicastAddress(InternetAddress(rcvIPAddress));
+	ipMCast.localAddress(InternetAddress());
+	outDebug(String("Add membership ")+ipMCast.toString());
+	if(!socket.addMembership(ipMCast))
+	{
+	  int error=WSAGetLastError();
+	  outDebug(String("[Error setting option ADDMEMBERSHIP.] Code=")+String().fromInt(error));
+		socket.destroy();
+		return;
+	}   
+  outDebug(String("Joined ")+InternetAddress(rcvIPAddress).toString());
+	if(!socket.setMulticastLoopback(false))
+	{
+	  int error=WSAGetLastError();
+	  outDebug(String("[Error setting option MULTICASTLOOPBACK.] Code=")+String().fromInt(error));
+	}
+  outDebug(String("Loopback is disabled."));
+	outDebug("Listening for messages...");
+  while(true)
+	{
+	  char buffer[1024];
+		INETSocketAddress from;
+	  if(!socket.receiveFrom(buffer,sizeof(buffer),0,from))
+		{
+		  outDebug("Receive Failed.");
+			socket.destroy();
+			return;
+		}
+		else 
+		{
+		  SystemTime sysTime;
+		  outDebug(String("[")+sysTime.toString()+String("]")+String("Received from [")+from.toString()+String("] \"")+String(buffer)+String("\""));
+		}
+	}
+	socket.destroy();
+}
+
+void mcastsnd(int destPort,String destIPAddress,Array<BYTE> &message,int frequency,int ttl)
+{
+  WSASystem wsaSystem;
+	SocketAddress socketAddressLocal;
+  Socket socket;
+
+	if(!socket.create(Socket::DomainInternet,Socket::SocketDGram,Socket::ProtocolNone))
+	{
+    outDebug("[Socket creation failed]");
+		return;
+	}
+	if(!socket.reuseAddress(true))
+	{
+	  outDebug("[Error setting option REUSEADDR.]");
+		socket.destroy();
+		return;
+	}
+	outDebug("Socket created.");
+	socketAddressLocal.family(AF_INET);
+	socketAddressLocal.internetAddress(InternetAddress());
+	socketAddressLocal.port(htons(0));
+	if(!socket.bind(socketAddressLocal))
+	{
+	  outDebug("Bind failed.");
+		socket.destroy();
+		return;
+	}
+	outDebug("Socket is bound.");
+	if(!socket.setMulticastTTL(ttl))
+	{
+	  outDebug("[Error setting option SET_MULTICAST_TTL.]");
+		socket.destroy();
+		return;
+	}
+  INETSocketAddress inetSocketAddress;
+	inetSocketAddress.family(AF_INET);
+	inetSocketAddress.port(destPort);
+	inetSocketAddress.internetAddress(InternetAddress(destIPAddress));
+  while(true)
+	{
+	  if(!socket.sendTo((void*)&message[0],message.size(),inetSocketAddress))
+		{
+		  outDebug("Send Failed.");
+			socket.destroy();
+			return;
+		}
+		else 
+		{
+		  SystemTime sysTime;
+		  outDebug(sysTime.toString()+"Data sent.");
+		}
+		::Sleep(frequency);
+	}
+	socket.destroy();
+}
+
+void outDebug(const String &string)
+{
+  String outString(string+String("\n"));
+  cout << ((String&)string).str() << endl;
+  ::OutputDebugString(outString.str());
+}