llcp_sock.c 22.5 KB
Newer Older
S
Samuel Ortiz 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * 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
15
 * along with this program; if not, see <http://www.gnu.org/licenses/>.
S
Samuel Ortiz 已提交
16 17 18 19 20 21 22 23
 */

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

#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/nfc.h>
24
#include <linux/sched/signal.h>
S
Samuel Ortiz 已提交
25

26
#include "nfc.h"
S
Samuel Ortiz 已提交
27 28
#include "llcp.h"

29 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
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 已提交
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79
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;

80 81
	if (!addr || alen < offsetofend(struct sockaddr, sa_family) ||
	    addr->sa_family != AF_NFC)
S
Samuel Ortiz 已提交
82 83
		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);
122 123 124 125
	if (!llcp_sock->service_name) {
		ret = -ENOMEM;
		goto put_dev;
	}
S
Samuel Ortiz 已提交
126
	llcp_sock->ssap = nfc_llcp_get_sdp_ssap(local, llcp_sock);
127
	if (llcp_sock->ssap == LLCP_SAP_MAX) {
128 129
		kfree(llcp_sock->service_name);
		llcp_sock->service_name = NULL;
130
		ret = -EADDRINUSE;
S
Samuel Ortiz 已提交
131
		goto put_dev;
132
	}
S
Samuel Ortiz 已提交
133

134 135
	llcp_sock->reserved_ssap = llcp_sock->ssap;

S
Samuel Ortiz 已提交
136
	nfc_llcp_sock_link(&local->sockets, sk);
S
Samuel Ortiz 已提交
137 138 139 140 141 142 143 144 145 146 147 148 149

	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 已提交
150 151 152 153 154 155 156 157 158 159
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;

160 161
	if (!addr || alen < offsetofend(struct sockaddr, sa_family) ||
	    addr->sa_family != AF_NFC)
T
Thierry Escande 已提交
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 199 200 201 202 203 204
		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 已提交
205 206 207 208 209 210 211 212 213
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);

214 215
	if ((sock->type != SOCK_SEQPACKET && sock->type != SOCK_STREAM) ||
	    sk->sk_state != LLCP_BOUND) {
S
Samuel Ortiz 已提交
216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231
		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;
}

232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287
static int nfc_llcp_setsockopt(struct socket *sock, int level, int optname,
			       char __user *optval, unsigned int optlen)
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	u32 opt;
	int err = 0;

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

	if (level != SOL_NFC)
		return -ENOPROTOOPT;

	lock_sock(sk);

	switch (optname) {
	case NFC_LLCP_RW:
		if (sk->sk_state == LLCP_CONNECTED ||
		    sk->sk_state == LLCP_BOUND ||
		    sk->sk_state == LLCP_LISTEN) {
			err = -EINVAL;
			break;
		}

		if (get_user(opt, (u32 __user *) optval)) {
			err = -EFAULT;
			break;
		}

		if (opt > LLCP_MAX_RW) {
			err = -EINVAL;
			break;
		}

		llcp_sock->rw = (u8) opt;

		break;

	case NFC_LLCP_MIUX:
		if (sk->sk_state == LLCP_CONNECTED ||
		    sk->sk_state == LLCP_BOUND ||
		    sk->sk_state == LLCP_LISTEN) {
			err = -EINVAL;
			break;
		}

		if (get_user(opt, (u32 __user *) optval)) {
			err = -EFAULT;
			break;
		}

		if (opt > LLCP_MAX_MIUX) {
			err = -EINVAL;
			break;
		}

288
		llcp_sock->miux = cpu_to_be16((u16) opt);
289 290 291 292 293 294 295 296 297 298

		break;

	default:
		err = -ENOPROTOOPT;
		break;
	}

	release_sock(sk);

299 300 301
	pr_debug("%p rw %d miux %d\n", llcp_sock,
		 llcp_sock->rw, llcp_sock->miux);

302 303 304 305 306 307
	return err;
}

