socket.c 62.6 KB
Newer Older
P
Per Liden 已提交
1
/*
2
 * net/tipc/socket.c: TIPC socket API
3
 *
4
 * Copyright (c) 2001-2007, 2012-2014, Ericsson AB
5
 * Copyright (c) 2004-2008, 2010-2013, Wind River Systems
P
Per Liden 已提交
6 7
 * All rights reserved.
 *
P
Per Liden 已提交
8
 * Redistribution and use in source and binary forms, with or without
P
Per Liden 已提交
9 10
 * modification, are permitted provided that the following conditions are met:
 *
P
Per Liden 已提交
11 12 13 14 15 16 17 18
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the names of the copyright holders nor the names of its
 *    contributors may be used to endorse or promote products derived from
 *    this software without specific prior written permission.
P
Per Liden 已提交
19
 *
P
Per Liden 已提交
20 21 22 23 24 25 26 27 28 29 30 31 32 33
 * Alternatively, this software may be distributed under the terms of the
 * GNU General Public License ("GPL") version 2 as published by the Free
 * Software Foundation.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
P
Per Liden 已提交
34 35 36 37
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include "core.h"
J
Jon Paul Maloy 已提交
38
#include "ref.h"
39
#include "port.h"
40
#include "name_table.h"
E
Erik Hugne 已提交
41
#include "node.h"
42
#include "link.h"
43
#include <linux/export.h>
44
#include "config.h"
45

P
Per Liden 已提交
46 47 48
#define SS_LISTENING	-1	/* socket is listening */
#define SS_READY	-2	/* socket is connectionless */

49
#define CONN_TIMEOUT_DEFAULT	8000	/* default connect timeout = 8s */
50
#define CONN_PROBING_INTERVAL 3600000	/* [ms] => 1 h */
51
#define TIPC_FWD_MSG	        1
P
Per Liden 已提交
52

53
static int tipc_backlog_rcv(struct sock *sk, struct sk_buff *skb);
54
static void tipc_data_ready(struct sock *sk);
55
static void tipc_write_space(struct sock *sk);
56 57
static int tipc_release(struct socket *sock);
static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags);
58
static int tipc_wait_for_sndmsg(struct socket *sock, long *timeo_p);
59
static void tipc_sk_timeout(unsigned long ref);
J
Jon Paul Maloy 已提交
60 61 62 63
static int tipc_sk_publish(struct tipc_port *port, uint scope,
			   struct tipc_name_seq const *seq);
static int tipc_sk_withdraw(struct tipc_port *port, uint scope,
			    struct tipc_name_seq const *seq);
P
Per Liden 已提交
64

65 66 67
static const struct proto_ops packet_ops;
static const struct proto_ops stream_ops;
static const struct proto_ops msg_ops;
P
Per Liden 已提交
68 69

static struct proto tipc_proto;
70
static struct proto tipc_proto_kern;
P
Per Liden 已提交
71

72
/*
73 74 75 76 77 78 79 80 81 82 83 84 85 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 114 115 116
 * Revised TIPC socket locking policy:
 *
 * Most socket operations take the standard socket lock when they start
 * and hold it until they finish (or until they need to sleep).  Acquiring
 * this lock grants the owner exclusive access to the fields of the socket
 * data structures, with the exception of the backlog queue.  A few socket
 * operations can be done without taking the socket lock because they only
 * read socket information that never changes during the life of the socket.
 *
 * Socket operations may acquire the lock for the associated TIPC port if they
 * need to perform an operation on the port.  If any routine needs to acquire
 * both the socket lock and the port lock it must take the socket lock first
 * to avoid the risk of deadlock.
 *
 * The dispatcher handling incoming messages cannot grab the socket lock in
 * the standard fashion, since invoked it runs at the BH level and cannot block.
 * Instead, it checks to see if the socket lock is currently owned by someone,
 * and either handles the message itself or adds it to the socket's backlog
 * queue; in the latter case the queued message is processed once the process
 * owning the socket lock releases it.
 *
 * NOTE: Releasing the socket lock while an operation is sleeping overcomes
 * the problem of a blocked socket operation preventing any other operations
 * from occurring.  However, applications must be careful if they have
 * multiple threads trying to send (or receive) on the same socket, as these
 * operations might interfere with each other.  For example, doing a connect
 * and a receive at the same time might allow the receive to consume the
 * ACK message meant for the connect.  While additional work could be done
 * to try and overcome this, it doesn't seem to be worthwhile at the present.
 *
 * NOTE: Releasing the socket lock while an operation is sleeping also ensures
 * that another operation that must be performed in a non-blocking manner is
 * not delayed for very long because the lock has already been taken.
 *
 * NOTE: This code assumes that certain fields of a port/socket pair are
 * constant over its lifetime; such fields can be examined without taking
 * the socket lock and/or port lock, and do not need to be re-read even
 * after resuming processing after waiting.  These fields include:
 *   - socket type
 *   - pointer to socket sk structure (aka tipc_sock structure)
 *   - pointer to port structure
 *   - port reference
 */

117 118
#include "socket.h"

119 120 121 122
/**
 * advance_rx_queue - discard first buffer in socket receive queue
 *
 * Caller must hold socket lock
P
Per Liden 已提交
123
 */
124
static void advance_rx_queue(struct sock *sk)
P
Per Liden 已提交
125
{
126
	kfree_skb(__skb_dequeue(&sk->sk_receive_queue));
P
Per Liden 已提交
127 128 129
}

/**
130 131 132
 * reject_rx_queue - reject all buffers in socket receive queue
 *
 * Caller must hold socket lock
P
Per Liden 已提交
133
 */
134
static void reject_rx_queue(struct sock *sk)
P
Per Liden 已提交
135
{
136
	struct sk_buff *buf;
137
	u32 dnode;
138

139 140
	while ((buf = __skb_dequeue(&sk->sk_receive_queue))) {
		if (tipc_msg_reverse(buf, &dnode, TIPC_ERR_NO_PORT))
141
			tipc_link_xmit(buf, dnode, 0);
142
	}
P
Per Liden 已提交
143 144
}

J
Jon Paul Maloy 已提交
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
/* tipc_sk_peer_msg - verify if message was sent by connected port's peer
 *
 * Handles cases where the node's network address has changed from
 * the default of <0.0.0> to its configured setting.
 */
static bool tipc_sk_peer_msg(struct tipc_sock *tsk, struct tipc_msg *msg)
{
	u32 peer_port = tipc_port_peerport(&tsk->port);
	u32 orig_node;
	u32 peer_node;

	if (unlikely(!tsk->port.connected))
		return false;

	if (unlikely(msg_origport(msg) != peer_port))
		return false;

	orig_node = msg_orignode(msg);
	peer_node = tipc_port_peernode(&tsk->port);

	if (likely(orig_node == peer_node))
		return true;

	if (!orig_node && (peer_node == tipc_own_addr))
		return true;

	if (!peer_node && (orig_node == tipc_own_addr))
		return true;

	return false;
}

P
Per Liden 已提交
177
/**
178
 * tipc_sk_create - create a TIPC socket
179
 * @net: network namespace (must be default network)
P
Per Liden 已提交
180 181
 * @sock: pre-allocated socket structure
 * @protocol: protocol indicator (must be 0)
182
 * @kern: caused by kernel or by userspace?
183
 *
184 185
 * This routine creates additional data structures used by the TIPC socket,
 * initializes them, and links them together.
P
Per Liden 已提交
186 187 188
 *
 * Returns 0 on success, errno otherwise
 */
189 190
static int tipc_sk_create(struct net *net, struct socket *sock,
			  int protocol, int kern)
P
Per Liden 已提交
191
{
192 193
	const struct proto_ops *ops;
	socket_state state;
P
Per Liden 已提交
194
	struct sock *sk;
195 196
	struct tipc_sock *tsk;
	struct tipc_port *port;
197
	struct tipc_msg *msg;
198
	u32 ref;
199 200

	/* Validate arguments */
P
Per Liden 已提交
201 202 203 204 205
	if (unlikely(protocol != 0))
		return -EPROTONOSUPPORT;

	switch (sock->type) {
	case SOCK_STREAM:
206 207
		ops = &stream_ops;
		state = SS_UNCONNECTED;
P
Per Liden 已提交
208 209
		break;
	case SOCK_SEQPACKET:
210 211
		ops = &packet_ops;
		state = SS_UNCONNECTED;
P
Per Liden 已提交
212 213 214
		break;
	case SOCK_DGRAM:
	case SOCK_RDM:
215 216
		ops = &msg_ops;
		state = SS_READY;
P
Per Liden 已提交
217
		break;
218 219
	default:
		return -EPROTOTYPE;
P
Per Liden 已提交
220 221
	}

222
	/* Allocate socket's protocol area */
223 224 225 226 227
	if (!kern)
		sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto);
	else
		sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto_kern);

228
	if (sk == NULL)
P
Per Liden 已提交
229 230
		return -ENOMEM;

231 232
	tsk = tipc_sk(sk);
	port = &tsk->port;
J
Jon Paul Maloy 已提交
233
	ref = tipc_ref_acquire(tsk);
234
	if (!ref) {
235
		pr_warn("Socket create failed; reference table exhausted\n");
236 237
		return -ENOMEM;
	}
238 239 240 241 242 243 244 245
	port->max_pkt = MAX_PKT_DEFAULT;
	port->ref = ref;
	INIT_LIST_HEAD(&port->publications);

	msg = &port->phdr;
	tipc_msg_init(msg, TIPC_LOW_IMPORTANCE, TIPC_NAMED_MSG,
		      NAMED_H_SIZE, 0);
	msg_set_origport(msg, ref);
P
Per Liden 已提交
246

247 248 249 250
	/* Finish initializing socket data structures */
	sock->ops = ops;
	sock->state = state;
	sock_init_data(sock, sk);
251
	k_init_timer(&port->timer, (Handler)tipc_sk_timeout, ref);
252
	sk->sk_backlog_rcv = tipc_backlog_rcv;
253
	sk->sk_rcvbuf = sysctl_tipc_rmem[1];
254 255
	sk->sk_data_ready = tipc_data_ready;
	sk->sk_write_space = tipc_write_space;
256
	tsk->conn_timeout = CONN_TIMEOUT_DEFAULT;
257
	tsk->sent_unacked = 0;
258
	atomic_set(&tsk->dupl_rcvcnt, 0);
259

260
	if (sock->state == SS_READY) {
261
		tipc_port_set_unreturnable(port, true);
262
		if (sock->type == SOCK_DGRAM)
263
			tipc_port_set_unreliable(port, true);
264
	}
P
Per Liden 已提交
265 266 267
	return 0;
}

268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301
/**
 * tipc_sock_create_local - create TIPC socket from inside TIPC module
 * @type: socket type - SOCK_RDM or SOCK_SEQPACKET
 *
 * We cannot use sock_creat_kern here because it bumps module user count.
 * Since socket owner and creator is the same module we must make sure
 * that module count remains zero for module local sockets, otherwise
 * we cannot do rmmod.
 *
 * Returns 0 on success, errno otherwise
 */
int tipc_sock_create_local(int type, struct socket **res)
{
	int rc;

	rc = sock_create_lite(AF_TIPC, type, 0, res);
	if (rc < 0) {
		pr_err("Failed to create kernel socket\n");
		return rc;
	}
	tipc_sk_create(&init_net, *res, 0, 1);

	return 0;
}

/**
 * tipc_sock_release_local - release socket created by tipc_sock_create_local
 * @sock: the socket to be released.
 *
 * Module reference count is not incremented when such sockets are created,
 * so we must keep it from being decremented when they are released.
 */
void tipc_sock_release_local(struct socket *sock)
{
302
	tipc_release(sock);
303 304 305 306 307 308 309 310 311 312 313 314 315 316 317
	sock->ops = NULL;
	sock_release(sock);
}

/**
 * tipc_sock_accept_local - accept a connection on a socket created
 * with tipc_sock_create_local. Use this function to avoid that
 * module reference count is inadvertently incremented.
 *
 * @sock:    the accepting socket
 * @newsock: reference to the new socket to be created
 * @flags:   socket flags
 */

int tipc_sock_accept_local(struct socket *sock, struct socket **newsock,
318
			   int flags)
