Linux Netlink

来源:互联网 发布:java方向算法书籍推荐 编辑:程序博客网 时间:2024/06/05 05:12

1. Netlink简介

    Netlink 是一种特殊的 socket,它是一种在内核用户间进行双向数据传输的一种方式,用户态应用使用标准的 socket API 就可以使用 Netlink 提供的强大功能,内核态需要使用专门的内核 API 来使用 Netlink。

    系统已定义的Netlink种类如下:

[cpp] view plaincopy
  1. /* kernel/include/linux/netlink.h */  
  2.   
  3. #define NETLINK_ROUTE       0   /* Routing/device hook                  */  
  4. #define NETLINK_UNUSED      1   /* Unused number                */  
  5. #define NETLINK_USERSOCK    2   /* Reserved for user mode socket protocols  */  
  6. #define NETLINK_FIREWALL    3   /* Firewalling hook             */  
  7. #define NETLINK_INET_DIAG   4   /* INET socket monitoring           */  
  8. #define NETLINK_NFLOG       5   /* netfilter/iptables ULOG */  
  9. #define NETLINK_XFRM        6   /* ipsec */  
  10. #define NETLINK_SELINUX     7   /* SELinux event notifications */  
  11. #define NETLINK_ISCSI       8   /* Open-iSCSI */  
  12. #define NETLINK_AUDIT       9   /* auditing */  
  13. #define NETLINK_FIB_LOOKUP  10  /* 转发信息表查询  */  
  14. #define NETLINK_CONNECTOR   11      /* netlink connector */  
  15. #define NETLINK_NETFILTER   12  /* netfilter subsystem */  
  16. #define NETLINK_IP6_FW      13      /* IPV6 firewall */  
  17. #define NETLINK_DNRTMSG     14  /* DECnet routing messages */  
  18. #define NETLINK_KOBJECT_UEVENT  15  /* Kernel event to userspace */  
  19. #define NETLINK_GENERIC     16      /* general netlink */  
  20. /* leave room for NETLINK_DM (DM Events) */  
  21. #define NETLINK_SCSITRANSPORT   18  /* SCSI Transports */  
  22. #define NETLINK_ECRYPTFS    19  
  23. #define NETLINK_RDMA        20  
  24.   
  25. #define MAX_LINKS 32      


2. Netlink优点

    Netlink 相对于系统调用、ioctl 以及 /proc 文件系统而言具有以下优点:
    1) 代码简化

        为了使用 netlink,用户仅需要在 include/linux/netlink.h 中增加一个新类型的 netlink 协议定义即可, 如 #define NETLINK_TEST 21。然后,内核和用户态应用就可以立即通过 socket API 使用该 netlink 协议类型进行数据交换。但系统调用需要增加新的系统调用,ioctl 则需要增加设备或文件, 那需要不少代码,proc 文件系统则需要在 /proc 下添加新的文件或目录,那将使本来就混乱的 /proc 更加混乱。
    2) 异步机制

        netlink是一种异步通信机制,在内核与用户态应用之间传递的消息保存在socket缓存队列中,发送消息只是把消息保存在接收者的socket的接 收队列,而不需要等待接收者收到消息,但系统调用与 ioctl 则是同步通信机制,如果传递的数据太长,将影响调度粒度。
    3) 模块化

        使用 netlink 的内核部分可以采用模块的方式实现,使用 netlink 的应用部分和内核部分没有编译时依赖,但系统调用就有依赖,而且新的系统调用的实现必须静态地连接到内核中,它无法在模块中实现,使用新系统调用的应用在编译时需要依赖内核。
    4) 支持多播 

        netlink 支持多播,内核模块或应用可以把消息多播给一个netlink组,属于该neilink 组的任何内核模块或应用都能接收到该消息,内核事件向用户态的通知机制就使用了这一特性,任何对内核事件感兴趣的应用都能收到该子系统发送的内核事件。
   5) 内核可发起会话
       内核可以使用 netlink 首先发起会话,但系统调用和 ioctl 只能由用户应用发起调用。
   6) 使用标准的Socket API

       netlink 使用标准的 socket API,因此很容易使用,但系统调用和 ioctl则需要专门的培训才能使用。


