ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-18 12:48:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix network server

Browse Source
This commit is contained in:
van Hauser
2020-08-31 16:50:26 +02:00
parent 909262f6c5
commit 4398b9b517
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/afl_network_proxy/afl-network-server.c
View File

@ -74,6 +74,7 @@ static u8 *in_data; /* Input data for trimming */
static u8 *buf2; static u8 *buf2;
static s32 in_len; static s32 in_len;
static s32 buf2_len;
static u32 map_size = MAP_SIZE; static u32 map_size = MAP_SIZE;
static volatile u8 stop_soon; /* Ctrl-C pressed? */ static volatile u8 stop_soon; /* Ctrl-C pressed? */
@ -381,6 +382,7 @@ int recv_testcase(int s, void **buf) {
if (clen < 1) if (clen < 1)
FATAL("did not receive valid compressed len information: %u", clen); FATAL("did not receive valid compressed len information: %u", clen);
buf2 = afl_realloc((void **)&buf2, clen); buf2 = afl_realloc((void **)&buf2, clen);
buf2_len = clen;
if (unlikely(!buf2)) { PFATAL("Alloc"); } if (unlikely(!buf2)) { PFATAL("Alloc"); }
received = 0; received = 0;
while (received < clen && while (received < clen &&
@ -641,6 +643,7 @@ int main(int argc, char **argv_orig, char **envp) {
compressor = libdeflate_alloc_compressor(1); compressor = libdeflate_alloc_compressor(1);
decompressor = libdeflate_alloc_decompressor(); decompressor = libdeflate_alloc_decompressor();
buf2 = afl_realloc((void **)&buf2, map_size + 16); buf2 = afl_realloc((void **)&buf2, map_size + 16);
buf2_len = map_size + 16;
if (unlikely(!buf2)) { PFATAL("alloc"); } if (unlikely(!buf2)) { PFATAL("alloc"); }
lenptr = (u32 *)(buf2 + 4); lenptr = (u32 *)(buf2 + 4);
fprintf(stderr, "Compiled with compression support\n"); fprintf(stderr, "Compiled with compression support\n");