/*加载库文件*/#pragma comment( lib, "ws2_32.lib" )/*加载头文件*/#include <winsock2.h>#include <ws2tcpip.h>#include <stdio.h>#include <stdlib.h>/*定义多播常量*/#define MCASTADDR "224.3.5.8"#define MCASTPORT 25000#define BUFSIZE 1024#define MCOUNT 10/*定义广播常量*/#define BCASTPORT 5050#define BCOUNT 10/*定义广播全局变量*/SOCKET socketBro;
SOCKET socketRec;
struct sockaddr_in addrBro;
struct sockaddr_in addrRec;
BOOL broadSendFlag;
BOOL broadFlag;
DWORD bCount;
DWORD bcastAddr;
short bPort;
/*定义多播全局变量*/SOCKET socketMul;
SOCKET sockJoin;
struct sockaddr_in addrLocal;
struct sockaddr_in addrMul;
BOOL multiSendFlag;
BOOL bLoopBack;
BOOL multiFlag;
DWORD dwInterface;
DWORD dwMulticastGroup;
DWORD mCount;
short mPort;
/*自定义函数*/void initial();
void GetArgments(int argc, char **argv);
void userHelpAll();
void userHelpBro();
void userHelpMul();
void broadcastSend();
void broadcastRec();
void mulControl();
void multicastSend();
void multicastRec();
/*初始化全局变量函数*/void initial(){
/*初始化广播全局变量*/ bPort = BCASTPORT;
bCount = BCOUNT;
bcastAddr = INADDR_BROADCAST;
broadSendFlag = FALSE;
broadFlag = FALSE;
multiFlag = FALSE;
/*初始化多播全局变量*/ dwInterface = INADDR_ANY;
dwMulticastGroup = inet_addr(MCASTADDR);
mPort = MCASTPORT;
mCount = MCOUNT;
multiSendFlag = FALSE;
bLoopBack = FALSE;
}
/*参数获取函数*/void GetArgments(int argc, char **argv){
int i;
/*如果参数个数小于2个*/ if(argc<=1) {
userHelpAll();
return ;
}
/*获取广播选项*/ if(argv[1][0]=='-'&&argv[1][1]=='b') {
/*广播标志设置为真*/ broadFlag = TRUE;
for(i=2;
i < argc ;
i++) {
if (argv[i][0] == '-') {
switch (tolower(argv[i][1])) {
/*如果是发送者*/ case 's': broadSendFlag = TRUE;
break;
/*广播的地址*/ case 'h': if (strlen(argv[i]) > 3) bcastAddr = inet_addr(&argv[i][3]);
break;
/*广播的端口号*/ case 'p': if (strlen(argv[i]) > 3) bPort = atoi(&argv[i][3]);
break;
/*广播(接收或者发送)的数量*/ case 'n': bCount = atoi(&argv[i][3]);
break;
/*其他情况显示用户帮助，终止程序*/ default: {
userHelpBro();
ExitProcess(-1);
}
break;
}
}
}
return ;
}
/*获取多播选项*/ if(argv[1][0]=='-'&&argv[1][1]=='m') {
/*多播标志设置为真*/ multiFlag = TRUE;
for(i=2;
i < argc ;
i++) {
if (argv[i][0] == '-') {
switch (tolower(argv[i][1])) {
/*如果是发送者*/ case 's': multiSendFlag = TRUE;
break;
/*多播地址*/ case 'h': if (strlen(argv[i]) > 3) dwMulticastGroup = inet_addr(&argv[i][3]);
break;
/*本地接口地址*/ case 'i': if (strlen(argv[i]) > 3) dwInterface = inet_addr(&argv[i][3]);
break;
/*多播端口号*/ case 'p': if (strlen(argv[i]) > 3) mPort = atoi(&argv[i][3]);
break;
/*环回标志设置为真*/ case 'l': bLoopBack = TRUE;
break;
/*发送(接收)的数量*/ case 'n': mCount = atoi(&argv[i][3]);
break;
/*其他情况，显示用户帮助，终止程序*/ default: userHelpMul();
break;
}
}
}
}
return;
}
/*全局用户帮助函数*/void userHelpAll(){
printf("Please choose broadcast[-b] or multicast[-m] !n");
printf("userHelpAll: -b [-s][p][-h][-n] | -m[-s][-h][-p][-i][-l][-n]n");
userHelpBro();
userHelpMul();
}
/*广播用户帮助函数*/void userHelpBro(){
printf("Broadcast: -b -s:str -p:int -h:str -n:intn");
printf(" -b Start the broadcast program.n");
printf(" -s Act as server (send data);
otherwisen");
printf(" receive data. Default is receiver.n");
printf(" -p:int Port number to usen ");
printf(" The default port is 5050.n");
printf(" -h:str The decimal broadcast IP address.n");
printf(" -n:int The Number of messages to send/receive.n");
printf(" The default number is 10.n");
}
/*多播用户帮助函数*/void userHelpMul(){
printf("Multicast: -m -s -h:str -p:int -i:str -l -n:intn");
printf(" -m Start the multicast program.n");
printf(" -s Act as server (send data);
otherwisen");
printf(" receive data. Default is receiver.n");
printf(" -h:str The decimal multicast IP address to joinn");
printf(" The default group is: %sn", MCASTADDR);
printf(" -p:int Port number to usen");
printf(" The default port is: %dn", MCASTPORT);
printf(" -i:str Local interface to bind to;
by default n");
printf(" use INADDRY_ANYn");
printf(" -l Disable loopbackn");
printf(" -n:int Number of messages to send/receiven");
ExitProcess(-1);
}
/*广播消息发送函数*/void broadcastSend(){
/*设置广播的消息*/ char *smsg="The message received is from sender!";
BOOL opt=TRUE;
int nlen=sizeof(addrBro);
int ret;
DWORD i=0;
/*创建UDP套接字*/ socketBro=WSASocket(AF_INET,SOCK_DGRAM,0,NULL,0,WSA_FLAG_OVERLAPPED);
/*如果创建失败*/ if(socketBro==INVALID_SOCKET) {
printf("Create socket failed:%dn",WSAGetLastError());
WSACleanup();
return;
}
/*设置广播地址各个选项*/ addrBro.sin_family=AF_INET;
addrBro.sin_addr.s_addr=bcastAddr;
addrBro.sin_port=htons(bPort);
/*设置该套接字为广播类型*/ if (setsockopt(socketBro,SOL_SOCKET,SO_BROADCAST,(char FAR *)&opt, sizeof(opt))==SOCKET_ERROR) /*如果设置失败*/ {
printf("setsockopt failed:%d",WSAGetLastError());
closesocket(socketBro);
WSACleanup();
return;
}
/*循环发送消息*/ while(i<bCount) {
/*延迟1秒*/ Sleep(1000);
/*从广播地址发送消息*/ ret=sendto(socketBro,smsg,256,0,(struct sockaddr*)&addrBro,nlen);
/*如果发送失败*/ if(ret==SOCKET_ERROR) printf("Send failed:%d",WSAGetLastError());
/*如果发送成功*/ else {
printf("Send message %d!n",i);
}
i++;
}
/*发送完毕后关闭套接字、释放占用资源*/ closesocket(socketBro);
WSACleanup();
}
/*广播消息接收函数*/void broadcastRec(){
BOOL optval = TRUE;
int addrBroLen;
char buf[256];
DWORD i=0;
/*该地址用来绑定套接字*/ addrRec.sin_family=AF_INET;
addrRec.sin_addr.s_addr=0;
addrRec.sin_port=htons(bPort);
/*该地址用来接收网路上广播的消息*/ addrBro.sin_family=AF_INET;
addrBro.sin_addr.s_addr=bcastAddr;
addrBro.sin_port=htons(bPort);
addrBroLen=sizeof(addrBro);
//创建UDP套接字 socketRec=socket(AF_INET,SOCK_DGRAM,0);
/*如果创建失败*/ if(socketRec==INVALID_SOCKET) {
printf("Create socket error:%d",WSAGetLastError());
WSACleanup();
return;
}
/*设置该套接字为可重用类型*/ if(setsockopt(socketRec,SOL_SOCKET,SO_REUSEADDR,(char FAR *)&optval, sizeof(optval))==SOCKET_ERROR) /*如果设置失败*/ {
printf("setsockopt failed:%d",WSAGetLastError());
closesocket(socketRec);
WSACleanup();
return;
}
/*绑定套接字和地址*/ if(bind(socketRec,(struct sockaddr *)&addrRec, sizeof(struct sockaddr_in))==SOCKET_ERROR) /*如果绑定失败*/ {
printf("bind failed with: %dn", WSAGetLastError());
closesocket(socketRec);
WSACleanup();
return ;
}
/*从广播地址接收消息*/ while(i<bCount) {
recvfrom(socketRec,buf,256,0,(struct sockaddr FAR *)&addrBro,(int FAR *)&addrBroLen);
/*延迟2秒钟*/ Sleep(2000);
/*输出接收到缓冲区的消息*/ printf("%sn",buf);
/*情况缓冲区*/ ZeroMemory(buf,256);
i++;
}
/*接收完毕后关闭套接字、释放占用资源*/ closesocket(socketRec);
WSACleanup();
}
/*多播控制函数*/void mulControl(){
int optval;
/*创建UDP套接字，用于多播*/ if ((socketMul = WSASocket(AF_INET, SOCK_DGRAM, 0, NULL, 0, WSA_FLAG_MULTIPOINT_C_LEAF | WSA_FLAG_MULTIPOINT_D_LEAF | WSA_FLAG_OVERLAPPED)) == INVALID_SOCKET) {
printf("socket failed with: %dn", WSAGetLastError());
WSACleanup();
return ;
}
/*设置本地接口地址*/ addrLocal.sin_family = AF_INET;
addrLocal.sin_port = htons(mPort);
addrLocal.sin_addr.s_addr = dwInterface;
/*将UDP套接字绑定到本地地址上*/ if (bind(socketMul, (struct sockaddr *)&addrLocal, sizeof(addrLocal)) == SOCKET_ERROR) /*如果绑定失败*/ {
printf("bind failed with: %dn", WSAGetLastError());
closesocket(socketMul);
WSACleanup();
return ;
}
/*设置多播地址各个选项*/ addrMul.sin_family = AF_INET;
addrMul.sin_port = htons(mPort);
addrMul.sin_addr.s_addr = dwMulticastGroup;
/*重新设置TTL值*/ optval = 8;
/*设置多播数据的TTL(存在时间)值。默认情况下，TTL值是1*/ if (setsockopt(socketMul, IPPROTO_IP, IP_MULTICAST_TTL, (char *)&optval, sizeof(int)) == SOCKET_ERROR) /*如果设置失败*/ {
printf("setsockopt(IP_MULTICAST_TTL) failed: %dn",WSAGetLastError());
closesocket(socketMul);
WSACleanup();
return ;
}
/*如果指定了返还选项*/ if (bLoopBack) {
/*设置返还选项为假，禁止将发送的数据返还给本地接口*/ optval = 0;
if (setsockopt(socketMul, IPPROTO_IP, IP_MULTICAST_LOOP, (char *)&optval, sizeof(optval)) == SOCKET_ERROR) /*如果设置失败*/ {
printf("setsockopt(IP_MULTICAST_LOOP) failed: %dn", WSAGetLastError());
closesocket(socketMul);
WSACleanup();
return ;
}
}
/*加入多播组*/ if ((sockJoin = WSAJoinLeaf(socketMul, (SOCKADDR *)&addrMul, sizeof(addrMul), NULL, NULL, NULL, NULL, JL_BOTH)) == INVALID_SOCKET) /*如果加入不成功*/ {
printf("WSAJoinLeaf() failed: %dn", WSAGetLastError());
closesocket(socketMul);
WSACleanup();
return ;
}
}
/*多播消息发送函数*/void multicastSend(){
TCHAR sendbuf[BUFSIZE];
DWORD i;
int ret;
mulControl();
/*发送mCount条消息*/ for(i = 0;
i < mCount;
i++) {
/*将待发送的消息写入发送缓冲区*/ sprintf(sendbuf, "server 1: This is a test: %d", i);
ret=sendto(socketMul, (char *)sendbuf, strlen(sendbuf), 0, (struct sockaddr *)&addrMul, sizeof(addrMul));
/*如果发送失败*/ if(ret==SOCKET_ERROR) {
printf("sendto failed with: %dn",WSAGetLastError());
closesocket(sockJoin);
closesocket(socketMul);
WSACleanup();
return ;
}
/*如果发送成功*/ else printf("Send message %dn",i);
Sleep(500);
}
/*关闭套接字、释放占用资源*/ closesocket(socketMul);
WSACleanup();
}
/*多播消息接收函数*/void multicastRec(){
DWORD i;
struct sockaddr_in from;
TCHAR recvbuf[BUFSIZE];
int ret;
int len = sizeof(struct sockaddr_in);
mulControl();
/*接收mCount条消息*/ for(i = 0;
i < mCount;
i++) {
/*将接收的消息写入接收缓冲区*/ if ((ret = recvfrom(socketMul, recvbuf, BUFSIZE, 0, (struct sockaddr *)&from, &len)) == SOCKET_ERROR) /*如果接收不成功*/ {
printf("recvfrom failed with: %dn",WSAGetLastError());
closesocket(sockJoin);
closesocket(socketMul);
WSACleanup();
return ;
}
/*接收成功，输出接收的消息*/ recvbuf[ret] = 0;
printf("RECV: '%s' from <%s>n", recvbuf,inet_ntoa(from.sin_addr));
}
/*关闭套接字、释放占用资源*/ closesocket(socketMul);
WSACleanup();
}
/*主函数*/int main(int argc, char **argv){
WSADATA wsd;
initial();
GetArgments(argc, argv);
/*初始化Winsock*/ if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0) {
printf("WSAStartup() failedn");
return -1;
}
/*如果是执行广播程序*/ if(broadFlag) {
/*以发送者身份发送消息*/ if(broadSendFlag) {
broadcastSend();
return 0;
}
/*以接收者身份接收消息*/ else {
broadcastRec();
return 0;
}
}
/*如果是执行多播程序*/ if(multiFlag) {
/*以发送者身份发送消息*/ if(multiSendFlag) {
multicastSend();
return 0;
}
/*以接收者身份接收消息*/ else {
multicastRec();
return 0;
}
}
return 0;
}