net/llc/af_llc.c

   1 /*
   2  * af_llc.c - LLC User Interface SAPs
   3  * Description:
   4  *   Functions in this module are implementation of socket based llc
   5  *   communications for the Linux operating system. Support of llc class
   6  *   one and class two is provided via SOCK_DGRAM and SOCK_STREAM
   7  *   respectively.
   8  *
   9  *   An llc2 connection is (mac + sap), only one llc2 sap connection
  10  *   is allowed per mac. Though one sap may have multiple mac + sap
  11  *   connections.
  12  *
  13  * Copyright (c) 2001 by Jay Schulist <[email protected]>
  14  *               2002-2003 by Arnaldo Carvalho de Melo <[email protected]>
  15  *
  16  * This program can be redistributed or modified under the terms of the
  17  * GNU General Public License as published by the Free Software Foundation.
  18  * This program is distributed without any warranty or implied warranty
  19  * of merchantability or fitness for a particular purpose.
  20  *
  21  * See the GNU General Public License for more details.
  22  */
  23 #include <linux/compiler.h>
  24 #include <linux/kernel.h>
  25 #include <linux/module.h>
  26 #include <linux/rtnetlink.h>
  27 #include <linux/init.h>
  28 #include <linux/slab.h>
  29 #include <linux/sched/signal.h>
  30
  31 #include <net/llc.h>
  32 #include <net/llc_sap.h>
  33 #include <net/llc_pdu.h>
  34 #include <net/llc_conn.h>
  35 #include <net/tcp_states.h>
  36
  37 /* remember: uninitialized global data is zeroed because its in .bss */
  38 static u16 llc_ui_sap_last_autoport = LLC_SAP_DYN_START;
  39 static u16 llc_ui_sap_link_no_max[256];
  40 static struct sockaddr_llc llc_ui_addrnull;
  41 static const struct proto_ops llc_ui_ops;
  42
  43 static bool llc_ui_wait_for_conn(struct sock *sk, long timeout);
  44 static int llc_ui_wait_for_disc(struct sock *sk, long timeout);
  45 static int llc_ui_wait_for_busy_core(struct sock *sk, long timeout);
  46
  47 #if 0
  48 #define dprintk(args...) printk(KERN_DEBUG args)
  49 #else
  50 #define dprintk(args...)
  51 #endif
  52
  53 /* Maybe we'll add some more in the future. */
  54 #define LLC_CMSG_PKTINFO        1
  55
  56
  57 /**
  58  *      llc_ui_next_link_no - return the next unused link number for a sap
  59  *      @sap: Address of sap to get link number from.
  60  *
  61  *      Return the next unused link number for a given sap.
  62  */
  63 static inline u16 llc_ui_next_link_no(int sap)
  64 {
  65         return llc_ui_sap_link_no_max[sap]++;
  66 }
  67
  68 /**
  69  *      llc_proto_type - return eth protocol for ARP header type
  70  *      @arphrd: ARP header type.
  71  *
  72  *      Given an ARP header type return the corresponding ethernet protocol.
  73  */
  74 static inline __be16 llc_proto_type(u16 arphrd)
  75 {
  76         return htons(ETH_P_802_2);
  77 }
  78
  79 /**
  80  *      llc_ui_addr_null - determines if a address structure is null
  81  *      @addr: Address to test if null.
  82  */
  83 static inline u8 llc_ui_addr_null(struct sockaddr_llc *addr)
  84 {
  85         return !memcmp(addr, &llc_ui_addrnull, sizeof(*addr));
  86 }
  87
  88 /**
  89  *      llc_ui_header_len - return length of llc header based on operation
  90  *      @sk: Socket which contains a valid llc socket type.
  91  *      @addr: Complete sockaddr_llc structure received from the user.
  92  *
  93  *      Provide the length of the llc header depending on what kind of
  94  *      operation the user would like to perform and the type of socket.
  95  *      Returns the correct llc header length.
  96  */
  97 static inline u8 llc_ui_header_len(struct sock *sk, struct sockaddr_llc *addr)
  98 {
  99         u8 rc = LLC_PDU_LEN_U;
 100
 101         if (addr->sllc_test || addr->sllc_xid)
 102                 rc = LLC_PDU_LEN_U;
 103         else if (sk->sk_type == SOCK_STREAM)
 104                 rc = LLC_PDU_LEN_I;
 105         return rc;
 106 }
 107
 108 /**
 109  *      llc_ui_send_data - send data via reliable llc2 connection
 110  *      @sk: Connection the socket is using.
 111  *      @skb: Data the user wishes to send.
 112  *      @noblock: can we block waiting for data?
 113  *
 114  *      Send data via reliable llc2 connection.
 115  *      Returns 0 upon success, non-zero if action did not succeed.
 116  */
 117 static int llc_ui_send_data(struct sock* sk, struct sk_buff *skb, int noblock)
 118 {
 119         struct llc_sock* llc = llc_sk(sk);
 120         int rc = 0;
 121
 122         if (unlikely(llc_data_accept_state(llc->state) ||
 123                      llc->remote_busy_flag ||
 124                      llc->p_flag)) {
 125                 long timeout = sock_sndtimeo(sk, noblock);
 126
 127                 rc = llc_ui_wait_for_busy_core(sk, timeout);
 128         }
 129         if (unlikely(!rc))
 130                 rc = llc_build_and_send_pkt(sk, skb);
 131         return rc;
 132 }
 133
 134 static void llc_ui_sk_init(struct socket *sock, struct sock *sk)
 135 {
 136         sock_graft(sk, sock);
 137         sk->sk_type     = sock->type;
 138         sock->ops       = &llc_ui_ops;
 139 }
 140
 141 static struct proto llc_proto = {
 142         .name     = "LLC",
 143         .owner    = THIS_MODULE,
 144         .obj_size = sizeof(struct llc_sock),
 145         .slab_flags = SLAB_TYPESAFE_BY_RCU,
 146 };
 147
 148 /**
 149  *      llc_ui_create - alloc and init a new llc_ui socket
 150  *      @net: network namespace (must be default network)
 151  *      @sock: Socket to initialize and attach allocated sk to.
 152  *      @protocol: Unused.
 153  *      @kern: on behalf of kernel or userspace
 154  *
 155  *      Allocate and initialize a new llc_ui socket, validate the user wants a
 156  *      socket type we have available.
 157  *      Returns 0 upon success, negative upon failure.
 158  */
 159 static int llc_ui_create(struct net *net, struct socket *sock, int protocol,
 160                          int kern)
 161 {
 162         struct sock *sk;
 163         int rc = -ESOCKTNOSUPPORT;
 164
 165         if (!ns_capable(net->user_ns, CAP_NET_RAW))
 166                 return -EPERM;
 167
 168         if (!net_eq(net, &init_net))
 169                 return -EAFNOSUPPORT;
 170
 171         if (likely(sock->type == SOCK_DGRAM || sock->type == SOCK_STREAM)) {
 172                 rc = -ENOMEM;
 173                 sk = llc_sk_alloc(net, PF_LLC, GFP_KERNEL, &llc_proto, kern);
 174                 if (sk) {
 175                         rc = 0;
 176                         llc_ui_sk_init(sock, sk);
 177                 }
 178         }
 179         return rc;
 180 }
 181
 182 /**
 183  *      llc_ui_release - shutdown socket
 184  *      @sock: Socket to release.
 185  *
 186  *      Shutdown and deallocate an existing socket.
 187  */
 188 static int llc_ui_release(struct socket *sock)
 189 {
 190         struct sock *sk = sock->sk;
 191         struct llc_sock *llc;
 192
 193         if (unlikely(sk == NULL))
 194                 goto out;
 195         sock_hold(sk);
 196         lock_sock(sk);
 197         llc = llc_sk(sk);
 198         dprintk("%s: closing local(%02X) remote(%02X)\n", __func__,
 199                 llc->laddr.lsap, llc->daddr.lsap);
 200         if (!llc_send_disc(sk))
 201                 llc_ui_wait_for_disc(sk, sk->sk_rcvtimeo);
 202         if (!sock_flag(sk, SOCK_ZAPPED)) {
 203                 struct llc_sap *sap = llc->sap;
 204
 205                 /* Hold this for release_sock(), so that llc_backlog_rcv()
 206                  * could still use it.
 207                  */
 208                 llc_sap_hold(sap);
 209                 llc_sap_remove_socket(llc->sap, sk);
 210                 release_sock(sk);
 211                 llc_sap_put(sap);
 212         } else {
 213                 release_sock(sk);
 214         }
 215         if (llc->dev)
 216                 dev_put(llc->dev);
 217         sock_put(sk);
 218         llc_sk_free(sk);
 219 out:
 220         return 0;
 221 }
 222
 223 /**
 224  *      llc_ui_autoport - provide dynamically allocate SAP number
 225  *
 226  *      Provide the caller with a dynamically allocated SAP number according
 227  *      to the rules that are set in this function. Returns: 0, upon failure,
 228  *      SAP number otherwise.
 229  */
 230 static int llc_ui_autoport(void)
 231 {
 232         struct llc_sap *sap;
 233         int i, tries = 0;
 234
 235         while (tries < LLC_SAP_DYN_TRIES) {
 236                 for (i = llc_ui_sap_last_autoport;
 237                      i < LLC_SAP_DYN_STOP; i += 2) {
 238                         sap = llc_sap_find(i);
 239                         if (!sap) {
 240                                 llc_ui_sap_last_autoport = i + 2;
 241                                 goto out;
 242                         }
 243                         llc_sap_put(sap);
 244                 }
 245                 llc_ui_sap_last_autoport = LLC_SAP_DYN_START;
 246                 tries++;
 247         }
 248         i = 0;
 249 out:
 250         return i;
 251 }
 252
 253 /**
 254  *      llc_ui_autobind - automatically bind a socket to a sap
 255  *      @sock: socket to bind
 256  *      @addr: address to connect to
 257  *
 258  *      Used by llc_ui_connect and llc_ui_sendmsg when the user hasn't
 259  *      specifically used llc_ui_bind to bind to an specific address/sap
 260  *
 261  *      Returns: 0 upon success, negative otherwise.
 262  */
 263 static int llc_ui_autobind(struct socket *sock, struct sockaddr_llc *addr)
 264 {
 265         struct sock *sk = sock->sk;
 266         struct llc_sock *llc = llc_sk(sk);
 267         struct llc_sap *sap;
 268         int rc = -EINVAL;
 269
 270         if (!sock_flag(sk, SOCK_ZAPPED))
 271                 goto out;
 272         rc = -ENODEV;
 273         if (sk->sk_bound_dev_if) {
 274                 llc->dev = dev_get_by_index(&init_net, sk->sk_bound_dev_if);
 275                 if (llc->dev && addr->sllc_arphrd != llc->dev->type) {
 276                         dev_put(llc->dev);
 277                         llc->dev = NULL;
 278                 }
 279         } else
 280                 llc->dev = dev_getfirstbyhwtype(&init_net, addr->sllc_arphrd);
 281         if (!llc->dev)
 282                 goto out;
 283         rc = -EUSERS;
 284         llc->laddr.lsap = llc_ui_autoport();
 285         if (!llc->laddr.lsap)
 286                 goto out;
 287         rc = -EBUSY; /* some other network layer is using the sap */
 288         sap = llc_sap_open(llc->laddr.lsap, NULL);
 289         if (!sap)
 290                 goto out;
 291         memcpy(llc->laddr.mac, llc->dev->dev_addr, IFHWADDRLEN);
 292         memcpy(&llc->addr, addr, sizeof(llc->addr));
 293         /* assign new connection to its SAP */
 294         llc_sap_add_socket(sap, sk);
 295         sock_reset_flag(sk, SOCK_ZAPPED);
 296         rc = 0;
 297 out:
 298         return rc;
 299 }
 300
 301 /**
 302  *      llc_ui_bind - bind a socket to a specific address.
 303  *      @sock: Socket to bind an address to.
 304  *      @uaddr: Address the user wants the socket bound to.
 305  *      @addrlen: Length of the uaddr structure.
 306  *
 307  *      Bind a socket to a specific address. For llc a user is able to bind to
 308  *      a specific sap only or mac + sap.
 309  *      If the user desires to bind to a specific mac + sap, it is possible to
 310  *      have multiple sap connections via multiple macs.
 311  *      Bind and autobind for that matter must enforce the correct sap usage
 312  *      otherwise all hell will break loose.
 313  *      Returns: 0 upon success, negative otherwise.
 314  */
 315 static int llc_ui_bind(struct socket *sock, struct sockaddr *uaddr, int addrlen)
 316 {
 317         struct sockaddr_llc *addr = (struct sockaddr_llc *)uaddr;
 318         struct sock *sk = sock->sk;
 319         struct llc_sock *llc = llc_sk(sk);
 320         struct llc_sap *sap;
 321         int rc = -EINVAL;
 322
 323         lock_sock(sk);
 324         if (unlikely(!sock_flag(sk, SOCK_ZAPPED) || addrlen != sizeof(*addr)))
 325                 goto out;
 326         rc = -EAFNOSUPPORT;
 327         if (unlikely(addr->sllc_family != AF_LLC))
 328                 goto out;
 329         dprintk("%s: binding %02X\n", __func__, addr->sllc_sap);
 330         rc = -ENODEV;
 331         rcu_read_lock();
 332         if (sk->sk_bound_dev_if) {
 333                 llc->dev = dev_get_by_index_rcu(&init_net, sk->sk_bound_dev_if);
 334                 if (llc->dev) {
 335                         if (!addr->sllc_arphrd)
 336                                 addr->sllc_arphrd = llc->dev->type;
 337                         if (is_zero_ether_addr(addr->sllc_mac))
 338                                 memcpy(addr->sllc_mac, llc->dev->dev_addr,
 339                                        IFHWADDRLEN);
 340                         if (addr->sllc_arphrd != llc->dev->type ||
 341                             !ether_addr_equal(addr->sllc_mac,
 342                                               llc->dev->dev_addr)) {
 343                                 rc = -EINVAL;
 344                                 llc->dev = NULL;
 345                         }
 346                 }
 347         } else
 348                 llc->dev = dev_getbyhwaddr_rcu(&init_net, addr->sllc_arphrd,
 349                                            addr->sllc_mac);
 350         if (llc->dev)
 351                 dev_hold(llc->dev);
 352         rcu_read_unlock();
 353         if (!llc->dev)
 354                 goto out;
 355         if (!addr->sllc_sap) {
 356                 rc = -EUSERS;
 357                 addr->sllc_sap = llc_ui_autoport();
 358                 if (!addr->sllc_sap)
 359                         goto out;
 360         }
 361         sap = llc_sap_find(addr->sllc_sap);
 362         if (!sap) {
 363                 sap = llc_sap_open(addr->sllc_sap, NULL);
 364                 rc = -EBUSY; /* some other network layer is using the sap */
 365                 if (!sap)
 366                         goto out;
 367         } else {
 368                 struct llc_addr laddr, daddr;
 369                 struct sock *ask;
 370
 371                 memset(&laddr, 0, sizeof(laddr));
 372                 memset(&daddr, 0, sizeof(daddr));
 373                 /*
 374                  * FIXME: check if the address is multicast,
 375                  *        only SOCK_DGRAM can do this.
 376                  */
 377                 memcpy(laddr.mac, addr->sllc_mac, IFHWADDRLEN);
 378                 laddr.lsap = addr->sllc_sap;
 379                 rc = -EADDRINUSE; /* mac + sap clash. */
 380                 ask = llc_lookup_established(sap, &daddr, &laddr);
 381                 if (ask) {
 382                         sock_put(ask);
 383                         goto out_put;
 384                 }
 385         }
 386         llc->laddr.lsap = addr->sllc_sap;
 387         memcpy(llc->laddr.mac, addr->sllc_mac, IFHWADDRLEN);
 388         memcpy(&llc->addr, addr, sizeof(llc->addr));
 389         /* assign new connection to its SAP */
 390         llc_sap_add_socket(sap, sk);
 391         sock_reset_flag(sk, SOCK_ZAPPED);
 392         rc = 0;
 393 out_put:
 394         llc_sap_put(sap);
 395 out:
 396         release_sock(sk);
 397         return rc;
 398 }
 399
 400 /**
 401  *      llc_ui_shutdown - shutdown a connect llc2 socket.
 402  *      @sock: Socket to shutdown.
 403  *      @how: What part of the socket to shutdown.
 404  *
 405  *      Shutdown a connected llc2 socket. Currently this function only supports
 406  *      shutting down both sends and receives (2), we could probably make this
 407  *      function such that a user can shutdown only half the connection but not
 408  *      right now.
 409  *      Returns: 0 upon success, negative otherwise.
 410  */
 411 static int llc_ui_shutdown(struct socket *sock, int how)
 412 {
 413         struct sock *sk = sock->sk;
 414         int rc = -ENOTCONN;
 415
 416         lock_sock(sk);
 417         if (unlikely(sk->sk_state != TCP_ESTABLISHED))
 418                 goto out;
 419         rc = -EINVAL;
 420         if (how != 2)
 421                 goto out;
 422         rc = llc_send_disc(sk);
 423         if (!rc)
 424                 rc = llc_ui_wait_for_disc(sk, sk->sk_rcvtimeo);
 425         /* Wake up anyone sleeping in poll */
 426         sk->sk_state_change(sk);
 427 out:
 428         release_sock(sk);
 429         return rc;
 430 }
 431
 432 /**
 433  *      llc_ui_connect - Connect to a remote llc2 mac + sap.
 434  *      @sock: Socket which will be connected to the remote destination.
 435  *      @uaddr: Remote and possibly the local address of the new connection.
 436  *      @addrlen: Size of uaddr structure.
 437  *      @flags: Operational flags specified by the user.
 438  *
 439  *      Connect to a remote llc2 mac + sap. The caller must specify the
 440  *      destination mac and address to connect to. If the user hasn't previously
 441  *      called bind(2) with a smac the address of the first interface of the
 442  *      specified arp type will be used.
 443  *      This function will autobind if user did not previously call bind.
 444  *      Returns: 0 upon success, negative otherwise.
 445  */
 446 static int llc_ui_connect(struct socket *sock, struct sockaddr *uaddr,
 447                           int addrlen, int flags)
 448 {
 449         struct sock *sk = sock->sk;
 450         struct llc_sock *llc = llc_sk(sk);
 451         struct sockaddr_llc *addr = (struct sockaddr_llc *)uaddr;
 452         int rc = -EINVAL;
 453
 454         lock_sock(sk);
 455         if (unlikely(addrlen != sizeof(*addr)))
 456                 goto out;
 457         rc = -EAFNOSUPPORT;
 458         if (unlikely(addr->sllc_family != AF_LLC))
 459                 goto out;
 460         if (unlikely(sk->sk_type != SOCK_STREAM))
 461                 goto out;
 462         rc = -EALREADY;
 463         if (unlikely(sock->state == SS_CONNECTING))
 464                 goto out;
 465         /* bind connection to sap if user hasn't done it. */
 466         if (sock_flag(sk, SOCK_ZAPPED)) {
 467                 /* bind to sap with null dev, exclusive */
 468                 rc = llc_ui_autobind(sock, addr);
 469                 if (rc)
 470                         goto out;
 471         }
 472         llc->daddr.lsap = addr->sllc_sap;
 473         memcpy(llc->daddr.mac, addr->sllc_mac, IFHWADDRLEN);
 474         sock->state = SS_CONNECTING;
 475         sk->sk_state   = TCP_SYN_SENT;
 476         llc->link   = llc_ui_next_link_no(llc->sap->laddr.lsap);
 477         rc = llc_establish_connection(sk, llc->dev->dev_addr,
 478                                       addr->sllc_mac, addr->sllc_sap);
 479         if (rc) {
 480                 dprintk("%s: llc_ui_send_conn failed :-(\n", __func__);
 481                 sock->state  = SS_UNCONNECTED;
 482                 sk->sk_state = TCP_CLOSE;
 483                 goto out;
 484         }
 485
 486         if (sk->sk_state == TCP_SYN_SENT) {
 487                 const long timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
 488
 489                 if (!timeo || !llc_ui_wait_for_conn(sk, timeo))
 490                         goto out;
 491
 492                 rc = sock_intr_errno(timeo);
 493                 if (signal_pending(current))
 494                         goto out;
 495         }
 496
 497         if (sk->sk_state == TCP_CLOSE)
 498                 goto sock_error;
 499
 500         sock->state = SS_CONNECTED;
 501         rc = 0;
 502 out:
 503         release_sock(sk);
 504         return rc;
 505 sock_error:
 506         rc = sock_error(sk) ? : -ECONNABORTED;
 507         sock->state = SS_UNCONNECTED;
 508         goto out;
 509 }
 510
 511 /**
 512  *      llc_ui_listen - allow a normal socket to accept incoming connections
 513  *      @sock: Socket to allow incoming connections on.
 514  *      @backlog: Number of connections to queue.
 515  *
 516  *      Allow a normal socket to accept incoming connections.
 517  *      Returns 0 upon success, negative otherwise.
 518  */
 519 static int llc_ui_listen(struct socket *sock, int backlog)
 520 {
 521         struct sock *sk = sock->sk;
 522         int rc = -EINVAL;
 523
 524         lock_sock(sk);
 525         if (unlikely(sock->state != SS_UNCONNECTED))
 526                 goto out;
 527         rc = -EOPNOTSUPP;
 528         if (unlikely(sk->sk_type != SOCK_STREAM))
 529                 goto out;
 530         rc = -EAGAIN;
 531         if (sock_flag(sk, SOCK_ZAPPED))
 532                 goto out;
 533         rc = 0;
 534         if (!(unsigned int)backlog)     /* BSDism */
 535                 backlog = 1;
 536         sk->sk_max_ack_backlog = backlog;
 537         if (sk->sk_state != TCP_LISTEN) {
 538                 sk->sk_ack_backlog = 0;
 539                 sk->sk_state       = TCP_LISTEN;
 540         }
 541         sk->sk_socket->flags |= __SO_ACCEPTCON;
 542 out:
 543         release_sock(sk);
 544         return rc;
 545 }
 546
 547 static int llc_ui_wait_for_disc(struct sock *sk, long timeout)
 548 {
 549         DEFINE_WAIT_FUNC(wait, woken_wake_function);
 550         int rc = 0;
 551
 552         add_wait_queue(sk_sleep(sk), &wait);
 553         while (1) {
 554                 if (sk_wait_event(sk, &timeout, sk->sk_state == TCP_CLOSE, &wait))
 555                         break;
 556                 rc = -ERESTARTSYS;
 557                 if (signal_pending(current))
 558                         break;
 559                 rc = -EAGAIN;
 560                 if (!timeout)
 561                         break;
 562                 rc = 0;
 563         }
 564         remove_wait_queue(sk_sleep(sk), &wait);
 565         return rc;
 566 }
 567
 568 static bool llc_ui_wait_for_conn(struct sock *sk, long timeout)
 569 {
 570         DEFINE_WAIT_FUNC(wait, woken_wake_function);
 571
 572         add_wait_queue(sk_sleep(sk), &wait);
 573         while (1) {
 574                 if (sk_wait_event(sk, &timeout, sk->sk_state != TCP_SYN_SENT, &wait))
 575                         break;
 576                 if (signal_pending(current) || !timeout)
 577                         break;
 578         }
 579         remove_wait_queue(sk_sleep(sk), &wait);
 580         return timeout;
 581 }
 582
 583 static int llc_ui_wait_for_busy_core(struct sock *sk, long timeout)
 584 {
 585         DEFINE_WAIT_FUNC(wait, woken_wake_function);
 586         struct llc_sock *llc = llc_sk(sk);
 587         int rc;
 588
 589         add_wait_queue(sk_sleep(sk), &wait);
 590         while (1) {
 591                 rc = 0;
 592                 if (sk_wait_event(sk, &timeout,
 593                                   (sk->sk_shutdown & RCV_SHUTDOWN) ||
 594                                   (!llc_data_accept_state(llc->state) &&
 595                                    !llc->remote_busy_flag &&
 596                                    !llc->p_flag), &wait))
 597                         break;
 598                 rc = -ERESTARTSYS;
 599                 if (signal_pending(current))
 600                         break;
 601                 rc = -EAGAIN;
 602                 if (!timeout)
 603                         break;
 604         }
 605         remove_wait_queue(sk_sleep(sk), &wait);
 606         return rc;
 607 }
 608
 609 static int llc_wait_data(struct sock *sk, long timeo)
 610 {
 611         int rc;
 612
 613         while (1) {
 614                 /*
 615                  * POSIX 1003.1g mandates this order.
 616                  */
 617                 rc = sock_error(sk);
 618                 if (rc)
 619                         break;
 620                 rc = 0;
 621                 if (sk->sk_shutdown & RCV_SHUTDOWN)
 622                         break;
 623                 rc = -EAGAIN;
 624                 if (!timeo)
 625                         break;
 626                 rc = sock_intr_errno(timeo);
 627                 if (signal_pending(current))
 628                         break;
 629                 rc = 0;
 630                 if (sk_wait_data(sk, &timeo, NULL))
 631                         break;
 632         }
 633         return rc;
 634 }
 635
 636 static void llc_cmsg_rcv(struct msghdr *msg, struct sk_buff *skb)
 637 {
 638         struct llc_sock *llc = llc_sk(skb->sk);
 639
 640         if (llc->cmsg_flags & LLC_CMSG_PKTINFO) {
 641                 struct llc_pktinfo info;
 642
 643                 memset(&info, 0, sizeof(info));
 644                 info.lpi_ifindex = llc_sk(skb->sk)->dev->ifindex;
 645                 llc_pdu_decode_dsap(skb, &info.lpi_sap);
 646                 llc_pdu_decode_da(skb, info.lpi_mac);
 647                 put_cmsg(msg, SOL_LLC, LLC_OPT_PKTINFO, sizeof(info), &info);
 648         }
 649 }
 650
 651 /**
 652  *      llc_ui_accept - accept a new incoming connection.
 653  *      @sock: Socket which connections arrive on.
 654  *      @newsock: Socket to move incoming connection to.
 655  *      @flags: User specified operational flags.
 656  *      @kern: If the socket is kernel internal
 657  *
 658  *      Accept a new incoming connection.
 659  *      Returns 0 upon success, negative otherwise.
 660  */
 661 static int llc_ui_accept(struct socket *sock, struct socket *newsock, int flags,
 662                          bool kern)
 663 {
 664         struct sock *sk = sock->sk, *newsk;
 665         struct llc_sock *llc, *newllc;
 666         struct sk_buff *skb;
 667         int rc = -EOPNOTSUPP;
 668
 669         dprintk("%s: accepting on %02X\n", __func__,
 670                 llc_sk(sk)->laddr.lsap);
 671         lock_sock(sk);
 672         if (unlikely(sk->sk_type != SOCK_STREAM))
 673                 goto out;
 674         rc = -EINVAL;
 675         if (unlikely(sock->state != SS_UNCONNECTED ||
 676                      sk->sk_state != TCP_LISTEN))
 677                 goto out;
 678         /* wait for a connection to arrive. */
 679         if (skb_queue_empty(&sk->sk_receive_queue)) {
 680                 rc = llc_wait_data(sk, sk->sk_rcvtimeo);
 681                 if (rc)
 682                         goto out;
 683         }
 684         dprintk("%s: got a new connection on %02X\n", __func__,
 685                 llc_sk(sk)->laddr.lsap);
 686         skb = skb_dequeue(&sk->sk_receive_queue);
 687         rc = -EINVAL;
 688         if (!skb->sk)
 689                 goto frees;
 690         rc = 0;
 691         newsk = skb->sk;
 692         /* attach connection to a new socket. */
 693         llc_ui_sk_init(newsock, newsk);
 694         sock_reset_flag(newsk, SOCK_ZAPPED);
 695         newsk->sk_state         = TCP_ESTABLISHED;
 696         newsock->state          = SS_CONNECTED;
 697         llc                     = llc_sk(sk);
 698         newllc                  = llc_sk(newsk);
 699         memcpy(&newllc->addr, &llc->addr, sizeof(newllc->addr));
 700         newllc->link = llc_ui_next_link_no(newllc->laddr.lsap);
 701
 702         /* put original socket back into a clean listen state. */
 703         sk->sk_state = TCP_LISTEN;
 704         sk->sk_ack_backlog--;
 705         dprintk("%s: ok success on %02X, client on %02X\n", __func__,
 706                 llc_sk(sk)->addr.sllc_sap, newllc->daddr.lsap);
 707 frees:
 708         kfree_skb(skb);
 709 out:
 710         release_sock(sk);
 711         return rc;
 712 }
 713
 714 /**
 715  *      llc_ui_recvmsg - copy received data to the socket user.
 716  *      @sock: Socket to copy data from.
 717  *      @msg: Various user space related information.
 718  *      @len: Size of user buffer.
 719  *      @flags: User specified flags.
 720  *
 721  *      Copy received data to the socket user.
 722  *      Returns non-negative upon success, negative otherwise.
 723  */
 724 static int llc_ui_recvmsg(struct socket *sock, struct msghdr *msg, size_t len,
 725                           int flags)
 726 {
 727         DECLARE_SOCKADDR(struct sockaddr_llc *, uaddr, msg->msg_name);
 728         const int nonblock = flags & MSG_DONTWAIT;
 729         struct sk_buff *skb = NULL;
 730         struct sock *sk = sock->sk;
 731         struct llc_sock *llc = llc_sk(sk);
 732         size_t copied = 0;
 733         u32 peek_seq = 0;
 734         u32 *seq, skb_len;
 735         unsigned long used;
 736         int target;     /* Read at least this many bytes */
 737         long timeo;
 738
 739         lock_sock(sk);
 740         copied = -ENOTCONN;
 741         if (unlikely(sk->sk_type == SOCK_STREAM && sk->sk_state == TCP_LISTEN))
 742                 goto out;
 743
 744         timeo = sock_rcvtimeo(sk, nonblock);
 745
 746         seq = &llc->copied_seq;
 747         if (flags & MSG_PEEK) {
 748                 peek_seq = llc->copied_seq;
 749                 seq = &peek_seq;
 750         }
 751
 752         target = sock_rcvlowat(sk, flags & MSG_WAITALL, len);
 753         copied = 0;
 754
 755         do {
 756                 u32 offset;
 757
 758                 /*
 759                  * We need to check signals first, to get correct SIGURG
 760                  * handling. FIXME: Need to check this doesn't impact 1003.1g
 761                  * and move it down to the bottom of the loop
 762                  */
 763                 if (signal_pending(current)) {
 764                         if (copied)
 765                                 break;
 766                         copied = timeo ? sock_intr_errno(timeo) : -EAGAIN;
 767                         break;
 768                 }
 769
 770                 /* Next get a buffer. */
 771
 772                 skb = skb_peek(&sk->sk_receive_queue);
 773                 if (skb) {
 774                         offset = *seq;
 775                         goto found_ok_skb;
 776                 }
 777                 /* Well, if we have backlog, try to process it now yet. */
 778
 779                 if (copied >= target && !sk->sk_backlog.tail)
 780                         break;
 781
 782                 if (copied) {
 783                         if (sk->sk_err ||
 784                             sk->sk_state == TCP_CLOSE ||
 785                             (sk->sk_shutdown & RCV_SHUTDOWN) ||
 786                             !timeo ||
 787                             (flags & MSG_PEEK))
 788                                 break;
 789                 } else {
 790                         if (sock_flag(sk, SOCK_DONE))
 791                                 break;
 792
 793                         if (sk->sk_err) {
 794                                 copied = sock_error(sk);
 795                                 break;
 796                         }
 797                         if (sk->sk_shutdown & RCV_SHUTDOWN)
 798                                 break;
 799
 800                         if (sk->sk_type == SOCK_STREAM && sk->sk_state == TCP_CLOSE) {
 801                                 if (!sock_flag(sk, SOCK_DONE)) {
 802                                         /*
 803                                          * This occurs when user tries to read
 804                                          * from never connected socket.
 805                                          */
 806                                         copied = -ENOTCONN;
 807                                         break;
 808                                 }
 809                                 break;
 810                         }
 811                         if (!timeo) {
 812                                 copied = -EAGAIN;
 813                                 break;
 814                         }
 815                 }
 816
 817                 if (copied >= target) { /* Do not sleep, just process backlog. */
 818                         release_sock(sk);
 819                         lock_sock(sk);
 820                 } else
 821                         sk_wait_data(sk, &timeo, NULL);
 822
 823                 if ((flags & MSG_PEEK) && peek_seq != llc->copied_seq) {
 824                         net_dbg_ratelimited("LLC(%s:%d): Application bug, race in MSG_PEEK\n",
 825                                             current->comm,
 826                                             task_pid_nr(current));
 827                         peek_seq = llc->copied_seq;
 828                 }
 829                 continue;
 830         found_ok_skb:
 831                 skb_len = skb->len;
 832                 /* Ok so how much can we use? */
 833                 used = skb->len - offset;
 834                 if (len < used)
 835                         used = len;
 836
 837                 if (!(flags & MSG_TRUNC)) {
 838                         int rc = skb_copy_datagram_msg(skb, offset, msg, used);
 839                         if (rc) {
 840                                 /* Exception. Bailout! */
 841                                 if (!copied)
 842                                         copied = -EFAULT;
 843                                 break;
 844                         }
 845                 }
 846
 847                 *seq += used;
 848                 copied += used;
 849                 len -= used;
 850
 851                 /* For non stream protcols we get one packet per recvmsg call */
 852                 if (sk->sk_type != SOCK_STREAM)
 853                         goto copy_uaddr;
 854
 855                 if (!(flags & MSG_PEEK)) {
 856                         skb_unlink(skb, &sk->sk_receive_queue);
 857                         kfree_skb(skb);
 858                         *seq = 0;
 859                 }
 860
 861                 /* Partial read */
 862                 if (used + offset < skb_len)
 863                         continue;
 864         } while (len > 0);
 865
 866 out:
 867         release_sock(sk);
 868         return copied;
 869 copy_uaddr:
 870         if (uaddr != NULL && skb != NULL) {
 871                 memcpy(uaddr, llc_ui_skb_cb(skb), sizeof(*uaddr));
 872                 msg->msg_namelen = sizeof(*uaddr);
 873         }
 874         if (llc_sk(sk)->cmsg_flags)
 875                 llc_cmsg_rcv(msg, skb);
 876
 877         if (!(flags & MSG_PEEK)) {
 878                 skb_unlink(skb, &sk->sk_receive_queue);
 879                 kfree_skb(skb);
 880                 *seq = 0;
 881         }
 882
 883         goto out;
 884 }
 885
 886 /**
 887  *      llc_ui_sendmsg - Transmit data provided by the socket user.
 888  *      @sock: Socket to transmit data from.
 889  *      @msg: Various user related information.
 890  *      @len: Length of data to transmit.
 891  *
 892  *      Transmit data provided by the socket user.
 893  *      Returns non-negative upon success, negative otherwise.
 894  */
 895 static int llc_ui_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
 896 {
 897         struct sock *sk = sock->sk;
 898         struct llc_sock *llc = llc_sk(sk);
 899         DECLARE_SOCKADDR(struct sockaddr_llc *, addr, msg->msg_name);
 900         int flags = msg->msg_flags;
 901         int noblock = flags & MSG_DONTWAIT;
 902         struct sk_buff *skb;
 903         size_t size = 0;
 904         int rc = -EINVAL, copied = 0, hdrlen;
 905
 906         dprintk("%s: sending from %02X to %02X\n", __func__,
 907                 llc->laddr.lsap, llc->daddr.lsap);
 908         lock_sock(sk);
 909         if (addr) {
 910                 if (msg->msg_namelen < sizeof(*addr))
 911                         goto release;
 912         } else {
 913                 if (llc_ui_addr_null(&llc->addr))
 914                         goto release;
 915                 addr = &llc->addr;
 916         }
 917         /* must bind connection to sap if user hasn't done it. */
 918         if (sock_flag(sk, SOCK_ZAPPED)) {
 919                 /* bind to sap with null dev, exclusive. */
 920                 rc = llc_ui_autobind(sock, addr);
 921                 if (rc)
 922                         goto release;
 923         }
 924         hdrlen = llc->dev->hard_header_len + llc_ui_header_len(sk, addr);
 925         size = hdrlen + len;
 926         if (size > llc->dev->mtu)
 927                 size = llc->dev->mtu;
 928         copied = size - hdrlen;
 929         rc = -EINVAL;
 930         if (copied < 0)
 931                 goto release;
 932         release_sock(sk);
 933         skb = sock_alloc_send_skb(sk, size, noblock, &rc);
 934         lock_sock(sk);
 935         if (!skb)
 936                 goto release;
 937         skb->dev      = llc->dev;
 938         skb->protocol = llc_proto_type(addr->sllc_arphrd);
 939         skb_reserve(skb, hdrlen);
 940         rc = memcpy_from_msg(skb_put(skb, copied), msg, copied);
 941         if (rc)
 942                 goto out;
 943         if (sk->sk_type == SOCK_DGRAM || addr->sllc_ua) {
 944                 llc_build_and_send_ui_pkt(llc->sap, skb, addr->sllc_mac,
 945                                           addr->sllc_sap);
 946                 goto out;
 947         }
 948         if (addr->sllc_test) {
 949                 llc_build_and_send_test_pkt(llc->sap, skb, addr->sllc_mac,
 950                                             addr->sllc_sap);
 951                 goto out;
 952         }
 953         if (addr->sllc_xid) {
 954                 llc_build_and_send_xid_pkt(llc->sap, skb, addr->sllc_mac,
 955                                            addr->sllc_sap);
 956                 goto out;
 957         }
 958         rc = -ENOPROTOOPT;
 959         if (!(sk->sk_type == SOCK_STREAM && !addr->sllc_ua))
 960                 goto out;
 961         rc = llc_ui_send_data(sk, skb, noblock);
 962 out:
 963         if (rc) {
 964                 kfree_skb(skb);
 965 release:
 966                 dprintk("%s: failed sending from %02X to %02X: %d\n",
 967                         __func__, llc->laddr.lsap, llc->daddr.lsap, rc);
 968         }
 969         release_sock(sk);
 970         return rc ? : copied;
 971 }
 972
 973 /**
 974  *      llc_ui_getname - return the address info of a socket
 975  *      @sock: Socket to get address of.
 976  *      @uaddr: Address structure to return information.
 977  *      @uaddrlen: Length of address structure.
 978  *      @peer: Does user want local or remote address information.
 979  *
 980  *      Return the address information of a socket.
 981  */
 982 static int llc_ui_getname(struct socket *sock, struct sockaddr *uaddr,
 983                           int peer)
 984 {
 985         struct sockaddr_llc sllc;
 986         struct sock *sk = sock->sk;
 987         struct llc_sock *llc = llc_sk(sk);
 988         int rc = -EBADF;
 989
 990         memset(&sllc, 0, sizeof(sllc));
 991         lock_sock(sk);
 992         if (sock_flag(sk, SOCK_ZAPPED))
 993                 goto out;
 994         if (peer) {
 995                 rc = -ENOTCONN;
 996                 if (sk->sk_state != TCP_ESTABLISHED)
 997                         goto out;
 998                 if(llc->dev)
 999                         sllc.sllc_arphrd = llc->dev->type;
1000                 sllc.sllc_sap = llc->daddr.lsap;
1001                 memcpy(&sllc.sllc_mac, &llc->daddr.mac, IFHWADDRLEN);
1002         } else {
1003                 rc = -EINVAL;
1004                 if (!llc->sap)
1005                         goto out;
1006                 sllc.sllc_sap = llc->sap->laddr.lsap;
1007
1008                 if (llc->dev) {
1009                         sllc.sllc_arphrd = llc->dev->type;
1010                         memcpy(&sllc.sllc_mac, llc->dev->dev_addr,
1011                                IFHWADDRLEN);
1012                 }
1013         }
1014         sllc.sllc_family = AF_LLC;
1015         memcpy(uaddr, &sllc, sizeof(sllc));
1016         rc = sizeof(sllc);
1017 out:
1018         release_sock(sk);
1019         return rc;
1020 }
1021
1022 /**
1023  *      llc_ui_ioctl - io controls for PF_LLC
1024  *      @sock: Socket to get/set info
1025  *      @cmd: command
1026  *      @arg: optional argument for cmd
1027  *
1028  *      get/set info on llc sockets
1029  */
1030 static int llc_ui_ioctl(struct socket *sock, unsigned int cmd,
1031                         unsigned long arg)
1032 {
1033         return -ENOIOCTLCMD;
1034 }
1035
1036 /**
1037  *      llc_ui_setsockopt - set various connection specific parameters.
1038  *      @sock: Socket to set options on.
1039  *      @level: Socket level user is requesting operations on.
1040  *      @optname: Operation name.
1041  *      @optval: User provided operation data.
1042  *      @optlen: Length of optval.
1043  *
1044  *      Set various connection specific parameters.
1045  */
1046 static int llc_ui_setsockopt(struct socket *sock, int level, int optname,
1047                              char __user *optval, unsigned int optlen)
1048 {
1049         struct sock *sk = sock->sk;
1050         struct llc_sock *llc = llc_sk(sk);
1051         unsigned int opt;
1052         int rc = -EINVAL;
1053
1054         lock_sock(sk);
1055         if (unlikely(level != SOL_LLC || optlen != sizeof(int)))
1056                 goto out;
1057         rc = get_user(opt, (int __user *)optval);
1058         if (rc)
1059                 goto out;
1060         rc = -EINVAL;
1061         switch (optname) {
1062         case LLC_OPT_RETRY:
1063                 if (opt > LLC_OPT_MAX_RETRY)
1064                         goto out;
1065                 llc->n2 = opt;
1066                 break;
1067         case LLC_OPT_SIZE:
1068                 if (opt > LLC_OPT_MAX_SIZE)
1069                         goto out;
1070                 llc->n1 = opt;
1071                 break;
1072         case LLC_OPT_ACK_TMR_EXP:
1073                 if (opt > LLC_OPT_MAX_ACK_TMR_EXP)
1074                         goto out;
1075                 llc->ack_timer.expire = opt * HZ;
1076                 break;
1077         case LLC_OPT_P_TMR_EXP:
1078                 if (opt > LLC_OPT_MAX_P_TMR_EXP)
1079                         goto out;
1080                 llc->pf_cycle_timer.expire = opt * HZ;
1081                 break;
1082         case LLC_OPT_REJ_TMR_EXP:
1083                 if (opt > LLC_OPT_MAX_REJ_TMR_EXP)
1084                         goto out;
1085                 llc->rej_sent_timer.expire = opt * HZ;
1086                 break;
1087         case LLC_OPT_BUSY_TMR_EXP:
1088                 if (opt > LLC_OPT_MAX_BUSY_TMR_EXP)
1089                         goto out;
1090                 llc->busy_state_timer.expire = opt * HZ;
1091                 break;
1092         case LLC_OPT_TX_WIN:
1093                 if (opt > LLC_OPT_MAX_WIN)
1094                         goto out;
1095                 llc->k = opt;
1096                 break;
1097         case LLC_OPT_RX_WIN:
1098                 if (opt > LLC_OPT_MAX_WIN)
1099                         goto out;
1100                 llc->rw = opt;
1101                 break;
1102         case LLC_OPT_PKTINFO:
1103                 if (opt)
1104                         llc->cmsg_flags |= LLC_CMSG_PKTINFO;
1105                 else
1106                         llc->cmsg_flags &= ~LLC_CMSG_PKTINFO;
1107                 break;
1108         default:
1109                 rc = -ENOPROTOOPT;
1110                 goto out;
1111         }
1112         rc = 0;
1113 out:
1114         release_sock(sk);
1115         return rc;
1116 }
1117
1118 /**
1119  *      llc_ui_getsockopt - get connection specific socket info
1120  *      @sock: Socket to get information from.
1121  *      @level: Socket level user is requesting operations on.
1122  *      @optname: Operation name.
1123  *      @optval: Variable to return operation data in.
1124  *      @optlen: Length of optval.
1125  *
1126  *      Get connection specific socket information.
1127  */
1128 static int llc_ui_getsockopt(struct socket *sock, int level, int optname,
1129                              char __user *optval, int __user *optlen)
1130 {
1131         struct sock *sk = sock->sk;
1132         struct llc_sock *llc = llc_sk(sk);
1133         int val = 0, len = 0, rc = -EINVAL;
1134
1135         lock_sock(sk);
1136         if (unlikely(level != SOL_LLC))
1137                 goto out;
1138         rc = get_user(len, optlen);
1139         if (rc)
1140                 goto out;
1141         rc = -EINVAL;
1142         if (len != sizeof(int))
1143                 goto out;
1144         switch (optname) {
1145         case LLC_OPT_RETRY:
1146                 val = llc->n2;                                  break;
1147         case LLC_OPT_SIZE:
1148                 val = llc->n1;                                  break;
1149         case LLC_OPT_ACK_TMR_EXP:
1150                 val = llc->ack_timer.expire / HZ;               break;
1151         case LLC_OPT_P_TMR_EXP:
1152                 val = llc->pf_cycle_timer.expire / HZ;          break;
1153         case LLC_OPT_REJ_TMR_EXP:
1154                 val = llc->rej_sent_timer.expire / HZ;          break;
1155         case LLC_OPT_BUSY_TMR_EXP:
1156                 val = llc->busy_state_timer.expire / HZ;        break;
1157         case LLC_OPT_TX_WIN:
1158                 val = llc->k;                           break;
1159         case LLC_OPT_RX_WIN:
1160                 val = llc->rw;                          break;
1161         case LLC_OPT_PKTINFO:
1162                 val = (llc->cmsg_flags & LLC_CMSG_PKTINFO) != 0;
1163                 break;
1164         default:
1165                 rc = -ENOPROTOOPT;
1166                 goto out;
1167         }
1168         rc = 0;
1169         if (put_user(len, optlen) || copy_to_user(optval, &val, len))
1170                 rc = -EFAULT;
1171 out:
1172         release_sock(sk);
1173         return rc;
1174 }
1175
1176 static const struct net_proto_family llc_ui_family_ops = {
1177         .family = PF_LLC,
1178         .create = llc_ui_create,
1179         .owner  = THIS_MODULE,
1180 };
1181
1182 static const struct proto_ops llc_ui_ops = {
1183         .family      = PF_LLC,
1184         .owner       = THIS_MODULE,
1185         .release     = llc_ui_release,
1186         .bind        = llc_ui_bind,
1187         .connect     = llc_ui_connect,
1188         .socketpair  = sock_no_socketpair,
1189         .accept      = llc_ui_accept,
1190         .getname     = llc_ui_getname,
1191         .poll        = datagram_poll,
1192         .ioctl       = llc_ui_ioctl,
1193         .listen      = llc_ui_listen,
1194         .shutdown    = llc_ui_shutdown,
1195         .setsockopt  = llc_ui_setsockopt,
1196         .getsockopt  = llc_ui_getsockopt,
1197         .sendmsg     = llc_ui_sendmsg,
1198         .recvmsg     = llc_ui_recvmsg,
1199         .mmap        = sock_no_mmap,
1200         .sendpage    = sock_no_sendpage,
1201 };
1202
1203 static const char llc_proc_err_msg[] __initconst =
1204         KERN_CRIT "LLC: Unable to register the proc_fs entries\n";
1205 static const char llc_sysctl_err_msg[] __initconst =
1206         KERN_CRIT "LLC: Unable to register the sysctl entries\n";
1207 static const char llc_sock_err_msg[] __initconst =
1208         KERN_CRIT "LLC: Unable to register the network family\n";
1209
1210 static int __init llc2_init(void)
1211 {
1212         int rc = proto_register(&llc_proto, 0);
1213
1214         if (rc != 0)
1215                 goto out;
1216
1217         llc_build_offset_table();
1218         llc_station_init();
1219         llc_ui_sap_last_autoport = LLC_SAP_DYN_START;
1220         rc = llc_proc_init();
1221         if (rc != 0) {
1222                 printk(llc_proc_err_msg);
1223                 goto out_station;
1224         }
1225         rc = llc_sysctl_init();
1226         if (rc) {
1227                 printk(llc_sysctl_err_msg);
1228                 goto out_proc;
1229         }
1230         rc = sock_register(&llc_ui_family_ops);
1231         if (rc) {
1232                 printk(llc_sock_err_msg);
1233                 goto out_sysctl;
1234         }
1235         llc_add_pack(LLC_DEST_SAP, llc_sap_handler);
1236         llc_add_pack(LLC_DEST_CONN, llc_conn_handler);
1237 out:
1238         return rc;
1239 out_sysctl:
1240         llc_sysctl_exit();
1241 out_proc:
1242         llc_proc_exit();
1243 out_station:
1244         llc_station_exit();
1245         proto_unregister(&llc_proto);
1246         goto out;
1247 }
1248
1249 static void __exit llc2_exit(void)
1250 {
1251         llc_station_exit();
1252         llc_remove_pack(LLC_DEST_SAP);
1253         llc_remove_pack(LLC_DEST_CONN);
1254         sock_unregister(PF_LLC);
1255         llc_proc_exit();
1256         llc_sysctl_exit();
1257         proto_unregister(&llc_proto);
1258 }
1259
1260 module_init(llc2_init);
1261 module_exit(llc2_exit);
1262
1263 MODULE_LICENSE("GPL");
1264 MODULE_AUTHOR("Procom 1997, Jay Schullist 2001, Arnaldo C. Melo 2001-2003");
1265 MODULE_DESCRIPTION("IEEE 802.2 PF_LLC support");
1266 MODULE_ALIAS_NETPROTO(PF_LLC);