1 /*
2 * ngtcp2
3 *
4 * Copyright (c) 2017 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_idtr.h"
26
27 #include <assert.h>
28
ngtcp2_idtr_init(ngtcp2_idtr * idtr,int server,const ngtcp2_mem * mem)29 int ngtcp2_idtr_init(ngtcp2_idtr *idtr, int server, const ngtcp2_mem *mem) {
30 int rv;
31
32 rv = ngtcp2_gaptr_init(&idtr->gap, mem);
33 if (rv != 0) {
34 return rv;
35 }
36
37 idtr->server = server;
38
39 return 0;
40 }
41
ngtcp2_idtr_free(ngtcp2_idtr * idtr)42 void ngtcp2_idtr_free(ngtcp2_idtr *idtr) {
43 if (idtr == NULL) {
44 return;
45 }
46
47 ngtcp2_gaptr_free(&idtr->gap);
48 }
49
50 /*
51 * id_from_stream_id translates |stream_id| to id space used by
52 * ngtcp2_idtr.
53 */
id_from_stream_id(int64_t stream_id)54 static uint64_t id_from_stream_id(int64_t stream_id) {
55 return (uint64_t)(stream_id >> 2);
56 }
57
ngtcp2_idtr_open(ngtcp2_idtr * idtr,int64_t stream_id)58 int ngtcp2_idtr_open(ngtcp2_idtr *idtr, int64_t stream_id) {
59 uint64_t q;
60
61 assert((idtr->server && (stream_id % 2)) ||
62 (!idtr->server && (stream_id % 2)) == 0);
63
64 q = id_from_stream_id(stream_id);
65
66 if (ngtcp2_gaptr_is_pushed(&idtr->gap, q, 1)) {
67 return NGTCP2_ERR_STREAM_IN_USE;
68 }
69
70 return ngtcp2_gaptr_push(&idtr->gap, q, 1);
71 }
72
ngtcp2_idtr_is_open(ngtcp2_idtr * idtr,int64_t stream_id)73 int ngtcp2_idtr_is_open(ngtcp2_idtr *idtr, int64_t stream_id) {
74 uint64_t q;
75
76 assert((idtr->server && (stream_id % 2)) ||
77 (!idtr->server && (stream_id % 2)) == 0);
78
79 q = id_from_stream_id(stream_id);
80
81 return ngtcp2_gaptr_is_pushed(&idtr->gap, q, 1);
82 }
83
ngtcp2_idtr_first_gap(ngtcp2_idtr * idtr)84 uint64_t ngtcp2_idtr_first_gap(ngtcp2_idtr *idtr) {
85 return ngtcp2_gaptr_first_gap_offset(&idtr->gap);
86 }
87