chtls_cm.c 51 KB

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