Free any allocated memory upon allocation failure

diff --git a/libarchive/archive_write_disk_set_standard_lookup.c b/libarchive/archive_write_disk_set_standard_lookup.c
index cf544dd7b608500ecedb832798b710b2efae742f..3b868fbad195e2b4e7f0090b238b5db6baa447b4 100644 (file)
--- a/libarchive/archive_write_disk_set_standard_lookup.c
+++ b/libarchive/archive_write_disk_set_standard_lookup.c
@@ -86,6 +86,11 @@ archive_write_disk_set_standard_lookup(struct archive *a)
 {
        struct bucket *ucache = malloc(cache_size * sizeof(struct bucket));
        struct bucket *gcache = malloc(cache_size * sizeof(struct bucket));
+       if (ucache == NULL || gcache == NULL) {
+               free(ucache);
+               free(gcache);
+               return (ARCHIVE_FATAL);
+       }
        memset(ucache, 0, cache_size * sizeof(struct bucket));
        memset(gcache, 0, cache_size * sizeof(struct bucket));
        archive_write_disk_set_group_lookup(a, gcache, lookup_gid, cleanup);