linux c监控U盘插拔事件

在linux下U盘插拔事件的分发采用的是netlink机制。netlink是一种linux下内核态和用户态进程间通讯(IPC)的方法，常用的还有ioctl调用(通过HDIO_GET_IDENTITY获取硬盘序列号),proc文件映射(/proc/cpuinfo可以看到CPU相关信息).相比于后面两者，netlink采用了socket通信方法，于是一下子就多了超多的优点。比如socket可以实现异步通信，允许对多个进程进行广播。最重要的是内核态和用户态可以全双工通信，来取代ioctl和proc的客户端轮询。

直接上代码示范如何监听netlink下面的NETLINK_KOBJECT_UEVENT消息获取U盘插拔事件。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <sys/un.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <linux/types.h>
#include <linux/netlink.h>
#include <errno.h>
#include <unistd.h>
#include <regex>
#include <iostream>
static int init_hotplug_sock(void)
{
    struct sockaddr_nl snl;
    const int buffersize = 16 * 1024 * 1024;
    int retval;
    memset(&snl, 0x00, sizeof(struct sockaddr_nl));
    snl.nl_family = AF_NETLINK;
    snl.nl_pid = getpid();
    snl.nl_groups = 1;
    int hotplug_sock = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
    if (hotplug_sock == -1)
    {
        printf("error getting socket: %s", strerror(errno));
        return -1;
    }
    /* set receive buffersize */
    setsockopt(hotplug_sock, SOL_SOCKET, SO_RCVBUFFORCE, &buffersize, sizeof(buffersize));
    retval = bind(hotplug_sock, (struct sockaddr *) &snl, sizeof(struct sockaddr_nl));
    if (retval < 0) {
        printf("bind failed: %s", strerror(errno));
        close(hotplug_sock);
        hotplug_sock = -1;
        return -1;
    }
    return hotplug_sock;
}

main函数调用时函数如下：

int main(int argc, char* argv[])
{
         int hotplug_sock       = init_hotplug_sock();
         while(1)
         {
               char buf[UEVENT_BUFFER_SIZE*2] = {0};
               recv(hotplug_sock, &buf, sizeof(buf), 0);
               std::cout<< buf << std::endl;
         }
         return 0;
}

经过以上代码其实已经可以实现U盘的插拔检测功能，但是可以发现屏幕输出的信息特别的多，我们只需要保留自己关心的信息，比如我只需要知道插入的U盘输入sdb还是sdc之类，所以加了下面一个正则过滤函数：

               std::cmatch regex_result;
               if(std::regex_match(buf,regex_result,std::regex("(\\w+)@.*?/block/(s\\w{2})")))
               {
                   for(auto x=regex_result.begin()+1;x!=regex_result.end();++x)
                   {
                       if(!x->str().empty())
                       {
                            std::cout << x->str()<< "  ";
                       }
                   }
                   std::cout << std::endl;
               }

至此屏幕上只会输出诸如add sdc,remove sdb 之类的信息。