现在的位置: 首页 > 综合 > 正文

Linux remote USB device detection

2013年06月12日 ⁄ 综合 ⁄ 共 8896字 ⁄ 字号 小 中 大 ⁄ 评论关闭

Program flow:

 

Server code:

  1. /*
  2.  * usbdetect_server.c
  3.  *
  4.  * Copyright (C) 2008 Breathomn <breathomn@sohu.com>
  5.  *
  6.  */
  7. #include <stdio.h>
  8. #include <stdlib.h>
  9. #include <string.h>
  10. #include <ctype.h>
  11. #include <sys/un.h>
  12. #include <sys/ioctl.h>
  13. #include <sys/socket.h>
  14. #include <linux/types.h>
  15. #include <linux/netlink.h>
  16. #include <errno.h>
  17. #include <unistd.h>
  18. #include <arpa/inet.h>
  19. #include <netinet/in.h>
  21. #define UEVENT_BUFFER_SIZE 2048
  22. #define PORT 9606
  24. static char adstr[41] = "add@/devices/pci0000:00/";
  25. static char rmstr[44] = "remove@/devices/pci0000:00/";
  27. /* command array */
  28. char cmd[40];
  30. /* communication package */
  31. typedef struct bus_msg
  32. {
  33.     char msgtype;
  34.     char busname[4];
  35. } BUS_MSG;
  38. static int init_hotplug_sock()
  39. {
  40.     const int buffersize = 1024;
  41.     int ret;
  43.     struct sockaddr_nl snl;
  44.     bzero(&snl, sizeof(struct sockaddr_nl));
  45.     snl.nl_family = AF_NETLINK;
  46.     snl.nl_pid = getpid();
  47.     snl.nl_groups = 1;
  49.     int s = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
  50.     if (s == -1) 
  51.     {
  52.         perror("socket");
  53.         return -1;
  54.     }
  55.     setsockopt(s, SOL_SOCKET, SO_RCVBUF, &buffersize, sizeof(buffersize));
  57.     ret = bind(s, (struct sockaddr *)&snl, sizeof(struct sockaddr_nl));
  58.     if (ret < 0) 
  59.     {
  60.         perror("bind");
  61.         close(s);
  62.         return -1;
  63.     }
  65.     return s;
  66. }
  68. char* get_busname(char* buf)
  69. {
  70.     int len = strlen(buf);
  71.     char *current = buf + len - 1;
  72.     while(*(current - 1) != '/') current--;
  73.     return current;
  74. }
  76. static int send_buf(struct in_addr destip, char *buf, int len)
  77. {
  78.     struct sockaddr_in si_other;
  79.     int s, ret;
  80.     if ((s = socket (AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
  81.         perror ("socket");
  82.     memset ((char *) &si_other, sizeof (si_other), 0);
  83.     si_other.sin_family = AF_INET;
  84.     si_other.sin_port = htons (PORT);
  85.     si_other.sin_addr.s_addr = destip.s_addr;
  86.     ret = sendto (s, buf, len, 0, (struct sockaddr *) &si_other, sizeof (si_other));
  87.     if (ret == -1)
  88.         perror ("sendto() error");
  89.     else if (ret != len)
  90.         perror ("sendapmsg UDP message not complete!");
  91.     else
  92.     {
  93.         printf ("send a udp message to %s/n", inet_ntoa (destip));
  94.     }
  95.     close (s);
  96.     return 1;
  97. }
  99. int main(int argc, char* argv[])
  100. {
  101.     /* Two switch specified for filtering the same adstr & rmstr */
  102.     unsigned char insertflag = 0x00;
  103.     unsigned char removeflag = 0x00;
  105.     int hotplug_sock = init_hotplug_sock();
  107.     while(1)
  108.     {
  109.         /* Netlink message buffer */
  110.         char buf[UEVENT_BUFFER_SIZE * 2] = {0};
  111.         int adeql = 0, rmeql = 0, recvlen = 0;
  112.         int buflen;
  113.         int i;
  115.         /* sendbuf specified for sending busname to the other end */
  116.         char *sendbuf;
  117.         BUS_MSG *bus_msg;
  118.         int sendbuf_len;
  120.         /* Allocate a send buffer and put it zero */
  121.         sendbuf_len = sizeof(struct bus_msg);
  122.         sendbuf = (char *)malloc(sendbuf_len);
  123.         bzero (sendbuf, sendbuf_len);
  124.         bus_msg = (BUS_MSG*)sendbuf;
  126.         recv(hotplug_sock, &buf, sizeof(buf), 0);
  128. //      printf("%s/n", buf);
  129.         buflen = strlen(buf);
  131.         /* Match plugged string */
  132.         adeql = memcmp(buf, adstr, strlen(adstr));
  134.         /* Match unplugged string */
  135.         rmeql = memcmp(buf, rmstr, strlen(rmstr));
  137.         /* Any device plugged into server, notify client */
  138.         if (adeql == 0)
  139.         {
  140.             insertflag = ~insertflag;
  141.             if (insertflag == 0xff)
  142.             {
  143.                 printf("Device plugged!/n");
  144.                 bus_msg->msgtype = '1';
  145.                 char* str = get_busname(buf);
  146.                 sprintf(cmd, "bind_usbip ");
  147.                 strcat(cmd, str);
  148.                 system(cmd);
  149.                 strncpy(bus_msg->busname, str, 3);
  150.                 struct in_addr destip;
  151.                 inet_aton ("127.0.0.1", &destip);
  152.                 send_buf (destip, sendbuf, sendbuf_len);
  153.                 printf("********************************/n");
  154.             }
  155.         }
  157.         /* Any device unplugged from server, notify client */
  158.         if (rmeql == 0)
  159.         {
  160.             removeflag = ~removeflag;
  161.             if (removeflag == 0x00)
  162.             {
  163.                 printf("Device unplugged!/n");
  164.                 bus_msg->msgtype = '0';
  165.                 char* str = get_busname(buf);
  166.                 sprintf(cmd, "unbind_usbip ");
  167.                 strcat(cmd, str);
  168.                 system(cmd);
  169.                 strncpy(bus_msg->busname, str, 3);
  170.                 struct in_addr destip;
  171.                 inet_aton ("127.0.0.1", &destip);
  172.                 send_buf (destip, sendbuf, sendbuf_len);
  173.                 printf("********************************/n");
  174.             }
  175.         }
  176.         free(sendbuf);
  177.     }
  178.     return 0;
  179. }

Client code:

  1. /*
  2.  * usbdetect_client.c
  3.  *
  4.  * Copyright (C) 2008 Breathomn <breathomn@sohu.com>
  5.  *
  6.  */
  7. #include <arpa/inet.h>
  8. #include <netinet/in.h>
  9. #include <stdlib.h>
  10. #include <stdio.h>
  11. #include <sys/types.h>
  12. #include <sys/socket.h>
  13. #include <unistd.h>
  14. #include <sys/time.h>
  15. #include <sys/select.h>
  16. #include <string.h>
  17. #include <errno.h>
  19. #define PORT 9606
  20. #define RECVBUFLEN 5
  22. typedef struct bus_msg
  23. {
  24.     char msgtype;
  25.     char busname[4];
  26. } BUS_MSG;
  28. typedef struct device_table
  29. {
  30.     u_int8_t portnum;
  31.     char busname[4];
  32. } DEVICE_TABLE;
  33. DEVICE_TABLE local_table[8];
  35. static void init_table()
  36. {
  37.     int i;
  38.     for (i = 0; i < 8; i++)
  39.         local_table[i].portnum = 9;
  40. }
  42. static int add_to_table(BUS_MSG *bus_msg)
  43. {
  44.     int i;
  45.     for (i = 0; i < 8; i++)
  46.     {
  47.         if (local_table[i].portnum == 9)
  48.         {
  49.             local_table[i].portnum = i;
  50.             strcpy(local_table[i].busname, bus_msg->busname);
  51.             return 0;
  52.         }
  53.     }
  54.     return -1;
  55.     
  56. }
  58. static int del_from_table(BUS_MSG *bus_msg)
  59. {
  60.     int i;
  61.     for (i = 0; i < 8; i++)
  62.     {
  63.         if (strcmp(local_table[i].busname, bus_msg->busname) == 0)
  64.         {
  65.             local_table[i].portnum = 9;
  66.             strcpy(local_table[i].busname, "");
  67.             return i;
  68.         }
  69.     }
  70.     return -1;
  71. }
  73. static int get_port_from_table(BUS_MSG *bus_msg)
  74. {
  75.     int i;
  76.     for (i = 0; i < 8; i++)
  77.     {
  78.         if (strcmp(local_table[i].busname, bus_msg->busname) == 0)
  79.         {
  80.             return (int)local_table[i].portnum;
  81.         }
  82.     }
  83.     return -1;
  84. }
  86. static void print_table()
  87. {
  88.     int i;
  89.     printf("******************** Local Table ***************************/n");
  90.     for (i = 0; i < 8; i++)
  91.         printf("%d/t%s/n", local_table[i].portnum, local_table[i].busname);
  92.     printf("************************************************************/n");
  93. }
  95. int main()
  96. {
  97.     struct sockaddr_in si_me, si_other;
  98.     int s;
  99.     socklen_t slen = sizeof(si_other), ret;
  100.     char buf[RECVBUFLEN];
  101.     char cmd[40];
  102.     int tableret;
  103.     fd_set readsock;
  104.     BUS_MSG *bus_msg;
  106.     init_table();
  107.     print_table();
  109.     /* Initialize the socket to recv messages */
  110.     if ((s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP)) == -1)
  111.         perror ("socket");
  112.     memset ((char *)&si_me, 0, sizeof(si_me));
  113.     si_me.sin_family = AF_INET;
  114.     si_me.sin_port = htons (PORT);
  115.     si_me.sin_addr.s_addr = htonl (INADDR_ANY);
  116.     if (bind (s, (struct sockaddr *)&si_me, sizeof (si_me)) == -1)
  117.         perror("bind");
  118.     while (1)
  119.     {
  120.         memset(buf, 0, sizeof(buf));
  121.         FD_ZERO(&readsock);
  122.         FD_SET(s, &readsock);
  123.         ret = select(FD_SETSIZE, &readsock, NULL, NULL, NULL);
  124.         if (ret == -1)
  125.             perror("error in select() at main loop");
  126.         if (ret > 0)
  127.         {
  128.             if (FD_ISSET(s, &readsock))
  129.             {
  130.                 if (recvfrom(s, buf, RECVBUFLEN, 0, (struct sockaddr *)&si_other, &slen) == -1)
  131.                     perror("recvfrom()");
  132.                 bus_msg = (BUS_MSG*)buf;
  133.                 /* Any device plugged into server */
  134.                 if (bus_msg->msgtype == '1')
  135.                 {
  136.                     tableret = add_to_table(bus_msg);
  137.                     if (tableret != 0)
  138.                     {
  139.                         printf("No hub any more./n");
  140.                         return -1;
  141.                     }
  142.                     print_table();
  144.                     printf("Received a bus (%s) from %s/n", bus_msg->busname, inet_ntoa(si_other.sin_addr));
  145.                     sprintf(cmd, "usbip_attach ");
  146.                     strcat(cmd, inet_ntoa(si_other.sin_addr));
  147.                     strcat(cmd, " ");
  148.                     strcat(cmd, bus_msg->busname);
  149.                     system(cmd);
  150.                 }
  152.                 /* Any device unplugged from server */
  153.                 if (bus_msg->msgtype == '0')
  154.                 {
  155.                     char str[2];
  156.                     tableret = del_from_table(bus_msg);
  157.                     if (tableret == -1)
  158.                     {
  159.                         printf("Device not found./n");
  160.                         return -1;
  161.                     }
  162.                     print_table();
  164.                     printf("Received a bus (%s) from %s/n", bus_msg->busname, inet_ntoa(si_other.sin_addr));
  165.                     sprintf(cmd, "usbip_detach ");
  166.                     sprintf(str, "%d", tableret);
  167.                     strcat(cmd, str);
  168.                     system(cmd);
  169.                 }
  170.             }
  171.         }
  172.     }
  173.     close(s);
  174. }

 

返回
【上篇】
【下篇】

作者:

抱歉!评论已关闭.

返回首页

Copyright © 2013-2018 学步园  保留所有权利.
软文销售 QQ客服：2265327166 点击这里给我发消息（其他合作也可洽谈）
必威体育
必威电竞