sock.c 19.0 KB
Newer Older
S
Samuel Ortiz 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
/*
 * Copyright (C) 2011  Intel Corporation. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the
 * Free Software Foundation, Inc.,
 * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#define pr_fmt(fmt) "llcp: %s: " fmt, __func__

#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/nfc.h>

#include "../nfc.h"
#include "llcp.h"

30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
static int sock_wait_state(struct sock *sk, int state, unsigned long timeo)
{
	DECLARE_WAITQUEUE(wait, current);
	int err = 0;

	pr_debug("sk %p", sk);

	add_wait_queue(sk_sleep(sk), &wait);
	set_current_state(TASK_INTERRUPTIBLE);

	while (sk->sk_state != state) {
		if (!timeo) {
			err = -EINPROGRESS;
			break;
		}

		if (signal_pending(current)) {
			err = sock_intr_errno(timeo);
			break;
		}

		release_sock(sk);
		timeo = schedule_timeout(timeo);
		lock_sock(sk);
		set_current_state(TASK_INTERRUPTIBLE);

		err = sock_error(sk);
		if (err)
			break;
	}

	__set_current_state(TASK_RUNNING);
	remove_wait_queue(sk_sleep(sk), &wait);
	return err;
}

S
Samuel Ortiz 已提交
66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
static struct proto llcp_sock_proto = {
	.name     = "NFC_LLCP",
	.owner    = THIS_MODULE,
	.obj_size = sizeof(struct nfc_llcp_sock),
};

static int llcp_sock_bind(struct socket *sock, struct sockaddr *addr, int alen)
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	struct nfc_llcp_local *local;
	struct nfc_dev *dev;
	struct sockaddr_nfc_llcp llcp_addr;
	int len, ret = 0;

	if (!addr || addr->sa_family != AF_NFC)
		return -EINVAL;

84 85
	pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);

S
Samuel Ortiz 已提交
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113
	memset(&llcp_addr, 0, sizeof(llcp_addr));
	len = min_t(unsigned int, sizeof(llcp_addr), alen);
	memcpy(&llcp_addr, addr, len);

	/* This is going to be a listening socket, dsap must be 0 */
	if (llcp_addr.dsap != 0)
		return -EINVAL;

	lock_sock(sk);

	if (sk->sk_state != LLCP_CLOSED) {
		ret = -EBADFD;
		goto error;
	}

	dev = nfc_get_device(llcp_addr.dev_idx);
	if (dev == NULL) {
		ret = -ENODEV;
		goto error;
	}

	local = nfc_llcp_find_local(dev);
	if (local == NULL) {
		ret = -ENODEV;
		goto put_dev;
	}

	llcp_sock->dev = dev;
114
	llcp_sock->local = nfc_llcp_local_get(local);
S
Samuel Ortiz 已提交
115 116
	llcp_sock->nfc_protocol = llcp_addr.nfc_protocol;
	llcp_sock->service_name_len = min_t(unsigned int,
S
Samuel Ortiz 已提交
117 118
					    llcp_addr.service_name_len,
					    NFC_LLCP_MAX_SERVICE_NAME);
S
Samuel Ortiz 已提交
119
	llcp_sock->service_name = kmemdup(llcp_addr.service_name,
S
Samuel Ortiz 已提交
120 121
					  llcp_sock->service_name_len,
					  GFP_KERNEL);
S
Samuel Ortiz 已提交
122 123

	llcp_sock->ssap = nfc_llcp_get_sdp_ssap(local, llcp_sock);
124 125
	if (llcp_sock->ssap == LLCP_SAP_MAX) {
		ret = -EADDRINUSE;
S
Samuel Ortiz 已提交
126
		goto put_dev;
127
	}
S
Samuel Ortiz 已提交
128

129 130
	llcp_sock->reserved_ssap = llcp_sock->ssap;

S
Samuel Ortiz 已提交
131
	nfc_llcp_sock_link(&local->sockets, sk);
S
Samuel Ortiz 已提交
132 133 134 135 136 137 138 139 140 141 142 143 144

	pr_debug("Socket bound to SAP %d\n", llcp_sock->ssap);

	sk->sk_state = LLCP_BOUND;

put_dev:
	nfc_put_device(dev);

error:
	release_sock(sk);
	return ret;
}

