rpmalloc: Add missing NULL checks to the allocation paths.

Fixes #15258.
This commit is contained in:
Augustin Cavalier
2019-08-17 12:29:20 -04:00
parent cc7e844c12
commit f70930539e
@@ -1351,6 +1351,9 @@ _memory_allocate_from_heap_fallback(heap_t* heap, uint32_t class_idx) {
//Find a span in one of the cache levels
active_span = _memory_heap_extract_new_span(heap, 1, class_idx);
if (!active_span)
return active_span;
//Mark span as owned by this heap and set base data, return first block
return _memory_span_set_new_active(heap, heap_class, active_span, class_idx);
}
@@ -1393,6 +1396,9 @@ _memory_allocate_large(heap_t* heap, size_t size) {
//Find a span in one of the cache levels
span_t* span = _memory_heap_extract_new_span(heap, span_count, SIZE_CLASS_COUNT);
if (!span)
return span;
//Mark span as owned by this heap and set base data
assert(span->span_count == span_count);
span->size_class = (uint32_t)(SIZE_CLASS_COUNT + idx);