sock.h 63 KB
Newer Older
Linus Torvalds's avatar
Linus Torvalds committed
1 2 3 4 5 6 7 8 9
/*
 * INET		An implementation of the TCP/IP protocol suite for the LINUX
 *		operating system.  INET is implemented using the  BSD Socket
 *		interface as the means of communication with the user level.
 *
 *		Definitions for the AF_INET socket handler.
 *
 * Version:	@(#)sock.h	1.0.4	05/13/93
 *
10
 * Authors:	Ross Biro
Linus Torvalds's avatar
Linus Torvalds committed
11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
 *		Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG>
 *		Corey Minyard <wf-rch!minyard@relay.EU.net>
 *		Florian La Roche <flla@stud.uni-sb.de>
 *
 * Fixes:
 *		Alan Cox	:	Volatiles in skbuff pointers. See
 *					skbuff comments. May be overdone,
 *					better to prove they can be removed
 *					than the reverse.
 *		Alan Cox	:	Added a zapped field for tcp to note
 *					a socket is reset and must stay shut up
 *		Alan Cox	:	New fields for options
 *	Pauline Middelink	:	identd support
 *		Alan Cox	:	Eliminate low level recv/recvfrom
 *		David S. Miller	:	New socket lookup architecture.
 *              Steve Whitehouse:       Default routines for sock_ops
 *              Arnaldo C. Melo :	removed net_pinfo, tp_pinfo and made
 *              			protinfo be just a void pointer, as the
 *              			protocol specific parts were moved to
 *              			respective headers and ipv4/v6, etc now
 *              			use private slabcaches for its socks
 *              Pedro Hortas	:	New flags field for socket options
 *
 *
 *		This program is free software; you can redistribute it and/or
 *		modify it under the terms of the GNU General Public License
 *		as published by the Free Software Foundation; either version
 *		2 of the License, or (at your option) any later version.
 */
#ifndef _SOCK_H
#define _SOCK_H

43
#include <linux/hardirq.h>
44
#include <linux/kernel.h>
Linus Torvalds's avatar
Linus Torvalds committed
45
#include <linux/list.h>
46
#include <linux/list_nulls.h>
Linus Torvalds's avatar
Linus Torvalds committed
47 48
#include <linux/timer.h>
#include <linux/cache.h>
49
#include <linux/bitops.h>
50
#include <linux/lockdep.h>
Linus Torvalds's avatar
Linus Torvalds committed
51 52
#include <linux/netdevice.h>
#include <linux/skbuff.h>	/* struct sk_buff */
53
#include <linux/mm.h>
Linus Torvalds's avatar
Linus Torvalds committed
54
#include <linux/security.h>
55
#include <linux/slab.h>
56
#include <linux/uaccess.h>
57
#include <linux/page_counter.h>
58
#include <linux/memcontrol.h>
59
#include <linux/static_key.h>
Al Viro's avatar
Al Viro committed
60
#include <linux/sched.h>
61
#include <linux/wait.h>
62
#include <linux/cgroup-defs.h>
Linus Torvalds's avatar
Linus Torvalds committed
63 64

#include <linux/filter.h>
65
#include <linux/rculist_nulls.h>
66
#include <linux/poll.h>
Linus Torvalds's avatar
Linus Torvalds committed
67

68
#include <linux/atomic.h>
Linus Torvalds's avatar
Linus Torvalds committed
69 70
#include <net/dst.h>
#include <net/checksum.h>
71
#include <net/tcp_states.h>
72
#include <linux/net_tstamp.h>
Linus Torvalds's avatar
Linus Torvalds committed
73 74 75 76 77 78 79 80 81 82 83 84 85

/*
 * This structure really needs to be cleaned up.
 * Most of it is for TCP, and not used by any of
 * the other protocols.
 */

/* Define this to get the SOCK_DBG debugging facility. */
#define SOCK_DEBUGGING
#ifdef SOCK_DEBUGGING
#define SOCK_DEBUG(sk, msg...) do { if ((sk) && sock_flag((sk), SOCK_DBG)) \
					printk(KERN_DEBUG msg); } while (0)
#else
86
/* Validate arguments and do nothing */
87
static inline __printf(2, 3)
88
void SOCK_DEBUG(const struct sock *sk, const char *msg, ...)
89 90
{
}
Linus Torvalds's avatar
Linus Torvalds committed
91 92 93 94 95 96 97 98
#endif

/* This is the per-socket lock.  The spinlock provides a synchronization
 * between user contexts and software interrupt processing, whereas the
 * mini-semaphore synchronizes multiple users amongst themselves.
 */
typedef struct {
	spinlock_t		slock;
99
	int			owned;
Linus Torvalds's avatar
Linus Torvalds committed
100
	wait_queue_head_t	wq;
101 102 103 104 105 106 107 108 109
	/*
	 * We express the mutex-alike socket_lock semantics
	 * to the lock validator by explicitly managing
	 * the slock as a lock variant (in addition to
	 * the slock itself):
	 */
#ifdef CONFIG_DEBUG_LOCK_ALLOC
	struct lockdep_map dep_map;
#endif
Linus Torvalds's avatar
Linus Torvalds committed
110 111 112
} socket_lock_t;

struct sock;
113
struct proto;
114
struct net;
Linus Torvalds's avatar
Linus Torvalds committed
115

116 117 118
typedef __u32 __bitwise __portpair;
typedef __u64 __bitwise __addrpair;

Linus Torvalds's avatar
Linus Torvalds committed
119
/**
120
 *	struct sock_common - minimal network layer representation of sockets
121 122
 *	@skc_daddr: Foreign IPv4 addr
 *	@skc_rcv_saddr: Bound local IPv4 addr
Eric Dumazet's avatar
Eric Dumazet committed
123
 *	@skc_hash: hash value used with various protocol lookup tables
124
 *	@skc_u16hashes: two u16 hash values used by UDP lookup tables
125 126
 *	@skc_dport: placeholder for inet_dport/tw_dport
 *	@skc_num: placeholder for inet_num/tw_num
127 128 129
 *	@skc_family: network address family
 *	@skc_state: Connection state
 *	@skc_reuse: %SO_REUSEADDR setting
130
 *	@skc_reuseport: %SO_REUSEPORT setting
131 132
 *	@skc_bound_dev_if: bound device index if != 0
 *	@skc_bind_node: bind hash linkage for various protocol lookup tables
133
 *	@skc_portaddr_node: second hash linkage for UDP/UDP-Lite protocol
134
 *	@skc_prot: protocol handlers inside a network family
135
 *	@skc_net: reference to the network namespace of this socket
136 137 138
 *	@skc_node: main hash linkage for various protocol lookup tables
 *	@skc_nulls_node: main hash linkage for TCP/UDP/UDP-Lite protocol
 *	@skc_tx_queue_mapping: tx queue number for this connection
139 140 141
 *	@skc_flags: place holder for sk_flags
 *		%SO_LINGER (l_onoff), %SO_BROADCAST, %SO_KEEPALIVE,
 *		%SO_OOBINLINE settings, %SO_TIMESTAMPING settings
142
 *	@skc_incoming_cpu: record/match cpu processing incoming packets
143
 *	@skc_refcnt: reference count
144 145
 *
 *	This is the minimal network layer representation of sockets, the header
146 147
 *	for struct sock and struct inet_timewait_sock.
 */
