socket.c 60.9 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"
38
#include "port.h"
39
#include "name_table.h"
E
Erik Hugne 已提交
40
#include "node.h"
41
#include "link.h"
42
#include <linux/export.h>
43
#include "config.h"
44

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

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

52
static int tipc_backlog_rcv(struct sock *sk, struct sk_buff *skb);
53
static void tipc_data_ready(struct sock *sk);
54
static void tipc_write_space(struct sock *sk);
55 56
static int tipc_release(struct socket *sock);
static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags);
57
static int tipc_wait_for_sndmsg(struct socket *sock, long *timeo_p);
58
static void tipc_sk_timeout(unsigned long ref);
P
Per Liden 已提交
59

60 61 62
static const struct proto_ops packet_ops;
static const struct proto_ops stream_ops;
static const struct proto_ops msg_ops;
P
Per Liden 已提交
63 64

static struct proto tipc_proto;
65
static struct proto tipc_proto_kern;
P
Per Liden 已提交
66

67
/*
68 69 70 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
 * 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
 */

112 113
#include "socket.h"

114 115 116 117
/* tipc_sk_lock_next: find & lock next socket in registry from given port number
*/
static struct tipc_sock *tipc_sk_lock_next(u32 *ref)
{
118
	return (struct tipc_sock *)tipc_ref_lock_next(ref);
119 120
}

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

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

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

/**
148
 * tipc_sk_create - create a TIPC socket
149
 * @net: network namespace (must be default network)
P
Per Liden 已提交
150 151
 * @sock: pre-allocated socket structure
 * @protocol: protocol indicator (must be 0)
152
 * @kern: caused by kernel or by userspace?
153
 *
154 155
 * This routine creates additional data structures used by the TIPC socket,
 * initializes them, and links them together.
P
Per Liden 已提交
156 157 158
 *
 * Returns 0 on success, errno otherwise
 */
159 160
static int tipc_sk_create(struct net *net, struct socket *sock,
			  int protocol, int kern)
P
Per Liden 已提交
161
{
162 163
	const struct proto_ops *ops;
	socket_state state;
P
Per Liden 已提交
164
	struct sock *sk;
165 166
	struct tipc_sock *tsk;
	struct tipc_port *port;
167
	struct tipc_msg *msg;
168
	u32 ref;
169 170

	/* Validate arguments */
P
Per Liden 已提交
171 172 173 174 175
	if (unlikely(protocol != 0))
		return -EPROTONOSUPPORT;

	switch (sock->type) {
	case SOCK_STREAM:
176 177
		ops = &stream_ops;
		state = SS_UNCONNECTED;
P
Per Liden 已提交
178 179
		break;
	case SOCK_SEQPACKET:
180 181
		ops = &packet_ops;
		state = SS_UNCONNECTED;
P
Per Liden 已提交
182 183 184
		break;
	case SOCK_DGRAM:
	case SOCK_RDM:
185 186
		ops = &msg_ops;
		state = SS_READY;
P
Per Liden 已提交
187
		break;
188 189
	default:
		return -EPROTOTYPE;
P
Per Liden 已提交
190 191
	}

192
	/* Allocate socket's protocol area */
193 194 195 196 197
	if (!kern)
		sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto);
	else
		sk = sk_alloc(net, AF_TIPC, GFP_KERNEL, &tipc_proto_kern);

198
	if (sk == NULL)
P
Per Liden 已提交
199 200
		return -ENOMEM;

201 202
	tsk = tipc_sk(sk);
	port = &tsk->port;
203
	ref = tipc_ref_acquire(tsk, &port->lock);
204
	if (!ref) {
205
		pr_warn("Socket create failed; reference table exhausted\n");
206 207
		return -ENOMEM;
	}
208 209 210 211 212 213 214 215
	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 已提交
216

217 218 219 220
	/* Finish initializing socket data structures */
	sock->ops = ops;
	sock->state = state;
	sock_init_data(sock, sk);
221
	k_init_timer(&port->timer, (Handler)tipc_sk_timeout, ref);
222
	sk->sk_backlog_rcv = tipc_backlog_rcv;
223
	sk->sk_rcvbuf = sysctl_tipc_rmem[1];
224 225
	sk->sk_data_ready = tipc_data_ready;
	sk->sk_write_space = tipc_write_space;
226
	tsk->conn_timeout = CONN_TIMEOUT_DEFAULT;
227
	tsk->sent_unacked = 0;
228
	atomic_set(&tsk->dupl_rcvcnt, 0);
229
	tipc_port_unlock(port);
230

231
	if (sock->state == SS_READY) {
232
		tipc_port_set_unreturnable(port, true);
233
		if (sock->type == SOCK_DGRAM)
234
			tipc_port_set_unreliable(port, true);
235
	}
P
Per Liden 已提交
236 237 238
	return 0;
}

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
/**
 * 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)
{
273
	tipc_release(sock);
274 275 276 277 278 279 280 281 282 283 284 285 286 287 288
	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,
289
			   int flags)
290 291 292 293 294 295 296 297 298
{
	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;

299
	ret = tipc_accept(sock, *newsock, flags);
300 301 302 303 304 305 306 307
	if (ret < 0) {
		sock_release(*newsock);
		return ret;
	}
	(*newsock)->ops = sock->ops;
	return ret;
}

P
Per Liden 已提交
308
/**
309
 * tipc_release - destroy a TIPC socket
P
Per Liden 已提交
310 311 312 313 314 315 316
 * @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.)
317
 *
P
Per Liden 已提交
318 319 320 321 322 323
 * 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
 */
324
static int tipc_release(struct socket *sock)
P
Per Liden 已提交
325 326
{
	struct sock *sk = sock->sk;
327 328
	struct tipc_sock *tsk;
	struct tipc_port *port;
P
Per Liden 已提交
329
	struct sk_buff *buf;
330
	u32 dnode;
P
Per Liden 已提交
331

332 333 334 335 336
	/*
	 * 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 已提交
337
		return 0;
338

339 340
	tsk = tipc_sk(sk);
	port = &tsk->port;
341 342 343 344 345 346
	lock_sock(sk);

	/*
	 * Reject all unreceived messages, except on an active connection
	 * (which disconnects locally & sends a 'FIN+' to peer)
	 */
347
	dnode = tipc_port_peernode(port);
P
Per Liden 已提交
348
	while (sock->state != SS_DISCONNECTING) {
349 350
		buf = __skb_dequeue(&sk->sk_receive_queue);
		if (buf == NULL)
P
Per Liden 已提交
351
			break;
Y
Ying Xue 已提交
352
		if (TIPC_SKB_CB(buf)->handle != NULL)
353
			kfree_skb(buf);
354 355 356 357
		else {
			if ((sock->state == SS_CONNECTING) ||
			    (sock->state == SS_CONNECTED)) {
				sock->state = SS_DISCONNECTING;
358
				port->connected = 0;
359
				tipc_node_remove_conn(dnode, port->ref);
360
			}
361
			if (tipc_msg_reverse(buf, &dnode, TIPC_ERR_NO_PORT))
362
				tipc_link_xmit(buf, dnode, 0);
363
		}
P
Per Liden 已提交
364 365
	}

366 367 368 369 370 371 372 373 374 375 376 377 378 379 380
	tipc_withdraw(port, 0, NULL);
	spin_lock_bh(port->lock);
	tipc_ref_discard(port->ref);
	spin_unlock_bh(port->lock);
	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 已提交
381

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

385 386 387
	/* Reject any messages that accumulated in backlog queue */
	sock->state = SS_DISCONNECTING;
	release_sock(sk);
P
Per Liden 已提交
388
	sock_put(sk);
389
	sock->sk = NULL;
P
Per Liden 已提交
390

391
	return 0;
P
Per Liden 已提交
392 393 394
}

/**
395
 * tipc_bind - associate or disassocate TIPC name(s) with a socket
P
Per Liden 已提交
396 397 398
 * @sock: socket structure
 * @uaddr: socket address describing name(s) and desired operation
 * @uaddr_len: size of socket address data structure
399
 *
P
Per Liden 已提交
400 401 402
 * 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.
403
 *
P
Per Liden 已提交
404
 * Returns 0 on success, errno otherwise
405 406 407
 *
 * NOTE: This routine doesn't need to take the socket lock since it doesn't
 *       access any non-constant socket information.
P
Per Liden 已提交
408
 */
