1 /*
2  * nghttp2 - HTTP/2 C Library
3  *
4  * Copyright (c) 2014 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_callbacks.h"
26 #include <string.h>
27 #include <stdlib.h>
28 #ifdef INFRA_MEM_STATS
29 #include "infra_mem_stats.h"
30 #endif
31 
32 extern void *HAL_Malloc(uint32_t size);
33 extern void HAL_Free(void *ptr);
34 
35 #if INFRA_MEM_STATS
36 #define NGHTTP2_CB_MALLOC(size) LITE_malloc(size, MEM_MAGIC, "nghttp2.cb")
37 #define NGHTTP2_CB_FREE(ptr)    LITE_free(ptr)
38 #else
39 #define NGHTTP2_CB_MALLOC(size) HAL_Malloc(size)
40 #define NGHTTP2_CB_FREE(ptr)   \
41     {                          \
42         HAL_Free((void *)ptr); \
43         ptr = NULL;            \
44     }
45 #endif
46 
nghttp2_session_callbacks_new(nghttp2_session_callbacks ** callbacks_ptr)47 int nghttp2_session_callbacks_new(nghttp2_session_callbacks **callbacks_ptr)
48 {
49     *callbacks_ptr = NGHTTP2_CB_MALLOC(sizeof(nghttp2_session_callbacks));
50 
51     if (*callbacks_ptr == NULL) {
52         return NGHTTP2_ERR_NOMEM;
53     }
54 
55     memset(*callbacks_ptr, 0, sizeof(nghttp2_session_callbacks));
56     return 0;
57 }
58 
nghttp2_session_callbacks_del(nghttp2_session_callbacks * callbacks)59 void nghttp2_session_callbacks_del(nghttp2_session_callbacks *callbacks)
60 {
61     NGHTTP2_CB_FREE(callbacks);
62 }
63 
nghttp2_session_callbacks_set_send_callback(nghttp2_session_callbacks * cbs,nghttp2_send_callback send_callback)64 void nghttp2_session_callbacks_set_send_callback(
65     nghttp2_session_callbacks *cbs, nghttp2_send_callback send_callback)
66 {
67     cbs->send_callback = send_callback;
68 }
69 
nghttp2_session_callbacks_set_recv_callback(nghttp2_session_callbacks * cbs,nghttp2_recv_callback recv_callback)70 void nghttp2_session_callbacks_set_recv_callback(
71     nghttp2_session_callbacks *cbs, nghttp2_recv_callback recv_callback)
72 {
73     cbs->recv_callback = recv_callback;
74 }
75 
nghttp2_session_callbacks_set_on_frame_recv_callback(nghttp2_session_callbacks * cbs,nghttp2_on_frame_recv_callback on_frame_recv_callback)76 void nghttp2_session_callbacks_set_on_frame_recv_callback(
77     nghttp2_session_callbacks *cbs,
78     nghttp2_on_frame_recv_callback on_frame_recv_callback)
79 {
80     cbs->on_frame_recv_callback = on_frame_recv_callback;
81 }
82 
nghttp2_session_callbacks_set_on_invalid_frame_recv_callback(nghttp2_session_callbacks * cbs,nghttp2_on_invalid_frame_recv_callback on_invalid_frame_recv_callback)83 void nghttp2_session_callbacks_set_on_invalid_frame_recv_callback(
84     nghttp2_session_callbacks *cbs,
85     nghttp2_on_invalid_frame_recv_callback on_invalid_frame_recv_callback)
86 {
87     cbs->on_invalid_frame_recv_callback = on_invalid_frame_recv_callback;
88 }
89 
nghttp2_session_callbacks_set_on_data_chunk_recv_callback(nghttp2_session_callbacks * cbs,nghttp2_on_data_chunk_recv_callback on_data_chunk_recv_callback)90 void nghttp2_session_callbacks_set_on_data_chunk_recv_callback(
91     nghttp2_session_callbacks *cbs,
92     nghttp2_on_data_chunk_recv_callback on_data_chunk_recv_callback)
93 {
94     cbs->on_data_chunk_recv_callback = on_data_chunk_recv_callback;
95 }
96 
nghttp2_session_callbacks_set_before_frame_send_callback(nghttp2_session_callbacks * cbs,nghttp2_before_frame_send_callback before_frame_send_callback)97 void nghttp2_session_callbacks_set_before_frame_send_callback(
98     nghttp2_session_callbacks *cbs,
99     nghttp2_before_frame_send_callback before_frame_send_callback)
100 {
101     cbs->before_frame_send_callback = before_frame_send_callback;
102 }
103 
nghttp2_session_callbacks_set_on_frame_send_callback(nghttp2_session_callbacks * cbs,nghttp2_on_frame_send_callback on_frame_send_callback)104 void nghttp2_session_callbacks_set_on_frame_send_callback(
105     nghttp2_session_callbacks *cbs,
106     nghttp2_on_frame_send_callback on_frame_send_callback)
107 {
108     cbs->on_frame_send_callback = on_frame_send_callback;
109 }
110 
nghttp2_session_callbacks_set_on_frame_not_send_callback(nghttp2_session_callbacks * cbs,nghttp2_on_frame_not_send_callback on_frame_not_send_callback)111 void nghttp2_session_callbacks_set_on_frame_not_send_callback(
112     nghttp2_session_callbacks *cbs,
113     nghttp2_on_frame_not_send_callback on_frame_not_send_callback)
114 {
115     cbs->on_frame_not_send_callback = on_frame_not_send_callback;
116 }
117 
nghttp2_session_callbacks_set_on_stream_close_callback(nghttp2_session_callbacks * cbs,nghttp2_on_stream_close_callback on_stream_close_callback)118 void nghttp2_session_callbacks_set_on_stream_close_callback(
119     nghttp2_session_callbacks *cbs,
120     nghttp2_on_stream_close_callback on_stream_close_callback)
121 {
122     cbs->on_stream_close_callback = on_stream_close_callback;
123 }
124 
nghttp2_session_callbacks_set_on_begin_headers_callback(nghttp2_session_callbacks * cbs,nghttp2_on_begin_headers_callback on_begin_headers_callback)125 void nghttp2_session_callbacks_set_on_begin_headers_callback(
126     nghttp2_session_callbacks *cbs,
127     nghttp2_on_begin_headers_callback on_begin_headers_callback)
128 {
129     cbs->on_begin_headers_callback = on_begin_headers_callback;
130 }
131 
nghttp2_session_callbacks_set_on_header_callback(nghttp2_session_callbacks * cbs,nghttp2_on_header_callback on_header_callback)132 void nghttp2_session_callbacks_set_on_header_callback(
133     nghttp2_session_callbacks *cbs,
134     nghttp2_on_header_callback on_header_callback)
135 {
136     cbs->on_header_callback = on_header_callback;
137 }
138 
nghttp2_session_callbacks_set_on_header_callback2(nghttp2_session_callbacks * cbs,nghttp2_on_header_callback2 on_header_callback2)139 void nghttp2_session_callbacks_set_on_header_callback2(
140     nghttp2_session_callbacks *cbs,
141     nghttp2_on_header_callback2 on_header_callback2)
142 {
143     cbs->on_header_callback2 = on_header_callback2;
144 }
145 
nghttp2_session_callbacks_set_on_invalid_header_callback(nghttp2_session_callbacks * cbs,nghttp2_on_invalid_header_callback on_invalid_header_callback)146 void nghttp2_session_callbacks_set_on_invalid_header_callback(
147     nghttp2_session_callbacks *cbs,
148     nghttp2_on_invalid_header_callback on_invalid_header_callback)
149 {
150     cbs->on_invalid_header_callback = on_invalid_header_callback;
151 }
152 
nghttp2_session_callbacks_set_on_invalid_header_callback2(nghttp2_session_callbacks * cbs,nghttp2_on_invalid_header_callback2 on_invalid_header_callback2)153 void nghttp2_session_callbacks_set_on_invalid_header_callback2(
154     nghttp2_session_callbacks *cbs,
155     nghttp2_on_invalid_header_callback2 on_invalid_header_callback2)
156 {
157     cbs->on_invalid_header_callback2 = on_invalid_header_callback2;
158 }
159 
nghttp2_session_callbacks_set_select_padding_callback(nghttp2_session_callbacks * cbs,nghttp2_select_padding_callback select_padding_callback)160 void nghttp2_session_callbacks_set_select_padding_callback(
161     nghttp2_session_callbacks *cbs,
162     nghttp2_select_padding_callback select_padding_callback)
163 {
164     cbs->select_padding_callback = select_padding_callback;
165 }
166 
nghttp2_session_callbacks_set_data_source_read_length_callback(nghttp2_session_callbacks * cbs,nghttp2_data_source_read_length_callback data_source_read_length_callback)167 void nghttp2_session_callbacks_set_data_source_read_length_callback(
168     nghttp2_session_callbacks *cbs,
169     nghttp2_data_source_read_length_callback data_source_read_length_callback)
170 {
171     cbs->read_length_callback = data_source_read_length_callback;
172 }
173 
nghttp2_session_callbacks_set_on_begin_frame_callback(nghttp2_session_callbacks * cbs,nghttp2_on_begin_frame_callback on_begin_frame_callback)174 void nghttp2_session_callbacks_set_on_begin_frame_callback(
175     nghttp2_session_callbacks *cbs,
176     nghttp2_on_begin_frame_callback on_begin_frame_callback)
177 {
178     cbs->on_begin_frame_callback = on_begin_frame_callback;
179 }
180 
nghttp2_session_callbacks_set_send_data_callback(nghttp2_session_callbacks * cbs,nghttp2_send_data_callback send_data_callback)181 void nghttp2_session_callbacks_set_send_data_callback(
182     nghttp2_session_callbacks *cbs,
183     nghttp2_send_data_callback send_data_callback)
184 {
185     cbs->send_data_callback = send_data_callback;
186 }
187 
nghttp2_session_callbacks_set_pack_extension_callback(nghttp2_session_callbacks * cbs,nghttp2_pack_extension_callback pack_extension_callback)188 void nghttp2_session_callbacks_set_pack_extension_callback(
189     nghttp2_session_callbacks *cbs,
190     nghttp2_pack_extension_callback pack_extension_callback)
191 {
192     cbs->pack_extension_callback = pack_extension_callback;
193 }
194 
nghttp2_session_callbacks_set_unpack_extension_callback(nghttp2_session_callbacks * cbs,nghttp2_unpack_extension_callback unpack_extension_callback)195 void nghttp2_session_callbacks_set_unpack_extension_callback(
196     nghttp2_session_callbacks *cbs,
197     nghttp2_unpack_extension_callback unpack_extension_callback)
198 {
199     cbs->unpack_extension_callback = unpack_extension_callback;
200 }
201 
nghttp2_session_callbacks_set_on_extension_chunk_recv_callback(nghttp2_session_callbacks * cbs,nghttp2_on_extension_chunk_recv_callback on_extension_chunk_recv_callback)202 void nghttp2_session_callbacks_set_on_extension_chunk_recv_callback(
203     nghttp2_session_callbacks *cbs,
204     nghttp2_on_extension_chunk_recv_callback on_extension_chunk_recv_callback)
205 {
206     cbs->on_extension_chunk_recv_callback = on_extension_chunk_recv_callback;
207 }
208 
nghttp2_session_callbacks_set_error_callback(nghttp2_session_callbacks * cbs,nghttp2_error_callback error_callback)209 void nghttp2_session_callbacks_set_error_callback(
210     nghttp2_session_callbacks *cbs, nghttp2_error_callback error_callback)
211 {
212     cbs->error_callback = error_callback;
213 }
214 
nghttp2_session_callbacks_set_error_callback2(nghttp2_session_callbacks * cbs,nghttp2_error_callback2 error_callback2)215 void nghttp2_session_callbacks_set_error_callback2(
216     nghttp2_session_callbacks *cbs, nghttp2_error_callback2 error_callback2)
217 {
218     cbs->error_callback2 = error_callback2;
219 }
220