tcp.c 72 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434
  1. /*
  2. * This file is part of the UCB release of Plan 9. It is subject to the license
  3. * terms in the LICENSE file found in the top-level directory of this
  4. * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
  5. * part of the UCB release of Plan 9, including this file, may be copied,
  6. * modified, propagated, or distributed except according to the terms contained
  7. * in the LICENSE file.
  8. */
  9. #include "u.h"
  10. #include "../port/lib.h"
  11. #include "mem.h"
  12. #include "dat.h"
  13. #include "fns.h"
  14. #include "../port/error.h"
  15. #include "ip.h"
  16. enum {
  17. QMAX = 64 * 1024 - 1,
  18. IP_TCPPROTO = 6,
  19. TCP4_IPLEN = 8,
  20. TCP4_PHDRSIZE = 12,
  21. TCP4_HDRSIZE = 20,
  22. TCP4_TCBPHDRSZ = 40,
  23. TCP4_PKT = TCP4_IPLEN + TCP4_PHDRSIZE,
  24. TCP6_IPLEN = 0,
  25. TCP6_PHDRSIZE = 40,
  26. TCP6_HDRSIZE = 20,
  27. TCP6_TCBPHDRSZ = 60,
  28. TCP6_PKT = TCP6_IPLEN + TCP6_PHDRSIZE,
  29. TcptimerOFF = 0,
  30. TcptimerON = 1,
  31. TcptimerDONE = 2,
  32. MAX_TIME = (1 << 20), /* Forever */
  33. TCP_ACK = 50, /* Timed ack sequence in ms */
  34. MAXBACKMS = 9 * 60 * 1000, /* longest backoff time (ms) before hangup */
  35. URG = 0x20, /* Data marked urgent */
  36. ACK = 0x10, /* Acknowledge is valid */
  37. PSH = 0x08, /* Whole data pipe is pushed */
  38. RST = 0x04, /* Reset connection */
  39. SYN = 0x02, /* Pkt. is synchronise */
  40. FIN = 0x01, /* Start close down */
  41. EOLOPT = 0,
  42. NOOPOPT = 1,
  43. MSSOPT = 2,
  44. MSS_LENGTH = 4, /* Maximum segment size */
  45. WSOPT = 3,
  46. WS_LENGTH = 3, /* Bits to scale window size by */
  47. MSL2 = 10,
  48. MSPTICK = 50, /* Milliseconds per timer tick */
  49. DEF_MSS = 1460, /* Default maximum segment */
  50. DEF_MSS6 = 1280, /* Default maximum segment (min) for v6 */
  51. DEF_RTT = 500, /* Default round trip */
  52. DEF_KAT = 120000, /* Default time (ms) between keep alives */
  53. TCP_LISTEN = 0, /* Listen connection */
  54. TCP_CONNECT = 1, /* Outgoing connection */
  55. SYNACK_RXTIMER = 250, /* ms between SYNACK retransmits */
  56. TCPREXMTTHRESH = 3, /* dupack threshhold for rxt */
  57. FORCE = 1,
  58. CLONE = 2,
  59. RETRAN = 4,
  60. ACTIVE = 8,
  61. SYNACK = 16,
  62. LOGAGAIN = 3,
  63. LOGDGAIN = 2,
  64. Closed = 0, /* Connection states */
  65. Listen,
  66. Syn_sent,
  67. Syn_received,
  68. Established,
  69. Finwait1,
  70. Finwait2,
  71. Close_wait,
  72. Closing,
  73. Last_ack,
  74. Time_wait,
  75. Maxlimbo = 1000, /* maximum procs waiting for response to SYN ACK */
  76. NLHT = 256, /* hash table size, must be a power of 2 */
  77. LHTMASK = NLHT - 1,
  78. /*
  79. * window is 64kb * 2ⁿ
  80. * these factors determine the ultimate bandwidth-delay product.
  81. * 64kb * 2⁵ = 2mb, or 2× overkill for 100mbps * 70ms.
  82. */
  83. Maxqscale = 4, /* maximum queuing scale */
  84. Defadvscale = 4, /* default advertisement */
  85. };
  86. /* Must correspond to the enumeration above */
  87. char *tcpstates[] =
  88. {
  89. "Closed", "Listen", "Syn_sent", "Syn_received",
  90. "Established", "Finwait1", "Finwait2", "Close_wait",
  91. "Closing", "Last_ack", "Time_wait"};
  92. typedef struct Tcptimer Tcptimer;
  93. struct Tcptimer {
  94. Tcptimer *next;
  95. Tcptimer *prev;
  96. Tcptimer *readynext;
  97. int state;
  98. int start;
  99. int count;
  100. void (*func)(void *);
  101. void *arg;
  102. };
  103. /*
  104. * v4 and v6 pseudo headers used for
  105. * checksuming tcp
  106. */
  107. typedef struct Tcp4hdr Tcp4hdr;
  108. struct Tcp4hdr {
  109. u8 vihl; /* Version and header length */
  110. u8 tos; /* Type of service */
  111. u8 length[2]; /* packet length */
  112. u8 id[2]; /* Identification */
  113. u8 frag[2]; /* Fragment information */
  114. u8 Unused;
  115. u8 proto;
  116. u8 tcplen[2];
  117. u8 tcpsrc[4];
  118. u8 tcpdst[4];
  119. /* same as v6 from here on */
  120. u8 tcpsport[2];
  121. u8 tcpdport[2];
  122. u8 tcpseq[4];
  123. u8 tcpack[4];
  124. u8 tcpflag[2];
  125. u8 tcpwin[2];
  126. u8 tcpcksum[2];
  127. u8 tcpurg[2];
  128. /* Options segment */
  129. u8 tcpopt[1];
  130. };
  131. typedef struct Tcp6hdr Tcp6hdr;
  132. struct Tcp6hdr {
  133. u8 vcf[4];
  134. u8 ploadlen[2];
  135. u8 proto;
  136. u8 ttl;
  137. u8 tcpsrc[IPaddrlen];
  138. u8 tcpdst[IPaddrlen];
  139. /* same as v4 from here on */
  140. u8 tcpsport[2];
  141. u8 tcpdport[2];
  142. u8 tcpseq[4];
  143. u8 tcpack[4];
  144. u8 tcpflag[2];
  145. u8 tcpwin[2];
  146. u8 tcpcksum[2];
  147. u8 tcpurg[2];
  148. /* Options segment */
  149. u8 tcpopt[1];
  150. };
  151. /*
  152. * this represents the control info
  153. * for a single packet. It is derived from
  154. * a packet in ntohtcp{4,6}() and stuck into
  155. * a packet in htontcp{4,6}().
  156. */
  157. typedef struct Tcp Tcp;
  158. struct Tcp {
  159. u16 source;
  160. u16 dest;
  161. u32 seq;
  162. u32 ack;
  163. u8 flags;
  164. u8 update;
  165. u16 ws; /* window scale option */
  166. u32 wnd; /* prescaled window*/
  167. u16 urg;
  168. u16 mss; /* max segment size option (if not zero) */
  169. u16 len; /* size of data */
  170. };
  171. /*
  172. * this header is malloc'd to thread together fragments
  173. * waiting to be coalesced
  174. */
  175. typedef struct Reseq Reseq;
  176. struct Reseq {
  177. Reseq *next;
  178. Tcp seg;
  179. Block *bp;
  180. u16 length;
  181. };
  182. /*
  183. * the qlock in the Conv locks this structure
  184. */
  185. typedef struct Tcpctl Tcpctl;
  186. struct Tcpctl {
  187. u8 state; /* Connection state */
  188. u8 type; /* Listening or active connection */
  189. u8 code; /* Icmp code */
  190. struct {
  191. u32 una; /* Unacked data pointer */
  192. u32 nxt; /* Next sequence expected */
  193. u32 ptr; /* Data pointer */
  194. u32 wnd; /* Tcp send window */
  195. u32 urg; /* Urgent data pointer */
  196. u32 wl2;
  197. u32 scale; /* how much to right shift window */
  198. /* in xmitted packets */
  199. /* to implement tahoe and reno TCP */
  200. u32 dupacks; /* number of duplicate acks rcvd */
  201. u32 partialack;
  202. int recovery; /* loss recovery flag */
  203. int retransmit; /* retransmit 1 packet @ una flag */
  204. int rto;
  205. u32 rxt; /* right window marker for recovery */
  206. /* "recover" rfc3782 */
  207. } snd;
  208. struct {
  209. u32 nxt; /* Receive pointer to next u8 slot */
  210. u32 wnd; /* Receive window incoming */
  211. u32 wsnt; /* Last wptr sent. important to */
  212. /* track for large bdp */
  213. u32 wptr;
  214. u32 urg; /* Urgent pointer */
  215. u32 ackptr; /* last acked sequence */
  216. int blocked;
  217. u32 scale; /* how much to left shift window in */
  218. /* rcv'd packets */
  219. } rcv;
  220. u32 iss; /* Initial sequence number */
  221. u32 cwind; /* Congestion window */
  222. u32 abcbytes; /* appropriate byte counting rfc 3465 */
  223. u32 scale; /* desired snd.scale */
  224. u32 ssthresh; /* Slow start threshold */
  225. int resent; /* Bytes just resent */
  226. int irs; /* Initial received squence */
  227. u16 mss; /* Maximum segment size */
  228. int rerecv; /* Overlap of data rerecevived */
  229. u32 window; /* Our receive window (queue) */
  230. u32 qscale; /* Log2 of our receive window (queue) */
  231. u8 backoff; /* Exponential backoff counter */
  232. int backedoff; /* ms we've backed off for rexmits */
  233. u8 flags; /* State flags */
  234. Reseq *reseq; /* Resequencing queue */
  235. int nreseq;
  236. int reseqlen;
  237. Tcptimer timer; /* Activity timer */
  238. Tcptimer acktimer; /* Acknowledge timer */
  239. Tcptimer rtt_timer; /* Round trip timer */
  240. Tcptimer katimer; /* keep alive timer */
  241. u32 rttseq; /* Round trip sequence */
  242. int srtt; /* Smoothed round trip */
  243. int mdev; /* Mean deviation of round trip */
  244. int kacounter; /* count down for keep alive */
  245. u32 sndsyntime; /* time syn sent */
  246. u32 time; /* time Finwait2 or Syn_received was sent */
  247. u32 timeuna; /* snd.una when time was set */
  248. int nochecksum; /* non-zero means don't send checksums */
  249. int flgcnt; /* number of flags in the sequence (FIN,SEQ) */
  250. union {
  251. Tcp4hdr tcp4hdr;
  252. Tcp6hdr tcp6hdr;
  253. } protohdr; /* prototype header */
  254. };
  255. /*
  256. * New calls are put in limbo rather than having a conversation structure
  257. * allocated. Thus, a SYN attack results in lots of limbo'd calls but not
  258. * any real Conv structures mucking things up. Calls in limbo rexmit their
  259. * SYN ACK every SYNACK_RXTIMER ms up to 4 times, i.e., they disappear after 1 second.
  260. *
  261. * In particular they aren't on a listener's queue so that they don't figure
  262. * in the input queue limit.
  263. *
  264. * If 1/2 of a T3 was attacking SYN packets, we'ld have a permanent queue
  265. * of 70000 limbo'd calls. Not great for a linear list but doable. Therefore
  266. * there is no hashing of this list.
  267. */
  268. typedef struct Limbo Limbo;
  269. struct Limbo {
  270. Limbo *next;
  271. u8 laddr[IPaddrlen];
  272. u8 raddr[IPaddrlen];
  273. u16 lport;
  274. u16 rport;
  275. u32 irs; /* initial received sequence */
  276. u32 iss; /* initial sent sequence */
  277. u16 mss; /* mss from the other end */
  278. u16 rcvscale; /* how much to scale rcvd windows */
  279. u16 sndscale; /* how much to scale sent windows */
  280. u32 lastsend; /* last time we sent a synack */
  281. u8 version; /* v4 or v6 */
  282. u8 rexmits; /* number of retransmissions */
  283. };
  284. int tcp_irtt = DEF_RTT; /* Initial guess at round trip time */
  285. enum {
  286. /* MIB stats */
  287. MaxConn,
  288. Mss,
  289. ActiveOpens,
  290. PassiveOpens,
  291. EstabResets,
  292. CurrEstab,
  293. InSegs,
  294. OutSegs,
  295. RetransSegs,
  296. RetransSegsSent,
  297. RetransTimeouts,
  298. InErrs,
  299. OutRsts,
  300. /* non-MIB stats */
  301. CsumErrs,
  302. HlenErrs,
  303. LenErrs,
  304. Resequenced,
  305. OutOfOrder,
  306. ReseqBytelim,
  307. ReseqPktlim,
  308. Delayack,
  309. Wopenack,
  310. Recovery,
  311. RecoveryDone,
  312. RecoveryRTO,
  313. RecoveryNoSeq,
  314. RecoveryCwind,
  315. RecoveryPA,
  316. Nstats
  317. };
  318. static char *statnames[Nstats] =
  319. {
  320. [MaxConn] = "MaxConn",
  321. [Mss] = "MaxSegment",
  322. [ActiveOpens] = "ActiveOpens",
  323. [PassiveOpens] = "PassiveOpens",
  324. [EstabResets] = "EstabResets",
  325. [CurrEstab] = "CurrEstab",
  326. [InSegs] = "InSegs",
  327. [OutSegs] = "OutSegs",
  328. [RetransSegs] = "RetransSegs",
  329. [RetransSegsSent] = "RetransSegsSent",
  330. [RetransTimeouts] = "RetransTimeouts",
  331. [InErrs] = "InErrs",
  332. [OutRsts] = "OutRsts",
  333. [CsumErrs] = "CsumErrs",
  334. [HlenErrs] = "HlenErrs",
  335. [LenErrs] = "LenErrs",
  336. [OutOfOrder] = "OutOfOrder",
  337. [Resequenced] = "Resequenced",
  338. [ReseqBytelim] = "ReseqBytelim",
  339. [ReseqPktlim] = "ReseqPktlim",
  340. [Delayack] = "Delayack",
  341. [Wopenack] = "Wopenack",
  342. [Recovery] = "Recovery",
  343. [RecoveryDone] = "RecoveryDone",
  344. [RecoveryRTO] = "RecoveryRTO",
  345. [RecoveryNoSeq] = "RecoveryNoSeq",
  346. [RecoveryCwind] = "RecoveryCwind",
  347. [RecoveryPA] = "RecoveryPA",
  348. };
  349. typedef struct Tcppriv Tcppriv;
  350. struct Tcppriv {
  351. /* List of active timers */
  352. QLock tl;
  353. Tcptimer *timers;
  354. /* hash table for matching conversations */
  355. Ipht ht;
  356. /* calls in limbo waiting for an ACK to our SYN ACK */
  357. int nlimbo;
  358. Limbo *lht[NLHT];
  359. /* for keeping track of tcpackproc */
  360. QLock apl;
  361. int ackprocstarted;
  362. u64 stats[Nstats];
  363. };
  364. /*
  365. * Setting tcpporthogdefense to non-zero enables Dong Lin's
  366. * solution to hijacked systems staking out port's as a form
  367. * of DoS attack.
  368. *
  369. * To avoid stateless Conv hogs, we pick a sequence number at random. If
  370. * that number gets acked by the other end, we shut down the connection.
  371. * Look for tcpporthogdefense in the code.
  372. */
  373. int tcpporthogdefense = 0;
  374. static int addreseq(Fs *, Tcpctl *, Tcppriv *, Tcp *, Block *, u16);
  375. static int dumpreseq(Tcpctl *);
  376. static void getreseq(Tcpctl *, Tcp *, Block **, u16 *);
  377. static void limbo(Conv *, u8 *, u8 *, Tcp *, int);
  378. static void limborexmit(Proto *);
  379. static void localclose(Conv *, char *);
  380. static void procsyn(Conv *, Tcp *);
  381. static void tcpacktimer(void *);
  382. static void tcpiput(Proto *, Ipifc *, Block *);
  383. static void tcpkeepalive(void *);
  384. static void tcpoutput(Conv *);
  385. static void tcprcvwin(Conv *);
  386. static void tcprxmit(Conv *);
  387. static void tcpsetkacounter(Tcpctl *);
  388. static void tcpsetscale(Conv *, Tcpctl *, u16, u16);
  389. static void tcpsettimer(Tcpctl *);
  390. static void tcpsndsyn(Conv *, Tcpctl *);
  391. static void tcpstart(Conv *, int);
  392. static void tcpsynackrtt(Conv *);
  393. static void tcptimeout(void *);
  394. static int tcptrim(Tcpctl *, Tcp *, Block **, u16 *);
  395. static void
  396. tcpsetstate(Conv *s, u8 newstate)
  397. {
  398. Tcpctl *tcb;
  399. u8 oldstate;
  400. Tcppriv *tpriv;
  401. tpriv = s->p->priv;
  402. tcb = (Tcpctl *)s->ptcl;
  403. oldstate = tcb->state;
  404. if(oldstate == newstate)
  405. return;
  406. if(oldstate == Established)
  407. tpriv->stats[CurrEstab]--;
  408. if(newstate == Established)
  409. tpriv->stats[CurrEstab]++;
  410. switch(newstate){
  411. case Closed:
  412. qclose(s->rq);
  413. qclose(s->wq);
  414. qclose(s->eq);
  415. break;
  416. case Close_wait: /* Remote closes */
  417. qhangup(s->rq, nil);
  418. break;
  419. }
  420. tcb->state = newstate;
  421. if(oldstate == Syn_sent && newstate != Closed)
  422. Fsconnected(s, nil);
  423. }
  424. static char *
  425. tcpconnect(Conv *c, char **argv, int argc)
  426. {
  427. char *e;
  428. Tcpctl *tcb;
  429. tcb = (Tcpctl *)(c->ptcl);
  430. if(tcb->state != Closed)
  431. return Econinuse;
  432. e = Fsstdconnect(c, argv, argc);
  433. if(e != nil)
  434. return e;
  435. tcpstart(c, TCP_CONNECT);
  436. return nil;
  437. }
  438. static int
  439. tcpstate(Conv *c, char *state, int n)
  440. {
  441. Tcpctl *s;
  442. s = (Tcpctl *)(c->ptcl);
  443. return snprint(state, n,
  444. "%s qin %d qout %d rq %d.%d srtt %d mdev %d sst %lu cwin %lu "
  445. "swin %lu>>%d rwin %lu>>%d qscale %d timer.start %d "
  446. "timer.count %d rerecv %d katimer.start %d katimer.count %d\n",
  447. tcpstates[s->state],
  448. c->rq ? qlen(c->rq) : 0,
  449. c->wq ? qlen(c->wq) : 0,
  450. s->nreseq, s->reseqlen,
  451. s->srtt, s->mdev, s->ssthresh,
  452. s->cwind, s->snd.wnd, s->rcv.scale, s->rcv.wnd, s->snd.scale,
  453. s->qscale,
  454. s->timer.start, s->timer.count, s->rerecv,
  455. s->katimer.start, s->katimer.count);
  456. }
  457. static int
  458. tcpinuse(Conv *c)
  459. {
  460. Tcpctl *s;
  461. s = (Tcpctl *)(c->ptcl);
  462. return s->state != Closed;
  463. }
  464. static char *
  465. tcpannounce(Conv *c, char **argv, int argc)
  466. {
  467. char *e;
  468. Tcpctl *tcb;
  469. tcb = (Tcpctl *)(c->ptcl);
  470. if(tcb->state != Closed)
  471. return Econinuse;
  472. e = Fsstdannounce(c, argv, argc);
  473. if(e != nil)
  474. return e;
  475. tcpstart(c, TCP_LISTEN);
  476. Fsconnected(c, nil);
  477. return nil;
  478. }
  479. /*
  480. * tcpclose is always called with the q locked
  481. */
  482. static void
  483. tcpclose(Conv *c)
  484. {
  485. Tcpctl *tcb;
  486. tcb = (Tcpctl *)c->ptcl;
  487. qhangup(c->rq, nil);
  488. qhangup(c->wq, nil);
  489. qhangup(c->eq, nil);
  490. qflush(c->rq);
  491. switch(tcb->state){
  492. case Listen:
  493. /*
  494. * reset any incoming calls to this listener
  495. */
  496. Fsconnected(c, "Hangup");
  497. localclose(c, nil);
  498. break;
  499. case Closed:
  500. case Syn_sent:
  501. localclose(c, nil);
  502. break;
  503. case Syn_received:
  504. case Established:
  505. tcb->flgcnt++;
  506. tcb->snd.nxt++;
  507. tcpsetstate(c, Finwait1);
  508. tcpoutput(c);
  509. break;
  510. case Close_wait:
  511. tcb->flgcnt++;
  512. tcb->snd.nxt++;
  513. tcpsetstate(c, Last_ack);
  514. tcpoutput(c);
  515. break;
  516. }
  517. }
  518. static void
  519. tcpkick(void *x)
  520. {
  521. Proc *up = externup();
  522. Conv *s = x;
  523. Tcpctl *tcb;
  524. tcb = (Tcpctl *)s->ptcl;
  525. if(waserror()){
  526. qunlock(&s->ql);
  527. nexterror();
  528. }
  529. qlock(&s->ql);
  530. switch(tcb->state){
  531. case Syn_sent:
  532. case Syn_received:
  533. case Established:
  534. case Close_wait:
  535. /*
  536. * Push data
  537. */
  538. tcpoutput(s);
  539. break;
  540. default:
  541. localclose(s, "Hangup");
  542. break;
  543. }
  544. qunlock(&s->ql);
  545. poperror();
  546. }
  547. static int seq_lt(u32, u32);
  548. static void
  549. tcprcvwin(Conv *s) /* Call with tcb locked */
  550. {
  551. int w;
  552. Tcpctl *tcb;
  553. tcb = (Tcpctl *)s->ptcl;
  554. w = tcb->window - qlen(s->rq);
  555. if(w < 0)
  556. w = 0;
  557. /* RFC 1122 § 4.2.2.17 do not move right edge of window left */
  558. if(seq_lt(tcb->rcv.nxt + w, tcb->rcv.wptr))
  559. w = tcb->rcv.wptr - tcb->rcv.nxt;
  560. if(w != tcb->rcv.wnd)
  561. if(w >> tcb->rcv.scale == 0 || (tcb->window > 4 * tcb->mss && w < tcb->mss / 4)){
  562. tcb->rcv.blocked = 1;
  563. netlog(s->p->f, Logtcp, "tcprcvwin: window %lu qlen %d ws %u lport %d\n",
  564. tcb->window, qlen(s->rq), tcb->rcv.scale, s->lport);
  565. }
  566. tcb->rcv.wnd = w;
  567. tcb->rcv.wptr = tcb->rcv.nxt + w;
  568. }
  569. static void
  570. tcpacktimer(void *v)
  571. {
  572. Proc *up = externup();
  573. Tcpctl *tcb;
  574. Conv *s;
  575. s = v;
  576. tcb = (Tcpctl *)s->ptcl;
  577. if(waserror()){
  578. qunlock(&s->ql);
  579. nexterror();
  580. }
  581. qlock(&s->ql);
  582. if(tcb->state != Closed){
  583. tcb->flags |= FORCE;
  584. tcpoutput(s);
  585. }
  586. qunlock(&s->ql);
  587. poperror();
  588. }
  589. static void
  590. tcpcongestion(Tcpctl *tcb)
  591. {
  592. u32 inflight;
  593. inflight = tcb->snd.nxt - tcb->snd.una;
  594. if(inflight > tcb->cwind)
  595. inflight = tcb->cwind;
  596. tcb->ssthresh = inflight / 2;
  597. if(tcb->ssthresh < 2 * tcb->mss)
  598. tcb->ssthresh = 2 * tcb->mss;
  599. }
  600. enum {
  601. L = 2, /* aggressive slow start; legal values ∈ (1.0, 2.0) */
  602. };
  603. static void
  604. tcpabcincr(Tcpctl *tcb, u32 acked)
  605. {
  606. u32 limit;
  607. tcb->abcbytes += acked;
  608. if(tcb->cwind < tcb->ssthresh){
  609. /* slow start */
  610. if(tcb->snd.rto)
  611. limit = tcb->mss;
  612. else
  613. limit = L * tcb->mss;
  614. tcb->cwind += MIN(tcb->abcbytes, limit);
  615. tcb->abcbytes = 0;
  616. } else {
  617. tcb->snd.rto = 0;
  618. /* avoidance */
  619. if(tcb->abcbytes >= tcb->cwind){
  620. tcb->abcbytes -= tcb->cwind;
  621. tcb->cwind += tcb->mss;
  622. }
  623. }
  624. }
  625. static void
  626. tcpcreate(Conv *c)
  627. {
  628. c->rq = qopen(QMAX, Qcoalesce, tcpacktimer, c);
  629. c->wq = qopen(QMAX, Qkick, tcpkick, c);
  630. }
  631. static void
  632. timerstate(Tcppriv *priv, Tcptimer *t, int newstate)
  633. {
  634. if(newstate != TcptimerON){
  635. if(t->state == TcptimerON){
  636. /* unchain */
  637. if(priv->timers == t){
  638. priv->timers = t->next;
  639. if(t->prev != nil)
  640. panic("timerstate1");
  641. }
  642. if(t->next)
  643. t->next->prev = t->prev;
  644. if(t->prev)
  645. t->prev->next = t->next;
  646. t->next = t->prev = nil;
  647. }
  648. } else {
  649. if(t->state != TcptimerON){
  650. /* chain */
  651. if(t->prev != nil || t->next != nil)
  652. panic("timerstate2");
  653. t->prev = nil;
  654. t->next = priv->timers;
  655. if(t->next)
  656. t->next->prev = t;
  657. priv->timers = t;
  658. }
  659. }
  660. t->state = newstate;
  661. }
  662. static void
  663. tcpackproc(void *a)
  664. {
  665. Proc *up = externup();
  666. Tcptimer *t, *tp, *timeo;
  667. Proto *tcp;
  668. Tcppriv *priv;
  669. int loop;
  670. tcp = a;
  671. priv = tcp->priv;
  672. for(;;){
  673. tsleep(&up->sleep, return0, 0, MSPTICK);
  674. qlock(&priv->tl);
  675. timeo = nil;
  676. loop = 0;
  677. for(t = priv->timers; t != nil; t = tp){
  678. if(loop++ > 10000)
  679. panic("tcpackproc1");
  680. tp = t->next;
  681. if(t->state == TcptimerON){
  682. t->count--;
  683. if(t->count == 0){
  684. timerstate(priv, t, TcptimerDONE);
  685. t->readynext = timeo;
  686. timeo = t;
  687. }
  688. }
  689. }
  690. qunlock(&priv->tl);
  691. loop = 0;
  692. for(t = timeo; t != nil; t = t->readynext){
  693. if(loop++ > 10000)
  694. panic("tcpackproc2");
  695. if(t->state == TcptimerDONE && t->func != nil && !waserror()){
  696. (*t->func)(t->arg);
  697. poperror();
  698. }
  699. }
  700. limborexmit(tcp);
  701. }
  702. }
  703. static void
  704. tcpgo(Tcppriv *priv, Tcptimer *t)
  705. {
  706. if(t == nil || t->start == 0)
  707. return;
  708. qlock(&priv->tl);
  709. t->count = t->start;
  710. timerstate(priv, t, TcptimerON);
  711. qunlock(&priv->tl);
  712. }
  713. static void
  714. tcphalt(Tcppriv *priv, Tcptimer *t)
  715. {
  716. if(t == nil)
  717. return;
  718. qlock(&priv->tl);
  719. timerstate(priv, t, TcptimerOFF);
  720. qunlock(&priv->tl);
  721. }
  722. static int
  723. backoff(int n)
  724. {
  725. return 1 << n;
  726. }
  727. static void
  728. localclose(Conv *s, char *reason) /* called with tcb locked */
  729. {
  730. Tcpctl *tcb;
  731. Tcppriv *tpriv;
  732. tpriv = s->p->priv;
  733. tcb = (Tcpctl *)s->ptcl;
  734. iphtrem(&tpriv->ht, s);
  735. tcphalt(tpriv, &tcb->timer);
  736. tcphalt(tpriv, &tcb->rtt_timer);
  737. tcphalt(tpriv, &tcb->acktimer);
  738. tcphalt(tpriv, &tcb->katimer);
  739. /* Flush reassembly queue; nothing more can arrive */
  740. dumpreseq(tcb);
  741. if(tcb->state == Syn_sent)
  742. Fsconnected(s, reason);
  743. if(s->state == Announced)
  744. wakeup(&s->listenr);
  745. qhangup(s->rq, reason);
  746. qhangup(s->wq, reason);
  747. tcpsetstate(s, Closed);
  748. }
  749. /* mtu (- TCP + IP hdr len) of 1st hop */
  750. static int
  751. tcpmtu(Proto *tcp, u8 *addr, int version, u32 *scale)
  752. {
  753. Ipifc *ifc;
  754. int mtu;
  755. ifc = findipifc(tcp->f, addr, 0);
  756. switch(version){
  757. default:
  758. case V4:
  759. mtu = DEF_MSS;
  760. if(ifc != nil)
  761. mtu = ifc->maxtu - ifc->medium->hsize - (TCP4_PKT + TCP4_HDRSIZE);
  762. break;
  763. case V6:
  764. mtu = DEF_MSS6;
  765. if(ifc != nil)
  766. mtu = ifc->maxtu - ifc->medium->hsize - (TCP6_PKT + TCP6_HDRSIZE);
  767. break;
  768. }
  769. /*
  770. * set the ws. it doesn't commit us to anything.
  771. * ws is the ultimate limit to the bandwidth-delay product.
  772. */
  773. *scale = Defadvscale;
  774. return mtu;
  775. }
  776. static void
  777. inittcpctl(Conv *s, int mode)
  778. {
  779. Tcpctl *tcb;
  780. Tcp4hdr *h4;
  781. Tcp6hdr *h6;
  782. Tcppriv *tpriv;
  783. int mss;
  784. tcb = (Tcpctl *)s->ptcl;
  785. memset(tcb, 0, sizeof(Tcpctl));
  786. tcb->ssthresh = QMAX; /* reset by tcpsetscale() */
  787. tcb->srtt = tcp_irtt << LOGAGAIN;
  788. tcb->mdev = 0;
  789. /* setup timers */
  790. tcb->timer.start = tcp_irtt / MSPTICK;
  791. tcb->timer.func = tcptimeout;
  792. tcb->timer.arg = s;
  793. tcb->rtt_timer.start = MAX_TIME;
  794. tcb->acktimer.start = TCP_ACK / MSPTICK;
  795. tcb->acktimer.func = tcpacktimer;
  796. tcb->acktimer.arg = s;
  797. tcb->katimer.start = DEF_KAT / MSPTICK;
  798. tcb->katimer.func = tcpkeepalive;
  799. tcb->katimer.arg = s;
  800. mss = DEF_MSS;
  801. /* create a prototype(pseudo) header */
  802. if(mode != TCP_LISTEN){
  803. if(ipcmp(s->laddr, IPnoaddr) == 0)
  804. findlocalip(s->p->f, s->laddr, s->raddr);
  805. switch(s->ipversion){
  806. case V4:
  807. h4 = &tcb->protohdr.tcp4hdr;
  808. memset(h4, 0, sizeof(*h4));
  809. h4->proto = IP_TCPPROTO;
  810. hnputs(h4->tcpsport, s->lport);
  811. hnputs(h4->tcpdport, s->rport);
  812. v6tov4(h4->tcpsrc, s->laddr);
  813. v6tov4(h4->tcpdst, s->raddr);
  814. break;
  815. case V6:
  816. h6 = &tcb->protohdr.tcp6hdr;
  817. memset(h6, 0, sizeof(*h6));
  818. h6->proto = IP_TCPPROTO;
  819. hnputs(h6->tcpsport, s->lport);
  820. hnputs(h6->tcpdport, s->rport);
  821. ipmove(h6->tcpsrc, s->laddr);
  822. ipmove(h6->tcpdst, s->raddr);
  823. mss = DEF_MSS6;
  824. break;
  825. default:
  826. panic("inittcpctl: version %d", s->ipversion);
  827. }
  828. }
  829. tcb->mss = tcb->cwind = mss;
  830. tcb->abcbytes = 0;
  831. tpriv = s->p->priv;
  832. tpriv->stats[Mss] = tcb->mss;
  833. /* default is no window scaling */
  834. tcpsetscale(s, tcb, 0, 0);
  835. }
  836. /*
  837. * called with s qlocked
  838. */
  839. static void
  840. tcpstart(Conv *s, int mode)
  841. {
  842. Tcpctl *tcb;
  843. Tcppriv *tpriv;
  844. char kpname[KNAMELEN];
  845. tpriv = s->p->priv;
  846. if(tpriv->ackprocstarted == 0){
  847. qlock(&tpriv->apl);
  848. if(tpriv->ackprocstarted == 0){
  849. snprint(kpname, sizeof kpname, "#I%dtcpack", s->p->f->dev);
  850. kproc(kpname, tcpackproc, s->p);
  851. tpriv->ackprocstarted = 1;
  852. }
  853. qunlock(&tpriv->apl);
  854. }
  855. tcb = (Tcpctl *)s->ptcl;
  856. inittcpctl(s, mode);
  857. iphtadd(&tpriv->ht, s);
  858. switch(mode){
  859. case TCP_LISTEN:
  860. tpriv->stats[PassiveOpens]++;
  861. tcb->flags |= CLONE;
  862. tcpsetstate(s, Listen);
  863. break;
  864. case TCP_CONNECT:
  865. tpriv->stats[ActiveOpens]++;
  866. tcb->flags |= ACTIVE;
  867. tcpsndsyn(s, tcb);
  868. tcpsetstate(s, Syn_sent);
  869. tcpoutput(s);
  870. break;
  871. }
  872. }
  873. #if 0
  874. static char*
  875. tcpflag(char *buf, char *e, u16 flag)
  876. {
  877. char *p;
  878. p = seprint(buf, e, "%d", flag>>10); /* Head len */
  879. if(flag & URG)
  880. p = seprint(p, e, " URG");
  881. if(flag & ACK)
  882. p = seprint(p, e, " ACK");
  883. if(flag & PSH)
  884. p = seprint(p, e, " PSH");
  885. if(flag & RST)
  886. p = seprint(p, e, " RST");
  887. if(flag & SYN)
  888. p = seprint(p, e, " SYN");
  889. if(flag & FIN)
  890. p = seprint(p, e, " FIN");
  891. USED(p);
  892. return buf;
  893. }
  894. #endif
  895. static Block *
  896. htontcp6(Tcp *tcph, Block *data, Tcp6hdr *ph, Tcpctl *tcb)
  897. {
  898. int dlen;
  899. Tcp6hdr *h;
  900. u16 csum;
  901. u16 hdrlen, optpad = 0;
  902. u8 *opt;
  903. hdrlen = TCP6_HDRSIZE;
  904. if(tcph->flags & SYN){
  905. if(tcph->mss)
  906. hdrlen += MSS_LENGTH;
  907. if(tcph->ws)
  908. hdrlen += WS_LENGTH;
  909. optpad = hdrlen & 3;
  910. if(optpad)
  911. optpad = 4 - optpad;
  912. hdrlen += optpad;
  913. }
  914. if(data){
  915. dlen = blocklen(data);
  916. data = padblock(data, hdrlen + TCP6_PKT);
  917. if(data == nil)
  918. return nil;
  919. } else {
  920. dlen = 0;
  921. data = allocb(hdrlen + TCP6_PKT + 64); /* the 64 pad is to meet mintu's */
  922. if(data == nil)
  923. return nil;
  924. data->wp += hdrlen + TCP6_PKT;
  925. }
  926. /* copy in pseudo ip header plus port numbers */
  927. h = (Tcp6hdr *)(data->rp);
  928. memmove(h, ph, TCP6_TCBPHDRSZ);
  929. /* compose pseudo tcp header, do cksum calculation */
  930. hnputl(h->vcf, hdrlen + dlen);
  931. h->ploadlen[0] = h->ploadlen[1] = h->proto = 0;
  932. h->ttl = ph->proto;
  933. /* copy in variable bits */
  934. hnputl(h->tcpseq, tcph->seq);
  935. hnputl(h->tcpack, tcph->ack);
  936. hnputs(h->tcpflag, (hdrlen << 10) | tcph->flags);
  937. hnputs(h->tcpwin, tcph->wnd >> (tcb != nil ? tcb->snd.scale : 0));
  938. hnputs(h->tcpurg, tcph->urg);
  939. if(tcph->flags & SYN){
  940. opt = h->tcpopt;
  941. if(tcph->mss != 0){
  942. *opt++ = MSSOPT;
  943. *opt++ = MSS_LENGTH;
  944. hnputs(opt, tcph->mss);
  945. opt += 2;
  946. }
  947. if(tcph->ws != 0){
  948. *opt++ = WSOPT;
  949. *opt++ = WS_LENGTH;
  950. *opt++ = tcph->ws;
  951. }
  952. while(optpad-- > 0)
  953. *opt++ = NOOPOPT;
  954. }
  955. if(tcb != nil && tcb->nochecksum){
  956. h->tcpcksum[0] = h->tcpcksum[1] = 0;
  957. } else {
  958. csum = ptclcsum(data, TCP6_IPLEN, hdrlen + dlen + TCP6_PHDRSIZE);
  959. hnputs(h->tcpcksum, csum);
  960. }
  961. /* move from pseudo header back to normal ip header */
  962. memset(h->vcf, 0, 4);
  963. h->vcf[0] = IP_VER6;
  964. hnputs(h->ploadlen, hdrlen + dlen);
  965. h->proto = ph->proto;
  966. return data;
  967. }
  968. static Block *
  969. htontcp4(Tcp *tcph, Block *data, Tcp4hdr *ph, Tcpctl *tcb)
  970. {
  971. int dlen;
  972. Tcp4hdr *h;
  973. u16 csum;
  974. u16 hdrlen, optpad = 0;
  975. u8 *opt;
  976. hdrlen = TCP4_HDRSIZE;
  977. if(tcph->flags & SYN){
  978. if(tcph->mss)
  979. hdrlen += MSS_LENGTH;
  980. if(1)
  981. hdrlen += WS_LENGTH;
  982. optpad = hdrlen & 3;
  983. if(optpad)
  984. optpad = 4 - optpad;
  985. hdrlen += optpad;
  986. }
  987. if(data){
  988. dlen = blocklen(data);
  989. data = padblock(data, hdrlen + TCP4_PKT);
  990. if(data == nil)
  991. return nil;
  992. } else {
  993. dlen = 0;
  994. data = allocb(hdrlen + TCP4_PKT + 64); /* the 64 pad is to meet mintu's */
  995. if(data == nil)
  996. return nil;
  997. data->wp += hdrlen + TCP4_PKT;
  998. }
  999. /* copy in pseudo ip header plus port numbers */
  1000. h = (Tcp4hdr *)(data->rp);
  1001. memmove(h, ph, TCP4_TCBPHDRSZ);
  1002. /* copy in variable bits */
  1003. hnputs(h->tcplen, hdrlen + dlen);
  1004. hnputl(h->tcpseq, tcph->seq);
  1005. hnputl(h->tcpack, tcph->ack);
  1006. hnputs(h->tcpflag, (hdrlen << 10) | tcph->flags);
  1007. hnputs(h->tcpwin, tcph->wnd >> (tcb != nil ? tcb->snd.scale : 0));
  1008. hnputs(h->tcpurg, tcph->urg);
  1009. if(tcph->flags & SYN){
  1010. opt = h->tcpopt;
  1011. if(tcph->mss != 0){
  1012. *opt++ = MSSOPT;
  1013. *opt++ = MSS_LENGTH;
  1014. hnputs(opt, tcph->mss);
  1015. opt += 2;
  1016. }
  1017. /* always offer. rfc1323 §2.2 */
  1018. if(1){
  1019. *opt++ = WSOPT;
  1020. *opt++ = WS_LENGTH;
  1021. *opt++ = tcph->ws;
  1022. }
  1023. while(optpad-- > 0)
  1024. *opt++ = NOOPOPT;
  1025. }
  1026. if(tcb != nil && tcb->nochecksum){
  1027. h->tcpcksum[0] = h->tcpcksum[1] = 0;
  1028. } else {
  1029. csum = ptclcsum(data, TCP4_IPLEN, hdrlen + dlen + TCP4_PHDRSIZE);
  1030. hnputs(h->tcpcksum, csum);
  1031. }
  1032. return data;
  1033. }
  1034. static int
  1035. ntohtcp6(Tcp *tcph, Block **bpp)
  1036. {
  1037. Tcp6hdr *h;
  1038. u8 *optr;
  1039. u16 hdrlen;
  1040. u16 optlen;
  1041. int n;
  1042. *bpp = pullupblock(*bpp, TCP6_PKT + TCP6_HDRSIZE);
  1043. if(*bpp == nil)
  1044. return -1;
  1045. h = (Tcp6hdr *)((*bpp)->rp);
  1046. tcph->source = nhgets(h->tcpsport);
  1047. tcph->dest = nhgets(h->tcpdport);
  1048. tcph->seq = nhgetl(h->tcpseq);
  1049. tcph->ack = nhgetl(h->tcpack);
  1050. hdrlen = (h->tcpflag[0] >> 2) & ~3;
  1051. if(hdrlen < TCP6_HDRSIZE){
  1052. freeblist(*bpp);
  1053. return -1;
  1054. }
  1055. tcph->flags = h->tcpflag[1];
  1056. tcph->wnd = nhgets(h->tcpwin);
  1057. tcph->urg = nhgets(h->tcpurg);
  1058. tcph->mss = 0;
  1059. tcph->ws = 0;
  1060. tcph->update = 0;
  1061. tcph->len = nhgets(h->ploadlen) - hdrlen;
  1062. *bpp = pullupblock(*bpp, hdrlen + TCP6_PKT);
  1063. if(*bpp == nil)
  1064. return -1;
  1065. optr = h->tcpopt;
  1066. n = hdrlen - TCP6_HDRSIZE;
  1067. while(n > 0 && *optr != EOLOPT){
  1068. if(*optr == NOOPOPT){
  1069. n--;
  1070. optr++;
  1071. continue;
  1072. }
  1073. optlen = optr[1];
  1074. if(optlen < 2 || optlen > n)
  1075. break;
  1076. switch(*optr){
  1077. case MSSOPT:
  1078. if(optlen == MSS_LENGTH)
  1079. tcph->mss = nhgets(optr + 2);
  1080. break;
  1081. case WSOPT:
  1082. if(optlen == WS_LENGTH && *(optr + 2) <= 14)
  1083. tcph->ws = *(optr + 2);
  1084. break;
  1085. }
  1086. n -= optlen;
  1087. optr += optlen;
  1088. }
  1089. return hdrlen;
  1090. }
  1091. static int
  1092. ntohtcp4(Tcp *tcph, Block **bpp)
  1093. {
  1094. Tcp4hdr *h;
  1095. u8 *optr;
  1096. u16 hdrlen;
  1097. u16 optlen;
  1098. int n;
  1099. *bpp = pullupblock(*bpp, TCP4_PKT + TCP4_HDRSIZE);
  1100. if(*bpp == nil)
  1101. return -1;
  1102. h = (Tcp4hdr *)((*bpp)->rp);
  1103. tcph->source = nhgets(h->tcpsport);
  1104. tcph->dest = nhgets(h->tcpdport);
  1105. tcph->seq = nhgetl(h->tcpseq);
  1106. tcph->ack = nhgetl(h->tcpack);
  1107. hdrlen = (h->tcpflag[0] >> 2) & ~3;
  1108. if(hdrlen < TCP4_HDRSIZE){
  1109. freeblist(*bpp);
  1110. return -1;
  1111. }
  1112. tcph->flags = h->tcpflag[1];
  1113. tcph->wnd = nhgets(h->tcpwin);
  1114. tcph->urg = nhgets(h->tcpurg);
  1115. tcph->mss = 0;
  1116. tcph->ws = 0;
  1117. tcph->update = 0;
  1118. tcph->len = nhgets(h->length) - (hdrlen + TCP4_PKT);
  1119. *bpp = pullupblock(*bpp, hdrlen + TCP4_PKT);
  1120. if(*bpp == nil)
  1121. return -1;
  1122. optr = h->tcpopt;
  1123. n = hdrlen - TCP4_HDRSIZE;
  1124. while(n > 0 && *optr != EOLOPT){
  1125. if(*optr == NOOPOPT){
  1126. n--;
  1127. optr++;
  1128. continue;
  1129. }
  1130. optlen = optr[1];
  1131. if(optlen < 2 || optlen > n)
  1132. break;
  1133. switch(*optr){
  1134. case MSSOPT:
  1135. if(optlen == MSS_LENGTH)
  1136. tcph->mss = nhgets(optr + 2);
  1137. break;
  1138. case WSOPT:
  1139. if(optlen == WS_LENGTH && *(optr + 2) <= 14)
  1140. tcph->ws = *(optr + 2);
  1141. break;
  1142. }
  1143. n -= optlen;
  1144. optr += optlen;
  1145. }
  1146. return hdrlen;
  1147. }
  1148. /*
  1149. * For outgoing calls, generate an initial sequence
  1150. * number and put a SYN on the send queue
  1151. */
  1152. static void
  1153. tcpsndsyn(Conv *s, Tcpctl *tcb)
  1154. {
  1155. Tcppriv *tpriv;
  1156. tcb->iss = (nrand(1 << 16) << 16) | nrand(1 << 16);
  1157. tcb->rttseq = tcb->iss;
  1158. tcb->snd.wl2 = tcb->iss;
  1159. tcb->snd.una = tcb->iss;
  1160. tcb->snd.rxt = tcb->iss;
  1161. tcb->snd.ptr = tcb->rttseq;
  1162. tcb->snd.nxt = tcb->rttseq;
  1163. tcb->flgcnt++;
  1164. tcb->flags |= FORCE;
  1165. tcb->sndsyntime = NOW;
  1166. /* set desired mss and scale */
  1167. tcb->mss = tcpmtu(s->p, s->laddr, s->ipversion, &tcb->scale);
  1168. tpriv = s->p->priv;
  1169. tpriv->stats[Mss] = tcb->mss;
  1170. }
  1171. void
  1172. sndrst(Proto *tcp, u8 *source, u8 *dest, u16 length, Tcp *seg,
  1173. u8 version, char *reason)
  1174. {
  1175. Block *hbp;
  1176. u8 rflags;
  1177. Tcppriv *tpriv;
  1178. Tcp4hdr ph4;
  1179. Tcp6hdr ph6;
  1180. netlog(tcp->f, Logtcp, "sndrst: %s\n", reason);
  1181. tpriv = tcp->priv;
  1182. if(seg->flags & RST)
  1183. return;
  1184. /* make pseudo header */
  1185. switch(version){
  1186. case V4:
  1187. memset(&ph4, 0, sizeof(ph4));
  1188. ph4.vihl = IP_VER4;
  1189. v6tov4(ph4.tcpsrc, dest);
  1190. v6tov4(ph4.tcpdst, source);
  1191. ph4.proto = IP_TCPPROTO;
  1192. hnputs(ph4.tcplen, TCP4_HDRSIZE);
  1193. hnputs(ph4.tcpsport, seg->dest);
  1194. hnputs(ph4.tcpdport, seg->source);
  1195. break;
  1196. case V6:
  1197. memset(&ph6, 0, sizeof(ph6));
  1198. ph6.vcf[0] = IP_VER6;
  1199. ipmove(ph6.tcpsrc, dest);
  1200. ipmove(ph6.tcpdst, source);
  1201. ph6.proto = IP_TCPPROTO;
  1202. hnputs(ph6.ploadlen, TCP6_HDRSIZE);
  1203. hnputs(ph6.tcpsport, seg->dest);
  1204. hnputs(ph6.tcpdport, seg->source);
  1205. break;
  1206. default:
  1207. panic("sndrst: version %d", version);
  1208. }
  1209. tpriv->stats[OutRsts]++;
  1210. rflags = RST;
  1211. /* convince the other end that this reset is in band */
  1212. if(seg->flags & ACK){
  1213. seg->seq = seg->ack;
  1214. seg->ack = 0;
  1215. } else {
  1216. rflags |= ACK;
  1217. seg->ack = seg->seq;
  1218. seg->seq = 0;
  1219. if(seg->flags & SYN)
  1220. seg->ack++;
  1221. seg->ack += length;
  1222. if(seg->flags & FIN)
  1223. seg->ack++;
  1224. }
  1225. seg->flags = rflags;
  1226. seg->wnd = 0;
  1227. seg->urg = 0;
  1228. seg->mss = 0;
  1229. seg->ws = 0;
  1230. switch(version){
  1231. case V4:
  1232. hbp = htontcp4(seg, nil, &ph4, nil);
  1233. if(hbp == nil)
  1234. return;
  1235. ipoput4(tcp->f, hbp, 0, MAXTTL, DFLTTOS, nil);
  1236. break;
  1237. case V6:
  1238. hbp = htontcp6(seg, nil, &ph6, nil);
  1239. if(hbp == nil)
  1240. return;
  1241. ipoput6(tcp->f, hbp, 0, MAXTTL, DFLTTOS, nil);
  1242. break;
  1243. default:
  1244. panic("sndrst2: version %d", version);
  1245. }
  1246. }
  1247. /*
  1248. * send a reset to the remote side and close the conversation
  1249. * called with s qlocked
  1250. */
  1251. static char *
  1252. tcphangup(Conv *s)
  1253. {
  1254. Proc *up = externup();
  1255. Tcp seg;
  1256. Tcpctl *tcb;
  1257. Block *hbp;
  1258. tcb = (Tcpctl *)s->ptcl;
  1259. if(waserror())
  1260. return commonerror();
  1261. if(ipcmp(s->raddr, IPnoaddr) != 0){
  1262. if(!waserror()){
  1263. memset(&seg, 0, sizeof seg);
  1264. seg.flags = RST | ACK;
  1265. seg.ack = tcb->rcv.nxt;
  1266. tcb->rcv.ackptr = seg.ack;
  1267. seg.seq = tcb->snd.ptr;
  1268. seg.wnd = 0;
  1269. seg.urg = 0;
  1270. seg.mss = 0;
  1271. seg.ws = 0;
  1272. switch(s->ipversion){
  1273. case V4:
  1274. tcb->protohdr.tcp4hdr.vihl = IP_VER4;
  1275. hbp = htontcp4(&seg, nil, &tcb->protohdr.tcp4hdr, tcb);
  1276. ipoput4(s->p->f, hbp, 0, s->ttl, s->tos, s);
  1277. break;
  1278. case V6:
  1279. tcb->protohdr.tcp6hdr.vcf[0] = IP_VER6;
  1280. hbp = htontcp6(&seg, nil, &tcb->protohdr.tcp6hdr, tcb);
  1281. ipoput6(s->p->f, hbp, 0, s->ttl, s->tos, s);
  1282. break;
  1283. default:
  1284. panic("tcphangup: version %d", s->ipversion);
  1285. }
  1286. poperror();
  1287. }
  1288. }
  1289. localclose(s, nil);
  1290. poperror();
  1291. return nil;
  1292. }
  1293. /*
  1294. * (re)send a SYN ACK
  1295. */
  1296. static int
  1297. sndsynack(Proto *tcp, Limbo *lp)
  1298. {
  1299. Block *hbp;
  1300. Tcp4hdr ph4;
  1301. Tcp6hdr ph6;
  1302. Tcp seg;
  1303. u32 scale;
  1304. /* make pseudo header */
  1305. switch(lp->version){
  1306. case V4:
  1307. memset(&ph4, 0, sizeof(ph4));
  1308. ph4.vihl = IP_VER4;
  1309. v6tov4(ph4.tcpsrc, lp->laddr);
  1310. v6tov4(ph4.tcpdst, lp->raddr);
  1311. ph4.proto = IP_TCPPROTO;
  1312. hnputs(ph4.tcplen, TCP4_HDRSIZE);
  1313. hnputs(ph4.tcpsport, lp->lport);
  1314. hnputs(ph4.tcpdport, lp->rport);
  1315. break;
  1316. case V6:
  1317. memset(&ph6, 0, sizeof(ph6));
  1318. ph6.vcf[0] = IP_VER6;
  1319. ipmove(ph6.tcpsrc, lp->laddr);
  1320. ipmove(ph6.tcpdst, lp->raddr);
  1321. ph6.proto = IP_TCPPROTO;
  1322. hnputs(ph6.ploadlen, TCP6_HDRSIZE);
  1323. hnputs(ph6.tcpsport, lp->lport);
  1324. hnputs(ph6.tcpdport, lp->rport);
  1325. break;
  1326. default:
  1327. panic("sndrst: version %d", lp->version);
  1328. }
  1329. memset(&seg, 0, sizeof seg);
  1330. seg.seq = lp->iss;
  1331. seg.ack = lp->irs + 1;
  1332. seg.flags = SYN | ACK;
  1333. seg.urg = 0;
  1334. seg.mss = tcpmtu(tcp, lp->laddr, lp->version, &scale);
  1335. seg.wnd = QMAX;
  1336. /* if the other side set scale, we should too */
  1337. if(lp->rcvscale){
  1338. seg.ws = scale;
  1339. lp->sndscale = scale;
  1340. } else {
  1341. seg.ws = 0;
  1342. lp->sndscale = 0;
  1343. }
  1344. switch(lp->version){
  1345. case V4:
  1346. hbp = htontcp4(&seg, nil, &ph4, nil);
  1347. if(hbp == nil)
  1348. return -1;
  1349. ipoput4(tcp->f, hbp, 0, MAXTTL, DFLTTOS, nil);
  1350. break;
  1351. case V6:
  1352. hbp = htontcp6(&seg, nil, &ph6, nil);
  1353. if(hbp == nil)
  1354. return -1;
  1355. ipoput6(tcp->f, hbp, 0, MAXTTL, DFLTTOS, nil);
  1356. break;
  1357. default:
  1358. panic("sndsnack: version %d", lp->version);
  1359. }
  1360. lp->lastsend = NOW;
  1361. return 0;
  1362. }
  1363. #define hashipa(a, p) (((a)[IPaddrlen - 2] + (a)[IPaddrlen - 1] + p) & LHTMASK)
  1364. /*
  1365. * put a call into limbo and respond with a SYN ACK
  1366. *
  1367. * called with proto locked
  1368. */
  1369. static void
  1370. limbo(Conv *s, u8 *source, u8 *dest, Tcp *seg, int version)
  1371. {
  1372. Limbo *lp, **l;
  1373. Tcppriv *tpriv;
  1374. int h;
  1375. tpriv = s->p->priv;
  1376. h = hashipa(source, seg->source);
  1377. for(l = &tpriv->lht[h]; *l != nil; l = &lp->next){
  1378. lp = *l;
  1379. if(lp->lport != seg->dest || lp->rport != seg->source || lp->version != version)
  1380. continue;
  1381. if(ipcmp(lp->raddr, source) != 0)
  1382. continue;
  1383. if(ipcmp(lp->laddr, dest) != 0)
  1384. continue;
  1385. /* each new SYN restarts the retransmits */
  1386. lp->irs = seg->seq;
  1387. break;
  1388. }
  1389. lp = *l;
  1390. if(lp == nil){
  1391. if(tpriv->nlimbo >= Maxlimbo && tpriv->lht[h]){
  1392. lp = tpriv->lht[h];
  1393. tpriv->lht[h] = lp->next;
  1394. lp->next = nil;
  1395. } else {
  1396. lp = malloc(sizeof(*lp));
  1397. if(lp == nil)
  1398. return;
  1399. tpriv->nlimbo++;
  1400. }
  1401. *l = lp;
  1402. lp->version = version;
  1403. ipmove(lp->laddr, dest);
  1404. ipmove(lp->raddr, source);
  1405. lp->lport = seg->dest;
  1406. lp->rport = seg->source;
  1407. lp->mss = seg->mss;
  1408. lp->rcvscale = seg->ws;
  1409. lp->irs = seg->seq;
  1410. lp->iss = (nrand(1 << 16) << 16) | nrand(1 << 16);
  1411. }
  1412. if(sndsynack(s->p, lp) < 0){
  1413. *l = lp->next;
  1414. tpriv->nlimbo--;
  1415. free(lp);
  1416. }
  1417. }
  1418. /*
  1419. * resend SYN ACK's once every SYNACK_RXTIMER ms.
  1420. */
  1421. static void
  1422. limborexmit(Proto *tcp)
  1423. {
  1424. Tcppriv *tpriv;
  1425. Limbo **l, *lp;
  1426. int h;
  1427. int seen;
  1428. u32 now;
  1429. tpriv = tcp->priv;
  1430. if(!canqlock(&tcp->ql))
  1431. return;
  1432. seen = 0;
  1433. now = NOW;
  1434. for(h = 0; h < NLHT && seen < tpriv->nlimbo; h++){
  1435. for(l = &tpriv->lht[h]; *l != nil && seen < tpriv->nlimbo;){
  1436. lp = *l;
  1437. seen++;
  1438. if(now - lp->lastsend < (lp->rexmits + 1) * SYNACK_RXTIMER)
  1439. continue;
  1440. /* time it out after 1 second */
  1441. if(++(lp->rexmits) > 5){
  1442. tpriv->nlimbo--;
  1443. *l = lp->next;
  1444. free(lp);
  1445. continue;
  1446. }
  1447. /* if we're being attacked, don't bother resending SYN ACK's */
  1448. if(tpriv->nlimbo > 100)
  1449. continue;
  1450. if(sndsynack(tcp, lp) < 0){
  1451. tpriv->nlimbo--;
  1452. *l = lp->next;
  1453. free(lp);
  1454. continue;
  1455. }
  1456. l = &lp->next;
  1457. }
  1458. }
  1459. qunlock(&tcp->ql);
  1460. }
  1461. /*
  1462. * lookup call in limbo. if found, throw it out.
  1463. *
  1464. * called with proto locked
  1465. */
  1466. static void
  1467. limborst(Conv *s, Tcp *segp, u8 *src, u8 *dst, u8 version)
  1468. {
  1469. Limbo *lp, **l;
  1470. int h;
  1471. Tcppriv *tpriv;
  1472. tpriv = s->p->priv;
  1473. /* find a call in limbo */
  1474. h = hashipa(src, segp->source);
  1475. for(l = &tpriv->lht[h]; *l != nil; l = &lp->next){
  1476. lp = *l;
  1477. if(lp->lport != segp->dest || lp->rport != segp->source || lp->version != version)
  1478. continue;
  1479. if(ipcmp(lp->laddr, dst) != 0)
  1480. continue;
  1481. if(ipcmp(lp->raddr, src) != 0)
  1482. continue;
  1483. /* RST can only follow the SYN */
  1484. if(segp->seq == lp->irs + 1){
  1485. tpriv->nlimbo--;
  1486. *l = lp->next;
  1487. free(lp);
  1488. }
  1489. break;
  1490. }
  1491. }
  1492. static void
  1493. initialwindow(Tcpctl *tcb)
  1494. {
  1495. /* RFC 3390 initial window */
  1496. if(tcb->mss < 1095)
  1497. tcb->cwind = 4 * tcb->mss;
  1498. else if(tcb->mss < 2190)
  1499. tcb->cwind = 2 * 2190;
  1500. else
  1501. tcb->cwind = 2 * tcb->mss;
  1502. }
  1503. /*
  1504. * come here when we finally get an ACK to our SYN-ACK.
  1505. * lookup call in limbo. if found, create a new conversation
  1506. *
  1507. * called with proto locked
  1508. */
  1509. static Conv *
  1510. tcpincoming(Conv *s, Tcp *segp, u8 *src, u8 *dst, u8 version)
  1511. {
  1512. Conv *new;
  1513. Tcpctl *tcb;
  1514. Tcppriv *tpriv;
  1515. Tcp4hdr *h4;
  1516. Tcp6hdr *h6;
  1517. Limbo *lp, **l;
  1518. int h;
  1519. /* unless it's just an ack, it can't be someone coming out of limbo */
  1520. if((segp->flags & SYN) || (segp->flags & ACK) == 0)
  1521. return nil;
  1522. tpriv = s->p->priv;
  1523. /* find a call in limbo */
  1524. h = hashipa(src, segp->source);
  1525. for(l = &tpriv->lht[h]; (lp = *l) != nil; l = &lp->next){
  1526. netlog(s->p->f, Logtcp, "tcpincoming s %I!%u/%I!%u d %I!%u/%I!%u v %d/%d\n",
  1527. src, segp->source, lp->raddr, lp->rport,
  1528. dst, segp->dest, lp->laddr, lp->lport,
  1529. version, lp->version);
  1530. if(lp->lport != segp->dest || lp->rport != segp->source || lp->version != version)
  1531. continue;
  1532. if(ipcmp(lp->laddr, dst) != 0)
  1533. continue;
  1534. if(ipcmp(lp->raddr, src) != 0)
  1535. continue;
  1536. /* we're assuming no data with the initial SYN */
  1537. if(segp->seq != lp->irs + 1 || segp->ack != lp->iss + 1){
  1538. netlog(s->p->f, Logtcp, "tcpincoming s %lx/%lx a %lx %lx\n",
  1539. segp->seq, lp->irs + 1, segp->ack, lp->iss + 1);
  1540. lp = nil;
  1541. } else {
  1542. tpriv->nlimbo--;
  1543. *l = lp->next;
  1544. }
  1545. break;
  1546. }
  1547. if(lp == nil)
  1548. return nil;
  1549. new = Fsnewcall(s, src, segp->source, dst, segp->dest, version);
  1550. if(new == nil)
  1551. return nil;
  1552. memmove(new->ptcl, s->ptcl, sizeof(Tcpctl));
  1553. tcb = (Tcpctl *)new->ptcl;
  1554. tcb->flags &= ~CLONE;
  1555. tcb->timer.arg = new;
  1556. tcb->timer.state = TcptimerOFF;
  1557. tcb->acktimer.arg = new;
  1558. tcb->acktimer.state = TcptimerOFF;
  1559. tcb->katimer.arg = new;
  1560. tcb->katimer.state = TcptimerOFF;
  1561. tcb->rtt_timer.arg = new;
  1562. tcb->rtt_timer.state = TcptimerOFF;
  1563. tcb->irs = lp->irs;
  1564. tcb->rcv.nxt = tcb->irs + 1;
  1565. tcb->rcv.wptr = tcb->rcv.nxt;
  1566. tcb->rcv.wsnt = 0;
  1567. tcb->rcv.urg = tcb->rcv.nxt;
  1568. tcb->iss = lp->iss;
  1569. tcb->rttseq = tcb->iss;
  1570. tcb->snd.wl2 = tcb->iss;
  1571. tcb->snd.una = tcb->iss + 1;
  1572. tcb->snd.ptr = tcb->iss + 1;
  1573. tcb->snd.nxt = tcb->iss + 1;
  1574. tcb->snd.rxt = tcb->iss + 1;
  1575. tcb->flgcnt = 0;
  1576. tcb->flags |= SYNACK;
  1577. /* set desired mss and scale */
  1578. tcb->mss = tcpmtu(s->p, s->laddr, s->ipversion, &tcb->scale);
  1579. /* our sending max segment size cannot be bigger than what he asked for */
  1580. if(lp->mss != 0 && lp->mss < tcb->mss){
  1581. tcb->mss = lp->mss;
  1582. }
  1583. tpriv->stats[Mss] = tcb->mss;
  1584. /* window scaling */
  1585. tcpsetscale(new, tcb, lp->rcvscale, lp->sndscale);
  1586. /* congestion window */
  1587. tcb->snd.wnd = segp->wnd;
  1588. initialwindow(tcb);
  1589. /* set initial round trip time */
  1590. tcb->sndsyntime = lp->lastsend + lp->rexmits * SYNACK_RXTIMER;
  1591. tcpsynackrtt(new);
  1592. free(lp);
  1593. /* set up proto header */
  1594. switch(version){
  1595. case V4:
  1596. h4 = &tcb->protohdr.tcp4hdr;
  1597. memset(h4, 0, sizeof(*h4));
  1598. h4->proto = IP_TCPPROTO;
  1599. hnputs(h4->tcpsport, new->lport);
  1600. hnputs(h4->tcpdport, new->rport);
  1601. v6tov4(h4->tcpsrc, dst);
  1602. v6tov4(h4->tcpdst, src);
  1603. break;
  1604. case V6:
  1605. h6 = &tcb->protohdr.tcp6hdr;
  1606. memset(h6, 0, sizeof(*h6));
  1607. h6->proto = IP_TCPPROTO;
  1608. hnputs(h6->tcpsport, new->lport);
  1609. hnputs(h6->tcpdport, new->rport);
  1610. ipmove(h6->tcpsrc, dst);
  1611. ipmove(h6->tcpdst, src);
  1612. break;
  1613. default:
  1614. panic("tcpincoming: version %d", new->ipversion);
  1615. }
  1616. tcpsetstate(new, Established);
  1617. iphtadd(&tpriv->ht, new);
  1618. return new;
  1619. }
  1620. static int
  1621. seq_within(u32 x, u32 low, u32 high)
  1622. {
  1623. if(low <= high){
  1624. if(low <= x && x <= high)
  1625. return 1;
  1626. } else {
  1627. if(x >= low || x <= high)
  1628. return 1;
  1629. }
  1630. return 0;
  1631. }
  1632. static int
  1633. seq_lt(u32 x, u32 y)
  1634. {
  1635. return (int)(x - y) < 0;
  1636. }
  1637. static int
  1638. seq_le(u32 x, u32 y)
  1639. {
  1640. return (int)(x - y) <= 0;
  1641. }
  1642. static int
  1643. seq_gt(u32 x, u32 y)
  1644. {
  1645. return (int)(x - y) > 0;
  1646. }
  1647. static int
  1648. seq_ge(u32 x, u32 y)
  1649. {
  1650. return (int)(x - y) >= 0;
  1651. }
  1652. /*
  1653. * use the time between the first SYN and it's ack as the
  1654. * initial round trip time
  1655. */
  1656. static void
  1657. tcpsynackrtt(Conv *s)
  1658. {
  1659. Tcpctl *tcb;
  1660. int delta;
  1661. Tcppriv *tpriv;
  1662. tcb = (Tcpctl *)s->ptcl;
  1663. tpriv = s->p->priv;
  1664. delta = NOW - tcb->sndsyntime;
  1665. tcb->srtt = delta << LOGAGAIN;
  1666. tcb->mdev = delta << LOGDGAIN;
  1667. /* halt round trip timer */
  1668. tcphalt(tpriv, &tcb->rtt_timer);
  1669. }
  1670. static void
  1671. update(Conv *s, Tcp *seg)
  1672. {
  1673. int rtt, delta;
  1674. Tcpctl *tcb;
  1675. u32 acked;
  1676. Tcppriv *tpriv;
  1677. if(seg->update)
  1678. return;
  1679. seg->update = 1;
  1680. tpriv = s->p->priv;
  1681. tcb = (Tcpctl *)s->ptcl;
  1682. /* catch zero-window updates, update window & recover */
  1683. if(tcb->snd.wnd == 0 && seg->wnd > 0 &&
  1684. seq_lt(seg->ack, tcb->snd.ptr)){
  1685. netlog(s->p->f, Logtcp, "tcp: zwu ack %lu una %lu ptr %lu win %lu\n",
  1686. seg->ack, tcb->snd.una, tcb->snd.ptr, seg->wnd);
  1687. tcb->snd.wnd = seg->wnd;
  1688. goto recovery;
  1689. }
  1690. /* newreno fast retransmit */
  1691. if(seg->ack == tcb->snd.una && tcb->snd.una != tcb->snd.nxt &&
  1692. ++tcb->snd.dupacks == 3) { /* was TCPREXMTTHRESH */
  1693. recovery:
  1694. if(tcb->snd.recovery){
  1695. tpriv->stats[RecoveryCwind]++;
  1696. tcb->cwind += tcb->mss;
  1697. } else if(seq_le(tcb->snd.rxt, seg->ack)){
  1698. tpriv->stats[Recovery]++;
  1699. tcb->abcbytes = 0;
  1700. tcb->snd.recovery = 1;
  1701. tcb->snd.partialack = 0;
  1702. tcb->snd.rxt = tcb->snd.nxt;
  1703. tcpcongestion(tcb);
  1704. tcb->cwind = tcb->ssthresh + 3 * tcb->mss;
  1705. netlog(s->p->f, Logtcpwin, "recovery inflate %ld ss %ld @%lu\n",
  1706. tcb->cwind, tcb->ssthresh, tcb->snd.rxt);
  1707. tcprxmit(s);
  1708. } else {
  1709. tpriv->stats[RecoveryNoSeq]++;
  1710. netlog(s->p->f, Logtcpwin, "!recov %lu not ≤ %lu %ld\n",
  1711. tcb->snd.rxt, seg->ack, tcb->snd.rxt - seg->ack);
  1712. /* don't enter fast retransmit, don't change ssthresh */
  1713. }
  1714. } else if(tcb->snd.recovery){
  1715. tpriv->stats[RecoveryCwind]++;
  1716. tcb->cwind += tcb->mss;
  1717. }
  1718. /*
  1719. * update window
  1720. */
  1721. if(seq_gt(seg->ack, tcb->snd.wl2) || (tcb->snd.wl2 == seg->ack && seg->wnd > tcb->snd.wnd)){
  1722. /* clear dupack if we advance wl2 */
  1723. if(tcb->snd.wl2 != seg->ack)
  1724. tcb->snd.dupacks = 0;
  1725. tcb->snd.wnd = seg->wnd;
  1726. tcb->snd.wl2 = seg->ack;
  1727. }
  1728. if(!seq_gt(seg->ack, tcb->snd.una)){
  1729. /*
  1730. * don't let us hangup if sending into a closed window and
  1731. * we're still getting acks
  1732. */
  1733. if((tcb->flags & RETRAN) && tcb->snd.wnd == 0)
  1734. tcb->backedoff = MAXBACKMS / 4;
  1735. return;
  1736. }
  1737. /* Compute the new send window size */
  1738. acked = seg->ack - tcb->snd.una;
  1739. /* avoid slow start and timers for SYN acks */
  1740. if((tcb->flags & SYNACK) == 0){
  1741. tcb->flags |= SYNACK;
  1742. acked--;
  1743. tcb->flgcnt--;
  1744. goto done;
  1745. }
  1746. /*
  1747. * congestion control
  1748. */
  1749. if(tcb->snd.recovery){
  1750. if(seq_ge(seg->ack, tcb->snd.rxt)){
  1751. /* recovery finished; deflate window */
  1752. tpriv->stats[RecoveryDone]++;
  1753. tcb->snd.dupacks = 0;
  1754. tcb->snd.recovery = 0;
  1755. tcb->cwind = (tcb->snd.nxt - tcb->snd.una) + tcb->mss;
  1756. if(tcb->ssthresh < tcb->cwind)
  1757. tcb->cwind = tcb->ssthresh;
  1758. netlog(s->p->f, Logtcpwin, "recovery deflate %ld %ld\n",
  1759. tcb->cwind, tcb->ssthresh);
  1760. } else {
  1761. /* partial ack; we lost more than one segment */
  1762. tpriv->stats[RecoveryPA]++;
  1763. if(tcb->cwind > acked)
  1764. tcb->cwind -= acked;
  1765. else {
  1766. netlog(s->p->f, Logtcpwin, "partial ack neg\n");
  1767. tcb->cwind = tcb->mss;
  1768. }
  1769. netlog(s->p->f, Logtcpwin, "partial ack %ld left %ld cwind %ld\n",
  1770. acked, tcb->snd.rxt - seg->ack, tcb->cwind);
  1771. if(acked >= tcb->mss)
  1772. tcb->cwind += tcb->mss;
  1773. tcb->snd.partialack++;
  1774. }
  1775. } else
  1776. tcpabcincr(tcb, acked);
  1777. /* Adjust the timers according to the round trip time */
  1778. /* TODO: fix sloppy treatment of overflow cases here. */
  1779. if(tcb->rtt_timer.state == TcptimerON && seq_ge(seg->ack, tcb->rttseq)){
  1780. tcphalt(tpriv, &tcb->rtt_timer);
  1781. if((tcb->flags & RETRAN) == 0){
  1782. tcb->backoff = 0;
  1783. tcb->backedoff = 0;
  1784. rtt = tcb->rtt_timer.start - tcb->rtt_timer.count;
  1785. if(rtt == 0)
  1786. rtt = 1; /* else all close sys's will rexmit in 0 time */
  1787. rtt *= MSPTICK;
  1788. if(tcb->srtt == 0){
  1789. tcb->srtt = rtt << LOGAGAIN;
  1790. tcb->mdev = rtt << LOGDGAIN;
  1791. } else {
  1792. delta = rtt - (tcb->srtt >> LOGAGAIN);
  1793. tcb->srtt += delta;
  1794. if(tcb->srtt <= 0)
  1795. tcb->srtt = 1;
  1796. delta = abs(delta) - (tcb->mdev >> LOGDGAIN);
  1797. tcb->mdev += delta;
  1798. if(tcb->mdev <= 0)
  1799. tcb->mdev = 1;
  1800. }
  1801. tcpsettimer(tcb);
  1802. }
  1803. }
  1804. done:
  1805. if(qdiscard(s->wq, acked) < acked)
  1806. tcb->flgcnt--;
  1807. tcb->snd.una = seg->ack;
  1808. /* newreno fast recovery */
  1809. if(tcb->snd.recovery)
  1810. tcprxmit(s);
  1811. if(seq_gt(seg->ack, tcb->snd.urg))
  1812. tcb->snd.urg = seg->ack;
  1813. if(tcb->snd.una != tcb->snd.nxt){
  1814. /* `impatient' variant */
  1815. if(!tcb->snd.recovery || tcb->snd.partialack == 1){
  1816. tcb->time = NOW;
  1817. tcb->timeuna = tcb->snd.una;
  1818. tcpgo(tpriv, &tcb->timer);
  1819. }
  1820. } else
  1821. tcphalt(tpriv, &tcb->timer);
  1822. if(seq_lt(tcb->snd.ptr, tcb->snd.una))
  1823. tcb->snd.ptr = tcb->snd.una;
  1824. if(!tcb->snd.recovery)
  1825. tcb->flags &= ~RETRAN;
  1826. tcb->backoff = 0;
  1827. tcb->backedoff = 0;
  1828. }
  1829. static void
  1830. tcpiput(Proto *tcp, Ipifc *ipifc, Block *bp)
  1831. {
  1832. Proc *up = externup();
  1833. Tcp seg;
  1834. Tcp4hdr *h4;
  1835. Tcp6hdr *h6;
  1836. int hdrlen;
  1837. Tcpctl *tcb;
  1838. u16 length, csum;
  1839. u8 source[IPaddrlen], dest[IPaddrlen];
  1840. Conv *s;
  1841. Fs *f;
  1842. Tcppriv *tpriv;
  1843. u8 version;
  1844. f = tcp->f;
  1845. tpriv = tcp->priv;
  1846. tpriv->stats[InSegs]++;
  1847. h4 = (Tcp4hdr *)(bp->rp);
  1848. h6 = (Tcp6hdr *)(bp->rp);
  1849. if((h4->vihl & 0xF0) == IP_VER4){
  1850. version = V4;
  1851. length = nhgets(h4->length);
  1852. v4tov6(dest, h4->tcpdst);
  1853. v4tov6(source, h4->tcpsrc);
  1854. h4->Unused = 0;
  1855. hnputs(h4->tcplen, length - TCP4_PKT);
  1856. if(!(bp->flag & Btcpck) && (h4->tcpcksum[0] || h4->tcpcksum[1]) &&
  1857. ptclcsum(bp, TCP4_IPLEN, length - TCP4_IPLEN)){
  1858. tpriv->stats[CsumErrs]++;
  1859. tpriv->stats[InErrs]++;
  1860. netlog(f, Logtcp, "bad tcp proto cksum\n");
  1861. freeblist(bp);
  1862. return;
  1863. }
  1864. hdrlen = ntohtcp4(&seg, &bp);
  1865. if(hdrlen < 0){
  1866. tpriv->stats[HlenErrs]++;
  1867. tpriv->stats[InErrs]++;
  1868. netlog(f, Logtcp, "bad tcp hdr len\n");
  1869. return;
  1870. }
  1871. /* trim the packet to the size claimed by the datagram */
  1872. length -= hdrlen + TCP4_PKT;
  1873. bp = trimblock(bp, hdrlen + TCP4_PKT, length);
  1874. if(bp == nil){
  1875. tpriv->stats[LenErrs]++;
  1876. tpriv->stats[InErrs]++;
  1877. netlog(f, Logtcp, "tcp len < 0 after trim\n");
  1878. return;
  1879. }
  1880. } else {
  1881. int ttl = h6->ttl;
  1882. int proto = h6->proto;
  1883. version = V6;
  1884. length = nhgets(h6->ploadlen);
  1885. ipmove(dest, h6->tcpdst);
  1886. ipmove(source, h6->tcpsrc);
  1887. h6->ploadlen[0] = h6->ploadlen[1] = h6->proto = 0;
  1888. h6->ttl = proto;
  1889. hnputl(h6->vcf, length);
  1890. if((h6->tcpcksum[0] || h6->tcpcksum[1]) &&
  1891. (csum = ptclcsum(bp, TCP6_IPLEN, length + TCP6_PHDRSIZE)) != 0){
  1892. tpriv->stats[CsumErrs]++;
  1893. tpriv->stats[InErrs]++;
  1894. netlog(f, Logtcp,
  1895. "bad tcpv6 proto cksum: got %#x, computed %#x\n",
  1896. h6->tcpcksum[0] << 8 | h6->tcpcksum[1], csum);
  1897. freeblist(bp);
  1898. return;
  1899. }
  1900. h6->ttl = ttl;
  1901. h6->proto = proto;
  1902. hnputs(h6->ploadlen, length);
  1903. hdrlen = ntohtcp6(&seg, &bp);
  1904. if(hdrlen < 0){
  1905. tpriv->stats[HlenErrs]++;
  1906. tpriv->stats[InErrs]++;
  1907. netlog(f, Logtcp, "bad tcpv6 hdr len\n");
  1908. return;
  1909. }
  1910. /* trim the packet to the size claimed by the datagram */
  1911. length -= hdrlen;
  1912. bp = trimblock(bp, hdrlen + TCP6_PKT, length);
  1913. if(bp == nil){
  1914. tpriv->stats[LenErrs]++;
  1915. tpriv->stats[InErrs]++;
  1916. netlog(f, Logtcp, "tcpv6 len < 0 after trim\n");
  1917. return;
  1918. }
  1919. }
  1920. /* lock protocol while searching for a conversation */
  1921. qlock(&tcp->ql);
  1922. /* Look for a matching conversation */
  1923. s = iphtlook(&tpriv->ht, source, seg.source, dest, seg.dest);
  1924. if(s == nil){
  1925. netlog(f, Logtcp, "iphtlook(src %I!%d, dst %I!%d) failed\n",
  1926. source, seg.source, dest, seg.dest);
  1927. reset:
  1928. qunlock(&tcp->ql);
  1929. sndrst(tcp, source, dest, length, &seg, version, "no conversation");
  1930. freeblist(bp);
  1931. return;
  1932. }
  1933. /* if it's a listener, look for the right flags and get a new conv */
  1934. tcb = (Tcpctl *)s->ptcl;
  1935. if(tcb->state == Listen){
  1936. if(seg.flags & RST){
  1937. limborst(s, &seg, source, dest, version);
  1938. qunlock(&tcp->ql);
  1939. freeblist(bp);
  1940. return;
  1941. }
  1942. /* if this is a new SYN, put the call into limbo */
  1943. if((seg.flags & SYN) && (seg.flags & ACK) == 0){
  1944. limbo(s, source, dest, &seg, version);
  1945. qunlock(&tcp->ql);
  1946. freeblist(bp);
  1947. return;
  1948. }
  1949. /*
  1950. * if there's a matching call in limbo, tcpincoming will
  1951. * return it in state Syn_received
  1952. */
  1953. s = tcpincoming(s, &seg, source, dest, version);
  1954. if(s == nil)
  1955. goto reset;
  1956. }
  1957. /* The rest of the input state machine is run with the control block
  1958. * locked and implements the state machine directly out of the RFC.
  1959. * Out-of-band data is ignored - it was always a bad idea.
  1960. */
  1961. tcb = (Tcpctl *)s->ptcl;
  1962. if(waserror()){
  1963. qunlock(&s->ql);
  1964. nexterror();
  1965. }
  1966. qlock(&s->ql);
  1967. qunlock(&tcp->ql);
  1968. /* fix up window */
  1969. seg.wnd <<= tcb->rcv.scale;
  1970. /* every input packet in puts off the keep alive time out */
  1971. tcpsetkacounter(tcb);
  1972. switch(tcb->state){
  1973. case Closed:
  1974. sndrst(tcp, source, dest, length, &seg, version, "sending to Closed");
  1975. goto raise;
  1976. case Syn_sent:
  1977. if(seg.flags & ACK){
  1978. if(!seq_within(seg.ack, tcb->iss + 1, tcb->snd.nxt)){
  1979. sndrst(tcp, source, dest, length, &seg, version,
  1980. "bad seq in Syn_sent");
  1981. goto raise;
  1982. }
  1983. }
  1984. if(seg.flags & RST){
  1985. if(seg.flags & ACK)
  1986. localclose(s, Econrefused);
  1987. goto raise;
  1988. }
  1989. if(seg.flags & SYN){
  1990. procsyn(s, &seg);
  1991. if(seg.flags & ACK){
  1992. update(s, &seg);
  1993. tcpsynackrtt(s);
  1994. tcpsetstate(s, Established);
  1995. tcpsetscale(s, tcb, seg.ws, tcb->scale);
  1996. } else {
  1997. tcb->time = NOW;
  1998. tcpsetstate(s, Syn_received); /* DLP - shouldn't this be a reset? */
  1999. }
  2000. if(length != 0 || (seg.flags & FIN))
  2001. break;
  2002. freeblist(bp);
  2003. goto output;
  2004. } else
  2005. freeblist(bp);
  2006. qunlock(&s->ql);
  2007. poperror();
  2008. return;
  2009. case Syn_received:
  2010. /* doesn't matter if it's the correct ack, we're just trying to set timing */
  2011. if(seg.flags & ACK)
  2012. tcpsynackrtt(s);
  2013. break;
  2014. }
  2015. /*
  2016. * One DOS attack is to open connections to us and then forget about them,
  2017. * thereby tying up a conv at no long term cost to the attacker.
  2018. * This is an attempt to defeat these stateless DOS attacks. See
  2019. * corresponding code in tcpsendka().
  2020. */
  2021. if(tcb->state != Syn_received && (seg.flags & RST) == 0){
  2022. if(tcpporthogdefense && seq_within(seg.ack, tcb->snd.una - (1 << 31), tcb->snd.una - (1 << 29))){
  2023. print("stateless hog %I.%d->%I.%d f %x %lx - %lx - %lx\n",
  2024. source, seg.source, dest, seg.dest, seg.flags,
  2025. tcb->snd.una - (1 << 31), seg.ack, tcb->snd.una - (1 << 29));
  2026. localclose(s, "stateless hog");
  2027. }
  2028. }
  2029. /* Cut the data to fit the receive window */
  2030. tcprcvwin(s);
  2031. if(tcptrim(tcb, &seg, &bp, &length) == -1){
  2032. if(seg.seq + 1 != tcb->rcv.nxt || length != 1)
  2033. netlog(f, Logtcp, "tcp: trim: !inwind: seq %lu-%lu win "
  2034. "%lu-%lu l %d from %I\n",
  2035. seg.seq,
  2036. seg.seq + length - 1, tcb->rcv.nxt,
  2037. tcb->rcv.nxt + tcb->rcv.wnd - 1, length, s->raddr);
  2038. update(s, &seg);
  2039. if(qlen(s->wq) + tcb->flgcnt == 0 && tcb->state == Closing){
  2040. tcphalt(tpriv, &tcb->rtt_timer);
  2041. tcphalt(tpriv, &tcb->acktimer);
  2042. tcphalt(tpriv, &tcb->katimer);
  2043. tcpsetstate(s, Time_wait);
  2044. tcb->timer.start = MSL2 * (1000 / MSPTICK);
  2045. tcpgo(tpriv, &tcb->timer);
  2046. }
  2047. if(!(seg.flags & RST)){
  2048. tcb->flags |= FORCE;
  2049. goto output;
  2050. }
  2051. qunlock(&s->ql);
  2052. poperror();
  2053. return;
  2054. }
  2055. /* Cannot accept so answer with a rst */
  2056. if(length && tcb->state == Closed){
  2057. sndrst(tcp, source, dest, length, &seg, version, "sending to Closed");
  2058. goto raise;
  2059. }
  2060. /* The segment is beyond the current receive pointer so
  2061. * queue the data in the resequence queue
  2062. */
  2063. if(seg.seq != tcb->rcv.nxt)
  2064. if(length != 0 || (seg.flags & (SYN | FIN))){
  2065. update(s, &seg);
  2066. if(addreseq(f, tcb, tpriv, &seg, bp, length) < 0)
  2067. print("reseq %I.%d -> %I.%d\n", s->raddr, s->rport,
  2068. s->laddr, s->lport);
  2069. tcb->flags |= FORCE; /* force duplicate ack; RFC 5681 §3.2 */
  2070. goto output;
  2071. }
  2072. if(tcb->nreseq > 0)
  2073. tcb->flags |= FORCE; /* filled hole in seq. space; RFC 5681 §3.2 */
  2074. /*
  2075. * keep looping till we've processed this packet plus any
  2076. * adjacent packets in the resequence queue
  2077. */
  2078. for(;;){
  2079. if(seg.flags & RST){
  2080. if(tcb->state == Established){
  2081. tpriv->stats[EstabResets]++;
  2082. if(tcb->rcv.nxt != seg.seq)
  2083. netlog(f, Logtcp, "out of order RST "
  2084. "rcvd: %I.%d -> %I.%d, rcv.nxt "
  2085. "%lx seq %lx\n",
  2086. s->raddr, s->rport, s->laddr,
  2087. s->lport, tcb->rcv.nxt, seg.seq);
  2088. }
  2089. localclose(s, Econrefused);
  2090. goto raise;
  2091. }
  2092. if((seg.flags & ACK) == 0)
  2093. goto raise;
  2094. switch(tcb->state){
  2095. case Syn_received:
  2096. if(!seq_within(seg.ack, tcb->snd.una + 1, tcb->snd.nxt)){
  2097. sndrst(tcp, source, dest, length, &seg, version,
  2098. "bad seq in Syn_received");
  2099. goto raise;
  2100. }
  2101. update(s, &seg);
  2102. tcpsetstate(s, Established);
  2103. case Established:
  2104. case Close_wait:
  2105. update(s, &seg);
  2106. break;
  2107. case Finwait1:
  2108. update(s, &seg);
  2109. if(qlen(s->wq) + tcb->flgcnt == 0){
  2110. tcphalt(tpriv, &tcb->rtt_timer);
  2111. tcphalt(tpriv, &tcb->acktimer);
  2112. tcpsetkacounter(tcb);
  2113. tcb->time = NOW;
  2114. tcpsetstate(s, Finwait2);
  2115. tcb->katimer.start = MSL2 * (1000 / MSPTICK);
  2116. tcpgo(tpriv, &tcb->katimer);
  2117. }
  2118. break;
  2119. case Finwait2:
  2120. update(s, &seg);
  2121. break;
  2122. case Closing:
  2123. update(s, &seg);
  2124. if(qlen(s->wq) + tcb->flgcnt == 0){
  2125. tcphalt(tpriv, &tcb->rtt_timer);
  2126. tcphalt(tpriv, &tcb->acktimer);
  2127. tcphalt(tpriv, &tcb->katimer);
  2128. tcpsetstate(s, Time_wait);
  2129. tcb->timer.start = MSL2 * (1000 / MSPTICK);
  2130. tcpgo(tpriv, &tcb->timer);
  2131. }
  2132. break;
  2133. case Last_ack:
  2134. update(s, &seg);
  2135. if(qlen(s->wq) + tcb->flgcnt == 0){
  2136. localclose(s, nil);
  2137. goto raise;
  2138. }
  2139. case Time_wait:
  2140. tcb->flags |= FORCE;
  2141. if(tcb->timer.state != TcptimerON)
  2142. tcpgo(tpriv, &tcb->timer);
  2143. }
  2144. if((seg.flags & URG) && seg.urg){
  2145. if(seq_gt(seg.urg + seg.seq, tcb->rcv.urg)){
  2146. tcb->rcv.urg = seg.urg + seg.seq;
  2147. pullblock(&bp, seg.urg);
  2148. }
  2149. } else if(seq_gt(tcb->rcv.nxt, tcb->rcv.urg))
  2150. tcb->rcv.urg = tcb->rcv.nxt;
  2151. if(length == 0){
  2152. if(bp != nil)
  2153. freeblist(bp);
  2154. } else {
  2155. switch(tcb->state){
  2156. default:
  2157. /* Ignore segment text */
  2158. if(bp != nil)
  2159. freeblist(bp);
  2160. break;
  2161. case Syn_received:
  2162. case Established:
  2163. case Finwait1:
  2164. /* If we still have some data place on
  2165. * receive queue
  2166. */
  2167. if(bp){
  2168. bp = packblock(bp);
  2169. if(bp == nil)
  2170. panic("tcp packblock");
  2171. qpassnolim(s->rq, bp);
  2172. bp = nil;
  2173. }
  2174. tcb->rcv.nxt += length;
  2175. /*
  2176. * turn on the acktimer if there's something
  2177. * to ack
  2178. */
  2179. if(tcb->acktimer.state != TcptimerON)
  2180. tcpgo(tpriv, &tcb->acktimer);
  2181. break;
  2182. case Finwait2:
  2183. /* no process to read the data, send a reset */
  2184. if(bp != nil)
  2185. freeblist(bp);
  2186. sndrst(tcp, source, dest, length, &seg, version,
  2187. "send to Finwait2");
  2188. qunlock(&s->ql);
  2189. poperror();
  2190. return;
  2191. }
  2192. }
  2193. if(seg.flags & FIN){
  2194. tcb->flags |= FORCE;
  2195. switch(tcb->state){
  2196. case Syn_received:
  2197. case Established:
  2198. tcb->rcv.nxt++;
  2199. tcpsetstate(s, Close_wait);
  2200. break;
  2201. case Finwait1:
  2202. tcb->rcv.nxt++;
  2203. if(qlen(s->wq) + tcb->flgcnt == 0){
  2204. tcphalt(tpriv, &tcb->rtt_timer);
  2205. tcphalt(tpriv, &tcb->acktimer);
  2206. tcphalt(tpriv, &tcb->katimer);
  2207. tcpsetstate(s, Time_wait);
  2208. tcb->timer.start = MSL2 * (1000 / MSPTICK);
  2209. tcpgo(tpriv, &tcb->timer);
  2210. } else
  2211. tcpsetstate(s, Closing);
  2212. break;
  2213. case Finwait2:
  2214. tcb->rcv.nxt++;
  2215. tcphalt(tpriv, &tcb->rtt_timer);
  2216. tcphalt(tpriv, &tcb->acktimer);
  2217. tcphalt(tpriv, &tcb->katimer);
  2218. tcpsetstate(s, Time_wait);
  2219. tcb->timer.start = MSL2 * (1000 / MSPTICK);
  2220. tcpgo(tpriv, &tcb->timer);
  2221. break;
  2222. case Close_wait:
  2223. case Closing:
  2224. case Last_ack:
  2225. break;
  2226. case Time_wait:
  2227. tcpgo(tpriv, &tcb->timer);
  2228. break;
  2229. }
  2230. }
  2231. /*
  2232. * get next adjacent segment from the resequence queue.
  2233. * dump/trim any overlapping segments
  2234. */
  2235. for(;;){
  2236. if(tcb->reseq == nil)
  2237. goto output;
  2238. if(seq_ge(tcb->rcv.nxt, tcb->reseq->seg.seq) == 0)
  2239. goto output;
  2240. getreseq(tcb, &seg, &bp, &length);
  2241. tcprcvwin(s);
  2242. if(tcptrim(tcb, &seg, &bp, &length) == 0){
  2243. tcb->flags |= FORCE;
  2244. break;
  2245. }
  2246. }
  2247. }
  2248. output:
  2249. tcpoutput(s);
  2250. qunlock(&s->ql);
  2251. poperror();
  2252. return;
  2253. raise:
  2254. qunlock(&s->ql);
  2255. poperror();
  2256. freeblist(bp);
  2257. tcpkick(s);
  2258. }
  2259. /*
  2260. * always enters and exits with the s locked. We drop
  2261. * the lock to ipoput the packet so some care has to be
  2262. * taken by callers.
  2263. */
  2264. static void
  2265. tcpoutput(Conv *s)
  2266. {
  2267. Tcp seg;
  2268. u32 msgs;
  2269. Tcpctl *tcb;
  2270. Block *hbp, *bp;
  2271. int sndcnt;
  2272. u32 ssize, dsize, sent;
  2273. Fs *f;
  2274. Tcppriv *tpriv;
  2275. u8 version;
  2276. f = s->p->f;
  2277. tpriv = s->p->priv;
  2278. version = s->ipversion;
  2279. tcb = (Tcpctl *)s->ptcl;
  2280. /* force ack every 2*mss */
  2281. if((tcb->flags & FORCE) == 0 &&
  2282. tcb->rcv.nxt - tcb->rcv.ackptr >= 2 * tcb->mss){
  2283. tpriv->stats[Delayack]++;
  2284. tcb->flags |= FORCE;
  2285. }
  2286. /* force ack if window opening */
  2287. if((tcb->flags & FORCE) == 0){
  2288. tcprcvwin(s);
  2289. if((int)(tcb->rcv.wptr - tcb->rcv.wsnt) >= 2 * tcb->mss){
  2290. tpriv->stats[Wopenack]++;
  2291. tcb->flags |= FORCE;
  2292. }
  2293. }
  2294. for(msgs = 0; msgs < 100; msgs++){
  2295. switch(tcb->state){
  2296. case Listen:
  2297. case Closed:
  2298. case Finwait2:
  2299. return;
  2300. }
  2301. /* Don't send anything else until our SYN has been acked */
  2302. if(tcb->snd.ptr != tcb->iss && (tcb->flags & SYNACK) == 0)
  2303. break;
  2304. /* force an ack when a window has opened up */
  2305. tcprcvwin(s);
  2306. if(tcb->rcv.blocked && tcb->rcv.wnd > 0){
  2307. tcb->rcv.blocked = 0;
  2308. tcb->flags |= FORCE;
  2309. }
  2310. sndcnt = qlen(s->wq) + tcb->flgcnt;
  2311. sent = tcb->snd.ptr - tcb->snd.una;
  2312. ssize = sndcnt;
  2313. if(tcb->snd.wnd == 0){
  2314. /* zero window probe */
  2315. if(sent > 0 && !(tcb->flags & FORCE))
  2316. break; /* already probing, rto re-probes */
  2317. if(ssize < sent)
  2318. ssize = 0;
  2319. else {
  2320. ssize -= sent;
  2321. if(ssize > 0)
  2322. ssize = 1;
  2323. }
  2324. } else {
  2325. /* calculate usable segment size */
  2326. if(ssize > tcb->cwind)
  2327. ssize = tcb->cwind;
  2328. if(ssize > tcb->snd.wnd)
  2329. ssize = tcb->snd.wnd;
  2330. if(ssize < sent)
  2331. ssize = 0;
  2332. else {
  2333. ssize -= sent;
  2334. if(ssize > tcb->mss)
  2335. ssize = tcb->mss;
  2336. }
  2337. }
  2338. dsize = ssize;
  2339. seg.urg = 0;
  2340. if(!(tcb->flags & FORCE))
  2341. if(ssize == 0 ||
  2342. (ssize < tcb->mss && tcb->snd.nxt == tcb->snd.ptr &&
  2343. sent > TCPREXMTTHRESH * tcb->mss))
  2344. break;
  2345. tcb->flags &= ~FORCE;
  2346. /* By default we will generate an ack */
  2347. tcphalt(tpriv, &tcb->acktimer);
  2348. seg.source = s->lport;
  2349. seg.dest = s->rport;
  2350. seg.flags = ACK;
  2351. seg.mss = 0;
  2352. seg.ws = 0;
  2353. seg.update = 0;
  2354. switch(tcb->state){
  2355. case Syn_sent:
  2356. seg.flags = 0;
  2357. if(tcb->snd.ptr == tcb->iss){
  2358. seg.flags |= SYN;
  2359. dsize--;
  2360. seg.mss = tcb->mss;
  2361. seg.ws = tcb->scale;
  2362. }
  2363. break;
  2364. case Syn_received:
  2365. /*
  2366. * don't send any data with a SYN/ACK packet
  2367. * because Linux rejects the packet in its
  2368. * attempt to solve the SYN attack problem
  2369. */
  2370. if(tcb->snd.ptr == tcb->iss){
  2371. seg.flags |= SYN;
  2372. dsize = 0;
  2373. ssize = 1;
  2374. seg.mss = tcb->mss;
  2375. seg.ws = tcb->scale;
  2376. }
  2377. break;
  2378. }
  2379. seg.seq = tcb->snd.ptr;
  2380. seg.ack = tcb->rcv.nxt;
  2381. seg.wnd = tcb->rcv.wnd;
  2382. /* Pull out data to send */
  2383. bp = nil;
  2384. if(dsize != 0){
  2385. bp = qcopy(s->wq, dsize, sent);
  2386. if(BLEN(bp) != dsize){
  2387. seg.flags |= FIN;
  2388. dsize--;
  2389. }
  2390. }
  2391. if(sent + dsize == sndcnt && dsize)
  2392. seg.flags |= PSH;
  2393. tcb->snd.ptr += ssize;
  2394. /* Pull up the send pointer so we can accept acks
  2395. * for this window
  2396. */
  2397. if(seq_gt(tcb->snd.ptr, tcb->snd.nxt))
  2398. tcb->snd.nxt = tcb->snd.ptr;
  2399. /* Build header, link data and compute cksum */
  2400. switch(version){
  2401. case V4:
  2402. tcb->protohdr.tcp4hdr.vihl = IP_VER4;
  2403. hbp = htontcp4(&seg, bp, &tcb->protohdr.tcp4hdr, tcb);
  2404. if(hbp == nil){
  2405. freeblist(bp);
  2406. return;
  2407. }
  2408. break;
  2409. case V6:
  2410. tcb->protohdr.tcp6hdr.vcf[0] = IP_VER6;
  2411. hbp = htontcp6(&seg, bp, &tcb->protohdr.tcp6hdr, tcb);
  2412. if(hbp == nil){
  2413. freeblist(bp);
  2414. return;
  2415. }
  2416. break;
  2417. default:
  2418. hbp = nil; /* to suppress a warning */
  2419. panic("tcpoutput: version %d", version);
  2420. }
  2421. /* Start the transmission timers if there is new data and we
  2422. * expect acknowledges
  2423. */
  2424. if(ssize != 0){
  2425. if(tcb->timer.state != TcptimerON){
  2426. tcb->time = NOW;
  2427. tcb->timeuna = tcb->snd.una;
  2428. tcpgo(tpriv, &tcb->timer);
  2429. }
  2430. /* If round trip timer isn't running, start it.
  2431. * measure the longest packet only in case the
  2432. * transmission time dominates RTT
  2433. */
  2434. if(tcb->snd.retransmit == 0)
  2435. if(tcb->rtt_timer.state != TcptimerON)
  2436. if(ssize == tcb->mss){
  2437. tcpgo(tpriv, &tcb->rtt_timer);
  2438. tcb->rttseq = tcb->snd.ptr;
  2439. }
  2440. }
  2441. tpriv->stats[OutSegs]++;
  2442. if(tcb->snd.retransmit)
  2443. tpriv->stats[RetransSegsSent]++;
  2444. tcb->rcv.ackptr = seg.ack;
  2445. tcb->rcv.wsnt = tcb->rcv.wptr;
  2446. /* put off the next keep alive */
  2447. tcpgo(tpriv, &tcb->katimer);
  2448. switch(version){
  2449. case V4:
  2450. if(ipoput4(f, hbp, 0, s->ttl, s->tos, s) < 0){
  2451. /* a negative return means no route */
  2452. localclose(s, "no route");
  2453. }
  2454. break;
  2455. case V6:
  2456. if(ipoput6(f, hbp, 0, s->ttl, s->tos, s) < 0){
  2457. /* a negative return means no route */
  2458. localclose(s, "no route");
  2459. }
  2460. break;
  2461. default:
  2462. panic("tcpoutput2: version %d", version);
  2463. }
  2464. if((msgs % 4) == 3){
  2465. qunlock(&s->ql);
  2466. qlock(&s->ql);
  2467. }
  2468. }
  2469. }
  2470. /*
  2471. * the BSD convention (hack?) for keep alives. resend last u8 acked.
  2472. */
  2473. static void
  2474. tcpsendka(Conv *s)
  2475. {
  2476. Tcp seg;
  2477. Tcpctl *tcb;
  2478. Block *hbp, *dbp;
  2479. tcb = (Tcpctl *)s->ptcl;
  2480. dbp = nil;
  2481. memset(&seg, 0, sizeof seg);
  2482. seg.urg = 0;
  2483. seg.source = s->lport;
  2484. seg.dest = s->rport;
  2485. seg.flags = ACK | PSH;
  2486. seg.mss = 0;
  2487. seg.ws = 0;
  2488. if(tcpporthogdefense)
  2489. seg.seq = tcb->snd.una - (1 << 30) - nrand(1 << 20);
  2490. else
  2491. seg.seq = tcb->snd.una - 1;
  2492. seg.ack = tcb->rcv.nxt;
  2493. tcb->rcv.ackptr = seg.ack;
  2494. tcprcvwin(s);
  2495. seg.wnd = tcb->rcv.wnd;
  2496. if(tcb->state == Finwait2){
  2497. seg.flags |= FIN;
  2498. } else {
  2499. dbp = allocb(1);
  2500. dbp->wp++;
  2501. }
  2502. if(isv4(s->raddr)){
  2503. /* Build header, link data and compute cksum */
  2504. tcb->protohdr.tcp4hdr.vihl = IP_VER4;
  2505. hbp = htontcp4(&seg, dbp, &tcb->protohdr.tcp4hdr, tcb);
  2506. if(hbp == nil){
  2507. freeblist(dbp);
  2508. return;
  2509. }
  2510. ipoput4(s->p->f, hbp, 0, s->ttl, s->tos, s);
  2511. } else {
  2512. /* Build header, link data and compute cksum */
  2513. tcb->protohdr.tcp6hdr.vcf[0] = IP_VER6;
  2514. hbp = htontcp6(&seg, dbp, &tcb->protohdr.tcp6hdr, tcb);
  2515. if(hbp == nil){
  2516. freeblist(dbp);
  2517. return;
  2518. }
  2519. ipoput6(s->p->f, hbp, 0, s->ttl, s->tos, s);
  2520. }
  2521. }
  2522. /*
  2523. * set connection to time out after 12 minutes
  2524. */
  2525. static void
  2526. tcpsetkacounter(Tcpctl *tcb)
  2527. {
  2528. tcb->kacounter = (12 * 60 * 1000) / (tcb->katimer.start * MSPTICK);
  2529. if(tcb->kacounter < 3)
  2530. tcb->kacounter = 3;
  2531. }
  2532. /*
  2533. * if we've timed out, close the connection
  2534. * otherwise, send a keepalive and restart the timer
  2535. */
  2536. static void
  2537. tcpkeepalive(void *v)
  2538. {
  2539. Proc *up = externup();
  2540. Tcpctl *tcb;
  2541. Conv *s;
  2542. s = v;
  2543. tcb = (Tcpctl *)s->ptcl;
  2544. if(waserror()){
  2545. qunlock(&s->ql);
  2546. nexterror();
  2547. }
  2548. qlock(&s->ql);
  2549. if(tcb->state != Closed){
  2550. if(--(tcb->kacounter) <= 0){
  2551. localclose(s, Etimedout);
  2552. } else {
  2553. tcpsendka(s);
  2554. tcpgo(s->p->priv, &tcb->katimer);
  2555. }
  2556. }
  2557. qunlock(&s->ql);
  2558. poperror();
  2559. }
  2560. /*
  2561. * start keepalive timer
  2562. */
  2563. static char *
  2564. tcpstartka(Conv *s, char **f, int n)
  2565. {
  2566. Tcpctl *tcb;
  2567. int x;
  2568. tcb = (Tcpctl *)s->ptcl;
  2569. if(tcb->state != Established)
  2570. return "connection must be in Establised state";
  2571. if(n > 1){
  2572. x = atoi(f[1]);
  2573. if(x >= MSPTICK)
  2574. tcb->katimer.start = x / MSPTICK;
  2575. }
  2576. tcpsetkacounter(tcb);
  2577. tcpgo(s->p->priv, &tcb->katimer);
  2578. return nil;
  2579. }
  2580. /*
  2581. * turn checksums on/off
  2582. */
  2583. static char *
  2584. tcpsetchecksum(Conv *s, char **f, int i)
  2585. {
  2586. Tcpctl *tcb;
  2587. tcb = (Tcpctl *)s->ptcl;
  2588. tcb->nochecksum = !atoi(f[1]);
  2589. return nil;
  2590. }
  2591. /*
  2592. * retransmit (at most) one segment at snd.una.
  2593. * preserve cwind & snd.ptr
  2594. */
  2595. static void
  2596. tcprxmit(Conv *s)
  2597. {
  2598. Tcpctl *tcb;
  2599. Tcppriv *tpriv;
  2600. u32 tcwind, tptr;
  2601. tcb = (Tcpctl *)s->ptcl;
  2602. tcb->flags |= RETRAN | FORCE;
  2603. tptr = tcb->snd.ptr;
  2604. tcwind = tcb->cwind;
  2605. tcb->snd.ptr = tcb->snd.una;
  2606. tcb->cwind = tcb->mss;
  2607. tcb->snd.retransmit = 1;
  2608. tcpoutput(s);
  2609. tcb->snd.retransmit = 0;
  2610. tcb->cwind = tcwind;
  2611. tcb->snd.ptr = tptr;
  2612. tpriv = s->p->priv;
  2613. tpriv->stats[RetransSegs]++;
  2614. }
  2615. /*
  2616. * TODO: RFC 4138 F-RTO
  2617. */
  2618. static void
  2619. tcptimeout(void *arg)
  2620. {
  2621. Proc *up = externup();
  2622. Conv *s;
  2623. Tcpctl *tcb;
  2624. int maxback;
  2625. Tcppriv *tpriv;
  2626. s = (Conv *)arg;
  2627. tpriv = s->p->priv;
  2628. tcb = (Tcpctl *)s->ptcl;
  2629. if(waserror()){
  2630. qunlock(&s->ql);
  2631. nexterror();
  2632. }
  2633. qlock(&s->ql);
  2634. switch(tcb->state){
  2635. default:
  2636. tcb->backoff++;
  2637. if(tcb->state == Syn_sent)
  2638. maxback = MAXBACKMS / 2;
  2639. else
  2640. maxback = MAXBACKMS;
  2641. tcb->backedoff += tcb->timer.start * MSPTICK;
  2642. if(tcb->backedoff >= maxback){
  2643. localclose(s, Etimedout);
  2644. break;
  2645. }
  2646. netlog(s->p->f, Logtcprxmt, "rxm %d/%d %ldms %lu rto %d %lu %s\n",
  2647. tcb->srtt, tcb->mdev, NOW - tcb->time,
  2648. tcb->snd.una - tcb->timeuna, tcb->snd.rto, tcb->snd.ptr,
  2649. tcpstates[s->state]);
  2650. tcpsettimer(tcb);
  2651. if(tcb->snd.rto == 0)
  2652. tcpcongestion(tcb);
  2653. tcprxmit(s);
  2654. tcb->snd.ptr = tcb->snd.una;
  2655. tcb->cwind = tcb->mss;
  2656. tcb->snd.rto = 1;
  2657. tpriv->stats[RetransTimeouts]++;
  2658. if(tcb->snd.recovery){
  2659. tcb->snd.dupacks = 0; /* reno rto */
  2660. tcb->snd.recovery = 0;
  2661. tpriv->stats[RecoveryRTO]++;
  2662. tcb->snd.rxt = tcb->snd.nxt;
  2663. netlog(s->p->f, Logtcpwin,
  2664. "rto recovery rxt @%lu\n", tcb->snd.nxt);
  2665. }
  2666. tcb->abcbytes = 0;
  2667. break;
  2668. case Time_wait:
  2669. localclose(s, nil);
  2670. break;
  2671. case Closed:
  2672. break;
  2673. }
  2674. qunlock(&s->ql);
  2675. poperror();
  2676. }
  2677. static int
  2678. inwindow(Tcpctl *tcb, int seq)
  2679. {
  2680. return seq_within(seq, tcb->rcv.nxt, tcb->rcv.nxt + tcb->rcv.wnd - 1);
  2681. }
  2682. /*
  2683. * set up state for a received SYN (or SYN ACK) packet
  2684. */
  2685. static void
  2686. procsyn(Conv *s, Tcp *seg)
  2687. {
  2688. Tcpctl *tcb;
  2689. Tcppriv *tpriv;
  2690. tcb = (Tcpctl *)s->ptcl;
  2691. tcb->flags |= FORCE;
  2692. tcb->rcv.nxt = seg->seq + 1;
  2693. tcb->rcv.wptr = tcb->rcv.nxt;
  2694. tcb->rcv.wsnt = 0;
  2695. tcb->rcv.urg = tcb->rcv.nxt;
  2696. tcb->irs = seg->seq;
  2697. /* our sending max segment size cannot be bigger than what he asked for */
  2698. if(seg->mss != 0 && seg->mss < tcb->mss){
  2699. tcb->mss = seg->mss;
  2700. tpriv = s->p->priv;
  2701. tpriv->stats[Mss] = tcb->mss;
  2702. }
  2703. tcb->snd.wnd = seg->wnd;
  2704. initialwindow(tcb);
  2705. }
  2706. static int
  2707. dumpreseq(Tcpctl *tcb)
  2708. {
  2709. Reseq *r, *next;
  2710. for(r = tcb->reseq; r != nil; r = next){
  2711. next = r->next;
  2712. freeblist(r->bp);
  2713. free(r);
  2714. }
  2715. tcb->reseq = nil;
  2716. tcb->nreseq = 0;
  2717. tcb->reseqlen = 0;
  2718. return -1;
  2719. }
  2720. static void
  2721. logreseq(Fs *f, Reseq *r, u32 n)
  2722. {
  2723. char *s;
  2724. for(; r != nil; r = r->next){
  2725. s = nil;
  2726. if(r->next == nil && r->seg.seq != n)
  2727. s = "hole/end";
  2728. else if(r->next == nil)
  2729. s = "end";
  2730. else if(r->seg.seq != n)
  2731. s = "hole";
  2732. if(s != nil)
  2733. netlog(f, Logtcp, "%s %lu-%lu (%ld) %#x\n", s,
  2734. n, r->seg.seq, r->seg.seq - n, r->seg.flags);
  2735. n = r->seg.seq + r->seg.len;
  2736. }
  2737. }
  2738. static int
  2739. addreseq(Fs *f, Tcpctl *tcb, Tcppriv *tpriv, Tcp *seg, Block *bp, u16 length)
  2740. {
  2741. Reseq *rp, **rr;
  2742. int qmax;
  2743. rp = malloc(sizeof *rp);
  2744. if(rp == nil){
  2745. freeblist(bp); /* bp always consumed by addreseq */
  2746. return 0;
  2747. }
  2748. rp->seg = *seg;
  2749. rp->bp = bp;
  2750. rp->length = length;
  2751. tcb->reseqlen += length;
  2752. tcb->nreseq++;
  2753. /* Place on reassembly list sorting by starting seq number */
  2754. for(rr = &tcb->reseq;; rr = &(*rr)->next)
  2755. if(*rr == nil || seq_lt(seg->seq, (*rr)->seg.seq)){
  2756. rp->next = *rr;
  2757. *rr = rp;
  2758. tpriv->stats[Resequenced]++;
  2759. if(rp->next != nil)
  2760. tpriv->stats[OutOfOrder]++;
  2761. break;
  2762. }
  2763. qmax = tcb->window;
  2764. if(tcb->reseqlen > qmax){
  2765. netlog(f, Logtcp, "tcp: reseq: queue > window: %d > %d; %d packets\n",
  2766. tcb->reseqlen, qmax, tcb->nreseq);
  2767. logreseq(f, tcb->reseq, tcb->rcv.nxt);
  2768. tpriv->stats[ReseqBytelim]++;
  2769. return dumpreseq(tcb);
  2770. }
  2771. qmax = tcb->window / tcb->mss; /* ~190 for qscale=2, 390 for qscale=3 */
  2772. if(tcb->nreseq > qmax){
  2773. netlog(f, Logtcp, "resequence queue > packets: %d %d; %d bytes\n",
  2774. tcb->nreseq, qmax, tcb->reseqlen);
  2775. logreseq(f, tcb->reseq, tcb->rcv.nxt);
  2776. tpriv->stats[ReseqPktlim]++;
  2777. return dumpreseq(tcb);
  2778. }
  2779. return 0;
  2780. }
  2781. static void
  2782. getreseq(Tcpctl *tcb, Tcp *seg, Block **bp, u16 *length)
  2783. {
  2784. Reseq *rp;
  2785. rp = tcb->reseq;
  2786. if(rp == nil)
  2787. return;
  2788. tcb->reseq = rp->next;
  2789. *seg = rp->seg;
  2790. *bp = rp->bp;
  2791. *length = rp->length;
  2792. tcb->nreseq--;
  2793. tcb->reseqlen -= rp->length;
  2794. free(rp);
  2795. }
  2796. static int
  2797. tcptrim(Tcpctl *tcb, Tcp *seg, Block **bp, u16 *length)
  2798. {
  2799. u16 len;
  2800. u8 accept;
  2801. int dupcnt, excess;
  2802. accept = 0;
  2803. len = *length;
  2804. if(seg->flags & SYN)
  2805. len++;
  2806. if(seg->flags & FIN)
  2807. len++;
  2808. if(tcb->rcv.wnd == 0){
  2809. if(len == 0 && seg->seq == tcb->rcv.nxt)
  2810. return 0;
  2811. } else {
  2812. /* Some part of the segment should be in the window */
  2813. if(inwindow(tcb, seg->seq))
  2814. accept++;
  2815. else if(len != 0){
  2816. if(inwindow(tcb, seg->seq + len - 1) ||
  2817. seq_within(tcb->rcv.nxt, seg->seq, seg->seq + len - 1))
  2818. accept++;
  2819. }
  2820. }
  2821. if(!accept){
  2822. freeblist(*bp);
  2823. return -1;
  2824. }
  2825. dupcnt = tcb->rcv.nxt - seg->seq;
  2826. if(dupcnt > 0){
  2827. tcb->rerecv += dupcnt;
  2828. if(seg->flags & SYN){
  2829. seg->flags &= ~SYN;
  2830. seg->seq++;
  2831. if(seg->urg > 1)
  2832. seg->urg--;
  2833. else
  2834. seg->flags &= ~URG;
  2835. dupcnt--;
  2836. }
  2837. if(dupcnt > 0){
  2838. pullblock(bp, (u16)dupcnt);
  2839. seg->seq += dupcnt;
  2840. *length -= dupcnt;
  2841. if(seg->urg > dupcnt)
  2842. seg->urg -= dupcnt;
  2843. else {
  2844. seg->flags &= ~URG;
  2845. seg->urg = 0;
  2846. }
  2847. }
  2848. }
  2849. excess = seg->seq + *length - (tcb->rcv.nxt + tcb->rcv.wnd);
  2850. if(excess > 0){
  2851. tcb->rerecv += excess;
  2852. *length -= excess;
  2853. *bp = trimblock(*bp, 0, *length);
  2854. if(*bp == nil)
  2855. panic("presotto is a boofhead");
  2856. seg->flags &= ~FIN;
  2857. }
  2858. return 0;
  2859. }
  2860. static void
  2861. tcpadvise(Proto *tcp, Block *bp, char *msg)
  2862. {
  2863. Tcp4hdr *h4;
  2864. Tcp6hdr *h6;
  2865. Tcpctl *tcb;
  2866. u8 source[IPaddrlen];
  2867. u8 dest[IPaddrlen];
  2868. u16 psource, pdest;
  2869. Conv *s, **p;
  2870. h4 = (Tcp4hdr *)(bp->rp);
  2871. h6 = (Tcp6hdr *)(bp->rp);
  2872. if((h4->vihl & 0xF0) == IP_VER4){
  2873. v4tov6(dest, h4->tcpdst);
  2874. v4tov6(source, h4->tcpsrc);
  2875. psource = nhgets(h4->tcpsport);
  2876. pdest = nhgets(h4->tcpdport);
  2877. } else {
  2878. ipmove(dest, h6->tcpdst);
  2879. ipmove(source, h6->tcpsrc);
  2880. psource = nhgets(h6->tcpsport);
  2881. pdest = nhgets(h6->tcpdport);
  2882. }
  2883. /* Look for a connection */
  2884. qlock(&tcp->ql);
  2885. for(p = tcp->conv; *p; p++){
  2886. s = *p;
  2887. tcb = (Tcpctl *)s->ptcl;
  2888. if(s->rport == pdest)
  2889. if(s->lport == psource)
  2890. if(tcb->state != Closed)
  2891. if(ipcmp(s->raddr, dest) == 0)
  2892. if(ipcmp(s->laddr, source) == 0){
  2893. qlock(&s->ql);
  2894. qunlock(&tcp->ql);
  2895. switch(tcb->state){
  2896. case Syn_sent:
  2897. localclose(s, msg);
  2898. break;
  2899. }
  2900. qunlock(&s->ql);
  2901. freeblist(bp);
  2902. return;
  2903. }
  2904. }
  2905. qunlock(&tcp->ql);
  2906. freeblist(bp);
  2907. }
  2908. static char *
  2909. tcpporthogdefensectl(char *val)
  2910. {
  2911. if(strcmp(val, "on") == 0)
  2912. tcpporthogdefense = 1;
  2913. else if(strcmp(val, "off") == 0)
  2914. tcpporthogdefense = 0;
  2915. else
  2916. return "unknown value for tcpporthogdefense";
  2917. return nil;
  2918. }
  2919. /* called with c qlocked */
  2920. static char *
  2921. tcpctl(Conv *c, char **f, int n)
  2922. {
  2923. if(n == 1 && strcmp(f[0], "hangup") == 0)
  2924. return tcphangup(c);
  2925. if(n >= 1 && strcmp(f[0], "keepalive") == 0)
  2926. return tcpstartka(c, f, n);
  2927. if(n >= 1 && strcmp(f[0], "checksum") == 0)
  2928. return tcpsetchecksum(c, f, n);
  2929. if(n >= 1 && strcmp(f[0], "tcpporthogdefense") == 0)
  2930. return tcpporthogdefensectl(f[1]);
  2931. return "unknown control request";
  2932. }
  2933. static int
  2934. tcpstats(Proto *tcp, char *buf, int len)
  2935. {
  2936. Tcppriv *priv;
  2937. char *p, *e;
  2938. int i;
  2939. priv = tcp->priv;
  2940. p = buf;
  2941. e = p + len;
  2942. for(i = 0; i < Nstats; i++)
  2943. p = seprint(p, e, "%s: %llu\n", statnames[i], priv->stats[i]);
  2944. return p - buf;
  2945. }
  2946. /*
  2947. * garbage collect any stale conversations:
  2948. * - SYN received but no SYN-ACK after 5 seconds (could be the SYN attack)
  2949. * - Finwait2 after 5 minutes
  2950. *
  2951. * this is called whenever we run out of channels. Both checks are
  2952. * of questionable validity so we try to use them only when we're
  2953. * up against the wall.
  2954. */
  2955. static int
  2956. tcpgc(Proto *tcp)
  2957. {
  2958. Conv *c, **pp, **ep;
  2959. int n;
  2960. Tcpctl *tcb;
  2961. n = 0;
  2962. ep = &tcp->conv[tcp->nc];
  2963. for(pp = tcp->conv; pp < ep; pp++){
  2964. c = *pp;
  2965. if(c == nil)
  2966. break;
  2967. if(!canqlock(&c->ql))
  2968. continue;
  2969. tcb = (Tcpctl *)c->ptcl;
  2970. switch(tcb->state){
  2971. case Syn_received:
  2972. if(NOW - tcb->time > 5000){
  2973. localclose(c, Etimedout);
  2974. n++;
  2975. }
  2976. break;
  2977. case Finwait2:
  2978. if(NOW - tcb->time > 5 * 60 * 1000){
  2979. localclose(c, Etimedout);
  2980. n++;
  2981. }
  2982. break;
  2983. }
  2984. qunlock(&c->ql);
  2985. }
  2986. return n;
  2987. }
  2988. static void
  2989. tcpsettimer(Tcpctl *tcb)
  2990. {
  2991. int x;
  2992. /* round trip dependency */
  2993. x = backoff(tcb->backoff) *
  2994. (tcb->mdev + (tcb->srtt >> LOGAGAIN) + MSPTICK) / MSPTICK;
  2995. /* bounded twixt 0.3 and 64 seconds */
  2996. if(x < 300 / MSPTICK)
  2997. x = 300 / MSPTICK;
  2998. else if(x > (64000 / MSPTICK))
  2999. x = 64000 / MSPTICK;
  3000. tcb->timer.start = x;
  3001. }
  3002. void
  3003. tcpinit(Fs *fs)
  3004. {
  3005. Proto *tcp;
  3006. Tcppriv *tpriv;
  3007. tcp = smalloc(sizeof(Proto));
  3008. tpriv = tcp->priv = smalloc(sizeof(Tcppriv));
  3009. tcp->name = "tcp";
  3010. tcp->connect = tcpconnect;
  3011. tcp->announce = tcpannounce;
  3012. tcp->ctl = tcpctl;
  3013. tcp->state = tcpstate;
  3014. tcp->create = tcpcreate;
  3015. tcp->close = tcpclose;
  3016. tcp->rcv = tcpiput;
  3017. tcp->advise = tcpadvise;
  3018. tcp->stats = tcpstats;
  3019. tcp->inuse = tcpinuse;
  3020. tcp->gc = tcpgc;
  3021. tcp->ipproto = IP_TCPPROTO;
  3022. tcp->nc = scalednconv();
  3023. tcp->ptclsize = sizeof(Tcpctl);
  3024. tpriv->stats[MaxConn] = tcp->nc;
  3025. Fsproto(fs, tcp);
  3026. }
  3027. static void
  3028. tcpsetscale(Conv *s, Tcpctl *tcb, u16 rcvscale, u16 sndscale)
  3029. {
  3030. /*
  3031. * guess at reasonable queue sizes. there's no current way
  3032. * to know how many nic receive buffers we can safely tie up in the
  3033. * tcp stack, and we don't adjust our queues to maximize throughput
  3034. * and minimize bufferbloat. n.b. the offer (rcvscale) needs to be
  3035. * respected, but we still control our own buffer commitment by
  3036. * keeping a seperate qscale.
  3037. */
  3038. tcb->rcv.scale = rcvscale & 0xff;
  3039. tcb->snd.scale = sndscale & 0xff;
  3040. tcb->qscale = rcvscale & 0xff;
  3041. if(rcvscale > Maxqscale)
  3042. tcb->qscale = Maxqscale;
  3043. if(rcvscale != tcb->rcv.scale)
  3044. netlog(s->p->f, Logtcp, "tcpsetscale: window %lu "
  3045. "qlen %d >> window %u lport %d\n",
  3046. tcb->window, qlen(s->rq), QMAX << tcb->qscale, s->lport);
  3047. tcb->window = QMAX << tcb->qscale;
  3048. tcb->ssthresh = tcb->window;
  3049. /*
  3050. * it's important to set wq large enough to cover the full
  3051. * bandwidth-delay product. it's possible to be in loss
  3052. * recovery with a big window, and we need to keep sending
  3053. * into the inflated window. the difference can be huge
  3054. * for even modest (70ms) ping times.
  3055. */
  3056. qsetlimit(s->rq, tcb->window);
  3057. qsetlimit(s->wq, tcb->window);
  3058. tcprcvwin(s);
  3059. }