tcp.c 64 KB

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