github/RRG-Proxmark3
1
0
Fork 0
mirror of https://github.com/RfidResearchGroup/proxmark3.git synced 2025-08-19 21:03:48 -07:00
Code Issues Projects Releases Packages Wiki Activity Actions

checking if calloc failed

Browse source
This commit is contained in:
iceman1001 2023-07-24 03:16:18 +02:00
commit b7900816bd
1 changed files with 18 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tools/fpga_compress/fpga_compress.c
View file

@ -144,12 +144,15 @@ typedef struct lz4_stream_s {
// Call it either with opened infile + outsize=0 // Call it either with opened infile + outsize=0
// or with opened infile, opened outfiles, num_outfiles and valid outsize // or with opened infile, opened outfiles, num_outfiles and valid outsize
static int zlib_decompress(FILE *infile, FILE *outfiles[], uint8_t num_outfiles, long *outsize) { static int zlib_decompress(FILE *infile, FILE *outfiles[], uint8_t num_outfiles, long *outsize) {
if (num_outfiles > 10) { if (num_outfiles > 10) {
return (EXIT_FAILURE); return (EXIT_FAILURE);
} }
LZ4_streamDecode_t lz4StreamDecode_body = {{ 0 }};
char outbuf[FPGA_RING_BUFFER_BYTES];
LZ4_streamDecode_t lz4StreamDecode_body = {{ 0 }};
char outbuf[FPGA_RING_BUFFER_BYTES] = {0};
// file size
fseek(infile, 0L, SEEK_END); fseek(infile, 0L, SEEK_END);
long infile_size = ftell(infile); long infile_size = ftell(infile);
fseek(infile, 0L, SEEK_SET); fseek(infile, 0L, SEEK_SET);
@ -168,17 +171,30 @@ static int zlib_decompress(FILE *infile, FILE *outfiles[], uint8_t num_outfiles,
char *outbufall = NULL; char *outbufall = NULL;
if (*outsize > 0) { if (*outsize > 0) {
outbufall = calloc(*outsize, sizeof(char)); outbufall = calloc(*outsize, sizeof(char));
if (outbufall == NULL) {
return (EXIT_FAILURE);
}
} }
char *inbuf = calloc(infile_size, sizeof(char)); char *inbuf = calloc(infile_size, sizeof(char));
if (inbuf == NULL) {
if (outbufall) {
free(outbufall);
}
return (EXIT_FAILURE);
}
size_t num_read = fread(inbuf, sizeof(char), infile_size, infile); size_t num_read = fread(inbuf, sizeof(char), infile_size, infile);
if (num_read != infile_size) { if (num_read != infile_size) {
if (*outsize > 0) { if (*outsize > 0) {
fclose(infile); fclose(infile);
for (uint16_t j = 0; j < num_outfiles; j++) { for (uint16_t j = 0; j < num_outfiles; j++) {
fclose(outfiles[j]); fclose(outfiles[j]);
} }
} }
if (outbufall) { if (outbufall) {
free(outbufall); free(outbufall);
} }