net: add option to enable error queue packets waking select
[linux-3.10.git] / net / nfc / llcp / sock.c
1 /*
2  * Copyright (C) 2011  Intel Corporation. All rights reserved.
3  *
4  * This program is free software; you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation; either version 2 of the License, or
7  * (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the
16  * Free Software Foundation, Inc.,
17  * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
18  */
19
20 #define pr_fmt(fmt) "llcp: %s: " fmt, __func__
21
22 #include <linux/init.h>
23 #include <linux/kernel.h>
24 #include <linux/module.h>
25 #include <linux/nfc.h>
26
27 #include "../nfc.h"
28 #include "llcp.h"
29
30 static int sock_wait_state(struct sock *sk, int state, unsigned long timeo)
31 {
32         DECLARE_WAITQUEUE(wait, current);
33         int err = 0;
34
35         pr_debug("sk %p", sk);
36
37         add_wait_queue(sk_sleep(sk), &wait);
38         set_current_state(TASK_INTERRUPTIBLE);
39
40         while (sk->sk_state != state) {
41                 if (!timeo) {
42                         err = -EINPROGRESS;
43                         break;
44                 }
45
46                 if (signal_pending(current)) {
47                         err = sock_intr_errno(timeo);
48                         break;
49                 }
50
51                 release_sock(sk);
52                 timeo = schedule_timeout(timeo);
53                 lock_sock(sk);
54                 set_current_state(TASK_INTERRUPTIBLE);
55
56                 err = sock_error(sk);
57                 if (err)
58                         break;
59         }
60
61         __set_current_state(TASK_RUNNING);
62         remove_wait_queue(sk_sleep(sk), &wait);
63         return err;
64 }
65
66 static struct proto llcp_sock_proto = {
67         .name     = "NFC_LLCP",
68         .owner    = THIS_MODULE,
69         .obj_size = sizeof(struct nfc_llcp_sock),
70 };
71
72 static int llcp_sock_bind(struct socket *sock, struct sockaddr *addr, int alen)
73 {
74         struct sock *sk = sock->sk;
75         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
76         struct nfc_llcp_local *local;
77         struct nfc_dev *dev;
78         struct sockaddr_nfc_llcp llcp_addr;
79         int len, ret = 0;
80
81         if (!addr || addr->sa_family != AF_NFC)
82                 return -EINVAL;
83
84         pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);
85
86         memset(&llcp_addr, 0, sizeof(llcp_addr));
87         len = min_t(unsigned int, sizeof(llcp_addr), alen);
88         memcpy(&llcp_addr, addr, len);
89
90         /* This is going to be a listening socket, dsap must be 0 */
91         if (llcp_addr.dsap != 0)
92                 return -EINVAL;
93
94         lock_sock(sk);
95
96         if (sk->sk_state != LLCP_CLOSED) {
97                 ret = -EBADFD;
98                 goto error;
99         }
100
101         dev = nfc_get_device(llcp_addr.dev_idx);
102         if (dev == NULL) {
103                 ret = -ENODEV;
104                 goto error;
105         }
106
107         local = nfc_llcp_find_local(dev);
108         if (local == NULL) {
109                 ret = -ENODEV;
110                 goto put_dev;
111         }
112
113         llcp_sock->dev = dev;
114         llcp_sock->local = nfc_llcp_local_get(local);
115         llcp_sock->nfc_protocol = llcp_addr.nfc_protocol;
116         llcp_sock->service_name_len = min_t(unsigned int,
117                                             llcp_addr.service_name_len,
118                                             NFC_LLCP_MAX_SERVICE_NAME);
119         llcp_sock->service_name = kmemdup(llcp_addr.service_name,
120                                           llcp_sock->service_name_len,
121                                           GFP_KERNEL);
122
123         llcp_sock->ssap = nfc_llcp_get_sdp_ssap(local, llcp_sock);
124         if (llcp_sock->ssap == LLCP_SAP_MAX) {
125                 ret = -EADDRINUSE;
126                 goto put_dev;
127         }
128
129         llcp_sock->reserved_ssap = llcp_sock->ssap;
130
131         nfc_llcp_sock_link(&local->sockets, sk);
132
133         pr_debug("Socket bound to SAP %d\n", llcp_sock->ssap);
134
135         sk->sk_state = LLCP_BOUND;
136
137 put_dev:
138         nfc_put_device(dev);
139
140 error:
141         release_sock(sk);
142         return ret;
143 }
144
145 static int llcp_raw_sock_bind(struct socket *sock, struct sockaddr *addr,
146                               int alen)
147 {
148         struct sock *sk = sock->sk;
149         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
150         struct nfc_llcp_local *local;
151         struct nfc_dev *dev;
152         struct sockaddr_nfc_llcp llcp_addr;
153         int len, ret = 0;
154
155         if (!addr || addr->sa_family != AF_NFC)
156                 return -EINVAL;
157
158         pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);
159
160         memset(&llcp_addr, 0, sizeof(llcp_addr));
161         len = min_t(unsigned int, sizeof(llcp_addr), alen);
162         memcpy(&llcp_addr, addr, len);
163
164         lock_sock(sk);
165
166         if (sk->sk_state != LLCP_CLOSED) {
167                 ret = -EBADFD;
168                 goto error;
169         }
170
171         dev = nfc_get_device(llcp_addr.dev_idx);
172         if (dev == NULL) {
173                 ret = -ENODEV;
174                 goto error;
175         }
176
177         local = nfc_llcp_find_local(dev);
178         if (local == NULL) {
179                 ret = -ENODEV;
180                 goto put_dev;
181         }
182
183         llcp_sock->dev = dev;
184         llcp_sock->local = nfc_llcp_local_get(local);
185         llcp_sock->nfc_protocol = llcp_addr.nfc_protocol;
186
187         nfc_llcp_sock_link(&local->raw_sockets, sk);
188
189         sk->sk_state = LLCP_BOUND;
190
191 put_dev:
192         nfc_put_device(dev);
193
194 error:
195         release_sock(sk);
196         return ret;
197 }
198
199 static int llcp_sock_listen(struct socket *sock, int backlog)
200 {
201         struct sock *sk = sock->sk;
202         int ret = 0;
203
204         pr_debug("sk %p backlog %d\n", sk, backlog);
205
206         lock_sock(sk);
207
208         if ((sock->type != SOCK_SEQPACKET && sock->type != SOCK_STREAM) ||
209             sk->sk_state != LLCP_BOUND) {
210                 ret = -EBADFD;
211                 goto error;
212         }
213
214         sk->sk_max_ack_backlog = backlog;
215         sk->sk_ack_backlog = 0;
216
217         pr_debug("Socket listening\n");
218         sk->sk_state = LLCP_LISTEN;
219
220 error:
221         release_sock(sk);
222
223         return ret;
224 }
225
226 static int nfc_llcp_setsockopt(struct socket *sock, int level, int optname,
227                                char __user *optval, unsigned int optlen)
228 {
229         struct sock *sk = sock->sk;
230         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
231         u32 opt;
232         int err = 0;
233
234         pr_debug("%p optname %d\n", sk, optname);
235
236         if (level != SOL_NFC)
237                 return -ENOPROTOOPT;
238
239         lock_sock(sk);
240
241         switch (optname) {
242         case NFC_LLCP_RW:
243                 if (sk->sk_state == LLCP_CONNECTED ||
244                     sk->sk_state == LLCP_BOUND ||
245                     sk->sk_state == LLCP_LISTEN) {
246                         err = -EINVAL;
247                         break;
248                 }
249
250                 if (get_user(opt, (u32 __user *) optval)) {
251                         err = -EFAULT;
252                         break;
253                 }
254
255                 if (opt > LLCP_MAX_RW) {
256                         err = -EINVAL;
257                         break;
258                 }
259
260                 llcp_sock->rw = (u8) opt;
261
262                 break;
263
264         case NFC_LLCP_MIUX:
265                 if (sk->sk_state == LLCP_CONNECTED ||
266                     sk->sk_state == LLCP_BOUND ||
267                     sk->sk_state == LLCP_LISTEN) {
268                         err = -EINVAL;
269                         break;
270                 }
271
272                 if (get_user(opt, (u32 __user *) optval)) {
273                         err = -EFAULT;
274                         break;
275                 }
276
277                 if (opt > LLCP_MAX_MIUX) {
278                         err = -EINVAL;
279                         break;
280                 }
281
282                 llcp_sock->miux = (u16) opt;
283
284                 break;
285
286         default:
287                 err = -ENOPROTOOPT;
288                 break;
289         }
290
291         release_sock(sk);
292
293         pr_debug("%p rw %d miux %d\n", llcp_sock,
294                  llcp_sock->rw, llcp_sock->miux);
295
296         return err;
297 }
298
299 static int nfc_llcp_getsockopt(struct socket *sock, int level, int optname,
300                                char __user *optval, int __user *optlen)
301 {
302         struct sock *sk = sock->sk;
303         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
304         int len, err = 0;
305
306         pr_debug("%p optname %d\n", sk, optname);
307
308         if (level != SOL_NFC)
309                 return -ENOPROTOOPT;
310
311         if (get_user(len, optlen))
312                 return -EFAULT;
313
314         len = min_t(u32, len, sizeof(u32));
315
316         lock_sock(sk);
317
318         switch (optname) {
319         case NFC_LLCP_RW:
320                 if (put_user(llcp_sock->rw, (u32 __user *) optval))
321                         err = -EFAULT;
322
323                 break;
324
325         case NFC_LLCP_MIUX:
326                 if (put_user(llcp_sock->miux, (u32 __user *) optval))
327                         err = -EFAULT;
328
329                 break;
330
331         default:
332                 err = -ENOPROTOOPT;
333                 break;
334         }
335
336         release_sock(sk);
337
338         if (put_user(len, optlen))
339                 return -EFAULT;
340
341         return err;
342 }
343
344 void nfc_llcp_accept_unlink(struct sock *sk)
345 {
346         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
347
348         pr_debug("state %d\n", sk->sk_state);
349
350         list_del_init(&llcp_sock->accept_queue);
351         sk_acceptq_removed(llcp_sock->parent);
352         llcp_sock->parent = NULL;
353
354         sock_put(sk);
355 }
356
357 void nfc_llcp_accept_enqueue(struct sock *parent, struct sock *sk)
358 {
359         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
360         struct nfc_llcp_sock *llcp_sock_parent = nfc_llcp_sock(parent);
361
362         /* Lock will be free from unlink */
363         sock_hold(sk);
364
365         list_add_tail(&llcp_sock->accept_queue,
366                       &llcp_sock_parent->accept_queue);
367         llcp_sock->parent = parent;
368         sk_acceptq_added(parent);
369 }
370
371 struct sock *nfc_llcp_accept_dequeue(struct sock *parent,
372                                      struct socket *newsock)
373 {
374         struct nfc_llcp_sock *lsk, *n, *llcp_parent;
375         struct sock *sk;
376
377         llcp_parent = nfc_llcp_sock(parent);
378
379         list_for_each_entry_safe(lsk, n, &llcp_parent->accept_queue,
380                                  accept_queue) {
381                 sk = &lsk->sk;
382                 lock_sock(sk);
383
384                 if (sk->sk_state == LLCP_CLOSED) {
385                         release_sock(sk);
386                         nfc_llcp_accept_unlink(sk);
387                         continue;
388                 }
389
390                 if (sk->sk_state == LLCP_CONNECTED || !newsock) {
391                         nfc_llcp_accept_unlink(sk);
392                         if (newsock)
393                                 sock_graft(sk, newsock);
394
395                         release_sock(sk);
396
397                         pr_debug("Returning sk state %d\n", sk->sk_state);
398
399                         sk_acceptq_removed(parent);
400
401                         return sk;
402                 }
403
404                 release_sock(sk);
405         }
406
407         return NULL;
408 }
409
410 static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
411                             int flags)
412 {
413         DECLARE_WAITQUEUE(wait, current);
414         struct sock *sk = sock->sk, *new_sk;
415         long timeo;
416         int ret = 0;
417
418         pr_debug("parent %p\n", sk);
419
420         lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
421
422         if (sk->sk_state != LLCP_LISTEN) {
423                 ret = -EBADFD;
424                 goto error;
425         }
426
427         timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
428
429         /* Wait for an incoming connection. */
430         add_wait_queue_exclusive(sk_sleep(sk), &wait);
431         while (!(new_sk = nfc_llcp_accept_dequeue(sk, newsock))) {
432                 set_current_state(TASK_INTERRUPTIBLE);
433
434                 if (!timeo) {
435                         ret = -EAGAIN;
436                         break;
437                 }
438
439                 if (signal_pending(current)) {
440                         ret = sock_intr_errno(timeo);
441                         break;
442                 }
443
444                 release_sock(sk);
445                 timeo = schedule_timeout(timeo);
446                 lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
447         }
448         __set_current_state(TASK_RUNNING);
449         remove_wait_queue(sk_sleep(sk), &wait);
450
451         if (ret)
452                 goto error;
453
454         newsock->state = SS_CONNECTED;
455
456         pr_debug("new socket %p\n", new_sk);
457
458 error:
459         release_sock(sk);
460
461         return ret;
462 }
463
464 static int llcp_sock_getname(struct socket *sock, struct sockaddr *uaddr,
465                              int *len, int peer)
466 {
467         struct sock *sk = sock->sk;
468         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
469         DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, llcp_addr, uaddr);
470
471         if (llcp_sock == NULL || llcp_sock->dev == NULL)
472                 return -EBADFD;
473
474         pr_debug("%p %d %d %d\n", sk, llcp_sock->target_idx,
475                  llcp_sock->dsap, llcp_sock->ssap);
476
477         uaddr->sa_family = AF_NFC;
478
479         *len = sizeof(struct sockaddr_nfc_llcp);
480
481         llcp_addr->dev_idx = llcp_sock->dev->idx;
482         llcp_addr->target_idx = llcp_sock->target_idx;
483         llcp_addr->dsap = llcp_sock->dsap;
484         llcp_addr->ssap = llcp_sock->ssap;
485         llcp_addr->service_name_len = llcp_sock->service_name_len;
486         memcpy(llcp_addr->service_name, llcp_sock->service_name,
487                llcp_addr->service_name_len);
488
489         return 0;
490 }
491
492 static inline unsigned int llcp_accept_poll(struct sock *parent)
493 {
494         struct nfc_llcp_sock *llcp_sock, *n, *parent_sock;
495         struct sock *sk;
496
497         parent_sock = nfc_llcp_sock(parent);
498
499         list_for_each_entry_safe(llcp_sock, n, &parent_sock->accept_queue,
500                                  accept_queue) {
501                 sk = &llcp_sock->sk;
502
503                 if (sk->sk_state == LLCP_CONNECTED)
504                         return POLLIN | POLLRDNORM;
505         }
506
507         return 0;
508 }
509
510 static unsigned int llcp_sock_poll(struct file *file, struct socket *sock,
511                                    poll_table *wait)
512 {
513         struct sock *sk = sock->sk;
514         unsigned int mask = 0;
515
516         pr_debug("%p\n", sk);
517
518         sock_poll_wait(file, sk_sleep(sk), wait);
519
520         if (sk->sk_state == LLCP_LISTEN)
521                 return llcp_accept_poll(sk);
522
523         if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
524                 mask |= POLLERR |
525                         sock_flag(sk, SOCK_SELECT_ERR_QUEUE) ? POLLPRI : 0;
526
527         if (!skb_queue_empty(&sk->sk_receive_queue))
528                 mask |= POLLIN | POLLRDNORM;
529
530         if (sk->sk_state == LLCP_CLOSED)
531                 mask |= POLLHUP;
532
533         if (sk->sk_shutdown & RCV_SHUTDOWN)
534                 mask |= POLLRDHUP | POLLIN | POLLRDNORM;
535
536         if (sk->sk_shutdown == SHUTDOWN_MASK)
537                 mask |= POLLHUP;
538
539         if (sock_writeable(sk))
540                 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
541         else
542                 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
543
544         pr_debug("mask 0x%x\n", mask);
545
546         return mask;
547 }
548
549 static int llcp_sock_release(struct socket *sock)
550 {
551         struct sock *sk = sock->sk;
552         struct nfc_llcp_local *local;
553         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
554         int err = 0;
555
556         if (!sk)
557                 return 0;
558
559         pr_debug("%p\n", sk);
560
561         local = llcp_sock->local;
562         if (local == NULL) {
563                 err = -ENODEV;
564                 goto out;
565         }
566
567         lock_sock(sk);
568
569         /* Send a DISC */
570         if (sk->sk_state == LLCP_CONNECTED)
571                 nfc_llcp_disconnect(llcp_sock);
572
573         if (sk->sk_state == LLCP_LISTEN) {
574                 struct nfc_llcp_sock *lsk, *n;
575                 struct sock *accept_sk;
576
577                 list_for_each_entry_safe(lsk, n, &llcp_sock->accept_queue,
578                                          accept_queue) {
579                         accept_sk = &lsk->sk;
580                         lock_sock(accept_sk);
581
582                         nfc_llcp_disconnect(lsk);
583                         nfc_llcp_accept_unlink(accept_sk);
584
585                         release_sock(accept_sk);
586
587                         sock_orphan(accept_sk);
588                 }
589         }
590
591         if (llcp_sock->reserved_ssap < LLCP_SAP_MAX)
592                 nfc_llcp_put_ssap(llcp_sock->local, llcp_sock->ssap);
593
594         release_sock(sk);
595
596         if (sock->type == SOCK_RAW)
597                 nfc_llcp_sock_unlink(&local->raw_sockets, sk);
598         else
599                 nfc_llcp_sock_unlink(&local->sockets, sk);
600
601 out:
602         sock_orphan(sk);
603         sock_put(sk);
604
605         return err;
606 }
607
608 static int llcp_sock_connect(struct socket *sock, struct sockaddr *_addr,
609                              int len, int flags)
610 {
611         struct sock *sk = sock->sk;
612         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
613         struct sockaddr_nfc_llcp *addr = (struct sockaddr_nfc_llcp *)_addr;
614         struct nfc_dev *dev;
615         struct nfc_llcp_local *local;
616         int ret = 0;
617
618         pr_debug("sock %p sk %p flags 0x%x\n", sock, sk, flags);
619
620         if (!addr || len < sizeof(struct sockaddr_nfc) ||
621             addr->sa_family != AF_NFC)
622                 return -EINVAL;
623
624         if (addr->service_name_len == 0 && addr->dsap == 0)
625                 return -EINVAL;
626
627         pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr->dev_idx,
628                  addr->target_idx, addr->nfc_protocol);
629
630         lock_sock(sk);
631
632         if (sk->sk_state == LLCP_CONNECTED) {
633                 ret = -EISCONN;
634                 goto error;
635         }
636
637         dev = nfc_get_device(addr->dev_idx);
638         if (dev == NULL) {
639                 ret = -ENODEV;
640                 goto error;
641         }
642
643         local = nfc_llcp_find_local(dev);
644         if (local == NULL) {
645                 ret = -ENODEV;
646                 goto put_dev;
647         }
648
649         device_lock(&dev->dev);
650         if (dev->dep_link_up == false) {
651                 ret = -ENOLINK;
652                 device_unlock(&dev->dev);
653                 goto put_dev;
654         }
655         device_unlock(&dev->dev);
656
657         if (local->rf_mode == NFC_RF_INITIATOR &&
658             addr->target_idx != local->target_idx) {
659                 ret = -ENOLINK;
660                 goto put_dev;
661         }
662
663         llcp_sock->dev = dev;
664         llcp_sock->local = nfc_llcp_local_get(local);
665         llcp_sock->remote_miu = llcp_sock->local->remote_miu;
666         llcp_sock->ssap = nfc_llcp_get_local_ssap(local);
667         if (llcp_sock->ssap == LLCP_SAP_MAX) {
668                 ret = -ENOMEM;
669                 goto put_dev;
670         }
671
672         llcp_sock->reserved_ssap = llcp_sock->ssap;
673
674         if (addr->service_name_len == 0)
675                 llcp_sock->dsap = addr->dsap;
676         else
677                 llcp_sock->dsap = LLCP_SAP_SDP;
678         llcp_sock->nfc_protocol = addr->nfc_protocol;
679         llcp_sock->service_name_len = min_t(unsigned int,
680                                             addr->service_name_len,
681                                             NFC_LLCP_MAX_SERVICE_NAME);
682         llcp_sock->service_name = kmemdup(addr->service_name,
683                                           llcp_sock->service_name_len,
684                                           GFP_KERNEL);
685
686         nfc_llcp_sock_link(&local->connecting_sockets, sk);
687
688         ret = nfc_llcp_send_connect(llcp_sock);
689         if (ret)
690                 goto sock_unlink;
691
692         ret = sock_wait_state(sk, LLCP_CONNECTED,
693                               sock_sndtimeo(sk, flags & O_NONBLOCK));
694         if (ret)
695                 goto sock_unlink;
696
697         release_sock(sk);
698
699         return 0;
700
701 sock_unlink:
702         nfc_llcp_put_ssap(local, llcp_sock->ssap);
703
704         nfc_llcp_sock_unlink(&local->connecting_sockets, sk);
705
706 put_dev:
707         nfc_put_device(dev);
708
709 error:
710         release_sock(sk);
711         return ret;
712 }
713
714 static int llcp_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
715                              struct msghdr *msg, size_t len)
716 {
717         struct sock *sk = sock->sk;
718         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
719         int ret;
720
721         pr_debug("sock %p sk %p", sock, sk);
722
723         ret = sock_error(sk);
724         if (ret)
725                 return ret;
726
727         if (msg->msg_flags & MSG_OOB)
728                 return -EOPNOTSUPP;
729
730         lock_sock(sk);
731
732         if (sk->sk_type == SOCK_DGRAM) {
733                 struct sockaddr_nfc_llcp *addr =
734                         (struct sockaddr_nfc_llcp *)msg->msg_name;
735
736                 if (msg->msg_namelen < sizeof(*addr)) {
737                         release_sock(sk);
738                         return -EINVAL;
739                 }
740
741                 release_sock(sk);
742
743                 return nfc_llcp_send_ui_frame(llcp_sock, addr->dsap, addr->ssap,
744                                               msg, len);
745         }
746
747         if (sk->sk_state != LLCP_CONNECTED) {
748                 release_sock(sk);
749                 return -ENOTCONN;
750         }
751
752         release_sock(sk);
753
754         return nfc_llcp_send_i_frame(llcp_sock, msg, len);
755 }
756
757 static int llcp_sock_recvmsg(struct kiocb *iocb, struct socket *sock,
758                              struct msghdr *msg, size_t len, int flags)
759 {
760         int noblock = flags & MSG_DONTWAIT;
761         struct sock *sk = sock->sk;
762         unsigned int copied, rlen;
763         struct sk_buff *skb, *cskb;
764         int err = 0;
765
766         pr_debug("%p %zu\n", sk, len);
767
768         lock_sock(sk);
769
770         if (sk->sk_state == LLCP_CLOSED &&
771             skb_queue_empty(&sk->sk_receive_queue)) {
772                 release_sock(sk);
773                 return 0;
774         }
775
776         release_sock(sk);
777
778         if (flags & (MSG_OOB))
779                 return -EOPNOTSUPP;
780
781         skb = skb_recv_datagram(sk, flags, noblock, &err);
782         if (!skb) {
783                 pr_err("Recv datagram failed state %d %d %d",
784                        sk->sk_state, err, sock_error(sk));
785
786                 if (sk->sk_shutdown & RCV_SHUTDOWN)
787                         return 0;
788
789                 return err;
790         }
791
792         rlen = skb->len;                /* real length of skb */
793         copied = min_t(unsigned int, rlen, len);
794
795         cskb = skb;
796         if (skb_copy_datagram_iovec(cskb, 0, msg->msg_iov, copied)) {
797                 if (!(flags & MSG_PEEK))
798                         skb_queue_head(&sk->sk_receive_queue, skb);
799                 return -EFAULT;
800         }
801
802         sock_recv_timestamp(msg, sk, skb);
803
804         if (sk->sk_type == SOCK_DGRAM && msg->msg_name) {
805                 struct nfc_llcp_ui_cb *ui_cb = nfc_llcp_ui_skb_cb(skb);
806                 struct sockaddr_nfc_llcp *sockaddr =
807                         (struct sockaddr_nfc_llcp *) msg->msg_name;
808
809                 msg->msg_namelen = sizeof(struct sockaddr_nfc_llcp);
810
811                 pr_debug("Datagram socket %d %d\n", ui_cb->dsap, ui_cb->ssap);
812
813                 sockaddr->sa_family = AF_NFC;
814                 sockaddr->nfc_protocol = NFC_PROTO_NFC_DEP;
815                 sockaddr->dsap = ui_cb->dsap;
816                 sockaddr->ssap = ui_cb->ssap;
817         }
818
819         /* Mark read part of skb as used */
820         if (!(flags & MSG_PEEK)) {
821
822                 /* SOCK_STREAM: re-queue skb if it contains unreceived data */
823                 if (sk->sk_type == SOCK_STREAM ||
824                     sk->sk_type == SOCK_DGRAM ||
825                     sk->sk_type == SOCK_RAW) {
826                         skb_pull(skb, copied);
827                         if (skb->len) {
828                                 skb_queue_head(&sk->sk_receive_queue, skb);
829                                 goto done;
830                         }
831                 }
832
833                 kfree_skb(skb);
834         }
835
836         /* XXX Queue backlogged skbs */
837
838 done:
839         /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
840         if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
841                 copied = rlen;
842
843         return copied;
844 }
845
846 static const struct proto_ops llcp_sock_ops = {
847         .family         = PF_NFC,
848         .owner          = THIS_MODULE,
849         .bind           = llcp_sock_bind,
850         .connect        = llcp_sock_connect,
851         .release        = llcp_sock_release,
852         .socketpair     = sock_no_socketpair,
853         .accept         = llcp_sock_accept,
854         .getname        = llcp_sock_getname,
855         .poll           = llcp_sock_poll,
856         .ioctl          = sock_no_ioctl,
857         .listen         = llcp_sock_listen,
858         .shutdown       = sock_no_shutdown,
859         .setsockopt     = nfc_llcp_setsockopt,
860         .getsockopt     = nfc_llcp_getsockopt,
861         .sendmsg        = llcp_sock_sendmsg,
862         .recvmsg        = llcp_sock_recvmsg,
863         .mmap           = sock_no_mmap,
864 };
865
866 static const struct proto_ops llcp_rawsock_ops = {
867         .family         = PF_NFC,
868         .owner          = THIS_MODULE,
869         .bind           = llcp_raw_sock_bind,
870         .connect        = sock_no_connect,
871         .release        = llcp_sock_release,
872         .socketpair     = sock_no_socketpair,
873         .accept         = sock_no_accept,
874         .getname        = llcp_sock_getname,
875         .poll           = llcp_sock_poll,
876         .ioctl          = sock_no_ioctl,
877         .listen         = sock_no_listen,
878         .shutdown       = sock_no_shutdown,
879         .setsockopt     = sock_no_setsockopt,
880         .getsockopt     = sock_no_getsockopt,
881         .sendmsg        = sock_no_sendmsg,
882         .recvmsg        = llcp_sock_recvmsg,
883         .mmap           = sock_no_mmap,
884 };
885
886 static void llcp_sock_destruct(struct sock *sk)
887 {
888         struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
889
890         pr_debug("%p\n", sk);
891
892         if (sk->sk_state == LLCP_CONNECTED)
893                 nfc_put_device(llcp_sock->dev);
894
895         skb_queue_purge(&sk->sk_receive_queue);
896
897         nfc_llcp_sock_free(llcp_sock);
898
899         if (!sock_flag(sk, SOCK_DEAD)) {
900                 pr_err("Freeing alive NFC LLCP socket %p\n", sk);
901                 return;
902         }
903 }
904
905 struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp)
906 {
907         struct sock *sk;
908         struct nfc_llcp_sock *llcp_sock;
909
910         sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto);
911         if (!sk)
912                 return NULL;
913
914         llcp_sock = nfc_llcp_sock(sk);
915
916         sock_init_data(sock, sk);
917         sk->sk_state = LLCP_CLOSED;
918         sk->sk_protocol = NFC_SOCKPROTO_LLCP;
919         sk->sk_type = type;
920         sk->sk_destruct = llcp_sock_destruct;
921
922         llcp_sock->ssap = 0;
923         llcp_sock->dsap = LLCP_SAP_SDP;
924         llcp_sock->rw = LLCP_MAX_RW + 1;
925         llcp_sock->miux = LLCP_MAX_MIUX + 1;
926         llcp_sock->remote_rw = LLCP_DEFAULT_RW;
927         llcp_sock->remote_miu = LLCP_DEFAULT_MIU;
928         llcp_sock->send_n = llcp_sock->send_ack_n = 0;
929         llcp_sock->recv_n = llcp_sock->recv_ack_n = 0;
930         llcp_sock->remote_ready = 1;
931         llcp_sock->reserved_ssap = LLCP_SAP_MAX;
932         skb_queue_head_init(&llcp_sock->tx_queue);
933         skb_queue_head_init(&llcp_sock->tx_pending_queue);
934         INIT_LIST_HEAD(&llcp_sock->accept_queue);
935
936         if (sock != NULL)
937                 sock->state = SS_UNCONNECTED;
938
939         return sk;
940 }
941
942 void nfc_llcp_sock_free(struct nfc_llcp_sock *sock)
943 {
944         kfree(sock->service_name);
945
946         skb_queue_purge(&sock->tx_queue);
947         skb_queue_purge(&sock->tx_pending_queue);
948
949         list_del_init(&sock->accept_queue);
950
951         sock->parent = NULL;
952
953         nfc_llcp_local_put(sock->local);
954 }
955
956 static int llcp_sock_create(struct net *net, struct socket *sock,
957                             const struct nfc_protocol *nfc_proto)
958 {
959         struct sock *sk;
960
961         pr_debug("%p\n", sock);
962
963         if (sock->type != SOCK_STREAM &&
964             sock->type != SOCK_DGRAM &&
965             sock->type != SOCK_RAW)
966                 return -ESOCKTNOSUPPORT;
967
968         if (sock->type == SOCK_RAW)
969                 sock->ops = &llcp_rawsock_ops;
970         else
971                 sock->ops = &llcp_sock_ops;
972
973         sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC);
974         if (sk == NULL)
975                 return -ENOMEM;
976
977         return 0;
978 }
979
980 static const struct nfc_protocol llcp_nfc_proto = {
981         .id       = NFC_SOCKPROTO_LLCP,
982         .proto    = &llcp_sock_proto,
983         .owner    = THIS_MODULE,
984         .create   = llcp_sock_create
985 };
986
987 int __init nfc_llcp_sock_init(void)
988 {
989         return nfc_proto_register(&llcp_nfc_proto);
990 }
991
992 void nfc_llcp_sock_exit(void)
993 {
994         nfc_proto_unregister(&llcp_nfc_proto);
995 }