spiel_ut_common.c

Go to the documentation of this file.

/* -*- C -*- */
/*
 * Copyright (c) 2013-2020 Seagate Technology LLC and/or its Affiliates
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * For any questions about this software or licensing,
 * please email opensource@seagate.com or cortx-questions@seagate.com.
 *
 */


#define M0_TRACE_SUBSYSTEM M0_TRACE_SUBSYS_SPIEL
#include "lib/trace.h"

#include "spiel/ut/spiel_ut_common.h"
#include "spiel/spiel.h"
#include "ut/misc.h" /* M0_UT_CONF_PROFILE */
#include "ut/ut.h"

struct  m0_spiel_ut_reqh ut_reqh;
const char    *confd_addr[] = { SERVER_ENDPOINT_ADDR, NULL };
const char    *rm_addr      = SERVER_ENDPOINT_ADDR;

static int m0_spiel__ut_rms_start(struct m0_reqh *reqh)
{
        struct m0_reqh_service *service;
        struct m0_fid           fid;

        m0_fid_tgenerate(&fid, M0_CONF_SERVICE_TYPE.cot_ftype.ft_id);
        return m0_reqh_service_setup(&service, &m0_rms_type, reqh, NULL, &fid);
}

M0_INTERNAL int m0_spiel__ut_reqh_init(struct m0_spiel_ut_reqh *spl_reqh,
                                       const char              *ep_addr)
{
        enum { NR_TMS = 1 };
        int rc;

        M0_SET0(spl_reqh);
        rc = m0_net_domain_init(&spl_reqh->sur_net_dom,
                                 m0_net_xprt_default_get());
        if (rc != 0)
                return rc;

        rc = m0_rpc_net_buffer_pool_setup(&spl_reqh->sur_net_dom,
                                          &spl_reqh->sur_buf_pool,
                                          m0_rpc_bufs_nr(
                                             M0_NET_TM_RECV_QUEUE_DEF_LEN,
                                             NR_TMS),
                                          NR_TMS);
        if (rc != 0)
                goto net;

        rc = M0_REQH_INIT(&spl_reqh->sur_reqh,
                          .rhia_dtm     = (void *)1,
                          .rhia_mdstore = (void *)1,
                          .rhia_fid     = &g_process_fid,
        );
        if (rc != 0)
                goto buf_pool;

        rc = m0_rpc_machine_init(&spl_reqh->sur_rmachine,
                                 &spl_reqh->sur_net_dom, ep_addr,
                                 &spl_reqh->sur_reqh, &spl_reqh->sur_buf_pool,
                                 M0_BUFFER_ANY_COLOUR,
                                 M0_RPC_DEF_MAX_RPC_MSG_SIZE,
                                 M0_NET_TM_RECV_QUEUE_DEF_LEN);
        if (rc != 0)
                goto reqh;
        m0_reqh_start(&spl_reqh->sur_reqh);
        rc = m0_spiel__ut_rms_start(&spl_reqh->sur_reqh);
        if (rc != 0)
                goto rm;
        return 0;
rm:
        m0_reqh_services_terminate(&spl_reqh->sur_reqh);
reqh:
        m0_reqh_fini(&spl_reqh->sur_reqh);
buf_pool:
        m0_rpc_net_buffer_pool_cleanup(&spl_reqh->sur_buf_pool);
net:
        m0_net_domain_fini(&spl_reqh->sur_net_dom);
        return rc;
}

M0_INTERNAL void m0_spiel__ut_reqh_fini(struct m0_spiel_ut_reqh *spl_reqh)
{
        m0_reqh_services_terminate(&spl_reqh->sur_reqh);
        m0_rpc_machine_fini(&spl_reqh->sur_rmachine);
        m0_reqh_fini(&spl_reqh->sur_reqh);
        m0_rpc_net_buffer_pool_cleanup(&spl_reqh->sur_buf_pool);
        m0_net_domain_fini(&spl_reqh->sur_net_dom);
}

