home *** CD-ROM | disk | FTP | other *** search
/ ftp.parl.clemson.edu / 2015-02-07.ftp.parl.clemson.edu.tar / ftp.parl.clemson.edu / pub / pvfs2 / orangefs-2.8.3-20110323.tar.gz / orangefs-2.8.3-20110323.tar / orangefs / test / io / dev / dev-mod.c < prev    next >
C/C++ Source or Header  |  2004-07-28  |  5KB  |  225 lines
  1. /*
  2.  * (C) 2001 Clemson University and The University of Chicago
  3.  *
  4.  * See COPYING in top-level directory.
  5.  */
  6.  
  7. /* this is just a trivial module that provides a character device for 
  8.  * testing the pvfs2/src/io/dev/ code
  9.  */
  10.  
  11. #include <linux/module.h>
  12. #include <linux/config.h>
  13. #include <linux/kernel.h>
  14. #include <linux/version.h>
  15. #include <linux/fs.h>
  16. #include <linux/major.h>
  17. #include <linux/smp_lock.h>
  18. #include <linux/poll.h>
  19. #include <linux/slab.h>
  20. #include <asm/uaccess.h>
  21.  
  22. #include "pint-dev-shared.h"
  23.  
  24. #ifdef MODULE_LICENSE
  25. MODULE_LICENSE("GPL");
  26. #endif
  27.  
  28. EXPORT_NO_SYMBOLS;
  29.  
  30. ssize_t    pdev_read(struct file *, char *, size_t, loff_t *);
  31. ssize_t    pdev_writev(struct file *, const struct iovec *, unsigned long, 
  32.     loff_t *);
  33. int    pdev_open(struct inode *, struct file *);
  34. int    pdev_release(struct inode *, struct file *);
  35. int    pdev_ioctl(struct inode *, struct file *, unsigned int, unsigned long);
  36. unsigned int pdev_poll(struct file *my_file, struct poll_table_struct *wait);
  37. static int __init pdev_init(void);
  38. static void __exit pdev_exit(void);
  39.  
  40. static struct file_operations pdev_fops = {
  41.     read:        pdev_read,
  42.     writev:        pdev_writev,
  43.     open:        pdev_open,
  44.     release:           pdev_release,
  45.     ioctl:        pdev_ioctl,
  46.     poll:        pdev_poll,
  47. };
  48.  
  49. static int pdev_major = 0;
  50. /* note: just picking these arbitrarily for testing */
  51. int32_t pdev_magic = 1234;
  52. int32_t pdev_max_upsize = 1024;
  53. int32_t pdev_max_downsize = 1024;
  54.  
  55. static int __init pdev_init(void)
  56. {
  57.  
  58.     int ret;
  59.    
  60. #ifdef HAVE_DEVFS
  61.     if ((ret = devfs_register_chrdev(0, "pvfs2-req", &pdev_fops)) < 0)
  62. #else
  63.     if ((ret = register_chrdev(0, "pvfs2-req", &pdev_fops)) < 0)
  64. #endif
  65.     {
  66.     printk("PDEV: failed to assign major number\n");
  67.     return -ENODEV;
  68.     }
  69.  
  70.     pdev_major = ret;
  71.  
  72. #ifdef HAVE_DEVFS
  73.     devfs_handle = devfs_register(NULL, "pvfs2-req", DEVFS_FL_DEFAULT,
  74.     pdev_major, 0, (S_IFCHR | S_IRUSR | S_IWUSR), &pdev_fops, NULL);
  75. #endif
  76.  
  77.     printk("PDEV: init successful\n");
  78.  
  79.     return 0;
  80. }
  81.  
  82. static void __exit pdev_exit(void)
  83. {
  84.  
  85. #ifdef HAVE_DEVFS
  86.     devfs_unregister(devfs_handle);
  87.     devfs_unregister_chrdev(pdev_major, "pvfs2-req");
  88. #else
  89.     unregister_chrdev(pdev_major, "pvfs2-req");
  90. #endif
  91.  
  92.     printk("PDEV: exit successful\n");
  93.     return;
  94. }
  95.  
  96. int pdev_open(struct inode *inode, struct file *filp)
  97. {
  98.     printk("PDEV: pdev_open()\n");
  99.     MOD_INC_USE_COUNT;
  100.     return(0);
  101. }
  102.  
  103. int pdev_release(struct inode *inode, struct file *filp)
  104. {
  105.     printk("PDEV: pdev_release()\n");
  106.     MOD_DEC_USE_COUNT;
  107.     return(0);
  108. }
  109.  
  110. int pdev_ioctl(struct inode *inode, 
  111.           struct file *flip,
  112.           unsigned int command, 
  113.           unsigned long arg)
  114. {
  115.     printk("PDEV: pdev_ioctl()\n");
  116.  
  117.     switch(command)
  118.     {
  119.     case(PVFS_DEV_GET_MAGIC):
  120.         copy_to_user((void*)arg, &pdev_magic, sizeof(int32_t)); 
  121.         return(0);
  122.     case(PVFS_DEV_GET_MAX_UPSIZE):
  123.         copy_to_user((void*)arg, &pdev_max_upsize, sizeof(int32_t)); 
  124.         return(0);
  125.     case(PVFS_DEV_GET_MAX_DOWNSIZE):
  126.         copy_to_user((void*)arg, &pdev_max_downsize, sizeof(int32_t)); 
  127.         return(0);
  128.     default:
  129.         return(-ENOSYS);
  130.     }
  131.  
  132.     return(-ENOSYS);
  133. } 
  134.  
  135. ssize_t    pdev_read(struct file *filp, char * buf, size_t size, loff_t *offp)
  136. {
  137.     char test_string[] = "Hello world.";
  138.     int64_t test_tag = 5;
  139.     void* tmp_buf = buf;
  140.  
  141.     printk("PDEV: pdev_read()\n");
  142.  
  143.     if(size < (strlen(test_string) + 1 + sizeof(int32_t) + sizeof(int64_t)))
  144.     {
  145.     return(-EMSGSIZE);
  146.     }
  147.  
  148.     /* copy out magic number */
  149.     copy_to_user(tmp_buf, &pdev_magic, sizeof(int32_t));
  150.     tmp_buf = (void*)((unsigned long)tmp_buf + sizeof(int32_t));
  151.     /* copy out tag */
  152.     copy_to_user(tmp_buf, &test_tag, sizeof(int64_t));
  153.     tmp_buf = (void*)((unsigned long)tmp_buf + sizeof(int64_t));
  154.     /* copy out message payload */
  155.     copy_to_user(tmp_buf, test_string, (strlen(test_string) + 1));
  156.  
  157.     return(strlen(test_string) + 1 + sizeof(int32_t) + sizeof(int64_t));
  158. }
  159.  
  160. ssize_t    pdev_writev(struct file *filp, const struct iovec* iov, 
  161.     unsigned long count, loff_t *offp)
  162. {
  163.     void* buffer = NULL;
  164.     void* offset;
  165.     int remaining = pdev_max_downsize;
  166.     int i;
  167.     int32_t* magic;
  168.     int64_t* tag;
  169.     char* payload = NULL;
  170.  
  171.     printk("PDEV: pdev_writev()\n");
  172.  
  173.     /* for simplicity's sake, lets always dump to a contiguous buffer and 
  174.      * then pull the message apart- logic is a bit complicated for handling 
  175.      * iovec directly 
  176.      */
  177.     buffer = kmalloc(pdev_max_downsize, GFP_KERNEL);
  178.     if(!buffer)
  179.     return(-ENOMEM);
  180.  
  181.     offset = buffer;
  182.     for(i=0; i<count; i++)
  183.     {
  184.     if(iov[i].iov_len > remaining)
  185.         return(-EMSGSIZE);
  186.     copy_from_user(offset, iov[i].iov_base, iov[i].iov_len);
  187.     remaining -= iov[i].iov_len;
  188.     offset = (void*)((unsigned long)offset + iov[i].iov_len);
  189.     }
  190.  
  191.     /* now, lets see what we got */
  192.     magic = (int32_t*)buffer;
  193.     tag = (int64_t*)((unsigned long)buffer + sizeof(int32_t));
  194.     payload = (char*)((unsigned long)tag + sizeof(int64_t));
  195.  
  196.     if(*magic != pdev_magic)
  197.     return(-EINVAL);
  198.  
  199.     printk("PDEV: magic: %d, tag: %d, payload: %s\n", (int)*magic, (int)*tag, 
  200.     payload);
  201.  
  202.     kfree(buffer);
  203.  
  204.     return(pdev_max_downsize - remaining);
  205. }
  206.  
  207. unsigned int pdev_poll(struct file *my_file, struct 
  208.     poll_table_struct *wait){
  209.  
  210.     /* the test module is always ready for IO ... */
  211.     return(POLLOUT|POLLWRNORM|POLLIN|POLLRDNORM);
  212. }
  213.  
  214. module_init(pdev_init);
  215. module_exit(pdev_exit);
  216.  
  217. /*
  218.  * Local variables:
  219.  *  c-indent-level: 4
  220.  *  c-basic-offset: 4
  221.  * End:
  222.  *
  223.  * vim: ts=8 sts=4 sw=4 expandtab
  224.  */
  225.