From: Erik van der Kouwe <erik@minix3.org>
Date: Fri, 4 Dec 2009 07:26:56 +0000 (+0000)
Subject: Cleanup getsockopt and add SO_TYPE
X-Git-Tag: v3.1.6~171
X-Git-Url: http://zhaoyanbai.com/repos/dnssec-keyfromlabel.html?a=commitdiff_plain;h=150dfbe96d0100dad8e6d6b5b65bc99e1c5e5800;p=minix.git

Cleanup getsockopt and add SO_TYPE
---

diff --git a/include/sys/socket.h b/include/sys/socket.h
index c51938e92..cf7221c4a 100644
--- a/include/sys/socket.h
+++ b/include/sys/socket.h
@@ -24,6 +24,7 @@ sys/socket.h
 #define SO_DEBUG	0x0001
 #define SO_REUSEADDR	0x0004
 #define SO_KEEPALIVE	0x0008
+#define SO_TYPE	0x0010	/* get socket type, SOCK_STREAM or SOCK_DGRAM */
 
 #define SO_SNDBUF	0x1001	/* send buffer size */
 #define SO_RCVBUF	0x1002	/* receive buffer size */
diff --git a/lib/ip/getsockopt.c b/lib/ip/getsockopt.c
index f72106264..d40deddcc 100644
--- a/lib/ip/getsockopt.c
+++ b/lib/ip/getsockopt.c
@@ -19,6 +19,8 @@ static int _tcp_getsockopt(int socket, int level, int option_name,
 	void *_RESTRICT option_value, socklen_t *_RESTRICT option_len);
 static int _udp_getsockopt(int socket, int level, int option_name,
 	void *_RESTRICT option_value, socklen_t *_RESTRICT option_len);
+static void getsockopt_copy(void *return_value, size_t return_len,
+	void *_RESTRICT option_value, socklen_t *_RESTRICT option_len);
 
 int getsockopt(int socket, int level, int option_name,
         void *_RESTRICT option_value, socklen_t *_RESTRICT option_len)
@@ -58,6 +60,19 @@ int getsockopt(int socket, int level, int option_name,
 	return -1;
 }
 
+static void getsockopt_copy(void *return_value, size_t return_len,
+	void *_RESTRICT option_value, socklen_t *_RESTRICT option_len)
+{
+	/* copy as much data as possible */
+	if (*option_len < return_len)
+		memcpy(option_value, return_value, *option_len);
+	else
+		memcpy(option_value, return_value, return_len);
+
+	/* return length */
+	*option_len = return_len;
+}
+
 static int _tcp_getsockopt(int socket, int level, int option_name,
 	void *_RESTRICT option_value, socklen_t *_RESTRICT option_len)
 {
@@ -65,56 +80,41 @@ static int _tcp_getsockopt(int socket, int level, int option_name,
 
 	if (level == SOL_SOCKET && option_name == SO_KEEPALIVE)
 	{
-		i= 1;	/* Keepalive is always on */
-		if (*option_len < sizeof(i))
-			memcpy(option_value, &i, *option_len);
-		else
-			memcpy(option_value, &i, sizeof(i));
-		*option_len= sizeof(i);
+		i = 1;	/* Keepalive is always on */
+		getsockopt_copy(&i, sizeof(i), option_value, option_len);
 		return 0;
 	}
 	if (level == SOL_SOCKET && option_name == SO_ERROR)
 	{
-		r= ioctl(socket, NWIOTCPGERROR, &err);
+		r = ioctl(socket, NWIOTCPGERROR, &err);
 		if (r != 0)
 			return r;
-		if (*option_len < sizeof(err))
-			memcpy(option_value, &err, *option_len);
-		else
-			memcpy(option_value, &err, sizeof(err));
-		*option_len= sizeof(err);
+
+		getsockopt_copy(&err, sizeof(err), option_value, option_len);
 		return 0;
 	}
 	if (level == SOL_SOCKET && option_name == SO_RCVBUF)
 	{
-		i= 32*1024;	/* Receive buffer in the current
-				 * implementation
-				 */
-		if (*option_len < sizeof(i))
-			memcpy(option_value, &i, *option_len);
-		else
-			memcpy(option_value, &i, sizeof(i));
-		*option_len= sizeof(i);
+		i = 32 * 1024;	/* Receive buffer in the current implementation */
+		getsockopt_copy(&i, sizeof(i), option_value, option_len);
 		return 0;
 	}
 	if (level == SOL_SOCKET && option_name == SO_SNDBUF)
 	{
-		i= 32*1024;	/* Send buffer in the current implementation */
-		if (*option_len < sizeof(i))
-			memcpy(option_value, &i, *option_len);
-		else
-			memcpy(option_value, &i, sizeof(i));
-		*option_len= sizeof(i);
+		i = 32 * 1024;	/* Send buffer in the current implementation */
+		getsockopt_copy(&i, sizeof(i), option_value, option_len);
+		return 0;
+	}
+	if (level == SOL_SOCKET && option_name == SO_TYPE)
+	{
+		i = SOCK_STREAM;	/* this is a TCP socket */
+		getsockopt_copy(&i, sizeof(i), option_value, option_len);
 		return 0;
 	}
 	if (level == IPPROTO_TCP && option_name == TCP_NODELAY)
 	{
-		i= 0;	/* nodelay is always off */
-		if (*option_len < sizeof(i))
-			memcpy(option_value, &i, *option_len);
-		else
-			memcpy(option_value, &i, sizeof(i));
-		*option_len= sizeof(i);
+		i = 0;	/* nodelay is always off */
+		getsockopt_copy(&i, sizeof(i), option_value, option_len);
 		return 0;
 	}
 #if DEBUG
@@ -131,6 +131,12 @@ static int _udp_getsockopt(int socket, int level, int option_name,
 {
 	int i;
 
+	if (level == SOL_SOCKET && option_name == SO_TYPE)
+	{
+		i = SOCK_DGRAM;	/* this is a TCP socket */
+		getsockopt_copy(&i, sizeof(i), option_value, option_len);
+		return 0;
+	}
 #if DEBUG
 	fprintf(stderr, "_udp_getsocketopt: level %d, name %d\n",
 		level, option_name);