From 6737b7efe2f4ef7ad4e8f6b2716ada469ef3913d Mon Sep 17 00:00:00 2001
From: Kaz Kylheku <kaz@kylheku.com>
Date: Tue, 15 Mar 2016 04:47:16 -0700
Subject: Implement socket timeouts.

* lib.c (timeout_error_s): New symbol variable.
(obj_init): Intern timeout-error, init new variable.

* lib.h (timeout_error_s): Declared.

* socket.c (sock_timeout, sock_send_timeout,
sock_recv_timeout): New static functions.
(sock_load_init): Register sock-send-timeout and
sock-recv-timeout intrinsics.

* stream.c (stdio_maybe_read_error, stdio_maybe_error):
Convert EAGAIN into timeout_error_s.

* txr.1: Documented.
---
 socket.c | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

(limited to 'socket.c')

diff --git a/socket.c b/socket.c
index d7d49b4c..b114f704 100644
--- a/socket.c
+++ b/socket.c
@@ -814,6 +814,36 @@ static val sock_shutdown(val sock, val how)
   return t;
 }
 
+#if defined SO_SNDTIMEO && defined SO_RCVTIMEO
+static val sock_timeout(val sock, val usec, val name, int which)
+{
+  cnum fd = c_num(stream_fd(sock));
+  cnum u = c_num(usec);
+  struct timeval tv;
+
+  tv.tv_sec = u / 1000000;
+  tv.tv_usec = u % 1000000;
+
+  if (setsockopt(fd, SOL_SOCKET, which, &tv, sizeof tv) != 0)
+    uw_throwf(socket_error_s, lit("~a failed on ~s: ~d/~s"),
+	      name, sock, num(errno),
+	      string_utf8(strerror(errno)), nao);
+
+  return sock;
+}
+
+static val sock_send_timeout(val sock, val usec)
+{
+  return sock_timeout(sock, usec, lit("sock-send-timeout"), SO_SNDTIMEO);
+}
+
+static val sock_recv_timeout(val sock, val usec)
+{
+  return sock_timeout(sock, usec, lit("sock-recv-timeout"), SO_RCVTIMEO);
+}
+#endif
+
+
 val open_sockfd(val fd, val family, val type, val mode_str_in)
 {
   if (type == num_fast(SOCK_DGRAM)) {
@@ -892,6 +922,11 @@ void sock_load_init(void)
   reg_fun(intern(lit("sock-accept"), user_package), func_n2o(sock_accept, 1));
   reg_fun(intern(lit("sock-shutdown"), user_package), func_n2o(sock_shutdown, 1));
 
+#if defined SO_SNDTIMEO && defined SO_RCVTIMEO
+  reg_fun(intern(lit("sock-send-timeout"), user_package), func_n2(sock_send_timeout));
+  reg_fun(intern(lit("sock-recv-timeout"), user_package), func_n2(sock_recv_timeout));
+#endif
+
   fill_stream_ops(&dgram_strm_ops);
   dgram_strm_ops.get_sock_family = dgram_get_sock_family;
   dgram_strm_ops.get_sock_type = dgram_get_sock_type;
-- 
cgit v1.2.3