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
33void ngtcp2_cid_zero(ngtcp2_cid *cid) { memset(cid, 0, sizeof(*cid)); }
34
35void 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
44int 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
49int 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
57int ngtcp2_cid_empty(const ngtcp2_cid *cid) { return cid->datalen == 0; }
58
59void 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
67void 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
73void 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
91void 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
98void ngtcp2_dcid_set_path(ngtcp2_dcid *dcid, const ngtcp2_path *path) {
99  ngtcp2_path_copy(&dcid->ps.path, path);
100}
101
102void 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
115void 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
126int 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
140int 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