Linus Torvalds's avatar
Linus Torvalds committed
148
struct sock_common {
149
	/* skc_daddr and skc_rcv_saddr must be grouped on a 8 bytes aligned
Eric Dumazet's avatar
Eric Dumazet committed
150
	 * address on 64bit arches : cf INET_MATCH()
Eric Dumazet's avatar
Eric Dumazet committed
151
	 */
152
	union {
153
		__addrpair	skc_addrpair;
154 155 156 157 158
		struct {
			__be32	skc_daddr;
			__be32	skc_rcv_saddr;
		};
	};
159 160 161 162
	union  {
		unsigned int	skc_hash;
		__u16		skc_u16hashes[2];
	};
163 164
	/* skc_dport && skc_num must be grouped as well */
	union {
165
		__portpair	skc_portpair;
166 167 168 169 170 171
		struct {
			__be16	skc_dport;
			__u16	skc_num;
		};
	};

Eric Dumazet's avatar
Eric Dumazet committed
172 173
	unsigned short		skc_family;
	volatile unsigned char	skc_state;
174
	unsigned char		skc_reuse:4;
175 176
	unsigned char		skc_reuseport:1;
	unsigned char		skc_ipv6only:1;
177
	unsigned char		skc_net_refcnt:1;
Eric Dumazet's avatar
Eric Dumazet committed
178
	int			skc_bound_dev_if;
179 180
	union {
		struct hlist_node	skc_bind_node;
181
		struct hlist_node	skc_portaddr_node;
182
	};
183
	struct proto		*skc_prot;
184
	possible_net_t		skc_net;
185 186 187 188 189 190

#if IS_ENABLED(CONFIG_IPV6)
	struct in6_addr		skc_v6_daddr;
	struct in6_addr		skc_v6_rcv_saddr;
#endif

191 192
	atomic64_t		skc_cookie;

193 194 195 196 197 198 199 200 201 202
	/* following fields are padding to force
	 * offset(struct sock, sk_refcnt) == 128 on 64bit arches
	 * assuming IPV6 is enabled. We use this padding differently
	 * for different kind of 'sockets'
	 */
	union {
		unsigned long	skc_flags;
		struct sock	*skc_listener; /* request_sock */
		struct inet_timewait_death_row *skc_tw_dr; /* inet_timewait_sock */
	};
203 204 205 206
	/*
	 * fields between dontcopy_begin/dontcopy_end
	 * are not copied in sock_copy()
	 */
207
	/* private: */
208
	int			skc_dontcopy_begin[0];
209
	/* public: */
210 211 212 213 214
	union {
		struct hlist_node	skc_node;
		struct hlist_nulls_node skc_nulls_node;
	};
	int			skc_tx_queue_mapping;
215 216 217
	union {
		int		skc_incoming_cpu;
		u32		skc_rcv_wnd;
218
		u32		skc_tw_rcv_nxt; /* struct tcp_timewait_sock  */
219
	};
220

221
	atomic_t		skc_refcnt;
222
	/* private: */
223
	int                     skc_dontcopy_end[0];
224 225 226
	union {
		u32		skc_rxhash;
		u32		skc_window_clamp;
227
		u32		skc_tw_snd_nxt; /* struct tcp_timewait_sock */
228
	};
229
	/* public: */
Linus Torvalds's avatar
Linus Torvalds committed
230 231 232 233
};

/**
  *	struct sock - network layer representation of sockets
234
  *	@__sk_common: shared layout with inet_timewait_sock
235 236 237 238
  *	@sk_shutdown: mask of %SEND_SHUTDOWN and/or %RCV_SHUTDOWN
  *	@sk_userlocks: %SO_SNDBUF and %SO_RCVBUF settings
  *	@sk_lock:	synchronizer
  *	@sk_rcvbuf: size of receive buffer in bytes
239
  *	@sk_wq: sock wait queue and async head
240
  *	@sk_rx_dst: receive input route used by early demux
241 242 243 244 245 246 247 248
  *	@sk_dst_cache: destination cache
  *	@sk_policy: flow policy
  *	@sk_receive_queue: incoming packets
  *	@sk_wmem_alloc: transmit queue bytes committed
  *	@sk_write_queue: Packet sending queue
  *	@sk_omem_alloc: "o" is "option" or "other"
  *	@sk_wmem_queued: persistent queue size
  *	@sk_forward_alloc: space allocated forward
249
  *	@sk_napi_id: id of the last napi context to receive data for sk
250
  *	@sk_ll_usec: usecs to busypoll when there is no data
251
  *	@sk_allocation: allocation mode
252
  *	@sk_pacing_rate: Pacing rate (if supported by transport/packet scheduler)
253
  *	@sk_max_pacing_rate: Maximum pacing rate (%SO_MAX_PACING_RATE)
254
  *	@sk_sndbuf: size of send buffer in bytes
255 256
  *	@sk_no_check_tx: %SO_NO_CHECK setting, set checksum in TX packets
  *	@sk_no_check_rx: allow zero checksum in RX packets
257
  *	@sk_route_caps: route capabilities (e.g. %NETIF_F_TSO)
258
  *	@sk_route_nocaps: forbidden route capabilities (e.g NETIF_F_GSO_MASK)
259
  *	@sk_gso_type: GSO type (e.g. %SKB_GSO_TCPV4)
260
  *	@sk_gso_max_size: Maximum GSO segment size to build
261
  *	@sk_gso_max_segs: Maximum number of GSO segments
262 263 264 265
  *	@sk_lingertime: %SO_LINGER l_linger setting
  *	@sk_backlog: always used with the per-socket spinlock held
  *	@sk_callback_lock: used with the callbacks in the end of this struct
  *	@sk_error_queue: rarely used
266 267
  *	@sk_prot_creator: sk_prot of original sock creator (see ipv6_setsockopt,
  *			  IPV6_ADDRFORM for instance)
268
  *	@sk_err: last error
269 270
  *	@sk_err_soft: errors that don't cause failure but are the cause of a
  *		      persistent failure not just 'timed out'
Eric Dumazet's avatar
Eric Dumazet committed
271
  *	@sk_drops: raw/udp drops counter
272 273 274 275 276
  *	@sk_ack_backlog: current listen backlog
  *	@sk_max_ack_backlog: listen backlog set in listen()
  *	@sk_priority: %SO_PRIORITY setting
  *	@sk_type: socket type (%SOCK_STREAM, etc)
  *	@sk_protocol: which protocol this socket belongs in this network family
277 278
  *	@sk_peer_pid: &struct pid for this socket's peer
  *	@sk_peer_cred: %SO_PEERCRED setting
279 280 281
  *	@sk_rcvlowat: %SO_RCVLOWAT setting
  *	@sk_rcvtimeo: %SO_RCVTIMEO setting
  *	@sk_sndtimeo: %SO_SNDTIMEO setting
282
  *	@sk_txhash: computed flow hash for use on transmit
283 284 285
  *	@sk_filter: socket filtering instructions
  *	@sk_timer: sock cleanup timer
  *	@sk_stamp: time stamp of last packet received
286
  *	@sk_tsflags: SO_TIMESTAMPING socket options
287
  *	@sk_tskey: counter to disambiguate concurrent tstamp requests
288 289
  *	@sk_socket: Identd and reporting IO signals
  *	@sk_user_data: RPC layer private data
290
  *	@sk_frag: cached page frag
291
  *	@sk_peek_off: current peek_offset value
292
  *	@sk_send_head: front of stuff to transmit
293
  *	@sk_security: used by security modules
294
  *	@sk_mark: generic packet mark
295
  *	@sk_cgrp_data: cgroup data for this cgroup
296
  *	@sk_memcg: this socket's memory cgroup association
297 298 299 300 301 302 303
  *	@sk_write_pending: a write to stream socket waits to start
  *	@sk_state_change: callback to indicate change in the state of the sock
  *	@sk_data_ready: callback to indicate there is data to be processed
  *	@sk_write_space: callback to indicate there is bf sending space available
  *	@sk_error_report: callback to indicate errors (e.g. %MSG_ERRQUEUE)
  *	@sk_backlog_rcv: callback to process the backlog
  *	@sk_destruct: called at sock freeing time, i.e. when all refcnt == 0
304
  *	@sk_reuseport_cb: reuseport group container
Linus Torvalds's avatar
Linus Torvalds committed
305 306 307
 */
