util: bitmap: Add option to allocate bitmap without reporting error

The virBitmapNew() function reports only OOM errors. Split out the
internals into a "quiet" function and add a wrapper that reports the
error.
This commit is contained in:
Peter Krempa 2015-01-30 11:19:59 +01:00
parent f0363aa145
commit bc2d8e5b06
3 changed files with 36 additions and 14 deletions

View File

@ -1048,6 +1048,7 @@ virBitmapLastSetBit;
virBitmapNew;
virBitmapNewCopy;
virBitmapNewData;
virBitmapNewQuiet;
virBitmapNextClearBit;
virBitmapNextSetBit;
virBitmapOverlaps;

View File

@ -54,31 +54,29 @@ struct _virBitmap {
/**
* virBitmapNew:
* virBitmapNewQuiet:
* @size: number of bits
*
* Allocate a bitmap capable of containing @size bits.
*
* Returns a pointer to the allocated bitmap or NULL if
* memory cannot be allocated.
* Returns a pointer to the allocated bitmap or NULL if memory cannot be
* allocated. Does not report libvirt errors.
*/
virBitmapPtr virBitmapNew(size_t size)
virBitmapPtr
virBitmapNewQuiet(size_t size)
{
virBitmapPtr bitmap;
size_t sz;
if (SIZE_MAX - VIR_BITMAP_BITS_PER_UNIT < size || size == 0) {
virReportOOMError();
return NULL;
}
sz = (size + VIR_BITMAP_BITS_PER_UNIT - 1) /
VIR_BITMAP_BITS_PER_UNIT;
if (VIR_ALLOC(bitmap) < 0)
if (SIZE_MAX - VIR_BITMAP_BITS_PER_UNIT < size || size == 0)
return NULL;
if (VIR_ALLOC_N(bitmap->map, sz) < 0) {
sz = (size + VIR_BITMAP_BITS_PER_UNIT - 1) / VIR_BITMAP_BITS_PER_UNIT;
if (VIR_ALLOC_QUIET(bitmap) < 0)
return NULL;
if (VIR_ALLOC_N_QUIET(bitmap->map, sz) < 0) {
VIR_FREE(bitmap);
return NULL;
}
@ -88,6 +86,28 @@ virBitmapPtr virBitmapNew(size_t size)
return bitmap;
}
/**
* virBitmapNew:
* @size: number of bits
*
* Allocate a bitmap capable of containing @size bits.
*
* Returns a pointer to the allocated bitmap or NULL if memory cannot be
* allocated. Reports libvirt errors.
*/
virBitmapPtr
virBitmapNew(size_t size)
{
virBitmapPtr ret;
if (!(ret = virBitmapNewQuiet(size)))
virReportOOMError();
return ret;
}
/**
* virBitmapFree:
* @bitmap: previously allocated bitmap

View File

@ -35,6 +35,7 @@ typedef virBitmap *virBitmapPtr;
/*
* Allocate a bitmap capable of containing @size bits.
*/
virBitmapPtr virBitmapNewQuiet(size_t size) ATTRIBUTE_RETURN_CHECK;
virBitmapPtr virBitmapNew(size_t size) ATTRIBUTE_RETURN_CHECK;
/*