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