Message ID | 1401801062-9154-9-git-send-email-kwolf@redhat.com |
---|---|
State | New |
Headers | show |
The Tuesday 03 Jun 2014 à 15:10:49 (+0200), Kevin Wolf wrote : > 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 nfs block driver. > > Signed-off-by: Kevin Wolf <kwolf@redhat.com> > Reviewed-by: Stefan Hajnoczi <stefanha@redhat.com> > --- > block/nfs.c | 6 +++++- > 1 file changed, 5 insertions(+), 1 deletion(-) > > diff --git a/block/nfs.c b/block/nfs.c > index 539bd95..b37316f 100644 > --- a/block/nfs.c > +++ b/block/nfs.c > @@ -165,7 +165,11 @@ static int coroutine_fn nfs_co_writev(BlockDriverState *bs, > > nfs_co_init_task(client, &task); > > - buf = g_malloc(nb_sectors * BDRV_SECTOR_SIZE); > + buf = g_try_malloc(nb_sectors * BDRV_SECTOR_SIZE); > + if (nb_sectors && buf == NULL) { > + return -ENOMEM; > + } > + > qemu_iovec_to_buf(iov, 0, buf, nb_sectors * BDRV_SECTOR_SIZE); > > if (nfs_pwrite_async(client->context, client->fh, > -- > 1.8.3.1 > > Reviewed-by: Benoit Canet <benoit@irqsave.net>
diff --git a/block/nfs.c b/block/nfs.c index 539bd95..b37316f 100644 --- a/block/nfs.c +++ b/block/nfs.c @@ -165,7 +165,11 @@ static int coroutine_fn nfs_co_writev(BlockDriverState *bs, nfs_co_init_task(client, &task); - buf = g_malloc(nb_sectors * BDRV_SECTOR_SIZE); + buf = g_try_malloc(nb_sectors * BDRV_SECTOR_SIZE); + if (nb_sectors && buf == NULL) { + return -ENOMEM; + } + qemu_iovec_to_buf(iov, 0, buf, nb_sectors * BDRV_SECTOR_SIZE); if (nfs_pwrite_async(client->context, client->fh,