tcp.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720
  1. /*
  2. * Copyright (c) 2006, 2018 Oracle and/or its affiliates. All rights reserved.
  3. *
  4. * This software is available to you under a choice of one of two
  5. * licenses. You may choose to be licensed under the terms of the GNU
  6. * General Public License (GPL) Version 2, available from the file
  7. * COPYING in the main directory of this source tree, or the
  8. * OpenIB.org BSD license below:
  9. *
  10. * Redistribution and use in source and binary forms, with or
  11. * without modification, are permitted provided that the following
  12. * conditions are met:
  13. *
  14. * - Redistributions of source code must retain the above
  15. * copyright notice, this list of conditions and the following
  16. * disclaimer.
  17. *
  18. * - Redistributions in binary form must reproduce the above
  19. * copyright notice, this list of conditions and the following
  20. * disclaimer in the documentation and/or other materials
  21. * provided with the distribution.
  22. *
  23. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  24. * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  25. * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  26. * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  27. * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  28. * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  29. * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  30. * SOFTWARE.
  31. *
  32. */
  33. #include <linux/kernel.h>
  34. #include <linux/slab.h>
  35. #include <linux/in.h>
  36. #include <linux/module.h>
  37. #include <net/tcp.h>
  38. #include <net/net_namespace.h>
  39. #include <net/netns/generic.h>
  40. #include <net/tcp.h>
  41. #include <net/addrconf.h>
  42. #include "rds.h"
  43. #include "tcp.h"
  44. /* only for info exporting */
  45. static DEFINE_SPINLOCK(rds_tcp_tc_list_lock);
  46. static LIST_HEAD(rds_tcp_tc_list);
  47. /* rds_tcp_tc_count counts only IPv4 connections.
  48. * rds6_tcp_tc_count counts both IPv4 and IPv6 connections.
  49. */
  50. static unsigned int rds_tcp_tc_count;
  51. static unsigned int rds6_tcp_tc_count;
  52. /* Track rds_tcp_connection structs so they can be cleaned up */
  53. static DEFINE_SPINLOCK(rds_tcp_conn_lock);
  54. static LIST_HEAD(rds_tcp_conn_list);
  55. static atomic_t rds_tcp_unloading = ATOMIC_INIT(0);
  56. static struct kmem_cache *rds_tcp_conn_slab;
  57. static int rds_tcp_skbuf_handler(struct ctl_table *ctl, int write,
  58. void __user *buffer, size_t *lenp,
  59. loff_t *fpos);
  60. static int rds_tcp_min_sndbuf = SOCK_MIN_SNDBUF;
  61. static int rds_tcp_min_rcvbuf = SOCK_MIN_RCVBUF;
  62. static struct ctl_table rds_tcp_sysctl_table[] = {
  63. #define RDS_TCP_SNDBUF 0
  64. {
  65. .procname = "rds_tcp_sndbuf",
  66. /* data is per-net pointer */
  67. .maxlen = sizeof(int),
  68. .mode = 0644,
  69. .proc_handler = rds_tcp_skbuf_handler,
  70. .extra1 = &rds_tcp_min_sndbuf,
  71. },
  72. #define RDS_TCP_RCVBUF 1
  73. {
  74. .procname = "rds_tcp_rcvbuf",
  75. /* data is per-net pointer */
  76. .maxlen = sizeof(int),
  77. .mode = 0644,
  78. .proc_handler = rds_tcp_skbuf_handler,
  79. .extra1 = &rds_tcp_min_rcvbuf,
  80. },
  81. { }
  82. };
  83. /* doing it this way avoids calling tcp_sk() */
  84. void rds_tcp_nonagle(struct socket *sock)
  85. {
  86. int val = 1;
  87. kernel_setsockopt(sock, SOL_TCP, TCP_NODELAY, (void *)&val,
  88. sizeof(val));
  89. }
  90. u32 rds_tcp_write_seq(struct rds_tcp_connection *tc)
  91. {
  92. /* seq# of the last byte of data in tcp send buffer */
  93. return tcp_sk(tc->t_sock->sk)->write_seq;
  94. }
  95. u32 rds_tcp_snd_una(struct rds_tcp_connection *tc)
  96. {
  97. return tcp_sk(tc->t_sock->sk)->snd_una;
  98. }
  99. void rds_tcp_restore_callbacks(struct socket *sock,
  100. struct rds_tcp_connection *tc)
  101. {
  102. rdsdebug("restoring sock %p callbacks from tc %p\n", sock, tc);
  103. write_lock_bh(&sock->sk->sk_callback_lock);
  104. /* done under the callback_lock to serialize with write_space */
  105. spin_lock(&rds_tcp_tc_list_lock);
  106. list_del_init(&tc->t_list_item);
  107. rds6_tcp_tc_count--;
  108. if (!tc->t_cpath->cp_conn->c_isv6)
  109. rds_tcp_tc_count--;
  110. spin_unlock(&rds_tcp_tc_list_lock);
  111. tc->t_sock = NULL;
  112. sock->sk->sk_write_space = tc->t_orig_write_space;
  113. sock->sk->sk_data_ready = tc->t_orig_data_ready;
  114. sock->sk->sk_state_change = tc->t_orig_state_change;
  115. sock->sk->sk_user_data = NULL;
  116. write_unlock_bh(&sock->sk->sk_callback_lock);
  117. }
  118. /*
  119. * rds_tcp_reset_callbacks() switches the to the new sock and
  120. * returns the existing tc->t_sock.
  121. *
  122. * The only functions that set tc->t_sock are rds_tcp_set_callbacks
  123. * and rds_tcp_reset_callbacks. Send and receive trust that
  124. * it is set. The absence of RDS_CONN_UP bit protects those paths
  125. * from being called while it isn't set.
  126. */
  127. void rds_tcp_reset_callbacks(struct socket *sock,
  128. struct rds_conn_path *cp)
  129. {
  130. struct rds_tcp_connection *tc = cp->cp_transport_data;
  131. struct socket *osock = tc->t_sock;
  132. if (!osock)
  133. goto newsock;
  134. /* Need to resolve a duelling SYN between peers.
  135. * We have an outstanding SYN to this peer, which may
  136. * potentially have transitioned to the RDS_CONN_UP state,
  137. * so we must quiesce any send threads before resetting
  138. * cp_transport_data. We quiesce these threads by setting
  139. * cp_state to something other than RDS_CONN_UP, and then
  140. * waiting for any existing threads in rds_send_xmit to
  141. * complete release_in_xmit(). (Subsequent threads entering
  142. * rds_send_xmit() will bail on !rds_conn_up().
  143. *
  144. * However an incoming syn-ack at this point would end up
  145. * marking the conn as RDS_CONN_UP, and would again permit
  146. * rds_send_xmi() threads through, so ideally we would
  147. * synchronize on RDS_CONN_UP after lock_sock(), but cannot
  148. * do that: waiting on !RDS_IN_XMIT after lock_sock() may
  149. * end up deadlocking with tcp_sendmsg(), and the RDS_IN_XMIT
  150. * would not get set. As a result, we set c_state to
  151. * RDS_CONN_RESETTTING, to ensure that rds_tcp_state_change
  152. * cannot mark rds_conn_path_up() in the window before lock_sock()
  153. */
  154. atomic_set(&cp->cp_state, RDS_CONN_RESETTING);
  155. wait_event(cp->cp_waitq, !test_bit(RDS_IN_XMIT, &cp->cp_flags));
  156. lock_sock(osock->sk);
  157. /* reset receive side state for rds_tcp_data_recv() for osock */
  158. cancel_delayed_work_sync(&cp->cp_send_w);
  159. cancel_delayed_work_sync(&cp->cp_recv_w);
  160. if (tc->t_tinc) {
  161. rds_inc_put(&tc->t_tinc->ti_inc);
  162. tc->t_tinc = NULL;
  163. }
  164. tc->t_tinc_hdr_rem = sizeof(struct rds_header);
  165. tc->t_tinc_data_rem = 0;
  166. rds_tcp_restore_callbacks(osock, tc);
  167. release_sock(osock->sk);
  168. sock_release(osock);
  169. newsock:
  170. rds_send_path_reset(cp);
  171. lock_sock(sock->sk);
  172. rds_tcp_set_callbacks(sock, cp);
  173. release_sock(sock->sk);
  174. }
  175. /* Add tc to rds_tcp_tc_list and set tc->t_sock. See comments
  176. * above rds_tcp_reset_callbacks for notes about synchronization
  177. * with data path
  178. */
  179. void rds_tcp_set_callbacks(struct socket *sock, struct rds_conn_path *cp)
  180. {
  181. struct rds_tcp_connection *tc = cp->cp_transport_data;
  182. rdsdebug("setting sock %p callbacks to tc %p\n", sock, tc);
  183. write_lock_bh(&sock->sk->sk_callback_lock);
  184. /* done under the callback_lock to serialize with write_space */
  185. spin_lock(&rds_tcp_tc_list_lock);
  186. list_add_tail(&tc->t_list_item, &rds_tcp_tc_list);
  187. rds6_tcp_tc_count++;
  188. if (!tc->t_cpath->cp_conn->c_isv6)
  189. rds_tcp_tc_count++;
  190. spin_unlock(&rds_tcp_tc_list_lock);
  191. /* accepted sockets need our listen data ready undone */
  192. if (sock->sk->sk_data_ready == rds_tcp_listen_data_ready)
  193. sock->sk->sk_data_ready = sock->sk->sk_user_data;
  194. tc->t_sock = sock;
  195. tc->t_cpath = cp;
  196. tc->t_orig_data_ready = sock->sk->sk_data_ready;
  197. tc->t_orig_write_space = sock->sk->sk_write_space;
  198. tc->t_orig_state_change = sock->sk->sk_state_change;
  199. sock->sk->sk_user_data = cp;
  200. sock->sk->sk_data_ready = rds_tcp_data_ready;
  201. sock->sk->sk_write_space = rds_tcp_write_space;
  202. sock->sk->sk_state_change = rds_tcp_state_change;
  203. write_unlock_bh(&sock->sk->sk_callback_lock);
  204. }
  205. /* Handle RDS_INFO_TCP_SOCKETS socket option. It only returns IPv4
  206. * connections for backward compatibility.
  207. */
  208. static void rds_tcp_tc_info(struct socket *rds_sock, unsigned int len,
  209. struct rds_info_iterator *iter,
  210. struct rds_info_lengths *lens)
  211. {
  212. struct rds_info_tcp_socket tsinfo;
  213. struct rds_tcp_connection *tc;
  214. unsigned long flags;
  215. spin_lock_irqsave(&rds_tcp_tc_list_lock, flags);
  216. if (len / sizeof(tsinfo) < rds_tcp_tc_count)
  217. goto out;
  218. list_for_each_entry(tc, &rds_tcp_tc_list, t_list_item) {
  219. struct inet_sock *inet = inet_sk(tc->t_sock->sk);
  220. if (tc->t_cpath->cp_conn->c_isv6)
  221. continue;
  222. tsinfo.local_addr = inet->inet_saddr;
  223. tsinfo.local_port = inet->inet_sport;
  224. tsinfo.peer_addr = inet->inet_daddr;
  225. tsinfo.peer_port = inet->inet_dport;
  226. tsinfo.hdr_rem = tc->t_tinc_hdr_rem;
  227. tsinfo.data_rem = tc->t_tinc_data_rem;
  228. tsinfo.last_sent_nxt = tc->t_last_sent_nxt;
  229. tsinfo.last_expected_una = tc->t_last_expected_una;
  230. tsinfo.last_seen_una = tc->t_last_seen_una;
  231. rds_info_copy(iter, &tsinfo, sizeof(tsinfo));
  232. }
  233. out:
  234. lens->nr = rds_tcp_tc_count;
  235. lens->each = sizeof(tsinfo);
  236. spin_unlock_irqrestore(&rds_tcp_tc_list_lock, flags);
  237. }
  238. /* Handle RDS6_INFO_TCP_SOCKETS socket option. It returns both IPv4 and
  239. * IPv6 connections. IPv4 connection address is returned in an IPv4 mapped
  240. * address.
  241. */
  242. static void rds6_tcp_tc_info(struct socket *sock, unsigned int len,
  243. struct rds_info_iterator *iter,
  244. struct rds_info_lengths *lens)
  245. {
  246. struct rds6_info_tcp_socket tsinfo6;
  247. struct rds_tcp_connection *tc;
  248. unsigned long flags;
  249. spin_lock_irqsave(&rds_tcp_tc_list_lock, flags);
  250. if (len / sizeof(tsinfo6) < rds6_tcp_tc_count)
  251. goto out;
  252. list_for_each_entry(tc, &rds_tcp_tc_list, t_list_item) {
  253. struct sock *sk = tc->t_sock->sk;
  254. struct inet_sock *inet = inet_sk(sk);
  255. tsinfo6.local_addr = sk->sk_v6_rcv_saddr;
  256. tsinfo6.local_port = inet->inet_sport;
  257. tsinfo6.peer_addr = sk->sk_v6_daddr;
  258. tsinfo6.peer_port = inet->inet_dport;
  259. tsinfo6.hdr_rem = tc->t_tinc_hdr_rem;
  260. tsinfo6.data_rem = tc->t_tinc_data_rem;
  261. tsinfo6.last_sent_nxt = tc->t_last_sent_nxt;
  262. tsinfo6.last_expected_una = tc->t_last_expected_una;
  263. tsinfo6.last_seen_una = tc->t_last_seen_una;
  264. rds_info_copy(iter, &tsinfo6, sizeof(tsinfo6));
  265. }
  266. out:
  267. lens->nr = rds6_tcp_tc_count;
  268. lens->each = sizeof(tsinfo6);
  269. spin_unlock_irqrestore(&rds_tcp_tc_list_lock, flags);
  270. }
  271. static int rds_tcp_laddr_check(struct net *net, const struct in6_addr *addr,
  272. __u32 scope_id)
  273. {
  274. struct net_device *dev = NULL;
  275. int ret;
  276. if (ipv6_addr_v4mapped(addr)) {
  277. if (inet_addr_type(net, addr->s6_addr32[3]) == RTN_LOCAL)
  278. return 0;
  279. return -EADDRNOTAVAIL;
  280. }
  281. /* If the scope_id is specified, check only those addresses
  282. * hosted on the specified interface.
  283. */
  284. if (scope_id != 0) {
  285. rcu_read_lock();
  286. dev = dev_get_by_index_rcu(net, scope_id);
  287. /* scope_id is not valid... */
  288. if (!dev) {
  289. rcu_read_unlock();
  290. return -EADDRNOTAVAIL;
  291. }
  292. rcu_read_unlock();
  293. }
  294. ret = ipv6_chk_addr(net, addr, dev, 0);
  295. if (ret)
  296. return 0;
  297. return -EADDRNOTAVAIL;
  298. }
  299. static void rds_tcp_conn_free(void *arg)
  300. {
  301. struct rds_tcp_connection *tc = arg;
  302. unsigned long flags;
  303. rdsdebug("freeing tc %p\n", tc);
  304. spin_lock_irqsave(&rds_tcp_conn_lock, flags);
  305. if (!tc->t_tcp_node_detached)
  306. list_del(&tc->t_tcp_node);
  307. spin_unlock_irqrestore(&rds_tcp_conn_lock, flags);
  308. kmem_cache_free(rds_tcp_conn_slab, tc);
  309. }
  310. static int rds_tcp_conn_alloc(struct rds_connection *conn, gfp_t gfp)
  311. {
  312. struct rds_tcp_connection *tc;
  313. int i, j;
  314. int ret = 0;
  315. for (i = 0; i < RDS_MPATH_WORKERS; i++) {
  316. tc = kmem_cache_alloc(rds_tcp_conn_slab, gfp);
  317. if (!tc) {
  318. ret = -ENOMEM;
  319. goto fail;
  320. }
  321. mutex_init(&tc->t_conn_path_lock);
  322. tc->t_sock = NULL;
  323. tc->t_tinc = NULL;
  324. tc->t_tinc_hdr_rem = sizeof(struct rds_header);
  325. tc->t_tinc_data_rem = 0;
  326. conn->c_path[i].cp_transport_data = tc;
  327. tc->t_cpath = &conn->c_path[i];
  328. tc->t_tcp_node_detached = true;
  329. rdsdebug("rds_conn_path [%d] tc %p\n", i,
  330. conn->c_path[i].cp_transport_data);
  331. }
  332. spin_lock_irq(&rds_tcp_conn_lock);
  333. for (i = 0; i < RDS_MPATH_WORKERS; i++) {
  334. tc = conn->c_path[i].cp_transport_data;
  335. tc->t_tcp_node_detached = false;
  336. list_add_tail(&tc->t_tcp_node, &rds_tcp_conn_list);
  337. }
  338. spin_unlock_irq(&rds_tcp_conn_lock);
  339. fail:
  340. if (ret) {
  341. for (j = 0; j < i; j++)
  342. rds_tcp_conn_free(conn->c_path[j].cp_transport_data);
  343. }
  344. return ret;
  345. }
  346. static bool list_has_conn(struct list_head *list, struct rds_connection *conn)
  347. {
  348. struct rds_tcp_connection *tc, *_tc;
  349. list_for_each_entry_safe(tc, _tc, list, t_tcp_node) {
  350. if (tc->t_cpath->cp_conn == conn)
  351. return true;
  352. }
  353. return false;
  354. }
  355. static void rds_tcp_set_unloading(void)
  356. {
  357. atomic_set(&rds_tcp_unloading, 1);
  358. }
  359. static bool rds_tcp_is_unloading(struct rds_connection *conn)
  360. {
  361. return atomic_read(&rds_tcp_unloading) != 0;
  362. }
  363. static void rds_tcp_destroy_conns(void)
  364. {
  365. struct rds_tcp_connection *tc, *_tc;
  366. LIST_HEAD(tmp_list);
  367. /* avoid calling conn_destroy with irqs off */
  368. spin_lock_irq(&rds_tcp_conn_lock);
  369. list_for_each_entry_safe(tc, _tc, &rds_tcp_conn_list, t_tcp_node) {
  370. if (!list_has_conn(&tmp_list, tc->t_cpath->cp_conn))
  371. list_move_tail(&tc->t_tcp_node, &tmp_list);
  372. }
  373. spin_unlock_irq(&rds_tcp_conn_lock);
  374. list_for_each_entry_safe(tc, _tc, &tmp_list, t_tcp_node)
  375. rds_conn_destroy(tc->t_cpath->cp_conn);
  376. }
  377. static void rds_tcp_exit(void);
  378. struct rds_transport rds_tcp_transport = {
  379. .laddr_check = rds_tcp_laddr_check,
  380. .xmit_path_prepare = rds_tcp_xmit_path_prepare,
  381. .xmit_path_complete = rds_tcp_xmit_path_complete,
  382. .xmit = rds_tcp_xmit,
  383. .recv_path = rds_tcp_recv_path,
  384. .conn_alloc = rds_tcp_conn_alloc,
  385. .conn_free = rds_tcp_conn_free,
  386. .conn_path_connect = rds_tcp_conn_path_connect,
  387. .conn_path_shutdown = rds_tcp_conn_path_shutdown,
  388. .inc_copy_to_user = rds_tcp_inc_copy_to_user,
  389. .inc_free = rds_tcp_inc_free,
  390. .stats_info_copy = rds_tcp_stats_info_copy,
  391. .exit = rds_tcp_exit,
  392. .t_owner = THIS_MODULE,
  393. .t_name = "tcp",
  394. .t_type = RDS_TRANS_TCP,
  395. .t_prefer_loopback = 1,
  396. .t_mp_capable = 1,
  397. .t_unloading = rds_tcp_is_unloading,
  398. };
  399. static unsigned int rds_tcp_netid;
  400. /* per-network namespace private data for this module */
  401. struct rds_tcp_net {
  402. struct socket *rds_tcp_listen_sock;
  403. struct work_struct rds_tcp_accept_w;
  404. struct ctl_table_header *rds_tcp_sysctl;
  405. struct ctl_table *ctl_table;
  406. int sndbuf_size;
  407. int rcvbuf_size;
  408. };
  409. /* All module specific customizations to the RDS-TCP socket should be done in
  410. * rds_tcp_tune() and applied after socket creation.
  411. */
  412. void rds_tcp_tune(struct socket *sock)
  413. {
  414. struct sock *sk = sock->sk;
  415. struct net *net = sock_net(sk);
  416. struct rds_tcp_net *rtn = net_generic(net, rds_tcp_netid);
  417. rds_tcp_nonagle(sock);
  418. lock_sock(sk);
  419. if (rtn->sndbuf_size > 0) {
  420. sk->sk_sndbuf = rtn->sndbuf_size;
  421. sk->sk_userlocks |= SOCK_SNDBUF_LOCK;
  422. }
  423. if (rtn->rcvbuf_size > 0) {
  424. sk->sk_sndbuf = rtn->rcvbuf_size;
  425. sk->sk_userlocks |= SOCK_RCVBUF_LOCK;
  426. }
  427. release_sock(sk);
  428. }
  429. static void rds_tcp_accept_worker(struct work_struct *work)
  430. {
  431. struct rds_tcp_net *rtn = container_of(work,
  432. struct rds_tcp_net,
  433. rds_tcp_accept_w);
  434. while (rds_tcp_accept_one(rtn->rds_tcp_listen_sock) == 0)
  435. cond_resched();
  436. }
  437. void rds_tcp_accept_work(struct sock *sk)
  438. {
  439. struct net *net = sock_net(sk);
  440. struct rds_tcp_net *rtn = net_generic(net, rds_tcp_netid);
  441. queue_work(rds_wq, &rtn->rds_tcp_accept_w);
  442. }
  443. static __net_init int rds_tcp_init_net(struct net *net)
  444. {
  445. struct rds_tcp_net *rtn = net_generic(net, rds_tcp_netid);
  446. struct ctl_table *tbl;
  447. int err = 0;
  448. memset(rtn, 0, sizeof(*rtn));
  449. /* {snd, rcv}buf_size default to 0, which implies we let the
  450. * stack pick the value, and permit auto-tuning of buffer size.
  451. */
  452. if (net == &init_net) {
  453. tbl = rds_tcp_sysctl_table;
  454. } else {
  455. tbl = kmemdup(rds_tcp_sysctl_table,
  456. sizeof(rds_tcp_sysctl_table), GFP_KERNEL);
  457. if (!tbl) {
  458. pr_warn("could not set allocate syctl table\n");
  459. return -ENOMEM;
  460. }
  461. rtn->ctl_table = tbl;
  462. }
  463. tbl[RDS_TCP_SNDBUF].data = &rtn->sndbuf_size;
  464. tbl[RDS_TCP_RCVBUF].data = &rtn->rcvbuf_size;
  465. rtn->rds_tcp_sysctl = register_net_sysctl(net, "net/rds/tcp", tbl);
  466. if (!rtn->rds_tcp_sysctl) {
  467. pr_warn("could not register sysctl\n");
  468. err = -ENOMEM;
  469. goto fail;
  470. }
  471. rtn->rds_tcp_listen_sock = rds_tcp_listen_init(net, true);
  472. if (!rtn->rds_tcp_listen_sock) {
  473. pr_warn("could not set up IPv6 listen sock\n");
  474. /* Try IPv4 as some systems disable IPv6 */
  475. rtn->rds_tcp_listen_sock = rds_tcp_listen_init(net, false);
  476. if (!rtn->rds_tcp_listen_sock) {
  477. unregister_net_sysctl_table(rtn->rds_tcp_sysctl);
  478. rtn->rds_tcp_sysctl = NULL;
  479. err = -EAFNOSUPPORT;
  480. goto fail;
  481. }
  482. }
  483. INIT_WORK(&rtn->rds_tcp_accept_w, rds_tcp_accept_worker);
  484. return 0;
  485. fail:
  486. if (net != &init_net)
  487. kfree(tbl);
  488. return err;
  489. }
  490. static void rds_tcp_kill_sock(struct net *net)
  491. {
  492. struct rds_tcp_connection *tc, *_tc;
  493. LIST_HEAD(tmp_list);
  494. struct rds_tcp_net *rtn = net_generic(net, rds_tcp_netid);
  495. struct socket *lsock = rtn->rds_tcp_listen_sock;
  496. rtn->rds_tcp_listen_sock = NULL;
  497. rds_tcp_listen_stop(lsock, &rtn->rds_tcp_accept_w);
  498. spin_lock_irq(&rds_tcp_conn_lock);
  499. list_for_each_entry_safe(tc, _tc, &rds_tcp_conn_list, t_tcp_node) {
  500. struct net *c_net = read_pnet(&tc->t_cpath->cp_conn->c_net);
  501. if (net != c_net || !tc->t_sock)
  502. continue;
  503. if (!list_has_conn(&tmp_list, tc->t_cpath->cp_conn)) {
  504. list_move_tail(&tc->t_tcp_node, &tmp_list);
  505. } else {
  506. list_del(&tc->t_tcp_node);
  507. tc->t_tcp_node_detached = true;
  508. }
  509. }
  510. spin_unlock_irq(&rds_tcp_conn_lock);
  511. list_for_each_entry_safe(tc, _tc, &tmp_list, t_tcp_node)
  512. rds_conn_destroy(tc->t_cpath->cp_conn);
  513. }
  514. static void __net_exit rds_tcp_exit_net(struct net *net)
  515. {
  516. struct rds_tcp_net *rtn = net_generic(net, rds_tcp_netid);
  517. rds_tcp_kill_sock(net);
  518. if (rtn->rds_tcp_sysctl)
  519. unregister_net_sysctl_table(rtn->rds_tcp_sysctl);
  520. if (net != &init_net && rtn->ctl_table)
  521. kfree(rtn->ctl_table);
  522. }
  523. static struct pernet_operations rds_tcp_net_ops = {
  524. .init = rds_tcp_init_net,
  525. .exit = rds_tcp_exit_net,
  526. .id = &rds_tcp_netid,
  527. .size = sizeof(struct rds_tcp_net),
  528. };
  529. void *rds_tcp_listen_sock_def_readable(struct net *net)
  530. {
  531. struct rds_tcp_net *rtn = net_generic(net, rds_tcp_netid);
  532. struct socket *lsock = rtn->rds_tcp_listen_sock;
  533. if (!lsock)
  534. return NULL;
  535. return lsock->sk->sk_user_data;
  536. }
  537. /* when sysctl is used to modify some kernel socket parameters,this
  538. * function resets the RDS connections in that netns so that we can
  539. * restart with new parameters. The assumption is that such reset
  540. * events are few and far-between.
  541. */
  542. static void rds_tcp_sysctl_reset(struct net *net)
  543. {
  544. struct rds_tcp_connection *tc, *_tc;
  545. spin_lock_irq(&rds_tcp_conn_lock);
  546. list_for_each_entry_safe(tc, _tc, &rds_tcp_conn_list, t_tcp_node) {
  547. struct net *c_net = read_pnet(&tc->t_cpath->cp_conn->c_net);
  548. if (net != c_net || !tc->t_sock)
  549. continue;
  550. /* reconnect with new parameters */
  551. rds_conn_path_drop(tc->t_cpath, false);
  552. }
  553. spin_unlock_irq(&rds_tcp_conn_lock);
  554. }
  555. static int rds_tcp_skbuf_handler(struct ctl_table *ctl, int write,
  556. void __user *buffer, size_t *lenp,
  557. loff_t *fpos)
  558. {
  559. struct net *net = current->nsproxy->net_ns;
  560. int err;
  561. err = proc_dointvec_minmax(ctl, write, buffer, lenp, fpos);
  562. if (err < 0) {
  563. pr_warn("Invalid input. Must be >= %d\n",
  564. *(int *)(ctl->extra1));
  565. return err;
  566. }
  567. if (write)
  568. rds_tcp_sysctl_reset(net);
  569. return 0;
  570. }
  571. static void rds_tcp_exit(void)
  572. {
  573. rds_tcp_set_unloading();
  574. synchronize_rcu();
  575. rds_info_deregister_func(RDS_INFO_TCP_SOCKETS, rds_tcp_tc_info);
  576. rds_info_deregister_func(RDS6_INFO_TCP_SOCKETS, rds6_tcp_tc_info);
  577. unregister_pernet_device(&rds_tcp_net_ops);
  578. rds_tcp_destroy_conns();
  579. rds_trans_unregister(&rds_tcp_transport);
  580. rds_tcp_recv_exit();
  581. kmem_cache_destroy(rds_tcp_conn_slab);
  582. }
  583. module_exit(rds_tcp_exit);
  584. static int rds_tcp_init(void)
  585. {
  586. int ret;
  587. rds_tcp_conn_slab = kmem_cache_create("rds_tcp_connection",
  588. sizeof(struct rds_tcp_connection),
  589. 0, 0, NULL);
  590. if (!rds_tcp_conn_slab) {
  591. ret = -ENOMEM;
  592. goto out;
  593. }
  594. ret = rds_tcp_recv_init();
  595. if (ret)
  596. goto out_slab;
  597. ret = register_pernet_device(&rds_tcp_net_ops);
  598. if (ret)
  599. goto out_recv;
  600. rds_trans_register(&rds_tcp_transport);
  601. rds_info_register_func(RDS_INFO_TCP_SOCKETS, rds_tcp_tc_info);
  602. rds_info_register_func(RDS6_INFO_TCP_SOCKETS, rds6_tcp_tc_info);
  603. goto out;
  604. out_recv:
  605. rds_tcp_recv_exit();
  606. out_slab:
  607. kmem_cache_destroy(rds_tcp_conn_slab);
  608. out:
  609. return ret;
  610. }
  611. module_init(rds_tcp_init);
  612. MODULE_AUTHOR("Oracle Corporation <rds-devel@oss.oracle.com>");
  613. MODULE_DESCRIPTION("RDS: TCP transport");
  614. MODULE_LICENSE("Dual BSD/GPL");