T
Thierry Escande 已提交
145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198
static int llcp_raw_sock_bind(struct socket *sock, struct sockaddr *addr,
			      int alen)
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	struct nfc_llcp_local *local;
	struct nfc_dev *dev;
	struct sockaddr_nfc_llcp llcp_addr;
	int len, ret = 0;

	if (!addr || addr->sa_family != AF_NFC)
		return -EINVAL;

	pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);

	memset(&llcp_addr, 0, sizeof(llcp_addr));
	len = min_t(unsigned int, sizeof(llcp_addr), alen);
	memcpy(&llcp_addr, addr, len);

	lock_sock(sk);

	if (sk->sk_state != LLCP_CLOSED) {
		ret = -EBADFD;
		goto error;
	}

	dev = nfc_get_device(llcp_addr.dev_idx);
	if (dev == NULL) {
		ret = -ENODEV;
		goto error;
	}

	local = nfc_llcp_find_local(dev);
	if (local == NULL) {
		ret = -ENODEV;
		goto put_dev;
	}

	llcp_sock->dev = dev;
	llcp_sock->local = nfc_llcp_local_get(local);
	llcp_sock->nfc_protocol = llcp_addr.nfc_protocol;

	nfc_llcp_sock_link(&local->raw_sockets, sk);

	sk->sk_state = LLCP_BOUND;

put_dev:
	nfc_put_device(dev);

error:
	release_sock(sk);
	return ret;
}

S
Samuel Ortiz 已提交
199 200 201 202 203 204 205 206 207
static int llcp_sock_listen(struct socket *sock, int backlog)
{
	struct sock *sk = sock->sk;
	int ret = 0;

	pr_debug("sk %p backlog %d\n", sk, backlog);

	lock_sock(sk);

208 209
	if ((sock->type != SOCK_SEQPACKET && sock->type != SOCK_STREAM) ||
	    sk->sk_state != LLCP_BOUND) {
S
Samuel Ortiz 已提交
210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247
		ret = -EBADFD;
		goto error;
	}

	sk->sk_max_ack_backlog = backlog;
	sk->sk_ack_backlog = 0;

	pr_debug("Socket listening\n");
	sk->sk_state = LLCP_LISTEN;

error:
	release_sock(sk);

	return ret;
}

void nfc_llcp_accept_unlink(struct sock *sk)
{
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);

	pr_debug("state %d\n", sk->sk_state);

	list_del_init(&llcp_sock->accept_queue);
	sk_acceptq_removed(llcp_sock->parent);
	llcp_sock->parent = NULL;

	sock_put(sk);
}

void nfc_llcp_accept_enqueue(struct sock *parent, struct sock *sk)
{
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	struct nfc_llcp_sock *llcp_sock_parent = nfc_llcp_sock(parent);

	/* Lock will be free from unlink */
	sock_hold(sk);

	list_add_tail(&llcp_sock->accept_queue,
S
Samuel Ortiz 已提交
248
		      &llcp_sock_parent->accept_queue);
S
Samuel Ortiz 已提交
249 250 251 252 253
	llcp_sock->parent = parent;
	sk_acceptq_added(parent);
}

struct sock *nfc_llcp_accept_dequeue(struct sock *parent,
S
Samuel Ortiz 已提交
254
				     struct socket *newsock)
S
Samuel Ortiz 已提交
255 256 257 258 259 260 261
{
	struct nfc_llcp_sock *lsk, *n, *llcp_parent;
	struct sock *sk;

	llcp_parent = nfc_llcp_sock(parent);

	list_for_each_entry_safe(lsk, n, &llcp_parent->accept_queue,
S
Samuel Ortiz 已提交
262
				 accept_queue) {
S
Samuel Ortiz 已提交
263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280
		sk = &lsk->sk;
		lock_sock(sk);

		if (sk->sk_state == LLCP_CLOSED) {
			release_sock(sk);
			nfc_llcp_accept_unlink(sk);
			continue;
		}

		if (sk->sk_state == LLCP_CONNECTED || !newsock) {
			nfc_llcp_accept_unlink(sk);
			if (newsock)
				sock_graft(sk, newsock);

			release_sock(sk);

			pr_debug("Returning sk state %d\n", sk->sk_state);

281 282
			sk_acceptq_removed(parent);

S
Samuel Ortiz 已提交
283 284 285 286 287 288 289 290 291 292
			return sk;
		}

		release_sock(sk);
	}

	return NULL;
}

static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
S
Samuel Ortiz 已提交
293
			    int flags)
