1. 程式人生 > >linux app應用如何檢測USB裝置熱插拔

linux app應用如何檢測USB裝置熱插拔

框架是這樣的。原理就是建立一個socket捕獲核心發過來的netlink訊息,很簡單的。

#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 <arpa/inet.h>
#include <netinet/in.h>

#define UEVENT_BUFFER_SIZE 2048

static int init_hotplug_sock()
{
    const int buffersize = 1024;
    int ret;

    struct sockaddr_nl snl;
    bzero(&snl, sizeof(struct sockaddr_nl));
    snl.nl_family = AF_NETLINK;
    snl.nl_pid = getpid();
    snl.nl_groups = 1;

    int s = socket(PF_NETLINK, SOCK_DGRAM, NETLINK_KOBJECT_UEVENT);
    if (s == -1) 
    {
        perror("socket");
        return -1;
    }
    setsockopt(s, SOL_SOCKET, SO_RCVBUF, &buffersize, sizeof(buffersize));

    ret = bind(s, (struct sockaddr *)&snl, sizeof(struct sockaddr_nl));
    if (ret < 0) 
    {
        perror("bind");
        close(s);
        return -1;
    }

    return s;
}

int main(int argc, char* argv[])
{
    int hotplug_sock = init_hotplug_sock();

    while(1)
    {
        /* Netlink message buffer */
        char buf[UEVENT_BUFFER_SIZE * 2] = {0};
        recv(hotplug_sock, &buf, sizeof(buf), 0);
printf("%s\n", buf);

        /* USB 裝置的插拔會出現字元資訊,通過比較不同的資訊確定特定裝置的插拔,在這新增比較程式碼 */

    }
    return 0;
}