/src/nghttp2/lib/nghttp2_rcbuf.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* |
2 | | * nghttp2 - HTTP/2 C Library |
3 | | * |
4 | | * Copyright (c) 2016 Tatsuhiro Tsujikawa |
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 "nghttp2_rcbuf.h" |
26 | | |
27 | | #include <string.h> |
28 | | #include <assert.h> |
29 | | |
30 | | #include "nghttp2_mem.h" |
31 | | #include "nghttp2_helper.h" |
32 | | |
33 | | int nghttp2_rcbuf_new(nghttp2_rcbuf **rcbuf_ptr, size_t size, |
34 | 0 | nghttp2_mem *mem) { |
35 | 0 | uint8_t *p; |
36 | |
|
37 | 0 | p = nghttp2_mem_malloc(mem, sizeof(nghttp2_rcbuf) + size); |
38 | 0 | if (p == NULL) { |
39 | 0 | return NGHTTP2_ERR_NOMEM; |
40 | 0 | } |
41 | | |
42 | 0 | *rcbuf_ptr = (void *)p; |
43 | |
|
44 | 0 | (*rcbuf_ptr)->mem_user_data = mem->mem_user_data; |
45 | 0 | (*rcbuf_ptr)->free = mem->free; |
46 | 0 | (*rcbuf_ptr)->base = p + sizeof(nghttp2_rcbuf); |
47 | 0 | (*rcbuf_ptr)->len = size; |
48 | 0 | (*rcbuf_ptr)->ref = 1; |
49 | |
|
50 | 0 | return 0; |
51 | 0 | } |
52 | | |
53 | | int nghttp2_rcbuf_new2(nghttp2_rcbuf **rcbuf_ptr, const uint8_t *src, |
54 | 0 | size_t srclen, nghttp2_mem *mem) { |
55 | 0 | int rv; |
56 | |
|
57 | 0 | rv = nghttp2_rcbuf_new(rcbuf_ptr, srclen + 1, mem); |
58 | 0 | if (rv != 0) { |
59 | 0 | return rv; |
60 | 0 | } |
61 | | |
62 | 0 | (*rcbuf_ptr)->len = srclen; |
63 | 0 | *nghttp2_cpymem((*rcbuf_ptr)->base, src, srclen) = '\0'; |
64 | |
|
65 | 0 | return 0; |
66 | 0 | } |
67 | | |
68 | | /* |
69 | | * Frees |rcbuf| itself, regardless of its reference cout. |
70 | | */ |
71 | 0 | void nghttp2_rcbuf_del(nghttp2_rcbuf *rcbuf) { |
72 | 0 | nghttp2_mem_free2(rcbuf->free, rcbuf, rcbuf->mem_user_data); |
73 | 0 | } |
74 | | |
75 | 0 | void nghttp2_rcbuf_incref(nghttp2_rcbuf *rcbuf) { |
76 | 0 | if (rcbuf->ref == -1) { |
77 | 0 | return; |
78 | 0 | } |
79 | | |
80 | 0 | ++rcbuf->ref; |
81 | 0 | } |
82 | | |
83 | 0 | void nghttp2_rcbuf_decref(nghttp2_rcbuf *rcbuf) { |
84 | 0 | if (rcbuf == NULL || rcbuf->ref == -1) { |
85 | 0 | return; |
86 | 0 | } |
87 | | |
88 | 0 | assert(rcbuf->ref > 0); |
89 | | |
90 | 0 | if (--rcbuf->ref == 0) { |
91 | 0 | nghttp2_rcbuf_del(rcbuf); |
92 | 0 | } |
93 | 0 | } |
94 | | |
95 | 0 | nghttp2_vec nghttp2_rcbuf_get_buf(nghttp2_rcbuf *rcbuf) { |
96 | 0 | nghttp2_vec res = {rcbuf->base, rcbuf->len}; |
97 | 0 | return res; |
98 | 0 | } |
99 | | |
100 | 0 | int nghttp2_rcbuf_is_static(const nghttp2_rcbuf *rcbuf) { |
101 | 0 | return rcbuf->ref == -1; |
102 | 0 | } |