2016-11-24 73 views
2

我想通過一個netlink套接字,當USB設備(dis)連接時,從內核接收uevent。我有一個python腳本,做同樣的工作,它的工作原理,但我需要在C相同的功能到目前爲止,我有這樣的:如何使用Netlink套接字接收內核事件?

#include <sys/types.h> 
#include <sys/socket.h> 
#include <linux/netlink.h> 
#include <unistd.h> 
#include <stdlib.h> 
#include <stdio.h> 
#include <string.h> 

#define NL_MAX_PAYLOAD 8192 

int main() { 
    int nl_socket; 
    struct sockaddr_nl src_addr, dest_addr; 
    struct nlmsghdr* nlh; 
    struct msghdr msg; 
    struct iovec iov; 

    // Prepare source address 
    memset(&src_addr, 0, sizeof(src_addr)); 
    src_addr.nl_family = AF_NETLINK; 
    src_addr.nl_pid = getpid(); 

    // Prepare destination address 
    memset(&dest_addr, 0, sizeof(dest_addr)); 
    dest_addr.nl_family = AF_NETLINK; 
    dest_addr.nl_pid = 0; 
    dest_addr.nl_groups = 0; 

    // Prepare netlink message 
    nlh = (struct nlmsghdr*) malloc(NLMSG_SPACE(NL_MAX_PAYLOAD)); 
    memset(nlh, 0, NLMSG_SPACE(NL_MAX_PAYLOAD)); 
    nlh->nlmsg_len = NLMSG_SPACE(NL_MAX_PAYLOAD); 
    nlh->nlmsg_pid = getpid(); 
    nlh->nlmsg_flags = 0; 
    iov.iov_base = (void *) nlh; 
    iov.iov_len = nlh->nlmsg_len; 
    msg.msg_iov = &iov; 
    msg.msg_iovlen = 1; 

    nl_socket = socket(AF_NETLINK, SOCK_DGRAM,  NETLINK_KOBJECT_UEVENT); 
    if (nl_socket < 0) { 
     printf("Failed to create socket for DeviceFinder"); 
     exit(1); 
    } 

    bind(nl_socket, (struct sockaddr*) &src_addr, sizeof(src_addr)); 

    while (1) { 
     int r = recvmsg(nl_socket, &msg, 0); 
     printf("%i\n", r); 
     if (r < 0) { 
      perror(""); 
     } 
    } 
} 

這是斷章取義,以創建一個小的,工作示例。

問題是我只是沒有收到任何東西。我沒有得到任何錯誤,recvmsg的調用在我(不)插入設備時從未返回。爲什麼?

回答

1

接收部分是不正確的,而不是用我使用的recv的recvmmsg,我已經修改了計劃了一下,以下是打印熱插拔事件代碼:

#include <sys/types.h> 
#include <sys/socket.h> 
#include <linux/netlink.h> 
#include <unistd.h> 
#include <stdlib.h> 
#include <stdio.h> 
#include <string.h> 

#define NL_MAX_PAYLOAD 8192 

int main() { 
    int nl_socket; 
    struct sockaddr_nl src_addr; 
    char msg[NL_MAX_PAYLOAD]; 
    int ret; 

    // Prepare source address 
    memset(&src_addr, 0, sizeof(src_addr)); 
    src_addr.nl_family = AF_NETLINK; 
    src_addr.nl_pid = getpid(); 
    src_addr.nl_groups = -1; 

    nl_socket = socket(AF_NETLINK, (SOCK_DGRAM | SOCK_CLOEXEC),  NETLINK_KOBJECT_UEVENT); 
    if (nl_socket < 0) { 
     printf("Failed to create socket for DeviceFinder"); 
     exit(1); 
    } 

    ret = bind(nl_socket, (struct sockaddr*) &src_addr, sizeof(src_addr)); 
    if (ret) { 
     printf("Failed to bind netlink socket.."); 
     close(nl_socket); 
     return 1; 
    } 

    printf("Waiting for events now...\n"); 
    while (1) { 
     int r = recv(nl_socket, msg, sizeof(msg), MSG_DONTWAIT); 
     if (r == -1) 
      continue; 
     if (r < 0) { 
      perror(""); 
      continue; 
     } 
     printf("length:%i\n msg:%s", r, msg); 

    } 
} 
+0

謝謝,這工作完美,儘管由於某種原因很多消息是空的。那些我需要的確定。 –

0

這很好地利用recvmsg比的recv(只對於連接的鏈接).ref man recvmsg。 和程序接近成功。

src_addr.nl_family = AF_NETLINK; 
src_addr.nl_pid = getpid(); 
src_addr.nl_groups = -1; <--add this so can receive from kernel broadcast 
相關問題