S
Samuel Ortiz 已提交
294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
{
	DECLARE_WAITQUEUE(wait, current);
	struct sock *sk = sock->sk, *new_sk;
	long timeo;
	int ret = 0;

	pr_debug("parent %p\n", sk);

	lock_sock_nested(sk, SINGLE_DEPTH_NESTING);

	if (sk->sk_state != LLCP_LISTEN) {
		ret = -EBADFD;
		goto error;
	}

	timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);

	/* Wait for an incoming connection. */
	add_wait_queue_exclusive(sk_sleep(sk), &wait);
	while (!(new_sk = nfc_llcp_accept_dequeue(sk, newsock))) {
		set_current_state(TASK_INTERRUPTIBLE);

		if (!timeo) {
			ret = -EAGAIN;
			break;
		}

		if (signal_pending(current)) {
			ret = sock_intr_errno(timeo);
			break;
		}

		release_sock(sk);
		timeo = schedule_timeout(timeo);
		lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
	}
	__set_current_state(TASK_RUNNING);
	remove_wait_queue(sk_sleep(sk), &wait);

	if (ret)
		goto error;

	newsock->state = SS_CONNECTED;

	pr_debug("new socket %p\n", new_sk);

error:
	release_sock(sk);

	return ret;
}

S
Samuel Ortiz 已提交
346
static int llcp_sock_getname(struct socket *sock, struct sockaddr *uaddr,
S
Samuel Ortiz 已提交
347 348 349 350
			     int *len, int peer)
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
S
Samuel Ortiz 已提交
351
	DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, llcp_addr, uaddr);
S
Samuel Ortiz 已提交
352

353 354 355
	if (llcp_sock == NULL || llcp_sock->dev == NULL)
		return -EBADFD;

S
Samuel Ortiz 已提交
356 357
	pr_debug("%p %d %d %d\n", sk, llcp_sock->target_idx,
		 llcp_sock->dsap, llcp_sock->ssap);
S
Samuel Ortiz 已提交
358

S
Samuel Ortiz 已提交
359 360
	uaddr->sa_family = AF_NFC;

S
Samuel Ortiz 已提交
361 362 363
	*len = sizeof(struct sockaddr_nfc_llcp);

	llcp_addr->dev_idx = llcp_sock->dev->idx;
S
Samuel Ortiz 已提交
364
	llcp_addr->target_idx = llcp_sock->target_idx;
S
Samuel Ortiz 已提交
365 366 367 368
	llcp_addr->dsap = llcp_sock->dsap;
	llcp_addr->ssap = llcp_sock->ssap;
	llcp_addr->service_name_len = llcp_sock->service_name_len;
	memcpy(llcp_addr->service_name, llcp_sock->service_name,
S
Samuel Ortiz 已提交
369
	       llcp_addr->service_name_len);
S
Samuel Ortiz 已提交
370 371 372 373 374 375 376 377 378 379 380 381

	return 0;
}

static inline unsigned int llcp_accept_poll(struct sock *parent)
{
	struct nfc_llcp_sock *llcp_sock, *n, *parent_sock;
	struct sock *sk;

	parent_sock = nfc_llcp_sock(parent);

	list_for_each_entry_safe(llcp_sock, n, &parent_sock->accept_queue,
S
Samuel Ortiz 已提交
382
				 accept_queue) {
S
Samuel Ortiz 已提交
383 384 385 386 387 388 389 390 391 392
		sk = &llcp_sock->sk;

		if (sk->sk_state == LLCP_CONNECTED)
			return POLLIN | POLLRDNORM;
	}

	return 0;
}

static unsigned int llcp_sock_poll(struct file *file, struct socket *sock,
S
Samuel Ortiz 已提交
393
				   poll_table *wait)
S
Samuel Ortiz 已提交
394 395 396 397 398 399 400 401 402 403 404 405 406 407 408
{
	struct sock *sk = sock->sk;
	unsigned int mask = 0;

	pr_debug("%p\n", sk);

	sock_poll_wait(file, sk_sleep(sk), wait);

	if (sk->sk_state == LLCP_LISTEN)
		return llcp_accept_poll(sk);

	if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
		mask |= POLLERR;

	if (!skb_queue_empty(&sk->sk_receive_queue))
S
Samuel Ortiz 已提交
409
		mask |= POLLIN | POLLRDNORM;
S
Samuel Ortiz 已提交
410 411 412 413

	if (sk->sk_state == LLCP_CLOSED)
		mask |= POLLHUP;

S
Samuel Ortiz 已提交
414 415 416 417 418 419 420 421 422 423 424 425 426
	if (sk->sk_shutdown & RCV_SHUTDOWN)
		mask |= POLLRDHUP | POLLIN | POLLRDNORM;

	if (sk->sk_shutdown == SHUTDOWN_MASK)
		mask |= POLLHUP;

	if (sock_writeable(sk))
		mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
	else
		set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);

	pr_debug("mask 0x%x\n", mask);

