VeltronNET/unifork.old
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add safety check to extracting

Browse source
This commit is contained in:
Pk11 2021-05-25 22:17:20 -05:00
commit f365f8710d
1 changed files with 11 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

12
source/utils/extract.cpp
View file

@ -124,7 +124,17 @@ Result extractArchive(const std::string &archivePath, const std::string &wantedF
return EXTRACT_ERROR_ARCHIVE;
}
fwrite(buf, 1, size, file);
size_t written = fwrite(buf, 1, size, file);
/* Failed to write, likely out of space. */
if (written != size) {
fclose(file);
delete[] buf;
archive_read_close(a);
archive_read_free(a);
return EXTRACT_ERROR_WRITEFILE;
}
sizeLeft -= size;
writeOffset += size;
}