call_object.c 15.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11
/* RxRPC individual remote procedure call handling
 *
 * Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
 * Written by David Howells (dhowells@redhat.com)
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version
 * 2 of the License, or (at your option) any later version.
 */

12 13
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

14
#include <linux/slab.h>
15 16
#include <linux/module.h>
#include <linux/circ_buf.h>
17
#include <linux/spinlock_types.h>
18 19 20 21
#include <net/sock.h>
#include <net/af_rxrpc.h>
#include "ar-internal.h"

22
const char *const rxrpc_call_states[NR__RXRPC_CALL_STATES] = {
23
	[RXRPC_CALL_UNINITIALISED]		= "Uninit  ",
24
	[RXRPC_CALL_CLIENT_AWAIT_CONN]		= "ClWtConn",
25 26 27
	[RXRPC_CALL_CLIENT_SEND_REQUEST]	= "ClSndReq",
	[RXRPC_CALL_CLIENT_AWAIT_REPLY]		= "ClAwtRpl",
	[RXRPC_CALL_CLIENT_RECV_REPLY]		= "ClRcvRpl",
28
	[RXRPC_CALL_SERVER_PREALLOC]		= "SvPrealc",
29 30 31 32 33 34 35
	[RXRPC_CALL_SERVER_SECURING]		= "SvSecure",
	[RXRPC_CALL_SERVER_ACCEPTING]		= "SvAccept",
	[RXRPC_CALL_SERVER_RECV_REQUEST]	= "SvRcvReq",
	[RXRPC_CALL_SERVER_ACK_REQUEST]		= "SvAckReq",
	[RXRPC_CALL_SERVER_SEND_REPLY]		= "SvSndRpl",
	[RXRPC_CALL_SERVER_AWAIT_ACK]		= "SvAwtACK",
	[RXRPC_CALL_COMPLETE]			= "Complete",
36 37 38 39
};

const char *const rxrpc_call_completions[NR__RXRPC_CALL_COMPLETIONS] = {
	[RXRPC_CALL_SUCCEEDED]			= "Complete",
40 41
	[RXRPC_CALL_REMOTELY_ABORTED]		= "RmtAbort",
	[RXRPC_CALL_LOCALLY_ABORTED]		= "LocAbort",
42
	[RXRPC_CALL_LOCAL_ERROR]		= "LocError",
43 44 45
	[RXRPC_CALL_NETWORK_ERROR]		= "NetError",
};

46 47 48 49
struct kmem_cache *rxrpc_call_jar;
LIST_HEAD(rxrpc_calls);
DEFINE_RWLOCK(rxrpc_call_lock);

50 51 52 53 54 55
static void rxrpc_call_timer_expired(unsigned long _call)
{
	struct rxrpc_call *call = (struct rxrpc_call *)_call;

	_enter("%d", call->debug_id);

56 57
	if (call->state < RXRPC_CALL_COMPLETE)
		rxrpc_set_timer(call, rxrpc_timer_expired, ktime_get_real());
58
}
59

60 61 62 63 64 65 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
/*
 * find an extant server call
 * - called in process context with IRQs enabled
 */
struct rxrpc_call *rxrpc_find_call_by_user_ID(struct rxrpc_sock *rx,
					      unsigned long user_call_ID)
{
	struct rxrpc_call *call;
	struct rb_node *p;

	_enter("%p,%lx", rx, user_call_ID);

	read_lock(&rx->call_lock);

	p = rx->calls.rb_node;
	while (p) {
		call = rb_entry(p, struct rxrpc_call, sock_node);

		if (user_call_ID < call->user_call_ID)
			p = p->rb_left;
		else if (user_call_ID > call->user_call_ID)
			p = p->rb_right;
		else
			goto found_extant_call;
	}

	read_unlock(&rx->call_lock);
	_leave(" = NULL");
	return NULL;

found_extant_call:
91
	rxrpc_get_call(call, rxrpc_call_got);
92 93 94 95 96
	read_unlock(&rx->call_lock);
	_leave(" = %p [%d]", call, atomic_read(&call->usage));
	return call;
}

97 98 99
/*
 * allocate a new call
 */
