Merge pull request #6083 from devreal/rdma-plug-memleak

Plug two memory leaks in rdma osc

Author: hjelmn

ompi/mca/osc/rdma/osc_rdma_component.c

@@ -586,16 +586,16 @@ static int allocate_state_shared (ompi_osc_rdma_module_t *module, void **base, s
             }
         }
 
-        /* allocate the shared memory segment */
-        ret = opal_asprintf (&data_file, "%s" OPAL_PATH_SEP "osc_rdma.%s.%x.%d",
-                        mca_osc_rdma_component.backing_directory, ompi_process_info.nodename,
-                        OMPI_PROC_MY_NAME->jobid, ompi_comm_get_cid(module->comm));
-        if (0 > ret) {
-            ret = OMPI_ERR_OUT_OF_RESOURCE;
-            break;
-        }
-
         if (0 == local_rank) {
+            /* allocate the shared memory segment */
+            ret = opal_asprintf (&data_file, "%s" OPAL_PATH_SEP "osc_rdma.%s.%x.%d",
+                            mca_osc_rdma_component.backing_directory, ompi_process_info.nodename,
+                            OMPI_PROC_MY_NAME->jobid, ompi_comm_get_cid(module->comm));
+            if (0 > ret) {
+                ret = OMPI_ERR_OUT_OF_RESOURCE;
+                break;
+            }
+
             /* allocate enough space for the state + data for all local ranks */
             ret = opal_shmem_segment_create (&module->seg_ds, data_file, total_size);
             free (data_file);

ompi/mca/osc/rdma/osc_rdma_module.c

@@ -111,7 +111,7 @@ int ompi_osc_rdma_free(ompi_win_t *win)
 
         OBJ_DESTRUCT(&module->peer_hash);
     } else if (NULL != module->comm) {
-        for (int i = 0 ; i < ompi_comm_rank (module->comm) ; ++i) {
+        for (int i = 0 ; i < ompi_comm_size (module->comm) ; ++i) {
             if (NULL != module->peer_array[i]) {
                 OBJ_RELEASE(module->peer_array[i]);
             }