/work/toxcore/events/dht_get_nodes_response.c

Source (jump to first uncovered line)
/* SPDX-License-Identifier: GPL-3.0-or-later
 * Copyright © 2022 The TokTok team.
 */

#include "events_alloc.h"

#include <assert.h>
#include <string.h>

#include "../bin_pack.h"
#include "../bin_unpack.h"
#include "../ccompat.h"
#include "../mem.h"
#include "../tox.h"
#include "../tox_events.h"
#include "../tox_private.h"


/*****************************************************
 *
 * :: struct and accessors
 *
 *****************************************************/


struct Tox_Event_Dht_Get_Nodes_Response {
    uint8_t public_key[TOX_PUBLIC_KEY_SIZE];
    uint8_t *ip;
    uint32_t ip_length;
    uint16_t port;
};

non_null()
static bool tox_event_dht_get_nodes_response_set_public_key(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response, const uint8_t public_key[TOX_PUBLIC_KEY_SIZE])
{
    memcpy(dht_get_nodes_response->public_key, public_key, TOX_PUBLIC_KEY_SIZE);
    return true;
}
const uint8_t *tox_event_dht_get_nodes_response_get_public_key(const Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
{
    return dht_get_nodes_response->public_key;
}

non_null()
static bool tox_event_dht_get_nodes_response_set_ip(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response,
        const char *ip, uint32_t ip_length, const Memory *mem)
{
    if (dht_get_nodes_response->ip != nullptr) {
        mem_delete(mem, dht_get_nodes_response->ip);
        dht_get_nodes_response->ip = nullptr;
        dht_get_nodes_response->ip_length = 0;
    }

    uint8_t *ip_tmp = (uint8_t *)mem_balloc(mem, ip_length);

    if (ip_tmp == nullptr) {
        return false;
    }

    memcpy(ip_tmp, ip, ip_length);
    dht_get_nodes_response->ip = ip_tmp;
    dht_get_nodes_response->ip_length = ip_length;
    return true;
}
uint32_t tox_event_dht_get_nodes_response_get_ip_length(const Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
{
    return dht_get_nodes_response->ip_length;
}
const uint8_t *tox_event_dht_get_nodes_response_get_ip(const Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
{
    return dht_get_nodes_response->ip;
}

non_null()
static bool tox_event_dht_get_nodes_response_set_port(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response, uint16_t port)
{
    dht_get_nodes_response->port = port;
    return true;
}
uint16_t tox_event_dht_get_nodes_response_get_port(const Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
{
    return dht_get_nodes_response->port;
}

non_null()
static void tox_event_dht_get_nodes_response_construct(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
{
    *dht_get_nodes_response = (Tox_Event_Dht_Get_Nodes_Response) {
        0
    };
}
non_null()
static void tox_event_dht_get_nodes_response_destruct(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response, const Memory *mem)
{
    mem_delete(mem, dht_get_nodes_response->ip);
}

bool tox_event_dht_get_nodes_response_pack(
    const Tox_Event_Dht_Get_Nodes_Response *event, Bin_Pack *bp)
{
    return bin_pack_array(bp, 3)
           && bin_pack_bin(bp, event->public_key, TOX_PUBLIC_KEY_SIZE)
           && bin_pack_bin(bp, event->ip, event->ip_length)
           && bin_pack_u16(bp, event->port);
}

non_null()
static bool tox_event_dht_get_nodes_response_unpack_into(
    Tox_Event_Dht_Get_Nodes_Response *event, Bin_Unpack *bu)
{
    if (!bin_unpack_array_fixed(bu, 3, nullptr)) {
        return false;
    }

    return bin_unpack_bin_fixed(bu, event->public_key, TOX_PUBLIC_KEY_SIZE)
           && bin_unpack_bin(bu, &event->ip, &event->ip_length)
           && bin_unpack_u16(bu, &event->port);
}

const Tox_Event_Dht_Get_Nodes_Response *tox_event_get_dht_get_nodes_response(
    const Tox_Event *event)
{
    return event->type == TOX_EVENT_DHT_GET_NODES_RESPONSE ? event->data.dht_get_nodes_response : nullptr;
}

Tox_Event_Dht_Get_Nodes_Response *tox_event_dht_get_nodes_response_new(const Memory *mem)
{
    Tox_Event_Dht_Get_Nodes_Response *const dht_get_nodes_response =
        (Tox_Event_Dht_Get_Nodes_Response *)mem_alloc(mem, sizeof(Tox_Event_Dht_Get_Nodes_Response));

    if (dht_get_nodes_response == nullptr) {
        return nullptr;
    }

    tox_event_dht_get_nodes_response_construct(dht_get_nodes_response);
    return dht_get_nodes_response;
}

void tox_event_dht_get_nodes_response_free(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response, const Memory *mem)
{
    if (dht_get_nodes_response != nullptr) {
        tox_event_dht_get_nodes_response_destruct(dht_get_nodes_response, mem);
    }
    mem_delete(mem, dht_get_nodes_response);
}

non_null()
static Tox_Event_Dht_Get_Nodes_Response *tox_events_add_dht_get_nodes_response(Tox_Events *events, const Memory *mem)
{
    Tox_Event_Dht_Get_Nodes_Response *const dht_get_nodes_response = tox_event_dht_get_nodes_response_new(mem);

    if (dht_get_nodes_response == nullptr) {
        return nullptr;
    }

    Tox_Event event;
    event.type = TOX_EVENT_DHT_GET_NODES_RESPONSE;
    event.data.dht_get_nodes_response = dht_get_nodes_response;

    tox_events_add(events, &event);
    return dht_get_nodes_response;
}

bool tox_event_dht_get_nodes_response_unpack(
    Tox_Event_Dht_Get_Nodes_Response **event, Bin_Unpack *bu, const Memory *mem)
{
    *event = tox_event_dht_get_nodes_response_new(mem);

    if (*event == nullptr) {
        return false;
    }

    return tox_event_dht_get_nodes_response_unpack_into(*event, bu);
}

non_null()
static Tox_Event_Dht_Get_Nodes_Response *tox_event_dht_get_nodes_response_alloc(void *user_data)
{
    Tox_Events_State *state = tox_events_alloc(user_data);
    assert(state != nullptr);

    if (state->events == nullptr) {
        return nullptr;
    }

    Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response = tox_events_add_dht_get_nodes_response(state->events, state->mem);

    if (dht_get_nodes_response == nullptr) {
        state->error = TOX_ERR_EVENTS_ITERATE_MALLOC;
        return nullptr;
    }

    return dht_get_nodes_response;
}


/*****************************************************
 *
 * :: event handler
 *
 *****************************************************/


void tox_events_handle_dht_get_nodes_response(
        Tox *tox, const uint8_t public_key[TOX_PUBLIC_KEY_SIZE],
        const char *ip, uint16_t port, void *user_data)
{
    Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response = tox_event_dht_get_nodes_response_alloc(user_data);

    if (dht_get_nodes_response == nullptr) {
        return;
    }

    const size_t ip_length = strlen(ip);
    if (ip_length >= UINT32_MAX) {
        return;
    }

    const Tox_System *sys = tox_get_system(tox);

    tox_event_dht_get_nodes_response_set_public_key(dht_get_nodes_response, public_key);
    tox_event_dht_get_nodes_response_set_ip(dht_get_nodes_response, ip, ip_length + 1, sys->mem);
    tox_event_dht_get_nodes_response_set_port(dht_get_nodes_response, port);
}

Coverage Report

Created: 2024-01-26 01:52

Line	Count	Source (jump to first uncovered line)
1		/* SPDX-License-Identifier: GPL-3.0-or-later
2		* Copyright © 2022 The TokTok team.
3		*/
4
5		#include "events_alloc.h"
6
7		#include <assert.h>
8		#include <string.h>
9
10		#include "../bin_pack.h"
11		#include "../bin_unpack.h"
12		#include "../ccompat.h"
13		#include "../mem.h"
14		#include "../tox.h"
15		#include "../tox_events.h"
16		#include "../tox_private.h"
17
18
19		/*****************************************************
20		*
21		* :: struct and accessors
22		*
23		*****************************************************/
24
25
26		struct Tox_Event_Dht_Get_Nodes_Response {
27		uint8_t public_key[TOX_PUBLIC_KEY_SIZE];
28		uint8_t *ip;
29		uint32_t ip_length;
30		uint16_t port;
31		};
32
33		non_null()
34		static bool tox_event_dht_get_nodes_response_set_public_key(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response, const uint8_t public_key[TOX_PUBLIC_KEY_SIZE])
35	167k	{
36	167k	memcpy(dht_get_nodes_response->public_key, public_key, TOX_PUBLIC_KEY_SIZE);
37	167k	return true;
38	167k	}
39		const uint8_t tox_event_dht_get_nodes_response_get_public_key(const Tox_Event_Dht_Get_Nodes_Response dht_get_nodes_response)
40	42.5k	{
41	42.5k	return dht_get_nodes_response->public_key;
42	42.5k	}
43
44		non_null()
45		static bool tox_event_dht_get_nodes_response_set_ip(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response,
46		const char ip, uint32_t ip_length, const Memory mem)
47	167k	{
48	167k	if (dht_get_nodes_response->ip != nullptr) {
49	0	mem_delete(mem, dht_get_nodes_response->ip);
50	0	dht_get_nodes_response->ip = nullptr;
51	0	dht_get_nodes_response->ip_length = 0;
52	0	}
53
54	167k	uint8_t ip_tmp = (uint8_t )mem_balloc(mem, ip_length);
55
56	167k	if (ip_tmp == nullptr) {
57	19	return false;
58	19	}
59
60	167k	memcpy(ip_tmp, ip, ip_length);
61	167k	dht_get_nodes_response->ip = ip_tmp;
62	167k	dht_get_nodes_response->ip_length = ip_length;
63	167k	return true;
64	167k	}
65		uint32_t tox_event_dht_get_nodes_response_get_ip_length(const Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
66	0	{
67	0	return dht_get_nodes_response->ip_length;
68	0	}
69		const uint8_t tox_event_dht_get_nodes_response_get_ip(const Tox_Event_Dht_Get_Nodes_Response dht_get_nodes_response)
70	42.5k	{
71	42.5k	return dht_get_nodes_response->ip;
72	42.5k	}
73
74		non_null()
75		static bool tox_event_dht_get_nodes_response_set_port(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response, uint16_t port)
76	167k	{
77	167k	dht_get_nodes_response->port = port;
78	167k	return true;
79	167k	}
80		uint16_t tox_event_dht_get_nodes_response_get_port(const Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
81	42.5k	{
82	42.5k	return dht_get_nodes_response->port;
83	42.5k	}
84
85		non_null()
86		static void tox_event_dht_get_nodes_response_construct(Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response)
87	168k	{
88	168k	*dht_get_nodes_response = (Tox_Event_Dht_Get_Nodes_Response) {
89	168k	0
90	168k	};
91	168k	}
92		non_null()
93		static void tox_event_dht_get_nodes_response_destruct(Tox_Event_Dht_Get_Nodes_Response dht_get_nodes_response, const Memory mem)
94	167k	{
95	167k	mem_delete(mem, dht_get_nodes_response->ip);
96	167k	}
97
98		bool tox_event_dht_get_nodes_response_pack(
99		const Tox_Event_Dht_Get_Nodes_Response event, Bin_Pack bp)
100	238	{
101	238	return bin_pack_array(bp, 3)
102	238	&& bin_pack_bin(bp, event->public_key, TOX_PUBLIC_KEY_SIZE)
103	238	&& bin_pack_bin(bp, event->ip, event->ip_length)
104	238	&& bin_pack_u16(bp, event->port);
105	238	}
106
107		non_null()
108		static bool tox_event_dht_get_nodes_response_unpack_into(
109		Tox_Event_Dht_Get_Nodes_Response event, Bin_Unpack bu)
110	35	{
111	35	if (!bin_unpack_array_fixed(bu, 3, nullptr)) {
112	1	return false;
113	1	}
114
115	34	return bin_unpack_bin_fixed(bu, event->public_key, TOX_PUBLIC_KEY_SIZE)
116	34	&& bin_unpack_bin(bu, &event->ip, &event->ip_length)
117	34	&& bin_unpack_u16(bu, &event->port);
118	35	}
119
120		const Tox_Event_Dht_Get_Nodes_Response *tox_event_get_dht_get_nodes_response(
121		const Tox_Event *event)
122	0	{
123	0	return event->type == TOX_EVENT_DHT_GET_NODES_RESPONSE ? event->data.dht_get_nodes_response : nullptr;
124	0	}
125
126		Tox_Event_Dht_Get_Nodes_Response tox_event_dht_get_nodes_response_new(const Memory mem)
127	168k	{
128	168k	Tox_Event_Dht_Get_Nodes_Response *const dht_get_nodes_response =
129	168k	(Tox_Event_Dht_Get_Nodes_Response *)mem_alloc(mem, sizeof(Tox_Event_Dht_Get_Nodes_Response));
130
131	168k	if (dht_get_nodes_response == nullptr) {
132	24	return nullptr;
133	24	}
134
135	168k	tox_event_dht_get_nodes_response_construct(dht_get_nodes_response);
136	168k	return dht_get_nodes_response;
137	168k	}
138
139		void tox_event_dht_get_nodes_response_free(Tox_Event_Dht_Get_Nodes_Response dht_get_nodes_response, const Memory mem)
140	167k	{
141	167k	if (dht_get_nodes_response != nullptr) {
142	167k	tox_event_dht_get_nodes_response_destruct(dht_get_nodes_response, mem);
143	167k	}
144	167k	mem_delete(mem, dht_get_nodes_response);
145	167k	}
146
147		non_null()
148		static Tox_Event_Dht_Get_Nodes_Response tox_events_add_dht_get_nodes_response(Tox_Events events, const Memory *mem)
149	167k	{
150	167k	Tox_Event_Dht_Get_Nodes_Response *const dht_get_nodes_response = tox_event_dht_get_nodes_response_new(mem);
151
152	167k	if (dht_get_nodes_response == nullptr) {
153	24	return nullptr;
154	24	}
155
156	167k	Tox_Event event;
157	167k	event.type = TOX_EVENT_DHT_GET_NODES_RESPONSE;
158	167k	event.data.dht_get_nodes_response = dht_get_nodes_response;
159
160	167k	tox_events_add(events, &event);
161	167k	return dht_get_nodes_response;
162	167k	}
163
164		bool tox_event_dht_get_nodes_response_unpack(
165		Tox_Event_Dht_Get_Nodes_Response *event, Bin_Unpack bu, const Memory *mem)
166	35	{
167	35	*event = tox_event_dht_get_nodes_response_new(mem);
168
169	35	if (*event == nullptr) {
170	0	return false;
171	0	}
172
173	35	return tox_event_dht_get_nodes_response_unpack_into(*event, bu);
174	35	}
175
176		non_null()
177		static Tox_Event_Dht_Get_Nodes_Response tox_event_dht_get_nodes_response_alloc(void user_data)
178	168k	{
179	168k	Tox_Events_State *state = tox_events_alloc(user_data);
180	168k	assert(state != nullptr);
181
182	168k	if (state->events == nullptr) {
183	27	return nullptr;
184	27	}
185
186	167k	Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response = tox_events_add_dht_get_nodes_response(state->events, state->mem);
187
188	167k	if (dht_get_nodes_response == nullptr) {
189	24	state->error = TOX_ERR_EVENTS_ITERATE_MALLOC;
190	24	return nullptr;
191	24	}
192
193	167k	return dht_get_nodes_response;
194	167k	}
195
196
197		/*****************************************************
198		*
199		* :: event handler
200		*
201		*****************************************************/
202
203
204		void tox_events_handle_dht_get_nodes_response(
205		Tox *tox, const uint8_t public_key[TOX_PUBLIC_KEY_SIZE],
206		const char ip, uint16_t port, void user_data)
207	168k	{
208	168k	Tox_Event_Dht_Get_Nodes_Response *dht_get_nodes_response = tox_event_dht_get_nodes_response_alloc(user_data);
209
210	168k	if (dht_get_nodes_response == nullptr) {
211	51	return;
212	51	}
213
214	167k	const size_t ip_length = strlen(ip);
215	167k	if (ip_length >= UINT32_MAX) {
216	0	return;
217	0	}
218
219	167k	const Tox_System *sys = tox_get_system(tox);
220
221	167k	tox_event_dht_get_nodes_response_set_public_key(dht_get_nodes_response, public_key);
222	167k	tox_event_dht_get_nodes_response_set_ip(dht_get_nodes_response, ip, ip_length + 1, sys->mem);
223	167k	tox_event_dht_get_nodes_response_set_port(dht_get_nodes_response, port);
224	167k	}