levelzero: add missing calloc return value check

tkatila · tkatila · commit 80a5529ad3e2 · 2024-09-20T13:59:00.000+03:00
Signed-off-by: Tuomas Katila &lt;tuomas.katila@intel.com&gt;
diff --git a/cmd/gpu_levelzero/zes.c b/cmd/gpu_levelzero/zes.c
@@ -113,6 +113,9 @@ static ze_result_t enumerate_zes_devices(void)
     }
 
     zes_handles = calloc(count, sizeof(zes_device_handle_t));
+    if (zes_handles == NULL) {
+        return ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY;
+    }
 
     res = zesDeviceGet(handle, &count, zes_handles);
     if (res != ZE_RESULT_SUCCESS) {

Original file line number	Diff line number	Diff line change
`@@ -113,6 +113,9 @@ static ze_result_t enumerate_zes_devices(void)`
`113`	`113`	`}`
`114`	`114`
`115`	`115`	`zes_handles = calloc(count, sizeof(zes_device_handle_t));`
	`116`	`+ if (zes_handles == NULL) {`
	`117`	`+ return ZE_RESULT_ERROR_OUT_OF_HOST_MEMORY;`
	`118`	`+ }`
`116`	`119`
`117`	`120`	`res = zesDeviceGet(handle, &count, zes_handles);`
`118`	`121`	`if (res != ZE_RESULT_SUCCESS) {`