Netlink Socket

原文出處:http://blog.csdn.net/accp_2008123456/article/details/5818194

 

簡介
Netlink
是一種特殊的 socket,它是 Linux 所特有的,類似於 BSD 中的AF_ROUTE 但又遠比它的功能強大,目前在最新的 Linux 內核(2.6.14)中使用netlink 進行應用與內核通信的應用很多,包括:

路由 daemonNETLINK_ROUTE

1-wire 子系統(NETLINK_W1

用戶態 socket 協議(NETLINK_USERSOCK

防火牆(NETLINK_FIREWALL

socket 監視(NETLINK_INET_DIAG

netfilter 日誌(NETLINK_NFLOG

ipsec 安全策略(NETLINK_XFRM

SELinux 事件通知(NETLINK_SELINUX

iSCSI 子系統(NETLINK_ISCSI

進程審計(NETLINK_AUDIT

轉發信息表查詢(NETLINK_FIB_LOOKUP

netlink connector(NETLINK_CONNECTOR)

netfilter 子系統(NETLINK_NETFILTER

IPv6 防火牆(NETLINK_IP6_FW

DECnet 路由信息(NETLINK_DNRTMSG

內核事件向用戶態通知(NETLINK_KOBJECT_UEVENT

通用 netlinkNETLINK_GENERIC)。

Netlink
是一種在內核與用戶應用間進行雙向數據傳輸的非常好的方式,用戶態應用使用標準的 socket API 就可以使用 netlink 提供的強大功能,內核態需要使用專門的內核 API 來使用 netlink

Netlink
相對於系統調用,ioctl 以及 /proc 文件系統而言具有以下優點:

1.
爲了使用 netlink,用戶僅需要在 include/linux/netlink.h 中增加一個新類型的 netlink 協議定義即可, #define NETLINK_MYTEST 17 然後,內核和用戶態應用就可以立即通過 socket API 使用該 netlink 協議類型進行數據交換。但系統調用需要增加新的系統調用,ioctl 則需要增加設備或文件,那需要不少代碼,proc 文件系統則需要在 /proc 下添加新的文件或目錄,那將使本來就混亂的 /proc 更加混亂。

2. Netlink
是一種異步通信機制,在內核與用戶態應用之間傳遞的消息保存在socket緩存隊列中,發送消息只是把消息保存在接收者的socket的接收隊列,而不需要等待接收者收到消息,但系統調用與 ioctl 則是同步通信機制,如果傳遞的數據太長,將影響調度粒度。

3.
使用 netlink 的內核部分可以採用模塊的方式實現,使用 netlink 的應用部分和內核部分沒有編譯時依賴,但系統調用就有依賴,而且新的系統調用的實現必須靜態地連接到內核中,它無法在模塊中實現,使用新系統調用的應用在編譯時需要依賴內核。

4. Netlink
支持多播,內核模塊或應用可以把消息多播給一個netlink組,屬於該neilink 組的任何內核模塊或應用都能接收到該消息,內核事件向用戶態的通知機制就使用了這一特性,任何對內核事件感興趣的應用都能收到該子系統發送的內核事件,在後面的文章中將介紹這一機制的使用。

5.
內核可以使用 netlink 首先發起會話,但系統調用和 ioctl 只能由用戶應用發起調用。

6. Netlink
使用標準的 socket API,因此很容易使用,但系統調用和 ioctl則需要專門的培訓才能使用。


用戶態使用 netlink

用戶態應用使用標準的socket APIs, socket(), bind(), sendmsg(), recvmsg() close() 就能很容易地使用 netlink socket,查詢手冊頁可以瞭解這些函數的使用細節,本文只是講解使用 netlink 的用戶應該如何使用這些函數。注意,使用 netlink 的應用必須包含頭文件 linux/netlink.h。當然 socket 需要的頭文件也必不可少,sys/socket.h

爲了創建一個 netlink socket,用戶需要使用如下參數調用 socket():

Cpp代碼

1.#define NETLINK_ROUTE 0 /* Routing/device hook */

2.#define NETLINK_W1 1 /* 1-wire subsystem */

3.#define NETLINK_USERSOCK 2 /* Reserved for user mode socket protocols */

4.#define NETLINK_FIREWALL 3 /* Firewalling hook */

5.#define NETLINK_INET_DIAG 4 /* INET socket monitoring */

6.#define NETLINK_NFLOG 5 /* netfilter/iptables ULOG */

7.#define NETLINK_XFRM 6 /* ipsec */

8.#define NETLINK_SELINUX 7 /* SELinux event notifications */

9.#define NETLINK_ISCSI 8 /* Open-iSCSI */

10.  #define NETLINK_AUDIT 9 /* auditing */

11.  #define NETLINK_FIB_LOOKUP 10

12.  #define NETLINK_CONNECTOR 11

13.  #define NETLINK_NETFILTER 12 /* netfilter subsystem */

14.  #define NETLINK_IP6_FW 13

15.  #define NETLINK_DNRTMSG 14 /* DECnet routing messages */

16.  #define NETLINK_KOBJECT_UEVENT 15 /* Kernel messages to userspace */

17.  #define NETLINK_GENERIC 16


對於每一個netlink協議類型,可以有多達 32多播組,每一個多播組用一個位表示,netlink 的多播特性使得發送消息給同一個組僅需要一次系統調用,因而對於需要多撥消息的應用而言,大大地降低了系統調用的次數。

函數 bind() 用於把一個打開的 netlink socket netlink socket 地址綁定在一起。netlink socket 的地址結構如下:

Cpp代碼

1.pthread_self() << 16 | getpid();

 

因此字段 nl_pid 實際上未必是進程 ID,它只是用於區分不同的接收者或發送者的一個標識,用戶可以根據自己需要設置該字段。函數 bind 的調用方式如下:

Cpp代碼

1.struct msghdr msg;

2.memset(&msg, 0, sizeof(msg));

3.msg.msg_name = (void *)&(nladdr);

4.msg.msg_namelen = sizeof(nladdr);

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

struct nlmsghdr
netlink socket 自己的消息頭,這用於多路複用和多路分解 netlink 定義的所有協議類型以及其它一些控制,netlink 的內核實現將利用這個消息頭來多路複用和多路分解已經其它的一些控制,因此它也被稱爲netlink 控制塊。因此,應用在發送 netlink 消息時必須提供該消息頭。

Cpp代碼

1./* Flags values */

2.#define NLM_F_REQUEST 1 /* It is request message */

3.#define NLM_F_MULTI 2 /* Multipart message, terminated by NLMSG_DONE */

4.#define NLM_F_ACK 4 /* Reply with ack, with zero or error code */

5.#define NLM_F_ECHO 8 /* Echo this request */

6. 

7./* Modifiers to GET request */

8.#define NLM_F_ROOT 0x100 /* specify tree root */

9.#define NLM_F_MATCH 0x200 /* return all matching */

10.  #define NLM_F_ATOMIC 0x400 /* atomic GET */

11.  #define NLM_F_DUMP (NLM_F_ROOT|NLM_F_MATCH)

12.   

13.  /* Modifiers to NEW request */

14.  #define NLM_F_REPLACE 0x100 /* Override existing */

15.  #define NLM_F_EXCL 0x200 /* Do not touch, if it exists */

16.  #define NLM_F_CREATE 0x400 /* Create, if it does not exist */

17.  #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 需要它進行一些複雜的操作),字段 nlmsg_seq nlmsg_pid 用於應用追蹤消息,前者表示順序號,後者爲消息來源進程 ID。下面是一個示例:

Cpp代碼

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代碼

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代碼

1.#define NLMSG_LENGTH(len) ((len)+NLMSG_ALIGN(sizeof(struct nlmsghdr)))

NLMSG_LENGTH(len)用於計算數據部分長度爲len時實際的消息長度。它一般用於分配消息緩存。

Cpp代碼

1.#define NLMSG_DATA(nlh) ((void*)(((char*)nlh) + NLMSG_LENGTH(0)))

NLMSG_DATA(nlh)用於取得消息的數據部分的首地址,設置和讀取消息數據部分時需要使用該宏。

Cpp代碼

1.#define NLMSG_OK(nlh,len) ((len) >= (int)sizeof(struct nlmsghdr) && (nlh)->nlmsg_len >= sizeof(struct nlmsghdr) && (nlh)->nlmsg_len <= (len))

NLMSG_OK(nlh,len)用於判斷消息是否有len這麼長。

Cpp代碼

1.#define NETLINK_MYTEST 17


只要增加這個定義之後,用戶就可以在內核的任何地方引用該協議。

在內核中,爲了創建一個netlink socket用戶需要調用如下函數:

Cpp代碼

1.void input (struct sock *sk, int len)

2.{

3.struct sk_buff *skb;

4.struct nlmsghdr *nlh = NULL;

5.u8 *data = NULL;

6.while ((skb = skb_dequeue(&sk->receive_queue)) != NULL)

7.{

8./* process netlink message pointed by skb->data */

9.nlh = (struct nlmsghdr *)skb->data;

10.  data = NLMSG_DATA(nlh);

11.  /* process netlink message with header pointed by

12.  * nlh and data pointed by data

13.  */

14.  }

15.  }


函數input()會在發送進程執行sendmsg()時被調用,這樣處理消息比較及時,但是,如果消息特別長時,這樣處理將增加系統調用sendmsg()的執行時間,對於這種情況,可以定義一個內核線程專門負責消息接收,而函數input的工作只是喚醒該內核線程,這樣sendmsg將很快返回。

函數skb = skb_dequeue(&sk->receive_queue)用於取得socket sk的接收隊列上的消息,返回爲一個struct sk_buff的結構,skb->data指向實際的netlink消息。

函數skb_recv_datagram(nl_sk)也用於在netlink socket nl_sk上接收消息,與skb_dequeue的不同指出是,如果socket的接收隊列上沒有消息,它將導致調用進程睡眠在等待隊列nl_sk->sk_sleep,因此它必須在進程上下文使用,剛纔講的內核線程就可以採用這種方式來接收消息。

下面的函數input就是這種使用的示例:

Cpp代碼

1.#define NETLINK_CB(skb) (*(struct netlink_skb_parms*)&((skb)->cb))


來方便消息的地址設置。下面是一個消息地址設置的例子:

Cpp代碼

1.int netlink_unicast(struct sock *sk, struct sk_buff *skb, u32 pid, int nonblock);


參數sk爲函數netlink_kernel_create()返回的socket,參數skb存放消息,它的data字段指向要發送的netlink消息結構,而skb的控制塊保存了消息的地址信息,前面的宏NETLINK_CB(skb)就用於方便設置該控制塊,參數pid爲接收消息進程的pid,參數nonblock表示該函數是否爲非阻塞,如果爲1,該函數將在沒有接收緩存可利用時立即返回,而如果爲0,該函數在沒有接收緩存可利用時睡眠。

內核模塊或子系統也可以使用函數netlink_broadcast來發送廣播消息:

Cpp代碼

1.void sock_release(struct socket * sock);


注意函數netlink_kernel_create()返回的類型爲struct sock,因此函數sock_release應該這種調用:

Cpp代碼

1.sock_release(sk->sk_socket);


sk
爲函數netlink_kernel_create()的返回值。

源代碼包(見附件)中給出了一個使用 netlink 的示例,它包括一個內核模塊 netlink-exam-kern.c 和兩個應用程序 netlink-exam-user-recv.c, netlink-exam-user-send.c。內核模塊必須先插入到內核,然後在一個終端上運行用戶態接收程序,在另一個終端上運行用戶態發送程序,發送程序讀取參數指定的文本文件並把它作爲 netlink 消息的內容發送給內核模塊,內核模塊接受該消息保存到內核緩存中,它也通過proc接口出口到 procfs,因此用戶也能夠通過 /proc/netlink_exam_buffer 看到全部的內容,同時內核也把該消息發送給用戶態接收程序,用戶態接收程序將把接收到的內容輸出到屏幕上。

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章