rdma_sync_event_responder_sample.c

Go to the documentation of this file.

/*
 * Copyright (c) 2023 NVIDIA CORPORATION AND AFFILIATES.  All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification, are permitted
 * provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright notice, this list of
 *       conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright notice, this list of
 *       conditions and the following disclaimer in the documentation and/or other materials
 *       provided with the distribution.
 *     * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used
 *       to endorse or promote products derived from this software without specific prior written
 *       permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
 * STRICT LIABILITY, OR TOR (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */
 
#include <doca_error.h>
#include <doca_log.h>
#include <doca_buf_inventory.h>
#include <doca_buf.h>
#include <doca_ctx.h>
 
#include "rdma_common.h"
 
DOCA_LOG_REGISTER(RDMA_READ_RESPONDER::SAMPLE);
 
#define EXAMPLE_SET_VALUE (0xD0CA) /* Example value to use for setting sync event */
 
/*
 * Destroy DOCA sync event resources
 *
 * @resources [in]: DOCA RDMA resources, including the sync_event to destroy
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
doca_error_t rdma_sync_event_responder_destroy_sync_event_resources(struct rdma_resources *resources)
{
    doca_error_t tmp_result, result = DOCA_SUCCESS;
 
    resources->sync_event_descriptor = NULL;
 
    /* Stop and destroy sync event if exists */
    if (resources->sync_event != NULL) {
        tmp_result = doca_sync_event_stop(resources->sync_event);
        if (tmp_result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("Failed to stop DOCA sync event: %s", doca_error_get_descr(tmp_result));
            DOCA_ERROR_PROPAGATE(result, tmp_result);
        }
 
        tmp_result = doca_sync_event_destroy(resources->sync_event);
        if (tmp_result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("Failed to destroy DOCA sync event: %s", doca_error_get_descr(tmp_result));
            DOCA_ERROR_PROPAGATE(result, tmp_result);
        }
 
        resources->sync_event = NULL;
    }
 
    return result;
}
 
/*
 * Allocate DOCA sync event resources
 *
 * @resources [in/out]: DOCA RDMA resources, including the sync_event to allocate
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
doca_error_t rdma_sync_event_responder_allocate_sync_event_resources(struct rdma_resources *resources)
{
    doca_error_t result;
 
    result = doca_sync_event_create(&resources->sync_event);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to create DOCA sync event: %s", doca_error_get_descr(result));
        goto destroy_se;
    }
 
    result = doca_sync_event_add_publisher_location_remote_net(resources->sync_event);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to set publisher for DOCA sync event: %s", doca_error_get_descr(result));
        goto destroy_se;
    }
 
    result = doca_sync_event_add_subscriber_location_cpu(resources->sync_event, resources->doca_device);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to set subscriber for DOCA sync event: %s", doca_error_get_descr(result));
        goto destroy_se;
    }
 
    result = doca_sync_event_start(resources->sync_event);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to start DOCA sync event: %s", doca_error_get_descr(result));
        goto destroy_se;
    }
 
    return DOCA_SUCCESS;
 
destroy_se:
    (void)rdma_sync_event_responder_destroy_sync_event_resources(resources);
 
    return result;
}
 
/*
 * Write the connection details and the mmap details for the requester to read,
 * and read the connection details of the requester
 * In DC transport mode it is only needed to read the remote connection details
 *
 * @cfg [in]: Configuration parameters
 * @resources [in/out]: DOCA RDMA resources
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
static doca_error_t write_read_connection(struct rdma_config *cfg, struct rdma_resources *resources)
{
    doca_error_t result = DOCA_SUCCESS;
 
    /* Write the RDMA connection details */
    result = write_file(cfg->local_connection_desc_path,
                (char *)resources->rdma_conn_descriptor,
                resources->rdma_conn_descriptor_size);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to write the RDMA connection details: %s", doca_error_get_descr(result));
        return result;
    }
 
    /* Write the RDMA connection details */
    result = write_file(cfg->remote_resource_desc_path,
                (char *)resources->sync_event_descriptor,
                resources->sync_event_descriptor_size);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to write sync event export blob: %s", doca_error_get_descr(result));
        return result;
    }
 
    DOCA_LOG_INFO("You can now copy %s and %s to the requester",
              cfg->local_connection_desc_path,
              cfg->remote_resource_desc_path);
 
    if (cfg->transport_type == DOCA_RDMA_TRANSPORT_TYPE_DC) {
        return result;
    }
    DOCA_LOG_INFO("Please copy %s from the requester and then press enter", cfg->remote_connection_desc_path);
 
    /* Wait for enter */
    wait_for_enter();
 
    /* Read the remote RDMA connection details */
    result = read_file(cfg->remote_connection_desc_path,
               (char **)&resources->remote_rdma_conn_descriptor,
               &resources->remote_rdma_conn_descriptor_size);
    if (result != DOCA_SUCCESS)
        DOCA_LOG_ERR("Failed to read the remote RDMA connection details: %s", doca_error_get_descr(result));
 
    return result;
}
 
