OSDN Git Service

sock: sockc cookie initializer
authorWillem de Bruijn <willemb@google.com>
Fri, 6 Jul 2018 14:12:56 +0000 (10:12 -0400)
committerDavid S. Miller <davem@davemloft.net>
Sat, 7 Jul 2018 01:58:49 +0000 (10:58 +0900)
Initialize the cookie in one location to reduce code duplication and
avoid bugs from inconsistent initialization, such as that fixed in
commit 9887cba19978 ("ip: limit use of gso_size to udp").

Signed-off-by: Willem de Bruijn <willemb@google.com>
Signed-off-by: David S. Miller <davem@davemloft.net>
include/net/sock.h
net/ipv4/tcp.c
net/packet/af_packet.c

index e0eac9e..83b7475 100644 (file)
@@ -1600,6 +1600,12 @@ struct sockcm_cookie {
        u16 tsflags;
 };
 
+static inline void sockcm_init(struct sockcm_cookie *sockc,
+                              const struct sock *sk)
+{
+       *sockc = (struct sockcm_cookie) { .tsflags = sk->sk_tsflags };
+}
+
 int __sock_cmsg_send(struct sock *sk, struct msghdr *msg, struct cmsghdr *cmsg,
                     struct sockcm_cookie *sockc);
 int sock_cmsg_send(struct sock *sk, struct msghdr *msg,
index bf461fa..850dc8f 100644 (file)
@@ -1241,7 +1241,7 @@ int tcp_sendmsg_locked(struct sock *sk, struct msghdr *msg, size_t size)
                /* 'common' sending to sendq */
        }
 
-       sockc.tsflags = sk->sk_tsflags;
+       sockcm_init(&sockc, sk);
        if (msg->msg_controllen) {
                err = sock_cmsg_send(sk, msg, &sockc);
                if (unlikely(err)) {
index 3428f77..47931eb 100644 (file)
@@ -1951,8 +1951,7 @@ retry:
                goto out_unlock;
        }
 
-       sockc.transmit_time = 0;
-       sockc.tsflags = sk->sk_tsflags;
+       sockcm_init(&sockc, sk);
        if (msg->msg_controllen) {
                err = sock_cmsg_send(sk, msg, &sockc);
                if (unlikely(err))
@@ -2636,8 +2635,7 @@ static int tpacket_snd(struct packet_sock *po, struct msghdr *msg)
        if (unlikely(!(dev->flags & IFF_UP)))
                goto out_put;
 
-       sockc.transmit_time = 0;
-       sockc.tsflags = po->sk.sk_tsflags;
+       sockcm_init(&sockc, &po->sk);
        if (msg->msg_controllen) {
                err = sock_cmsg_send(&po->sk, msg, &sockc);
                if (unlikely(err))
@@ -2833,8 +2831,7 @@ static int packet_snd(struct socket *sock, struct msghdr *msg, size_t len)
        if (unlikely(!(dev->flags & IFF_UP)))
                goto out_unlock;
 
-       sockc.transmit_time = 0;
-       sockc.tsflags = sk->sk_tsflags;
+       sockcm_init(&sockc, sk);
        sockc.mark = sk->sk_mark;
        if (msg->msg_controllen) {
                err = sock_cmsg_send(sk, msg, &sockc);