嵌入式

IOCTL获取和配置IP地址/子网掩码/网关

2019-07-13 08:29发布

站内文章 / 嵌入式Linux
14414 0 1210
原文 点击打开链接
OS：LINUX

这部分代码包含：
1.获取IP地址，子网掩码，物理地址。
2.配置IP地址，子网掩码，网关等。
3.IP地址合法验证和子网掩码验证。

TIPS：部分代码摘自网上，代码已测试通过。
  1. #include <sys/socket.h>
  2. #include <arpa/inet.h>
  3. #include <net/if.h>
  4. #include <net/if_arp.h>
  5. #include <unistd.h>
  6. #include <errno.h>
  7. #include <sys/time.h>
  8. #include <sys/stat.h>
  9. #include <stdio.h>
  10. #include <netdb.h>
  11. #include <netinet/in.h>
  12. #include <sys/ioctl.h>
  13. #include <ctype.h>
  14. #include <net/route.h>
  15. #include <fcntl.h>
  16. #include <string.h>
  17. #include <pthread.h>
  18. #if defined(__GLIBC__) && __GLIBC__ >=2 && __GLIBC_MINOR__ >= 1
  19. #include <netpacket/packet.h>
  20. #include <net/ethernet.h>
  21. #else
  22. #include <sys/types.h>
  23. #include <netinet/if_ether.h>
  24. #endif

  26. #include "xxx_ip.h"

  28. static int set_addr(unsigned char ip[16], int flag);
  29. static int get_addr(unsigned char ip[16], int flag);

  31. int get_ip(unsigned char ip[16])
  32. {
  33.     return get_addr(ip, SIOCGIFADDR);
  34. }

  36. int get_ip_netmask(unsigned char ip[16])
  37. {
  38.     return get_addr(ip, SIOCGIFNETMASK);
  39. }

  41. int get_mac(unsigned char addr[6])
  42. {
  43.     return get_addr(addr, SIOCGIFHWADDR);
  44. }

  46. static int get_addr(unsigned char *addr, int flag)
  47. {
  48.     SINT32 sockfd = 0;
  49.     struct sockaddr_in *sin;
  50.     struct ifreq ifr;

  52.     if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) < 0)
  53.     {
  54.         perror("socket error! ");
  55.         return FALSE;
  56.     }

  58.     memset(&ifr, 0, sizeof(ifr));
  59.     snprintf(ifr.ifr_name, (sizeof(ifr.ifr_name) - 1), "%s", DEFAULT_ETH);

  61.     if(ioctl(sockfd, flag, &ifr) < 0 )
  62.     {
  63.         perror("ioctl error! ");
  64.         close(sockfd);
  65.         return FALSE;
  66.     }
  67.     close(sockfd);

  69.     if (SIOCGIFHWADDR == flag){
  70.         memcpy((void *)addr, (const void *)&ifr.ifr_ifru.ifru_hwaddr.sa_data, 6);
  71.         /*printf("mac address: %2.2x:%2.2x:%2.2x:%2.2x:%2.2x:%2.2x ", addr[0],addr[1],addr[2],addr[3],addr[4],addr[5]);*/
  72.     }else{
  73.         sin = (struct sockaddr_in *)&ifr.ifr_addr;
  74.         snprintf((char *)addr, IP_LENGTH, "%s", inet_ntoa(sin->sin_addr));
  75.     }

  77.     return TRUE;
  78. }

  80. int is_valid_ip(unsigned char ipaddr[16])
  81. {
  82.     int ret = 0;
  83.     struct in_addr inp;
  84.     ret = inet_aton(ipaddr, &inp);
  85.     if (0 == ret)
  86.     {
  87.         return FALSE;
  88.     }
  89.     else
  90.     {
  91.         printf("inet_aton:ip=%lu ",ntohl(inp.s_addr));
  92.     }

  94.     return TRUE;
  95. }

  97. /*
  98.  * 先验证是否为合法IP，然后将掩码转化成32无符号整型，取反为000...00111...1，
  99.  * 然后再加1为00...01000...0，此时为2^n，如果满足就为合法掩码
  100.  *
  101.  * */
  102. int is_valid_netmask(unsigned char netmask[16])
  103. {
  104.     if(is_valid_ip(netmask) > 0)
  105.     {
  106.         unsigned int b = 0, i, n[4];
  107.         sscanf(netmask, "%u.%u.%u.%u", &n[3], &n[2], &n[1], &n[0]);
  108.         for(i = 0; i < 4; ++i) //将子网掩码存入32位无符号整型
  109.             b += n[i] << (i * 8);
  110.         b = ~b + 1;
  111.         if((b & (b - 1)) == 0) //判断是否为2^n
  112.             return TRUE;
  113.     }

  115.     return FALSE;
  116. }


  119. int set_ip_netmask(unsigned char ip[16])
  120. {
  121.     return set_addr(ip, SIOCSIFNETMASK);
  122. }

  124. int set_ip(unsigned char ip[16])
  125. {
  126.     return set_addr(ip, SIOCSIFADDR);
  127. }

  129. static int set_addr(unsigned char ip[16], int flag)
  130. {
  131.     struct ifreq ifr;
  132.     struct sockaddr_in sin;
  133.     int sockfd;

  135.     if (is_valid_ip(ip) < 0)
  136.     {
  137.         printf("ip was invalid! ");
  138.         return FALSE;
  139.     }

  141.     sockfd = socket(AF_INET, SOCK_DGRAM, 0);
  142.     if(sockfd == -1){
  143.         fprintf(stderr, "Could not get socket. ");
  144.         perror("eth0 ");
  145.         return FALSE;
  146.     }

  148.     snprintf(ifr.ifr_name, (sizeof(ifr.ifr_name) - 1), "%s", DEFAULT_ETH);

  150.     /* Read interface flags */
  151.     if (ioctl(sockfd, SIOCGIFFLAGS, &ifr) < 0) {
  152.         fprintf(stderr, "ifdown: shutdown ");
  153.         perror(ifr.ifr_name);
  154.         return FALSE;
  155.     }

  157.     memset(&sin, 0, sizeof(struct sockaddr));
  158.     sin.sin_family = AF_INET;
  159.     inet_aton(ip, &sin.sin_addr.s_addr);
  160.     memcpy(&ifr.ifr_addr, &sin, sizeof(struct sockaddr));
  161.     if (ioctl(sockfd, flag, &ifr) < 0){
  162.         fprintf(stderr, "Cannot set IP address. ");
  163.         perror(ifr.ifr_name);
  164.         return FALSE;
  165.     }

  167.     return TRUE;
  168. }


  171. int set_gateway(unsigned char ip[16])
  172. {
  173.     int sockFd;
  174.     struct sockaddr_in sockaddr;
  175.     struct rtentry rt;

  177.     if (is_valid_ip(ip) < 0)
  178.     {
  179.         printf("gateway was invalid! ");
  180.         return FALSE;
  181.     }

  183.     sockFd = socket(AF_INET, SOCK_DGRAM, 0);
  184.     if (sockFd < 0)
  185.     {
  186.         perror("Socket create error. ");
  187.         return FALSE;
  188.     }

  190.     memset(&rt, 0, sizeof(struct rtentry));
  191.     memset(&sockaddr, 0, sizeof(struct sockaddr_in));
  192.     sockaddr.sin_family = AF_INET;
  193.     sockaddr.sin_port = 0;
  194.     if(inet_aton(ip, &sockaddr.sin_addr)<0)
  195.     {
  196.         perror("inet_aton error " );
  197.         close(sockFd);
  198.         return FALSE;
  199.     }

  201.     memcpy ( &rt.rt_gateway, &sockaddr, sizeof(struct sockaddr_in));
  202.     ((struct sockaddr_in *)&rt.rt_dst)->sin_family=AF_INET;
  203.     ((struct sockaddr_in *)&rt.rt_genmask)->sin_family=AF_INET;
  204.     rt.rt_flags = RTF_GATEWAY;
  205.     if (ioctl(sockFd, SIOCADDRT, &rt)<0)
  206.     {
  207.         perror("ioctl(SIOCADDRT) error in set_default_route ");
  208.         close(sockFd);
  209.         return FALSE;
  210.     }

  212.     return TRUE;
  213. }

热门文章