Coverage Report

Created: 2024-01-26 01:52

/work/toxcore/events/friend_lossless_packet.c
Line
Count
Source (jump to first uncovered line)
1
/* SPDX-License-Identifier: GPL-3.0-or-later
2
 * Copyright © 2023-2024 The TokTok team.
3
 */
4
5
#include "events_alloc.h"
6
7
#include <assert.h>
8
#include <stdlib.h>
9
#include <string.h>
10
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_events.h"
17
18
19
/*****************************************************
20
 *
21
 * :: struct and accessors
22
 *
23
 *****************************************************/
24
25
26
struct Tox_Event_Friend_Lossless_Packet {
27
    uint32_t friend_number;
28
    uint8_t *data;
29
    uint32_t data_length;
30
};
31
32
non_null()
33
static void tox_event_friend_lossless_packet_set_friend_number(Tox_Event_Friend_Lossless_Packet *friend_lossless_packet,
34
        uint32_t friend_number)
35
7
{
36
7
    assert(friend_lossless_packet != nullptr);
37
7
    friend_lossless_packet->friend_number = friend_number;
38
7
}
39
uint32_t tox_event_friend_lossless_packet_get_friend_number(const Tox_Event_Friend_Lossless_Packet *friend_lossless_packet)
40
0
{
41
0
    assert(friend_lossless_packet != nullptr);
42
0
    return friend_lossless_packet->friend_number;
43
0
}
44
45
non_null(1) nullable(2)
46
static bool tox_event_friend_lossless_packet_set_data(Tox_Event_Friend_Lossless_Packet *friend_lossless_packet,
47
        const uint8_t *data, uint32_t data_length)
48
7
{
49
7
    assert(friend_lossless_packet != nullptr);
50
51
7
    if (friend_lossless_packet->data != nullptr) {
52
0
        free(friend_lossless_packet->data);
53
0
        friend_lossless_packet->data = nullptr;
54
0
        friend_lossless_packet->data_length = 0;
55
0
    }
56
57
7
    if (data == nullptr) {
58
0
        assert(data_length == 0);
59
0
        return true;
60
0
    }
61
62
7
    uint8_t *data_copy = (uint8_t *)malloc(data_length);
63
64
7
    if (data_copy == nullptr) {
65
2
        return false;
66
2
    }
67
68
5
    memcpy(data_copy, data, data_length);
69
5
    friend_lossless_packet->data = data_copy;
70
5
    friend_lossless_packet->data_length = data_length;
71
5
    return true;
72
7
}
73
uint32_t tox_event_friend_lossless_packet_get_data_length(const Tox_Event_Friend_Lossless_Packet *friend_lossless_packet)
74
5
{
75
5
    assert(friend_lossless_packet != nullptr);
76
5
    return friend_lossless_packet->data_length;
77
5
}
78
const uint8_t *tox_event_friend_lossless_packet_get_data(const Tox_Event_Friend_Lossless_Packet *friend_lossless_packet)
79
5
{
80
5
    assert(friend_lossless_packet != nullptr);
81
5
    return friend_lossless_packet->data;
82
5
}
83
84
non_null()
85
static void tox_event_friend_lossless_packet_construct(Tox_Event_Friend_Lossless_Packet *friend_lossless_packet)
86
27
{
87
27
    *friend_lossless_packet = (Tox_Event_Friend_Lossless_Packet) {
88
27
        0
89
27
    };
90
27
}
91
non_null()
92
static void tox_event_friend_lossless_packet_destruct(Tox_Event_Friend_Lossless_Packet *friend_lossless_packet, const Memory *mem)
93
24
{
94
24
    free(friend_lossless_packet->data);
95
24
}
96
97
bool tox_event_friend_lossless_packet_pack(
98
    const Tox_Event_Friend_Lossless_Packet *event, Bin_Pack *bp)
99
22
{
100
22
    return bin_pack_array(bp, 2)
101
22
           && bin_pack_u32(bp, event->friend_number)
102
22
           && bin_pack_bin(bp, event->data, event->data_length);
103
22
}
104
105
non_null()
106
static bool tox_event_friend_lossless_packet_unpack_into(
107
    Tox_Event_Friend_Lossless_Packet *event, Bin_Unpack *bu)
108
20
{
109
20
    assert(event != nullptr);
110
20
    if (!bin_unpack_array_fixed(bu, 2, nullptr)) {
111
1
        return false;
112
1
    }
113
114
19
    return bin_unpack_u32(bu, &event->friend_number)
115
19
           && bin_unpack_bin(bu, &event->data, &event->data_length);
116
20
}
117
118
119
/*****************************************************
120
 *
121
 * :: new/free/add/get/size/unpack
122
 *
123
 *****************************************************/
