mtl/ofi: Check ompi_ret in ompi_mtl_ofi_send_generic

Luke Robison · Luke Robison · commit eb0ca9756b46 · 2024-01-09T16:15:27.000Z
Correct Coverity issue CID 1515755.

Signed-off-by: Luke Robison &lt;lrbison@amazon.com&gt;
diff --git a/ompi/mca/mtl/ofi/mtl_ofi.h b/ompi/mca/mtl/ofi/mtl_ofi.h
@@ -865,11 +865,17 @@ ompi_mtl_ofi_send_generic(struct mca_mtl_base_module_t *mtl,
         if (mtl_comm->c_index_vec[dest].c_index_state == MCA_MTL_OFI_CID_NOT_EXCHANGED) {
             mtl_comm->c_index_vec[dest].c_index_state = MCA_MTL_OFI_CID_EXCHANGING;
             ompi_ret = ompi_mtl_ofi_send_excid(mtl, comm, dest, ofi_cq_data, true);
+            if (OPAL_UNLIKELY(OMPI_SUCCESS != ompi_ret)) {
+                return ompi_ret;
+            }
         }
 
         if (mtl_comm->c_index_vec[dest].c_index_state > MCA_MTL_OFI_CID_EXCHANGED) {
              while (mtl_comm->c_index_vec[dest].c_index_state > MCA_MTL_OFI_CID_EXCHANGED) {
                 ompi_ret = ompi_mtl_ofi_post_recv_excid_buffer(true, comm, dest);
+                if (OPAL_UNLIKELY(OMPI_SUCCESS != ompi_ret)) {
+                    return ompi_ret;
+                }
             }
         }
         c_index_for_tag = mtl_comm->c_index_vec[dest].c_index;

Original file line number	Diff line number	Diff line change
`@@ -865,11 +865,17 @@ ompi_mtl_ofi_send_generic(struct mca_mtl_base_module_t *mtl,`
`865`	`865`	`if (mtl_comm->c_index_vec[dest].c_index_state == MCA_MTL_OFI_CID_NOT_EXCHANGED) {`
`866`	`866`	`mtl_comm->c_index_vec[dest].c_index_state = MCA_MTL_OFI_CID_EXCHANGING;`
`867`	`867`	`ompi_ret = ompi_mtl_ofi_send_excid(mtl, comm, dest, ofi_cq_data, true);`
	`868`	`+ if (OPAL_UNLIKELY(OMPI_SUCCESS != ompi_ret)) {`
	`869`	`+ return ompi_ret;`
	`870`	`+ }`
`868`	`871`	`}`
`869`	`872`
`870`	`873`	`if (mtl_comm->c_index_vec[dest].c_index_state > MCA_MTL_OFI_CID_EXCHANGED) {`
`871`	`874`	`while (mtl_comm->c_index_vec[dest].c_index_state > MCA_MTL_OFI_CID_EXCHANGED) {`
`872`	`875`	`ompi_ret = ompi_mtl_ofi_post_recv_excid_buffer(true, comm, dest);`
	`876`	`+ if (OPAL_UNLIKELY(OMPI_SUCCESS != ompi_ret)) {`
	`877`	`+ return ompi_ret;`
	`878`	`+ }`
`873`	`879`	`}`
`874`	`880`	`}`
`875`	`881`	`c_index_for_tag = mtl_comm->c_index_vec[dest].c_index;`