Zbigniew Jędrzejewski-Szmek 126222
From 4a0ab3a8fd59d38dabb9a7096939d8d8e8adc62f Mon Sep 17 00:00:00 2001
Zbigniew Jędrzejewski-Szmek 126222
From: Lennart Poettering <lennart@poettering.net>
Zbigniew Jędrzejewski-Szmek 126222
Date: Mon, 16 Dec 2013 17:04:36 +0100
Zbigniew Jędrzejewski-Szmek 126222
Subject: [PATCH] util: try harder to increase the send/recv buffers of sockets
Zbigniew Jędrzejewski-Szmek 126222
Zbigniew Jędrzejewski-Szmek 126222
If we have the priviliges we will try SO_SNDBUFFORCE/SO_RCVBUFFORCE and
Zbigniew Jędrzejewski-Szmek 126222
only fall back to SO_SNDBUF/SO_RCVBUF if that fails.
Zbigniew Jędrzejewski-Szmek 126222
---
Zbigniew Jędrzejewski-Szmek 126222
 src/shared/util.c | 25 ++++++++++++-------------
Zbigniew Jędrzejewski-Szmek 126222
 1 file changed, 12 insertions(+), 13 deletions(-)
Zbigniew Jędrzejewski-Szmek 126222
Zbigniew Jędrzejewski-Szmek 126222
diff --git a/src/shared/util.c b/src/shared/util.c
Zbigniew Jędrzejewski-Szmek 126222
index 02bae90..8824b9b 100644
Zbigniew Jędrzejewski-Szmek 126222
--- a/src/shared/util.c
Zbigniew Jędrzejewski-Szmek 126222
+++ b/src/shared/util.c
Zbigniew Jędrzejewski-Szmek 126222
@@ -4928,15 +4928,15 @@ int fd_inc_sndbuf(int fd, size_t n) {
Zbigniew Jędrzejewski-Szmek 126222
         socklen_t l = sizeof(value);
Zbigniew Jędrzejewski-Szmek 126222
 
Zbigniew Jędrzejewski-Szmek 126222
         r = getsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, &l);
Zbigniew Jędrzejewski-Szmek 126222
-        if (r >= 0 &&
Zbigniew Jędrzejewski-Szmek 126222
-            l == sizeof(value) &&
Zbigniew Jędrzejewski-Szmek 126222
-            (size_t) value >= n*2)
Zbigniew Jędrzejewski-Szmek 126222
+        if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
Zbigniew Jędrzejewski-Szmek 126222
                 return 0;
Zbigniew Jędrzejewski-Szmek 126222
 
Zbigniew Jędrzejewski-Szmek 126222
+        /* If we have the privileges we will ignore the kernel limit. */
Zbigniew Jędrzejewski-Szmek 126222
+
Zbigniew Jędrzejewski-Szmek 126222
         value = (int) n;
Zbigniew Jędrzejewski-Szmek 126222
-        r = setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value));
Zbigniew Jędrzejewski-Szmek 126222
-        if (r < 0)
Zbigniew Jędrzejewski-Szmek 126222
-                return -errno;
Zbigniew Jędrzejewski-Szmek 126222
+        if (setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &value, sizeof(value)) < 0)
Zbigniew Jędrzejewski-Szmek 126222
+                if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
Zbigniew Jędrzejewski-Szmek 126222
+                        return -errno;
Zbigniew Jędrzejewski-Szmek 126222
 
Zbigniew Jędrzejewski-Szmek 126222
         return 1;
Zbigniew Jędrzejewski-Szmek 126222
 }
Zbigniew Jędrzejewski-Szmek 126222
@@ -4946,16 +4946,15 @@ int fd_inc_rcvbuf(int fd, size_t n) {
Zbigniew Jędrzejewski-Szmek 126222
         socklen_t l = sizeof(value);
Zbigniew Jędrzejewski-Szmek 126222
 
Zbigniew Jędrzejewski-Szmek 126222
         r = getsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, &l);
Zbigniew Jędrzejewski-Szmek 126222
-        if (r >= 0 &&
Zbigniew Jędrzejewski-Szmek 126222
-            l == sizeof(value) &&
Zbigniew Jędrzejewski-Szmek 126222
-            (size_t) value >= n*2)
Zbigniew Jędrzejewski-Szmek 126222
+        if (r >= 0 && l == sizeof(value) && (size_t) value >= n*2)
Zbigniew Jędrzejewski-Szmek 126222
                 return 0;
Zbigniew Jędrzejewski-Szmek 126222
 
Zbigniew Jędrzejewski-Szmek 126222
-        value = (int) n;
Zbigniew Jędrzejewski-Szmek 126222
-        r = setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value));
Zbigniew Jędrzejewski-Szmek 126222
-        if (r < 0)
Zbigniew Jędrzejewski-Szmek 126222
-                return -errno;
Zbigniew Jędrzejewski-Szmek 126222
+        /* If we have the privileges we will ignore the kernel limit. */
Zbigniew Jędrzejewski-Szmek 126222
 
Zbigniew Jędrzejewski-Szmek 126222
+        value = (int) n;
Zbigniew Jędrzejewski-Szmek 126222
+        if (setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &value, sizeof(value)) < 0)
Zbigniew Jędrzejewski-Szmek 126222
+                if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
Zbigniew Jędrzejewski-Szmek 126222
+                        return -errno;
Zbigniew Jędrzejewski-Szmek 126222
         return 1;
Zbigniew Jędrzejewski-Szmek 126222
 }
Zbigniew Jędrzejewski-Szmek 126222