409 410
static int tipc_bind(struct socket *sock, struct sockaddr *uaddr,
		     int uaddr_len)
P
Per Liden 已提交
411
{
412
	struct sock *sk = sock->sk;
P
Per Liden 已提交
413
	struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
414
	struct tipc_sock *tsk = tipc_sk(sk);
415
	int res = -EINVAL;
P
Per Liden 已提交
416

417 418
	lock_sock(sk);
	if (unlikely(!uaddr_len)) {
419
		res = tipc_withdraw(&tsk->port, 0, NULL);
420 421
		goto exit;
	}
422

423 424 425 426 427 428 429 430
	if (uaddr_len < sizeof(struct sockaddr_tipc)) {
		res = -EINVAL;
		goto exit;
	}
	if (addr->family != AF_TIPC) {
		res = -EAFNOSUPPORT;
		goto exit;
	}
P
Per Liden 已提交
431 432 433

	if (addr->addrtype == TIPC_ADDR_NAME)
		addr->addr.nameseq.upper = addr->addr.nameseq.lower;
434 435 436 437
	else if (addr->addrtype != TIPC_ADDR_NAMESEQ) {
		res = -EAFNOSUPPORT;
		goto exit;
	}
438

439
	if ((addr->addr.nameseq.type < TIPC_RESERVED_TYPES) &&
440
	    (addr->addr.nameseq.type != TIPC_TOP_SRV) &&
441 442 443 444
	    (addr->addr.nameseq.type != TIPC_CFG_SRV)) {
		res = -EACCES;
		goto exit;
	}
445

446
	res = (addr->scope > 0) ?
447 448
		tipc_publish(&tsk->port, addr->scope, &addr->addr.nameseq) :
		tipc_withdraw(&tsk->port, -addr->scope, &addr->addr.nameseq);
449 450 451
exit:
	release_sock(sk);
	return res;
P
Per Liden 已提交
452 453
}

454
/**
455
 * tipc_getname - get port ID of socket or peer socket
P
Per Liden 已提交
456 457 458
 * @sock: socket structure
 * @uaddr: area for returned socket address
 * @uaddr_len: area for returned length of socket address
459
 * @peer: 0 = own ID, 1 = current peer ID, 2 = current/former peer ID
460
 *
P
Per Liden 已提交
461
 * Returns 0 on success, errno otherwise
462
 *
463 464
 * NOTE: This routine doesn't need to take the socket lock since it only
 *       accesses socket information that is unchanging (or which changes in
465
 *       a completely predictable manner).
P
Per Liden 已提交
466
 */
467 468
static int tipc_getname(struct socket *sock, struct sockaddr *uaddr,
			int *uaddr_len, int peer)
P
Per Liden 已提交
469 470
{
	struct sockaddr_tipc *addr = (struct sockaddr_tipc *)uaddr;
471
	struct tipc_sock *tsk = tipc_sk(sock->sk);
P
Per Liden 已提交
472

473
	memset(addr, 0, sizeof(*addr));
474
	if (peer) {
475 476 477
		if ((sock->state != SS_CONNECTED) &&
			((peer != 2) || (sock->state != SS_DISCONNECTING)))
			return -ENOTCONN;
478 479
		addr->addr.id.ref = tipc_port_peerport(&tsk->port);
		addr->addr.id.node = tipc_port_peernode(&tsk->port);
480
	} else {
481
		addr->addr.id.ref = tsk->port.ref;
A
Allan Stephens 已提交
482
		addr->addr.id.node = tipc_own_addr;
483
	}
P
Per Liden 已提交
484 485 486 487 488 489 490

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

491
	return 0;
P
Per Liden 已提交
492 493 494
}

/**
495
 * tipc_poll - read and possibly block on pollmask
P
Per Liden 已提交
496 497 498 499
 * @file: file structure associated with the socket
 * @sock: socket for which to calculate the poll bits
 * @wait: ???
 *
500 501 502 503 504 505 506 507 508
 * 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:
509 510 511 512
 *
 * socket state		flags set
 * ------------		---------
 * unconnected		no read flags
513
 *			POLLOUT if port is not congested
514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529 530 531 532
 *
 * 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 已提交
533
 */
534 535
static unsigned int tipc_poll(struct file *file, struct socket *sock,
			      poll_table *wait)
P
Per Liden 已提交
536
{
537
	struct sock *sk = sock->sk;
538
	struct tipc_sock *tsk = tipc_sk(sk);
539
	u32 mask = 0;
540

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

543
	switch ((int)sock->state) {
544
	case SS_UNCONNECTED:
545
		if (!tsk->link_cong)
546 547
			mask |= POLLOUT;
		break;
548 549
	case SS_READY:
	case SS_CONNECTED:
550
		if (!tsk->link_cong && !tipc_sk_conn_cong(tsk))
551 552 553 554 555 556 557 558 559 560 561
			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;
	}
562 563

	return mask;
P
Per Liden 已提交
564 565
}

566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596
/**
 * 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();
597
	rc = tipc_msg_build(mhdr, iov, 0, dsz, mtu, &buf);
598 599 600 601 602 603 604 605 606 607 608 609 610
	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;
611
		tipc_sk(sk)->link_cong = 1;
612 613 614 615 616 617 618
		rc = tipc_wait_for_sndmsg(sock, &timeo);
		if (rc)
			kfree_skb_list(buf);
	} while (!rc);
	return rc;
}

619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658
/* 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);
}

659 660 661 662 663 664 665 666
/**
 * 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.
 */
667 668
static int tipc_sk_proto_rcv(struct tipc_sock *tsk, u32 *dnode,
			     struct sk_buff *buf)
669 670 671
{
	struct tipc_msg *msg = buf_msg(buf);
	struct tipc_port *port = &tsk->port;
672
	int conn_cong;
673 674 675 676 677 678 679 680

	/* Ignore if connection cannot be validated: */
	if (!port->connected || !tipc_port_peer_msg(port, msg))
		goto exit;

	port->probing_state = TIPC_CONN_OK;

	if (msg_type(msg) == CONN_ACK) {
681 682 683
		conn_cong = tipc_sk_conn_cong(tsk);
		tsk->sent_unacked -= msg_msgcnt(msg);
		if (conn_cong)
684
			tsk->sk.sk_write_space(&tsk->sk);
685 686 687 688 689 690 691 692 693 694 695 696
	} 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;
}

697
/**
P
Per Liden 已提交
698 699 700
 * dest_name_check - verify user is permitted to send to specified port name
 * @dest: destination address
 * @m: descriptor for message to be sent
701
 *
P
Per Liden 已提交
702 703
 * Prevents restricted configuration commands from being issued by
 * unauthorized users.
704
 *
P
Per Liden 已提交
705 706
 * Returns 0 if permission is granted, otherwise errno
 */
S
Sam Ravnborg 已提交
707
static int dest_name_check(struct sockaddr_tipc *dest, struct msghdr *m)
P
Per Liden 已提交
708 709 710
{
	struct tipc_cfg_msg_hdr hdr;

711 712
	if (unlikely(dest->addrtype == TIPC_ADDR_ID))
		return 0;
713 714 715 716 717 718
	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 已提交
719

720 721
	if (!m->msg_iovlen || (m->msg_iov[0].iov_len < sizeof(hdr)))
		return -EMSGSIZE;
722
	if (copy_from_user(&hdr, m->msg_iov[0].iov_base, sizeof(hdr)))
P
Per Liden 已提交
723
		return -EFAULT;
724
	if ((ntohs(hdr.tcm_type) & 0xC000) && (!capable(CAP_NET_ADMIN)))
P
Per Liden 已提交
725
		return -EACCES;
726

P
Per Liden 已提交
727 728 729
	return 0;
}

730 731 732
static int tipc_wait_for_sndmsg(struct socket *sock, long *timeo_p)
{
	struct sock *sk = sock->sk;
733
	struct tipc_sock *tsk = tipc_sk(sk);
734 735 736 737 738 739 740 741 742 743 744 745 746 747 748
	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);
749
		done = sk_wait_event(sk, timeo_p, !tsk->link_cong);