100
struct rxrpc_call *rxrpc_alloc_call(gfp_t gfp)
101 102 103 104 105 106 107
{
	struct rxrpc_call *call;

	call = kmem_cache_zalloc(rxrpc_call_jar, gfp);
	if (!call)
		return NULL;

108 109
	call->rxtx_buffer = kcalloc(RXRPC_RXTX_BUFF_SIZE,
				    sizeof(struct sk_buff *),
110
				    gfp);
111 112
	if (!call->rxtx_buffer)
		goto nomem;
113

114 115 116 117
	call->rxtx_annotations = kcalloc(RXRPC_RXTX_BUFF_SIZE, sizeof(u8), gfp);
	if (!call->rxtx_annotations)
		goto nomem_2;

118
	mutex_init(&call->user_mutex);
119 120
	setup_timer(&call->timer, rxrpc_call_timer_expired,
		    (unsigned long)call);
121
	INIT_WORK(&call->processor, &rxrpc_process_call);
122
	INIT_LIST_HEAD(&call->link);
123
	INIT_LIST_HEAD(&call->chan_wait_link);
124
	INIT_LIST_HEAD(&call->accept_link);
125 126
	INIT_LIST_HEAD(&call->recvmsg_link);
	INIT_LIST_HEAD(&call->sock_link);
127
	init_waitqueue_head(&call->waitq);
128 129 130 131 132 133 134
	spin_lock_init(&call->lock);
	rwlock_init(&call->state_lock);
	atomic_set(&call->usage, 1);
	call->debug_id = atomic_inc_return(&rxrpc_debug_id);

	memset(&call->sock_node, 0xed, sizeof(call->sock_node));

135
	/* Leave space in the ring to handle a maxed-out jumbo packet */
136
	call->rx_winsize = rxrpc_rx_window_size;
137 138
	call->tx_winsize = 16;
	call->rx_expect_next = 1;
D
David Howells 已提交
139 140 141 142 143 144 145 146

	if (RXRPC_TX_SMSS > 2190)
		call->cong_cwnd = 2;
	else if (RXRPC_TX_SMSS > 1095)
		call->cong_cwnd = 3;
	else
		call->cong_cwnd = 4;
	call->cong_ssthresh = RXRPC_RXTX_BUFF_SIZE - 1;
147
	return call;
148 149 150 151 152 153

nomem_2:
	kfree(call->rxtx_buffer);
nomem:
	kmem_cache_free(rxrpc_call_jar, call);
	return NULL;
154 155 156
}

/*
157
 * Allocate a new client call.
158
 */
159
static struct rxrpc_call *rxrpc_alloc_client_call(struct sockaddr_rxrpc *srx,
160
						  gfp_t gfp)
161 162
{
	struct rxrpc_call *call;
D
David Howells 已提交
163
	ktime_t now;
164 165 166 167 168 169

	_enter("");

	call = rxrpc_alloc_call(gfp);
	if (!call)
		return ERR_PTR(-ENOMEM);
170 171
	call->state = RXRPC_CALL_CLIENT_AWAIT_CONN;
	call->service_id = srx->srx_service;
D
David Howells 已提交
172
	call->tx_phase = true;
D
David Howells 已提交
173 174 175
	now = ktime_get_real();
	call->acks_latest_ts = now;
	call->cong_tstamp = now;
176 177 178 179 180 181

	_leave(" = %p", call);
	return call;
}

/*
182
 * Initiate the call ack/resend/expiry timer.
183
 */
184
static void rxrpc_start_call_timer(struct rxrpc_call *call)
185
{
186
	ktime_t now = ktime_get_real(), expire_at;
187

188
	expire_at = ktime_add_ms(now, rxrpc_max_call_lifetime);
189 190
	call->expire_at = expire_at;
	call->ack_at = expire_at;
191
	call->ping_at = expire_at;
192
	call->resend_at = expire_at;
193 194
	call->timer.expires = jiffies + LONG_MAX / 2;
	rxrpc_set_timer(call, rxrpc_timer_begin, now);
195 196 197
}

/*
198 199 200
 * Set up a call for the given parameters.
 * - Called with the socket lock held, which it must release.
 * - If it returns a call, the call's lock will need releasing by the caller.
201
 */