319 320 321 322 323 324 325 326 327
{
	struct sock *sk = sock->sk;
	int ret;

	ret = sock_create_lite(sk->sk_family, sk->sk_type,
			       sk->sk_protocol, newsock);
	if (ret < 0)
		return ret;

328
	ret = tipc_accept(sock, *newsock, flags);
329 330 331 332 333 334 335 336
	if (ret < 0) {
		sock_release(*newsock);
		return ret;
	}
	(*newsock)->ops = sock->ops;
	return ret;
}

P
Per Liden 已提交
337
/**
338
 * tipc_release - destroy a TIPC socket
P
Per Liden 已提交
339 340 341 342 343 344 345
 * @sock: socket to destroy
 *
 * This routine cleans up any messages that are still queued on the socket.
 * For DGRAM and RDM socket types, all queued messages are rejected.
 * For SEQPACKET and STREAM socket types, the first message is rejected
 * and any others are discarded.  (If the first message on a STREAM socket
 * is partially-read, it is discarded and the next one is rejected instead.)
346
 *
P
Per Liden 已提交
347 348 349 350 351 352
 * NOTE: Rejected messages are not necessarily returned to the sender!  They
 * are returned or discarded according to the "destination droppable" setting
 * specified for the message by the sender.
 *
 * Returns 0 on success, errno otherwise
 */
353
static int tipc_release(struct socket *sock)
P
Per Liden 已提交
354 355
{
	struct sock *sk = sock->sk;
356 357
	struct tipc_sock *tsk;
	struct tipc_port *port;
P
Per Liden 已提交
358
	struct sk_buff *buf;
359
	u32 dnode;
P
Per Liden 已提交
360

361 362 363 364 365
	/*
	 * Exit if socket isn't fully initialized (occurs when a failed accept()
	 * releases a pre-allocated child socket that was never used)
	 */
	if (sk == NULL)
P
Per Liden 已提交
366
		return 0;
367

368 369
	tsk = tipc_sk(sk);
	port = &tsk->port;
370 371 372 373 374 375
	lock_sock(sk);

	/*
	 * Reject all unreceived messages, except on an active connection
	 * (which disconnects locally & sends a 'FIN+' to peer)
	 */
376
	dnode = tipc_port_peernode(port);
P
Per Liden 已提交
377
	while (sock->state != SS_DISCONNECTING) {
378 379
		buf = __skb_dequeue(&sk->sk_receive_queue);
		if (buf == NULL)
P
Per Liden 已提交
380
			break;
Y
Ying Xue 已提交
381
		if (TIPC_SKB_CB(buf)->handle != NULL)
382
			kfree_skb(buf);
383 384 385 386
		else {
			if ((sock->state == SS_CONNECTING) ||
			    (sock->state == SS_CONNECTED)) {
				sock->state = SS_DISCONNECTING;
387
				port->connected = 0;
388
				tipc_node_remove_conn(dnode, port->ref);
389
			}
390
			if (tipc_msg_reverse(buf, &dnode, TIPC_ERR_NO_PORT))
391
				tipc_link_xmit(buf, dnode, 0);
392
		}
P
Per Liden 已提交
393 394
	}

J
Jon Paul Maloy 已提交
395
	tipc_sk_withdraw(port, 0, NULL);
396 397 398 399 400 401 402 403 404 405 406 407
	tipc_ref_discard(port->ref);
	k_cancel_timer(&port->timer);
	if (port->connected) {
		buf = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE, TIPC_CONN_MSG,
				      SHORT_H_SIZE, 0, dnode, tipc_own_addr,
				      tipc_port_peerport(port),
				      port->ref, TIPC_ERR_NO_PORT);
		if (buf)
			tipc_link_xmit(buf, dnode, port->ref);
		tipc_node_remove_conn(dnode, port->ref);
	}
	k_term_timer(&port->timer);
P
Per Liden 已提交
408

409
	/* Discard any remaining (connection-based) messages in receive queue */
410
	__skb_queue_purge(&sk->sk_receive_queue);
P
Per Liden 已提交
411

412 413 414
	/* Reject any messages that accumulated in backlog queue */
	sock->state = SS_DISCONNECTING;
	release_sock(sk);
P
Per Liden 已提交
415
	sock_put(sk);
416
	sock->sk = NULL;
P
Per Liden 已提交
417

418
	return 0;
P
Per Liden 已提交
419 420 421
}

/**
422
 * tipc_bind - associate or disassocate TIPC name(s) with a socket
P
Per Liden 已提交
423 424 425
 * @sock: socket structure
 * @uaddr: socket address describing name(s) and desired operation
 * @uaddr_len: size of socket address data structure
426
 *
P
Per Liden 已提交
427 428 429
 * Name and name sequence binding is indicated using a positive scope value;
 * a negative scope value unbinds the specified name.  Specifying no name
 * (i.e. a socket address length of 0) unbinds all names from the socket.
430
 *
P
Per Liden 已提交
431
 * Returns 0 on success, errno otherwise
432 433 434
 *
 * NOTE: This routine doesn't need to take the socket lock since it doesn't
 *       access any non-constant socket information.
P
Per Liden 已提交
435
 */
436 437
static int tipc_bind(struct socket *sock, struct sockaddr *uaddr,
		     int uaddr_len)
P
Per Liden 已提交
438
{
439
	struct sock *sk = sock->sk;
P
Per Liden 已提交
440
	struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
441
	struct tipc_sock *tsk = tipc_sk(sk);
442
	int res = -EINVAL;
P
Per Liden 已提交
443

444 445
	lock_sock(sk);
	if (unlikely(!uaddr_len)) {
J
Jon Paul Maloy 已提交
446
		res = tipc_sk_withdraw(&tsk->port, 0, NULL);
447 448
		goto exit;
	}
449

450 451 452 453 454 455 456 457
	if (uaddr_len < sizeof(struct sockaddr_tipc)) {
		res = -EINVAL;
		goto exit;
	}
	if (addr->family != AF_TIPC) {
		res = -EAFNOSUPPORT;
		goto exit;
	}
P
Per Liden 已提交
458 459 460

	if (addr->addrtype == TIPC_ADDR_NAME)
		addr->addr.nameseq.upper = addr->addr.nameseq.lower;
461 462 463 464
	else if (addr->addrtype != TIPC_ADDR_NAMESEQ) {
		res = -EAFNOSUPPORT;
		goto exit;
	}
465

466
	if ((addr->addr.nameseq.type < TIPC_RESERVED_TYPES) &&
467
	    (addr->addr.nameseq.type != TIPC_TOP_SRV) &&
468 469 470 471
	    (addr->addr.nameseq.type != TIPC_CFG_SRV)) {
		res = -EACCES;
		goto exit;
	}
472

473
	res = (addr->scope > 0) ?
J
Jon Paul Maloy 已提交
474 475
		tipc_sk_publish(&tsk->port, addr->scope, &addr->addr.nameseq) :
		tipc_sk_withdraw(&tsk->port, -addr->scope, &addr->addr.nameseq);
476 477 478
exit:
	release_sock(sk);
	return res;
P
Per Liden 已提交
479 480
}

481
/**
482
 * tipc_getname - get port ID of socket or peer socket
P
Per Liden 已提交
483 484 485
 * @sock: socket structure
 * @uaddr: area for returned socket address
 * @uaddr_len: area for returned length of socket address
486
 * @peer: 0 = own ID, 1 = current peer ID, 2 = current/former peer ID
487
 *
P
Per Liden 已提交
488
 * Returns 0 on success, errno otherwise
489
 *
490 491
 * NOTE: This routine doesn't need to take the socket lock since it only
 *       accesses socket information that is unchanging (or which changes in
492
 *       a completely predictable manner).
P
Per Liden 已提交
493
 */
494 495
static int tipc_getname(struct socket *sock, struct sockaddr *uaddr,
			int *uaddr_len, int peer)
P
Per Liden 已提交
496 497
{
	struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
498
	struct tipc_sock *tsk = tipc_sk(sock->sk);
P
Per Liden 已提交
499

500
	memset(addr, 0, sizeof(*addr));
501
	if (peer) {
502 503 504
		if ((sock->state != SS_CONNECTED) &&
			((peer != 2) || (sock->state != SS_DISCONNECTING)))
			return -ENOTCONN;
505 506
		addr->addr.id.ref = tipc_port_peerport(&tsk->port);
		addr->addr.id.node = tipc_port_peernode(&tsk->port);
507
	} else {
508
		addr->addr.id.ref = tsk->port.ref;
A
Allan Stephens 已提交
509
		addr->addr.id.node = tipc_own_addr;
510
	}
P
Per Liden 已提交
511 512 513 514 515 516 517

	*uaddr_len = sizeof(*addr);
	addr->addrtype = TIPC_ADDR_ID;
	addr->family = AF_TIPC;
	addr->scope = 0;
	addr->addr.name.domain = 0;

518
	return 0;
P
Per Liden 已提交
519 520 521
}

/**
522
 * tipc_poll - read and possibly block on pollmask
P
Per Liden 已提交
523 524 525 526
 * @file: file structure associated with the socket
 * @sock: socket for which to calculate the poll bits
 * @wait: ???
 *
527 528 529 530 531 532 533 534 535
 * Returns pollmask value
 *
 * COMMENTARY:
 * It appears that the usual socket locking mechanisms are not useful here
 * since the pollmask info is potentially out-of-date the moment this routine
 * exits.  TCP and other protocols seem to rely on higher level poll routines
 * to handle any preventable race conditions, so TIPC will do the same ...
 *
 * TIPC sets the returned events as follows:
536 537 538 539
 *
 * socket state		flags set
 * ------------		---------
 * unconnected		no read flags
540
 *			POLLOUT if port is not congested
541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557 558 559
 *
 * connecting		POLLIN/POLLRDNORM if ACK/NACK in rx queue
 *			no write flags
 *
 * connected		POLLIN/POLLRDNORM if data in rx queue
 *			POLLOUT if port is not congested
 *
 * disconnecting	POLLIN/POLLRDNORM/POLLHUP
 *			no write flags
 *
 * listening		POLLIN if SYN in rx queue
 *			no write flags
 *
 * ready		POLLIN/POLLRDNORM if data in rx queue
 * [connectionless]	POLLOUT (since port cannot be congested)
 *
 * IMPORTANT: The fact that a read or write operation is indicated does NOT
 * imply that the operation will succeed, merely that it should be performed
 * and will not block.
P
Per Liden 已提交
560
 */
561 562
static unsigned int tipc_poll(struct file *file, struct socket *sock,
			      poll_table *wait)
P
Per Liden 已提交
563
{
564
	struct sock *sk = sock->sk;
565
	struct tipc_sock *tsk = tipc_sk(sk);
566
	u32 mask = 0;
567

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

570
	switch ((int)sock->state) {
571
	case SS_UNCONNECTED:
572
		if (!tsk->link_cong)
573 574
			mask |= POLLOUT;
		break;
575 576
	case SS_READY:
	case SS_CONNECTED:
577
		if (!tsk->link_cong && !tipc_sk_conn_cong(tsk))
578 579 580 581 582 583 584 585 586 587 588
			mask |= POLLOUT;
		/* fall thru' */
	case SS_CONNECTING:
	case SS_LISTENING:
		if (!skb_queue_empty(&sk->sk_receive_queue))
			mask |= (POLLIN | POLLRDNORM);
		break;
	case SS_DISCONNECTING:
		mask = (POLLIN | POLLRDNORM | POLLHUP);
		break;
	}
589 590

	return mask;
P
Per Liden 已提交
591 592
}

593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623
/**
 * tipc_sendmcast - send multicast message
 * @sock: socket structure
 * @seq: destination address
 * @iov: message data to send
 * @dsz: total length of message data
 * @timeo: timeout to wait for wakeup
 *
 * Called from function tipc_sendmsg(), which has done all sanity checks
 * Returns the number of bytes sent on success, or errno
 */
static int tipc_sendmcast(struct  socket *sock, struct tipc_name_seq *seq,
			  struct iovec *iov, size_t dsz, long timeo)
{
	struct sock *sk = sock->sk;
	struct tipc_msg *mhdr = &tipc_sk(sk)->port.phdr;
	struct sk_buff *buf;
	uint mtu;
	int rc;