3. 用户态使用Netlink

3.1 创建socket

    用户态应用使用标准的socket APIs, socket(), bind(), sendmsg(), recvmsg() 和 close() 就能使用 netlink socket。注: 使用 netlink 的应用必须包含头文件 linux/netlink.h。当然 socket 需要的头文件也必不可少,sys/socket.h。

     为了创建一个 netlink socket,用户需要使用如下参数调用 socket():

      socket(AF_NETLINK, SOCK_RAW,netlink_type)

      第一个参数:必须是 AF_NETLINK 或 PF_NETLINK,在 Linux 中,它们俩实际为一个东西,它表示要使用netlink;

      第二个参数:必须是SOCK_RAW或SOCK_DGRAM;

      第三个参数:指定netlink协议类型,如前面讲的用户自定义协议类型NETLINK_TEST及系统定义的协议类型, NETLINK_GENERIC是一个通用的协议类型,它是专门为用户使用的,因此,用户可以直接使用它,而不必再添加新的协议类型。

      对于每一个netlink协议类型,可以有多达 32多播组,每一个多播组用一个位表示,netlink 的多播特性使得发送消息给同一个组仅需要一次系统调用,因而对于需要多拨消息的应用而言,大大地降低了系统调用的次数。

3.2 bind地址

      函数 bind() 用于把一个打开的 netlink socket 与 netlink 源 socket 地址绑定在一起。netlink socket 的地址结构如下:

[cpp] view plaincopy
  1. struct sockaddr_nl {  
  2.     sa_family_t nl_family;  /* AF_NETLINK   */  
  3.     unsigned short  nl_pad;     /* zero     */  
  4.     __u32       nl_pid;     /* port ID, note: kernel nl_pid is 0 */  
  5.         __u32       nl_groups;  /* multicast groups mask */  
  6. }  
       • nl_family:必须设置为 AF_NETLINK 或着 PF_NETLINK;

       • nl_pad:当前没有使用,因此要总是设置为;

       • nl_pid:为接收或发送消息的进程的 ID,如果希望内核处理消息为多播消息,就把该字段设置为 0,否则设置为处理消息的进程 ID;

       • nl_groups:用于指定多播组,bind 函数用于把调用进程加入到该字段指定的多播组,如果设置为 0,表示调用者不加入任何多播组。

       传递给 bind 函数的地址的 nl_pid 字段应当设置为本进程的进程 ID,这相当于 netlink socket 的本地地址。但是,对于一个进程的多个线程使用 netlink socket 的情况,字段 nl_pid 则可以设置为其它的值,如:

[cpp] view plaincopy
  1. pthread_self() << 16 | getpid();  

       因此字段 nl_pid 实际上未必是进程 ID,它只是用于区分不同的接收者或发送者的一个标识,用户可以根据自己需要设置该字段。函数 bind 的调用方式如下:

[cpp] view plaincopy
  1. bind(fd, (struct sockaddr*)&nladdr, sizeof(struct sockaddr_nl));  

       fd为前面的 socket 调用返回的文件描述符,参数 nladdr 为 struct sockaddr_nl 类型的地址。

3.3 发送消息

      为了发送一个 netlink 消息给内核或其他用户态应用,需要填充目标 netlink socket 地址,此时,字段 nl_pid 和 nl_groups 分别表示接收消息者的进程 ID 与多播组。如果字段 nl_pid 设置为 0,表示消息接收者为内核或多播组,如果 nl_groups为 0,表示该消息为单播消息,否则表示多播消息。

     使用函数 sendmsg 发送 netlink 消息时还需要引用结构 struct msghdr、struct nlmsghdr 和 struct iovec,结构 struct msghdr 需如下设置:

[cpp] view plaincopy
  1. struct msghdr {  
  2.     void    *   msg_name;   /* Socket name          */  
  3.     int     msg_namelen;    /* Length of name       */  
  4.     struct iovec *  msg_iov;    /* Data blocks          */  
  5.     __kernel_size_t msg_iovlen; /* Number of blocks     */  
  6.     void    *   msg_control;    /* Per protocol magic (eg BSD file descriptor passing) */  
  7.     __kernel_size_t msg_controllen; /* Length of cmsg list */  
  8.     unsigned    msg_flags;  
  9. };  


