net/ipv4/ip_sockglue.c

   1 /*
   2  * INET         An implementation of the TCP/IP protocol suite for the LINUX
   3  *              operating system.  INET is implemented using the  BSD Socket
   4  *              interface as the means of communication with the user level.
   5  *
   6  *              The IP to API glue.
   7  *
   8  * Authors:     see ip.c
   9  *
  10  * Fixes:
  11  *              Many            :       Split from ip.c , see ip.c for history.
  12  *              Martin Mares    :       TOS setting fixed.
  13  *              Alan Cox        :       Fixed a couple of oopses in Martin's
  14  *                                      TOS tweaks.
  15  *              Mike McLagan    :       Routing by source
  16  */
  17
  18 #include <linux/module.h>
  19 #include <linux/types.h>
  20 #include <linux/mm.h>
  21 #include <linux/skbuff.h>
  22 #include <linux/ip.h>
  23 #include <linux/icmp.h>
  24 #include <linux/inetdevice.h>
  25 #include <linux/netdevice.h>
  26 #include <linux/slab.h>
  27 #include <net/sock.h>
  28 #include <net/ip.h>
  29 #include <net/icmp.h>
  30 #include <net/tcp_states.h>
  31 #include <linux/udp.h>
  32 #include <linux/igmp.h>
  33 #include <linux/netfilter.h>
  34 #include <linux/route.h>
  35 #include <linux/mroute.h>
  36 #include <net/inet_ecn.h>
  37 #include <net/route.h>
  38 #include <net/xfrm.h>
  39 #include <net/compat.h>
  40 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
  41 #include <net/transp_v6.h>
  42 #endif
  43
  44 #include <linux/errqueue.h>
  45 #include <asm/uaccess.h>
  46
  47 #define IP_CMSG_PKTINFO         1
  48 #define IP_CMSG_TTL             2
  49 #define IP_CMSG_TOS             4
  50 #define IP_CMSG_RECVOPTS        8
  51 #define IP_CMSG_RETOPTS         16
  52 #define IP_CMSG_PASSSEC         32
  53 #define IP_CMSG_ORIGDSTADDR     64
  54
  55 /*
  56  *      SOL_IP control messages.
  57  */
  58
  59 static void ip_cmsg_recv_pktinfo(struct msghdr *msg, struct sk_buff *skb)
  60 {
  61         struct in_pktinfo info;
  62         struct rtable *rt = skb_rtable(skb);
  63
  64         info.ipi_addr.s_addr = ip_hdr(skb)->daddr;
  65         if (rt) {
  66                 info.ipi_ifindex = rt->rt_iif;
  67                 info.ipi_spec_dst.s_addr = rt->rt_spec_dst;
  68         } else {
  69                 info.ipi_ifindex = 0;
  70                 info.ipi_spec_dst.s_addr = 0;
  71         }
  72
  73         put_cmsg(msg, SOL_IP, IP_PKTINFO, sizeof(info), &info);
  74 }
  75
  76 static void ip_cmsg_recv_ttl(struct msghdr *msg, struct sk_buff *skb)
  77 {
  78         int ttl = ip_hdr(skb)->ttl;
  79         put_cmsg(msg, SOL_IP, IP_TTL, sizeof(int), &ttl);
  80 }
  81
  82 static void ip_cmsg_recv_tos(struct msghdr *msg, struct sk_buff *skb)
  83 {
  84         put_cmsg(msg, SOL_IP, IP_TOS, 1, &ip_hdr(skb)->tos);
  85 }
  86
  87 static void ip_cmsg_recv_opts(struct msghdr *msg, struct sk_buff *skb)
  88 {
  89         if (IPCB(skb)->opt.optlen == 0)
  90                 return;
  91
  92         put_cmsg(msg, SOL_IP, IP_RECVOPTS, IPCB(skb)->opt.optlen,
  93                  ip_hdr(skb) + 1);
  94 }
  95
  96
  97 static void ip_cmsg_recv_retopts(struct msghdr *msg, struct sk_buff *skb)
  98 {
  99         unsigned char optbuf[sizeof(struct ip_options) + 40];
 100         struct ip_options * opt = (struct ip_options *)optbuf;
 101
 102         if (IPCB(skb)->opt.optlen == 0)
 103                 return;
 104
 105         if (ip_options_echo(opt, skb)) {
 106                 msg->msg_flags |= MSG_CTRUNC;
 107                 return;
 108         }
 109         ip_options_undo(opt);
 110
 111         put_cmsg(msg, SOL_IP, IP_RETOPTS, opt->optlen, opt->__data);
 112 }
 113
 114 static void ip_cmsg_recv_security(struct msghdr *msg, struct sk_buff *skb)
 115 {
 116         char *secdata;
 117         u32 seclen, secid;
 118         int err;
 119
 120         err = security_socket_getpeersec_dgram(NULL, skb, &secid);
 121         if (err)
 122                 return;
 123
 124         err = security_secid_to_secctx(secid, &secdata, &seclen);
 125         if (err)
 126                 return;
 127
 128         put_cmsg(msg, SOL_IP, SCM_SECURITY, seclen, secdata);
 129         security_release_secctx(secdata, seclen);
 130 }
 131
 132 static void ip_cmsg_recv_dstaddr(struct msghdr *msg, struct sk_buff *skb)
 133 {
 134         struct sockaddr_in sin;
 135         const struct iphdr *iph = ip_hdr(skb);
 136         __be16 *ports = (__be16 *)skb_transport_header(skb);
 137
 138         if (skb_transport_offset(skb) + 4 > skb->len)
 139                 return;
 140
 141         /* All current transport protocols have the port numbers in the
 142          * first four bytes of the transport header and this function is
 143          * written with this assumption in mind.
 144          */
 145
 146         sin.sin_family = AF_INET;
 147         sin.sin_addr.s_addr = iph->daddr;
 148         sin.sin_port = ports[1];
 149         memset(sin.sin_zero, 0, sizeof(sin.sin_zero));
 150
 151         put_cmsg(msg, SOL_IP, IP_ORIGDSTADDR, sizeof(sin), &sin);
 152 }
 153
 154 void ip_cmsg_recv(struct msghdr *msg, struct sk_buff *skb)
 155 {
 156         struct inet_sock *inet = inet_sk(skb->sk);
 157         unsigned flags = inet->cmsg_flags;
 158
 159         /* Ordered by supposed usage frequency */
 160         if (flags & 1)
 161                 ip_cmsg_recv_pktinfo(msg, skb);
 162         if ((flags >>= 1) == 0)
 163                 return;
 164
 165         if (flags & 1)
 166                 ip_cmsg_recv_ttl(msg, skb);
 167         if ((flags >>= 1) == 0)
 168                 return;
 169
 170         if (flags & 1)
 171                 ip_cmsg_recv_tos(msg, skb);
 172         if ((flags >>= 1) == 0)
 173                 return;
 174
 175         if (flags & 1)
 176                 ip_cmsg_recv_opts(msg, skb);
 177         if ((flags >>= 1) == 0)
 178                 return;
 179
 180         if (flags & 1)
 181                 ip_cmsg_recv_retopts(msg, skb);
 182         if ((flags >>= 1) == 0)
 183                 return;
 184
 185         if (flags & 1)
 186                 ip_cmsg_recv_security(msg, skb);
 187
 188         if ((flags >>= 1) == 0)
 189                 return;
 190         if (flags & 1)
 191                 ip_cmsg_recv_dstaddr(msg, skb);
 192
 193 }
 194 EXPORT_SYMBOL(ip_cmsg_recv);
 195
 196 int ip_cmsg_send(struct net *net, struct msghdr *msg, struct ipcm_cookie *ipc)
 197 {
 198         int err;
 199         struct cmsghdr *cmsg;
 200
 201         for (cmsg = CMSG_FIRSTHDR(msg); cmsg; cmsg = CMSG_NXTHDR(msg, cmsg)) {
 202                 if (!CMSG_OK(msg, cmsg))
 203                         return -EINVAL;
 204                 if (cmsg->cmsg_level != SOL_IP)
 205                         continue;
 206                 switch (cmsg->cmsg_type) {
 207                 case IP_RETOPTS:
 208                         err = cmsg->cmsg_len - CMSG_ALIGN(sizeof(struct cmsghdr));
 209
 210                         /* Our caller is responsible for freeing ipc->opt */
 211                         err = ip_options_get(net, &ipc->opt, CMSG_DATA(cmsg),
 212                                              err < 40 ? err : 40);
 213                         if (err)
 214                                 return err;
 215                         break;
 216                 case IP_PKTINFO:
 217                 {
 218                         struct in_pktinfo *info;
 219                         if (cmsg->cmsg_len != CMSG_LEN(sizeof(struct in_pktinfo)))
 220                                 return -EINVAL;
 221                         info = (struct in_pktinfo *)CMSG_DATA(cmsg);
 222                         ipc->oif = info->ipi_ifindex;
 223                         ipc->addr = info->ipi_spec_dst.s_addr;
 224                         break;
 225                 }
 226                 default:
 227                         return -EINVAL;
 228                 }
 229         }
 230         return 0;
 231 }
 232
 233
 234 /* Special input handler for packets caught by router alert option.
 235    They are selected only by protocol field, and then processed likely
 236    local ones; but only if someone wants them! Otherwise, router
 237    not running rsvpd will kill RSVP.
 238
 239    It is user level problem, what it will make with them.
 240    I have no idea, how it will masquearde or NAT them (it is joke, joke :-)),
 241    but receiver should be enough clever f.e. to forward mtrace requests,
 242    sent to multicast group to reach destination designated router.
 243  */
 244 struct ip_ra_chain __rcu *ip_ra_chain;
 245 static DEFINE_SPINLOCK(ip_ra_lock);
 246
 247
 248 static void ip_ra_destroy_rcu(struct rcu_head *head)
 249 {
 250         struct ip_ra_chain *ra = container_of(head, struct ip_ra_chain, rcu);
 251
 252         sock_put(ra->saved_sk);
 253         kfree(ra);
 254 }
 255
 256 int ip_ra_control(struct sock *sk, unsigned char on,
 257                   void (*destructor)(struct sock *))
 258 {
 259         struct ip_ra_chain *ra, *new_ra;
 260         struct ip_ra_chain __rcu **rap;
 261
 262         if (sk->sk_type != SOCK_RAW || inet_sk(sk)->inet_num == IPPROTO_RAW)
 263                 return -EINVAL;
 264
 265         new_ra = on ? kmalloc(sizeof(*new_ra), GFP_KERNEL) : NULL;
 266
 267         spin_lock_bh(&ip_ra_lock);
 268         for (rap = &ip_ra_chain;
 269              (ra = rcu_dereference_protected(*rap,
 270                         lockdep_is_held(&ip_ra_lock))) != NULL;
 271              rap = &ra->next) {
 272                 if (ra->sk == sk) {
 273                         if (on) {
 274                                 spin_unlock_bh(&ip_ra_lock);
 275                                 kfree(new_ra);
 276                                 return -EADDRINUSE;
 277                         }
 278                         /* dont let ip_call_ra_chain() use sk again */
 279                         ra->sk = NULL;
 280                         rcu_assign_pointer(*rap, ra->next);
 281                         spin_unlock_bh(&ip_ra_lock);
 282
 283                         if (ra->destructor)
 284                                 ra->destructor(sk);
 285                         /*
 286                          * Delay sock_put(sk) and kfree(ra) after one rcu grace
 287                          * period. This guarantee ip_call_ra_chain() dont need
 288                          * to mess with socket refcounts.
 289                          */
 290                         ra->saved_sk = sk;
 291                         call_rcu(&ra->rcu, ip_ra_destroy_rcu);
 292                         return 0;
 293                 }
 294         }
 295         if (new_ra == NULL) {
 296                 spin_unlock_bh(&ip_ra_lock);
 297                 return -ENOBUFS;
 298         }
 299         new_ra->sk = sk;
 300         new_ra->destructor = destructor;
 301
 302         new_ra->next = ra;
 303         rcu_assign_pointer(*rap, new_ra);
 304         sock_hold(sk);
 305         spin_unlock_bh(&ip_ra_lock);
 306
 307         return 0;
 308 }
 309
 310 void ip_icmp_error(struct sock *sk, struct sk_buff *skb, int err,
 311                    __be16 port, u32 info, u8 *payload)
 312 {
 313         struct sock_exterr_skb *serr;
 314
 315         skb = skb_clone(skb, GFP_ATOMIC);
 316         if (!skb)
 317                 return;
 318
 319         serr = SKB_EXT_ERR(skb);
 320         serr->ee.ee_errno = err;
 321         serr->ee.ee_origin = SO_EE_ORIGIN_ICMP;
 322         serr->ee.ee_type = icmp_hdr(skb)->type;
 323         serr->ee.ee_code = icmp_hdr(skb)->code;
 324         serr->ee.ee_pad = 0;
 325         serr->ee.ee_info = info;
 326         serr->ee.ee_data = 0;
 327         serr->addr_offset = (u8 *)&(((struct iphdr *)(icmp_hdr(skb) + 1))->daddr) -
 328                                    skb_network_header(skb);
 329         serr->port = port;
 330
 331         if (skb_pull(skb, payload - skb->data) != NULL) {
 332                 skb_reset_transport_header(skb);
 333                 if (sock_queue_err_skb(sk, skb) == 0)
 334                         return;
 335         }
 336         kfree_skb(skb);
 337 }
 338
 339 void ip_local_error(struct sock *sk, int err, __be32 daddr, __be16 port, u32 info)
 340 {
 341         struct inet_sock *inet = inet_sk(sk);
 342         struct sock_exterr_skb *serr;
 343         struct iphdr *iph;
 344         struct sk_buff *skb;
 345
 346         if (!inet->recverr)
 347                 return;
 348
 349         skb = alloc_skb(sizeof(struct iphdr), GFP_ATOMIC);
 350         if (!skb)
 351                 return;
 352
 353         skb_put(skb, sizeof(struct iphdr));
 354         skb_reset_network_header(skb);
 355         iph = ip_hdr(skb);
 356         iph->daddr = daddr;
 357
 358         serr = SKB_EXT_ERR(skb);
 359         serr->ee.ee_errno = err;
 360         serr->ee.ee_origin = SO_EE_ORIGIN_LOCAL;
 361         serr->ee.ee_type = 0;
 362         serr->ee.ee_code = 0;
 363         serr->ee.ee_pad = 0;
 364         serr->ee.ee_info = info;
 365         serr->ee.ee_data = 0;
 366         serr->addr_offset = (u8 *)&iph->daddr - skb_network_header(skb);
 367         serr->port = port;
 368
 369         __skb_pull(skb, skb_tail_pointer(skb) - skb->data);
 370         skb_reset_transport_header(skb);
 371
 372         if (sock_queue_err_skb(sk, skb))
 373                 kfree_skb(skb);
 374 }
 375
 376 /*
 377  *      Handle MSG_ERRQUEUE
 378  */
 379 int ip_recv_error(struct sock *sk, struct msghdr *msg, int len, int *addr_len)
 380 {
 381         struct sock_exterr_skb *serr;
 382         struct sk_buff *skb, *skb2;
 383         struct sockaddr_in *sin;
 384         struct {
 385                 struct sock_extended_err ee;
 386                 struct sockaddr_in       offender;
 387         } errhdr;
 388         int err;
 389         int copied;
 390
 391         err = -EAGAIN;
 392         skb = skb_dequeue(&sk->sk_error_queue);
 393         if (skb == NULL)
 394                 goto out;
 395
 396         copied = skb->len;
 397         if (copied > len) {
 398                 msg->msg_flags |= MSG_TRUNC;
 399                 copied = len;
 400         }
 401         err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, copied);
 402         if (err)
 403                 goto out_free_skb;
 404
 405         sock_recv_timestamp(msg, sk, skb);
 406
 407         serr = SKB_EXT_ERR(skb);
 408
 409         sin = (struct sockaddr_in *)msg->msg_name;
 410         if (sin) {
 411                 sin->sin_family = AF_INET;
 412                 sin->sin_addr.s_addr = *(__be32 *)(skb_network_header(skb) +
 413                                                    serr->addr_offset);
 414                 sin->sin_port = serr->port;
 415                 memset(&sin->sin_zero, 0, sizeof(sin->sin_zero));
 416                 *addr_len = sizeof(*sin);
 417         }
 418
 419         memcpy(&errhdr.ee, &serr->ee, sizeof(struct sock_extended_err));
 420         sin = &errhdr.offender;
 421         memset(sin, 0, sizeof(*sin));
 422         if (serr->ee.ee_origin == SO_EE_ORIGIN_ICMP) {
 423                 sin->sin_family = AF_INET;
 424                 sin->sin_addr.s_addr = ip_hdr(skb)->saddr;
 425                 if (inet_sk(sk)->cmsg_flags)
 426                         ip_cmsg_recv(msg, skb);
 427         }
 428
 429         put_cmsg(msg, SOL_IP, IP_RECVERR, sizeof(errhdr), &errhdr);
 430
 431         /* Now we could try to dump offended packet options */
 432
 433         msg->msg_flags |= MSG_ERRQUEUE;
 434         err = copied;
 435
 436         /* Reset and regenerate socket error */
 437         spin_lock_bh(&sk->sk_error_queue.lock);
 438         sk->sk_err = 0;
 439         skb2 = skb_peek(&sk->sk_error_queue);
 440         if (skb2 != NULL) {
 441                 sk->sk_err = SKB_EXT_ERR(skb2)->ee.ee_errno;
 442                 spin_unlock_bh(&sk->sk_error_queue.lock);
 443                 sk->sk_error_report(sk);
 444         } else
 445                 spin_unlock_bh(&sk->sk_error_queue.lock);
 446
 447 out_free_skb:
 448         kfree_skb(skb);
 449 out:
 450         return err;
 451 }
 452
 453
 454 static void opt_kfree_rcu(struct rcu_head *head)
 455 {
 456         kfree(container_of(head, struct ip_options_rcu, rcu));
 457 }
 458
 459 /*
 460  *      Socket option code for IP. This is the end of the line after any
 461  *      TCP,UDP etc options on an IP socket.
 462  */
 463
 464 static int do_ip_setsockopt(struct sock *sk, int level,
 465                             int optname, char __user *optval, unsigned int optlen)
 466 {
 467         struct inet_sock *inet = inet_sk(sk);
 468         int val = 0, err;
 469
 470         switch (optname) {
 471         case IP_PKTINFO:
 472         case IP_RECVTTL:
 473         case IP_RECVOPTS:
 474         case IP_RECVTOS:
 475         case IP_RETOPTS:
 476         case IP_TOS:
 477         case IP_TTL:
 478         case IP_HDRINCL:
 479         case IP_MTU_DISCOVER:
 480         case IP_RECVERR:
 481         case IP_ROUTER_ALERT:
 482         case IP_FREEBIND:
 483         case IP_PASSSEC:
 484         case IP_TRANSPARENT:
 485         case IP_MINTTL:
 486         case IP_NODEFRAG:
 487         case IP_MULTICAST_TTL:
 488         case IP_MULTICAST_ALL:
 489         case IP_MULTICAST_LOOP:
 490         case IP_RECVORIGDSTADDR:
 491                 if (optlen >= sizeof(int)) {
 492                         if (get_user(val, (int __user *) optval))
 493                                 return -EFAULT;
 494                 } else if (optlen >= sizeof(char)) {
 495                         unsigned char ucval;
 496
 497                         if (get_user(ucval, (unsigned char __user *) optval))
 498                                 return -EFAULT;
 499                         val = (int) ucval;
 500                 }
 501         }
 502
 503         /* If optlen==0, it is equivalent to val == 0 */
 504
 505         if (ip_mroute_opt(optname))
 506                 return ip_mroute_setsockopt(sk, optname, optval, optlen);
 507
 508         err = 0;
 509         lock_sock(sk);
 510
 511         switch (optname) {
 512         case IP_OPTIONS:
 513         {
 514                 struct ip_options_rcu *old, *opt = NULL;
 515
 516                 if (optlen > 40)
 517                         goto e_inval;
 518                 err = ip_options_get_from_user(sock_net(sk), &opt,
 519                                                optval, optlen);
 520                 if (err)
 521                         break;
 522                 old = rcu_dereference_protected(inet->inet_opt,
 523                                                 sock_owned_by_user(sk));
 524                 if (inet->is_icsk) {
 525                         struct inet_connection_sock *icsk = inet_csk(sk);
 526 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
 527                         if (sk->sk_family == PF_INET ||
 528                             (!((1 << sk->sk_state) &
 529                                (TCPF_LISTEN | TCPF_CLOSE)) &&
 530                              inet->inet_daddr != LOOPBACK4_IPV6)) {
 531 #endif
 532                                 if (old)
 533                                         icsk->icsk_ext_hdr_len -= old->opt.optlen;
 534                                 if (opt)
 535                                         icsk->icsk_ext_hdr_len += opt->opt.optlen;
 536                                 icsk->icsk_sync_mss(sk, icsk->icsk_pmtu_cookie);
 537 #if defined(CONFIG_IPV6) || defined(CONFIG_IPV6_MODULE)
 538                         }
 539 #endif
 540                 }
 541                 rcu_assign_pointer(inet->inet_opt, opt);
 542                 if (old)
 543                         call_rcu(&old->rcu, opt_kfree_rcu);
 544                 break;
 545         }
 546         case IP_PKTINFO:
 547                 if (val)
 548                         inet->cmsg_flags |= IP_CMSG_PKTINFO;
 549                 else
 550                         inet->cmsg_flags &= ~IP_CMSG_PKTINFO;
 551                 break;
 552         case IP_RECVTTL:
 553                 if (val)
 554                         inet->cmsg_flags |=  IP_CMSG_TTL;
 555                 else
 556                         inet->cmsg_flags &= ~IP_CMSG_TTL;
 557                 break;
 558         case IP_RECVTOS:
 559                 if (val)
 560                         inet->cmsg_flags |=  IP_CMSG_TOS;
 561                 else
 562                         inet->cmsg_flags &= ~IP_CMSG_TOS;
 563                 break;
 564         case IP_RECVOPTS:
 565                 if (val)
 566                         inet->cmsg_flags |=  IP_CMSG_RECVOPTS;
 567                 else
 568                         inet->cmsg_flags &= ~IP_CMSG_RECVOPTS;
 569                 break;
 570         case IP_RETOPTS:
 571                 if (val)
 572                         inet->cmsg_flags |= IP_CMSG_RETOPTS;
 573                 else
 574                         inet->cmsg_flags &= ~IP_CMSG_RETOPTS;
 575                 break;
 576         case IP_PASSSEC:
 577                 if (val)
 578                         inet->cmsg_flags |= IP_CMSG_PASSSEC;
 579                 else
 580                         inet->cmsg_flags &= ~IP_CMSG_PASSSEC;
 581                 break;
 582         case IP_RECVORIGDSTADDR:
 583                 if (val)
 584                         inet->cmsg_flags |= IP_CMSG_ORIGDSTADDR;
 585                 else
 586                         inet->cmsg_flags &= ~IP_CMSG_ORIGDSTADDR;
 587                 break;
 588         case IP_TOS:    /* This sets both TOS and Precedence */
 589                 if (sk->sk_type == SOCK_STREAM) {
 590                         val &= ~INET_ECN_MASK;
 591                         val |= inet->tos & INET_ECN_MASK;
 592                 }
 593                 if (inet->tos != val) {
 594                         inet->tos = val;
 595                         sk->sk_priority = rt_tos2priority(val);
 596                         sk_dst_reset(sk);
 597                 }
 598                 break;
 599         case IP_TTL:
 600                 if (optlen < 1)
 601                         goto e_inval;
 602                 if (val != -1 && (val < 1 || val > 255))
 603                         goto e_inval;
 604                 inet->uc_ttl = val;
 605                 break;
 606         case IP_HDRINCL:
 607                 if (sk->sk_type != SOCK_RAW) {
 608                         err = -ENOPROTOOPT;
 609                         break;
 610                 }
 611                 inet->hdrincl = val ? 1 : 0;
 612                 break;
 613         case IP_NODEFRAG:
 614                 if (sk->sk_type != SOCK_RAW) {
 615                         err = -ENOPROTOOPT;
 616                         break;
 617                 }
 618                 inet->nodefrag = val ? 1 : 0;
 619                 break;
 620         case IP_MTU_DISCOVER:
 621                 if (val < IP_PMTUDISC_DONT || val > IP_PMTUDISC_PROBE)
 622                         goto e_inval;
 623                 inet->pmtudisc = val;
 624                 break;
 625         case IP_RECVERR:
 626                 inet->recverr = !!val;
 627                 if (!val)
 628                         skb_queue_purge(&sk->sk_error_queue);
 629                 break;
 630         case IP_MULTICAST_TTL:
 631                 if (sk->sk_type == SOCK_STREAM)
 632                         goto e_inval;
 633                 if (optlen < 1)
 634                         goto e_inval;
 635                 if (val == -1)
 636                         val = 1;
 637                 if (val < 0 || val > 255)
 638                         goto e_inval;
 639                 inet->mc_ttl = val;
 640                 break;
 641         case IP_MULTICAST_LOOP:
 642                 if (optlen < 1)
 643                         goto e_inval;
 644                 inet->mc_loop = !!val;
 645                 break;
 646         case IP_MULTICAST_IF:
 647         {
 648                 struct ip_mreqn mreq;
 649                 struct net_device *dev = NULL;
 650
 651                 if (sk->sk_type == SOCK_STREAM)
 652                         goto e_inval;
 653                 /*
 654                  *      Check the arguments are allowable
 655                  */
 656
 657                 if (optlen < sizeof(struct in_addr))
 658                         goto e_inval;
 659
 660                 err = -EFAULT;
 661                 if (optlen >= sizeof(struct ip_mreqn)) {
 662                         if (copy_from_user(&mreq, optval, sizeof(mreq)))
 663                                 break;
 664                 } else {
 665                         memset(&mreq, 0, sizeof(mreq));
 666                         if (optlen >= sizeof(struct in_addr) &&
 667                             copy_from_user(&mreq.imr_address, optval,
 668                                            sizeof(struct in_addr)))
 669                                 break;
 670                 }
 671
 672                 if (!mreq.imr_ifindex) {
 673                         if (mreq.imr_address.s_addr == htonl(INADDR_ANY)) {
 674                                 inet->mc_index = 0;
 675                                 inet->mc_addr  = 0;
 676                                 err = 0;
 677                                 break;
 678                         }
 679                         dev = ip_dev_find(sock_net(sk), mreq.imr_address.s_addr);
 680                         if (dev)
 681                                 mreq.imr_ifindex = dev->ifindex;
 682                 } else
 683                         dev = dev_get_by_index(sock_net(sk), mreq.imr_ifindex);
 684
 685
 686                 err = -EADDRNOTAVAIL;
 687                 if (!dev)
 688                         break;
 689                 dev_put(dev);
 690
 691                 err = -EINVAL;
 692                 if (sk->sk_bound_dev_if &&
 693                     mreq.imr_ifindex != sk->sk_bound_dev_if)
 694                         break;
 695
 696                 inet->mc_index = mreq.imr_ifindex;
 697                 inet->mc_addr  = mreq.imr_address.s_addr;
 698                 err = 0;
 699                 break;
 700         }
 701
 702         case IP_ADD_MEMBERSHIP:
 703         case IP_DROP_MEMBERSHIP:
 704         {
 705                 struct ip_mreqn mreq;
 706
 707                 err = -EPROTO;
 708                 if (inet_sk(sk)->is_icsk)
 709                         break;
 710
 711                 if (optlen < sizeof(struct ip_mreq))
 712                         goto e_inval;
 713                 err = -EFAULT;
 714                 if (optlen >= sizeof(struct ip_mreqn)) {
 715                         if (copy_from_user(&mreq, optval, sizeof(mreq)))
 716                                 break;
 717                 } else {
 718                         memset(&mreq, 0, sizeof(mreq));
 719                         if (copy_from_user(&mreq, optval, sizeof(struct ip_mreq)))
 720                                 break;
 721                 }
 722
 723                 if (optname == IP_ADD_MEMBERSHIP)
 724                         err = ip_mc_join_group(sk, &mreq);
 725                 else
 726                         err = ip_mc_leave_group(sk, &mreq);
 727                 break;
 728         }
 729         case IP_MSFILTER:
 730         {
 731                 struct ip_msfilter *msf;
 732
 733                 if (optlen < IP_MSFILTER_SIZE(0))
 734                         goto e_inval;
 735                 if (optlen > sysctl_optmem_max) {
 736                         err = -ENOBUFS;
 737                         break;
 738                 }
 739                 msf = kmalloc(optlen, GFP_KERNEL);
 740                 if (!msf) {
 741                         err = -ENOBUFS;
 742                         break;
 743                 }
 744                 err = -EFAULT;
 745                 if (copy_from_user(msf, optval, optlen)) {
 746                         kfree(msf);
 747                         break;
 748                 }
 749                 /* numsrc >= (1G-4) overflow in 32 bits */
 750                 if (msf->imsf_numsrc >= 0x3ffffffcU ||
 751                     msf->imsf_numsrc > sysctl_igmp_max_msf) {
 752                         kfree(msf);
 753                         err = -ENOBUFS;
 754                         break;
 755                 }
 756                 if (IP_MSFILTER_SIZE(msf->imsf_numsrc) > optlen) {
 757                         kfree(msf);
 758                         err = -EINVAL;
 759                         break;
 760                 }
 761                 err = ip_mc_msfilter(sk, msf, 0);
 762                 kfree(msf);
 763                 break;
 764         }
 765         case IP_BLOCK_SOURCE:
 766         case IP_UNBLOCK_SOURCE:
 767         case IP_ADD_SOURCE_MEMBERSHIP:
 768         case IP_DROP_SOURCE_MEMBERSHIP:
 769         {
 770                 struct ip_mreq_source mreqs;
 771                 int omode, add;
 772
 773                 if (optlen != sizeof(struct ip_mreq_source))
 774                         goto e_inval;
 775                 if (copy_from_user(&mreqs, optval, sizeof(mreqs))) {
 776                         err = -EFAULT;
 777                         break;
 778                 }
 779                 if (optname == IP_BLOCK_SOURCE) {
 780                         omode = MCAST_EXCLUDE;
 781                         add = 1;
 782                 } else if (optname == IP_UNBLOCK_SOURCE) {
 783                         omode = MCAST_EXCLUDE;
 784                         add = 0;
 785                 } else if (optname == IP_ADD_SOURCE_MEMBERSHIP) {
 786                         struct ip_mreqn mreq;
 787
 788                         mreq.imr_multiaddr.s_addr = mreqs.imr_multiaddr;
 789                         mreq.imr_address.s_addr = mreqs.imr_interface;
 790                         mreq.imr_ifindex = 0;
 791                         err = ip_mc_join_group(sk, &mreq);
 792                         if (err && err != -EADDRINUSE)
 793                                 break;
 794                         omode = MCAST_INCLUDE;
 795                         add = 1;
 796                 } else /* IP_DROP_SOURCE_MEMBERSHIP */ {
 797                         omode = MCAST_INCLUDE;
 798                         add = 0;
 799                 }
 800                 err = ip_mc_source(add, omode, sk, &mreqs, 0);
 801                 break;
 802         }
 803         case MCAST_JOIN_GROUP:
 804         case MCAST_LEAVE_GROUP:
 805         {
 806                 struct group_req greq;
 807                 struct sockaddr_in *psin;
 808                 struct ip_mreqn mreq;
 809
 810                 if (optlen < sizeof(struct group_req))
 811                         goto e_inval;
 812                 err = -EFAULT;
 813                 if (copy_from_user(&greq, optval, sizeof(greq)))
 814                         break;
 815                 psin = (struct sockaddr_in *)&greq.gr_group;
 816                 if (psin->sin_family != AF_INET)
 817                         goto e_inval;
 818                 memset(&mreq, 0, sizeof(mreq));
 819                 mreq.imr_multiaddr = psin->sin_addr;
 820                 mreq.imr_ifindex = greq.gr_interface;
 821
 822                 if (optname == MCAST_JOIN_GROUP)
 823                         err = ip_mc_join_group(sk, &mreq);
 824                 else
 825                         err = ip_mc_leave_group(sk, &mreq);
 826                 break;
 827         }
 828         case MCAST_JOIN_SOURCE_GROUP:
 829         case MCAST_LEAVE_SOURCE_GROUP:
 830         case MCAST_BLOCK_SOURCE:
 831         case MCAST_UNBLOCK_SOURCE:
 832         {
 833                 struct group_source_req greqs;
 834                 struct ip_mreq_source mreqs;
 835                 struct sockaddr_in *psin;
 836                 int omode, add;
 837
 838                 if (optlen != sizeof(struct group_source_req))
 839                         goto e_inval;
 840                 if (copy_from_user(&greqs, optval, sizeof(greqs))) {
 841                         err = -EFAULT;
 842                         break;
 843                 }
 844                 if (greqs.gsr_group.ss_family != AF_INET ||
 845                     greqs.gsr_source.ss_family != AF_INET) {
 846                         err = -EADDRNOTAVAIL;
 847                         break;
 848                 }
 849                 psin = (struct sockaddr_in *)&greqs.gsr_group;
 850                 mreqs.imr_multiaddr = psin->sin_addr.s_addr;
 851                 psin = (struct sockaddr_in *)&greqs.gsr_source;
 852                 mreqs.imr_sourceaddr = psin->sin_addr.s_addr;
 853                 mreqs.imr_interface = 0; /* use index for mc_source */
 854
 855                 if (optname == MCAST_BLOCK_SOURCE) {
 856                         omode = MCAST_EXCLUDE;
 857                         add = 1;
 858                 } else if (optname == MCAST_UNBLOCK_SOURCE) {
 859                         omode = MCAST_EXCLUDE;
 860                         add = 0;
 861                 } else if (optname == MCAST_JOIN_SOURCE_GROUP) {
 862                         struct ip_mreqn mreq;
 863
 864                         psin = (struct sockaddr_in *)&greqs.gsr_group;
 865                         mreq.imr_multiaddr = psin->sin_addr;
 866                         mreq.imr_address.s_addr = 0;
 867                         mreq.imr_ifindex = greqs.gsr_interface;
 868                         err = ip_mc_join_group(sk, &mreq);
 869                         if (err && err != -EADDRINUSE)
 870                                 break;
 871                         greqs.gsr_interface = mreq.imr_ifindex;
 872                         omode = MCAST_INCLUDE;
 873                         add = 1;
 874                 } else /* MCAST_LEAVE_SOURCE_GROUP */ {
 875                         omode = MCAST_INCLUDE;
 876                         add = 0;
 877                 }
 878                 err = ip_mc_source(add, omode, sk, &mreqs,
 879                                    greqs.gsr_interface);
 880                 break;
 881         }
 882         case MCAST_MSFILTER:
 883         {
 884                 struct sockaddr_in *psin;
 885                 struct ip_msfilter *msf = NULL;
 886                 struct group_filter *gsf = NULL;
 887                 int msize, i, ifindex;
 888
 889                 if (optlen < GROUP_FILTER_SIZE(0))
 890                         goto e_inval;
 891                 if (optlen > sysctl_optmem_max) {
 892                         err = -ENOBUFS;
 893                         break;
 894                 }
 895                 gsf = kmalloc(optlen, GFP_KERNEL);
 896                 if (!gsf) {
 897                         err = -ENOBUFS;
 898                         break;
 899                 }
 900                 err = -EFAULT;
 901                 if (copy_from_user(gsf, optval, optlen))
 902                         goto mc_msf_out;
 903
 904                 /* numsrc >= (4G-140)/128 overflow in 32 bits */
 905                 if (gsf->gf_numsrc >= 0x1ffffff ||
 906                     gsf->gf_numsrc > sysctl_igmp_max_msf) {
 907                         err = -ENOBUFS;
 908                         goto mc_msf_out;
 909                 }
 910                 if (GROUP_FILTER_SIZE(gsf->gf_numsrc) > optlen) {
 911                         err = -EINVAL;
 912                         goto mc_msf_out;
 913                 }
 914                 msize = IP_MSFILTER_SIZE(gsf->gf_numsrc);
 915                 msf = kmalloc(msize, GFP_KERNEL);
 916                 if (!msf) {
 917                         err = -ENOBUFS;
 918                         goto mc_msf_out;
 919                 }
 920                 ifindex = gsf->gf_interface;
 921                 psin = (struct sockaddr_in *)&gsf->gf_group;
 922                 if (psin->sin_family != AF_INET) {
 923                         err = -EADDRNOTAVAIL;
 924                         goto mc_msf_out;
 925                 }
 926                 msf->imsf_multiaddr = psin->sin_addr.s_addr;
 927                 msf->imsf_interface = 0;
 928                 msf->imsf_fmode = gsf->gf_fmode;
 929                 msf->imsf_numsrc = gsf->gf_numsrc;
 930                 err = -EADDRNOTAVAIL;
 931                 for (i = 0; i < gsf->gf_numsrc; ++i) {
 932                         psin = (struct sockaddr_in *)&gsf->gf_slist[i];
 933
 934                         if (psin->sin_family != AF_INET)
 935                                 goto mc_msf_out;
 936                         msf->imsf_slist[i] = psin->sin_addr.s_addr;
 937                 }
 938                 kfree(gsf);
 939                 gsf = NULL;
 940
 941                 err = ip_mc_msfilter(sk, msf, ifindex);
 942 mc_msf_out:
 943                 kfree(msf);
 944                 kfree(gsf);
 945                 break;
 946         }
 947         case IP_MULTICAST_ALL:
 948                 if (optlen < 1)
 949                         goto e_inval;
 950                 if (val != 0 && val != 1)
 951                         goto e_inval;
 952                 inet->mc_all = val;
 953                 break;
 954         case IP_ROUTER_ALERT:
 955                 err = ip_ra_control(sk, val ? 1 : 0, NULL);
 956                 break;
 957
 958         case IP_FREEBIND:
 959                 if (optlen < 1)
 960                         goto e_inval;
 961                 inet->freebind = !!val;
 962                 break;
 963
 964         case IP_IPSEC_POLICY:
 965         case IP_XFRM_POLICY:
 966                 err = -EPERM;
 967                 if (!capable(CAP_NET_ADMIN))
 968                         break;
 969                 err = xfrm_user_policy(sk, optname, optval, optlen);
 970                 break;
 971
 972         case IP_TRANSPARENT:
 973                 if (!!val && !capable(CAP_NET_RAW) && !capable(CAP_NET_ADMIN)) {
 974                         err = -EPERM;
 975                         break;
 976                 }
 977                 if (optlen < 1)
 978                         goto e_inval;
 979                 inet->transparent = !!val;
 980                 break;
 981
 982         case IP_MINTTL:
 983                 if (optlen < 1)
 984                         goto e_inval;
 985                 if (val < 0 || val > 255)
 986                         goto e_inval;
 987                 inet->min_ttl = val;
 988                 break;
 989
 990         default:
 991                 err = -ENOPROTOOPT;
 992                 break;
 993         }
 994         release_sock(sk);
 995         return err;
 996
 997 e_inval:
 998         release_sock(sk);
 999         return -EINVAL;
1000 }
1001
1002 /**
1003  * ip_queue_rcv_skb - Queue an skb into sock receive queue
1004  * @sk: socket
1005  * @skb: buffer
1006  *
1007  * Queues an skb into socket receive queue. If IP_CMSG_PKTINFO option
1008  * is not set, we drop skb dst entry now, while dst cache line is hot.
1009  */
1010 int ip_queue_rcv_skb(struct sock *sk, struct sk_buff *skb)
1011 {
1012         if (!(inet_sk(sk)->cmsg_flags & IP_CMSG_PKTINFO) &&
1013             !IPCB(skb)->opt.optlen)
1014                 skb_dst_drop(skb);
1015         return sock_queue_rcv_skb(sk, skb);
1016 }
1017 EXPORT_SYMBOL(ip_queue_rcv_skb);
1018
1019 int ip_setsockopt(struct sock *sk, int level,
1020                 int optname, char __user *optval, unsigned int optlen)
1021 {
1022         int err;
1023
1024         if (level != SOL_IP)
1025                 return -ENOPROTOOPT;
1026
1027         err = do_ip_setsockopt(sk, level, optname, optval, optlen);
1028 #ifdef CONFIG_NETFILTER
1029         /* we need to exclude all possible ENOPROTOOPTs except default case */
1030         if (err == -ENOPROTOOPT && optname != IP_HDRINCL &&
1031                         optname != IP_IPSEC_POLICY &&
1032                         optname != IP_XFRM_POLICY &&
1033                         !ip_mroute_opt(optname))
1034                 err = nf_setsockopt(sk, PF_INET, optname, optval, optlen);
1035 #endif
1036         return err;
1037 }
1038 EXPORT_SYMBOL(ip_setsockopt);
1039
1040 #ifdef CONFIG_COMPAT
1041 int compat_ip_setsockopt(struct sock *sk, int level, int optname,
1042                          char __user *optval, unsigned int optlen)
1043 {
1044         int err;
1045
1046         if (level != SOL_IP)
1047                 return -ENOPROTOOPT;
1048
1049         if (optname >= MCAST_JOIN_GROUP && optname <= MCAST_MSFILTER)
1050                 return compat_mc_setsockopt(sk, level, optname, optval, optlen,
1051                         ip_setsockopt);
1052
1053         err = do_ip_setsockopt(sk, level, optname, optval, optlen);
1054 #ifdef CONFIG_NETFILTER
1055         /* we need to exclude all possible ENOPROTOOPTs except default case */
1056         if (err == -ENOPROTOOPT && optname != IP_HDRINCL &&
1057                         optname != IP_IPSEC_POLICY &&
1058                         optname != IP_XFRM_POLICY &&
1059                         !ip_mroute_opt(optname))
1060                 err = compat_nf_setsockopt(sk, PF_INET, optname, optval,
1061                                            optlen);
1062 #endif
1063         return err;
1064 }
1065 EXPORT_SYMBOL(compat_ip_setsockopt);
1066 #endif
1067
1068 /*
1069  *      Get the options. Note for future reference. The GET of IP options gets
1070  *      the _received_ ones. The set sets the _sent_ ones.
1071  */
1072
1073 static int do_ip_getsockopt(struct sock *sk, int level, int optname,
1074                             char __user *optval, int __user *optlen, unsigned flags)
1075 {
1076         struct inet_sock *inet = inet_sk(sk);
1077         int val;
1078         int len;
1079
1080         if (level != SOL_IP)
1081                 return -EOPNOTSUPP;
1082
1083         if (ip_mroute_opt(optname))
1084                 return ip_mroute_getsockopt(sk, optname, optval, optlen);
1085
1086         if (get_user(len, optlen))
1087                 return -EFAULT;
1088         if (len < 0)
1089                 return -EINVAL;
1090
1091         lock_sock(sk);
1092
1093         switch (optname) {
1094         case IP_OPTIONS:
1095         {
1096                 unsigned char optbuf[sizeof(struct ip_options)+40];
1097                 struct ip_options *opt = (struct ip_options *)optbuf;
1098                 struct ip_options_rcu *inet_opt;
1099
1100                 inet_opt = rcu_dereference_protected(inet->inet_opt,
1101                                                      sock_owned_by_user(sk));
1102                 opt->optlen = 0;
1103                 if (inet_opt)
1104                         memcpy(optbuf, &inet_opt->opt,
1105                                sizeof(struct ip_options) +
1106                                inet_opt->opt.optlen);
1107                 release_sock(sk);
1108
1109                 if (opt->optlen == 0)
1110                         return put_user(0, optlen);
1111
1112                 ip_options_undo(opt);
1113
1114                 len = min_t(unsigned int, len, opt->optlen);
1115                 if (put_user(len, optlen))
1116                         return -EFAULT;
1117                 if (copy_to_user(optval, opt->__data, len))
1118                         return -EFAULT;
1119                 return 0;
1120         }
1121         case IP_PKTINFO:
1122                 val = (inet->cmsg_flags & IP_CMSG_PKTINFO) != 0;
1123                 break;
1124         case IP_RECVTTL:
1125                 val = (inet->cmsg_flags & IP_CMSG_TTL) != 0;
1126                 break;
1127         case IP_RECVTOS:
1128                 val = (inet->cmsg_flags & IP_CMSG_TOS) != 0;
1129                 break;
1130         case IP_RECVOPTS:
1131                 val = (inet->cmsg_flags & IP_CMSG_RECVOPTS) != 0;
1132                 break;
1133         case IP_RETOPTS:
1134                 val = (inet->cmsg_flags & IP_CMSG_RETOPTS) != 0;
1135                 break;
1136         case IP_PASSSEC:
1137                 val = (inet->cmsg_flags & IP_CMSG_PASSSEC) != 0;
1138                 break;
1139         case IP_RECVORIGDSTADDR:
1140                 val = (inet->cmsg_flags & IP_CMSG_ORIGDSTADDR) != 0;
1141                 break;
1142         case IP_TOS:
1143                 val = inet->tos;
1144                 break;
1145         case IP_TTL:
1146                 val = (inet->uc_ttl == -1 ?
1147                        sysctl_ip_default_ttl :
1148                        inet->uc_ttl);
1149                 break;
1150         case IP_HDRINCL:
1151                 val = inet->hdrincl;
1152                 break;
1153         case IP_NODEFRAG:
1154                 val = inet->nodefrag;
1155                 break;
1156         case IP_MTU_DISCOVER:
1157                 val = inet->pmtudisc;
1158                 break;
1159         case IP_MTU:
1160         {
1161                 struct dst_entry *dst;
1162                 val = 0;
1163                 dst = sk_dst_get(sk);
1164                 if (dst) {
1165                         val = dst_mtu(dst);
1166                         dst_release(dst);
1167                 }
1168                 if (!val) {
1169                         release_sock(sk);
1170                         return -ENOTCONN;
1171                 }
1172                 break;
1173         }
1174         case IP_RECVERR:
1175                 val = inet->recverr;
1176                 break;
1177         case IP_MULTICAST_TTL:
1178                 val = inet->mc_ttl;
1179                 break;
1180         case IP_MULTICAST_LOOP:
1181                 val = inet->mc_loop;
1182                 break;
1183         case IP_MULTICAST_IF:
1184         {
1185                 struct in_addr addr;
1186                 len = min_t(unsigned int, len, sizeof(struct in_addr));
1187                 addr.s_addr = inet->mc_addr;
1188                 release_sock(sk);
1189
1190                 if (put_user(len, optlen))
1191                         return -EFAULT;
1192                 if (copy_to_user(optval, &addr, len))
1193                         return -EFAULT;
1194                 return 0;
1195         }
1196         case IP_MSFILTER:
1197         {
1198                 struct ip_msfilter msf;
1199                 int err;
1200
1201                 if (len < IP_MSFILTER_SIZE(0)) {
1202                         release_sock(sk);
1203                         return -EINVAL;
1204                 }
1205                 if (copy_from_user(&msf, optval, IP_MSFILTER_SIZE(0))) {
1206                         release_sock(sk);
1207                         return -EFAULT;
1208                 }
1209                 err = ip_mc_msfget(sk, &msf,
1210                                    (struct ip_msfilter __user *)optval, optlen);
1211                 release_sock(sk);
1212                 return err;
1213         }
1214         case MCAST_MSFILTER:
1215         {
1216                 struct group_filter gsf;
1217                 int err;
1218
1219                 if (len < GROUP_FILTER_SIZE(0)) {
1220                         release_sock(sk);
1221                         return -EINVAL;
1222                 }
1223                 if (copy_from_user(&gsf, optval, GROUP_FILTER_SIZE(0))) {
1224                         release_sock(sk);
1225                         return -EFAULT;
1226                 }
1227                 err = ip_mc_gsfget(sk, &gsf,
1228                                    (struct group_filter __user *)optval,
1229                                    optlen);
1230                 release_sock(sk);
1231                 return err;
1232         }
1233         case IP_MULTICAST_ALL:
1234                 val = inet->mc_all;
1235                 break;
1236         case IP_PKTOPTIONS:
1237         {
1238                 struct msghdr msg;
1239
1240                 release_sock(sk);
1241
1242                 if (sk->sk_type != SOCK_STREAM)
1243                         return -ENOPROTOOPT;
1244
1245                 msg.msg_control = optval;
1246                 msg.msg_controllen = len;
1247                 msg.msg_flags = flags;
1248
1249                 if (inet->cmsg_flags & IP_CMSG_PKTINFO) {
1250                         struct in_pktinfo info;
1251
1252                         info.ipi_addr.s_addr = inet->inet_rcv_saddr;
1253                         info.ipi_spec_dst.s_addr = inet->inet_rcv_saddr;
1254                         info.ipi_ifindex = inet->mc_index;
1255                         put_cmsg(&msg, SOL_IP, IP_PKTINFO, sizeof(info), &info);
1256                 }
1257                 if (inet->cmsg_flags & IP_CMSG_TTL) {
1258                         int hlim = inet->mc_ttl;
1259                         put_cmsg(&msg, SOL_IP, IP_TTL, sizeof(hlim), &hlim);
1260                 }
1261                 len -= msg.msg_controllen;
1262                 return put_user(len, optlen);
1263         }
1264         case IP_FREEBIND:
1265                 val = inet->freebind;
1266                 break;
1267         case IP_TRANSPARENT:
1268                 val = inet->transparent;
1269                 break;
1270         case IP_MINTTL:
1271                 val = inet->min_ttl;
1272                 break;
1273         default:
1274                 release_sock(sk);
1275                 return -ENOPROTOOPT;
1276         }
1277         release_sock(sk);
1278
1279         if (len < sizeof(int) && len > 0 && val >= 0 && val <= 255) {
1280                 unsigned char ucval = (unsigned char)val;
1281                 len = 1;
1282                 if (put_user(len, optlen))
1283                         return -EFAULT;
1284                 if (copy_to_user(optval, &ucval, 1))
1285                         return -EFAULT;
1286         } else {
1287                 len = min_t(unsigned int, sizeof(int), len);
1288                 if (put_user(len, optlen))
1289                         return -EFAULT;
1290                 if (copy_to_user(optval, &val, len))
1291                         return -EFAULT;
1292         }
1293         return 0;
1294 }
1295
1296 int ip_getsockopt(struct sock *sk, int level,
1297                   int optname, char __user *optval, int __user *optlen)
1298 {
1299         int err;
1300
1301         err = do_ip_getsockopt(sk, level, optname, optval, optlen, 0);
1302 #ifdef CONFIG_NETFILTER
1303         /* we need to exclude all possible ENOPROTOOPTs except default case */
1304         if (err == -ENOPROTOOPT && optname != IP_PKTOPTIONS &&
1305                         !ip_mroute_opt(optname)) {
1306                 int len;
1307
1308                 if (get_user(len, optlen))
1309                         return -EFAULT;
1310
1311                 err = nf_getsockopt(sk, PF_INET, optname, optval, &len);
1312                 if (err >= 0)
1313                         err = put_user(len, optlen);
1314                 return err;
1315         }
1316 #endif
1317         return err;
1318 }
1319 EXPORT_SYMBOL(ip_getsockopt);
1320
1321 #ifdef CONFIG_COMPAT
1322 int compat_ip_getsockopt(struct sock *sk, int level, int optname,
1323                          char __user *optval, int __user *optlen)
1324 {
1325         int err;
1326
1327         if (optname == MCAST_MSFILTER)
1328                 return compat_mc_getsockopt(sk, level, optname, optval, optlen,
1329                         ip_getsockopt);
1330
1331         err = do_ip_getsockopt(sk, level, optname, optval, optlen,
1332                 MSG_CMSG_COMPAT);
1333
1334 #ifdef CONFIG_NETFILTER
1335         /* we need to exclude all possible ENOPROTOOPTs except default case */
1336         if (err == -ENOPROTOOPT && optname != IP_PKTOPTIONS &&
1337                         !ip_mroute_opt(optname)) {
1338                 int len;
1339
1340                 if (get_user(len, optlen))
1341                         return -EFAULT;
1342
1343                 err = compat_nf_getsockopt(sk, PF_INET, optname, optval, &len);
1344                 if (err >= 0)
1345                         err = put_user(len, optlen);
1346                 return err;
1347         }
1348 #endif
1349         return err;
1350 }
1351 EXPORT_SYMBOL(compat_ip_getsockopt);
1352 #endif