S
Samuel Ortiz 已提交
427 428 429 430 431 432 433 434
	return mask;
}

static int llcp_sock_release(struct socket *sock)
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_local *local;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
435
	int err = 0;
S
Samuel Ortiz 已提交
436 437 438 439 440 441 442

	if (!sk)
		return 0;

	pr_debug("%p\n", sk);

	local = llcp_sock->local;
443 444 445 446
	if (local == NULL) {
		err = -ENODEV;
		goto out;
	}
S
Samuel Ortiz 已提交
447 448 449 450 451 452 453 454 455 456 457 458

	lock_sock(sk);

	/* Send a DISC */
	if (sk->sk_state == LLCP_CONNECTED)
		nfc_llcp_disconnect(llcp_sock);

	if (sk->sk_state == LLCP_LISTEN) {
		struct nfc_llcp_sock *lsk, *n;
		struct sock *accept_sk;

		list_for_each_entry_safe(lsk, n, &llcp_sock->accept_queue,
S
Samuel Ortiz 已提交
459
					 accept_queue) {
S
Samuel Ortiz 已提交
460 461 462 463 464 465 466 467 468 469 470 471
			accept_sk = &lsk->sk;
			lock_sock(accept_sk);

			nfc_llcp_disconnect(lsk);
			nfc_llcp_accept_unlink(accept_sk);

			release_sock(accept_sk);

			sock_orphan(accept_sk);
		}
	}

472 473
	if (llcp_sock->reserved_ssap < LLCP_SAP_MAX)
		nfc_llcp_put_ssap(llcp_sock->local, llcp_sock->ssap);
S
Samuel Ortiz 已提交
474 475 476

	release_sock(sk);

T
Thierry Escande 已提交
477 478 479 480
	if (sock->type == SOCK_RAW)
		nfc_llcp_sock_unlink(&local->raw_sockets, sk);
	else
		nfc_llcp_sock_unlink(&local->sockets, sk);
S
Samuel Ortiz 已提交
481

482
out:
S
Samuel Ortiz 已提交
483 484 485
	sock_orphan(sk);
	sock_put(sk);

486
	return err;
S
Samuel Ortiz 已提交
487 488 489
}

static int llcp_sock_connect(struct socket *sock, struct sockaddr *_addr,
S
Samuel Ortiz 已提交
490
			     int len, int flags)
S
Samuel Ortiz 已提交
491 492 493 494 495 496 497 498 499 500 501
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	struct sockaddr_nfc_llcp *addr = (struct sockaddr_nfc_llcp *)_addr;
	struct nfc_dev *dev;
	struct nfc_llcp_local *local;
	int ret = 0;

	pr_debug("sock %p sk %p flags 0x%x\n", sock, sk, flags);

	if (!addr || len < sizeof(struct sockaddr_nfc) ||
502
	    addr->sa_family != AF_NFC)
S
Samuel Ortiz 已提交
503 504
		return -EINVAL;

505
	if (addr->service_name_len == 0 && addr->dsap == 0)
S
Samuel Ortiz 已提交
506 507 508
		return -EINVAL;

	pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr->dev_idx,
S
Samuel Ortiz 已提交
509
		 addr->target_idx, addr->nfc_protocol);
S
Samuel Ortiz 已提交
510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537 538

	lock_sock(sk);

	if (sk->sk_state == LLCP_CONNECTED) {
		ret = -EISCONN;
		goto error;
	}

	dev = nfc_get_device(addr->dev_idx);
	if (dev == NULL) {
		ret = -ENODEV;
		goto error;
	}

	local = nfc_llcp_find_local(dev);
	if (local == NULL) {
		ret = -ENODEV;
		goto put_dev;
	}

	device_lock(&dev->dev);
	if (dev->dep_link_up == false) {
		ret = -ENOLINK;
		device_unlock(&dev->dev);
		goto put_dev;
	}
	device_unlock(&dev->dev);

	if (local->rf_mode == NFC_RF_INITIATOR &&
S
Samuel Ortiz 已提交
539
	    addr->target_idx != local->target_idx) {
S
Samuel Ortiz 已提交
540 541 542 543 544
		ret = -ENOLINK;
		goto put_dev;
	}

	llcp_sock->dev = dev;
