[<prev] [next>] [thread-next>] [day] [month] [year] [list]
Message-ID: <20100304032612.GB9016@home.unix.ba>
Date: Thu, 4 Mar 2010 04:26:13 +0100
From: Kenan Kalajdzic <kenan@...x.ba>
To: netdev@...r.kernel.org
Cc: davem@...emloft.net
Subject: [PATCH] socket: Merge getsockname and getpeername into a single
function
The code of getsockname is almost identical to getpeername. This patch removes
duplicate code and merges both functions into a single common function.
Signed-off-by: Kenan Kalajdzic <kenan@...x.ba>
---
net/socket.c | 50 +++++++++++++++++---------------------------------
1 files changed, 17 insertions(+), 33 deletions(-)
diff --git a/net/socket.c b/net/socket.c
index 769c386..526e1f5 100644
--- a/net/socket.c
+++ b/net/socket.c
@@ -1593,12 +1593,12 @@ out:
}
/*
- * Get the local address ('name') of a socket object. Move the obtained
- * name to user space.
+ * Common code for getsockname and getpeername system calls
+ * Get local or remote address ('name') of a socket object.
+ * Move the obtained name to user space.
*/
-
-SYSCALL_DEFINE3(getsockname, int, fd, struct sockaddr __user *, usockaddr,
- int __user *, usockaddr_len)
+static int common_getsockpeername(int fd, struct sockaddr __user *usockaddr,
+ int __user * usockaddr_len, int remote)
{
struct socket *sock;
struct sockaddr_storage address;
@@ -1608,50 +1608,34 @@ SYSCALL_DEFINE3(getsockname, int, fd, struct sockaddr __user *, usockaddr,
if (!sock)
goto out;
- err = security_socket_getsockname(sock);
+ if (remote)
+ err = security_socket_getpeername(sock);
+ else
+ err = security_socket_getsockname(sock);
if (err)
goto out_put;
- err = sock->ops->getname(sock, (struct sockaddr *)&address, &len, 0);
+ err = sock->ops->getname(sock, (struct sockaddr *)&address, &len, remote);
if (err)
goto out_put;
- err = move_addr_to_user((struct sockaddr *)&address, len, usockaddr, usockaddr_len);
+ err = move_addr_to_user((struct sockaddr *)&address, len, usockaddr, usockaddr_len);
out_put:
fput_light(sock->file, fput_needed);
out:
return err;
}
-/*
- * Get the remote address ('name') of a socket object. Move the obtained
- * name to user space.
- */
+SYSCALL_DEFINE3(getsockname, int, fd, struct sockaddr __user *, usockaddr,
+ int __user *, usockaddr_len)
+{
+ return common_getsockpeername(fd, usockaddr, usockaddr_len, 0);
+}
SYSCALL_DEFINE3(getpeername, int, fd, struct sockaddr __user *, usockaddr,
int __user *, usockaddr_len)
{
- struct socket *sock;
- struct sockaddr_storage address;
- int len, err, fput_needed;
-
- sock = sockfd_lookup_light(fd, &err, &fput_needed);
- if (sock != NULL) {
- err = security_socket_getpeername(sock);
- if (err) {
- fput_light(sock->file, fput_needed);
- return err;
- }
-
- err =
- sock->ops->getname(sock, (struct sockaddr *)&address, &len,
- 1);
- if (!err)
- err = move_addr_to_user((struct sockaddr *)&address, len, usockaddr,
- usockaddr_len);
- fput_light(sock->file, fput_needed);
- }
- return err;
+ return common_getsockpeername(fd, usockaddr, usockaddr_len, 1);
}
/*
--
1.6.4
--
To unsubscribe from this list: send the line "unsubscribe netdev" in
the body of a message to majordomo@...r.kernel.org
More majordomo info at http://vger.kernel.org/majordomo-info.html
Powered by blists - more mailing lists