diff --git a/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf.cpp b/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf.cpp index ed2a649e2d..a48b385f77 100644 --- a/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf.cpp +++ b/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf.cpp @@ -69,6 +69,9 @@ static int recvmsg_fd(int socket) { } struct cmsghdr *controlHeader = CMSG_FIRSTHDR(&msgHeader); + if (CMSG_DATA(controlHeader) == nullptr) { + return -1; + } memmove(&fd, CMSG_DATA(controlHeader), sizeof(int)); return fd; } diff --git a/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf_p2p.cpp b/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf_p2p.cpp index e26ffe4490..d7577c8d71 100644 --- a/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf_p2p.cpp +++ b/level_zero/core/test/black_box_tests/zello_ipc_copy_dma_buf_p2p.cpp @@ -69,6 +69,9 @@ static int recvmsg_fd(int socket) { } struct cmsghdr *controlHeader = CMSG_FIRSTHDR(&msgHeader); + if (CMSG_DATA(controlHeader) == nullptr) { + return -1; + } memmove(&fd, CMSG_DATA(controlHeader), sizeof(int)); return fd; }