	msg_set_type(mhdr, TIPC_MCAST_MSG);
	msg_set_lookup_scope(mhdr, TIPC_CLUSTER_SCOPE);
	msg_set_destport(mhdr, 0);
	msg_set_destnode(mhdr, 0);
	msg_set_nametype(mhdr, seq->type);
	msg_set_namelower(mhdr, seq->lower);
	msg_set_nameupper(mhdr, seq->upper);
	msg_set_hdr_sz(mhdr, MCAST_H_SIZE);

new_mtu:
	mtu = tipc_bclink_get_mtu();
624
	rc = tipc_msg_build(mhdr, iov, 0, dsz, mtu, &buf);
625 626 627 628 629 630 631 632 633 634 635 636 637
	if (unlikely(rc < 0))
		return rc;

	do {
		rc = tipc_bclink_xmit(buf);
		if (likely(rc >= 0)) {
			rc = dsz;
			break;
		}
		if (rc == -EMSGSIZE)
			goto new_mtu;
		if (rc != -ELINKCONG)
			break;
638
		tipc_sk(sk)->link_cong = 1;
639 640 641 642 643 644 645
		rc = tipc_wait_for_sndmsg(sock, &timeo);
		if (rc)
			kfree_skb_list(buf);
	} while (!rc);
	return rc;
}

646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
/* tipc_sk_mcast_rcv - Deliver multicast message to all destination sockets
 */
void tipc_sk_mcast_rcv(struct sk_buff *buf)
{
	struct tipc_msg *msg = buf_msg(buf);
	struct tipc_port_list dports = {0, NULL, };
	struct tipc_port_list *item;
	struct sk_buff *b;
	uint i, last, dst = 0;
	u32 scope = TIPC_CLUSTER_SCOPE;

	if (in_own_node(msg_orignode(msg)))
		scope = TIPC_NODE_SCOPE;

	/* Create destination port list: */
	tipc_nametbl_mc_translate(msg_nametype(msg),
				  msg_namelower(msg),
				  msg_nameupper(msg),
				  scope,
				  &dports);
	last = dports.count;
	if (!last) {
		kfree_skb(buf);
		return;
	}

	for (item = &dports; item; item = item->next) {
		for (i = 0; i < PLSIZE && ++dst <= last; i++) {
			b = (dst != last) ? skb_clone(buf, GFP_ATOMIC) : buf;
			if (!b) {
				pr_warn("Failed do clone mcast rcv buffer\n");
				continue;
			}
			msg_set_destport(msg, item->ports[i]);
			tipc_sk_rcv(b);
		}
	}
	tipc_port_list_free(&dports);
}

686 687 688 689 690 691 692 693
/**
 * tipc_sk_proto_rcv - receive a connection mng protocol message
 * @tsk: receiving socket
 * @dnode: node to send response message to, if any
 * @buf: buffer containing protocol message
 * Returns 0 (TIPC_OK) if message was consumed, 1 (TIPC_FWD_MSG) if
 * (CONN_PROBE_REPLY) message should be forwarded.
 */
694 695
static int tipc_sk_proto_rcv(struct tipc_sock *tsk, u32 *dnode,
			     struct sk_buff *buf)
696 697 698
{
	struct tipc_msg *msg = buf_msg(buf);
	struct tipc_port *port = &tsk->port;
699
	int conn_cong;
700 701

	/* Ignore if connection cannot be validated: */
J
Jon Paul Maloy 已提交
702
	if (!tipc_sk_peer_msg(tsk, msg))
703 704 705 706 707
		goto exit;

	port->probing_state = TIPC_CONN_OK;

	if (msg_type(msg) == CONN_ACK) {
708 709 710
		conn_cong = tipc_sk_conn_cong(tsk);
		tsk->sent_unacked -= msg_msgcnt(msg);
		if (conn_cong)
711
			tsk->sk.sk_write_space(&tsk->sk);
712 713 714 715 716 717 718 719 720 721 722 723
	} else if (msg_type(msg) == CONN_PROBE) {
		if (!tipc_msg_reverse(buf, dnode, TIPC_OK))
			return TIPC_OK;
		msg_set_type(msg, CONN_PROBE_REPLY);
		return TIPC_FWD_MSG;
	}
	/* Do nothing if msg_type() == CONN_PROBE_REPLY */
exit:
	kfree_skb(buf);
	return TIPC_OK;
}

724
/**
P
Per Liden 已提交
725 726 727
 * dest_name_check - verify user is permitted to send to specified port name
 * @dest: destination address
 * @m: descriptor for message to be sent
728
 *
P
Per Liden 已提交
729 730
 * Prevents restricted configuration commands from being issued by
 * unauthorized users.
731
 *
P
Per Liden 已提交
732 733
 * Returns 0 if permission is granted, otherwise errno
 */
S
Sam Ravnborg 已提交
734
static int dest_name_check(struct sockaddr_tipc *dest, struct msghdr *m)
P
Per Liden 已提交
735 736 737
{
	struct tipc_cfg_msg_hdr hdr;

738 739
	if (unlikely(dest->addrtype == TIPC_ADDR_ID))
		return 0;
740 741 742 743 744 745
	if (likely(dest->addr.name.name.type >= TIPC_RESERVED_TYPES))
		return 0;
	if (likely(dest->addr.name.name.type == TIPC_TOP_SRV))
		return 0;
	if (likely(dest->addr.name.name.type != TIPC_CFG_SRV))
		return -EACCES;
P
Per Liden 已提交
746

747 748
	if (!m->msg_iovlen || (m->msg_iov[0].iov_len < sizeof(hdr)))
		return -EMSGSIZE;
749
	if (copy_from_user(&hdr, m->msg_iov[0].iov_base, sizeof(hdr)))
P
Per Liden 已提交
750
		return -EFAULT;
751
	if ((ntohs(hdr.tcm_type) & 0xC000) && (!capable(CAP_NET_ADMIN)))
P
Per Liden 已提交
752
		return -EACCES;
753

P
Per Liden 已提交
754 755 756
	return 0;
}

757 758 759
static int tipc_wait_for_sndmsg(struct socket *sock, long *timeo_p)
{
	struct sock *sk = sock->sk;
760
	struct tipc_sock *tsk = tipc_sk(sk);
761 762 763 764 765 766 767 768 769 770 771 772 773 774 775
	DEFINE_WAIT(wait);
	int done;

	do {
		int err = sock_error(sk);
		if (err)
			return err;
		if (sock->state == SS_DISCONNECTING)
			return -EPIPE;
		if (!*timeo_p)
			return -EAGAIN;
		if (signal_pending(current))
			return sock_intr_errno(*timeo_p);

		prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
776
		done = sk_wait_event(sk, timeo_p, !tsk->link_cong);
777 778 779 780 781
		finish_wait(sk_sleep(sk), &wait);
	} while (!done);
	return 0;
}

P
Per Liden 已提交
782
/**
783
 * tipc_sendmsg - send message in connectionless manner
784
 * @iocb: if NULL, indicates that socket lock is already held
P
Per Liden 已提交
785 786
 * @sock: socket structure
 * @m: message to send
787
 * @dsz: amount of user data to be sent
788
 *
P
Per Liden 已提交
789
 * Message must have an destination specified explicitly.
790
 * Used for SOCK_RDM and SOCK_DGRAM messages,
P
Per Liden 已提交
791 792
 * and for 'SYN' messages on SOCK_SEQPACKET and SOCK_STREAM connections.
 * (Note: 'SYN+' is prohibited on SOCK_STREAM.)
793
 *
P
Per Liden 已提交
794 795
 * Returns the number of bytes sent on success, or errno otherwise
 */
796
static int tipc_sendmsg(struct kiocb *iocb, struct socket *sock,
797
			struct msghdr *m, size_t dsz)
P
Per Liden 已提交
798
{
799
	DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
800
	struct sock *sk = sock->sk;
801
	struct tipc_sock *tsk = tipc_sk(sk);
802
	struct tipc_port *port = &tsk->port;
803 804 805 806 807 808
	struct tipc_msg *mhdr = &port->phdr;
	struct iovec *iov = m->msg_iov;
	u32 dnode, dport;
	struct sk_buff *buf;
	struct tipc_name_seq *seq = &dest->addr.nameseq;
	u32 mtu;
809
	long timeo;
810
	int rc = -EINVAL;
P
Per Liden 已提交
811 812 813

	if (unlikely(!dest))
		return -EDESTADDRREQ;
814

815 816
	if (unlikely((m->msg_namelen < sizeof(*dest)) ||
		     (dest->family != AF_TIPC)))
P
Per Liden 已提交
817
		return -EINVAL;
818 819

	if (dsz > TIPC_MAX_USER_MSG_SIZE)
820
		return -EMSGSIZE;
P
Per Liden 已提交
821

822 823 824
	if (iocb)
		lock_sock(sk);

825
	if (unlikely(sock->state != SS_READY)) {
826
		if (sock->state == SS_LISTENING) {
827
			rc = -EPIPE;
828 829 830
			goto exit;
		}
		if (sock->state != SS_UNCONNECTED) {
831
			rc = -EISCONN;
832 833
			goto exit;
		}
834
		if (tsk->port.published) {
835
			rc = -EOPNOTSUPP;
836 837
			goto exit;
		}
838
		if (dest->addrtype == TIPC_ADDR_NAME) {
839 840
			tsk->port.conn_type = dest->addr.name.name.type;
			tsk->port.conn_instance = dest->addr.name.name.instance;
841
		}
P
Per Liden 已提交
842
	}
843 844 845
	rc = dest_name_check(dest, m);
	if (rc)
		goto exit;
P
Per Liden 已提交
846

847
	timeo = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868

	if (dest->addrtype == TIPC_ADDR_MCAST) {
		rc = tipc_sendmcast(sock, seq, iov, dsz, timeo);
		goto exit;
	} else if (dest->addrtype == TIPC_ADDR_NAME) {
		u32 type = dest->addr.name.name.type;
		u32 inst = dest->addr.name.name.instance;
		u32 domain = dest->addr.name.domain;

		dnode = domain;
		msg_set_type(mhdr, TIPC_NAMED_MSG);
		msg_set_hdr_sz(mhdr, NAMED_H_SIZE);
		msg_set_nametype(mhdr, type);
		msg_set_nameinst(mhdr, inst);
		msg_set_lookup_scope(mhdr, tipc_addr_scope(domain));
		dport = tipc_nametbl_translate(type, inst, &dnode);
		msg_set_destnode(mhdr, dnode);
		msg_set_destport(mhdr, dport);
		if (unlikely(!dport && !dnode)) {
			rc = -EHOSTUNREACH;
			goto exit;
869
		}
870 871 872 873 874 875 876 877 878 879 880
	} else if (dest->addrtype == TIPC_ADDR_ID) {
		dnode = dest->addr.id.node;
		msg_set_type(mhdr, TIPC_DIRECT_MSG);
		msg_set_lookup_scope(mhdr, 0);
		msg_set_destnode(mhdr, dnode);
		msg_set_destport(mhdr, dest->addr.id.ref);
		msg_set_hdr_sz(mhdr, BASIC_H_SIZE);
	}

new_mtu:
	mtu = tipc_node_get_mtu(dnode, tsk->port.ref);
881
	rc = tipc_msg_build(mhdr, iov, 0, dsz, mtu, &buf);
882 883 884 885
	if (rc < 0)
		goto exit;

	do {
886
		TIPC_SKB_CB(buf)->wakeup_pending = tsk->link_cong;
887
		rc = tipc_link_xmit(buf, dnode, tsk->port.ref);
888 889
		if (likely(rc >= 0)) {
			if (sock->state != SS_READY)
890
				sock->state = SS_CONNECTING;
891
			rc = dsz;
892
			break;
893
		}
894 895 896
		if (rc == -EMSGSIZE)
			goto new_mtu;
		if (rc != -ELINKCONG)
897
			break;
898
		tsk->link_cong = 1;
899
		rc = tipc_wait_for_sndmsg(sock, &timeo);
900 901
		if (rc)
			kfree_skb_list(buf);
902
	} while (!rc);
903 904 905
exit:
	if (iocb)
		release_sock(sk);
906 907

	return rc;
P
Per Liden 已提交
908 909
}

910 911 912
static int tipc_wait_for_sndpkt(struct socket *sock, long *timeo_p)
{
	struct sock *sk = sock->sk;
913
	struct tipc_sock *tsk = tipc_sk(sk);
914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931
	DEFINE_WAIT(wait);
	int done;

	do {
		int err = sock_error(sk);
		if (err)
			return err;
		if (sock->state == SS_DISCONNECTING)
			return -EPIPE;
		else if (sock->state != SS_CONNECTED)
			return -ENOTCONN;
		if (!*timeo_p)
			return -EAGAIN;
		if (signal_pending(current))
			return sock_intr_errno(*timeo_p);

		prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
		done = sk_wait_event(sk, timeo_p,
932 933 934
				     (!tsk->link_cong &&
				      !tipc_sk_conn_cong(tsk)) ||
				     !tsk->port.connected);
935 936 937 938 939
		finish_wait(sk_sleep(sk), &wait);
	} while (!done);
	return 0;
}

