1 files changed, 7 insertions, 18 deletions

diff --git a/net/socket.c b/net/socket.c
index 92fab9e1c602..aeeab388cc38 100644
--- a/net/socket.c
+++ b/net/socket.c
@@ -1594,16 +1594,11 @@ asmlinkage long sys_sendto(int fd, void __user *buff, size_t len,
         struct msghdr msg;
         struct iovec iov;
         int fput_needed;
-        struct file *sock_file;
 
-        sock_file = fget_light(fd, &fput_needed);
-        err = -EBADF;
-        if (!sock_file)
+        sock = sockfd_lookup_light(fd, &err, &fput_needed);
+        if (!sock)
                 goto out;
 
-        sock = sock_from_file(sock_file, &err);
-        if (!sock)
-                goto out_put;
         iov.iov_base = buff;
         iov.iov_len = len;
         msg.msg_name = NULL;
@@ -1625,7 +1620,7 @@ asmlinkage long sys_sendto(int fd, void __user *buff, size_t len,
         err = sock_sendmsg(sock, &msg, len);
 
 out_put:
-        fput_light(sock_file, fput_needed);
+        fput_light(sock->file, fput_needed);
 out:
         return err;
 }
@@ -1654,17 +1649,11 @@ asmlinkage long sys_recvfrom(int fd, void __user *ubuf, size_t size,
         struct msghdr msg;
         char address[MAX_SOCK_ADDR];
         int err, err2;
-        struct file *sock_file;
         int fput_needed;
 
-        sock_file = fget_light(fd, &fput_needed);
-        err = -EBADF;
-        if (!sock_file)
-                goto out;
-
-        sock = sock_from_file(sock_file, &err);
+        sock = sockfd_lookup_light(fd, &err, &fput_needed);
         if (!sock)
-                goto out_put;
+                goto out;
 
         msg.msg_control = NULL;
         msg.msg_controllen = 0;
@@ -1683,8 +1672,8 @@ asmlinkage long sys_recvfrom(int fd, void __user *ubuf, size_t size,
                 if (err2 < 0)
                         err = err2;
         }
-out_put:
-        fput_light(sock_file, fput_needed);
+
+        fput_light(sock->file, fput_needed);
 out:
         return err;
 }