tcp.c 60 KB

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