940
/**
941 942
 * tipc_send_stream - send stream-oriented data
 * @iocb: (unused)
P
Per Liden 已提交
943
 * @sock: socket structure
944 945
 * @m: data to send
 * @dsz: total length of data to be transmitted
946
 *
947
 * Used for SOCK_STREAM data.
948
 *
949 950
 * Returns the number of bytes sent on success (or partial success),
 * or errno if no data sent
P
Per Liden 已提交
951
 */
952 953
static int tipc_send_stream(struct kiocb *iocb, struct socket *sock,
			    struct msghdr *m, size_t dsz)
P
Per Liden 已提交
954
{
955
	struct sock *sk = sock->sk;
956
	struct tipc_sock *tsk = tipc_sk(sk);
957 958 959
	struct tipc_port *port = &tsk->port;
	struct tipc_msg *mhdr = &port->phdr;
	struct sk_buff *buf;
960
	DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
961 962
	u32 ref = port->ref;
	int rc = -EINVAL;
963
	long timeo;
964 965
	u32 dnode;
	uint mtu, send, sent = 0;
P
Per Liden 已提交
966 967

	/* Handle implied connection establishment */
968 969 970
	if (unlikely(dest)) {
		rc = tipc_sendmsg(iocb, sock, m, dsz);
		if (dsz && (dsz == rc))
971
			tsk->sent_unacked = 1;
972 973 974
		return rc;
	}
	if (dsz > (uint)INT_MAX)
975 976
		return -EMSGSIZE;

977 978
	if (iocb)
		lock_sock(sk);
P
Per Liden 已提交
979

980 981
	if (unlikely(sock->state != SS_CONNECTED)) {
		if (sock->state == SS_DISCONNECTING)
982
			rc = -EPIPE;
983
		else
984
			rc = -ENOTCONN;
985 986
		goto exit;
	}
987

988
	timeo = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
989 990 991 992 993
	dnode = tipc_port_peernode(port);

next:
	mtu = port->max_pkt;
	send = min_t(uint, dsz - sent, TIPC_MAX_USER_MSG_SIZE);
994
	rc = tipc_msg_build(mhdr, m->msg_iov, sent, send, mtu, &buf);
995 996
	if (unlikely(rc < 0))
		goto exit;
997
	do {
998
		if (likely(!tipc_sk_conn_cong(tsk))) {
999
			rc = tipc_link_xmit(buf, dnode, ref);
1000
			if (likely(!rc)) {
1001
				tsk->sent_unacked++;
1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012
				sent += send;
				if (sent == dsz)
					break;
				goto next;
			}
			if (rc == -EMSGSIZE) {
				port->max_pkt = tipc_node_get_mtu(dnode, ref);
				goto next;
			}
			if (rc != -ELINKCONG)
				break;
1013
			tsk->link_cong = 1;
1014 1015
		}
		rc = tipc_wait_for_sndpkt(sock, &timeo);
1016 1017
		if (rc)
			kfree_skb_list(buf);
1018
	} while (!rc);
1019
exit:
1020 1021
	if (iocb)
		release_sock(sk);
1022
	return sent ? sent : rc;
P
Per Liden 已提交
1023 1024
}

1025
/**
1026 1027
 * tipc_send_packet - send a connection-oriented message
 * @iocb: if NULL, indicates that socket lock is already held
P
Per Liden 已提交
1028
 * @sock: socket structure
1029 1030
 * @m: message to send
 * @dsz: length of data to be transmitted
1031
 *
1032
 * Used for SOCK_SEQPACKET messages.
1033
 *
1034
 * Returns the number of bytes sent on success, or errno otherwise
P
Per Liden 已提交
1035
 */
1036 1037
static int tipc_send_packet(struct kiocb *iocb, struct socket *sock,
			    struct msghdr *m, size_t dsz)
P
Per Liden 已提交
1038
{
1039 1040
	if (dsz > TIPC_MAX_USER_MSG_SIZE)
		return -EMSGSIZE;
P
Per Liden 已提交
1041

1042
	return tipc_send_stream(iocb, sock, m, dsz);
P
Per Liden 已提交
1043 1044
}

1045
/* tipc_sk_finish_conn - complete the setup of a connection
P
Per Liden 已提交
1046
 */
1047 1048
static void tipc_sk_finish_conn(struct tipc_port *port, u32 peer_port,
				u32 peer_node)
P
Per Liden 已提交
1049
{
1050
	struct tipc_msg *msg = &port->phdr;
P
Per Liden 已提交
1051

1052 1053 1054 1055 1056
	msg_set_destnode(msg, peer_node);
	msg_set_destport(msg, peer_port);
	msg_set_type(msg, TIPC_CONN_MSG);
	msg_set_lookup_scope(msg, 0);
	msg_set_hdr_sz(msg, SHORT_H_SIZE);
1057

1058 1059 1060 1061 1062 1063
	port->probing_interval = CONN_PROBING_INTERVAL;
	port->probing_state = TIPC_CONN_OK;
	port->connected = 1;
	k_start_timer(&port->timer, port->probing_interval);
	tipc_node_add_conn(peer_node, port->ref, peer_port);
	port->max_pkt = tipc_node_get_mtu(peer_node, port->ref);
P
Per Liden 已提交
1064 1065 1066 1067 1068 1069
}

/**
 * set_orig_addr - capture sender's address for received message
 * @m: descriptor for message info
 * @msg: received message header
1070
 *
P
Per Liden 已提交
1071 1072
 * Note: Address is not captured if not requested by receiver.
 */
S
Sam Ravnborg 已提交
1073
static void set_orig_addr(struct msghdr *m, struct tipc_msg *msg)
P
Per Liden 已提交
1074
{
1075
	DECLARE_SOCKADDR(struct sockaddr_tipc *, addr, m->msg_name);
P
Per Liden 已提交
1076

1077
	if (addr) {
P
Per Liden 已提交
1078 1079
		addr->family = AF_TIPC;
		addr->addrtype = TIPC_ADDR_ID;
1080
		memset(&addr->addr, 0, sizeof(addr->addr));
P
Per Liden 已提交
1081 1082
		addr->addr.id.ref = msg_origport(msg);
		addr->addr.id.node = msg_orignode(msg);
1083 1084
		addr->addr.name.domain = 0;	/* could leave uninitialized */
		addr->scope = 0;		/* could leave uninitialized */
P
Per Liden 已提交
1085 1086 1087 1088 1089
		m->msg_namelen = sizeof(struct sockaddr_tipc);
	}
}

/**
1090
 * anc_data_recv - optionally capture ancillary data for received message
P
Per Liden 已提交
1091 1092 1093
 * @m: descriptor for message info
 * @msg: received message header
 * @tport: TIPC port associated with message
1094
 *
P
Per Liden 已提交
1095
 * Note: Ancillary data is not captured if not requested by receiver.
1096
 *
P
Per Liden 已提交
1097 1098
 * Returns 0 if successful, otherwise errno
 */
S
Sam Ravnborg 已提交
1099
static int anc_data_recv(struct msghdr *m, struct tipc_msg *msg,
1100
			 struct tipc_port *tport)
P
Per Liden 已提交
1101 1102 1103 1104
{
	u32 anc_data[3];
	u32 err;
	u32 dest_type;
1105
	int has_name;
P
Per Liden 已提交
1106 1107 1108 1109 1110 1111 1112 1113 1114 1115
	int res;

	if (likely(m->msg_controllen == 0))
		return 0;

	/* Optionally capture errored message object(s) */
	err = msg ? msg_errcode(msg) : 0;
	if (unlikely(err)) {
		anc_data[0] = err;
		anc_data[1] = msg_data_sz(msg);
1116 1117
		res = put_cmsg(m, SOL_TIPC, TIPC_ERRINFO, 8, anc_data);
		if (res)
P
Per Liden 已提交
1118
			return res;
1119 1120 1121 1122 1123 1124
		if (anc_data[1]) {
			res = put_cmsg(m, SOL_TIPC, TIPC_RETDATA, anc_data[1],
				       msg_data(msg));
			if (res)
				return res;
		}
P
Per Liden 已提交
1125 1126 1127 1128 1129 1130
	}

	/* Optionally capture message destination object */
	dest_type = msg ? msg_type(msg) : TIPC_DIRECT_MSG;
	switch (dest_type) {
	case TIPC_NAMED_MSG:
1131
		has_name = 1;
P
Per Liden 已提交
1132 1133 1134 1135 1136
		anc_data[0] = msg_nametype(msg);
		anc_data[1] = msg_namelower(msg);
		anc_data[2] = msg_namelower(msg);
		break;
	case TIPC_MCAST_MSG:
1137
		has_name = 1;
P
Per Liden 已提交
1138 1139 1140 1141 1142
		anc_data[0] = msg_nametype(msg);
		anc_data[1] = msg_namelower(msg);
		anc_data[2] = msg_nameupper(msg);
		break;
	case TIPC_CONN_MSG:
1143
		has_name = (tport->conn_type != 0);
P
Per Liden 已提交
1144 1145 1146 1147 1148
		anc_data[0] = tport->conn_type;
		anc_data[1] = tport->conn_instance;
		anc_data[2] = tport->conn_instance;
		break;
	default:
1149
		has_name = 0;
P
Per Liden 已提交
1150
	}
1151 1152 1153 1154 1155
	if (has_name) {
		res = put_cmsg(m, SOL_TIPC, TIPC_DESTNAME, 12, anc_data);
		if (res)
			return res;
	}
P
Per Liden 已提交
1156 1157 1158 1159

	return 0;
}

1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
static void tipc_sk_send_ack(struct tipc_port *port, uint ack)
{
	struct sk_buff *buf = NULL;
	struct tipc_msg *msg;
	u32 peer_port = tipc_port_peerport(port);
	u32 dnode = tipc_port_peernode(port);

	if (!port->connected)
		return;
	buf = tipc_msg_create(CONN_MANAGER, CONN_ACK, INT_H_SIZE, 0, dnode,
			      tipc_own_addr, peer_port, port->ref, TIPC_OK);
	if (!buf)
		return;
	msg = buf_msg(buf);
	msg_set_msgcnt(msg, ack);
	tipc_link_xmit(buf, dnode, msg_link_selector(msg));
}

1178
static int tipc_wait_for_rcvmsg(struct socket *sock, long *timeop)
Y
Ying Xue 已提交
1179 1180 1181
{
	struct sock *sk = sock->sk;
	DEFINE_WAIT(wait);
1182
	long timeo = *timeop;
Y
Ying Xue 已提交
1183 1184 1185 1186
	int err;

	for (;;) {
		prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
1187
		if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
Y
Ying Xue 已提交
1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206
			if (sock->state == SS_DISCONNECTING) {
				err = -ENOTCONN;
				break;
			}
			release_sock(sk);
			timeo = schedule_timeout(timeo);
			lock_sock(sk);
		}
		err = 0;
		if (!skb_queue_empty(&sk->sk_receive_queue))
			break;
		err = sock_intr_errno(timeo);
		if (signal_pending(current))
			break;
		err = -EAGAIN;
		if (!timeo)
			break;
	}
	finish_wait(sk_sleep(sk), &wait);
1207
	*timeop = timeo;
Y
Ying Xue 已提交
1208 1209 1210
	return err;
}

1211
/**
1212
 * tipc_recvmsg - receive packet-oriented message
P
Per Liden 已提交
1213 1214 1215 1216
 * @iocb: (unused)
 * @m: descriptor for message info
 * @buf_len: total size of user buffer area
 * @flags: receive flags
1217
 *
P
Per Liden 已提交
1218 1219 1220 1221 1222
 * Used for SOCK_DGRAM, SOCK_RDM, and SOCK_SEQPACKET messages.
 * If the complete message doesn't fit in user area, truncate it.
 *
 * Returns size of returned message data, errno otherwise
 */
1223 1224
static int tipc_recvmsg(struct kiocb *iocb, struct socket *sock,
			struct msghdr *m, size_t buf_len, int flags)