750 751 752 753 754
		finish_wait(sk_sleep(sk), &wait);
	} while (!done);
	return 0;
}

P
Per Liden 已提交
755
/**
756
 * tipc_sendmsg - send message in connectionless manner
757
 * @iocb: if NULL, indicates that socket lock is already held
P
Per Liden 已提交
758 759
 * @sock: socket structure
 * @m: message to send
760
 * @dsz: amount of user data to be sent
761
 *
P
Per Liden 已提交
762
 * Message must have an destination specified explicitly.
763
 * Used for SOCK_RDM and SOCK_DGRAM messages,
P
Per Liden 已提交
764 765
 * and for 'SYN' messages on SOCK_SEQPACKET and SOCK_STREAM connections.
 * (Note: 'SYN+' is prohibited on SOCK_STREAM.)
766
 *
P
Per Liden 已提交
767 768
 * Returns the number of bytes sent on success, or errno otherwise
 */
769
static int tipc_sendmsg(struct kiocb *iocb, struct socket *sock,
770
			struct msghdr *m, size_t dsz)
P
Per Liden 已提交
771
{
772
	DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
773
	struct sock *sk = sock->sk;
774
	struct tipc_sock *tsk = tipc_sk(sk);
775
	struct tipc_port *port = &tsk->port;
776 777 778 779 780 781
	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;
782
	long timeo;
783
	int rc = -EINVAL;
P
Per Liden 已提交
784 785 786

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

788 789
	if (unlikely((m->msg_namelen < sizeof(*dest)) ||
		     (dest->family != AF_TIPC)))
P
Per Liden 已提交
790
		return -EINVAL;
791 792

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

795 796 797
	if (iocb)
		lock_sock(sk);

798
	if (unlikely(sock->state != SS_READY)) {
799
		if (sock->state == SS_LISTENING) {
800
			rc = -EPIPE;
801 802 803
			goto exit;
		}
		if (sock->state != SS_UNCONNECTED) {
804
			rc = -EISCONN;
805 806
			goto exit;
		}
807
		if (tsk->port.published) {
808
			rc = -EOPNOTSUPP;
809 810
			goto exit;
		}
811
		if (dest->addrtype == TIPC_ADDR_NAME) {
812 813
			tsk->port.conn_type = dest->addr.name.name.type;
			tsk->port.conn_instance = dest->addr.name.name.instance;
814
		}
P
Per Liden 已提交
815
	}
816 817 818
	rc = dest_name_check(dest, m);
	if (rc)
		goto exit;
P
Per Liden 已提交
819

820
	timeo = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841

	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;
842
		}
843 844 845 846 847 848 849 850 851 852 853
	} 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);
854
	rc = tipc_msg_build(mhdr, iov, 0, dsz, mtu, &buf);
855 856 857 858
	if (rc < 0)
		goto exit;

	do {
859
		TIPC_SKB_CB(buf)->wakeup_pending = tsk->link_cong;
860
		rc = tipc_link_xmit(buf, dnode, tsk->port.ref);
861 862
		if (likely(rc >= 0)) {
			if (sock->state != SS_READY)
863
				sock->state = SS_CONNECTING;
864
			rc = dsz;
865
			break;
866
		}
867 868 869
		if (rc == -EMSGSIZE)
			goto new_mtu;
		if (rc != -ELINKCONG)
870
			break;
871
		tsk->link_cong = 1;
872
		rc = tipc_wait_for_sndmsg(sock, &timeo);
873 874
		if (rc)
			kfree_skb_list(buf);
875
	} while (!rc);
876 877 878
exit:
	if (iocb)
		release_sock(sk);
879 880

	return rc;
P
Per Liden 已提交
881 882
}

883 884 885
static int tipc_wait_for_sndpkt(struct socket *sock, long *timeo_p)
{
	struct sock *sk = sock->sk;
886
	struct tipc_sock *tsk = tipc_sk(sk);
887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904
	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,
905 906 907
				     (!tsk->link_cong &&
				      !tipc_sk_conn_cong(tsk)) ||
				     !tsk->port.connected);
908 909 910 911 912
		finish_wait(sk_sleep(sk), &wait);
	} while (!done);
	return 0;
}

913
/**
914 915
 * tipc_send_stream - send stream-oriented data
 * @iocb: (unused)
P
Per Liden 已提交
916
 * @sock: socket structure
917 918
 * @m: data to send
 * @dsz: total length of data to be transmitted
919
 *
920
 * Used for SOCK_STREAM data.
921
 *
922 923
 * Returns the number of bytes sent on success (or partial success),
 * or errno if no data sent
P
Per Liden 已提交
924
 */
925 926
static int tipc_send_stream(struct kiocb *iocb, struct socket *sock,
			    struct msghdr *m, size_t dsz)
P
Per Liden 已提交
927
{
928
	struct sock *sk = sock->sk;
929
	struct tipc_sock *tsk = tipc_sk(sk);
930 931 932
	struct tipc_port *port = &tsk->port;
	struct tipc_msg *mhdr = &port->phdr;
	struct sk_buff *buf;
933
	DECLARE_SOCKADDR(struct sockaddr_tipc *, dest, m->msg_name);
934 935
	u32 ref = port->ref;
	int rc = -EINVAL;
936
	long timeo;
937 938
	u32 dnode;
	uint mtu, send, sent = 0;
P
Per Liden 已提交
939 940

	/* Handle implied connection establishment */
941 942 943
	if (unlikely(dest)) {
		rc = tipc_sendmsg(iocb, sock, m, dsz);
		if (dsz && (dsz == rc))
944
			tsk->sent_unacked = 1;
945 946 947
		return rc;
	}
	if (dsz > (uint)INT_MAX)
948 949
		return -EMSGSIZE;

950 951
	if (iocb)
		lock_sock(sk);
P
Per Liden 已提交
952

953 954
	if (unlikely(sock->state != SS_CONNECTED)) {
		if (sock->state == SS_DISCONNECTING)
955
			rc = -EPIPE;
956
		else
957
			rc = -ENOTCONN;
958 959
		goto exit;
	}
960

961
	timeo = sock_sndtimeo(sk, m->msg_flags & MSG_DONTWAIT);
962 963 964 965 966
	dnode = tipc_port_peernode(port);

next:
	mtu = port->max_pkt;
	send = min_t(uint, dsz - sent, TIPC_MAX_USER_MSG_SIZE);
967
	rc = tipc_msg_build(mhdr, m->msg_iov, sent, send, mtu, &buf);
968 969
	if (unlikely(rc < 0))
		goto exit;
970
	do {
971
		if (likely(!tipc_sk_conn_cong(tsk))) {
972
			rc = tipc_link_xmit(buf, dnode, ref);
973
			if (likely(!rc)) {
974
				tsk->sent_unacked++;
975 976 977 978 979 980 981 982 983 984 985
				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;
986
			tsk->link_cong = 1;
987 988
		}
		rc = tipc_wait_for_sndpkt(sock, &timeo);
989 990
		if (rc)
			kfree_skb_list(buf);
991
	} while (!rc);
992
exit:
993 994
	if (iocb)
		release_sock(sk);
995
	return sent ? sent : rc;
P
Per Liden 已提交
996 997
}

998
/**
999 1000
 * tipc_send_packet - send a connection-oriented message
 * @iocb: if NULL, indicates that socket lock is already held
P
Per Liden 已提交
1001
 * @sock: socket structure
1002 1003
 * @m: message to send
 * @dsz: length of data to be transmitted
1004
 *
1005
 * Used for SOCK_SEQPACKET messages.
1006
 *
1007
 * Returns the number of bytes sent on success, or errno otherwise
P
Per Liden 已提交
1008
 */
1009 1010
static int tipc_send_packet(struct kiocb *iocb, struct socket *sock,
			    struct msghdr *m, size_t dsz)
P
Per Liden 已提交
1011
{
1012 1013
	if (dsz > TIPC_MAX_USER_MSG_SIZE)
		return -EMSGSIZE;
P
Per Liden 已提交
1014

1015
	return tipc_send_stream(iocb, sock, m, dsz);
P
Per Liden 已提交
1016 1017
}

1018
/* tipc_sk_finish_conn - complete the setup of a connection
P
Per Liden 已提交
1019
 */
1020 1021
static void tipc_sk_finish_conn(struct tipc_port *port, u32 peer_port,
				u32 peer_node)
P
Per Liden 已提交
1022
{
1023
	struct tipc_msg *msg = &port->phdr;
P
Per Liden 已提交
1024

1025 1026 1027 1028 1029
	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);
1030

1031 1032 1033 1034 1035 1036
	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 已提交
1037 1038 1039 1040 1041 1042
}