545
	llcp_sock->local = nfc_llcp_local_get(local);
546
	llcp_sock->miu = llcp_sock->local->remote_miu;
S
Samuel Ortiz 已提交
547 548 549 550 551
	llcp_sock->ssap = nfc_llcp_get_local_ssap(local);
	if (llcp_sock->ssap == LLCP_SAP_MAX) {
		ret = -ENOMEM;
		goto put_dev;
	}
552 553 554

	llcp_sock->reserved_ssap = llcp_sock->ssap;

S
Samuel Ortiz 已提交
555 556 557 558 559 560
	if (addr->service_name_len == 0)
		llcp_sock->dsap = addr->dsap;
	else
		llcp_sock->dsap = LLCP_SAP_SDP;
	llcp_sock->nfc_protocol = addr->nfc_protocol;
	llcp_sock->service_name_len = min_t(unsigned int,
S
Samuel Ortiz 已提交
561 562
					    addr->service_name_len,
					    NFC_LLCP_MAX_SERVICE_NAME);
S
Samuel Ortiz 已提交
563
	llcp_sock->service_name = kmemdup(addr->service_name,
S
Samuel Ortiz 已提交
564 565
					  llcp_sock->service_name_len,
					  GFP_KERNEL);
S
Samuel Ortiz 已提交
566

S
Samuel Ortiz 已提交
567
	nfc_llcp_sock_link(&local->connecting_sockets, sk);
S
Samuel Ortiz 已提交
568 569 570

	ret = nfc_llcp_send_connect(llcp_sock);
	if (ret)
S
Samuel Ortiz 已提交
571
		goto sock_unlink;
S
Samuel Ortiz 已提交
572

573 574 575
	ret = sock_wait_state(sk, LLCP_CONNECTED,
			      sock_sndtimeo(sk, flags & O_NONBLOCK));
	if (ret)
S
Samuel Ortiz 已提交
576
		goto sock_unlink;
S
Samuel Ortiz 已提交
577 578

	release_sock(sk);
579

S
Samuel Ortiz 已提交
580 581
	return 0;

S
Samuel Ortiz 已提交
582 583 584 585 586
sock_unlink:
	nfc_llcp_put_ssap(local, llcp_sock->ssap);

	nfc_llcp_sock_unlink(&local->connecting_sockets, sk);

S
Samuel Ortiz 已提交
587 588 589 590 591 592 593 594
put_dev:
	nfc_put_device(dev);

error:
	release_sock(sk);
	return ret;
}

595
static int llcp_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
S
Samuel Ortiz 已提交
596
			     struct msghdr *msg, size_t len)
597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	int ret;

	pr_debug("sock %p sk %p", sock, sk);

	ret = sock_error(sk);
	if (ret)
		return ret;

	if (msg->msg_flags & MSG_OOB)
		return -EOPNOTSUPP;

	lock_sock(sk);

613 614 615 616 617 618 619 620 621 622 623 624 625 626 627
	if (sk->sk_type == SOCK_DGRAM) {
		struct sockaddr_nfc_llcp *addr =
			(struct sockaddr_nfc_llcp *)msg->msg_name;

		if (msg->msg_namelen < sizeof(*addr)) {
			release_sock(sk);
			return -EINVAL;
		}

		release_sock(sk);

		return nfc_llcp_send_ui_frame(llcp_sock, addr->dsap, addr->ssap,
					      msg, len);
	}

628 629 630 631 632 633 634 635 636 637
	if (sk->sk_state != LLCP_CONNECTED) {
		release_sock(sk);
		return -ENOTCONN;
	}

	release_sock(sk);

	return nfc_llcp_send_i_frame(llcp_sock, msg, len);
}

S
Samuel Ortiz 已提交
638 639 640 641 642 643 644 645 646 647 648 649 650 651
static int llcp_sock_recvmsg(struct kiocb *iocb, struct socket *sock,
			     struct msghdr *msg, size_t len, int flags)
{
	int noblock = flags & MSG_DONTWAIT;
	struct sock *sk = sock->sk;
	unsigned int copied, rlen;
	struct sk_buff *skb, *cskb;
	int err = 0;

