/*

 * Copyright 2004-2019 the Pacemaker project contributors

 *

 * The version control history for this file may have further details.

 *

 * This source code is licensed under the GNU General Public License version 2

 * or later (GPLv2+) WITHOUT ANY WARRANTY.

 */

#include <crm_internal.h>

#include <unistd.h>  /* sleep */

#include <crm/common/alerts_internal.h>

#include <crm/common/xml.h>

#include <crm/crm.h>

#include <crm/msg_xml.h>

#include <pacemaker-controld.h>

int cib_retries = 0;

static void

do_cib_updated(const char *event, xmlNode * msg)

{

    if (crm_patchset_contains_alert(msg, TRUE)) {

        mainloop_set_trigger(config_read);

    }

}

static void

do_cib_replaced(const char *event, xmlNode * msg)

{

    crm_debug("Updating the CIB after a replace: DC=%s", AM_I_DC ? "true" : "false");

    if (AM_I_DC == FALSE) {

        return;

    } else if (fsa_state == S_FINALIZE_JOIN && is_set(fsa_input_register, R_CIB_ASKED)) {

        /* no need to restart the join - we asked for this replace op */

        return;

    }

    /* start the join process again so we get everyone's LRM status */

    populate_cib_nodes(node_update_quick|node_update_all, __FUNCTION__);

    register_fsa_input(C_FSA_INTERNAL, I_ELECTION, NULL);

}

/* A_CIB_STOP, A_CIB_START, O_CIB_RESTART */

void

do_cib_control(long long action,

               enum crmd_fsa_cause cause,

               enum crmd_fsa_state cur_state,

               enum crmd_fsa_input current_input, fsa_data_t * msg_data)

{

    CRM_ASSERT(fsa_cib_conn != NULL);

    if (action & A_CIB_STOP) {

        if (fsa_cib_conn->state != cib_disconnected && last_resource_update != 0) {

            crm_info("Waiting for resource update %d to complete", last_resource_update);

            crmd_fsa_stall(FALSE);

            return;

        }

        crm_info("Disconnecting from the CIB manager");

        clear_bit(fsa_input_register, R_CIB_CONNECTED);

        fsa_cib_conn->cmds->del_notify_callback(fsa_cib_conn, T_CIB_DIFF_NOTIFY, do_cib_updated);

        if (fsa_cib_conn->state != cib_disconnected) {

            fsa_cib_conn->cmds->set_slave(fsa_cib_conn, cib_scope_local);

            fsa_cib_conn->cmds->signoff(fsa_cib_conn);

        }

        crm_notice("Disconnected from the CIB manager");

    }

    if (action & A_CIB_START) {

        int rc = pcmk_ok;

        if (cur_state == S_STOPPING) {

            crm_err("Ignoring request to connect to the CIB manager after shutdown");

            return;

        }

        rc = fsa_cib_conn->cmds->signon(fsa_cib_conn, CRM_SYSTEM_CRMD, cib_command_nonblocking);

        if (rc != pcmk_ok) {

            /* a short wait that usually avoids stalling the FSA */

            sleep(1);

            rc = fsa_cib_conn->cmds->signon(fsa_cib_conn, CRM_SYSTEM_CRMD, cib_command_nonblocking);

        }

        if (rc != pcmk_ok) {

            crm_info("Could not connect to the CIB manager: %s", pcmk_strerror(rc));

        } else if (pcmk_ok !=

                   fsa_cib_conn->cmds->set_connection_dnotify(fsa_cib_conn,

                                                              crmd_cib_connection_destroy)) {

            crm_err("Could not set dnotify callback");

        } else if (pcmk_ok !=

                   fsa_cib_conn->cmds->add_notify_callback(fsa_cib_conn, T_CIB_REPLACE_NOTIFY,

                                                           do_cib_replaced)) {

            crm_err("Could not set CIB notification callback (replace)");

        } else if (pcmk_ok !=

                   fsa_cib_conn->cmds->add_notify_callback(fsa_cib_conn, T_CIB_DIFF_NOTIFY,

                                                           do_cib_updated)) {

            crm_err("Could not set CIB notification callback (update)");

        } else {

            set_bit(fsa_input_register, R_CIB_CONNECTED);

            cib_retries = 0;

        }

        if (is_not_set(fsa_input_register, R_CIB_CONNECTED)) {

            cib_retries++;

            crm_warn("Couldn't complete CIB registration %d"

                     " times... pause and retry", cib_retries);

            if (cib_retries < 30) {

                controld_start_timer(wait_timer);

                crmd_fsa_stall(FALSE);

            } else {

                crm_err("Could not complete CIB"

                        " registration  %d times..." " hard error", cib_retries);

                register_fsa_error(C_FSA_INTERNAL, I_ERROR, NULL);

            }

        }

    }

}

/*!

 * \internal

 * \brief Get CIB call options to use local scope if master unavailable

 *

 * \return CIB call options

 */

int crmd_cib_smart_opt()

{

    int call_opt = cib_quorum_override;

    if (fsa_state == S_ELECTION || fsa_state == S_PENDING) {

        crm_info("Sending update to local CIB in state: %s", fsa_state2string(fsa_state));

        call_opt |= cib_scope_local;

    }

    return call_opt;

}

/*!

 * \internal

 * \brief Check whether an action type should be recorded in the CIB

 *

 * \param[in] action  Action type

 *

 * \return TRUE if action should be recorded, FALSE otherwise

 */

bool

controld_action_is_recordable(const char *action)

{

    if (safe_str_eq(action, CRMD_ACTION_CANCEL)

        || safe_str_eq(action, CRMD_ACTION_DELETE)

        || safe_str_eq(action, CRMD_ACTION_NOTIFY)

        || safe_str_eq(action, CRMD_ACTION_METADATA)) {

        return FALSE;

    }

    return TRUE;

}