11da177e4SLinus Torvalds /* 21da177e4SLinus Torvalds * INET An implementation of the TCP/IP protocol suite for the LINUX 31da177e4SLinus Torvalds * operating system. INET is implemented using the BSD Socket 41da177e4SLinus Torvalds * interface as the means of communication with the user level. 51da177e4SLinus Torvalds * 61da177e4SLinus Torvalds * PF_INET protocol family socket handler. 71da177e4SLinus Torvalds * 802c30a84SJesper Juhl * Authors: Ross Biro 91da177e4SLinus Torvalds * Fred N. van Kempen, <waltje@uWalt.NL.Mugnet.ORG> 101da177e4SLinus Torvalds * Florian La Roche, <flla@stud.uni-sb.de> 111da177e4SLinus Torvalds * Alan Cox, <A.Cox@swansea.ac.uk> 121da177e4SLinus Torvalds * 131da177e4SLinus Torvalds * Changes (see also sock.c) 141da177e4SLinus Torvalds * 151da177e4SLinus Torvalds * piggy, 161da177e4SLinus Torvalds * Karl Knutson : Socket protocol table 171da177e4SLinus Torvalds * A.N.Kuznetsov : Socket death error in accept(). 181da177e4SLinus Torvalds * John Richardson : Fix non blocking error in connect() 191da177e4SLinus Torvalds * so sockets that fail to connect 201da177e4SLinus Torvalds * don't return -EINPROGRESS. 211da177e4SLinus Torvalds * Alan Cox : Asynchronous I/O support 221da177e4SLinus Torvalds * Alan Cox : Keep correct socket pointer on sock 231da177e4SLinus Torvalds * structures 241da177e4SLinus Torvalds * when accept() ed 251da177e4SLinus Torvalds * Alan Cox : Semantics of SO_LINGER aren't state 261da177e4SLinus Torvalds * moved to close when you look carefully. 271da177e4SLinus Torvalds * With this fixed and the accept bug fixed 281da177e4SLinus Torvalds * some RPC stuff seems happier. 291da177e4SLinus Torvalds * Niibe Yutaka : 4.4BSD style write async I/O 301da177e4SLinus Torvalds * Alan Cox, 311da177e4SLinus Torvalds * Tony Gale : Fixed reuse semantics. 321da177e4SLinus Torvalds * Alan Cox : bind() shouldn't abort existing but dead 331da177e4SLinus Torvalds * sockets. Stops FTP netin:.. I hope. 341da177e4SLinus Torvalds * Alan Cox : bind() works correctly for RAW sockets. 351da177e4SLinus Torvalds * Note that FreeBSD at least was broken 361da177e4SLinus Torvalds * in this respect so be careful with 371da177e4SLinus Torvalds * compatibility tests... 381da177e4SLinus Torvalds * Alan Cox : routing cache support 391da177e4SLinus Torvalds * Alan Cox : memzero the socket structure for 401da177e4SLinus Torvalds * compactness. 411da177e4SLinus Torvalds * Matt Day : nonblock connect error handler 421da177e4SLinus Torvalds * Alan Cox : Allow large numbers of pending sockets 431da177e4SLinus Torvalds * (eg for big web sites), but only if 441da177e4SLinus Torvalds * specifically application requested. 451da177e4SLinus Torvalds * Alan Cox : New buffering throughout IP. Used 461da177e4SLinus Torvalds * dumbly. 471da177e4SLinus Torvalds * Alan Cox : New buffering now used smartly. 481da177e4SLinus Torvalds * Alan Cox : BSD rather than common sense 491da177e4SLinus Torvalds * interpretation of listen. 501da177e4SLinus Torvalds * Germano Caronni : Assorted small races. 511da177e4SLinus Torvalds * Alan Cox : sendmsg/recvmsg basic support. 521da177e4SLinus Torvalds * Alan Cox : Only sendmsg/recvmsg now supported. 531da177e4SLinus Torvalds * Alan Cox : Locked down bind (see security list). 541da177e4SLinus Torvalds * Alan Cox : Loosened bind a little. 551da177e4SLinus Torvalds * Mike McLagan : ADD/DEL DLCI Ioctls 561da177e4SLinus Torvalds * Willy Konynenberg : Transparent proxying support. 571da177e4SLinus Torvalds * David S. Miller : New socket lookup architecture. 581da177e4SLinus Torvalds * Some other random speedups. 591da177e4SLinus Torvalds * Cyrus Durgin : Cleaned up file for kmod hacks. 601da177e4SLinus Torvalds * Andi Kleen : Fix inet_stream_connect TCP race. 611da177e4SLinus Torvalds * 621da177e4SLinus Torvalds * This program is free software; you can redistribute it and/or 631da177e4SLinus Torvalds * modify it under the terms of the GNU General Public License 641da177e4SLinus Torvalds * as published by the Free Software Foundation; either version 651da177e4SLinus Torvalds * 2 of the License, or (at your option) any later version. 661da177e4SLinus Torvalds */ 671da177e4SLinus Torvalds 68f4c50d99SHerbert Xu #include <linux/err.h> 691da177e4SLinus Torvalds #include <linux/errno.h> 701da177e4SLinus Torvalds #include <linux/types.h> 711da177e4SLinus Torvalds #include <linux/socket.h> 721da177e4SLinus Torvalds #include <linux/in.h> 731da177e4SLinus Torvalds #include <linux/kernel.h> 741da177e4SLinus Torvalds #include <linux/module.h> 751da177e4SLinus Torvalds #include <linux/sched.h> 761da177e4SLinus Torvalds #include <linux/timer.h> 771da177e4SLinus Torvalds #include <linux/string.h> 781da177e4SLinus Torvalds #include <linux/sockios.h> 791da177e4SLinus Torvalds #include <linux/net.h> 804fc268d2SRandy Dunlap #include <linux/capability.h> 811da177e4SLinus Torvalds #include <linux/fcntl.h> 821da177e4SLinus Torvalds #include <linux/mm.h> 831da177e4SLinus Torvalds #include <linux/interrupt.h> 841da177e4SLinus Torvalds #include <linux/stat.h> 851da177e4SLinus Torvalds #include <linux/init.h> 861da177e4SLinus Torvalds #include <linux/poll.h> 871da177e4SLinus Torvalds #include <linux/netfilter_ipv4.h> 88b3da2cf3SDavid S. Miller #include <linux/random.h> 895a0e3ad6STejun Heo #include <linux/slab.h> 901da177e4SLinus Torvalds 911da177e4SLinus Torvalds #include <asm/uaccess.h> 921da177e4SLinus Torvalds #include <asm/system.h> 931da177e4SLinus Torvalds 941da177e4SLinus Torvalds #include <linux/inet.h> 951da177e4SLinus Torvalds #include <linux/igmp.h> 9614c85021SArnaldo Carvalho de Melo #include <linux/inetdevice.h> 971da177e4SLinus Torvalds #include <linux/netdevice.h> 9873cc19f1SHerbert Xu #include <net/checksum.h> 991da177e4SLinus Torvalds #include <net/ip.h> 1001da177e4SLinus Torvalds #include <net/protocol.h> 1011da177e4SLinus Torvalds #include <net/arp.h> 1021da177e4SLinus Torvalds #include <net/route.h> 1031da177e4SLinus Torvalds #include <net/ip_fib.h> 104295f7324SArnaldo Carvalho de Melo #include <net/inet_connection_sock.h> 1051da177e4SLinus Torvalds #include <net/tcp.h> 1061da177e4SLinus Torvalds #include <net/udp.h> 107ba4e58ecSGerrit Renker #include <net/udplite.h> 108c319b4d7SVasiliy Kulikov #include <net/ping.h> 1091da177e4SLinus Torvalds #include <linux/skbuff.h> 1101da177e4SLinus Torvalds #include <net/sock.h> 1111da177e4SLinus Torvalds #include <net/raw.h> 1121da177e4SLinus Torvalds #include <net/icmp.h> 1131da177e4SLinus Torvalds #include <net/ipip.h> 1141da177e4SLinus Torvalds #include <net/inet_common.h> 1151da177e4SLinus Torvalds #include <net/xfrm.h> 1169b4661bdSPavel Emelyanov #include <net/net_namespace.h> 1171da177e4SLinus Torvalds #ifdef CONFIG_IP_MROUTE 1181da177e4SLinus Torvalds #include <linux/mroute.h> 1191da177e4SLinus Torvalds #endif 1201da177e4SLinus Torvalds 1211da177e4SLinus Torvalds 1221da177e4SLinus Torvalds /* The inetsw table contains everything that inet_create needs to 1231da177e4SLinus Torvalds * build a new socket. 1241da177e4SLinus Torvalds */ 1251da177e4SLinus Torvalds static struct list_head inetsw[SOCK_MAX]; 1261da177e4SLinus Torvalds static DEFINE_SPINLOCK(inetsw_lock); 1271da177e4SLinus Torvalds 1289ba63979SPavel Emelyanov struct ipv4_config ipv4_config; 1299ba63979SPavel Emelyanov EXPORT_SYMBOL(ipv4_config); 1309ba63979SPavel Emelyanov 1311da177e4SLinus Torvalds /* New destruction routine */ 1321da177e4SLinus Torvalds 1331da177e4SLinus Torvalds void inet_sock_destruct(struct sock *sk) 1341da177e4SLinus Torvalds { 1351da177e4SLinus Torvalds struct inet_sock *inet = inet_sk(sk); 1361da177e4SLinus Torvalds 1371da177e4SLinus Torvalds __skb_queue_purge(&sk->sk_receive_queue); 1381da177e4SLinus Torvalds __skb_queue_purge(&sk->sk_error_queue); 1391da177e4SLinus Torvalds 14095766fffSHideo Aoki sk_mem_reclaim(sk); 14195766fffSHideo Aoki 1421da177e4SLinus Torvalds if (sk->sk_type == SOCK_STREAM && sk->sk_state != TCP_CLOSE) { 1433d1427f8SEric Dumazet pr_err("Attempt to release TCP socket in state %d %p\n", 1441da177e4SLinus Torvalds sk->sk_state, sk); 1451da177e4SLinus Torvalds return; 1461da177e4SLinus Torvalds } 1471da177e4SLinus Torvalds if (!sock_flag(sk, SOCK_DEAD)) { 1483d1427f8SEric Dumazet pr_err("Attempt to release alive inet socket %p\n", sk); 1491da177e4SLinus Torvalds return; 1501da177e4SLinus Torvalds } 1511da177e4SLinus Torvalds 152547b792cSIlpo Järvinen WARN_ON(atomic_read(&sk->sk_rmem_alloc)); 153547b792cSIlpo Järvinen WARN_ON(atomic_read(&sk->sk_wmem_alloc)); 154547b792cSIlpo Järvinen WARN_ON(sk->sk_wmem_queued); 155547b792cSIlpo Järvinen WARN_ON(sk->sk_forward_alloc); 1561da177e4SLinus Torvalds 157f6d8bd05SEric Dumazet kfree(rcu_dereference_protected(inet->inet_opt, 1)); 158b6c6712aSEric Dumazet dst_release(rcu_dereference_check(sk->sk_dst_cache, 1)); 159e6848976SArnaldo Carvalho de Melo sk_refcnt_debug_dec(sk); 1601da177e4SLinus Torvalds } 1613d1427f8SEric Dumazet EXPORT_SYMBOL(inet_sock_destruct); 1621da177e4SLinus Torvalds 1631da177e4SLinus Torvalds /* 1641da177e4SLinus Torvalds * The routines beyond this point handle the behaviour of an AF_INET 1651da177e4SLinus Torvalds * socket object. Mostly it punts to the subprotocols of IP to do 1661da177e4SLinus Torvalds * the work. 1671da177e4SLinus Torvalds */ 1681da177e4SLinus Torvalds 1691da177e4SLinus Torvalds /* 1701da177e4SLinus Torvalds * Automatically bind an unbound socket. 1711da177e4SLinus Torvalds */ 1721da177e4SLinus Torvalds 1731da177e4SLinus Torvalds static int inet_autobind(struct sock *sk) 1741da177e4SLinus Torvalds { 1751da177e4SLinus Torvalds struct inet_sock *inet; 1761da177e4SLinus Torvalds /* We may need to bind the socket. */ 1771da177e4SLinus Torvalds lock_sock(sk); 1781da177e4SLinus Torvalds inet = inet_sk(sk); 179c720c7e8SEric Dumazet if (!inet->inet_num) { 1801da177e4SLinus Torvalds if (sk->sk_prot->get_port(sk, 0)) { 1811da177e4SLinus Torvalds release_sock(sk); 1821da177e4SLinus Torvalds return -EAGAIN; 1831da177e4SLinus Torvalds } 184c720c7e8SEric Dumazet inet->inet_sport = htons(inet->inet_num); 1851da177e4SLinus Torvalds } 1861da177e4SLinus Torvalds release_sock(sk); 1871da177e4SLinus Torvalds return 0; 1881da177e4SLinus Torvalds } 1891da177e4SLinus Torvalds 1901da177e4SLinus Torvalds /* 1911da177e4SLinus Torvalds * Move a socket into listening state. 1921da177e4SLinus Torvalds */ 1931da177e4SLinus Torvalds int inet_listen(struct socket *sock, int backlog) 1941da177e4SLinus Torvalds { 1951da177e4SLinus Torvalds struct sock *sk = sock->sk; 1961da177e4SLinus Torvalds unsigned char old_state; 1971da177e4SLinus Torvalds int err; 1981da177e4SLinus Torvalds 1991da177e4SLinus Torvalds lock_sock(sk); 2001da177e4SLinus Torvalds 2011da177e4SLinus Torvalds err = -EINVAL; 2021da177e4SLinus Torvalds if (sock->state != SS_UNCONNECTED || sock->type != SOCK_STREAM) 2031da177e4SLinus Torvalds goto out; 2041da177e4SLinus Torvalds 2051da177e4SLinus Torvalds old_state = sk->sk_state; 2061da177e4SLinus Torvalds if (!((1 << old_state) & (TCPF_CLOSE | TCPF_LISTEN))) 2071da177e4SLinus Torvalds goto out; 2081da177e4SLinus Torvalds 2091da177e4SLinus Torvalds /* Really, if the socket is already in listen state 2101da177e4SLinus Torvalds * we can only allow the backlog to be adjusted. 2111da177e4SLinus Torvalds */ 2121da177e4SLinus Torvalds if (old_state != TCP_LISTEN) { 21372a3effaSEric Dumazet err = inet_csk_listen_start(sk, backlog); 2141da177e4SLinus Torvalds if (err) 2151da177e4SLinus Torvalds goto out; 2161da177e4SLinus Torvalds } 2171da177e4SLinus Torvalds sk->sk_max_ack_backlog = backlog; 2181da177e4SLinus Torvalds err = 0; 2191da177e4SLinus Torvalds 2201da177e4SLinus Torvalds out: 2211da177e4SLinus Torvalds release_sock(sk); 2221da177e4SLinus Torvalds return err; 2231da177e4SLinus Torvalds } 2243d1427f8SEric Dumazet EXPORT_SYMBOL(inet_listen); 2251da177e4SLinus Torvalds 226be776281SEric Dumazet u32 inet_ehash_secret __read_mostly; 227b3da2cf3SDavid S. Miller EXPORT_SYMBOL(inet_ehash_secret); 228b3da2cf3SDavid S. Miller 229be776281SEric Dumazet /* 230be776281SEric Dumazet * inet_ehash_secret must be set exactly once 231be776281SEric Dumazet */ 232b3da2cf3SDavid S. Miller void build_ehash_secret(void) 233b3da2cf3SDavid S. Miller { 234be776281SEric Dumazet u32 rnd; 23549e8ab03SEric Dumazet 236be776281SEric Dumazet do { 237be776281SEric Dumazet get_random_bytes(&rnd, sizeof(rnd)); 238be776281SEric Dumazet } while (rnd == 0); 23949e8ab03SEric Dumazet 24049e8ab03SEric Dumazet cmpxchg(&inet_ehash_secret, 0, rnd); 241b3da2cf3SDavid S. Miller } 242b3da2cf3SDavid S. Miller EXPORT_SYMBOL(build_ehash_secret); 243b3da2cf3SDavid S. Miller 2442342fd7eSDenis V. Lunev static inline int inet_netns_ok(struct net *net, int protocol) 2452342fd7eSDenis V. Lunev { 2462342fd7eSDenis V. Lunev int hash; 24732613090SAlexey Dobriyan const struct net_protocol *ipprot; 2482342fd7eSDenis V. Lunev 24904f258ceSEric Dumazet if (net_eq(net, &init_net)) 2502342fd7eSDenis V. Lunev return 1; 2512342fd7eSDenis V. Lunev 2522342fd7eSDenis V. Lunev hash = protocol & (MAX_INET_PROTOS - 1); 2532342fd7eSDenis V. Lunev ipprot = rcu_dereference(inet_protos[hash]); 2542342fd7eSDenis V. Lunev 2552342fd7eSDenis V. Lunev if (ipprot == NULL) 2562342fd7eSDenis V. Lunev /* raw IP is OK */ 2572342fd7eSDenis V. Lunev return 1; 2582342fd7eSDenis V. Lunev return ipprot->netns_ok; 2592342fd7eSDenis V. Lunev } 2602342fd7eSDenis V. Lunev 2611da177e4SLinus Torvalds /* 2621da177e4SLinus Torvalds * Create an inet socket. 2631da177e4SLinus Torvalds */ 2641da177e4SLinus Torvalds 2653f378b68SEric Paris static int inet_create(struct net *net, struct socket *sock, int protocol, 2663f378b68SEric Paris int kern) 2671da177e4SLinus Torvalds { 2681da177e4SLinus Torvalds struct sock *sk; 2691da177e4SLinus Torvalds struct inet_protosw *answer; 2701da177e4SLinus Torvalds struct inet_sock *inet; 2711da177e4SLinus Torvalds struct proto *answer_prot; 2721da177e4SLinus Torvalds unsigned char answer_flags; 2731da177e4SLinus Torvalds char answer_no_check; 274bb97d31fSArnaldo Carvalho de Melo int try_loading_module = 0; 27586c8f9d1SHerbert Xu int err; 2761da177e4SLinus Torvalds 27704f258ceSEric Dumazet if (unlikely(!inet_ehash_secret)) 27804f258ceSEric Dumazet if (sock->type != SOCK_RAW && sock->type != SOCK_DGRAM) 279b3da2cf3SDavid S. Miller build_ehash_secret(); 280b3da2cf3SDavid S. Miller 2811da177e4SLinus Torvalds sock->state = SS_UNCONNECTED; 2821da177e4SLinus Torvalds 2831da177e4SLinus Torvalds /* Look for the requested type/protocol pair. */ 284bb97d31fSArnaldo Carvalho de Melo lookup_protocol: 28586c8f9d1SHerbert Xu err = -ESOCKTNOSUPPORT; 2861da177e4SLinus Torvalds rcu_read_lock(); 287696adfe8SPaul E. McKenney list_for_each_entry_rcu(answer, &inetsw[sock->type], list) { 2881da177e4SLinus Torvalds 289696adfe8SPaul E. McKenney err = 0; 2901da177e4SLinus Torvalds /* Check the non-wild match. */ 2911da177e4SLinus Torvalds if (protocol == answer->protocol) { 2921da177e4SLinus Torvalds if (protocol != IPPROTO_IP) 2931da177e4SLinus Torvalds break; 2941da177e4SLinus Torvalds } else { 2951da177e4SLinus Torvalds /* Check for the two wild cases. */ 2961da177e4SLinus Torvalds if (IPPROTO_IP == protocol) { 2971da177e4SLinus Torvalds protocol = answer->protocol; 2981da177e4SLinus Torvalds break; 2991da177e4SLinus Torvalds } 3001da177e4SLinus Torvalds if (IPPROTO_IP == answer->protocol) 3011da177e4SLinus Torvalds break; 3021da177e4SLinus Torvalds } 30386c8f9d1SHerbert Xu err = -EPROTONOSUPPORT; 3041da177e4SLinus Torvalds } 3051da177e4SLinus Torvalds 306696adfe8SPaul E. McKenney if (unlikely(err)) { 307bb97d31fSArnaldo Carvalho de Melo if (try_loading_module < 2) { 308bb97d31fSArnaldo Carvalho de Melo rcu_read_unlock(); 309bb97d31fSArnaldo Carvalho de Melo /* 310bb97d31fSArnaldo Carvalho de Melo * Be more specific, e.g. net-pf-2-proto-132-type-1 311bb97d31fSArnaldo Carvalho de Melo * (net-pf-PF_INET-proto-IPPROTO_SCTP-type-SOCK_STREAM) 312bb97d31fSArnaldo Carvalho de Melo */ 313bb97d31fSArnaldo Carvalho de Melo if (++try_loading_module == 1) 314bb97d31fSArnaldo Carvalho de Melo request_module("net-pf-%d-proto-%d-type-%d", 315bb97d31fSArnaldo Carvalho de Melo PF_INET, protocol, sock->type); 316bb97d31fSArnaldo Carvalho de Melo /* 317bb97d31fSArnaldo Carvalho de Melo * Fall back to generic, e.g. net-pf-2-proto-132 318bb97d31fSArnaldo Carvalho de Melo * (net-pf-PF_INET-proto-IPPROTO_SCTP) 319bb97d31fSArnaldo Carvalho de Melo */ 320bb97d31fSArnaldo Carvalho de Melo else 321bb97d31fSArnaldo Carvalho de Melo request_module("net-pf-%d-proto-%d", 322bb97d31fSArnaldo Carvalho de Melo PF_INET, protocol); 323bb97d31fSArnaldo Carvalho de Melo goto lookup_protocol; 324bb97d31fSArnaldo Carvalho de Melo } else 3251da177e4SLinus Torvalds goto out_rcu_unlock; 326bb97d31fSArnaldo Carvalho de Melo } 327bb97d31fSArnaldo Carvalho de Melo 3281da177e4SLinus Torvalds err = -EPERM; 329c84b3268SEric Paris if (sock->type == SOCK_RAW && !kern && !capable(CAP_NET_RAW)) 3301da177e4SLinus Torvalds goto out_rcu_unlock; 3311da177e4SLinus Torvalds 3322342fd7eSDenis V. Lunev err = -EAFNOSUPPORT; 3332342fd7eSDenis V. Lunev if (!inet_netns_ok(net, protocol)) 3342342fd7eSDenis V. Lunev goto out_rcu_unlock; 3352342fd7eSDenis V. Lunev 3361da177e4SLinus Torvalds sock->ops = answer->ops; 3371da177e4SLinus Torvalds answer_prot = answer->prot; 3381da177e4SLinus Torvalds answer_no_check = answer->no_check; 3391da177e4SLinus Torvalds answer_flags = answer->flags; 3401da177e4SLinus Torvalds rcu_read_unlock(); 3411da177e4SLinus Torvalds 342547b792cSIlpo Järvinen WARN_ON(answer_prot->slab == NULL); 3431da177e4SLinus Torvalds 3441da177e4SLinus Torvalds err = -ENOBUFS; 3456257ff21SPavel Emelyanov sk = sk_alloc(net, PF_INET, GFP_KERNEL, answer_prot); 3461da177e4SLinus Torvalds if (sk == NULL) 3471da177e4SLinus Torvalds goto out; 3481da177e4SLinus Torvalds 3491da177e4SLinus Torvalds err = 0; 3501da177e4SLinus Torvalds sk->sk_no_check = answer_no_check; 3511da177e4SLinus Torvalds if (INET_PROTOSW_REUSE & answer_flags) 3521da177e4SLinus Torvalds sk->sk_reuse = 1; 3531da177e4SLinus Torvalds 3541da177e4SLinus Torvalds inet = inet_sk(sk); 355469de9b9SPaul Moore inet->is_icsk = (INET_PROTOSW_ICSK & answer_flags) != 0; 3561da177e4SLinus Torvalds 3577b2ff18eSJiri Olsa inet->nodefrag = 0; 3587b2ff18eSJiri Olsa 3591da177e4SLinus Torvalds if (SOCK_RAW == sock->type) { 360c720c7e8SEric Dumazet inet->inet_num = protocol; 3611da177e4SLinus Torvalds if (IPPROTO_RAW == protocol) 3621da177e4SLinus Torvalds inet->hdrincl = 1; 3631da177e4SLinus Torvalds } 3641da177e4SLinus Torvalds 3651da177e4SLinus Torvalds if (ipv4_config.no_pmtu_disc) 3661da177e4SLinus Torvalds inet->pmtudisc = IP_PMTUDISC_DONT; 3671da177e4SLinus Torvalds else 3681da177e4SLinus Torvalds inet->pmtudisc = IP_PMTUDISC_WANT; 3691da177e4SLinus Torvalds 370c720c7e8SEric Dumazet inet->inet_id = 0; 3711da177e4SLinus Torvalds 3721da177e4SLinus Torvalds sock_init_data(sock, sk); 3731da177e4SLinus Torvalds 3741da177e4SLinus Torvalds sk->sk_destruct = inet_sock_destruct; 3751da177e4SLinus Torvalds sk->sk_protocol = protocol; 3761da177e4SLinus Torvalds sk->sk_backlog_rcv = sk->sk_prot->backlog_rcv; 3771da177e4SLinus Torvalds 3781da177e4SLinus Torvalds inet->uc_ttl = -1; 3791da177e4SLinus Torvalds inet->mc_loop = 1; 3801da177e4SLinus Torvalds inet->mc_ttl = 1; 381f771bef9SNivedita Singhvi inet->mc_all = 1; 3821da177e4SLinus Torvalds inet->mc_index = 0; 3831da177e4SLinus Torvalds inet->mc_list = NULL; 3844c507d28SJiri Benc inet->rcv_tos = 0; 3851da177e4SLinus Torvalds 386e6848976SArnaldo Carvalho de Melo sk_refcnt_debug_inc(sk); 3871da177e4SLinus Torvalds 388c720c7e8SEric Dumazet if (inet->inet_num) { 3891da177e4SLinus Torvalds /* It assumes that any protocol which allows 3901da177e4SLinus Torvalds * the user to assign a number at socket 3911da177e4SLinus Torvalds * creation time automatically 3921da177e4SLinus Torvalds * shares. 3931da177e4SLinus Torvalds */ 394c720c7e8SEric Dumazet inet->inet_sport = htons(inet->inet_num); 3951da177e4SLinus Torvalds /* Add to protocol hash chains. */ 3961da177e4SLinus Torvalds sk->sk_prot->hash(sk); 3971da177e4SLinus Torvalds } 3981da177e4SLinus Torvalds 3991da177e4SLinus Torvalds if (sk->sk_prot->init) { 4001da177e4SLinus Torvalds err = sk->sk_prot->init(sk); 4011da177e4SLinus Torvalds if (err) 4021da177e4SLinus Torvalds sk_common_release(sk); 4031da177e4SLinus Torvalds } 4041da177e4SLinus Torvalds out: 4051da177e4SLinus Torvalds return err; 4061da177e4SLinus Torvalds out_rcu_unlock: 4071da177e4SLinus Torvalds rcu_read_unlock(); 4081da177e4SLinus Torvalds goto out; 4091da177e4SLinus Torvalds } 4101da177e4SLinus Torvalds 4111da177e4SLinus Torvalds 4121da177e4SLinus Torvalds /* 4131da177e4SLinus Torvalds * The peer socket should always be NULL (or else). When we call this 4141da177e4SLinus Torvalds * function we are destroying the object and from then on nobody 4151da177e4SLinus Torvalds * should refer to it. 4161da177e4SLinus Torvalds */ 4171da177e4SLinus Torvalds int inet_release(struct socket *sock) 4181da177e4SLinus Torvalds { 4191da177e4SLinus Torvalds struct sock *sk = sock->sk; 4201da177e4SLinus Torvalds 4211da177e4SLinus Torvalds if (sk) { 4221da177e4SLinus Torvalds long timeout; 4231da177e4SLinus Torvalds 424c58dc01bSDavid S. Miller sock_rps_reset_flow(sk); 425fec5e652STom Herbert 4261da177e4SLinus Torvalds /* Applications forget to leave groups before exiting */ 4271da177e4SLinus Torvalds ip_mc_drop_socket(sk); 4281da177e4SLinus Torvalds 4291da177e4SLinus Torvalds /* If linger is set, we don't return until the close 4301da177e4SLinus Torvalds * is complete. Otherwise we return immediately. The 4311da177e4SLinus Torvalds * actually closing is done the same either way. 4321da177e4SLinus Torvalds * 4331da177e4SLinus Torvalds * If the close is due to the process exiting, we never 4341da177e4SLinus Torvalds * linger.. 4351da177e4SLinus Torvalds */ 4361da177e4SLinus Torvalds timeout = 0; 4371da177e4SLinus Torvalds if (sock_flag(sk, SOCK_LINGER) && 4381da177e4SLinus Torvalds !(current->flags & PF_EXITING)) 4391da177e4SLinus Torvalds timeout = sk->sk_lingertime; 4401da177e4SLinus Torvalds sock->sk = NULL; 4411da177e4SLinus Torvalds sk->sk_prot->close(sk, timeout); 4421da177e4SLinus Torvalds } 4431da177e4SLinus Torvalds return 0; 4441da177e4SLinus Torvalds } 4453d1427f8SEric Dumazet EXPORT_SYMBOL(inet_release); 4461da177e4SLinus Torvalds 4471da177e4SLinus Torvalds /* It is off by default, see below. */ 448ab32ea5dSBrian Haley int sysctl_ip_nonlocal_bind __read_mostly; 4493d1427f8SEric Dumazet EXPORT_SYMBOL(sysctl_ip_nonlocal_bind); 4501da177e4SLinus Torvalds 4511da177e4SLinus Torvalds int inet_bind(struct socket *sock, struct sockaddr *uaddr, int addr_len) 4521da177e4SLinus Torvalds { 4531da177e4SLinus Torvalds struct sockaddr_in *addr = (struct sockaddr_in *)uaddr; 4541da177e4SLinus Torvalds struct sock *sk = sock->sk; 4551da177e4SLinus Torvalds struct inet_sock *inet = inet_sk(sk); 4561da177e4SLinus Torvalds unsigned short snum; 4571da177e4SLinus Torvalds int chk_addr_ret; 4581da177e4SLinus Torvalds int err; 4591da177e4SLinus Torvalds 4601da177e4SLinus Torvalds /* If the socket has its own bind function then use it. (RAW) */ 4611da177e4SLinus Torvalds if (sk->sk_prot->bind) { 4621da177e4SLinus Torvalds err = sk->sk_prot->bind(sk, uaddr, addr_len); 4631da177e4SLinus Torvalds goto out; 4641da177e4SLinus Torvalds } 4651da177e4SLinus Torvalds err = -EINVAL; 4661da177e4SLinus Torvalds if (addr_len < sizeof(struct sockaddr_in)) 4671da177e4SLinus Torvalds goto out; 4681da177e4SLinus Torvalds 469c349a528SMarcus Meissner if (addr->sin_family != AF_INET) { 47029c486dfSEric Dumazet /* Compatibility games : accept AF_UNSPEC (mapped to AF_INET) 47129c486dfSEric Dumazet * only if s_addr is INADDR_ANY. 47229c486dfSEric Dumazet */ 473c349a528SMarcus Meissner err = -EAFNOSUPPORT; 47429c486dfSEric Dumazet if (addr->sin_family != AF_UNSPEC || 47529c486dfSEric Dumazet addr->sin_addr.s_addr != htonl(INADDR_ANY)) 476d0733d2eSMarcus Meissner goto out; 477c349a528SMarcus Meissner } 478d0733d2eSMarcus Meissner 4793b1e0a65SYOSHIFUJI Hideaki chk_addr_ret = inet_addr_type(sock_net(sk), addr->sin_addr.s_addr); 4801da177e4SLinus Torvalds 4811da177e4SLinus Torvalds /* Not specified by any standard per-se, however it breaks too 4821da177e4SLinus Torvalds * many applications when removed. It is unfortunate since 4831da177e4SLinus Torvalds * allowing applications to make a non-local bind solves 4841da177e4SLinus Torvalds * several problems with systems using dynamic addressing. 4851da177e4SLinus Torvalds * (ie. your servers still start up even if your ISDN link 4861da177e4SLinus Torvalds * is temporarily down) 4871da177e4SLinus Torvalds */ 4881da177e4SLinus Torvalds err = -EADDRNOTAVAIL; 4891da177e4SLinus Torvalds if (!sysctl_ip_nonlocal_bind && 490b9fb1506STóth László Attila !(inet->freebind || inet->transparent) && 491e6f1cebfSAl Viro addr->sin_addr.s_addr != htonl(INADDR_ANY) && 4921da177e4SLinus Torvalds chk_addr_ret != RTN_LOCAL && 4931da177e4SLinus Torvalds chk_addr_ret != RTN_MULTICAST && 4941da177e4SLinus Torvalds chk_addr_ret != RTN_BROADCAST) 4951da177e4SLinus Torvalds goto out; 4961da177e4SLinus Torvalds 4971da177e4SLinus Torvalds snum = ntohs(addr->sin_port); 4981da177e4SLinus Torvalds err = -EACCES; 4991da177e4SLinus Torvalds if (snum && snum < PROT_SOCK && !capable(CAP_NET_BIND_SERVICE)) 5001da177e4SLinus Torvalds goto out; 5011da177e4SLinus Torvalds 5021da177e4SLinus Torvalds /* We keep a pair of addresses. rcv_saddr is the one 5031da177e4SLinus Torvalds * used by hash lookups, and saddr is used for transmit. 5041da177e4SLinus Torvalds * 5051da177e4SLinus Torvalds * In the BSD API these are the same except where it 5061da177e4SLinus Torvalds * would be illegal to use them (multicast/broadcast) in 5071da177e4SLinus Torvalds * which case the sending device address is used. 5081da177e4SLinus Torvalds */ 5091da177e4SLinus Torvalds lock_sock(sk); 5101da177e4SLinus Torvalds 5111da177e4SLinus Torvalds /* Check these errors (active socket, double bind). */ 5121da177e4SLinus Torvalds err = -EINVAL; 513c720c7e8SEric Dumazet if (sk->sk_state != TCP_CLOSE || inet->inet_num) 5141da177e4SLinus Torvalds goto out_release_sock; 5151da177e4SLinus Torvalds 516c720c7e8SEric Dumazet inet->inet_rcv_saddr = inet->inet_saddr = addr->sin_addr.s_addr; 5171da177e4SLinus Torvalds if (chk_addr_ret == RTN_MULTICAST || chk_addr_ret == RTN_BROADCAST) 518c720c7e8SEric Dumazet inet->inet_saddr = 0; /* Use device */ 5191da177e4SLinus Torvalds 5201da177e4SLinus Torvalds /* Make sure we are allowed to bind here. */ 5211da177e4SLinus Torvalds if (sk->sk_prot->get_port(sk, snum)) { 522c720c7e8SEric Dumazet inet->inet_saddr = inet->inet_rcv_saddr = 0; 5231da177e4SLinus Torvalds err = -EADDRINUSE; 5241da177e4SLinus Torvalds goto out_release_sock; 5251da177e4SLinus Torvalds } 5261da177e4SLinus Torvalds 527c720c7e8SEric Dumazet if (inet->inet_rcv_saddr) 5281da177e4SLinus Torvalds sk->sk_userlocks |= SOCK_BINDADDR_LOCK; 5291da177e4SLinus Torvalds if (snum) 5301da177e4SLinus Torvalds sk->sk_userlocks |= SOCK_BINDPORT_LOCK; 531c720c7e8SEric Dumazet inet->inet_sport = htons(inet->inet_num); 532c720c7e8SEric Dumazet inet->inet_daddr = 0; 533c720c7e8SEric Dumazet inet->inet_dport = 0; 5341da177e4SLinus Torvalds sk_dst_reset(sk); 5351da177e4SLinus Torvalds err = 0; 5361da177e4SLinus Torvalds out_release_sock: 5371da177e4SLinus Torvalds release_sock(sk); 5381da177e4SLinus Torvalds out: 5391da177e4SLinus Torvalds return err; 5401da177e4SLinus Torvalds } 5413d1427f8SEric Dumazet EXPORT_SYMBOL(inet_bind); 5421da177e4SLinus Torvalds 5431da177e4SLinus Torvalds int inet_dgram_connect(struct socket *sock, struct sockaddr * uaddr, 5441da177e4SLinus Torvalds int addr_len, int flags) 5451da177e4SLinus Torvalds { 5461da177e4SLinus Torvalds struct sock *sk = sock->sk; 5471da177e4SLinus Torvalds 5486503d961SChangli Gao if (addr_len < sizeof(uaddr->sa_family)) 5496503d961SChangli Gao return -EINVAL; 5501da177e4SLinus Torvalds if (uaddr->sa_family == AF_UNSPEC) 5511da177e4SLinus Torvalds return sk->sk_prot->disconnect(sk, flags); 5521da177e4SLinus Torvalds 553c720c7e8SEric Dumazet if (!inet_sk(sk)->inet_num && inet_autobind(sk)) 5541da177e4SLinus Torvalds return -EAGAIN; 5551da177e4SLinus Torvalds return sk->sk_prot->connect(sk, (struct sockaddr *)uaddr, addr_len); 5561da177e4SLinus Torvalds } 5573d1427f8SEric Dumazet EXPORT_SYMBOL(inet_dgram_connect); 5581da177e4SLinus Torvalds 5591da177e4SLinus Torvalds static long inet_wait_for_connect(struct sock *sk, long timeo) 5601da177e4SLinus Torvalds { 5611da177e4SLinus Torvalds DEFINE_WAIT(wait); 5621da177e4SLinus Torvalds 563aa395145SEric Dumazet prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE); 5641da177e4SLinus Torvalds 5651da177e4SLinus Torvalds /* Basic assumption: if someone sets sk->sk_err, he _must_ 5661da177e4SLinus Torvalds * change state of the socket from TCP_SYN_*. 5671da177e4SLinus Torvalds * Connect() does not allow to get error notifications 5681da177e4SLinus Torvalds * without closing the socket. 5691da177e4SLinus Torvalds */ 5701da177e4SLinus Torvalds while ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV)) { 5711da177e4SLinus Torvalds release_sock(sk); 5721da177e4SLinus Torvalds timeo = schedule_timeout(timeo); 5731da177e4SLinus Torvalds lock_sock(sk); 5741da177e4SLinus Torvalds if (signal_pending(current) || !timeo) 5751da177e4SLinus Torvalds break; 576aa395145SEric Dumazet prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE); 5771da177e4SLinus Torvalds } 578aa395145SEric Dumazet finish_wait(sk_sleep(sk), &wait); 5791da177e4SLinus Torvalds return timeo; 5801da177e4SLinus Torvalds } 5811da177e4SLinus Torvalds 5821da177e4SLinus Torvalds /* 5831da177e4SLinus Torvalds * Connect to a remote host. There is regrettably still a little 5841da177e4SLinus Torvalds * TCP 'magic' in here. 5851da177e4SLinus Torvalds */ 5861da177e4SLinus Torvalds int inet_stream_connect(struct socket *sock, struct sockaddr *uaddr, 5871da177e4SLinus Torvalds int addr_len, int flags) 5881da177e4SLinus Torvalds { 5891da177e4SLinus Torvalds struct sock *sk = sock->sk; 5901da177e4SLinus Torvalds int err; 5911da177e4SLinus Torvalds long timeo; 5921da177e4SLinus Torvalds 5936503d961SChangli Gao if (addr_len < sizeof(uaddr->sa_family)) 5946503d961SChangli Gao return -EINVAL; 5956503d961SChangli Gao 5961da177e4SLinus Torvalds lock_sock(sk); 5971da177e4SLinus Torvalds 5981da177e4SLinus Torvalds if (uaddr->sa_family == AF_UNSPEC) { 5991da177e4SLinus Torvalds err = sk->sk_prot->disconnect(sk, flags); 6001da177e4SLinus Torvalds sock->state = err ? SS_DISCONNECTING : SS_UNCONNECTED; 6011da177e4SLinus Torvalds goto out; 6021da177e4SLinus Torvalds } 6031da177e4SLinus Torvalds 6041da177e4SLinus Torvalds switch (sock->state) { 6051da177e4SLinus Torvalds default: 6061da177e4SLinus Torvalds err = -EINVAL; 6071da177e4SLinus Torvalds goto out; 6081da177e4SLinus Torvalds case SS_CONNECTED: 6091da177e4SLinus Torvalds err = -EISCONN; 6101da177e4SLinus Torvalds goto out; 6111da177e4SLinus Torvalds case SS_CONNECTING: 6121da177e4SLinus Torvalds err = -EALREADY; 6131da177e4SLinus Torvalds /* Fall out of switch with err, set for this state */ 6141da177e4SLinus Torvalds break; 6151da177e4SLinus Torvalds case SS_UNCONNECTED: 6161da177e4SLinus Torvalds err = -EISCONN; 6171da177e4SLinus Torvalds if (sk->sk_state != TCP_CLOSE) 6181da177e4SLinus Torvalds goto out; 6191da177e4SLinus Torvalds 6201da177e4SLinus Torvalds err = sk->sk_prot->connect(sk, uaddr, addr_len); 6211da177e4SLinus Torvalds if (err < 0) 6221da177e4SLinus Torvalds goto out; 6231da177e4SLinus Torvalds 6241da177e4SLinus Torvalds sock->state = SS_CONNECTING; 6251da177e4SLinus Torvalds 6261da177e4SLinus Torvalds /* Just entered SS_CONNECTING state; the only 6271da177e4SLinus Torvalds * difference is that return value in non-blocking 6281da177e4SLinus Torvalds * case is EINPROGRESS, rather than EALREADY. 6291da177e4SLinus Torvalds */ 6301da177e4SLinus Torvalds err = -EINPROGRESS; 6311da177e4SLinus Torvalds break; 6321da177e4SLinus Torvalds } 6331da177e4SLinus Torvalds 6341da177e4SLinus Torvalds timeo = sock_sndtimeo(sk, flags & O_NONBLOCK); 6351da177e4SLinus Torvalds 6361da177e4SLinus Torvalds if ((1 << sk->sk_state) & (TCPF_SYN_SENT | TCPF_SYN_RECV)) { 6371da177e4SLinus Torvalds /* Error code is set above */ 6381da177e4SLinus Torvalds if (!timeo || !inet_wait_for_connect(sk, timeo)) 6391da177e4SLinus Torvalds goto out; 6401da177e4SLinus Torvalds 6411da177e4SLinus Torvalds err = sock_intr_errno(timeo); 6421da177e4SLinus Torvalds if (signal_pending(current)) 6431da177e4SLinus Torvalds goto out; 6441da177e4SLinus Torvalds } 6451da177e4SLinus Torvalds 6461da177e4SLinus Torvalds /* Connection was closed by RST, timeout, ICMP error 6471da177e4SLinus Torvalds * or another process disconnected us. 6481da177e4SLinus Torvalds */ 6491da177e4SLinus Torvalds if (sk->sk_state == TCP_CLOSE) 6501da177e4SLinus Torvalds goto sock_error; 6511da177e4SLinus Torvalds 6521da177e4SLinus Torvalds /* sk->sk_err may be not zero now, if RECVERR was ordered by user 6531da177e4SLinus Torvalds * and error was received after socket entered established state. 6541da177e4SLinus Torvalds * Hence, it is handled normally after connect() return successfully. 6551da177e4SLinus Torvalds */ 6561da177e4SLinus Torvalds 6571da177e4SLinus Torvalds sock->state = SS_CONNECTED; 6581da177e4SLinus Torvalds err = 0; 6591da177e4SLinus Torvalds out: 6601da177e4SLinus Torvalds release_sock(sk); 6611da177e4SLinus Torvalds return err; 6621da177e4SLinus Torvalds 6631da177e4SLinus Torvalds sock_error: 6641da177e4SLinus Torvalds err = sock_error(sk) ? : -ECONNABORTED; 6651da177e4SLinus Torvalds sock->state = SS_UNCONNECTED; 6661da177e4SLinus Torvalds if (sk->sk_prot->disconnect(sk, flags)) 6671da177e4SLinus Torvalds sock->state = SS_DISCONNECTING; 6681da177e4SLinus Torvalds goto out; 6691da177e4SLinus Torvalds } 6703d1427f8SEric Dumazet EXPORT_SYMBOL(inet_stream_connect); 6711da177e4SLinus Torvalds 6721da177e4SLinus Torvalds /* 6731da177e4SLinus Torvalds * Accept a pending connection. The TCP layer now gives BSD semantics. 6741da177e4SLinus Torvalds */ 6751da177e4SLinus Torvalds 6761da177e4SLinus Torvalds int inet_accept(struct socket *sock, struct socket *newsock, int flags) 6771da177e4SLinus Torvalds { 6781da177e4SLinus Torvalds struct sock *sk1 = sock->sk; 6791da177e4SLinus Torvalds int err = -EINVAL; 6801da177e4SLinus Torvalds struct sock *sk2 = sk1->sk_prot->accept(sk1, flags, &err); 6811da177e4SLinus Torvalds 6821da177e4SLinus Torvalds if (!sk2) 6831da177e4SLinus Torvalds goto do_err; 6841da177e4SLinus Torvalds 6851da177e4SLinus Torvalds lock_sock(sk2); 6861da177e4SLinus Torvalds 6871eddceadSEric Dumazet sock_rps_record_flow(sk2); 688547b792cSIlpo Järvinen WARN_ON(!((1 << sk2->sk_state) & 689547b792cSIlpo Järvinen (TCPF_ESTABLISHED | TCPF_CLOSE_WAIT | TCPF_CLOSE))); 6901da177e4SLinus Torvalds 6911da177e4SLinus Torvalds sock_graft(sk2, newsock); 6921da177e4SLinus Torvalds 6931da177e4SLinus Torvalds newsock->state = SS_CONNECTED; 6941da177e4SLinus Torvalds err = 0; 6951da177e4SLinus Torvalds release_sock(sk2); 6961da177e4SLinus Torvalds do_err: 6971da177e4SLinus Torvalds return err; 6981da177e4SLinus Torvalds } 6993d1427f8SEric Dumazet EXPORT_SYMBOL(inet_accept); 7001da177e4SLinus Torvalds 7011da177e4SLinus Torvalds 7021da177e4SLinus Torvalds /* 7031da177e4SLinus Torvalds * This does both peername and sockname. 7041da177e4SLinus Torvalds */ 7051da177e4SLinus Torvalds int inet_getname(struct socket *sock, struct sockaddr *uaddr, 7061da177e4SLinus Torvalds int *uaddr_len, int peer) 7071da177e4SLinus Torvalds { 7081da177e4SLinus Torvalds struct sock *sk = sock->sk; 7091da177e4SLinus Torvalds struct inet_sock *inet = inet_sk(sk); 71038bfd8f5SCyrill Gorcunov DECLARE_SOCKADDR(struct sockaddr_in *, sin, uaddr); 7111da177e4SLinus Torvalds 7121da177e4SLinus Torvalds sin->sin_family = AF_INET; 7131da177e4SLinus Torvalds if (peer) { 714c720c7e8SEric Dumazet if (!inet->inet_dport || 7151da177e4SLinus Torvalds (((1 << sk->sk_state) & (TCPF_CLOSE | TCPF_SYN_SENT)) && 7161da177e4SLinus Torvalds peer == 1)) 7171da177e4SLinus Torvalds return -ENOTCONN; 718c720c7e8SEric Dumazet sin->sin_port = inet->inet_dport; 719c720c7e8SEric Dumazet sin->sin_addr.s_addr = inet->inet_daddr; 7201da177e4SLinus Torvalds } else { 721c720c7e8SEric Dumazet __be32 addr = inet->inet_rcv_saddr; 7221da177e4SLinus Torvalds if (!addr) 723c720c7e8SEric Dumazet addr = inet->inet_saddr; 724c720c7e8SEric Dumazet sin->sin_port = inet->inet_sport; 7251da177e4SLinus Torvalds sin->sin_addr.s_addr = addr; 7261da177e4SLinus Torvalds } 7271da177e4SLinus Torvalds memset(sin->sin_zero, 0, sizeof(sin->sin_zero)); 7281da177e4SLinus Torvalds *uaddr_len = sizeof(*sin); 7291da177e4SLinus Torvalds return 0; 7301da177e4SLinus Torvalds } 7313d1427f8SEric Dumazet EXPORT_SYMBOL(inet_getname); 7321da177e4SLinus Torvalds 7331da177e4SLinus Torvalds int inet_sendmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg, 7341da177e4SLinus Torvalds size_t size) 7351da177e4SLinus Torvalds { 7361da177e4SLinus Torvalds struct sock *sk = sock->sk; 7371da177e4SLinus Torvalds 738c58dc01bSDavid S. Miller sock_rps_record_flow(sk); 739fec5e652STom Herbert 7401da177e4SLinus Torvalds /* We may need to bind the socket. */ 7417ba42910SChangli Gao if (!inet_sk(sk)->inet_num && !sk->sk_prot->no_autobind && 7427ba42910SChangli Gao inet_autobind(sk)) 7431da177e4SLinus Torvalds return -EAGAIN; 7441da177e4SLinus Torvalds 7451da177e4SLinus Torvalds return sk->sk_prot->sendmsg(iocb, sk, msg, size); 7461da177e4SLinus Torvalds } 7473d1427f8SEric Dumazet EXPORT_SYMBOL(inet_sendmsg); 7481da177e4SLinus Torvalds 7497ba42910SChangli Gao ssize_t inet_sendpage(struct socket *sock, struct page *page, int offset, 7503d1427f8SEric Dumazet size_t size, int flags) 7511da177e4SLinus Torvalds { 7521da177e4SLinus Torvalds struct sock *sk = sock->sk; 7531da177e4SLinus Torvalds 754c58dc01bSDavid S. Miller sock_rps_record_flow(sk); 755fec5e652STom Herbert 7561da177e4SLinus Torvalds /* We may need to bind the socket. */ 7577ba42910SChangli Gao if (!inet_sk(sk)->inet_num && !sk->sk_prot->no_autobind && 7587ba42910SChangli Gao inet_autobind(sk)) 7591da177e4SLinus Torvalds return -EAGAIN; 7601da177e4SLinus Torvalds 7611da177e4SLinus Torvalds if (sk->sk_prot->sendpage) 7621da177e4SLinus Torvalds return sk->sk_prot->sendpage(sk, page, offset, size, flags); 7631da177e4SLinus Torvalds return sock_no_sendpage(sock, page, offset, size, flags); 7641da177e4SLinus Torvalds } 7657ba42910SChangli Gao EXPORT_SYMBOL(inet_sendpage); 7661da177e4SLinus Torvalds 767fec5e652STom Herbert int inet_recvmsg(struct kiocb *iocb, struct socket *sock, struct msghdr *msg, 768fec5e652STom Herbert size_t size, int flags) 769fec5e652STom Herbert { 770fec5e652STom Herbert struct sock *sk = sock->sk; 771fec5e652STom Herbert int addr_len = 0; 772fec5e652STom Herbert int err; 773fec5e652STom Herbert 774c58dc01bSDavid S. Miller sock_rps_record_flow(sk); 775fec5e652STom Herbert 776fec5e652STom Herbert err = sk->sk_prot->recvmsg(iocb, sk, msg, size, flags & MSG_DONTWAIT, 777fec5e652STom Herbert flags & ~MSG_DONTWAIT, &addr_len); 778fec5e652STom Herbert if (err >= 0) 779fec5e652STom Herbert msg->msg_namelen = addr_len; 780fec5e652STom Herbert return err; 781fec5e652STom Herbert } 782fec5e652STom Herbert EXPORT_SYMBOL(inet_recvmsg); 7831da177e4SLinus Torvalds 7841da177e4SLinus Torvalds int inet_shutdown(struct socket *sock, int how) 7851da177e4SLinus Torvalds { 7861da177e4SLinus Torvalds struct sock *sk = sock->sk; 7871da177e4SLinus Torvalds int err = 0; 7881da177e4SLinus Torvalds 7891da177e4SLinus Torvalds /* This should really check to make sure 7901da177e4SLinus Torvalds * the socket is a TCP socket. (WHY AC...) 7911da177e4SLinus Torvalds */ 7921da177e4SLinus Torvalds how++; /* maps 0->1 has the advantage of making bit 1 rcvs and 7931da177e4SLinus Torvalds 1->2 bit 2 snds. 7941da177e4SLinus Torvalds 2->3 */ 7951da177e4SLinus Torvalds if ((how & ~SHUTDOWN_MASK) || !how) /* MAXINT->0 */ 7961da177e4SLinus Torvalds return -EINVAL; 7971da177e4SLinus Torvalds 7981da177e4SLinus Torvalds lock_sock(sk); 7991da177e4SLinus Torvalds if (sock->state == SS_CONNECTING) { 8001da177e4SLinus Torvalds if ((1 << sk->sk_state) & 8011da177e4SLinus Torvalds (TCPF_SYN_SENT | TCPF_SYN_RECV | TCPF_CLOSE)) 8021da177e4SLinus Torvalds sock->state = SS_DISCONNECTING; 8031da177e4SLinus Torvalds else 8041da177e4SLinus Torvalds sock->state = SS_CONNECTED; 8051da177e4SLinus Torvalds } 8061da177e4SLinus Torvalds 8071da177e4SLinus Torvalds switch (sk->sk_state) { 8081da177e4SLinus Torvalds case TCP_CLOSE: 8091da177e4SLinus Torvalds err = -ENOTCONN; 8101da177e4SLinus Torvalds /* Hack to wake up other listeners, who can poll for 8111da177e4SLinus Torvalds POLLHUP, even on eg. unconnected UDP sockets -- RR */ 8121da177e4SLinus Torvalds default: 8131da177e4SLinus Torvalds sk->sk_shutdown |= how; 8141da177e4SLinus Torvalds if (sk->sk_prot->shutdown) 8151da177e4SLinus Torvalds sk->sk_prot->shutdown(sk, how); 8161da177e4SLinus Torvalds break; 8171da177e4SLinus Torvalds 8181da177e4SLinus Torvalds /* Remaining two branches are temporary solution for missing 8191da177e4SLinus Torvalds * close() in multithreaded environment. It is _not_ a good idea, 8201da177e4SLinus Torvalds * but we have no choice until close() is repaired at VFS level. 8211da177e4SLinus Torvalds */ 8221da177e4SLinus Torvalds case TCP_LISTEN: 8231da177e4SLinus Torvalds if (!(how & RCV_SHUTDOWN)) 8241da177e4SLinus Torvalds break; 8251da177e4SLinus Torvalds /* Fall through */ 8261da177e4SLinus Torvalds case TCP_SYN_SENT: 8271da177e4SLinus Torvalds err = sk->sk_prot->disconnect(sk, O_NONBLOCK); 8281da177e4SLinus Torvalds sock->state = err ? SS_DISCONNECTING : SS_UNCONNECTED; 8291da177e4SLinus Torvalds break; 8301da177e4SLinus Torvalds } 8311da177e4SLinus Torvalds 8321da177e4SLinus Torvalds /* Wake up anyone sleeping in poll. */ 8331da177e4SLinus Torvalds sk->sk_state_change(sk); 8341da177e4SLinus Torvalds release_sock(sk); 8351da177e4SLinus Torvalds return err; 8361da177e4SLinus Torvalds } 8373d1427f8SEric Dumazet EXPORT_SYMBOL(inet_shutdown); 8381da177e4SLinus Torvalds 8391da177e4SLinus Torvalds /* 8401da177e4SLinus Torvalds * ioctl() calls you can issue on an INET socket. Most of these are 8411da177e4SLinus Torvalds * device configuration and stuff and very rarely used. Some ioctls 8421da177e4SLinus Torvalds * pass on to the socket itself. 8431da177e4SLinus Torvalds * 8441da177e4SLinus Torvalds * NOTE: I like the idea of a module for the config stuff. ie ifconfig 8451da177e4SLinus Torvalds * loads the devconfigure module does its configuring and unloads it. 8461da177e4SLinus Torvalds * There's a good 20K of config code hanging around the kernel. 8471da177e4SLinus Torvalds */ 8481da177e4SLinus Torvalds 8491da177e4SLinus Torvalds int inet_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg) 8501da177e4SLinus Torvalds { 8511da177e4SLinus Torvalds struct sock *sk = sock->sk; 8521da177e4SLinus Torvalds int err = 0; 8533b1e0a65SYOSHIFUJI Hideaki struct net *net = sock_net(sk); 8541da177e4SLinus Torvalds 8551da177e4SLinus Torvalds switch (cmd) { 8561da177e4SLinus Torvalds case SIOCGSTAMP: 8571da177e4SLinus Torvalds err = sock_get_timestamp(sk, (struct timeval __user *)arg); 8581da177e4SLinus Torvalds break; 859ae40eb1eSEric Dumazet case SIOCGSTAMPNS: 860ae40eb1eSEric Dumazet err = sock_get_timestampns(sk, (struct timespec __user *)arg); 861ae40eb1eSEric Dumazet break; 8621da177e4SLinus Torvalds case SIOCADDRT: 8631da177e4SLinus Torvalds case SIOCDELRT: 8641da177e4SLinus Torvalds case SIOCRTMSG: 865e5b13cb1SDenis V. Lunev err = ip_rt_ioctl(net, cmd, (void __user *)arg); 8661da177e4SLinus Torvalds break; 8671da177e4SLinus Torvalds case SIOCDARP: 8681da177e4SLinus Torvalds case SIOCGARP: 8691da177e4SLinus Torvalds case SIOCSARP: 870e5b13cb1SDenis V. Lunev err = arp_ioctl(net, cmd, (void __user *)arg); 8711da177e4SLinus Torvalds break; 8721da177e4SLinus Torvalds case SIOCGIFADDR: 8731da177e4SLinus Torvalds case SIOCSIFADDR: 8741da177e4SLinus Torvalds case SIOCGIFBRDADDR: 8751da177e4SLinus Torvalds case SIOCSIFBRDADDR: 8761da177e4SLinus Torvalds case SIOCGIFNETMASK: 8771da177e4SLinus Torvalds case SIOCSIFNETMASK: 8781da177e4SLinus Torvalds case SIOCGIFDSTADDR: 8791da177e4SLinus Torvalds case SIOCSIFDSTADDR: 8801da177e4SLinus Torvalds case SIOCSIFPFLAGS: 8811da177e4SLinus Torvalds case SIOCGIFPFLAGS: 8821da177e4SLinus Torvalds case SIOCSIFFLAGS: 883e5b13cb1SDenis V. Lunev err = devinet_ioctl(net, cmd, (void __user *)arg); 8841da177e4SLinus Torvalds break; 8851da177e4SLinus Torvalds default: 886b5e5fa5eSChristoph Hellwig if (sk->sk_prot->ioctl) 887b5e5fa5eSChristoph Hellwig err = sk->sk_prot->ioctl(sk, cmd, arg); 888b5e5fa5eSChristoph Hellwig else 889b5e5fa5eSChristoph Hellwig err = -ENOIOCTLCMD; 8901da177e4SLinus Torvalds break; 8911da177e4SLinus Torvalds } 8921da177e4SLinus Torvalds return err; 8931da177e4SLinus Torvalds } 8943d1427f8SEric Dumazet EXPORT_SYMBOL(inet_ioctl); 8951da177e4SLinus Torvalds 896709b46e8SEric W. Biederman #ifdef CONFIG_COMPAT 897686dc6b6SGerrit Renker static int inet_compat_ioctl(struct socket *sock, unsigned int cmd, unsigned long arg) 898709b46e8SEric W. Biederman { 899709b46e8SEric W. Biederman struct sock *sk = sock->sk; 900709b46e8SEric W. Biederman int err = -ENOIOCTLCMD; 901709b46e8SEric W. Biederman 902709b46e8SEric W. Biederman if (sk->sk_prot->compat_ioctl) 903709b46e8SEric W. Biederman err = sk->sk_prot->compat_ioctl(sk, cmd, arg); 904709b46e8SEric W. Biederman 905709b46e8SEric W. Biederman return err; 906709b46e8SEric W. Biederman } 907709b46e8SEric W. Biederman #endif 908709b46e8SEric W. Biederman 90990ddc4f0SEric Dumazet const struct proto_ops inet_stream_ops = { 9101da177e4SLinus Torvalds .family = PF_INET, 9111da177e4SLinus Torvalds .owner = THIS_MODULE, 9121da177e4SLinus Torvalds .release = inet_release, 9131da177e4SLinus Torvalds .bind = inet_bind, 9141da177e4SLinus Torvalds .connect = inet_stream_connect, 9151da177e4SLinus Torvalds .socketpair = sock_no_socketpair, 9161da177e4SLinus Torvalds .accept = inet_accept, 9171da177e4SLinus Torvalds .getname = inet_getname, 9181da177e4SLinus Torvalds .poll = tcp_poll, 9191da177e4SLinus Torvalds .ioctl = inet_ioctl, 9201da177e4SLinus Torvalds .listen = inet_listen, 9211da177e4SLinus Torvalds .shutdown = inet_shutdown, 9221da177e4SLinus Torvalds .setsockopt = sock_common_setsockopt, 9231da177e4SLinus Torvalds .getsockopt = sock_common_getsockopt, 9247ba42910SChangli Gao .sendmsg = inet_sendmsg, 925fec5e652STom Herbert .recvmsg = inet_recvmsg, 926543d9cfeSArnaldo Carvalho de Melo .mmap = sock_no_mmap, 9277ba42910SChangli Gao .sendpage = inet_sendpage, 9289c55e01cSJens Axboe .splice_read = tcp_splice_read, 9293fdadf7dSDmitry Mishin #ifdef CONFIG_COMPAT 9303fdadf7dSDmitry Mishin .compat_setsockopt = compat_sock_common_setsockopt, 9313fdadf7dSDmitry Mishin .compat_getsockopt = compat_sock_common_getsockopt, 932709b46e8SEric W. Biederman .compat_ioctl = inet_compat_ioctl, 9333fdadf7dSDmitry Mishin #endif 9341da177e4SLinus Torvalds }; 9353d1427f8SEric Dumazet EXPORT_SYMBOL(inet_stream_ops); 9361da177e4SLinus Torvalds 93790ddc4f0SEric Dumazet const struct proto_ops inet_dgram_ops = { 9381da177e4SLinus Torvalds .family = PF_INET, 9391da177e4SLinus Torvalds .owner = THIS_MODULE, 9401da177e4SLinus Torvalds .release = inet_release, 9411da177e4SLinus Torvalds .bind = inet_bind, 9421da177e4SLinus Torvalds .connect = inet_dgram_connect, 9431da177e4SLinus Torvalds .socketpair = sock_no_socketpair, 9441da177e4SLinus Torvalds .accept = sock_no_accept, 9451da177e4SLinus Torvalds .getname = inet_getname, 9461da177e4SLinus Torvalds .poll = udp_poll, 9471da177e4SLinus Torvalds .ioctl = inet_ioctl, 9481da177e4SLinus Torvalds .listen = sock_no_listen, 9491da177e4SLinus Torvalds .shutdown = inet_shutdown, 9501da177e4SLinus Torvalds .setsockopt = sock_common_setsockopt, 9511da177e4SLinus Torvalds .getsockopt = sock_common_getsockopt, 9521da177e4SLinus Torvalds .sendmsg = inet_sendmsg, 953fec5e652STom Herbert .recvmsg = inet_recvmsg, 9541da177e4SLinus Torvalds .mmap = sock_no_mmap, 9551da177e4SLinus Torvalds .sendpage = inet_sendpage, 956543d9cfeSArnaldo Carvalho de Melo #ifdef CONFIG_COMPAT 957543d9cfeSArnaldo Carvalho de Melo .compat_setsockopt = compat_sock_common_setsockopt, 958543d9cfeSArnaldo Carvalho de Melo .compat_getsockopt = compat_sock_common_getsockopt, 959709b46e8SEric W. Biederman .compat_ioctl = inet_compat_ioctl, 960543d9cfeSArnaldo Carvalho de Melo #endif 9611da177e4SLinus Torvalds }; 9623d1427f8SEric Dumazet EXPORT_SYMBOL(inet_dgram_ops); 9631da177e4SLinus Torvalds 9641da177e4SLinus Torvalds /* 9651da177e4SLinus Torvalds * For SOCK_RAW sockets; should be the same as inet_dgram_ops but without 9661da177e4SLinus Torvalds * udp_poll 9671da177e4SLinus Torvalds */ 96890ddc4f0SEric Dumazet static const struct proto_ops inet_sockraw_ops = { 9691da177e4SLinus Torvalds .family = PF_INET, 9701da177e4SLinus Torvalds .owner = THIS_MODULE, 9711da177e4SLinus Torvalds .release = inet_release, 9721da177e4SLinus Torvalds .bind = inet_bind, 9731da177e4SLinus Torvalds .connect = inet_dgram_connect, 9741da177e4SLinus Torvalds .socketpair = sock_no_socketpair, 9751da177e4SLinus Torvalds .accept = sock_no_accept, 9761da177e4SLinus Torvalds .getname = inet_getname, 9771da177e4SLinus Torvalds .poll = datagram_poll, 9781da177e4SLinus Torvalds .ioctl = inet_ioctl, 9791da177e4SLinus Torvalds .listen = sock_no_listen, 9801da177e4SLinus Torvalds .shutdown = inet_shutdown, 9811da177e4SLinus Torvalds .setsockopt = sock_common_setsockopt, 9821da177e4SLinus Torvalds .getsockopt = sock_common_getsockopt, 9831da177e4SLinus Torvalds .sendmsg = inet_sendmsg, 984fec5e652STom Herbert .recvmsg = inet_recvmsg, 9851da177e4SLinus Torvalds .mmap = sock_no_mmap, 9861da177e4SLinus Torvalds .sendpage = inet_sendpage, 987543d9cfeSArnaldo Carvalho de Melo #ifdef CONFIG_COMPAT 988543d9cfeSArnaldo Carvalho de Melo .compat_setsockopt = compat_sock_common_setsockopt, 989543d9cfeSArnaldo Carvalho de Melo .compat_getsockopt = compat_sock_common_getsockopt, 990709b46e8SEric W. Biederman .compat_ioctl = inet_compat_ioctl, 991543d9cfeSArnaldo Carvalho de Melo #endif 9921da177e4SLinus Torvalds }; 9931da177e4SLinus Torvalds 994ec1b4cf7SStephen Hemminger static const struct net_proto_family inet_family_ops = { 9951da177e4SLinus Torvalds .family = PF_INET, 9961da177e4SLinus Torvalds .create = inet_create, 9971da177e4SLinus Torvalds .owner = THIS_MODULE, 9981da177e4SLinus Torvalds }; 9991da177e4SLinus Torvalds 10001da177e4SLinus Torvalds /* Upon startup we insert all the elements in inetsw_array[] into 10011da177e4SLinus Torvalds * the linked list inetsw. 10021da177e4SLinus Torvalds */ 10031da177e4SLinus Torvalds static struct inet_protosw inetsw_array[] = 10041da177e4SLinus Torvalds { 10051da177e4SLinus Torvalds { 10061da177e4SLinus Torvalds .type = SOCK_STREAM, 10071da177e4SLinus Torvalds .protocol = IPPROTO_TCP, 10081da177e4SLinus Torvalds .prot = &tcp_prot, 10091da177e4SLinus Torvalds .ops = &inet_stream_ops, 10101da177e4SLinus Torvalds .no_check = 0, 1011d83d8461SArnaldo Carvalho de Melo .flags = INET_PROTOSW_PERMANENT | 1012d83d8461SArnaldo Carvalho de Melo INET_PROTOSW_ICSK, 10131da177e4SLinus Torvalds }, 10141da177e4SLinus Torvalds 10151da177e4SLinus Torvalds { 10161da177e4SLinus Torvalds .type = SOCK_DGRAM, 10171da177e4SLinus Torvalds .protocol = IPPROTO_UDP, 10181da177e4SLinus Torvalds .prot = &udp_prot, 10191da177e4SLinus Torvalds .ops = &inet_dgram_ops, 10201da177e4SLinus Torvalds .no_check = UDP_CSUM_DEFAULT, 10211da177e4SLinus Torvalds .flags = INET_PROTOSW_PERMANENT, 10221da177e4SLinus Torvalds }, 10231da177e4SLinus Torvalds 1024c319b4d7SVasiliy Kulikov { 1025c319b4d7SVasiliy Kulikov .type = SOCK_DGRAM, 1026c319b4d7SVasiliy Kulikov .protocol = IPPROTO_ICMP, 1027c319b4d7SVasiliy Kulikov .prot = &ping_prot, 1028c319b4d7SVasiliy Kulikov .ops = &inet_dgram_ops, 1029c319b4d7SVasiliy Kulikov .no_check = UDP_CSUM_DEFAULT, 1030c319b4d7SVasiliy Kulikov .flags = INET_PROTOSW_REUSE, 1031c319b4d7SVasiliy Kulikov }, 10321da177e4SLinus Torvalds 10331da177e4SLinus Torvalds { 10341da177e4SLinus Torvalds .type = SOCK_RAW, 10351da177e4SLinus Torvalds .protocol = IPPROTO_IP, /* wild card */ 10361da177e4SLinus Torvalds .prot = &raw_prot, 10371da177e4SLinus Torvalds .ops = &inet_sockraw_ops, 10381da177e4SLinus Torvalds .no_check = UDP_CSUM_DEFAULT, 10391da177e4SLinus Torvalds .flags = INET_PROTOSW_REUSE, 10401da177e4SLinus Torvalds } 10411da177e4SLinus Torvalds }; 10421da177e4SLinus Torvalds 1043c40f6fffSDenis Cheng #define INETSW_ARRAY_LEN ARRAY_SIZE(inetsw_array) 10441da177e4SLinus Torvalds 10451da177e4SLinus Torvalds void inet_register_protosw(struct inet_protosw *p) 10461da177e4SLinus Torvalds { 10471da177e4SLinus Torvalds struct list_head *lh; 10481da177e4SLinus Torvalds struct inet_protosw *answer; 10491da177e4SLinus Torvalds int protocol = p->protocol; 10501da177e4SLinus Torvalds struct list_head *last_perm; 10511da177e4SLinus Torvalds 10521da177e4SLinus Torvalds spin_lock_bh(&inetsw_lock); 10531da177e4SLinus Torvalds 10541da177e4SLinus Torvalds if (p->type >= SOCK_MAX) 10551da177e4SLinus Torvalds goto out_illegal; 10561da177e4SLinus Torvalds 10571da177e4SLinus Torvalds /* If we are trying to override a permanent protocol, bail. */ 10581da177e4SLinus Torvalds answer = NULL; 10591da177e4SLinus Torvalds last_perm = &inetsw[p->type]; 10601da177e4SLinus Torvalds list_for_each(lh, &inetsw[p->type]) { 10611da177e4SLinus Torvalds answer = list_entry(lh, struct inet_protosw, list); 10621da177e4SLinus Torvalds 10631da177e4SLinus Torvalds /* Check only the non-wild match. */ 10641da177e4SLinus Torvalds if (INET_PROTOSW_PERMANENT & answer->flags) { 10651da177e4SLinus Torvalds if (protocol == answer->protocol) 10661da177e4SLinus Torvalds break; 10671da177e4SLinus Torvalds last_perm = lh; 10681da177e4SLinus Torvalds } 10691da177e4SLinus Torvalds 10701da177e4SLinus Torvalds answer = NULL; 10711da177e4SLinus Torvalds } 10721da177e4SLinus Torvalds if (answer) 10731da177e4SLinus Torvalds goto out_permanent; 10741da177e4SLinus Torvalds 10751da177e4SLinus Torvalds /* Add the new entry after the last permanent entry if any, so that 10761da177e4SLinus Torvalds * the new entry does not override a permanent entry when matched with 10771da177e4SLinus Torvalds * a wild-card protocol. But it is allowed to override any existing 10781da177e4SLinus Torvalds * non-permanent entry. This means that when we remove this entry, the 10791da177e4SLinus Torvalds * system automatically returns to the old behavior. 10801da177e4SLinus Torvalds */ 10811da177e4SLinus Torvalds list_add_rcu(&p->list, last_perm); 10821da177e4SLinus Torvalds out: 10831da177e4SLinus Torvalds spin_unlock_bh(&inetsw_lock); 10841da177e4SLinus Torvalds 10851da177e4SLinus Torvalds return; 10861da177e4SLinus Torvalds 10871da177e4SLinus Torvalds out_permanent: 1088058bd4d2SJoe Perches pr_err("Attempt to override permanent protocol %d\n", protocol); 10891da177e4SLinus Torvalds goto out; 10901da177e4SLinus Torvalds 10911da177e4SLinus Torvalds out_illegal: 1092058bd4d2SJoe Perches pr_err("Ignoring attempt to register invalid socket type %d\n", 10931da177e4SLinus Torvalds p->type); 10941da177e4SLinus Torvalds goto out; 10951da177e4SLinus Torvalds } 10963d1427f8SEric Dumazet EXPORT_SYMBOL(inet_register_protosw); 10971da177e4SLinus Torvalds 10981da177e4SLinus Torvalds void inet_unregister_protosw(struct inet_protosw *p) 10991da177e4SLinus Torvalds { 11001da177e4SLinus Torvalds if (INET_PROTOSW_PERMANENT & p->flags) { 1101058bd4d2SJoe Perches pr_err("Attempt to unregister permanent protocol %d\n", 11021da177e4SLinus Torvalds p->protocol); 11031da177e4SLinus Torvalds } else { 11041da177e4SLinus Torvalds spin_lock_bh(&inetsw_lock); 11051da177e4SLinus Torvalds list_del_rcu(&p->list); 11061da177e4SLinus Torvalds spin_unlock_bh(&inetsw_lock); 11071da177e4SLinus Torvalds 11081da177e4SLinus Torvalds synchronize_net(); 11091da177e4SLinus Torvalds } 11101da177e4SLinus Torvalds } 11113d1427f8SEric Dumazet EXPORT_SYMBOL(inet_unregister_protosw); 11121da177e4SLinus Torvalds 111332519f11SArnaldo Carvalho de Melo /* 111432519f11SArnaldo Carvalho de Melo * Shall we try to damage output packets if routing dev changes? 111532519f11SArnaldo Carvalho de Melo */ 111632519f11SArnaldo Carvalho de Melo 1117ab32ea5dSBrian Haley int sysctl_ip_dynaddr __read_mostly; 111832519f11SArnaldo Carvalho de Melo 111932519f11SArnaldo Carvalho de Melo static int inet_sk_reselect_saddr(struct sock *sk) 112032519f11SArnaldo Carvalho de Melo { 112132519f11SArnaldo Carvalho de Melo struct inet_sock *inet = inet_sk(sk); 1122c720c7e8SEric Dumazet __be32 old_saddr = inet->inet_saddr; 1123c720c7e8SEric Dumazet __be32 daddr = inet->inet_daddr; 11246e869138SDavid S. Miller struct flowi4 *fl4; 1125b23dd4feSDavid S. Miller struct rtable *rt; 1126b23dd4feSDavid S. Miller __be32 new_saddr; 1127f6d8bd05SEric Dumazet struct ip_options_rcu *inet_opt; 112832519f11SArnaldo Carvalho de Melo 1129f6d8bd05SEric Dumazet inet_opt = rcu_dereference_protected(inet->inet_opt, 1130f6d8bd05SEric Dumazet sock_owned_by_user(sk)); 1131f6d8bd05SEric Dumazet if (inet_opt && inet_opt->opt.srr) 1132f6d8bd05SEric Dumazet daddr = inet_opt->opt.faddr; 113332519f11SArnaldo Carvalho de Melo 113432519f11SArnaldo Carvalho de Melo /* Query new route. */ 11356e869138SDavid S. Miller fl4 = &inet->cork.fl.u.ip4; 11366e869138SDavid S. Miller rt = ip_route_connect(fl4, daddr, 0, RT_CONN_FLAGS(sk), 1137b23dd4feSDavid S. Miller sk->sk_bound_dev_if, sk->sk_protocol, 1138abdf7e72SDavid S. Miller inet->inet_sport, inet->inet_dport, sk, false); 1139b23dd4feSDavid S. Miller if (IS_ERR(rt)) 1140b23dd4feSDavid S. Miller return PTR_ERR(rt); 114132519f11SArnaldo Carvalho de Melo 1142d8d1f30bSChangli Gao sk_setup_caps(sk, &rt->dst); 114332519f11SArnaldo Carvalho de Melo 11446e869138SDavid S. Miller new_saddr = fl4->saddr; 114532519f11SArnaldo Carvalho de Melo 114632519f11SArnaldo Carvalho de Melo if (new_saddr == old_saddr) 114732519f11SArnaldo Carvalho de Melo return 0; 114832519f11SArnaldo Carvalho de Melo 114932519f11SArnaldo Carvalho de Melo if (sysctl_ip_dynaddr > 1) { 1150058bd4d2SJoe Perches pr_info("%s(): shifting inet->saddr from %pI4 to %pI4\n", 1151673d57e7SHarvey Harrison __func__, &old_saddr, &new_saddr); 115232519f11SArnaldo Carvalho de Melo } 115332519f11SArnaldo Carvalho de Melo 1154c720c7e8SEric Dumazet inet->inet_saddr = inet->inet_rcv_saddr = new_saddr; 115532519f11SArnaldo Carvalho de Melo 115632519f11SArnaldo Carvalho de Melo /* 115732519f11SArnaldo Carvalho de Melo * XXX The only one ugly spot where we need to 115832519f11SArnaldo Carvalho de Melo * XXX really change the sockets identity after 115932519f11SArnaldo Carvalho de Melo * XXX it has entered the hashes. -DaveM 116032519f11SArnaldo Carvalho de Melo * 116132519f11SArnaldo Carvalho de Melo * Besides that, it does not check for connection 116232519f11SArnaldo Carvalho de Melo * uniqueness. Wait for troubles. 116332519f11SArnaldo Carvalho de Melo */ 116432519f11SArnaldo Carvalho de Melo __sk_prot_rehash(sk); 116532519f11SArnaldo Carvalho de Melo return 0; 116632519f11SArnaldo Carvalho de Melo } 116732519f11SArnaldo Carvalho de Melo 116832519f11SArnaldo Carvalho de Melo int inet_sk_rebuild_header(struct sock *sk) 116932519f11SArnaldo Carvalho de Melo { 117032519f11SArnaldo Carvalho de Melo struct inet_sock *inet = inet_sk(sk); 117132519f11SArnaldo Carvalho de Melo struct rtable *rt = (struct rtable *)__sk_dst_check(sk, 0); 11723ca3c68eSAl Viro __be32 daddr; 1173f6d8bd05SEric Dumazet struct ip_options_rcu *inet_opt; 11746e869138SDavid S. Miller struct flowi4 *fl4; 117532519f11SArnaldo Carvalho de Melo int err; 117632519f11SArnaldo Carvalho de Melo 117732519f11SArnaldo Carvalho de Melo /* Route is OK, nothing to do. */ 117832519f11SArnaldo Carvalho de Melo if (rt) 117932519f11SArnaldo Carvalho de Melo return 0; 118032519f11SArnaldo Carvalho de Melo 118132519f11SArnaldo Carvalho de Melo /* Reroute. */ 1182f6d8bd05SEric Dumazet rcu_read_lock(); 1183f6d8bd05SEric Dumazet inet_opt = rcu_dereference(inet->inet_opt); 1184c720c7e8SEric Dumazet daddr = inet->inet_daddr; 1185f6d8bd05SEric Dumazet if (inet_opt && inet_opt->opt.srr) 1186f6d8bd05SEric Dumazet daddr = inet_opt->opt.faddr; 1187f6d8bd05SEric Dumazet rcu_read_unlock(); 11886e869138SDavid S. Miller fl4 = &inet->cork.fl.u.ip4; 11896e869138SDavid S. Miller rt = ip_route_output_ports(sock_net(sk), fl4, sk, daddr, inet->inet_saddr, 119078fbfd8aSDavid S. Miller inet->inet_dport, inet->inet_sport, 119178fbfd8aSDavid S. Miller sk->sk_protocol, RT_CONN_FLAGS(sk), 119278fbfd8aSDavid S. Miller sk->sk_bound_dev_if); 1193b23dd4feSDavid S. Miller if (!IS_ERR(rt)) { 1194b23dd4feSDavid S. Miller err = 0; 1195d8d1f30bSChangli Gao sk_setup_caps(sk, &rt->dst); 1196b23dd4feSDavid S. Miller } else { 1197b23dd4feSDavid S. Miller err = PTR_ERR(rt); 1198b23dd4feSDavid S. Miller 119932519f11SArnaldo Carvalho de Melo /* Routing failed... */ 120032519f11SArnaldo Carvalho de Melo sk->sk_route_caps = 0; 120132519f11SArnaldo Carvalho de Melo /* 120232519f11SArnaldo Carvalho de Melo * Other protocols have to map its equivalent state to TCP_SYN_SENT. 120332519f11SArnaldo Carvalho de Melo * DCCP maps its DCCP_REQUESTING state to TCP_SYN_SENT. -acme 120432519f11SArnaldo Carvalho de Melo */ 120532519f11SArnaldo Carvalho de Melo if (!sysctl_ip_dynaddr || 120632519f11SArnaldo Carvalho de Melo sk->sk_state != TCP_SYN_SENT || 120732519f11SArnaldo Carvalho de Melo (sk->sk_userlocks & SOCK_BINDADDR_LOCK) || 120832519f11SArnaldo Carvalho de Melo (err = inet_sk_reselect_saddr(sk)) != 0) 120932519f11SArnaldo Carvalho de Melo sk->sk_err_soft = -err; 121032519f11SArnaldo Carvalho de Melo } 121132519f11SArnaldo Carvalho de Melo 121232519f11SArnaldo Carvalho de Melo return err; 121332519f11SArnaldo Carvalho de Melo } 121432519f11SArnaldo Carvalho de Melo EXPORT_SYMBOL(inet_sk_rebuild_header); 121532519f11SArnaldo Carvalho de Melo 1216a430a43dSHerbert Xu static int inet_gso_send_check(struct sk_buff *skb) 1217a430a43dSHerbert Xu { 1218b71d1d42SEric Dumazet const struct iphdr *iph; 121932613090SAlexey Dobriyan const struct net_protocol *ops; 1220a430a43dSHerbert Xu int proto; 1221a430a43dSHerbert Xu int ihl; 1222a430a43dSHerbert Xu int err = -EINVAL; 1223a430a43dSHerbert Xu 1224a430a43dSHerbert Xu if (unlikely(!pskb_may_pull(skb, sizeof(*iph)))) 1225a430a43dSHerbert Xu goto out; 1226a430a43dSHerbert Xu 1227eddc9ec5SArnaldo Carvalho de Melo iph = ip_hdr(skb); 1228a430a43dSHerbert Xu ihl = iph->ihl * 4; 1229a430a43dSHerbert Xu if (ihl < sizeof(*iph)) 1230a430a43dSHerbert Xu goto out; 1231a430a43dSHerbert Xu 1232a430a43dSHerbert Xu if (unlikely(!pskb_may_pull(skb, ihl))) 1233a430a43dSHerbert Xu goto out; 1234a430a43dSHerbert Xu 1235badff6d0SArnaldo Carvalho de Melo __skb_pull(skb, ihl); 1236badff6d0SArnaldo Carvalho de Melo skb_reset_transport_header(skb); 1237eddc9ec5SArnaldo Carvalho de Melo iph = ip_hdr(skb); 1238a430a43dSHerbert Xu proto = iph->protocol & (MAX_INET_PROTOS - 1); 1239a430a43dSHerbert Xu err = -EPROTONOSUPPORT; 1240a430a43dSHerbert Xu 1241a430a43dSHerbert Xu rcu_read_lock(); 1242a430a43dSHerbert Xu ops = rcu_dereference(inet_protos[proto]); 1243a430a43dSHerbert Xu if (likely(ops && ops->gso_send_check)) 1244a430a43dSHerbert Xu err = ops->gso_send_check(skb); 1245a430a43dSHerbert Xu rcu_read_unlock(); 1246a430a43dSHerbert Xu 1247a430a43dSHerbert Xu out: 1248a430a43dSHerbert Xu return err; 1249a430a43dSHerbert Xu } 1250a430a43dSHerbert Xu 1251c8f44affSMichał Mirosław static struct sk_buff *inet_gso_segment(struct sk_buff *skb, 1252c8f44affSMichał Mirosław netdev_features_t features) 1253f4c50d99SHerbert Xu { 1254f4c50d99SHerbert Xu struct sk_buff *segs = ERR_PTR(-EINVAL); 1255f4c50d99SHerbert Xu struct iphdr *iph; 125632613090SAlexey Dobriyan const struct net_protocol *ops; 1257f4c50d99SHerbert Xu int proto; 1258f4c50d99SHerbert Xu int ihl; 1259f4c50d99SHerbert Xu int id; 1260d7ca4cc0SSridhar Samudrala unsigned int offset = 0; 1261f4c50d99SHerbert Xu 1262d212f87bSStephen Hemminger if (!(features & NETIF_F_V4_CSUM)) 1263d212f87bSStephen Hemminger features &= ~NETIF_F_SG; 1264d212f87bSStephen Hemminger 1265bbcf467dSHerbert Xu if (unlikely(skb_shinfo(skb)->gso_type & 1266bbcf467dSHerbert Xu ~(SKB_GSO_TCPV4 | 1267bbcf467dSHerbert Xu SKB_GSO_UDP | 1268bbcf467dSHerbert Xu SKB_GSO_DODGY | 1269bbcf467dSHerbert Xu SKB_GSO_TCP_ECN | 1270bbcf467dSHerbert Xu 0))) 1271bbcf467dSHerbert Xu goto out; 1272bbcf467dSHerbert Xu 1273bbcf467dSHerbert Xu if (unlikely(!pskb_may_pull(skb, sizeof(*iph)))) 1274f4c50d99SHerbert Xu goto out; 1275f4c50d99SHerbert Xu 1276eddc9ec5SArnaldo Carvalho de Melo iph = ip_hdr(skb); 1277f4c50d99SHerbert Xu ihl = iph->ihl * 4; 1278f4c50d99SHerbert Xu if (ihl < sizeof(*iph)) 1279f4c50d99SHerbert Xu goto out; 1280f4c50d99SHerbert Xu 1281bbcf467dSHerbert Xu if (unlikely(!pskb_may_pull(skb, ihl))) 1282f4c50d99SHerbert Xu goto out; 1283f4c50d99SHerbert Xu 1284badff6d0SArnaldo Carvalho de Melo __skb_pull(skb, ihl); 1285badff6d0SArnaldo Carvalho de Melo skb_reset_transport_header(skb); 1286eddc9ec5SArnaldo Carvalho de Melo iph = ip_hdr(skb); 1287f4c50d99SHerbert Xu id = ntohs(iph->id); 1288f4c50d99SHerbert Xu proto = iph->protocol & (MAX_INET_PROTOS - 1); 1289f4c50d99SHerbert Xu segs = ERR_PTR(-EPROTONOSUPPORT); 1290f4c50d99SHerbert Xu 1291f4c50d99SHerbert Xu rcu_read_lock(); 1292f4c50d99SHerbert Xu ops = rcu_dereference(inet_protos[proto]); 1293bbcf467dSHerbert Xu if (likely(ops && ops->gso_segment)) 1294576a30ebSHerbert Xu segs = ops->gso_segment(skb, features); 1295f4c50d99SHerbert Xu rcu_read_unlock(); 1296f4c50d99SHerbert Xu 1297801678c5SHirofumi Nakagawa if (!segs || IS_ERR(segs)) 1298f4c50d99SHerbert Xu goto out; 1299f4c50d99SHerbert Xu 1300f4c50d99SHerbert Xu skb = segs; 1301f4c50d99SHerbert Xu do { 1302eddc9ec5SArnaldo Carvalho de Melo iph = ip_hdr(skb); 1303d7ca4cc0SSridhar Samudrala if (proto == IPPROTO_UDP) { 1304d7ca4cc0SSridhar Samudrala iph->id = htons(id); 1305d7ca4cc0SSridhar Samudrala iph->frag_off = htons(offset >> 3); 1306d7ca4cc0SSridhar Samudrala if (skb->next != NULL) 1307d7ca4cc0SSridhar Samudrala iph->frag_off |= htons(IP_MF); 1308d7ca4cc0SSridhar Samudrala offset += (skb->len - skb->mac_len - iph->ihl * 4); 1309d7ca4cc0SSridhar Samudrala } else 1310f4c50d99SHerbert Xu iph->id = htons(id++); 1311f4c50d99SHerbert Xu iph->tot_len = htons(skb->len - skb->mac_len); 1312f4c50d99SHerbert Xu iph->check = 0; 1313d56f90a7SArnaldo Carvalho de Melo iph->check = ip_fast_csum(skb_network_header(skb), iph->ihl); 1314f4c50d99SHerbert Xu } while ((skb = skb->next)); 1315f4c50d99SHerbert Xu 1316f4c50d99SHerbert Xu out: 1317f4c50d99SHerbert Xu return segs; 1318f4c50d99SHerbert Xu } 1319f4c50d99SHerbert Xu 132073cc19f1SHerbert Xu static struct sk_buff **inet_gro_receive(struct sk_buff **head, 132173cc19f1SHerbert Xu struct sk_buff *skb) 132273cc19f1SHerbert Xu { 132332613090SAlexey Dobriyan const struct net_protocol *ops; 132473cc19f1SHerbert Xu struct sk_buff **pp = NULL; 132573cc19f1SHerbert Xu struct sk_buff *p; 1326b71d1d42SEric Dumazet const struct iphdr *iph; 1327a5b1cf28SHerbert Xu unsigned int hlen; 1328a5b1cf28SHerbert Xu unsigned int off; 13291075f3f6SHerbert Xu unsigned int id; 133073cc19f1SHerbert Xu int flush = 1; 133173cc19f1SHerbert Xu int proto; 133273cc19f1SHerbert Xu 1333a5b1cf28SHerbert Xu off = skb_gro_offset(skb); 1334a5b1cf28SHerbert Xu hlen = off + sizeof(*iph); 1335a5b1cf28SHerbert Xu iph = skb_gro_header_fast(skb, off); 1336a5b1cf28SHerbert Xu if (skb_gro_header_hard(skb, hlen)) { 1337a5b1cf28SHerbert Xu iph = skb_gro_header_slow(skb, hlen, off); 133886911732SHerbert Xu if (unlikely(!iph)) 133973cc19f1SHerbert Xu goto out; 1340a5b1cf28SHerbert Xu } 134173cc19f1SHerbert Xu 134273cc19f1SHerbert Xu proto = iph->protocol & (MAX_INET_PROTOS - 1); 134373cc19f1SHerbert Xu 134473cc19f1SHerbert Xu rcu_read_lock(); 134573cc19f1SHerbert Xu ops = rcu_dereference(inet_protos[proto]); 134673cc19f1SHerbert Xu if (!ops || !ops->gro_receive) 134773cc19f1SHerbert Xu goto out_unlock; 134873cc19f1SHerbert Xu 1349a5ad24beSHerbert Xu if (*(u8 *)iph != 0x45) 135073cc19f1SHerbert Xu goto out_unlock; 135173cc19f1SHerbert Xu 135273cc19f1SHerbert Xu if (unlikely(ip_fast_csum((u8 *)iph, iph->ihl))) 135373cc19f1SHerbert Xu goto out_unlock; 135473cc19f1SHerbert Xu 13550eae88f3SEric Dumazet id = ntohl(*(__be32 *)&iph->id); 13560eae88f3SEric Dumazet flush = (u16)((ntohl(*(__be32 *)iph) ^ skb_gro_len(skb)) | (id ^ IP_DF)); 13571075f3f6SHerbert Xu id >>= 16; 135873cc19f1SHerbert Xu 135973cc19f1SHerbert Xu for (p = *head; p; p = p->next) { 136073cc19f1SHerbert Xu struct iphdr *iph2; 136173cc19f1SHerbert Xu 136273cc19f1SHerbert Xu if (!NAPI_GRO_CB(p)->same_flow) 136373cc19f1SHerbert Xu continue; 136473cc19f1SHerbert Xu 136573cc19f1SHerbert Xu iph2 = ip_hdr(p); 136673cc19f1SHerbert Xu 1367a5ad24beSHerbert Xu if ((iph->protocol ^ iph2->protocol) | 1368a5ad24beSHerbert Xu (iph->tos ^ iph2->tos) | 13690eae88f3SEric Dumazet ((__force u32)iph->saddr ^ (__force u32)iph2->saddr) | 13700eae88f3SEric Dumazet ((__force u32)iph->daddr ^ (__force u32)iph2->daddr)) { 137173cc19f1SHerbert Xu NAPI_GRO_CB(p)->same_flow = 0; 137273cc19f1SHerbert Xu continue; 137373cc19f1SHerbert Xu } 137473cc19f1SHerbert Xu 137573cc19f1SHerbert Xu /* All fields must match except length and checksum. */ 137673cc19f1SHerbert Xu NAPI_GRO_CB(p)->flush |= 1377a5ad24beSHerbert Xu (iph->ttl ^ iph2->ttl) | 1378a5ad24beSHerbert Xu ((u16)(ntohs(iph2->id) + NAPI_GRO_CB(p)->count) ^ id); 137973cc19f1SHerbert Xu 138073cc19f1SHerbert Xu NAPI_GRO_CB(p)->flush |= flush; 138173cc19f1SHerbert Xu } 138273cc19f1SHerbert Xu 138373cc19f1SHerbert Xu NAPI_GRO_CB(skb)->flush |= flush; 138486911732SHerbert Xu skb_gro_pull(skb, sizeof(*iph)); 138586911732SHerbert Xu skb_set_transport_header(skb, skb_gro_offset(skb)); 138673cc19f1SHerbert Xu 138773cc19f1SHerbert Xu pp = ops->gro_receive(head, skb); 138873cc19f1SHerbert Xu 138973cc19f1SHerbert Xu out_unlock: 139073cc19f1SHerbert Xu rcu_read_unlock(); 139173cc19f1SHerbert Xu 139273cc19f1SHerbert Xu out: 139373cc19f1SHerbert Xu NAPI_GRO_CB(skb)->flush |= flush; 139473cc19f1SHerbert Xu 139573cc19f1SHerbert Xu return pp; 139673cc19f1SHerbert Xu } 139773cc19f1SHerbert Xu 139873cc19f1SHerbert Xu static int inet_gro_complete(struct sk_buff *skb) 139973cc19f1SHerbert Xu { 140032613090SAlexey Dobriyan const struct net_protocol *ops; 140173cc19f1SHerbert Xu struct iphdr *iph = ip_hdr(skb); 140273cc19f1SHerbert Xu int proto = iph->protocol & (MAX_INET_PROTOS - 1); 140373cc19f1SHerbert Xu int err = -ENOSYS; 140473cc19f1SHerbert Xu __be16 newlen = htons(skb->len - skb_network_offset(skb)); 140573cc19f1SHerbert Xu 140673cc19f1SHerbert Xu csum_replace2(&iph->check, iph->tot_len, newlen); 140773cc19f1SHerbert Xu iph->tot_len = newlen; 140873cc19f1SHerbert Xu 140973cc19f1SHerbert Xu rcu_read_lock(); 141073cc19f1SHerbert Xu ops = rcu_dereference(inet_protos[proto]); 141173cc19f1SHerbert Xu if (WARN_ON(!ops || !ops->gro_complete)) 141273cc19f1SHerbert Xu goto out_unlock; 141373cc19f1SHerbert Xu 141473cc19f1SHerbert Xu err = ops->gro_complete(skb); 141573cc19f1SHerbert Xu 141673cc19f1SHerbert Xu out_unlock: 141773cc19f1SHerbert Xu rcu_read_unlock(); 141873cc19f1SHerbert Xu 141973cc19f1SHerbert Xu return err; 142073cc19f1SHerbert Xu } 142173cc19f1SHerbert Xu 1422eee4fe4dSDenis V. Lunev int inet_ctl_sock_create(struct sock **sk, unsigned short family, 14235677242fSDenis V. Lunev unsigned short type, unsigned char protocol, 14245677242fSDenis V. Lunev struct net *net) 14253d58b5faSDenis V. Lunev { 1426eee4fe4dSDenis V. Lunev struct socket *sock; 1427eee4fe4dSDenis V. Lunev int rc = sock_create_kern(family, type, protocol, &sock); 14283d58b5faSDenis V. Lunev 14293d58b5faSDenis V. Lunev if (rc == 0) { 1430eee4fe4dSDenis V. Lunev *sk = sock->sk; 1431eee4fe4dSDenis V. Lunev (*sk)->sk_allocation = GFP_ATOMIC; 14323d58b5faSDenis V. Lunev /* 14333d58b5faSDenis V. Lunev * Unhash it so that IP input processing does not even see it, 14343d58b5faSDenis V. Lunev * we do not wish this socket to see incoming packets. 14353d58b5faSDenis V. Lunev */ 1436eee4fe4dSDenis V. Lunev (*sk)->sk_prot->unhash(*sk); 14375677242fSDenis V. Lunev 14385677242fSDenis V. Lunev sk_change_net(*sk, net); 14393d58b5faSDenis V. Lunev } 14403d58b5faSDenis V. Lunev return rc; 14413d58b5faSDenis V. Lunev } 14423d58b5faSDenis V. Lunev EXPORT_SYMBOL_GPL(inet_ctl_sock_create); 14433d58b5faSDenis V. Lunev 14447d720c3eSTejun Heo unsigned long snmp_fold_field(void __percpu *mib[], int offt) 14455e0f0435SHerbert Xu { 14465e0f0435SHerbert Xu unsigned long res = 0; 14478f0ea0feSEric Dumazet int i, j; 14485e0f0435SHerbert Xu 14495e0f0435SHerbert Xu for_each_possible_cpu(i) { 14508f0ea0feSEric Dumazet for (j = 0; j < SNMP_ARRAY_SZ; j++) 14518f0ea0feSEric Dumazet res += *(((unsigned long *) per_cpu_ptr(mib[j], i)) + offt); 14525e0f0435SHerbert Xu } 14535e0f0435SHerbert Xu return res; 14545e0f0435SHerbert Xu } 14555e0f0435SHerbert Xu EXPORT_SYMBOL_GPL(snmp_fold_field); 14565e0f0435SHerbert Xu 14574ce3c183SEric Dumazet #if BITS_PER_LONG==32 14584ce3c183SEric Dumazet 14594ce3c183SEric Dumazet u64 snmp_fold_field64(void __percpu *mib[], int offt, size_t syncp_offset) 14604ce3c183SEric Dumazet { 14614ce3c183SEric Dumazet u64 res = 0; 14624ce3c183SEric Dumazet int cpu; 14634ce3c183SEric Dumazet 14644ce3c183SEric Dumazet for_each_possible_cpu(cpu) { 14658f0ea0feSEric Dumazet void *bhptr; 14664ce3c183SEric Dumazet struct u64_stats_sync *syncp; 14678f0ea0feSEric Dumazet u64 v; 14684ce3c183SEric Dumazet unsigned int start; 14694ce3c183SEric Dumazet 14708f0ea0feSEric Dumazet bhptr = per_cpu_ptr(mib[0], cpu); 14714ce3c183SEric Dumazet syncp = (struct u64_stats_sync *)(bhptr + syncp_offset); 14724ce3c183SEric Dumazet do { 14734ce3c183SEric Dumazet start = u64_stats_fetch_begin_bh(syncp); 14748f0ea0feSEric Dumazet v = *(((u64 *) bhptr) + offt); 14754ce3c183SEric Dumazet } while (u64_stats_fetch_retry_bh(syncp, start)); 14764ce3c183SEric Dumazet 14778f0ea0feSEric Dumazet res += v; 14784ce3c183SEric Dumazet } 14794ce3c183SEric Dumazet return res; 14804ce3c183SEric Dumazet } 14814ce3c183SEric Dumazet EXPORT_SYMBOL_GPL(snmp_fold_field64); 14824ce3c183SEric Dumazet #endif 14834ce3c183SEric Dumazet 14841823e4c8SEric Dumazet int snmp_mib_init(void __percpu *ptr[2], size_t mibsize, size_t align) 14855e0f0435SHerbert Xu { 14865e0f0435SHerbert Xu BUG_ON(ptr == NULL); 14871823e4c8SEric Dumazet ptr[0] = __alloc_percpu(mibsize, align); 14885e0f0435SHerbert Xu if (!ptr[0]) 14898f0ea0feSEric Dumazet return -ENOMEM; 14908f0ea0feSEric Dumazet #if SNMP_ARRAY_SZ == 2 14911823e4c8SEric Dumazet ptr[1] = __alloc_percpu(mibsize, align); 14928f0ea0feSEric Dumazet if (!ptr[1]) { 14935e0f0435SHerbert Xu free_percpu(ptr[0]); 14945e0f0435SHerbert Xu ptr[0] = NULL; 14955e0f0435SHerbert Xu return -ENOMEM; 14965e0f0435SHerbert Xu } 14978f0ea0feSEric Dumazet #endif 14988f0ea0feSEric Dumazet return 0; 14998f0ea0feSEric Dumazet } 15005e0f0435SHerbert Xu EXPORT_SYMBOL_GPL(snmp_mib_init); 15015e0f0435SHerbert Xu 15028f0ea0feSEric Dumazet void snmp_mib_free(void __percpu *ptr[SNMP_ARRAY_SZ]) 15035e0f0435SHerbert Xu { 15048f0ea0feSEric Dumazet int i; 15058f0ea0feSEric Dumazet 15065e0f0435SHerbert Xu BUG_ON(ptr == NULL); 15078f0ea0feSEric Dumazet for (i = 0; i < SNMP_ARRAY_SZ; i++) { 15088f0ea0feSEric Dumazet free_percpu(ptr[i]); 15098f0ea0feSEric Dumazet ptr[i] = NULL; 15108f0ea0feSEric Dumazet } 15115e0f0435SHerbert Xu } 15125e0f0435SHerbert Xu EXPORT_SYMBOL_GPL(snmp_mib_free); 15135e0f0435SHerbert Xu 15141da177e4SLinus Torvalds #ifdef CONFIG_IP_MULTICAST 151532613090SAlexey Dobriyan static const struct net_protocol igmp_protocol = { 15161da177e4SLinus Torvalds .handler = igmp_rcv, 1517b4ee07dfSAlexey Dobriyan .netns_ok = 1, 15181da177e4SLinus Torvalds }; 15191da177e4SLinus Torvalds #endif 15201da177e4SLinus Torvalds 152132613090SAlexey Dobriyan static const struct net_protocol tcp_protocol = { 15221da177e4SLinus Torvalds .handler = tcp_v4_rcv, 15231da177e4SLinus Torvalds .err_handler = tcp_v4_err, 1524a430a43dSHerbert Xu .gso_send_check = tcp_v4_gso_send_check, 1525f4c50d99SHerbert Xu .gso_segment = tcp_tso_segment, 1526bf296b12SHerbert Xu .gro_receive = tcp4_gro_receive, 1527bf296b12SHerbert Xu .gro_complete = tcp4_gro_complete, 15281da177e4SLinus Torvalds .no_policy = 1, 152992f1fecbSDenis V. Lunev .netns_ok = 1, 15301da177e4SLinus Torvalds }; 15311da177e4SLinus Torvalds 153232613090SAlexey Dobriyan static const struct net_protocol udp_protocol = { 15331da177e4SLinus Torvalds .handler = udp_rcv, 15341da177e4SLinus Torvalds .err_handler = udp_err, 1535d7ca4cc0SSridhar Samudrala .gso_send_check = udp4_ufo_send_check, 1536d7ca4cc0SSridhar Samudrala .gso_segment = udp4_ufo_fragment, 15371da177e4SLinus Torvalds .no_policy = 1, 153892f1fecbSDenis V. Lunev .netns_ok = 1, 15391da177e4SLinus Torvalds }; 15401da177e4SLinus Torvalds 154132613090SAlexey Dobriyan static const struct net_protocol icmp_protocol = { 15421da177e4SLinus Torvalds .handler = icmp_rcv, 1543c319b4d7SVasiliy Kulikov .err_handler = ping_err, 15448b7817f3SHerbert Xu .no_policy = 1, 154592f1fecbSDenis V. Lunev .netns_ok = 1, 15461da177e4SLinus Torvalds }; 15471da177e4SLinus Torvalds 15489b4661bdSPavel Emelyanov static __net_init int ipv4_mib_init_net(struct net *net) 15499b4661bdSPavel Emelyanov { 15507d720c3eSTejun Heo if (snmp_mib_init((void __percpu **)net->mib.tcp_statistics, 15511823e4c8SEric Dumazet sizeof(struct tcp_mib), 15521823e4c8SEric Dumazet __alignof__(struct tcp_mib)) < 0) 155357ef42d5SPavel Emelyanov goto err_tcp_mib; 15547d720c3eSTejun Heo if (snmp_mib_init((void __percpu **)net->mib.ip_statistics, 15551823e4c8SEric Dumazet sizeof(struct ipstats_mib), 15561823e4c8SEric Dumazet __alignof__(struct ipstats_mib)) < 0) 1557a20f5799SPavel Emelyanov goto err_ip_mib; 15587d720c3eSTejun Heo if (snmp_mib_init((void __percpu **)net->mib.net_statistics, 15591823e4c8SEric Dumazet sizeof(struct linux_mib), 15601823e4c8SEric Dumazet __alignof__(struct linux_mib)) < 0) 156161a7e260SPavel Emelyanov goto err_net_mib; 15627d720c3eSTejun Heo if (snmp_mib_init((void __percpu **)net->mib.udp_statistics, 15631823e4c8SEric Dumazet sizeof(struct udp_mib), 15641823e4c8SEric Dumazet __alignof__(struct udp_mib)) < 0) 15652f275f91SPavel Emelyanov goto err_udp_mib; 15667d720c3eSTejun Heo if (snmp_mib_init((void __percpu **)net->mib.udplite_statistics, 15671823e4c8SEric Dumazet sizeof(struct udp_mib), 15681823e4c8SEric Dumazet __alignof__(struct udp_mib)) < 0) 1569386019d3SPavel Emelyanov goto err_udplite_mib; 15707d720c3eSTejun Heo if (snmp_mib_init((void __percpu **)net->mib.icmp_statistics, 15711823e4c8SEric Dumazet sizeof(struct icmp_mib), 15721823e4c8SEric Dumazet __alignof__(struct icmp_mib)) < 0) 1573b60538a0SPavel Emelyanov goto err_icmp_mib; 1574acb32ba3SEric Dumazet net->mib.icmpmsg_statistics = kzalloc(sizeof(struct icmpmsg_mib), 1575acb32ba3SEric Dumazet GFP_KERNEL); 1576acb32ba3SEric Dumazet if (!net->mib.icmpmsg_statistics) 1577923c6586SPavel Emelyanov goto err_icmpmsg_mib; 157857ef42d5SPavel Emelyanov 157957ef42d5SPavel Emelyanov tcp_mib_init(net); 15809b4661bdSPavel Emelyanov return 0; 158157ef42d5SPavel Emelyanov 1582923c6586SPavel Emelyanov err_icmpmsg_mib: 15837d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.icmp_statistics); 1584b60538a0SPavel Emelyanov err_icmp_mib: 15857d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.udplite_statistics); 1586386019d3SPavel Emelyanov err_udplite_mib: 15877d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.udp_statistics); 15882f275f91SPavel Emelyanov err_udp_mib: 15897d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.net_statistics); 159061a7e260SPavel Emelyanov err_net_mib: 15917d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.ip_statistics); 1592a20f5799SPavel Emelyanov err_ip_mib: 15937d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.tcp_statistics); 159457ef42d5SPavel Emelyanov err_tcp_mib: 159557ef42d5SPavel Emelyanov return -ENOMEM; 15969b4661bdSPavel Emelyanov } 15979b4661bdSPavel Emelyanov 15989b4661bdSPavel Emelyanov static __net_exit void ipv4_mib_exit_net(struct net *net) 15999b4661bdSPavel Emelyanov { 1600acb32ba3SEric Dumazet kfree(net->mib.icmpmsg_statistics); 16017d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.icmp_statistics); 16027d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.udplite_statistics); 16037d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.udp_statistics); 16047d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.net_statistics); 16057d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.ip_statistics); 16067d720c3eSTejun Heo snmp_mib_free((void __percpu **)net->mib.tcp_statistics); 16079b4661bdSPavel Emelyanov } 16089b4661bdSPavel Emelyanov 16099b4661bdSPavel Emelyanov static __net_initdata struct pernet_operations ipv4_mib_ops = { 16109b4661bdSPavel Emelyanov .init = ipv4_mib_init_net, 16119b4661bdSPavel Emelyanov .exit = ipv4_mib_exit_net, 16129b4661bdSPavel Emelyanov }; 16139b4661bdSPavel Emelyanov 16141da177e4SLinus Torvalds static int __init init_ipv4_mibs(void) 16151da177e4SLinus Torvalds { 1616d89cbbb1SPavel Emelyanov return register_pernet_subsys(&ipv4_mib_ops); 16171da177e4SLinus Torvalds } 16181da177e4SLinus Torvalds 16191da177e4SLinus Torvalds static int ipv4_proc_init(void); 16201da177e4SLinus Torvalds 162130e224d7SHerbert Xu /* 162230e224d7SHerbert Xu * IP protocol layer initialiser 162330e224d7SHerbert Xu */ 162430e224d7SHerbert Xu 16257546dd97SStephen Hemminger static struct packet_type ip_packet_type __read_mostly = { 162609640e63SHarvey Harrison .type = cpu_to_be16(ETH_P_IP), 162730e224d7SHerbert Xu .func = ip_rcv, 1628a430a43dSHerbert Xu .gso_send_check = inet_gso_send_check, 1629f4c50d99SHerbert Xu .gso_segment = inet_gso_segment, 163073cc19f1SHerbert Xu .gro_receive = inet_gro_receive, 163173cc19f1SHerbert Xu .gro_complete = inet_gro_complete, 163230e224d7SHerbert Xu }; 163330e224d7SHerbert Xu 16341da177e4SLinus Torvalds static int __init inet_init(void) 16351da177e4SLinus Torvalds { 16361da177e4SLinus Torvalds struct sk_buff *dummy_skb; 16371da177e4SLinus Torvalds struct inet_protosw *q; 16381da177e4SLinus Torvalds struct list_head *r; 16391da177e4SLinus Torvalds int rc = -EINVAL; 16401da177e4SLinus Torvalds 1641ef047f5eSYOSHIFUJI Hideaki BUILD_BUG_ON(sizeof(struct inet_skb_parm) > sizeof(dummy_skb->cb)); 16421da177e4SLinus Torvalds 1643e3826f1eSAmerigo Wang sysctl_local_reserved_ports = kzalloc(65536 / 8, GFP_KERNEL); 1644e3826f1eSAmerigo Wang if (!sysctl_local_reserved_ports) 1645e3826f1eSAmerigo Wang goto out; 1646e3826f1eSAmerigo Wang 16471da177e4SLinus Torvalds rc = proto_register(&tcp_prot, 1); 16481da177e4SLinus Torvalds if (rc) 1649e3826f1eSAmerigo Wang goto out_free_reserved_ports; 16501da177e4SLinus Torvalds 16511da177e4SLinus Torvalds rc = proto_register(&udp_prot, 1); 16521da177e4SLinus Torvalds if (rc) 16531da177e4SLinus Torvalds goto out_unregister_tcp_proto; 16541da177e4SLinus Torvalds 16551da177e4SLinus Torvalds rc = proto_register(&raw_prot, 1); 16561da177e4SLinus Torvalds if (rc) 16571da177e4SLinus Torvalds goto out_unregister_udp_proto; 16581da177e4SLinus Torvalds 1659c319b4d7SVasiliy Kulikov rc = proto_register(&ping_prot, 1); 1660c319b4d7SVasiliy Kulikov if (rc) 1661c319b4d7SVasiliy Kulikov goto out_unregister_raw_proto; 1662c319b4d7SVasiliy Kulikov 16631da177e4SLinus Torvalds /* 16641da177e4SLinus Torvalds * Tell SOCKET that we are alive... 16651da177e4SLinus Torvalds */ 16661da177e4SLinus Torvalds 16671da177e4SLinus Torvalds (void)sock_register(&inet_family_ops); 16681da177e4SLinus Torvalds 1669bd7b1533SAl Viro #ifdef CONFIG_SYSCTL 1670bd7b1533SAl Viro ip_static_sysctl_init(); 1671bd7b1533SAl Viro #endif 1672bd7b1533SAl Viro 16733dc43e3eSGlauber Costa tcp_prot.sysctl_mem = init_net.ipv4.sysctl_tcp_mem; 16743dc43e3eSGlauber Costa 16751da177e4SLinus Torvalds /* 16761da177e4SLinus Torvalds * Add all the base protocols. 16771da177e4SLinus Torvalds */ 16781da177e4SLinus Torvalds 16791da177e4SLinus Torvalds if (inet_add_protocol(&icmp_protocol, IPPROTO_ICMP) < 0) 1680058bd4d2SJoe Perches pr_crit("%s: Cannot add ICMP protocol\n", __func__); 16811da177e4SLinus Torvalds if (inet_add_protocol(&udp_protocol, IPPROTO_UDP) < 0) 1682058bd4d2SJoe Perches pr_crit("%s: Cannot add UDP protocol\n", __func__); 16831da177e4SLinus Torvalds if (inet_add_protocol(&tcp_protocol, IPPROTO_TCP) < 0) 1684058bd4d2SJoe Perches pr_crit("%s: Cannot add TCP protocol\n", __func__); 16851da177e4SLinus Torvalds #ifdef CONFIG_IP_MULTICAST 16861da177e4SLinus Torvalds if (inet_add_protocol(&igmp_protocol, IPPROTO_IGMP) < 0) 1687058bd4d2SJoe Perches pr_crit("%s: Cannot add IGMP protocol\n", __func__); 16881da177e4SLinus Torvalds #endif 16891da177e4SLinus Torvalds 16901da177e4SLinus Torvalds /* Register the socket-side information for inet_create. */ 16911da177e4SLinus Torvalds for (r = &inetsw[0]; r < &inetsw[SOCK_MAX]; ++r) 16921da177e4SLinus Torvalds INIT_LIST_HEAD(r); 16931da177e4SLinus Torvalds 16941da177e4SLinus Torvalds for (q = inetsw_array; q < &inetsw_array[INETSW_ARRAY_LEN]; ++q) 16951da177e4SLinus Torvalds inet_register_protosw(q); 16961da177e4SLinus Torvalds 16971da177e4SLinus Torvalds /* 16981da177e4SLinus Torvalds * Set the ARP module up 16991da177e4SLinus Torvalds */ 17001da177e4SLinus Torvalds 17011da177e4SLinus Torvalds arp_init(); 17021da177e4SLinus Torvalds 17031da177e4SLinus Torvalds /* 17041da177e4SLinus Torvalds * Set the IP module up 17051da177e4SLinus Torvalds */ 17061da177e4SLinus Torvalds 17071da177e4SLinus Torvalds ip_init(); 17081da177e4SLinus Torvalds 17099b0f976fSDenis V. Lunev tcp_v4_init(); 17101da177e4SLinus Torvalds 17111da177e4SLinus Torvalds /* Setup TCP slab cache for open requests. */ 17121da177e4SLinus Torvalds tcp_init(); 17131da177e4SLinus Torvalds 171495766fffSHideo Aoki /* Setup UDP memory threshold */ 171595766fffSHideo Aoki udp_init(); 171695766fffSHideo Aoki 1717ba4e58ecSGerrit Renker /* Add UDP-Lite (RFC 3828) */ 1718ba4e58ecSGerrit Renker udplite4_register(); 17191da177e4SLinus Torvalds 1720c319b4d7SVasiliy Kulikov ping_init(); 1721c319b4d7SVasiliy Kulikov 17221da177e4SLinus Torvalds /* 17231da177e4SLinus Torvalds * Set the ICMP layer up 17241da177e4SLinus Torvalds */ 17251da177e4SLinus Torvalds 1726a5710d65SDenis V. Lunev if (icmp_init() < 0) 1727a5710d65SDenis V. Lunev panic("Failed to create the ICMP control socket.\n"); 17281da177e4SLinus Torvalds 17291da177e4SLinus Torvalds /* 17301da177e4SLinus Torvalds * Initialise the multicast router 17311da177e4SLinus Torvalds */ 17321da177e4SLinus Torvalds #if defined(CONFIG_IP_MROUTE) 173303d2f897SWang Chen if (ip_mr_init()) 1734058bd4d2SJoe Perches pr_crit("%s: Cannot init ipv4 mroute\n", __func__); 17351da177e4SLinus Torvalds #endif 17361da177e4SLinus Torvalds /* 17371da177e4SLinus Torvalds * Initialise per-cpu ipv4 mibs 17381da177e4SLinus Torvalds */ 17391da177e4SLinus Torvalds 17401da177e4SLinus Torvalds if (init_ipv4_mibs()) 1741058bd4d2SJoe Perches pr_crit("%s: Cannot init ipv4 mibs\n", __func__); 17421da177e4SLinus Torvalds 17431da177e4SLinus Torvalds ipv4_proc_init(); 17441da177e4SLinus Torvalds 17451da177e4SLinus Torvalds ipfrag_init(); 17461da177e4SLinus Torvalds 174730e224d7SHerbert Xu dev_add_pack(&ip_packet_type); 174830e224d7SHerbert Xu 17491da177e4SLinus Torvalds rc = 0; 17501da177e4SLinus Torvalds out: 17511da177e4SLinus Torvalds return rc; 1752c319b4d7SVasiliy Kulikov out_unregister_raw_proto: 1753c319b4d7SVasiliy Kulikov proto_unregister(&raw_prot); 17541da177e4SLinus Torvalds out_unregister_udp_proto: 17551da177e4SLinus Torvalds proto_unregister(&udp_prot); 1756321efff7SOlaf Kirch out_unregister_tcp_proto: 1757321efff7SOlaf Kirch proto_unregister(&tcp_prot); 1758e3826f1eSAmerigo Wang out_free_reserved_ports: 1759e3826f1eSAmerigo Wang kfree(sysctl_local_reserved_ports); 17601da177e4SLinus Torvalds goto out; 17611da177e4SLinus Torvalds } 17621da177e4SLinus Torvalds 1763a536e077SHeiko Carstens fs_initcall(inet_init); 17641da177e4SLinus Torvalds 17651da177e4SLinus Torvalds /* ------------------------------------------------------------------------ */ 17661da177e4SLinus Torvalds 17671da177e4SLinus Torvalds #ifdef CONFIG_PROC_FS 17681da177e4SLinus Torvalds static int __init ipv4_proc_init(void) 17691da177e4SLinus Torvalds { 17701da177e4SLinus Torvalds int rc = 0; 17711da177e4SLinus Torvalds 17721da177e4SLinus Torvalds if (raw_proc_init()) 17731da177e4SLinus Torvalds goto out_raw; 17741da177e4SLinus Torvalds if (tcp4_proc_init()) 17751da177e4SLinus Torvalds goto out_tcp; 17761da177e4SLinus Torvalds if (udp4_proc_init()) 17771da177e4SLinus Torvalds goto out_udp; 1778c319b4d7SVasiliy Kulikov if (ping_proc_init()) 1779c319b4d7SVasiliy Kulikov goto out_ping; 17801da177e4SLinus Torvalds if (ip_misc_proc_init()) 17811da177e4SLinus Torvalds goto out_misc; 17821da177e4SLinus Torvalds out: 17831da177e4SLinus Torvalds return rc; 17841da177e4SLinus Torvalds out_misc: 1785c319b4d7SVasiliy Kulikov ping_proc_exit(); 1786c319b4d7SVasiliy Kulikov out_ping: 17871da177e4SLinus Torvalds udp4_proc_exit(); 17881da177e4SLinus Torvalds out_udp: 17891da177e4SLinus Torvalds tcp4_proc_exit(); 17901da177e4SLinus Torvalds out_tcp: 17911da177e4SLinus Torvalds raw_proc_exit(); 17921da177e4SLinus Torvalds out_raw: 17931da177e4SLinus Torvalds rc = -ENOMEM; 17941da177e4SLinus Torvalds goto out; 17951da177e4SLinus Torvalds } 17961da177e4SLinus Torvalds 17971da177e4SLinus Torvalds #else /* CONFIG_PROC_FS */ 17981da177e4SLinus Torvalds static int __init ipv4_proc_init(void) 17991da177e4SLinus Torvalds { 18001da177e4SLinus Torvalds return 0; 18011da177e4SLinus Torvalds } 18021da177e4SLinus Torvalds #endif /* CONFIG_PROC_FS */ 18031da177e4SLinus Torvalds 18041da177e4SLinus Torvalds MODULE_ALIAS_NETPROTO(PF_INET); 18051da177e4SLinus Torvalds 1806