M0_INTERNAL int m0_spiel__ut_rpc_server_start(struct m0_rpc_server_ctx *rpc_srv,
                                              const char          *ha_ep,
                                              const char          *confdb_path)
{
        enum {
                LOG_NAME_MAX_LEN     = 128,
                EP_MAX_LEN           = 24,
                RPC_SIZE_MAX_LEN     = 32,
        };

        char                log_name[LOG_NAME_MAX_LEN];
        char                full_ep[EP_MAX_LEN];
        char                max_rpc_size[RPC_SIZE_MAX_LEN];

        snprintf(full_ep, EP_MAX_LEN, "%s:%s", M0_NET_XPRT_PREFIX_DEFAULT,
                 ha_ep);
        snprintf(max_rpc_size, RPC_SIZE_MAX_LEN,
                 "%d", M0_RPC_DEF_MAX_RPC_MSG_SIZE);

#define NAME(ext) "ut_spiel" ext
        char *argv[] = {
                NAME(""), "-T", "AD", "-D", NAME(".db"), "-S", NAME(".stob"),
                "-A", "linuxstob:"NAME("-addb_stob"), "-w", "10",
                "-e", full_ep, "-H", (char *)ha_ep, "-f", M0_UT_CONF_PROCESS,
                "-m", max_rpc_size, "-G", full_ep,
                "-c", (char *)confdb_path
        };
#undef NAME

        M0_SET0(rpc_srv);

        rpc_srv->rsx_xprts    = m0_net_all_xprt_get();
        rpc_srv->rsx_xprts_nr = m0_net_xprt_nr();
        rpc_srv->rsx_argv     = argv;
        /* if !run_io_srv then remove last row of argv array */
        rpc_srv->rsx_argc     = ARRAY_SIZE(argv);
        snprintf(log_name, LOG_NAME_MAX_LEN, "confd_%s.log", ha_ep);
        rpc_srv->rsx_log_file_name = log_name;

        return m0_rpc_server_start(rpc_srv);
}

M0_INTERNAL void m0_spiel__ut_rpc_server_stop(struct m0_rpc_server_ctx *rpc_srv)
{
        m0_rpc_server_stop(rpc_srv);
}

M0_INTERNAL void m0_spiel__ut_init(struct m0_spiel *spiel,
                                   const char      *confd_path,
                                   bool             cmd_iface)
{
        int         rc;
        const char *client_ep = CLIENT_ENDPOINT_ADDR;
        const char *profile = M0_UT_CONF_PROFILE;

        rc = m0_spiel__ut_reqh_init(&ut_reqh, client_ep);
        M0_UT_ASSERT(rc == 0);

        rc = m0_spiel__ut_rpc_server_start(&ut_reqh.sur_confd_srv,
                                           confd_addr[0],
                                           confd_path);
        M0_UT_ASSERT(rc == 0);

        rc = m0_spiel_init(spiel, &ut_reqh.sur_reqh);
        M0_UT_ASSERT(rc == 0);
        rc = m0_spiel_cmd_profile_set(spiel, profile);
        M0_UT_ASSERT(rc == 0);
        if (cmd_iface) {
                rc = m0_spiel_rconfc_start(spiel, NULL);
                M0_UT_ASSERT(rc == 0);
        }
}

M0_INTERNAL void m0_spiel__ut_fini(struct m0_spiel *spiel, bool cmd_iface)
{
        if (cmd_iface)
                m0_spiel_rconfc_stop(spiel);
        m0_spiel_fini(spiel);
        m0_spiel__ut_rpc_server_stop(&ut_reqh.sur_confd_srv);
        m0_spiel__ut_reqh_fini(&ut_reqh);
}

#undef M0_TRACE_SUBSYSTEM

/*
 *  Local variables:
 *  c-indentation-style: "K&R"
 *  c-basic-offset: 8
 *  tab-width: 8
 *  fill-column: 80
 *  scroll-step: 1
 *  End:
 */