diff options
-rw-r--r-- | src/wsbm_manager.c | 2 | ||||
-rw-r--r-- | src/wsbm_userpool.c | 2 |
2 files changed, 3 insertions, 1 deletions
diff --git a/src/wsbm_manager.c b/src/wsbm_manager.c index a86b894..e95309d 100644 --- a/src/wsbm_manager.c +++ b/src/wsbm_manager.c @@ -1011,6 +1011,8 @@ wsbmBOCreateList(int target, int hasKernelBuffers) struct _WsbmBufferList *list = calloc(sizeof(*list), 1); int ret; + if (!list) + return NULL; list->hasKernelBuffers = hasKernelBuffers; if (hasKernelBuffers) { ret = validateCreateList(target, &list->kernelBuffers, 0); diff --git a/src/wsbm_userpool.c b/src/wsbm_userpool.c index f7833b7..9c7b79a 100644 --- a/src/wsbm_userpool.c +++ b/src/wsbm_userpool.c @@ -339,7 +339,7 @@ pool_validate(struct _WsbmBufStorage *buf, uint64_t set_flags, vBuf->node = wsbmMMGetBlock(vBuf->node, vBuf->size, vBuf->alignment); - if (!err) { + if (!err && vBuf->node) { vBuf->kBuf.placement = WSBM_PL_FLAG_VRAM; vBuf->kBuf.gpuOffset = p->vramOffset + vBuf->node->start; vBuf->map = (void *)(p->vramMap + vBuf->node->start); |