linux netlink套接字学习资料

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <fcntl.h>
#include <errno.h>
#include <string.h>
#include <asm/types.h>
#include <sys/socket.h>
#include <linux/netlink.h>
#include <linux/inet_diag.h>
#include <netinet/tcp.h>

#define NETLINK_TEST 18
#define MAX_PAYLOAD 1024
struct req {
	struct nlmsghdr nlh;
	char [MAX_PAYLOAD];
};

void eprint(int err_no, char *str, int line)
{
	printf("Error %d in line %d:%s() with %s\n", err_no, line, str, strerror(errno));
}
int main()
{
int sock_fd;
sock_fd = socket(PF_NETLINK, SOCK_RAW, NETLINK_TEST);
if (sock_fd < 0) {
	eprint(errno, "socket", __LINE__);
    return errno;
}
//将本地套接字与源地址绑定
struct sockaddr_nl src_addr;
memset(&src_addr, 0, sizeof(src_addr));
src_addr.nl_family = AF_NETLINK;
src_addr.nl_pid = getpid();                 //nl_pid字段指明发送消息一方的pid
src_addr.nl_groups = 0;						//nl_groups表示多播组的掩码  这里我们并没有涉及多播。因此默觉得0

if (bind(sock_fd, (struct sockaddr *)&src_addr, sizeof(src_addr)) < 0) {
	eprint(errno, "bind", __LINE__);
	return errno;
}

//绑定了套接字之后。用户进程终于发送的是msghdr结构的消息，因此必须对这个结构进行初始化，
//而msghdr结构又与
//初始化msghdr结构 sockaddr_nl。iovec和nlmsghdr三个结构相关。因此必须依次对这些数据结构进行初始化
/*
struct msghdr {
	void	*	msg_name;	// Socket name			
	int		msg_namelen;	//Length of name		
	struct iovec *	msg_iov;	// Data blocks			
	__kernel_size_t	msg_iovlen;	// Number of blocks		
	void 	*	msg_control;	// Per protocol magic (eg BSD file descriptor passing) 
	__kernel_size_t	msg_controllen;	// Length of cmsg list 
	unsigned	msg_flags;
};*/
/*
struct sockaddr_nl {
	__kernel_sa_family_t	nl_family;	// AF_NETLINK	
	unsigned short	nl_pad;		//zero		
	__u32		nl_pid;		// port ID	
       	__u32		nl_groups;	// multicast groups mask 
};*/

struct sockaddr_nl dest_addr;
memset(&dest_addr, 0, sizeof(dest_addr));
dest_addr.nl_family = AF_NETLINK;
dest_addr.nl_pid = 0;   //即nl_pid必须为0，表示接收方为内核。                 dest_addr.nl_groups = 0;                 //req类型的数据报进行初始化。即依次对其封装的两个数据结构初始化：struct req r;//自己定义协议数据结构  使用netlink进行用户进程和内核的数据交互时 用到r.nlh.nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);				/*这里的nlmsg_len为为sizeof(structnlmsghdr)+MAX_PAYLOAD的总和。宏NLMSG_SPACE会自己主动将两者的长度相加#define NLMSG_LENGTH(len) ((len) + NLMSG_HDRLEN)#define NLMSG_SPACE(len) NLMSG_ALIGN(NLMSG_LENGTH(len)) */r.nlh.nlmsg_pid = getpid();r.nlh.nlmsg_flags = 0;memset(r.buf, 0, MAX_PAYLOAD);strcpy(NLMSG_DATA(&(r.nlh)), "hello, I am edsionte!");	//#define NLMSG_DATA(nlh)  ((void*)(((char*)nlh) + NLMSG_LENGTH(0)))//接下来对缓冲区向量iov进行初始化，让iov_base字段指向数据报结构，而iov_len为数据报长度。struct iovec iov;iov.iov_base = (void *)&r;iov.iov_len = sizeof(r);//一切就绪后，将目的套接字地址与当前要发送的消息msg绑定，即将目的套接字地址复制给msg_name。再将要发送的数据iov与msg_iov绑定，假设一次///性要发送多个数据包，则创建一个iovec类型的数组。struct msghdr msg;msg.msg_name = (void *)&dest_addr;msg.msg_namelen = sizeof(dest_addr);msg.msg_iov = &iov;msg.msg_iovlen = 1;//.向内核发送消息if (sendmsg(sock_fd, &msg, 0) < 0) {    eprint(errno, "sendmsg", __LINE__);    return errno;}//接收内核发来的消息memset(&r.nlh, 0, NLMSG_SPACE(MAX_PAYLOAD));if (recvmsg(sock_fd, &msg, 0) < 0) {        eprint(errno, "recvmsg", __LINE__);        return errno;}printf("Received message payload:%s\n", (char *)NLMSG_DATA(r.nlh));close(sock_fd);}