组播报文收发测试

3060阅读　0评论2014-07-07　fireaxe
分类：LINUX



点击(此处)折叠或打开

  1. #include <sys/types.h>
  2. #include <stdio.h>
  3. #include <stdlib.h>
  4. #include <string.h>

  6. #ifdef WIN_DEVCPP
  7. #include <winsock.h>
  8. #include <io.h>
  9. #pragma comment(lib,"ws2_32.lib")
  10. int WSAStartup
  11. (
  12.     WORD wVersionRequested,
  13.     LPWSADATA lpWSAData
  14. );
  15. typedef unsigned short in_port_t;
  16. #else
  17. #include <sys/socket.h>
  18. #include <arpa/inet.h>

  20. #define TRUE (1)
  21. #define FALSE (0)
  22. #define ERROR (-1)
  23. typedef unsigned long BOOL;
  24. #endif /* WIN_DEVCPP */

  26. #define BUFLEN 255
  27. #define TC_BOARD_IP "192.168.2.3"
  28. #define TC_HOST_IP "192.168.2.253"
  29. #define TC_BOARD_PORT 7602
  30. #define TC_HOST_PORT 7601

  32. #define FIRST_SHAKE "the first handshake"
  33. #define SECOND_SHAKE "the second handshake"
  34. #define THIRD_SHAKE "the third handshake"

  36. /*********************************************************************
  37. *filename: eth_test.c
  38. *********************************************************************/
  39. #ifdef WIN_DEVCPP
  40. int inet_aton( const char * pString, struct in_addr * inetAddress )
  41. {
  42.     u_long rtnAddress;

  44.     if (inetAddress == NULL)
  45.         return (ERROR);

  47.     rtnAddress = inet_addr (pString);
  48.     inetAddress->s_addr = rtnAddress;
  49.     return (0);
  50. }
  51. #endif /* WIN_DEVCPP */

  53. int main(int argc, char **argv)
  54. {
  55. #ifdef WIN_DEVCPP
  56.     WSADATA wsaData;
  57. #endif /* WIN_DEVCPP */
  58.     int retVal;
  59.     struct sockaddr_in to;
  60.     struct sockaddr_in from;
  61.     struct sockaddr_in local;
  62.     int sockfd;
  63.     char buff[BUFLEN + 1];
  64.     struct in_addr localIpAddr;
  65.     struct ip_mreq mreq;
  66.     int pkgLen;
  67.     BOOL isMultiCAST;
  68.     struct in_addr targetIp;
  69.     struct in_addr selfIp;

  71.     printf("argc=%d\n",argc);

  73. #ifdef WIN_DEVCPP
  74.     retVal = WSAStartup( MAKEWORD(2, 2), &wsaData );
  75.     if (retVal == SOCKET_ERROR) {
  76.         printf("WSAStartup error\n");
  77.         exit(33);
  78.     }
  79. #endif /* WIN_DEVCPP */

  81.     if ((argc == 2) && (0 == strcmp(argv[1], "help"))) {
  82.         printf("Usage: mcast_server [targetIp [selfIp]]\n");
  83.         printf("default: mcast_server %s %s\n", TC_BOARD_IP, TC_HOST_IP);
  84.         exit (0);
  85.     }

  87.     if (argc >= 2) {
  88.         if (0 > inet_aton(argv[1], &targetIp))
  89.             exit (1);
  90.     } else {
  91.         if (0 > inet_aton(TC_BOARD_IP, &targetIp))
  92.             exit (2);
  93.     }

  95.     if (0xe == (htonl(targetIp.s_addr) >> 28)) {
  96.         isMultiCAST = TRUE;
  97.         printf("## multi-cast ##\n");
  98.     } else {
  99.         isMultiCAST = FALSE;
  100.         printf("## uni-cast ##\n");
  101.     }

  103.     if (argc >= 3) {
  104.         if (0 > inet_aton(argv[2], &selfIp))
  105.             exit (3);
  106.     } else {
  107.         if (0 > inet_aton(TC_HOST_IP, &selfIp))
  108.             exit (4);
  109.     }

  111.     printf("set targetIp to %s, port to %d\n", (char *)inet_ntoa(targetIp), TC_BOARD_PORT);
  112.     printf("set selfIp to %s, port to %d\n", (char *)inet_ntoa(selfIp), TC_HOST_PORT);

  114.     /* create socket */
  115.     sockfd = socket(AF_INET, SOCK_DGRAM, 0);
  116.     if (sockfd < 0) {
  117.         perror("socket");
  118.         printf("socket creating error\n");
  119.         exit(1);
  120.     }

  122.     /* set self info */
  123.     memset(&local, 0, sizeof(local));
  124.     local.sin_family = AF_INET;
  125.     local.sin_port = htons(TC_HOST_PORT);
  126.     local.sin_addr.s_addr = INADDR_ANY;

  128.     /*bind socket*/
  129.     printf("bind socket to %s:%d\n", (char *)inet_ntoa(selfIp), TC_HOST_PORT);
  130.     retVal = bind(sockfd,(struct sockaddr*)&local, sizeof(local));
  131.     if (retVal < 0) {
  132.         perror("bind");
  133.         printf("bind error!\n");
  134.         close(sockfd);
  135.         exit(5);
  136.     }

  138.     if (isMultiCAST) {
  139.          /* join multi-cast group */
  140.         mreq.imr_multiaddr = targetIp;
  141.         mreq.imr_interface = selfIp;

  143.         retVal = setsockopt(sockfd, IPPROTO_IP, IP_ADD_MEMBERSHIP, (const char*)&mreq, sizeof(mreq));
  144.         if (retVal == ERROR) {
  145.             perror("ADD_MEMBERSHIP");
  146.             printf("setsockopt join error!\n");
  147.             close(sockfd);
  148.             exit(6);
  149.         }

  151.         /* enable multi-cast */
  152.         localIpAddr = selfIp;
  153.         retVal = setsockopt (sockfd, IPPROTO_IP, IP_MULTICAST_IF,
  154.             (char *)&localIpAddr, sizeof(localIpAddr));
  155.         if (retVal == ERROR) {
  156.             perror("MULTICAST_IF");
  157.             printf("setsockopt enable multi-cast error!\n");
  158.             close(sockfd);
  159.             exit(7);
  160.         }
  161.     }

  163.     /* set dst info */
  164.     memset(&to, 0, sizeof(to));
  165.     to.sin_family = AF_INET;
  166.     to.sin_port = htons(TC_BOARD_PORT);
  167.     to.sin_addr = targetIp;

  169.     /* send the first handshake */
  170.     retVal = sendto(sockfd, FIRST_SHAKE, sizeof(FIRST_SHAKE), 0, (struct sockaddr *)&to, sizeof(to));
  171.     if (retVal == ERROR) {
  172.         perror("sendto");
  173.         printf("sendto error!\n");
  174.         close(sockfd);
  175.         exit(8);
  176.     }

  178.     printf(">>>> [%s:%d] %s is sent >>>>\n",
  179.         (char *)inet_ntoa(to.sin_addr), htons(to.sin_port), FIRST_SHAKE);

  181.     /* receive the second handshake */
  182.     do {
  183.          memset(buff, 0, sizeof(buff));
  184.         memset(&from, 0, sizeof(from));
  185.         pkgLen = sizeof(from);
  186.         retVal = recvfrom(sockfd, buff, sizeof(buff) - 1, 0,
  187.             (struct sockaddr*)&from, &pkgLen);
  188.         if (retVal < 0) {
  189.             perror("recvfrom");
  190.             printf("recvfrom error!\n");
  191.             close(sockfd);
  192.             exit(9);
  193.          }

  195.         printf("received msg: len=%d, ip=%s, port=%d, msg=%s\n", retVal,
  196.             (char *)inet_ntoa(from.sin_addr), htons(from.sin_port), buff);
  197.     } while ((retVal != sizeof(SECOND_SHAKE)) || (0 != strcmp(buff, SECOND_SHAKE)));

  199.     printf("<<<< [%s:%d] %s is received <<<<\n",
  200.         (char *)inet_ntoa(from.sin_addr), htons(from.sin_port), SECOND_SHAKE);

  202.     /* send the first handshake */
  203.     retVal = sendto(sockfd, THIRD_SHAKE, sizeof(THIRD_SHAKE), 0,
  204.             (struct sockaddr *)&to, sizeof(to));
  205.     if (retVal == ERROR) {
  206.         perror("sendto");
  207.         printf("sendto error!\n");
  208.         close(sockfd);
  209.         exit(10);
  210.     }

  212.     printf(">>>> [%s:%d] %s is sent >>>>\n",
  213.         (char *)inet_ntoa(to.sin_addr), htons(to.sin_port), THIRD_SHAKE);

  215.     if (isMultiCAST)
  216.         {
  217.          retVal = setsockopt(sockfd, IPPROTO_IP, IP_DROP_MEMBERSHIP, (const char*)&mreq, sizeof(mreq));
  218.         if (retVal == ERROR) {
  219.             perror("DROP_MEMBERSHIP");
  220.             printf("setsockopt join error!\n");
  221.             close(sockfd);
  222.             exit(11);
  223.         }
  224.     }

  226.     close(sockfd);
  227. }

上一篇：undefined reference to `inet_addr@4'
下一篇：如何编译配置u-boot