	pr_debug("%p %zu\n", sk, len);

	lock_sock(sk);

	if (sk->sk_state == LLCP_CLOSED &&
S
Samuel Ortiz 已提交
652
	    skb_queue_empty(&sk->sk_receive_queue)) {
S
Samuel Ortiz 已提交
653 654 655 656 657 658 659 660 661 662 663 664
		release_sock(sk);
		return 0;
	}

	release_sock(sk);

	if (flags & (MSG_OOB))
		return -EOPNOTSUPP;

	skb = skb_recv_datagram(sk, flags, noblock, &err);
	if (!skb) {
		pr_err("Recv datagram failed state %d %d %d",
S
Samuel Ortiz 已提交
665
		       sk->sk_state, err, sock_error(sk));
S
Samuel Ortiz 已提交
666 667 668 669 670 671 672

		if (sk->sk_shutdown & RCV_SHUTDOWN)
			return 0;

		return err;
	}

S
Samuel Ortiz 已提交
673
	rlen = skb->len;		/* real length of skb */
S
Samuel Ortiz 已提交
674 675 676
	copied = min_t(unsigned int, rlen, len);

	cskb = skb;
S
Samuel Ortiz 已提交
677
	if (skb_copy_datagram_iovec(cskb, 0, msg->msg_iov, copied)) {
S
Samuel Ortiz 已提交
678 679 680 681 682
		if (!(flags & MSG_PEEK))
			skb_queue_head(&sk->sk_receive_queue, skb);
		return -EFAULT;
	}

683 684
	sock_recv_timestamp(msg, sk, skb);

685 686
	if (sk->sk_type == SOCK_DGRAM && msg->msg_name) {
		struct nfc_llcp_ui_cb *ui_cb = nfc_llcp_ui_skb_cb(skb);
687 688
		struct sockaddr_nfc_llcp *sockaddr =
			(struct sockaddr_nfc_llcp *) msg->msg_name;
689

690
		msg->msg_namelen = sizeof(struct sockaddr_nfc_llcp);
691

692
		pr_debug("Datagram socket %d %d\n", ui_cb->dsap, ui_cb->ssap);
693

694 695 696 697
		sockaddr->sa_family = AF_NFC;
		sockaddr->nfc_protocol = NFC_PROTO_NFC_DEP;
		sockaddr->dsap = ui_cb->dsap;
		sockaddr->ssap = ui_cb->ssap;
698 699
	}

S
Samuel Ortiz 已提交
700 701 702 703
	/* Mark read part of skb as used */
	if (!(flags & MSG_PEEK)) {

		/* SOCK_STREAM: re-queue skb if it contains unreceived data */
704 705 706
		if (sk->sk_type == SOCK_STREAM ||
		    sk->sk_type == SOCK_DGRAM ||
		    sk->sk_type == SOCK_RAW) {
S
Samuel Ortiz 已提交
707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741
			skb_pull(skb, copied);
			if (skb->len) {
				skb_queue_head(&sk->sk_receive_queue, skb);
				goto done;
			}
		}

		kfree_skb(skb);
	}