/**
 * set_orig_addr - capture sender's address for received message
 * @m: descriptor for message info
 * @msg: received message header
1043
 *
P
Per Liden 已提交
1044 1045
 * Note: Address is not captured if not requested by receiver.
 */
S
Sam Ravnborg 已提交
1046
static void set_orig_addr(struct msghdr *m, struct tipc_msg *msg)
P
Per Liden 已提交
1047
{
1048
	DECLARE_SOCKADDR(struct sockaddr_tipc *, addr, m->msg_name);
P
Per Liden 已提交
1049

1050
	if (addr) {
P
Per Liden 已提交
1051 1052
		addr->family = AF_TIPC;
		addr->addrtype = TIPC_ADDR_ID;
1053
		memset(&addr->addr, 0, sizeof(addr->addr));
P
Per Liden 已提交
1054 1055
		addr->addr.id.ref = msg_origport(msg);
		addr->addr.id.node = msg_orignode(msg);
1056 1057
		addr->addr.name.domain = 0;	/* could leave uninitialized */
		addr->scope = 0;		/* could leave uninitialized */
P
Per Liden 已提交
1058 1059 1060 1061 1062
		m->msg_namelen = sizeof(struct sockaddr_tipc);
	}
}

/**
1063
 * anc_data_recv - optionally capture ancillary data for received message
P
Per Liden 已提交
1064 1065 1066
 * @m: descriptor for message info
 * @msg: received message header
 * @tport: TIPC port associated with message
1067
 *
P
Per Liden 已提交
1068
 * Note: Ancillary data is not captured if not requested by receiver.
1069
 *
P
Per Liden 已提交
1070 1071
 * Returns 0 if successful, otherwise errno
 */
S
Sam Ravnborg 已提交
1072
static int anc_data_recv(struct msghdr *m, struct tipc_msg *msg,
1073
			 struct tipc_port *tport)
P
Per Liden 已提交
1074 1075 1076 1077
{
	u32 anc_data[3];
	u32 err;
	u32 dest_type;
1078
	int has_name;
P
Per Liden 已提交
1079 1080 1081 1082 1083 1084 1085 1086 1087 1088
	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);
1089 1090
		res = put_cmsg(m, SOL_TIPC, TIPC_ERRINFO, 8, anc_data);
		if (res)
P
Per Liden 已提交
1091
			return res;
1092 1093 1094 1095 1096 1097
		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 已提交
1098 1099 1100 1101 1102 1103
	}

	/* Optionally capture message destination object */
	dest_type = msg ? msg_type(msg) : TIPC_DIRECT_MSG;
	switch (dest_type) {
	case TIPC_NAMED_MSG:
1104
		has_name = 1;
P
Per Liden 已提交
1105 1106 1107 1108 1109
		anc_data[0] = msg_nametype(msg);
		anc_data[1] = msg_namelower(msg);
		anc_data[2] = msg_namelower(msg);
		break;
	case TIPC_MCAST_MSG:
1110
		has_name = 1;
P
Per Liden 已提交
1111 1112 1113 1114 1115
		anc_data[0] = msg_nametype(msg);
		anc_data[1] = msg_namelower(msg);
		anc_data[2] = msg_nameupper(msg);
		break;
	case TIPC_CONN_MSG:
1116
		has_name = (tport->conn_type != 0);
P
Per Liden 已提交
1117 1118 1119 1120 1121
		anc_data[0] = tport->conn_type;
		anc_data[1] = tport->conn_instance;
		anc_data[2] = tport->conn_instance;
		break;
	default:
1122
		has_name = 0;
P
Per Liden 已提交
1123
	}
1124 1125 1126 1127 1128
	if (has_name) {
		res = put_cmsg(m, SOL_TIPC, TIPC_DESTNAME, 12, anc_data);
		if (res)
			return res;
	}
P
Per Liden 已提交
1129 1130 1131 1132

	return 0;
}

1133 1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150
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));
}

1151
static int tipc_wait_for_rcvmsg(struct socket *sock, long *timeop)
Y
Ying Xue 已提交
1152 1153 1154
{
	struct sock *sk = sock->sk;
	DEFINE_WAIT(wait);
1155
	long timeo = *timeop;
Y
Ying Xue 已提交
1156 1157 1158 1159
	int err;

	for (;;) {
		prepare_to_wait(sk_sleep(sk), &wait, TASK_INTERRUPTIBLE);
1160
		if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
Y
Ying Xue 已提交
1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179
			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);
1180
	*timeop = timeo;
Y
Ying Xue 已提交
1181 1182 1183
	return err;
}

1184
/**
1185
 * tipc_recvmsg - receive packet-oriented message
P
Per Liden 已提交
1186 1187 1188 1189
 * @iocb: (unused)
 * @m: descriptor for message info
 * @buf_len: total size of user buffer area
 * @flags: receive flags
1190
 *
P
Per Liden 已提交
1191 1192 1193 1194 1195
 * 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
 */
1196 1197
static int tipc_recvmsg(struct kiocb *iocb, struct socket *sock,
			struct msghdr *m, size_t buf_len, int flags)
P
Per Liden 已提交
1198
{
1199
	struct sock *sk = sock->sk;
1200 1201
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
1202 1203
	struct sk_buff *buf;
	struct tipc_msg *msg;
Y
Ying Xue 已提交
1204
	long timeo;
P
Per Liden 已提交
1205 1206 1207 1208
	unsigned int sz;
	u32 err;
	int res;

1209
	/* Catch invalid receive requests */
P
Per Liden 已提交
1210 1211 1212
	if (unlikely(!buf_len))
		return -EINVAL;

1213
	lock_sock(sk);
P
Per Liden 已提交
1214

1215 1216
	if (unlikely(sock->state == SS_UNCONNECTED)) {
		res = -ENOTCONN;
P
Per Liden 已提交
1217 1218 1219
		goto exit;
	}

Y
Ying Xue 已提交
1220
	timeo = sock_rcvtimeo(sk, flags & MSG_DONTWAIT);
1221
restart:
P
Per Liden 已提交
1222

1223
	/* Look for a message in receive queue; wait if necessary */
1224
	res = tipc_wait_for_rcvmsg(sock, &timeo);
Y
Ying Xue 已提交
1225 1226
	if (res)
		goto exit;
P
Per Liden 已提交
1227

1228 1229
	/* Look at first message in receive queue */
	buf = skb_peek(&sk->sk_receive_queue);
P
Per Liden 已提交
1230 1231 1232 1233 1234 1235
	msg = buf_msg(buf);
	sz = msg_data_sz(msg);
	err = msg_errcode(msg);

	/* Discard an empty non-errored message & try again */
	if ((!sz) && (!err)) {
1236
		advance_rx_queue(sk);
P
Per Liden 已提交
1237 1238 1239 1240 1241 1242 1243
		goto restart;
	}

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

	/* Capture ancillary data (optional) */
1244
	res = anc_data_recv(m, msg, port);
1245
	if (res)
P
Per Liden 已提交
1246 1247 1248 1249 1250 1251 1252 1253
		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;
		}
1254 1255 1256
		res = skb_copy_datagram_iovec(buf, msg_hdr_sz(msg),
					      m->msg_iov, sz);
		if (res)
P
Per Liden 已提交
1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268
			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))) {
1269
		if ((sock->state != SS_READY) &&
1270
		    (++tsk->rcv_unacked >= TIPC_CONNACK_INTV)) {
1271
			tipc_sk_send_ack(port, tsk->rcv_unacked);
1272 1273
			tsk->rcv_unacked = 0;
		}
1274
		advance_rx_queue(sk);
1275
	}
P
Per Liden 已提交
1276
exit:
1277
	release_sock(sk);
P
Per Liden 已提交
1278 1279 1280
	return res;
}

