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