Commit | Line | Data |
---|---|---|
45025bce | 1 | /* RxRPC virtual connection handler, common bits. |
17926a79 | 2 | * |
45025bce | 3 | * Copyright (C) 2007, 2016 Red Hat, Inc. All Rights Reserved. |
17926a79 DH |
4 | * Written by David Howells (dhowells@redhat.com) |
5 | * | |
6 | * This program is free software; you can redistribute it and/or | |
7 | * modify it under the terms of the GNU General Public License | |
8 | * as published by the Free Software Foundation; either version | |
9 | * 2 of the License, or (at your option) any later version. | |
10 | */ | |
11 | ||
9b6d5398 JP |
12 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
13 | ||
17926a79 | 14 | #include <linux/module.h> |
5a0e3ad6 | 15 | #include <linux/slab.h> |
17926a79 DH |
16 | #include <linux/net.h> |
17 | #include <linux/skbuff.h> | |
17926a79 DH |
18 | #include "ar-internal.h" |
19 | ||
5873c083 DH |
20 | /* |
21 | * Time till a connection expires after last use (in seconds). | |
22 | */ | |
f859ab61 DH |
23 | unsigned int __read_mostly rxrpc_connection_expiry = 10 * 60; |
24 | unsigned int __read_mostly rxrpc_closed_conn_expiry = 10; | |
5873c083 | 25 | |
45025bce DH |
26 | static void rxrpc_destroy_connection(struct rcu_head *); |
27 | ||
3136ef49 DH |
28 | static void rxrpc_connection_timer(struct timer_list *timer) |
29 | { | |
30 | struct rxrpc_connection *conn = | |
31 | container_of(timer, struct rxrpc_connection, timer); | |
32 | ||
33 | rxrpc_queue_conn(conn); | |
34 | } | |
35 | ||
17926a79 DH |
36 | /* |
37 | * allocate a new connection | |
38 | */ | |
c6d2b8d7 | 39 | struct rxrpc_connection *rxrpc_alloc_connection(gfp_t gfp) |
17926a79 DH |
40 | { |
41 | struct rxrpc_connection *conn; | |
42 | ||
43 | _enter(""); | |
44 | ||
45 | conn = kzalloc(sizeof(struct rxrpc_connection), gfp); | |
46 | if (conn) { | |
45025bce | 47 | INIT_LIST_HEAD(&conn->cache_link); |
999b69f8 | 48 | spin_lock_init(&conn->channel_lock); |
45025bce | 49 | INIT_LIST_HEAD(&conn->waiting_calls); |
3136ef49 | 50 | timer_setup(&conn->timer, &rxrpc_connection_timer, 0); |
17926a79 | 51 | INIT_WORK(&conn->processor, &rxrpc_process_connection); |
4d028b2c | 52 | INIT_LIST_HEAD(&conn->proc_link); |
999b69f8 | 53 | INIT_LIST_HEAD(&conn->link); |
17926a79 | 54 | skb_queue_head_init(&conn->rx_queue); |
e0e4d82f | 55 | conn->security = &rxrpc_no_security; |
17926a79 | 56 | spin_lock_init(&conn->state_lock); |
17926a79 | 57 | conn->debug_id = atomic_inc_return(&rxrpc_debug_id); |
17926a79 | 58 | conn->size_align = 4; |
f51b4480 | 59 | conn->idle_timestamp = jiffies; |
17926a79 DH |
60 | } |
61 | ||
16c61add | 62 | _leave(" = %p{%d}", conn, conn ? conn->debug_id : 0); |
17926a79 DH |
63 | return conn; |
64 | } | |
65 | ||
17926a79 | 66 | /* |
8496af50 DH |
67 | * Look up a connection in the cache by protocol parameters. |
68 | * | |
69 | * If successful, a pointer to the connection is returned, but no ref is taken. | |
70 | * NULL is returned if there is no match. | |
71 | * | |
0099dc58 DH |
72 | * When searching for a service call, if we find a peer but no connection, we |
73 | * return that through *_peer in case we need to create a new service call. | |
74 | * | |
8496af50 | 75 | * The caller must be holding the RCU read lock. |
17926a79 | 76 | */ |
8496af50 | 77 | struct rxrpc_connection *rxrpc_find_connection_rcu(struct rxrpc_local *local, |
0099dc58 DH |
78 | struct sk_buff *skb, |
79 | struct rxrpc_peer **_peer) | |
17926a79 DH |
80 | { |
81 | struct rxrpc_connection *conn; | |
1291e9d1 | 82 | struct rxrpc_conn_proto k; |
42886ffe | 83 | struct rxrpc_skb_priv *sp = rxrpc_skb(skb); |
1291e9d1 DH |
84 | struct sockaddr_rxrpc srx; |
85 | struct rxrpc_peer *peer; | |
17926a79 | 86 | |
8496af50 | 87 | _enter(",%x", sp->hdr.cid & RXRPC_CIDMASK); |
17926a79 | 88 | |
5a790b73 | 89 | if (rxrpc_extract_addr_from_skb(&srx, skb) < 0) |
1291e9d1 | 90 | goto not_found; |
17926a79 | 91 | |
46894a13 DH |
92 | if (srx.transport.family != local->srx.transport.family && |
93 | (srx.transport.family == AF_INET && | |
94 | local->srx.transport.family != AF_INET6)) { | |
1291e9d1 DH |
95 | pr_warn_ratelimited("AF_RXRPC: Protocol mismatch %u not %u\n", |
96 | srx.transport.family, | |
97 | local->srx.transport.family); | |
98 | goto not_found; | |
99 | } | |
100 | ||
101 | k.epoch = sp->hdr.epoch; | |
102 | k.cid = sp->hdr.cid & RXRPC_CIDMASK; | |
17926a79 | 103 | |
dc71db34 | 104 | if (rxrpc_to_server(sp)) { |
1291e9d1 DH |
105 | /* We need to look up service connections by the full protocol |
106 | * parameter set. We look up the peer first as an intermediate | |
107 | * step and then the connection from the peer's tree. | |
108 | */ | |
109 | peer = rxrpc_lookup_peer_rcu(local, &srx); | |
110 | if (!peer) | |
111 | goto not_found; | |
0099dc58 | 112 | *_peer = peer; |
8496af50 DH |
113 | conn = rxrpc_find_service_conn_rcu(peer, skb); |
114 | if (!conn || atomic_read(&conn->usage) == 0) | |
115 | goto not_found; | |
116 | _leave(" = %p", conn); | |
117 | return conn; | |
4a3388c8 | 118 | } else { |
8496af50 DH |
119 | /* Look up client connections by connection ID alone as their |
120 | * IDs are unique for this machine. | |
121 | */ | |
1291e9d1 | 122 | conn = idr_find(&rxrpc_client_conn_ids, |
8496af50 DH |
123 | sp->hdr.cid >> RXRPC_CIDSHIFT); |
124 | if (!conn || atomic_read(&conn->usage) == 0) { | |
125 | _debug("no conn"); | |
126 | goto not_found; | |
127 | } | |
128 | ||
129 | if (conn->proto.epoch != k.epoch || | |
1291e9d1 DH |
130 | conn->params.local != local) |
131 | goto not_found; | |
132 | ||
133 | peer = conn->params.peer; | |
134 | switch (srx.transport.family) { | |
135 | case AF_INET: | |
136 | if (peer->srx.transport.sin.sin_port != | |
137 | srx.transport.sin.sin_port || | |
138 | peer->srx.transport.sin.sin_addr.s_addr != | |
139 | srx.transport.sin.sin_addr.s_addr) | |
140 | goto not_found; | |
141 | break; | |
d1912747 | 142 | #ifdef CONFIG_AF_RXRPC_IPV6 |
75b54cb5 DH |
143 | case AF_INET6: |
144 | if (peer->srx.transport.sin6.sin6_port != | |
145 | srx.transport.sin6.sin6_port || | |
146 | memcmp(&peer->srx.transport.sin6.sin6_addr, | |
147 | &srx.transport.sin6.sin6_addr, | |
148 | sizeof(struct in6_addr)) != 0) | |
149 | goto not_found; | |
150 | break; | |
d1912747 | 151 | #endif |
1291e9d1 DH |
152 | default: |
153 | BUG(); | |
154 | } | |
155 | ||
1291e9d1 DH |
156 | _leave(" = %p", conn); |
157 | return conn; | |
17926a79 DH |
158 | } |
159 | ||
1291e9d1 | 160 | not_found: |
17926a79 DH |
161 | _leave(" = NULL"); |
162 | return NULL; | |
17926a79 DH |
163 | } |
164 | ||
999b69f8 DH |
165 | /* |
166 | * Disconnect a call and clear any channel it occupies when that call | |
a1399f8b DH |
167 | * terminates. The caller must hold the channel_lock and must release the |
168 | * call's ref on the connection. | |
999b69f8 | 169 | */ |
45025bce DH |
170 | void __rxrpc_disconnect_call(struct rxrpc_connection *conn, |
171 | struct rxrpc_call *call) | |
999b69f8 | 172 | { |
01a90a45 DH |
173 | struct rxrpc_channel *chan = |
174 | &conn->channels[call->cid & RXRPC_CHANNELMASK]; | |
999b69f8 | 175 | |
01a90a45 | 176 | _enter("%d,%x", conn->debug_id, call->cid); |
999b69f8 | 177 | |
a1399f8b DH |
178 | if (rcu_access_pointer(chan->call) == call) { |
179 | /* Save the result of the call so that we can repeat it if necessary | |
180 | * through the channel, whilst disposing of the actual call record. | |
181 | */ | |
b1d9f7fd | 182 | trace_rxrpc_disconnect_call(call); |
17e9e23b DH |
183 | switch (call->completion) { |
184 | case RXRPC_CALL_SUCCEEDED: | |
248f219c | 185 | chan->last_seq = call->rx_hard_ack; |
18bfeba5 | 186 | chan->last_type = RXRPC_PACKET_TYPE_ACK; |
17e9e23b DH |
187 | break; |
188 | case RXRPC_CALL_LOCALLY_ABORTED: | |
189 | chan->last_abort = call->abort_code; | |
190 | chan->last_type = RXRPC_PACKET_TYPE_ABORT; | |
191 | break; | |
192 | default: | |
193 | chan->last_abort = RX_USER_ABORT; | |
194 | chan->last_type = RXRPC_PACKET_TYPE_ABORT; | |
195 | break; | |
18bfeba5 | 196 | } |
17e9e23b | 197 | |
18bfeba5 | 198 | /* Sync with rxrpc_conn_retransmit(). */ |
a1399f8b DH |
199 | smp_wmb(); |
200 | chan->last_call = chan->call_id; | |
201 | chan->call_id = chan->call_counter; | |
e653cfe4 | 202 | |
a1399f8b | 203 | rcu_assign_pointer(chan->call, NULL); |
999b69f8 | 204 | } |
e653cfe4 | 205 | |
a1399f8b DH |
206 | _leave(""); |
207 | } | |
208 | ||
209 | /* | |
210 | * Disconnect a call and clear any channel it occupies when that call | |
211 | * terminates. | |
212 | */ | |
213 | void rxrpc_disconnect_call(struct rxrpc_call *call) | |
214 | { | |
215 | struct rxrpc_connection *conn = call->conn; | |
216 | ||
f7aec129 DH |
217 | call->peer->cong_cwnd = call->cong_cwnd; |
218 | ||
248f219c | 219 | spin_lock_bh(&conn->params.peer->lock); |
f3344303 | 220 | hlist_del_rcu(&call->error_link); |
248f219c DH |
221 | spin_unlock_bh(&conn->params.peer->lock); |
222 | ||
45025bce DH |
223 | if (rxrpc_is_client_call(call)) |
224 | return rxrpc_disconnect_client_call(call); | |
225 | ||
a1399f8b | 226 | spin_lock(&conn->channel_lock); |
45025bce | 227 | __rxrpc_disconnect_call(conn, call); |
e653cfe4 DH |
228 | spin_unlock(&conn->channel_lock); |
229 | ||
230 | call->conn = NULL; | |
f51b4480 | 231 | conn->idle_timestamp = jiffies; |
e653cfe4 | 232 | rxrpc_put_connection(conn); |
999b69f8 DH |
233 | } |
234 | ||
45025bce DH |
235 | /* |
236 | * Kill off a connection. | |
237 | */ | |
238 | void rxrpc_kill_connection(struct rxrpc_connection *conn) | |
239 | { | |
2baec2c3 DH |
240 | struct rxrpc_net *rxnet = conn->params.local->rxnet; |
241 | ||
45025bce DH |
242 | ASSERT(!rcu_access_pointer(conn->channels[0].call) && |
243 | !rcu_access_pointer(conn->channels[1].call) && | |
244 | !rcu_access_pointer(conn->channels[2].call) && | |
245 | !rcu_access_pointer(conn->channels[3].call)); | |
246 | ASSERT(list_empty(&conn->cache_link)); | |
247 | ||
2baec2c3 | 248 | write_lock(&rxnet->conn_lock); |
45025bce | 249 | list_del_init(&conn->proc_link); |
2baec2c3 | 250 | write_unlock(&rxnet->conn_lock); |
45025bce DH |
251 | |
252 | /* Drain the Rx queue. Note that even though we've unpublished, an | |
253 | * incoming packet could still be being added to our Rx queue, so we | |
254 | * will need to drain it again in the RCU cleanup handler. | |
255 | */ | |
256 | rxrpc_purge_queue(&conn->rx_queue); | |
257 | ||
258 | /* Leave final destruction to RCU. The connection processor work item | |
259 | * must carry a ref on the connection to prevent us getting here whilst | |
260 | * it is queued or running. | |
261 | */ | |
262 | call_rcu(&conn->rcu, rxrpc_destroy_connection); | |
263 | } | |
264 | ||
17926a79 | 265 | /* |
363deeab DH |
266 | * Queue a connection's work processor, getting a ref to pass to the work |
267 | * queue. | |
17926a79 | 268 | */ |
363deeab | 269 | bool rxrpc_queue_conn(struct rxrpc_connection *conn) |
17926a79 | 270 | { |
363deeab | 271 | const void *here = __builtin_return_address(0); |
bfc18e38 | 272 | int n = atomic_fetch_add_unless(&conn->usage, 1, 0); |
363deeab DH |
273 | if (n == 0) |
274 | return false; | |
275 | if (rxrpc_queue_work(&conn->processor)) | |
276 | trace_rxrpc_conn(conn, rxrpc_conn_queued, n + 1, here); | |
277 | else | |
278 | rxrpc_put_connection(conn); | |
279 | return true; | |
280 | } | |
281 | ||
282 | /* | |
283 | * Note the re-emergence of a connection. | |
284 | */ | |
285 | void rxrpc_see_connection(struct rxrpc_connection *conn) | |
286 | { | |
287 | const void *here = __builtin_return_address(0); | |
288 | if (conn) { | |
289 | int n = atomic_read(&conn->usage); | |
290 | ||
291 | trace_rxrpc_conn(conn, rxrpc_conn_seen, n, here); | |
292 | } | |
293 | } | |
294 | ||
295 | /* | |
296 | * Get a ref on a connection. | |
297 | */ | |
298 | void rxrpc_get_connection(struct rxrpc_connection *conn) | |
299 | { | |
300 | const void *here = __builtin_return_address(0); | |
301 | int n = atomic_inc_return(&conn->usage); | |
302 | ||
303 | trace_rxrpc_conn(conn, rxrpc_conn_got, n, here); | |
304 | } | |
305 | ||
306 | /* | |
307 | * Try to get a ref on a connection. | |
308 | */ | |
309 | struct rxrpc_connection * | |
310 | rxrpc_get_connection_maybe(struct rxrpc_connection *conn) | |
311 | { | |
312 | const void *here = __builtin_return_address(0); | |
313 | ||
314 | if (conn) { | |
bfc18e38 | 315 | int n = atomic_fetch_add_unless(&conn->usage, 1, 0); |
363deeab DH |
316 | if (n > 0) |
317 | trace_rxrpc_conn(conn, rxrpc_conn_got, n + 1, here); | |
318 | else | |
319 | conn = NULL; | |
320 | } | |
321 | return conn; | |
322 | } | |
323 | ||
3d18cbb7 DH |
324 | /* |
325 | * Set the service connection reap timer. | |
326 | */ | |
327 | static void rxrpc_set_service_reap_timer(struct rxrpc_net *rxnet, | |
328 | unsigned long reap_at) | |
329 | { | |
330 | if (rxnet->live) | |
331 | timer_reduce(&rxnet->service_conn_reap_timer, reap_at); | |
332 | } | |
333 | ||
363deeab DH |
334 | /* |
335 | * Release a service connection | |
336 | */ | |
337 | void rxrpc_put_service_conn(struct rxrpc_connection *conn) | |
338 | { | |
339 | const void *here = __builtin_return_address(0); | |
340 | int n; | |
341 | ||
342 | n = atomic_dec_return(&conn->usage); | |
343 | trace_rxrpc_conn(conn, rxrpc_conn_put_service, n, here); | |
344 | ASSERTCMP(n, >=, 0); | |
3d18cbb7 DH |
345 | if (n == 1) |
346 | rxrpc_set_service_reap_timer(conn->params.local->rxnet, | |
347 | jiffies + rxrpc_connection_expiry); | |
17926a79 DH |
348 | } |
349 | ||
350 | /* | |
351 | * destroy a virtual connection | |
352 | */ | |
dee46364 | 353 | static void rxrpc_destroy_connection(struct rcu_head *rcu) |
17926a79 | 354 | { |
dee46364 DH |
355 | struct rxrpc_connection *conn = |
356 | container_of(rcu, struct rxrpc_connection, rcu); | |
357 | ||
358 | _enter("{%d,u=%d}", conn->debug_id, atomic_read(&conn->usage)); | |
17926a79 DH |
359 | |
360 | ASSERTCMP(atomic_read(&conn->usage), ==, 0); | |
361 | ||
362 | _net("DESTROY CONN %d", conn->debug_id); | |
363 | ||
3136ef49 | 364 | del_timer_sync(&conn->timer); |
17926a79 DH |
365 | rxrpc_purge_queue(&conn->rx_queue); |
366 | ||
e0e4d82f | 367 | conn->security->clear(conn); |
19ffa01c | 368 | key_put(conn->params.key); |
e0e4d82f | 369 | key_put(conn->server_key); |
aa390bbe | 370 | rxrpc_put_peer(conn->params.peer); |
31f5f9a1 DH |
371 | |
372 | if (atomic_dec_and_test(&conn->params.local->rxnet->nr_conns)) | |
5bb053be | 373 | wake_up_var(&conn->params.local->rxnet->nr_conns); |
aa390bbe | 374 | rxrpc_put_local(conn->params.local); |
e0e4d82f | 375 | |
17926a79 DH |
376 | kfree(conn); |
377 | _leave(""); | |
378 | } | |
379 | ||
380 | /* | |
45025bce | 381 | * reap dead service connections |
17926a79 | 382 | */ |
2baec2c3 | 383 | void rxrpc_service_connection_reaper(struct work_struct *work) |
17926a79 DH |
384 | { |
385 | struct rxrpc_connection *conn, *_p; | |
2baec2c3 | 386 | struct rxrpc_net *rxnet = |
3d18cbb7 | 387 | container_of(work, struct rxrpc_net, service_conn_reaper); |
f859ab61 | 388 | unsigned long expire_at, earliest, idle_timestamp, now; |
17926a79 DH |
389 | |
390 | LIST_HEAD(graveyard); | |
391 | ||
392 | _enter(""); | |
393 | ||
f51b4480 | 394 | now = jiffies; |
f859ab61 | 395 | earliest = now + MAX_JIFFY_OFFSET; |
17926a79 | 396 | |
2baec2c3 DH |
397 | write_lock(&rxnet->conn_lock); |
398 | list_for_each_entry_safe(conn, _p, &rxnet->service_conns, link) { | |
001c1122 DH |
399 | ASSERTCMP(atomic_read(&conn->usage), >, 0); |
400 | if (likely(atomic_read(&conn->usage) > 1)) | |
17926a79 | 401 | continue; |
00e90712 DH |
402 | if (conn->state == RXRPC_CONN_SERVICE_PREALLOC) |
403 | continue; | |
17926a79 | 404 | |
f859ab61 DH |
405 | if (rxnet->live) { |
406 | idle_timestamp = READ_ONCE(conn->idle_timestamp); | |
407 | expire_at = idle_timestamp + rxrpc_connection_expiry * HZ; | |
408 | if (conn->params.local->service_closed) | |
409 | expire_at = idle_timestamp + rxrpc_closed_conn_expiry * HZ; | |
410 | ||
411 | _debug("reap CONN %d { u=%d,t=%ld }", | |
412 | conn->debug_id, atomic_read(&conn->usage), | |
413 | (long)expire_at - (long)now); | |
414 | ||
415 | if (time_before(now, expire_at)) { | |
416 | if (time_before(expire_at, earliest)) | |
417 | earliest = expire_at; | |
418 | continue; | |
419 | } | |
17926a79 | 420 | } |
001c1122 DH |
421 | |
422 | /* The usage count sits at 1 whilst the object is unused on the | |
423 | * list; we reduce that to 0 to make the object unavailable. | |
424 | */ | |
425 | if (atomic_cmpxchg(&conn->usage, 1, 0) != 1) | |
426 | continue; | |
88f2a825 | 427 | trace_rxrpc_conn(conn, rxrpc_conn_reap_service, 0, NULL); |
001c1122 DH |
428 | |
429 | if (rxrpc_conn_is_client(conn)) | |
45025bce | 430 | BUG(); |
001c1122 DH |
431 | else |
432 | rxrpc_unpublish_service_conn(conn); | |
433 | ||
434 | list_move_tail(&conn->link, &graveyard); | |
17926a79 | 435 | } |
2baec2c3 | 436 | write_unlock(&rxnet->conn_lock); |
17926a79 | 437 | |
f859ab61 DH |
438 | if (earliest != now + MAX_JIFFY_OFFSET) { |
439 | _debug("reschedule reaper %ld", (long)earliest - (long)now); | |
f51b4480 | 440 | ASSERT(time_after(earliest, now)); |
3d7682af | 441 | rxrpc_set_service_reap_timer(rxnet, earliest); |
17926a79 DH |
442 | } |
443 | ||
17926a79 DH |
444 | while (!list_empty(&graveyard)) { |
445 | conn = list_entry(graveyard.next, struct rxrpc_connection, | |
446 | link); | |
447 | list_del_init(&conn->link); | |
448 | ||
449 | ASSERTCMP(atomic_read(&conn->usage), ==, 0); | |
45025bce | 450 | rxrpc_kill_connection(conn); |
17926a79 DH |
451 | } |
452 | ||
453 | _leave(""); | |
454 | } | |
455 | ||
456 | /* | |
45025bce DH |
457 | * preemptively destroy all the service connection records rather than |
458 | * waiting for them to time out | |
17926a79 | 459 | */ |
2baec2c3 | 460 | void rxrpc_destroy_all_connections(struct rxrpc_net *rxnet) |
17926a79 | 461 | { |
dee46364 DH |
462 | struct rxrpc_connection *conn, *_p; |
463 | bool leak = false; | |
464 | ||
17926a79 DH |
465 | _enter(""); |
466 | ||
31f5f9a1 | 467 | atomic_dec(&rxnet->nr_conns); |
2baec2c3 | 468 | rxrpc_destroy_all_client_connections(rxnet); |
45025bce | 469 | |
3d18cbb7 DH |
470 | del_timer_sync(&rxnet->service_conn_reap_timer); |
471 | rxrpc_queue_work(&rxnet->service_conn_reaper); | |
dee46364 DH |
472 | flush_workqueue(rxrpc_workqueue); |
473 | ||
2baec2c3 DH |
474 | write_lock(&rxnet->conn_lock); |
475 | list_for_each_entry_safe(conn, _p, &rxnet->service_conns, link) { | |
dee46364 DH |
476 | pr_err("AF_RXRPC: Leaked conn %p {%d}\n", |
477 | conn, atomic_read(&conn->usage)); | |
478 | leak = true; | |
479 | } | |
2baec2c3 | 480 | write_unlock(&rxnet->conn_lock); |
dee46364 DH |
481 | BUG_ON(leak); |
482 | ||
2baec2c3 | 483 | ASSERT(list_empty(&rxnet->conn_proc_list)); |
17926a79 | 484 | |
31f5f9a1 DH |
485 | /* We need to wait for the connections to be destroyed by RCU as they |
486 | * pin things that we still need to get rid of. | |
487 | */ | |
5bb053be | 488 | wait_var_event(&rxnet->nr_conns, !atomic_read(&rxnet->nr_conns)); |
17926a79 DH |
489 | _leave(""); |
490 | } |