generic-library/isa-l
1
0
Fork 0
mirror of https://github.com/intel/isa-l.git synced 2025-02-25 07:41:06 +01:00
Code Issues Projects Releases Wiki Activity

igzip: Fix 0 length file and looping errors in igzip_inflate_test

Browse Source 
Change-Id: I328f241ba07d8a0ae4fbc4c7de2ea8913912a188
Signed-off-by: Roy Oursler <roy.j.oursler@intel.com>
This commit is contained in:
Roy Oursler 2017-06-05 13:05:20 -07:00 committed by Xiaodong Liu
parent fc1467deb2
commit 1a7c640ef9
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
igzip/igzip_inflate_test.c
View File

@ -229,6 +229,12 @@ int main(int argc, char **argv)
if (argc == 1)
printf("Error, no input file\n");
for (i = 1; i < argc; i++) {
file = NULL;
uncompressed_stream = NULL;
compressed_stream = NULL;
uncompressed_test_stream = NULL;
file = fopen(argv[i], "r");
if (file == NULL) {
printf("Error opening file %s\n", argv[i]);
@ -246,23 +252,25 @@ int main(int argc, char **argv)
}
compressed_length = compressBound(file_length);
if (file_length != 0) {
uncompressed_stream = malloc(file_length);
uncompressed_test_stream = malloc(file_length);
}
compressed_stream = malloc(compressed_length);
if (uncompressed_stream == NULL && file_length != 0) {
printf("\nFailed to allocate memory\n");
printf("\nFailed to allocate input memory\n");
exit(0);
}
if (compressed_stream == NULL) {
printf("\nFailed to allocate memory\n");
printf("\nFailed to allocate output memory\n");
exit(0);
}
if (uncompressed_test_stream == NULL) {
printf("\nFailed to allocate memory\n");
if (uncompressed_test_stream == NULL && file_length != 0) {
printf("\nFailed to allocate decompressed memory\n");
exit(0);
}
@ -285,7 +293,8 @@ int main(int argc, char **argv)
fflush(0);
fclose(file);
free(compressed_stream);
if (compressed_stream != NULL)
free(compressed_stream);
if (uncompressed_stream != NULL)
free(uncompressed_stream);
if (uncompressed_test_stream != NULL)