From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from eggs.gnu.org ([2001:4830:134:3::10]:54979) by lists.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1Wn9Nf-0000v1-Nr for qemu-devel@nongnu.org; Wed, 21 May 2014 12:28:46 -0400 Received: from Debian-exim by eggs.gnu.org with spam-scanned (Exim 4.71) (envelope-from ) id 1Wn9NZ-0002Dh-4G for qemu-devel@nongnu.org; Wed, 21 May 2014 12:28:39 -0400 Received: from mx1.redhat.com ([209.132.183.28]:6984) by eggs.gnu.org with esmtp (Exim 4.71) (envelope-from ) id 1Wn9NY-0002DZ-Rv for qemu-devel@nongnu.org; Wed, 21 May 2014 12:28:33 -0400 Received: from int-mx02.intmail.prod.int.phx2.redhat.com (int-mx02.intmail.prod.int.phx2.redhat.com [10.5.11.12]) by mx1.redhat.com (8.14.4/8.14.4) with ESMTP id s4LGSWMw005611 (version=TLSv1/SSLv3 cipher=DHE-RSA-AES256-SHA bits=256 verify=OK) for ; Wed, 21 May 2014 12:28:32 -0400 From: Kevin Wolf Date: Wed, 21 May 2014 18:28:05 +0200 Message-Id: <1400689698-3096-8-git-send-email-kwolf@redhat.com> In-Reply-To: <1400689698-3096-1-git-send-email-kwolf@redhat.com> References: <1400689698-3096-1-git-send-email-kwolf@redhat.com> Subject: [Qemu-devel] [PATCH 07/20] iscsi: Handle failure for potentially large allocations List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , To: qemu-devel@nongnu.org Cc: kwolf@redhat.com, stefanha@redhat.com Some code in the block layer makes potentially huge allocations. Failure is not completely unexpected there, so avoid aborting qemu and handle out-of-memory situations gracefully. This patch addresses the allocations in the iscsi block driver. Signed-off-by: Kevin Wolf --- block/iscsi.c | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/block/iscsi.c b/block/iscsi.c index 52355b8..c047ca8 100644 --- a/block/iscsi.c +++ b/block/iscsi.c @@ -296,7 +296,10 @@ static int coroutine_fn iscsi_co_writev(BlockDriverState *bs, data = iov->iov[0].iov_base; } else { size_t size = MIN(nb_sectors * BDRV_SECTOR_SIZE, iov->size); - buf = g_malloc(size); + buf = g_try_malloc(size); + if (buf == NULL) { + return -ENOMEM; + } qemu_iovec_to_buf(iov, 0, buf, size); data = buf; } @@ -550,7 +553,11 @@ static BlockDriverAIOCB *iscsi_aio_ioctl(BlockDriverState *bs, #else struct iovec *iov = (struct iovec *)acb->ioh->dxferp; - acb->buf = g_malloc(acb->ioh->dxfer_len); + acb->buf = g_try_malloc(acb->ioh->dxfer_len); + if (acb->buf == NULL) { + qemu_aio_release(acb); + return NULL; + } data.data = acb->buf; data.size = iov_to_buf(iov, acb->ioh->iovec_count, 0, acb->buf, acb->ioh->dxfer_len); @@ -823,7 +830,10 @@ coroutine_fn iscsi_co_write_zeroes(BlockDriverState *bs, int64_t sector_num, nb_blocks = sector_qemu2lun(nb_sectors, iscsilun); if (iscsilun->zeroblock == NULL) { - iscsilun->zeroblock = g_malloc0(iscsilun->block_size); + iscsilun->zeroblock = g_try_malloc0(iscsilun->block_size); + if (iscsilun->zeroblock == NULL) { + return -ENOMEM; + } } iscsi_co_init_iscsitask(iscsilun, &iTask); -- 1.8.3.1