virtio-blk: handle malloc failure

Signed-off-by: David Scott <[email protected]>
moby · Mar 6, 2020 · 11dae8f · 11dae8f
1 parent 6d0f4d1
commit 11dae8f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/lib/pci_virtio_block.c b/src/lib/pci_virtio_block.c
@@ -398,6 +398,10 @@ pci_vtblk_init(struct pci_devinst *pi, char *opts)
 
 	/* Customise the capabilities per-device */
 	vi_consts = (struct virtio_consts*)malloc(sizeof(struct virtio_consts));
+	if (vi_consts == NULL){
+		perror("Failed to allocate buffer for virtio_consts");
+		return (1);
+	}
 	*vi_consts = vtblk_vi_consts_defaults;
 	if (!blockif_candelete(bctxt)){
 		/* Don't tell the guest we can discard if the backing device doesn't support it */