static int nfc_llcp_getsockopt(struct socket *sock, int level, int optname,
			       char __user *optval, int __user *optlen)
{
308
	struct nfc_llcp_local *local;
309 310 311
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
	int len, err = 0;
312
	u16 miux, remote_miu;
313
	u8 rw;
314 315 316 317 318 319 320 321 322

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

	if (level != SOL_NFC)
		return -ENOPROTOOPT;

	if (get_user(len, optlen))
		return -EFAULT;

323 324 325 326
	local = llcp_sock->local;
	if (!local)
		return -ENODEV;

327 328 329 330 331 332
	len = min_t(u32, len, sizeof(u32));

	lock_sock(sk);

	switch (optname) {
	case NFC_LLCP_RW:
333 334
		rw = llcp_sock->rw > LLCP_MAX_RW ? local->rw : llcp_sock->rw;
		if (put_user(rw, (u32 __user *) optval))
335 336 337 338 339
			err = -EFAULT;

		break;

	case NFC_LLCP_MIUX:
340 341 342 343
		miux = be16_to_cpu(llcp_sock->miux) > LLCP_MAX_MIUX ?
			be16_to_cpu(local->miux) : be16_to_cpu(llcp_sock->miux);

		if (put_user(miux, (u32 __user *) optval))
344 345 346 347
			err = -EFAULT;

		break;

348 349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364
	case NFC_LLCP_REMOTE_MIU:
		remote_miu = llcp_sock->remote_miu > LLCP_MAX_MIU ?
				local->remote_miu : llcp_sock->remote_miu;

		if (put_user(remote_miu, (u32 __user *) optval))
			err = -EFAULT;

		break;

	case NFC_LLCP_REMOTE_LTO:
		if (put_user(local->remote_lto / 10, (u32 __user *) optval))
			err = -EFAULT;

		break;

	case NFC_LLCP_REMOTE_RW:
		if (put_user(llcp_sock->remote_rw, (u32 __user *) optval))
365 366 367 368 369 370 371 372 373 374 375 376 377 378 379 380 381
			err = -EFAULT;

		break;

	default:
		err = -ENOPROTOOPT;
		break;
	}

	release_sock(sk);

	if (put_user(len, optlen))
		return -EFAULT;

	return err;
}

S
Samuel Ortiz 已提交
382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403
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 已提交
404
		      &llcp_sock_parent->accept_queue);
S
Samuel Ortiz 已提交
405 406 407 408 409
	llcp_sock->parent = parent;
	sk_acceptq_added(parent);
}

struct sock *nfc_llcp_accept_dequeue(struct sock *parent,
S
Samuel Ortiz 已提交
410
				     struct socket *newsock)
S
Samuel Ortiz 已提交
411 412 413 414 415 416 417
{
	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 已提交
418
				 accept_queue) {
S
Samuel Ortiz 已提交
419 420 421 422 423 424 425 426 427 428
		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) {
429 430 431
			list_del_init(&lsk->accept_queue);
			sock_put(sk);

S
Samuel Ortiz 已提交
432 433 434 435 436 437 438
			if (newsock)
				sock_graft(sk, newsock);

			release_sock(sk);

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

439 440
			sk_acceptq_removed(parent);

S
Samuel Ortiz 已提交
441 442 443 444 445 446 447 448 449 450
			return sk;
		}

		release_sock(sk);
	}

	return NULL;
}

static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
451
			    int flags, bool kern)
S
Samuel Ortiz 已提交
452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503
{
	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 已提交
504
static int llcp_sock_getname(struct socket *sock, struct sockaddr *uaddr,
505
			     int peer)
S
Samuel Ortiz 已提交
506 507 508
{
	struct sock *sk = sock->sk;
	struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
S
Samuel Ortiz 已提交
509
	DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, llcp_addr, uaddr);
S
Samuel Ortiz 已提交
510

511 512 513
	if (llcp_sock == NULL || llcp_sock->dev == NULL)
		return -EBADFD;

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

517
	memset(llcp_addr, 0, sizeof(*llcp_addr));
S
Samuel Ortiz 已提交
518

519 520 521 522 523
	lock_sock(sk);
	if (!llcp_sock->dev) {
		release_sock(sk);
		return -EBADFD;
	}
524
	llcp_addr->sa_family = AF_NFC;
S
Samuel Ortiz 已提交
525
	llcp_addr->dev_idx = llcp_sock->dev->idx;
S
Samuel Ortiz 已提交
526
	llcp_addr->target_idx = llcp_sock->target_idx;
527
	llcp_addr->nfc_protocol = llcp_sock->nfc_protocol;
S
Samuel Ortiz 已提交
528 529 530 531
	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 已提交
532
	       llcp_addr->service_name_len);
533
	release_sock(sk);
S
Samuel Ortiz 已提交
534

535
	return sizeof(struct sockaddr_nfc_llcp);
S
Samuel Ortiz 已提交
536 537
}

