C语言UDP传输系统源码

/*加载库文件*/#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: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 <%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() 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;}