From 6046bfd5e34054f61c682ec7e425e907a8f1e051 Mon Sep 17 00:00:00 2001 From: David Sterba Date: Fri, 6 Nov 2015 17:57:41 +0100 Subject: btrfs-progs: handle memory allocation failure in add_file_items Signed-off-by: David Sterba --- mkfs.c | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'mkfs.c') diff --git a/mkfs.c b/mkfs.c index 0ea39f3..7316cb1 100644 --- a/mkfs.c +++ b/mkfs.c @@ -648,6 +648,12 @@ static int add_file_items(struct btrfs_trans_handle *trans, if (st->st_size <= BTRFS_MAX_INLINE_DATA_SIZE(root)) { char *buffer = malloc(st->st_size); + + if (!buffer) { + ret = -ENOMEM; + goto end; + } + ret_read = pread64(fd, buffer, st->st_size, bytes_read); if (ret_read == -1) { fprintf(stderr, "%s read failed\n", path_name); -- cgit v1.2.1