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