smc_core.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715
  1. // SPDX-License-Identifier: GPL-2.0
  2. /*
  3. * Shared Memory Communications over RDMA (SMC-R) and RoCE
  4. *
  5. * Basic Transport Functions exploiting Infiniband API
  6. *
  7. * Copyright IBM Corp. 2016
  8. *
  9. * Author(s): Ursula Braun <ubraun@linux.vnet.ibm.com>
  10. */
  11. #include <linux/socket.h>
  12. #include <linux/if_vlan.h>
  13. #include <linux/random.h>
  14. #include <linux/workqueue.h>
  15. #include <net/tcp.h>
  16. #include <net/sock.h>
  17. #include <rdma/ib_verbs.h>
  18. #include "smc.h"
  19. #include "smc_clc.h"
  20. #include "smc_core.h"
  21. #include "smc_ib.h"
  22. #include "smc_wr.h"
  23. #include "smc_llc.h"
  24. #include "smc_cdc.h"
  25. #include "smc_close.h"
  26. #define SMC_LGR_NUM_INCR 256
  27. #define SMC_LGR_FREE_DELAY_SERV (600 * HZ)
  28. #define SMC_LGR_FREE_DELAY_CLNT (SMC_LGR_FREE_DELAY_SERV + 10)
  29. static u32 smc_lgr_num; /* unique link group number */
  30. /* Register connection's alert token in our lookup structure.
  31. * To use rbtrees we have to implement our own insert core.
  32. * Requires @conns_lock
  33. * @smc connection to register
  34. * Returns 0 on success, != otherwise.
  35. */
  36. static void smc_lgr_add_alert_token(struct smc_connection *conn)
  37. {
  38. struct rb_node **link, *parent = NULL;
  39. u32 token = conn->alert_token_local;
  40. link = &conn->lgr->conns_all.rb_node;
  41. while (*link) {
  42. struct smc_connection *cur = rb_entry(*link,
  43. struct smc_connection, alert_node);
  44. parent = *link;
  45. if (cur->alert_token_local > token)
  46. link = &parent->rb_left;
  47. else
  48. link = &parent->rb_right;
  49. }
  50. /* Put the new node there */
  51. rb_link_node(&conn->alert_node, parent, link);
  52. rb_insert_color(&conn->alert_node, &conn->lgr->conns_all);
  53. }
  54. /* Register connection in link group by assigning an alert token
  55. * registered in a search tree.
  56. * Requires @conns_lock
  57. * Note that '0' is a reserved value and not assigned.
  58. */
  59. static void smc_lgr_register_conn(struct smc_connection *conn)
  60. {
  61. struct smc_sock *smc = container_of(conn, struct smc_sock, conn);
  62. static atomic_t nexttoken = ATOMIC_INIT(0);
  63. /* find a new alert_token_local value not yet used by some connection
  64. * in this link group
  65. */
  66. sock_hold(&smc->sk); /* sock_put in smc_lgr_unregister_conn() */
  67. while (!conn->alert_token_local) {
  68. conn->alert_token_local = atomic_inc_return(&nexttoken);
  69. if (smc_lgr_find_conn(conn->alert_token_local, conn->lgr))
  70. conn->alert_token_local = 0;
  71. }
  72. smc_lgr_add_alert_token(conn);
  73. conn->lgr->conns_num++;
  74. }
  75. /* Unregister connection and reset the alert token of the given connection<
  76. */
  77. static void __smc_lgr_unregister_conn(struct smc_connection *conn)
  78. {
  79. struct smc_sock *smc = container_of(conn, struct smc_sock, conn);
  80. struct smc_link_group *lgr = conn->lgr;
  81. rb_erase(&conn->alert_node, &lgr->conns_all);
  82. lgr->conns_num--;
  83. conn->alert_token_local = 0;
  84. conn->lgr = NULL;
  85. sock_put(&smc->sk); /* sock_hold in smc_lgr_register_conn() */
  86. }
  87. /* Unregister connection and trigger lgr freeing if applicable
  88. */
  89. static void smc_lgr_unregister_conn(struct smc_connection *conn)
  90. {
  91. struct smc_link_group *lgr = conn->lgr;
  92. int reduced = 0;
  93. write_lock_bh(&lgr->conns_lock);
  94. if (conn->alert_token_local) {
  95. reduced = 1;
  96. __smc_lgr_unregister_conn(conn);
  97. }
  98. write_unlock_bh(&lgr->conns_lock);
  99. if (!reduced || lgr->conns_num)
  100. return;
  101. /* client link group creation always follows the server link group
  102. * creation. For client use a somewhat higher removal delay time,
  103. * otherwise there is a risk of out-of-sync link groups.
  104. */
  105. mod_delayed_work(system_wq, &lgr->free_work,
  106. lgr->role == SMC_CLNT ? SMC_LGR_FREE_DELAY_CLNT :
  107. SMC_LGR_FREE_DELAY_SERV);
  108. }
  109. static void smc_lgr_free_work(struct work_struct *work)
  110. {
  111. struct smc_link_group *lgr = container_of(to_delayed_work(work),
  112. struct smc_link_group,
  113. free_work);
  114. bool conns;
  115. spin_lock_bh(&smc_lgr_list.lock);
  116. read_lock_bh(&lgr->conns_lock);
  117. conns = RB_EMPTY_ROOT(&lgr->conns_all);
  118. read_unlock_bh(&lgr->conns_lock);
  119. if (!conns) { /* number of lgr connections is no longer zero */
  120. spin_unlock_bh(&smc_lgr_list.lock);
  121. return;
  122. }
  123. list_del_init(&lgr->list); /* remove from smc_lgr_list */
  124. spin_unlock_bh(&smc_lgr_list.lock);
  125. smc_lgr_free(lgr);
  126. }
  127. /* create a new SMC link group */
  128. static int smc_lgr_create(struct smc_sock *smc, __be32 peer_in_addr,
  129. struct smc_ib_device *smcibdev, u8 ibport,
  130. char *peer_systemid, unsigned short vlan_id)
  131. {
  132. struct smc_link_group *lgr;
  133. struct smc_link *lnk;
  134. u8 rndvec[3];
  135. int rc = 0;
  136. int i;
  137. lgr = kzalloc(sizeof(*lgr), GFP_KERNEL);
  138. if (!lgr) {
  139. rc = -ENOMEM;
  140. goto out;
  141. }
  142. lgr->role = smc->listen_smc ? SMC_SERV : SMC_CLNT;
  143. lgr->sync_err = false;
  144. lgr->daddr = peer_in_addr;
  145. memcpy(lgr->peer_systemid, peer_systemid, SMC_SYSTEMID_LEN);
  146. lgr->vlan_id = vlan_id;
  147. rwlock_init(&lgr->sndbufs_lock);
  148. rwlock_init(&lgr->rmbs_lock);
  149. for (i = 0; i < SMC_RMBE_SIZES; i++) {
  150. INIT_LIST_HEAD(&lgr->sndbufs[i]);
  151. INIT_LIST_HEAD(&lgr->rmbs[i]);
  152. }
  153. smc_lgr_num += SMC_LGR_NUM_INCR;
  154. memcpy(&lgr->id, (u8 *)&smc_lgr_num, SMC_LGR_ID_SIZE);
  155. INIT_DELAYED_WORK(&lgr->free_work, smc_lgr_free_work);
  156. lgr->conns_all = RB_ROOT;
  157. lnk = &lgr->lnk[SMC_SINGLE_LINK];
  158. /* initialize link */
  159. lnk->smcibdev = smcibdev;
  160. lnk->ibport = ibport;
  161. lnk->path_mtu = smcibdev->pattr[ibport - 1].active_mtu;
  162. if (!smcibdev->initialized)
  163. smc_ib_setup_per_ibdev(smcibdev);
  164. get_random_bytes(rndvec, sizeof(rndvec));
  165. lnk->psn_initial = rndvec[0] + (rndvec[1] << 8) + (rndvec[2] << 16);
  166. rc = smc_wr_alloc_link_mem(lnk);
  167. if (rc)
  168. goto free_lgr;
  169. rc = smc_ib_create_protection_domain(lnk);
  170. if (rc)
  171. goto free_link_mem;
  172. rc = smc_ib_create_queue_pair(lnk);
  173. if (rc)
  174. goto dealloc_pd;
  175. rc = smc_wr_create_link(lnk);
  176. if (rc)
  177. goto destroy_qp;
  178. init_completion(&lnk->llc_confirm);
  179. init_completion(&lnk->llc_confirm_resp);
  180. smc->conn.lgr = lgr;
  181. rwlock_init(&lgr->conns_lock);
  182. spin_lock_bh(&smc_lgr_list.lock);
  183. list_add(&lgr->list, &smc_lgr_list.list);
  184. spin_unlock_bh(&smc_lgr_list.lock);
  185. return 0;
  186. destroy_qp:
  187. smc_ib_destroy_queue_pair(lnk);
  188. dealloc_pd:
  189. smc_ib_dealloc_protection_domain(lnk);
  190. free_link_mem:
  191. smc_wr_free_link_mem(lnk);
  192. free_lgr:
  193. kfree(lgr);
  194. out:
  195. return rc;
  196. }
  197. static void smc_buf_unuse(struct smc_connection *conn)
  198. {
  199. if (conn->sndbuf_desc) {
  200. conn->sndbuf_desc->used = 0;
  201. conn->sndbuf_size = 0;
  202. }
  203. if (conn->rmb_desc) {
  204. conn->rmb_desc->reused = true;
  205. conn->rmb_desc->used = 0;
  206. conn->rmbe_size = 0;
  207. }
  208. }
  209. /* remove a finished connection from its link group */
  210. void smc_conn_free(struct smc_connection *conn)
  211. {
  212. struct smc_link_group *lgr = conn->lgr;
  213. if (!lgr)
  214. return;
  215. smc_cdc_tx_dismiss_slots(conn);
  216. smc_lgr_unregister_conn(conn);
  217. smc_buf_unuse(conn);
  218. }
  219. static void smc_link_clear(struct smc_link *lnk)
  220. {
  221. lnk->peer_qpn = 0;
  222. smc_ib_modify_qp_reset(lnk);
  223. smc_wr_free_link(lnk);
  224. smc_ib_destroy_queue_pair(lnk);
  225. smc_ib_dealloc_protection_domain(lnk);
  226. smc_wr_free_link_mem(lnk);
  227. }
  228. static void smc_buf_free(struct smc_buf_desc *buf_desc, struct smc_link *lnk,
  229. bool is_rmb)
  230. {
  231. if (is_rmb) {
  232. if (buf_desc->mr_rx[SMC_SINGLE_LINK])
  233. smc_ib_put_memory_region(
  234. buf_desc->mr_rx[SMC_SINGLE_LINK]);
  235. smc_ib_buf_unmap_sg(lnk->smcibdev, buf_desc,
  236. DMA_FROM_DEVICE);
  237. } else {
  238. smc_ib_buf_unmap_sg(lnk->smcibdev, buf_desc,
  239. DMA_TO_DEVICE);
  240. }
  241. sg_free_table(&buf_desc->sgt[SMC_SINGLE_LINK]);
  242. if (buf_desc->cpu_addr)
  243. free_pages((unsigned long)buf_desc->cpu_addr, buf_desc->order);
  244. kfree(buf_desc);
  245. }
  246. static void __smc_lgr_free_bufs(struct smc_link_group *lgr, bool is_rmb)
  247. {
  248. struct smc_link *lnk = &lgr->lnk[SMC_SINGLE_LINK];
  249. struct smc_buf_desc *buf_desc, *bf_desc;
  250. struct list_head *buf_list;
  251. int i;
  252. for (i = 0; i < SMC_RMBE_SIZES; i++) {
  253. if (is_rmb)
  254. buf_list = &lgr->rmbs[i];
  255. else
  256. buf_list = &lgr->sndbufs[i];
  257. list_for_each_entry_safe(buf_desc, bf_desc, buf_list,
  258. list) {
  259. list_del(&buf_desc->list);
  260. smc_buf_free(buf_desc, lnk, is_rmb);
  261. }
  262. }
  263. }
  264. static void smc_lgr_free_bufs(struct smc_link_group *lgr)
  265. {
  266. /* free send buffers */
  267. __smc_lgr_free_bufs(lgr, false);
  268. /* free rmbs */
  269. __smc_lgr_free_bufs(lgr, true);
  270. }
  271. /* remove a link group */
  272. void smc_lgr_free(struct smc_link_group *lgr)
  273. {
  274. smc_lgr_free_bufs(lgr);
  275. smc_link_clear(&lgr->lnk[SMC_SINGLE_LINK]);
  276. kfree(lgr);
  277. }
  278. /* terminate linkgroup abnormally */
  279. void smc_lgr_terminate(struct smc_link_group *lgr)
  280. {
  281. struct smc_connection *conn;
  282. struct smc_sock *smc;
  283. struct rb_node *node;
  284. spin_lock_bh(&smc_lgr_list.lock);
  285. if (list_empty(&lgr->list)) {
  286. /* termination already triggered */
  287. spin_unlock_bh(&smc_lgr_list.lock);
  288. return;
  289. }
  290. /* do not use this link group for new connections */
  291. list_del_init(&lgr->list);
  292. spin_unlock_bh(&smc_lgr_list.lock);
  293. write_lock_bh(&lgr->conns_lock);
  294. node = rb_first(&lgr->conns_all);
  295. while (node) {
  296. conn = rb_entry(node, struct smc_connection, alert_node);
  297. smc = container_of(conn, struct smc_sock, conn);
  298. sock_hold(&smc->sk);
  299. __smc_lgr_unregister_conn(conn);
  300. schedule_work(&conn->close_work);
  301. sock_put(&smc->sk);
  302. node = rb_first(&lgr->conns_all);
  303. }
  304. write_unlock_bh(&lgr->conns_lock);
  305. }
  306. /* Determine vlan of internal TCP socket.
  307. * @vlan_id: address to store the determined vlan id into
  308. */
  309. static int smc_vlan_by_tcpsk(struct socket *clcsock, unsigned short *vlan_id)
  310. {
  311. struct dst_entry *dst = sk_dst_get(clcsock->sk);
  312. int rc = 0;
  313. *vlan_id = 0;
  314. if (!dst) {
  315. rc = -ENOTCONN;
  316. goto out;
  317. }
  318. if (!dst->dev) {
  319. rc = -ENODEV;
  320. goto out_rel;
  321. }
  322. if (is_vlan_dev(dst->dev))
  323. *vlan_id = vlan_dev_vlan_id(dst->dev);
  324. out_rel:
  325. dst_release(dst);
  326. out:
  327. return rc;
  328. }
  329. /* determine the link gid matching the vlan id of the link group */
  330. static int smc_link_determine_gid(struct smc_link_group *lgr)
  331. {
  332. struct smc_link *lnk = &lgr->lnk[SMC_SINGLE_LINK];
  333. struct ib_gid_attr gattr;
  334. union ib_gid gid;
  335. int i;
  336. if (!lgr->vlan_id) {
  337. lnk->gid = lnk->smcibdev->gid[lnk->ibport - 1];
  338. return 0;
  339. }
  340. for (i = 0; i < lnk->smcibdev->pattr[lnk->ibport - 1].gid_tbl_len;
  341. i++) {
  342. if (ib_query_gid(lnk->smcibdev->ibdev, lnk->ibport, i, &gid,
  343. &gattr))
  344. continue;
  345. if (gattr.ndev &&
  346. (vlan_dev_vlan_id(gattr.ndev) == lgr->vlan_id)) {
  347. lnk->gid = gid;
  348. return 0;
  349. }
  350. }
  351. return -ENODEV;
  352. }
  353. /* create a new SMC connection (and a new link group if necessary) */
  354. int smc_conn_create(struct smc_sock *smc, __be32 peer_in_addr,
  355. struct smc_ib_device *smcibdev, u8 ibport,
  356. struct smc_clc_msg_local *lcl, int srv_first_contact)
  357. {
  358. struct smc_connection *conn = &smc->conn;
  359. struct smc_link_group *lgr;
  360. unsigned short vlan_id;
  361. enum smc_lgr_role role;
  362. int local_contact = SMC_FIRST_CONTACT;
  363. int rc = 0;
  364. role = smc->listen_smc ? SMC_SERV : SMC_CLNT;
  365. rc = smc_vlan_by_tcpsk(smc->clcsock, &vlan_id);
  366. if (rc)
  367. return rc;
  368. if ((role == SMC_CLNT) && srv_first_contact)
  369. /* create new link group as well */
  370. goto create;
  371. /* determine if an existing link group can be reused */
  372. spin_lock_bh(&smc_lgr_list.lock);
  373. list_for_each_entry(lgr, &smc_lgr_list.list, list) {
  374. write_lock_bh(&lgr->conns_lock);
  375. if (!memcmp(lgr->peer_systemid, lcl->id_for_peer,
  376. SMC_SYSTEMID_LEN) &&
  377. !memcmp(lgr->lnk[SMC_SINGLE_LINK].peer_gid, &lcl->gid,
  378. SMC_GID_SIZE) &&
  379. !memcmp(lgr->lnk[SMC_SINGLE_LINK].peer_mac, lcl->mac,
  380. sizeof(lcl->mac)) &&
  381. !lgr->sync_err &&
  382. (lgr->role == role) &&
  383. (lgr->vlan_id == vlan_id) &&
  384. ((role == SMC_CLNT) ||
  385. (lgr->conns_num < SMC_RMBS_PER_LGR_MAX))) {
  386. /* link group found */
  387. local_contact = SMC_REUSE_CONTACT;
  388. conn->lgr = lgr;
  389. smc_lgr_register_conn(conn); /* add smc conn to lgr */
  390. write_unlock_bh(&lgr->conns_lock);
  391. break;
  392. }
  393. write_unlock_bh(&lgr->conns_lock);
  394. }
  395. spin_unlock_bh(&smc_lgr_list.lock);
  396. if (role == SMC_CLNT && !srv_first_contact &&
  397. (local_contact == SMC_FIRST_CONTACT)) {
  398. /* Server reuses a link group, but Client wants to start
  399. * a new one
  400. * send out_of_sync decline, reason synchr. error
  401. */
  402. return -ENOLINK;
  403. }
  404. create:
  405. if (local_contact == SMC_FIRST_CONTACT) {
  406. rc = smc_lgr_create(smc, peer_in_addr, smcibdev, ibport,
  407. lcl->id_for_peer, vlan_id);
  408. if (rc)
  409. goto out;
  410. smc_lgr_register_conn(conn); /* add smc conn to lgr */
  411. rc = smc_link_determine_gid(conn->lgr);
  412. }
  413. conn->local_tx_ctrl.common.type = SMC_CDC_MSG_TYPE;
  414. conn->local_tx_ctrl.len = sizeof(struct smc_cdc_msg);
  415. #ifndef KERNEL_HAS_ATOMIC64
  416. spin_lock_init(&conn->acurs_lock);
  417. #endif
  418. out:
  419. return rc ? rc : local_contact;
  420. }
  421. /* try to reuse a sndbuf or rmb description slot for a certain
  422. * buffer size; if not available, return NULL
  423. */
  424. static inline
  425. struct smc_buf_desc *smc_buf_get_slot(struct smc_link_group *lgr,
  426. int compressed_bufsize,
  427. rwlock_t *lock,
  428. struct list_head *buf_list)
  429. {
  430. struct smc_buf_desc *buf_slot;
  431. read_lock_bh(lock);
  432. list_for_each_entry(buf_slot, buf_list, list) {
  433. if (cmpxchg(&buf_slot->used, 0, 1) == 0) {
  434. read_unlock_bh(lock);
  435. return buf_slot;
  436. }
  437. }
  438. read_unlock_bh(lock);
  439. return NULL;
  440. }
  441. /* one of the conditions for announcing a receiver's current window size is
  442. * that it "results in a minimum increase in the window size of 10% of the
  443. * receive buffer space" [RFC7609]
  444. */
  445. static inline int smc_rmb_wnd_update_limit(int rmbe_size)
  446. {
  447. return min_t(int, rmbe_size / 10, SOCK_MIN_SNDBUF / 2);
  448. }
  449. static struct smc_buf_desc *smc_new_buf_create(struct smc_link_group *lgr,
  450. bool is_rmb, int bufsize)
  451. {
  452. struct smc_buf_desc *buf_desc;
  453. struct smc_link *lnk;
  454. int rc;
  455. /* try to alloc a new buffer */
  456. buf_desc = kzalloc(sizeof(*buf_desc), GFP_KERNEL);
  457. if (!buf_desc)
  458. return ERR_PTR(-ENOMEM);
  459. buf_desc->cpu_addr =
  460. (void *)__get_free_pages(GFP_KERNEL | __GFP_NOWARN |
  461. __GFP_NOMEMALLOC |
  462. __GFP_NORETRY | __GFP_ZERO,
  463. get_order(bufsize));
  464. if (!buf_desc->cpu_addr) {
  465. kfree(buf_desc);
  466. return ERR_PTR(-EAGAIN);
  467. }
  468. buf_desc->order = get_order(bufsize);
  469. /* build the sg table from the pages */
  470. lnk = &lgr->lnk[SMC_SINGLE_LINK];
  471. rc = sg_alloc_table(&buf_desc->sgt[SMC_SINGLE_LINK], 1,
  472. GFP_KERNEL);
  473. if (rc) {
  474. smc_buf_free(buf_desc, lnk, is_rmb);
  475. return ERR_PTR(rc);
  476. }
  477. sg_set_buf(buf_desc->sgt[SMC_SINGLE_LINK].sgl,
  478. buf_desc->cpu_addr, bufsize);
  479. /* map sg table to DMA address */
  480. rc = smc_ib_buf_map_sg(lnk->smcibdev, buf_desc,
  481. is_rmb ? DMA_FROM_DEVICE : DMA_TO_DEVICE);
  482. /* SMC protocol depends on mapping to one DMA address only */
  483. if (rc != 1) {
  484. smc_buf_free(buf_desc, lnk, is_rmb);
  485. return ERR_PTR(-EAGAIN);
  486. }
  487. /* create a new memory region for the RMB */
  488. if (is_rmb) {
  489. rc = smc_ib_get_memory_region(lnk->roce_pd,
  490. IB_ACCESS_REMOTE_WRITE |
  491. IB_ACCESS_LOCAL_WRITE,
  492. buf_desc);
  493. if (rc) {
  494. smc_buf_free(buf_desc, lnk, is_rmb);
  495. return ERR_PTR(rc);
  496. }
  497. }
  498. return buf_desc;
  499. }
  500. static int __smc_buf_create(struct smc_sock *smc, bool is_rmb)
  501. {
  502. struct smc_connection *conn = &smc->conn;
  503. struct smc_link_group *lgr = conn->lgr;
  504. struct smc_buf_desc *buf_desc = NULL;
  505. struct list_head *buf_list;
  506. int bufsize, bufsize_short;
  507. int sk_buf_size;
  508. rwlock_t *lock;
  509. if (is_rmb)
  510. /* use socket recv buffer size (w/o overhead) as start value */
  511. sk_buf_size = smc->sk.sk_rcvbuf / 2;
  512. else
  513. /* use socket send buffer size (w/o overhead) as start value */
  514. sk_buf_size = smc->sk.sk_sndbuf / 2;
  515. for (bufsize_short = smc_compress_bufsize(smc->sk.sk_sndbuf / 2);
  516. bufsize_short >= 0; bufsize_short--) {
  517. if (is_rmb) {
  518. lock = &lgr->rmbs_lock;
  519. buf_list = &lgr->rmbs[bufsize_short];
  520. } else {
  521. lock = &lgr->sndbufs_lock;
  522. buf_list = &lgr->sndbufs[bufsize_short];
  523. }
  524. bufsize = smc_uncompress_bufsize(bufsize_short);
  525. if ((1 << get_order(bufsize)) > SG_MAX_SINGLE_ALLOC)
  526. continue;
  527. /* check for reusable slot in the link group */
  528. buf_desc = smc_buf_get_slot(lgr, bufsize_short, lock, buf_list);
  529. if (buf_desc) {
  530. memset(buf_desc->cpu_addr, 0, bufsize);
  531. break; /* found reusable slot */
  532. }
  533. buf_desc = smc_new_buf_create(lgr, is_rmb, bufsize);
  534. if (PTR_ERR(buf_desc) == -ENOMEM)
  535. break;
  536. if (IS_ERR(buf_desc))
  537. continue;
  538. buf_desc->used = 1;
  539. write_lock_bh(lock);
  540. list_add(&buf_desc->list, buf_list);
  541. write_unlock_bh(lock);
  542. break; /* found */
  543. }
  544. if (IS_ERR(buf_desc))
  545. return -ENOMEM;
  546. if (is_rmb) {
  547. conn->rmb_desc = buf_desc;
  548. conn->rmbe_size = bufsize;
  549. conn->rmbe_size_short = bufsize_short;
  550. smc->sk.sk_rcvbuf = bufsize * 2;
  551. atomic_set(&conn->bytes_to_rcv, 0);
  552. conn->rmbe_update_limit = smc_rmb_wnd_update_limit(bufsize);
  553. } else {
  554. conn->sndbuf_desc = buf_desc;
  555. conn->sndbuf_size = bufsize;
  556. smc->sk.sk_sndbuf = bufsize * 2;
  557. atomic_set(&conn->sndbuf_space, bufsize);
  558. }
  559. return 0;
  560. }
  561. void smc_sndbuf_sync_sg_for_cpu(struct smc_connection *conn)
  562. {
  563. struct smc_link_group *lgr = conn->lgr;
  564. smc_ib_sync_sg_for_cpu(lgr->lnk[SMC_SINGLE_LINK].smcibdev,
  565. conn->sndbuf_desc, DMA_TO_DEVICE);
  566. }
  567. void smc_sndbuf_sync_sg_for_device(struct smc_connection *conn)
  568. {
  569. struct smc_link_group *lgr = conn->lgr;
  570. smc_ib_sync_sg_for_device(lgr->lnk[SMC_SINGLE_LINK].smcibdev,
  571. conn->sndbuf_desc, DMA_TO_DEVICE);
  572. }
  573. void smc_rmb_sync_sg_for_cpu(struct smc_connection *conn)
  574. {
  575. struct smc_link_group *lgr = conn->lgr;
  576. smc_ib_sync_sg_for_cpu(lgr->lnk[SMC_SINGLE_LINK].smcibdev,
  577. conn->rmb_desc, DMA_FROM_DEVICE);
  578. }
  579. void smc_rmb_sync_sg_for_device(struct smc_connection *conn)
  580. {
  581. struct smc_link_group *lgr = conn->lgr;
  582. smc_ib_sync_sg_for_device(lgr->lnk[SMC_SINGLE_LINK].smcibdev,
  583. conn->rmb_desc, DMA_FROM_DEVICE);
  584. }
  585. /* create the send and receive buffer for an SMC socket;
  586. * receive buffers are called RMBs;
  587. * (even though the SMC protocol allows more than one RMB-element per RMB,
  588. * the Linux implementation uses just one RMB-element per RMB, i.e. uses an
  589. * extra RMB for every connection in a link group
  590. */
  591. int smc_buf_create(struct smc_sock *smc)
  592. {
  593. int rc;
  594. /* create send buffer */
  595. rc = __smc_buf_create(smc, false);
  596. if (rc)
  597. return rc;
  598. /* create rmb */
  599. rc = __smc_buf_create(smc, true);
  600. if (rc)
  601. smc_buf_free(smc->conn.sndbuf_desc,
  602. &smc->conn.lgr->lnk[SMC_SINGLE_LINK], false);
  603. return rc;
  604. }
  605. static inline int smc_rmb_reserve_rtoken_idx(struct smc_link_group *lgr)
  606. {
  607. int i;
  608. for_each_clear_bit(i, lgr->rtokens_used_mask, SMC_RMBS_PER_LGR_MAX) {
  609. if (!test_and_set_bit(i, lgr->rtokens_used_mask))
  610. return i;
  611. }
  612. return -ENOSPC;
  613. }
  614. /* save rkey and dma_addr received from peer during clc handshake */
  615. int smc_rmb_rtoken_handling(struct smc_connection *conn,
  616. struct smc_clc_msg_accept_confirm *clc)
  617. {
  618. u64 dma_addr = be64_to_cpu(clc->rmb_dma_addr);
  619. struct smc_link_group *lgr = conn->lgr;
  620. u32 rkey = ntohl(clc->rmb_rkey);
  621. int i;
  622. for (i = 0; i < SMC_RMBS_PER_LGR_MAX; i++) {
  623. if ((lgr->rtokens[i][SMC_SINGLE_LINK].rkey == rkey) &&
  624. (lgr->rtokens[i][SMC_SINGLE_LINK].dma_addr == dma_addr) &&
  625. test_bit(i, lgr->rtokens_used_mask)) {
  626. conn->rtoken_idx = i;
  627. return 0;
  628. }
  629. }
  630. conn->rtoken_idx = smc_rmb_reserve_rtoken_idx(lgr);
  631. if (conn->rtoken_idx < 0)
  632. return conn->rtoken_idx;
  633. lgr->rtokens[conn->rtoken_idx][SMC_SINGLE_LINK].rkey = rkey;
  634. lgr->rtokens[conn->rtoken_idx][SMC_SINGLE_LINK].dma_addr = dma_addr;
  635. return 0;
  636. }