tcp.c 65 KB

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