NFC: Socket linked list
[deliverable/linux.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 pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);
82
83 if (!addr || addr->sa_family != AF_NFC)
84 return -EINVAL;
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_MAX_SAP)
125 goto put_dev;
126
127 nfc_llcp_sock_link(&local->sockets, sk);
128
129 pr_debug("Socket bound to SAP %d\n", llcp_sock->ssap);
130
131 sk->sk_state = LLCP_BOUND;
132
133 put_dev:
134 nfc_put_device(dev);
135
136 error:
137 release_sock(sk);
138 return ret;
139 }
140
141 static int llcp_sock_listen(struct socket *sock, int backlog)
142 {
143 struct sock *sk = sock->sk;
144 int ret = 0;
145
146 pr_debug("sk %p backlog %d\n", sk, backlog);
147
148 lock_sock(sk);
149
150 if ((sock->type != SOCK_SEQPACKET && sock->type != SOCK_STREAM)
151 || sk->sk_state != LLCP_BOUND) {
152 ret = -EBADFD;
153 goto error;
154 }
155
156 sk->sk_max_ack_backlog = backlog;
157 sk->sk_ack_backlog = 0;
158
159 pr_debug("Socket listening\n");
160 sk->sk_state = LLCP_LISTEN;
161
162 error:
163 release_sock(sk);
164
165 return ret;
166 }
167
168 void nfc_llcp_accept_unlink(struct sock *sk)
169 {
170 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
171
172 pr_debug("state %d\n", sk->sk_state);
173
174 list_del_init(&llcp_sock->accept_queue);
175 sk_acceptq_removed(llcp_sock->parent);
176 llcp_sock->parent = NULL;
177
178 sock_put(sk);
179 }
180
181 void nfc_llcp_accept_enqueue(struct sock *parent, struct sock *sk)
182 {
183 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
184 struct nfc_llcp_sock *llcp_sock_parent = nfc_llcp_sock(parent);
185
186 /* Lock will be free from unlink */
187 sock_hold(sk);
188
189 list_add_tail(&llcp_sock->accept_queue,
190 &llcp_sock_parent->accept_queue);
191 llcp_sock->parent = parent;
192 sk_acceptq_added(parent);
193 }
194
195 struct sock *nfc_llcp_accept_dequeue(struct sock *parent,
196 struct socket *newsock)
197 {
198 struct nfc_llcp_sock *lsk, *n, *llcp_parent;
199 struct sock *sk;
200
201 llcp_parent = nfc_llcp_sock(parent);
202
203 list_for_each_entry_safe(lsk, n, &llcp_parent->accept_queue,
204 accept_queue) {
205 sk = &lsk->sk;
206 lock_sock(sk);
207
208 if (sk->sk_state == LLCP_CLOSED) {
209 release_sock(sk);
210 nfc_llcp_accept_unlink(sk);
211 continue;
212 }
213
214 if (sk->sk_state == LLCP_CONNECTED || !newsock) {
215 nfc_llcp_accept_unlink(sk);
216 if (newsock)
217 sock_graft(sk, newsock);
218
219 release_sock(sk);
220
221 pr_debug("Returning sk state %d\n", sk->sk_state);
222
223 return sk;
224 }
225
226 release_sock(sk);
227 }
228
229 return NULL;
230 }
231
232 static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
233 int flags)
234 {
235 DECLARE_WAITQUEUE(wait, current);
236 struct sock *sk = sock->sk, *new_sk;
237 long timeo;
238 int ret = 0;
239
240 pr_debug("parent %p\n", sk);
241
242 lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
243
244 if (sk->sk_state != LLCP_LISTEN) {
245 ret = -EBADFD;
246 goto error;
247 }
248
249 timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
250
251 /* Wait for an incoming connection. */
252 add_wait_queue_exclusive(sk_sleep(sk), &wait);
253 while (!(new_sk = nfc_llcp_accept_dequeue(sk, newsock))) {
254 set_current_state(TASK_INTERRUPTIBLE);
255
256 if (!timeo) {
257 ret = -EAGAIN;
258 break;
259 }
260
261 if (signal_pending(current)) {
262 ret = sock_intr_errno(timeo);
263 break;
264 }
265
266 release_sock(sk);
267 timeo = schedule_timeout(timeo);
268 lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
269 }
270 __set_current_state(TASK_RUNNING);
271 remove_wait_queue(sk_sleep(sk), &wait);
272
273 if (ret)
274 goto error;
275
276 newsock->state = SS_CONNECTED;
277
278 pr_debug("new socket %p\n", new_sk);
279
280 error:
281 release_sock(sk);
282
283 return ret;
284 }
285
286 static int llcp_sock_getname(struct socket *sock, struct sockaddr *addr,
287 int *len, int peer)
288 {
289 struct sockaddr_nfc_llcp *llcp_addr = (struct sockaddr_nfc_llcp *)addr;
290 struct sock *sk = sock->sk;
291 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
292
293 pr_debug("%p\n", sk);
294
295 addr->sa_family = AF_NFC;
296 *len = sizeof(struct sockaddr_nfc_llcp);
297
298 llcp_addr->dev_idx = llcp_sock->dev->idx;
299 llcp_addr->dsap = llcp_sock->dsap;
300 llcp_addr->ssap = llcp_sock->ssap;
301 llcp_addr->service_name_len = llcp_sock->service_name_len;
302 memcpy(llcp_addr->service_name, llcp_sock->service_name,
303 llcp_addr->service_name_len);
304
305 return 0;
306 }
307
308 static inline unsigned int llcp_accept_poll(struct sock *parent)
309 {
310 struct nfc_llcp_sock *llcp_sock, *n, *parent_sock;
311 struct sock *sk;
312
313 parent_sock = nfc_llcp_sock(parent);
314
315 list_for_each_entry_safe(llcp_sock, n, &parent_sock->accept_queue,
316 accept_queue) {
317 sk = &llcp_sock->sk;
318
319 if (sk->sk_state == LLCP_CONNECTED)
320 return POLLIN | POLLRDNORM;
321 }
322
323 return 0;
324 }
325
326 static unsigned int llcp_sock_poll(struct file *file, struct socket *sock,
327 poll_table *wait)
328 {
329 struct sock *sk = sock->sk;
330 unsigned int mask = 0;
331
332 pr_debug("%p\n", sk);
333
334 sock_poll_wait(file, sk_sleep(sk), wait);
335
336 if (sk->sk_state == LLCP_LISTEN)
337 return llcp_accept_poll(sk);
338
339 if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
340 mask |= POLLERR;
341
342 if (!skb_queue_empty(&sk->sk_receive_queue))
343 mask |= POLLIN | POLLRDNORM;
344
345 if (sk->sk_state == LLCP_CLOSED)
346 mask |= POLLHUP;
347
348 if (sk->sk_shutdown & RCV_SHUTDOWN)
349 mask |= POLLRDHUP | POLLIN | POLLRDNORM;
350
351 if (sk->sk_shutdown == SHUTDOWN_MASK)
352 mask |= POLLHUP;
353
354 if (sock_writeable(sk))
355 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
356 else
357 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
358
359 pr_debug("mask 0x%x\n", mask);
360
361 return mask;
362 }
363
364 static int llcp_sock_release(struct socket *sock)
365 {
366 struct sock *sk = sock->sk;
367 struct nfc_llcp_local *local;
368 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
369 int err = 0;
370
371 if (!sk)
372 return 0;
373
374 pr_debug("%p\n", sk);
375
376 local = llcp_sock->local;
377 if (local == NULL) {
378 err = -ENODEV;
379 goto out;
380 }
381
382 lock_sock(sk);
383
384 /* Send a DISC */
385 if (sk->sk_state == LLCP_CONNECTED)
386 nfc_llcp_disconnect(llcp_sock);
387
388 if (sk->sk_state == LLCP_LISTEN) {
389 struct nfc_llcp_sock *lsk, *n;
390 struct sock *accept_sk;
391
392 list_for_each_entry_safe(lsk, n, &llcp_sock->accept_queue,
393 accept_queue) {
394 accept_sk = &lsk->sk;
395 lock_sock(accept_sk);
396
397 nfc_llcp_disconnect(lsk);
398 nfc_llcp_accept_unlink(accept_sk);
399
400 release_sock(accept_sk);
401
402 sock_orphan(accept_sk);
403 }
404 }
405
406 nfc_llcp_put_ssap(llcp_sock->local, llcp_sock->ssap);
407
408 release_sock(sk);
409
410 nfc_llcp_sock_unlink(&local->sockets, sk);
411
412 out:
413 sock_orphan(sk);
414 sock_put(sk);
415
416 return err;
417 }
418
419 static int llcp_sock_connect(struct socket *sock, struct sockaddr *_addr,
420 int len, int flags)
421 {
422 struct sock *sk = sock->sk;
423 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
424 struct sockaddr_nfc_llcp *addr = (struct sockaddr_nfc_llcp *)_addr;
425 struct nfc_dev *dev;
426 struct nfc_llcp_local *local;
427 int ret = 0;
428
429 pr_debug("sock %p sk %p flags 0x%x\n", sock, sk, flags);
430
431 if (!addr || len < sizeof(struct sockaddr_nfc) ||
432 addr->sa_family != AF_NFC) {
433 pr_err("Invalid socket\n");
434 return -EINVAL;
435 }
436
437 if (addr->service_name_len == 0 && addr->dsap == 0) {
438 pr_err("Missing service name or dsap\n");
439 return -EINVAL;
440 }
441
442 pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr->dev_idx,
443 addr->target_idx, addr->nfc_protocol);
444
445 lock_sock(sk);
446
447 if (sk->sk_state == LLCP_CONNECTED) {
448 ret = -EISCONN;
449 goto error;
450 }
451
452 dev = nfc_get_device(addr->dev_idx);
453 if (dev == NULL) {
454 ret = -ENODEV;
455 goto error;
456 }
457
458 local = nfc_llcp_find_local(dev);
459 if (local == NULL) {
460 ret = -ENODEV;
461 goto put_dev;
462 }
463
464 device_lock(&dev->dev);
465 if (dev->dep_link_up == false) {
466 ret = -ENOLINK;
467 device_unlock(&dev->dev);
468 goto put_dev;
469 }
470 device_unlock(&dev->dev);
471
472 if (local->rf_mode == NFC_RF_INITIATOR &&
473 addr->target_idx != local->target_idx) {
474 ret = -ENOLINK;
475 goto put_dev;
476 }
477
478 llcp_sock->dev = dev;
479 llcp_sock->local = nfc_llcp_local_get(local);
480 llcp_sock->ssap = nfc_llcp_get_local_ssap(local);
481 if (llcp_sock->ssap == LLCP_SAP_MAX) {
482 ret = -ENOMEM;
483 goto put_dev;
484 }
485 if (addr->service_name_len == 0)
486 llcp_sock->dsap = addr->dsap;
487 else
488 llcp_sock->dsap = LLCP_SAP_SDP;
489 llcp_sock->nfc_protocol = addr->nfc_protocol;
490 llcp_sock->service_name_len = min_t(unsigned int,
491 addr->service_name_len,
492 NFC_LLCP_MAX_SERVICE_NAME);
493 llcp_sock->service_name = kmemdup(addr->service_name,
494 llcp_sock->service_name_len,
495 GFP_KERNEL);
496
497 nfc_llcp_sock_link(&local->connecting_sockets, sk);
498
499 ret = nfc_llcp_send_connect(llcp_sock);
500 if (ret)
501 goto sock_unlink;
502
503 ret = sock_wait_state(sk, LLCP_CONNECTED,
504 sock_sndtimeo(sk, flags & O_NONBLOCK));
505 if (ret)
506 goto sock_unlink;
507
508 release_sock(sk);
509
510 return 0;
511
512 sock_unlink:
513 nfc_llcp_put_ssap(local, llcp_sock->ssap);
514
515 nfc_llcp_sock_unlink(&local->connecting_sockets, sk);
516
517 put_dev:
518 nfc_put_device(dev);
519
520 error:
521 release_sock(sk);
522 return ret;
523 }
524
525 static int llcp_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
526 struct msghdr *msg, size_t len)
527 {
528 struct sock *sk = sock->sk;
529 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
530 int ret;
531
532 pr_debug("sock %p sk %p", sock, sk);
533
534 ret = sock_error(sk);
535 if (ret)
536 return ret;
537
538 if (msg->msg_flags & MSG_OOB)
539 return -EOPNOTSUPP;
540
541 lock_sock(sk);
542
543 if (sk->sk_state != LLCP_CONNECTED) {
544 release_sock(sk);
545 return -ENOTCONN;
546 }
547
548 release_sock(sk);
549
550 return nfc_llcp_send_i_frame(llcp_sock, msg, len);
551 }
552
553 static int llcp_sock_recvmsg(struct kiocb *iocb, struct socket *sock,
554 struct msghdr *msg, size_t len, int flags)
555 {
556 int noblock = flags & MSG_DONTWAIT;
557 struct sock *sk = sock->sk;
558 unsigned int copied, rlen;
559 struct sk_buff *skb, *cskb;
560 int err = 0;
561
562 pr_debug("%p %zu\n", sk, len);
563
564 lock_sock(sk);
565
566 if (sk->sk_state == LLCP_CLOSED &&
567 skb_queue_empty(&sk->sk_receive_queue)) {
568 release_sock(sk);
569 return 0;
570 }
571
572 release_sock(sk);
573
574 if (flags & (MSG_OOB))
575 return -EOPNOTSUPP;
576
577 skb = skb_recv_datagram(sk, flags, noblock, &err);
578 if (!skb) {
579 pr_err("Recv datagram failed state %d %d %d",
580 sk->sk_state, err, sock_error(sk));
581
582 if (sk->sk_shutdown & RCV_SHUTDOWN)
583 return 0;
584
585 return err;
586 }
587
588 rlen = skb->len; /* real length of skb */
589 copied = min_t(unsigned int, rlen, len);
590
591 cskb = skb;
592 if (memcpy_toiovec(msg->msg_iov, cskb->data, copied)) {
593 if (!(flags & MSG_PEEK))
594 skb_queue_head(&sk->sk_receive_queue, skb);
595 return -EFAULT;
596 }
597
598 /* Mark read part of skb as used */
599 if (!(flags & MSG_PEEK)) {
600
601 /* SOCK_STREAM: re-queue skb if it contains unreceived data */
602 if (sk->sk_type == SOCK_STREAM) {
603 skb_pull(skb, copied);
604 if (skb->len) {
605 skb_queue_head(&sk->sk_receive_queue, skb);
606 goto done;
607 }
608 }
609
610 kfree_skb(skb);
611 }
612
613 /* XXX Queue backlogged skbs */
614
615 done:
616 /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
617 if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
618 copied = rlen;
619
620 return copied;
621 }
622
623 static const struct proto_ops llcp_sock_ops = {
624 .family = PF_NFC,
625 .owner = THIS_MODULE,
626 .bind = llcp_sock_bind,
627 .connect = llcp_sock_connect,
628 .release = llcp_sock_release,
629 .socketpair = sock_no_socketpair,
630 .accept = llcp_sock_accept,
631 .getname = llcp_sock_getname,
632 .poll = llcp_sock_poll,
633 .ioctl = sock_no_ioctl,
634 .listen = llcp_sock_listen,
635 .shutdown = sock_no_shutdown,
636 .setsockopt = sock_no_setsockopt,
637 .getsockopt = sock_no_getsockopt,
638 .sendmsg = llcp_sock_sendmsg,
639 .recvmsg = llcp_sock_recvmsg,
640 .mmap = sock_no_mmap,
641 };
642
643 static void llcp_sock_destruct(struct sock *sk)
644 {
645 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
646
647 pr_debug("%p\n", sk);
648
649 if (sk->sk_state == LLCP_CONNECTED)
650 nfc_put_device(llcp_sock->dev);
651
652 skb_queue_purge(&sk->sk_receive_queue);
653
654 nfc_llcp_sock_free(llcp_sock);
655
656 if (!sock_flag(sk, SOCK_DEAD)) {
657 pr_err("Freeing alive NFC LLCP socket %p\n", sk);
658 return;
659 }
660 }
661
662 struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp)
663 {
664 struct sock *sk;
665 struct nfc_llcp_sock *llcp_sock;
666
667 sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto);
668 if (!sk)
669 return NULL;
670
671 llcp_sock = nfc_llcp_sock(sk);
672
673 sock_init_data(sock, sk);
674 sk->sk_state = LLCP_CLOSED;
675 sk->sk_protocol = NFC_SOCKPROTO_LLCP;
676 sk->sk_type = type;
677 sk->sk_destruct = llcp_sock_destruct;
678
679 llcp_sock->ssap = 0;
680 llcp_sock->dsap = LLCP_SAP_SDP;
681 llcp_sock->send_n = llcp_sock->send_ack_n = 0;
682 llcp_sock->recv_n = llcp_sock->recv_ack_n = 0;
683 llcp_sock->remote_ready = 1;
684 skb_queue_head_init(&llcp_sock->tx_queue);
685 skb_queue_head_init(&llcp_sock->tx_pending_queue);
686 skb_queue_head_init(&llcp_sock->tx_backlog_queue);
687 INIT_LIST_HEAD(&llcp_sock->accept_queue);
688
689 if (sock != NULL)
690 sock->state = SS_UNCONNECTED;
691
692 return sk;
693 }
694
695 void nfc_llcp_sock_free(struct nfc_llcp_sock *sock)
696 {
697 kfree(sock->service_name);
698
699 skb_queue_purge(&sock->tx_queue);
700 skb_queue_purge(&sock->tx_pending_queue);
701 skb_queue_purge(&sock->tx_backlog_queue);
702
703 list_del_init(&sock->accept_queue);
704
705 sock->parent = NULL;
706
707 nfc_llcp_local_put(sock->local);
708 }
709
710 static int llcp_sock_create(struct net *net, struct socket *sock,
711 const struct nfc_protocol *nfc_proto)
712 {
713 struct sock *sk;
714
715 pr_debug("%p\n", sock);
716
717 if (sock->type != SOCK_STREAM && sock->type != SOCK_DGRAM)
718 return -ESOCKTNOSUPPORT;
719
720 sock->ops = &llcp_sock_ops;
721
722 sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC);
723 if (sk == NULL)
724 return -ENOMEM;
725
726 return 0;
727 }
728
729 static const struct nfc_protocol llcp_nfc_proto = {
730 .id = NFC_SOCKPROTO_LLCP,
731 .proto = &llcp_sock_proto,
732 .owner = THIS_MODULE,
733 .create = llcp_sock_create
734 };
735
736 int __init nfc_llcp_sock_init(void)
737 {
738 return nfc_proto_register(&llcp_nfc_proto);
739 }
740
741 void nfc_llcp_sock_exit(void)
742 {
743 nfc_proto_unregister(&llcp_nfc_proto);
744 }
This page took 0.04399 seconds and 6 git commands to generate.