/*
 * Export and receive connection and sync event details, and connect to the remote RDMA
 *
 * @resources [in]: RDMA resources
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
static doca_error_t rdma_sync_event_responder_export_and_connect(struct rdma_resources *resources)
{
    doca_error_t result;
 
    if (resources->cfg->use_rdma_cm == true)
        return rdma_cm_connect(resources);
 
    /* Export DOCA RDMA */
    result = doca_rdma_export(resources->rdma,
                  &(resources->rdma_conn_descriptor),
                  &(resources->rdma_conn_descriptor_size),
                  &(resources->connections[0]));
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to export DOCA RDMA: %s", doca_error_get_descr(result));
        return result;
    }
 
    /* Export RDMA sync event */
    result = doca_sync_event_export_to_remote_net(resources->sync_event,
                              (const uint8_t **)&(resources->sync_event_descriptor),
                              &(resources->sync_event_descriptor_size));
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to export DOCA sync event for RDMA: %s", doca_error_get_descr(result));
        return result;
    }
 
    /* write and read connection details from the requester */
    result = write_read_connection(resources->cfg, resources);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to write and read connection details from the requester: %s",
                 doca_error_get_descr(result));
        return result;
    }
 
    if (resources->cfg->transport_type == DOCA_RDMA_TRANSPORT_TYPE_DC) {
        return result;
    }
    /* Connect RDMA */
    result = doca_rdma_connect(resources->rdma,
                   resources->remote_rdma_conn_descriptor,
                   resources->remote_rdma_conn_descriptor_size,
                   resources->connections[0]);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to connect the responder's DOCA RDMA to the requester's DOCA RDMA: %s",
                 doca_error_get_descr(result));
        return result;
    }
 
    return DOCA_SUCCESS;
}
 
/*
 * Handle the event - wait for a signal, update the sync even and wait for completion.
 *
 * @sync_event [in]: DOCA sync event
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
static doca_error_t rdma_sync_event_responder_handle_event(struct doca_sync_event *sync_event)
{
    doca_error_t result;
 
    DOCA_LOG_INFO("Waiting for sync event to be signaled from remote");
    result = doca_sync_event_wait_gt(sync_event, EXAMPLE_SET_VALUE - 1, UINT64_MAX);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Waiting for sync event to be signaled from remote failed");
        return result;
    }
 
    DOCA_LOG_INFO("Signaling sync event");
    result = doca_sync_event_update_set(sync_event, EXAMPLE_SET_VALUE + 1);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Signaling sync event failed");
        return result;
    }
 
    DOCA_LOG_INFO("Waiting for sync event to be notified for completion");
    result = doca_sync_event_wait_gt(sync_event, UINT64_MAX - 1, UINT64_MAX);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Waiting for sync event to be notified for completion failed");
        return result;
    }
 
    return DOCA_SUCCESS;
}
 
/*
 * Responder wait for requester to finish
 *
 * @resources [in]: RDMA resources
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
static doca_error_t responder_wait_for_requester_finish(struct rdma_resources *resources)
{
    doca_error_t result = DOCA_SUCCESS, tmp_result;
 
    tmp_result = rdma_sync_event_responder_handle_event(resources->sync_event);
    if (tmp_result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Rdma_sync_event_responder_handle_event() failed: %s", doca_error_get_descr(tmp_result));
        DOCA_ERROR_PROPAGATE(result, tmp_result);
    }
 
    if (resources->cfg->use_rdma_cm == true) {
        tmp_result = rdma_cm_disconnect(resources);
        if (tmp_result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("Failed to disconnect RDMA connection: %s", doca_error_get_descr(tmp_result));
            DOCA_ERROR_PROPAGATE(result, tmp_result);
        }
    }
 
    /* Stop the context */
    tmp_result = doca_ctx_stop(resources->rdma_ctx);
    if ((tmp_result != DOCA_SUCCESS) && (tmp_result != DOCA_ERROR_IN_PROGRESS)) {
        DOCA_LOG_ERR("Doca_ctx_stop() failed: %s", doca_error_get_descr(tmp_result));
        DOCA_ERROR_PROPAGATE(result, tmp_result);
    }
 
    return result;
}
 
