Add missing param checks in mem target functions

Author: kswiecicki

src/memory_target.c

@@ -79,6 +79,10 @@ umf_result_t umfMemoryTargetClone(umf_memory_target_handle_t memoryTarget,
 
 umf_result_t umfMemoryTargetGetCapacity(umf_memory_target_handle_t memoryTarget,
                                         size_t *capacity) {
+    if (!memoryTarget || !capacity) {
+        return UMF_RESULT_ERROR_INVALID_ARGUMENT;
+    }
+
     assert(memoryTarget);
     assert(capacity);
     return memoryTarget->ops->get_capacity(memoryTarget->priv, capacity);
@@ -88,6 +92,10 @@ umf_result_t
 umfMemoryTargetGetBandwidth(umf_memory_target_handle_t srcMemoryTarget,
                             umf_memory_target_handle_t dstMemoryTarget,
                             size_t *bandwidth) {
+    if (!srcMemoryTarget || !dstMemoryTarget || !bandwidth) {
+        return UMF_RESULT_ERROR_INVALID_ARGUMENT;
+    }
+
     assert(srcMemoryTarget);
     assert(dstMemoryTarget);
     assert(bandwidth);

src/memory_targets/memory_target_numa.c

@@ -125,6 +125,10 @@ static umf_result_t numa_clone(void *memTarget, void **outMemTarget) {
 }
 
 static umf_result_t numa_get_capacity(void *memTarget, size_t *capacity) {
+    if (!memTarget || !capacity) {
+        return UMF_RESULT_ERROR_INVALID_ARGUMENT;
+    }
+
     hwloc_topology_t topology = umfGetTopology();
     if (!topology) {
         return UMF_RESULT_ERROR_NOT_SUPPORTED;
@@ -147,6 +151,10 @@ static umf_result_t numa_get_capacity(void *memTarget, size_t *capacity) {
 static umf_result_t numa_get_bandwidth(void *srcMemoryTarget,
                                        void *dstMemoryTarget,
                                        size_t *bandwidth) {
+    if (!srcMemoryTarget || !dstMemoryTarget || !bandwidth) {
+        return UMF_RESULT_ERROR_INVALID_ARGUMENT;
+    }
+
     hwloc_topology_t topology = umfGetTopology();
     if (!topology) {
         return UMF_RESULT_ERROR_NOT_SUPPORTED;