2 * Copyright (C) 2011 Intel Corporation. All rights reserved.
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.
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.
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.
20 #define pr_fmt(fmt) "llcp: %s: " fmt, __func__
22 #include <linux/init.h>
23 #include <linux/kernel.h>
24 #include <linux/module.h>
25 #include <linux/nfc.h>
30 static int sock_wait_state(struct sock *sk, int state, unsigned long timeo)
32 DECLARE_WAITQUEUE(wait, current);
35 pr_debug("sk %p", sk);
37 add_wait_queue(sk_sleep(sk), &wait);
38 set_current_state(TASK_INTERRUPTIBLE);
40 while (sk->sk_state != state) {
46 if (signal_pending(current)) {
47 err = sock_intr_errno(timeo);
52 timeo = schedule_timeout(timeo);
54 set_current_state(TASK_INTERRUPTIBLE);
61 __set_current_state(TASK_RUNNING);
62 remove_wait_queue(sk_sleep(sk), &wait);
66 static struct proto llcp_sock_proto = {
69 .obj_size = sizeof(struct nfc_llcp_sock),
72 static int llcp_sock_bind(struct socket *sock, struct sockaddr *addr, int alen)
74 struct sock *sk = sock->sk;
75 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
76 struct nfc_llcp_local *local;
78 struct sockaddr_nfc_llcp llcp_addr;
81 pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);
83 if (!addr || addr->sa_family != AF_NFC)
86 memset(&llcp_addr, 0, sizeof(llcp_addr));
87 len = min_t(unsigned int, sizeof(llcp_addr), alen);
88 memcpy(&llcp_addr, addr, len);
90 /* This is going to be a listening socket, dsap must be 0 */
91 if (llcp_addr.dsap != 0)
96 if (sk->sk_state != LLCP_CLOSED) {
101 dev = nfc_get_device(llcp_addr.dev_idx);
107 local = nfc_llcp_find_local(dev);
113 llcp_sock->dev = dev;
114 llcp_sock->local = 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,
123 llcp_sock->ssap = nfc_llcp_get_sdp_ssap(local, llcp_sock);
124 if (llcp_sock->ssap == LLCP_MAX_SAP)
127 local->sockets[llcp_sock->ssap] = llcp_sock;
129 pr_debug("Socket bound to SAP %d\n", llcp_sock->ssap);
131 sk->sk_state = LLCP_BOUND;
141 static int llcp_sock_listen(struct socket *sock, int backlog)
143 struct sock *sk = sock->sk;
146 pr_debug("sk %p backlog %d\n", sk, backlog);
150 if ((sock->type != SOCK_SEQPACKET && sock->type != SOCK_STREAM)
151 || sk->sk_state != LLCP_BOUND) {
156 sk->sk_max_ack_backlog = backlog;
157 sk->sk_ack_backlog = 0;
159 pr_debug("Socket listening\n");
160 sk->sk_state = LLCP_LISTEN;
168 void nfc_llcp_accept_unlink(struct sock *sk)
170 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
172 pr_debug("state %d\n", sk->sk_state);
174 list_del_init(&llcp_sock->accept_queue);
175 sk_acceptq_removed(llcp_sock->parent);
176 llcp_sock->parent = NULL;
181 void nfc_llcp_accept_enqueue(struct sock *parent, struct sock *sk)
183 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
184 struct nfc_llcp_sock *llcp_sock_parent = nfc_llcp_sock(parent);
186 /* Lock will be free from unlink */
189 list_add_tail(&llcp_sock->accept_queue,
190 &llcp_sock_parent->accept_queue);
191 llcp_sock->parent = parent;
192 sk_acceptq_added(parent);
195 struct sock *nfc_llcp_accept_dequeue(struct sock *parent,
196 struct socket *newsock)
198 struct nfc_llcp_sock *lsk, *n, *llcp_parent;
201 llcp_parent = nfc_llcp_sock(parent);
203 list_for_each_entry_safe(lsk, n, &llcp_parent->accept_queue,
208 if (sk->sk_state == LLCP_CLOSED) {
210 nfc_llcp_accept_unlink(sk);
214 if (sk->sk_state == LLCP_CONNECTED || !newsock) {
215 nfc_llcp_accept_unlink(sk);
217 sock_graft(sk, newsock);
221 pr_debug("Returning sk state %d\n", sk->sk_state);
232 static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
235 DECLARE_WAITQUEUE(wait, current);
236 struct sock *sk = sock->sk, *new_sk;
240 pr_debug("parent %p\n", sk);
242 lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
244 if (sk->sk_state != LLCP_LISTEN) {
249 timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
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);
261 if (signal_pending(current)) {
262 ret = sock_intr_errno(timeo);
267 timeo = schedule_timeout(timeo);
268 lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
270 __set_current_state(TASK_RUNNING);
271 remove_wait_queue(sk_sleep(sk), &wait);
276 newsock->state = SS_CONNECTED;
278 pr_debug("new socket %p\n", new_sk);
286 static int llcp_sock_getname(struct socket *sock, struct sockaddr *addr,
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);
293 pr_debug("%p\n", sk);
295 if (llcp_sock == NULL || llcp_sock->dev == NULL)
298 addr->sa_family = AF_NFC;
299 *len = sizeof(struct sockaddr_nfc_llcp);
301 llcp_addr->dev_idx = llcp_sock->dev->idx;
302 llcp_addr->dsap = llcp_sock->dsap;
303 llcp_addr->ssap = llcp_sock->ssap;
304 llcp_addr->service_name_len = llcp_sock->service_name_len;
305 memcpy(llcp_addr->service_name, llcp_sock->service_name,
306 llcp_addr->service_name_len);
311 static inline unsigned int llcp_accept_poll(struct sock *parent)
313 struct nfc_llcp_sock *llcp_sock, *n, *parent_sock;
316 parent_sock = nfc_llcp_sock(parent);
318 list_for_each_entry_safe(llcp_sock, n, &parent_sock->accept_queue,
322 if (sk->sk_state == LLCP_CONNECTED)
323 return POLLIN | POLLRDNORM;
329 static unsigned int llcp_sock_poll(struct file *file, struct socket *sock,
332 struct sock *sk = sock->sk;
333 unsigned int mask = 0;
335 pr_debug("%p\n", sk);
337 sock_poll_wait(file, sk_sleep(sk), wait);
339 if (sk->sk_state == LLCP_LISTEN)
340 return llcp_accept_poll(sk);
342 if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
345 if (!skb_queue_empty(&sk->sk_receive_queue))
346 mask |= POLLIN | POLLRDNORM;
348 if (sk->sk_state == LLCP_CLOSED)
351 if (sk->sk_shutdown & RCV_SHUTDOWN)
352 mask |= POLLRDHUP | POLLIN | POLLRDNORM;
354 if (sk->sk_shutdown == SHUTDOWN_MASK)
357 if (sock_writeable(sk))
358 mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
360 set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
362 pr_debug("mask 0x%x\n", mask);
367 static int llcp_sock_release(struct socket *sock)
369 struct sock *sk = sock->sk;
370 struct nfc_llcp_local *local;
371 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
377 pr_debug("%p\n", sk);
379 local = llcp_sock->local;
385 mutex_lock(&local->socket_lock);
387 if (llcp_sock == local->sockets[llcp_sock->ssap])
388 local->sockets[llcp_sock->ssap] = NULL;
390 list_del_init(&llcp_sock->list);
392 mutex_unlock(&local->socket_lock);
397 if (sk->sk_state == LLCP_CONNECTED)
398 nfc_llcp_disconnect(llcp_sock);
400 if (sk->sk_state == LLCP_LISTEN) {
401 struct nfc_llcp_sock *lsk, *n;
402 struct sock *accept_sk;
404 list_for_each_entry_safe(lsk, n, &llcp_sock->accept_queue,
406 accept_sk = &lsk->sk;
407 lock_sock(accept_sk);
409 nfc_llcp_disconnect(lsk);
410 nfc_llcp_accept_unlink(accept_sk);
412 release_sock(accept_sk);
414 sock_orphan(accept_sk);
418 /* Freeing the SAP */
419 if ((sk->sk_state == LLCP_CONNECTED
420 && llcp_sock->ssap > LLCP_LOCAL_SAP_OFFSET) ||
421 sk->sk_state == LLCP_BOUND || sk->sk_state == LLCP_LISTEN)
422 nfc_llcp_put_ssap(llcp_sock->local, llcp_sock->ssap);
433 static int llcp_sock_connect(struct socket *sock, struct sockaddr *_addr,
436 struct sock *sk = sock->sk;
437 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
438 struct sockaddr_nfc_llcp *addr = (struct sockaddr_nfc_llcp *)_addr;
440 struct nfc_llcp_local *local;
443 pr_debug("sock %p sk %p flags 0x%x\n", sock, sk, flags);
445 if (!addr || len < sizeof(struct sockaddr_nfc) ||
446 addr->sa_family != AF_NFC) {
447 pr_err("Invalid socket\n");
451 if (addr->service_name_len == 0 && addr->dsap == 0) {
452 pr_err("Missing service name or dsap\n");
456 pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr->dev_idx,
457 addr->target_idx, addr->nfc_protocol);
461 if (sk->sk_state == LLCP_CONNECTED) {
466 dev = nfc_get_device(addr->dev_idx);
472 local = nfc_llcp_find_local(dev);
478 device_lock(&dev->dev);
479 if (dev->dep_link_up == false) {
481 device_unlock(&dev->dev);
484 device_unlock(&dev->dev);
486 if (local->rf_mode == NFC_RF_INITIATOR &&
487 addr->target_idx != local->target_idx) {
492 llcp_sock->dev = dev;
493 llcp_sock->local = local;
494 llcp_sock->ssap = nfc_llcp_get_local_ssap(local);
495 if (llcp_sock->ssap == LLCP_SAP_MAX) {
499 if (addr->service_name_len == 0)
500 llcp_sock->dsap = addr->dsap;
502 llcp_sock->dsap = LLCP_SAP_SDP;
503 llcp_sock->nfc_protocol = addr->nfc_protocol;
504 llcp_sock->service_name_len = min_t(unsigned int,
505 addr->service_name_len,
506 NFC_LLCP_MAX_SERVICE_NAME);
507 llcp_sock->service_name = kmemdup(addr->service_name,
508 llcp_sock->service_name_len,
511 local->sockets[llcp_sock->ssap] = llcp_sock;
513 ret = nfc_llcp_send_connect(llcp_sock);
517 ret = sock_wait_state(sk, LLCP_CONNECTED,
518 sock_sndtimeo(sk, flags & O_NONBLOCK));
534 static int llcp_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
535 struct msghdr *msg, size_t len)
537 struct sock *sk = sock->sk;
538 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
541 pr_debug("sock %p sk %p", sock, sk);
543 ret = sock_error(sk);
547 if (msg->msg_flags & MSG_OOB)
552 if (sk->sk_state != LLCP_CONNECTED) {
559 return nfc_llcp_send_i_frame(llcp_sock, msg, len);
562 static int llcp_sock_recvmsg(struct kiocb *iocb, struct socket *sock,
563 struct msghdr *msg, size_t len, int flags)
565 int noblock = flags & MSG_DONTWAIT;
566 struct sock *sk = sock->sk;
567 unsigned int copied, rlen;
568 struct sk_buff *skb, *cskb;
571 pr_debug("%p %zu\n", sk, len);
575 if (sk->sk_state == LLCP_CLOSED &&
576 skb_queue_empty(&sk->sk_receive_queue)) {
583 if (flags & (MSG_OOB))
586 skb = skb_recv_datagram(sk, flags, noblock, &err);
588 pr_err("Recv datagram failed state %d %d %d",
589 sk->sk_state, err, sock_error(sk));
591 if (sk->sk_shutdown & RCV_SHUTDOWN)
597 rlen = skb->len; /* real length of skb */
598 copied = min_t(unsigned int, rlen, len);
601 if (memcpy_toiovec(msg->msg_iov, cskb->data, copied)) {
602 if (!(flags & MSG_PEEK))
603 skb_queue_head(&sk->sk_receive_queue, skb);
607 /* Mark read part of skb as used */
608 if (!(flags & MSG_PEEK)) {
610 /* SOCK_STREAM: re-queue skb if it contains unreceived data */
611 if (sk->sk_type == SOCK_STREAM) {
612 skb_pull(skb, copied);
614 skb_queue_head(&sk->sk_receive_queue, skb);
622 /* XXX Queue backlogged skbs */
625 /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
626 if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
632 static const struct proto_ops llcp_sock_ops = {
634 .owner = THIS_MODULE,
635 .bind = llcp_sock_bind,
636 .connect = llcp_sock_connect,
637 .release = llcp_sock_release,
638 .socketpair = sock_no_socketpair,
639 .accept = llcp_sock_accept,
640 .getname = llcp_sock_getname,
641 .poll = llcp_sock_poll,
642 .ioctl = sock_no_ioctl,
643 .listen = llcp_sock_listen,
644 .shutdown = sock_no_shutdown,
645 .setsockopt = sock_no_setsockopt,
646 .getsockopt = sock_no_getsockopt,
647 .sendmsg = llcp_sock_sendmsg,
648 .recvmsg = llcp_sock_recvmsg,
649 .mmap = sock_no_mmap,
652 static void llcp_sock_destruct(struct sock *sk)
654 struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
656 pr_debug("%p\n", sk);
658 if (sk->sk_state == LLCP_CONNECTED)
659 nfc_put_device(llcp_sock->dev);
661 skb_queue_purge(&sk->sk_receive_queue);
663 nfc_llcp_sock_free(llcp_sock);
665 if (!sock_flag(sk, SOCK_DEAD)) {
666 pr_err("Freeing alive NFC LLCP socket %p\n", sk);
671 struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp)
674 struct nfc_llcp_sock *llcp_sock;
676 sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto);
680 llcp_sock = nfc_llcp_sock(sk);
682 sock_init_data(sock, sk);
683 sk->sk_state = LLCP_CLOSED;
684 sk->sk_protocol = NFC_SOCKPROTO_LLCP;
686 sk->sk_destruct = llcp_sock_destruct;
689 llcp_sock->dsap = LLCP_SAP_SDP;
690 llcp_sock->send_n = llcp_sock->send_ack_n = 0;
691 llcp_sock->recv_n = llcp_sock->recv_ack_n = 0;
692 llcp_sock->remote_ready = 1;
693 skb_queue_head_init(&llcp_sock->tx_queue);
694 skb_queue_head_init(&llcp_sock->tx_pending_queue);
695 skb_queue_head_init(&llcp_sock->tx_backlog_queue);
696 INIT_LIST_HEAD(&llcp_sock->list);
697 INIT_LIST_HEAD(&llcp_sock->accept_queue);
700 sock->state = SS_UNCONNECTED;
705 void nfc_llcp_sock_free(struct nfc_llcp_sock *sock)
707 struct nfc_llcp_local *local = sock->local;
709 kfree(sock->service_name);
711 skb_queue_purge(&sock->tx_queue);
712 skb_queue_purge(&sock->tx_pending_queue);
713 skb_queue_purge(&sock->tx_backlog_queue);
715 list_del_init(&sock->accept_queue);
717 if (local != NULL && sock == local->sockets[sock->ssap])
718 local->sockets[sock->ssap] = NULL;
720 list_del_init(&sock->list);
725 static int llcp_sock_create(struct net *net, struct socket *sock,
726 const struct nfc_protocol *nfc_proto)
730 pr_debug("%p\n", sock);
732 if (sock->type != SOCK_STREAM && sock->type != SOCK_DGRAM)
733 return -ESOCKTNOSUPPORT;
735 sock->ops = &llcp_sock_ops;
737 sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC);
744 static const struct nfc_protocol llcp_nfc_proto = {
745 .id = NFC_SOCKPROTO_LLCP,
746 .proto = &llcp_sock_proto,
747 .owner = THIS_MODULE,
748 .create = llcp_sock_create
751 int __init nfc_llcp_sock_init(void)
753 return nfc_proto_register(&llcp_nfc_proto);
756 void nfc_llcp_sock_exit(void)
758 nfc_proto_unregister(&llcp_nfc_proto);