P
Per Liden 已提交
1225
{
1226
	struct sock *sk = sock->sk;
1227 1228
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
1229 1230
	struct sk_buff *buf;
	struct tipc_msg *msg;
Y
Ying Xue 已提交
1231
	long timeo;
P
Per Liden 已提交
1232 1233 1234 1235
	unsigned int sz;
	u32 err;
	int res;

1236
	/* Catch invalid receive requests */
P
Per Liden 已提交
1237 1238 1239
	if (unlikely(!buf_len))
		return -EINVAL;

1240
	lock_sock(sk);
P
Per Liden 已提交
1241

1242 1243
	if (unlikely(sock->state == SS_UNCONNECTED)) {
		res = -ENOTCONN;
P
Per Liden 已提交
1244 1245 1246
		goto exit;
	}

Y
Ying Xue 已提交
1247
	timeo = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
1248
restart:
P
Per Liden 已提交
1249

1250
	/* Look for a message in receive queue; wait if necessary */
1251
	res = tipc_wait_for_rcvmsg(sock, &timeo);
Y
Ying Xue 已提交
1252 1253
	if (res)
		goto exit;
P
Per Liden 已提交
1254

1255 1256
	/* Look at first message in receive queue */
	buf = skb_peek(&sk->sk_receive_queue);
P
Per Liden 已提交
1257 1258 1259 1260 1261 1262
	msg = buf_msg(buf);
	sz = msg_data_sz(msg);
	err = msg_errcode(msg);

	/* Discard an empty non-errored message & try again */
	if ((!sz) && (!err)) {
1263
		advance_rx_queue(sk);
P
Per Liden 已提交
1264 1265 1266 1267 1268 1269 1270
		goto restart;
	}

	/* Capture sender's address (optional) */
	set_orig_addr(m, msg);

	/* Capture ancillary data (optional) */
1271
	res = anc_data_recv(m, msg, port);
1272
	if (res)
P
Per Liden 已提交
1273 1274 1275 1276 1277 1278 1279 1280
		goto exit;

	/* Capture message data (if valid) & compute return value (always) */
	if (!err) {
		if (unlikely(buf_len < sz)) {
			sz = buf_len;
			m->msg_flags |= MSG_TRUNC;
		}
1281 1282 1283
		res = skb_copy_datagram_iovec(buf, msg_hdr_sz(msg),
					      m->msg_iov, sz);
		if (res)
P
Per Liden 已提交
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295
			goto exit;
		res = sz;
	} else {
		if ((sock->state == SS_READY) ||
		    ((err == TIPC_CONN_SHUTDOWN) || m->msg_control))
			res = 0;
		else
			res = -ECONNRESET;
	}

	/* Consume received message (optional) */
	if (likely(!(flags & MSG_PEEK))) {
1296
		if ((sock->state != SS_READY) &&
1297
		    (++tsk->rcv_unacked >= TIPC_CONNACK_INTV)) {
1298
			tipc_sk_send_ack(port, tsk->rcv_unacked);
1299 1300
			tsk->rcv_unacked = 0;
		}
1301
		advance_rx_queue(sk);
1302
	}
P
Per Liden 已提交
1303
exit:
1304
	release_sock(sk);
P
Per Liden 已提交
1305 1306 1307
	return res;
}

1308
/**
1309
 * tipc_recv_stream - receive stream-oriented data
P
Per Liden 已提交
1310 1311 1312 1313
 * @iocb: (unused)
 * @m: descriptor for message info
 * @buf_len: total size of user buffer area
 * @flags: receive flags
1314 1315
 *
 * Used for SOCK_STREAM messages only.  If not enough data is available
P
Per Liden 已提交
1316 1317 1318 1319
 * will optionally wait for more; never truncates data.
 *
 * Returns size of returned message data, errno otherwise
 */
1320 1321
static int tipc_recv_stream(struct kiocb *iocb, struct socket *sock,
			    struct msghdr *m, size_t buf_len, int flags)
P
Per Liden 已提交
1322
{
1323
	struct sock *sk = sock->sk;
1324 1325
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
1326 1327
	struct sk_buff *buf;
	struct tipc_msg *msg;
Y
Ying Xue 已提交
1328
	long timeo;
P
Per Liden 已提交
1329
	unsigned int sz;
1330
	int sz_to_copy, target, needed;
P
Per Liden 已提交
1331 1332
	int sz_copied = 0;
	u32 err;
1333
	int res = 0;
P
Per Liden 已提交
1334

1335
	/* Catch invalid receive attempts */
P
Per Liden 已提交
1336 1337 1338
	if (unlikely(!buf_len))
		return -EINVAL;

1339
	lock_sock(sk);
P
Per Liden 已提交
1340

Y
Ying Xue 已提交
1341
	if (unlikely(sock->state == SS_UNCONNECTED)) {
1342
		res = -ENOTCONN;
P
Per Liden 已提交
1343 1344 1345
		goto exit;
	}

1346
	target = sock_rcvlowat(sk, flags & MSG_WAITALL, buf_len);
Y
Ying Xue 已提交
1347
	timeo = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
P
Per Liden 已提交
1348

1349
restart:
1350
	/* Look for a message in receive queue; wait if necessary */
1351
	res = tipc_wait_for_rcvmsg(sock, &timeo);
Y
Ying Xue 已提交
1352 1353
	if (res)
		goto exit;
P
Per Liden 已提交
1354

1355 1356
	/* Look at first message in receive queue */
	buf = skb_peek(&sk->sk_receive_queue);
P
Per Liden 已提交
1357 1358 1359 1360 1361 1362
	msg = buf_msg(buf);
	sz = msg_data_sz(msg);
	err = msg_errcode(msg);

	/* Discard an empty non-errored message & try again */
	if ((!sz) && (!err)) {
1363
		advance_rx_queue(sk);
P
Per Liden 已提交
1364 1365 1366 1367 1368 1369
		goto restart;
	}

	/* Optionally capture sender's address & ancillary data of first msg */
	if (sz_copied == 0) {
		set_orig_addr(m, msg);
1370
		res = anc_data_recv(m, msg, port);
1371
		if (res)
P
Per Liden 已提交
1372 1373 1374 1375 1376
			goto exit;
	}

	/* Capture message data (if valid) & compute return value (always) */
	if (!err) {
1377
		u32 offset = (u32)(unsigned long)(TIPC_SKB_CB(buf)->handle);
P
Per Liden 已提交
1378

1379
		sz -= offset;
P
Per Liden 已提交
1380 1381
		needed = (buf_len - sz_copied);
		sz_to_copy = (sz <= needed) ? sz : needed;
1382 1383 1384 1385

		res = skb_copy_datagram_iovec(buf, msg_hdr_sz(msg) + offset,
					      m->msg_iov, sz_to_copy);
		if (res)
P
Per Liden 已提交
1386
			goto exit;
1387

P
Per Liden 已提交
1388 1389 1390 1391
		sz_copied += sz_to_copy;

		if (sz_to_copy < sz) {
			if (!(flags & MSG_PEEK))
1392 1393
				TIPC_SKB_CB(buf)->handle =
				(void *)(unsigned long)(offset + sz_to_copy);
P
Per Liden 已提交
1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407
			goto exit;
		}
	} else {
		if (sz_copied != 0)
			goto exit; /* can't add error msg to valid data */

		if ((err == TIPC_CONN_SHUTDOWN) || m->msg_control)
			res = 0;
		else
			res = -ECONNRESET;
	}

	/* Consume received message (optional) */
	if (likely(!(flags & MSG_PEEK))) {
1408
		if (unlikely(++tsk->rcv_unacked >= TIPC_CONNACK_INTV)) {
1409
			tipc_sk_send_ack(port, tsk->rcv_unacked);
1410 1411
			tsk->rcv_unacked = 0;
		}
1412
		advance_rx_queue(sk);
1413
	}
P
Per Liden 已提交
1414 1415

	/* Loop around if more data is required */
1416 1417
	if ((sz_copied < buf_len) &&	/* didn't get all requested data */
	    (!skb_queue_empty(&sk->sk_receive_queue) ||
1418
	    (sz_copied < target)) &&	/* and more is ready or required */
1419 1420
	    (!(flags & MSG_PEEK)) &&	/* and aren't just peeking at data */
	    (!err))			/* and haven't reached a FIN */
P
Per Liden 已提交
1421 1422 1423
		goto restart;

exit:
1424
	release_sock(sk);
1425
	return sz_copied ? sz_copied : res;
P
Per Liden 已提交
1426 1427
}

1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448
/**
 * tipc_write_space - wake up thread if port congestion is released
 * @sk: socket
 */
static void tipc_write_space(struct sock *sk)
{
	struct socket_wq *wq;

	rcu_read_lock();
	wq = rcu_dereference(sk->sk_wq);
	if (wq_has_sleeper(wq))
		wake_up_interruptible_sync_poll(&wq->wait, POLLOUT |
						POLLWRNORM | POLLWRBAND);
	rcu_read_unlock();
}

/**
 * tipc_data_ready - wake up threads to indicate messages have been received
 * @sk: socket
 * @len: the length of messages
 */
1449
static void tipc_data_ready(struct sock *sk)
1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460
{
	struct socket_wq *wq;

	rcu_read_lock();
	wq = rcu_dereference(sk->sk_wq);
	if (wq_has_sleeper(wq))
		wake_up_interruptible_sync_poll(&wq->wait, POLLIN |
						POLLRDNORM | POLLRDBAND);
	rcu_read_unlock();
}

1461 1462
/**
 * filter_connect - Handle all incoming messages for a connection-based socket
1463
 * @tsk: TIPC socket
1464 1465
 * @msg: message
 *
1466
 * Returns 0 (TIPC_OK) if everyting ok, -TIPC_ERR_NO_PORT otherwise
1467
 */
1468
static int filter_connect(struct tipc_sock *tsk, struct sk_buff **buf)
1469
{
1470 1471
	struct sock *sk = &tsk->sk;
	struct tipc_port *port = &tsk->port;
1472
	struct socket *sock = sk->sk_socket;
1473
	struct tipc_msg *msg = buf_msg(*buf);
1474

1475
	int retval = -TIPC_ERR_NO_PORT;
1476 1477 1478 1479 1480 1481 1482

	if (msg_mcast(msg))
		return retval;

	switch ((int)sock->state) {
	case SS_CONNECTED:
		/* Accept only connection-based messages sent by peer */
J
Jon Paul Maloy 已提交
1483
		if (tipc_sk_peer_msg(tsk, msg)) {
1484 1485
			if (unlikely(msg_errcode(msg))) {
				sock->state = SS_DISCONNECTING;
1486 1487 1488 1489
				port->connected = 0;
				/* let timer expire on it's own */
				tipc_node_remove_conn(tipc_port_peernode(port),
						      port->ref);
1490 1491 1492 1493 1494 1495
			}
			retval = TIPC_OK;
		}
		break;
	case SS_CONNECTING:
		/* Accept only ACK or NACK message */
1496 1497 1498 1499

		if (unlikely(!msg_connected(msg)))
			break;

1500 1501
		if (unlikely(msg_errcode(msg))) {
			sock->state = SS_DISCONNECTING;
1502
			sk->sk_err = ECONNREFUSED;
1503 1504 1505 1506
			retval = TIPC_OK;
			break;
		}

1507
		if (unlikely(msg_importance(msg) > TIPC_CRITICAL_IMPORTANCE)) {
1508
			sock->state = SS_DISCONNECTING;
1509
			sk->sk_err = EINVAL;
1510
			retval = TIPC_OK;
1511 1512 1513
			break;
		}

1514 1515 1516 1517
		tipc_sk_finish_conn(port, msg_origport(msg), msg_orignode(msg));
		msg_set_importance(&port->phdr, msg_importance(msg));
		sock->state = SS_CONNECTED;

1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529
		/* If an incoming message is an 'ACK-', it should be
		 * discarded here because it doesn't contain useful
		 * data. In addition, we should try to wake up
		 * connect() routine if sleeping.
		 */
		if (msg_data_sz(msg) == 0) {
			kfree_skb(*buf);
			*buf = NULL;
			if (waitqueue_active(sk_sleep(sk)))
				wake_up_interruptible(sk_sleep(sk));
		}
		retval = TIPC_OK;
1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544
		break;
	case SS_LISTENING:
	case SS_UNCONNECTED:
		/* Accept only SYN message */
		if (!msg_connected(msg) && !(msg_errcode(msg)))
			retval = TIPC_OK;
		break;
	case SS_DISCONNECTING:
		break;
	default:
		pr_err("Unknown socket state %u\n", sock->state);
	}
	return retval;
}