A
Al Viro 已提交
538
static inline __poll_t llcp_accept_poll(struct sock *parent)
S
Samuel Ortiz 已提交
539
{
540
	struct nfc_llcp_sock *llcp_sock, *parent_sock;
S
Samuel Ortiz 已提交
541 542 543 544
	struct sock *sk;

	parent_sock = nfc_llcp_sock(parent);

545 546
	list_for_each_entry(llcp_sock, &parent_sock->accept_queue,
			    accept_queue) {
S
Samuel Ortiz 已提交
547 548 549
		sk = &llcp_sock->sk;

		if (sk->sk_state == LLCP_CONNECTED)
550
			return EPOLLIN | EPOLLRDNORM;
S
Samuel Ortiz 已提交
551 552 553 554 555
	}

	return 0;
}

556 557
static __poll_t llcp_sock_poll(struct file *file, struct socket *sock,
				   poll_table *wait)
S
Samuel Ortiz 已提交
558 559
{
	struct sock *sk = sock->sk;
A
Al Viro 已提交
560
	__poll_t mask = 0;
S
Samuel Ortiz 已提交
561 562 563

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

564
	sock_poll_wait(file, sock, wait);
565

S
Samuel Ortiz 已提交
566 567 568
	if (sk->sk_state == LLCP_LISTEN)
		return llcp_accept_poll(sk);

569
	if (sk->sk_err || !skb_queue_empty_lockless(&sk->sk_error_queue))
570 571
		mask |= EPOLLERR |
			(sock_flag(sk, SOCK_SELECT_ERR_QUEUE) ? EPOLLPRI : 0);
S
Samuel Ortiz 已提交
572

573
	if (!skb_queue_empty_lockless(&sk->sk_receive_queue))
574
		mask |= EPOLLIN | EPOLLRDNORM;
S
Samuel Ortiz 已提交
575 576

	if (sk->sk_state == LLCP_CLOSED)
577
		mask |= EPOLLHUP;
S
Samuel Ortiz 已提交
578

S
Samuel Ortiz 已提交
579
	if (sk->sk_shutdown & RCV_SHUTDOWN)
580
		mask |= EPOLLRDHUP | EPOLLIN | EPOLLRDNORM;
S
Samuel Ortiz 已提交
581 582

	if (sk->sk_shutdown == SHUTDOWN_MASK)
583
		mask |= EPOLLHUP;
S
Samuel Ortiz 已提交
584

585
	if (sock_writeable(sk) && sk->sk_state == LLCP_CONNECTED)
586
		mask |= EPOLLOUT | EPOLLWRNORM | EPOLLWRBAND;
S
Samuel Ortiz 已提交
587
	else
588
		sk_set_bit(SOCKWQ_ASYNC_NOSPACE, sk);
S
Samuel Ortiz 已提交
589 590 591

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

S
Samuel Ortiz 已提交
592 593 594 595 596 597 598 599
	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);
600
	int err = 0;
S
Samuel Ortiz 已提交
601 602 603 604 605 606 607

	if (!sk)
		return 0;

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

	local = llcp_sock->local;
608 609 610 611
	if (local == NULL) {
		err = -ENODEV;
		goto out;
	}
S
Samuel Ortiz 已提交
612 613 614 615 616

	lock_sock(sk);

	/* Send a DISC */
	if (sk->sk_state == LLCP_CONNECTED)
617
		nfc_llcp_send_disconnect(llcp_sock);
S
Samuel Ortiz 已提交
618 619 620 621 622 623

	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 已提交
624
					 accept_queue) {
S
Samuel Ortiz 已提交
625 626 627
			accept_sk = &lsk->sk;
			lock_sock(accept_sk);

628
			nfc_llcp_send_disconnect(lsk);
S
Samuel Ortiz 已提交
629 630 631 632 633 634
			nfc_llcp_accept_unlink(accept_sk);

			release_sock(accept_sk);
		}
	}

635 636
	if (llcp_sock->reserved_ssap < LLCP_SAP_MAX)
		nfc_llcp_put_ssap(llcp_sock->local, llcp_sock->ssap);
S
Samuel Ortiz 已提交
637 638 639

	release_sock(sk);

640 641 642 643 644 645 646
	/* Keep this sock alive and therefore do not remove it from the sockets
	 * list until the DISC PDU has been actually sent. Otherwise we would
	 * reply with DM PDUs before sending the DISC one.
	 */
	if (sk->sk_state == LLCP_DISCONNECTING)
		return err;

T
Thierry Escande 已提交
647 648 649 650
	if (sock->type == SOCK_RAW)
		nfc_llcp_sock_unlink(&local->raw_sockets, sk);
	else
		nfc_llcp_sock_unlink(&local->sockets, sk);
S
Samuel Ortiz 已提交
651