struct sock {
	/*
308
	 * Now struct inet_timewait_sock also uses sock_common, so please just
Linus Torvalds's avatar
Linus Torvalds committed
309 310 311
	 * don't add nothing before this first member (__sk_common) --acme
	 */
	struct sock_common	__sk_common;
Eric Dumazet's avatar
Eric Dumazet committed
312 313 314
#define sk_node			__sk_common.skc_node
#define sk_nulls_node		__sk_common.skc_nulls_node
#define sk_refcnt		__sk_common.skc_refcnt
315
#define sk_tx_queue_mapping	__sk_common.skc_tx_queue_mapping
Eric Dumazet's avatar
Eric Dumazet committed
316

317 318
#define sk_dontcopy_begin	__sk_common.skc_dontcopy_begin
#define sk_dontcopy_end		__sk_common.skc_dontcopy_end
Eric Dumazet's avatar
Eric Dumazet committed
319
#define sk_hash			__sk_common.skc_hash
320
#define sk_portpair		__sk_common.skc_portpair
Eric Dumazet's avatar
Eric Dumazet committed
321 322
#define sk_num			__sk_common.skc_num
#define sk_dport		__sk_common.skc_dport
323 324 325
#define sk_addrpair		__sk_common.skc_addrpair
#define sk_daddr		__sk_common.skc_daddr
#define sk_rcv_saddr		__sk_common.skc_rcv_saddr
Linus Torvalds's avatar
Linus Torvalds committed
326 327 328
#define sk_family		__sk_common.skc_family
#define sk_state		__sk_common.skc_state
#define sk_reuse		__sk_common.skc_reuse
329
#define sk_reuseport		__sk_common.skc_reuseport
330
#define sk_ipv6only		__sk_common.skc_ipv6only
331
#define sk_net_refcnt		__sk_common.skc_net_refcnt
Linus Torvalds's avatar
Linus Torvalds committed
332 333
#define sk_bound_dev_if		__sk_common.skc_bound_dev_if
#define sk_bind_node		__sk_common.skc_bind_node
334
#define sk_prot			__sk_common.skc_prot
335
#define sk_net			__sk_common.skc_net
336 337
#define sk_v6_daddr		__sk_common.skc_v6_daddr
#define sk_v6_rcv_saddr	__sk_common.skc_v6_rcv_saddr
338
#define sk_cookie		__sk_common.skc_cookie
339
#define sk_incoming_cpu		__sk_common.skc_incoming_cpu
340
#define sk_flags		__sk_common.skc_flags
341
#define sk_rxhash		__sk_common.skc_rxhash
342

Linus Torvalds's avatar
Linus Torvalds committed
343
	socket_lock_t		sk_lock;
344
	struct sk_buff_head	sk_receive_queue;
345 346 347 348
	/*
	 * The backlog queue is special, it is always used with
	 * the per-socket spinlock held and requires low latency
	 * access. Therefore we special case it's implementation.
349 350 351
	 * Note : rmem_alloc is in this structure to fill a hole
	 * on 64bit arches, not because its logically part of
	 * backlog.
352 353
	 */
	struct {
354 355 356 357
		atomic_t	rmem_alloc;
		int		len;
		struct sk_buff	*head;
		struct sk_buff	*tail;
358
	} sk_backlog;
359 360
#define sk_rmem_alloc sk_backlog.rmem_alloc
	int			sk_forward_alloc;
361

362
	__u32			sk_txhash;
363
#ifdef CONFIG_NET_RX_BUSY_POLL
364
	unsigned int		sk_napi_id;
365
	unsigned int		sk_ll_usec;
366 367 368 369 370
#endif
	atomic_t		sk_drops;
	int			sk_rcvbuf;

	struct sk_filter __rcu	*sk_filter;
371 372 373 374
	union {
		struct socket_wq __rcu	*sk_wq;
		struct socket_wq	*sk_wq_raw;
	};
375
#ifdef CONFIG_XFRM
376
	struct xfrm_policy __rcu *sk_policy[2];
377
#endif
Eric Dumazet's avatar
Eric Dumazet committed
378
	struct dst_entry	*sk_rx_dst;
379
	struct dst_entry __rcu	*sk_dst_cache;
Eric Dumazet's avatar
Eric Dumazet committed
380
	/* Note: 32bit hole on 64bit arches */
Linus Torvalds's avatar
Linus Torvalds committed
381 382
	atomic_t		sk_wmem_alloc;
	atomic_t		sk_omem_alloc;
383
	int			sk_sndbuf;
Linus Torvalds's avatar
Linus Torvalds committed
384
	struct sk_buff_head	sk_write_queue;
385 386 387 388 389

