USe interruptible version of down.

[cascardo/kernel/samples/char2/.git] / hellochar.c

diff --git a/hellochar.c b/hellochar.c
index 576968b..628f53c 100644 (file)
--- a/hellochar.c
+++ b/hellochar.c
@@ -18,23 +18,127 @@
 
 #include <linux/module.h>
 #include <linux/fs.h>
+#include <linux/cdev.h>
+#include <linux/slab.h>
+#include <linux/semaphore.h>
+#include <asm/uaccess.h>
 
 MODULE_LICENSE("GPL");
-static int major = 10;
-module_param_named(major, major, int, S_IRUGO | S_IWUSR);
 
-static int __init ch_init(void)
+#define MAXLEN 4000
+static dev_t devnum;
+static struct cdev *dev;
+static const char default_greeting[] = "Hello, World!\n";
+
+struct hello_buffer {
+       size_t len;
+       char buffer[0];
+};
+static struct hello_buffer *hello;
+static DECLARE_MUTEX(hello_mtx);
+
+static int hello_open(struct inode *ino, struct file *fp)
+{
+       if (down_interruptible(&hello_mtx))
+               return -ERESTARTSYS;
+       if (fp->f_flags & O_TRUNC) {
+               memset(hello->buffer, 0, MAXLEN);
+               hello->len = 0;
+       }
+       if (fp->f_flags & O_APPEND)
+               fp->f_pos = hello->len;
+       up(&hello_mtx);
+       return 0;
+}
+
+static ssize_t hello_read(struct file *fp, char __user *buf, size_t sz,
+       loff_t *pos)
+{
+       int r;
+       if (down_interruptible(&hello_mtx))
+               return -ERESTARTSYS;
+       if (sz + *pos > hello->len)
+               sz = hello->len - *pos;
+       r = copy_to_user(buf, hello->buffer + *pos, sz);
+       up(&hello_mtx);
+       if (r)
+               return -EFAULT;
+       *pos += sz;
+       return sz;
+}
+
+static ssize_t hello_write(struct file *fp, const char __user *buf, size_t sz,
+       loff_t *pos)
 {
        int r;
-       r = register_chrdev_region(MKDEV(major, 0), 256, "hello");
+       if (down_interruptible(&hello_mtx))
+               return -ERESTARTSYS;
+       if (sz + *pos > MAXLEN)
+               sz = MAXLEN - *pos;
+       r = copy_from_user(hello->buffer + *pos, buf, sz);
+       if (r) {
+               up(&hello_mtx);
+               return -EFAULT;
+       }
+       *pos += sz;
+       if (hello->len < *pos)
+               hello->len = *pos;
+       up(&hello_mtx);
+       return sz;
+}
+
+static int hello_release(struct inode *ino, struct file *fp)
+{
+       return 0;
+}
+
+static const struct file_operations hello_fops = {
+       .owner = THIS_MODULE,
+       .open = hello_open,
+       .release = hello_release,
+       .read = hello_read,
+       .write = hello_write,
+};
+
+static int __init ch_init(void)
+{
+       int r = 0;
+       hello = kzalloc(sizeof(*hello) + MAXLEN, GFP_KERNEL);
+       if (!hello) {
+               r = -ENOMEM;
+               goto out;
+       }
+       memcpy(hello->buffer, default_greeting, sizeof(default_greeting));
+       hello->len = sizeof(default_greeting);
+       r = alloc_chrdev_region(&devnum, 0, 256, "hello");
+       if (r)
+               goto reg_out;
+       dev = cdev_alloc();
+       if (!dev) {
+               r = -ENOMEM;
+               goto cdev_out;
+       }
+       dev->ops = &hello_fops;
+       r = cdev_add(dev, devnum, 256);
        if (r)
-               return r;
+               goto add_out;
+       printk(KERN_DEBUG "Allocate major %d\n", MAJOR(devnum));
        return 0;
+add_out:
+       kfree(dev);
+cdev_out:
+       unregister_chrdev_region(devnum, 256);
+reg_out:
+       kfree(hello);
+out:
+       return r;
 }
 
 static void __exit ch_exit(void)
 {
-       unregister_chrdev_region(MKDEV(major, 0), 256);
+       cdev_del(dev);
+       unregister_chrdev_region(devnum, 256);
+       kfree(hello);
 }
 
 module_init(ch_init);