/*
 * RDMA sync event responder state change callback
 * This function represents the state machine for this RDMA program
 *
 * @user_data [in]: doca_data from the context
 * @ctx [in]: DOCA context
 * @prev_state [in]: Previous DOCA context state
 * @next_state [in]: Next DOCA context state
 */
static void rdma_sync_event_responder_state_change_callback(const union doca_data user_data,
                                struct doca_ctx *ctx,
                                enum doca_ctx_states prev_state,
                                enum doca_ctx_states next_state)
{
    struct rdma_resources *resources = (struct rdma_resources *)user_data.ptr;
    doca_error_t result = DOCA_SUCCESS;
    (void)prev_state;
    (void)ctx;
 
    switch (next_state) {
    case DOCA_CTX_STATE_STARTING:
        DOCA_LOG_INFO("RDMA context entered starting state");
        break;
    case DOCA_CTX_STATE_RUNNING:
        DOCA_LOG_INFO("RDMA context is running");
 
        result = rdma_sync_event_responder_export_and_connect(resources);
        if (result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("rdma_sync_event_responder_export_and_connect() failed: %s",
                     doca_error_get_descr(result));
            break;
        } else
            DOCA_LOG_INFO("RDMA context finished initialization");
 
        if (resources->cfg->use_rdma_cm == true)
            break;
 
        result = responder_wait_for_requester_finish(resources);
        if (result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("responder_wait_for_requester_finish() failed: %s", doca_error_get_descr(result));
        }
        break;
    case DOCA_CTX_STATE_STOPPING:
        DOCA_LOG_INFO("RDMA context entered into stopping state. Any inflight tasks will be flushed");
        break;
    case DOCA_CTX_STATE_IDLE:
        DOCA_LOG_INFO("RDMA context has been stopped");
 
        /* We can stop progressing the PE */
        resources->run_pe_progress = false;
        break;
    default:
        break;
    }
 
    /* If something failed - update that an error was encountered and stop the ctx */
    if (result != DOCA_SUCCESS) {
        DOCA_ERROR_PROPAGATE(resources->first_encountered_error, result);
        (void)doca_ctx_stop(ctx);
    }
}
 
/*
 * Responder side
 *
 * @cfg [in]: Configuration parameters
 * @return: DOCA_SUCCESS on success and DOCA_ERROR otherwise
 */
