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; }