202
struct rxrpc_call *rxrpc_new_client_call(struct rxrpc_sock *rx,
203
					 struct rxrpc_conn_parameters *cp,
204
					 struct sockaddr_rxrpc *srx,
205 206
					 unsigned long user_call_ID,
					 gfp_t gfp)
207
	__releases(&rx->sk.sk_lock.slock)
208
{
209 210
	struct rxrpc_call *call, *xcall;
	struct rb_node *parent, **pp;
D
David Howells 已提交
211
	const void *here = __builtin_return_address(0);
212
	int ret;
213

214
	_enter("%p,%lx", rx, user_call_ID);
215

216
	call = rxrpc_alloc_client_call(srx, gfp);
217
	if (IS_ERR(call)) {
218
		release_sock(&rx->sk);
219 220
		_leave(" = %ld", PTR_ERR(call));
		return call;
221 222
	}

223 224
	trace_rxrpc_call(call, rxrpc_call_new_client, atomic_read(&call->usage),
			 here, (const void *)user_call_ID);
D
David Howells 已提交
225

226 227 228 229 230
	/* We need to protect a partially set up call against the user as we
	 * will be acting outside the socket lock.
	 */
	mutex_lock(&call->user_mutex);

231
	/* Publish the call, even though it is incompletely set up as yet */
232 233 234 235 236 237
	write_lock(&rx->call_lock);

	pp = &rx->calls.rb_node;
	parent = NULL;
	while (*pp) {
		parent = *pp;
238
		xcall = rb_entry(parent, struct rxrpc_call, sock_node);
239

240
		if (user_call_ID < xcall->user_call_ID)
241
			pp = &(*pp)->rb_left;
242
		else if (user_call_ID > xcall->user_call_ID)
243 244
			pp = &(*pp)->rb_right;
		else
245
			goto error_dup_user_ID;
246 247
	}

248
	rcu_assign_pointer(call->socket, rx);
249 250
	call->user_call_ID = user_call_ID;
	__set_bit(RXRPC_CALL_HAS_USERID, &call->flags);
251
	rxrpc_get_call(call, rxrpc_call_got_userid);
252 253
	rb_link_node(&call->sock_node, parent, pp);
	rb_insert_color(&call->sock_node, &rx->calls);
254 255
	list_add(&call->sock_link, &rx->sock_calls);

256 257
	write_unlock(&rx->call_lock);

258
	write_lock(&rxrpc_call_lock);
259
	list_add_tail(&call->link, &rxrpc_calls);
260
	write_unlock(&rxrpc_call_lock);
261

262 263 264
	/* From this point on, the call is protected by its own lock. */
	release_sock(&rx->sk);

265 266 267 268
	/* Set up or get a connection record and set the protocol parameters,
	 * including channel number and call ID.
	 */
	ret = rxrpc_connect_call(call, cp, srx, gfp);
269 270 271
	if (ret < 0)
		goto error;

272
	trace_rxrpc_call(call, rxrpc_call_connected, atomic_read(&call->usage),
273
			 here, NULL);
274

275 276 277 278 279 280 281
	spin_lock_bh(&call->conn->params.peer->lock);
	hlist_add_head(&call->error_link,
		       &call->conn->params.peer->error_targets);
	spin_unlock_bh(&call->conn->params.peer->lock);

	rxrpc_start_call_timer(call);

282 283 284 285 286
	_net("CALL new %d on CONN %d", call->debug_id, call->conn->debug_id);

	_leave(" = %p [new]", call);
	return call;

287 288 289 290 291
	/* We unexpectedly found the user ID in the list after taking
	 * the call_lock.  This shouldn't happen unless the user races
	 * with itself and tries to add the same user ID twice at the
	 * same time in different threads.
	 */
292
error_dup_user_ID:
293
	write_unlock(&rx->call_lock);
294
	release_sock(&rx->sk);
295
	ret = -EEXIST;
296 297 298 299

error:
	__rxrpc_set_call_completion(call, RXRPC_CALL_LOCAL_ERROR,
				    RX_CALL_DEAD, ret);
300 301
	trace_rxrpc_call(call, rxrpc_call_error, atomic_read(&call->usage),
			 here, ERR_PTR(ret));
302
	rxrpc_release_call(rx, call);
303
	mutex_unlock(&call->user_mutex);
304 305 306
	rxrpc_put_call(call, rxrpc_call_put);
	_leave(" = %d", ret);
	return ERR_PTR(ret);
307 308 309
}

