OSDN Git Service

IB/hfi1: Refactor hfi_user_exp_rcv_setup() IOCTL
authorMichael J. Ruhl <michael.j.ruhl@intel.com>
Tue, 26 Sep 2017 14:04:16 +0000 (07:04 -0700)
committerDoug Ledford <dledford@redhat.com>
Fri, 29 Sep 2017 15:08:06 +0000 (11:08 -0400)
The IOCTL is a bit unwieldy.  Refactor to a common pattern.
Refactor the _TID_UPDATE IOCTL.

Reviewed-by: Ira Weiny <ira.weiny@intel.com>
Signed-off-by: Michael J. Ruhl <michael.j.ruhl@intel.com>
Signed-off-by: Dennis Dalessandro <dennis.dalessandro@intel.com>
Signed-off-by: Doug Ledford <dledford@redhat.com>
drivers/infiniband/hw/hfi1/file_ops.c

index 8618a3b..38630a6 100644 (file)
@@ -86,6 +86,8 @@ static int init_user_ctxt(struct hfi1_filedata *fd,
 static void user_init(struct hfi1_ctxtdata *uctxt);
 static int get_ctxt_info(struct hfi1_filedata *fd, unsigned long arg, u32 len);
 static int get_base_info(struct hfi1_filedata *fd, unsigned long arg, u32 len);
+static int user_exp_rcv_setup(struct hfi1_filedata *fd, unsigned long arg,
+                             u32 len);
 static int setup_base_ctxt(struct hfi1_filedata *fd,
                           struct hfi1_ctxtdata *uctxt);
 static int setup_subctxt(struct hfi1_ctxtdata *uctxt);
@@ -251,27 +253,7 @@ static long hfi1_file_ioctl(struct file *fp, unsigned int cmd,
                break;
 
        case HFI1_IOCTL_TID_UPDATE:
-               if (copy_from_user(&tinfo,
-                                  (struct hfi11_tid_info __user *)arg,
-                                  sizeof(tinfo)))
-                       return -EFAULT;
-
-               ret = hfi1_user_exp_rcv_setup(fd, &tinfo);
-               if (!ret) {
-                       /*
-                        * Copy the number of tidlist entries we used
-                        * and the length of the buffer we registered.
-                        */
-                       addr = arg + offsetof(struct hfi1_tid_info, tidcnt);
-                       if (copy_to_user((void __user *)addr, &tinfo.tidcnt,
-                                        sizeof(tinfo.tidcnt)))
-                               return -EFAULT;
-
-                       addr = arg + offsetof(struct hfi1_tid_info, length);
-                       if (copy_to_user((void __user *)addr, &tinfo.length,
-                                        sizeof(tinfo.length)))
-                               ret = -EFAULT;
-               }
+               ret = user_exp_rcv_setup(fd, arg, _IOC_SIZE(cmd));
                break;
 
        case HFI1_IOCTL_TID_FREE:
@@ -1417,6 +1399,48 @@ static int get_base_info(struct hfi1_filedata *fd, unsigned long arg, u32 len)
        return 0;
 }
 
+/**
+ * user_exp_rcv_setup - Set up the given tid rcv list
+ * @fd: file data of the current driver instance
+ * @arg: ioctl argumnent for user space information
+ * @len: length of data structure associated with ioctl command
+ *
+ * Wrapper to validate ioctl information before doing _rcv_setup.
+ *
+ */
+static int user_exp_rcv_setup(struct hfi1_filedata *fd, unsigned long arg,
+                             u32 len)
+{
+       int ret;
+       unsigned long addr;
+       struct hfi1_tid_info tinfo;
+
+       if (sizeof(tinfo) != len)
+               return -EINVAL;
+
+       if (copy_from_user(&tinfo, (void __user *)arg, (sizeof(tinfo))))
+               return -EFAULT;
+
+       ret = hfi1_user_exp_rcv_setup(fd, &tinfo);
+       if (!ret) {
+               /*
+                * Copy the number of tidlist entries we used
+                * and the length of the buffer we registered.
+                */
+               addr = arg + offsetof(struct hfi1_tid_info, tidcnt);
+               if (copy_to_user((void __user *)addr, &tinfo.tidcnt,
+                                sizeof(tinfo.tidcnt)))
+                       return -EFAULT;
+
+               addr = arg + offsetof(struct hfi1_tid_info, length);
+               if (copy_to_user((void __user *)addr, &tinfo.length,
+                                sizeof(tinfo.length)))
+                       ret = -EFAULT;
+       }
+
+       return ret;
+}
+
 static unsigned int poll_urgent(struct file *fp,
                                struct poll_table_struct *pt)
 {