Forward port SOCKSENDBUF, SOCKRECVBUF features from 2.10.11.

[ircu2.10.12-pk.git] / ircd / os_linux.c

diff --git a/ircd/os_linux.c b/ircd/os_linux.c
index e3dc5e7d616d6b5de099382cb23e431433ea63f0..2c24bb106806df0be6a055cb610384089e6802f6 100644 (file)
--- a/ircd/os_linux.c
+++ b/ircd/os_linux.c
@@ -19,11 +19,17 @@
  * $Id$
  *
  */
+#include "config.h"
+
+#define _XOPEN_SOURCE  /* make limits.h #define IOV_MAX */
+
 #include "ircd_osdep.h"
+#include "msgq.h"
 
 #include <assert.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <limits.h>
 #include <stdio.h>
 #include <netinet/in.h>
 #include <string.h>
@@ -32,10 +38,9 @@
 #include <sys/socket.h>
 #include <sys/types.h>
 #include <sys/times.h>
+#include <sys/uio.h>
 #include <sys/param.h>
-#if 0
 #include <unistd.h>
-#endif
 
 /*
  * This is part of the STATS replies. There is no offical numeric for this
@@ -145,11 +150,20 @@ int os_set_reuseaddr(int fd)
   return (0 == setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)));
 }
 
-int os_set_sockbufs(int fd, unsigned int size)
+int os_set_sockbufs(int fd, unsigned int ssize, unsigned int rsize)
+{
+  unsigned int sopt = ssize;
+  unsigned int ropt = rsize;
+  return (0 == setsockopt(fd, SOL_SOCKET, SO_RCVBUF, 
+                          (const char*) &ropt, sizeof(ropt)) &&
+          0 == setsockopt(fd, SOL_SOCKET, SO_SNDBUF, 
+                          (const char*) &sopt, sizeof(sopt)));
+}
+
+int os_set_tos(int fd,int tos)
 {
-  unsigned int opt = size;
-  return (0 == setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &opt, sizeof(opt)) &&
-          0 == setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &opt, sizeof(opt)));
+  unsigned int opt = tos;
+  return (0 == setsockopt(fd, SOL_IP, IP_TOS, &opt, sizeof(opt)));
 }
 
 int os_disable_options(int fd)
@@ -253,14 +267,48 @@ IOResult os_send_nonb(int fd, const char* buf, unsigned int length,
   return IO_FAILURE;
 }
 
-
-int os_connect_nonb(int fd, const struct sockaddr_in* sin)
+/*
+ * os_sendv_nonb - non blocking writev to a connection
+ * returns:
+ *  1  if data was written
+ *    count_out contains amount written
+ *   
+ *  0  if write call blocked, recoverable error
+ *  -1 if an unrecoverable error occurred
+ */
+IOResult os_sendv_nonb(int fd, struct MsgQ* buf, unsigned int* count_in,
+                      unsigned int* count_out)
 {
-  if (connect(fd, (const struct sockaddr*) sin, sizeof(struct sockaddr_in))) {
-    if (errno != EINPROGRESS)
-      return 0;
+  int res;
+  int count;
+  struct iovec iov[IOV_MAX];
+
+  assert(0 != buf);
+  assert(0 != count_in);
+  assert(0 != count_out);
+
+  *count_in = 0;
+  *count_out = 0;
+  errno = 0;
+
+  count = msgq_mapiov(buf, iov, IOV_MAX, count_in);
+
+  if (-1 < (res = writev(fd, iov, count))) {
+    *count_out = (unsigned) res;
+    return IO_SUCCESS;
   }
-  return 1;
+  else if (EAGAIN == errno || ENOMEM == errno || ENOBUFS == errno)
+    return IO_BLOCKED;
+
+  return IO_FAILURE;
+}
+
+
+IOResult os_connect_nonb(int fd, const struct sockaddr_in* sin)
+{
+  if (connect(fd, (const struct sockaddr*) sin, sizeof(struct sockaddr_in)))
+    return (errno == EINPROGRESS) ? IO_BLOCKED : IO_FAILURE;
+  return IO_SUCCESS;
 }
       
 int os_get_sockname(int fd, struct sockaddr_in* sin_out)