1281
/**
1282
 * tipc_recv_stream - receive stream-oriented data
P
Per Liden 已提交
1283 1284 1285 1286
 * @iocb: (unused)
 * @m: descriptor for message info
 * @buf_len: total size of user buffer area
 * @flags: receive flags
1287 1288
 *
 * Used for SOCK_STREAM messages only.  If not enough data is available
P
Per Liden 已提交
1289 1290 1291 1292
 * will optionally wait for more; never truncates data.
 *
 * Returns size of returned message data, errno otherwise
 */
1293 1294
static int tipc_recv_stream(struct kiocb *iocb, struct socket *sock,
			    struct msghdr *m, size_t buf_len, int flags)
P
Per Liden 已提交
1295
{
1296
	struct sock *sk = sock->sk;
1297 1298
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
1299 1300
	struct sk_buff *buf;
	struct tipc_msg *msg;
Y
Ying Xue 已提交
1301
	long timeo;
P
Per Liden 已提交
1302
	unsigned int sz;
1303
	int sz_to_copy, target, needed;
P
Per Liden 已提交
1304 1305
	int sz_copied = 0;
	u32 err;
1306
	int res = 0;
P
Per Liden 已提交
1307

1308
	/* Catch invalid receive attempts */
P
Per Liden 已提交
1309 1310 1311
	if (unlikely(!buf_len))
		return -EINVAL;

1312
	lock_sock(sk);
P
Per Liden 已提交
1313

Y
Ying Xue 已提交
1314
	if (unlikely(sock->state == SS_UNCONNECTED)) {
1315
		res = -ENOTCONN;
P
Per Liden 已提交
1316 1317 1318
		goto exit;
	}

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

1322
restart:
1323
	/* Look for a message in receive queue; wait if necessary */
1324
	res = tipc_wait_for_rcvmsg(sock, &timeo);
Y
Ying Xue 已提交
1325 1326
	if (res)
		goto exit;
P
Per Liden 已提交
1327

1328 1329
	/* Look at first message in receive queue */
	buf = skb_peek(&sk->sk_receive_queue);
P
Per Liden 已提交
1330 1331 1332 1333 1334 1335
	msg = buf_msg(buf);
	sz = msg_data_sz(msg);
	err = msg_errcode(msg);

	/* Discard an empty non-errored message & try again */
	if ((!sz) && (!err)) {
1336
		advance_rx_queue(sk);
P
Per Liden 已提交
1337 1338 1339 1340 1341 1342
		goto restart;
	}

	/* Optionally capture sender's address & ancillary data of first msg */
	if (sz_copied == 0) {
		set_orig_addr(m, msg);
1343
		res = anc_data_recv(m, msg, port);
1344
		if (res)
P
Per Liden 已提交
1345 1346 1347 1348 1349
			goto exit;
	}

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

1352
		sz -= offset;
P
Per Liden 已提交
1353 1354
		needed = (buf_len - sz_copied);
		sz_to_copy = (sz <= needed) ? sz : needed;
1355 1356 1357 1358

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

P
Per Liden 已提交
1361 1362 1363 1364
		sz_copied += sz_to_copy;

		if (sz_to_copy < sz) {
			if (!(flags & MSG_PEEK))
1365 1366
				TIPC_SKB_CB(buf)->handle =
				(void *)(unsigned long)(offset + sz_to_copy);
P
Per Liden 已提交
1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
			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))) {
1381
		if (unlikely(++tsk->rcv_unacked >= TIPC_CONNACK_INTV)) {
1382
			tipc_sk_send_ack(port, tsk->rcv_unacked);
1383 1384
			tsk->rcv_unacked = 0;
		}
1385
		advance_rx_queue(sk);
1386
	}
P
Per Liden 已提交
1387 1388

	/* Loop around if more data is required */
1389 1390
	if ((sz_copied < buf_len) &&	/* didn't get all requested data */
	    (!skb_queue_empty(&sk->sk_receive_queue) ||
1391
	    (sz_copied < target)) &&	/* and more is ready or required */
1392 1393
	    (!(flags & MSG_PEEK)) &&	/* and aren't just peeking at data */
	    (!err))			/* and haven't reached a FIN */
P
Per Liden 已提交
1394 1395 1396
		goto restart;

exit:
1397
	release_sock(sk);
1398
	return sz_copied ? sz_copied : res;
P
Per Liden 已提交
1399 1400
}

1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421
/**
 * 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
 */
1422
static void tipc_data_ready(struct sock *sk)
1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433
{
	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();
}

1434 1435
/**
 * filter_connect - Handle all incoming messages for a connection-based socket
1436
 * @tsk: TIPC socket
1437 1438
 * @msg: message
 *
1439
 * Returns 0 (TIPC_OK) if everyting ok, -TIPC_ERR_NO_PORT otherwise
1440
 */
