llc: do not use sk_eat_skb()
[muen/linux.git] / 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 <jschlst@samba.org>
14  *               2002-2003 by Arnaldo Carvalho de Melo <acme@conectiva.com.br>
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         dprintk("%s: binding %02X\n", __func__, addr->sllc_sap);
324
325         lock_sock(sk);
326         if (unlikely(!sock_flag(sk, SOCK_ZAPPED) || addrlen != sizeof(*addr)))
327                 goto out;
328         rc = -EAFNOSUPPORT;
329         if (unlikely(addr->sllc_family != AF_LLC))
330                 goto out;
331         rc = -ENODEV;
332         rcu_read_lock();
333         if (sk->sk_bound_dev_if) {
334                 llc->dev = dev_get_by_index_rcu(&init_net, sk->sk_bound_dev_if);
335                 if (llc->dev) {
336                         if (!addr->sllc_arphrd)
337                                 addr->sllc_arphrd = llc->dev->type;
338                         if (is_zero_ether_addr(addr->sllc_mac))
339                                 memcpy(addr->sllc_mac, llc->dev->dev_addr,
340                                        IFHWADDRLEN);
341                         if (addr->sllc_arphrd != llc->dev->type ||
342                             !ether_addr_equal(addr->sllc_mac,
343                                               llc->dev->dev_addr)) {
344                                 rc = -EINVAL;
345                                 llc->dev = NULL;
346                         }
347                 }
348         } else
349                 llc->dev = dev_getbyhwaddr_rcu(&init_net, addr->sllc_arphrd,
350                                            addr->sllc_mac);
351         if (llc->dev)
352                 dev_hold(llc->dev);
353         rcu_read_unlock();
354         if (!llc->dev)
355                 goto out;
356         if (!addr->sllc_sap) {
357                 rc = -EUSERS;
358                 addr->sllc_sap = llc_ui_autoport();
359                 if (!addr->sllc_sap)
360                         goto out;
361         }
362         sap = llc_sap_find(addr->sllc_sap);
363         if (!sap) {
364                 sap = llc_sap_open(addr->sllc_sap, NULL);
365                 rc = -EBUSY; /* some other network layer is using the sap */
366                 if (!sap)
367                         goto out;
368         } else {
369                 struct llc_addr laddr, daddr;
370                 struct sock *ask;
371
372                 memset(&laddr, 0, sizeof(laddr));
373                 memset(&daddr, 0, sizeof(daddr));
374                 /*
375                  * FIXME: check if the address is multicast,
376                  *        only SOCK_DGRAM can do this.
377                  */
378                 memcpy(laddr.mac, addr->sllc_mac, IFHWADDRLEN);
379                 laddr.lsap = addr->sllc_sap;
380                 rc = -EADDRINUSE; /* mac + sap clash. */
381                 ask = llc_lookup_established(sap, &daddr, &laddr);
382                 if (ask) {
383                         sock_put(ask);
384                         goto out_put;
385                 }
386         }
387         llc->laddr.lsap = addr->sllc_sap;
388         memcpy(llc->laddr.mac, addr->sllc_mac, IFHWADDRLEN);
389         memcpy(&llc->addr, addr, sizeof(llc->addr));
390         /* assign new connection to its SAP */
391         llc_sap_add_socket(sap, sk);
392         sock_reset_flag(sk, SOCK_ZAPPED);
393         rc = 0;
394 out_put:
395         llc_sap_put(sap);
396 out:
397         release_sock(sk);
398         return rc;
399 }
400
401 /**
402  *      llc_ui_shutdown - shutdown a connect llc2 socket.
403  *      @sock: Socket to shutdown.
404  *      @how: What part of the socket to shutdown.
405  *
406  *      Shutdown a connected llc2 socket. Currently this function only supports
407  *      shutting down both sends and receives (2), we could probably make this
408  *      function such that a user can shutdown only half the connection but not
409  *      right now.
410  *      Returns: 0 upon success, negative otherwise.
411  */
412 static int llc_ui_shutdown(struct socket *sock, int how)
413 {
414         struct sock *sk = sock->sk;
415         int rc = -ENOTCONN;
416
417         lock_sock(sk);
418         if (unlikely(sk->sk_state != TCP_ESTABLISHED))
419                 goto out;
420         rc = -EINVAL;
421         if (how != 2)
422                 goto out;
423         rc = llc_send_disc(sk);
424         if (!rc)
425                 rc = llc_ui_wait_for_disc(sk, sk->sk_rcvtimeo);
426         /* Wake up anyone sleeping in poll */
427         sk->sk_state_change(sk);
428 out:
429         release_sock(sk);
430         return rc;
431 }
432
433 /**
434  *      llc_ui_connect - Connect to a remote llc2 mac + sap.
435  *      @sock: Socket which will be connected to the remote destination.
436  *      @uaddr: Remote and possibly the local address of the new connection.
437  *      @addrlen: Size of uaddr structure.
438  *      @flags: Operational flags specified by the user.
439  *
440  *      Connect to a remote llc2 mac + sap. The caller must specify the
441  *      destination mac and address to connect to. If the user hasn't previously
442  *      called bind(2) with a smac the address of the first interface of the
443  *      specified arp type will be used.
444  *      This function will autobind if user did not previously call bind.
445  *      Returns: 0 upon success, negative otherwise.
446  */
447 static int llc_ui_connect(struct socket *sock, struct sockaddr *uaddr,
448                           int addrlen, int flags)
449 {
450         struct sock *sk = sock->sk;
451         struct llc_sock *llc = llc_sk(sk);
452         struct sockaddr_llc *addr = (struct sockaddr_llc *)uaddr;
453         int rc = -EINVAL;
454
455         lock_sock(sk);
456         if (unlikely(addrlen != sizeof(*addr)))
457                 goto out;
458         rc = -EAFNOSUPPORT;
459         if (unlikely(addr->sllc_family != AF_LLC))
460                 goto out;
461         if (unlikely(sk->sk_type != SOCK_STREAM))
462                 goto out;
463         rc = -EALREADY;
464         if (unlikely(sock->state == SS_CONNECTING))
465                 goto out;
466         /* bind connection to sap if user hasn't done it. */
467         if (sock_flag(sk, SOCK_ZAPPED)) {
468                 /* bind to sap with null dev, exclusive */
469                 rc = llc_ui_autobind(sock, addr);
470                 if (rc)
471                         goto out;
472         }
473         llc->daddr.lsap = addr->sllc_sap;
474         memcpy(llc->daddr.mac, addr->sllc_mac, IFHWADDRLEN);
475         sock->state = SS_CONNECTING;
476         sk->sk_state   = TCP_SYN_SENT;
477         llc->link   = llc_ui_next_link_no(llc->sap->laddr.lsap);
478         rc = llc_establish_connection(sk, llc->dev->dev_addr,
479                                       addr->sllc_mac, addr->sllc_sap);
480         if (rc) {
481                 dprintk("%s: llc_ui_send_conn failed :-(\n", __func__);
482                 sock->state  = SS_UNCONNECTED;
483                 sk->sk_state = TCP_CLOSE;
484                 goto out;
485         }
486
487         if (sk->sk_state == TCP_SYN_SENT) {
488                 const long timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
489
490                 if (!timeo || !llc_ui_wait_for_conn(sk, timeo))
491                         goto out;
492
493                 rc = sock_intr_errno(timeo);
494                 if (signal_pending(current))
495                         goto out;
496         }
497
498         if (sk->sk_state == TCP_CLOSE)
499                 goto sock_error;
500
501         sock->state = SS_CONNECTED;
502         rc = 0;
503 out:
504         release_sock(sk);
505         return rc;
506 sock_error:
507         rc = sock_error(sk) ? : -ECONNABORTED;
508         sock->state = SS_UNCONNECTED;
509         goto out;
510 }
511
512 /**
513  *      llc_ui_listen - allow a normal socket to accept incoming connections
514  *      @sock: Socket to allow incoming connections on.
515  *      @backlog: Number of connections to queue.
516  *
517  *      Allow a normal socket to accept incoming connections.
518  *      Returns 0 upon success, negative otherwise.
519  */
520 static int llc_ui_listen(struct socket *sock, int backlog)
521 {
522         struct sock *sk = sock->sk;
523         int rc = -EINVAL;
524
525         lock_sock(sk);
526         if (unlikely(sock->state != SS_UNCONNECTED))
527                 goto out;
528         rc = -EOPNOTSUPP;
529         if (unlikely(sk->sk_type != SOCK_STREAM))
530                 goto out;
531         rc = -EAGAIN;
532         if (sock_flag(sk, SOCK_ZAPPED))
533                 goto out;
534         rc = 0;
535         if (!(unsigned int)backlog)     /* BSDism */
536                 backlog = 1;
537         sk->sk_max_ack_backlog = backlog;
538         if (sk->sk_state != TCP_LISTEN) {
539                 sk->sk_ack_backlog = 0;
540                 sk->sk_state       = TCP_LISTEN;
541         }
542         sk->sk_socket->flags |= __SO_ACCEPTCON;
543 out:
544         release_sock(sk);
545         return rc;
546 }
547
548 static int llc_ui_wait_for_disc(struct sock *sk, long timeout)
549 {
550         DEFINE_WAIT_FUNC(wait, woken_wake_function);
551         int rc = 0;
552
553         add_wait_queue(sk_sleep(sk), &wait);
554         while (1) {
555                 if (sk_wait_event(sk, &timeout, sk->sk_state == TCP_CLOSE, &wait))
556                         break;
557                 rc = -ERESTARTSYS;
558                 if (signal_pending(current))
559                         break;
560                 rc = -EAGAIN;
561                 if (!timeout)
562                         break;
563                 rc = 0;
564         }
565         remove_wait_queue(sk_sleep(sk), &wait);
566         return rc;
567 }
568
569 static bool llc_ui_wait_for_conn(struct sock *sk, long timeout)
570 {
571         DEFINE_WAIT_FUNC(wait, woken_wake_function);
572
573         add_wait_queue(sk_sleep(sk), &wait);
574         while (1) {
575                 if (sk_wait_event(sk, &timeout, sk->sk_state != TCP_SYN_SENT, &wait))
576                         break;
577                 if (signal_pending(current) || !timeout)
578                         break;
579         }
580         remove_wait_queue(sk_sleep(sk), &wait);
581         return timeout;
582 }
583
584 static int llc_ui_wait_for_busy_core(struct sock *sk, long timeout)
585 {
586         DEFINE_WAIT_FUNC(wait, woken_wake_function);
587         struct llc_sock *llc = llc_sk(sk);
588         int rc;
589
590         add_wait_queue(sk_sleep(sk), &wait);
591         while (1) {
592                 rc = 0;
593                 if (sk_wait_event(sk, &timeout,
594                                   (sk->sk_shutdown & RCV_SHUTDOWN) ||
595                                   (!llc_data_accept_state(llc->state) &&
596                                    !llc->remote_busy_flag &&
597                                    !llc->p_flag), &wait))
598                         break;
599                 rc = -ERESTARTSYS;
600                 if (signal_pending(current))
601                         break;
602                 rc = -EAGAIN;
603                 if (!timeout)
604                         break;
605         }
606         remove_wait_queue(sk_sleep(sk), &wait);
607         return rc;
608 }
609
610 static int llc_wait_data(struct sock *sk, long timeo)
611 {
612         int rc;
613
614         while (1) {
615                 /*
616                  * POSIX 1003.1g mandates this order.
617                  */
618                 rc = sock_error(sk);
619                 if (rc)
620                         break;
621                 rc = 0;
622                 if (sk->sk_shutdown & RCV_SHUTDOWN)
623                         break;
624                 rc = -EAGAIN;
625                 if (!timeo)
626                         break;
627                 rc = sock_intr_errno(timeo);
628                 if (signal_pending(current))
629                         break;
630                 rc = 0;
631                 if (sk_wait_data(sk, &timeo, NULL))
632                         break;
633         }
634         return rc;
635 }
636
637 static void llc_cmsg_rcv(struct msghdr *msg, struct sk_buff *skb)
638 {
639         struct llc_sock *llc = llc_sk(skb->sk);
640
641         if (llc->cmsg_flags & LLC_CMSG_PKTINFO) {
642                 struct llc_pktinfo info;
643
644                 memset(&info, 0, sizeof(info));
645                 info.lpi_ifindex = llc_sk(skb->sk)->dev->ifindex;
646                 llc_pdu_decode_dsap(skb, &info.lpi_sap);
647                 llc_pdu_decode_da(skb, info.lpi_mac);
648                 put_cmsg(msg, SOL_LLC, LLC_OPT_PKTINFO, sizeof(info), &info);
649         }
650 }
651
652 /**
653  *      llc_ui_accept - accept a new incoming connection.
654  *      @sock: Socket which connections arrive on.
655  *      @newsock: Socket to move incoming connection to.
656  *      @flags: User specified operational flags.
657  *      @kern: If the socket is kernel internal
658  *
659  *      Accept a new incoming connection.
660  *      Returns 0 upon success, negative otherwise.
661  */
662 static int llc_ui_accept(struct socket *sock, struct socket *newsock, int flags,
663                          bool kern)
664 {
665         struct sock *sk = sock->sk, *newsk;
666         struct llc_sock *llc, *newllc;
667         struct sk_buff *skb;
668         int rc = -EOPNOTSUPP;
669
670         dprintk("%s: accepting on %02X\n", __func__,
671                 llc_sk(sk)->laddr.lsap);
672         lock_sock(sk);
673         if (unlikely(sk->sk_type != SOCK_STREAM))
674                 goto out;
675         rc = -EINVAL;
676         if (unlikely(sock->state != SS_UNCONNECTED ||
677                      sk->sk_state != TCP_LISTEN))
678                 goto out;
679         /* wait for a connection to arrive. */
680         if (skb_queue_empty(&sk->sk_receive_queue)) {
681                 rc = llc_wait_data(sk, sk->sk_rcvtimeo);
682                 if (rc)
683                         goto out;
684         }
685         dprintk("%s: got a new connection on %02X\n", __func__,
686                 llc_sk(sk)->laddr.lsap);
687         skb = skb_dequeue(&sk->sk_receive_queue);
688         rc = -EINVAL;
689         if (!skb->sk)
690                 goto frees;
691         rc = 0;
692         newsk = skb->sk;
693         /* attach connection to a new socket. */
694         llc_ui_sk_init(newsock, newsk);
695         sock_reset_flag(newsk, SOCK_ZAPPED);
696         newsk->sk_state         = TCP_ESTABLISHED;
697         newsock->state          = SS_CONNECTED;
698         llc                     = llc_sk(sk);
699         newllc                  = llc_sk(newsk);
700         memcpy(&newllc->addr, &llc->addr, sizeof(newllc->addr));
701         newllc->link = llc_ui_next_link_no(newllc->laddr.lsap);
702
703         /* put original socket back into a clean listen state. */
704         sk->sk_state = TCP_LISTEN;
705         sk->sk_ack_backlog--;
706         dprintk("%s: ok success on %02X, client on %02X\n", __func__,
707                 llc_sk(sk)->addr.sllc_sap, newllc->daddr.lsap);
708 frees:
709         kfree_skb(skb);
710 out:
711         release_sock(sk);
712         return rc;
713 }
714
715 /**
716  *      llc_ui_recvmsg - copy received data to the socket user.
717  *      @sock: Socket to copy data from.
718  *      @msg: Various user space related information.
719  *      @len: Size of user buffer.
720  *      @flags: User specified flags.
721  *
722  *      Copy received data to the socket user.
723  *      Returns non-negative upon success, negative otherwise.
724  */
725 static int llc_ui_recvmsg(struct socket *sock, struct msghdr *msg, size_t len,
726                           int flags)
727 {
728         DECLARE_SOCKADDR(struct sockaddr_llc *, uaddr, msg->msg_name);
729         const int nonblock = flags & MSG_DONTWAIT;
730         struct sk_buff *skb = NULL;
731         struct sock *sk = sock->sk;
732         struct llc_sock *llc = llc_sk(sk);
733         size_t copied = 0;
734         u32 peek_seq = 0;
735         u32 *seq, skb_len;
736         unsigned long used;
737         int target;     /* Read at least this many bytes */
738         long timeo;
739
740         lock_sock(sk);
741         copied = -ENOTCONN;
742         if (unlikely(sk->sk_type == SOCK_STREAM && sk->sk_state == TCP_LISTEN))
743                 goto out;
744
745         timeo = sock_rcvtimeo(sk, nonblock);
746
747         seq = &llc->copied_seq;
748         if (flags & MSG_PEEK) {
749                 peek_seq = llc->copied_seq;
750                 seq = &peek_seq;
751         }
752
753         target = sock_rcvlowat(sk, flags & MSG_WAITALL, len);
754         copied = 0;
755
756         do {
757                 u32 offset;
758
759                 /*
760                  * We need to check signals first, to get correct SIGURG
761                  * handling. FIXME: Need to check this doesn't impact 1003.1g
762                  * and move it down to the bottom of the loop
763                  */
764                 if (signal_pending(current)) {
765                         if (copied)
766                                 break;
767                         copied = timeo ? sock_intr_errno(timeo) : -EAGAIN;
768                         break;
769                 }
770
771                 /* Next get a buffer. */
772
773                 skb = skb_peek(&sk->sk_receive_queue);
774                 if (skb) {
775                         offset = *seq;
776                         goto found_ok_skb;
777                 }
778                 /* Well, if we have backlog, try to process it now yet. */
779
780                 if (copied >= target && !sk->sk_backlog.tail)
781                         break;
782
783                 if (copied) {
784                         if (sk->sk_err ||
785                             sk->sk_state == TCP_CLOSE ||
786                             (sk->sk_shutdown & RCV_SHUTDOWN) ||
787                             !timeo ||
788                             (flags & MSG_PEEK))
789                                 break;
790                 } else {
791                         if (sock_flag(sk, SOCK_DONE))
792                                 break;
793
794                         if (sk->sk_err) {
795                                 copied = sock_error(sk);
796                                 break;
797                         }
798                         if (sk->sk_shutdown & RCV_SHUTDOWN)
799                                 break;
800
801                         if (sk->sk_type == SOCK_STREAM && sk->sk_state == TCP_CLOSE) {
802                                 if (!sock_flag(sk, SOCK_DONE)) {
803                                         /*
804                                          * This occurs when user tries to read
805                                          * from never connected socket.
806                                          */
807                                         copied = -ENOTCONN;
808                                         break;
809                                 }
810                                 break;
811                         }
812                         if (!timeo) {
813                                 copied = -EAGAIN;
814                                 break;
815                         }
816                 }
817
818                 if (copied >= target) { /* Do not sleep, just process backlog. */
819                         release_sock(sk);
820                         lock_sock(sk);
821                 } else
822                         sk_wait_data(sk, &timeo, NULL);
823
824                 if ((flags & MSG_PEEK) && peek_seq != llc->copied_seq) {
825                         net_dbg_ratelimited("LLC(%s:%d): Application bug, race in MSG_PEEK\n",
826                                             current->comm,
827                                             task_pid_nr(current));
828                         peek_seq = llc->copied_seq;
829                 }
830                 continue;
831         found_ok_skb:
832                 skb_len = skb->len;
833                 /* Ok so how much can we use? */
834                 used = skb->len - offset;
835                 if (len < used)
836                         used = len;
837
838                 if (!(flags & MSG_TRUNC)) {
839                         int rc = skb_copy_datagram_msg(skb, offset, msg, used);
840                         if (rc) {
841                                 /* Exception. Bailout! */
842                                 if (!copied)
843                                         copied = -EFAULT;
844                                 break;
845                         }
846                 }
847
848                 *seq += used;
849                 copied += used;
850                 len -= used;
851
852                 /* For non stream protcols we get one packet per recvmsg call */
853                 if (sk->sk_type != SOCK_STREAM)
854                         goto copy_uaddr;
855
856                 if (!(flags & MSG_PEEK)) {
857                         skb_unlink(skb, &sk->sk_receive_queue);
858                         kfree_skb(skb);
859                         *seq = 0;
860                 }
861
862                 /* Partial read */
863                 if (used + offset < skb_len)
864                         continue;
865         } while (len > 0);
866
867 out:
868         release_sock(sk);
869         return copied;
870 copy_uaddr:
871         if (uaddr != NULL && skb != NULL) {
872                 memcpy(uaddr, llc_ui_skb_cb(skb), sizeof(*uaddr));
873                 msg->msg_namelen = sizeof(*uaddr);
874         }
875         if (llc_sk(sk)->cmsg_flags)
876                 llc_cmsg_rcv(msg, skb);
877
878         if (!(flags & MSG_PEEK)) {
879                 skb_unlink(skb, &sk->sk_receive_queue);
880                 kfree_skb(skb);
881                 *seq = 0;
882         }
883
884         goto out;
885 }
886
887 /**
888  *      llc_ui_sendmsg - Transmit data provided by the socket user.
889  *      @sock: Socket to transmit data from.
890  *      @msg: Various user related information.
891  *      @len: Length of data to transmit.
892  *
893  *      Transmit data provided by the socket user.
894  *      Returns non-negative upon success, negative otherwise.
895  */
896 static int llc_ui_sendmsg(struct socket *sock, struct msghdr *msg, size_t len)
897 {
898         struct sock *sk = sock->sk;
899         struct llc_sock *llc = llc_sk(sk);
900         DECLARE_SOCKADDR(struct sockaddr_llc *, addr, msg->msg_name);
901         int flags = msg->msg_flags;
902         int noblock = flags & MSG_DONTWAIT;
903         struct sk_buff *skb;
904         size_t size = 0;
905         int rc = -EINVAL, copied = 0, hdrlen;
906
907         dprintk("%s: sending from %02X to %02X\n", __func__,
908                 llc->laddr.lsap, llc->daddr.lsap);
909         lock_sock(sk);
910         if (addr) {
911                 if (msg->msg_namelen < sizeof(*addr))
912                         goto release;
913         } else {
914                 if (llc_ui_addr_null(&llc->addr))
915                         goto release;
916                 addr = &llc->addr;
917         }
918         /* must bind connection to sap if user hasn't done it. */
919         if (sock_flag(sk, SOCK_ZAPPED)) {
920                 /* bind to sap with null dev, exclusive. */
921                 rc = llc_ui_autobind(sock, addr);
922                 if (rc)
923                         goto release;
924         }
925         hdrlen = llc->dev->hard_header_len + llc_ui_header_len(sk, addr);
926         size = hdrlen + len;
927         if (size > llc->dev->mtu)
928                 size = llc->dev->mtu;
929         copied = size - hdrlen;
930         rc = -EINVAL;
931         if (copied < 0)
932                 goto release;
933         release_sock(sk);
934         skb = sock_alloc_send_skb(sk, size, noblock, &rc);
935         lock_sock(sk);
936         if (!skb)
937                 goto release;
938         skb->dev      = llc->dev;
939         skb->protocol = llc_proto_type(addr->sllc_arphrd);
940         skb_reserve(skb, hdrlen);
941         rc = memcpy_from_msg(skb_put(skb, copied), msg, copied);
942         if (rc)
943                 goto out;
944         if (sk->sk_type == SOCK_DGRAM || addr->sllc_ua) {
945                 llc_build_and_send_ui_pkt(llc->sap, skb, addr->sllc_mac,
946                                           addr->sllc_sap);
947                 goto out;
948         }
949         if (addr->sllc_test) {
950                 llc_build_and_send_test_pkt(llc->sap, skb, addr->sllc_mac,
951                                             addr->sllc_sap);
952                 goto out;
953         }
954         if (addr->sllc_xid) {
955                 llc_build_and_send_xid_pkt(llc->sap, skb, addr->sllc_mac,
956                                            addr->sllc_sap);
957                 goto out;
958         }
959         rc = -ENOPROTOOPT;
960         if (!(sk->sk_type == SOCK_STREAM && !addr->sllc_ua))
961                 goto out;
962         rc = llc_ui_send_data(sk, skb, noblock);
963 out:
964         if (rc) {
965                 kfree_skb(skb);
966 release:
967                 dprintk("%s: failed sending from %02X to %02X: %d\n",
968                         __func__, llc->laddr.lsap, llc->daddr.lsap, rc);
969         }
970         release_sock(sk);
971         return rc ? : copied;
972 }
973
974 /**
975  *      llc_ui_getname - return the address info of a socket
976  *      @sock: Socket to get address of.
977  *      @uaddr: Address structure to return information.
978  *      @uaddrlen: Length of address structure.
979  *      @peer: Does user want local or remote address information.
980  *
981  *      Return the address information of a socket.
982  */
983 static int llc_ui_getname(struct socket *sock, struct sockaddr *uaddr,
984                           int peer)
985 {
986         struct sockaddr_llc sllc;
987         struct sock *sk = sock->sk;
988         struct llc_sock *llc = llc_sk(sk);
989         int rc = -EBADF;
990
991         memset(&sllc, 0, sizeof(sllc));
992         lock_sock(sk);
993         if (sock_flag(sk, SOCK_ZAPPED))
994                 goto out;
995         if (peer) {
996                 rc = -ENOTCONN;
997                 if (sk->sk_state != TCP_ESTABLISHED)
998                         goto out;
999                 if(llc->dev)
1000                         sllc.sllc_arphrd = llc->dev->type;
1001                 sllc.sllc_sap = llc->daddr.lsap;
1002                 memcpy(&sllc.sllc_mac, &llc->daddr.mac, IFHWADDRLEN);
1003         } else {
1004                 rc = -EINVAL;
1005                 if (!llc->sap)
1006                         goto out;
1007                 sllc.sllc_sap = llc->sap->laddr.lsap;
1008
1009                 if (llc->dev) {
1010                         sllc.sllc_arphrd = llc->dev->type;
1011                         memcpy(&sllc.sllc_mac, llc->dev->dev_addr,
1012                                IFHWADDRLEN);
1013                 }
1014         }
1015         sllc.sllc_family = AF_LLC;
1016         memcpy(uaddr, &sllc, sizeof(sllc));
1017         rc = sizeof(sllc);
1018 out:
1019         release_sock(sk);
1020         return rc;
1021 }
1022
1023 /**
1024  *      llc_ui_ioctl - io controls for PF_LLC
1025  *      @sock: Socket to get/set info
1026  *      @cmd: command
1027  *      @arg: optional argument for cmd
1028  *
1029  *      get/set info on llc sockets
1030  */
1031 static int llc_ui_ioctl(struct socket *sock, unsigned int cmd,
1032                         unsigned long arg)
1033 {
1034         return -ENOIOCTLCMD;
1035 }
1036
1037 /**
1038  *      llc_ui_setsockopt - set various connection specific parameters.
1039  *      @sock: Socket to set options on.
1040  *      @level: Socket level user is requesting operations on.
1041  *      @optname: Operation name.
1042  *      @optval: User provided operation data.
1043  *      @optlen: Length of optval.
1044  *
1045  *      Set various connection specific parameters.
1046  */
1047 static int llc_ui_setsockopt(struct socket *sock, int level, int optname,
1048                              char __user *optval, unsigned int optlen)
1049 {
1050         struct sock *sk = sock->sk;
1051         struct llc_sock *llc = llc_sk(sk);
1052         unsigned int opt;
1053         int rc = -EINVAL;
1054
1055         lock_sock(sk);
1056         if (unlikely(level != SOL_LLC || optlen != sizeof(int)))
1057                 goto out;
1058         rc = get_user(opt, (int __user *)optval);
1059         if (rc)
1060                 goto out;
1061         rc = -EINVAL;
1062         switch (optname) {
1063         case LLC_OPT_RETRY:
1064                 if (opt > LLC_OPT_MAX_RETRY)
1065                         goto out;
1066                 llc->n2 = opt;
1067                 break;
1068         case LLC_OPT_SIZE:
1069                 if (opt > LLC_OPT_MAX_SIZE)
1070                         goto out;
1071                 llc->n1 = opt;
1072                 break;
1073         case LLC_OPT_ACK_TMR_EXP:
1074                 if (opt > LLC_OPT_MAX_ACK_TMR_EXP)
1075                         goto out;
1076                 llc->ack_timer.expire = opt * HZ;
1077                 break;
1078         case LLC_OPT_P_TMR_EXP:
1079                 if (opt > LLC_OPT_MAX_P_TMR_EXP)
1080                         goto out;
1081                 llc->pf_cycle_timer.expire = opt * HZ;
1082                 break;
1083         case LLC_OPT_REJ_TMR_EXP:
1084                 if (opt > LLC_OPT_MAX_REJ_TMR_EXP)
1085                         goto out;
1086                 llc->rej_sent_timer.expire = opt * HZ;
1087                 break;
1088         case LLC_OPT_BUSY_TMR_EXP:
1089                 if (opt > LLC_OPT_MAX_BUSY_TMR_EXP)
1090                         goto out;
1091                 llc->busy_state_timer.expire = opt * HZ;
1092                 break;
1093         case LLC_OPT_TX_WIN:
1094                 if (opt > LLC_OPT_MAX_WIN)
1095                         goto out;
1096                 llc->k = opt;
1097                 break;
1098         case LLC_OPT_RX_WIN:
1099                 if (opt > LLC_OPT_MAX_WIN)
1100                         goto out;
1101                 llc->rw = opt;
1102                 break;
1103         case LLC_OPT_PKTINFO:
1104                 if (opt)
1105                         llc->cmsg_flags |= LLC_CMSG_PKTINFO;
1106                 else
1107                         llc->cmsg_flags &= ~LLC_CMSG_PKTINFO;
1108                 break;
1109         default:
1110                 rc = -ENOPROTOOPT;
1111                 goto out;
1112         }
1113         rc = 0;
1114 out:
1115         release_sock(sk);
1116         return rc;
1117 }
1118
1119 /**
1120  *      llc_ui_getsockopt - get connection specific socket info
1121  *      @sock: Socket to get information from.
1122  *      @level: Socket level user is requesting operations on.
1123  *      @optname: Operation name.
1124  *      @optval: Variable to return operation data in.
1125  *      @optlen: Length of optval.
1126  *
1127  *      Get connection specific socket information.
1128  */
1129 static int llc_ui_getsockopt(struct socket *sock, int level, int optname,
1130                              char __user *optval, int __user *optlen)
1131 {
1132         struct sock *sk = sock->sk;
1133         struct llc_sock *llc = llc_sk(sk);
1134         int val = 0, len = 0, rc = -EINVAL;
1135
1136         lock_sock(sk);
1137         if (unlikely(level != SOL_LLC))
1138                 goto out;
1139         rc = get_user(len, optlen);
1140         if (rc)
1141                 goto out;
1142         rc = -EINVAL;
1143         if (len != sizeof(int))
1144                 goto out;
1145         switch (optname) {
1146         case LLC_OPT_RETRY:
1147                 val = llc->n2;                                  break;
1148         case LLC_OPT_SIZE:
1149                 val = llc->n1;                                  break;
1150         case LLC_OPT_ACK_TMR_EXP:
1151                 val = llc->ack_timer.expire / HZ;               break;
1152         case LLC_OPT_P_TMR_EXP:
1153                 val = llc->pf_cycle_timer.expire / HZ;          break;
1154         case LLC_OPT_REJ_TMR_EXP:
1155                 val = llc->rej_sent_timer.expire / HZ;          break;
1156         case LLC_OPT_BUSY_TMR_EXP:
1157                 val = llc->busy_state_timer.expire / HZ;        break;
1158         case LLC_OPT_TX_WIN:
1159                 val = llc->k;                           break;
1160         case LLC_OPT_RX_WIN:
1161                 val = llc->rw;                          break;
1162         case LLC_OPT_PKTINFO:
1163                 val = (llc->cmsg_flags & LLC_CMSG_PKTINFO) != 0;
1164                 break;
1165         default:
1166                 rc = -ENOPROTOOPT;
1167                 goto out;
1168         }
1169         rc = 0;
1170         if (put_user(len, optlen) || copy_to_user(optval, &val, len))
1171                 rc = -EFAULT;
1172 out:
1173         release_sock(sk);
1174         return rc;
1175 }
1176
1177 static const struct net_proto_family llc_ui_family_ops = {
1178         .family = PF_LLC,
1179         .create = llc_ui_create,
1180         .owner  = THIS_MODULE,
1181 };
1182
1183 static const struct proto_ops llc_ui_ops = {
1184         .family      = PF_LLC,
1185         .owner       = THIS_MODULE,
1186         .release     = llc_ui_release,
1187         .bind        = llc_ui_bind,
1188         .connect     = llc_ui_connect,
1189         .socketpair  = sock_no_socketpair,
1190         .accept      = llc_ui_accept,
1191         .getname     = llc_ui_getname,
1192         .poll        = datagram_poll,
1193         .ioctl       = llc_ui_ioctl,
1194         .listen      = llc_ui_listen,
1195         .shutdown    = llc_ui_shutdown,
1196         .setsockopt  = llc_ui_setsockopt,
1197         .getsockopt  = llc_ui_getsockopt,
1198         .sendmsg     = llc_ui_sendmsg,
1199         .recvmsg     = llc_ui_recvmsg,
1200         .mmap        = sock_no_mmap,
1201         .sendpage    = sock_no_sendpage,
1202 };
1203
1204 static const char llc_proc_err_msg[] __initconst =
1205         KERN_CRIT "LLC: Unable to register the proc_fs entries\n";
1206 static const char llc_sysctl_err_msg[] __initconst =
1207         KERN_CRIT "LLC: Unable to register the sysctl entries\n";
1208 static const char llc_sock_err_msg[] __initconst =
1209         KERN_CRIT "LLC: Unable to register the network family\n";
1210
1211 static int __init llc2_init(void)
1212 {
1213         int rc = proto_register(&llc_proto, 0);
1214
1215         if (rc != 0)
1216                 goto out;
1217
1218         llc_build_offset_table();
1219         llc_station_init();
1220         llc_ui_sap_last_autoport = LLC_SAP_DYN_START;
1221         rc = llc_proc_init();
1222         if (rc != 0) {
1223                 printk(llc_proc_err_msg);
1224                 goto out_station;
1225         }
1226         rc = llc_sysctl_init();
1227         if (rc) {
1228                 printk(llc_sysctl_err_msg);
1229                 goto out_proc;
1230         }
1231         rc = sock_register(&llc_ui_family_ops);
1232         if (rc) {
1233                 printk(llc_sock_err_msg);
1234                 goto out_sysctl;
1235         }
1236         llc_add_pack(LLC_DEST_SAP, llc_sap_handler);
1237         llc_add_pack(LLC_DEST_CONN, llc_conn_handler);
1238 out:
1239         return rc;
1240 out_sysctl:
1241         llc_sysctl_exit();
1242 out_proc:
1243         llc_proc_exit();
1244 out_station:
1245         llc_station_exit();
1246         proto_unregister(&llc_proto);
1247         goto out;
1248 }
1249
1250 static void __exit llc2_exit(void)
1251 {
1252         llc_station_exit();
1253         llc_remove_pack(LLC_DEST_SAP);
1254         llc_remove_pack(LLC_DEST_CONN);
1255         sock_unregister(PF_LLC);
1256         llc_proc_exit();
1257         llc_sysctl_exit();
1258         proto_unregister(&llc_proto);
1259 }
1260
1261 module_init(llc2_init);
1262 module_exit(llc2_exit);
1263
1264 MODULE_LICENSE("GPL");
1265 MODULE_AUTHOR("Procom 1997, Jay Schullist 2001, Arnaldo C. Melo 2001-2003");
1266 MODULE_DESCRIPTION("IEEE 802.2 PF_LLC support");
1267 MODULE_ALIAS_NETPROTO(PF_LLC);