kcp.go 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844
  1. // Package kcp - A Fast and Reliable ARQ Protocol
  2. //
  3. // Acknowledgement:
  4. // skywind3000@github for inventing the KCP protocol
  5. // xtaci@github for translating to Golang
  6. package kcp
  7. import (
  8. "encoding/binary"
  9. "github.com/v2ray/v2ray-core/common/alloc"
  10. )
  11. const (
  12. IKCP_RTO_NDL = 30 // no delay min rto
  13. IKCP_RTO_MIN = 100 // normal min rto
  14. IKCP_RTO_DEF = 200
  15. IKCP_RTO_MAX = 60000
  16. IKCP_CMD_PUSH = 81 // cmd: push data
  17. IKCP_CMD_ACK = 82 // cmd: ack
  18. IKCP_CMD_WASK = 83 // cmd: window probe (ask)
  19. IKCP_CMD_WINS = 84 // cmd: window size (tell)
  20. IKCP_ASK_SEND = 1 // need to send IKCP_CMD_WASK
  21. IKCP_ASK_TELL = 2 // need to send IKCP_CMD_WINS
  22. IKCP_WND_SND = 32
  23. IKCP_WND_RCV = 32
  24. IKCP_MTU_DEF = 1350
  25. IKCP_ACK_FAST = 3
  26. IKCP_INTERVAL = 100
  27. IKCP_OVERHEAD = 24
  28. IKCP_DEADLINK = 20
  29. IKCP_THRESH_INIT = 2
  30. IKCP_THRESH_MIN = 2
  31. IKCP_PROBE_INIT = 7000 // 7 secs to probe window size
  32. IKCP_PROBE_LIMIT = 120000 // up to 120 secs to probe window
  33. )
  34. // Output is a closure which captures conn and calls conn.Write
  35. type Output func(buf []byte)
  36. /* encode 8 bits unsigned int */
  37. func ikcp_encode8u(p []byte, c byte) []byte {
  38. p[0] = c
  39. return p[1:]
  40. }
  41. /* decode 8 bits unsigned int */
  42. func ikcp_decode8u(p []byte, c *byte) []byte {
  43. *c = p[0]
  44. return p[1:]
  45. }
  46. /* encode 16 bits unsigned int (lsb) */
  47. func ikcp_encode16u(p []byte, w uint16) []byte {
  48. binary.LittleEndian.PutUint16(p, w)
  49. return p[2:]
  50. }
  51. /* decode 16 bits unsigned int (lsb) */
  52. func ikcp_decode16u(p []byte, w *uint16) []byte {
  53. *w = binary.LittleEndian.Uint16(p)
  54. return p[2:]
  55. }
  56. /* encode 32 bits unsigned int (lsb) */
  57. func ikcp_encode32u(p []byte, l uint32) []byte {
  58. binary.LittleEndian.PutUint32(p, l)
  59. return p[4:]
  60. }
  61. /* decode 32 bits unsigned int (lsb) */
  62. func ikcp_decode32u(p []byte, l *uint32) []byte {
  63. *l = binary.LittleEndian.Uint32(p)
  64. return p[4:]
  65. }
  66. func _imin_(a, b uint32) uint32 {
  67. if a <= b {
  68. return a
  69. } else {
  70. return b
  71. }
  72. }
  73. func _imax_(a, b uint32) uint32 {
  74. if a >= b {
  75. return a
  76. } else {
  77. return b
  78. }
  79. }
  80. func _itimediff(later, earlier uint32) int32 {
  81. return (int32)(later - earlier)
  82. }
  83. // Segment defines a KCP segment
  84. type Segment struct {
  85. conv uint32
  86. cmd uint32
  87. frg uint32
  88. wnd uint32
  89. ts uint32
  90. sn uint32
  91. una uint32
  92. resendts uint32
  93. rto uint32
  94. fastack uint32
  95. xmit uint32
  96. data *alloc.Buffer
  97. }
  98. // encode a segment into buffer
  99. func (seg *Segment) encode(ptr []byte) []byte {
  100. ptr = ikcp_encode32u(ptr, seg.conv)
  101. ptr = ikcp_encode8u(ptr, uint8(seg.cmd))
  102. ptr = ikcp_encode8u(ptr, uint8(seg.frg))
  103. ptr = ikcp_encode16u(ptr, uint16(seg.wnd))
  104. ptr = ikcp_encode32u(ptr, seg.ts)
  105. ptr = ikcp_encode32u(ptr, seg.sn)
  106. ptr = ikcp_encode32u(ptr, seg.una)
  107. ptr = ikcp_encode32u(ptr, uint32(seg.data.Len()))
  108. return ptr
  109. }
  110. func (this *Segment) Release() {
  111. this.data.Release()
  112. this.data = nil
  113. }
  114. // NewSegment creates a KCP segment
  115. func NewSegment() *Segment {
  116. return &Segment{
  117. data: alloc.NewSmallBuffer().Clear(),
  118. }
  119. }
  120. // KCP defines a single KCP connection
  121. type KCP struct {
  122. conv, mtu, mss, state uint32
  123. snd_una, snd_nxt, rcv_nxt uint32
  124. ts_recent, ts_lastack, ssthresh uint32
  125. rx_rttval, rx_srtt, rx_rto, rx_minrto uint32
  126. snd_wnd, rcv_wnd, rmt_wnd, cwnd, probe uint32
  127. current, interval, ts_flush, xmit uint32
  128. nodelay, updated uint32
  129. ts_probe, probe_wait uint32
  130. dead_link, incr uint32
  131. snd_queue []*Segment
  132. rcv_queue []*Segment
  133. snd_buf []*Segment
  134. rcv_buf []*Segment
  135. acklist []uint32
  136. buffer []byte
  137. fastresend int32
  138. congestionControl bool
  139. output Output
  140. }
  141. // NewKCP create a new kcp control object, 'conv' must equal in two endpoint
  142. // from the same connection.
  143. func NewKCP(conv uint32, mtu uint32, output Output) *KCP {
  144. kcp := new(KCP)
  145. kcp.conv = conv
  146. kcp.snd_wnd = IKCP_WND_SND
  147. kcp.rcv_wnd = IKCP_WND_RCV
  148. kcp.rmt_wnd = IKCP_WND_RCV
  149. kcp.mtu = mtu
  150. kcp.mss = kcp.mtu - IKCP_OVERHEAD
  151. kcp.buffer = make([]byte, (kcp.mtu+IKCP_OVERHEAD)*3)
  152. kcp.rx_rto = IKCP_RTO_DEF
  153. kcp.rx_minrto = IKCP_RTO_MIN
  154. kcp.interval = IKCP_INTERVAL
  155. kcp.ts_flush = IKCP_INTERVAL
  156. kcp.ssthresh = IKCP_THRESH_INIT
  157. kcp.dead_link = IKCP_DEADLINK
  158. kcp.output = output
  159. return kcp
  160. }
  161. // Recv is user/upper level recv: returns size, returns below zero for EAGAIN
  162. func (kcp *KCP) Recv(buffer []byte) (n int) {
  163. if len(kcp.rcv_queue) == 0 {
  164. return -1
  165. }
  166. var fast_recover bool
  167. if len(kcp.rcv_queue) >= int(kcp.rcv_wnd) {
  168. fast_recover = true
  169. }
  170. // merge fragment
  171. count := 0
  172. for _, seg := range kcp.rcv_queue {
  173. dataLen := seg.data.Len()
  174. if dataLen > len(buffer) {
  175. break
  176. }
  177. copy(buffer, seg.data.Value)
  178. seg.Release()
  179. buffer = buffer[dataLen:]
  180. n += dataLen
  181. count++
  182. }
  183. kcp.rcv_queue = kcp.rcv_queue[count:]
  184. // move available data from rcv_buf -> rcv_queue
  185. count = 0
  186. for _, seg := range kcp.rcv_buf {
  187. if seg.sn == kcp.rcv_nxt && len(kcp.rcv_queue) < int(kcp.rcv_wnd) {
  188. kcp.rcv_queue = append(kcp.rcv_queue, seg)
  189. kcp.rcv_nxt++
  190. count++
  191. } else {
  192. break
  193. }
  194. }
  195. kcp.rcv_buf = kcp.rcv_buf[count:]
  196. // fast recover
  197. if len(kcp.rcv_queue) < int(kcp.rcv_wnd) && fast_recover {
  198. // ready to send back IKCP_CMD_WINS in ikcp_flush
  199. // tell remote my window size
  200. kcp.probe |= IKCP_ASK_TELL
  201. }
  202. return
  203. }
  204. // Send is user/upper level send, returns below zero for error
  205. func (kcp *KCP) Send(buffer []byte) int {
  206. var count int
  207. if len(buffer) == 0 {
  208. return -1
  209. }
  210. if len(buffer) < int(kcp.mss) {
  211. count = 1
  212. } else {
  213. count = (len(buffer) + int(kcp.mss) - 1) / int(kcp.mss)
  214. }
  215. if count > 255 {
  216. return -2
  217. }
  218. if count == 0 {
  219. count = 1
  220. }
  221. for i := 0; i < count; i++ {
  222. var size int
  223. if len(buffer) > int(kcp.mss) {
  224. size = int(kcp.mss)
  225. } else {
  226. size = len(buffer)
  227. }
  228. seg := NewSegment()
  229. seg.data.Append(buffer[:size])
  230. seg.frg = uint32(count - i - 1)
  231. kcp.snd_queue = append(kcp.snd_queue, seg)
  232. buffer = buffer[size:]
  233. }
  234. return 0
  235. }
  236. // https://tools.ietf.org/html/rfc6298
  237. func (kcp *KCP) update_ack(rtt int32) {
  238. var rto uint32 = 0
  239. if kcp.rx_srtt == 0 {
  240. kcp.rx_srtt = uint32(rtt)
  241. kcp.rx_rttval = uint32(rtt) / 2
  242. } else {
  243. delta := rtt - int32(kcp.rx_srtt)
  244. if delta < 0 {
  245. delta = -delta
  246. }
  247. kcp.rx_rttval = (3*kcp.rx_rttval + uint32(delta)) / 4
  248. kcp.rx_srtt = (7*kcp.rx_srtt + uint32(rtt)) / 8
  249. if kcp.rx_srtt < 1 {
  250. kcp.rx_srtt = 1
  251. }
  252. }
  253. rto = kcp.rx_srtt + _imax_(1, 4*kcp.rx_rttval)
  254. if rto > IKCP_RTO_MAX {
  255. rto = IKCP_RTO_MAX
  256. }
  257. if rto < kcp.rx_minrto {
  258. rto = kcp.rx_minrto
  259. }
  260. kcp.rx_rto = rto
  261. }
  262. func (kcp *KCP) shrink_buf() {
  263. if len(kcp.snd_buf) > 0 {
  264. seg := kcp.snd_buf[0]
  265. kcp.snd_una = seg.sn
  266. } else {
  267. kcp.snd_una = kcp.snd_nxt
  268. }
  269. }
  270. func (kcp *KCP) parse_ack(sn uint32) {
  271. if _itimediff(sn, kcp.snd_una) < 0 || _itimediff(sn, kcp.snd_nxt) >= 0 {
  272. return
  273. }
  274. for k, seg := range kcp.snd_buf {
  275. if sn == seg.sn {
  276. kcp.snd_buf = append(kcp.snd_buf[:k], kcp.snd_buf[k+1:]...)
  277. seg.Release()
  278. break
  279. }
  280. if _itimediff(sn, seg.sn) < 0 {
  281. break
  282. }
  283. }
  284. }
  285. func (kcp *KCP) parse_fastack(sn uint32) {
  286. if _itimediff(sn, kcp.snd_una) < 0 || _itimediff(sn, kcp.snd_nxt) >= 0 {
  287. return
  288. }
  289. for _, seg := range kcp.snd_buf {
  290. if _itimediff(sn, seg.sn) < 0 {
  291. break
  292. } else if sn != seg.sn {
  293. seg.fastack++
  294. }
  295. }
  296. }
  297. func (kcp *KCP) parse_una(una uint32) {
  298. count := 0
  299. for _, seg := range kcp.snd_buf {
  300. if _itimediff(una, seg.sn) > 0 {
  301. seg.Release()
  302. count++
  303. } else {
  304. break
  305. }
  306. }
  307. kcp.snd_buf = kcp.snd_buf[count:]
  308. }
  309. // ack append
  310. func (kcp *KCP) ack_push(sn, ts uint32) {
  311. kcp.acklist = append(kcp.acklist, sn, ts)
  312. }
  313. func (kcp *KCP) ack_get(p int) (sn, ts uint32) {
  314. return kcp.acklist[p*2+0], kcp.acklist[p*2+1]
  315. }
  316. func (kcp *KCP) parse_data(newseg *Segment) {
  317. sn := newseg.sn
  318. if _itimediff(sn, kcp.rcv_nxt+kcp.rcv_wnd) >= 0 ||
  319. _itimediff(sn, kcp.rcv_nxt) < 0 {
  320. return
  321. }
  322. n := len(kcp.rcv_buf) - 1
  323. insert_idx := 0
  324. repeat := false
  325. for i := n; i >= 0; i-- {
  326. seg := kcp.rcv_buf[i]
  327. if seg.sn == sn {
  328. repeat = true
  329. break
  330. }
  331. if _itimediff(sn, seg.sn) > 0 {
  332. insert_idx = i + 1
  333. break
  334. }
  335. }
  336. if !repeat {
  337. if insert_idx == n+1 {
  338. kcp.rcv_buf = append(kcp.rcv_buf, newseg)
  339. } else {
  340. kcp.rcv_buf = append(kcp.rcv_buf, &Segment{})
  341. copy(kcp.rcv_buf[insert_idx+1:], kcp.rcv_buf[insert_idx:])
  342. kcp.rcv_buf[insert_idx] = newseg
  343. }
  344. }
  345. // move available data from rcv_buf -> rcv_queue
  346. count := 0
  347. for k, seg := range kcp.rcv_buf {
  348. if seg.sn == kcp.rcv_nxt && len(kcp.rcv_queue) < int(kcp.rcv_wnd) {
  349. kcp.rcv_queue = append(kcp.rcv_queue, kcp.rcv_buf[k])
  350. kcp.rcv_nxt++
  351. count++
  352. } else {
  353. break
  354. }
  355. }
  356. kcp.rcv_buf = kcp.rcv_buf[count:]
  357. }
  358. // Input when you received a low level packet (eg. UDP packet), call it
  359. func (kcp *KCP) Input(data []byte) int {
  360. una := kcp.snd_una
  361. if len(data) < IKCP_OVERHEAD {
  362. return -1
  363. }
  364. var maxack uint32
  365. var flag int
  366. for {
  367. var ts, sn, length, una, conv uint32
  368. var wnd uint16
  369. var cmd, frg uint8
  370. if len(data) < int(IKCP_OVERHEAD) {
  371. break
  372. }
  373. data = ikcp_decode32u(data, &conv)
  374. if conv != kcp.conv {
  375. return -1
  376. }
  377. data = ikcp_decode8u(data, &cmd)
  378. data = ikcp_decode8u(data, &frg)
  379. data = ikcp_decode16u(data, &wnd)
  380. data = ikcp_decode32u(data, &ts)
  381. data = ikcp_decode32u(data, &sn)
  382. data = ikcp_decode32u(data, &una)
  383. data = ikcp_decode32u(data, &length)
  384. if len(data) < int(length) {
  385. return -2
  386. }
  387. if cmd != IKCP_CMD_PUSH && cmd != IKCP_CMD_ACK &&
  388. cmd != IKCP_CMD_WASK && cmd != IKCP_CMD_WINS {
  389. return -3
  390. }
  391. kcp.rmt_wnd = uint32(wnd)
  392. kcp.parse_una(una)
  393. kcp.shrink_buf()
  394. if cmd == IKCP_CMD_ACK {
  395. if _itimediff(kcp.current, ts) >= 0 {
  396. kcp.update_ack(_itimediff(kcp.current, ts))
  397. }
  398. kcp.parse_ack(sn)
  399. kcp.shrink_buf()
  400. if flag == 0 {
  401. flag = 1
  402. maxack = sn
  403. } else if _itimediff(sn, maxack) > 0 {
  404. maxack = sn
  405. }
  406. } else if cmd == IKCP_CMD_PUSH {
  407. if _itimediff(sn, kcp.rcv_nxt+kcp.rcv_wnd) < 0 {
  408. kcp.ack_push(sn, ts)
  409. if _itimediff(sn, kcp.rcv_nxt) >= 0 {
  410. seg := NewSegment()
  411. seg.conv = conv
  412. seg.cmd = uint32(cmd)
  413. seg.frg = uint32(frg)
  414. seg.wnd = uint32(wnd)
  415. seg.ts = ts
  416. seg.sn = sn
  417. seg.una = una
  418. seg.data.Append(data[:length])
  419. kcp.parse_data(seg)
  420. }
  421. }
  422. } else if cmd == IKCP_CMD_WASK {
  423. // ready to send back IKCP_CMD_WINS in Ikcp_flush
  424. // tell remote my window size
  425. kcp.probe |= IKCP_ASK_TELL
  426. } else if cmd == IKCP_CMD_WINS {
  427. // do nothing
  428. } else {
  429. return -3
  430. }
  431. data = data[length:]
  432. }
  433. if flag != 0 {
  434. kcp.parse_fastack(maxack)
  435. }
  436. if _itimediff(kcp.snd_una, una) > 0 {
  437. if kcp.cwnd < kcp.rmt_wnd {
  438. mss := kcp.mss
  439. if kcp.cwnd < kcp.ssthresh {
  440. kcp.cwnd++
  441. kcp.incr += mss
  442. } else {
  443. if kcp.incr < mss {
  444. kcp.incr = mss
  445. }
  446. kcp.incr += (mss*mss)/kcp.incr + (mss / 16)
  447. if (kcp.cwnd+1)*mss <= kcp.incr {
  448. kcp.cwnd++
  449. }
  450. }
  451. if kcp.cwnd > kcp.rmt_wnd {
  452. kcp.cwnd = kcp.rmt_wnd
  453. kcp.incr = kcp.rmt_wnd * mss
  454. }
  455. }
  456. }
  457. return 0
  458. }
  459. func (kcp *KCP) wnd_unused() int32 {
  460. if len(kcp.rcv_queue) < int(kcp.rcv_wnd) {
  461. return int32(int(kcp.rcv_wnd) - len(kcp.rcv_queue))
  462. }
  463. return 0
  464. }
  465. // flush pending data
  466. func (kcp *KCP) flush() {
  467. current := kcp.current
  468. buffer := kcp.buffer
  469. change := 0
  470. lost := false
  471. if kcp.updated == 0 {
  472. return
  473. }
  474. var seg Segment
  475. seg.conv = kcp.conv
  476. seg.cmd = IKCP_CMD_ACK
  477. seg.wnd = uint32(kcp.wnd_unused())
  478. seg.una = kcp.rcv_nxt
  479. // flush acknowledges
  480. count := len(kcp.acklist) / 2
  481. ptr := buffer
  482. for i := 0; i < count; i++ {
  483. size := len(buffer) - len(ptr)
  484. if size+IKCP_OVERHEAD > int(kcp.mtu) {
  485. kcp.output(buffer[:size])
  486. ptr = buffer
  487. }
  488. seg.sn, seg.ts = kcp.ack_get(i)
  489. ptr = seg.encode(ptr)
  490. }
  491. kcp.acklist = nil
  492. // probe window size (if remote window size equals zero)
  493. if kcp.rmt_wnd == 0 {
  494. if kcp.probe_wait == 0 {
  495. kcp.probe_wait = IKCP_PROBE_INIT
  496. kcp.ts_probe = kcp.current + kcp.probe_wait
  497. } else {
  498. if _itimediff(kcp.current, kcp.ts_probe) >= 0 {
  499. if kcp.probe_wait < IKCP_PROBE_INIT {
  500. kcp.probe_wait = IKCP_PROBE_INIT
  501. }
  502. kcp.probe_wait += kcp.probe_wait / 2
  503. if kcp.probe_wait > IKCP_PROBE_LIMIT {
  504. kcp.probe_wait = IKCP_PROBE_LIMIT
  505. }
  506. kcp.ts_probe = kcp.current + kcp.probe_wait
  507. kcp.probe |= IKCP_ASK_SEND
  508. }
  509. }
  510. } else {
  511. kcp.ts_probe = 0
  512. kcp.probe_wait = 0
  513. }
  514. // flush window probing commands
  515. if (kcp.probe & IKCP_ASK_SEND) != 0 {
  516. seg.cmd = IKCP_CMD_WASK
  517. size := len(buffer) - len(ptr)
  518. if size+IKCP_OVERHEAD > int(kcp.mtu) {
  519. kcp.output(buffer[:size])
  520. ptr = buffer
  521. }
  522. ptr = seg.encode(ptr)
  523. }
  524. // flush window probing commands
  525. if (kcp.probe & IKCP_ASK_TELL) != 0 {
  526. seg.cmd = IKCP_CMD_WINS
  527. size := len(buffer) - len(ptr)
  528. if size+IKCP_OVERHEAD > int(kcp.mtu) {
  529. kcp.output(buffer[:size])
  530. ptr = buffer
  531. }
  532. ptr = seg.encode(ptr)
  533. }
  534. kcp.probe = 0
  535. // calculate window size
  536. cwnd := _imin_(kcp.snd_wnd, kcp.rmt_wnd)
  537. if kcp.congestionControl {
  538. cwnd = _imin_(kcp.cwnd, cwnd)
  539. }
  540. count = 0
  541. for k := range kcp.snd_queue {
  542. if _itimediff(kcp.snd_nxt, kcp.snd_una+cwnd) >= 0 {
  543. break
  544. }
  545. newseg := kcp.snd_queue[k]
  546. newseg.conv = kcp.conv
  547. newseg.cmd = IKCP_CMD_PUSH
  548. newseg.wnd = seg.wnd
  549. newseg.ts = current
  550. newseg.sn = kcp.snd_nxt
  551. newseg.una = kcp.rcv_nxt
  552. newseg.resendts = current
  553. newseg.rto = kcp.rx_rto
  554. newseg.fastack = 0
  555. newseg.xmit = 0
  556. kcp.snd_buf = append(kcp.snd_buf, newseg)
  557. kcp.snd_nxt++
  558. count++
  559. }
  560. kcp.snd_queue = kcp.snd_queue[count:]
  561. // calculate resent
  562. resent := uint32(kcp.fastresend)
  563. if kcp.fastresend <= 0 {
  564. resent = 0xffffffff
  565. }
  566. rtomin := (kcp.rx_rto >> 3)
  567. if kcp.nodelay != 0 {
  568. rtomin = 0
  569. }
  570. // flush data segments
  571. for _, segment := range kcp.snd_buf {
  572. needsend := false
  573. if segment.xmit == 0 {
  574. needsend = true
  575. segment.xmit++
  576. segment.rto = kcp.rx_rto
  577. segment.resendts = current + segment.rto + rtomin
  578. } else if _itimediff(current, segment.resendts) >= 0 {
  579. needsend = true
  580. segment.xmit++
  581. kcp.xmit++
  582. if kcp.nodelay == 0 {
  583. segment.rto += kcp.rx_rto
  584. } else {
  585. segment.rto += kcp.rx_rto / 2
  586. }
  587. segment.resendts = current + segment.rto
  588. lost = true
  589. } else if segment.fastack >= resent {
  590. needsend = true
  591. segment.xmit++
  592. segment.fastack = 0
  593. segment.resendts = current + segment.rto
  594. change++
  595. }
  596. if needsend {
  597. segment.ts = current
  598. segment.wnd = seg.wnd
  599. segment.una = kcp.rcv_nxt
  600. size := len(buffer) - len(ptr)
  601. need := IKCP_OVERHEAD + segment.data.Len()
  602. if size+need >= int(kcp.mtu) {
  603. kcp.output(buffer[:size])
  604. ptr = buffer
  605. }
  606. ptr = segment.encode(ptr)
  607. copy(ptr, segment.data.Value)
  608. ptr = ptr[segment.data.Len():]
  609. if segment.xmit >= kcp.dead_link {
  610. kcp.state = 0xFFFFFFFF
  611. }
  612. }
  613. }
  614. // flash remain segments
  615. size := len(buffer) - len(ptr)
  616. if size > 0 {
  617. kcp.output(buffer[:size])
  618. }
  619. // update ssthresh
  620. // rate halving, https://tools.ietf.org/html/rfc6937
  621. if change != 0 {
  622. inflight := kcp.snd_nxt - kcp.snd_una
  623. kcp.ssthresh = inflight / 2
  624. if kcp.ssthresh < IKCP_THRESH_MIN {
  625. kcp.ssthresh = IKCP_THRESH_MIN
  626. }
  627. kcp.cwnd = kcp.ssthresh + resent
  628. kcp.incr = kcp.cwnd * kcp.mss
  629. }
  630. // congestion control, https://tools.ietf.org/html/rfc5681
  631. if lost {
  632. kcp.ssthresh = cwnd / 2
  633. if kcp.ssthresh < IKCP_THRESH_MIN {
  634. kcp.ssthresh = IKCP_THRESH_MIN
  635. }
  636. kcp.cwnd = 1
  637. kcp.incr = kcp.mss
  638. }
  639. if kcp.cwnd < 1 {
  640. kcp.cwnd = 1
  641. kcp.incr = kcp.mss
  642. }
  643. }
  644. // Update updates state (call it repeatedly, every 10ms-100ms), or you can ask
  645. // ikcp_check when to call it again (without ikcp_input/_send calling).
  646. // 'current' - current timestamp in millisec.
  647. func (kcp *KCP) Update(current uint32) {
  648. var slap int32
  649. kcp.current = current
  650. if kcp.updated == 0 {
  651. kcp.updated = 1
  652. kcp.ts_flush = kcp.current
  653. }
  654. slap = _itimediff(kcp.current, kcp.ts_flush)
  655. if slap >= 10000 || slap < -10000 {
  656. kcp.ts_flush = kcp.current
  657. slap = 0
  658. }
  659. if slap >= 0 {
  660. kcp.ts_flush += kcp.interval
  661. if _itimediff(kcp.current, kcp.ts_flush) >= 0 {
  662. kcp.ts_flush = kcp.current + kcp.interval
  663. }
  664. kcp.flush()
  665. }
  666. }
  667. // Check determines when should you invoke ikcp_update:
  668. // returns when you should invoke ikcp_update in millisec, if there
  669. // is no ikcp_input/_send calling. you can call ikcp_update in that
  670. // time, instead of call update repeatly.
  671. // Important to reduce unnacessary ikcp_update invoking. use it to
  672. // schedule ikcp_update (eg. implementing an epoll-like mechanism,
  673. // or optimize ikcp_update when handling massive kcp connections)
  674. func (kcp *KCP) Check(current uint32) uint32 {
  675. ts_flush := kcp.ts_flush
  676. tm_flush := int32(0x7fffffff)
  677. tm_packet := int32(0x7fffffff)
  678. minimal := uint32(0)
  679. if kcp.updated == 0 {
  680. return current
  681. }
  682. if _itimediff(current, ts_flush) >= 10000 ||
  683. _itimediff(current, ts_flush) < -10000 {
  684. ts_flush = current
  685. }
  686. if _itimediff(current, ts_flush) >= 0 {
  687. return current
  688. }
  689. tm_flush = _itimediff(ts_flush, current)
  690. for _, seg := range kcp.snd_buf {
  691. diff := _itimediff(seg.resendts, current)
  692. if diff <= 0 {
  693. return current
  694. }
  695. if diff < tm_packet {
  696. tm_packet = diff
  697. }
  698. }
  699. minimal = uint32(tm_packet)
  700. if tm_packet >= tm_flush {
  701. minimal = uint32(tm_flush)
  702. }
  703. if minimal >= kcp.interval {
  704. minimal = kcp.interval
  705. }
  706. return current + minimal
  707. }
  708. // NoDelay options
  709. // fastest: ikcp_nodelay(kcp, 1, 20, 2, 1)
  710. // nodelay: 0:disable(default), 1:enable
  711. // interval: internal update timer interval in millisec, default is 100ms
  712. // resend: 0:disable fast resend(default), 1:enable fast resend
  713. // nc: 0:normal congestion control(default), 1:disable congestion control
  714. func (kcp *KCP) NoDelay(nodelay, interval, resend int, congestionControl bool) int {
  715. if nodelay >= 0 {
  716. kcp.nodelay = uint32(nodelay)
  717. if nodelay != 0 {
  718. kcp.rx_minrto = IKCP_RTO_NDL
  719. } else {
  720. kcp.rx_minrto = IKCP_RTO_MIN
  721. }
  722. }
  723. if interval >= 0 {
  724. if interval > 5000 {
  725. interval = 5000
  726. } else if interval < 10 {
  727. interval = 10
  728. }
  729. kcp.interval = uint32(interval)
  730. }
  731. if resend >= 0 {
  732. kcp.fastresend = int32(resend)
  733. }
  734. kcp.congestionControl = congestionControl
  735. return 0
  736. }
  737. // WndSize sets maximum window size: sndwnd=32, rcvwnd=32 by default
  738. func (kcp *KCP) WndSize(sndwnd, rcvwnd int) int {
  739. if sndwnd > 0 {
  740. kcp.snd_wnd = uint32(sndwnd)
  741. }
  742. if rcvwnd > 0 {
  743. kcp.rcv_wnd = uint32(rcvwnd)
  744. }
  745. return 0
  746. }
  747. // WaitSnd gets how many packet is waiting to be sent
  748. func (kcp *KCP) WaitSnd() int {
  749. return len(kcp.snd_buf) + len(kcp.snd_queue)
  750. }