From 24e6690ee874d320a34cab086af06f2522f9bb36 Mon Sep 17 00:00:00 2001
From: Lucas Bajolet <r4pass@hotmail.com>
Date: Mon, 7 Apr 2014 11:23:11 -0400
Subject: [PATCH] lib/socket: Added sockopts to socket lib

Signed-off-by: Lucas Bajolet <r4pass@hotmail.com>
---
 lib/socket/socket.nit   |    2 ++
 lib/socket/socket_c.nit |   32 +++++++++++++++++++++++++++++++-
 2 files changed, 33 insertions(+), 1 deletion(-)

diff --git a/lib/socket/socket.nit b/lib/socket/socket.nit
index e295860..b8f4acc 100644
--- a/lib/socket/socket.nit
+++ b/lib/socket/socket.nit
@@ -39,6 +39,7 @@ class Socket
 			still_alive = false
 			return
 		end
+		socket.setsockopt(new FFSocketOptLevels.socket, new FFSocketOptNames.reuseaddr, 1)
 		var hostname = socket.gethostbyname(thost)
 		addrin = new FFSocketAddrIn.with_hostent(hostname, tport)
 		address = addrin.address
@@ -53,6 +54,7 @@ class Socket
 			still_alive = false
 			return
 		end
+		socket.setsockopt(new FFSocketOptLevels.socket, new FFSocketOptNames.reuseaddr, 1)
 		addrin = new FFSocketAddrIn.with(tport, new FFSocketAddressFamilies.af_inet)
 		address = addrin.address
 		port = addrin.port
diff --git a/lib/socket/socket_c.nit b/lib/socket/socket_c.nit
index 984bf63..11f518e 100644
--- a/lib/socket/socket_c.nit
+++ b/lib/socket/socket_c.nit
@@ -130,6 +130,15 @@ extern FFSocket `{ S_DESCRIPTOR* `}
 		return NativeString_to_s(c);
 	`}
 
+	# Sets an option for the socket
+	fun setsockopt(level: FFSocketOptLevels, option_name: FFSocketOptNames, option_value: Int) `{
+		int err = setsockopt(*recv, level, option_name, &option_value, sizeof(int));
+		if(err != 0){
+			perror("Error on setsockopts : ");
+			exit(1);
+		}
+	`}
+
 	fun bind(addrIn: FFSocketAddrIn): Int `{ return bind(*recv, (S_ADDR*)addrIn, sizeof(*addrIn)); `}
 	fun listen(size: Int): Int `{ return listen(*recv, size); `}
 
@@ -311,7 +320,28 @@ extern FFSocketProtocolFamilies `{ int `}
 	new pf_inet6 `{ return PF_INET6; `}
 	new pf_max `{ return PF_MAX; `}
 end
-
+# Level on which to set options
+extern FFSocketOptLevels `{ int `}
+	# Dummy for IP (As defined in C)
+	new ip `{ return IPPROTO_IP;`}
+	# Control message protocol
+	new icmp `{ return IPPROTO_ICMP;`}
+	# Use TCP
+	new tcp `{ return IPPROTO_TCP; `}
+	# Socket level options
+	new socket `{ return SOL_SOCKET; `}
+end
+# Options for socket, use with setsockopt
+extern FFSocketOptNames `{ int `}
+	# Enables debugging information
+	new debug `{ return SO_DEBUG; `}
+	# Authorizes the broadcasting of messages
+	new broadcast `{ return SO_BROADCAST; `}
+	# Authorizes the reuse of the local address
+	new reuseaddr `{ return SO_REUSEADDR; `}
+	# Authorizes the use of keep-alive packets in a connection
+	new keepalive `{ return SO_KEEPALIVE; `}
+end
 # Used for the poll function of a socket, mix several Poll values to check for events on more than one type of event
 extern FFSocketPollValues `{ int `}
 	new pollin `{ return POLLIN; `}           # Data other than high-priority data may be read without blocking.
-- 
1.7.9.5