1 /* AF_RXRPC implementation
2 *
3 * Copyright (C) 2007 Red Hat, Inc. All Rights Reserved.
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
12 #include <linux/module.h>
13 #include <linux/kernel.h>
14 #include <linux/net.h>
15 #include <linux/slab.h>
16 #include <linux/skbuff.h>
17 #include <linux/poll.h>
18 #include <linux/proc_fs.h>
19 #include <linux/key-type.h>
20 #include <net/net_namespace.h>
21 #include <net/sock.h>
22 #include <net/af_rxrpc.h>
23 #include "ar-internal.h"
24
25 MODULE_DESCRIPTION("RxRPC network protocol");
26 MODULE_AUTHOR("Red Hat, Inc.");
27 MODULE_LICENSE("GPL");
28 MODULE_ALIAS_NETPROTO(PF_RXRPC);
29
30 unsigned int rxrpc_debug; // = RXRPC_DEBUG_KPROTO;
31 module_param_named(debug, rxrpc_debug, uint, S_IWUSR | S_IRUGO);
32 MODULE_PARM_DESC(debug, "RxRPC debugging mask");
33
34 static int sysctl_rxrpc_max_qlen __read_mostly = 10;
35
36 static struct proto rxrpc_proto;
37 static const struct proto_ops rxrpc_rpc_ops;
38
39 /* local epoch for detecting local-end reset */
40 __be32 rxrpc_epoch;
41
42 /* current debugging ID */
43 atomic_t rxrpc_debug_id;
44
45 /* count of skbs currently in use */
46 atomic_t rxrpc_n_skbs;
47
48 struct workqueue_struct *rxrpc_workqueue;
49
50 static void rxrpc_sock_destructor(struct sock *);
51
52 /*
53 * see if an RxRPC socket is currently writable
54 */
rxrpc_writable(struct sock * sk)55 static inline int rxrpc_writable(struct sock *sk)
56 {
57 return atomic_read(&sk->sk_wmem_alloc) < (size_t) sk->sk_sndbuf;
58 }
59
60 /*
61 * wait for write bufferage to become available
62 */
rxrpc_write_space(struct sock * sk)63 static void rxrpc_write_space(struct sock *sk)
64 {
65 _enter("%p", sk);
66 rcu_read_lock();
67 if (rxrpc_writable(sk)) {
68 struct socket_wq *wq = rcu_dereference(sk->sk_wq);
69
70 if (wq_has_sleeper(wq))
71 wake_up_interruptible(&wq->wait);
72 sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
73 }
74 rcu_read_unlock();
75 }
76
77 /*
78 * validate an RxRPC address
79 */
rxrpc_validate_address(struct rxrpc_sock * rx,struct sockaddr_rxrpc * srx,int len)80 static int rxrpc_validate_address(struct rxrpc_sock *rx,
81 struct sockaddr_rxrpc *srx,
82 int len)
83 {
84 if (len < sizeof(struct sockaddr_rxrpc))
85 return -EINVAL;
86
87 if (srx->srx_family != AF_RXRPC)
88 return -EAFNOSUPPORT;
89
90 if (srx->transport_type != SOCK_DGRAM)
91 return -ESOCKTNOSUPPORT;
92
93 len -= offsetof(struct sockaddr_rxrpc, transport);
94 if (srx->transport_len < sizeof(sa_family_t) ||
95 srx->transport_len > len)
96 return -EINVAL;
97
98 if (srx->transport.family != rx->proto)
99 return -EAFNOSUPPORT;
100
101 switch (srx->transport.family) {
102 case AF_INET:
103 _debug("INET: %x @ %pI4",
104 ntohs(srx->transport.sin.sin_port),
105 &srx->transport.sin.sin_addr);
106 if (srx->transport_len > 8)
107 memset((void *)&srx->transport + 8, 0,
108 srx->transport_len - 8);
109 break;
110
111 case AF_INET6:
112 default:
113 return -EAFNOSUPPORT;
114 }
115
116 return 0;
117 }
118
119 /*
120 * bind a local address to an RxRPC socket
121 */
rxrpc_bind(struct socket * sock,struct sockaddr * saddr,int len)122 static int rxrpc_bind(struct socket *sock, struct sockaddr *saddr, int len)
123 {
124 struct sockaddr_rxrpc *srx = (struct sockaddr_rxrpc *) saddr;
125 struct sock *sk = sock->sk;
126 struct rxrpc_local *local;
127 struct rxrpc_sock *rx = rxrpc_sk(sk), *prx;
128 __be16 service_id;
129 int ret;
130
131 _enter("%p,%p,%d", rx, saddr, len);
132
133 ret = rxrpc_validate_address(rx, srx, len);
134 if (ret < 0)
135 goto error;
136
137 lock_sock(&rx->sk);
138
139 if (rx->sk.sk_state != RXRPC_UNCONNECTED) {
140 ret = -EINVAL;
141 goto error_unlock;
142 }
143
144 memcpy(&rx->srx, srx, sizeof(rx->srx));
145
146 /* find a local transport endpoint if we don't have one already */
147 local = rxrpc_lookup_local(&rx->srx);
148 if (IS_ERR(local)) {
149 ret = PTR_ERR(local);
150 goto error_unlock;
151 }
152
153 rx->local = local;
154 if (srx->srx_service) {
155 service_id = htons(srx->srx_service);
156 write_lock_bh(&local->services_lock);
157 list_for_each_entry(prx, &local->services, listen_link) {
158 if (prx->service_id == service_id)
159 goto service_in_use;
160 }
161
162 rx->service_id = service_id;
163 list_add_tail(&rx->listen_link, &local->services);
164 write_unlock_bh(&local->services_lock);
165
166 rx->sk.sk_state = RXRPC_SERVER_BOUND;
167 } else {
168 rx->sk.sk_state = RXRPC_CLIENT_BOUND;
169 }
170
171 release_sock(&rx->sk);
172 _leave(" = 0");
173 return 0;
174
175 service_in_use:
176 ret = -EADDRINUSE;
177 write_unlock_bh(&local->services_lock);
178 error_unlock:
179 release_sock(&rx->sk);
180 error:
181 _leave(" = %d", ret);
182 return ret;
183 }
184
185 /*
186 * set the number of pending calls permitted on a listening socket
187 */
rxrpc_listen(struct socket * sock,int backlog)188 static int rxrpc_listen(struct socket *sock, int backlog)
189 {
190 struct sock *sk = sock->sk;
191 struct rxrpc_sock *rx = rxrpc_sk(sk);
192 int ret;
193
194 _enter("%p,%d", rx, backlog);
195
196 lock_sock(&rx->sk);
197
198 switch (rx->sk.sk_state) {
199 case RXRPC_UNCONNECTED:
200 ret = -EADDRNOTAVAIL;
201 break;
202 case RXRPC_CLIENT_BOUND:
203 case RXRPC_CLIENT_CONNECTED:
204 default:
205 ret = -EBUSY;
206 break;
207 case RXRPC_SERVER_BOUND:
208 ASSERT(rx->local != NULL);
209 sk->sk_max_ack_backlog = backlog;
210 rx->sk.sk_state = RXRPC_SERVER_LISTENING;
211 ret = 0;
212 break;
213 }
214
215 release_sock(&rx->sk);
216 _leave(" = %d", ret);
217 return ret;
218 }
219
220 /*
221 * find a transport by address
222 */
rxrpc_name_to_transport(struct socket * sock,struct sockaddr * addr,int addr_len,int flags,gfp_t gfp)223 static struct rxrpc_transport *rxrpc_name_to_transport(struct socket *sock,
224 struct sockaddr *addr,
225 int addr_len, int flags,
226 gfp_t gfp)
227 {
228 struct sockaddr_rxrpc *srx = (struct sockaddr_rxrpc *) addr;
229 struct rxrpc_transport *trans;
230 struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
231 struct rxrpc_peer *peer;
232
233 _enter("%p,%p,%d,%d", rx, addr, addr_len, flags);
234
235 ASSERT(rx->local != NULL);
236 ASSERT(rx->sk.sk_state > RXRPC_UNCONNECTED);
237
238 if (rx->srx.transport_type != srx->transport_type)
239 return ERR_PTR(-ESOCKTNOSUPPORT);
240 if (rx->srx.transport.family != srx->transport.family)
241 return ERR_PTR(-EAFNOSUPPORT);
242
243 /* find a remote transport endpoint from the local one */
244 peer = rxrpc_get_peer(srx, gfp);
245 if (IS_ERR(peer))
246 return ERR_CAST(peer);
247
248 /* find a transport */
249 trans = rxrpc_get_transport(rx->local, peer, gfp);
250 rxrpc_put_peer(peer);
251 _leave(" = %p", trans);
252 return trans;
253 }
254
255 /**
256 * rxrpc_kernel_begin_call - Allow a kernel service to begin a call
257 * @sock: The socket on which to make the call
258 * @srx: The address of the peer to contact (defaults to socket setting)
259 * @key: The security context to use (defaults to socket setting)
260 * @user_call_ID: The ID to use
261 *
262 * Allow a kernel service to begin a call on the nominated socket. This just
263 * sets up all the internal tracking structures and allocates connection and
264 * call IDs as appropriate. The call to be used is returned.
265 *
266 * The default socket destination address and security may be overridden by
267 * supplying @srx and @key.
268 */
rxrpc_kernel_begin_call(struct socket * sock,struct sockaddr_rxrpc * srx,struct key * key,unsigned long user_call_ID,gfp_t gfp)269 struct rxrpc_call *rxrpc_kernel_begin_call(struct socket *sock,
270 struct sockaddr_rxrpc *srx,
271 struct key *key,
272 unsigned long user_call_ID,
273 gfp_t gfp)
274 {
275 struct rxrpc_conn_bundle *bundle;
276 struct rxrpc_transport *trans;
277 struct rxrpc_call *call;
278 struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
279 __be16 service_id;
280
281 _enter(",,%x,%lx", key_serial(key), user_call_ID);
282
283 lock_sock(&rx->sk);
284
285 if (srx) {
286 trans = rxrpc_name_to_transport(sock, (struct sockaddr *) srx,
287 sizeof(*srx), 0, gfp);
288 if (IS_ERR(trans)) {
289 call = ERR_CAST(trans);
290 trans = NULL;
291 goto out_notrans;
292 }
293 } else {
294 trans = rx->trans;
295 if (!trans) {
296 call = ERR_PTR(-ENOTCONN);
297 goto out_notrans;
298 }
299 atomic_inc(&trans->usage);
300 }
301
302 service_id = rx->service_id;
303 if (srx)
304 service_id = htons(srx->srx_service);
305
306 if (!key)
307 key = rx->key;
308 if (key && !key->payload.data)
309 key = NULL; /* a no-security key */
310
311 bundle = rxrpc_get_bundle(rx, trans, key, service_id, gfp);
312 if (IS_ERR(bundle)) {
313 call = ERR_CAST(bundle);
314 goto out;
315 }
316
317 call = rxrpc_get_client_call(rx, trans, bundle, user_call_ID, true,
318 gfp);
319 rxrpc_put_bundle(trans, bundle);
320 out:
321 rxrpc_put_transport(trans);
322 out_notrans:
323 release_sock(&rx->sk);
324 _leave(" = %p", call);
325 return call;
326 }
327
328 EXPORT_SYMBOL(rxrpc_kernel_begin_call);
329
330 /**
331 * rxrpc_kernel_end_call - Allow a kernel service to end a call it was using
332 * @call: The call to end
333 *
334 * Allow a kernel service to end a call it was using. The call must be
335 * complete before this is called (the call should be aborted if necessary).
336 */
rxrpc_kernel_end_call(struct rxrpc_call * call)337 void rxrpc_kernel_end_call(struct rxrpc_call *call)
338 {
339 _enter("%d{%d}", call->debug_id, atomic_read(&call->usage));
340 rxrpc_remove_user_ID(call->socket, call);
341 rxrpc_put_call(call);
342 }
343
344 EXPORT_SYMBOL(rxrpc_kernel_end_call);
345
346 /**
347 * rxrpc_kernel_intercept_rx_messages - Intercept received RxRPC messages
348 * @sock: The socket to intercept received messages on
349 * @interceptor: The function to pass the messages to
350 *
351 * Allow a kernel service to intercept messages heading for the Rx queue on an
352 * RxRPC socket. They get passed to the specified function instead.
353 * @interceptor should free the socket buffers it is given. @interceptor is
354 * called with the socket receive queue spinlock held and softirqs disabled -
355 * this ensures that the messages will be delivered in the right order.
356 */
rxrpc_kernel_intercept_rx_messages(struct socket * sock,rxrpc_interceptor_t interceptor)357 void rxrpc_kernel_intercept_rx_messages(struct socket *sock,
358 rxrpc_interceptor_t interceptor)
359 {
360 struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
361
362 _enter("");
363 rx->interceptor = interceptor;
364 }
365
366 EXPORT_SYMBOL(rxrpc_kernel_intercept_rx_messages);
367
368 /*
369 * connect an RxRPC socket
370 * - this just targets it at a specific destination; no actual connection
371 * negotiation takes place
372 */
rxrpc_connect(struct socket * sock,struct sockaddr * addr,int addr_len,int flags)373 static int rxrpc_connect(struct socket *sock, struct sockaddr *addr,
374 int addr_len, int flags)
375 {
376 struct sockaddr_rxrpc *srx = (struct sockaddr_rxrpc *) addr;
377 struct sock *sk = sock->sk;
378 struct rxrpc_transport *trans;
379 struct rxrpc_local *local;
380 struct rxrpc_sock *rx = rxrpc_sk(sk);
381 int ret;
382
383 _enter("%p,%p,%d,%d", rx, addr, addr_len, flags);
384
385 ret = rxrpc_validate_address(rx, srx, addr_len);
386 if (ret < 0) {
387 _leave(" = %d [bad addr]", ret);
388 return ret;
389 }
390
391 lock_sock(&rx->sk);
392
393 switch (rx->sk.sk_state) {
394 case RXRPC_UNCONNECTED:
395 /* find a local transport endpoint if we don't have one already */
396 ASSERTCMP(rx->local, ==, NULL);
397 rx->srx.srx_family = AF_RXRPC;
398 rx->srx.srx_service = 0;
399 rx->srx.transport_type = srx->transport_type;
400 rx->srx.transport_len = sizeof(sa_family_t);
401 rx->srx.transport.family = srx->transport.family;
402 local = rxrpc_lookup_local(&rx->srx);
403 if (IS_ERR(local)) {
404 release_sock(&rx->sk);
405 return PTR_ERR(local);
406 }
407 rx->local = local;
408 rx->sk.sk_state = RXRPC_CLIENT_BOUND;
409 case RXRPC_CLIENT_BOUND:
410 break;
411 case RXRPC_CLIENT_CONNECTED:
412 release_sock(&rx->sk);
413 return -EISCONN;
414 default:
415 release_sock(&rx->sk);
416 return -EBUSY; /* server sockets can't connect as well */
417 }
418
419 trans = rxrpc_name_to_transport(sock, addr, addr_len, flags,
420 GFP_KERNEL);
421 if (IS_ERR(trans)) {
422 release_sock(&rx->sk);
423 _leave(" = %ld", PTR_ERR(trans));
424 return PTR_ERR(trans);
425 }
426
427 rx->trans = trans;
428 rx->service_id = htons(srx->srx_service);
429 rx->sk.sk_state = RXRPC_CLIENT_CONNECTED;
430
431 release_sock(&rx->sk);
432 return 0;
433 }
434
435 /*
436 * send a message through an RxRPC socket
437 * - in a client this does a number of things:
438 * - finds/sets up a connection for the security specified (if any)
439 * - initiates a call (ID in control data)
440 * - ends the request phase of a call (if MSG_MORE is not set)
441 * - sends a call data packet
442 * - may send an abort (abort code in control data)
443 */
rxrpc_sendmsg(struct kiocb * iocb,struct socket * sock,struct msghdr * m,size_t len)444 static int rxrpc_sendmsg(struct kiocb *iocb, struct socket *sock,
445 struct msghdr *m, size_t len)
446 {
447 struct rxrpc_transport *trans;
448 struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
449 int ret;
450
451 _enter(",{%d},,%zu", rx->sk.sk_state, len);
452
453 if (m->msg_flags & MSG_OOB)
454 return -EOPNOTSUPP;
455
456 if (m->msg_name) {
457 ret = rxrpc_validate_address(rx, m->msg_name, m->msg_namelen);
458 if (ret < 0) {
459 _leave(" = %d [bad addr]", ret);
460 return ret;
461 }
462 }
463
464 trans = NULL;
465 lock_sock(&rx->sk);
466
467 if (m->msg_name) {
468 ret = -EISCONN;
469 trans = rxrpc_name_to_transport(sock, m->msg_name,
470 m->msg_namelen, 0, GFP_KERNEL);
471 if (IS_ERR(trans)) {
472 ret = PTR_ERR(trans);
473 trans = NULL;
474 goto out;
475 }
476 } else {
477 trans = rx->trans;
478 if (trans)
479 atomic_inc(&trans->usage);
480 }
481
482 switch (rx->sk.sk_state) {
483 case RXRPC_SERVER_LISTENING:
484 if (!m->msg_name) {
485 ret = rxrpc_server_sendmsg(iocb, rx, m, len);
486 break;
487 }
488 case RXRPC_SERVER_BOUND:
489 case RXRPC_CLIENT_BOUND:
490 if (!m->msg_name) {
491 ret = -ENOTCONN;
492 break;
493 }
494 case RXRPC_CLIENT_CONNECTED:
495 ret = rxrpc_client_sendmsg(iocb, rx, trans, m, len);
496 break;
497 default:
498 ret = -ENOTCONN;
499 break;
500 }
501
502 out:
503 release_sock(&rx->sk);
504 if (trans)
505 rxrpc_put_transport(trans);
506 _leave(" = %d", ret);
507 return ret;
508 }
509
510 /*
511 * set RxRPC socket options
512 */
rxrpc_setsockopt(struct socket * sock,int level,int optname,char __user * optval,unsigned int optlen)513 static int rxrpc_setsockopt(struct socket *sock, int level, int optname,
514 char __user *optval, unsigned int optlen)
515 {
516 struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
517 unsigned int min_sec_level;
518 int ret;
519
520 _enter(",%d,%d,,%d", level, optname, optlen);
521
522 lock_sock(&rx->sk);
523 ret = -EOPNOTSUPP;
524
525 if (level == SOL_RXRPC) {
526 switch (optname) {
527 case RXRPC_EXCLUSIVE_CONNECTION:
528 ret = -EINVAL;
529 if (optlen != 0)
530 goto error;
531 ret = -EISCONN;
532 if (rx->sk.sk_state != RXRPC_UNCONNECTED)
533 goto error;
534 set_bit(RXRPC_SOCK_EXCLUSIVE_CONN, &rx->flags);
535 goto success;
536
537 case RXRPC_SECURITY_KEY:
538 ret = -EINVAL;
539 if (rx->key)
540 goto error;
541 ret = -EISCONN;
542 if (rx->sk.sk_state != RXRPC_UNCONNECTED)
543 goto error;
544 ret = rxrpc_request_key(rx, optval, optlen);
545 goto error;
546
547 case RXRPC_SECURITY_KEYRING:
548 ret = -EINVAL;
549 if (rx->key)
550 goto error;
551 ret = -EISCONN;
552 if (rx->sk.sk_state != RXRPC_UNCONNECTED)
553 goto error;
554 ret = rxrpc_server_keyring(rx, optval, optlen);
555 goto error;
556
557 case RXRPC_MIN_SECURITY_LEVEL:
558 ret = -EINVAL;
559 if (optlen != sizeof(unsigned int))
560 goto error;
561 ret = -EISCONN;
562 if (rx->sk.sk_state != RXRPC_UNCONNECTED)
563 goto error;
564 ret = get_user(min_sec_level,
565 (unsigned int __user *) optval);
566 if (ret < 0)
567 goto error;
568 ret = -EINVAL;
569 if (min_sec_level > RXRPC_SECURITY_MAX)
570 goto error;
571 rx->min_sec_level = min_sec_level;
572 goto success;
573
574 default:
575 break;
576 }
577 }
578
579 success:
580 ret = 0;
581 error:
582 release_sock(&rx->sk);
583 return ret;
584 }
585
586 /*
587 * permit an RxRPC socket to be polled
588 */
rxrpc_poll(struct file * file,struct socket * sock,poll_table * wait)589 static unsigned int rxrpc_poll(struct file *file, struct socket *sock,
590 poll_table *wait)
591 {
592 unsigned int mask;
593 struct sock *sk = sock->sk;
594
595 sock_poll_wait(file, sk_sleep(sk), wait);
596 mask = 0;
597
598 /* the socket is readable if there are any messages waiting on the Rx
599 * queue */
600 if (!skb_queue_empty(&sk->sk_receive_queue))
601 mask |= POLLIN | POLLRDNORM;
602
603 /* the socket is writable if there is space to add new data to the
604 * socket; there is no guarantee that any particular call in progress
605 * on the socket may have space in the Tx ACK window */
606 if (rxrpc_writable(sk))
607 mask |= POLLOUT | POLLWRNORM;
608
609 return mask;
610 }
611
612 /*
613 * create an RxRPC socket
614 */
rxrpc_create(struct net * net,struct socket * sock,int protocol,int kern)615 static int rxrpc_create(struct net *net, struct socket *sock, int protocol,
616 int kern)
617 {
618 struct rxrpc_sock *rx;
619 struct sock *sk;
620
621 _enter("%p,%d", sock, protocol);
622
623 if (!net_eq(net, &init_net))
624 return -EAFNOSUPPORT;
625
626 /* we support transport protocol UDP only */
627 if (protocol != PF_INET)
628 return -EPROTONOSUPPORT;
629
630 if (sock->type != SOCK_DGRAM)
631 return -ESOCKTNOSUPPORT;
632
633 sock->ops = &rxrpc_rpc_ops;
634 sock->state = SS_UNCONNECTED;
635
636 sk = sk_alloc(net, PF_RXRPC, GFP_KERNEL, &rxrpc_proto);
637 if (!sk)
638 return -ENOMEM;
639
640 sock_init_data(sock, sk);
641 sk->sk_state = RXRPC_UNCONNECTED;
642 sk->sk_write_space = rxrpc_write_space;
643 sk->sk_max_ack_backlog = sysctl_rxrpc_max_qlen;
644 sk->sk_destruct = rxrpc_sock_destructor;
645
646 rx = rxrpc_sk(sk);
647 rx->proto = protocol;
648 rx->calls = RB_ROOT;
649
650 INIT_LIST_HEAD(&rx->listen_link);
651 INIT_LIST_HEAD(&rx->secureq);
652 INIT_LIST_HEAD(&rx->acceptq);
653 rwlock_init(&rx->call_lock);
654 memset(&rx->srx, 0, sizeof(rx->srx));
655
656 _leave(" = 0 [%p]", rx);
657 return 0;
658 }
659
660 /*
661 * RxRPC socket destructor
662 */
rxrpc_sock_destructor(struct sock * sk)663 static void rxrpc_sock_destructor(struct sock *sk)
664 {
665 _enter("%p", sk);
666
667 rxrpc_purge_queue(&sk->sk_receive_queue);
668
669 WARN_ON(atomic_read(&sk->sk_wmem_alloc));
670 WARN_ON(!sk_unhashed(sk));
671 WARN_ON(sk->sk_socket);
672
673 if (!sock_flag(sk, SOCK_DEAD)) {
674 printk("Attempt to release alive rxrpc socket: %p\n", sk);
675 return;
676 }
677 }
678
679 /*
680 * release an RxRPC socket
681 */
rxrpc_release_sock(struct sock * sk)682 static int rxrpc_release_sock(struct sock *sk)
683 {
684 struct rxrpc_sock *rx = rxrpc_sk(sk);
685
686 _enter("%p{%d,%d}", sk, sk->sk_state, atomic_read(&sk->sk_refcnt));
687
688 /* declare the socket closed for business */
689 sock_orphan(sk);
690 sk->sk_shutdown = SHUTDOWN_MASK;
691
692 spin_lock_bh(&sk->sk_receive_queue.lock);
693 sk->sk_state = RXRPC_CLOSE;
694 spin_unlock_bh(&sk->sk_receive_queue.lock);
695
696 ASSERTCMP(rx->listen_link.next, !=, LIST_POISON1);
697
698 if (!list_empty(&rx->listen_link)) {
699 write_lock_bh(&rx->local->services_lock);
700 list_del(&rx->listen_link);
701 write_unlock_bh(&rx->local->services_lock);
702 }
703
704 /* try to flush out this socket */
705 rxrpc_release_calls_on_socket(rx);
706 flush_workqueue(rxrpc_workqueue);
707 rxrpc_purge_queue(&sk->sk_receive_queue);
708
709 if (rx->conn) {
710 rxrpc_put_connection(rx->conn);
711 rx->conn = NULL;
712 }
713
714 if (rx->bundle) {
715 rxrpc_put_bundle(rx->trans, rx->bundle);
716 rx->bundle = NULL;
717 }
718 if (rx->trans) {
719 rxrpc_put_transport(rx->trans);
720 rx->trans = NULL;
721 }
722 if (rx->local) {
723 rxrpc_put_local(rx->local);
724 rx->local = NULL;
725 }
726
727 key_put(rx->key);
728 rx->key = NULL;
729 key_put(rx->securities);
730 rx->securities = NULL;
731 sock_put(sk);
732
733 _leave(" = 0");
734 return 0;
735 }
736
737 /*
738 * release an RxRPC BSD socket on close() or equivalent
739 */
rxrpc_release(struct socket * sock)740 static int rxrpc_release(struct socket *sock)
741 {
742 struct sock *sk = sock->sk;
743
744 _enter("%p{%p}", sock, sk);
745
746 if (!sk)
747 return 0;
748
749 sock->sk = NULL;
750
751 return rxrpc_release_sock(sk);
752 }
753
754 /*
755 * RxRPC network protocol
756 */
757 static const struct proto_ops rxrpc_rpc_ops = {
758 .family = PF_UNIX,
759 .owner = THIS_MODULE,
760 .release = rxrpc_release,
761 .bind = rxrpc_bind,
762 .connect = rxrpc_connect,
763 .socketpair = sock_no_socketpair,
764 .accept = sock_no_accept,
765 .getname = sock_no_getname,
766 .poll = rxrpc_poll,
767 .ioctl = sock_no_ioctl,
768 .listen = rxrpc_listen,
769 .shutdown = sock_no_shutdown,
770 .setsockopt = rxrpc_setsockopt,
771 .getsockopt = sock_no_getsockopt,
772 .sendmsg = rxrpc_sendmsg,
773 .recvmsg = rxrpc_recvmsg,
774 .mmap = sock_no_mmap,
775 .sendpage = sock_no_sendpage,
776 };
777
778 static struct proto rxrpc_proto = {
779 .name = "RXRPC",
780 .owner = THIS_MODULE,
781 .obj_size = sizeof(struct rxrpc_sock),
782 .max_header = sizeof(struct rxrpc_header),
783 };
784
785 static const struct net_proto_family rxrpc_family_ops = {
786 .family = PF_RXRPC,
787 .create = rxrpc_create,
788 .owner = THIS_MODULE,
789 };
790
791 /*
792 * initialise and register the RxRPC protocol
793 */
af_rxrpc_init(void)794 static int __init af_rxrpc_init(void)
795 {
796 int ret = -1;
797
798 BUILD_BUG_ON(sizeof(struct rxrpc_skb_priv) > FIELD_SIZEOF(struct sk_buff, cb));
799
800 rxrpc_epoch = htonl(get_seconds());
801
802 ret = -ENOMEM;
803 rxrpc_call_jar = kmem_cache_create(
804 "rxrpc_call_jar", sizeof(struct rxrpc_call), 0,
805 SLAB_HWCACHE_ALIGN, NULL);
806 if (!rxrpc_call_jar) {
807 printk(KERN_NOTICE "RxRPC: Failed to allocate call jar\n");
808 goto error_call_jar;
809 }
810
811 rxrpc_workqueue = alloc_workqueue("krxrpcd", 0, 1);
812 if (!rxrpc_workqueue) {
813 printk(KERN_NOTICE "RxRPC: Failed to allocate work queue\n");
814 goto error_work_queue;
815 }
816
817 ret = proto_register(&rxrpc_proto, 1);
818 if (ret < 0) {
819 printk(KERN_CRIT "RxRPC: Cannot register protocol\n");
820 goto error_proto;
821 }
822
823 ret = sock_register(&rxrpc_family_ops);
824 if (ret < 0) {
825 printk(KERN_CRIT "RxRPC: Cannot register socket family\n");
826 goto error_sock;
827 }
828
829 ret = register_key_type(&key_type_rxrpc);
830 if (ret < 0) {
831 printk(KERN_CRIT "RxRPC: Cannot register client key type\n");
832 goto error_key_type;
833 }
834
835 ret = register_key_type(&key_type_rxrpc_s);
836 if (ret < 0) {
837 printk(KERN_CRIT "RxRPC: Cannot register server key type\n");
838 goto error_key_type_s;
839 }
840
841 ret = rxrpc_sysctl_init();
842 if (ret < 0) {
843 printk(KERN_CRIT "RxRPC: Cannot register sysctls\n");
844 goto error_sysctls;
845 }
846
847 #ifdef CONFIG_PROC_FS
848 proc_create("rxrpc_calls", 0, init_net.proc_net, &rxrpc_call_seq_fops);
849 proc_create("rxrpc_conns", 0, init_net.proc_net,
850 &rxrpc_connection_seq_fops);
851 #endif
852 return 0;
853
854 error_sysctls:
855 unregister_key_type(&key_type_rxrpc_s);
856 error_key_type_s:
857 unregister_key_type(&key_type_rxrpc);
858 error_key_type:
859 sock_unregister(PF_RXRPC);
860 error_sock:
861 proto_unregister(&rxrpc_proto);
862 error_proto:
863 destroy_workqueue(rxrpc_workqueue);
864 error_work_queue:
865 kmem_cache_destroy(rxrpc_call_jar);
866 error_call_jar:
867 return ret;
868 }
869
870 /*
871 * unregister the RxRPC protocol
872 */
af_rxrpc_exit(void)873 static void __exit af_rxrpc_exit(void)
874 {
875 _enter("");
876 rxrpc_sysctl_exit();
877 unregister_key_type(&key_type_rxrpc_s);
878 unregister_key_type(&key_type_rxrpc);
879 sock_unregister(PF_RXRPC);
880 proto_unregister(&rxrpc_proto);
881 rxrpc_destroy_all_calls();
882 rxrpc_destroy_all_connections();
883 rxrpc_destroy_all_transports();
884 rxrpc_destroy_all_peers();
885 rxrpc_destroy_all_locals();
886
887 ASSERTCMP(atomic_read(&rxrpc_n_skbs), ==, 0);
888
889 _debug("flush scheduled work");
890 flush_workqueue(rxrpc_workqueue);
891 remove_proc_entry("rxrpc_conns", init_net.proc_net);
892 remove_proc_entry("rxrpc_calls", init_net.proc_net);
893 destroy_workqueue(rxrpc_workqueue);
894 kmem_cache_destroy(rxrpc_call_jar);
895 _leave("");
896 }
897
898 module_init(af_rxrpc_init);
899 module_exit(af_rxrpc_exit);
900