[cpp] view plaincopy
  1. struct msghdr msg;  
  2. struct sockaddr_nl nladdr;  
  3. memset(&msg, 0, sizeof(msg));  
  4. msg.msg_name = (void *)&(nladdr);  
  5. msg.msg_namelen = sizeof(nladdr);  

    其中 nladdr 为消息接收者的 netlink 地址。

    struct nlmsghdr 为 netlink socket 自己的消息头,这用于多路复用和多路分解 netlink 定义的所有协议类型以及其它一些控制,netlink 的内核实现将利用这个消息头来多路复用和多路分解已经其它的一些控制,因此它也被称为netlink 控制块。因此,应用在发送 netlink 消息时必须提供该消息头。

[cpp] view plaincopy
  1. struct nlmsghdr {  
  2.     __u32       nlmsg_len;  /* Length of message including header */  
  3.     __u16       nlmsg_type; /* Message content */  
  4.     __u16       nlmsg_flags;    /* Additional flags */  
  5.     __u32       nlmsg_seq;  /* Sequence number */  
  6.     __u32       nlmsg_pid;  /* Sending process port ID */  
  7. };  

       • nlmsg_len:指定消息的总长度,包括紧跟该结构的数据部分长度以及该结构的大小

       • nlmsg_type:用于应用内部定义消息的类型,它对 netlink 内核实现是透明的,因此大部分情况下设置为 0

       • nlmsg_seq:用于应用追踪消息,表示顺序号

       • nlmsg_pid:用于应用追踪消息,表示消息来源进程 ID

       • nlmsg_flags:用于设置消息标志,可用的标志包括:

[cpp] view plaincopy
  1. /* Flags values */  
  2.   
  3. #define NLM_F_REQUEST       1   /* It is request message.   */  
  4. #define NLM_F_MULTI     2   /* Multipart message, terminated by NLMSG_DONE */  
  5. #define NLM_F_ACK       4   /* Reply with ack, with zero or error code */  
  6. #define NLM_F_ECHO      8   /* Echo this request        */  
  7.   
  8. /* Modifiers to GET request */  
  9. #define NLM_F_ROOT  0x100   /* specify tree root    */  
  10. #define NLM_F_MATCH 0x200   /* return all matching  */  
  11. #define NLM_F_ATOMIC    0x400   /* atomic GET       */  
  12. #define NLM_F_DUMP  (NLM_F_ROOT|NLM_F_MATCH)  
  13.   
  14. /* Modifiers to NEW request */  
  15. #define NLM_F_REPLACE   0x100   /* Override existing        */  
  16. #define NLM_F_EXCL  0x200   /* Do not touch, if it exists   */  
  17. #define NLM_F_CREATE    0x400   /* Create, if it does not exist */  
  18. #define NLM_F_APPEND    0x800   /* Add to end of list       */  

 • NLM_F_REQUEST:用于表示消息是一个请求,所有应用首先发起的消息都应设置该标志。
 • NLM_F_MULTI:用于指示该消息是一个多部分消息的一部分,后续的消息可以通过宏NLMSG_NEXT来获得。
 • NLM_F_ACK:表示该消息是前一个请求消息的响应,顺序号与进程ID可以把请求与响应关联起来。
 • NLM_F_ECHO:表示该消息是相关的一个包的回传。
 • NLM_F_ROOT:被许多 netlink 协议的各种数据获取操作使用,该标志指示被请求的数据表应当整体返回用户应用,而不是一个条目一个条目地返回。有该标志的请求通常导致响应消息设置 NLM_F_MULTI标志。注意,当设置了该标志时,请求是协议特定的,因此,需要在字段 nlmsg_type 中指定协议类型。
 • NLM_F_MATCH: 表示该协议特定的请求只需要一个数据子集,数据子集由指定的协议特定的过滤器来匹配。
 • NLM_F_ATOMIC: 指示请求返回的数据应当原子地收集,这预防数据在获取期间被修改。
 • NLM_F_DUMP: 未实现。
 • NLM_F_REPLACE: 用于取代在数据表中的现有条目。
 • NLM_F_EXCL:用于和 CREATE 和 APPEND 配合使用,如果条目已经存在,将失败。
 • NLM_F_CREATE: 指示应当在指定的表中创建一个条目。
 • NLM_F_APPEND: 指示在表末尾添加新的条目。
 
      内核需要读取和修改这些标志,对于一般的使用,用户把它设置为 0 就可以,只是一些高级应用(如 netfilter 和路由 daemon 需要它进行一些复杂的操作),下面是一个示例:

[cpp] view plaincopy
  1. #define MAX_MSGSIZE 1024  
  2. char buffer[] = "An example message";  
  3. struct nlmsghdr nlhdr;  
  4. nlhdr = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_MSGSIZE));  
  5. strcpy(NLMSG_DATA(nlhdr),buffer);  
  6. nlhdr->nlmsg_len = NLMSG_LENGTH(strlen(buffer));  
  7. nlhdr->nlmsg_pid = getpid();  /* self pid */  
  8. nlhdr->nlmsg_flags = 0;  

结构 struct iovec 用于把多个消息通过一次系统调用来发送,下面是该结构使用示例:

[cpp] view plaincopy
  1. struct iovec  
  2. {  
  3.     void __user *iov_base;  /* BSD uses caddr_t (1003.1g requires void *),save nlmsghdr */  
  4.     __kernel_size_t iov_len; /* Must be size_t (1003.1g), save nlmsghdr->nlmsg_len */  
  5. };  

[cpp] view plaincopy
  1. struct iovec iov;  
  2. iov.iov_base = (void *)nlhdr;  
  3. iov.iov_len = nlh->nlmsg_len;  
  4. msg.msg_iov = &iov;  
  5. msg.msg_iovlen = 1;  

在完成以上步骤后,消息就可以通过下面语句直接发送:

[cpp] view plaincopy
  1. sendmsg(fd, &msg, 0);  

3.4 接收消息

      应用接收消息时需要首先分配一个足够大的缓存来保存消息头以及消息的数据部分,然后填充消息头,之后就可以直接调用函数 recvmsg() 来接收消息,示例如下所示:

[cpp] view plaincopy
  1. #define MAX_NL_MSG_LEN 1024  
  2. struct sockaddr_nl nladdr;  
  3. struct msghdr msg;  
  4. struct iovec iov;  
  5. struct nlmsghdr * nlhdr;  
  6. nlhdr = (struct nlmsghdr *)malloc(MAX_NL_MSG_LEN);  
  7. iov.iov_base = (void *)nlhdr;  
  8. iov.iov_len = MAX_NL_MSG_LEN;  
  9. msg.msg_name = (void *)&(nladdr);  
  10. msg.msg_namelen = sizeof(nladdr);  
  11. msg.msg_iov = &iov;  
  12. msg.msg_iovlen = 1;  
  13. recvmsg(fd, &msg, 0);  

    注意:fd为socket调用打开的netlink socket描述符。在消息接收后,nlhdr指向接收到的消息的消息头,nladdr保存了接收到的消息的目标地址,宏NLMSG_DATA(nlhdr)返回指向消息的数据部分的指针。在linux/netlink.h中定义了一些方便对消息进行处理的宏,这些宏包括:

[cpp] view plaincopy
  1. #define NLMSG_ALIGNTO   4U  
  2. //用于得到不小于len且字节对齐的最小数值  
  3. #define NLMSG_ALIGN(len) ( ((len)+NLMSG_ALIGNTO-1) & ~(NLMSG_ALIGNTO-1) )  
  4.   
  5. // 用于计算nlmsghdr字节对齐的长度   
  6. #define NLMSG_HDRLEN     ((int) NLMSG_ALIGN(sizeof(struct nlmsghdr)))  
  7.   
  8. // 用于计算数据部分长度为len时实际的消息长度。它一般用于分配消息缓存  
  9. #define NLMSG_LENGTH(len) ((len)+NLMSG_ALIGN(NLMSG_HDRLEN))  
  10.   
  11. // 返回不小于NLMSG_LENGTH(len)且字节对齐的最小数值,它也用于分配消息缓存  
  12. #define NLMSG_SPACE(len) NLMSG_ALIGN(NLMSG_LENGTH(len))  
  13.   
  14. // 用于取得消息的数据部分的首地址,设置和读取消息数据部分时需要使用该宏  
  15. #define NLMSG_DATA(nlh)  ((void*)(((char*)nlh) + NLMSG_LENGTH(0)))  
  16.   
  17. // 用于得到下一个消息的首地址,同时len也减少为剩余消息的总长度,该宏一般在一个消息被分成几个部分发送或接收时使用  
  18. #define NLMSG_NEXT(nlh,len)  ((len) -= NLMSG_ALIGN((nlh)->nlmsg_len), \  
  19.                   (struct nlmsghdr*)(((char*)(nlh)) + NLMSG_ALIGN((nlh)->nlmsg_len)))  
  20.   
  21. // 用于判断消息是否有len这么长  
  22. #define NLMSG_OK(nlh,len) ((len) >= (int)sizeof(struct nlmsghdr) && \  
  23.                (nlh)->nlmsg_len >= sizeof(struct nlmsghdr) && \  
  24.                (nlh)->nlmsg_len <= (len))  
  25.   
  26. // 用于返回payload的长度  
  27. #define NLMSG_PAYLOAD(nlh,len) ((nlh)->nlmsg_len - NLMSG_SPACE((len)))  
  28.   
  29. #define NLMSG_NOOP      0x1 /* Nothing.     */  
  30. #define NLMSG_ERROR     0x2 /* Error        */  
  31. #define NLMSG_DONE      0x3 /* End of a dump    */  
  32. #define NLMSG_OVERRUN       0x4 /* Data lost        */  

4. 内核态使用Netlink

    Netlink的内核实现在net/netlink/af_netlink.c中,内核模块要想使用netlink,也必须包含头文件linux /netlink.h。内核使用netlink需要专门的API,这完全不同于用户态应用对netlink的使用。如果用户需要增加新的netlink协议类型,必须通过修改linux/netlink.h来实现,当然,目前的netlink实现已经包含了一个通用的协议类型 NETLINK_GENERIC以方便用户使用,用户可以直接使用它而不必增加新的协议类型。前面讲到,为了增加新的netlink协议类型,用户仅需增 加如下定义到linux/netlink.h就可以:

[cpp] view plaincopy
  1. #define NETLINK_TEST 21  


    只要增加这个定义之后,用户就可以在内核的任何地方引用该协议。

4.1 创建netlink socket

    在内核中,为了创建一个netlink socket用户需要调用如下函数:

[cpp] view plaincopy
  1. struct sock * netlink_kernel_create(struct net *net, int unit, unsigned int groups,  
  2.                             void (*input)(struct sk_buff *skb),  
  3.                             struct mutex *cb_mutex, struct module *module)  

    •  net:

       是一个网络名字空间namespace,在不同的名字空间里面可以有自己的转发信息库,有自己的一套net_device等等。默认情况下都是使用 init_net这个全局变量,下面是内核中调用netlink_kernel_create()函数的一个示例:

[cpp] view plaincopy
  1. static struct sock *audit_sock;  
  2. audit_sock = netlink_kernel_create(&init_net, NETLINK_AUDIT, 0,  
  3.               audit_receive, NULL, THIS_MODULE);  

    •  unit:表示netlink协议类型,如NETLINK_TEST、NETLINK_SELINUX
    •  groups:多播地址
    •  input:为内核模块定义的netlink消息处理函数,当有消 息到达这个netlink socket时,该input函数指针就会被引用,且只有此函数返回时,调用者的sendmsg才能返回。
    •  cb_mutex:为访问数据时的互斥信号量
    •  module: 一般为THIS_MODULE

4.2 发送单播消息netlink_unicast