1441
static int filter_connect(struct tipc_sock *tsk, struct sk_buff **buf)
1442
{
1443 1444
	struct sock *sk = &tsk->sk;
	struct tipc_port *port = &tsk->port;
1445
	struct socket *sock = sk->sk_socket;
1446
	struct tipc_msg *msg = buf_msg(*buf);
1447

1448
	int retval = -TIPC_ERR_NO_PORT;
1449 1450 1451 1452 1453 1454 1455

	if (msg_mcast(msg))
		return retval;

	switch ((int)sock->state) {
	case SS_CONNECTED:
		/* Accept only connection-based messages sent by peer */
1456
		if (msg_connected(msg) && tipc_port_peer_msg(port, msg)) {
1457 1458
			if (unlikely(msg_errcode(msg))) {
				sock->state = SS_DISCONNECTING;
1459 1460 1461 1462
				port->connected = 0;
				/* let timer expire on it's own */
				tipc_node_remove_conn(tipc_port_peernode(port),
						      port->ref);
1463 1464 1465 1466 1467 1468
			}
			retval = TIPC_OK;
		}
		break;
	case SS_CONNECTING:
		/* Accept only ACK or NACK message */
1469 1470 1471 1472

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

1473 1474
		if (unlikely(msg_errcode(msg))) {
			sock->state = SS_DISCONNECTING;
1475
			sk->sk_err = ECONNREFUSED;
1476 1477 1478 1479
			retval = TIPC_OK;
			break;
		}

1480
		if (unlikely(msg_importance(msg) > TIPC_CRITICAL_IMPORTANCE)) {
1481
			sock->state = SS_DISCONNECTING;
1482
			sk->sk_err = EINVAL;
1483
			retval = TIPC_OK;
1484 1485 1486
			break;
		}

1487 1488 1489 1490
		tipc_sk_finish_conn(port, msg_origport(msg), msg_orignode(msg));
		msg_set_importance(&port->phdr, msg_importance(msg));
		sock->state = SS_CONNECTED;

1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502
		/* 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;
1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517
		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;
}

1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528
/**
 * 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:
 *
1529 1530 1531 1532
 * TIPC_LOW_IMPORTANCE       (4 MB)
 * TIPC_MEDIUM_IMPORTANCE    (8 MB)
 * TIPC_HIGH_IMPORTANCE      (16 MB)
 * TIPC_CRITICAL_IMPORTANCE  (32 MB)
1533 1534 1535 1536 1537 1538 1539 1540
 *
 * 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))
1541 1542 1543 1544
		return sysctl_tipc_rmem[2];

	return sk->sk_rcvbuf >> TIPC_CRITICAL_IMPORTANCE <<
		msg_importance(msg);
1545 1546
}

1547
/**
1548 1549
 * filter_rcv - validate incoming message
 * @sk: socket
P
Per Liden 已提交
1550
 * @buf: message
1551
 *
1552 1553 1554 1555
 * 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.
1556
 *
1557
 * Returns 0 (TIPC_OK) if message was consumed, -TIPC error code if message
1558
 * to be rejected, 1 (TIPC_FWD_MSG) if (CONN_MANAGER) message to be forwarded
P
Per Liden 已提交
1559
 */
1560
static int filter_rcv(struct sock *sk, struct sk_buff *buf)
P
Per Liden 已提交
1561
{
1562
	struct socket *sock = sk->sk_socket;
1563
	struct tipc_sock *tsk = tipc_sk(sk);
P
Per Liden 已提交
1564
	struct tipc_msg *msg = buf_msg(buf);
1565
	unsigned int limit = rcvbuf_limit(sk, buf);
1566
	u32 onode;
1567
	int rc = TIPC_OK;
P
Per Liden 已提交
1568

1569 1570
	if (unlikely(msg_user(msg) == CONN_MANAGER))
		return tipc_sk_proto_rcv(tsk, &onode, buf);
1571

1572 1573 1574 1575 1576 1577 1578
	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 已提交
1579
	/* Reject message if it is wrong sort of message for socket */
1580
	if (msg_type(msg) > TIPC_DIRECT_MSG)
1581
		return -TIPC_ERR_NO_PORT;
1582

P
Per Liden 已提交
1583
	if (sock->state == SS_READY) {
1584
		if (msg_connected(msg))
1585
			return -TIPC_ERR_NO_PORT;
P
Per Liden 已提交
1586
	} else {
1587 1588 1589
		rc = filter_connect(tsk, &buf);
		if (rc != TIPC_OK || buf == NULL)
			return rc;
P
Per Liden 已提交
1590 1591 1592
	}

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

1596
	/* Enqueue message */
Y
Ying Xue 已提交
1597
	TIPC_SKB_CB(buf)->handle = NULL;
1598
	__skb_queue_tail(&sk->sk_receive_queue, buf);
1599
	skb_set_owner_r(buf, sk);
1600

1601
	sk->sk_data_ready(sk);
1602 1603
	return TIPC_OK;
}
P
Per Liden 已提交
1604

1605
/**
1606
 * tipc_backlog_rcv - handle incoming message from backlog queue
1607 1608 1609 1610 1611 1612 1613
 * @sk: socket
 * @buf: message
 *
 * Caller must hold socket lock, but not port lock.
 *
 * Returns 0
 */
1614
static int tipc_backlog_rcv(struct sock *sk, struct sk_buff *buf)
1615
{
1616
	int rc;
1617
	u32 onode;
1618
	struct tipc_sock *tsk = tipc_sk(sk);
1619
	uint truesize = buf->truesize;
1620

1621
	rc = filter_rcv(sk, buf);
1622

1623 1624 1625 1626 1627 1628 1629 1630 1631
	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;

1632
	tipc_link_xmit(buf, onode, 0);
1633

1634 1635 1636 1637
	return 0;
}

/**
1638
 * tipc_sk_rcv - handle incoming message
1639 1640 1641
 * @buf: buffer containing arriving message
 * Consumes buffer
 * Returns 0 if success, or errno: -EHOSTUNREACH
1642
 */
1643
int tipc_sk_rcv(struct sk_buff *buf)
1644
{
1645 1646 1647 1648
	struct tipc_sock *tsk;
	struct tipc_port *port;
	struct sock *sk;
	u32 dport = msg_destport(buf_msg(buf));
1649
	int rc = TIPC_OK;
1650
	uint limit;
1651
	u32 dnode;
1652

1653
	/* Validate destination and message */
1654 1655
	tsk = tipc_port_lock(dport);
	if (unlikely(!tsk)) {
1656
		rc = tipc_msg_eval(buf, &dnode);
1657 1658
		goto exit;
	}
1659
	port = &tsk->port;
1660 1661 1662
	sk = &tsk->sk;

	/* Queue message */
1663
	bh_lock_sock(sk);
1664

1665
	if (!sock_owned_by_user(sk)) {
1666
		rc = filter_rcv(sk, buf);
1667
	} else {
1668 1669 1670 1671
		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))
1672
			rc = -TIPC_ERR_OVERLOAD;
1673 1674
	}
	bh_unlock_sock(sk);
1675
	tipc_port_unlock(port);
1676

1677
	if (likely(!rc))
1678 1679
		return 0;
exit:
1680
	if ((rc < 0) && !tipc_msg_reverse(buf, &dnode, -rc))
1681
		return -EHOSTUNREACH;
1682

1683
	tipc_link_xmit(buf, dnode, 0);
1684
	return (rc < 0) ? -EHOSTUNREACH : 0;
P
Per Liden 已提交
1685 1686
}

Y
Ying Xue 已提交
1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708
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 已提交
1709
/**
1710
 * tipc_connect - establish a connection to another TIPC port
P
Per Liden 已提交
1711 1712 1713
 * @sock: socket structure
 * @dest: socket address for destination port
 * @destlen: size of socket address data structure
1714
 * @flags: file-related flags associated with socket
P
Per Liden 已提交
1715 1716 1717
 *
 * Returns 0 on success, errno otherwise
 */
1718 1719
static int tipc_connect(struct socket *sock, struct sockaddr *dest,
			int destlen, int flags)
P
Per Liden 已提交
1720
{
1721
	struct sock *sk = sock->sk;
1722 1723
	struct sockaddr_tipc *dst = (struct sockaddr_tipc *)dest;
	struct msghdr m = {NULL,};
Y
Ying Xue 已提交
1724 1725
	long timeout = (flags & O_NONBLOCK) ? 0 : tipc_sk(sk)->conn_timeout;
	socket_state previous;
1726 1727
	int res;

1728 1729
	lock_sock(sk);

1730
	/* For now, TIPC does not allow use of connect() with DGRAM/RDM types */
1731 1732 1733 1734
	if (sock->state == SS_READY) {
		res = -EOPNOTSUPP;
		goto exit;
	}
1735 1736 1737 1738 1739 1740 1741

	/*
	 * 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
	 */
1742 1743 1744 1745 1746
	if (dst->addrtype == TIPC_ADDR_MCAST) {
		res = -EINVAL;
		goto exit;
	}

Y
Ying Xue 已提交
1747
	previous = sock->state;
1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759
	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;

1760
		res = tipc_sendmsg(NULL, sock, &m, 0);
1761 1762 1763 1764 1765 1766 1767 1768 1769
		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 已提交
1770 1771 1772 1773 1774 1775 1776
		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);
1777 1778 1779 1780 1781 1782
		break;
	case SS_CONNECTED:
		res = -EISCONN;
		break;
	default:
		res = -EINVAL;
Y
Ying Xue 已提交
1783
		break;
1784
	}
1785 1786
exit:
	release_sock(sk);
1787
	return res;
P
Per Liden 已提交
1788 1789
}

1790
/**
1791
 * tipc_listen - allow socket to listen for incoming connections
P
Per Liden 已提交
1792 1793
 * @sock: socket structure
 * @len: (unused)
1794
 *
P
Per Liden 已提交
1795 1796
 * Returns 0 on success, errno otherwise
 */
1797
static int tipc_listen(struct socket *sock, int len)
P
Per Liden 已提交
1798
{
1799 1800 1801 1802
	struct sock *sk = sock->sk;
	int res;

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

1804
	if (sock->state != SS_UNCONNECTED)
1805 1806 1807 1808 1809 1810 1811 1812
		res = -EINVAL;
	else {
		sock->state = SS_LISTENING;
		res = 0;
	}

	release_sock(sk);
	return res;
P
Per Liden 已提交
1813 1814
}

Y
Ying Xue 已提交
1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828
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);
1829
		if (timeo && skb_queue_empty(&sk->sk_receive_queue)) {
Y
Ying Xue 已提交
1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850
			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;
}

1851
/**
1852
 * tipc_accept - wait for connection request
P
Per Liden 已提交
1853 1854 1855
 * @sock: listening socket
 * @newsock: new socket that is to be connected
 * @flags: file-related flags associated with socket
1856
 *
P
Per Liden 已提交
1857 1858
 * Returns 0 on success, errno otherwise
 */
