熟悉netlink和内核多线程编程的进来看看！

lihw_2002

下面是参照网上的程序写的一段基于netlink的通信程序（内核），有些地方还不明白，先拿来用用．调试时能接收用户空间数据，但只要一调用向用户空间发送信息的函数就down机，大虾指点一下吧，谢谢啦！

struct {   __u32 pid;   rwlock_t lock; }user_proc; static struct sock *netlink_sock_recv; static struct sock *netlink_sock_send; static int exit_flag = 0; static DECLARE_COMPLETION(exit_completion); static DECLARE_COMPLETION(send_completion); static DECLARE_COMPLETION(sendp_completion); static void recv_handler(struct sock * sk, int length){         wake_up(sk->sk_sleep); } static int process_recv_thread(void * data){         struct sk_buff * skb = NULL;         struct nlmsghdr * nlhdr = NULL;      DEFINE_WAIT(wait);         daemonize("mynetlink");         while (exit_flag == 0) {                 prepare_to_wait(netlink_sock_recv->sk_sleep, &wait, TASK_INTERRUPTIBLE);                 schedule();                 finish_wait(netlink_sock_recv->sk_sleep, &wait);                 while ((skb = skb_dequeue(&netlink_sock_recv->sk_receive_queue))                          != NULL) {                         nlhdr = (struct nlmsghdr *)skb->data;                         if (nlhdr->nlmsg_len < sizeof(struct nlmsghdr)) {                                 printk("Corrupt netlink message.\n");                                 continue;                         }                packet=NLMSG_DATA(nlhdr);             //PROCESS  DATA                                         }         }         complete(&exit_completion);         return 0; } static void send_handler(struct sock *sk, int len){       wake_up(sk->sk_sleep); } static int process_send_thread(void *data){     struct sk_buff * skb = NULL;       struct nlmsghdr * nlhdr = NULL;      DEFINE_WAIT(waits);     daemonize("mynetlinksend");         while (exit_flag == 0) {         prepare_to_wait(netlink_sock_send->sk_sleep, &waits, TASK_INTERRUPTIBLE);               schedule();               finish_wait(netlink_sock_send->sk_sleep, &waits);               while ((skb = skb_dequeue(&netlink_sock_send->sk_receive_queue))!= NULL) {             nlhdr = (struct nlmsghdr *)skb->data;                  if (nlhdr->nlmsg_len < sizeof(struct nlmsghdr)) {                          printk("Corrupt netlink message.\n");                         continue;                       }             //packet=NLMSG_DATA(nlhdr);             if(nlhdr->nlmsg_type == ACC_U_PID)               {             write_lock_bh(&user_proc.lock);             user_proc.pid = nlhdr->nlmsg_pid;             write_unlock_bh(&user_proc.lock);               }             else if(nlhdr->nlmsg_type == ACC_CLOSE)               {             write_lock_bh(&user_proc.lock);             if(nlhdr->nlmsg_pid == user_proc.pid)               user_proc.pid = 0;             write_unlock_bh(&user_proc.lock);               }                    }         }     complete(&send_completion);     return 0; } static int send_to_user(struct info *info){   int ret;   int size;   unsigned char *old_tail;   struct sk_buff *skb;   struct nlmsghdr *nlh;   struct info *sinfo;   size = NLMSG_SPACE(sizeof(*info));   skb = alloc_skb(size, GFP_ATOMIC);   old_tail = skb->tail;   nlh = NLMSG_PUT(skb, 0, 0, ACC_K_MSG, size-sizeof(*nlh));   sinfo= NLMSG_DATA(nlh);   memset(sinfo, 0, sizeof(struct info));     //填充sinfo   nlh->nlmsg_len = skb->tail - old_tail;   NETLINK_CB(skb).dst_group = 0;   read_lock_bh(&user_proc.lock);   ret = netlink_unicast(netlink_sock_send, skb, user_proc.pid, MSG_DONTWAIT);   read_unlock_bh(&user_proc.lock);   return ret; nlmsg_failure:   if(skb)     kfree_skb(skb);   return -1; } //周期性的向用户空间发送数据 static int send_thread(void *data){     struct info sinfo;     daemonize("mynetlinksend");     schedule();     while(exit_flag == 0){         //填充sinfo?         send_to_user(&sinfo);         sleep(50000);     }      complete(&sendp_completion); } static int __init netlink_exam_init(void){         netlink_sock_recv= netlink_kernel_create(NL_RECV, 0, recv_handler, THIS_MODULE);         if (!netlink_sock_recv) {                 printk("Fail to create netlink socket.\n");                 return 1;         }     netlink_sock_send = netlink_kernel_create(NL_SEND, 0, send_handler, THIS_MODULE);     if (!netlink_sock_send) {                 printk("Fail to create netlink socket.\n");                 return 1;         }         kernel_thread(process_recv_thread, NULL, CLONE_KERNEL);      kernel_thread(process_send_thread, NULL, CLONE_KERNEL);         kernel_thread(send_thread,NULL,CLONE_KERNEL);         return 0; } static void __exit netlink_exam_exit(void){         exit_flag = 1;         wake_up(netlink_sock_recv->sk_sleep);      wait_for_completion(&exit_completion);         sock_release(netlink_sock_recv->sk_socket);         wake_up(netlink_sock_send->sk_sleep);      wait_for_completion(&send_completion);         sock_release(netlink_sock_send->sk_socket);     wait_for_completion(&sendp_completion); }

[ 本帖最后由 lihw_2002 于 2008-2-28 14:11 编辑 ]