124
125
const Tox_Event_Friend_Lossless_Packet *tox_event_get_friend_lossless_packet(const Tox_Event *event)
126
0
{
127
0
    return event->type == TOX_EVENT_FRIEND_LOSSLESS_PACKET ? event->data.friend_lossless_packet : nullptr;
128
0
}
129
130
Tox_Event_Friend_Lossless_Packet *tox_event_friend_lossless_packet_new(const Memory *mem)
131
30
{
132
30
    Tox_Event_Friend_Lossless_Packet *const friend_lossless_packet =
133
30
        (Tox_Event_Friend_Lossless_Packet *)mem_alloc(mem, sizeof(Tox_Event_Friend_Lossless_Packet));
134
135
30
    if (friend_lossless_packet == nullptr) {
136
3
        return nullptr;
137
3
    }
138
139
27
    tox_event_friend_lossless_packet_construct(friend_lossless_packet);
140
27
    return friend_lossless_packet;
141
30
}
142
143
void tox_event_friend_lossless_packet_free(Tox_Event_Friend_Lossless_Packet *friend_lossless_packet, const Memory *mem)
144
25
{
145
25
    if (friend_lossless_packet != nullptr) {
146
24
        tox_event_friend_lossless_packet_destruct(friend_lossless_packet, mem);
147
24
    }
148
25
    mem_delete(mem, friend_lossless_packet);
149
25
}
150
151
non_null()
152
static Tox_Event_Friend_Lossless_Packet *tox_events_add_friend_lossless_packet(Tox_Events *events, const Memory *mem)
153
9
{
154
9
    Tox_Event_Friend_Lossless_Packet *const friend_lossless_packet = tox_event_friend_lossless_packet_new(mem);
155
156
9
    if (friend_lossless_packet == nullptr) {
157
2
        return nullptr;
158
2
    }
159
160
7
    Tox_Event event;
161
7
    event.type = TOX_EVENT_FRIEND_LOSSLESS_PACKET;
162
7
    event.data.friend_lossless_packet = friend_lossless_packet;
163
164
7
    tox_events_add(events, &event);
165
7
    return friend_lossless_packet;
166
9
}
167
168
bool tox_event_friend_lossless_packet_unpack(
169
    Tox_Event_Friend_Lossless_Packet **event, Bin_Unpack *bu, const Memory *mem)
170
21
{
171
21
    assert(event != nullptr);
172
21
    assert(*event == nullptr);
173
21
    *event = tox_event_friend_lossless_packet_new(mem);
174
175
21
    if (*event == nullptr) {
176
1
        return false;
177
1
    }
178
179
20
    return tox_event_friend_lossless_packet_unpack_into(*event, bu);
180
21
}
181
182
non_null()
183
static Tox_Event_Friend_Lossless_Packet *tox_event_friend_lossless_packet_alloc(void *user_data)
184
11
{
185
11
    Tox_Events_State *state = tox_events_alloc(user_data);
186
11
    assert(state != nullptr);
187
188
11
    if (state->events == nullptr) {
189
2
        return nullptr;
190
2
    }
191
192
9
    Tox_Event_Friend_Lossless_Packet *friend_lossless_packet = tox_events_add_friend_lossless_packet(state->events, state->mem);
193
194
9
    if (friend_lossless_packet == nullptr) {
195
2
        state->error = TOX_ERR_EVENTS_ITERATE_MALLOC;
196
2
        return nullptr;
197
2
    }
198
199
7
    return friend_lossless_packet;
200
9
}
201
202
203
/*****************************************************
204
 *
205
 * :: event handler
206
 *
207
 *****************************************************/
208
209
210
void tox_events_handle_friend_lossless_packet(Tox *tox, uint32_t friend_number, const uint8_t *data, size_t length,
211
        void *user_data)
212
11
{
213
11
    Tox_Event_Friend_Lossless_Packet *friend_lossless_packet = tox_event_friend_lossless_packet_alloc(user_data);
214
215
11
    if (friend_lossless_packet == nullptr) {
216
4
        return;
217
4
    }
218
219
7
    tox_event_friend_lossless_packet_set_friend_number(friend_lossless_packet, friend_number);
220
7
    tox_event_friend_lossless_packet_set_data(friend_lossless_packet, data, length);
221
7
}