summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorZhiqiang Liu <liuzhiqiang26@huawei.com>2021-02-19 15:28:38 +0800
committerZhiqiang Liu <liuzhiqiang26@huawei.com>2021-02-19 15:28:47 +0800
commitaa8c962c174dd4439d9abfb3d7ae0d2d69e96438 (patch)
tree6c496483982c1bd0d7fe548c92dfd79658a47692
parent0a3b79c82d11f9d9a65351fd9cc130da50f1da65 (diff)
downloadliburing-aa8c962c174dd4439d9abfb3d7ae0d2d69e96438.tar.gz
liburing-aa8c962c174dd4439d9abfb3d7ae0d2d69e96438.tar.bz2
setup: check whether malloc succ before using it
In io_uring_get_probe_ring func, we will call malloc() to alloc memory for probe. Considering malloc may return NULL, we should check whether malloc() succ before using it. Signed-off-by: Zhiqiang Liu <liuzhiqiang26@huawei.com>
-rw-r--r--src/setup.c3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/setup.c b/src/setup.c
index 062eaa0..94fec90 100644
--- a/src/setup.c
+++ b/src/setup.c
@@ -185,7 +185,10 @@ struct io_uring_probe *io_uring_get_probe_ring(struct io_uring *ring)
size_t len = sizeof(*probe) + 256 * sizeof(struct io_uring_probe_op);
probe = malloc(len);
+ if (!probe)
+ return NULL;
memset(probe, 0, len);
+
r = io_uring_register_probe(ring, probe, 256);
if (r < 0)
goto fail;