在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
之类的信息。