drivers/uio/uio.c

   1 /*
   2  * drivers/uio/uio.c
   3  *
   4  * Copyright(C) 2005, Benedikt Spranger <b.spranger@linutronix.de>
   5  * Copyright(C) 2005, Thomas Gleixner <tglx@linutronix.de>
   6  * Copyright(C) 2006, Hans J. Koch <hjk@linutronix.de>
   7  * Copyright(C) 2006, Greg Kroah-Hartman <greg@kroah.com>
   8  *
   9  * Userspace IO
  10  *
  11  * Base Functions
  12  *
  13  * Licensed under the GPLv2 only.
  14  */
  15
  16 #include <linux/module.h>
  17 #include <linux/init.h>
  18 #include <linux/poll.h>
  19 #include <linux/device.h>
  20 #include <linux/mm.h>
  21 #include <linux/idr.h>
  22 #include <linux/string.h>
  23 #include <linux/kobject.h>
  24 #include <linux/uio_driver.h>
  25
  26 #define UIO_MAX_DEVICES 255
  27
  28 struct uio_device {
  29         struct module           *owner;
  30         struct device           *dev;
  31         int                     minor;
  32         atomic_t                event;
  33         struct fasync_struct    *async_queue;
  34         wait_queue_head_t       wait;
  35         int                     vma_count;
  36         struct uio_info         *info;
  37         struct kobject          *map_dir;
  38 };
  39
  40 static int uio_major;
  41 static DEFINE_IDR(uio_idr);
  42 static struct file_operations uio_fops;
  43
  44 /* UIO class infrastructure */
  45 static struct uio_class {
  46         struct kref kref;
  47         struct class *class;
  48 } *uio_class;
  49
  50 /*
  51  * attributes
  52  */
  53
  54 struct uio_map {
  55         struct kobject kobj;
  56         struct uio_mem *mem;
  57 };
  58 #define to_map(map) container_of(map, struct uio_map, kobj)
  59
  60
  61 static ssize_t map_attr_show(struct kobject *kobj, struct kobj_attribute *attr,
  62                              char *buf)
  63 {
  64         struct uio_map *map = to_map(kobj);
  65         struct uio_mem *mem = map->mem;
  66
  67         if (strncmp(attr->attr.name, "addr", 4) == 0)
  68                 return sprintf(buf, "0x%lx\n", mem->addr);
  69
  70         if (strncmp(attr->attr.name, "size", 4) == 0)
  71                 return sprintf(buf, "0x%lx\n", mem->size);
  72
  73         return -ENODEV;
  74 }
  75
  76 static struct kobj_attribute attr_attribute =
  77         __ATTR(addr, S_IRUGO, map_attr_show, NULL);
  78 static struct kobj_attribute size_attribute =
  79         __ATTR(size, S_IRUGO, map_attr_show, NULL);
  80
  81 static struct attribute *attrs[] = {
  82         &attr_attribute.attr,
  83         &size_attribute.attr,
  84         NULL,   /* need to NULL terminate the list of attributes */
  85 };
  86
  87 static void map_release(struct kobject *kobj)
  88 {
  89         struct uio_map *map = to_map(kobj);
  90         kfree(map);
  91 }
  92
  93 static struct kobj_type map_attr_type = {
  94         .release        = map_release,
  95         .default_attrs  = attrs,
  96 };
  97
  98 static ssize_t show_name(struct device *dev,
  99                          struct device_attribute *attr, char *buf)
 100 {
 101         struct uio_device *idev = dev_get_drvdata(dev);
 102         if (idev)
 103                 return sprintf(buf, "%s\n", idev->info->name);
 104         else
 105                 return -ENODEV;
 106 }
 107 static DEVICE_ATTR(name, S_IRUGO, show_name, NULL);
 108
 109 static ssize_t show_version(struct device *dev,
 110                             struct device_attribute *attr, char *buf)
 111 {
 112         struct uio_device *idev = dev_get_drvdata(dev);
 113         if (idev)
 114                 return sprintf(buf, "%s\n", idev->info->version);
 115         else
 116                 return -ENODEV;
 117 }
 118 static DEVICE_ATTR(version, S_IRUGO, show_version, NULL);
 119
 120 static ssize_t show_event(struct device *dev,
 121                           struct device_attribute *attr, char *buf)
 122 {
 123         struct uio_device *idev = dev_get_drvdata(dev);
 124         if (idev)
 125                 return sprintf(buf, "%u\n",
 126                                 (unsigned int)atomic_read(&idev->event));
 127         else
 128                 return -ENODEV;
 129 }
 130 static DEVICE_ATTR(event, S_IRUGO, show_event, NULL);
 131
 132 static struct attribute *uio_attrs[] = {
 133         &dev_attr_name.attr,
 134         &dev_attr_version.attr,
 135         &dev_attr_event.attr,
 136         NULL,
 137 };
 138
 139 static struct attribute_group uio_attr_grp = {
 140         .attrs = uio_attrs,
 141 };
 142
 143 /*
 144  * device functions
 145  */
 146 static int uio_dev_add_attributes(struct uio_device *idev)
 147 {
 148         int ret;
 149         int mi;
 150         int map_found = 0;
 151         struct uio_mem *mem;
 152         struct uio_map *map;
 153
 154         ret = sysfs_create_group(&idev->dev->kobj, &uio_attr_grp);
 155         if (ret)
 156                 goto err_group;
 157
 158         for (mi = 0; mi < MAX_UIO_MAPS; mi++) {
 159                 mem = &idev->info->mem[mi];
 160                 if (mem->size == 0)
 161                         break;
 162                 if (!map_found) {
 163                         map_found = 1;
 164                         idev->map_dir = kobject_create_and_add("maps",
 165                                                         &idev->dev->kobj);
 166                         if (!idev->map_dir)
 167                                 goto err;
 168                 }
 169                 map = kzalloc(sizeof(*map), GFP_KERNEL);
 170                 if (!map)
 171                         goto err;
 172                 kobject_init(&map->kobj, &map_attr_type);
 173                 map->mem = mem;
 174                 mem->map = map;
 175                 ret = kobject_add(&map->kobj, idev->map_dir, "map%d", mi);
 176                 if (ret)
 177                         goto err;
 178                 ret = kobject_uevent(&map->kobj, KOBJ_ADD);
 179                 if (ret)
 180                         goto err;
 181         }
 182
 183         return 0;
 184
 185 err:
 186         for (mi--; mi>=0; mi--) {
 187                 mem = &idev->info->mem[mi];
 188                 map = mem->map;
 189                 kobject_put(&map->kobj);
 190         }
 191         kobject_put(idev->map_dir);
 192         sysfs_remove_group(&idev->dev->kobj, &uio_attr_grp);
 193 err_group:
 194         dev_err(idev->dev, "error creating sysfs files (%d)\n", ret);
 195         return ret;
 196 }
 197
 198 static void uio_dev_del_attributes(struct uio_device *idev)
 199 {
 200         int mi;
 201         struct uio_mem *mem;
 202         for (mi = 0; mi < MAX_UIO_MAPS; mi++) {
 203                 mem = &idev->info->mem[mi];
 204                 if (mem->size == 0)
 205                         break;
 206                 kobject_put(&mem->map->kobj);
 207         }
 208         kobject_put(idev->map_dir);
 209         sysfs_remove_group(&idev->dev->kobj, &uio_attr_grp);
 210 }
 211
 212 static int uio_get_minor(struct uio_device *idev)
 213 {
 214         static DEFINE_MUTEX(minor_lock);
 215         int retval = -ENOMEM;
 216         int id;
 217
 218         mutex_lock(&minor_lock);
 219         if (idr_pre_get(&uio_idr, GFP_KERNEL) == 0)
 220                 goto exit;
 221
 222         retval = idr_get_new(&uio_idr, idev, &id);
 223         if (retval < 0) {
 224                 if (retval == -EAGAIN)
 225                         retval = -ENOMEM;
 226                 goto exit;
 227         }
 228         idev->minor = id & MAX_ID_MASK;
 229 exit:
 230         mutex_unlock(&minor_lock);
 231         return retval;
 232 }
 233
 234 static void uio_free_minor(struct uio_device *idev)
 235 {
 236         idr_remove(&uio_idr, idev->minor);
 237 }
 238
 239 /**
 240  * uio_event_notify - trigger an interrupt event
 241  * @info: UIO device capabilities
 242  */
 243 void uio_event_notify(struct uio_info *info)
 244 {
 245         struct uio_device *idev = info->uio_dev;
 246
 247         atomic_inc(&idev->event);
 248         wake_up_interruptible(&idev->wait);
 249         kill_fasync(&idev->async_queue, SIGIO, POLL_IN);
 250 }
 251 EXPORT_SYMBOL_GPL(uio_event_notify);
 252
 253 /**
 254  * uio_interrupt - hardware interrupt handler
 255  * @irq: IRQ number, can be UIO_IRQ_CYCLIC for cyclic timer
 256  * @dev_id: Pointer to the devices uio_device structure
 257  */
 258 static irqreturn_t uio_interrupt(int irq, void *dev_id)
 259 {
 260         struct uio_device *idev = (struct uio_device *)dev_id;
 261         irqreturn_t ret = idev->info->handler(irq, idev->info);
 262
 263         if (ret == IRQ_HANDLED)
 264                 uio_event_notify(idev->info);
 265
 266         return ret;
 267 }
 268
 269 struct uio_listener {
 270         struct uio_device *dev;
 271         s32 event_count;
 272 };
 273
 274 static int uio_open(struct inode *inode, struct file *filep)
 275 {
 276         struct uio_device *idev;
 277         struct uio_listener *listener;
 278         int ret = 0;
 279
 280         idev = idr_find(&uio_idr, iminor(inode));
 281         if (!idev)
 282                 return -ENODEV;
 283
 284         listener = kmalloc(sizeof(*listener), GFP_KERNEL);
 285         if (!listener)
 286                 return -ENOMEM;
 287
 288         listener->dev = idev;
 289         listener->event_count = atomic_read(&idev->event);
 290         filep->private_data = listener;
 291
 292         if (idev->info->open) {
 293                 if (!try_module_get(idev->owner))
 294                         return -ENODEV;
 295                 ret = idev->info->open(idev->info, inode);
 296                 module_put(idev->owner);
 297         }
 298
 299         if (ret)
 300                 kfree(listener);
 301
 302         return ret;
 303 }
 304
 305 static int uio_fasync(int fd, struct file *filep, int on)
 306 {
 307         struct uio_listener *listener = filep->private_data;
 308         struct uio_device *idev = listener->dev;
 309
 310         return fasync_helper(fd, filep, on, &idev->async_queue);
 311 }
 312
 313 static int uio_release(struct inode *inode, struct file *filep)
 314 {
 315         int ret = 0;
 316         struct uio_listener *listener = filep->private_data;
 317         struct uio_device *idev = listener->dev;
 318
 319         if (idev->info->release) {
 320                 if (!try_module_get(idev->owner))
 321                         return -ENODEV;
 322                 ret = idev->info->release(idev->info, inode);
 323                 module_put(idev->owner);
 324         }
 325         if (filep->f_flags & FASYNC)
 326                 ret = uio_fasync(-1, filep, 0);
 327         kfree(listener);
 328         return ret;
 329 }
 330
 331 static unsigned int uio_poll(struct file *filep, poll_table *wait)
 332 {
 333         struct uio_listener *listener = filep->private_data;
 334         struct uio_device *idev = listener->dev;
 335
 336         if (idev->info->irq == UIO_IRQ_NONE)
 337                 return -EIO;
 338
 339         poll_wait(filep, &idev->wait, wait);
 340         if (listener->event_count != atomic_read(&idev->event))
 341                 return POLLIN | POLLRDNORM;
 342         return 0;
 343 }
 344
 345 static ssize_t uio_read(struct file *filep, char __user *buf,
 346                         size_t count, loff_t *ppos)
 347 {
 348         struct uio_listener *listener = filep->private_data;
 349         struct uio_device *idev = listener->dev;
 350         DECLARE_WAITQUEUE(wait, current);
 351         ssize_t retval;
 352         s32 event_count;
 353
 354         if (idev->info->irq == UIO_IRQ_NONE)
 355                 return -EIO;
 356
 357         if (count != sizeof(s32))
 358                 return -EINVAL;
 359
 360         add_wait_queue(&idev->wait, &wait);
 361
 362         do {
 363                 set_current_state(TASK_INTERRUPTIBLE);
 364
 365                 event_count = atomic_read(&idev->event);
 366                 if (event_count != listener->event_count) {
 367                         if (copy_to_user(buf, &event_count, count))
 368                                 retval = -EFAULT;
 369                         else {
 370                                 listener->event_count = event_count;
 371                                 retval = count;
 372                         }
 373                         break;
 374                 }
 375
 376                 if (filep->f_flags & O_NONBLOCK) {
 377                         retval = -EAGAIN;
 378                         break;
 379                 }
 380
 381                 if (signal_pending(current)) {
 382                         retval = -ERESTARTSYS;
 383                         break;
 384                 }
 385                 schedule();
 386         } while (1);
 387
 388         __set_current_state(TASK_RUNNING);
 389         remove_wait_queue(&idev->wait, &wait);
 390
 391         return retval;
 392 }
 393
 394 static int uio_find_mem_index(struct vm_area_struct *vma)
 395 {
 396         int mi;
 397         struct uio_device *idev = vma->vm_private_data;
 398
 399         for (mi = 0; mi < MAX_UIO_MAPS; mi++) {
 400                 if (idev->info->mem[mi].size == 0)
 401                         return -1;
 402                 if (vma->vm_pgoff == mi)
 403                         return mi;
 404         }
 405         return -1;
 406 }
 407
 408 static void uio_vma_open(struct vm_area_struct *vma)
 409 {
 410         struct uio_device *idev = vma->vm_private_data;
 411         idev->vma_count++;
 412 }
 413
 414 static void uio_vma_close(struct vm_area_struct *vma)
 415 {
 416         struct uio_device *idev = vma->vm_private_data;
 417         idev->vma_count--;
 418 }
 419
 420 static struct page *uio_vma_nopage(struct vm_area_struct *vma,
 421                                    unsigned long address, int *type)
 422 {
 423         struct uio_device *idev = vma->vm_private_data;
 424         struct page* page = NOPAGE_SIGBUS;
 425
 426         int mi = uio_find_mem_index(vma);
 427         if (mi < 0)
 428                 return page;
 429
 430         if (idev->info->mem[mi].memtype == UIO_MEM_LOGICAL)
 431                 page = virt_to_page(idev->info->mem[mi].addr);
 432         else
 433                 page = vmalloc_to_page((void*)idev->info->mem[mi].addr);
 434         get_page(page);
 435         if (type)
 436                 *type = VM_FAULT_MINOR;
 437         return page;
 438 }
 439
 440 static struct vm_operations_struct uio_vm_ops = {
 441         .open = uio_vma_open,
 442         .close = uio_vma_close,
 443         .nopage = uio_vma_nopage,
 444 };
 445
 446 static int uio_mmap_physical(struct vm_area_struct *vma)
 447 {
 448         struct uio_device *idev = vma->vm_private_data;
 449         int mi = uio_find_mem_index(vma);
 450         if (mi < 0)
 451                 return -EINVAL;
 452
 453         vma->vm_flags |= VM_IO | VM_RESERVED;
 454
 455         return remap_pfn_range(vma,
 456                                vma->vm_start,
 457                                idev->info->mem[mi].addr >> PAGE_SHIFT,
 458                                vma->vm_end - vma->vm_start,
 459                                vma->vm_page_prot);
 460 }
 461
 462 static int uio_mmap_logical(struct vm_area_struct *vma)
 463 {
 464         vma->vm_flags |= VM_RESERVED;
 465         vma->vm_ops = &uio_vm_ops;
 466         uio_vma_open(vma);
 467         return 0;
 468 }
 469
 470 static int uio_mmap(struct file *filep, struct vm_area_struct *vma)
 471 {
 472         struct uio_listener *listener = filep->private_data;
 473         struct uio_device *idev = listener->dev;
 474         int mi;
 475         unsigned long requested_pages, actual_pages;
 476         int ret = 0;
 477
 478         if (vma->vm_end < vma->vm_start)
 479                 return -EINVAL;
 480
 481         vma->vm_private_data = idev;
 482
 483         mi = uio_find_mem_index(vma);
 484         if (mi < 0)
 485                 return -EINVAL;
 486
 487         requested_pages = (vma->vm_end - vma->vm_start) >> PAGE_SHIFT;
 488         actual_pages = (idev->info->mem[mi].size + PAGE_SIZE -1) >> PAGE_SHIFT;
 489         if (requested_pages > actual_pages)
 490                 return -EINVAL;
 491
 492         if (idev->info->mmap) {
 493                 if (!try_module_get(idev->owner))
 494                         return -ENODEV;
 495                 ret = idev->info->mmap(idev->info, vma);
 496                 module_put(idev->owner);
 497                 return ret;
 498         }
 499
 500         switch (idev->info->mem[mi].memtype) {
 501                 case UIO_MEM_PHYS:
 502                         return uio_mmap_physical(vma);
 503                 case UIO_MEM_LOGICAL:
 504                 case UIO_MEM_VIRTUAL:
 505                         return uio_mmap_logical(vma);
 506                 default:
 507                         return -EINVAL;
 508         }
 509 }
 510
 511 static struct file_operations uio_fops = {
 512         .owner          = THIS_MODULE,
 513         .open           = uio_open,
 514         .release        = uio_release,
 515         .read           = uio_read,
 516         .mmap           = uio_mmap,
 517         .poll           = uio_poll,
 518         .fasync         = uio_fasync,
 519 };
 520
 521 static int uio_major_init(void)
 522 {
 523         uio_major = register_chrdev(0, "uio", &uio_fops);
 524         if (uio_major < 0)
 525                 return uio_major;
 526         return 0;
 527 }
 528
 529 static void uio_major_cleanup(void)
 530 {
 531         unregister_chrdev(uio_major, "uio");
 532 }
 533
 534 static int init_uio_class(void)
 535 {
 536         int ret = 0;
 537
 538         if (uio_class != NULL) {
 539                 kref_get(&uio_class->kref);
 540                 goto exit;
 541         }
 542
 543         /* This is the first time in here, set everything up properly */
 544         ret = uio_major_init();
 545         if (ret)
 546                 goto exit;
 547
 548         uio_class = kzalloc(sizeof(*uio_class), GFP_KERNEL);
 549         if (!uio_class) {
 550                 ret = -ENOMEM;
 551                 goto err_kzalloc;
 552         }
 553
 554         kref_init(&uio_class->kref);
 555         uio_class->class = class_create(THIS_MODULE, "uio");
 556         if (IS_ERR(uio_class->class)) {
 557                 ret = IS_ERR(uio_class->class);
 558                 printk(KERN_ERR "class_create failed for uio\n");
 559                 goto err_class_create;
 560         }
 561         return 0;
 562
 563 err_class_create:
 564         kfree(uio_class);
 565         uio_class = NULL;
 566 err_kzalloc:
 567         uio_major_cleanup();
 568 exit:
 569         return ret;
 570 }
 571
 572 static void release_uio_class(struct kref *kref)
 573 {
 574         /* Ok, we cheat as we know we only have one uio_class */
 575         class_destroy(uio_class->class);
 576         kfree(uio_class);
 577         uio_major_cleanup();
 578         uio_class = NULL;
 579 }
 580
 581 static void uio_class_destroy(void)
 582 {
 583         if (uio_class)
 584                 kref_put(&uio_class->kref, release_uio_class);
 585 }
 586
 587 /**
 588  * uio_register_device - register a new userspace IO device
 589  * @owner:      module that creates the new device
 590  * @parent:     parent device
 591  * @info:       UIO device capabilities
 592  *
 593  * returns zero on success or a negative error code.
 594  */
 595 int __uio_register_device(struct module *owner,
 596                           struct device *parent,
 597                           struct uio_info *info)
 598 {
 599         struct uio_device *idev;
 600         int ret = 0;
 601
 602         if (!parent || !info || !info->name || !info->version)
 603                 return -EINVAL;
 604
 605         info->uio_dev = NULL;
 606
 607         ret = init_uio_class();
 608         if (ret)
 609                 return ret;
 610
 611         idev = kzalloc(sizeof(*idev), GFP_KERNEL);
 612         if (!idev) {
 613                 ret = -ENOMEM;
 614                 goto err_kzalloc;
 615         }
 616
 617         idev->owner = owner;
 618         idev->info = info;
 619         init_waitqueue_head(&idev->wait);
 620         atomic_set(&idev->event, 0);
 621
 622         ret = uio_get_minor(idev);
 623         if (ret)
 624                 goto err_get_minor;
 625
 626         idev->dev = device_create(uio_class->class, parent,
 627                                   MKDEV(uio_major, idev->minor),
 628                                   "uio%d", idev->minor);
 629         if (IS_ERR(idev->dev)) {
 630                 printk(KERN_ERR "UIO: device register failed\n");
 631                 ret = PTR_ERR(idev->dev);
 632                 goto err_device_create;
 633         }
 634         dev_set_drvdata(idev->dev, idev);
 635
 636         ret = uio_dev_add_attributes(idev);
 637         if (ret)
 638                 goto err_uio_dev_add_attributes;
 639
 640         info->uio_dev = idev;
 641
 642         if (idev->info->irq >= 0) {
 643                 ret = request_irq(idev->info->irq, uio_interrupt,
 644                                   idev->info->irq_flags, idev->info->name, idev);
 645                 if (ret)
 646                         goto err_request_irq;
 647         }
 648
 649         return 0;
 650
 651 err_request_irq:
 652         uio_dev_del_attributes(idev);
 653 err_uio_dev_add_attributes:
 654         device_destroy(uio_class->class, MKDEV(uio_major, idev->minor));
 655 err_device_create:
 656         uio_free_minor(idev);
 657 err_get_minor:
 658         kfree(idev);
 659 err_kzalloc:
 660         uio_class_destroy();
 661         return ret;
 662 }
 663 EXPORT_SYMBOL_GPL(__uio_register_device);
 664
 665 /**
 666  * uio_unregister_device - unregister a industrial IO device
 667  * @info:       UIO device capabilities
 668  *
 669  */
 670 void uio_unregister_device(struct uio_info *info)
 671 {
 672         struct uio_device *idev;
 673
 674         if (!info || !info->uio_dev)
 675                 return;
 676
 677         idev = info->uio_dev;
 678
 679         uio_free_minor(idev);
 680
 681         if (info->irq >= 0)
 682                 free_irq(info->irq, idev);
 683
 684         uio_dev_del_attributes(idev);
 685
 686         dev_set_drvdata(idev->dev, NULL);
 687         device_destroy(uio_class->class, MKDEV(uio_major, idev->minor));
 688         kfree(idev);
 689         uio_class_destroy();
 690
 691         return;
 692 }
 693 EXPORT_SYMBOL_GPL(uio_unregister_device);
 694
 695 static int __init uio_init(void)
 696 {
 697         return 0;
 698 }
 699
 700 static void __exit uio_exit(void)
 701 {
 702 }
 703
 704 module_init(uio_init)
 705 module_exit(uio_exit)
 706 MODULE_LICENSE("GPL v2");