测试IPv6 ready logo   rfc 3315的时候,遇到一个问题,要求在收到ICMPv6 RA的时候,DHCPv6  Client要发Solicit消息。在平常的应用中,都是启动DHCPv6  Client后,client每隔一段时间就会发送Solicit,而并不是要在收到RA的时候发送。于是就修改了下kernel和client的代码,使得在kernel收到RA消息时,发送消息给DHCPv6  Client,使client发送Solicit。

思路为:使用netlink_kernel_create建立消息机制。在DHCPv6 Client初始化的时候,新建类型为NETLINK_RTK_ICMP6_RA  的netlink socket,监听kernel的消息,并且把自己的pid发送给kernel, kernel 收到消息后,存储pid为全局变量。当kernel收到RA包后,将发送消息给pid的用户进程。用户进程收到消息,发送solicit。

Kernel中接收ICMPv6 RA的代码在linux-3.10/net/ipv6/ndisc.c。

Kernel:

  1. 初始化IPC

在ndisc_init()中加上进程通信的socket。

//create netlink socket

struct netlink_kernel_cfg cfg = {

.input    = nl_icmp_input,

};

printk("%s: before netlink_kernel_create\n", __func__);

icmp6_ra_sock = netlink_kernel_create(&init_net, NETLINK_RTK_ICMP6_RA, &cfg);

if (icmp6_ra_sock == NULL) {

printk(KERN_ERR "Netlink[Kernel] Cannot create netlink socket for ipv6 ra.\n");

return -EAFNOSUPPORT;

}

printk("Netlink[Kernel] create netlink socket for ipv6 ra ok.\n");

其中NETLINK_RTK_ICMP6_RA  为在netlink.h中定义的一个类型。nl_icmp_input用于接收用户层发来的信息。

2. 接收DHCPv6 Client发来的pid 信息

static void nl_icmp_input(struct sk_buff  *__skb)

{

struct sk_buff *skb;

struct nlmsghdr *nlh;

char str[100];

struct completion cmpl;

int i=10;

skb = skb_get (__skb);

if(skb->len >= NLMSG_SPACE(0))

{

nlh = nlmsg_hdr(skb);

memcpy(str, NLMSG_DATA(nlh), sizeof(str));

printk("Message received:%s\n",str) ;

DHCP6C_PID = nlh->nlmsg_pid;

printk("DHCP PID = %d:%s\n", DHCP6C_PID) ;

kfree_skb(skb);

}

return;

}

3. 收到RA后,发消息给用户进程

在ndisc_router_discovery中,确认RA包ok后,发送消息。

if (icmp6_ra_sock) {

printk("%s: ra socket is not null\n", __func__);

payloadLen = NLMSG_SPACE(payloadLen);

/*Alloc skb ,this check helps to call the fucntion from interrupt context */

if(in_atomic())

{

skb_ra = alloc_skb(payloadLen, GFP_ATOMIC);

}

else

{

skb_ra = alloc_skb(payloadLen, GFP_KERNEL);

}

if(!skb_ra)

{

printk(KERN_ERR "failed to alloc skb in %s",__FUNCTION__);

return;

}

nl_msgHdr = (struct nlmsghdr *)skb_ra->data;

nl_msgHdr->nlmsg_type = MESSAGE_ICMP6_RA;

nl_msgHdr->nlmsg_pid=0;/*from kernel */

nl_msgHdr->nlmsg_len = payloadLen;

nl_msgHdr->nlmsg_flags =0;

NETLINK_CB(skb_ra).portid = 0; /*from kernel */

skb_ra->len = payloadLen;

err = netlink_unicast(icmp6_ra_sock, skb_ra, DHCP6C_PID, MSG_DONTWAIT); //send to dhcp6c

if (err < 0)

printk("%s: nfnetlink_unicast icmpv6 ra socket fail\n", __func__);

else

printk("%s: nfnetlink_unicast icmpv6 ra socket ok\n", __func__);

}

用户层:

  1. Dhcp6c 初始化,并发送pid给kernel

int init_ra_monitor_fd()

{

int ret = 0;

struct sockaddr_nl src_addr, dest_addr;

struct nlmsghdr *nlh = NULL;

struct iovec iov;

struct msghdr msg;

int state_smg = 0;

int pid = getpid();

if ((rasock = socket(AF_NETLINK, SOCK_RAW, NETLINK_RTK_ICMP6_RA)) < 0)

{

printf("Could not open netlink socket for kernel monitor\n");

return 1;

}

else

{

printf("kernelMonitorFd=%d\n", rasock);

}

memset(&src_addr, 0, sizeof(src_addr));

src_addr.nl_family = AF_NETLINK;

src_addr.nl_pid = pid;

src_addr.nl_groups = 0; // multicast

if (bind(rasock, (struct sockaddr *)&src_addr, sizeof(src_addr)) < 0)

{

printf("Could not bind netlink socket for kernel monitor\n");

close(rasock);

rasock = -1;

return 1;

}

/* Tell kernel dhcp6c pid, kernel will send messages to dhcp6c when it receive ICMPv6 RA */

/* Fix IPv6 Ready logo rfc3315 */

memset(&dest_addr, 0, sizeof(dest_addr));

dest_addr.nl_family = AF_NETLINK;

dest_addr.nl_pid = 0;

dest_addr.nl_groups = 0;

nlh = (struct nlmsghdr *)malloc(NLMSG_SPACE(MAX_PAYLOAD));

if(!nlh){

printf("malloc nlmsghdr error!\n");

close(rasock);

return -1;

}

nlh->nlmsg_len = NLMSG_SPACE(MAX_PAYLOAD);

nlh->nlmsg_pid = pid;

nlh->nlmsg_flags = 0;

sprintf(NLMSG_DATA(nlh), "%d", pid);

iov.iov_base = (void *)nlh;

iov.iov_len = NLMSG_SPACE(MAX_PAYLOAD);

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

msg.msg_name = (void *)&dest_addr;

msg.msg_namelen = sizeof(dest_addr);

msg.msg_iov = &iov;

msg.msg_iovlen = 1;

printf("state_smg\n");

state_smg = sendmsg(rasock, &msg, 0);

if(state_smg == -1)

{

printf("get error sendmsg = %s\n",strerror(errno));

}

}

2. 等待接收消息

// add to read sockets

if (rasock >= 0) {

FD_SET(rasock, &r);

maxsock = (rasock > maxsock) ? rasock : maxsock;

}

//wait

ret = select(maxsock + 1, &r, NULL, NULL, w);

if (FD_ISSET(rasock, &r)) {

printf("%s: have receive the ra message\n", __func__);

dhcp6_ra_recv(); //send solicit

}

05-15 22:39