attrd_client.c

Go to the documentation of this file.

/*
 * Copyright 2011-2021 the Pacemaker project contributors
 *
 * The version control history for this file may have further details.
 *
 * This source code is licensed under the GNU Lesser General Public License
 * version 2.1 or later (LGPLv2.1+) WITHOUT ANY WARRANTY.
 */

#ifndef _GNU_SOURCE
#  define _GNU_SOURCE
#endif

#include <crm_internal.h>

#include <stdio.h>

#include <crm/crm.h>
#include <crm/msg_xml.h>
#include <crm/common/attrd_internal.h>

static xmlNode *
create_attrd_op(const char *user_name)
{
    xmlNode *attrd_op = create_xml_node(NULL, __func__);

    crm_xml_add(attrd_op, F_TYPE, T_ATTRD);
    crm_xml_add(attrd_op, F_ORIG, (crm_system_name? crm_system_name: "unknown"));
    crm_xml_add(attrd_op, PCMK__XA_ATTR_USER, user_name);

    return attrd_op;
}

static int
send_attrd_op(crm_ipc_t *ipc, xmlNode *attrd_op)
{
    int rc = -ENOTCONN; // initially handled as legacy return code
    int max = 5;

    static gboolean connected = TRUE;
    static crm_ipc_t *local_ipc = NULL;
    static enum crm_ipc_flags flags = crm_ipc_flags_none;

    if (ipc == NULL && local_ipc == NULL) {
        local_ipc = crm_ipc_new(T_ATTRD, 0);
        pcmk__set_ipc_flags(flags, "client", crm_ipc_client_response);
        connected = FALSE;
    }

    if (ipc == NULL) {
        ipc = local_ipc;
    }

    while (max > 0) {
        if (connected == FALSE) {
            crm_info("Connecting to cluster... %d retries remaining", max);
            connected = crm_ipc_connect(ipc);
        }

        if (connected) {
            rc = crm_ipc_send(ipc, attrd_op, flags, 0, NULL);
        } else {
            crm_perror(LOG_INFO, "Connection to cluster attribute manager failed");
        }

        if (ipc != local_ipc) {
            break;

        } else if (rc > 0) {
            break;

        } else if (rc == -EAGAIN || rc == -EALREADY) {
            sleep(5 - max);
            max--;

        } else {
            crm_ipc_close(ipc);
            connected = FALSE;
            sleep(5 - max);
            max--;
        }
    }

    if (rc > 0) {
        rc = pcmk_ok;
    }
    return pcmk_legacy2rc(rc);
}

int
pcmk__node_attr_request(crm_ipc_t *ipc, char command, const char *host,
                        const char *name, const char *value,
                        const char *section, const char *set,
                        const char *dampen, const char *user_name, int options)
{
    int rc = pcmk_rc_ok;
    const char *task = NULL;
    const char *name_as = NULL;
    const char *display_host = (host ? host : "localhost");
    const char *display_command = NULL; /* for commands without name/value */
    xmlNode *update = create_attrd_op(user_name);

    /* remap common aliases */
    if (pcmk__str_eq(section, "reboot", pcmk__str_casei)) {
        section = XML_CIB_TAG_STATUS;

    } else if (pcmk__str_eq(section, "forever", pcmk__str_casei)) {
        section = XML_CIB_TAG_NODES;
    }

    if (name == NULL && command == 'U') {
        command = 'R';
    }

    switch (command) {
        case 'u':
            task = PCMK__ATTRD_CMD_UPDATE;
            name_as = PCMK__XA_ATTR_PATTERN;
            break;
        case 'D':
        case 'U':
        case 'v':
            task = PCMK__ATTRD_CMD_UPDATE;
            name_as = PCMK__XA_ATTR_NAME;
            break;
        case 'R':
            task = PCMK__ATTRD_CMD_REFRESH;
            display_command = "refresh";
            break;
        case 'B':
            task = PCMK__ATTRD_CMD_UPDATE_BOTH;
            name_as = PCMK__XA_ATTR_NAME;
            break;
        case 'Y':
            task = PCMK__ATTRD_CMD_UPDATE_DELAY;
            name_as = PCMK__XA_ATTR_NAME;
            break;
        case 'Q':
            task = PCMK__ATTRD_CMD_QUERY;
            name_as = PCMK__XA_ATTR_NAME;
            break;
        case 'C':
            task = PCMK__ATTRD_CMD_PEER_REMOVE;
            display_command = "purge";
            break;
    }

    if (name_as != NULL) {
        if (name == NULL) {
            rc = EINVAL;
            goto done;
        }
        crm_xml_add(update, name_as, name);
    }

    crm_xml_add(update, PCMK__XA_TASK, task);
    crm_xml_add(update, PCMK__XA_ATTR_VALUE, value);
    crm_xml_add(update, PCMK__XA_ATTR_DAMPENING, dampen);
    crm_xml_add(update, PCMK__XA_ATTR_SECTION, section);
    crm_xml_add(update, PCMK__XA_ATTR_NODE_NAME, host);
    crm_xml_add(update, PCMK__XA_ATTR_SET, set);
    crm_xml_add_int(update, PCMK__XA_ATTR_IS_REMOTE,
                    pcmk_is_set(options, pcmk__node_attr_remote));
    crm_xml_add_int(update, PCMK__XA_ATTR_IS_PRIVATE,
                    pcmk_is_set(options, pcmk__node_attr_private));

    rc = send_attrd_op(ipc, update);

done:
    free_xml(update);

    if (display_command) {
        crm_debug("Asked pacemaker-attrd to %s %s: %s (%d)",
                  display_command, display_host, pcmk_rc_str(rc), rc);
    } else {
        crm_debug("Asked pacemaker-attrd to update %s=%s for %s: %s (%d)",
                  name, value, display_host, pcmk_rc_str(rc), rc);
    }
    return rc;
}