1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555
/**
 * rcvbuf_limit - get proper overload limit of socket receive queue
 * @sk: socket
 * @buf: message
 *
 * For all connection oriented messages, irrespective of importance,
 * the default overload value (i.e. 67MB) is set as limit.
 *
 * For all connectionless messages, by default new queue limits are
 * as belows:
 *
1556 1557 1558 1559
 * TIPC_LOW_IMPORTANCE       (4 MB)
 * TIPC_MEDIUM_IMPORTANCE    (8 MB)
 * TIPC_HIGH_IMPORTANCE      (16 MB)
 * TIPC_CRITICAL_IMPORTANCE  (32 MB)
1560 1561 1562 1563 1564 1565 1566 1567
 *
 * Returns overload limit according to corresponding message importance
 */
static unsigned int rcvbuf_limit(struct sock *sk, struct sk_buff *buf)
{
	struct tipc_msg *msg = buf_msg(buf);

	if (msg_connected(msg))
1568 1569 1570 1571
		return sysctl_tipc_rmem[2];

	return sk->sk_rcvbuf >> TIPC_CRITICAL_IMPORTANCE <<
		msg_importance(msg);
1572 1573
}

1574
/**
1575 1576
 * filter_rcv - validate incoming message
 * @sk: socket
P
Per Liden 已提交
1577
 * @buf: message
1578
 *
1579 1580 1581 1582
 * Enqueues message on receive queue if acceptable; optionally handles
 * disconnect indication for a connected socket.
 *
 * Called with socket lock already taken; port lock may also be taken.
1583
 *
1584
 * Returns 0 (TIPC_OK) if message was consumed, -TIPC error code if message
1585
 * to be rejected, 1 (TIPC_FWD_MSG) if (CONN_MANAGER) message to be forwarded
P
Per Liden 已提交
1586
 */
1587
static int filter_rcv(struct sock *sk, struct sk_buff *buf)
P
Per Liden 已提交
1588
{
1589
	struct socket *sock = sk->sk_socket;
1590
	struct tipc_sock *tsk = tipc_sk(sk);
P
Per Liden 已提交
1591
	struct tipc_msg *msg = buf_msg(buf);
1592
	unsigned int limit = rcvbuf_limit(sk, buf);
1593
	u32 onode;
1594
	int rc = TIPC_OK;
P
Per Liden 已提交
1595

1596 1597
	if (unlikely(msg_user(msg) == CONN_MANAGER))
		return tipc_sk_proto_rcv(tsk, &onode, buf);
1598

1599 1600 1601 1602 1603 1604 1605
	if (unlikely(msg_user(msg) == SOCK_WAKEUP)) {
		kfree_skb(buf);
		tsk->link_cong = 0;
		sk->sk_write_space(sk);
		return TIPC_OK;
	}

P
Per Liden 已提交
1606
	/* Reject message if it is wrong sort of message for socket */
1607
	if (msg_type(msg) > TIPC_DIRECT_MSG)
1608
		return -TIPC_ERR_NO_PORT;
1609

P
Per Liden 已提交
1610
	if (sock->state == SS_READY) {
1611
		if (msg_connected(msg))
1612
			return -TIPC_ERR_NO_PORT;
P
Per Liden 已提交
1613
	} else {
1614 1615 1616
		rc = filter_connect(tsk, &buf);
		if (rc != TIPC_OK || buf == NULL)
			return rc;
P
Per Liden 已提交
1617 1618 1619
	}

	/* Reject message if there isn't room to queue it */
1620
	if (sk_rmem_alloc_get(sk) + buf->truesize >= limit)
1621
		return -TIPC_ERR_OVERLOAD;
P
Per Liden 已提交
1622

1623
	/* Enqueue message */
Y
Ying Xue 已提交
1624
	TIPC_SKB_CB(buf)->handle = NULL;
1625
	__skb_queue_tail(&sk->sk_receive_queue, buf);
1626
	skb_set_owner_r(buf, sk);
1627

1628
	sk->sk_data_ready(sk);
1629 1630
	return TIPC_OK;
}
P
Per Liden 已提交
1631

1632
/**
1633
 * tipc_backlog_rcv - handle incoming message from backlog queue
1634 1635 1636 1637 1638 1639 1640
 * @sk: socket
 * @buf: message
 *
 * Caller must hold socket lock, but not port lock.
 *
 * Returns 0
 */
1641
static int tipc_backlog_rcv(struct sock *sk, struct sk_buff *buf)
1642
{
1643
	int rc;
1644
	u32 onode;
1645
	struct tipc_sock *tsk = tipc_sk(sk);
1646
	uint truesize = buf->truesize;
1647

1648
	rc = filter_rcv(sk, buf);
1649

1650 1651 1652 1653 1654 1655 1656 1657 1658
	if (likely(!rc)) {
		if (atomic_read(&tsk->dupl_rcvcnt) < TIPC_CONN_OVERLOAD_LIMIT)
			atomic_add(truesize, &tsk->dupl_rcvcnt);
		return 0;
	}

	if ((rc < 0) && !tipc_msg_reverse(buf, &onode, -rc))
		return 0;

1659
	tipc_link_xmit(buf, onode, 0);
1660

1661 1662 1663 1664
	return 0;
}

/**
1665
 * tipc_sk_rcv - handle incoming message
1666 1667 1668
 * @buf: buffer containing arriving message
 * Consumes buffer
 * Returns 0 if success, or errno: -EHOSTUNREACH
1669
 */
1670
int tipc_sk_rcv(struct sk_buff *buf)
1671
{
1672 1673 1674 1675
	struct tipc_sock *tsk;
	struct tipc_port *port;
	struct sock *sk;
	u32 dport = msg_destport(buf_msg(buf));
1676
	int rc = TIPC_OK;
1677
	uint limit;
1678
	u32 dnode;
1679

1680
	/* Validate destination and message */
J
Jon Paul Maloy 已提交
1681
	tsk = tipc_sk_get(dport);
1682
	if (unlikely(!tsk)) {
1683
		rc = tipc_msg_eval(buf, &dnode);
1684 1685
		goto exit;
	}
1686
	port = &tsk->port;
1687 1688 1689
	sk = &tsk->sk;

	/* Queue message */
1690
	bh_lock_sock(sk);
1691

1692
	if (!sock_owned_by_user(sk)) {
1693
		rc = filter_rcv(sk, buf);
1694
	} else {
1695 1696 1697 1698
		if (sk->sk_backlog.len == 0)
			atomic_set(&tsk->dupl_rcvcnt, 0);
		limit = rcvbuf_limit(sk, buf) + atomic_read(&tsk->dupl_rcvcnt);
		if (sk_add_backlog(sk, buf, limit))
1699
			rc = -TIPC_ERR_OVERLOAD;
1700 1701
	}
	bh_unlock_sock(sk);
J
Jon Paul Maloy 已提交
1702
	tipc_sk_put(tsk);
1703
	if (likely(!rc))
1704 1705
		return 0;
exit:
1706
	if ((rc < 0) && !tipc_msg_reverse(buf, &dnode, -rc))
1707
		return -EHOSTUNREACH;
1708

1709
	tipc_link_xmit(buf, dnode, 0);
1710
	return (rc < 0) ? -EHOSTUNREACH : 0;
P
Per Liden 已提交
1711 1712
}

Y
Ying Xue 已提交
1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734
static int tipc_wait_for_connect(struct socket *sock, long *timeo_p)
{
	struct sock *sk = sock->sk;
	DEFINE_WAIT(wait);
	int done;

	do {
		int err = sock_error(sk);
		if (err)
			return err;
		if (!*timeo_p)
			return -ETIMEDOUT;
		if (signal_pending(current))
			return sock_intr_errno(*timeo_p);

		prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
		done = sk_wait_event(sk, timeo_p, sock->state != SS_CONNECTING);
		finish_wait(sk_sleep(sk), &wait);
	} while (!done);
	return 0;
}

P
Per Liden 已提交
1735
/**
1736
 * tipc_connect - establish a connection to another TIPC port
P
Per Liden 已提交
1737 1738 1739
 * @sock: socket structure
 * @dest: socket address for destination port
 * @destlen: size of socket address data structure
1740
 * @flags: file-related flags associated with socket
P
Per Liden 已提交
1741 1742 1743
 *
 * Returns 0 on success, errno otherwise
 */
1744 1745
static int tipc_connect(struct socket *sock, struct sockaddr *dest,
			int destlen, int flags)
P
Per Liden 已提交
1746
{
1747
	struct sock *sk = sock->sk;
1748 1749
	struct sockaddr_tipc *dst = (struct sockaddr_tipc *)dest;
	struct msghdr m = {NULL,};
Y
Ying Xue 已提交
1750 1751
	long timeout = (flags & O_NONBLOCK) ? 0 : tipc_sk(sk)->conn_timeout;
	socket_state previous;
1752 1753
	int res;

1754 1755
	lock_sock(sk);

1756
	/* For now, TIPC does not allow use of connect() with DGRAM/RDM types */
1757 1758 1759 1760
	if (sock->state == SS_READY) {
		res = -EOPNOTSUPP;
		goto exit;
	}
1761 1762 1763 1764 1765 1766 1767

	/*
	 * Reject connection attempt using multicast address
	 *
	 * Note: send_msg() validates the rest of the address fields,
	 *       so there's no need to do it here
	 */
1768 1769 1770 1771 1772
	if (dst->addrtype == TIPC_ADDR_MCAST) {
		res = -EINVAL;
		goto exit;
	}

Y
Ying Xue 已提交
1773
	previous = sock->state;
1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785
	switch (sock->state) {
	case SS_UNCONNECTED:
		/* Send a 'SYN-' to destination */
		m.msg_name = dest;
		m.msg_namelen = destlen;

		/* If connect is in non-blocking case, set MSG_DONTWAIT to
		 * indicate send_msg() is never blocked.
		 */
		if (!timeout)
			m.msg_flags = MSG_DONTWAIT;

1786
		res = tipc_sendmsg(NULL, sock, &m, 0);
1787 1788 1789 1790 1791 1792 1793 1794 1795
		if ((res < 0) && (res != -EWOULDBLOCK))
			goto exit;

		/* Just entered SS_CONNECTING state; the only
		 * difference is that return value in non-blocking
		 * case is EINPROGRESS, rather than EALREADY.
		 */
		res = -EINPROGRESS;
	case SS_CONNECTING:
Y
Ying Xue 已提交
1796 1797 1798 1799 1800 1801 1802
		if (previous == SS_CONNECTING)
			res = -EALREADY;
		if (!timeout)
			goto exit;
		timeout = msecs_to_jiffies(timeout);
		/* Wait until an 'ACK' or 'RST' arrives, or a timeout occurs */
		res = tipc_wait_for_connect(sock, &timeout);
1803 1804 1805 1806 1807 1808
		break;
	case SS_CONNECTED:
		res = -EISCONN;
		break;
	default:
		res = -EINVAL;
Y
Ying Xue 已提交
1809
		break;
1810
	}
1811 1812
exit:
	release_sock(sk);
1813
	return res;
P
Per Liden 已提交
1814 1815
}

1816
/**
1817
 * tipc_listen - allow socket to listen for incoming connections
P
Per Liden 已提交
1818 1819
 * @sock: socket structure
 * @len: (unused)
1820
 *
P
Per Liden 已提交
1821 1822
 * Returns 0 on success, errno otherwise
 */
1823
static int tipc_listen(struct socket *sock, int len)
P
Per Liden 已提交
1824
{
1825 1826 1827 1828
	struct sock *sk = sock->sk;
	int res;

	lock_sock(sk);
P
Per Liden 已提交
1829

1830
	if (sock->state != SS_UNCONNECTED)
1831 1832 1833 1834 1835 1836 1837 1838
		res = -EINVAL;
	else {
		sock->state = SS_LISTENING;
		res = 0;
	}

	release_sock(sk);
	return res;
P
Per Liden 已提交
1839 1840
}

