e2fsck: fix to return ENOMEM in alloc_size_dir()

Two memory allocation return check is missed.

Signed-off-by: Wang Shilong <wshilong@ddn.com>
Signed-off-by: Theodore Ts'o <tytso@mit.edu>

diff --git a/e2fsck/rehash.c b/e2fsck/rehash.c
index 2c908be..392cfe9 100644 (file)
--- a/e2fsck/rehash.c
+++ b/e2fsck/rehash.c
@@ -276,7 +276,11 @@ static errcode_t alloc_size_dir(ext2_filsys fs, struct out_dir *outdir,
                outdir->hashes = new_mem;
        } else {
                outdir->buf = malloc(blocks * fs->blocksize);
+               if (!outdir->buf)
+                       return ENOMEM;
                outdir->hashes = malloc(blocks * sizeof(ext2_dirhash_t));
+               if (!outdir->hashes)
+                       return ENOMEM;
                outdir->num = 0;
        }
        outdir->max = blocks;