1 /*
2 * ngtcp2
3 *
4 * Copyright (c) 2018 ngtcp2 contributors
5 *
6 * Permission is hereby granted, free of charge, to any person obtaining
7 * a copy of this software and associated documentation files (the
8 * "Software"), to deal in the Software without restriction, including
9 * without limitation the rights to use, copy, modify, merge, publish,
10 * distribute, sublicense, and/or sell copies of the Software, and to
11 * permit persons to whom the Software is furnished to do so, subject to
12 * the following conditions:
13 *
14 * The above copyright notice and this permission notice shall be
15 * included in all copies or substantial portions of the Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
18 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
19 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
20 * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
21 * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
22 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
23 * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
24 */
25 #include "ngtcp2_cid.h"
26
27 #include <assert.h>
28 #include <string.h>
29
30 #include "ngtcp2_path.h"
31 #include "ngtcp2_str.h"
32
ngtcp2_cid_zero(ngtcp2_cid * cid)33 void ngtcp2_cid_zero(ngtcp2_cid *cid) { memset(cid, 0, sizeof(*cid)); }
34
ngtcp2_cid_init(ngtcp2_cid * cid,const uint8_t * data,size_t datalen)35 void ngtcp2_cid_init(ngtcp2_cid *cid, const uint8_t *data, size_t datalen) {
36 assert(datalen <= NGTCP2_MAX_CIDLEN);
37
38 cid->datalen = datalen;
39 if (datalen) {
40 ngtcp2_cpymem(cid->data, data, datalen);
41 }
42 }
43
ngtcp2_cid_eq(const ngtcp2_cid * cid,const ngtcp2_cid * other)44 int ngtcp2_cid_eq(const ngtcp2_cid *cid, const ngtcp2_cid *other) {
45 return cid->datalen == other->datalen &&
46 0 == memcmp(cid->data, other->data, cid->datalen);
47 }
48
ngtcp2_cid_less(const ngtcp2_cid * lhs,const ngtcp2_cid * rhs)49 int ngtcp2_cid_less(const ngtcp2_cid *lhs, const ngtcp2_cid *rhs) {
50 int s = lhs->datalen < rhs->datalen;
51 size_t n = s ? lhs->datalen : rhs->datalen;
52 int c = memcmp(lhs->data, rhs->data, n);
53
54 return c < 0 || (c == 0 && s);
55 }
56
ngtcp2_cid_empty(const ngtcp2_cid * cid)57 int ngtcp2_cid_empty(const ngtcp2_cid *cid) { return cid->datalen == 0; }
58
ngtcp2_scid_init(ngtcp2_scid * scid,uint64_t seq,const ngtcp2_cid * cid)59 void ngtcp2_scid_init(ngtcp2_scid *scid, uint64_t seq, const ngtcp2_cid *cid) {
60 scid->pe.index = NGTCP2_PQ_BAD_INDEX;
61 scid->seq = seq;
62 scid->cid = *cid;
63 scid->retired_ts = UINT64_MAX;
64 scid->flags = NGTCP2_SCID_FLAG_NONE;
65 }
66
ngtcp2_scid_copy(ngtcp2_scid * dest,const ngtcp2_scid * src)67 void ngtcp2_scid_copy(ngtcp2_scid *dest, const ngtcp2_scid *src) {
68 ngtcp2_scid_init(dest, src->seq, &src->cid);
69 dest->retired_ts = src->retired_ts;
70 dest->flags = src->flags;
71 }
72
ngtcp2_dcid_init(ngtcp2_dcid * dcid,uint64_t seq,const ngtcp2_cid * cid,const uint8_t * token)73 void ngtcp2_dcid_init(ngtcp2_dcid *dcid, uint64_t seq, const ngtcp2_cid *cid,
74 const uint8_t *token) {
75 dcid->seq = seq;
76 dcid->cid = *cid;
77 if (token) {
78 memcpy(dcid->token, token, NGTCP2_STATELESS_RESET_TOKENLEN);
79 dcid->flags = NGTCP2_DCID_FLAG_TOKEN_PRESENT;
80 } else {
81 dcid->flags = NGTCP2_DCID_FLAG_NONE;
82 }
83 ngtcp2_path_storage_zero(&dcid->ps);
84 dcid->retired_ts = UINT64_MAX;
85 dcid->bound_ts = UINT64_MAX;
86 dcid->bytes_sent = 0;
87 dcid->bytes_recv = 0;
88 dcid->max_udp_payload_size = NGTCP2_MAX_UDP_PAYLOAD_SIZE;
89 }
90
ngtcp2_dcid_set_token(ngtcp2_dcid * dcid,const uint8_t * token)91 void ngtcp2_dcid_set_token(ngtcp2_dcid *dcid, const uint8_t *token) {
92 assert(token);
93
94 dcid->flags |= NGTCP2_DCID_FLAG_TOKEN_PRESENT;
95 memcpy(dcid->token, token, NGTCP2_STATELESS_RESET_TOKENLEN);
96 }
97
ngtcp2_dcid_set_path(ngtcp2_dcid * dcid,const ngtcp2_path * path)98 void ngtcp2_dcid_set_path(ngtcp2_dcid *dcid, const ngtcp2_path *path) {
99 ngtcp2_path_copy(&dcid->ps.path, path);
100 }
101
ngtcp2_dcid_copy(ngtcp2_dcid * dest,const ngtcp2_dcid * src)102 void ngtcp2_dcid_copy(ngtcp2_dcid *dest, const ngtcp2_dcid *src) {
103 ngtcp2_dcid_init(dest, src->seq, &src->cid,
104 (src->flags & NGTCP2_DCID_FLAG_TOKEN_PRESENT) ? src->token
105 : NULL);
106 ngtcp2_path_copy(&dest->ps.path, &src->ps.path);
107 dest->retired_ts = src->retired_ts;
108 dest->bound_ts = src->bound_ts;
109 dest->flags = src->flags;
110 dest->bytes_sent = src->bytes_sent;
111 dest->bytes_recv = src->bytes_recv;
112 dest->max_udp_payload_size = src->max_udp_payload_size;
113 }
114
ngtcp2_dcid_copy_cid_token(ngtcp2_dcid * dest,const ngtcp2_dcid * src)115 void ngtcp2_dcid_copy_cid_token(ngtcp2_dcid *dest, const ngtcp2_dcid *src) {
116 dest->seq = src->seq;
117 dest->cid = src->cid;
118 if (src->flags & NGTCP2_DCID_FLAG_TOKEN_PRESENT) {
119 dest->flags |= NGTCP2_DCID_FLAG_TOKEN_PRESENT;
120 memcpy(dest->token, src->token, NGTCP2_STATELESS_RESET_TOKENLEN);
121 } else if (dest->flags & NGTCP2_DCID_FLAG_TOKEN_PRESENT) {
122 dest->flags &= (uint8_t)~NGTCP2_DCID_FLAG_TOKEN_PRESENT;
123 }
124 }
125
ngtcp2_dcid_verify_uniqueness(ngtcp2_dcid * dcid,uint64_t seq,const ngtcp2_cid * cid,const uint8_t * token)126 int ngtcp2_dcid_verify_uniqueness(ngtcp2_dcid *dcid, uint64_t seq,
127 const ngtcp2_cid *cid, const uint8_t *token) {
128 if (dcid->seq == seq) {
129 return ngtcp2_cid_eq(&dcid->cid, cid) &&
130 (dcid->flags & NGTCP2_DCID_FLAG_TOKEN_PRESENT) &&
131 memcmp(dcid->token, token,
132 NGTCP2_STATELESS_RESET_TOKENLEN) == 0
133 ? 0
134 : NGTCP2_ERR_PROTO;
135 }
136
137 return !ngtcp2_cid_eq(&dcid->cid, cid) ? 0 : NGTCP2_ERR_PROTO;
138 }
139
ngtcp2_dcid_verify_stateless_reset_token(const ngtcp2_dcid * dcid,const uint8_t * token)140 int ngtcp2_dcid_verify_stateless_reset_token(const ngtcp2_dcid *dcid,
141 const uint8_t *token) {
142 return (dcid->flags & NGTCP2_DCID_FLAG_TOKEN_PRESENT) &&
143 ngtcp2_cmemeq(dcid->token, token,
144 NGTCP2_STATELESS_RESET_TOKENLEN)
145 ? 0
146 : NGTCP2_ERR_INVALID_ARGUMENT;
147 }
148