/*
310 311
 * Set up an incoming call.  call->conn points to the connection.
 * This is called in BH context and isn't allowed to fail.
312
 */
313 314 315
void rxrpc_incoming_call(struct rxrpc_sock *rx,
			 struct rxrpc_call *call,
			 struct sk_buff *skb)
316
{
317
	struct rxrpc_connection *conn = call->conn;
318
	struct rxrpc_skb_priv *sp = rxrpc_skb(skb);
319
	u32 chan;
320

321
	_enter(",%d", call->conn->debug_id);
D
David Howells 已提交
322

323 324 325 326 327 328 329
	rcu_assign_pointer(call->socket, rx);
	call->call_id		= sp->hdr.callNumber;
	call->service_id	= sp->hdr.serviceId;
	call->cid		= sp->hdr.cid;
	call->state		= RXRPC_CALL_SERVER_ACCEPTING;
	if (sp->hdr.securityIndex > 0)
		call->state	= RXRPC_CALL_SERVER_SECURING;
D
David Howells 已提交
330
	call->cong_tstamp	= skb->tstamp;
331 332 333 334 335 336

	/* Set the channel for this call.  We don't get channel_lock as we're
	 * only defending against the data_ready handler (which we're called
	 * from) and the RESPONSE packet parser (which is only really
	 * interested in call_counter and can cope with a disagreement with the
	 * call pointer).
337
	 */
338 339 340
	chan = sp->hdr.cid & RXRPC_CHANNELMASK;
	conn->channels[chan].call_counter = call->call_id;
	conn->channels[chan].call_id = call->call_id;
341
	rcu_assign_pointer(conn->channels[chan].call, call);
342

343 344 345
	spin_lock(&conn->params.peer->lock);
	hlist_add_head(&call->error_link, &conn->params.peer->error_targets);
	spin_unlock(&conn->params.peer->lock);
346 347 348

	_net("CALL incoming %d on CONN %d", call->debug_id, call->conn->debug_id);

349 350
	rxrpc_start_call_timer(call);
	_leave("");
351 352
}

353 354 355 356 357 358 359 360 361 362
/*
 * Queue a call's work processor, getting a ref to pass to the work queue.
 */
bool rxrpc_queue_call(struct rxrpc_call *call)
{
	const void *here = __builtin_return_address(0);
	int n = __atomic_add_unless(&call->usage, 1, 0);
	if (n == 0)
		return false;
	if (rxrpc_queue_work(&call->processor))
363
		trace_rxrpc_call(call, rxrpc_call_queued, n + 1, here, NULL);
364 365 366 367 368 369 370 371 372 373 374 375 376 377
	else
		rxrpc_put_call(call, rxrpc_call_put_noqueue);
	return true;
}

/*
 * Queue a call's work processor, passing the callers ref to the work queue.
 */
bool __rxrpc_queue_call(struct rxrpc_call *call)
{
	const void *here = __builtin_return_address(0);
	int n = atomic_read(&call->usage);
	ASSERTCMP(n, >=, 1);
	if (rxrpc_queue_work(&call->processor))
378
		trace_rxrpc_call(call, rxrpc_call_queued_ref, n, here, NULL);
379 380 381 382 383
	else
		rxrpc_put_call(call, rxrpc_call_put_noqueue);
	return true;
}

D
David Howells 已提交
384 385 386 387 388 389 390 391 392
/*
 * Note the re-emergence of a call.
 */
void rxrpc_see_call(struct rxrpc_call *call)
{
	const void *here = __builtin_return_address(0);
	if (call) {
		int n = atomic_read(&call->usage);

393
		trace_rxrpc_call(call, rxrpc_call_seen, n, here, NULL);
D
David Howells 已提交
394 395 396 397 398 399
	}
}

/*
 * Note the addition of a ref on a call.
 */
