转贴：Linux Netlink Socket Example

2. #include <asm/types.h>
4. #include <sys/socket.h>
5. #include <unistd.h>
6. #include <err.h>
7. #include <stdio.h>
8. #include <netinet/in.h>
10. #include <linux/netlink.h>
11. #include <linux/rtnetlink.h>
13. #if 0
14. //#define MYPROTO NETLINK_ARPD
15. #define MYMGRP RTMGRP_NEIGH
16. // if you want the above you'll find that the kernel must be compiled with CONFIG_ARPD, and
17. // that you need MYPROTO=NETLINK_ROUTE, since the kernel arp code {re,ab}uses rtnl (NETLINK_ROUTE)
19. #else
20. #define MYPROTO NETLINK_ROUTE
21. #define MYMGRP RTMGRP_IPV4_ROUTE
22. #endif
24. struct msgnames_t {
25.         int id;
26.         char *msg;
27. } typenames[] = {
28. #define MSG(x) { x, #x }
29.         MSG(RTM_NEWROUTE),
30.         MSG(RTM_DELROUTE),
31.         MSG(RTM_GETROUTE),
32. #undef MSG
33.         {0,0}
34. };
36. char *lookup_name(struct msgnames_t *db,int id)
37. {
38.         static char name[512];
39.         struct msgnames_t *msgnamesiter;
40.         for(msgnamesiter=db;msgnamesiter->msg;++msgnamesiter) {
41.                 if (msgnamesiter->id == id)
42.                         break;
43.         }
44.         if (msgnamesiter->msg) {
45.                 return msgnamesiter->msg;
46.         }
47.         snprintf(name,sizeof(name),"#%i",id);
48.         return name;
49. }
51. int open_netlink()
52. {
53.         int sock = socket(AF_NETLINK,SOCK_RAW,MYPROTO);
54.         struct sockaddr_nl addr;
56.         memset((void *)&addr, 0, sizeof(addr));
58.         if (sock<0)
59.                 return sock;
60.         addr.nl_family = AF_NETLINK;
61.         addr.nl_pid = getpid();
62.         addr.nl_groups = MYMGRP;
63.         if (bind(sock,(struct sockaddr *)&addr,sizeof(addr))<0)
64.                 return -1;
65.         return sock;
66. }
68. int read_event(int sock)
69. {
70.         struct sockaddr_nl nladdr;
71.         struct msghdr msg;
72.         struct iovec iov[2];
73.         struct nlmsghdr nlh;
74.         char buffer[65536];
75.         int ret;
76.         iov[0].iov_base = (void *)&nlh;
77.         iov[0].iov_len = sizeof(nlh);
78.         iov[1].iov_base = (void *)buffer;
79.         iov[1].iov_len = sizeof(buffer);
80.         msg.msg_name = (void *)&(nladdr);
81.         msg.msg_namelen = sizeof(nladdr);
82.         msg.msg_iov = iov;
83.         msg.msg_iovlen = sizeof(iov)/sizeof(iov[0]);
84.         ret=recvmsg(sock, &msg, 0);
85.         if (ret<0) {
86.                 return ret;
87.         }
88.         printf("Type: %i (%s)/n",(nlh.nlmsg_type),lookup_name(typenames,nlh.nlmsg_type));
89.         printf("Flag:");
90. #define FLAG(x) if (nlh.nlmsg_type & x) printf(" %s",#x)
91.         FLAG(NLM_F_REQUEST);
92.         FLAG(NLM_F_MULTI);
93.         FLAG(NLM_F_ACK);
94.         FLAG(NLM_F_ECHO);
95.         FLAG(NLM_F_REPLACE);
96.         FLAG(NLM_F_EXCL);
97.         FLAG(NLM_F_CREATE);
98.         FLAG(NLM_F_APPEND);
99. #undef FLAG
100.         printf("/n");
101.         printf("Seq : %i/n",nlh.nlmsg_seq);
102.         printf("Pid : %i/n",nlh.nlmsg_pid);
103.         printf("/n");
104.         return 0;
105. }
107. int main(int argc, char *argv[])
108. {         int nls = open_netlink();
109.         if (nls<0) {
110.                 err(1,"netlink");
111.         }
112.         while (1)
113.                 read_event(nls);
114.         return 0;
115. }