Fix ioctl return values:
 * `-EFAULT' when unable to fetch user supplied arguments,
 * `-ESRCH' when no target process is found,
 * `-ESTALE' when unable to get `struct files',
 * `-EBADF' when unable to get `struct file' for fd.
---
 lib/librte_vhost/eventfd_link/eventfd_link.c | 46 ++++++++++++++++++----------
 1 file changed, 30 insertions(+), 16 deletions(-)

diff --git a/lib/librte_vhost/eventfd_link/eventfd_link.c 
b/lib/librte_vhost/eventfd_link/eventfd_link.c
index 2a29999..0a06594 100644
--- a/lib/librte_vhost/eventfd_link/eventfd_link.c
+++ b/lib/librte_vhost/eventfd_link/eventfd_link.c
@@ -92,33 +92,38 @@ eventfd_link_ioctl_copy(unsigned long arg)
        struct files_struct *files;
        struct fdtable *fdt;
        struct eventfd_copy eventfd_copy;
+       long ret = -EFAULT;

-       if (copy_from_user(&eventfd_copy, argp,
-               sizeof(struct eventfd_copy)))
-               return -EFAULT;
+       if (copy_from_user(&eventfd_copy, argp, sizeof(struct eventfd_copy)))
+               goto out;

        /*
         * Find the task struct for the target pid
         */
+       ret = -ESRCH;
+
        task_target =
-               pid_task(find_vpid(eventfd_copy.target_pid), PIDTYPE_PID);
+               get_pid_task(find_vpid(eventfd_copy.target_pid), PIDTYPE_PID);
        if (task_target == NULL) {
-               pr_debug("Failed to get mem ctx for target pid\n");
-               return -EFAULT;
+               pr_info("Unable to find pid %d\n", eventfd_copy.target_pid);
+               goto out;
        }

+       ret = -ESTALE;
        files = get_files_struct(current);
        if (files == NULL) {
-               pr_debug("Failed to get files struct\n");
-               return -EFAULT;
+               pr_info("Failed to get current files struct\n");
+               goto out_task;
        }

+       ret = -EBADF;
        file = fget_from_files(files, eventfd_copy.source_fd);
-       put_files_struct(files);

        if (file == NULL) {
-               pr_debug("Failed to get file from source pid\n");
-               return 0;
+               pr_info("Failed to get fd %d from source\n",
+                       eventfd_copy.source_fd);
+               put_files_struct(files);
+               goto out_task;
        }

        /*
@@ -131,22 +136,27 @@ eventfd_link_ioctl_copy(unsigned long arg)
        fdt->fd[eventfd_copy.source_fd] = NULL;
        spin_unlock(&files->file_lock);

+       put_files_struct(files);
+
        /*
         * Find the file struct associated with the target fd.
         */

+       ret = -ESTALE;
        files = get_files_struct(task_target);
        if (files == NULL) {
-               pr_debug("Failed to get files struct\n");
-               return -EFAULT;
+               pr_info("Failed to get target files struct\n");
+               goto out_task;
        }

+       ret = -EBADF;
        file = fget_from_files(files, eventfd_copy.target_fd);
        put_files_struct(files);

        if (file == NULL) {
-               pr_debug("Failed to get file from target pid\n");
-               return 0;
+               pr_info("Failed to get fd %d from target\n",
+                       eventfd_copy.target_fd);
+               goto out_task;
        }

        /*
@@ -155,8 +165,12 @@ eventfd_link_ioctl_copy(unsigned long arg)
         */

        fd_install(eventfd_copy.source_fd, file);
+       ret = 0;

-       return 0;
+out_task:
+       put_task_struct(task_target);
+out:
+       return ret;
 }

 static long
-- 
1.9.1

Reply via email to