doca_error_t rdma_sync_event_responder(struct rdma_config *cfg)
{
    struct rdma_resources resources = {0};
    union doca_data ctx_user_data = {0};
    const uint32_t mmap_permissions = DOCA_ACCESS_FLAG_LOCAL_READ_WRITE | DOCA_ACCESS_FLAG_RDMA_READ |
                      DOCA_ACCESS_FLAG_RDMA_WRITE;
    const uint32_t rdma_permissions = DOCA_ACCESS_FLAG_RDMA_READ | DOCA_ACCESS_FLAG_RDMA_WRITE;
    doca_error_t result, tmp_result;
    struct timespec ts = {
        .tv_sec = 0,
        .tv_nsec = SLEEP_IN_NANOS,
    };
 
    /* Allocating resources */
    result = allocate_rdma_resources(cfg, mmap_permissions, rdma_permissions, NULL, &resources);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to allocate RDMA resources: %s", doca_error_get_descr(result));
        return result;
    }
 
    result = rdma_sync_event_responder_allocate_sync_event_resources(&resources);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to allocate sync event resources: %s", doca_error_get_descr(result));
        goto destroy_rdma_resources;
    }
 
    result = doca_ctx_set_state_changed_cb(resources.rdma_ctx, rdma_sync_event_responder_state_change_callback);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Unable to set state change callback for RDMA context: %s", doca_error_get_descr(result));
        goto destroy_sync_event_resources;
    }
 
    /* Include the program's resources in user data of context to be used in callbacks */
    ctx_user_data.ptr = &(resources);
    result = doca_ctx_set_user_data(resources.rdma_ctx, ctx_user_data);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to set context user data: %s", doca_error_get_descr(result));
        goto destroy_sync_event_resources;
    }
 
    if (cfg->use_rdma_cm == true) {
        resources.recv_sync_event_desc = true;
        resources.is_requester = false;
        resources.require_remote_mmap = true;
        resources.task_fn = responder_wait_for_requester_finish;
        result = config_rdma_cm_callback_and_negotiation_task(&resources,
                                      /* need_send_mmap_info */ true,
                                      /* need_recv_mmap_info */ false);
        if (result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("Failed to config RDMA CM callbacks and negotiation functions: %s",
                     doca_error_get_descr(result));
            goto destroy_sync_event_resources;
        }
    }
 
    /* Start RDMA context */
    result = doca_ctx_start(resources.rdma_ctx);
    if (result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to start RDMA context: %s", doca_error_get_descr(result));
        goto destroy_sync_event_resources;
    }
 
    /*
     * Run the progress engine which will run the state machine defined in
     * rdma_sync_event_responder_state_change_callback().
     * When the context moves to idle, the context change callback call will signal to stop running the progress
     * engine.
     */
    while (resources.run_pe_progress) {
        if (doca_pe_progress(resources.pe) == 0)
            nanosleep(&ts, &ts);
    }
 
    /* Assign the result we update in the callback */
    result = resources.first_encountered_error;
 
    tmp_result = request_stop_ctx(resources.pe, resources.rdma_ctx);
    if (tmp_result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to stop DOCA RDMA context: %s", doca_error_get_descr(tmp_result));
        DOCA_ERROR_PROPAGATE(result, tmp_result);
    }
 
destroy_sync_event_resources:
    tmp_result = rdma_sync_event_responder_destroy_sync_event_resources(&resources);
    if (tmp_result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to destroy DOCA Sync Event resources: %s", doca_error_get_descr(tmp_result));
        DOCA_ERROR_PROPAGATE(result, tmp_result);
    }
destroy_rdma_resources:
    if (resources.buf_inventory != NULL) {
        tmp_result = doca_buf_inventory_stop(resources.buf_inventory);
        if (tmp_result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("Failed to stop DOCA buffer inventory: %s", doca_error_get_descr(tmp_result));
            DOCA_ERROR_PROPAGATE(result, tmp_result);
        }
        tmp_result = doca_buf_inventory_destroy(resources.buf_inventory);
        if (tmp_result != DOCA_SUCCESS) {
            DOCA_LOG_ERR("Failed to destroy DOCA buffer inventory: %s", doca_error_get_descr(tmp_result));
            DOCA_ERROR_PROPAGATE(result, tmp_result);
        }
    }
    tmp_result = destroy_rdma_resources(&resources, cfg);
    if (tmp_result != DOCA_SUCCESS) {
        DOCA_LOG_ERR("Failed to destroy DOCA RDMA resources: %s", doca_error_get_descr(tmp_result));
        DOCA_ERROR_PROPAGATE(result, tmp_result);
    }
    return result;
}