	/*
	 * Because of non atomicity rules, all
	 * changes are protected by socket lock.
	 */
390
	kmemcheck_bitfield_begin(flags);
391
	unsigned int		sk_padding : 2,
392 393
				sk_no_check_tx : 1,
				sk_no_check_rx : 1,
394 395 396
				sk_userlocks : 4,
				sk_protocol  : 8,
				sk_type      : 16;
397
#define SK_PROTOCOL_MAX U8_MAX
398
	kmemcheck_bitfield_end(flags);
399

Linus Torvalds's avatar
Linus Torvalds committed
400
	int			sk_wmem_queued;
Al Viro's avatar
Al Viro committed
401
	gfp_t			sk_allocation;
402
	u32			sk_pacing_rate; /* bytes per second */
403
	u32			sk_max_pacing_rate;
404 405
	netdev_features_t	sk_route_caps;
	netdev_features_t	sk_route_nocaps;
406
	int			sk_gso_type;
407
	unsigned int		sk_gso_max_size;
408
	u16			sk_gso_max_segs;
409
	int			sk_rcvlowat;
Linus Torvalds's avatar
Linus Torvalds committed
410 411
	unsigned long	        sk_lingertime;
	struct sk_buff_head	sk_error_queue;
412
	struct proto		*sk_prot_creator;
Linus Torvalds's avatar
Linus Torvalds committed
413 414 415
	rwlock_t		sk_callback_lock;
	int			sk_err,
				sk_err_soft;
416 417
	u32			sk_ack_backlog;
	u32			sk_max_ack_backlog;
Linus Torvalds's avatar
Linus Torvalds committed
418
	__u32			sk_priority;
419
	__u32			sk_mark;
420 421
	struct pid		*sk_peer_pid;
	const struct cred	*sk_peer_cred;
Linus Torvalds's avatar
Linus Torvalds committed
422 423 424
	long			sk_rcvtimeo;
	long			sk_sndtimeo;
	struct timer_list	sk_timer;
425
	ktime_t			sk_stamp;
426
	u16			sk_tsflags;
427
	u8			sk_shutdown;
428
	u32			sk_tskey;
Linus Torvalds's avatar
Linus Torvalds committed
429 430
	struct socket		*sk_socket;
	void			*sk_user_data;
431
	struct page_frag	sk_frag;
Linus Torvalds's avatar
Linus Torvalds committed
432
	struct sk_buff		*sk_send_head;
433
	__s32			sk_peek_off;
Linus Torvalds's avatar
Linus Torvalds committed
434
	int			sk_write_pending;
435
#ifdef CONFIG_SECURITY
Linus Torvalds's avatar
Linus Torvalds committed
436
	void			*sk_security;
437
#endif
438
	struct sock_cgroup_data	sk_cgrp_data;
439
	struct mem_cgroup	*sk_memcg;
Linus Torvalds's avatar
Linus Torvalds committed
440
	void			(*sk_state_change)(struct sock *sk);
441
	void			(*sk_data_ready)(struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
442 443
	void			(*sk_write_space)(struct sock *sk);
	void			(*sk_error_report)(struct sock *sk);
444 445
	int			(*sk_backlog_rcv)(struct sock *sk,
						  struct sk_buff *skb);
Linus Torvalds's avatar
Linus Torvalds committed
446
	void                    (*sk_destruct)(struct sock *sk);
447
	struct sock_reuseport __rcu	*sk_reuseport_cb;
448
	struct rcu_head		sk_rcu;
Linus Torvalds's avatar
Linus Torvalds committed
449 450
};

451 452 453 454 455
#define __sk_user_data(sk) ((*((void __rcu **)&(sk)->sk_user_data)))

#define rcu_dereference_sk_user_data(sk)	rcu_dereference(__sk_user_data((sk)))
#define rcu_assign_sk_user_data(sk, ptr)	rcu_assign_pointer(__sk_user_data((sk)), ptr)

456 457 458 459 460 461 462 463 464 465 466
/*
 * SK_CAN_REUSE and SK_NO_REUSE on a socket mean that the socket is OK
 * or not whether his port will be reused by someone else. SK_FORCE_REUSE
 * on a socket means that the socket will reuse everybody else's port
 * without looking at the other's sk_reuse value.
 */

#define SK_NO_REUSE	0
#define SK_CAN_REUSE	1
#define SK_FORCE_REUSE	2

467 468
int sk_set_peek_off(struct sock *sk, int val);

469 470
static inline int sk_peek_offset(struct sock *sk, int flags)
{
471 472 473 474 475 476 477
	if (unlikely(flags & MSG_PEEK)) {
		s32 off = READ_ONCE(sk->sk_peek_off);
		if (off >= 0)
			return off;
	}

	return 0;
478 479 480 481
}

static inline void sk_peek_offset_bwd(struct sock *sk, int val)
{
482 483 484 485 486
	s32 off = READ_ONCE(sk->sk_peek_off);

	if (unlikely(off >= 0)) {
		off = max_t(s32, off - val, 0);
		WRITE_ONCE(sk->sk_peek_off, off);
487 488 489 490 491
	}
}

static inline void sk_peek_offset_fwd(struct sock *sk, int val)
{
492
	sk_peek_offset_bwd(sk, -val);
493 494
}

Linus Torvalds's avatar
Linus Torvalds committed
495 496 497
/*
 * Hashed lists helper routines
 */
Li Zefan's avatar
Li Zefan committed
498 499 500 501 502
static inline struct sock *sk_entry(const struct hlist_node *node)
{
	return hlist_entry(node, struct sock, sk_node);
}

503
static inline struct sock *__sk_head(const struct hlist_head *head)
Linus Torvalds's avatar
Linus Torvalds committed
504 505 506 507
{
	return hlist_entry(head->first, struct sock, sk_node);
}

508
static inline struct sock *sk_head(const struct hlist_head *head)
Linus Torvalds's avatar
Linus Torvalds committed
509 510 511 512
{
	return hlist_empty(head) ? NULL : __sk_head(head);
}

513 514 515 516 517 518 519 520 521 522
static inline struct sock *__sk_nulls_head(const struct hlist_nulls_head *head)
{
	return hlist_nulls_entry(head->first, struct sock, sk_nulls_node);
}

static inline struct sock *sk_nulls_head(const struct hlist_nulls_head *head)
{
	return hlist_nulls_empty(head) ? NULL : __sk_nulls_head(head);
}

523
static inline struct sock *sk_next(const struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
524 525 526 527 528
{
	return sk->sk_node.next ?
		hlist_entry(sk->sk_node.next, struct sock, sk_node) : NULL;
}

529 530 531 532 533 534 535 536
static inline struct sock *sk_nulls_next(const struct sock *sk)
{
	return (!is_a_nulls(sk->sk_nulls_node.next)) ?
		hlist_nulls_entry(sk->sk_nulls_node.next,
				  struct sock, sk_nulls_node) :
		NULL;
}

537
static inline bool sk_unhashed(const struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
538 539 540 541
{
	return hlist_unhashed(&sk->sk_node);
}

542
static inline bool sk_hashed(const struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
543
{
544
	return !sk_unhashed(sk);
Linus Torvalds's avatar
Linus Torvalds committed
545 546
}

547
static inline void sk_node_init(struct hlist_node *node)
Linus Torvalds's avatar
Linus Torvalds committed
548 549 550 551
{
	node->pprev = NULL;
}

552
static inline void sk_nulls_node_init(struct hlist_nulls_node *node)
553 554 555 556
{
	node->pprev = NULL;
}

557
static inline void __sk_del_node(struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
558 559 560 561
{
	__hlist_del(&sk->sk_node);
}

562
/* NB: equivalent to hlist_del_init_rcu */
563
static inline bool __sk_del_node_init(struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
564 565 566 567
{
	if (sk_hashed(sk)) {
		__sk_del_node(sk);
		sk_node_init(&sk->sk_node);
568
		return true;
Linus Torvalds's avatar
Linus Torvalds committed
569
	}
570
	return false;
Linus Torvalds's avatar
Linus Torvalds committed
571 572 573 574 575 576 577 578
}

/* Grab socket reference count. This operation is valid only
   when sk is ALREADY grabbed f.e. it is found in hash table
   or a list and the lookup is made under lock preventing hash table
   modifications.
 */

579
static __always_inline void sock_hold(struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
580 581 582 583 584 585 586
{
	atomic_inc(&sk->sk_refcnt);
}

/* Ungrab socket in the context, which assumes that socket refcnt
   cannot hit zero, f.e. it is true in context of any socketcall.
 */
587
static __always_inline void __sock_put(struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
588 589 590 591
{
	atomic_dec(&sk->sk_refcnt);
}

592
static inline bool sk_del_node_init(struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
593
{
594
	bool rc = __sk_del_node_init(sk);
Linus Torvalds's avatar
Linus Torvalds committed
595 596 597 598 599 600 601 602

	if (rc) {
		/* paranoid for a while -acme */
		WARN_ON(atomic_read(&sk->sk_refcnt) == 1);
		__sock_put(sk);
	}
	return rc;
}
603
#define sk_del_node_init_rcu(sk)	sk_del_node_init(sk)
Linus Torvalds's avatar
Linus Torvalds committed
604

605
static inline bool __sk_nulls_del_node_init_rcu(struct sock *sk)
606 607
{
	if (sk_hashed(sk)) {
608
		hlist_nulls_del_init_rcu(&sk->sk_nulls_node);
609
		return true;
610
	}
611
	return false;
612 613
}

614
static inline bool sk_nulls_del_node_init_rcu(struct sock *sk)
615
{
616
	bool rc = __sk_nulls_del_node_init_rcu(sk);
617 618 619 620 621 622 623 624 625

	if (rc) {
		/* paranoid for a while -acme */
		WARN_ON(atomic_read(&sk->sk_refcnt) == 1);
		__sock_put(sk);
	}
	return rc;
}

626
static inline void __sk_add_node(struct sock *sk, struct hlist_head *list)
Linus Torvalds's avatar
Linus Torvalds committed
627 628 629 630
{
	hlist_add_head(&sk->sk_node, list);
}

631
static inline void sk_add_node(struct sock *sk, struct hlist_head *list)
Linus Torvalds's avatar
Linus Torvalds committed
632 633 634 635 636
{
	sock_hold(sk);
	__sk_add_node(sk, list);
}

637
static inline void sk_add_node_rcu(struct sock *sk, struct hlist_head *list)
638 639
{
	sock_hold(sk);
640 641 642 643 644
	if (IS_ENABLED(CONFIG_IPV6) && sk->sk_reuseport &&
	    sk->sk_family == AF_INET6)
		hlist_add_tail_rcu(&sk->sk_node, list);
	else
		hlist_add_head_rcu(&sk->sk_node, list);
645 646
}

647
static inline void __sk_nulls_add_node_rcu(struct sock *sk, struct hlist_nulls_head *list)
648
{
649 650 651 652 653
	if (IS_ENABLED(CONFIG_IPV6) && sk->sk_reuseport &&
	    sk->sk_family == AF_INET6)
		hlist_nulls_add_tail_rcu(&sk->sk_nulls_node, list);
	else
		hlist_nulls_add_head_rcu(&sk->sk_nulls_node, list);
654 655
}

656
static inline void sk_nulls_add_node_rcu(struct sock *sk, struct hlist_nulls_head *list)
657 658
{
	sock_hold(sk);
659
	__sk_nulls_add_node_rcu(sk, list);
660 661
}

662
static inline void __sk_del_bind_node(struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
663 664 665 666
{
	__hlist_del(&sk->sk_bind_node);
}

667
static inline void sk_add_bind_node(struct sock *sk,
Linus Torvalds's avatar
Linus Torvalds committed
668 669 670 671 672
					struct hlist_head *list)
{
	hlist_add_head(&sk->sk_bind_node, list);
}

673 674 675 676
#define sk_for_each(__sk, list) \
	hlist_for_each_entry(__sk, list, sk_node)
#define sk_for_each_rcu(__sk, list) \
	hlist_for_each_entry_rcu(__sk, list, sk_node)
677 678 679 680
#define sk_nulls_for_each(__sk, node, list) \
	hlist_nulls_for_each_entry(__sk, node, list, sk_nulls_node)
#define sk_nulls_for_each_rcu(__sk, node, list) \
	hlist_nulls_for_each_entry_rcu(__sk, node, list, sk_nulls_node)
681 682
#define sk_for_each_from(__sk) \
	hlist_for_each_entry_from(__sk, sk_node)
683 684 685
#define sk_nulls_for_each_from(__sk, node) \
	if (__sk && ({ node = &(__sk)->sk_nulls_node; 1; })) \
		hlist_nulls_for_each_entry_from(__sk, node, sk_nulls_node)
686 687 688 689
#define sk_for_each_safe(__sk, tmp, list) \
	hlist_for_each_entry_safe(__sk, tmp, list, sk_node)
#define sk_for_each_bound(__sk, list) \
	hlist_for_each_entry(__sk, list, sk_bind_node)
Linus Torvalds's avatar
Linus Torvalds committed
690

691
/**
692
 * sk_for_each_entry_offset_rcu - iterate over a list at a given struct offset
693 694 695 696 697 698
 * @tpos:	the type * to use as a loop cursor.
 * @pos:	the &struct hlist_node to use as a loop cursor.
 * @head:	the head for your list.
 * @offset:	offset of hlist_node within the struct.
 *
 */
699 700 701
#define sk_for_each_entry_offset_rcu(tpos, pos, head, offset)		       \
	for (pos = rcu_dereference((head)->first);			       \
	     pos != NULL &&						       \
702
		({ tpos = (typeof(*tpos) *)((void *)pos - offset); 1;});       \
703
	     pos = rcu_dereference(pos->next))
704

705 706 707 708 709 710 711 712 713
static inline struct user_namespace *sk_user_ns(struct sock *sk)
{
	/* Careful only use this in a context where these parameters
	 * can not change and must all be valid, such as recvmsg from
	 * userspace.
	 */
	return sk->sk_socket->file->f_cred->user_ns;
}

Linus Torvalds's avatar
Linus Torvalds committed
714 715 716 717 718 719 720 721 722 723 724 725 726 727
/* Sock flags */
enum sock_flags {
	SOCK_DEAD,
	SOCK_DONE,
	SOCK_URGINLINE,
	SOCK_KEEPOPEN,
	SOCK_LINGER,
	SOCK_DESTROY,
	SOCK_BROADCAST,
	SOCK_TIMESTAMP,
	SOCK_ZAPPED,
	SOCK_USE_WRITE_QUEUE, /* whether to call sk->sk_write_space in sock_wfree */
	SOCK_DBG, /* %SO_DEBUG setting */
	SOCK_RCVTSTAMP, /* %SO_TIMESTAMP setting */
728
	SOCK_RCVTSTAMPNS, /* %SO_TIMESTAMPNS setting */
Linus Torvalds's avatar
Linus Torvalds committed
729 730
	SOCK_LOCALROUTE, /* route locally only, %SO_DONTROUTE setting */
	SOCK_QUEUE_SHRUNK, /* write queue has been shrunk recently */
731
	SOCK_MEMALLOC, /* VM depends on this socket for swapping */
732
	SOCK_TIMESTAMPING_RX_SOFTWARE,  /* %SOF_TIMESTAMPING_RX_SOFTWARE */
733
	SOCK_FASYNC, /* fasync() active */
734
	SOCK_RXQ_OVFL,
735
	SOCK_ZEROCOPY, /* buffers from userspace */
736
	SOCK_WIFI_STATUS, /* push wifi status to userspace */
737 738 739 740
	SOCK_NOFCS, /* Tell NIC not to do the Ethernet FCS.
		     * Will use last 4 bytes of packet sent from
		     * user-space instead.
		     */
741
	SOCK_FILTER_LOCKED, /* Filter cannot be changed anymore */
742
	SOCK_SELECT_ERR_QUEUE, /* Wake select on error queue */
743
	SOCK_RCU_FREE, /* wait rcu grace period in sk_destruct() */
Linus Torvalds's avatar
Linus Torvalds committed
744 745
};

746 747
#define SK_FLAGS_TIMESTAMP ((1UL << SOCK_TIMESTAMP) | (1UL << SOCK_TIMESTAMPING_RX_SOFTWARE))

748 749 750 751 752
static inline void sock_copy_flags(struct sock *nsk, struct sock *osk)
{
	nsk->sk_flags = osk->sk_flags;
}

Linus Torvalds's avatar
Linus Torvalds committed
753 754 755 756 757 758 759 760 761 762
static inline void sock_set_flag(struct sock *sk, enum sock_flags flag)
{
	__set_bit(flag, &sk->sk_flags);
}

static inline void sock_reset_flag(struct sock *sk, enum sock_flags flag)
{
	__clear_bit(flag, &sk->sk_flags);
}

Eric Dumazet's avatar
Eric Dumazet committed
763
static inline bool sock_flag(const struct sock *sk, enum sock_flags flag)
Linus Torvalds's avatar
Linus Torvalds committed
764 765 766 767
{
	return test_bit(flag, &sk->sk_flags);
}

768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
#ifdef CONFIG_NET
extern struct static_key memalloc_socks;
static inline int sk_memalloc_socks(void)
{
	return static_key_false(&memalloc_socks);
}
#else

static inline int sk_memalloc_socks(void)
{
	return 0;
}

#endif

783
static inline gfp_t sk_gfp_mask(const struct sock *sk, gfp_t gfp_mask)
784
{
785
	return gfp_mask | (sk->sk_allocation & __GFP_MEMALLOC);
786 787
}

Linus Torvalds's avatar
Linus Torvalds committed
788 789 790 791 792 793 794 795 796 797
static inline void sk_acceptq_removed(struct sock *sk)
{
	sk->sk_ack_backlog--;
}

static inline void sk_acceptq_added(struct sock *sk)
{
	sk->sk_ack_backlog++;
}

798
static inline bool sk_acceptq_is_full(const struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
799
{
800
	return sk->sk_ack_backlog > sk->sk_max_ack_backlog;
Linus Torvalds's avatar
Linus Torvalds committed
801 802 803 804 805
}

/*
 * Compute minimal free write space needed to queue new packets.
 */
806
static inline int sk_stream_min_wspace(const struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
807
{
808
	return sk->sk_wmem_queued >> 1;
Linus Torvalds's avatar
Linus Torvalds committed
809 810
}

811
static inline int sk_stream_wspace(const struct sock *sk)
Linus Torvalds's avatar
Linus Torvalds committed
812 813 814 815
{
	return sk->sk_sndbuf - sk->sk_wmem_queued;
}

816
void sk_stream_write_space(struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
817

818
/* OOB backlog add */
Zhu Yi's avatar
Zhu Yi committed
819
static inline void __sk_add_backlog(struct sock *sk, struct sk_buff *skb)
820
{
821
	/* dont let skb dst not refcounted, we are going to leave rcu lock */
822
	skb_dst_force_safe(skb);
823 824 825 826

	if (!sk->sk_backlog.tail)
		sk->sk_backlog.head = skb;
	else
827
		sk->sk_backlog.tail->next = skb;
828 829

	sk->sk_backlog.tail = skb;
830 831
	skb->next = NULL;
}
Linus Torvalds's avatar
Linus Torvalds committed
832

833 834
/*
 * Take into account size of receive queue and backlog queue
Eric Dumazet's avatar
Eric Dumazet committed
835 836
 * Do not take into account this skb truesize,
 * to allow even a single big packet to come.
837
 */
838
static inline bool sk_rcvqueues_full(const struct sock *sk, unsigned int limit)
839 840 841
{
	unsigned int qsize = sk->sk_backlog.len + atomic_read(&sk->sk_rmem_alloc);

842
	return qsize > limit;
843 844
}

845
/* The per-socket spinlock must be held here. */
846 847
static inline __must_check int sk_add_backlog(struct sock *sk, struct sk_buff *skb,
					      unsigned int limit)
848
{
849
	if (sk_rcvqueues_full(sk, limit))
850 851
		return -ENOBUFS;

852 853 854 855 856 857 858 859
	/*
	 * If the skb was allocated from pfmemalloc reserves, only
	 * allow SOCK_MEMALLOC sockets to use it as this socket is
	 * helping free memory
	 */
	if (skb_pfmemalloc(skb) && !sock_flag(sk, SOCK_MEMALLOC))
		return -ENOMEM;

Zhu Yi's avatar
Zhu Yi committed
860
	__sk_add_backlog(sk, skb);
861 862 863 864
	sk->sk_backlog.len += skb->truesize;
	return 0;
}

865
int __sk_backlog_rcv(struct sock *sk, struct sk_buff *skb);
866

867 868
static inline int sk_backlog_rcv(struct sock *sk, struct sk_buff *skb)
{
869 870 871
	if (sk_memalloc_socks() && skb_pfmemalloc(skb))
		return __sk_backlog_rcv(sk, skb);

872 873 874
	return sk->sk_backlog_rcv(sk, skb);
}

875 876 877 878 879
static inline void sk_incoming_cpu_update(struct sock *sk)
{
	sk->sk_incoming_cpu = raw_smp_processor_id();
}

880
static inline void sock_rps_record_flow_hash(__u32 hash)
881 882 883 884 885 886
{
#ifdef CONFIG_RPS
	struct rps_sock_flow_table *sock_flow_table;

	rcu_read_lock();
	sock_flow_table = rcu_dereference(rps_sock_flow_table);
887
	rps_record_sock_flow(sock_flow_table, hash);
888 889 890 891
	rcu_read_unlock();
#endif
}

892 893
static inline void sock_rps_record_flow(const struct sock *sk)
{
894
#ifdef CONFIG_RPS
895
	sock_rps_record_flow_hash(sk->sk_rxhash);
896
#endif
897 898
}

899 900
static inline void sock_rps_save_rxhash(struct sock *sk,
					const struct sk_buff *skb)
901 902
{
#ifdef CONFIG_RPS
903
	if (unlikely(sk->sk_rxhash != skb->hash))
904
		sk->sk_rxhash = skb->hash;
905 906 907
#endif
}

908 909 910 911 912 913 914
static inline void sock_rps_reset_rxhash(struct sock *sk)
{
#ifdef CONFIG_RPS
	sk->sk_rxhash = 0;
#endif
}

915 916 917 918 919 920 921
#define sk_wait_event(__sk, __timeo, __condition)			\
	({	int __rc;						\
		release_sock(__sk);					\
		__rc = __condition;					\
		if (!__rc) {						\
			*(__timeo) = schedule_timeout(*(__timeo));	\
		}							\
922
		sched_annotate_sleep();						\
923 924 925 926
		lock_sock(__sk);					\
		__rc = __condition;					\
		__rc;							\
	})
Linus Torvalds's avatar
Linus Torvalds committed
927

928 929 930 931 932 933 934
int sk_stream_wait_connect(struct sock *sk, long *timeo_p);
int sk_stream_wait_memory(struct sock *sk, long *timeo_p);
void sk_stream_wait_close(struct sock *sk, long timeo_p);
int sk_stream_error(struct sock *sk, int flags, int err);
void sk_stream_kill_queues(struct sock *sk);
void sk_set_memalloc(struct sock *sk);
void sk_clear_memalloc(struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
935

936 937 938 939 940 941 942 943 944 945 946
void __sk_flush_backlog(struct sock *sk);

static inline bool sk_flush_backlog(struct sock *sk)
{
	if (unlikely(READ_ONCE(sk->sk_backlog.tail))) {
		__sk_flush_backlog(sk);
		return true;
	}
	return false;
}

947
int sk_wait_data(struct sock *sk, long *timeo, const struct sk_buff *skb);
Linus Torvalds's avatar
Linus Torvalds committed
948

949
struct request_sock_ops;
950
struct timewait_sock_ops;
951
struct inet_hashinfo;
952
struct raw_hashinfo;
953
struct module;
954

955 956 957 958 959 960 961 962 963 964 965 966
/*
 * caches using SLAB_DESTROY_BY_RCU should let .next pointer from nulls nodes
 * un-modified. Special care is taken when initializing object to zero.
 */
static inline void sk_prot_clear_nulls(struct sock *sk, int size)
{
	if (offsetof(struct sock, sk_node.next) != 0)
		memset(sk, 0, offsetof(struct sock, sk_node.next));
	memset(&sk->sk_node.pprev, 0,
	       size - offsetof(struct sock, sk_node.pprev));
}

Linus Torvalds's avatar
Linus Torvalds committed
967 968 969 970
/* Networking protocol blocks we attach to sockets.
 * socket layer -> transport layer interface
 */
struct proto {
971
	void			(*close)(struct sock *sk,
Linus Torvalds's avatar
Linus Torvalds committed
972 973
					long timeout);
	int			(*connect)(struct sock *sk,
974
					struct sockaddr *uaddr,
Linus Torvalds's avatar
Linus Torvalds committed
975 976 977
					int addr_len);
	int			(*disconnect)(struct sock *sk, int flags);

978
	struct sock *		(*accept)(struct sock *sk, int flags, int *err);
Linus Torvalds's avatar
Linus Torvalds committed
979 980 981 982

	int			(*ioctl)(struct sock *sk, int cmd,
					 unsigned long arg);
	int			(*init)(struct sock *sk);
983
	void			(*destroy)(struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
984
	void			(*shutdown)(struct sock *sk, int how);
985
	int			(*setsockopt)(struct sock *sk, int level,
Linus Torvalds's avatar
Linus Torvalds committed
986
					int optname, char __user *optval,
987
					unsigned int optlen);
988 989 990
	int			(*getsockopt)(struct sock *sk, int level,
					int optname, char __user *optval,
					int __user *option);
991
#ifdef CONFIG_COMPAT
992 993 994
	int			(*compat_setsockopt)(struct sock *sk,
					int level,
					int optname, char __user *optval,
995
					unsigned int optlen);
996 997 998 999
	int			(*compat_getsockopt)(struct sock *sk,
					int level,
					int optname, char __user *optval,
					int __user *option);
1000 1001
	int			(*compat_ioctl)(struct sock *sk,
					unsigned int cmd, unsigned long arg);
1002
#endif
1003 1004 1005
	int			(*sendmsg)(struct sock *sk, struct msghdr *msg,
					   size_t len);
	int			(*recvmsg)(struct sock *sk, struct msghdr *msg,
1006 1007
					   size_t len, int noblock, int flags,
					   int *addr_len);
Linus Torvalds's avatar
Linus Torvalds committed
1008 1009
	int			(*sendpage)(struct sock *sk, struct page *page,
					int offset, size_t size, int flags);
1010
	int			(*bind)(struct sock *sk,
Linus Torvalds's avatar
Linus Torvalds committed
1011 1012
					struct sockaddr *uaddr, int addr_len);

1013
	int			(*backlog_rcv) (struct sock *sk,
Linus Torvalds's avatar
Linus Torvalds committed
1014 1015
						struct sk_buff *skb);

Eric Dumazet's avatar
Eric Dumazet committed
1016 1017
	void		(*release_cb)(struct sock *sk);

Linus Torvalds's avatar
Linus Torvalds committed
1018
	/* Keeping track of sk's, looking them up, and port selection methods. */
1019
	int			(*hash)(struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
1020
	void			(*unhash)(struct sock *sk);
1021
	void			(*rehash)(struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
1022
	int			(*get_port)(struct sock *sk, unsigned short snum);
1023
	void			(*clear_sk)(struct sock *sk, int size);
Linus Torvalds's avatar
Linus Torvalds committed
1024

1025
	/* Keeping track of sockets in use */
1026
#ifdef CONFIG_PROC_FS
1027
	unsigned int		inuse_idx;
1028
#endif
1029

1030
	bool			(*stream_memory_free)(const struct sock *sk);
Linus Torvalds's avatar
Linus Torvalds committed
1031
	/* Memory pressure */
1032
	void			(*enter_memory_pressure)(struct sock *sk);
1033
	atomic_long_t		*memory_allocated;	/* Current allocated memory. */
1034
	struct percpu_counter	*sockets_allocated;	/* Current number of sockets. */
Linus Torvalds's avatar
Linus Torvalds committed
1035 1036 1037
	/*
	 * Pressure flag: try to collapse.
	 * Technical note: it is used by multiple contexts non atomically.
1038
	 * All the __sk_mem_schedule() is of this nature: accounting
Linus Torvalds's avatar
Linus Torvalds committed
1039 1040 1041
	 * is strict, actions are advisory and have some latency.
	 */
	int			*memory_pressure;
1042
	long			*sysctl_mem;
Linus Torvalds's avatar
Linus Torvalds committed
1043 1044 1045
	int			*sysctl_wmem;
	int			*sysctl_rmem;
	int			max_header;
1046
	bool			no_autobind;
Linus Torvalds's avatar
Linus Torvalds committed
1047

1048
	struct kmem_cache	*slab;
Linus Torvalds's avatar
Linus Torvalds committed
1049
	unsigned int		obj_size;
1050
	int			slab_flags;
Linus Torvalds's avatar
Linus Torvalds committed
1051

1052
	struct percpu_counter	*orphan_count;
1053

1054
	struct request_sock_ops	*rsk_prot;
1055
	struct timewait_sock_ops *twsk_prot;
1056

1057 1058
	union {
		struct inet_hashinfo	*hashinfo;
1059
		struct udp_table	*udp_table;
1060
		struct raw_hashinfo	*raw_hash;
1061
	} h;
1062

Linus Torvalds's avatar
Linus Torvalds committed
1063 1064 1065 1066 1067
	struct module		*owner;

	char			name[32];

	struct list_head	node;
1068 1069
#ifdef SOCK_REFCNT_DEBUG
	atomic_t		socks;
1070
#endif
1071
	int			(*diag_destroy)(struct sock *sk, int err);
1072 1073
};

1074 1075
int proto_register(struct proto *prot, int alloc_slab);
void proto_unregister(struct proto *prot);
Linus Torvalds's avatar
Linus Torvalds committed
1076

1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089
#ifdef SOCK_REFCNT_DEBUG
static inline void sk_refcnt_debug_inc(struct sock *sk)
{
	atomic_inc(&sk->sk_prot->socks);
}

static inline void sk_refcnt_debug_dec(struct sock *sk)
{
	atomic_dec(&sk->sk_prot->socks);
	printk(KERN_DEBUG "%s socket %p released, %d are still alive\n",
	       sk->sk_prot->name, sk, atomic_read(&sk->sk_prot->socks));
}

1090
static inline void sk_refcnt_debug_release(const struct sock *sk)
1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
{
	if (atomic_read(&sk->sk_refcnt) != 1)
		printk(KERN_DEBUG "Destruction of the %s socket %p delayed, refcnt=%d\n",
		       sk->sk_prot->name, sk, atomic_read(&sk->sk_refcnt));
}
#else /* SOCK_REFCNT_DEBUG */
#define sk_refcnt_debug_inc(sk) do { } while (0)
#define sk_refcnt_debug_dec(sk) do { } while (0)
#define sk_refcnt_debug_release(sk) do { } while (0)
#endif /* SOCK_REFCNT_DEBUG */

1102 1103 1104 1105 1106 1107 1108 1109 1110
static inline bool sk_stream_memory_free(const struct sock *sk)
{
	if (sk->sk_wmem_queued >= sk->sk_sndbuf)
		return false;

	return sk->sk_prot->stream_memory_free ?
		sk->sk_prot->stream_memory_free(sk) : true;
}

1111 1112
static inline bool sk_stream_is_writeable(const struct sock *sk)
{
1113 1114
	return sk_stream_wspace(sk) >= sk_stream_min_wspace(sk) &&
	       sk_stream_memory_free(sk);
1115
}
1116

1117

1118 1119 1120 1121 1122 1123 1124 1125 1126
static inline bool sk_has_memory_pressure(const struct sock *sk)
{
	return sk->sk_prot->memory_pressure != NULL;
}

static inline bool sk_under_memory_pressure(const struct sock *sk)
{
	if (!sk->sk_prot->memory_pressure)
		return false;
1127

1128 1129
	if (mem_cgroup_sockets_enabled && sk->sk_memcg &&
	    mem_cgroup_under_socket_pressure(sk->sk_memcg))
1130
		return true;
1131

1132
	return !!*sk->sk_prot->memory_pressure;
1133 1134 1135 1136 1137 1138
}

static inline void sk_leave_memory_pressure(struct sock *sk)
{
	int *memory_pressure = sk->sk_prot->memory_pressure;

1139 1140 1141 1142
	if (!memory_pressure)
		return;

	if (*memory_pressure)
1143 1144 1145 1146 1147
		*memory_pressure = 0;
}

static inline void sk_enter_memory_pressure(struct sock *sk)
{
1148 1149 1150 1151
	if (!sk->sk_prot->enter_memory_pressure)
		return;

	sk->sk_prot->enter_memory_pressure(sk);
1152 1153 1154 1155
}

static inline long sk_prot_mem_limits(const struct sock *sk, int index)
{
1156
	return sk->sk_prot->sysctl_mem[index];
1157 1158
}

1159 1160 1161
static inline long
sk_memory_allocated(const struct sock *sk)
{
1162
	return atomic_long_read(sk->sk_prot->memory_allocated);
1163 1164 1165
}

static inline long
1166
sk_memory_allocated_add(struct sock *sk, int amt)
1167
{
1168
	return atomic_long_add_return(amt, sk->sk_prot->memory_allocated);
1169 1170 1171
}

static inline void
1172
sk_memory_allocated_sub(struct sock *sk, int amt)
1173
{
1174
	atomic_long_sub(amt, sk->sk_prot->memory_allocated);
1175 1176 1177 1178
}

static inline void sk_sockets_allocated_dec(struct sock *sk)
{
1179
	percpu_counter_dec(sk->sk_prot->sockets_allocated);
1180 1181 1182 1183
}

static inline void sk_sockets_allocated_inc(struct sock *sk)
{
1184
	percpu_counter_inc(sk->sk_prot->sockets_allocated);
1185 1186 1187 1188 1189
}

static inline int
sk_sockets_allocated_read_positive(struct sock *sk)
{
1190
	return percpu_counter_read_positive(sk->sk_prot->sockets_allocated);
1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
}

static inline int
proto_sockets_allocated_sum_positive(struct proto *prot)
{
	return percpu_counter_sum_positive(prot->sockets_allocated);
}

static inline long
proto_memory_allocated(struct proto *prot)
{
	return atomic_long_read(prot->memory_allocated);
}

static inline bool
proto_memory_pressure(struct proto *prot)
{
	if (!prot->memory_pressure)
		return false;
	return !!*prot->memory_pressure;
}

1213 1214

#ifdef CONFIG_PROC_FS
Linus Torvalds's avatar
Linus Torvalds committed
1215
/* Called with local bh disabled */
1216 1217
void sock_prot_inuse_add(struct net *net, struct proto *prot, int inc);
int sock_prot_inuse_get(struct net *net, struct proto *proto);
1218
#else
1219
static inline void sock_prot_inuse_add(struct net *net, struct proto *prot,
1220
		int inc)
1221 1222 1223 1224
{
}
#endif

Linus Torvalds's avatar
Linus Torvalds committed
1225

1226 1227 1228
/* With per-bucket locks this operation is not-atomic, so that
 * this version is not worse.
 */
1229
static inline int __sk_prot_rehash(struct sock *sk)
1230 1231
{
	sk->sk_prot->unhash(sk);
1232
	return sk->sk_prot->hash(sk);
1233 1234
}

1235 1236
void sk_prot_clear_portaddr_nulls(struct sock *sk, int size);

Linus Torvalds's avatar
Linus Torvalds committed
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246