zoukankan      html  css  js  c++  java
  • linux用户空间与内核空间通信——Netlink通信机制【转】

    转自:https://blog.csdn.net/zhao_h/article/details/80943226

    一:什么是Netlink通信机制

    Netlink是linux提供的用于内核和用户态进程之间的通信方式。

    但是注意虽然Netlink主要用于用户空间和内核空间的通信,但是也能用于用户空间的两个进程通信。只是进程间通信有其他很多

    方式,一般不用Netlink。除非需要用到Netlink的广播特性时。

    那么Netlink有什么优势呢?

    一般来说用户空间和内核空间的通信方式有三种:/proc、ioctl、Netlink。而前两种都是单向的,但是Netlink可以实现双工通信。

    Netlink协议基于BSD socket和AF_NETLINK地址簇(address family),使用32位的端口号寻址(以前称作PID),每个Netlink协议

    (或称作总线,man手册中则称之为netlink family),通常与一个或一组内核服务/组件相关联,如NETLINK_ROUTE用于获取和

    设置路由与链路信息、NETLINK_KOBJECT_UEVENT用于内核向用户空间的udev进程发送通知等。

    netlink具有以下特点:

    ① 支持全双工、异步通信(当然同步也支持)

    ② 用户空间可使用标准的BSD socket接口(但netlink并没有屏蔽掉协议包的构造与解析过程,推荐使用libnl等第三方库)

    ③ 在内核空间使用专用的内核API接口

    ④ 支持多播(因此支持“总线”式通信,可实现消息订阅)

    ⑤ 在内核端可用于进程上下文与中断上下文

    二:用户态数据结构

    首先看一下几个重要的数据结构的关系:

    1.struct msghdr

    msghdr这个结构在socket变成中就会用到,并不算Netlink专有的,这里不在过多说明。只说明一下如何更好理解这个结构的功能。我们

    知道socket消息的发送和接收函数一般有这几对:recv/send、readv/writev、recvfrom/sendto。当然还有recvmsg/sendmsg,

    前面三对函数各有各的特点功能,而recvmsg/sendmsg就是要囊括前面三对的所有功能,当然还有自己特殊的用途。msghdr的前两个

    成员就是为了满足recvfrom/sendto的功能,中间两个成员msg_iov和msg_iovlen则是为了满足readv/writev的功能,而最后的

    msg_flags则是为了满足recv/send中flag的功能,剩下的msg_control和msg_controllen则是满足recvmsg/sendmsg特有的功能。

    2.struct sockaddr_ln

    struct sockaddr_ln为Netlink的地址,和我们通常socket编程中的sockaddr_in作用一样,他们的结构对比如下:

    struct sockaddr_nl的详细定义和描述如下:

    1.  
      struct sockaddr_nl
    2.  
      {
    3.  
      sa_family_t nl_family; /*该字段总是为AF_NETLINK */
    4.  
      unsigned short nl_pad; /* 目前未用到,填充为0*/
    5.  
      __u32 nl_pid; /* process pid */
    6.  
      __u32 nl_groups; /* multicast groups mask */
    7.  
      };

    (1) nl_pid:在Netlink规范里,PID全称是Port-ID(32bits),其主要作用是用于唯一的标识一个基于netlink的socket通道。通常

    情况下nl_pid都设置为当前进程的进程号。前面我们也说过,Netlink不仅可以实现用户-内核空间的通信还可使现实用户空间两

    个进程之间,或内核空间两个进程之间的通信。该属性为0时一般指内核。

    (2) nl_groups:如果用户空间的进程希望加入某个多播组,则必须执行bind()系统调用。该字段指明了调用者希望加入的多播组

    号的掩码(注意不是组号,后面我们会详细讲解这个字段)。如果该字段为0则表示调用者不希望加入任何多播组。对于每个隶属于

    Netlink协议域的协议,最多可支持32个多播组(因为nl_groups的长度为32比特),每个多播组用一个比特来表示。

    3.struct nlmsghdr

    Netlink的报文由消息头和消息体构成,struct nlmsghdr即为消息头。消息头定义在文件里,由结构体nlmsghdr表示:

    1.  
      struct nlmsghdr
    2.  
      {
    3.  
      __u32 nlmsg_len; /* Length of message including header */
    4.  
      __u16 nlmsg_type; /* Message content */
    5.  
      __u16 nlmsg_flags; /* Additional flags */
    6.  
      __u32 nlmsg_seq; /* Sequence number */
    7.  
      __u32 nlmsg_pid; /* Sending process PID */
    8.  
      };

    消息头中各成员属性的解释及说明:

    (1) nlmsg_len:整个消息的长度,按字节计算。包括了Netlink消息头本身。

    (2) nlmsg_type:消息的类型,即是数据还是控制消息。目前(内核版本2.6.21)Netlink仅支持四种类型的控制消息,如下:

    a) NLMSG_NOOP-空消息,什么也不做;

    b) NLMSG_ERROR-指明该消息中包含一个错误;

    c) NLMSG_DONE-如果内核通过Netlink队列返回了多个消息,那么队列的最后一条消息的类型为NLMSG_DONE,其余所有消息的nlmsg_flags属性都被设置NLM_F_MULTI位有效。

    d) NLMSG_OVERRUN-暂时没用到。

    (3) nlmsg_flags:附加在消息上的额外说明信息,如上面提到的NLM_F_MULTI。

    三:用户空间Netlink socket API

    1.创建socket

    int socket(int domain, int type, int protocol)

    domain指代地址族,即AF_NETLINK;

    套接字类型为SOCK_RAW或SOCK_DGRAM,因为netlink是一个面向数据报的服务;

    protocol选择该套接字使用哪种netlink特征。

    以下是几种预定义的协议类型:

    NETLINK_ROUTE,

    NETLINK_FIREWALL,

    NETLINK_APRD,

    NETLINK_ROUTE6_FW。

    可以非常容易的添加自己的netlink协议。

    为每一个协议类型最多可以定义32个多播组。

    每一个多播组用一个bitmask来表示,1<<i(0<=i<= 31),这在一组进程和内核进程协同完成一项任务时非常有用。发送多播

    netlink消息可以减少系统调用的数量,同时减少用来维护多播组成员信息的负担。

    2.地址绑定bind()

    bind(fd, (struct sockaddr*)&, nladdr, sizeof(nladdr));

    3.发送netlink消息

    为了发送一条netlink消息到内核或者其他的用户空间进程,另外一个struct sockaddr_nl nladdr需要作为目的地址,这和使用

    sendmsg()发送一个UDP包是一样的。

    如果该消息是发送至内核的,那么nl_pid和nl_groups都置为0.

    如果消息是发送给另一个进程的单播消息,nl_pid是另外一个进程的pid值而nl_groups为零。

    如果消息是发送给一个或多个多播组的多播消息,所有的目的多播组必须bitmask必须or起来从而形成nl_groups域。

    sendmsg(fd, &, msg, 0);

    4.接收netlink消息

    一个接收程序必须分配一个足够大的内存用于保存netlink消息头和消息负载。然后其填充struct msghdr msg,再使用标准的

    recvmsg()函数来接收netlink消息。

    当消息被正确的接收之后,nlh应该指向刚刚接收到的netlink消息的头。nladdr应该包含接收消息的目的地址,其中包括了消息发送

    者的pid和多播组。同时,宏NLMSG_DATA(nlh),定义在netlink.h中,返回一个指向netlink消息负载的指针。调用close(fd)关闭fd

    描述符所标识的socket。

    recvmsg(fd, &, msg, 0);

    四:内核空间Netlink socket API

    1.创建 netlink socket

    1.  
      struct sock *netlink_kernel_create(struct net *net,
    2.  
           int unit,unsigned int groups,
    3.  
           void (*input)(struct sk_buff *skb),
    4.  
          struct mutex *cb_mutex,struct module *module);

    参数说明:

    (1) net:是一个网络名字空间namespace,在不同的名字空间里面可以有自己的转发信息库,有自己的一套net_device等等。

    默认情况下都是使用 init_net这个全局变量。

    (2) unit:表示netlink协议类型,如NETLINK_TEST、NETLINK_SELINUX。

    (3) groups:多播地址。

    (4) input:为内核模块定义的netlink消息处理函数,当有消 息到达这个netlink socket时,该input函数指针就会被引用,且只

    有此函数返回时,调用者的sendmsg才能返回。

    (5)  cb_mutex:为访问数据时的互斥信号量。

    (6) module: 一般为THIS_MODULE。

    2.发送单播消息 netlink_unicast

    int netlink_unicast(struct sock *ssk, struct sk_buff *skb, u32 pid, int nonblock)

    参数说明:

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

    (2) skb:存放消息,它的data字段指向要发送的netlink消息结构,而 skb的控制块保存了消息的地址信息,宏

    NETLINK_CB(skb)就用于方便设置该控制块。

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

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

    收缓存可利用定时睡眠。

    3.发送广播消息 netlink_broadcast

    1.  
      int netlink_broadcast(struct sock *ssk, struct sk_buff *skb, u32 pid, u32 group, gfp_t allocation)
    2.  
       

    前面的三个参数与 netlink_unicast相同,参数group为接收消息的多播组,该参数的每一个位代表一个多播组,因此如果发送给

    多个多播组,就把该参数设置为多个多播组组ID的位或。参数allocation为内核内存分配类型,一般地为GFP_ATOMIC或

    GFP_KERNEL,GFP_ATOMIC用于原子的上下文(即不可以睡眠),而GFP_KERNEL用于非原子上下文。

    4.释放 netlink socket

    int netlink_broadcast(struct sock *ssk, struct sk_buff *skb, u32 pid, u32 group, gfp_t allocation)

    五:用户态范例一

    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.  
      #define MAX_PAYLOAD 1024 // maximum payload size
    13.  
      #define NETLINK_TEST 25 //自定义的协议
    14.  
      int main(int argc, char* argv[])
    15.  
      {
    16.  
      int state;
    17.  
      struct sockaddr_nl src_addr, dest_addr;
    18.  
      struct nlmsghdr *nlh = NULL; //Netlink数据包头
    19.  
      struct iovec iov;
    20.  
      struct msghdr msg;
    21.  
      int sock_fd, retval;
    22.  
      int state_smg = 0;
    23.  
      // Create a socket
    24.  
      sock_fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_TEST);
    25.  
      if(sock_fd == -1){
    26.  
      printf("error getting socket: %s", strerror(errno));
    27.  
      return -1;
    28.  
      }
    29.  
      // To prepare binding
    30.  
      memset(&src_addr, 0, sizeof(src_addr));
    31.  
      src_addr.nl_family = AF_NETLINK;
    32.  
      src_addr.nl_pid = 100; //A:设置源端端口号
    33.  
      src_addr.nl_groups = 0;
    34.  
      //Bind
    35.  
      retval = bind(sock_fd, (struct sockaddr*)&src_addr, sizeof(src_addr));
    36.  
      if(retval < 0){
    37.  
      printf("bind failed: %s", strerror(errno));
    38.  
      close(sock_fd);
    39.  
      return -1;
    40.  
      }
    41.  
      // To orepare create mssage
    42.  
      nlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_PAYLOAD));
    43.  
      if(!nlh){
    44.  
      printf("malloc nlmsghdr error! ");
    45.  
      close(sock_fd);
    46.  
      return -1;
    47.  
      }
    48.  
      memset(&dest_addr,0,sizeof(dest_addr));
    49.  
      dest_addr.nl_family = AF_NETLINK;
    50.  
      dest_addr.nl_pid = 0; //B:设置目的端口号
    51.  
      dest_addr.nl_groups = 0;
    52.  
      nlh->nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);
    53.  
      nlh->nlmsg_pid = 100; //C:设置源端口
    54.  
      nlh->nlmsg_flags = 0;
    55.  
      strcpy(NLMSG_DATA(nlh),"Hello you!"); //设置消息体
    56.  
      iov.iov_base = (void *)nlh;
    57.  
      iov.iov_len = NLMSG_SPACE(MAX_PAYLOAD);
    58.  
      //Create mssage
    59.  
      memset(&msg, 0, sizeof(msg));
    60.  
      msg.msg_name = (void *)&dest_addr;
    61.  
      msg.msg_namelen = sizeof(dest_addr);
    62.  
      msg.msg_iov = &iov;
    63.  
      msg.msg_iovlen = 1;
    64.  
      //send message
    65.  
      printf("state_smg ");
    66.  
      state_smg = sendmsg(sock_fd,&msg,0);
    67.  
      if(state_smg == -1)
    68.  
      {
    69.  
      printf("get error sendmsg = %s ",strerror(errno));
    70.  
      }
    71.  
      memset(nlh,0,NLMSG_SPACE(MAX_PAYLOAD));
    72.  
      //receive message
    73.  
      printf("waiting received! ");
    74.  
      while(1){
    75.  
      printf("In while recvmsg ");
    76.  
      state = recvmsg(sock_fd, &msg, 0);
    77.  
      if(state<0)
    78.  
      {
    79.  
      printf("state<1");
    80.  
      }
    81.  
      printf("Received message: %s ",(char *) NLMSG_DATA(nlh));
    82.  
      }
    83.  
      close(sock_fd);
    84.  
      return 0;
    85.  
      }

    上面程序首先向内核发送一条消息;“Hello you”,然后进入循环一直等待读取内核的回复,并将收到的回复打印出来。如果

    看上面程序感觉很吃力,那么应该首先复习一下UDP中使用sendmsg的用法,特别时struct msghdr的结构要清楚,这里再赘

    述。下面主要分析与UDP发送数据包的不同点:

    1. socket地址结构不同,UDP为sockaddr_in,Netlink为struct sockaddr_nl;

    2. 与UDP发送数据相比,Netlink多了一个消息头结构struct nlmsghdr需要我们构造。

    注意代码注释中的A、B、C三处分别设置了pid。首先解释一下什么是pid,网上很多文章把这个字段说成是进程的pid,其实这

    完全是望文生义。这里的pid和进程pid没有什么关系,仅仅相当于UDP的port。对于UDP来说port和ip标示一个地址,那对我们

    的NETLINK_TEST协议(注意Netlink本身不是一个协议)来说,pid就唯一标示了一个地址。所以你如果用进程pid做为标示当然

    也是可以的。当然同样的pid对于NETLINK_TEST协议和内核定义的其他使用Netlink的协议是不冲突的(就像TCP的80端口和

    UDP的80端口)。

    下面分析这三处设置pid分别有什么作用,首先A和B位置的比较好理解,这是在地址(sockaddr_nl)上进行的设置,就是相当

    于设置源地址和目的地址(其实是端口),只是注意B处设置pid为0,0就代表是内核,可以理解为内核专用的pid,那么用户进

    程就不能用0做为自己的pid吗?这个只能说如果你非要用也是可以的,只是会产生一些问题,后面在分析。接下来看为什么C处

    的消息头仍然需要设置pid呢?这里首先要知道一个前提:内核不会像UDP一样根据我们设置的原、目的地址为我们构造消息

    头,所以我们不在包头写入我们自己的地址(pid),那内核怎么知道是谁发来的报文呢?当然如果内核只是处理消息不需要回

    复进程的话舍不设置这个消息头pid都可以。

    所以每个pid的设置功能不同:A处的设置是要设置发送者的源地址,有人会说既然源地址又不会自动填充到报文中,我们为什么

    还要设置这个,因为你还可能要接收回复啊。就像寄信,你连“门牌号”都没有,即使你在写信时候写上你的地址是100号,对

    方回信目的地址也是100号,但是邮局发现根本没有这个地址怎么可能把信送到你手里呢?所以A的主要作用是注册源地址,保证

    可以收到回复,如果不需要回复当然可以简单将pid设置为0;B处自然就是收信人的地址,pid为0代表内核的地址,假如有一个

    进程在101号上注册了地址,并调用了recvmsg,如果你将B处的pid设置为101,那数据包就发给了另一个进程,这就实现了使

    用Netlink进行进程间通信;C相当于你在信封上写的源地址,通常情况下这个应该和你的真实地址(A)处注册的源地址相同,

    当然你要是不想收到回信,又想恶搞一下或者有特殊需求,你可以写成其他进程注册的pid(比如101)。这和我们现实中寄信是

    一样的,你给你朋友写封情书,把写信人写成你的另一个好基友,然后后果你懂得……

    好了,有了这个例子我们就大概知道用户态怎么使用Netlink了。

    六:内核态程序范例一

    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.  
      #define NETLINK_TEST 25
    9.  
      #define MAX_MSGSIZE 1024
    10.  
      int stringlength(char *s);
    11.  
      int err;
    12.  
      struct sock *nl_sk = NULL;
    13.  
      int flag = 0;
    14.  
      //向用户态进程回发消息
    15.  
      void sendnlmsg(char *message, int pid)
    16.  
      {
    17.  
      struct sk_buff *skb_1;
    18.  
      struct nlmsghdr *nlh;
    19.  
      int len = NLMSG_SPACE(MAX_MSGSIZE);
    20.  
      int slen = 0;
    21.  
      if(!message || !nl_sk)
    22.  
      {
    23.  
      return ;
    24.  
      }
    25.  
      printk(KERN_ERR "pid:%d ",pid);
    26.  
      skb_1 = alloc_skb(len,GFP_KERNEL);
    27.  
      if(!skb_1)
    28.  
      {
    29.  
      printk(KERN_ERR "my_net_link:alloc_skb error ");
    30.  
      }
    31.  
      slen = stringlength(message);
    32.  
      nlh = nlmsg_put(skb_1,0,0,0,MAX_MSGSIZE,0);
    33.  
      NETLINK_CB(skb_1).pid = 0;
    34.  
      NETLINK_CB(skb_1).dst_group = 0;
    35.  
      message[slen]= '';
    36.  
      memcpy(NLMSG_DATA(nlh),message,slen+1);
    37.  
      printk("my_net_link:send message '%s'. ",(char *)NLMSG_DATA(nlh));
    38.  
      netlink_unicast(nl_sk,skb_1,pid,MSG_DONTWAIT);
    39.  
      }
    40.  
      int stringlength(char *s)
    41.  
      {
    42.  
      int slen = 0;
    43.  
      for(; *s; s++)
    44.  
      {
    45.  
      slen++;
    46.  
      }
    47.  
      return slen;
    48.  
      }
    49.  
      //接收用户态发来的消息
    50.  
      void nl_data_ready(struct sk_buff *__skb)
    51.  
      {
    52.  
      struct sk_buff *skb;
    53.  
      struct nlmsghdr *nlh;
    54.  
      char str[100];
    55.  
      struct completion cmpl;
    56.  
      printk("begin data_ready ");
    57.  
      int i=10;
    58.  
      int pid;
    59.  
      skb = skb_get (__skb);
    60.  
      if(skb->len >= NLMSG_SPACE(0))
    61.  
      {
    62.  
      nlh = nlmsg_hdr(skb);
    63.  
      memcpy(str, NLMSG_DATA(nlh), sizeof(str));
    64.  
      printk("Message received:%s ",str) ;
    65.  
      pid = nlh->nlmsg_pid;
    66.  
      while(i--)
    67.  
      {//我们使用completion做延时,每3秒钟向用户态回发一个消息
    68.  
      init_completion(&cmpl);
    69.  
      wait_for_completion_timeout(&cmpl,3 * HZ);
    70.  
      sendnlmsg("I am from kernel!",pid);
    71.  
      }
    72.  
      flag = 1;
    73.  
      kfree_skb(skb);
    74.  
      }
    75.  
      }
    76.  
      // Initialize netlink
    77.  
      int netlink_init(void)
    78.  
      {
    79.  
      nl_sk = netlink_kernel_create(&init_net, NETLINK_TEST, 1,
    80.  
      nl_data_ready, NULL, THIS_MODULE);
    81.  
      if(!nl_sk){
    82.  
      printk(KERN_ERR "my_net_link: create netlink socket error. ");
    83.  
      return 1;
    84.  
      }
    85.  
      printk("my_net_link_4: create netlink socket ok. ");
    86.  
      return 0;
    87.  
      }
    88.  
      static void netlink_exit(void)
    89.  
      {
    90.  
      if(nl_sk != NULL){
    91.  
      sock_release(nl_sk->sk_socket);
    92.  
      }
    93.  
      printk("my_net_link: self module exited ");
    94.  
      }
    95.  
      module_init(netlink_init);
    96.  
      module_exit(netlink_exit);
    97.  
      MODULE_AUTHOR("zhao_h");
    98.  
      MODULE_LICENSE("GPL");

    附上内核代码的Makefile文件:

    1.  
      ifneq ($(KERNELRELEASE),)
    2.  
      obj-m :=netl.o
    3.  
      else
    4.  
      KERNELDIR ?=/lib/modules/$(shell uname -r)/build
    5.  
      PWD :=$(shell pwd)
    6.  
      default:
    7.  
      $(MAKE) -C $(KERNELDIR) M=$(PWD) modules
    8.  
      endif

    我们将内核模块insmod后,运行用户态程序,结果如下:

    这个结果复合我们的预期,但是运行过程中打印出“state_smg”卡了好久才输出了后面的结果。这时候查看客户进程是处于D

    状态的(不了解D状态的同学可以google一下)。这是为什么呢?因为进程使用Netlink向内核发数据是同步,内核向进程发数

    据是异步。什么意思呢?也就是用户进程调用sendmsg发送消息后,内核会调用相应的接收函数,但是一定到这个接收函数执行

    完用户态的sendmsg才能够返回。我们在内核态的接收函数中调用了10次回发函数,每次都等待3秒钟,所以内核接收函数30秒

    后才返回,所以我们用户态程序的sendmsg也要等30秒后才返回。相反,内核回发的数据不用等待用户程序接收,这是因为内核

    所发的数据会暂时存放在一个队列中。

    再来回到之前的一个问题,用户态程序的源地址(pid)可以用0吗?我把上面的用户程序的A和C处pid都改为了0,结果一运行

    就死机了。为什么呢?我们看一下内核代码的逻辑,收到用户消息后,根据消息中的pid发送回去,而pid为0,内核并不认为这

    是用户程序,认为是自身,所有又将回发的10个消息发给了自己(内核),这样就陷入了一个死循环,而用户态这时候进程一直

    处于D。

    另外一个问题,如果同时启动两个用户进程会是什么情况?答案是再调用bind时出错:“Address already in use”,这个同

    UDP一样,同一个地址同一个port如果没有设置SO_REUSEADDR两次bind就会出错,之后我用同样的方式再Netlink的socket

    上设置了SO_REUSEADDR,但是并没有什么效果。

    七:用户态范例二

    之前我们说过UDP可以使用sendmsg/recvmsg也可以使用sendto/recvfrom,那么Netlink同样也可以使用sendto/

    recvfrom。具体实现如下:

    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.  
      #define MAX_PAYLOAD 1024 // maximum payload size
    13.  
      #define NETLINK_TEST 25
    14.  
      int main(int argc, char* argv[])
    15.  
      {
    16.  
      struct sockaddr_nl src_addr, dest_addr;
    17.  
      struct nlmsghdr *nlh = NULL;
    18.  
      int sock_fd, retval;
    19.  
      int state,state_smg = 0;
    20.  
      // Create a socket
    21.  
      sock_fd = socket(AF_NETLINK, SOCK_RAW, NETLINK_TEST);
    22.  
      if(sock_fd == -1){
    23.  
      printf("error getting socket: %s", strerror(errno));
    24.  
      return -1;
    25.  
      }
    26.  
      // To prepare binding
    27.  
      memset(&src_addr, 0, sizeof(src_addr));
    28.  
      src_addr.nl_family = AF_NETLINK;
    29.  
      src_addr.nl_pid = 100;
    30.  
      src_addr.nl_groups = 0;
    31.  
      //Bind
    32.  
      retval = bind(sock_fd, (struct sockaddr*)&src_addr, sizeof(src_addr));
    33.  
      if(retval < 0){
    34.  
      printf("bind failed: %s", strerror(errno));
    35.  
      close(sock_fd);
    36.  
      return -1;
    37.  
      }
    38.  
      // To orepare create mssage head
    39.  
      nlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_PAYLOAD));
    40.  
      if(!nlh){
    41.  
      printf("malloc nlmsghdr error! ");
    42.  
      close(sock_fd);
    43.  
      return -1;
    44.  
      }
    45.  
      memset(&dest_addr,0,sizeof(dest_addr));
    46.  
      dest_addr.nl_family = AF_NETLINK;
    47.  
      dest_addr.nl_pid = 0;
    48.  
      dest_addr.nl_groups = 0;
    49.  
      nlh->nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);
    50.  
      nlh->nlmsg_pid = 100;
    51.  
      nlh->nlmsg_flags = 0;
    52.  
      strcpy(NLMSG_DATA(nlh),"Hello you!");
    53.  
      //send message
    54.  
      printf("state_smg ");
    55.  
      sendto(sock_fd,nlh,NLMSG_LENGTH(MAX_PAYLOAD),0,(struct sockaddr*)(&dest_addr),sizeof(dest_addr));
    56.  
      if(state_smg == -1)
    57.  
      {
    58.  
      printf("get error sendmsg = %s ",strerror(errno));
    59.  
      }
    60.  
      memset(nlh,0,NLMSG_SPACE(MAX_PAYLOAD));
    61.  
      //receive message
    62.  
      printf("waiting received! ");
    63.  
      while(1){
    64.  
      printf("In while recvmsg ");
    65.  
      state=recvfrom(sock_fd,nlh,NLMSG_LENGTH(MAX_PAYLOAD),0,NULL,NULL);
    66.  
      if(state<0)
    67.  
      {
    68.  
      printf("state<1");
    69.  
      }
    70.  
      printf("Received message: %s ",(char *) NLMSG_DATA(nlh));
    71.  
      memset(nlh,0,NLMSG_SPACE(MAX_PAYLOAD));
    72.  
      }
    73.  
      close(sock_fd);
    74.  
      return 0;
    75.  
      }

    熟悉UDP编程的同学看到这个程序一定很熟悉,除了多了一个Netlink消息头的设置。但是我们发现程序中调用了bind函数,这

    个函数再UDP编程中的客户端不是必须的,因为我们不需要把UDP socket与某个地址关联,同时再发送UDP数据包时内核会为

    我们分配一个随即的端口。但是对于Netlink必须要有这一步bind,因为Netlink内核可不会为我们分配一个pid。再强调一遍消

    息头(nlmsghdr)中的pid是告诉内核接收端要回复的地址,但是这个地址存不存在内核并不关心,这个地址只有用户端调用了

    bind后才存在。

    我们看到这两个例子都是用户态首先发起的,那Netlink是否支持内核态主动发起的情况呢?

    当然是可以的,只是内核一般需要事件触发,这里,只要和用户态约定号一个地址(pid),内核直接调用netlink_unicast就可以了。

  • 相关阅读:
    数论分块与整除相
    P3254——DP&&入门
    P3384——树链剖分&&模板
    树链剖分模板
    P4145——线段树点修改&&模板题
    P1198最大数——线段树点修改&&模板题
    线段树(四)——两个标记(add和set)
    epoll 知识总结
    C++中的mutable关键字
    [LeetCode] Max Points on a Line
  • 原文地址:https://www.cnblogs.com/sky-heaven/p/14042082.html
Copyright © 2011-2022 走看看