multitest.cpp

windows下和linux下的组播测试程序。源码

// multiTest.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"

#include <WINSOCK2.H>
#pragma comment(lib,"ws2_32.lib")
#include <WS2tcpip.h>

#include <stdio.h>

#pragma warning(disable:4786)

#include   <winsock2.h>   
#include   <ws2tcpip.h>   
#include   <stdio.h>   

#define   BUFFSIZE   2048   
#define   RECV_IP_ADDR             "224.1.1.1"             
#define   DEST_PORT                65000   

int   main(int   argc,char   *argv[])   
{   
	int   sockfd;   
	int   sock_reuse   =   1;   
	struct   ip_mreq   multicast;   
	struct   sockaddr_in   recver_addr;   
	char   szError[100];   
	
	int   index   =   0,                                             //   Integer   index   
		iRecvLen;                                               //   Length   of   recv_sin   
	char   szMessageA[1024*320];                               //   ASCII   string     
	TCHAR   szMessageW[1024*320];                             //   Unicode   string   
	
	WSADATA   WSAData;   
	WORD     wVersionRequested;   
	wVersionRequested   =   MAKEWORD(2,   2);   
	if   (WSAStartup   (wVersionRequested   ,   &WSAData)   !=   0)     
	{   
		printf   ("recver:Initialize   Winsock   error!");   
		exit(1);   
	}   
	if   (LOBYTE(WSAData.wVersion)   !=   2   ||   HIBYTE(WSAData.wVersion)   !=   2   )   {   
		WSACleanup(   );   
		printf   ("setsockopt   failed!   Error:   %d",   WSAGetLastError   ());   
		exit(1);   
	}     
	
	multicast.imr_multiaddr.s_addr   =   inet_addr(RECV_IP_ADDR);   
	multicast.imr_interface.s_addr   =   htonl(INADDR_ANY);   
	recver_addr.sin_family   =   AF_INET;   
	recver_addr.sin_port   =   htons(DEST_PORT);   
	recver_addr.sin_addr.s_addr   =   INADDR_ANY;   
	
	if   ((sockfd=socket(AF_INET,SOCK_DGRAM,0))   <   0)   {   
		printf("recver:new   a   socket   error!\n");   
		exit(1);           
	}   
	if   (setsockopt(sockfd,   SOL_SOCKET,   SO_REUSEADDR,   (char   *)&sock_reuse,   sizeof(sock_reuse))   <   0)   {   
		printf("recver   :   socket   options   set   error");   
		exit(1);   
	}   
	
	if   (bind(sockfd,(struct   sockaddr   *)&recver_addr,sizeof(recver_addr))   <   0)   {   
		closesocket(sockfd);   
		printf("recver:bind   socket   error!\n");   
		exit(1);           
	}   
	
	if   (setsockopt(sockfd,IPPROTO_IP,IP_ADD_MEMBERSHIP,(char   *)&multicast,sizeof(multicast))   <   0   )   {   
		sprintf   (szError,   TEXT("setsockopt   failed!   Error:   %d"),   WSAGetLastError   ());   
		printf("%s\n",   szError);   
		closesocket(sockfd);   
		exit(1);   
	}   
	
	printf("Receive   on   %s:%d\n",   RECV_IP_ADDR,   DEST_PORT);   
	iRecvLen   =   sizeof   (recver_addr);   
	memset(szMessageA,   0,   sizeof(szMessageA)); 
	while(TRUE)
	{
		
		//   Receive   data   from   the   multicasting   group   server.   
		if   (recvfrom   (sockfd,     
			szMessageA,   
			sizeof(szMessageA),                     
			0,   
			(struct   sockaddr   FAR   *)   &recver_addr,   
			&iRecvLen)   ==   SOCKET_ERROR)   
		{   
			wsprintf   (szError,   TEXT("recvfrom   failed!   Error:   %d"),     
				WSAGetLastError   ());   
			MessageBox   (NULL,   szError,   TEXT("Error"),   MB_OK);   
			closesocket   (sockfd);   
			return   FALSE;   
		}   
		else   
		{   
			//   Convert   the   ASCII   string   to   a   Unicode   string. 
			/*
			for   (index   =   0;   index   <=   sizeof   (szMessageA);   index++)   
				szMessageW[index]   =   szMessageA[index];   
			*/
			//MessageBox   (NULL,   szMessageW,   TEXT("Info"),   MB_OK); 
			printf("rec message is :%s\n",szMessageA);
		}
		szMessageA[0] ='\0';
		memset(szMessageA,0,sizeof(szMessageA));
	}
	//   Disable   receiving   on   Sock   before   closing   it.   
	shutdown(sockfd,   0x00);   
	
	//   Close   Sock.   
	closesocket(sockfd);   
	
	WSACleanup();   
	return   0;   
}