1
0

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; virBitmapNew;
virBitmapNewCopy; virBitmapNewCopy;
virBitmapNewData; virBitmapNewData;
virBitmapNewQuiet;
virBitmapNextClearBit; virBitmapNextClearBit;
virBitmapNextSetBit; virBitmapNextSetBit;
virBitmapOverlaps; virBitmapOverlaps;

View File

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

View File

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