Y
Ying Xue 已提交
1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854
static int tipc_wait_for_accept(struct socket *sock, long timeo)
{
	struct sock *sk = sock->sk;
	DEFINE_WAIT(wait);
	int err;

	/* True wake-one mechanism for incoming connections: only
	 * one process gets woken up, not the 'whole herd'.
	 * Since we do not 'race & poll' for established sockets
	 * anymore, the common case will execute the loop only once.
	*/
	for (;;) {
		prepare_to_wait_exclusive(sk_sleep(sk), &wait,
					  TASK_INTERRUPTIBLE);
1855
		if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
Y
Ying Xue 已提交
1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876
			release_sock(sk);
			timeo = schedule_timeout(timeo);
			lock_sock(sk);
		}
		err = 0;
		if (!skb_queue_empty(&sk->sk_receive_queue))
			break;
		err = -EINVAL;
		if (sock->state != SS_LISTENING)
			break;
		err = sock_intr_errno(timeo);
		if (signal_pending(current))
			break;
		err = -EAGAIN;
		if (!timeo)
			break;
	}
	finish_wait(sk_sleep(sk), &wait);
	return err;
}

1877
/**
1878
 * tipc_accept - wait for connection request
P
Per Liden 已提交
1879 1880 1881
 * @sock: listening socket
 * @newsock: new socket that is to be connected
 * @flags: file-related flags associated with socket
1882
 *
P
Per Liden 已提交
1883 1884
 * Returns 0 on success, errno otherwise
 */
1885
static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags)
P
Per Liden 已提交
1886
{
1887
	struct sock *new_sk, *sk = sock->sk;
P
Per Liden 已提交
1888
	struct sk_buff *buf;
1889
	struct tipc_port *new_port;
1890
	struct tipc_msg *msg;
Y
Ying Xue 已提交
1891
	long timeo;
1892
	int res;
P
Per Liden 已提交
1893

1894
	lock_sock(sk);
P
Per Liden 已提交
1895

1896 1897
	if (sock->state != SS_LISTENING) {
		res = -EINVAL;
P
Per Liden 已提交
1898 1899
		goto exit;
	}
Y
Ying Xue 已提交
1900 1901 1902 1903
	timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
	res = tipc_wait_for_accept(sock, timeo);
	if (res)
		goto exit;
1904 1905 1906

	buf = skb_peek(&sk->sk_receive_queue);

1907
	res = tipc_sk_create(sock_net(sock->sk), new_sock, 0, 1);
1908 1909
	if (res)
		goto exit;
P
Per Liden 已提交
1910

1911
	new_sk = new_sock->sk;
1912
	new_port = &tipc_sk(new_sk)->port;
1913
	msg = buf_msg(buf);
P
Per Liden 已提交
1914

1915 1916 1917 1918 1919 1920 1921 1922 1923 1924
	/* we lock on new_sk; but lockdep sees the lock on sk */
	lock_sock_nested(new_sk, SINGLE_DEPTH_NESTING);

	/*
	 * Reject any stray messages received by new socket
	 * before the socket lock was taken (very, very unlikely)
	 */
	reject_rx_queue(new_sk);

	/* Connect new socket to it's peer */
1925
	tipc_sk_finish_conn(new_port, msg_origport(msg), msg_orignode(msg));
1926 1927
	new_sock->state = SS_CONNECTED;

1928
	tipc_port_set_importance(new_port, msg_importance(msg));
1929
	if (msg_named(msg)) {
1930 1931
		new_port->conn_type = msg_nametype(msg);
		new_port->conn_instance = msg_nameinst(msg);
P
Per Liden 已提交
1932
	}
1933 1934 1935 1936 1937 1938 1939 1940 1941

	/*
	 * Respond to 'SYN-' by discarding it & returning 'ACK'-.
	 * Respond to 'SYN+' by queuing it on new socket.
	 */
	if (!msg_data_sz(msg)) {
		struct msghdr m = {NULL,};

		advance_rx_queue(sk);
1942
		tipc_send_packet(NULL, new_sock, &m, 0);
1943 1944 1945
	} else {
		__skb_dequeue(&sk->sk_receive_queue);
		__skb_queue_head(&new_sk->sk_receive_queue, buf);
1946
		skb_set_owner_r(buf, new_sk);
1947 1948
	}
	release_sock(new_sk);
P
Per Liden 已提交
1949
exit:
1950
	release_sock(sk);
P
Per Liden 已提交
1951 1952 1953 1954
	return res;
}

/**
1955
 * tipc_shutdown - shutdown socket connection
P
Per Liden 已提交
1956
 * @sock: socket structure
1957
 * @how: direction to close (must be SHUT_RDWR)
P
Per Liden 已提交
1958 1959
 *
 * Terminates connection (if necessary), then purges socket's receive queue.
1960
 *
P
Per Liden 已提交
1961 1962
 * Returns 0 on success, errno otherwise
 */
1963
static int tipc_shutdown(struct socket *sock, int how)
P
Per Liden 已提交
1964
{
1965
	struct sock *sk = sock->sk;
1966 1967
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
1968
	struct sk_buff *buf;
1969
	u32 dnode;
P
Per Liden 已提交
1970 1971
	int res;

1972 1973
	if (how != SHUT_RDWR)
		return -EINVAL;
P
Per Liden 已提交
1974

1975
	lock_sock(sk);
P
Per Liden 已提交
1976 1977

	switch (sock->state) {
1978
	case SS_CONNECTING:
P
Per Liden 已提交
1979 1980 1981
	case SS_CONNECTED:

restart:
1982
		/* Disconnect and send a 'FIN+' or 'FIN-' message to peer */
1983 1984
		buf = __skb_dequeue(&sk->sk_receive_queue);
		if (buf) {
Y
Ying Xue 已提交
1985
			if (TIPC_SKB_CB(buf)->handle != NULL) {
1986
				kfree_skb(buf);
P
Per Liden 已提交
1987 1988
				goto restart;
			}
1989 1990
			if (tipc_msg_reverse(buf, &dnode, TIPC_CONN_SHUTDOWN))
				tipc_link_xmit(buf, dnode, port->ref);
1991
			tipc_node_remove_conn(dnode, port->ref);
1992
		} else {
1993 1994 1995 1996 1997 1998 1999
			dnode = tipc_port_peernode(port);
			buf = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE,
					      TIPC_CONN_MSG, SHORT_H_SIZE,
					      0, dnode, tipc_own_addr,
					      tipc_port_peerport(port),
					      port->ref, TIPC_CONN_SHUTDOWN);
			tipc_link_xmit(buf, dnode, port->ref);
P
Per Liden 已提交
2000
		}
2001
		port->connected = 0;
2002
		sock->state = SS_DISCONNECTING;
2003
		tipc_node_remove_conn(dnode, port->ref);
P
Per Liden 已提交
2004 2005 2006 2007
		/* fall through */

	case SS_DISCONNECTING:

2008
		/* Discard any unreceived messages */
2009
		__skb_queue_purge(&sk->sk_receive_queue);
2010 2011 2012

		/* Wake up anyone sleeping in poll */
		sk->sk_state_change(sk);
P
Per Liden 已提交
2013 2014 2015 2016 2017 2018 2019
		res = 0;
		break;

	default:
		res = -ENOTCONN;
	}

2020
	release_sock(sk);
P
Per Liden 已提交
2021 2022 2023
	return res;
}

2024 2025
static void tipc_sk_timeout(unsigned long ref)
{
J
Jon Paul Maloy 已提交
2026
	struct tipc_sock *tsk;
2027
	struct tipc_port *port;
2028 2029 2030 2031
	struct sock *sk;
	struct sk_buff *buf = NULL;
	u32 peer_port, peer_node;

J
Jon Paul Maloy 已提交
2032
	tsk = tipc_sk_get(ref);
2033
	if (!tsk)
J
Jon Paul Maloy 已提交
2034 2035
		goto exit;
	sk = &tsk->sk;
2036
	port = &tsk->port;
J
Jon Paul Maloy 已提交
2037 2038

	bh_lock_sock(sk);
2039
	if (!port->connected) {
J
Jon Paul Maloy 已提交
2040 2041
		bh_unlock_sock(sk);
		goto exit;
2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059
	}
	peer_port = tipc_port_peerport(port);
	peer_node = tipc_port_peernode(port);

	if (port->probing_state == TIPC_CONN_PROBING) {
		/* Previous probe not answered -> self abort */
		buf = tipc_msg_create(TIPC_CRITICAL_IMPORTANCE, TIPC_CONN_MSG,
				      SHORT_H_SIZE, 0, tipc_own_addr,
				      peer_node, ref, peer_port,
				      TIPC_ERR_NO_PORT);
	} else {
		buf = tipc_msg_create(CONN_MANAGER, CONN_PROBE, INT_H_SIZE,
				      0, peer_node, tipc_own_addr,
				      peer_port, ref, TIPC_OK);
		port->probing_state = TIPC_CONN_PROBING;
		k_start_timer(&port->timer, port->probing_interval);
	}
	bh_unlock_sock(sk);
J
Jon Paul Maloy 已提交
2060 2061 2062 2063
	if (buf)
		tipc_link_xmit(buf, peer_node, ref);
exit:
	tipc_sk_put(tsk);
2064 2065
}

J
Jon Paul Maloy 已提交
2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119
static int tipc_sk_publish(struct tipc_port *port, uint scope,
			   struct tipc_name_seq const *seq)
{
	struct publication *publ;
	u32 key;

	if (port->connected)
		return -EINVAL;
	key = port->ref + port->pub_count + 1;
	if (key == port->ref)
		return -EADDRINUSE;

	publ = tipc_nametbl_publish(seq->type, seq->lower, seq->upper,
				    scope, port->ref, key);
	if (unlikely(!publ))
		return -EINVAL;

	list_add(&publ->pport_list, &port->publications);
	port->pub_count++;
	port->published = 1;
	return 0;
}

static int tipc_sk_withdraw(struct tipc_port *port, uint scope,
			    struct tipc_name_seq const *seq)
{
	struct publication *publ;
	struct publication *safe;
	int rc = -EINVAL;

	list_for_each_entry_safe(publ, safe, &port->publications, pport_list) {
		if (seq) {
			if (publ->scope != scope)
				continue;
			if (publ->type != seq->type)
				continue;
			if (publ->lower != seq->lower)
				continue;
			if (publ->upper != seq->upper)
				break;
			tipc_nametbl_withdraw(publ->type, publ->lower,
					      publ->ref, publ->key);
			rc = 0;
			break;
		}
		tipc_nametbl_withdraw(publ->type, publ->lower,
				      publ->ref, publ->key);
		rc = 0;
	}
	if (list_empty(&port->publications))
		port->published = 0;
	return rc;
}

2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180
static int tipc_sk_show(struct tipc_port *port, char *buf,
			int len, int full_id)
{
	struct publication *publ;
	int ret;

	if (full_id)
		ret = tipc_snprintf(buf, len, "<%u.%u.%u:%u>:",
				    tipc_zone(tipc_own_addr),
				    tipc_cluster(tipc_own_addr),
				    tipc_node(tipc_own_addr), port->ref);
	else
		ret = tipc_snprintf(buf, len, "%-10u:", port->ref);

	if (port->connected) {
		u32 dport = tipc_port_peerport(port);
		u32 destnode = tipc_port_peernode(port);

		ret += tipc_snprintf(buf + ret, len - ret,
				     " connected to <%u.%u.%u:%u>",
				     tipc_zone(destnode),
				     tipc_cluster(destnode),
				     tipc_node(destnode), dport);
		if (port->conn_type != 0)
			ret += tipc_snprintf(buf + ret, len - ret,
					     " via {%u,%u}", port->conn_type,
					     port->conn_instance);
	} else if (port->published) {
		ret += tipc_snprintf(buf + ret, len - ret, " bound to");
		list_for_each_entry(publ, &port->publications, pport_list) {
			if (publ->lower == publ->upper)
				ret += tipc_snprintf(buf + ret, len - ret,
						     " {%u,%u}", publ->type,
						     publ->lower);
			else
				ret += tipc_snprintf(buf + ret, len - ret,
						     " {%u,%u,%u}", publ->type,
						     publ->lower, publ->upper);
		}
	}
	ret += tipc_snprintf(buf + ret, len - ret, "\n");
	return ret;
}

struct sk_buff *tipc_sk_socks_show(void)
{
	struct sk_buff *buf;
	struct tlv_desc *rep_tlv;
	char *pb;
	int pb_len;
	struct tipc_sock *tsk;
	int str_len = 0;
	u32 ref = 0;