	/* XXX Queue backlogged skbs */

done:
	/* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
	if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
		copied = rlen;

	return copied;
}

static const struct proto_ops llcp_sock_ops = {
	.family         = PF_NFC,
	.owner          = THIS_MODULE,
	.bind           = llcp_sock_bind,
	.connect        = llcp_sock_connect,
	.release        = llcp_sock_release,
	.socketpair     = sock_no_socketpair,
	.accept         = llcp_sock_accept,
	.getname        = llcp_sock_getname,
	.poll           = llcp_sock_poll,
	.ioctl          = sock_no_ioctl,
	.listen         = llcp_sock_listen,
	.shutdown       = sock_no_shutdown,
	.setsockopt     = sock_no_setsockopt,
	.getsockopt     = sock_no_getsockopt,
742
	.sendmsg        = llcp_sock_sendmsg,
S
Samuel Ortiz 已提交
743 744 745 746
	.recvmsg        = llcp_sock_recvmsg,
	.mmap           = sock_no_mmap,
};

T
Thierry Escande 已提交
747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766
static const struct proto_ops llcp_rawsock_ops = {
	.family         = PF_NFC,
	.owner          = THIS_MODULE,
	.bind           = llcp_raw_sock_bind,
	.connect        = sock_no_connect,
	.release        = llcp_sock_release,
	.socketpair     = sock_no_socketpair,
	.accept         = sock_no_accept,
	.getname        = llcp_sock_getname,
	.poll           = llcp_sock_poll,
	.ioctl          = sock_no_ioctl,
	.listen         = sock_no_listen,
	.shutdown       = sock_no_shutdown,
	.setsockopt     = sock_no_setsockopt,
	.getsockopt     = sock_no_getsockopt,
	.sendmsg        = sock_no_sendmsg,
	.recvmsg        = llcp_sock_recvmsg,
	.mmap           = sock_no_mmap,
};

S
Samuel Ortiz 已提交
767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804
static void llcp_sock_destruct(struct sock *sk)
{
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);

	pr_debug("%p\n", sk);

	if (sk->sk_state == LLCP_CONNECTED)
		nfc_put_device(llcp_sock->dev);

	skb_queue_purge(&sk->sk_receive_queue);

	nfc_llcp_sock_free(llcp_sock);

	if (!sock_flag(sk, SOCK_DEAD)) {
		pr_err("Freeing alive NFC LLCP socket %p\n", sk);
		return;
	}
}

struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp)
{
	struct sock *sk;
	struct nfc_llcp_sock *llcp_sock;

	sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto);
	if (!sk)
		return NULL;

	llcp_sock = nfc_llcp_sock(sk);

	sock_init_data(sock, sk);
	sk->sk_state = LLCP_CLOSED;
	sk->sk_protocol = NFC_SOCKPROTO_LLCP;
	sk->sk_type = type;
	sk->sk_destruct = llcp_sock_destruct;

	llcp_sock->ssap = 0;
	llcp_sock->dsap = LLCP_SAP_SDP;
805
	llcp_sock->rw = LLCP_DEFAULT_RW;
806
	llcp_sock->miu = LLCP_DEFAULT_MIU;
S
Samuel Ortiz 已提交
807 808 809
	llcp_sock->send_n = llcp_sock->send_ack_n = 0;
	llcp_sock->recv_n = llcp_sock->recv_ack_n = 0;
	llcp_sock->remote_ready = 1;
810
	llcp_sock->reserved_ssap = LLCP_SAP_MAX;
S
Samuel Ortiz 已提交
811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828
	skb_queue_head_init(&llcp_sock->tx_queue);
	skb_queue_head_init(&llcp_sock->tx_pending_queue);
	INIT_LIST_HEAD(&llcp_sock->accept_queue);

	if (sock != NULL)
		sock->state = SS_UNCONNECTED;

	return sk;
}

void nfc_llcp_sock_free(struct nfc_llcp_sock *sock)
{
	kfree(sock->service_name);

	skb_queue_purge(&sock->tx_queue);
	skb_queue_purge(&sock->tx_pending_queue);

	list_del_init(&sock->accept_queue);
829

S
Samuel Ortiz 已提交
830
	sock->parent = NULL;
831 832

	nfc_llcp_local_put(sock->local);
S
Samuel Ortiz 已提交
833 834 835
}

static int llcp_sock_create(struct net *net, struct socket *sock,
S
Samuel Ortiz 已提交
836
			    const struct nfc_protocol *nfc_proto)
S
Samuel Ortiz 已提交
837 838 839 840 841
{
	struct sock *sk;

	pr_debug("%p\n", sock);

T
Thierry Escande 已提交
842 843 844
	if (sock->type != SOCK_STREAM &&
	    sock->type != SOCK_DGRAM &&
	    sock->type != SOCK_RAW)
S
Samuel Ortiz 已提交
845 846
		return -ESOCKTNOSUPPORT;

T
Thierry Escande 已提交
847 848 849 850
	if (sock->type == SOCK_RAW)
		sock->ops = &llcp_rawsock_ops;
	else
		sock->ops = &llcp_sock_ops;
S
Samuel Ortiz 已提交
851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874

	sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC);
	if (sk == NULL)
		return -ENOMEM;

	return 0;
}

static const struct nfc_protocol llcp_nfc_proto = {
	.id	  = NFC_SOCKPROTO_LLCP,
	.proto    = &llcp_sock_proto,
	.owner    = THIS_MODULE,
	.create   = llcp_sock_create
};

int __init nfc_llcp_sock_init(void)
{
	return nfc_proto_register(&llcp_nfc_proto);
}

void nfc_llcp_sock_exit(void)
{
	nfc_proto_unregister(&llcp_nfc_proto);
}