C语言UDP传输系统源码

 /*加载库文件*/ #pragma comment( lib, "ws2_32.lib" ) /*加载头文件*/ #include  #include  #include  #include  /*定义多播常量*/ #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 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  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:int\n"); printf("      -b   Start the broadcast program.\n"); printf("      -s   Act as server (send data); otherwise\n"); printf("         receive data. Default is receiver.\n"); printf("      -p:int Port number to use\n "); 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:int\n"); printf("      -m   Start the multicast program.\n"); printf("      -s   Act as server (send data); otherwise\n"); printf("          receive data. Default is receiver.\n"); printf("      -h:str The decimal multicast IP address to join\n"); printf("          The default group is: %s\n", MCASTADDR); printf("      -p:int Port number to use\n"); printf("          The default port is: %d\n", MCASTPORT); printf("      -i:str Local interface to bind to; by default \n"); printf("          use INADDRY_ANY\n"); printf("      -l   Disable loopback\n"); printf("      -n:int Number of messages to send/receive\n"); 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:%d\n",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: %d\n", WSAGetLastError()); closesocket(socketRec); WSACleanup(); return ; } /*从广播地址接收消息*/ while(i<bCount) { recvfrom(socketRec,buf,256,0,(struct sockaddr FAR *)&addrBro,(int FAR *)&addrBroLen); /*延迟2秒钟*/ Sleep(2000); /*输出接收到缓冲区的消息*/ printf("%s\n",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: %d\n", 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: %d\n", 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: %d\n",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: %d\n", WSAGetLastError()); closesocket(socketMul); WSACleanup(); return ; } } /*加入多播组*/ if ((sockJoin = WSAJoinLeaf(socketMul, (SOCKADDR *)&addrMul, sizeof(addrMul), NULL, NULL, NULL, NULL, JL_BOTH)) == INVALID_SOCKET) /*如果加入不成功*/ { printf("WSAJoinLeaf() failed: %d\n", 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: %d\n",WSAGetLastError()); closesocket(sockJoin); closesocket(socketMul); WSACleanup(); return ; } /*如果发送成功*/ else printf("Send message %d\n",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: %d\n",WSAGetLastError()); closesocket(sockJoin); closesocket(socketMul); WSACleanup(); return ; } /*接收成功，输出接收的消息*/ recvbuf[ret] = 0; printf("RECV: '%s' from \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() failed\n"); 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; } 

来源gao@daima#com搞(%代@#码@网