[PATCH net-next 1/2] net: allow per netns sysctl_rmem and sysctl_wmem for protos

[Eric Dumazet]

As we want to gradually implement per netns sysctl_rmem and sysctl_wmem
on per protocol basis, add two new fields in struct proto,
and two new helpers : sk_get_wmem0() and sk_get_rmem0()

First user will be TCP. Then UDP and SCTP can be easily converted,
while DECNET probably wont get this support.

Signed-off-by: Eric Dumazet <eduma...@google.com>
---
 include/net/sock.h          | 22 ++++++++++++++++++++++
 include/trace/events/sock.h |  2 +-
 net/core/sock.c             | 10 ++++++----
 3 files changed, 29 insertions(+), 5 deletions(-)

diff --git a/include/net/sock.h b/include/net/sock.h
index 
6f1be9726e02b350162573117ed6eb6cebc627b4..688a823dccc306bd21f47da167c6922161af5a6a
 100644
--- a/include/net/sock.h
+++ b/include/net/sock.h
@@ -1101,8 +1101,12 @@ struct proto {
         */
        unsigned long           *memory_pressure;
        long                    *sysctl_mem;
+
        int                     *sysctl_wmem;
        int                     *sysctl_rmem;
+       u32                     sysctl_wmem_offset;
+       u32                     sysctl_rmem_offset;
+
        int                     max_header;
        bool                    no_autobind;
 
@@ -2390,4 +2394,22 @@ extern int sysctl_optmem_max;
 extern __u32 sysctl_wmem_default;
 extern __u32 sysctl_rmem_default;
 
+static inline int sk_get_wmem0(const struct sock *sk, const struct proto 
*proto)
+{
+       /* Does this proto have per netns sysctl_wmem ? */
+       if (proto->sysctl_wmem_offset)
+               return *(int *)((void *)sock_net(sk) + 
proto->sysctl_wmem_offset);
+
+       return *proto->sysctl_wmem;
+}
+
+static inline int sk_get_rmem0(const struct sock *sk, const struct proto 
*proto)
+{
+       /* Does this proto have per netns sysctl_rmem ? */
+       if (proto->sysctl_rmem_offset)
+               return *(int *)((void *)sock_net(sk) + 
proto->sysctl_rmem_offset);
+
+       return *proto->sysctl_rmem;
+}
+
 #endif /* _SOCK_H */
diff --git a/include/trace/events/sock.h b/include/trace/events/sock.h
index 
6d31c0520ef36f89af825acebedfa5dabcaa6f6a..ec4dade2446607bd124133f22366c63e5caba852
 100644
--- a/include/trace/events/sock.h
+++ b/include/trace/events/sock.h
@@ -48,7 +48,7 @@ TRACE_EVENT(sock_exceed_buf_limit,
                strncpy(__entry->name, prot->name, 32);
                __entry->sysctl_mem = prot->sysctl_mem;
                __entry->allocated = allocated;
-               __entry->sysctl_rmem = prot->sysctl_rmem[0];
+               __entry->sysctl_rmem = sk_get_rmem0(sk, prot);
                __entry->rmem_alloc = atomic_read(&sk->sk_rmem_alloc);
        ),
 
diff --git a/net/core/sock.c b/net/core/sock.c
index 
759400053110fee3ca1180536ebbe739e614eab6..c59bcf90d90536fedc7809e397f6bd414781b529
 100644
--- a/net/core/sock.c
+++ b/net/core/sock.c
@@ -2346,16 +2346,18 @@ int __sk_mem_raise_allocated(struct sock *sk, int size, 
int amt, int kind)
 
        /* guarantee minimum buffer size under pressure */
        if (kind == SK_MEM_RECV) {
-               if (atomic_read(&sk->sk_rmem_alloc) < prot->sysctl_rmem[0])
+               if (atomic_read(&sk->sk_rmem_alloc) < sk_get_rmem0(sk, prot))
                        return 1;
 
        } else { /* SK_MEM_SEND */
+               int wmem0 = sk_get_wmem0(sk, prot);
+
                if (sk->sk_type == SOCK_STREAM) {
-                       if (sk->sk_wmem_queued < prot->sysctl_wmem[0])
+                       if (sk->sk_wmem_queued < wmem0)
                                return 1;
-               } else if (refcount_read(&sk->sk_wmem_alloc) <
-                          prot->sysctl_wmem[0])
+               } else if (refcount_read(&sk->sk_wmem_alloc) < wmem0) {
                                return 1;
+               }
        }
 
        if (sk_has_memory_pressure(sk)) {
-- 
2.15.0.403.gc27cc4dac6-goog