652
out:
S
Samuel Ortiz 已提交
653 654 655
	sock_orphan(sk);
	sock_put(sk);

656
	return err;
S
Samuel Ortiz 已提交
657 658 659
}

static int llcp_sock_connect(struct socket *sock, struct sockaddr *_addr,
S
Samuel Ortiz 已提交
660
			     int len, int flags)
S
Samuel Ortiz 已提交
661 662 663 664 665 666 667 668 669 670
{
	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);

671
	if (!addr || len < sizeof(*addr) || addr->sa_family != AF_NFC)
S
Samuel Ortiz 已提交
672 673
		return -EINVAL;

674
	if (addr->service_name_len == 0 && addr->dsap == 0)
S
Samuel Ortiz 已提交
675 676 677
		return -EINVAL;

	pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr->dev_idx,
S
Samuel Ortiz 已提交
678
		 addr->target_idx, addr->nfc_protocol);
S
Samuel Ortiz 已提交
679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707

	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 已提交
708
	    addr->target_idx != local->target_idx) {
S
Samuel Ortiz 已提交
709 710 711 712 713
		ret = -ENOLINK;
		goto put_dev;
	}

	llcp_sock->dev = dev;
714
	llcp_sock->local = nfc_llcp_local_get(local);
S
Samuel Ortiz 已提交
715 716 717 718 719
	llcp_sock->ssap = nfc_llcp_get_local_ssap(local);
	if (llcp_sock->ssap == LLCP_SAP_MAX) {
		ret = -ENOMEM;
		goto put_dev;
	}
720 721 722

	llcp_sock->reserved_ssap = llcp_sock->ssap;

S
Samuel Ortiz 已提交
723 724 725 726 727 728
	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 已提交
729 730
					    addr->service_name_len,
					    NFC_LLCP_MAX_SERVICE_NAME);
S
Samuel Ortiz 已提交
731
	llcp_sock->service_name = kmemdup(addr->service_name,
S
Samuel Ortiz 已提交
732 733
					  llcp_sock->service_name_len,
					  GFP_KERNEL);
S
Samuel Ortiz 已提交
734

S
Samuel Ortiz 已提交
735
	nfc_llcp_sock_link(&local->connecting_sockets, sk);
S
Samuel Ortiz 已提交
736 737 738

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

741 742
	sk->sk_state = LLCP_CONNECTING;

743 744
	ret = sock_wait_state(sk, LLCP_CONNECTED,
			      sock_sndtimeo(sk, flags & O_NONBLOCK));
745
	if (ret && ret != -EINPROGRESS)
S
Samuel Ortiz 已提交
746
		goto sock_unlink;
S
Samuel Ortiz 已提交
747 748

	release_sock(sk);
749

750
	return ret;
S
Samuel Ortiz 已提交
751

S
Samuel Ortiz 已提交
752 753 754 755 756
sock_unlink:
	nfc_llcp_put_ssap(local, llcp_sock->ssap);

	nfc_llcp_sock_unlink(&local->connecting_sockets, sk);

S
Samuel Ortiz 已提交
757 758 759 760 761 762 763 764
put_dev:
	nfc_put_device(dev);

error:
	release_sock(sk);
	return ret;
}

765 766
static int llcp_sock_sendmsg(struct socket *sock, struct msghdr *msg,
			     size_t len)
767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
{
	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);

783
	if (sk->sk_type == SOCK_DGRAM) {
784 785
		DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, addr,
				 msg->msg_name);
786 787 788 789 790 791 792 793 794 795 796 797

		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);
	}

798 799 800 801 802 803 804 805 806 807
	if (sk->sk_state != LLCP_CONNECTED) {
		release_sock(sk);
		return -ENOTCONN;
	}

	release_sock(sk);

	return nfc_llcp_send_i_frame(llcp_sock, msg, len);
}

808 809
static int llcp_sock_recvmsg(struct socket *sock, struct msghdr *msg,
			     size_t len, int flags)
S
Samuel Ortiz 已提交
810 811 812 813 814 815 816 817 818 819 820 821
{
	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 已提交
822
	    skb_queue_empty(&sk->sk_receive_queue)) {
S
Samuel Ortiz 已提交
823 824 825 826 827 828 829 830 831 832 833 834
		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 已提交
835
		       sk->sk_state, err, sock_error(sk));
S
Samuel Ortiz 已提交
836 837 838 839 840 841 842

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

		return err;
	}

S
Samuel Ortiz 已提交
843
	rlen = skb->len;		/* real length of skb */
