options.c 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778
  1. /*
  2. * net/dccp/options.c
  3. *
  4. * An implementation of the DCCP protocol
  5. * Aristeu Sergio Rozanski Filho <aris@cathedrallabs.org>
  6. * Arnaldo Carvalho de Melo <acme@ghostprotocols.net>
  7. *
  8. * This program is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU General Public License
  10. * as published by the Free Software Foundation; either version
  11. * 2 of the License, or (at your option) any later version.
  12. */
  13. #include <linux/config.h>
  14. #include <linux/dccp.h>
  15. #include <linux/module.h>
  16. #include <linux/types.h>
  17. #include <linux/kernel.h>
  18. #include <linux/skbuff.h>
  19. #include "ccid.h"
  20. #include "dccp.h"
  21. static void dccp_ackpkts_check_rcv_ackvector(struct dccp_ackpkts *ap,
  22. struct sock *sk,
  23. const u64 ackno,
  24. const unsigned char len,
  25. const unsigned char *vector);
  26. /* stores the default values for new connection. may be changed with sysctl */
  27. static const struct dccp_options dccpo_default_values = {
  28. .dccpo_sequence_window = DCCPF_INITIAL_SEQUENCE_WINDOW,
  29. .dccpo_ccid = DCCPF_INITIAL_CCID,
  30. .dccpo_send_ack_vector = DCCPF_INITIAL_SEND_ACK_VECTOR,
  31. .dccpo_send_ndp_count = DCCPF_INITIAL_SEND_NDP_COUNT,
  32. };
  33. void dccp_options_init(struct dccp_options *dccpo)
  34. {
  35. memcpy(dccpo, &dccpo_default_values, sizeof(*dccpo));
  36. }
  37. static u32 dccp_decode_value_var(const unsigned char *bf, const u8 len)
  38. {
  39. u32 value = 0;
  40. if (len > 3)
  41. value += *bf++ << 24;
  42. if (len > 2)
  43. value += *bf++ << 16;
  44. if (len > 1)
  45. value += *bf++ << 8;
  46. if (len > 0)
  47. value += *bf;
  48. return value;
  49. }
  50. int dccp_parse_options(struct sock *sk, struct sk_buff *skb)
  51. {
  52. struct dccp_sock *dp = dccp_sk(sk);
  53. #ifdef DCCP_DEBUG
  54. const char *debug_prefix = dp->dccps_role == DCCP_ROLE_CLIENT ? "CLIENT rx opt: " :
  55. "server rx opt: ";
  56. #endif
  57. const struct dccp_hdr *dh = dccp_hdr(skb);
  58. const u8 pkt_type = DCCP_SKB_CB(skb)->dccpd_type;
  59. unsigned char *options = (unsigned char *)dh + dccp_hdr_len(skb);
  60. unsigned char *opt_ptr = options;
  61. const unsigned char *opt_end = (unsigned char *)dh + (dh->dccph_doff * 4);
  62. struct dccp_options_received *opt_recv = &dp->dccps_options_received;
  63. unsigned char opt, len;
  64. unsigned char *value;
  65. memset(opt_recv, 0, sizeof(*opt_recv));
  66. while (opt_ptr != opt_end) {
  67. opt = *opt_ptr++;
  68. len = 0;
  69. value = NULL;
  70. /* Check if this isn't a single byte option */
  71. if (opt > DCCPO_MAX_RESERVED) {
  72. if (opt_ptr == opt_end)
  73. goto out_invalid_option;
  74. len = *opt_ptr++;
  75. if (len < 3)
  76. goto out_invalid_option;
  77. /*
  78. * Remove the type and len fields, leaving
  79. * just the value size
  80. */
  81. len -= 2;
  82. value = opt_ptr;
  83. opt_ptr += len;
  84. if (opt_ptr > opt_end)
  85. goto out_invalid_option;
  86. }
  87. switch (opt) {
  88. case DCCPO_PADDING:
  89. break;
  90. case DCCPO_NDP_COUNT:
  91. if (len > 3)
  92. goto out_invalid_option;
  93. opt_recv->dccpor_ndp = dccp_decode_value_var(value, len);
  94. dccp_pr_debug("%sNDP count=%d\n", debug_prefix, opt_recv->dccpor_ndp);
  95. break;
  96. case DCCPO_ACK_VECTOR_0:
  97. if (len > DCCP_MAX_ACK_VECTOR_LEN)
  98. goto out_invalid_option;
  99. if (pkt_type == DCCP_PKT_DATA)
  100. continue;
  101. opt_recv->dccpor_ack_vector_len = len;
  102. opt_recv->dccpor_ack_vector_idx = value - options;
  103. dccp_pr_debug("%sACK vector 0, len=%d, ack_ackno=%llu\n",
  104. debug_prefix, len,
  105. (unsigned long long)
  106. DCCP_SKB_CB(skb)->dccpd_ack_seq);
  107. dccp_ackvector_print(DCCP_SKB_CB(skb)->dccpd_ack_seq,
  108. value, len);
  109. dccp_ackpkts_check_rcv_ackvector(dp->dccps_hc_rx_ackpkts, sk,
  110. DCCP_SKB_CB(skb)->dccpd_ack_seq,
  111. len, value);
  112. break;
  113. case DCCPO_TIMESTAMP:
  114. if (len != 4)
  115. goto out_invalid_option;
  116. opt_recv->dccpor_timestamp = ntohl(*(u32 *)value);
  117. dp->dccps_timestamp_echo = opt_recv->dccpor_timestamp;
  118. dp->dccps_timestamp_time = jiffies;
  119. dccp_pr_debug("%sTIMESTAMP=%u, ackno=%llu\n",
  120. debug_prefix, opt_recv->dccpor_timestamp,
  121. (unsigned long long)
  122. DCCP_SKB_CB(skb)->dccpd_ack_seq);
  123. break;
  124. case DCCPO_TIMESTAMP_ECHO:
  125. if (len < 4 || len > 8)
  126. goto out_invalid_option;
  127. opt_recv->dccpor_timestamp_echo = ntohl(*(u32 *)value);
  128. dccp_pr_debug("%sTIMESTAMP_ECHO=%u, len=%d, ackno=%llu, diff=%u\n",
  129. debug_prefix, opt_recv->dccpor_timestamp_echo,
  130. len + 2,
  131. (unsigned long long)
  132. DCCP_SKB_CB(skb)->dccpd_ack_seq,
  133. tcp_time_stamp - opt_recv->dccpor_timestamp_echo);
  134. opt_recv->dccpor_elapsed_time = dccp_decode_value_var(value + 4, len - 4);
  135. dccp_pr_debug("%sTIMESTAMP_ECHO ELAPSED_TIME=%d\n", debug_prefix,
  136. opt_recv->dccpor_elapsed_time);
  137. break;
  138. case DCCPO_ELAPSED_TIME:
  139. if (len > 4)
  140. goto out_invalid_option;
  141. if (pkt_type == DCCP_PKT_DATA)
  142. continue;
  143. opt_recv->dccpor_elapsed_time = dccp_decode_value_var(value, len);
  144. dccp_pr_debug("%sELAPSED_TIME=%d\n", debug_prefix,
  145. opt_recv->dccpor_elapsed_time);
  146. break;
  147. /*
  148. * From draft-ietf-dccp-spec-11.txt:
  149. *
  150. * Option numbers 128 through 191 are for options sent from the HC-
  151. * Sender to the HC-Receiver; option numbers 192 through 255 are for
  152. * options sent from the HC-Receiver to the HC-Sender.
  153. */
  154. case 128 ... 191: {
  155. const u16 idx = value - options;
  156. if (ccid_hc_rx_parse_options(dp->dccps_hc_rx_ccid, sk, opt, len, idx, value) != 0)
  157. goto out_invalid_option;
  158. }
  159. break;
  160. case 192 ... 255: {
  161. const u16 idx = value - options;
  162. if (ccid_hc_tx_parse_options(dp->dccps_hc_tx_ccid, sk, opt, len, idx, value) != 0)
  163. goto out_invalid_option;
  164. }
  165. break;
  166. default:
  167. pr_info("DCCP(%p): option %d(len=%d) not implemented, ignoring\n",
  168. sk, opt, len);
  169. break;
  170. }
  171. }
  172. return 0;
  173. out_invalid_option:
  174. DCCP_INC_STATS_BH(DCCP_MIB_INVALIDOPT);
  175. DCCP_SKB_CB(skb)->dccpd_reset_code = DCCP_RESET_CODE_OPTION_ERROR;
  176. pr_info("DCCP(%p): invalid option %d, len=%d\n", sk, opt, len);
  177. return -1;
  178. }
  179. static void dccp_encode_value_var(const u32 value, unsigned char *to,
  180. const unsigned int len)
  181. {
  182. if (len > 3)
  183. *to++ = (value & 0xFF000000) >> 24;
  184. if (len > 2)
  185. *to++ = (value & 0xFF0000) >> 16;
  186. if (len > 1)
  187. *to++ = (value & 0xFF00) >> 8;
  188. if (len > 0)
  189. *to++ = (value & 0xFF);
  190. }
  191. static inline int dccp_ndp_len(const int ndp)
  192. {
  193. return likely(ndp <= 0xFF) ? 1 : ndp <= 0xFFFF ? 2 : 3;
  194. }
  195. void dccp_insert_option(struct sock *sk, struct sk_buff *skb,
  196. const unsigned char option,
  197. const void *value, const unsigned char len)
  198. {
  199. unsigned char *to;
  200. if (DCCP_SKB_CB(skb)->dccpd_opt_len + len + 2 > DCCP_MAX_OPT_LEN) {
  201. LIMIT_NETDEBUG(KERN_INFO "DCCP: packet too small to insert %d option!\n", option);
  202. return;
  203. }
  204. DCCP_SKB_CB(skb)->dccpd_opt_len += len + 2;
  205. to = skb_push(skb, len + 2);
  206. *to++ = option;
  207. *to++ = len + 2;
  208. memcpy(to, value, len);
  209. }
  210. EXPORT_SYMBOL_GPL(dccp_insert_option);
  211. static void dccp_insert_option_ndp(struct sock *sk, struct sk_buff *skb)
  212. {
  213. struct dccp_sock *dp = dccp_sk(sk);
  214. int ndp = dp->dccps_ndp_count;
  215. if (dccp_non_data_packet(skb))
  216. ++dp->dccps_ndp_count;
  217. else
  218. dp->dccps_ndp_count = 0;
  219. if (ndp > 0) {
  220. unsigned char *ptr;
  221. const int ndp_len = dccp_ndp_len(ndp);
  222. const int len = ndp_len + 2;
  223. if (DCCP_SKB_CB(skb)->dccpd_opt_len + len > DCCP_MAX_OPT_LEN)
  224. return;
  225. DCCP_SKB_CB(skb)->dccpd_opt_len += len;
  226. ptr = skb_push(skb, len);
  227. *ptr++ = DCCPO_NDP_COUNT;
  228. *ptr++ = len;
  229. dccp_encode_value_var(ndp, ptr, ndp_len);
  230. }
  231. }
  232. static inline int dccp_elapsed_time_len(const u32 elapsed_time)
  233. {
  234. return elapsed_time == 0 ? 0 :
  235. elapsed_time <= 0xFF ? 1 :
  236. elapsed_time <= 0xFFFF ? 2 :
  237. elapsed_time <= 0xFFFFFF ? 3 : 4;
  238. }
  239. void dccp_insert_option_elapsed_time(struct sock *sk,
  240. struct sk_buff *skb,
  241. u32 elapsed_time)
  242. {
  243. #ifdef DCCP_DEBUG
  244. struct dccp_sock *dp = dccp_sk(sk);
  245. const char *debug_prefix = dp->dccps_role == DCCP_ROLE_CLIENT ? "CLIENT TX opt: " :
  246. "server TX opt: ";
  247. #endif
  248. const int elapsed_time_len = dccp_elapsed_time_len(elapsed_time);
  249. const int len = 2 + elapsed_time_len;
  250. unsigned char *to;
  251. /* If elapsed_time == 0... */
  252. if (elapsed_time_len == 2)
  253. return;
  254. if (DCCP_SKB_CB(skb)->dccpd_opt_len + len > DCCP_MAX_OPT_LEN) {
  255. LIMIT_NETDEBUG(KERN_INFO "DCCP: packet too small to insert elapsed time!\n");
  256. return;
  257. }
  258. DCCP_SKB_CB(skb)->dccpd_opt_len += len;
  259. to = skb_push(skb, len);
  260. *to++ = DCCPO_ELAPSED_TIME;
  261. *to++ = len;
  262. dccp_encode_value_var(elapsed_time, to, elapsed_time_len);
  263. dccp_pr_debug("%sELAPSED_TIME=%u, len=%d, seqno=%llu\n",
  264. debug_prefix, elapsed_time,
  265. len,
  266. (unsigned long long) DCCP_SKB_CB(skb)->dccpd_seq);
  267. }
  268. EXPORT_SYMBOL(dccp_insert_option_elapsed_time);
  269. static void dccp_insert_option_ack_vector(struct sock *sk, struct sk_buff *skb)
  270. {
  271. struct dccp_sock *dp = dccp_sk(sk);
  272. #ifdef DCCP_DEBUG
  273. const char *debug_prefix = dp->dccps_role == DCCP_ROLE_CLIENT ? "CLIENT TX opt: " :
  274. "server TX opt: ";
  275. #endif
  276. struct dccp_ackpkts *ap = dp->dccps_hc_rx_ackpkts;
  277. int len = ap->dccpap_buf_vector_len + 2;
  278. const u32 elapsed_time = jiffies_to_usecs(jiffies - ap->dccpap_time) / 10;
  279. unsigned char *to, *from;
  280. if (elapsed_time != 0)
  281. dccp_insert_option_elapsed_time(sk, skb, elapsed_time);
  282. if (DCCP_SKB_CB(skb)->dccpd_opt_len + len > DCCP_MAX_OPT_LEN) {
  283. LIMIT_NETDEBUG(KERN_INFO "DCCP: packet too small to insert ACK Vector!\n");
  284. return;
  285. }
  286. /*
  287. * XXX: now we have just one ack vector sent record, so
  288. * we have to wait for it to be cleared.
  289. *
  290. * Of course this is not acceptable, but this is just for
  291. * basic testing now.
  292. */
  293. if (ap->dccpap_ack_seqno != DCCP_MAX_SEQNO + 1)
  294. return;
  295. DCCP_SKB_CB(skb)->dccpd_opt_len += len;
  296. to = skb_push(skb, len);
  297. *to++ = DCCPO_ACK_VECTOR_0;
  298. *to++ = len;
  299. len = ap->dccpap_buf_vector_len;
  300. from = ap->dccpap_buf + ap->dccpap_buf_head;
  301. /* Check if buf_head wraps */
  302. if (ap->dccpap_buf_head + len > ap->dccpap_buf_len) {
  303. const unsigned int tailsize = ap->dccpap_buf_len - ap->dccpap_buf_head;
  304. memcpy(to, from, tailsize);
  305. to += tailsize;
  306. len -= tailsize;
  307. from = ap->dccpap_buf;
  308. }
  309. memcpy(to, from, len);
  310. /*
  311. * From draft-ietf-dccp-spec-11.txt:
  312. *
  313. * For each acknowledgement it sends, the HC-Receiver will add an
  314. * acknowledgement record. ack_seqno will equal the HC-Receiver
  315. * sequence number it used for the ack packet; ack_ptr will equal
  316. * buf_head; ack_ackno will equal buf_ackno; and ack_nonce will equal
  317. * buf_nonce.
  318. *
  319. * This implemention uses just one ack record for now.
  320. */
  321. ap->dccpap_ack_seqno = DCCP_SKB_CB(skb)->dccpd_seq;
  322. ap->dccpap_ack_ptr = ap->dccpap_buf_head;
  323. ap->dccpap_ack_ackno = ap->dccpap_buf_ackno;
  324. ap->dccpap_ack_nonce = ap->dccpap_buf_nonce;
  325. ap->dccpap_ack_vector_len = ap->dccpap_buf_vector_len;
  326. dccp_pr_debug("%sACK Vector 0, len=%d, ack_seqno=%llu, ack_ackno=%llu\n",
  327. debug_prefix, ap->dccpap_ack_vector_len,
  328. (unsigned long long) ap->dccpap_ack_seqno,
  329. (unsigned long long) ap->dccpap_ack_ackno);
  330. }
  331. static inline void dccp_insert_option_timestamp(struct sock *sk, struct sk_buff *skb)
  332. {
  333. const u32 now = htonl(tcp_time_stamp);
  334. dccp_insert_option(sk, skb, DCCPO_TIMESTAMP, &now, sizeof(now));
  335. }
  336. static void dccp_insert_option_timestamp_echo(struct sock *sk, struct sk_buff *skb)
  337. {
  338. struct dccp_sock *dp = dccp_sk(sk);
  339. #ifdef DCCP_DEBUG
  340. const char *debug_prefix = dp->dccps_role == DCCP_ROLE_CLIENT ? "CLIENT TX opt: " :
  341. "server TX opt: ";
  342. #endif
  343. u32 tstamp_echo;
  344. const u32 elapsed_time = jiffies_to_usecs(jiffies - dp->dccps_timestamp_time) / 10;
  345. const int elapsed_time_len = dccp_elapsed_time_len(elapsed_time);
  346. const int len = 6 + elapsed_time_len;
  347. unsigned char *to;
  348. if (DCCP_SKB_CB(skb)->dccpd_opt_len + len > DCCP_MAX_OPT_LEN) {
  349. LIMIT_NETDEBUG(KERN_INFO "DCCP: packet too small to insert timestamp echo!\n");
  350. return;
  351. }
  352. DCCP_SKB_CB(skb)->dccpd_opt_len += len;
  353. to = skb_push(skb, len);
  354. *to++ = DCCPO_TIMESTAMP_ECHO;
  355. *to++ = len;
  356. tstamp_echo = htonl(dp->dccps_timestamp_echo);
  357. memcpy(to, &tstamp_echo, 4);
  358. to += 4;
  359. dccp_encode_value_var(elapsed_time, to, elapsed_time_len);
  360. dccp_pr_debug("%sTIMESTAMP_ECHO=%u, len=%d, seqno=%llu\n",
  361. debug_prefix, dp->dccps_timestamp_echo,
  362. len,
  363. (unsigned long long) DCCP_SKB_CB(skb)->dccpd_seq);
  364. dp->dccps_timestamp_echo = 0;
  365. dp->dccps_timestamp_time = 0;
  366. }
  367. void dccp_insert_options(struct sock *sk, struct sk_buff *skb)
  368. {
  369. struct dccp_sock *dp = dccp_sk(sk);
  370. DCCP_SKB_CB(skb)->dccpd_opt_len = 0;
  371. if (dp->dccps_options.dccpo_send_ndp_count)
  372. dccp_insert_option_ndp(sk, skb);
  373. if (!dccp_packet_without_ack(skb)) {
  374. if (dp->dccps_options.dccpo_send_ack_vector &&
  375. dp->dccps_hc_rx_ackpkts->dccpap_buf_ackno != DCCP_MAX_SEQNO + 1)
  376. dccp_insert_option_ack_vector(sk, skb);
  377. dccp_insert_option_timestamp(sk, skb);
  378. if (dp->dccps_timestamp_echo != 0)
  379. dccp_insert_option_timestamp_echo(sk, skb);
  380. }
  381. ccid_hc_rx_insert_options(dp->dccps_hc_rx_ccid, sk, skb);
  382. ccid_hc_tx_insert_options(dp->dccps_hc_tx_ccid, sk, skb);
  383. /* XXX: insert other options when appropriate */
  384. if (DCCP_SKB_CB(skb)->dccpd_opt_len != 0) {
  385. /* The length of all options has to be a multiple of 4 */
  386. int padding = DCCP_SKB_CB(skb)->dccpd_opt_len % 4;
  387. if (padding != 0) {
  388. padding = 4 - padding;
  389. memset(skb_push(skb, padding), 0, padding);
  390. DCCP_SKB_CB(skb)->dccpd_opt_len += padding;
  391. }
  392. }
  393. }
  394. struct dccp_ackpkts *dccp_ackpkts_alloc(unsigned int len, int priority)
  395. {
  396. struct dccp_ackpkts *ap = kmalloc(sizeof(*ap) + len, priority);
  397. if (ap != NULL) {
  398. #ifdef DCCP_DEBUG
  399. memset(ap->dccpap_buf, 0xFF, len);
  400. #endif
  401. ap->dccpap_buf_len = len;
  402. ap->dccpap_buf_head = ap->dccpap_buf_tail = ap->dccpap_buf_len - 1;
  403. ap->dccpap_buf_ackno = ap->dccpap_ack_ackno = ap->dccpap_ack_seqno = DCCP_MAX_SEQNO + 1;
  404. ap->dccpap_buf_nonce = ap->dccpap_buf_nonce = 0;
  405. ap->dccpap_ack_ptr = 0;
  406. ap->dccpap_time = 0;
  407. ap->dccpap_buf_vector_len = ap->dccpap_ack_vector_len = 0;
  408. }
  409. return ap;
  410. }
  411. void dccp_ackpkts_free(struct dccp_ackpkts *ap)
  412. {
  413. if (ap != NULL) {
  414. #ifdef DCCP_DEBUG
  415. memset(ap, 0xFF, sizeof(*ap) + ap->dccpap_buf_len);
  416. #endif
  417. kfree(ap);
  418. }
  419. }
  420. static inline u8 dccp_ackpkts_state(const struct dccp_ackpkts *ap,
  421. const unsigned int index)
  422. {
  423. return ap->dccpap_buf[index] & DCCP_ACKPKTS_STATE_MASK;
  424. }
  425. static inline u8 dccp_ackpkts_len(const struct dccp_ackpkts *ap,
  426. const unsigned int index)
  427. {
  428. return ap->dccpap_buf[index] & DCCP_ACKPKTS_LEN_MASK;
  429. }
  430. /*
  431. * If several packets are missing, the HC-Receiver may prefer to enter multiple
  432. * bytes with run length 0, rather than a single byte with a larger run length;
  433. * this simplifies table updates if one of the missing packets arrives.
  434. */
  435. static inline int dccp_ackpkts_set_buf_head_state(struct dccp_ackpkts *ap,
  436. const unsigned int packets,
  437. const unsigned char state)
  438. {
  439. unsigned int gap;
  440. signed long new_head;
  441. if (ap->dccpap_buf_vector_len + packets > ap->dccpap_buf_len)
  442. return -ENOBUFS;
  443. gap = packets - 1;
  444. new_head = ap->dccpap_buf_head - packets;
  445. if (new_head < 0) {
  446. if (gap > 0) {
  447. memset(ap->dccpap_buf, DCCP_ACKPKTS_STATE_NOT_RECEIVED,
  448. gap + new_head + 1);
  449. gap = -new_head;
  450. }
  451. new_head += ap->dccpap_buf_len;
  452. }
  453. ap->dccpap_buf_head = new_head;
  454. if (gap > 0)
  455. memset(ap->dccpap_buf + ap->dccpap_buf_head + 1,
  456. DCCP_ACKPKTS_STATE_NOT_RECEIVED, gap);
  457. ap->dccpap_buf[ap->dccpap_buf_head] = state;
  458. ap->dccpap_buf_vector_len += packets;
  459. return 0;
  460. }
  461. /*
  462. * Implements the draft-ietf-dccp-spec-11.txt Appendix A
  463. */
  464. int dccp_ackpkts_add(struct dccp_ackpkts *ap, u64 ackno, u8 state)
  465. {
  466. /*
  467. * Check at the right places if the buffer is full, if it is, tell the
  468. * caller to start dropping packets till the HC-Sender acks our ACK
  469. * vectors, when we will free up space in dccpap_buf.
  470. *
  471. * We may well decide to do buffer compression, etc, but for now lets
  472. * just drop.
  473. *
  474. * From Appendix A:
  475. *
  476. * Of course, the circular buffer may overflow, either when the HC-
  477. * Sender is sending data at a very high rate, when the HC-Receiver's
  478. * acknowledgements are not reaching the HC-Sender, or when the HC-
  479. * Sender is forgetting to acknowledge those acks (so the HC-Receiver
  480. * is unable to clean up old state). In this case, the HC-Receiver
  481. * should either compress the buffer (by increasing run lengths when
  482. * possible), transfer its state to a larger buffer, or, as a last
  483. * resort, drop all received packets, without processing them
  484. * whatsoever, until its buffer shrinks again.
  485. */
  486. /* See if this is the first ackno being inserted */
  487. if (ap->dccpap_buf_vector_len == 0) {
  488. ap->dccpap_buf[ap->dccpap_buf_head] = state;
  489. ap->dccpap_buf_vector_len = 1;
  490. } else if (after48(ackno, ap->dccpap_buf_ackno)) {
  491. const u64 delta = dccp_delta_seqno(ap->dccpap_buf_ackno, ackno);
  492. /*
  493. * Look if the state of this packet is the same as the previous ackno
  494. * and if so if we can bump the head len.
  495. */
  496. if (delta == 1 &&
  497. dccp_ackpkts_state(ap, ap->dccpap_buf_head) == state &&
  498. dccp_ackpkts_len(ap, ap->dccpap_buf_head) < DCCP_ACKPKTS_LEN_MASK)
  499. ap->dccpap_buf[ap->dccpap_buf_head]++;
  500. else if (dccp_ackpkts_set_buf_head_state(ap, delta, state))
  501. return -ENOBUFS;
  502. } else {
  503. /*
  504. * A.1.2. Old Packets
  505. *
  506. * When a packet with Sequence Number S arrives, and S <= buf_ackno,
  507. * the HC-Receiver will scan the table for the byte corresponding to S.
  508. * (Indexing structures could reduce the complexity of this scan.)
  509. */
  510. u64 delta = dccp_delta_seqno(ackno, ap->dccpap_buf_ackno);
  511. unsigned int index = ap->dccpap_buf_head;
  512. while (1) {
  513. const u8 len = dccp_ackpkts_len(ap, index);
  514. const u8 state = dccp_ackpkts_state(ap, index);
  515. /*
  516. * valid packets not yet in dccpap_buf have a reserved entry, with
  517. * a len equal to 0
  518. */
  519. if (state == DCCP_ACKPKTS_STATE_NOT_RECEIVED &&
  520. len == 0 && delta == 0) { /* Found our reserved seat! */
  521. dccp_pr_debug("Found %llu reserved seat!\n",
  522. (unsigned long long) ackno);
  523. ap->dccpap_buf[index] = state;
  524. goto out;
  525. }
  526. /* len == 0 means one packet */
  527. if (delta < len + 1)
  528. goto out_duplicate;
  529. delta -= len + 1;
  530. if (++index == ap->dccpap_buf_len)
  531. index = 0;
  532. }
  533. }
  534. ap->dccpap_buf_ackno = ackno;
  535. ap->dccpap_time = jiffies;
  536. out:
  537. dccp_pr_debug("");
  538. dccp_ackpkts_print(ap);
  539. return 0;
  540. out_duplicate:
  541. /* Duplicate packet */
  542. dccp_pr_debug("Received a dup or already considered lost packet: %llu\n",
  543. (unsigned long long) ackno);
  544. return -EILSEQ;
  545. }
  546. #ifdef DCCP_DEBUG
  547. void dccp_ackvector_print(const u64 ackno, const unsigned char *vector, int len)
  548. {
  549. if (!dccp_debug)
  550. return;
  551. printk("ACK vector len=%d, ackno=%llu |", len,
  552. (unsigned long long) ackno);
  553. while (len--) {
  554. const u8 state = (*vector & DCCP_ACKPKTS_STATE_MASK) >> 6;
  555. const u8 rl = (*vector & DCCP_ACKPKTS_LEN_MASK);
  556. printk("%d,%d|", state, rl);
  557. ++vector;
  558. }
  559. printk("\n");
  560. }
  561. void dccp_ackpkts_print(const struct dccp_ackpkts *ap)
  562. {
  563. dccp_ackvector_print(ap->dccpap_buf_ackno,
  564. ap->dccpap_buf + ap->dccpap_buf_head,
  565. ap->dccpap_buf_vector_len);
  566. }
  567. #endif
  568. static void dccp_ackpkts_trow_away_ack_record(struct dccp_ackpkts *ap)
  569. {
  570. /*
  571. * As we're keeping track of the ack vector size
  572. * (dccpap_buf_vector_len) and the sent ack vector size
  573. * (dccpap_ack_vector_len) we don't need dccpap_buf_tail at all, but
  574. * keep this code here as in the future we'll implement a vector of ack
  575. * records, as suggested in draft-ietf-dccp-spec-11.txt Appendix A. -acme
  576. */
  577. #if 0
  578. ap->dccpap_buf_tail = ap->dccpap_ack_ptr + 1;
  579. if (ap->dccpap_buf_tail >= ap->dccpap_buf_len)
  580. ap->dccpap_buf_tail -= ap->dccpap_buf_len;
  581. #endif
  582. ap->dccpap_buf_vector_len -= ap->dccpap_ack_vector_len;
  583. }
  584. void dccp_ackpkts_check_rcv_ackno(struct dccp_ackpkts *ap, struct sock *sk,
  585. u64 ackno)
  586. {
  587. /* Check if we actually sent an ACK vector */
  588. if (ap->dccpap_ack_seqno == DCCP_MAX_SEQNO + 1)
  589. return;
  590. if (ackno == ap->dccpap_ack_seqno) {
  591. #ifdef DCCP_DEBUG
  592. struct dccp_sock *dp = dccp_sk(sk);
  593. const char *debug_prefix = dp->dccps_role == DCCP_ROLE_CLIENT ? "CLIENT rx ack: " :
  594. "server rx ack: ";
  595. #endif
  596. dccp_pr_debug("%sACK packet 0, len=%d, ack_seqno=%llu, ack_ackno=%llu, ACKED!\n",
  597. debug_prefix, 1,
  598. (unsigned long long) ap->dccpap_ack_seqno,
  599. (unsigned long long) ap->dccpap_ack_ackno);
  600. dccp_ackpkts_trow_away_ack_record(ap);
  601. ap->dccpap_ack_seqno = DCCP_MAX_SEQNO + 1;
  602. }
  603. }
  604. static void dccp_ackpkts_check_rcv_ackvector(struct dccp_ackpkts *ap,
  605. struct sock *sk, u64 ackno,
  606. const unsigned char len,
  607. const unsigned char *vector)
  608. {
  609. unsigned char i;
  610. /* Check if we actually sent an ACK vector */
  611. if (ap->dccpap_ack_seqno == DCCP_MAX_SEQNO + 1)
  612. return;
  613. /*
  614. * We're in the receiver half connection, so if the received an ACK vector
  615. * ackno (e.g. 50) before dccpap_ack_seqno (e.g. 52), we're not interested.
  616. *
  617. * Extra explanation with example:
  618. *
  619. * if we received an ACK vector with ackno 50, it can only be acking
  620. * 50, 49, 48, etc, not 52 (the seqno for the ACK vector we sent).
  621. */
  622. // dccp_pr_debug("is %llu < %llu? ", ackno, ap->dccpap_ack_seqno);
  623. if (before48(ackno, ap->dccpap_ack_seqno)) {
  624. // dccp_pr_debug_cat("yes\n");
  625. return;
  626. }
  627. // dccp_pr_debug_cat("no\n");
  628. i = len;
  629. while (i--) {
  630. const u8 rl = (*vector & DCCP_ACKPKTS_LEN_MASK);
  631. u64 ackno_end_rl;
  632. dccp_set_seqno(&ackno_end_rl, ackno - rl);
  633. // dccp_pr_debug("is %llu <= %llu <= %llu? ", ackno_end_rl, ap->dccpap_ack_seqno, ackno);
  634. if (between48(ap->dccpap_ack_seqno, ackno_end_rl, ackno)) {
  635. const u8 state = (*vector & DCCP_ACKPKTS_STATE_MASK) >> 6;
  636. // dccp_pr_debug_cat("yes\n");
  637. if (state != DCCP_ACKPKTS_STATE_NOT_RECEIVED) {
  638. #ifdef DCCP_DEBUG
  639. struct dccp_sock *dp = dccp_sk(sk);
  640. const char *debug_prefix = dp->dccps_role == DCCP_ROLE_CLIENT ? "CLIENT rx ack: " :
  641. "server rx ack: ";
  642. #endif
  643. dccp_pr_debug("%sACK vector 0, len=%d, ack_seqno=%llu, ack_ackno=%llu, ACKED!\n",
  644. debug_prefix, len,
  645. (unsigned long long)
  646. ap->dccpap_ack_seqno,
  647. (unsigned long long)
  648. ap->dccpap_ack_ackno);
  649. dccp_ackpkts_trow_away_ack_record(ap);
  650. }
  651. /*
  652. * If dccpap_ack_seqno was not received, no problem we'll
  653. * send another ACK vector.
  654. */
  655. ap->dccpap_ack_seqno = DCCP_MAX_SEQNO + 1;
  656. break;
  657. }
  658. // dccp_pr_debug_cat("no\n");
  659. dccp_set_seqno(&ackno, ackno_end_rl - 1);
  660. ++vector;
  661. }
  662. }