af_rxrpc.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813
  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. #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  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/random.h>
  18. #include <linux/poll.h>
  19. #include <linux/proc_fs.h>
  20. #include <linux/key-type.h>
  21. #include <net/net_namespace.h>
  22. #include <net/sock.h>
  23. #include <net/af_rxrpc.h>
  24. #define CREATE_TRACE_POINTS
  25. #include "ar-internal.h"
  26. MODULE_DESCRIPTION("RxRPC network protocol");
  27. MODULE_AUTHOR("Red Hat, Inc.");
  28. MODULE_LICENSE("GPL");
  29. MODULE_ALIAS_NETPROTO(PF_RXRPC);
  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. static struct proto rxrpc_proto;
  34. static const struct proto_ops rxrpc_rpc_ops;
  35. /* local epoch for detecting local-end reset */
  36. u32 rxrpc_epoch;
  37. /* current debugging ID */
  38. atomic_t rxrpc_debug_id;
  39. /* count of skbs currently in use */
  40. atomic_t rxrpc_n_skbs;
  41. struct workqueue_struct *rxrpc_workqueue;
  42. static void rxrpc_sock_destructor(struct sock *);
  43. /*
  44. * see if an RxRPC socket is currently writable
  45. */
  46. static inline int rxrpc_writable(struct sock *sk)
  47. {
  48. return atomic_read(&sk->sk_wmem_alloc) < (size_t) sk->sk_sndbuf;
  49. }
  50. /*
  51. * wait for write bufferage to become available
  52. */
  53. static void rxrpc_write_space(struct sock *sk)
  54. {
  55. _enter("%p", sk);
  56. rcu_read_lock();
  57. if (rxrpc_writable(sk)) {
  58. struct socket_wq *wq = rcu_dereference(sk->sk_wq);
  59. if (skwq_has_sleeper(wq))
  60. wake_up_interruptible(&wq->wait);
  61. sk_wake_async(sk, SOCK_WAKE_SPACE, POLL_OUT);
  62. }
  63. rcu_read_unlock();
  64. }
  65. /*
  66. * validate an RxRPC address
  67. */
  68. static int rxrpc_validate_address(struct rxrpc_sock *rx,
  69. struct sockaddr_rxrpc *srx,
  70. int len)
  71. {
  72. unsigned int tail;
  73. if (len < sizeof(struct sockaddr_rxrpc))
  74. return -EINVAL;
  75. if (srx->srx_family != AF_RXRPC)
  76. return -EAFNOSUPPORT;
  77. if (srx->transport_type != SOCK_DGRAM)
  78. return -ESOCKTNOSUPPORT;
  79. len -= offsetof(struct sockaddr_rxrpc, transport);
  80. if (srx->transport_len < sizeof(sa_family_t) ||
  81. srx->transport_len > len)
  82. return -EINVAL;
  83. if (srx->transport.family != rx->family)
  84. return -EAFNOSUPPORT;
  85. switch (srx->transport.family) {
  86. case AF_INET:
  87. if (srx->transport_len < sizeof(struct sockaddr_in))
  88. return -EINVAL;
  89. _debug("INET: %x @ %pI4",
  90. ntohs(srx->transport.sin.sin_port),
  91. &srx->transport.sin.sin_addr);
  92. tail = offsetof(struct sockaddr_rxrpc, transport.sin.__pad);
  93. break;
  94. case AF_INET6:
  95. default:
  96. return -EAFNOSUPPORT;
  97. }
  98. if (tail < len)
  99. memset((void *)srx + tail, 0, len - tail);
  100. return 0;
  101. }
  102. /*
  103. * bind a local address to an RxRPC socket
  104. */
  105. static int rxrpc_bind(struct socket *sock, struct sockaddr *saddr, int len)
  106. {
  107. struct sockaddr_rxrpc *srx = (struct sockaddr_rxrpc *)saddr;
  108. struct sock *sk = sock->sk;
  109. struct rxrpc_local *local;
  110. struct rxrpc_sock *rx = rxrpc_sk(sk), *prx;
  111. int ret;
  112. _enter("%p,%p,%d", rx, saddr, len);
  113. ret = rxrpc_validate_address(rx, srx, len);
  114. if (ret < 0)
  115. goto error;
  116. lock_sock(&rx->sk);
  117. if (rx->sk.sk_state != RXRPC_UNBOUND) {
  118. ret = -EINVAL;
  119. goto error_unlock;
  120. }
  121. memcpy(&rx->srx, srx, sizeof(rx->srx));
  122. local = rxrpc_lookup_local(&rx->srx);
  123. if (IS_ERR(local)) {
  124. ret = PTR_ERR(local);
  125. goto error_unlock;
  126. }
  127. if (rx->srx.srx_service) {
  128. write_lock_bh(&local->services_lock);
  129. list_for_each_entry(prx, &local->services, listen_link) {
  130. if (prx->srx.srx_service == rx->srx.srx_service)
  131. goto service_in_use;
  132. }
  133. rx->local = local;
  134. list_add_tail(&rx->listen_link, &local->services);
  135. write_unlock_bh(&local->services_lock);
  136. rx->sk.sk_state = RXRPC_SERVER_BOUND;
  137. } else {
  138. rx->local = local;
  139. rx->sk.sk_state = RXRPC_CLIENT_BOUND;
  140. }
  141. release_sock(&rx->sk);
  142. _leave(" = 0");
  143. return 0;
  144. service_in_use:
  145. write_unlock_bh(&local->services_lock);
  146. rxrpc_put_local(local);
  147. ret = -EADDRINUSE;
  148. error_unlock:
  149. release_sock(&rx->sk);
  150. error:
  151. _leave(" = %d", ret);
  152. return ret;
  153. }
  154. /*
  155. * set the number of pending calls permitted on a listening socket
  156. */
  157. static int rxrpc_listen(struct socket *sock, int backlog)
  158. {
  159. struct sock *sk = sock->sk;
  160. struct rxrpc_sock *rx = rxrpc_sk(sk);
  161. unsigned int max;
  162. int ret;
  163. _enter("%p,%d", rx, backlog);
  164. lock_sock(&rx->sk);
  165. switch (rx->sk.sk_state) {
  166. case RXRPC_UNBOUND:
  167. ret = -EADDRNOTAVAIL;
  168. break;
  169. case RXRPC_SERVER_BOUND:
  170. ASSERT(rx->local != NULL);
  171. max = READ_ONCE(rxrpc_max_backlog);
  172. ret = -EINVAL;
  173. if (backlog == INT_MAX)
  174. backlog = max;
  175. else if (backlog < 0 || backlog > max)
  176. break;
  177. sk->sk_max_ack_backlog = backlog;
  178. rx->sk.sk_state = RXRPC_SERVER_LISTENING;
  179. ret = 0;
  180. break;
  181. default:
  182. ret = -EBUSY;
  183. break;
  184. }
  185. release_sock(&rx->sk);
  186. _leave(" = %d", ret);
  187. return ret;
  188. }
  189. /**
  190. * rxrpc_kernel_begin_call - Allow a kernel service to begin a call
  191. * @sock: The socket on which to make the call
  192. * @srx: The address of the peer to contact
  193. * @key: The security context to use (defaults to socket setting)
  194. * @user_call_ID: The ID to use
  195. * @gfp: The allocation constraints
  196. * @notify_rx: Where to send notifications instead of socket queue
  197. *
  198. * Allow a kernel service to begin a call on the nominated socket. This just
  199. * sets up all the internal tracking structures and allocates connection and
  200. * call IDs as appropriate. The call to be used is returned.
  201. *
  202. * The default socket destination address and security may be overridden by
  203. * supplying @srx and @key.
  204. */
  205. struct rxrpc_call *rxrpc_kernel_begin_call(struct socket *sock,
  206. struct sockaddr_rxrpc *srx,
  207. struct key *key,
  208. unsigned long user_call_ID,
  209. gfp_t gfp,
  210. rxrpc_notify_rx_t notify_rx)
  211. {
  212. struct rxrpc_conn_parameters cp;
  213. struct rxrpc_call *call;
  214. struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
  215. int ret;
  216. _enter(",,%x,%lx", key_serial(key), user_call_ID);
  217. ret = rxrpc_validate_address(rx, srx, sizeof(*srx));
  218. if (ret < 0)
  219. return ERR_PTR(ret);
  220. lock_sock(&rx->sk);
  221. if (!key)
  222. key = rx->key;
  223. if (key && !key->payload.data[0])
  224. key = NULL; /* a no-security key */
  225. memset(&cp, 0, sizeof(cp));
  226. cp.local = rx->local;
  227. cp.key = key;
  228. cp.security_level = 0;
  229. cp.exclusive = false;
  230. cp.service_id = srx->srx_service;
  231. call = rxrpc_new_client_call(rx, &cp, srx, user_call_ID, gfp);
  232. if (!IS_ERR(call))
  233. call->notify_rx = notify_rx;
  234. release_sock(&rx->sk);
  235. _leave(" = %p", call);
  236. return call;
  237. }
  238. EXPORT_SYMBOL(rxrpc_kernel_begin_call);
  239. /**
  240. * rxrpc_kernel_end_call - Allow a kernel service to end a call it was using
  241. * @sock: The socket the call is on
  242. * @call: The call to end
  243. *
  244. * Allow a kernel service to end a call it was using. The call must be
  245. * complete before this is called (the call should be aborted if necessary).
  246. */
  247. void rxrpc_kernel_end_call(struct socket *sock, struct rxrpc_call *call)
  248. {
  249. _enter("%d{%d}", call->debug_id, atomic_read(&call->usage));
  250. rxrpc_release_call(rxrpc_sk(sock->sk), call);
  251. rxrpc_put_call(call, rxrpc_call_put);
  252. }
  253. EXPORT_SYMBOL(rxrpc_kernel_end_call);
  254. /**
  255. * rxrpc_kernel_new_call_notification - Get notifications of new calls
  256. * @sock: The socket to intercept received messages on
  257. * @notify_new_call: Function to be called when new calls appear
  258. *
  259. * Allow a kernel service to be given notifications about new calls.
  260. */
  261. void rxrpc_kernel_new_call_notification(
  262. struct socket *sock,
  263. rxrpc_notify_new_call_t notify_new_call)
  264. {
  265. struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
  266. rx->notify_new_call = notify_new_call;
  267. }
  268. EXPORT_SYMBOL(rxrpc_kernel_new_call_notification);
  269. /*
  270. * connect an RxRPC socket
  271. * - this just targets it at a specific destination; no actual connection
  272. * negotiation takes place
  273. */
  274. static int rxrpc_connect(struct socket *sock, struct sockaddr *addr,
  275. int addr_len, int flags)
  276. {
  277. struct sockaddr_rxrpc *srx = (struct sockaddr_rxrpc *)addr;
  278. struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
  279. int ret;
  280. _enter("%p,%p,%d,%d", rx, addr, addr_len, flags);
  281. ret = rxrpc_validate_address(rx, srx, addr_len);
  282. if (ret < 0) {
  283. _leave(" = %d [bad addr]", ret);
  284. return ret;
  285. }
  286. lock_sock(&rx->sk);
  287. ret = -EISCONN;
  288. if (test_bit(RXRPC_SOCK_CONNECTED, &rx->flags))
  289. goto error;
  290. switch (rx->sk.sk_state) {
  291. case RXRPC_UNBOUND:
  292. rx->sk.sk_state = RXRPC_CLIENT_UNBOUND;
  293. case RXRPC_CLIENT_UNBOUND:
  294. case RXRPC_CLIENT_BOUND:
  295. break;
  296. default:
  297. ret = -EBUSY;
  298. goto error;
  299. }
  300. rx->connect_srx = *srx;
  301. set_bit(RXRPC_SOCK_CONNECTED, &rx->flags);
  302. ret = 0;
  303. error:
  304. release_sock(&rx->sk);
  305. return ret;
  306. }
  307. /*
  308. * send a message through an RxRPC socket
  309. * - in a client this does a number of things:
  310. * - finds/sets up a connection for the security specified (if any)
  311. * - initiates a call (ID in control data)
  312. * - ends the request phase of a call (if MSG_MORE is not set)
  313. * - sends a call data packet
  314. * - may send an abort (abort code in control data)
  315. */
  316. static int rxrpc_sendmsg(struct socket *sock, struct msghdr *m, size_t len)
  317. {
  318. struct rxrpc_local *local;
  319. struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
  320. int ret;
  321. _enter(",{%d},,%zu", rx->sk.sk_state, len);
  322. if (m->msg_flags & MSG_OOB)
  323. return -EOPNOTSUPP;
  324. if (m->msg_name) {
  325. ret = rxrpc_validate_address(rx, m->msg_name, m->msg_namelen);
  326. if (ret < 0) {
  327. _leave(" = %d [bad addr]", ret);
  328. return ret;
  329. }
  330. }
  331. lock_sock(&rx->sk);
  332. switch (rx->sk.sk_state) {
  333. case RXRPC_UNBOUND:
  334. local = rxrpc_lookup_local(&rx->srx);
  335. if (IS_ERR(local)) {
  336. ret = PTR_ERR(local);
  337. goto error_unlock;
  338. }
  339. rx->local = local;
  340. rx->sk.sk_state = RXRPC_CLIENT_UNBOUND;
  341. /* Fall through */
  342. case RXRPC_CLIENT_UNBOUND:
  343. case RXRPC_CLIENT_BOUND:
  344. if (!m->msg_name &&
  345. test_bit(RXRPC_SOCK_CONNECTED, &rx->flags)) {
  346. m->msg_name = &rx->connect_srx;
  347. m->msg_namelen = sizeof(rx->connect_srx);
  348. }
  349. case RXRPC_SERVER_BOUND:
  350. case RXRPC_SERVER_LISTENING:
  351. ret = rxrpc_do_sendmsg(rx, m, len);
  352. break;
  353. default:
  354. ret = -EINVAL;
  355. break;
  356. }
  357. error_unlock:
  358. release_sock(&rx->sk);
  359. _leave(" = %d", ret);
  360. return ret;
  361. }
  362. /*
  363. * set RxRPC socket options
  364. */
  365. static int rxrpc_setsockopt(struct socket *sock, int level, int optname,
  366. char __user *optval, unsigned int optlen)
  367. {
  368. struct rxrpc_sock *rx = rxrpc_sk(sock->sk);
  369. unsigned int min_sec_level;
  370. int ret;
  371. _enter(",%d,%d,,%d", level, optname, optlen);
  372. lock_sock(&rx->sk);
  373. ret = -EOPNOTSUPP;
  374. if (level == SOL_RXRPC) {
  375. switch (optname) {
  376. case RXRPC_EXCLUSIVE_CONNECTION:
  377. ret = -EINVAL;
  378. if (optlen != 0)
  379. goto error;
  380. ret = -EISCONN;
  381. if (rx->sk.sk_state != RXRPC_UNBOUND)
  382. goto error;
  383. rx->exclusive = true;
  384. goto success;
  385. case RXRPC_SECURITY_KEY:
  386. ret = -EINVAL;
  387. if (rx->key)
  388. goto error;
  389. ret = -EISCONN;
  390. if (rx->sk.sk_state != RXRPC_UNBOUND)
  391. goto error;
  392. ret = rxrpc_request_key(rx, optval, optlen);
  393. goto error;
  394. case RXRPC_SECURITY_KEYRING:
  395. ret = -EINVAL;
  396. if (rx->key)
  397. goto error;
  398. ret = -EISCONN;
  399. if (rx->sk.sk_state != RXRPC_UNBOUND)
  400. goto error;
  401. ret = rxrpc_server_keyring(rx, optval, optlen);
  402. goto error;
  403. case RXRPC_MIN_SECURITY_LEVEL:
  404. ret = -EINVAL;
  405. if (optlen != sizeof(unsigned int))
  406. goto error;
  407. ret = -EISCONN;
  408. if (rx->sk.sk_state != RXRPC_UNBOUND)
  409. goto error;
  410. ret = get_user(min_sec_level,
  411. (unsigned int __user *) optval);
  412. if (ret < 0)
  413. goto error;
  414. ret = -EINVAL;
  415. if (min_sec_level > RXRPC_SECURITY_MAX)
  416. goto error;
  417. rx->min_sec_level = min_sec_level;
  418. goto success;
  419. default:
  420. break;
  421. }
  422. }
  423. success:
  424. ret = 0;
  425. error:
  426. release_sock(&rx->sk);
  427. return ret;
  428. }
  429. /*
  430. * permit an RxRPC socket to be polled
  431. */
  432. static unsigned int rxrpc_poll(struct file *file, struct socket *sock,
  433. poll_table *wait)
  434. {
  435. unsigned int mask;
  436. struct sock *sk = sock->sk;
  437. sock_poll_wait(file, sk_sleep(sk), wait);
  438. mask = 0;
  439. /* the socket is readable if there are any messages waiting on the Rx
  440. * queue */
  441. if (!skb_queue_empty(&sk->sk_receive_queue))
  442. mask |= POLLIN | POLLRDNORM;
  443. /* the socket is writable if there is space to add new data to the
  444. * socket; there is no guarantee that any particular call in progress
  445. * on the socket may have space in the Tx ACK window */
  446. if (rxrpc_writable(sk))
  447. mask |= POLLOUT | POLLWRNORM;
  448. return mask;
  449. }
  450. /*
  451. * create an RxRPC socket
  452. */
  453. static int rxrpc_create(struct net *net, struct socket *sock, int protocol,
  454. int kern)
  455. {
  456. struct rxrpc_sock *rx;
  457. struct sock *sk;
  458. _enter("%p,%d", sock, protocol);
  459. if (!net_eq(net, &init_net))
  460. return -EAFNOSUPPORT;
  461. /* we support transport protocol UDP/UDP6 only */
  462. if (protocol != PF_INET)
  463. return -EPROTONOSUPPORT;
  464. if (sock->type != SOCK_DGRAM)
  465. return -ESOCKTNOSUPPORT;
  466. sock->ops = &rxrpc_rpc_ops;
  467. sock->state = SS_UNCONNECTED;
  468. sk = sk_alloc(net, PF_RXRPC, GFP_KERNEL, &rxrpc_proto, kern);
  469. if (!sk)
  470. return -ENOMEM;
  471. sock_init_data(sock, sk);
  472. sock_set_flag(sk, SOCK_RCU_FREE);
  473. sk->sk_state = RXRPC_UNBOUND;
  474. sk->sk_write_space = rxrpc_write_space;
  475. sk->sk_max_ack_backlog = 0;
  476. sk->sk_destruct = rxrpc_sock_destructor;
  477. rx = rxrpc_sk(sk);
  478. rx->family = protocol;
  479. rx->calls = RB_ROOT;
  480. INIT_LIST_HEAD(&rx->listen_link);
  481. INIT_LIST_HEAD(&rx->secureq);
  482. INIT_LIST_HEAD(&rx->acceptq);
  483. rwlock_init(&rx->call_lock);
  484. memset(&rx->srx, 0, sizeof(rx->srx));
  485. _leave(" = 0 [%p]", rx);
  486. return 0;
  487. }
  488. /*
  489. * RxRPC socket destructor
  490. */
  491. static void rxrpc_sock_destructor(struct sock *sk)
  492. {
  493. _enter("%p", sk);
  494. rxrpc_purge_queue(&sk->sk_receive_queue);
  495. WARN_ON(atomic_read(&sk->sk_wmem_alloc));
  496. WARN_ON(!sk_unhashed(sk));
  497. WARN_ON(sk->sk_socket);
  498. if (!sock_flag(sk, SOCK_DEAD)) {
  499. printk("Attempt to release alive rxrpc socket: %p\n", sk);
  500. return;
  501. }
  502. }
  503. /*
  504. * release an RxRPC socket
  505. */
  506. static int rxrpc_release_sock(struct sock *sk)
  507. {
  508. struct rxrpc_sock *rx = rxrpc_sk(sk);
  509. _enter("%p{%d,%d}", sk, sk->sk_state, atomic_read(&sk->sk_refcnt));
  510. /* declare the socket closed for business */
  511. sock_orphan(sk);
  512. sk->sk_shutdown = SHUTDOWN_MASK;
  513. spin_lock_bh(&sk->sk_receive_queue.lock);
  514. sk->sk_state = RXRPC_CLOSE;
  515. spin_unlock_bh(&sk->sk_receive_queue.lock);
  516. ASSERTCMP(rx->listen_link.next, !=, LIST_POISON1);
  517. if (!list_empty(&rx->listen_link)) {
  518. write_lock_bh(&rx->local->services_lock);
  519. list_del(&rx->listen_link);
  520. write_unlock_bh(&rx->local->services_lock);
  521. }
  522. /* try to flush out this socket */
  523. rxrpc_release_calls_on_socket(rx);
  524. flush_workqueue(rxrpc_workqueue);
  525. rxrpc_purge_queue(&sk->sk_receive_queue);
  526. rxrpc_put_local(rx->local);
  527. rx->local = NULL;
  528. key_put(rx->key);
  529. rx->key = NULL;
  530. key_put(rx->securities);
  531. rx->securities = NULL;
  532. sock_put(sk);
  533. _leave(" = 0");
  534. return 0;
  535. }
  536. /*
  537. * release an RxRPC BSD socket on close() or equivalent
  538. */
  539. static int rxrpc_release(struct socket *sock)
  540. {
  541. struct sock *sk = sock->sk;
  542. _enter("%p{%p}", sock, sk);
  543. if (!sk)
  544. return 0;
  545. sock->sk = NULL;
  546. return rxrpc_release_sock(sk);
  547. }
  548. /*
  549. * RxRPC network protocol
  550. */
  551. static const struct proto_ops rxrpc_rpc_ops = {
  552. .family = PF_RXRPC,
  553. .owner = THIS_MODULE,
  554. .release = rxrpc_release,
  555. .bind = rxrpc_bind,
  556. .connect = rxrpc_connect,
  557. .socketpair = sock_no_socketpair,
  558. .accept = sock_no_accept,
  559. .getname = sock_no_getname,
  560. .poll = rxrpc_poll,
  561. .ioctl = sock_no_ioctl,
  562. .listen = rxrpc_listen,
  563. .shutdown = sock_no_shutdown,
  564. .setsockopt = rxrpc_setsockopt,
  565. .getsockopt = sock_no_getsockopt,
  566. .sendmsg = rxrpc_sendmsg,
  567. .recvmsg = rxrpc_recvmsg,
  568. .mmap = sock_no_mmap,
  569. .sendpage = sock_no_sendpage,
  570. };
  571. static struct proto rxrpc_proto = {
  572. .name = "RXRPC",
  573. .owner = THIS_MODULE,
  574. .obj_size = sizeof(struct rxrpc_sock),
  575. .max_header = sizeof(struct rxrpc_wire_header),
  576. };
  577. static const struct net_proto_family rxrpc_family_ops = {
  578. .family = PF_RXRPC,
  579. .create = rxrpc_create,
  580. .owner = THIS_MODULE,
  581. };
  582. /*
  583. * initialise and register the RxRPC protocol
  584. */
  585. static int __init af_rxrpc_init(void)
  586. {
  587. int ret = -1;
  588. BUILD_BUG_ON(sizeof(struct rxrpc_skb_priv) > FIELD_SIZEOF(struct sk_buff, cb));
  589. get_random_bytes(&rxrpc_epoch, sizeof(rxrpc_epoch));
  590. rxrpc_epoch |= RXRPC_RANDOM_EPOCH;
  591. get_random_bytes(&rxrpc_client_conn_ids.cur,
  592. sizeof(rxrpc_client_conn_ids.cur));
  593. rxrpc_client_conn_ids.cur &= 0x3fffffff;
  594. if (rxrpc_client_conn_ids.cur == 0)
  595. rxrpc_client_conn_ids.cur = 1;
  596. ret = -ENOMEM;
  597. rxrpc_call_jar = kmem_cache_create(
  598. "rxrpc_call_jar", sizeof(struct rxrpc_call), 0,
  599. SLAB_HWCACHE_ALIGN, NULL);
  600. if (!rxrpc_call_jar) {
  601. pr_notice("Failed to allocate call jar\n");
  602. goto error_call_jar;
  603. }
  604. rxrpc_workqueue = alloc_workqueue("krxrpcd", 0, 1);
  605. if (!rxrpc_workqueue) {
  606. pr_notice("Failed to allocate work queue\n");
  607. goto error_work_queue;
  608. }
  609. ret = rxrpc_init_security();
  610. if (ret < 0) {
  611. pr_crit("Cannot initialise security\n");
  612. goto error_security;
  613. }
  614. ret = proto_register(&rxrpc_proto, 1);
  615. if (ret < 0) {
  616. pr_crit("Cannot register protocol\n");
  617. goto error_proto;
  618. }
  619. ret = sock_register(&rxrpc_family_ops);
  620. if (ret < 0) {
  621. pr_crit("Cannot register socket family\n");
  622. goto error_sock;
  623. }
  624. ret = register_key_type(&key_type_rxrpc);
  625. if (ret < 0) {
  626. pr_crit("Cannot register client key type\n");
  627. goto error_key_type;
  628. }
  629. ret = register_key_type(&key_type_rxrpc_s);
  630. if (ret < 0) {
  631. pr_crit("Cannot register server key type\n");
  632. goto error_key_type_s;
  633. }
  634. ret = rxrpc_sysctl_init();
  635. if (ret < 0) {
  636. pr_crit("Cannot register sysctls\n");
  637. goto error_sysctls;
  638. }
  639. #ifdef CONFIG_PROC_FS
  640. proc_create("rxrpc_calls", 0, init_net.proc_net, &rxrpc_call_seq_fops);
  641. proc_create("rxrpc_conns", 0, init_net.proc_net,
  642. &rxrpc_connection_seq_fops);
  643. #endif
  644. return 0;
  645. error_sysctls:
  646. unregister_key_type(&key_type_rxrpc_s);
  647. error_key_type_s:
  648. unregister_key_type(&key_type_rxrpc);
  649. error_key_type:
  650. sock_unregister(PF_RXRPC);
  651. error_sock:
  652. proto_unregister(&rxrpc_proto);
  653. error_proto:
  654. rxrpc_exit_security();
  655. error_security:
  656. destroy_workqueue(rxrpc_workqueue);
  657. error_work_queue:
  658. kmem_cache_destroy(rxrpc_call_jar);
  659. error_call_jar:
  660. return ret;
  661. }
  662. /*
  663. * unregister the RxRPC protocol
  664. */
  665. static void __exit af_rxrpc_exit(void)
  666. {
  667. _enter("");
  668. rxrpc_sysctl_exit();
  669. unregister_key_type(&key_type_rxrpc_s);
  670. unregister_key_type(&key_type_rxrpc);
  671. sock_unregister(PF_RXRPC);
  672. proto_unregister(&rxrpc_proto);
  673. rxrpc_destroy_all_calls();
  674. rxrpc_destroy_all_connections();
  675. ASSERTCMP(atomic_read(&rxrpc_n_skbs), ==, 0);
  676. rxrpc_destroy_all_locals();
  677. remove_proc_entry("rxrpc_conns", init_net.proc_net);
  678. remove_proc_entry("rxrpc_calls", init_net.proc_net);
  679. destroy_workqueue(rxrpc_workqueue);
  680. rxrpc_exit_security();
  681. kmem_cache_destroy(rxrpc_call_jar);
  682. _leave("");
  683. }
  684. module_init(af_rxrpc_init);
  685. module_exit(af_rxrpc_exit);