400
void rxrpc_get_call(struct rxrpc_call *call, enum rxrpc_call_trace op)
D
David Howells 已提交
401 402 403 404
{
	const void *here = __builtin_return_address(0);
	int n = atomic_inc_return(&call->usage);

405
	trace_rxrpc_call(call, op, n, here, NULL);
D
David Howells 已提交
406 407 408
}

/*
409
 * Detach a call from its owning socket.
D
David Howells 已提交
410
 */
411
void rxrpc_release_call(struct rxrpc_sock *rx, struct rxrpc_call *call)
D
David Howells 已提交
412
{
413
	const void *here = __builtin_return_address(0);
414 415 416
	struct rxrpc_connection *conn = call->conn;
	bool put = false;
	int i;
D
David Howells 已提交
417

418
	_enter("{%d,%d}", call->debug_id, atomic_read(&call->usage));
D
David Howells 已提交
419

420 421
	trace_rxrpc_call(call, rxrpc_call_release, atomic_read(&call->usage),
			 here, (const void *)call->flags);
422

423
	ASSERTCMP(call->state, ==, RXRPC_CALL_COMPLETE);
D
David Howells 已提交
424

425 426 427 428 429
	spin_lock_bh(&call->lock);
	if (test_and_set_bit(RXRPC_CALL_RELEASED, &call->flags))
		BUG();
	spin_unlock_bh(&call->lock);

430
	del_timer_sync(&call->timer);
431

432 433
	/* Make sure we don't get any more notifications */
	write_lock_bh(&rx->recvmsg_lock);
434

435
	if (!list_empty(&call->recvmsg_link)) {
436 437
		_debug("unlinking once-pending call %p { e=%lx f=%lx }",
		       call, call->events, call->flags);
438 439 440 441 442 443 444 445 446 447 448 449 450 451 452
		list_del(&call->recvmsg_link);
		put = true;
	}

	/* list_empty() must return false in rxrpc_notify_socket() */
	call->recvmsg_link.next = NULL;
	call->recvmsg_link.prev = NULL;

	write_unlock_bh(&rx->recvmsg_lock);
	if (put)
		rxrpc_put_call(call, rxrpc_call_put);

	write_lock(&rx->call_lock);

	if (test_and_clear_bit(RXRPC_CALL_HAS_USERID, &call->flags)) {
453 454
		rb_erase(&call->sock_node, &rx->calls);
		memset(&call->sock_node, 0xdd, sizeof(call->sock_node));
455
		rxrpc_put_call(call, rxrpc_call_put_userid);
456 457
	}

458 459 460 461 462 463
	list_del(&call->sock_link);
	write_unlock(&rx->call_lock);

	_debug("RELEASE CALL %p (%d CONN %p)", call, call->debug_id, conn);

	if (conn)
464
		rxrpc_disconnect_call(call);
465

466
	for (i = 0; i < RXRPC_RXTX_BUFF_SIZE; i++) {
D
David Howells 已提交
467 468 469
		rxrpc_free_skb(call->rxtx_buffer[i],
			       (call->tx_phase ? rxrpc_skb_tx_cleaned :
				rxrpc_skb_rx_cleaned));
470
		call->rxtx_buffer[i] = NULL;
471 472 473 474 475 476 477 478 479 480 481 482 483 484
	}

	_leave("");
}

/*
 * release all the calls associated with a socket
 */
void rxrpc_release_calls_on_socket(struct rxrpc_sock *rx)
{
	struct rxrpc_call *call;

	_enter("%p", rx);

485 486 487 488 489 490 491 492
	while (!list_empty(&rx->to_be_accepted)) {
		call = list_entry(rx->to_be_accepted.next,
				  struct rxrpc_call, accept_link);
		list_del(&call->accept_link);
		rxrpc_abort_call("SKR", call, 0, RX_CALL_DEAD, ECONNRESET);
		rxrpc_put_call(call, rxrpc_call_put);
	}

493 494 495 496 497
	while (!list_empty(&rx->sock_calls)) {
		call = list_entry(rx->sock_calls.next,
				  struct rxrpc_call, sock_link);
		rxrpc_get_call(call, rxrpc_call_got);
		rxrpc_abort_call("SKT", call, 0, RX_CALL_DEAD, ECONNRESET);
498
		rxrpc_send_abort_packet(call);
499
		rxrpc_release_call(rx, call);
500
		rxrpc_put_call(call, rxrpc_call_put);
501 502
	}

503 504 505 506 507 508
	_leave("");
}

