diff --git a/libfdisk/src/ask.c b/libfdisk/src/ask.c index 31f95fb06..274f6ba7b 100644 --- a/libfdisk/src/ask.c +++ b/libfdisk/src/ask.c @@ -36,6 +36,10 @@ int fdisk_set_ask(struct fdisk_context *cxt, struct fdisk_ask *fdisk_new_ask(void) { struct fdisk_ask *ask = calloc(1, sizeof(struct fdisk_ask)); + + if (!ask) + return NULL; + DBG(ASK, ul_debugobj(ask, "alloc")); ask->refcount = 1; return ask; diff --git a/libfdisk/src/item.c b/libfdisk/src/item.c index 86fa0fb1f..671f9ad38 100644 --- a/libfdisk/src/item.c +++ b/libfdisk/src/item.c @@ -40,6 +40,9 @@ struct fdisk_labelitem *fdisk_new_labelitem(void) { struct fdisk_labelitem *li = calloc(1, sizeof(*li)); + if (!li) + return NULL; + li->refcount = 1; DBG(ITEM, ul_debugobj(li, "alloc")); return li;