	buf = tipc_cfg_reply_alloc(TLV_SPACE(ULTRA_STRING_MAX_LEN));
	if (!buf)
		return NULL;
	rep_tlv = (struct tlv_desc *)buf->data;
	pb = TLV_DATA(rep_tlv);
	pb_len = ULTRA_STRING_MAX_LEN;

J
Jon Paul Maloy 已提交
2181 2182 2183
	tsk = tipc_sk_get_next(&ref);
	for (; tsk; tsk = tipc_sk_get_next(&ref)) {
		lock_sock(&tsk->sk);
2184 2185
		str_len += tipc_sk_show(&tsk->port, pb + str_len,
					pb_len - str_len, 0);
J
Jon Paul Maloy 已提交
2186 2187
		release_sock(&tsk->sk);
		tipc_sk_put(tsk);
2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202
	}
	str_len += 1;	/* for "\0" */
	skb_put(buf, TLV_SPACE(str_len));
	TLV_SET(rep_tlv, TIPC_TLV_ULTRA_STRING, NULL, str_len);

	return buf;
}

/* tipc_sk_reinit: set non-zero address in all existing sockets
 *                 when we go from standalone to network mode.
 */
void tipc_sk_reinit(void)
{
	struct tipc_msg *msg;
	u32 ref = 0;
J
Jon Paul Maloy 已提交
2203
	struct tipc_sock *tsk = tipc_sk_get_next(&ref);
2204

J
Jon Paul Maloy 已提交
2205 2206
	for (; tsk; tsk = tipc_sk_get_next(&ref)) {
		lock_sock(&tsk->sk);
2207 2208 2209
		msg = &tsk->port.phdr;
		msg_set_prevnode(msg, tipc_own_addr);
		msg_set_orignode(msg, tipc_own_addr);
J
Jon Paul Maloy 已提交
2210 2211
		release_sock(&tsk->sk);
		tipc_sk_put(tsk);
2212 2213 2214
	}
}

P
Per Liden 已提交
2215
/**
2216
 * tipc_setsockopt - set socket option
P
Per Liden 已提交
2217 2218 2219 2220 2221
 * @sock: socket structure
 * @lvl: option level
 * @opt: option identifier
 * @ov: pointer to new option value
 * @ol: length of option value
2222 2223
 *
 * For stream sockets only, accepts and ignores all IPPROTO_TCP options
P
Per Liden 已提交
2224
 * (to ease compatibility).
2225
 *
P
Per Liden 已提交
2226 2227
 * Returns 0 on success, errno otherwise
 */
2228 2229
static int tipc_setsockopt(struct socket *sock, int lvl, int opt,
			   char __user *ov, unsigned int ol)
P
Per Liden 已提交
2230
{
2231
	struct sock *sk = sock->sk;
2232 2233
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
2234 2235 2236
	u32 value;
	int res;

2237 2238
	if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
		return 0;
P
Per Liden 已提交
2239 2240 2241 2242
	if (lvl != SOL_TIPC)
		return -ENOPROTOOPT;
	if (ol < sizeof(value))
		return -EINVAL;
2243 2244
	res = get_user(value, (u32 __user *)ov);
	if (res)
P
Per Liden 已提交
2245 2246
		return res;

2247
	lock_sock(sk);
2248

P
Per Liden 已提交
2249 2250
	switch (opt) {
	case TIPC_IMPORTANCE:
2251
		res = tipc_port_set_importance(port, value);
P
Per Liden 已提交
2252 2253 2254
		break;
	case TIPC_SRC_DROPPABLE:
		if (sock->type != SOCK_STREAM)
2255
			tipc_port_set_unreliable(port, value);
2256
		else
P
Per Liden 已提交
2257 2258 2259
			res = -ENOPROTOOPT;
		break;
	case TIPC_DEST_DROPPABLE:
2260
		tipc_port_set_unreturnable(port, value);
P
Per Liden 已提交
2261 2262
		break;
	case TIPC_CONN_TIMEOUT:
2263
		tipc_sk(sk)->conn_timeout = value;
2264
		/* no need to set "res", since already 0 at this point */
P
Per Liden 已提交
2265 2266 2267 2268 2269
		break;
	default:
		res = -EINVAL;
	}

2270 2271
	release_sock(sk);

P
Per Liden 已提交
2272 2273 2274 2275
	return res;
}

/**
2276
 * tipc_getsockopt - get socket option
P
Per Liden 已提交
2277 2278 2279 2280 2281
 * @sock: socket structure
 * @lvl: option level
 * @opt: option identifier
 * @ov: receptacle for option value
 * @ol: receptacle for length of option value
2282 2283
 *
 * For stream sockets only, returns 0 length result for all IPPROTO_TCP options
P
Per Liden 已提交
2284
 * (to ease compatibility).
2285
 *
P
Per Liden 已提交
2286 2287
 * Returns 0 on success, errno otherwise
 */
2288 2289
static int tipc_getsockopt(struct socket *sock, int lvl, int opt,
			   char __user *ov, int __user *ol)
P
Per Liden 已提交
2290
{
2291
	struct sock *sk = sock->sk;
2292 2293
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
2294
	int len;
P
Per Liden 已提交
2295
	u32 value;
2296
	int res;
P
Per Liden 已提交
2297

2298 2299
	if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
		return put_user(0, ol);
P
Per Liden 已提交
2300 2301
	if (lvl != SOL_TIPC)
		return -ENOPROTOOPT;
2302 2303
	res = get_user(len, ol);
	if (res)
2304
		return res;
P
Per Liden 已提交
2305

2306
	lock_sock(sk);
P
Per Liden 已提交
2307 2308 2309

	switch (opt) {
	case TIPC_IMPORTANCE:
2310
		value = tipc_port_importance(port);
P
Per Liden 已提交
2311 2312
		break;
	case TIPC_SRC_DROPPABLE:
2313
		value = tipc_port_unreliable(port);
P
Per Liden 已提交
2314 2315
		break;
	case TIPC_DEST_DROPPABLE:
2316
		value = tipc_port_unreturnable(port);
P
Per Liden 已提交
2317 2318
		break;
	case TIPC_CONN_TIMEOUT:
2319
		value = tipc_sk(sk)->conn_timeout;
2320
		/* no need to set "res", since already 0 at this point */
P
Per Liden 已提交
2321
		break;
2322
	case TIPC_NODE_RECVQ_DEPTH:
2323
		value = 0; /* was tipc_queue_size, now obsolete */
2324
		break;
2325
	case TIPC_SOCK_RECVQ_DEPTH:
2326 2327
		value = skb_queue_len(&sk->sk_receive_queue);
		break;
P
Per Liden 已提交
2328 2329 2330 2331
	default:
		res = -EINVAL;
	}

2332 2333
	release_sock(sk);

2334 2335
	if (res)
		return res;	/* "get" failed */
P
Per Liden 已提交
2336

2337 2338 2339 2340 2341 2342 2343
	if (len < sizeof(value))
		return -EINVAL;

	if (copy_to_user(ov, &value, sizeof(value)))
		return -EFAULT;

	return put_user(sizeof(value), ol);
P
Per Liden 已提交
2344 2345
}

2346
static int tipc_ioctl(struct socket *sk, unsigned int cmd, unsigned long arg)
E
Erik Hugne 已提交
2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365 2366
{
	struct tipc_sioc_ln_req lnr;
	void __user *argp = (void __user *)arg;

	switch (cmd) {
	case SIOCGETLINKNAME:
		if (copy_from_user(&lnr, argp, sizeof(lnr)))
			return -EFAULT;
		if (!tipc_node_get_linkname(lnr.bearer_id, lnr.peer,
					    lnr.linkname, TIPC_MAX_LINK_NAME)) {
			if (copy_to_user(argp, &lnr, sizeof(lnr)))
				return -EFAULT;
			return 0;
		}
		return -EADDRNOTAVAIL;
	default:
		return -ENOIOCTLCMD;
	}
}

2367 2368
/* Protocol switches for the various types of TIPC sockets */

2369
static const struct proto_ops msg_ops = {
2370
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2371
	.family		= AF_TIPC,
2372 2373 2374
	.release	= tipc_release,
	.bind		= tipc_bind,
	.connect	= tipc_connect,
2375
	.socketpair	= sock_no_socketpair,
2376
	.accept		= sock_no_accept,
2377 2378
	.getname	= tipc_getname,
	.poll		= tipc_poll,
E
Erik Hugne 已提交
2379
	.ioctl		= tipc_ioctl,
2380
	.listen		= sock_no_listen,
2381 2382 2383 2384 2385
	.shutdown	= tipc_shutdown,
	.setsockopt	= tipc_setsockopt,
	.getsockopt	= tipc_getsockopt,
	.sendmsg	= tipc_sendmsg,
	.recvmsg	= tipc_recvmsg,
2386 2387
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage
P
Per Liden 已提交
2388 2389
};

2390
static const struct proto_ops packet_ops = {
2391
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2392
	.family		= AF_TIPC,
2393 2394 2395
	.release	= tipc_release,
	.bind		= tipc_bind,
	.connect	= tipc_connect,
2396
	.socketpair	= sock_no_socketpair,
2397 2398 2399
	.accept		= tipc_accept,
	.getname	= tipc_getname,
	.poll		= tipc_poll,
E
Erik Hugne 已提交
2400
	.ioctl		= tipc_ioctl,
2401 2402 2403 2404 2405 2406
	.listen		= tipc_listen,
	.shutdown	= tipc_shutdown,
	.setsockopt	= tipc_setsockopt,
	.getsockopt	= tipc_getsockopt,
	.sendmsg	= tipc_send_packet,
	.recvmsg	= tipc_recvmsg,
2407 2408
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage
P
Per Liden 已提交
2409 2410
};

2411
static const struct proto_ops stream_ops = {
2412
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2413
	.family		= AF_TIPC,
2414 2415 2416
	.release	= tipc_release,
	.bind		= tipc_bind,
	.connect	= tipc_connect,
2417
	.socketpair	= sock_no_socketpair,
2418 2419 2420
	.accept		= tipc_accept,
	.getname	= tipc_getname,
	.poll		= tipc_poll,
E
Erik Hugne 已提交
2421
	.ioctl		= tipc_ioctl,
2422 2423 2424 2425 2426 2427
	.listen		= tipc_listen,
	.shutdown	= tipc_shutdown,
	.setsockopt	= tipc_setsockopt,
	.getsockopt	= tipc_getsockopt,
	.sendmsg	= tipc_send_stream,
	.recvmsg	= tipc_recv_stream,
2428 2429
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage
P
Per Liden 已提交
2430 2431
};

2432
static const struct net_proto_family tipc_family_ops = {
2433
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2434
	.family		= AF_TIPC,
2435
	.create		= tipc_sk_create
P
Per Liden 已提交
2436 2437 2438 2439 2440
};

static struct proto tipc_proto = {
	.name		= "TIPC",
	.owner		= THIS_MODULE,
2441 2442
	.obj_size	= sizeof(struct tipc_sock),
	.sysctl_rmem	= sysctl_tipc_rmem
P
Per Liden 已提交
2443 2444
};

2445 2446 2447 2448 2449 2450
static struct proto tipc_proto_kern = {
	.name		= "TIPC",
	.obj_size	= sizeof(struct tipc_sock),
	.sysctl_rmem	= sysctl_tipc_rmem
};

P
Per Liden 已提交
2451
/**
2452
 * tipc_socket_init - initialize TIPC socket interface
2453
 *
P
Per Liden 已提交
2454 2455
 * Returns 0 on success, errno otherwise
 */
2456
int tipc_socket_init(void)
P
Per Liden 已提交
2457 2458 2459
{
	int res;

2460
	res = proto_register(&tipc_proto, 1);
P
Per Liden 已提交
2461
	if (res) {
2462
		pr_err("Failed to register TIPC protocol type\n");
P
Per Liden 已提交
2463 2464 2465 2466 2467
		goto out;
	}

	res = sock_register(&tipc_family_ops);
	if (res) {
2468
		pr_err("Failed to register TIPC socket type\n");
P
Per Liden 已提交
2469 2470 2471 2472 2473 2474 2475 2476
		proto_unregister(&tipc_proto);
		goto out;
	}
 out:
	return res;
}

/**
2477
 * tipc_socket_stop - stop TIPC socket interface
P
Per Liden 已提交
2478
 */
2479
void tipc_socket_stop(void)
P
Per Liden 已提交
2480 2481 2482 2483
{
	sock_unregister(tipc_family_ops.family);
	proto_unregister(&tipc_proto);
}