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