S
Samuel Ortiz 已提交
844 845 846
	copied = min_t(unsigned int, rlen, len);

	cskb = skb;
847
	if (skb_copy_datagram_msg(cskb, 0, msg, copied)) {
S
Samuel Ortiz 已提交
848 849 850 851 852
		if (!(flags & MSG_PEEK))
			skb_queue_head(&sk->sk_receive_queue, skb);
		return -EFAULT;
	}

853 854
	sock_recv_timestamp(msg, sk, skb);

855 856
	if (sk->sk_type == SOCK_DGRAM && msg->msg_name) {
		struct nfc_llcp_ui_cb *ui_cb = nfc_llcp_ui_skb_cb(skb);
857 858
		DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, sockaddr,
				 msg->msg_name);
859

860
		msg->msg_namelen = sizeof(struct sockaddr_nfc_llcp);
861

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

864
		memset(sockaddr, 0, sizeof(*sockaddr));
865 866 867 868
		sockaddr->sa_family = AF_NFC;
		sockaddr->nfc_protocol = NFC_PROTO_NFC_DEP;
		sockaddr->dsap = ui_cb->dsap;
		sockaddr->ssap = ui_cb->ssap;
869 870
	}

S
Samuel Ortiz 已提交
871 872 873 874
	/* Mark read part of skb as used */
	if (!(flags & MSG_PEEK)) {

		/* SOCK_STREAM: re-queue skb if it contains unreceived data */
875 876 877
		if (sk->sk_type == SOCK_STREAM ||
		    sk->sk_type == SOCK_DGRAM ||
		    sk->sk_type == SOCK_RAW) {
S
Samuel Ortiz 已提交
878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906
			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,
907
	.poll           = llcp_sock_poll,
S
Samuel Ortiz 已提交
908 909 910
	.ioctl          = sock_no_ioctl,
	.listen         = llcp_sock_listen,
	.shutdown       = sock_no_shutdown,
911 912
	.setsockopt     = nfc_llcp_setsockopt,
	.getsockopt     = nfc_llcp_getsockopt,
913
	.sendmsg        = llcp_sock_sendmsg,
S
Samuel Ortiz 已提交
914 915 916 917
	.recvmsg        = llcp_sock_recvmsg,
	.mmap           = sock_no_mmap,
};

T
Thierry Escande 已提交
918 919 920 921 922 923 924 925 926
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,
927
	.poll           = llcp_sock_poll,
T
Thierry Escande 已提交
928 929 930 931 932 933 934 935 936 937
	.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 已提交
938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956
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;
	}
}

957
struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp, int kern)
S
Samuel Ortiz 已提交
958 959 960 961
{
	struct sock *sk;
	struct nfc_llcp_sock *llcp_sock;

962
	sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto, kern);
S
Samuel Ortiz 已提交
963 964 965 966 967 968 969 970 971 972 973 974 975
	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;
976
	llcp_sock->rw = LLCP_MAX_RW + 1;
977
	llcp_sock->miux = cpu_to_be16(LLCP_MAX_MIUX + 1);
S
Samuel Ortiz 已提交
978 979 980
	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;
981
	llcp_sock->reserved_ssap = LLCP_SAP_MAX;
982
	nfc_llcp_socket_remote_param_init(llcp_sock);
S
Samuel Ortiz 已提交
983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000
	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);
1001

S
Samuel Ortiz 已提交
1002
	sock->parent = NULL;
1003 1004

	nfc_llcp_local_put(sock->local);
S
Samuel Ortiz 已提交
1005 1006 1007
}

static int llcp_sock_create(struct net *net, struct socket *sock,
1008
			    const struct nfc_protocol *nfc_proto, int kern)
S
Samuel Ortiz 已提交
1009 1010 1011 1012 1013
{
	struct sock *sk;

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

T
Thierry Escande 已提交
1014 1015 1016
	if (sock->type != SOCK_STREAM &&
	    sock->type != SOCK_DGRAM &&
	    sock->type != SOCK_RAW)
S
Samuel Ortiz 已提交
1017 1018
		return -ESOCKTNOSUPPORT;

1019 1020 1021
	if (sock->type == SOCK_RAW) {
		if (!capable(CAP_NET_RAW))
			return -EPERM;
T
Thierry Escande 已提交
1022
		sock->ops = &llcp_rawsock_ops;
1023
	} else {
T
Thierry Escande 已提交
1024
		sock->ops = &llcp_sock_ops;
1025
	}
S
Samuel Ortiz 已提交
1026

1027
	sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC, kern);
S
Samuel Ortiz 已提交
1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049
	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);
}