[cpp] view plaincopy
  1. int netlink_unicast(struct sock *ssk, struct sk_buff *skb, u32 pid, int nonblock)  


    • ssk:为函数 netlink_kernel_create()返回的socket

    • skb:存放消息,它的data字段指向要发送的netlink消息结构,而 skb的控制块保存了消息的地址信息,宏NETLINK_CB(skb)就用于方便设置该控制块

    • pid:为接收此消息进程的pid,即目标地址,如果目标为组或内核,它设置为 0

    • nonblock:表示该函数是否为非阻塞,如果为1,该函数将在没有接收缓存可利用时立即返回;而如果为0,该函数在没有接收缓存可利用定时睡眠。


4.3 发送广播消息netlink_broadcast

[cpp] view plaincopy
  1. int netlink_broadcast(struct sock *ssk, struct sk_buff *skb, u32 pid, u32 group, gfp_t allocation)  

            前面的三个参数与 netlink_unicast相同,参数group为接收消息的多播组,该参数的每一个位代表一个多播组,因此如果发送给多个多播组,就把该参数设置为多个多播组组ID的位或。参数allocation为内核内存分配类型,一般地为GFP_ATOMIC或GFP_KERNEL,GFP_ATOMIC用于原子的上下文(即不可以睡眠),而GFP_KERNEL用于非原子上下文。 
[cpp] view plaincopy
  1. struct netlink_skb_parms {  
  2.     struct ucred        creds;      /* Skb credentials  */  
  3.     __u32           pid;            /* source address pid */  
  4.     __u32           dst_group;  
  5. };  
  6.   
  7. #define NETLINK_CB(skb)     (*(struct netlink_skb_parms*)&((skb)->cb))  
  8. #define NETLINK_CREDS(skb)  (&NETLINK_CB((skb)).creds)  

[cpp] view plaincopy
  1. NETLINK_CB(skb).pid = 0;  
  2. NETLINK_CB(skb).dst_group = 1;  

     NETLINK_CB(skb).pid:表示消息发送者进程 ID,也即源地址,对于内核,它为 0      

     NETLINK_CB(skb).dst_group:表示目标组地址,如果目标为某一进程或内核,dst_group 应当设置为 0

4.4 释放netlink socket

void netlink_kernel_release(struct sock *sk)


5. Netlink Kernel实现


[cpp] view plaincopy
  1. static const struct proto_ops netlink_ops = {  
  2.     .family =   PF_NETLINK,  
  3.     .owner =    THIS_MODULE,  
  4.     .release =  netlink_release,  
  5.     .bind =     netlink_bind,  
  6.     .connect =  netlink_connect,  
  7.     .socketpair =   sock_no_socketpair,  
  8.     .accept =   sock_no_accept,  
  9.     .getname =  netlink_getname,  
  10.     .poll =     datagram_poll,  
  11.     .ioctl =    sock_no_ioctl,  
  12.     .listen =   sock_no_listen,  
  13.     .shutdown = sock_no_shutdown,  
  14.     .setsockopt =   netlink_setsockopt,  
  15.     .getsockopt =   netlink_getsockopt,  
  16.     .sendmsg =  netlink_sendmsg,  
  17.     .recvmsg =  netlink_recvmsg,  
  18.     .mmap =     sock_no_mmap,  
  19.     .sendpage = sock_no_sendpage,  
  20. };  



6. 实现实例

       程序实现流程如下所示:

     1) 运行netlink内核模块;
     2) 运行用户态程序,向内核发送bind消息,通知内核自身进程id(在kernel态执行netlink_bind);
     3) 内核接收用户消息,记录其进程id;
     4) 内核向用户进程id发送netlink消息;
     5) 用户接收内核发送的netlink消息。


6.1 用户态程序