/*
 * release a call
 */
509
void rxrpc_put_call(struct rxrpc_call *call, enum rxrpc_call_trace op)
510
{
D
David Howells 已提交
511
	const void *here = __builtin_return_address(0);
512
	int n;
513

D
David Howells 已提交
514
	ASSERT(call != NULL);
515

D
David Howells 已提交
516
	n = atomic_dec_return(&call->usage);
517
	trace_rxrpc_call(call, op, n, here, NULL);
D
David Howells 已提交
518 519 520
	ASSERTCMP(n, >=, 0);
	if (n == 0) {
		_debug("call %d dead", call->debug_id);
521
		ASSERTCMP(call->state, ==, RXRPC_CALL_COMPLETE);
522

523 524 525
		write_lock(&rxrpc_call_lock);
		list_del_init(&call->link);
		write_unlock(&rxrpc_call_lock);
D
David Howells 已提交
526

527
		rxrpc_cleanup_call(call);
528 529 530
	}
}

531 532 533 534 535 536 537
/*
 * Final call destruction under RCU.
 */
static void rxrpc_rcu_destroy_call(struct rcu_head *rcu)
{
	struct rxrpc_call *call = container_of(rcu, struct rxrpc_call, rcu);

538
	rxrpc_put_peer(call->peer);
539 540
	kfree(call->rxtx_buffer);
	kfree(call->rxtx_annotations);
541 542 543
	kmem_cache_free(rxrpc_call_jar, call);
}

544 545 546
/*
 * clean up a call
 */
547
void rxrpc_cleanup_call(struct rxrpc_call *call)
548
{
549
	int i;
550

551
	_net("DESTROY CALL %d", call->debug_id);
552 553 554

	memset(&call->sock_node, 0xcd, sizeof(call->sock_node));

555
	del_timer_sync(&call->timer);
556

557
	ASSERTCMP(call->state, ==, RXRPC_CALL_COMPLETE);
558
	ASSERT(test_bit(RXRPC_CALL_RELEASED, &call->flags));
559
	ASSERTCMP(call->conn, ==, NULL);
560

561 562
	/* Clean up the Rx/Tx buffer */
	for (i = 0; i < RXRPC_RXTX_BUFF_SIZE; i++)
D
David Howells 已提交
563 564 565
		rxrpc_free_skb(call->rxtx_buffer[i],
			       (call->tx_phase ? rxrpc_skb_tx_cleaned :
				rxrpc_skb_rx_cleaned));
566

D
David Howells 已提交
567
	rxrpc_free_skb(call->tx_pending, rxrpc_skb_tx_cleaned);
568

569
	call_rcu(&call->rcu, rxrpc_rcu_destroy_call);
570 571 572
}

/*
573
 * Make sure that all calls are gone.
574 575 576 577 578 579
 */
void __exit rxrpc_destroy_all_calls(void)
{
	struct rxrpc_call *call;

	_enter("");
580 581 582

	if (list_empty(&rxrpc_calls))
		return;
583 584

	write_lock(&rxrpc_call_lock);
585 586 587 588 589

	while (!list_empty(&rxrpc_calls)) {
		call = list_entry(rxrpc_calls.next, struct rxrpc_call, link);
		_debug("Zapping call %p", call);

D
David Howells 已提交
590
		rxrpc_see_call(call);
591 592
		list_del_init(&call->link);

593
		pr_err("Call %p still in use (%d,%s,%lx,%lx)!\n",
594 595 596
		       call, atomic_read(&call->usage),
		       rxrpc_call_states[call->state],
		       call->flags, call->events);
597

598
		write_unlock(&rxrpc_call_lock);
599
		cond_resched();
600
		write_lock(&rxrpc_call_lock);
601 602
	}

603
	write_unlock(&rxrpc_call_lock);
604
}