4 * Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of version 2 of the GNU General Public License as
8 * published by the Free Software Foundation.
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
15 * You should have received a copy of the GNU General Public License
16 * along with this program; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin St - Fifth Floor, Boston, MA 02110-1301 USA.
18 * The full GNU General Public License is included in this distribution
19 * in the file called LICENSE.GPL.
21 * Contact Information:
25 #include <linux/module.h>
26 #include <linux/miscdevice.h>
27 #include <linux/netdevice.h>
28 #include <linux/pci.h>
29 #include <linux/kthread.h>
30 #include <linux/rwsem.h>
32 #include <exec-env/rte_kni_common.h>
34 #include <rte_config.h>
36 MODULE_LICENSE("Dual BSD/GPL");
37 MODULE_AUTHOR("Intel Corporation");
38 MODULE_DESCRIPTION("Kernel Module for managing kni devices");
40 #define KNI_RX_LOOP_NUM 1000
42 #define KNI_MAX_DEVICES 32
44 extern void kni_net_rx(struct kni_dev *kni);
45 extern void kni_net_init(struct net_device *dev);
46 extern void kni_net_config_lo_mode(char *lo_str);
47 extern void kni_net_poll_resp(struct kni_dev *kni);
48 extern void kni_set_ethtool_ops(struct net_device *netdev);
50 extern int ixgbe_kni_probe(struct pci_dev *pdev, struct net_device **lad_dev);
51 extern void ixgbe_kni_remove(struct pci_dev *pdev);
52 extern int igb_kni_probe(struct pci_dev *pdev, struct net_device **lad_dev);
53 extern void igb_kni_remove(struct pci_dev *pdev);
55 static int kni_open(struct inode *inode, struct file *file);
56 static int kni_release(struct inode *inode, struct file *file);
57 static int kni_ioctl(struct inode *inode, unsigned int ioctl_num,
58 unsigned long ioctl_param);
59 static int kni_compat_ioctl(struct inode *inode, unsigned int ioctl_num,
60 unsigned long ioctl_param);
61 static int kni_dev_remove(struct kni_dev *dev);
63 static int __init kni_parse_kthread_mode(void);
65 /* KNI processing for single kernel thread mode */
66 static int kni_thread_single(void *unused);
67 /* KNI processing for multiple kernel thread mode */
68 static int kni_thread_multiple(void *param);
70 static struct file_operations kni_fops = {
73 .release = kni_release,
74 .unlocked_ioctl = (void *)kni_ioctl,
75 .compat_ioctl = (void *)kni_compat_ioctl,
78 static struct miscdevice kni_misc = {
79 .minor = MISC_DYNAMIC_MINOR,
85 static char *lo_mode = NULL;
87 /* Kernel thread mode */
88 static char *kthread_mode = NULL;
89 static unsigned multiple_kthread_on = 0;
91 #define KNI_DEV_IN_USE_BIT_NUM 0 /* Bit number for device in use */
93 static volatile unsigned long device_in_use; /* device in use flag */
94 static struct task_struct *kni_kthread;
97 static DECLARE_RWSEM(kni_list_lock);
100 static struct list_head kni_list_head = LIST_HEAD_INIT(kni_list_head);
105 KNI_PRINT("######## DPDK kni module loading ########\n");
107 if (kni_parse_kthread_mode() < 0) {
108 KNI_ERR("Invalid parameter for kthread_mode\n");
112 if (misc_register(&kni_misc) != 0) {
113 KNI_ERR("Misc registration failed\n");
117 /* Clear the bit of device in use */
118 clear_bit(KNI_DEV_IN_USE_BIT_NUM, &device_in_use);
120 /* Configure the lo mode according to the input parameter */
121 kni_net_config_lo_mode(lo_mode);
123 KNI_PRINT("######## DPDK kni module loaded ########\n");
131 misc_deregister(&kni_misc);
132 KNI_PRINT("####### DPDK kni module unloaded #######\n");
136 kni_parse_kthread_mode(void)
141 if (strcmp(kthread_mode, "single") == 0)
143 else if (strcmp(kthread_mode, "multiple") == 0)
144 multiple_kthread_on = 1;
152 kni_open(struct inode *inode, struct file *file)
154 /* kni device can be opened by one user only, test and set bit */
155 if (test_and_set_bit(KNI_DEV_IN_USE_BIT_NUM, &device_in_use))
158 /* Create kernel thread for single mode */
159 if (multiple_kthread_on == 0) {
160 KNI_PRINT("Single kernel thread for all KNI devices\n");
161 /* Create kernel thread for RX */
162 kni_kthread = kthread_run(kni_thread_single, NULL,
164 if (IS_ERR(kni_kthread)) {
165 KNI_ERR("Unable to create kernel threaed\n");
166 return PTR_ERR(kni_kthread);
169 KNI_PRINT("Multiple kernel thread mode enabled\n");
171 KNI_PRINT("/dev/kni opened\n");
177 kni_release(struct inode *inode, struct file *file)
179 struct kni_dev *dev, *n;
181 /* Stop kernel thread for single mode */
182 if (multiple_kthread_on == 0) {
183 /* Stop kernel thread */
184 kthread_stop(kni_kthread);
188 down_write(&kni_list_lock);
189 list_for_each_entry_safe(dev, n, &kni_list_head, list) {
190 /* Stop kernel thread for multiple mode */
191 if (multiple_kthread_on && dev->pthread != NULL) {
192 kthread_stop(dev->pthread);
197 kni_vhost_backend_release(dev);
200 list_del(&dev->list);
202 up_write(&kni_list_lock);
204 /* Clear the bit of device in use */
205 clear_bit(KNI_DEV_IN_USE_BIT_NUM, &device_in_use);
207 KNI_PRINT("/dev/kni closed\n");
213 kni_thread_single(void *unused)
216 struct kni_dev *dev, *n;
218 while (!kthread_should_stop()) {
219 down_read(&kni_list_lock);
220 for (j = 0; j < KNI_RX_LOOP_NUM; j++) {
221 list_for_each_entry_safe(dev, n,
222 &kni_list_head, list) {
224 kni_chk_vhost_rx(dev);
228 kni_net_poll_resp(dev);
231 up_read(&kni_list_lock);
232 /* reschedule out for a while */
233 schedule_timeout_interruptible(usecs_to_jiffies( \
234 KNI_KTHREAD_RESCHEDULE_INTERVAL));
241 kni_thread_multiple(void *param)
244 struct kni_dev *dev = (struct kni_dev *)param;
246 while (!kthread_should_stop()) {
247 for (j = 0; j < KNI_RX_LOOP_NUM; j++) {
249 kni_chk_vhost_rx(dev);
253 kni_net_poll_resp(dev);
255 schedule_timeout_interruptible(usecs_to_jiffies( \
256 KNI_KTHREAD_RESCHEDULE_INTERVAL));
263 kni_dev_remove(struct kni_dev *dev)
268 switch (dev->device_id) {
269 #define RTE_PCI_DEV_ID_DECL_IGB(vend, dev) case (dev):
270 #include <rte_pci_dev_ids.h>
271 igb_kni_remove(dev->pci_dev);
273 #define RTE_PCI_DEV_ID_DECL_IXGBE(vend, dev) case (dev):
274 #include <rte_pci_dev_ids.h>
275 ixgbe_kni_remove(dev->pci_dev);
282 unregister_netdev(dev->net_dev);
283 free_netdev(dev->net_dev);
290 kni_check_param(struct kni_dev *kni, struct rte_kni_device_info *dev)
295 /* Check if network name has been used */
296 if (!strncmp(kni->name, dev->name, RTE_KNI_NAMESIZE)) {
297 KNI_ERR("KNI name %s duplicated\n", dev->name);
305 kni_ioctl_create(unsigned int ioctl_num, unsigned long ioctl_param)
308 struct rte_kni_device_info dev_info;
309 struct pci_dev *pci = NULL;
310 struct pci_dev *found_pci = NULL;
311 struct net_device *net_dev = NULL;
312 struct net_device *lad_dev = NULL;
313 struct kni_dev *kni, *dev, *n;
315 printk(KERN_INFO "KNI: Creating kni...\n");
316 /* Check the buffer size, to avoid warning */
317 if (_IOC_SIZE(ioctl_num) > sizeof(dev_info))
320 /* Copy kni info from user space */
321 ret = copy_from_user(&dev_info, (void *)ioctl_param, sizeof(dev_info));
323 KNI_ERR("copy_from_user in kni_ioctl_create");
328 * Check if the cpu core id is valid for binding,
329 * for multiple kernel thread mode.
331 if (multiple_kthread_on && dev_info.force_bind &&
332 !cpu_online(dev_info.core_id)) {
333 KNI_ERR("cpu %u is not online\n", dev_info.core_id);
337 /* Check if it has been created */
338 down_read(&kni_list_lock);
339 list_for_each_entry_safe(dev, n, &kni_list_head, list) {
340 if (kni_check_param(dev, &dev_info) < 0) {
341 up_read(&kni_list_lock);
345 up_read(&kni_list_lock);
347 net_dev = alloc_netdev(sizeof(struct kni_dev), dev_info.name,
349 if (net_dev == NULL) {
350 KNI_ERR("error allocating device \"%s\"\n", dev_info.name);
354 kni = netdev_priv(net_dev);
356 kni->net_dev = net_dev;
357 kni->group_id = dev_info.group_id;
358 kni->core_id = dev_info.core_id;
359 strncpy(kni->name, dev_info.name, RTE_KNI_NAMESIZE);
361 /* Translate user space info into kernel space info */
362 kni->tx_q = phys_to_virt(dev_info.tx_phys);
363 kni->rx_q = phys_to_virt(dev_info.rx_phys);
364 kni->alloc_q = phys_to_virt(dev_info.alloc_phys);
365 kni->free_q = phys_to_virt(dev_info.free_phys);
367 kni->req_q = phys_to_virt(dev_info.req_phys);
368 kni->resp_q = phys_to_virt(dev_info.resp_phys);
369 kni->sync_va = dev_info.sync_va;
370 kni->sync_kva = phys_to_virt(dev_info.sync_phys);
372 kni->mbuf_kva = phys_to_virt(dev_info.mbuf_phys);
373 kni->mbuf_va = dev_info.mbuf_va;
376 kni->vhost_queue = NULL;
377 kni->vq_status = BE_STOP;
379 kni->mbuf_size = dev_info.mbuf_size;
381 KNI_PRINT("tx_phys: 0x%016llx, tx_q addr: 0x%p\n",
382 (unsigned long long) dev_info.tx_phys, kni->tx_q);
383 KNI_PRINT("rx_phys: 0x%016llx, rx_q addr: 0x%p\n",
384 (unsigned long long) dev_info.rx_phys, kni->rx_q);
385 KNI_PRINT("alloc_phys: 0x%016llx, alloc_q addr: 0x%p\n",
386 (unsigned long long) dev_info.alloc_phys, kni->alloc_q);
387 KNI_PRINT("free_phys: 0x%016llx, free_q addr: 0x%p\n",
388 (unsigned long long) dev_info.free_phys, kni->free_q);
389 KNI_PRINT("req_phys: 0x%016llx, req_q addr: 0x%p\n",
390 (unsigned long long) dev_info.req_phys, kni->req_q);
391 KNI_PRINT("resp_phys: 0x%016llx, resp_q addr: 0x%p\n",
392 (unsigned long long) dev_info.resp_phys, kni->resp_q);
393 KNI_PRINT("mbuf_phys: 0x%016llx, mbuf_kva: 0x%p\n",
394 (unsigned long long) dev_info.mbuf_phys, kni->mbuf_kva);
395 KNI_PRINT("mbuf_va: 0x%p\n", dev_info.mbuf_va);
396 KNI_PRINT("mbuf_size: %u\n", kni->mbuf_size);
398 KNI_DBG("PCI: %02x:%02x.%02x %04x:%04x\n",
405 pci = pci_get_device(dev_info.vendor_id, dev_info.device_id, NULL);
407 /* Support Ethtool */
409 KNI_PRINT("pci_bus: %02x:%02x:%02x \n",
411 PCI_SLOT(pci->devfn),
412 PCI_FUNC(pci->devfn));
414 if ((pci->bus->number == dev_info.bus) &&
415 (PCI_SLOT(pci->devfn) == dev_info.devid) &&
416 (PCI_FUNC(pci->devfn) == dev_info.function)) {
418 switch (dev_info.device_id) {
419 #define RTE_PCI_DEV_ID_DECL_IGB(vend, dev) case (dev):
420 #include <rte_pci_dev_ids.h>
421 ret = igb_kni_probe(found_pci, &lad_dev);
423 #define RTE_PCI_DEV_ID_DECL_IXGBE(vend, dev) \
425 #include <rte_pci_dev_ids.h>
426 ret = ixgbe_kni_probe(found_pci, &lad_dev);
433 KNI_DBG("PCI found: pci=0x%p, lad_dev=0x%p\n",
436 kni->lad_dev = lad_dev;
437 kni_set_ethtool_ops(kni->net_dev);
439 KNI_ERR("Device not supported by ethtool");
443 kni->pci_dev = found_pci;
444 kni->device_id = dev_info.device_id;
447 pci = pci_get_device(dev_info.vendor_id,
448 dev_info.device_id, pci);
453 ret = register_netdev(net_dev);
455 KNI_ERR("error %i registering device \"%s\"\n",
466 * Create a new kernel thread for multiple mode, set its core affinity,
467 * and finally wake it up.
469 if (multiple_kthread_on) {
470 kni->pthread = kthread_create(kni_thread_multiple,
472 "kni_%s", kni->name);
473 if (IS_ERR(kni->pthread)) {
477 if (dev_info.force_bind)
478 kthread_bind(kni->pthread, kni->core_id);
479 wake_up_process(kni->pthread);
482 down_write(&kni_list_lock);
483 list_add(&kni->list, &kni_list_head);
484 up_write(&kni_list_lock);
490 kni_ioctl_release(unsigned int ioctl_num, unsigned long ioctl_param)
493 struct kni_dev *dev, *n;
494 struct rte_kni_device_info dev_info;
496 if (_IOC_SIZE(ioctl_num) > sizeof(dev_info))
499 ret = copy_from_user(&dev_info, (void *)ioctl_param, sizeof(dev_info));
501 KNI_ERR("copy_from_user in kni_ioctl_release");
505 /* Release the network device according to its name */
506 if (strlen(dev_info.name) == 0)
509 down_write(&kni_list_lock);
510 list_for_each_entry_safe(dev, n, &kni_list_head, list) {
511 if (strncmp(dev->name, dev_info.name, RTE_KNI_NAMESIZE) != 0)
514 if (multiple_kthread_on && dev->pthread != NULL) {
515 kthread_stop(dev->pthread);
520 kni_vhost_backend_release(dev);
523 list_del(&dev->list);
527 up_write(&kni_list_lock);
528 printk(KERN_INFO "KNI: %s release kni named %s\n",
529 (ret == 0 ? "Successfully" : "Unsuccessfully"), dev_info.name);
535 kni_ioctl(struct inode *inode,
536 unsigned int ioctl_num,
537 unsigned long ioctl_param)
541 KNI_DBG("IOCTL num=0x%0x param=0x%0lx \n", ioctl_num, ioctl_param);
544 * Switch according to the ioctl called
546 switch (_IOC_NR(ioctl_num)) {
547 case _IOC_NR(RTE_KNI_IOCTL_TEST):
548 /* For test only, not used */
550 case _IOC_NR(RTE_KNI_IOCTL_CREATE):
551 ret = kni_ioctl_create(ioctl_num, ioctl_param);
553 case _IOC_NR(RTE_KNI_IOCTL_RELEASE):
554 ret = kni_ioctl_release(ioctl_num, ioctl_param);
557 KNI_DBG("IOCTL default \n");
565 kni_compat_ioctl(struct inode *inode,
566 unsigned int ioctl_num,
567 unsigned long ioctl_param)
569 /* 32 bits app on 64 bits OS to be supported later */
570 KNI_PRINT("Not implemented.\n");
575 module_init(kni_init);
576 module_exit(kni_exit);
578 module_param(lo_mode, charp, S_IRUGO | S_IWUSR);
579 MODULE_PARM_DESC(lo_mode,
580 "KNI loopback mode (default=lo_mode_none):\n"
581 " lo_mode_none Kernel loopback disabled\n"
582 " lo_mode_fifo Enable kernel loopback with fifo\n"
583 " lo_mode_fifo_skb Enable kernel loopback with fifo and skb buffer\n"
587 module_param(kthread_mode, charp, S_IRUGO);
588 MODULE_PARM_DESC(kthread_mode,
589 "Kernel thread mode (default=single):\n"
590 " single Single kernel thread mode enabled.\n"
591 " multiple Multiple kernel thread mode enabled.\n"