/work/toxcore/events/dht_nodes_response.c
Line | Count | Source (jump to first uncovered line) |
1 | | /* SPDX-License-Identifier: GPL-3.0-or-later |
2 | | * Copyright © 2023-2026 The TokTok team. |
3 | | */ |
4 | | |
5 | | #include "events_alloc.h" |
6 | | |
7 | | #include <assert.h> |
8 | | #include <string.h> |
9 | | |
10 | | #include "../attributes.h" |
11 | | #include "../bin_pack.h" |
12 | | #include "../bin_unpack.h" |
13 | | #include "../ccompat.h" |
14 | | #include "../mem.h" |
15 | | #include "../tox.h" |
16 | | #include "../tox_event.h" |
17 | | #include "../tox_events.h" |
18 | | #include "../tox_struct.h" |
19 | | |
20 | | /***************************************************** |
21 | | * |
22 | | * :: struct and accessors |
23 | | * |
24 | | *****************************************************/ |
25 | | |
26 | | struct Tox_Event_Dht_Nodes_Response { |
27 | | uint8_t public_key[TOX_PUBLIC_KEY_SIZE]; |
28 | | char *_Nullable ip; |
29 | | uint32_t ip_length; |
30 | | uint16_t port; |
31 | | }; |
32 | | |
33 | | static bool tox_event_dht_nodes_response_set_public_key(Tox_Event_Dht_Nodes_Response *_Nonnull dht_nodes_response, const uint8_t public_key[TOX_PUBLIC_KEY_SIZE]) |
34 | 185k | { |
35 | 185k | assert(dht_nodes_response != nullptr); |
36 | 185k | memcpy(dht_nodes_response->public_key, public_key, TOX_PUBLIC_KEY_SIZE); |
37 | 185k | return true; |
38 | 185k | } |
39 | | const uint8_t *tox_event_dht_nodes_response_get_public_key(const Tox_Event_Dht_Nodes_Response *dht_nodes_response) |
40 | 101k | { |
41 | 101k | assert(dht_nodes_response != nullptr); |
42 | 101k | return dht_nodes_response->public_key; |
43 | 101k | } |
44 | | |
45 | | static bool tox_event_dht_nodes_response_set_ip(Tox_Event_Dht_Nodes_Response *_Nonnull dht_nodes_response, |
46 | | const Memory *_Nonnull mem, const char *_Nullable ip, uint32_t ip_length) |
47 | 185k | { |
48 | 185k | assert(dht_nodes_response != nullptr); |
49 | 185k | if (dht_nodes_response->ip != nullptr) { |
50 | 0 | mem_delete(mem, dht_nodes_response->ip); |
51 | 0 | dht_nodes_response->ip = nullptr; |
52 | 0 | dht_nodes_response->ip_length = 0; |
53 | 0 | } |
54 | | |
55 | 185k | if (ip == nullptr) { |
56 | 0 | assert(ip_length == 0); |
57 | 0 | return true; |
58 | 0 | } |
59 | | |
60 | 185k | if (ip_length == UINT32_MAX) { |
61 | 0 | return false; |
62 | 0 | } |
63 | | |
64 | 185k | char *ip_copy = (char *)mem_balloc(mem, ip_length + 1); |
65 | | |
66 | 185k | if (ip_copy == nullptr) { |
67 | 0 | return false; |
68 | 0 | } |
69 | | |
70 | 185k | memcpy(ip_copy, ip, ip_length); |
71 | 185k | ip_copy[ip_length] = 0; |
72 | 185k | dht_nodes_response->ip = ip_copy; |
73 | 185k | dht_nodes_response->ip_length = ip_length; |
74 | 185k | return true; |
75 | 185k | } |
76 | | uint32_t tox_event_dht_nodes_response_get_ip_length(const Tox_Event_Dht_Nodes_Response *dht_nodes_response) |
77 | 0 | { |
78 | 0 | assert(dht_nodes_response != nullptr); |
79 | 0 | return dht_nodes_response->ip_length; |
80 | 0 | } |
81 | | const char *tox_event_dht_nodes_response_get_ip(const Tox_Event_Dht_Nodes_Response *dht_nodes_response) |
82 | 101k | { |
83 | 101k | assert(dht_nodes_response != nullptr); |
84 | 101k | return dht_nodes_response->ip; |
85 | 101k | } |
86 | | |
87 | | static void tox_event_dht_nodes_response_set_port(Tox_Event_Dht_Nodes_Response *_Nonnull dht_nodes_response, uint16_t port) |
88 | 185k | { |
89 | 185k | assert(dht_nodes_response != nullptr); |
90 | 185k | dht_nodes_response->port = port; |
91 | 185k | } |
92 | | uint16_t tox_event_dht_nodes_response_get_port(const Tox_Event_Dht_Nodes_Response *dht_nodes_response) |
93 | 101k | { |
94 | 101k | assert(dht_nodes_response != nullptr); |
95 | 101k | return dht_nodes_response->port; |
96 | 101k | } |
97 | | |
98 | | static void tox_event_dht_nodes_response_construct(Tox_Event_Dht_Nodes_Response *_Nonnull dht_nodes_response) |
99 | 185k | { |
100 | 185k | *dht_nodes_response = (Tox_Event_Dht_Nodes_Response) { |
101 | 185k | { |
102 | 185k | 0 |
103 | 185k | } |
104 | 185k | }; |
105 | 185k | } |
106 | | static void tox_event_dht_nodes_response_destruct(Tox_Event_Dht_Nodes_Response *_Nonnull dht_nodes_response, const Memory *_Nonnull mem) |
107 | 185k | { |
108 | 185k | mem_delete(mem, dht_nodes_response->ip); |
109 | 185k | } |
110 | | |
111 | | bool tox_event_dht_nodes_response_pack( |
112 | | const Tox_Event_Dht_Nodes_Response *event, Bin_Pack *bp) |
113 | 244 | { |
114 | 244 | return bin_pack_array(bp, 3) |
115 | 244 | && bin_pack_bin(bp, event->public_key, TOX_PUBLIC_KEY_SIZE) |
116 | 244 | && bin_pack_str(bp, event->ip, event->ip_length) |
117 | 244 | && bin_pack_u16(bp, event->port); |
118 | 244 | } |
119 | | |
120 | | static bool tox_event_dht_nodes_response_unpack_into(Tox_Event_Dht_Nodes_Response *_Nonnull event, Bin_Unpack *_Nonnull bu) |
121 | 58 | { |
122 | 58 | assert(event != nullptr); |
123 | 58 | if (!bin_unpack_array_fixed(bu, 3, nullptr)) { |
124 | 2 | return false; |
125 | 2 | } |
126 | | |
127 | 56 | return bin_unpack_bin_fixed(bu, event->public_key, TOX_PUBLIC_KEY_SIZE) |
128 | 56 | && bin_unpack_str(bu, &event->ip, &event->ip_length) |
129 | 56 | && bin_unpack_u16(bu, &event->port); |
130 | 58 | } |
131 | | |
132 | | /***************************************************** |
133 | | * |
134 | | * :: new/free/add/get/size/unpack |
135 | | * |
136 | | *****************************************************/ |
137 | | |
138 | | const Tox_Event_Dht_Nodes_Response *tox_event_get_dht_nodes_response(const Tox_Event *event) |
139 | 0 | { |
140 | 0 | return event->type == TOX_EVENT_DHT_NODES_RESPONSE ? event->data.dht_nodes_response : nullptr; |
141 | 0 | } |
142 | | |
143 | | Tox_Event_Dht_Nodes_Response *tox_event_dht_nodes_response_new(const Memory *mem) |
144 | 185k | { |
145 | 185k | Tox_Event_Dht_Nodes_Response *const dht_nodes_response = |
146 | 185k | (Tox_Event_Dht_Nodes_Response *)mem_alloc(mem, sizeof(Tox_Event_Dht_Nodes_Response)); |
147 | | |
148 | 185k | if (dht_nodes_response == nullptr) { |
149 | 1 | return nullptr; |
150 | 1 | } |
151 | | |
152 | 185k | tox_event_dht_nodes_response_construct(dht_nodes_response); |
153 | 185k | return dht_nodes_response; |
154 | 185k | } |
155 | | |
156 | | void tox_event_dht_nodes_response_free(Tox_Event_Dht_Nodes_Response *dht_nodes_response, const Memory *mem) |
157 | 185k | { |
158 | 185k | if (dht_nodes_response != nullptr) { |
159 | 185k | tox_event_dht_nodes_response_destruct(dht_nodes_response, mem); |
160 | 185k | } |
161 | 185k | mem_delete(mem, dht_nodes_response); |
162 | 185k | } |
163 | | |
164 | | static Tox_Event_Dht_Nodes_Response *_Nullable tox_events_add_dht_nodes_response(Tox_Events *_Nonnull events, const Memory *_Nonnull mem) |
165 | 185k | { |
166 | 185k | Tox_Event_Dht_Nodes_Response *const dht_nodes_response = tox_event_dht_nodes_response_new(mem); |
167 | | |
168 | 185k | if (dht_nodes_response == nullptr) { |
169 | 0 | return nullptr; |
170 | 0 | } |
171 | | |
172 | 185k | Tox_Event event; |
173 | 185k | event.type = TOX_EVENT_DHT_NODES_RESPONSE; |
174 | 185k | event.data.dht_nodes_response = dht_nodes_response; |
175 | | |
176 | 185k | if (!tox_events_add(events, &event)) { |
177 | 0 | tox_event_dht_nodes_response_free(dht_nodes_response, mem); |
178 | 0 | return nullptr; |
179 | 0 | } |
180 | 185k | return dht_nodes_response; |
181 | 185k | } |
182 | | |
183 | | bool tox_event_dht_nodes_response_unpack( |
184 | | Tox_Event_Dht_Nodes_Response **event, Bin_Unpack *bu, const Memory *mem) |
185 | 59 | { |
186 | 59 | assert(event != nullptr); |
187 | 59 | assert(*event == nullptr); |
188 | 59 | *event = tox_event_dht_nodes_response_new(mem); |
189 | | |
190 | 59 | if (*event == nullptr) { |
191 | 1 | return false; |
192 | 1 | } |
193 | | |
194 | 58 | return tox_event_dht_nodes_response_unpack_into(*event, bu); |
195 | 59 | } |
196 | | |
197 | | static Tox_Event_Dht_Nodes_Response *_Nullable tox_event_dht_nodes_response_alloc(Tox_Events_State *_Nonnull state) |
198 | 185k | { |
199 | 185k | if (state->events == nullptr) { |
200 | 0 | return nullptr; |
201 | 0 | } |
202 | | |
203 | 185k | Tox_Event_Dht_Nodes_Response *dht_nodes_response = tox_events_add_dht_nodes_response(state->events, state->mem); |
204 | | |
205 | 185k | if (dht_nodes_response == nullptr) { |
206 | 0 | state->error = TOX_ERR_EVENTS_ITERATE_MALLOC; |
207 | 0 | return nullptr; |
208 | 0 | } |
209 | | |
210 | 185k | return dht_nodes_response; |
211 | 185k | } |
212 | | |
213 | | /***************************************************** |
214 | | * |
215 | | * :: event handler |
216 | | * |
217 | | *****************************************************/ |
218 | | |
219 | | void tox_events_handle_dht_nodes_response( |
220 | | Tox *tox, |
221 | | const uint8_t *public_key, |
222 | | const char *ip, uint32_t ip_length, |
223 | | uint16_t port, |
224 | | void *user_data) |
225 | 185k | { |
226 | 185k | Tox_Events_State *state = tox_events_alloc(user_data); |
227 | 185k | Tox_Event_Dht_Nodes_Response *dht_nodes_response = tox_event_dht_nodes_response_alloc(state); |
228 | | |
229 | 185k | if (dht_nodes_response == nullptr) { |
230 | 0 | return; |
231 | 0 | } |
232 | | |
233 | 185k | tox_event_dht_nodes_response_set_public_key(dht_nodes_response, public_key); |
234 | 185k | if (!tox_event_dht_nodes_response_set_ip(dht_nodes_response, state->mem, ip, ip_length)) { |
235 | 0 | tox_event_dht_nodes_response_free(dht_nodes_response, state->mem); |
236 | 0 | state->error = TOX_ERR_EVENTS_ITERATE_MALLOC; |
237 | 0 | return; |
238 | 0 | } |
239 | 185k | tox_event_dht_nodes_response_set_port(dht_nodes_response, port); |
240 | 185k | } |
241 | | |
242 | | void tox_events_handle_dht_nodes_response_dispatch(Tox *tox, const Tox_Event_Dht_Nodes_Response *event, void *user_data) |
243 | 0 | { |
244 | 0 | if (tox->dht_nodes_response_callback == nullptr) { |
245 | 0 | return; |
246 | 0 | } |
247 | | |
248 | 0 | tox_unlock(tox); |
249 | 0 | tox->dht_nodes_response_callback(tox, event->public_key, (const char *)event->ip, event->ip_length, event->port, user_data); |
250 | 0 | tox_lock(tox); |
251 | 0 | } |