chtls_cm.c 51 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146
  1. /*
  2. * Copyright (c) 2018 Chelsio Communications, Inc.
  3. *
  4. * This program is free software; you can redistribute it and/or modify
  5. * it under the terms of the GNU General Public License version 2 as
  6. * published by the Free Software Foundation.
  7. *
  8. * Written by: Atul Gupta (atul.gupta@chelsio.com)
  9. */
  10. #include <linux/module.h>
  11. #include <linux/list.h>
  12. #include <linux/workqueue.h>
  13. #include <linux/skbuff.h>
  14. #include <linux/timer.h>
  15. #include <linux/notifier.h>
  16. #include <linux/inetdevice.h>
  17. #include <linux/ip.h>
  18. #include <linux/tcp.h>
  19. #include <linux/sched/signal.h>
  20. #include <linux/kallsyms.h>
  21. #include <linux/kprobes.h>
  22. #include <linux/if_vlan.h>
  23. #include <net/tcp.h>
  24. #include <net/dst.h>
  25. #include "chtls.h"
  26. #include "chtls_cm.h"
  27. /*
  28. * State transitions and actions for close. Note that if we are in SYN_SENT
  29. * we remain in that state as we cannot control a connection while it's in
  30. * SYN_SENT; such connections are allowed to establish and are then aborted.
  31. */
  32. static unsigned char new_state[16] = {
  33. /* current state: new state: action: */
  34. /* (Invalid) */ TCP_CLOSE,
  35. /* TCP_ESTABLISHED */ TCP_FIN_WAIT1 | TCP_ACTION_FIN,
  36. /* TCP_SYN_SENT */ TCP_SYN_SENT,
  37. /* TCP_SYN_RECV */ TCP_FIN_WAIT1 | TCP_ACTION_FIN,
  38. /* TCP_FIN_WAIT1 */ TCP_FIN_WAIT1,
  39. /* TCP_FIN_WAIT2 */ TCP_FIN_WAIT2,
  40. /* TCP_TIME_WAIT */ TCP_CLOSE,
  41. /* TCP_CLOSE */ TCP_CLOSE,
  42. /* TCP_CLOSE_WAIT */ TCP_LAST_ACK | TCP_ACTION_FIN,
  43. /* TCP_LAST_ACK */ TCP_LAST_ACK,
  44. /* TCP_LISTEN */ TCP_CLOSE,
  45. /* TCP_CLOSING */ TCP_CLOSING,
  46. };
  47. static struct chtls_sock *chtls_sock_create(struct chtls_dev *cdev)
  48. {
  49. struct chtls_sock *csk = kzalloc(sizeof(*csk), GFP_ATOMIC);
  50. if (!csk)
  51. return NULL;
  52. csk->txdata_skb_cache = alloc_skb(TXDATA_SKB_LEN, GFP_ATOMIC);
  53. if (!csk->txdata_skb_cache) {
  54. kfree(csk);
  55. return NULL;
  56. }
  57. kref_init(&csk->kref);
  58. csk->cdev = cdev;
  59. skb_queue_head_init(&csk->txq);
  60. csk->wr_skb_head = NULL;
  61. csk->wr_skb_tail = NULL;
  62. csk->mss = MAX_MSS;
  63. csk->tlshws.ofld = 1;
  64. csk->tlshws.txkey = -1;
  65. csk->tlshws.rxkey = -1;
  66. csk->tlshws.mfs = TLS_MFS;
  67. skb_queue_head_init(&csk->tlshws.sk_recv_queue);
  68. return csk;
  69. }
  70. static void chtls_sock_release(struct kref *ref)
  71. {
  72. struct chtls_sock *csk =
  73. container_of(ref, struct chtls_sock, kref);
  74. kfree(csk);
  75. }
  76. static struct net_device *chtls_ipv4_netdev(struct chtls_dev *cdev,
  77. struct sock *sk)
  78. {
  79. struct net_device *ndev = cdev->ports[0];
  80. if (likely(!inet_sk(sk)->inet_rcv_saddr))
  81. return ndev;
  82. ndev = ip_dev_find(&init_net, inet_sk(sk)->inet_rcv_saddr);
  83. if (!ndev)
  84. return NULL;
  85. if (is_vlan_dev(ndev))
  86. return vlan_dev_real_dev(ndev);
  87. return ndev;
  88. }
  89. static void assign_rxopt(struct sock *sk, unsigned int opt)
  90. {
  91. const struct chtls_dev *cdev;
  92. struct chtls_sock *csk;
  93. struct tcp_sock *tp;
  94. csk = rcu_dereference_sk_user_data(sk);
  95. tp = tcp_sk(sk);
  96. cdev = csk->cdev;
  97. tp->tcp_header_len = sizeof(struct tcphdr);
  98. tp->rx_opt.mss_clamp = cdev->mtus[TCPOPT_MSS_G(opt)] - 40;
  99. tp->mss_cache = tp->rx_opt.mss_clamp;
  100. tp->rx_opt.tstamp_ok = TCPOPT_TSTAMP_G(opt);
  101. tp->rx_opt.snd_wscale = TCPOPT_SACK_G(opt);
  102. tp->rx_opt.wscale_ok = TCPOPT_WSCALE_OK_G(opt);
  103. SND_WSCALE(tp) = TCPOPT_SND_WSCALE_G(opt);
  104. if (!tp->rx_opt.wscale_ok)
  105. tp->rx_opt.rcv_wscale = 0;
  106. if (tp->rx_opt.tstamp_ok) {
  107. tp->tcp_header_len += TCPOLEN_TSTAMP_ALIGNED;
  108. tp->rx_opt.mss_clamp -= TCPOLEN_TSTAMP_ALIGNED;
  109. } else if (csk->opt2 & TSTAMPS_EN_F) {
  110. csk->opt2 &= ~TSTAMPS_EN_F;
  111. csk->mtu_idx = TCPOPT_MSS_G(opt);
  112. }
  113. }
  114. static void chtls_purge_receive_queue(struct sock *sk)
  115. {
  116. struct sk_buff *skb;
  117. while ((skb = __skb_dequeue(&sk->sk_receive_queue)) != NULL) {
  118. skb_dst_set(skb, (void *)NULL);
  119. kfree_skb(skb);
  120. }
  121. }
  122. static void chtls_purge_write_queue(struct sock *sk)
  123. {
  124. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  125. struct sk_buff *skb;
  126. while ((skb = __skb_dequeue(&csk->txq))) {
  127. sk->sk_wmem_queued -= skb->truesize;
  128. __kfree_skb(skb);
  129. }
  130. }
  131. static void chtls_purge_recv_queue(struct sock *sk)
  132. {
  133. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  134. struct chtls_hws *tlsk = &csk->tlshws;
  135. struct sk_buff *skb;
  136. while ((skb = __skb_dequeue(&tlsk->sk_recv_queue)) != NULL) {
  137. skb_dst_set(skb, NULL);
  138. kfree_skb(skb);
  139. }
  140. }
  141. static void abort_arp_failure(void *handle, struct sk_buff *skb)
  142. {
  143. struct cpl_abort_req *req = cplhdr(skb);
  144. struct chtls_dev *cdev;
  145. cdev = (struct chtls_dev *)handle;
  146. req->cmd = CPL_ABORT_NO_RST;
  147. cxgb4_ofld_send(cdev->lldi->ports[0], skb);
  148. }
  149. static struct sk_buff *alloc_ctrl_skb(struct sk_buff *skb, int len)
  150. {
  151. if (likely(skb && !skb_shared(skb) && !skb_cloned(skb))) {
  152. __skb_trim(skb, 0);
  153. refcount_add(2, &skb->users);
  154. } else {
  155. skb = alloc_skb(len, GFP_KERNEL | __GFP_NOFAIL);
  156. }
  157. return skb;
  158. }
  159. static void chtls_send_abort(struct sock *sk, int mode, struct sk_buff *skb)
  160. {
  161. struct cpl_abort_req *req;
  162. struct chtls_sock *csk;
  163. struct tcp_sock *tp;
  164. csk = rcu_dereference_sk_user_data(sk);
  165. tp = tcp_sk(sk);
  166. if (!skb)
  167. skb = alloc_ctrl_skb(csk->txdata_skb_cache, sizeof(*req));
  168. req = (struct cpl_abort_req *)skb_put(skb, sizeof(*req));
  169. INIT_TP_WR_CPL(req, CPL_ABORT_REQ, csk->tid);
  170. skb_set_queue_mapping(skb, (csk->txq_idx << 1) | CPL_PRIORITY_DATA);
  171. req->rsvd0 = htonl(tp->snd_nxt);
  172. req->rsvd1 = !csk_flag_nochk(csk, CSK_TX_DATA_SENT);
  173. req->cmd = mode;
  174. t4_set_arp_err_handler(skb, csk->cdev, abort_arp_failure);
  175. send_or_defer(sk, tp, skb, mode == CPL_ABORT_SEND_RST);
  176. }
  177. static void chtls_send_reset(struct sock *sk, int mode, struct sk_buff *skb)
  178. {
  179. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  180. if (unlikely(csk_flag_nochk(csk, CSK_ABORT_SHUTDOWN) ||
  181. !csk->cdev)) {
  182. if (sk->sk_state == TCP_SYN_RECV)
  183. csk_set_flag(csk, CSK_RST_ABORTED);
  184. goto out;
  185. }
  186. if (!csk_flag_nochk(csk, CSK_TX_DATA_SENT)) {
  187. struct tcp_sock *tp = tcp_sk(sk);
  188. if (send_tx_flowc_wr(sk, 0, tp->snd_nxt, tp->rcv_nxt) < 0)
  189. WARN_ONCE(1, "send tx flowc error");
  190. csk_set_flag(csk, CSK_TX_DATA_SENT);
  191. }
  192. csk_set_flag(csk, CSK_ABORT_RPL_PENDING);
  193. chtls_purge_write_queue(sk);
  194. csk_set_flag(csk, CSK_ABORT_SHUTDOWN);
  195. if (sk->sk_state != TCP_SYN_RECV)
  196. chtls_send_abort(sk, mode, skb);
  197. else
  198. goto out;
  199. return;
  200. out:
  201. if (skb)
  202. kfree_skb(skb);
  203. }
  204. static void release_tcp_port(struct sock *sk)
  205. {
  206. if (inet_csk(sk)->icsk_bind_hash)
  207. inet_put_port(sk);
  208. }
  209. static void tcp_uncork(struct sock *sk)
  210. {
  211. struct tcp_sock *tp = tcp_sk(sk);
  212. if (tp->nonagle & TCP_NAGLE_CORK) {
  213. tp->nonagle &= ~TCP_NAGLE_CORK;
  214. chtls_tcp_push(sk, 0);
  215. }
  216. }
  217. static void chtls_close_conn(struct sock *sk)
  218. {
  219. struct cpl_close_con_req *req;
  220. struct chtls_sock *csk;
  221. struct sk_buff *skb;
  222. unsigned int tid;
  223. unsigned int len;
  224. len = roundup(sizeof(struct cpl_close_con_req), 16);
  225. csk = rcu_dereference_sk_user_data(sk);
  226. tid = csk->tid;
  227. skb = alloc_skb(len, GFP_KERNEL | __GFP_NOFAIL);
  228. req = (struct cpl_close_con_req *)__skb_put(skb, len);
  229. memset(req, 0, len);
  230. req->wr.wr_hi = htonl(FW_WR_OP_V(FW_TP_WR) |
  231. FW_WR_IMMDLEN_V(sizeof(*req) -
  232. sizeof(req->wr)));
  233. req->wr.wr_mid = htonl(FW_WR_LEN16_V(DIV_ROUND_UP(sizeof(*req), 16)) |
  234. FW_WR_FLOWID_V(tid));
  235. OPCODE_TID(req) = htonl(MK_OPCODE_TID(CPL_CLOSE_CON_REQ, tid));
  236. tcp_uncork(sk);
  237. skb_entail(sk, skb, ULPCB_FLAG_NO_HDR | ULPCB_FLAG_NO_APPEND);
  238. if (sk->sk_state != TCP_SYN_SENT)
  239. chtls_push_frames(csk, 1);
  240. }
  241. /*
  242. * Perform a state transition during close and return the actions indicated
  243. * for the transition. Do not make this function inline, the main reason
  244. * it exists at all is to avoid multiple inlining of tcp_set_state.
  245. */
  246. static int make_close_transition(struct sock *sk)
  247. {
  248. int next = (int)new_state[sk->sk_state];
  249. tcp_set_state(sk, next & TCP_STATE_MASK);
  250. return next & TCP_ACTION_FIN;
  251. }
  252. void chtls_close(struct sock *sk, long timeout)
  253. {
  254. int data_lost, prev_state;
  255. struct chtls_sock *csk;
  256. csk = rcu_dereference_sk_user_data(sk);
  257. lock_sock(sk);
  258. sk->sk_shutdown |= SHUTDOWN_MASK;
  259. data_lost = skb_queue_len(&sk->sk_receive_queue);
  260. data_lost |= skb_queue_len(&csk->tlshws.sk_recv_queue);
  261. chtls_purge_recv_queue(sk);
  262. chtls_purge_receive_queue(sk);
  263. if (sk->sk_state == TCP_CLOSE) {
  264. goto wait;
  265. } else if (data_lost || sk->sk_state == TCP_SYN_SENT) {
  266. chtls_send_reset(sk, CPL_ABORT_SEND_RST, NULL);
  267. release_tcp_port(sk);
  268. goto unlock;
  269. } else if (sock_flag(sk, SOCK_LINGER) && !sk->sk_lingertime) {
  270. sk->sk_prot->disconnect(sk, 0);
  271. } else if (make_close_transition(sk)) {
  272. chtls_close_conn(sk);
  273. }
  274. wait:
  275. if (timeout)
  276. sk_stream_wait_close(sk, timeout);
  277. unlock:
  278. prev_state = sk->sk_state;
  279. sock_hold(sk);
  280. sock_orphan(sk);
  281. release_sock(sk);
  282. local_bh_disable();
  283. bh_lock_sock(sk);
  284. if (prev_state != TCP_CLOSE && sk->sk_state == TCP_CLOSE)
  285. goto out;
  286. if (sk->sk_state == TCP_FIN_WAIT2 && tcp_sk(sk)->linger2 < 0 &&
  287. !csk_flag(sk, CSK_ABORT_SHUTDOWN)) {
  288. struct sk_buff *skb;
  289. skb = alloc_skb(sizeof(struct cpl_abort_req), GFP_ATOMIC);
  290. if (skb)
  291. chtls_send_reset(sk, CPL_ABORT_SEND_RST, skb);
  292. }
  293. if (sk->sk_state == TCP_CLOSE)
  294. inet_csk_destroy_sock(sk);
  295. out:
  296. bh_unlock_sock(sk);
  297. local_bh_enable();
  298. sock_put(sk);
  299. }
  300. /*
  301. * Wait until a socket enters on of the given states.
  302. */
  303. static int wait_for_states(struct sock *sk, unsigned int states)
  304. {
  305. DECLARE_WAITQUEUE(wait, current);
  306. struct socket_wq _sk_wq;
  307. long current_timeo;
  308. int err = 0;
  309. current_timeo = 200;
  310. /*
  311. * We want this to work even when there's no associated struct socket.
  312. * In that case we provide a temporary wait_queue_head_t.
  313. */
  314. if (!sk->sk_wq) {
  315. init_waitqueue_head(&_sk_wq.wait);
  316. _sk_wq.fasync_list = NULL;
  317. init_rcu_head_on_stack(&_sk_wq.rcu);
  318. RCU_INIT_POINTER(sk->sk_wq, &_sk_wq);
  319. }
  320. add_wait_queue(sk_sleep(sk), &wait);
  321. while (!sk_in_state(sk, states)) {
  322. if (!current_timeo) {
  323. err = -EBUSY;
  324. break;
  325. }
  326. if (signal_pending(current)) {
  327. err = sock_intr_errno(current_timeo);
  328. break;
  329. }
  330. set_current_state(TASK_UNINTERRUPTIBLE);
  331. release_sock(sk);
  332. if (!sk_in_state(sk, states))
  333. current_timeo = schedule_timeout(current_timeo);
  334. __set_current_state(TASK_RUNNING);
  335. lock_sock(sk);
  336. }
  337. remove_wait_queue(sk_sleep(sk), &wait);
  338. if (rcu_dereference(sk->sk_wq) == &_sk_wq)
  339. sk->sk_wq = NULL;
  340. return err;
  341. }
  342. int chtls_disconnect(struct sock *sk, int flags)
  343. {
  344. struct chtls_sock *csk;
  345. struct tcp_sock *tp;
  346. int err;
  347. tp = tcp_sk(sk);
  348. csk = rcu_dereference_sk_user_data(sk);
  349. chtls_purge_recv_queue(sk);
  350. chtls_purge_receive_queue(sk);
  351. chtls_purge_write_queue(sk);
  352. if (sk->sk_state != TCP_CLOSE) {
  353. sk->sk_err = ECONNRESET;
  354. chtls_send_reset(sk, CPL_ABORT_SEND_RST, NULL);
  355. err = wait_for_states(sk, TCPF_CLOSE);
  356. if (err)
  357. return err;
  358. }
  359. chtls_purge_recv_queue(sk);
  360. chtls_purge_receive_queue(sk);
  361. tp->max_window = 0xFFFF << (tp->rx_opt.snd_wscale);
  362. return tcp_disconnect(sk, flags);
  363. }
  364. #define SHUTDOWN_ELIGIBLE_STATE (TCPF_ESTABLISHED | \
  365. TCPF_SYN_RECV | TCPF_CLOSE_WAIT)
  366. void chtls_shutdown(struct sock *sk, int how)
  367. {
  368. if ((how & SEND_SHUTDOWN) &&
  369. sk_in_state(sk, SHUTDOWN_ELIGIBLE_STATE) &&
  370. make_close_transition(sk))
  371. chtls_close_conn(sk);
  372. }
  373. void chtls_destroy_sock(struct sock *sk)
  374. {
  375. struct chtls_sock *csk;
  376. csk = rcu_dereference_sk_user_data(sk);
  377. chtls_purge_recv_queue(sk);
  378. csk->ulp_mode = ULP_MODE_NONE;
  379. chtls_purge_write_queue(sk);
  380. free_tls_keyid(sk);
  381. kref_put(&csk->kref, chtls_sock_release);
  382. sk->sk_prot = &tcp_prot;
  383. sk->sk_prot->destroy(sk);
  384. }
  385. static void reset_listen_child(struct sock *child)
  386. {
  387. struct chtls_sock *csk = rcu_dereference_sk_user_data(child);
  388. struct sk_buff *skb;
  389. skb = alloc_ctrl_skb(csk->txdata_skb_cache,
  390. sizeof(struct cpl_abort_req));
  391. chtls_send_reset(child, CPL_ABORT_SEND_RST, skb);
  392. sock_orphan(child);
  393. INC_ORPHAN_COUNT(child);
  394. if (child->sk_state == TCP_CLOSE)
  395. inet_csk_destroy_sock(child);
  396. }
  397. static void chtls_disconnect_acceptq(struct sock *listen_sk)
  398. {
  399. struct request_sock **pprev;
  400. pprev = ACCEPT_QUEUE(listen_sk);
  401. while (*pprev) {
  402. struct request_sock *req = *pprev;
  403. if (req->rsk_ops == &chtls_rsk_ops) {
  404. struct sock *child = req->sk;
  405. *pprev = req->dl_next;
  406. sk_acceptq_removed(listen_sk);
  407. reqsk_put(req);
  408. sock_hold(child);
  409. local_bh_disable();
  410. bh_lock_sock(child);
  411. release_tcp_port(child);
  412. reset_listen_child(child);
  413. bh_unlock_sock(child);
  414. local_bh_enable();
  415. sock_put(child);
  416. } else {
  417. pprev = &req->dl_next;
  418. }
  419. }
  420. }
  421. static int listen_hashfn(const struct sock *sk)
  422. {
  423. return ((unsigned long)sk >> 10) & (LISTEN_INFO_HASH_SIZE - 1);
  424. }
  425. static struct listen_info *listen_hash_add(struct chtls_dev *cdev,
  426. struct sock *sk,
  427. unsigned int stid)
  428. {
  429. struct listen_info *p = kmalloc(sizeof(*p), GFP_KERNEL);
  430. if (p) {
  431. int key = listen_hashfn(sk);
  432. p->sk = sk;
  433. p->stid = stid;
  434. spin_lock(&cdev->listen_lock);
  435. p->next = cdev->listen_hash_tab[key];
  436. cdev->listen_hash_tab[key] = p;
  437. spin_unlock(&cdev->listen_lock);
  438. }
  439. return p;
  440. }
  441. static int listen_hash_find(struct chtls_dev *cdev,
  442. struct sock *sk)
  443. {
  444. struct listen_info *p;
  445. int stid = -1;
  446. int key;
  447. key = listen_hashfn(sk);
  448. spin_lock(&cdev->listen_lock);
  449. for (p = cdev->listen_hash_tab[key]; p; p = p->next)
  450. if (p->sk == sk) {
  451. stid = p->stid;
  452. break;
  453. }
  454. spin_unlock(&cdev->listen_lock);
  455. return stid;
  456. }
  457. static int listen_hash_del(struct chtls_dev *cdev,
  458. struct sock *sk)
  459. {
  460. struct listen_info *p, **prev;
  461. int stid = -1;
  462. int key;
  463. key = listen_hashfn(sk);
  464. prev = &cdev->listen_hash_tab[key];
  465. spin_lock(&cdev->listen_lock);
  466. for (p = *prev; p; prev = &p->next, p = p->next)
  467. if (p->sk == sk) {
  468. stid = p->stid;
  469. *prev = p->next;
  470. kfree(p);
  471. break;
  472. }
  473. spin_unlock(&cdev->listen_lock);
  474. return stid;
  475. }
  476. static void cleanup_syn_rcv_conn(struct sock *child, struct sock *parent)
  477. {
  478. struct request_sock *req;
  479. struct chtls_sock *csk;
  480. csk = rcu_dereference_sk_user_data(child);
  481. req = csk->passive_reap_next;
  482. reqsk_queue_removed(&inet_csk(parent)->icsk_accept_queue, req);
  483. __skb_unlink((struct sk_buff *)&csk->synq, &csk->listen_ctx->synq);
  484. chtls_reqsk_free(req);
  485. csk->passive_reap_next = NULL;
  486. }
  487. static void chtls_reset_synq(struct listen_ctx *listen_ctx)
  488. {
  489. struct sock *listen_sk = listen_ctx->lsk;
  490. while (!skb_queue_empty(&listen_ctx->synq)) {
  491. struct chtls_sock *csk =
  492. container_of((struct synq *)__skb_dequeue
  493. (&listen_ctx->synq), struct chtls_sock, synq);
  494. struct sock *child = csk->sk;
  495. cleanup_syn_rcv_conn(child, listen_sk);
  496. sock_hold(child);
  497. local_bh_disable();
  498. bh_lock_sock(child);
  499. release_tcp_port(child);
  500. reset_listen_child(child);
  501. bh_unlock_sock(child);
  502. local_bh_enable();
  503. sock_put(child);
  504. }
  505. }
  506. int chtls_listen_start(struct chtls_dev *cdev, struct sock *sk)
  507. {
  508. struct net_device *ndev;
  509. struct listen_ctx *ctx;
  510. struct adapter *adap;
  511. struct port_info *pi;
  512. int stid;
  513. int ret;
  514. if (sk->sk_family != PF_INET)
  515. return -EAGAIN;
  516. rcu_read_lock();
  517. ndev = chtls_ipv4_netdev(cdev, sk);
  518. rcu_read_unlock();
  519. if (!ndev)
  520. return -EBADF;
  521. pi = netdev_priv(ndev);
  522. adap = pi->adapter;
  523. if (!(adap->flags & FULL_INIT_DONE))
  524. return -EBADF;
  525. if (listen_hash_find(cdev, sk) >= 0) /* already have it */
  526. return -EADDRINUSE;
  527. ctx = kmalloc(sizeof(*ctx), GFP_KERNEL);
  528. if (!ctx)
  529. return -ENOMEM;
  530. __module_get(THIS_MODULE);
  531. ctx->lsk = sk;
  532. ctx->cdev = cdev;
  533. ctx->state = T4_LISTEN_START_PENDING;
  534. skb_queue_head_init(&ctx->synq);
  535. stid = cxgb4_alloc_stid(cdev->tids, sk->sk_family, ctx);
  536. if (stid < 0)
  537. goto free_ctx;
  538. sock_hold(sk);
  539. if (!listen_hash_add(cdev, sk, stid))
  540. goto free_stid;
  541. ret = cxgb4_create_server(ndev, stid,
  542. inet_sk(sk)->inet_rcv_saddr,
  543. inet_sk(sk)->inet_sport, 0,
  544. cdev->lldi->rxq_ids[0]);
  545. if (ret > 0)
  546. ret = net_xmit_errno(ret);
  547. if (ret)
  548. goto del_hash;
  549. return 0;
  550. del_hash:
  551. listen_hash_del(cdev, sk);
  552. free_stid:
  553. cxgb4_free_stid(cdev->tids, stid, sk->sk_family);
  554. sock_put(sk);
  555. free_ctx:
  556. kfree(ctx);
  557. module_put(THIS_MODULE);
  558. return -EBADF;
  559. }
  560. void chtls_listen_stop(struct chtls_dev *cdev, struct sock *sk)
  561. {
  562. struct listen_ctx *listen_ctx;
  563. int stid;
  564. stid = listen_hash_del(cdev, sk);
  565. if (stid < 0)
  566. return;
  567. listen_ctx = (struct listen_ctx *)lookup_stid(cdev->tids, stid);
  568. chtls_reset_synq(listen_ctx);
  569. cxgb4_remove_server(cdev->lldi->ports[0], stid,
  570. cdev->lldi->rxq_ids[0], 0);
  571. chtls_disconnect_acceptq(sk);
  572. }
  573. static int chtls_pass_open_rpl(struct chtls_dev *cdev, struct sk_buff *skb)
  574. {
  575. struct cpl_pass_open_rpl *rpl = cplhdr(skb) + RSS_HDR;
  576. unsigned int stid = GET_TID(rpl);
  577. struct listen_ctx *listen_ctx;
  578. listen_ctx = (struct listen_ctx *)lookup_stid(cdev->tids, stid);
  579. if (!listen_ctx)
  580. return CPL_RET_BUF_DONE;
  581. if (listen_ctx->state == T4_LISTEN_START_PENDING) {
  582. listen_ctx->state = T4_LISTEN_STARTED;
  583. return CPL_RET_BUF_DONE;
  584. }
  585. if (rpl->status != CPL_ERR_NONE) {
  586. pr_info("Unexpected PASS_OPEN_RPL status %u for STID %u\n",
  587. rpl->status, stid);
  588. return CPL_RET_BUF_DONE;
  589. }
  590. cxgb4_free_stid(cdev->tids, stid, listen_ctx->lsk->sk_family);
  591. sock_put(listen_ctx->lsk);
  592. kfree(listen_ctx);
  593. module_put(THIS_MODULE);
  594. return 0;
  595. }
  596. static int chtls_close_listsrv_rpl(struct chtls_dev *cdev, struct sk_buff *skb)
  597. {
  598. struct cpl_close_listsvr_rpl *rpl = cplhdr(skb) + RSS_HDR;
  599. struct listen_ctx *listen_ctx;
  600. unsigned int stid;
  601. void *data;
  602. stid = GET_TID(rpl);
  603. data = lookup_stid(cdev->tids, stid);
  604. listen_ctx = (struct listen_ctx *)data;
  605. if (rpl->status != CPL_ERR_NONE) {
  606. pr_info("Unexpected CLOSE_LISTSRV_RPL status %u for STID %u\n",
  607. rpl->status, stid);
  608. return CPL_RET_BUF_DONE;
  609. }
  610. cxgb4_free_stid(cdev->tids, stid, listen_ctx->lsk->sk_family);
  611. sock_put(listen_ctx->lsk);
  612. kfree(listen_ctx);
  613. module_put(THIS_MODULE);
  614. return 0;
  615. }
  616. static void chtls_release_resources(struct sock *sk)
  617. {
  618. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  619. struct chtls_dev *cdev = csk->cdev;
  620. unsigned int tid = csk->tid;
  621. struct tid_info *tids;
  622. if (!cdev)
  623. return;
  624. tids = cdev->tids;
  625. kfree_skb(csk->txdata_skb_cache);
  626. csk->txdata_skb_cache = NULL;
  627. if (csk->l2t_entry) {
  628. cxgb4_l2t_release(csk->l2t_entry);
  629. csk->l2t_entry = NULL;
  630. }
  631. cxgb4_remove_tid(tids, csk->port_id, tid, sk->sk_family);
  632. sock_put(sk);
  633. }
  634. static void chtls_conn_done(struct sock *sk)
  635. {
  636. if (sock_flag(sk, SOCK_DEAD))
  637. chtls_purge_receive_queue(sk);
  638. sk_wakeup_sleepers(sk, 0);
  639. tcp_done(sk);
  640. }
  641. static void do_abort_syn_rcv(struct sock *child, struct sock *parent)
  642. {
  643. /*
  644. * If the server is still open we clean up the child connection,
  645. * otherwise the server already did the clean up as it was purging
  646. * its SYN queue and the skb was just sitting in its backlog.
  647. */
  648. if (likely(parent->sk_state == TCP_LISTEN)) {
  649. cleanup_syn_rcv_conn(child, parent);
  650. /* Without the below call to sock_orphan,
  651. * we leak the socket resource with syn_flood test
  652. * as inet_csk_destroy_sock will not be called
  653. * in tcp_done since SOCK_DEAD flag is not set.
  654. * Kernel handles this differently where new socket is
  655. * created only after 3 way handshake is done.
  656. */
  657. sock_orphan(child);
  658. percpu_counter_inc((child)->sk_prot->orphan_count);
  659. chtls_release_resources(child);
  660. chtls_conn_done(child);
  661. } else {
  662. if (csk_flag(child, CSK_RST_ABORTED)) {
  663. chtls_release_resources(child);
  664. chtls_conn_done(child);
  665. }
  666. }
  667. }
  668. static void pass_open_abort(struct sock *child, struct sock *parent,
  669. struct sk_buff *skb)
  670. {
  671. do_abort_syn_rcv(child, parent);
  672. kfree_skb(skb);
  673. }
  674. static void bl_pass_open_abort(struct sock *lsk, struct sk_buff *skb)
  675. {
  676. pass_open_abort(skb->sk, lsk, skb);
  677. }
  678. static void chtls_pass_open_arp_failure(struct sock *sk,
  679. struct sk_buff *skb)
  680. {
  681. const struct request_sock *oreq;
  682. struct chtls_sock *csk;
  683. struct chtls_dev *cdev;
  684. struct sock *parent;
  685. void *data;
  686. csk = rcu_dereference_sk_user_data(sk);
  687. cdev = csk->cdev;
  688. /*
  689. * If the connection is being aborted due to the parent listening
  690. * socket going away there's nothing to do, the ABORT_REQ will close
  691. * the connection.
  692. */
  693. if (csk_flag(sk, CSK_ABORT_RPL_PENDING)) {
  694. kfree_skb(skb);
  695. return;
  696. }
  697. oreq = csk->passive_reap_next;
  698. data = lookup_stid(cdev->tids, oreq->ts_recent);
  699. parent = ((struct listen_ctx *)data)->lsk;
  700. bh_lock_sock(parent);
  701. if (!sock_owned_by_user(parent)) {
  702. pass_open_abort(sk, parent, skb);
  703. } else {
  704. BLOG_SKB_CB(skb)->backlog_rcv = bl_pass_open_abort;
  705. __sk_add_backlog(parent, skb);
  706. }
  707. bh_unlock_sock(parent);
  708. }
  709. static void chtls_accept_rpl_arp_failure(void *handle,
  710. struct sk_buff *skb)
  711. {
  712. struct sock *sk = (struct sock *)handle;
  713. sock_hold(sk);
  714. process_cpl_msg(chtls_pass_open_arp_failure, sk, skb);
  715. sock_put(sk);
  716. }
  717. static unsigned int chtls_select_mss(const struct chtls_sock *csk,
  718. unsigned int pmtu,
  719. struct cpl_pass_accept_req *req)
  720. {
  721. struct chtls_dev *cdev;
  722. struct dst_entry *dst;
  723. unsigned int tcpoptsz;
  724. unsigned int iphdrsz;
  725. unsigned int mtu_idx;
  726. struct tcp_sock *tp;
  727. unsigned int mss;
  728. struct sock *sk;
  729. mss = ntohs(req->tcpopt.mss);
  730. sk = csk->sk;
  731. dst = __sk_dst_get(sk);
  732. cdev = csk->cdev;
  733. tp = tcp_sk(sk);
  734. tcpoptsz = 0;
  735. iphdrsz = sizeof(struct iphdr) + sizeof(struct tcphdr);
  736. if (req->tcpopt.tstamp)
  737. tcpoptsz += round_up(TCPOLEN_TIMESTAMP, 4);
  738. tp->advmss = dst_metric_advmss(dst);
  739. if (USER_MSS(tp) && tp->advmss > USER_MSS(tp))
  740. tp->advmss = USER_MSS(tp);
  741. if (tp->advmss > pmtu - iphdrsz)
  742. tp->advmss = pmtu - iphdrsz;
  743. if (mss && tp->advmss > mss)
  744. tp->advmss = mss;
  745. tp->advmss = cxgb4_best_aligned_mtu(cdev->lldi->mtus,
  746. iphdrsz + tcpoptsz,
  747. tp->advmss - tcpoptsz,
  748. 8, &mtu_idx);
  749. tp->advmss -= iphdrsz;
  750. inet_csk(sk)->icsk_pmtu_cookie = pmtu;
  751. return mtu_idx;
  752. }
  753. static unsigned int select_rcv_wnd(struct chtls_sock *csk)
  754. {
  755. unsigned int rcvwnd;
  756. unsigned int wnd;
  757. struct sock *sk;
  758. sk = csk->sk;
  759. wnd = tcp_full_space(sk);
  760. if (wnd < MIN_RCV_WND)
  761. wnd = MIN_RCV_WND;
  762. rcvwnd = MAX_RCV_WND;
  763. csk_set_flag(csk, CSK_UPDATE_RCV_WND);
  764. return min(wnd, rcvwnd);
  765. }
  766. static unsigned int select_rcv_wscale(int space, int wscale_ok, int win_clamp)
  767. {
  768. int wscale = 0;
  769. if (space > MAX_RCV_WND)
  770. space = MAX_RCV_WND;
  771. if (win_clamp && win_clamp < space)
  772. space = win_clamp;
  773. if (wscale_ok) {
  774. while (wscale < 14 && (65535 << wscale) < space)
  775. wscale++;
  776. }
  777. return wscale;
  778. }
  779. static void chtls_pass_accept_rpl(struct sk_buff *skb,
  780. struct cpl_pass_accept_req *req,
  781. unsigned int tid)
  782. {
  783. struct cpl_t5_pass_accept_rpl *rpl5;
  784. struct cxgb4_lld_info *lldi;
  785. const struct tcphdr *tcph;
  786. const struct tcp_sock *tp;
  787. struct chtls_sock *csk;
  788. unsigned int len;
  789. struct sock *sk;
  790. u32 opt2, hlen;
  791. u64 opt0;
  792. sk = skb->sk;
  793. tp = tcp_sk(sk);
  794. csk = sk->sk_user_data;
  795. csk->tid = tid;
  796. lldi = csk->cdev->lldi;
  797. len = roundup(sizeof(*rpl5), 16);
  798. rpl5 = __skb_put_zero(skb, len);
  799. INIT_TP_WR(rpl5, tid);
  800. OPCODE_TID(rpl5) = cpu_to_be32(MK_OPCODE_TID(CPL_PASS_ACCEPT_RPL,
  801. csk->tid));
  802. csk->mtu_idx = chtls_select_mss(csk, dst_mtu(__sk_dst_get(sk)),
  803. req);
  804. opt0 = TCAM_BYPASS_F |
  805. WND_SCALE_V((tp)->rx_opt.rcv_wscale) |
  806. MSS_IDX_V(csk->mtu_idx) |
  807. L2T_IDX_V(csk->l2t_entry->idx) |
  808. NAGLE_V(!(tp->nonagle & TCP_NAGLE_OFF)) |
  809. TX_CHAN_V(csk->tx_chan) |
  810. SMAC_SEL_V(csk->smac_idx) |
  811. DSCP_V(csk->tos >> 2) |
  812. ULP_MODE_V(ULP_MODE_TLS) |
  813. RCV_BUFSIZ_V(min(tp->rcv_wnd >> 10, RCV_BUFSIZ_M));
  814. opt2 = RX_CHANNEL_V(0) |
  815. RSS_QUEUE_VALID_F | RSS_QUEUE_V(csk->rss_qid);
  816. if (!is_t5(lldi->adapter_type))
  817. opt2 |= RX_FC_DISABLE_F;
  818. if (req->tcpopt.tstamp)
  819. opt2 |= TSTAMPS_EN_F;
  820. if (req->tcpopt.sack)
  821. opt2 |= SACK_EN_F;
  822. hlen = ntohl(req->hdr_len);
  823. tcph = (struct tcphdr *)((u8 *)(req + 1) +
  824. T6_ETH_HDR_LEN_G(hlen) + T6_IP_HDR_LEN_G(hlen));
  825. if (tcph->ece && tcph->cwr)
  826. opt2 |= CCTRL_ECN_V(1);
  827. opt2 |= CONG_CNTRL_V(CONG_ALG_NEWRENO);
  828. opt2 |= T5_ISS_F;
  829. opt2 |= T5_OPT_2_VALID_F;
  830. rpl5->opt0 = cpu_to_be64(opt0);
  831. rpl5->opt2 = cpu_to_be32(opt2);
  832. rpl5->iss = cpu_to_be32((prandom_u32() & ~7UL) - 1);
  833. set_wr_txq(skb, CPL_PRIORITY_SETUP, csk->port_id);
  834. t4_set_arp_err_handler(skb, sk, chtls_accept_rpl_arp_failure);
  835. cxgb4_l2t_send(csk->egress_dev, skb, csk->l2t_entry);
  836. }
  837. static void inet_inherit_port(struct inet_hashinfo *hash_info,
  838. struct sock *lsk, struct sock *newsk)
  839. {
  840. local_bh_disable();
  841. __inet_inherit_port(lsk, newsk);
  842. local_bh_enable();
  843. }
  844. static int chtls_backlog_rcv(struct sock *sk, struct sk_buff *skb)
  845. {
  846. if (skb->protocol) {
  847. kfree_skb(skb);
  848. return 0;
  849. }
  850. BLOG_SKB_CB(skb)->backlog_rcv(sk, skb);
  851. return 0;
  852. }
  853. static struct sock *chtls_recv_sock(struct sock *lsk,
  854. struct request_sock *oreq,
  855. void *network_hdr,
  856. const struct cpl_pass_accept_req *req,
  857. struct chtls_dev *cdev)
  858. {
  859. const struct tcphdr *tcph;
  860. struct inet_sock *newinet;
  861. const struct iphdr *iph;
  862. struct net_device *ndev;
  863. struct chtls_sock *csk;
  864. struct dst_entry *dst;
  865. struct neighbour *n;
  866. struct tcp_sock *tp;
  867. struct sock *newsk;
  868. u16 port_id;
  869. int rxq_idx;
  870. int step;
  871. iph = (const struct iphdr *)network_hdr;
  872. newsk = tcp_create_openreq_child(lsk, oreq, cdev->askb);
  873. if (!newsk)
  874. goto free_oreq;
  875. dst = inet_csk_route_child_sock(lsk, newsk, oreq);
  876. if (!dst)
  877. goto free_sk;
  878. tcph = (struct tcphdr *)(iph + 1);
  879. n = dst_neigh_lookup(dst, &iph->saddr);
  880. if (!n)
  881. goto free_sk;
  882. ndev = n->dev;
  883. if (!ndev)
  884. goto free_dst;
  885. port_id = cxgb4_port_idx(ndev);
  886. csk = chtls_sock_create(cdev);
  887. if (!csk)
  888. goto free_dst;
  889. csk->l2t_entry = cxgb4_l2t_get(cdev->lldi->l2t, n, ndev, 0);
  890. if (!csk->l2t_entry)
  891. goto free_csk;
  892. newsk->sk_user_data = csk;
  893. newsk->sk_backlog_rcv = chtls_backlog_rcv;
  894. tp = tcp_sk(newsk);
  895. newinet = inet_sk(newsk);
  896. newinet->inet_daddr = iph->saddr;
  897. newinet->inet_rcv_saddr = iph->daddr;
  898. newinet->inet_saddr = iph->daddr;
  899. oreq->ts_recent = PASS_OPEN_TID_G(ntohl(req->tos_stid));
  900. sk_setup_caps(newsk, dst);
  901. csk->sk = newsk;
  902. csk->passive_reap_next = oreq;
  903. csk->tx_chan = cxgb4_port_chan(ndev);
  904. csk->port_id = port_id;
  905. csk->egress_dev = ndev;
  906. csk->tos = PASS_OPEN_TOS_G(ntohl(req->tos_stid));
  907. csk->ulp_mode = ULP_MODE_TLS;
  908. step = cdev->lldi->nrxq / cdev->lldi->nchan;
  909. csk->rss_qid = cdev->lldi->rxq_ids[port_id * step];
  910. rxq_idx = port_id * step;
  911. csk->txq_idx = (rxq_idx < cdev->lldi->ntxq) ? rxq_idx :
  912. port_id * step;
  913. csk->sndbuf = newsk->sk_sndbuf;
  914. csk->smac_idx = cxgb4_tp_smt_idx(cdev->lldi->adapter_type,
  915. cxgb4_port_viid(ndev));
  916. tp->rcv_wnd = select_rcv_wnd(csk);
  917. RCV_WSCALE(tp) = select_rcv_wscale(tcp_full_space(newsk),
  918. WSCALE_OK(tp),
  919. tp->window_clamp);
  920. neigh_release(n);
  921. inet_inherit_port(&tcp_hashinfo, lsk, newsk);
  922. csk_set_flag(csk, CSK_CONN_INLINE);
  923. bh_unlock_sock(newsk); /* tcp_create_openreq_child ->sk_clone_lock */
  924. return newsk;
  925. free_csk:
  926. chtls_sock_release(&csk->kref);
  927. free_dst:
  928. dst_release(dst);
  929. free_sk:
  930. inet_csk_prepare_forced_close(newsk);
  931. tcp_done(newsk);
  932. free_oreq:
  933. chtls_reqsk_free(oreq);
  934. return NULL;
  935. }
  936. /*
  937. * Populate a TID_RELEASE WR. The skb must be already propely sized.
  938. */
  939. static void mk_tid_release(struct sk_buff *skb,
  940. unsigned int chan, unsigned int tid)
  941. {
  942. struct cpl_tid_release *req;
  943. unsigned int len;
  944. len = roundup(sizeof(struct cpl_tid_release), 16);
  945. req = (struct cpl_tid_release *)__skb_put(skb, len);
  946. memset(req, 0, len);
  947. set_wr_txq(skb, CPL_PRIORITY_SETUP, chan);
  948. INIT_TP_WR_CPL(req, CPL_TID_RELEASE, tid);
  949. }
  950. static int chtls_get_module(struct sock *sk)
  951. {
  952. struct inet_connection_sock *icsk = inet_csk(sk);
  953. if (!try_module_get(icsk->icsk_ulp_ops->owner))
  954. return -1;
  955. return 0;
  956. }
  957. static void chtls_pass_accept_request(struct sock *sk,
  958. struct sk_buff *skb)
  959. {
  960. struct cpl_t5_pass_accept_rpl *rpl;
  961. struct cpl_pass_accept_req *req;
  962. struct listen_ctx *listen_ctx;
  963. struct request_sock *oreq;
  964. struct sk_buff *reply_skb;
  965. struct chtls_sock *csk;
  966. struct chtls_dev *cdev;
  967. struct tcphdr *tcph;
  968. struct sock *newsk;
  969. struct ethhdr *eh;
  970. struct iphdr *iph;
  971. void *network_hdr;
  972. unsigned int stid;
  973. unsigned int len;
  974. unsigned int tid;
  975. req = cplhdr(skb) + RSS_HDR;
  976. tid = GET_TID(req);
  977. cdev = BLOG_SKB_CB(skb)->cdev;
  978. newsk = lookup_tid(cdev->tids, tid);
  979. stid = PASS_OPEN_TID_G(ntohl(req->tos_stid));
  980. if (newsk) {
  981. pr_info("tid (%d) already in use\n", tid);
  982. return;
  983. }
  984. len = roundup(sizeof(*rpl), 16);
  985. reply_skb = alloc_skb(len, GFP_ATOMIC);
  986. if (!reply_skb) {
  987. cxgb4_remove_tid(cdev->tids, 0, tid, sk->sk_family);
  988. kfree_skb(skb);
  989. return;
  990. }
  991. if (sk->sk_state != TCP_LISTEN)
  992. goto reject;
  993. if (inet_csk_reqsk_queue_is_full(sk))
  994. goto reject;
  995. if (sk_acceptq_is_full(sk))
  996. goto reject;
  997. oreq = inet_reqsk_alloc(&chtls_rsk_ops, sk, true);
  998. if (!oreq)
  999. goto reject;
  1000. oreq->rsk_rcv_wnd = 0;
  1001. oreq->rsk_window_clamp = 0;
  1002. oreq->cookie_ts = 0;
  1003. oreq->mss = 0;
  1004. oreq->ts_recent = 0;
  1005. eh = (struct ethhdr *)(req + 1);
  1006. iph = (struct iphdr *)(eh + 1);
  1007. if (iph->version != 0x4)
  1008. goto free_oreq;
  1009. network_hdr = (void *)(eh + 1);
  1010. tcph = (struct tcphdr *)(iph + 1);
  1011. tcp_rsk(oreq)->tfo_listener = false;
  1012. tcp_rsk(oreq)->rcv_isn = ntohl(tcph->seq);
  1013. chtls_set_req_port(oreq, tcph->source, tcph->dest);
  1014. inet_rsk(oreq)->ecn_ok = 0;
  1015. chtls_set_req_addr(oreq, iph->daddr, iph->saddr);
  1016. if (req->tcpopt.wsf <= 14) {
  1017. inet_rsk(oreq)->wscale_ok = 1;
  1018. inet_rsk(oreq)->snd_wscale = req->tcpopt.wsf;
  1019. }
  1020. inet_rsk(oreq)->ir_iif = sk->sk_bound_dev_if;
  1021. newsk = chtls_recv_sock(sk, oreq, network_hdr, req, cdev);
  1022. if (!newsk)
  1023. goto reject;
  1024. if (chtls_get_module(newsk))
  1025. goto reject;
  1026. inet_csk_reqsk_queue_added(sk);
  1027. reply_skb->sk = newsk;
  1028. chtls_install_cpl_ops(newsk);
  1029. cxgb4_insert_tid(cdev->tids, newsk, tid, newsk->sk_family);
  1030. csk = rcu_dereference_sk_user_data(newsk);
  1031. listen_ctx = (struct listen_ctx *)lookup_stid(cdev->tids, stid);
  1032. csk->listen_ctx = listen_ctx;
  1033. __skb_queue_tail(&listen_ctx->synq, (struct sk_buff *)&csk->synq);
  1034. chtls_pass_accept_rpl(reply_skb, req, tid);
  1035. kfree_skb(skb);
  1036. return;
  1037. free_oreq:
  1038. chtls_reqsk_free(oreq);
  1039. reject:
  1040. mk_tid_release(reply_skb, 0, tid);
  1041. cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
  1042. kfree_skb(skb);
  1043. }
  1044. /*
  1045. * Handle a CPL_PASS_ACCEPT_REQ message.
  1046. */
  1047. static int chtls_pass_accept_req(struct chtls_dev *cdev, struct sk_buff *skb)
  1048. {
  1049. struct cpl_pass_accept_req *req = cplhdr(skb) + RSS_HDR;
  1050. struct listen_ctx *ctx;
  1051. unsigned int stid;
  1052. unsigned int tid;
  1053. struct sock *lsk;
  1054. void *data;
  1055. stid = PASS_OPEN_TID_G(ntohl(req->tos_stid));
  1056. tid = GET_TID(req);
  1057. data = lookup_stid(cdev->tids, stid);
  1058. if (!data)
  1059. return 1;
  1060. ctx = (struct listen_ctx *)data;
  1061. lsk = ctx->lsk;
  1062. if (unlikely(tid >= cdev->tids->ntids)) {
  1063. pr_info("passive open TID %u too large\n", tid);
  1064. return 1;
  1065. }
  1066. BLOG_SKB_CB(skb)->cdev = cdev;
  1067. process_cpl_msg(chtls_pass_accept_request, lsk, skb);
  1068. return 0;
  1069. }
  1070. /*
  1071. * Completes some final bits of initialization for just established connections
  1072. * and changes their state to TCP_ESTABLISHED.
  1073. *
  1074. * snd_isn here is the ISN after the SYN, i.e., the true ISN + 1.
  1075. */
  1076. static void make_established(struct sock *sk, u32 snd_isn, unsigned int opt)
  1077. {
  1078. struct tcp_sock *tp = tcp_sk(sk);
  1079. tp->pushed_seq = snd_isn;
  1080. tp->write_seq = snd_isn;
  1081. tp->snd_nxt = snd_isn;
  1082. tp->snd_una = snd_isn;
  1083. inet_sk(sk)->inet_id = tp->write_seq ^ jiffies;
  1084. assign_rxopt(sk, opt);
  1085. if (tp->rcv_wnd > (RCV_BUFSIZ_M << 10))
  1086. tp->rcv_wup -= tp->rcv_wnd - (RCV_BUFSIZ_M << 10);
  1087. smp_mb();
  1088. tcp_set_state(sk, TCP_ESTABLISHED);
  1089. }
  1090. static void chtls_abort_conn(struct sock *sk, struct sk_buff *skb)
  1091. {
  1092. struct sk_buff *abort_skb;
  1093. abort_skb = alloc_skb(sizeof(struct cpl_abort_req), GFP_ATOMIC);
  1094. if (abort_skb)
  1095. chtls_send_reset(sk, CPL_ABORT_SEND_RST, abort_skb);
  1096. }
  1097. static struct sock *reap_list;
  1098. static DEFINE_SPINLOCK(reap_list_lock);
  1099. /*
  1100. * Process the reap list.
  1101. */
  1102. DECLARE_TASK_FUNC(process_reap_list, task_param)
  1103. {
  1104. spin_lock_bh(&reap_list_lock);
  1105. while (reap_list) {
  1106. struct sock *sk = reap_list;
  1107. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  1108. reap_list = csk->passive_reap_next;
  1109. csk->passive_reap_next = NULL;
  1110. spin_unlock(&reap_list_lock);
  1111. sock_hold(sk);
  1112. bh_lock_sock(sk);
  1113. chtls_abort_conn(sk, NULL);
  1114. sock_orphan(sk);
  1115. if (sk->sk_state == TCP_CLOSE)
  1116. inet_csk_destroy_sock(sk);
  1117. bh_unlock_sock(sk);
  1118. sock_put(sk);
  1119. spin_lock(&reap_list_lock);
  1120. }
  1121. spin_unlock_bh(&reap_list_lock);
  1122. }
  1123. static DECLARE_WORK(reap_task, process_reap_list);
  1124. static void add_to_reap_list(struct sock *sk)
  1125. {
  1126. struct chtls_sock *csk = sk->sk_user_data;
  1127. local_bh_disable();
  1128. bh_lock_sock(sk);
  1129. release_tcp_port(sk); /* release the port immediately */
  1130. spin_lock(&reap_list_lock);
  1131. csk->passive_reap_next = reap_list;
  1132. reap_list = sk;
  1133. if (!csk->passive_reap_next)
  1134. schedule_work(&reap_task);
  1135. spin_unlock(&reap_list_lock);
  1136. bh_unlock_sock(sk);
  1137. local_bh_enable();
  1138. }
  1139. static void add_pass_open_to_parent(struct sock *child, struct sock *lsk,
  1140. struct chtls_dev *cdev)
  1141. {
  1142. struct request_sock *oreq;
  1143. struct chtls_sock *csk;
  1144. if (lsk->sk_state != TCP_LISTEN)
  1145. return;
  1146. csk = child->sk_user_data;
  1147. oreq = csk->passive_reap_next;
  1148. csk->passive_reap_next = NULL;
  1149. reqsk_queue_removed(&inet_csk(lsk)->icsk_accept_queue, oreq);
  1150. __skb_unlink((struct sk_buff *)&csk->synq, &csk->listen_ctx->synq);
  1151. if (sk_acceptq_is_full(lsk)) {
  1152. chtls_reqsk_free(oreq);
  1153. add_to_reap_list(child);
  1154. } else {
  1155. refcount_set(&oreq->rsk_refcnt, 1);
  1156. inet_csk_reqsk_queue_add(lsk, oreq, child);
  1157. lsk->sk_data_ready(lsk);
  1158. }
  1159. }
  1160. static void bl_add_pass_open_to_parent(struct sock *lsk, struct sk_buff *skb)
  1161. {
  1162. struct sock *child = skb->sk;
  1163. skb->sk = NULL;
  1164. add_pass_open_to_parent(child, lsk, BLOG_SKB_CB(skb)->cdev);
  1165. kfree_skb(skb);
  1166. }
  1167. static int chtls_pass_establish(struct chtls_dev *cdev, struct sk_buff *skb)
  1168. {
  1169. struct cpl_pass_establish *req = cplhdr(skb) + RSS_HDR;
  1170. struct chtls_sock *csk;
  1171. struct sock *lsk, *sk;
  1172. unsigned int hwtid;
  1173. hwtid = GET_TID(req);
  1174. sk = lookup_tid(cdev->tids, hwtid);
  1175. if (!sk)
  1176. return (CPL_RET_UNKNOWN_TID | CPL_RET_BUF_DONE);
  1177. bh_lock_sock(sk);
  1178. if (unlikely(sock_owned_by_user(sk))) {
  1179. kfree_skb(skb);
  1180. } else {
  1181. unsigned int stid;
  1182. void *data;
  1183. csk = sk->sk_user_data;
  1184. csk->wr_max_credits = 64;
  1185. csk->wr_credits = 64;
  1186. csk->wr_unacked = 0;
  1187. make_established(sk, ntohl(req->snd_isn), ntohs(req->tcp_opt));
  1188. stid = PASS_OPEN_TID_G(ntohl(req->tos_stid));
  1189. sk->sk_state_change(sk);
  1190. if (unlikely(sk->sk_socket))
  1191. sk_wake_async(sk, 0, POLL_OUT);
  1192. data = lookup_stid(cdev->tids, stid);
  1193. lsk = ((struct listen_ctx *)data)->lsk;
  1194. bh_lock_sock(lsk);
  1195. if (unlikely(skb_queue_empty(&csk->listen_ctx->synq))) {
  1196. /* removed from synq */
  1197. bh_unlock_sock(lsk);
  1198. kfree_skb(skb);
  1199. goto unlock;
  1200. }
  1201. if (likely(!sock_owned_by_user(lsk))) {
  1202. kfree_skb(skb);
  1203. add_pass_open_to_parent(sk, lsk, cdev);
  1204. } else {
  1205. skb->sk = sk;
  1206. BLOG_SKB_CB(skb)->cdev = cdev;
  1207. BLOG_SKB_CB(skb)->backlog_rcv =
  1208. bl_add_pass_open_to_parent;
  1209. __sk_add_backlog(lsk, skb);
  1210. }
  1211. bh_unlock_sock(lsk);
  1212. }
  1213. unlock:
  1214. bh_unlock_sock(sk);
  1215. return 0;
  1216. }
  1217. /*
  1218. * Handle receipt of an urgent pointer.
  1219. */
  1220. static void handle_urg_ptr(struct sock *sk, u32 urg_seq)
  1221. {
  1222. struct tcp_sock *tp = tcp_sk(sk);
  1223. urg_seq--;
  1224. if (tp->urg_data && !after(urg_seq, tp->urg_seq))
  1225. return; /* duplicate pointer */
  1226. sk_send_sigurg(sk);
  1227. if (tp->urg_seq == tp->copied_seq && tp->urg_data &&
  1228. !sock_flag(sk, SOCK_URGINLINE) &&
  1229. tp->copied_seq != tp->rcv_nxt) {
  1230. struct sk_buff *skb = skb_peek(&sk->sk_receive_queue);
  1231. tp->copied_seq++;
  1232. if (skb && tp->copied_seq - ULP_SKB_CB(skb)->seq >= skb->len)
  1233. chtls_free_skb(sk, skb);
  1234. }
  1235. tp->urg_data = TCP_URG_NOTYET;
  1236. tp->urg_seq = urg_seq;
  1237. }
  1238. static void check_sk_callbacks(struct chtls_sock *csk)
  1239. {
  1240. struct sock *sk = csk->sk;
  1241. if (unlikely(sk->sk_user_data &&
  1242. !csk_flag_nochk(csk, CSK_CALLBACKS_CHKD)))
  1243. csk_set_flag(csk, CSK_CALLBACKS_CHKD);
  1244. }
  1245. /*
  1246. * Handles Rx data that arrives in a state where the socket isn't accepting
  1247. * new data.
  1248. */
  1249. static void handle_excess_rx(struct sock *sk, struct sk_buff *skb)
  1250. {
  1251. if (!csk_flag(sk, CSK_ABORT_SHUTDOWN))
  1252. chtls_abort_conn(sk, skb);
  1253. kfree_skb(skb);
  1254. }
  1255. static void chtls_recv_data(struct sock *sk, struct sk_buff *skb)
  1256. {
  1257. struct cpl_rx_data *hdr = cplhdr(skb) + RSS_HDR;
  1258. struct chtls_sock *csk;
  1259. struct tcp_sock *tp;
  1260. csk = rcu_dereference_sk_user_data(sk);
  1261. tp = tcp_sk(sk);
  1262. if (unlikely(sk->sk_shutdown & RCV_SHUTDOWN)) {
  1263. handle_excess_rx(sk, skb);
  1264. return;
  1265. }
  1266. ULP_SKB_CB(skb)->seq = ntohl(hdr->seq);
  1267. ULP_SKB_CB(skb)->psh = hdr->psh;
  1268. skb_ulp_mode(skb) = ULP_MODE_NONE;
  1269. skb_reset_transport_header(skb);
  1270. __skb_pull(skb, sizeof(*hdr) + RSS_HDR);
  1271. if (!skb->data_len)
  1272. __skb_trim(skb, ntohs(hdr->len));
  1273. if (unlikely(hdr->urg))
  1274. handle_urg_ptr(sk, tp->rcv_nxt + ntohs(hdr->urg));
  1275. if (unlikely(tp->urg_data == TCP_URG_NOTYET &&
  1276. tp->urg_seq - tp->rcv_nxt < skb->len))
  1277. tp->urg_data = TCP_URG_VALID |
  1278. skb->data[tp->urg_seq - tp->rcv_nxt];
  1279. if (unlikely(hdr->dack_mode != csk->delack_mode)) {
  1280. csk->delack_mode = hdr->dack_mode;
  1281. csk->delack_seq = tp->rcv_nxt;
  1282. }
  1283. tcp_hdr(skb)->fin = 0;
  1284. tp->rcv_nxt += skb->len;
  1285. __skb_queue_tail(&sk->sk_receive_queue, skb);
  1286. if (!sock_flag(sk, SOCK_DEAD)) {
  1287. check_sk_callbacks(csk);
  1288. sk->sk_data_ready(sk);
  1289. }
  1290. }
  1291. static int chtls_rx_data(struct chtls_dev *cdev, struct sk_buff *skb)
  1292. {
  1293. struct cpl_rx_data *req = cplhdr(skb) + RSS_HDR;
  1294. unsigned int hwtid = GET_TID(req);
  1295. struct sock *sk;
  1296. sk = lookup_tid(cdev->tids, hwtid);
  1297. if (unlikely(!sk)) {
  1298. pr_err("can't find conn. for hwtid %u.\n", hwtid);
  1299. return -EINVAL;
  1300. }
  1301. skb_dst_set(skb, NULL);
  1302. process_cpl_msg(chtls_recv_data, sk, skb);
  1303. return 0;
  1304. }
  1305. static void chtls_recv_pdu(struct sock *sk, struct sk_buff *skb)
  1306. {
  1307. struct cpl_tls_data *hdr = cplhdr(skb);
  1308. struct chtls_sock *csk;
  1309. struct chtls_hws *tlsk;
  1310. struct tcp_sock *tp;
  1311. csk = rcu_dereference_sk_user_data(sk);
  1312. tlsk = &csk->tlshws;
  1313. tp = tcp_sk(sk);
  1314. if (unlikely(sk->sk_shutdown & RCV_SHUTDOWN)) {
  1315. handle_excess_rx(sk, skb);
  1316. return;
  1317. }
  1318. ULP_SKB_CB(skb)->seq = ntohl(hdr->seq);
  1319. ULP_SKB_CB(skb)->flags = 0;
  1320. skb_ulp_mode(skb) = ULP_MODE_TLS;
  1321. skb_reset_transport_header(skb);
  1322. __skb_pull(skb, sizeof(*hdr));
  1323. if (!skb->data_len)
  1324. __skb_trim(skb,
  1325. CPL_TLS_DATA_LENGTH_G(ntohl(hdr->length_pkd)));
  1326. if (unlikely(tp->urg_data == TCP_URG_NOTYET && tp->urg_seq -
  1327. tp->rcv_nxt < skb->len))
  1328. tp->urg_data = TCP_URG_VALID |
  1329. skb->data[tp->urg_seq - tp->rcv_nxt];
  1330. tcp_hdr(skb)->fin = 0;
  1331. tlsk->pldlen = CPL_TLS_DATA_LENGTH_G(ntohl(hdr->length_pkd));
  1332. __skb_queue_tail(&tlsk->sk_recv_queue, skb);
  1333. }
  1334. static int chtls_rx_pdu(struct chtls_dev *cdev, struct sk_buff *skb)
  1335. {
  1336. struct cpl_tls_data *req = cplhdr(skb);
  1337. unsigned int hwtid = GET_TID(req);
  1338. struct sock *sk;
  1339. sk = lookup_tid(cdev->tids, hwtid);
  1340. if (unlikely(!sk)) {
  1341. pr_err("can't find conn. for hwtid %u.\n", hwtid);
  1342. return -EINVAL;
  1343. }
  1344. skb_dst_set(skb, NULL);
  1345. process_cpl_msg(chtls_recv_pdu, sk, skb);
  1346. return 0;
  1347. }
  1348. static void chtls_set_hdrlen(struct sk_buff *skb, unsigned int nlen)
  1349. {
  1350. struct tlsrx_cmp_hdr *tls_cmp_hdr = cplhdr(skb);
  1351. skb->hdr_len = ntohs((__force __be16)tls_cmp_hdr->length);
  1352. tls_cmp_hdr->length = ntohs((__force __be16)nlen);
  1353. }
  1354. static void chtls_rx_hdr(struct sock *sk, struct sk_buff *skb)
  1355. {
  1356. struct tlsrx_cmp_hdr *tls_hdr_pkt;
  1357. struct cpl_rx_tls_cmp *cmp_cpl;
  1358. struct sk_buff *skb_rec;
  1359. struct chtls_sock *csk;
  1360. struct chtls_hws *tlsk;
  1361. struct tcp_sock *tp;
  1362. cmp_cpl = cplhdr(skb);
  1363. csk = rcu_dereference_sk_user_data(sk);
  1364. tlsk = &csk->tlshws;
  1365. tp = tcp_sk(sk);
  1366. ULP_SKB_CB(skb)->seq = ntohl(cmp_cpl->seq);
  1367. ULP_SKB_CB(skb)->flags = 0;
  1368. skb_reset_transport_header(skb);
  1369. __skb_pull(skb, sizeof(*cmp_cpl));
  1370. tls_hdr_pkt = (struct tlsrx_cmp_hdr *)skb->data;
  1371. if (tls_hdr_pkt->res_to_mac_error & TLSRX_HDR_PKT_ERROR_M)
  1372. tls_hdr_pkt->type = CONTENT_TYPE_ERROR;
  1373. if (!skb->data_len)
  1374. __skb_trim(skb, TLS_HEADER_LENGTH);
  1375. tp->rcv_nxt +=
  1376. CPL_RX_TLS_CMP_PDULENGTH_G(ntohl(cmp_cpl->pdulength_length));
  1377. ULP_SKB_CB(skb)->flags |= ULPCB_FLAG_TLS_HDR;
  1378. skb_rec = __skb_dequeue(&tlsk->sk_recv_queue);
  1379. if (!skb_rec) {
  1380. __skb_queue_tail(&sk->sk_receive_queue, skb);
  1381. } else {
  1382. chtls_set_hdrlen(skb, tlsk->pldlen);
  1383. tlsk->pldlen = 0;
  1384. __skb_queue_tail(&sk->sk_receive_queue, skb);
  1385. __skb_queue_tail(&sk->sk_receive_queue, skb_rec);
  1386. }
  1387. if (!sock_flag(sk, SOCK_DEAD)) {
  1388. check_sk_callbacks(csk);
  1389. sk->sk_data_ready(sk);
  1390. }
  1391. }
  1392. static int chtls_rx_cmp(struct chtls_dev *cdev, struct sk_buff *skb)
  1393. {
  1394. struct cpl_rx_tls_cmp *req = cplhdr(skb);
  1395. unsigned int hwtid = GET_TID(req);
  1396. struct sock *sk;
  1397. sk = lookup_tid(cdev->tids, hwtid);
  1398. if (unlikely(!sk)) {
  1399. pr_err("can't find conn. for hwtid %u.\n", hwtid);
  1400. return -EINVAL;
  1401. }
  1402. skb_dst_set(skb, NULL);
  1403. process_cpl_msg(chtls_rx_hdr, sk, skb);
  1404. return 0;
  1405. }
  1406. static void chtls_timewait(struct sock *sk)
  1407. {
  1408. struct tcp_sock *tp = tcp_sk(sk);
  1409. tp->rcv_nxt++;
  1410. tp->rx_opt.ts_recent_stamp = ktime_get_seconds();
  1411. tp->srtt_us = 0;
  1412. tcp_time_wait(sk, TCP_TIME_WAIT, 0);
  1413. }
  1414. static void chtls_peer_close(struct sock *sk, struct sk_buff *skb)
  1415. {
  1416. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  1417. sk->sk_shutdown |= RCV_SHUTDOWN;
  1418. sock_set_flag(sk, SOCK_DONE);
  1419. switch (sk->sk_state) {
  1420. case TCP_SYN_RECV:
  1421. case TCP_ESTABLISHED:
  1422. tcp_set_state(sk, TCP_CLOSE_WAIT);
  1423. break;
  1424. case TCP_FIN_WAIT1:
  1425. tcp_set_state(sk, TCP_CLOSING);
  1426. break;
  1427. case TCP_FIN_WAIT2:
  1428. chtls_release_resources(sk);
  1429. if (csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING))
  1430. chtls_conn_done(sk);
  1431. else
  1432. chtls_timewait(sk);
  1433. break;
  1434. default:
  1435. pr_info("cpl_peer_close in bad state %d\n", sk->sk_state);
  1436. }
  1437. if (!sock_flag(sk, SOCK_DEAD)) {
  1438. sk->sk_state_change(sk);
  1439. /* Do not send POLL_HUP for half duplex close. */
  1440. if ((sk->sk_shutdown & SEND_SHUTDOWN) ||
  1441. sk->sk_state == TCP_CLOSE)
  1442. sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_HUP);
  1443. else
  1444. sk_wake_async(sk, SOCK_WAKE_WAITD, POLL_IN);
  1445. }
  1446. }
  1447. static void chtls_close_con_rpl(struct sock *sk, struct sk_buff *skb)
  1448. {
  1449. struct cpl_close_con_rpl *rpl = cplhdr(skb) + RSS_HDR;
  1450. struct chtls_sock *csk;
  1451. struct tcp_sock *tp;
  1452. csk = rcu_dereference_sk_user_data(sk);
  1453. tp = tcp_sk(sk);
  1454. tp->snd_una = ntohl(rpl->snd_nxt) - 1; /* exclude FIN */
  1455. switch (sk->sk_state) {
  1456. case TCP_CLOSING:
  1457. chtls_release_resources(sk);
  1458. if (csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING))
  1459. chtls_conn_done(sk);
  1460. else
  1461. chtls_timewait(sk);
  1462. break;
  1463. case TCP_LAST_ACK:
  1464. chtls_release_resources(sk);
  1465. chtls_conn_done(sk);
  1466. break;
  1467. case TCP_FIN_WAIT1:
  1468. tcp_set_state(sk, TCP_FIN_WAIT2);
  1469. sk->sk_shutdown |= SEND_SHUTDOWN;
  1470. if (!sock_flag(sk, SOCK_DEAD))
  1471. sk->sk_state_change(sk);
  1472. else if (tcp_sk(sk)->linger2 < 0 &&
  1473. !csk_flag_nochk(csk, CSK_ABORT_SHUTDOWN))
  1474. chtls_abort_conn(sk, skb);
  1475. break;
  1476. default:
  1477. pr_info("close_con_rpl in bad state %d\n", sk->sk_state);
  1478. }
  1479. kfree_skb(skb);
  1480. }
  1481. static struct sk_buff *get_cpl_skb(struct sk_buff *skb,
  1482. size_t len, gfp_t gfp)
  1483. {
  1484. if (likely(!skb_is_nonlinear(skb) && !skb_cloned(skb))) {
  1485. WARN_ONCE(skb->len < len, "skb alloc error");
  1486. __skb_trim(skb, len);
  1487. skb_get(skb);
  1488. } else {
  1489. skb = alloc_skb(len, gfp);
  1490. if (skb)
  1491. __skb_put(skb, len);
  1492. }
  1493. return skb;
  1494. }
  1495. static void set_abort_rpl_wr(struct sk_buff *skb, unsigned int tid,
  1496. int cmd)
  1497. {
  1498. struct cpl_abort_rpl *rpl = cplhdr(skb);
  1499. INIT_TP_WR_CPL(rpl, CPL_ABORT_RPL, tid);
  1500. rpl->cmd = cmd;
  1501. }
  1502. static void send_defer_abort_rpl(struct chtls_dev *cdev, struct sk_buff *skb)
  1503. {
  1504. struct cpl_abort_req_rss *req = cplhdr(skb);
  1505. struct sk_buff *reply_skb;
  1506. reply_skb = alloc_skb(sizeof(struct cpl_abort_rpl),
  1507. GFP_KERNEL | __GFP_NOFAIL);
  1508. __skb_put(reply_skb, sizeof(struct cpl_abort_rpl));
  1509. set_abort_rpl_wr(reply_skb, GET_TID(req),
  1510. (req->status & CPL_ABORT_NO_RST));
  1511. set_wr_txq(reply_skb, CPL_PRIORITY_DATA, req->status >> 1);
  1512. cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
  1513. kfree_skb(skb);
  1514. }
  1515. static void send_abort_rpl(struct sock *sk, struct sk_buff *skb,
  1516. struct chtls_dev *cdev, int status, int queue)
  1517. {
  1518. struct cpl_abort_req_rss *req = cplhdr(skb);
  1519. struct sk_buff *reply_skb;
  1520. struct chtls_sock *csk;
  1521. csk = rcu_dereference_sk_user_data(sk);
  1522. reply_skb = alloc_skb(sizeof(struct cpl_abort_rpl),
  1523. GFP_KERNEL);
  1524. if (!reply_skb) {
  1525. req->status = (queue << 1);
  1526. send_defer_abort_rpl(cdev, skb);
  1527. return;
  1528. }
  1529. set_abort_rpl_wr(reply_skb, GET_TID(req), status);
  1530. kfree_skb(skb);
  1531. set_wr_txq(reply_skb, CPL_PRIORITY_DATA, queue);
  1532. if (csk_conn_inline(csk)) {
  1533. struct l2t_entry *e = csk->l2t_entry;
  1534. if (e && sk->sk_state != TCP_SYN_RECV) {
  1535. cxgb4_l2t_send(csk->egress_dev, reply_skb, e);
  1536. return;
  1537. }
  1538. }
  1539. cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
  1540. }
  1541. /*
  1542. * Add an skb to the deferred skb queue for processing from process context.
  1543. */
  1544. static void t4_defer_reply(struct sk_buff *skb, struct chtls_dev *cdev,
  1545. defer_handler_t handler)
  1546. {
  1547. DEFERRED_SKB_CB(skb)->handler = handler;
  1548. spin_lock_bh(&cdev->deferq.lock);
  1549. __skb_queue_tail(&cdev->deferq, skb);
  1550. if (skb_queue_len(&cdev->deferq) == 1)
  1551. schedule_work(&cdev->deferq_task);
  1552. spin_unlock_bh(&cdev->deferq.lock);
  1553. }
  1554. static void chtls_send_abort_rpl(struct sock *sk, struct sk_buff *skb,
  1555. struct chtls_dev *cdev,
  1556. int status, int queue)
  1557. {
  1558. struct cpl_abort_req_rss *req = cplhdr(skb) + RSS_HDR;
  1559. struct sk_buff *reply_skb;
  1560. struct chtls_sock *csk;
  1561. unsigned int tid;
  1562. csk = rcu_dereference_sk_user_data(sk);
  1563. tid = GET_TID(req);
  1564. reply_skb = get_cpl_skb(skb, sizeof(struct cpl_abort_rpl), gfp_any());
  1565. if (!reply_skb) {
  1566. req->status = (queue << 1) | status;
  1567. t4_defer_reply(skb, cdev, send_defer_abort_rpl);
  1568. return;
  1569. }
  1570. set_abort_rpl_wr(reply_skb, tid, status);
  1571. set_wr_txq(reply_skb, CPL_PRIORITY_DATA, queue);
  1572. if (csk_conn_inline(csk)) {
  1573. struct l2t_entry *e = csk->l2t_entry;
  1574. if (e && sk->sk_state != TCP_SYN_RECV) {
  1575. cxgb4_l2t_send(csk->egress_dev, reply_skb, e);
  1576. return;
  1577. }
  1578. }
  1579. cxgb4_ofld_send(cdev->lldi->ports[0], reply_skb);
  1580. kfree_skb(skb);
  1581. }
  1582. /*
  1583. * This is run from a listener's backlog to abort a child connection in
  1584. * SYN_RCV state (i.e., one on the listener's SYN queue).
  1585. */
  1586. static void bl_abort_syn_rcv(struct sock *lsk, struct sk_buff *skb)
  1587. {
  1588. struct chtls_sock *csk;
  1589. struct sock *child;
  1590. int queue;
  1591. child = skb->sk;
  1592. csk = rcu_dereference_sk_user_data(child);
  1593. queue = csk->txq_idx;
  1594. skb->sk = NULL;
  1595. do_abort_syn_rcv(child, lsk);
  1596. send_abort_rpl(child, skb, BLOG_SKB_CB(skb)->cdev,
  1597. CPL_ABORT_NO_RST, queue);
  1598. }
  1599. static int abort_syn_rcv(struct sock *sk, struct sk_buff *skb)
  1600. {
  1601. const struct request_sock *oreq;
  1602. struct listen_ctx *listen_ctx;
  1603. struct chtls_sock *csk;
  1604. struct chtls_dev *cdev;
  1605. struct sock *psk;
  1606. void *ctx;
  1607. csk = sk->sk_user_data;
  1608. oreq = csk->passive_reap_next;
  1609. cdev = csk->cdev;
  1610. if (!oreq)
  1611. return -1;
  1612. ctx = lookup_stid(cdev->tids, oreq->ts_recent);
  1613. if (!ctx)
  1614. return -1;
  1615. listen_ctx = (struct listen_ctx *)ctx;
  1616. psk = listen_ctx->lsk;
  1617. bh_lock_sock(psk);
  1618. if (!sock_owned_by_user(psk)) {
  1619. int queue = csk->txq_idx;
  1620. do_abort_syn_rcv(sk, psk);
  1621. send_abort_rpl(sk, skb, cdev, CPL_ABORT_NO_RST, queue);
  1622. } else {
  1623. skb->sk = sk;
  1624. BLOG_SKB_CB(skb)->backlog_rcv = bl_abort_syn_rcv;
  1625. __sk_add_backlog(psk, skb);
  1626. }
  1627. bh_unlock_sock(psk);
  1628. return 0;
  1629. }
  1630. static void chtls_abort_req_rss(struct sock *sk, struct sk_buff *skb)
  1631. {
  1632. const struct cpl_abort_req_rss *req = cplhdr(skb) + RSS_HDR;
  1633. struct chtls_sock *csk = sk->sk_user_data;
  1634. int rst_status = CPL_ABORT_NO_RST;
  1635. int queue = csk->txq_idx;
  1636. if (is_neg_adv(req->status)) {
  1637. if (sk->sk_state == TCP_SYN_RECV)
  1638. chtls_set_tcb_tflag(sk, 0, 0);
  1639. kfree_skb(skb);
  1640. return;
  1641. }
  1642. csk_reset_flag(csk, CSK_ABORT_REQ_RCVD);
  1643. if (!csk_flag_nochk(csk, CSK_ABORT_SHUTDOWN) &&
  1644. !csk_flag_nochk(csk, CSK_TX_DATA_SENT)) {
  1645. struct tcp_sock *tp = tcp_sk(sk);
  1646. if (send_tx_flowc_wr(sk, 0, tp->snd_nxt, tp->rcv_nxt) < 0)
  1647. WARN_ONCE(1, "send_tx_flowc error");
  1648. csk_set_flag(csk, CSK_TX_DATA_SENT);
  1649. }
  1650. csk_set_flag(csk, CSK_ABORT_SHUTDOWN);
  1651. if (!csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING)) {
  1652. sk->sk_err = ETIMEDOUT;
  1653. if (!sock_flag(sk, SOCK_DEAD))
  1654. sk->sk_error_report(sk);
  1655. if (sk->sk_state == TCP_SYN_RECV && !abort_syn_rcv(sk, skb))
  1656. return;
  1657. chtls_release_resources(sk);
  1658. chtls_conn_done(sk);
  1659. }
  1660. chtls_send_abort_rpl(sk, skb, csk->cdev, rst_status, queue);
  1661. }
  1662. static void chtls_abort_rpl_rss(struct sock *sk, struct sk_buff *skb)
  1663. {
  1664. struct cpl_abort_rpl_rss *rpl = cplhdr(skb) + RSS_HDR;
  1665. struct chtls_sock *csk;
  1666. struct chtls_dev *cdev;
  1667. csk = rcu_dereference_sk_user_data(sk);
  1668. cdev = csk->cdev;
  1669. if (csk_flag_nochk(csk, CSK_ABORT_RPL_PENDING)) {
  1670. csk_reset_flag(csk, CSK_ABORT_RPL_PENDING);
  1671. if (!csk_flag_nochk(csk, CSK_ABORT_REQ_RCVD)) {
  1672. if (sk->sk_state == TCP_SYN_SENT) {
  1673. cxgb4_remove_tid(cdev->tids,
  1674. csk->port_id,
  1675. GET_TID(rpl),
  1676. sk->sk_family);
  1677. sock_put(sk);
  1678. }
  1679. chtls_release_resources(sk);
  1680. chtls_conn_done(sk);
  1681. }
  1682. }
  1683. kfree_skb(skb);
  1684. }
  1685. static int chtls_conn_cpl(struct chtls_dev *cdev, struct sk_buff *skb)
  1686. {
  1687. struct cpl_peer_close *req = cplhdr(skb) + RSS_HDR;
  1688. void (*fn)(struct sock *sk, struct sk_buff *skb);
  1689. unsigned int hwtid = GET_TID(req);
  1690. struct sock *sk;
  1691. u8 opcode;
  1692. opcode = ((const struct rss_header *)cplhdr(skb))->opcode;
  1693. sk = lookup_tid(cdev->tids, hwtid);
  1694. if (!sk)
  1695. goto rel_skb;
  1696. switch (opcode) {
  1697. case CPL_PEER_CLOSE:
  1698. fn = chtls_peer_close;
  1699. break;
  1700. case CPL_CLOSE_CON_RPL:
  1701. fn = chtls_close_con_rpl;
  1702. break;
  1703. case CPL_ABORT_REQ_RSS:
  1704. fn = chtls_abort_req_rss;
  1705. break;
  1706. case CPL_ABORT_RPL_RSS:
  1707. fn = chtls_abort_rpl_rss;
  1708. break;
  1709. default:
  1710. goto rel_skb;
  1711. }
  1712. process_cpl_msg(fn, sk, skb);
  1713. return 0;
  1714. rel_skb:
  1715. kfree_skb(skb);
  1716. return 0;
  1717. }
  1718. static struct sk_buff *dequeue_wr(struct sock *sk)
  1719. {
  1720. struct chtls_sock *csk = rcu_dereference_sk_user_data(sk);
  1721. struct sk_buff *skb = csk->wr_skb_head;
  1722. if (likely(skb)) {
  1723. /* Don't bother clearing the tail */
  1724. csk->wr_skb_head = WR_SKB_CB(skb)->next_wr;
  1725. WR_SKB_CB(skb)->next_wr = NULL;
  1726. }
  1727. return skb;
  1728. }
  1729. static void chtls_rx_ack(struct sock *sk, struct sk_buff *skb)
  1730. {
  1731. struct cpl_fw4_ack *hdr = cplhdr(skb) + RSS_HDR;
  1732. struct chtls_sock *csk = sk->sk_user_data;
  1733. struct tcp_sock *tp = tcp_sk(sk);
  1734. u32 credits = hdr->credits;
  1735. u32 snd_una;
  1736. snd_una = ntohl(hdr->snd_una);
  1737. csk->wr_credits += credits;
  1738. if (csk->wr_unacked > csk->wr_max_credits - csk->wr_credits)
  1739. csk->wr_unacked = csk->wr_max_credits - csk->wr_credits;
  1740. while (credits) {
  1741. struct sk_buff *pskb = csk->wr_skb_head;
  1742. u32 csum;
  1743. if (unlikely(!pskb)) {
  1744. if (csk->wr_nondata)
  1745. csk->wr_nondata -= credits;
  1746. break;
  1747. }
  1748. csum = (__force u32)pskb->csum;
  1749. if (unlikely(credits < csum)) {
  1750. pskb->csum = (__force __wsum)(csum - credits);
  1751. break;
  1752. }
  1753. dequeue_wr(sk);
  1754. credits -= csum;
  1755. kfree_skb(pskb);
  1756. }
  1757. if (hdr->seq_vld & CPL_FW4_ACK_FLAGS_SEQVAL) {
  1758. if (unlikely(before(snd_una, tp->snd_una))) {
  1759. kfree_skb(skb);
  1760. return;
  1761. }
  1762. if (tp->snd_una != snd_una) {
  1763. tp->snd_una = snd_una;
  1764. tp->rcv_tstamp = tcp_time_stamp(tp);
  1765. if (tp->snd_una == tp->snd_nxt &&
  1766. !csk_flag_nochk(csk, CSK_TX_FAILOVER))
  1767. csk_reset_flag(csk, CSK_TX_WAIT_IDLE);
  1768. }
  1769. }
  1770. if (hdr->seq_vld & CPL_FW4_ACK_FLAGS_CH) {
  1771. unsigned int fclen16 = roundup(failover_flowc_wr_len, 16);
  1772. csk->wr_credits -= fclen16;
  1773. csk_reset_flag(csk, CSK_TX_WAIT_IDLE);
  1774. csk_reset_flag(csk, CSK_TX_FAILOVER);
  1775. }
  1776. if (skb_queue_len(&csk->txq) && chtls_push_frames(csk, 0))
  1777. sk->sk_write_space(sk);
  1778. kfree_skb(skb);
  1779. }
  1780. static int chtls_wr_ack(struct chtls_dev *cdev, struct sk_buff *skb)
  1781. {
  1782. struct cpl_fw4_ack *rpl = cplhdr(skb) + RSS_HDR;
  1783. unsigned int hwtid = GET_TID(rpl);
  1784. struct sock *sk;
  1785. sk = lookup_tid(cdev->tids, hwtid);
  1786. if (unlikely(!sk)) {
  1787. pr_err("can't find conn. for hwtid %u.\n", hwtid);
  1788. return -EINVAL;
  1789. }
  1790. process_cpl_msg(chtls_rx_ack, sk, skb);
  1791. return 0;
  1792. }
  1793. chtls_handler_func chtls_handlers[NUM_CPL_CMDS] = {
  1794. [CPL_PASS_OPEN_RPL] = chtls_pass_open_rpl,
  1795. [CPL_CLOSE_LISTSRV_RPL] = chtls_close_listsrv_rpl,
  1796. [CPL_PASS_ACCEPT_REQ] = chtls_pass_accept_req,
  1797. [CPL_PASS_ESTABLISH] = chtls_pass_establish,
  1798. [CPL_RX_DATA] = chtls_rx_data,
  1799. [CPL_TLS_DATA] = chtls_rx_pdu,
  1800. [CPL_RX_TLS_CMP] = chtls_rx_cmp,
  1801. [CPL_PEER_CLOSE] = chtls_conn_cpl,
  1802. [CPL_CLOSE_CON_RPL] = chtls_conn_cpl,
  1803. [CPL_ABORT_REQ_RSS] = chtls_conn_cpl,
  1804. [CPL_ABORT_RPL_RSS] = chtls_conn_cpl,
  1805. [CPL_FW4_ACK] = chtls_wr_ack,
  1806. };