hellochar.c

   1 /*
   2  *  Copyright (C) 2010  Thadeu Lima de Souza Cascardo <cascardo@holoscopio.com>
   3  *
   4  *  This program is free software; you can redistribute it and/or modify
   5  *  it under the terms of the GNU General Public License as published by
   6  *  the Free Software Foundation; either version 2 of the License, or
   7  *  (at your option) any later version.
   8  *
   9  *  This program is distributed in the hope that it will be useful,
  10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  *  GNU General Public License for more details.
  13  *
  14  *  You should have received a copy of the GNU General Public License along
  15  *  with this program; if not, write to the Free Software Foundation, Inc.,
  16  *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <linux/module.h>
  20 #include <linux/fs.h>
  21 #include <linux/cdev.h>
  22 #include <linux/slab.h>
  23 #include <linux/mutex.h>
  24 #include <asm/uaccess.h>
  25 #include <linux/circ_buf.h>
  26
  27 MODULE_LICENSE("GPL");
  28
  29 #define MAXLEN 4096
  30 static dev_t devnum;
  31 static struct cdev *dev;
  32 static const char default_greeting[] = "Hello, World!\n";
  33
  34 static struct circ_buf *hello;
  35 static struct circ_buf hello_instance;
  36 static DEFINE_MUTEX(hello_mtx);
  37
  38 static int hello_open(struct inode *ino, struct file *fp)
  39 {
  40         return 0;
  41 }
  42
  43 static ssize_t hello_read(struct file *fp, char __user *buf, size_t sz,
  44         loff_t *pos)
  45 {
  46         int r;
  47         size_t fsz;
  48         size_t ssz;
  49         size_t len;
  50         if (mutex_lock_interruptible(&hello_mtx))
  51                 return -ERESTARTSYS;
  52         len = CIRC_CNT(hello->head, hello->tail, MAXLEN);
  53         if (sz > len)
  54                 sz = len;
  55         /* First, copy at most until the limit of the buffer */
  56         fsz = min(sz, CIRC_CNT_TO_END(hello->head, hello->tail, MAXLEN));
  57         r = copy_to_user(buf, hello->buf + hello->tail, fsz);
  58         ssz = sz - fsz;
  59         if (!r && !ssz)
  60                 r = copy_to_user(buf + fsz, hello->buf, ssz);
  61         if (!r)
  62                 hello->tail = (hello->tail + sz) & ~MAXLEN;
  63         mutex_unlock(&hello_mtx);
  64         if (r)
  65                 return -EFAULT;
  66         return sz;
  67 }
  68
  69 static ssize_t hello_write(struct file *fp, const char __user *buf, size_t sz,
  70         loff_t *pos)
  71 {
  72         int r;
  73         size_t fsz;
  74         size_t ssz;
  75         size_t len;
  76         if (mutex_lock_interruptible(&hello_mtx))
  77                 return -ERESTARTSYS;
  78         len = CIRC_SPACE(hello->head, hello->tail, MAXLEN);
  79         if (sz > len)
  80                 sz = len;
  81         fsz = min(sz, CIRC_SPACE_TO_END(hello->head, hello->tail, MAXLEN));
  82         r = copy_from_user(hello->buf + hello->head, buf, fsz);
  83         ssz = sz - fsz;
  84         if (!r && !ssz)
  85                 r = copy_from_user(hello->buf, buf + fsz, ssz);
  86         if (!r)
  87                 hello->head = (hello->head + sz) & ~MAXLEN;
  88         mutex_unlock(&hello_mtx);
  89         if (r)
  90                 return -EFAULT;
  91         return sz;
  92 }
  93
  94 static int hello_release(struct inode *ino, struct file *fp)
  95 {
  96         return 0;
  97 }
  98
  99 static const struct file_operations hello_fops = {
 100         .owner = THIS_MODULE,
 101         .open = hello_open,
 102         .release = hello_release,
 103         .read = hello_read,
 104         .write = hello_write,
 105 };
 106
 107 static int __init ch_init(void)
 108 {
 109         int r = 0;
 110         hello = &hello_instance;
 111         hello->buf = kzalloc(sizeof(*hello) + MAXLEN, GFP_KERNEL);
 112         if (!hello) {
 113                 r = -ENOMEM;
 114                 goto out;
 115         }
 116         memcpy(hello->buf, default_greeting, sizeof(default_greeting));
 117         hello->head = sizeof(default_greeting);
 118         r = alloc_chrdev_region(&devnum, 0, 256, "hello");
 119         if (r)
 120                 goto reg_out;
 121         dev = cdev_alloc();
 122         if (!dev) {
 123                 r = -ENOMEM;
 124                 goto cdev_out;
 125         }
 126         dev->ops = &hello_fops;
 127         r = cdev_add(dev, devnum, 256);
 128         if (r)
 129                 goto add_out;
 130         return 0;
 131 add_out:
 132         kfree(dev);
 133 cdev_out:
 134         unregister_chrdev_region(devnum, 256);
 135 reg_out:
 136         kfree(hello);
 137 out:
 138         return r;
 139 }
 140
 141 static void __exit ch_exit(void)
 142 {
 143         cdev_del(dev);
 144         unregister_chrdev_region(devnum, 256);
 145         kfree(hello->buf);
 146 }
 147
 148 module_init(ch_init);
 149 module_exit(ch_exit);