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<=1) { userHelpAll(); return ; } /*获取广播选项*/ if(argv[1][0]=='-'&&argv[1][1]=='b') { /*广播标志设置为真*/ broadFlag = TRUE; for(i=2; 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  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\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; }