[cpp] view plaincopy
  1. #include <sys/stat.h>  
  2. #include <unistd.h>  
  3. #include <stdio.h>  
  4. #include <stdlib.h>  
  5. #include <sys/socket.h>  
  6. #include <sys/types.h>  
  7. #include <string.h>  
  8. #include <asm/types.h>  
  9. #include <linux/netlink.h>  
  10. #include <linux/socket.h>  
  11. #include <errno.h>  
  12.   
  13. #define NETLINK_TEST 21  
  14. #define MAX_PAYLOAD 1024 // maximum payload size  
  15.   
  16. int main(int argc, char* argv[])  
  17. {  
  18.     int state;  
  19.     struct sockaddr_nl src_addr, dest_addr;  
  20.     struct nlmsghdr *nlh = NULL;  
  21.     struct iovec iov;  
  22.     struct msghdr msg;  
  23.     int sock_fd, retval;  
  24.     int state_smg = 0;  
  25.     // Create a socket  
  26.   
  27.     sock_fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_TEST);  
  28.     if(sock_fd == -1){  
  29.         printf("error getting socket: %s", strerror(errno));  
  30.         return -1;  
  31.     }  
  32.   
  33.     // To prepare binding  
  34.     memset(&msg,0,sizeof(msg));  
  35.     memset(&src_addr, 0, sizeof(src_addr));  
  36.     src_addr.nl_family = AF_NETLINK;  
  37.     src_addr.nl_pid = getpid(); // self pid  
  38.     src_addr.nl_groups = 0; // multi cast  
  39.   
  40.     retval = bind(sock_fd, (struct sockaddr*)&src_addr, sizeof(src_addr));  
  41.     if(retval < 0){  
  42.         printf("bind failed: %s", strerror(errno));  
  43.         close(sock_fd);  
  44.         return -1;  
  45.     }  
  46.   
  47.     // To prepare recvmsg  
  48.   
  49.     nlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_PAYLOAD));  
  50.     if(!nlh){  
  51.         printf("malloc nlmsghdr error!\n");  
  52.         close(sock_fd);  
  53.         return -1;  
  54.     }  
  55.   
  56.     memset(&dest_addr,0,sizeof(dest_addr));  
  57.     dest_addr.nl_family = AF_NETLINK;  
  58.     dest_addr.nl_pid = 0;  
  59.     dest_addr.nl_groups = 0;  
  60.   
  61.     nlh->nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);  
  62.     nlh->nlmsg_pid = getpid(); // self pid  
  63.     nlh->nlmsg_flags = 0;  
  64.     strcpy(NLMSG_DATA(nlh),"Hello you!");  
  65.   
  66.     iov.iov_base = (void *)nlh;  
  67.     iov.iov_len = NLMSG_SPACE(MAX_PAYLOAD);  
  68.     // iov.iov_len = nlh->nlmsg_len;  
  69.   
  70.     memset(&msg, 0, sizeof(msg));  
  71.      
  72.     msg.msg_name = (void *)&dest_addr;  
  73.     msg.msg_namelen = sizeof(dest_addr);  
  74.     msg.msg_iov = &iov;  
  75.     msg.msg_iovlen = 1;  
  76.   
  77.     printf("state_smg\n");  
  78.     state_smg = sendmsg(sock_fd,&msg,0);  
  79.   
  80.     if(state_smg == -1)  
  81.     {  
  82.         printf("get error sendmsg = %s\n",strerror(errno));  
  83.     }  
  84.   
  85.     memset(nlh,0,NLMSG_SPACE(MAX_PAYLOAD));  
  86.     printf("waiting received!\n");  
  87.     // Read message from kernel  
  88.   
  89.     while(1){  
  90.         printf("In while recvmsg\n");  
  91.         state = recvmsg(sock_fd, &msg, 0);  
  92.         if(state<0)  
  93.         {  
  94.             printf("state<1");  
  95.         }  
  96.         printf("In while\n");  
  97.         printf("Received message: %s\n",(char *) NLMSG_DATA(nlh));  
  98.     }  
  99.   
  100.     close(sock_fd);  
  101.   
  102.     return 0;  
  103. }  


6.2 内核态程序

