From: Cong Wang <[email protected]>

Currently only unix stream socket sets TCP_ESTABLISHED,
datagram socket can set this too when they connect to its
peer socket. At least __ip4_datagram_connect() does the same.

This will be used by the next patch to determine whether an
AF_UNIX datagram socket can be redirected in sockmap.

Cc: John Fastabend <[email protected]>
Cc: Daniel Borkmann <[email protected]>
Cc: Jakub Sitnicki <[email protected]>
Cc: Lorenz Bauer <[email protected]>
Signed-off-by: Cong Wang <[email protected]>
---
 net/unix/af_unix.c | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/net/unix/af_unix.c b/net/unix/af_unix.c
index 4ce12d3c369e..21c4406f879b 100644
--- a/net/unix/af_unix.c
+++ b/net/unix/af_unix.c
@@ -1206,6 +1206,8 @@ static int unix_dgram_connect(struct socket *sock, struct 
sockaddr *addr,
                unix_peer(sk) = other;
                unix_state_double_unlock(sk, other);
        }
+
+       sk->sk_state = other->sk_state = TCP_ESTABLISHED;
        return 0;
 
 out_unlock:
@@ -1438,12 +1440,10 @@ static int unix_socketpair(struct socket *socka, struct 
socket *sockb)
        init_peercred(ska);
        init_peercred(skb);
 
-       if (ska->sk_type != SOCK_DGRAM) {
-               ska->sk_state = TCP_ESTABLISHED;
-               skb->sk_state = TCP_ESTABLISHED;
-               socka->state  = SS_CONNECTED;
-               sockb->state  = SS_CONNECTED;
-       }
+       ska->sk_state = TCP_ESTABLISHED;
+       skb->sk_state = TCP_ESTABLISHED;
+       socka->state  = SS_CONNECTED;
+       sockb->state  = SS_CONNECTED;
        return 0;
 }
 
-- 
2.25.1

Reply via email to