int
pcmk__node_attr_request_clear(crm_ipc_t *ipc, const char *host,
                              const char *resource, const char *operation,
                              const char *interval_spec, const char *user_name,
                              int options)
{
    int rc = pcmk_rc_ok;
    xmlNode *clear_op = create_attrd_op(user_name);
    const char *interval_desc = NULL;
    const char *op_desc = NULL;

    crm_xml_add(clear_op, PCMK__XA_TASK, PCMK__ATTRD_CMD_CLEAR_FAILURE);
    crm_xml_add(clear_op, PCMK__XA_ATTR_NODE_NAME, host);
    crm_xml_add(clear_op, PCMK__XA_ATTR_RESOURCE, resource);
    crm_xml_add(clear_op, PCMK__XA_ATTR_OPERATION, operation);
    crm_xml_add(clear_op, PCMK__XA_ATTR_INTERVAL, interval_spec);
    crm_xml_add_int(clear_op, PCMK__XA_ATTR_IS_REMOTE,
                    pcmk_is_set(options, pcmk__node_attr_remote));

    rc = send_attrd_op(ipc, clear_op);
    free_xml(clear_op);

    if (operation) {
        interval_desc = interval_spec? interval_spec : "nonrecurring";
        op_desc = operation;
    } else {
        interval_desc = "all";
        op_desc = "operations";
    }
    crm_debug("Asked pacemaker-attrd to clear failure of %s %s for %s on %s: %s (%d)",
              interval_desc, op_desc, (resource? resource : "all resources"),
              (host? host : "all nodes"), pcmk_rc_str(rc), rc);
    return rc;
}

#define LRM_TARGET_ENV "OCF_RESKEY_" CRM_META "_" XML_LRM_ATTR_TARGET

const char *
pcmk__node_attr_target(const char *name)
{
    if (name == NULL || pcmk__strcase_any_of(name, "auto", "localhost", NULL)) {
        char *target_var = crm_meta_name(XML_RSC_ATTR_TARGET);
        char *phys_var = crm_meta_name(PCMK__ENV_PHYSICAL_HOST);
        const char *target = getenv(target_var);
        const char *host_physical = getenv(phys_var);

        // It is important to use the name by which the scheduler knows us
        if (host_physical && pcmk__str_eq(target, "host", pcmk__str_casei)) {
            name = host_physical;

        } else {
            const char *host_pcmk = getenv(LRM_TARGET_ENV);

            if (host_pcmk) {
                name = host_pcmk;
            }
        }
        free(target_var);
        free(phys_var);

        // TODO? Call get_local_node_name() if name == NULL
        // (currently would require linkage against libcrmcluster)
        return name;
    } else {
        return NULL;
    }
}

char *
pcmk_promotion_score_name(const char *rsc_id)
{
    if (rsc_id == NULL) {
        rsc_id = getenv("OCF_RESOURCE_INSTANCE");
        if (rsc_id == NULL) {
            return NULL;
        }
    }
    return crm_strdup_printf("master-%s", rsc_id);
}