recv.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731
  1. /*
  2. * Copyright (c) 2006 Oracle. 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 <net/sock.h>
  36. #include <linux/in.h>
  37. #include <linux/export.h>
  38. #include <linux/time.h>
  39. #include <linux/rds.h>
  40. #include "rds.h"
  41. void rds_inc_init(struct rds_incoming *inc, struct rds_connection *conn,
  42. __be32 saddr)
  43. {
  44. int i;
  45. atomic_set(&inc->i_refcount, 1);
  46. INIT_LIST_HEAD(&inc->i_item);
  47. inc->i_conn = conn;
  48. inc->i_saddr = saddr;
  49. inc->i_rdma_cookie = 0;
  50. inc->i_rx_tstamp.tv_sec = 0;
  51. inc->i_rx_tstamp.tv_usec = 0;
  52. for (i = 0; i < RDS_RX_MAX_TRACES; i++)
  53. inc->i_rx_lat_trace[i] = 0;
  54. }
  55. EXPORT_SYMBOL_GPL(rds_inc_init);
  56. void rds_inc_path_init(struct rds_incoming *inc, struct rds_conn_path *cp,
  57. __be32 saddr)
  58. {
  59. atomic_set(&inc->i_refcount, 1);
  60. INIT_LIST_HEAD(&inc->i_item);
  61. inc->i_conn = cp->cp_conn;
  62. inc->i_conn_path = cp;
  63. inc->i_saddr = saddr;
  64. inc->i_rdma_cookie = 0;
  65. inc->i_rx_tstamp.tv_sec = 0;
  66. inc->i_rx_tstamp.tv_usec = 0;
  67. }
  68. EXPORT_SYMBOL_GPL(rds_inc_path_init);
  69. static void rds_inc_addref(struct rds_incoming *inc)
  70. {
  71. rdsdebug("addref inc %p ref %d\n", inc, atomic_read(&inc->i_refcount));
  72. atomic_inc(&inc->i_refcount);
  73. }
  74. void rds_inc_put(struct rds_incoming *inc)
  75. {
  76. rdsdebug("put inc %p ref %d\n", inc, atomic_read(&inc->i_refcount));
  77. if (atomic_dec_and_test(&inc->i_refcount)) {
  78. BUG_ON(!list_empty(&inc->i_item));
  79. inc->i_conn->c_trans->inc_free(inc);
  80. }
  81. }
  82. EXPORT_SYMBOL_GPL(rds_inc_put);
  83. static void rds_recv_rcvbuf_delta(struct rds_sock *rs, struct sock *sk,
  84. struct rds_cong_map *map,
  85. int delta, __be16 port)
  86. {
  87. int now_congested;
  88. if (delta == 0)
  89. return;
  90. rs->rs_rcv_bytes += delta;
  91. if (delta > 0)
  92. rds_stats_add(s_recv_bytes_added_to_socket, delta);
  93. else
  94. rds_stats_add(s_recv_bytes_removed_from_socket, -delta);
  95. now_congested = rs->rs_rcv_bytes > rds_sk_rcvbuf(rs);
  96. rdsdebug("rs %p (%pI4:%u) recv bytes %d buf %d "
  97. "now_cong %d delta %d\n",
  98. rs, &rs->rs_bound_addr,
  99. ntohs(rs->rs_bound_port), rs->rs_rcv_bytes,
  100. rds_sk_rcvbuf(rs), now_congested, delta);
  101. /* wasn't -> am congested */
  102. if (!rs->rs_congested && now_congested) {
  103. rs->rs_congested = 1;
  104. rds_cong_set_bit(map, port);
  105. rds_cong_queue_updates(map);
  106. }
  107. /* was -> aren't congested */
  108. /* Require more free space before reporting uncongested to prevent
  109. bouncing cong/uncong state too often */
  110. else if (rs->rs_congested && (rs->rs_rcv_bytes < (rds_sk_rcvbuf(rs)/2))) {
  111. rs->rs_congested = 0;
  112. rds_cong_clear_bit(map, port);
  113. rds_cong_queue_updates(map);
  114. }
  115. /* do nothing if no change in cong state */
  116. }
  117. static void rds_conn_peer_gen_update(struct rds_connection *conn,
  118. u32 peer_gen_num)
  119. {
  120. int i;
  121. struct rds_message *rm, *tmp;
  122. unsigned long flags;
  123. WARN_ON(conn->c_trans->t_type != RDS_TRANS_TCP);
  124. if (peer_gen_num != 0) {
  125. if (conn->c_peer_gen_num != 0 &&
  126. peer_gen_num != conn->c_peer_gen_num) {
  127. for (i = 0; i < RDS_MPATH_WORKERS; i++) {
  128. struct rds_conn_path *cp;
  129. cp = &conn->c_path[i];
  130. spin_lock_irqsave(&cp->cp_lock, flags);
  131. cp->cp_next_tx_seq = 1;
  132. cp->cp_next_rx_seq = 0;
  133. list_for_each_entry_safe(rm, tmp,
  134. &cp->cp_retrans,
  135. m_conn_item) {
  136. set_bit(RDS_MSG_FLUSH, &rm->m_flags);
  137. }
  138. spin_unlock_irqrestore(&cp->cp_lock, flags);
  139. }
  140. }
  141. conn->c_peer_gen_num = peer_gen_num;
  142. }
  143. }
  144. /*
  145. * Process all extension headers that come with this message.
  146. */
  147. static void rds_recv_incoming_exthdrs(struct rds_incoming *inc, struct rds_sock *rs)
  148. {
  149. struct rds_header *hdr = &inc->i_hdr;
  150. unsigned int pos = 0, type, len;
  151. union {
  152. struct rds_ext_header_version version;
  153. struct rds_ext_header_rdma rdma;
  154. struct rds_ext_header_rdma_dest rdma_dest;
  155. } buffer;
  156. while (1) {
  157. len = sizeof(buffer);
  158. type = rds_message_next_extension(hdr, &pos, &buffer, &len);
  159. if (type == RDS_EXTHDR_NONE)
  160. break;
  161. /* Process extension header here */
  162. switch (type) {
  163. case RDS_EXTHDR_RDMA:
  164. rds_rdma_unuse(rs, be32_to_cpu(buffer.rdma.h_rdma_rkey), 0);
  165. break;
  166. case RDS_EXTHDR_RDMA_DEST:
  167. /* We ignore the size for now. We could stash it
  168. * somewhere and use it for error checking. */
  169. inc->i_rdma_cookie = rds_rdma_make_cookie(
  170. be32_to_cpu(buffer.rdma_dest.h_rdma_rkey),
  171. be32_to_cpu(buffer.rdma_dest.h_rdma_offset));
  172. break;
  173. }
  174. }
  175. }
  176. static void rds_recv_hs_exthdrs(struct rds_header *hdr,
  177. struct rds_connection *conn)
  178. {
  179. unsigned int pos = 0, type, len;
  180. union {
  181. struct rds_ext_header_version version;
  182. u16 rds_npaths;
  183. u32 rds_gen_num;
  184. } buffer;
  185. u32 new_peer_gen_num = 0;
  186. while (1) {
  187. len = sizeof(buffer);
  188. type = rds_message_next_extension(hdr, &pos, &buffer, &len);
  189. if (type == RDS_EXTHDR_NONE)
  190. break;
  191. /* Process extension header here */
  192. switch (type) {
  193. case RDS_EXTHDR_NPATHS:
  194. conn->c_npaths = min_t(int, RDS_MPATH_WORKERS,
  195. buffer.rds_npaths);
  196. break;
  197. case RDS_EXTHDR_GEN_NUM:
  198. new_peer_gen_num = buffer.rds_gen_num;
  199. break;
  200. default:
  201. pr_warn_ratelimited("ignoring unknown exthdr type "
  202. "0x%x\n", type);
  203. }
  204. }
  205. /* if RDS_EXTHDR_NPATHS was not found, default to a single-path */
  206. conn->c_npaths = max_t(int, conn->c_npaths, 1);
  207. rds_conn_peer_gen_update(conn, new_peer_gen_num);
  208. }
  209. /* rds_start_mprds() will synchronously start multiple paths when appropriate.
  210. * The scheme is based on the following rules:
  211. *
  212. * 1. rds_sendmsg on first connect attempt sends the probe ping, with the
  213. * sender's npaths (s_npaths)
  214. * 2. rcvr of probe-ping knows the mprds_paths = min(s_npaths, r_npaths). It
  215. * sends back a probe-pong with r_npaths. After that, if rcvr is the
  216. * smaller ip addr, it starts rds_conn_path_connect_if_down on all
  217. * mprds_paths.
  218. * 3. sender gets woken up, and can move to rds_conn_path_connect_if_down.
  219. * If it is the smaller ipaddr, rds_conn_path_connect_if_down can be
  220. * called after reception of the probe-pong on all mprds_paths.
  221. * Otherwise (sender of probe-ping is not the smaller ip addr): just call
  222. * rds_conn_path_connect_if_down on the hashed path. (see rule 4)
  223. * 4. when cp_index > 0, rds_connect_worker must only trigger
  224. * a connection if laddr < faddr.
  225. * 5. sender may end up queuing the packet on the cp. will get sent out later.
  226. * when connection is completed.
  227. */
  228. static void rds_start_mprds(struct rds_connection *conn)
  229. {
  230. int i;
  231. struct rds_conn_path *cp;
  232. if (conn->c_npaths > 1 && conn->c_laddr < conn->c_faddr) {
  233. for (i = 1; i < conn->c_npaths; i++) {
  234. cp = &conn->c_path[i];
  235. rds_conn_path_connect_if_down(cp);
  236. }
  237. }
  238. }
  239. /*
  240. * The transport must make sure that this is serialized against other
  241. * rx and conn reset on this specific conn.
  242. *
  243. * We currently assert that only one fragmented message will be sent
  244. * down a connection at a time. This lets us reassemble in the conn
  245. * instead of per-flow which means that we don't have to go digging through
  246. * flows to tear down partial reassembly progress on conn failure and
  247. * we save flow lookup and locking for each frag arrival. It does mean
  248. * that small messages will wait behind large ones. Fragmenting at all
  249. * is only to reduce the memory consumption of pre-posted buffers.
  250. *
  251. * The caller passes in saddr and daddr instead of us getting it from the
  252. * conn. This lets loopback, who only has one conn for both directions,
  253. * tell us which roles the addrs in the conn are playing for this message.
  254. */
  255. void rds_recv_incoming(struct rds_connection *conn, __be32 saddr, __be32 daddr,
  256. struct rds_incoming *inc, gfp_t gfp)
  257. {
  258. struct rds_sock *rs = NULL;
  259. struct sock *sk;
  260. unsigned long flags;
  261. struct rds_conn_path *cp;
  262. inc->i_conn = conn;
  263. inc->i_rx_jiffies = jiffies;
  264. if (conn->c_trans->t_mp_capable)
  265. cp = inc->i_conn_path;
  266. else
  267. cp = &conn->c_path[0];
  268. rdsdebug("conn %p next %llu inc %p seq %llu len %u sport %u dport %u "
  269. "flags 0x%x rx_jiffies %lu\n", conn,
  270. (unsigned long long)cp->cp_next_rx_seq,
  271. inc,
  272. (unsigned long long)be64_to_cpu(inc->i_hdr.h_sequence),
  273. be32_to_cpu(inc->i_hdr.h_len),
  274. be16_to_cpu(inc->i_hdr.h_sport),
  275. be16_to_cpu(inc->i_hdr.h_dport),
  276. inc->i_hdr.h_flags,
  277. inc->i_rx_jiffies);
  278. /*
  279. * Sequence numbers should only increase. Messages get their
  280. * sequence number as they're queued in a sending conn. They
  281. * can be dropped, though, if the sending socket is closed before
  282. * they hit the wire. So sequence numbers can skip forward
  283. * under normal operation. They can also drop back in the conn
  284. * failover case as previously sent messages are resent down the
  285. * new instance of a conn. We drop those, otherwise we have
  286. * to assume that the next valid seq does not come after a
  287. * hole in the fragment stream.
  288. *
  289. * The headers don't give us a way to realize if fragments of
  290. * a message have been dropped. We assume that frags that arrive
  291. * to a flow are part of the current message on the flow that is
  292. * being reassembled. This means that senders can't drop messages
  293. * from the sending conn until all their frags are sent.
  294. *
  295. * XXX we could spend more on the wire to get more robust failure
  296. * detection, arguably worth it to avoid data corruption.
  297. */
  298. if (be64_to_cpu(inc->i_hdr.h_sequence) < cp->cp_next_rx_seq &&
  299. (inc->i_hdr.h_flags & RDS_FLAG_RETRANSMITTED)) {
  300. rds_stats_inc(s_recv_drop_old_seq);
  301. goto out;
  302. }
  303. cp->cp_next_rx_seq = be64_to_cpu(inc->i_hdr.h_sequence) + 1;
  304. if (rds_sysctl_ping_enable && inc->i_hdr.h_dport == 0) {
  305. if (inc->i_hdr.h_sport == 0) {
  306. rdsdebug("ignore ping with 0 sport from 0x%x\n", saddr);
  307. goto out;
  308. }
  309. rds_stats_inc(s_recv_ping);
  310. rds_send_pong(cp, inc->i_hdr.h_sport);
  311. /* if this is a handshake ping, start multipath if necessary */
  312. if (RDS_HS_PROBE(inc->i_hdr.h_sport, inc->i_hdr.h_dport)) {
  313. rds_recv_hs_exthdrs(&inc->i_hdr, cp->cp_conn);
  314. rds_start_mprds(cp->cp_conn);
  315. }
  316. goto out;
  317. }
  318. if (inc->i_hdr.h_dport == RDS_FLAG_PROBE_PORT &&
  319. inc->i_hdr.h_sport == 0) {
  320. rds_recv_hs_exthdrs(&inc->i_hdr, cp->cp_conn);
  321. /* if this is a handshake pong, start multipath if necessary */
  322. rds_start_mprds(cp->cp_conn);
  323. wake_up(&cp->cp_conn->c_hs_waitq);
  324. goto out;
  325. }
  326. rs = rds_find_bound(daddr, inc->i_hdr.h_dport);
  327. if (!rs) {
  328. rds_stats_inc(s_recv_drop_no_sock);
  329. goto out;
  330. }
  331. /* Process extension headers */
  332. rds_recv_incoming_exthdrs(inc, rs);
  333. /* We can be racing with rds_release() which marks the socket dead. */
  334. sk = rds_rs_to_sk(rs);
  335. /* serialize with rds_release -> sock_orphan */
  336. write_lock_irqsave(&rs->rs_recv_lock, flags);
  337. if (!sock_flag(sk, SOCK_DEAD)) {
  338. rdsdebug("adding inc %p to rs %p's recv queue\n", inc, rs);
  339. rds_stats_inc(s_recv_queued);
  340. rds_recv_rcvbuf_delta(rs, sk, inc->i_conn->c_lcong,
  341. be32_to_cpu(inc->i_hdr.h_len),
  342. inc->i_hdr.h_dport);
  343. if (sock_flag(sk, SOCK_RCVTSTAMP))
  344. do_gettimeofday(&inc->i_rx_tstamp);
  345. rds_inc_addref(inc);
  346. inc->i_rx_lat_trace[RDS_MSG_RX_END] = local_clock();
  347. list_add_tail(&inc->i_item, &rs->rs_recv_queue);
  348. __rds_wake_sk_sleep(sk);
  349. } else {
  350. rds_stats_inc(s_recv_drop_dead_sock);
  351. }
  352. write_unlock_irqrestore(&rs->rs_recv_lock, flags);
  353. out:
  354. if (rs)
  355. rds_sock_put(rs);
  356. }
  357. EXPORT_SYMBOL_GPL(rds_recv_incoming);
  358. /*
  359. * be very careful here. This is being called as the condition in
  360. * wait_event_*() needs to cope with being called many times.
  361. */
  362. static int rds_next_incoming(struct rds_sock *rs, struct rds_incoming **inc)
  363. {
  364. unsigned long flags;
  365. if (!*inc) {
  366. read_lock_irqsave(&rs->rs_recv_lock, flags);
  367. if (!list_empty(&rs->rs_recv_queue)) {
  368. *inc = list_entry(rs->rs_recv_queue.next,
  369. struct rds_incoming,
  370. i_item);
  371. rds_inc_addref(*inc);
  372. }
  373. read_unlock_irqrestore(&rs->rs_recv_lock, flags);
  374. }
  375. return *inc != NULL;
  376. }
  377. static int rds_still_queued(struct rds_sock *rs, struct rds_incoming *inc,
  378. int drop)
  379. {
  380. struct sock *sk = rds_rs_to_sk(rs);
  381. int ret = 0;
  382. unsigned long flags;
  383. write_lock_irqsave(&rs->rs_recv_lock, flags);
  384. if (!list_empty(&inc->i_item)) {
  385. ret = 1;
  386. if (drop) {
  387. /* XXX make sure this i_conn is reliable */
  388. rds_recv_rcvbuf_delta(rs, sk, inc->i_conn->c_lcong,
  389. -be32_to_cpu(inc->i_hdr.h_len),
  390. inc->i_hdr.h_dport);
  391. list_del_init(&inc->i_item);
  392. rds_inc_put(inc);
  393. }
  394. }
  395. write_unlock_irqrestore(&rs->rs_recv_lock, flags);
  396. rdsdebug("inc %p rs %p still %d dropped %d\n", inc, rs, ret, drop);
  397. return ret;
  398. }
  399. /*
  400. * Pull errors off the error queue.
  401. * If msghdr is NULL, we will just purge the error queue.
  402. */
  403. int rds_notify_queue_get(struct rds_sock *rs, struct msghdr *msghdr)
  404. {
  405. struct rds_notifier *notifier;
  406. struct rds_rdma_notify cmsg = { 0 }; /* fill holes with zero */
  407. unsigned int count = 0, max_messages = ~0U;
  408. unsigned long flags;
  409. LIST_HEAD(copy);
  410. int err = 0;
  411. /* put_cmsg copies to user space and thus may sleep. We can't do this
  412. * with rs_lock held, so first grab as many notifications as we can stuff
  413. * in the user provided cmsg buffer. We don't try to copy more, to avoid
  414. * losing notifications - except when the buffer is so small that it wouldn't
  415. * even hold a single notification. Then we give him as much of this single
  416. * msg as we can squeeze in, and set MSG_CTRUNC.
  417. */
  418. if (msghdr) {
  419. max_messages = msghdr->msg_controllen / CMSG_SPACE(sizeof(cmsg));
  420. if (!max_messages)
  421. max_messages = 1;
  422. }
  423. spin_lock_irqsave(&rs->rs_lock, flags);
  424. while (!list_empty(&rs->rs_notify_queue) && count < max_messages) {
  425. notifier = list_entry(rs->rs_notify_queue.next,
  426. struct rds_notifier, n_list);
  427. list_move(&notifier->n_list, &copy);
  428. count++;
  429. }
  430. spin_unlock_irqrestore(&rs->rs_lock, flags);
  431. if (!count)
  432. return 0;
  433. while (!list_empty(&copy)) {
  434. notifier = list_entry(copy.next, struct rds_notifier, n_list);
  435. if (msghdr) {
  436. cmsg.user_token = notifier->n_user_token;
  437. cmsg.status = notifier->n_status;
  438. err = put_cmsg(msghdr, SOL_RDS, RDS_CMSG_RDMA_STATUS,
  439. sizeof(cmsg), &cmsg);
  440. if (err)
  441. break;
  442. }
  443. list_del_init(&notifier->n_list);
  444. kfree(notifier);
  445. }
  446. /* If we bailed out because of an error in put_cmsg,
  447. * we may be left with one or more notifications that we
  448. * didn't process. Return them to the head of the list. */
  449. if (!list_empty(&copy)) {
  450. spin_lock_irqsave(&rs->rs_lock, flags);
  451. list_splice(&copy, &rs->rs_notify_queue);
  452. spin_unlock_irqrestore(&rs->rs_lock, flags);
  453. }
  454. return err;
  455. }
  456. /*
  457. * Queue a congestion notification
  458. */
  459. static int rds_notify_cong(struct rds_sock *rs, struct msghdr *msghdr)
  460. {
  461. uint64_t notify = rs->rs_cong_notify;
  462. unsigned long flags;
  463. int err;
  464. err = put_cmsg(msghdr, SOL_RDS, RDS_CMSG_CONG_UPDATE,
  465. sizeof(notify), &notify);
  466. if (err)
  467. return err;
  468. spin_lock_irqsave(&rs->rs_lock, flags);
  469. rs->rs_cong_notify &= ~notify;
  470. spin_unlock_irqrestore(&rs->rs_lock, flags);
  471. return 0;
  472. }
  473. /*
  474. * Receive any control messages.
  475. */
  476. static int rds_cmsg_recv(struct rds_incoming *inc, struct msghdr *msg,
  477. struct rds_sock *rs)
  478. {
  479. int ret = 0;
  480. if (inc->i_rdma_cookie) {
  481. ret = put_cmsg(msg, SOL_RDS, RDS_CMSG_RDMA_DEST,
  482. sizeof(inc->i_rdma_cookie), &inc->i_rdma_cookie);
  483. if (ret)
  484. goto out;
  485. }
  486. if ((inc->i_rx_tstamp.tv_sec != 0) &&
  487. sock_flag(rds_rs_to_sk(rs), SOCK_RCVTSTAMP)) {
  488. ret = put_cmsg(msg, SOL_SOCKET, SCM_TIMESTAMP,
  489. sizeof(struct timeval),
  490. &inc->i_rx_tstamp);
  491. if (ret)
  492. goto out;
  493. }
  494. if (rs->rs_rx_traces) {
  495. struct rds_cmsg_rx_trace t;
  496. int i, j;
  497. inc->i_rx_lat_trace[RDS_MSG_RX_CMSG] = local_clock();
  498. t.rx_traces = rs->rs_rx_traces;
  499. for (i = 0; i < rs->rs_rx_traces; i++) {
  500. j = rs->rs_rx_trace[i];
  501. t.rx_trace_pos[i] = j;
  502. t.rx_trace[i] = inc->i_rx_lat_trace[j + 1] -
  503. inc->i_rx_lat_trace[j];
  504. }
  505. ret = put_cmsg(msg, SOL_RDS, RDS_CMSG_RXPATH_LATENCY,
  506. sizeof(t), &t);
  507. if (ret)
  508. goto out;
  509. }
  510. out:
  511. return ret;
  512. }
  513. int rds_recvmsg(struct socket *sock, struct msghdr *msg, size_t size,
  514. int msg_flags)
  515. {
  516. struct sock *sk = sock->sk;
  517. struct rds_sock *rs = rds_sk_to_rs(sk);
  518. long timeo;
  519. int ret = 0, nonblock = msg_flags & MSG_DONTWAIT;
  520. DECLARE_SOCKADDR(struct sockaddr_in *, sin, msg->msg_name);
  521. struct rds_incoming *inc = NULL;
  522. /* udp_recvmsg()->sock_recvtimeo() gets away without locking too.. */
  523. timeo = sock_rcvtimeo(sk, nonblock);
  524. rdsdebug("size %zu flags 0x%x timeo %ld\n", size, msg_flags, timeo);
  525. if (msg_flags & MSG_OOB)
  526. goto out;
  527. while (1) {
  528. struct iov_iter save;
  529. /* If there are pending notifications, do those - and nothing else */
  530. if (!list_empty(&rs->rs_notify_queue)) {
  531. ret = rds_notify_queue_get(rs, msg);
  532. break;
  533. }
  534. if (rs->rs_cong_notify) {
  535. ret = rds_notify_cong(rs, msg);
  536. break;
  537. }
  538. if (!rds_next_incoming(rs, &inc)) {
  539. if (nonblock) {
  540. ret = -EAGAIN;
  541. break;
  542. }
  543. timeo = wait_event_interruptible_timeout(*sk_sleep(sk),
  544. (!list_empty(&rs->rs_notify_queue) ||
  545. rs->rs_cong_notify ||
  546. rds_next_incoming(rs, &inc)), timeo);
  547. rdsdebug("recvmsg woke inc %p timeo %ld\n", inc,
  548. timeo);
  549. if (timeo > 0 || timeo == MAX_SCHEDULE_TIMEOUT)
  550. continue;
  551. ret = timeo;
  552. if (ret == 0)
  553. ret = -ETIMEDOUT;
  554. break;
  555. }
  556. rdsdebug("copying inc %p from %pI4:%u to user\n", inc,
  557. &inc->i_conn->c_faddr,
  558. ntohs(inc->i_hdr.h_sport));
  559. save = msg->msg_iter;
  560. ret = inc->i_conn->c_trans->inc_copy_to_user(inc, &msg->msg_iter);
  561. if (ret < 0)
  562. break;
  563. /*
  564. * if the message we just copied isn't at the head of the
  565. * recv queue then someone else raced us to return it, try
  566. * to get the next message.
  567. */
  568. if (!rds_still_queued(rs, inc, !(msg_flags & MSG_PEEK))) {
  569. rds_inc_put(inc);
  570. inc = NULL;
  571. rds_stats_inc(s_recv_deliver_raced);
  572. msg->msg_iter = save;
  573. continue;
  574. }
  575. if (ret < be32_to_cpu(inc->i_hdr.h_len)) {
  576. if (msg_flags & MSG_TRUNC)
  577. ret = be32_to_cpu(inc->i_hdr.h_len);
  578. msg->msg_flags |= MSG_TRUNC;
  579. }
  580. if (rds_cmsg_recv(inc, msg, rs)) {
  581. ret = -EFAULT;
  582. goto out;
  583. }
  584. rds_stats_inc(s_recv_delivered);
  585. if (sin) {
  586. sin->sin_family = AF_INET;
  587. sin->sin_port = inc->i_hdr.h_sport;
  588. sin->sin_addr.s_addr = inc->i_saddr;
  589. memset(sin->sin_zero, 0, sizeof(sin->sin_zero));
  590. msg->msg_namelen = sizeof(*sin);
  591. }
  592. break;
  593. }
  594. if (inc)
  595. rds_inc_put(inc);
  596. out:
  597. return ret;
  598. }
  599. /*
  600. * The socket is being shut down and we're asked to drop messages that were
  601. * queued for recvmsg. The caller has unbound the socket so the receive path
  602. * won't queue any more incoming fragments or messages on the socket.
  603. */
  604. void rds_clear_recv_queue(struct rds_sock *rs)
  605. {
  606. struct sock *sk = rds_rs_to_sk(rs);
  607. struct rds_incoming *inc, *tmp;
  608. unsigned long flags;
  609. write_lock_irqsave(&rs->rs_recv_lock, flags);
  610. list_for_each_entry_safe(inc, tmp, &rs->rs_recv_queue, i_item) {
  611. rds_recv_rcvbuf_delta(rs, sk, inc->i_conn->c_lcong,
  612. -be32_to_cpu(inc->i_hdr.h_len),
  613. inc->i_hdr.h_dport);
  614. list_del_init(&inc->i_item);
  615. rds_inc_put(inc);
  616. }
  617. write_unlock_irqrestore(&rs->rs_recv_lock, flags);
  618. }
  619. /*
  620. * inc->i_saddr isn't used here because it is only set in the receive
  621. * path.
  622. */
  623. void rds_inc_info_copy(struct rds_incoming *inc,
  624. struct rds_info_iterator *iter,
  625. __be32 saddr, __be32 daddr, int flip)
  626. {
  627. struct rds_info_message minfo;
  628. minfo.seq = be64_to_cpu(inc->i_hdr.h_sequence);
  629. minfo.len = be32_to_cpu(inc->i_hdr.h_len);
  630. if (flip) {
  631. minfo.laddr = daddr;
  632. minfo.faddr = saddr;
  633. minfo.lport = inc->i_hdr.h_dport;
  634. minfo.fport = inc->i_hdr.h_sport;
  635. } else {
  636. minfo.laddr = saddr;
  637. minfo.faddr = daddr;
  638. minfo.lport = inc->i_hdr.h_sport;
  639. minfo.fport = inc->i_hdr.h_dport;
  640. }
  641. minfo.flags = 0;
  642. rds_info_copy(iter, &minfo, sizeof(minfo));
  643. }