diff options
author | Zhiqiang Liu <liuzhiqiang26@huawei.com> | 2021-02-19 15:28:38 +0800 |
---|---|---|
committer | Zhiqiang Liu <liuzhiqiang26@huawei.com> | 2021-02-19 15:28:47 +0800 |
commit | aa8c962c174dd4439d9abfb3d7ae0d2d69e96438 (patch) | |
tree | 6c496483982c1bd0d7fe548c92dfd79658a47692 | |
parent | 0a3b79c82d11f9d9a65351fd9cc130da50f1da65 (diff) | |
download | liburing-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.c | 3 |
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; |