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
= 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
,
123 llcp_sock
->ssap
= nfc_llcp_get_sdp_ssap(local
, llcp_sock
);
124 if (llcp_sock
->ssap
== LLCP_MAX_SAP
)
127 nfc_llcp_sock_link(&local
->sockets
, sk
);
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 addr
->sa_family
= AF_NFC
;
296 *len
= sizeof(struct sockaddr_nfc_llcp
);
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
);
308 static inline unsigned int llcp_accept_poll(struct sock
*parent
)
310 struct nfc_llcp_sock
*llcp_sock
, *n
, *parent_sock
;
313 parent_sock
= nfc_llcp_sock(parent
);
315 list_for_each_entry_safe(llcp_sock
, n
, &parent_sock
->accept_queue
,
319 if (sk
->sk_state
== LLCP_CONNECTED
)
320 return POLLIN
| POLLRDNORM
;
326 static unsigned int llcp_sock_poll(struct file
*file
, struct socket
*sock
,
329 struct sock
*sk
= sock
->sk
;
330 unsigned int mask
= 0;
332 pr_debug("%p\n", sk
);
334 sock_poll_wait(file
, sk_sleep(sk
), wait
);
336 if (sk
->sk_state
== LLCP_LISTEN
)
337 return llcp_accept_poll(sk
);
339 if (sk
->sk_err
|| !skb_queue_empty(&sk
->sk_error_queue
))
342 if (!skb_queue_empty(&sk
->sk_receive_queue
))
343 mask
|= POLLIN
| POLLRDNORM
;
345 if (sk
->sk_state
== LLCP_CLOSED
)
348 if (sk
->sk_shutdown
& RCV_SHUTDOWN
)
349 mask
|= POLLRDHUP
| POLLIN
| POLLRDNORM
;
351 if (sk
->sk_shutdown
== SHUTDOWN_MASK
)
354 if (sock_writeable(sk
))
355 mask
|= POLLOUT
| POLLWRNORM
| POLLWRBAND
;
357 set_bit(SOCK_ASYNC_NOSPACE
, &sk
->sk_socket
->flags
);
359 pr_debug("mask 0x%x\n", mask
);
364 static int llcp_sock_release(struct socket
*sock
)
366 struct sock
*sk
= sock
->sk
;
367 struct nfc_llcp_local
*local
;
368 struct nfc_llcp_sock
*llcp_sock
= nfc_llcp_sock(sk
);
374 pr_debug("%p\n", sk
);
376 local
= llcp_sock
->local
;
385 if (sk
->sk_state
== LLCP_CONNECTED
)
386 nfc_llcp_disconnect(llcp_sock
);
388 if (sk
->sk_state
== LLCP_LISTEN
) {
389 struct nfc_llcp_sock
*lsk
, *n
;
390 struct sock
*accept_sk
;
392 list_for_each_entry_safe(lsk
, n
, &llcp_sock
->accept_queue
,
394 accept_sk
= &lsk
->sk
;
395 lock_sock(accept_sk
);
397 nfc_llcp_disconnect(lsk
);
398 nfc_llcp_accept_unlink(accept_sk
);
400 release_sock(accept_sk
);
402 sock_orphan(accept_sk
);
406 nfc_llcp_put_ssap(llcp_sock
->local
, llcp_sock
->ssap
);
410 nfc_llcp_sock_unlink(&local
->sockets
, sk
);
419 static int llcp_sock_connect(struct socket
*sock
, struct sockaddr
*_addr
,
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
;
426 struct nfc_llcp_local
*local
;
429 pr_debug("sock %p sk %p flags 0x%x\n", sock
, sk
, flags
);
431 if (!addr
|| len
< sizeof(struct sockaddr_nfc
) ||
432 addr
->sa_family
!= AF_NFC
) {
433 pr_err("Invalid socket\n");
437 if (addr
->service_name_len
== 0 && addr
->dsap
== 0) {
438 pr_err("Missing service name or dsap\n");
442 pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr
->dev_idx
,
443 addr
->target_idx
, addr
->nfc_protocol
);
447 if (sk
->sk_state
== LLCP_CONNECTED
) {
452 dev
= nfc_get_device(addr
->dev_idx
);
458 local
= nfc_llcp_find_local(dev
);
464 device_lock(&dev
->dev
);
465 if (dev
->dep_link_up
== false) {
467 device_unlock(&dev
->dev
);
470 device_unlock(&dev
->dev
);
472 if (local
->rf_mode
== NFC_RF_INITIATOR
&&
473 addr
->target_idx
!= local
->target_idx
) {
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
) {
485 if (addr
->service_name_len
== 0)
486 llcp_sock
->dsap
= addr
->dsap
;
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
,
497 nfc_llcp_sock_link(&local
->connecting_sockets
, sk
);
499 ret
= nfc_llcp_send_connect(llcp_sock
);
503 ret
= sock_wait_state(sk
, LLCP_CONNECTED
,
504 sock_sndtimeo(sk
, flags
& O_NONBLOCK
));
513 nfc_llcp_put_ssap(local
, llcp_sock
->ssap
);
515 nfc_llcp_sock_unlink(&local
->connecting_sockets
, sk
);
525 static int llcp_sock_sendmsg(struct kiocb
*iocb
, struct socket
*sock
,
526 struct msghdr
*msg
, size_t len
)
528 struct sock
*sk
= sock
->sk
;
529 struct nfc_llcp_sock
*llcp_sock
= nfc_llcp_sock(sk
);
532 pr_debug("sock %p sk %p", sock
, sk
);
534 ret
= sock_error(sk
);
538 if (msg
->msg_flags
& MSG_OOB
)
543 if (sk
->sk_state
!= LLCP_CONNECTED
) {
550 return nfc_llcp_send_i_frame(llcp_sock
, msg
, len
);
553 static int llcp_sock_recvmsg(struct kiocb
*iocb
, struct socket
*sock
,
554 struct msghdr
*msg
, size_t len
, int flags
)
556 int noblock
= flags
& MSG_DONTWAIT
;
557 struct sock
*sk
= sock
->sk
;
558 unsigned int copied
, rlen
;
559 struct sk_buff
*skb
, *cskb
;
562 pr_debug("%p %zu\n", sk
, len
);
566 if (sk
->sk_state
== LLCP_CLOSED
&&
567 skb_queue_empty(&sk
->sk_receive_queue
)) {
574 if (flags
& (MSG_OOB
))
577 skb
= skb_recv_datagram(sk
, flags
, noblock
, &err
);
579 pr_err("Recv datagram failed state %d %d %d",
580 sk
->sk_state
, err
, sock_error(sk
));
582 if (sk
->sk_shutdown
& RCV_SHUTDOWN
)
588 rlen
= skb
->len
; /* real length of skb */
589 copied
= min_t(unsigned int, rlen
, len
);
592 if (memcpy_toiovec(msg
->msg_iov
, cskb
->data
, copied
)) {
593 if (!(flags
& MSG_PEEK
))
594 skb_queue_head(&sk
->sk_receive_queue
, skb
);
598 /* Mark read part of skb as used */
599 if (!(flags
& MSG_PEEK
)) {
601 /* SOCK_STREAM: re-queue skb if it contains unreceived data */
602 if (sk
->sk_type
== SOCK_STREAM
) {
603 skb_pull(skb
, copied
);
605 skb_queue_head(&sk
->sk_receive_queue
, skb
);
613 /* XXX Queue backlogged skbs */
616 /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
617 if (sk
->sk_type
== SOCK_SEQPACKET
&& (flags
& MSG_TRUNC
))
623 static const struct proto_ops llcp_sock_ops
= {
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
,
643 static void llcp_sock_destruct(struct sock
*sk
)
645 struct nfc_llcp_sock
*llcp_sock
= nfc_llcp_sock(sk
);
647 pr_debug("%p\n", sk
);
649 if (sk
->sk_state
== LLCP_CONNECTED
)
650 nfc_put_device(llcp_sock
->dev
);
652 skb_queue_purge(&sk
->sk_receive_queue
);
654 nfc_llcp_sock_free(llcp_sock
);
656 if (!sock_flag(sk
, SOCK_DEAD
)) {
657 pr_err("Freeing alive NFC LLCP socket %p\n", sk
);
662 struct sock
*nfc_llcp_sock_alloc(struct socket
*sock
, int type
, gfp_t gfp
)
665 struct nfc_llcp_sock
*llcp_sock
;
667 sk
= sk_alloc(&init_net
, PF_NFC
, gfp
, &llcp_sock_proto
);
671 llcp_sock
= nfc_llcp_sock(sk
);
673 sock_init_data(sock
, sk
);
674 sk
->sk_state
= LLCP_CLOSED
;
675 sk
->sk_protocol
= NFC_SOCKPROTO_LLCP
;
677 sk
->sk_destruct
= llcp_sock_destruct
;
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
);
690 sock
->state
= SS_UNCONNECTED
;
695 void nfc_llcp_sock_free(struct nfc_llcp_sock
*sock
)
697 kfree(sock
->service_name
);
699 skb_queue_purge(&sock
->tx_queue
);
700 skb_queue_purge(&sock
->tx_pending_queue
);
701 skb_queue_purge(&sock
->tx_backlog_queue
);
703 list_del_init(&sock
->accept_queue
);
707 nfc_llcp_local_put(sock
->local
);
710 static int llcp_sock_create(struct net
*net
, struct socket
*sock
,
711 const struct nfc_protocol
*nfc_proto
)
715 pr_debug("%p\n", sock
);
717 if (sock
->type
!= SOCK_STREAM
&& sock
->type
!= SOCK_DGRAM
)
718 return -ESOCKTNOSUPPORT
;
720 sock
->ops
= &llcp_sock_ops
;
722 sk
= nfc_llcp_sock_alloc(sock
, sock
->type
, GFP_ATOMIC
);
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
736 int __init
nfc_llcp_sock_init(void)
738 return nfc_proto_register(&llcp_nfc_proto
);
741 void nfc_llcp_sock_exit(void)
743 nfc_proto_unregister(&llcp_nfc_proto
);