[cpp] view plaincopy
  1. #include <linux/init.h>  
  2. #include <linux/module.h>  
  3. #include <linux/timer.h>  
  4. #include <linux/time.h>  
  5. #include <linux/types.h>  
  6. #include <net/sock.h>  
  7. #include <net/netlink.h>   
  8.   
  9. #define NETLINK_TEST 21  
  10. #define MAX_MSGSIZE 1024  
  11. int stringlength(char *s);  
  12. void sendnlmsg(char * message);  
  13. static int pid; // user process pid  
  14. static int err;  
  15. static struct sock *nl_sk = NULL;  
  16. static int flag = 0;  
  17.   
  18. void sendnlmsg(char *message)  
  19. {  
  20.     struct sk_buff *skb_1;  
  21.     struct nlmsghdr *nlh;  
  22.     int len = NLMSG_SPACE(MAX_MSGSIZE);  
  23.     int slen = 0;  
  24.     if(!message || !nl_sk)  
  25.     {  
  26.         return ;  
  27.     }  
  28.       
  29.     skb_1 = alloc_skb(len,GFP_KERNEL);  
  30.     if(!skb_1)  
  31.     {  
  32.         printk(KERN_ERR "my_net_link:alloc_skb_1 error\n");  
  33.     }  
  34.       
  35.     slen = stringlength(message);  
  36.     nlh = nlmsg_put(skb_1,0,0,0,MAX_MSGSIZE,0);  
  37.   
  38.     NETLINK_CB(skb_1).pid = 0;  
  39.     NETLINK_CB(skb_1).dst_group = 0;  
  40.   
  41.     message[slen]= '\0';  
  42.     memcpy(NLMSG_DATA(nlh),message,slen+1);  
  43.     printk("my_net_link:send message '%s'.\n",(char *)NLMSG_DATA(nlh));  
  44.   
  45.     netlink_unicast(nl_sk,skb_1,pid,MSG_DONTWAIT);  
  46. }  
  47.   
  48. int stringlength(char *s)  
  49. {  
  50.     int slen = 0;  
  51.   
  52.     for(; *s; s++){  
  53.         slen++;  
  54.     }  
  55.   
  56.     return slen;  
  57. }  
  58.   
  59. void nl_data_ready(struct sk_buff *__skb)  
  60. {  
  61.     struct sk_buff *skb;  
  62.     struct nlmsghdr *nlh;  
  63.     char str[100];  
  64.     struct completion cmpl;  
  65.     int i=10;  
  66.     skb = skb_get (__skb);  
  67.       
  68.     if(skb->len >= NLMSG_SPACE(0))  
  69.     {  
  70.         nlh = nlmsg_hdr(skb);  
  71.   
  72.         memcpy(str, NLMSG_DATA(nlh), sizeof(str));  
  73.         printk("Message received:%s\n",str) ;  
  74.         pid = nlh->nlmsg_pid;  
  75.         while(i--)  
  76.         {  
  77.             init_completion(&cmpl);  
  78.             wait_for_completion_timeout(&cmpl,3 * HZ);  
  79.             sendnlmsg("I am from kernel!");  
  80.         }  
  81.         flag = 1;  
  82.         kfree_skb(skb);  
  83.     }  
  84. }  
  85.   
  86. // Initialize netlink  
  87.   
  88. int netlink_init(void)  
  89. {  
  90.   
  91.   
  92.     nl_sk = netlink_kernel_create(&init_net, NETLINK_TEST, 1,  
  93.                                  nl_data_ready, NULL, THIS_MODULE);  
  94.   
  95.     if(!nl_sk){  
  96.         printk(KERN_ERR "my_net_link: create netlink socket error.\n");  
  97.         return 1;  
  98.     }  
  99.   
  100.     printk("my_net_link_3: create netlink socket ok.\n");  
  101.   
  102.   
  103.     return 0;  
  104. }  
  105.   
  106. static void netlink_exit(void)  
  107. {  
  108.     if(nl_sk != NULL){  
  109.         netlink_kernel_release(nl_sk);  
  110.     }  
  111.   
  112.     printk("my_net_link: self module exited\n");  
  113. }  
  114.   
  115. module_init(netlink_init);  
  116. module_exit(netlink_exit);  
  117.   
  118. MODULE_AUTHOR("frankzfz");  
  119. MODULE_LICENSE("GPL");  



部分内容摘自:http://www.cnblogs.com/iceocean/articles/1594195.html

0 0
原创粉丝点击