1859
static int tipc_accept(struct socket *sock, struct socket *new_sock, int flags)
P
Per Liden 已提交
1860
{
1861
	struct sock *new_sk, *sk = sock->sk;
P
Per Liden 已提交
1862
	struct sk_buff *buf;
1863
	struct tipc_port *new_port;
1864
	struct tipc_msg *msg;
Y
Ying Xue 已提交
1865
	long timeo;
1866
	int res;
P
Per Liden 已提交
1867

1868
	lock_sock(sk);
P
Per Liden 已提交
1869

1870 1871
	if (sock->state != SS_LISTENING) {
		res = -EINVAL;
P
Per Liden 已提交
1872 1873
		goto exit;
	}
Y
Ying Xue 已提交
1874 1875 1876 1877
	timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
	res = tipc_wait_for_accept(sock, timeo);
	if (res)
		goto exit;
1878 1879 1880

	buf = skb_peek(&sk->sk_receive_queue);

1881
	res = tipc_sk_create(sock_net(sock->sk), new_sock, 0, 1);
1882 1883
	if (res)
		goto exit;
P
Per Liden 已提交
1884

1885
	new_sk = new_sock->sk;
1886
	new_port = &tipc_sk(new_sk)->port;
1887
	msg = buf_msg(buf);
P
Per Liden 已提交
1888

1889 1890 1891 1892 1893 1894 1895 1896 1897 1898
	/* 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 */
1899
	tipc_sk_finish_conn(new_port, msg_origport(msg), msg_orignode(msg));
1900 1901
	new_sock->state = SS_CONNECTED;

1902
	tipc_port_set_importance(new_port, msg_importance(msg));
1903
	if (msg_named(msg)) {
1904 1905
		new_port->conn_type = msg_nametype(msg);
		new_port->conn_instance = msg_nameinst(msg);
P
Per Liden 已提交
1906
	}
1907 1908 1909 1910 1911 1912 1913 1914 1915

	/*
	 * 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);
1916
		tipc_send_packet(NULL, new_sock, &m, 0);
1917 1918 1919
	} else {
		__skb_dequeue(&sk->sk_receive_queue);
		__skb_queue_head(&new_sk->sk_receive_queue, buf);
1920
		skb_set_owner_r(buf, new_sk);
1921 1922
	}
	release_sock(new_sk);
P
Per Liden 已提交
1923
exit:
1924
	release_sock(sk);
P
Per Liden 已提交
1925 1926 1927 1928
	return res;
}

/**
1929
 * tipc_shutdown - shutdown socket connection
P
Per Liden 已提交
1930
 * @sock: socket structure
1931
 * @how: direction to close (must be SHUT_RDWR)
P
Per Liden 已提交
1932 1933
 *
 * Terminates connection (if necessary), then purges socket's receive queue.
1934
 *
P
Per Liden 已提交
1935 1936
 * Returns 0 on success, errno otherwise
 */
1937
static int tipc_shutdown(struct socket *sock, int how)
P
Per Liden 已提交
1938
{
1939
	struct sock *sk = sock->sk;
1940 1941
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
1942
	struct sk_buff *buf;
1943
	u32 dnode;
P
Per Liden 已提交
1944 1945
	int res;

1946 1947
	if (how != SHUT_RDWR)
		return -EINVAL;
P
Per Liden 已提交
1948

1949
	lock_sock(sk);
P
Per Liden 已提交
1950 1951

	switch (sock->state) {
1952
	case SS_CONNECTING:
P
Per Liden 已提交
1953 1954 1955
	case SS_CONNECTED:

restart:
1956
		/* Disconnect and send a 'FIN+' or 'FIN-' message to peer */
1957 1958
		buf = __skb_dequeue(&sk->sk_receive_queue);
		if (buf) {
Y
Ying Xue 已提交
1959
			if (TIPC_SKB_CB(buf)->handle != NULL) {
1960
				kfree_skb(buf);
P
Per Liden 已提交
1961 1962
				goto restart;
			}
1963 1964
			if (tipc_msg_reverse(buf, &dnode, TIPC_CONN_SHUTDOWN))
				tipc_link_xmit(buf, dnode, port->ref);
1965
			tipc_node_remove_conn(dnode, port->ref);
1966
		} else {
1967 1968 1969 1970 1971 1972 1973
			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 已提交
1974
		}
1975
		port->connected = 0;
1976
		sock->state = SS_DISCONNECTING;
1977
		tipc_node_remove_conn(dnode, port->ref);
P
Per Liden 已提交
1978 1979 1980 1981
		/* fall through */

	case SS_DISCONNECTING:

1982
		/* Discard any unreceived messages */
1983
		__skb_queue_purge(&sk->sk_receive_queue);
1984 1985 1986

		/* Wake up anyone sleeping in poll */
		sk->sk_state_change(sk);
P
Per Liden 已提交
1987 1988 1989 1990 1991 1992 1993
		res = 0;
		break;

	default:
		res = -ENOTCONN;
	}

1994
	release_sock(sk);
P
Per Liden 已提交
1995 1996 1997
	return res;
}

1998 1999
static void tipc_sk_timeout(unsigned long ref)
{
2000 2001
	struct tipc_sock *tsk = tipc_port_lock(ref);
	struct tipc_port *port;
2002 2003 2004 2005 2006
	struct sock *sk;
	struct sk_buff *buf = NULL;
	struct tipc_msg *msg = NULL;
	u32 peer_port, peer_node;

2007
	if (!tsk)
2008 2009
		return;

2010
	port = &tsk->port;
2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041
	if (!port->connected) {
		tipc_port_unlock(port);
		return;
	}
	sk = &tsk->sk;
	bh_lock_sock(sk);
	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);
	tipc_port_unlock(port);
	if (!buf)
		return;

	msg = buf_msg(buf);
	tipc_link_xmit(buf, msg_destnode(msg),	msg_link_selector(msg));
}

2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062 2063 2064 2065 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 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136
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;

	tsk = tipc_sk_lock_next(&ref);
	for (; tsk; tsk = tipc_sk_lock_next(&ref)) {
		bh_lock_sock(&tsk->sk);
		str_len += tipc_sk_show(&tsk->port, pb + str_len,
					pb_len - str_len, 0);
		bh_unlock_sock(&tsk->sk);
		tipc_port_unlock(&tsk->port);
	}
	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;
	struct tipc_sock *tsk = tipc_sk_lock_next(&ref);

	for (; tsk; tsk = tipc_sk_lock_next(&ref)) {
		bh_lock_sock(&tsk->sk);
		msg = &tsk->port.phdr;
		msg_set_prevnode(msg, tipc_own_addr);
		msg_set_orignode(msg, tipc_own_addr);
		bh_unlock_sock(&tsk->sk);
		tipc_port_unlock(&tsk->port);
	}
}

P
Per Liden 已提交
2137
/**
2138
 * tipc_setsockopt - set socket option
P
Per Liden 已提交
2139 2140 2141 2142 2143
 * @sock: socket structure
 * @lvl: option level
 * @opt: option identifier
 * @ov: pointer to new option value
 * @ol: length of option value
2144 2145
 *
 * For stream sockets only, accepts and ignores all IPPROTO_TCP options
P
Per Liden 已提交
2146
 * (to ease compatibility).
2147
 *
P
Per Liden 已提交
2148 2149
 * Returns 0 on success, errno otherwise
 */
2150 2151
static int tipc_setsockopt(struct socket *sock, int lvl, int opt,
			   char __user *ov, unsigned int ol)
P
Per Liden 已提交
2152
{
2153
	struct sock *sk = sock->sk;
2154 2155
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
P
Per Liden 已提交
2156 2157 2158
	u32 value;
	int res;

2159 2160
	if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
		return 0;
P
Per Liden 已提交
2161 2162 2163 2164
	if (lvl != SOL_TIPC)
		return -ENOPROTOOPT;
	if (ol < sizeof(value))
		return -EINVAL;
2165 2166
	res = get_user(value, (u32 __user *)ov);
	if (res)
P
Per Liden 已提交
2167 2168
		return res;

2169
	lock_sock(sk);
2170

P
Per Liden 已提交
2171 2172
	switch (opt) {
	case TIPC_IMPORTANCE:
2173
		res = tipc_port_set_importance(port, value);
P
Per Liden 已提交
2174 2175 2176
		break;
	case TIPC_SRC_DROPPABLE:
		if (sock->type != SOCK_STREAM)
2177
			tipc_port_set_unreliable(port, value);
2178
		else
P
Per Liden 已提交
2179 2180 2181
			res = -ENOPROTOOPT;
		break;
	case TIPC_DEST_DROPPABLE:
2182
		tipc_port_set_unreturnable(port, value);
P
Per Liden 已提交
2183 2184
		break;
	case TIPC_CONN_TIMEOUT:
2185
		tipc_sk(sk)->conn_timeout = value;
2186
		/* no need to set "res", since already 0 at this point */
P
Per Liden 已提交
2187 2188 2189 2190 2191
		break;
	default:
		res = -EINVAL;
	}

2192 2193
	release_sock(sk);

P
Per Liden 已提交
2194 2195 2196 2197
	return res;
}

