linux下檢測U盤插入並讀取檔案
阿新 • • 發佈：2019-02-13
獲得U盤的插入或者拔取得資訊的傳統方法是在核心級執行hotplug程式，相關引數通過環境變數傳遞過來，再由hotplug通知其他關注hotplug的應用程式。這樣的做法效率有些低，現在通過一種特殊型別的socket netlink實現獲取U盤拔插的資訊。netlink專門用於核心空間和使用者空間的非同步通訊。
由於NETLINK是Linux內建功能，所以使用起來很簡單：建立一個AF_NETLINK協議族下NETLINK_KOBJECT_UEVENT型別的特殊檔案描述符（套結字）CppLive，然後利用setsocketopt允許該檔案描述符（套結字）複用其他埠，再利用band函式將自身程序繫結到特殊檔案描述符（套結字）CppLive，最後利用select在while迴圈內監聽CppLive是否可讀，如果可讀則呼叫recv接收Linux系統核心傳遞過來的資料並打印出來，這些輸出便是USB熱插拔資訊。當然你也可以個性化地處理來自核心的熱插拔資訊，讓程式變得更加智慧以及人性化。
建立NET_LINK套接字,與核心通訊實時檢測U狀態(與基本的套接字程式設計基本一致)
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#define UEVENT_BUFFER_SIZE 2048
static int init_hotplug_sock(void);
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);
printf("%s\n", buf);
}
return 0;
}
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;
}