Coverage Report

Created: 2026-10-24 08:16

/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
}