/**
2198
 * tipc_getsockopt - get socket option
P
Per Liden 已提交
2199 2200 2201 2202 2203
 * @sock: socket structure
 * @lvl: option level
 * @opt: option identifier
 * @ov: receptacle for option value
 * @ol: receptacle for length of option value
2204 2205
 *
 * For stream sockets only, returns 0 length result for all IPPROTO_TCP options
P
Per Liden 已提交
2206
 * (to ease compatibility).
2207
 *
P
Per Liden 已提交
2208 2209
 * Returns 0 on success, errno otherwise
 */
2210 2211
static int tipc_getsockopt(struct socket *sock, int lvl, int opt,
			   char __user *ov, int __user *ol)
P
Per Liden 已提交
2212
{
2213
	struct sock *sk = sock->sk;
2214 2215
	struct tipc_sock *tsk = tipc_sk(sk);
	struct tipc_port *port = &tsk->port;
2216
	int len;
P
Per Liden 已提交
2217
	u32 value;
2218
	int res;
P
Per Liden 已提交
2219

2220 2221
	if ((lvl == IPPROTO_TCP) && (sock->type == SOCK_STREAM))
		return put_user(0, ol);
P
Per Liden 已提交
2222 2223
	if (lvl != SOL_TIPC)
		return -ENOPROTOOPT;
2224 2225
	res = get_user(len, ol);
	if (res)
2226
		return res;
P
Per Liden 已提交
2227

2228
	lock_sock(sk);
P
Per Liden 已提交
2229 2230 2231

	switch (opt) {
	case TIPC_IMPORTANCE:
2232
		value = tipc_port_importance(port);
P
Per Liden 已提交
2233 2234
		break;
	case TIPC_SRC_DROPPABLE:
2235
		value = tipc_port_unreliable(port);
P
Per Liden 已提交
2236 2237
		break;
	case TIPC_DEST_DROPPABLE:
2238
		value = tipc_port_unreturnable(port);
P
Per Liden 已提交
2239 2240
		break;
	case TIPC_CONN_TIMEOUT:
2241
		value = tipc_sk(sk)->conn_timeout;
2242
		/* no need to set "res", since already 0 at this point */
P
Per Liden 已提交
2243
		break;
2244
	case TIPC_NODE_RECVQ_DEPTH:
2245
		value = 0; /* was tipc_queue_size, now obsolete */
2246
		break;
2247
	case TIPC_SOCK_RECVQ_DEPTH:
2248 2249
		value = skb_queue_len(&sk->sk_receive_queue);
		break;
P
Per Liden 已提交
2250 2251 2252 2253
	default:
		res = -EINVAL;
	}

2254 2255
	release_sock(sk);

2256 2257
	if (res)
		return res;	/* "get" failed */
P
Per Liden 已提交
2258

2259 2260 2261 2262 2263 2264 2265
	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 已提交
2266 2267
}

2268
static int tipc_ioctl(struct socket *sk, unsigned int cmd, unsigned long arg)
E
Erik Hugne 已提交
2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288
{
	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;
	}
}

2289 2290
/* Protocol switches for the various types of TIPC sockets */

2291
static const struct proto_ops msg_ops = {
2292
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2293
	.family		= AF_TIPC,
2294 2295 2296
	.release	= tipc_release,
	.bind		= tipc_bind,
	.connect	= tipc_connect,
2297
	.socketpair	= sock_no_socketpair,
2298
	.accept		= sock_no_accept,
2299 2300
	.getname	= tipc_getname,
	.poll		= tipc_poll,
E
Erik Hugne 已提交
2301
	.ioctl		= tipc_ioctl,
2302
	.listen		= sock_no_listen,
2303 2304 2305 2306 2307
	.shutdown	= tipc_shutdown,
	.setsockopt	= tipc_setsockopt,
	.getsockopt	= tipc_getsockopt,
	.sendmsg	= tipc_sendmsg,
	.recvmsg	= tipc_recvmsg,
2308 2309
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage
P
Per Liden 已提交
2310 2311
};

2312
static const struct proto_ops packet_ops = {
2313
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2314
	.family		= AF_TIPC,
2315 2316 2317
	.release	= tipc_release,
	.bind		= tipc_bind,
	.connect	= tipc_connect,
2318
	.socketpair	= sock_no_socketpair,
2319 2320 2321
	.accept		= tipc_accept,
	.getname	= tipc_getname,
	.poll		= tipc_poll,
E
Erik Hugne 已提交
2322
	.ioctl		= tipc_ioctl,
2323 2324 2325 2326 2327 2328
	.listen		= tipc_listen,
	.shutdown	= tipc_shutdown,
	.setsockopt	= tipc_setsockopt,
	.getsockopt	= tipc_getsockopt,
	.sendmsg	= tipc_send_packet,
	.recvmsg	= tipc_recvmsg,
2329 2330
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage
P
Per Liden 已提交
2331 2332
};

2333
static const struct proto_ops stream_ops = {
2334
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2335
	.family		= AF_TIPC,
2336 2337 2338
	.release	= tipc_release,
	.bind		= tipc_bind,
	.connect	= tipc_connect,
2339
	.socketpair	= sock_no_socketpair,
2340 2341 2342
	.accept		= tipc_accept,
	.getname	= tipc_getname,
	.poll		= tipc_poll,
E
Erik Hugne 已提交
2343
	.ioctl		= tipc_ioctl,
2344 2345 2346 2347 2348 2349
	.listen		= tipc_listen,
	.shutdown	= tipc_shutdown,
	.setsockopt	= tipc_setsockopt,
	.getsockopt	= tipc_getsockopt,
	.sendmsg	= tipc_send_stream,
	.recvmsg	= tipc_recv_stream,
2350 2351
	.mmap		= sock_no_mmap,
	.sendpage	= sock_no_sendpage
P
Per Liden 已提交
2352 2353
};

2354
static const struct net_proto_family tipc_family_ops = {
2355
	.owner		= THIS_MODULE,
P
Per Liden 已提交
2356
	.family		= AF_TIPC,
2357
	.create		= tipc_sk_create
P
Per Liden 已提交
2358 2359 2360 2361 2362
};

static struct proto tipc_proto = {
	.name		= "TIPC",
	.owner		= THIS_MODULE,
2363 2364
	.obj_size	= sizeof(struct tipc_sock),
	.sysctl_rmem	= sysctl_tipc_rmem
P
Per Liden 已提交
2365 2366
};

2367 2368 2369 2370 2371 2372
static struct proto tipc_proto_kern = {
	.name		= "TIPC",
	.obj_size	= sizeof(struct tipc_sock),
	.sysctl_rmem	= sysctl_tipc_rmem
};

P
Per Liden 已提交
2373
/**
2374
 * tipc_socket_init - initialize TIPC socket interface
2375
 *
P
Per Liden 已提交
2376 2377
 * Returns 0 on success, errno otherwise
 */
2378
int tipc_socket_init(void)
P
Per Liden 已提交
2379 2380 2381
{
	int res;

2382
	res = proto_register(&tipc_proto, 1);
P
Per Liden 已提交
2383
	if (res) {
2384
		pr_err("Failed to register TIPC protocol type\n");
P
Per Liden 已提交
2385 2386 2387 2388 2389
		goto out;
	}

	res = sock_register(&tipc_family_ops);
	if (res) {
2390
		pr_err("Failed to register TIPC socket type\n");
P
Per Liden 已提交
2391 2392 2393 2394 2395 2396 2397 2398
		proto_unregister(&tipc_proto);
		goto out;
	}
 out:
	return res;
}

/**
2399
 * tipc_socket_stop - stop TIPC socket interface
P
Per Liden 已提交
2400
 */
2401
void tipc_socket_stop(void)
P
Per Liden 已提交
2402 2403 2404 2405
{
	sock_unregister(tipc_family_ops.family);
	proto_unregister(&tipc_proto);
}