1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 /*
3 * INET An implementation of the TCP/IP protocol suite for the LINUX
4 * operating system. INET is implemented using the BSD Socket
5 * interface as the means of communication with the user level.
6 *
7 * Definitions for a generic INET TIMEWAIT sock
8 *
9 * From code originally in net/tcp.h
10 */
11 #ifndef _INET_TIMEWAIT_SOCK_
12 #define _INET_TIMEWAIT_SOCK_
13
14 #include <linux/list.h>
15 #include <linux/timer.h>
16 #include <linux/types.h>
17 #include <linux/workqueue.h>
18
19 #include <net/inet_sock.h>
20 #include <net/sock.h>
21 #include <net/tcp_states.h>
22 #include <net/timewait_sock.h>
23
24 #include <linux/atomic.h>
25
26 struct inet_bind_bucket;
27
28 /*
29 * This is a TIME_WAIT sock. It works around the memory consumption
30 * problems of sockets in such a state on heavily loaded servers, but
31 * without violating the protocol specification.
32 */
33 struct inet_timewait_sock {
34 /*
35 * Now struct sock also uses sock_common, so please just
36 * don't add nothing before this first member (__tw_common) --acme
37 */
38 struct sock_common __tw_common;
39 #define tw_family __tw_common.skc_family
40 #define tw_state __tw_common.skc_state
41 #define tw_reuse __tw_common.skc_reuse
42 #define tw_reuseport __tw_common.skc_reuseport
43 #define tw_ipv6only __tw_common.skc_ipv6only
44 #define tw_bound_dev_if __tw_common.skc_bound_dev_if
45 #define tw_node __tw_common.skc_nulls_node
46 #define tw_bind_node __tw_common.skc_bind_node
47 #define tw_refcnt __tw_common.skc_refcnt
48 #define tw_hash __tw_common.skc_hash
49 #define tw_prot __tw_common.skc_prot
50 #define tw_net __tw_common.skc_net
51 #define tw_daddr __tw_common.skc_daddr
52 #define tw_v6_daddr __tw_common.skc_v6_daddr
53 #define tw_rcv_saddr __tw_common.skc_rcv_saddr
54 #define tw_v6_rcv_saddr __tw_common.skc_v6_rcv_saddr
55 #define tw_dport __tw_common.skc_dport
56 #define tw_num __tw_common.skc_num
57 #define tw_cookie __tw_common.skc_cookie
58 #define tw_dr __tw_common.skc_tw_dr
59
60 __u32 tw_mark;
61 volatile unsigned char tw_substate;
62 unsigned char tw_rcv_wscale;
63
64 /* Socket demultiplex comparisons on incoming packets. */
65 /* these three are in inet_sock */
66 __be16 tw_sport;
67 /* And these are ours. */
68 unsigned int tw_transparent : 1,
69 tw_flowlabel : 20,
70 tw_usec_ts : 1,
71 tw_pad : 2, /* 2 bits hole */
72 tw_tos : 8;
73 u32 tw_txhash;
74 u32 tw_priority;
75 struct timer_list tw_timer;
76 struct inet_bind_bucket *tw_tb;
77 struct inet_bind2_bucket *tw_tb2;
78 };
79 #define tw_tclass tw_tos
80
inet_twsk(const struct sock * sk)81 static inline struct inet_timewait_sock *inet_twsk(const struct sock *sk)
82 {
83 return (struct inet_timewait_sock *)sk;
84 }
85
86 void inet_twsk_free(struct inet_timewait_sock *tw);
87 void inet_twsk_put(struct inet_timewait_sock *tw);
88
89 void inet_twsk_bind_unhash(struct inet_timewait_sock *tw,
90 struct inet_hashinfo *hashinfo);
91
92 struct inet_timewait_sock *inet_twsk_alloc(const struct sock *sk,
93 struct inet_timewait_death_row *dr,
94 const int state);
95
96 void inet_twsk_hashdance(struct inet_timewait_sock *tw, struct sock *sk,
97 struct inet_hashinfo *hashinfo);
98
99 void __inet_twsk_schedule(struct inet_timewait_sock *tw, int timeo,
100 bool rearm);
101
inet_twsk_schedule(struct inet_timewait_sock * tw,int timeo)102 static inline void inet_twsk_schedule(struct inet_timewait_sock *tw, int timeo)
103 {
104 __inet_twsk_schedule(tw, timeo, false);
105 }
106
inet_twsk_reschedule(struct inet_timewait_sock * tw,int timeo)107 static inline void inet_twsk_reschedule(struct inet_timewait_sock *tw, int timeo)
108 {
109 __inet_twsk_schedule(tw, timeo, true);
110 }
111
112 void inet_twsk_deschedule_put(struct inet_timewait_sock *tw);
113
114 void inet_twsk_purge(struct inet_hashinfo *hashinfo);
115
116 static inline
twsk_net(const struct inet_timewait_sock * twsk)117 struct net *twsk_net(const struct inet_timewait_sock *twsk)
118 {
119 return read_pnet(&twsk->tw_net);
120 }
121
122 static inline
twsk_net_set(struct inet_timewait_sock * twsk,struct net * net)123 void twsk_net_set(struct inet_timewait_sock *twsk, struct net *net)
124 {
125 write_pnet(&twsk->tw_net, net);
126 }
127 #endif /* _INET_TIMEWAIT_SOCK_ */
128