1/* 2 * cn_test.c 3 * 4 * 2004-2005 Copyright (c) Evgeniy Polyakov <johnpol@2ka.mipt.ru> 5 * All rights reserved. 6 * 7 * This program is free software; you can redistribute it and/or modify 8 * it under the terms of the GNU General Public License as published by 9 * the Free Software Foundation; either version 2 of the License, or 10 * (at your option) any later version. 11 * 12 * This program is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 15 * GNU General Public License for more details. 16 * 17 * You should have received a copy of the GNU General Public License 18 * along with this program; if not, write to the Free Software 19 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA 20 */ 21 22#include <linux/kernel.h> 23#include <linux/module.h> 24#include <linux/moduleparam.h> 25#include <linux/skbuff.h> 26#include <linux/timer.h> 27 28#include <linux/connector.h> 29 30static struct cb_id cn_test_id = { 0x123, 0x456 }; 31static char cn_test_name[] = "cn_test"; 32static struct sock *nls; 33static struct timer_list cn_test_timer; 34 35void cn_test_callback(void *data) 36{ 37 struct cn_msg *msg = (struct cn_msg *)data; 38 39 printk("%s: %lu: idx=%x, val=%x, seq=%u, ack=%u, len=%d: %s.\n", 40 __func__, jiffies, msg->id.idx, msg->id.val, 41 msg->seq, msg->ack, msg->len, (char *)msg->data); 42} 43 44static int cn_test_want_notify(void) 45{ 46 struct cn_ctl_msg *ctl; 47 struct cn_notify_req *req; 48 struct cn_msg *msg = NULL; 49 int size, size0; 50 struct sk_buff *skb; 51 struct nlmsghdr *nlh; 52 u32 group = 1; 53 54 size0 = sizeof(*msg) + sizeof(*ctl) + 3 * sizeof(*req); 55 56 size = NLMSG_SPACE(size0); 57 58 skb = alloc_skb(size, GFP_ATOMIC); 59 if (!skb) { 60 printk(KERN_ERR "Failed to allocate new skb with size=%u.\n", 61 size); 62 63 return -ENOMEM; 64 } 65 66 nlh = NLMSG_PUT(skb, 0, 0x123, NLMSG_DONE, size - sizeof(*nlh)); 67 68 msg = (struct cn_msg *)NLMSG_DATA(nlh); 69 70 memset(msg, 0, size0); 71 72 msg->id.idx = -1; 73 msg->id.val = -1; 74 msg->seq = 0x123; 75 msg->ack = 0x345; 76 msg->len = size0 - sizeof(*msg); 77 78 ctl = (struct cn_ctl_msg *)(msg + 1); 79 80 ctl->idx_notify_num = 1; 81 ctl->val_notify_num = 2; 82 ctl->group = group; 83 ctl->len = msg->len - sizeof(*ctl); 84 85 req = (struct cn_notify_req *)(ctl + 1); 86 87 /* 88 * Idx. 89 */ 90 req->first = cn_test_id.idx; 91 req->range = 10; 92 93 /* 94 * Val 0. 95 */ 96 req++; 97 req->first = cn_test_id.val; 98 req->range = 10; 99 100 /* 101 * Val 1. 102 */ 103 req++; 104 req->first = cn_test_id.val + 20; 105 req->range = 10; 106 107 NETLINK_CB(skb).dst_group = ctl->group; 108 //netlink_broadcast(nls, skb, 0, ctl->group, GFP_ATOMIC); 109 netlink_unicast(nls, skb, 0, 0); 110 111 printk(KERN_INFO "Request was sent. Group=0x%x.\n", ctl->group); 112 113 return 0; 114 115nlmsg_failure: 116 printk(KERN_ERR "Failed to send %u.%u\n", msg->seq, msg->ack); 117 kfree_skb(skb); 118 return -EINVAL; 119} 120 121static u32 cn_test_timer_counter; 122static void cn_test_timer_func(unsigned long __data) 123{ 124 struct cn_msg *m; 125 char data[32]; 126 127 m = kmalloc(sizeof(*m) + sizeof(data), GFP_ATOMIC); 128 if (m) { 129 memset(m, 0, sizeof(*m) + sizeof(data)); 130 131 memcpy(&m->id, &cn_test_id, sizeof(m->id)); 132 m->seq = cn_test_timer_counter; 133 m->len = sizeof(data); 134 135 m->len = 136 scnprintf(data, sizeof(data), "counter = %u", 137 cn_test_timer_counter) + 1; 138 139 memcpy(m + 1, data, m->len); 140 141 cn_netlink_send(m, 0, gfp_any()); 142 kfree(m); 143 } 144 145 cn_test_timer_counter++; 146 147 mod_timer(&cn_test_timer, jiffies + HZ); 148} 149 150static int cn_test_init(void) 151{ 152 int err; 153 154 err = cn_add_callback(&cn_test_id, cn_test_name, cn_test_callback); 155 if (err) 156 goto err_out; 157 cn_test_id.val++; 158 err = cn_add_callback(&cn_test_id, cn_test_name, cn_test_callback); 159 if (err) { 160 cn_del_callback(&cn_test_id); 161 goto err_out; 162 } 163 164 init_timer(&cn_test_timer); 165 cn_test_timer.function = cn_test_timer_func; 166 cn_test_timer.expires = jiffies + HZ; 167 cn_test_timer.data = 0; 168 add_timer(&cn_test_timer); 169 170 return 0; 171 172 err_out: 173 if (nls && nls->sk_socket) 174 sock_release(nls->sk_socket); 175 176 return err; 177} 178 179static void cn_test_fini(void) 180{ 181 del_timer_sync(&cn_test_timer); 182 cn_del_callback(&cn_test_id); 183 cn_test_id.val--; 184 cn_del_callback(&cn_test_id); 185 if (nls && nls->sk_socket) 186 sock_release(nls->sk_socket); 187} 188 189module_init(cn_test_init); 190module_exit(cn_test_fini); 191 192MODULE_LICENSE("GPL"); 193MODULE_AUTHOR("Evgeniy Polyakov <johnpol@2ka.mipt.ru>"); 194MODULE_DESCRIPTION("Connector's test module"); 195