Page MenuHomeClusterLabs Projects
Files F3686655

No OneTemporary
Actions

View Options

diff --git a/tools/crm_attribute.c b/tools/crm_attribute.c
index 8f90a3a54a..6a0c3ae195 100644
--- a/tools/crm_attribute.c
+++ b/tools/crm_attribute.c
@@ -1,575 +1,576 @@
/*
* Copyright (C) 2004 Andrew Beekhof <andrew@beekhof.net>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This software is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* General Public License for more details.
*
* You should have received a copy of the GNU General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include <crm_internal.h>
#include <sys/param.h>
#include <crm/crm.h>
#include <stdio.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdlib.h>
#include <errno.h>
#include <fcntl.h>
#include <libgen.h>
#include <time.h>
#include <clplumbing/uids.h>
#include <clplumbing/Gmain_timeout.h>
#include <crm/msg_xml.h>
#include <crm/common/xml.h>
#include <crm/common/ctrl.h>
#include <crm/common/ipc.h>
#include <crm/cib.h>
#include <sys/utsname.h>
#include <attrd.h>
#ifdef HAVE_GETOPT_H
# include <getopt.h>
#endif
void usage(const char *cmd, int exit_status);
gboolean BE_QUIET = FALSE;
gboolean DO_WRITE = TRUE;
gboolean DO_DELETE = FALSE;
char *dest_uname = NULL;
char *dest_node = NULL;
char *set_name = NULL;
char *attr_id = NULL;
char *attr_name = NULL;
const char *type = NULL;
const char *rsc_id = NULL;
const char *attr_value = NULL;
#define OPTARGS "V?GDQN:U:u:s:n:v:l:t:i:!r:"
static gboolean
set_via_attrd(const char *set, const char *id,
const char *name, const char *value)
{
gboolean rc = FALSE;
xmlNode *update = NULL;
+ const char *reason = "connection failed";
IPC_Channel *attrd = init_client_ipc_comms_nodispatch(T_ATTRD);
+
+ fprintf(stderr, "%d Trying %s=%s %s via %s\n", DO_DELETE, name, value, DO_DELETE?"delete":"update", T_ATTRD);
if(attrd != NULL) {
update = create_xml_node(NULL, __FUNCTION__);
crm_xml_add(update, F_TYPE, T_ATTRD);
crm_xml_add(update, F_ORIG, crm_system_name);
crm_xml_add(update, F_ATTRD_TASK, "update");
crm_xml_add(update, F_ATTRD_SECTION, XML_CIB_TAG_STATUS);
crm_xml_add(update, F_ATTRD_ATTRIBUTE, name);
if(set != NULL) {
crm_xml_add(update, F_ATTRD_SET, set);
}
if(id != NULL) {
crm_xml_add(update, F_ATTRD_KEY, id);
}
if(value != NULL) {
crm_xml_add(update, F_ATTRD_VALUE, value);
}
rc = send_ipc_message(attrd, update);
-
- } else {
- crm_info("Could not connect to %s", T_ATTRD);
+ reason = "message delivery";
}
if(rc == FALSE) {
- crm_info("Could not send %s=%s update via %s", name, value?"<none>":value, T_ATTRD);
- fprintf(stderr, "Could not send %s=%s update via %s\n", name, value?"<none>":value, T_ATTRD);
+ crm_info("Could not send %s=%s update via %s: %s", name, value?"<none>":value, T_ATTRD, reason);
+ fprintf(stderr, "Could not send %s=%s update via %s: %s\n", name, value?"<none>":value, T_ATTRD, reason);
}
free_xml(update);
return rc;
}
static int determine_host(cib_t *the_cib)
{
if(dest_uname == NULL) {
struct utsname name;
if(uname(&name) < 0) {
crm_perror(LOG_ERR,"uname(2) call failed");
return 1;
}
dest_uname = crm_strdup(name.nodename);
crm_info("Detected uname: %s", dest_uname);
}
if(dest_node == NULL && dest_uname != NULL) {
int rc = query_node_uuid(the_cib, dest_uname, &dest_node);
if(rc != cib_ok) {
fprintf(stderr,"Could not map uname=%s to a UUID: %s\n",
dest_uname, cib_error2string(rc));
return rc;
} else {
crm_info("Mapped %s to %s",
dest_uname, crm_str(dest_node));
}
}
return cib_ok;
}
int
main(int argc, char **argv)
{
cib_t * the_cib = NULL;
enum cib_errors rc = cib_ok;
int cib_opts = cib_sync_call;
int argerr = 0;
int flag;
#ifdef HAVE_GETOPT_H
int option_index = 0;
static struct option long_options[] = {
/* Top-level Options */
{"verbose", 0, 0, 'V'},
{"help", 0, 0, '?'},
{"quiet", 0, 0, 'Q'},
{"get-value", 0, 0, 'G'},
{"delete-attr", 0, 0, 'D'},
{"node", 1, 0, 'N'},
{"node-uname", 1, 0, 'U'}, /* legacy */
{"node-uuid", 1, 0, 'u'}, /* legacy */
{"set-name", 1, 0, 's'},
{"attr-id", 1, 0, 'i'},
{"attr-name", 1, 0, 'n'},
{"attr-value", 1, 0, 'v'},
{"resource-id", 1, 0, 'r'},
{"lifetime", 1, 0, 'l'},
{"type", 1, 0, 't'},
{"inhibit-policy-engine", 0, 0, '!'},
{0, 0, 0, 0}
};
#endif
crm_log_init(basename(argv[0]), LOG_ERR, FALSE, FALSE, argc, argv);
if(argc < 2) {
usage(crm_system_name, LSB_EXIT_EINVAL);
}
while (1) {
#ifdef HAVE_GETOPT_H
flag = getopt_long(argc, argv, OPTARGS,
long_options, &option_index);
#else
flag = getopt(argc, argv, OPTARGS);
#endif
if (flag == -1)
break;
switch(flag) {
case 'V':
cl_log_enable_stderr(TRUE);
alter_debug(DEBUG_INC);
break;
case '?':
usage(crm_system_name, LSB_EXIT_OK);
break;
case 'G':
DO_WRITE = FALSE;
break;
case 'Q':
BE_QUIET = TRUE;
break;
case 'D':
DO_DELETE = TRUE;
DO_WRITE = FALSE;
break;
case 'U':
case 'N':
crm_debug_2("Option %c => %s", flag, optarg);
dest_uname = crm_strdup(optarg);
break;
case 'u':
crm_debug_2("Option %c => %s", flag, optarg);
dest_node = crm_strdup(optarg);
break;
case 's':
crm_debug_2("Option %c => %s", flag, optarg);
set_name = crm_strdup(optarg);
break;
case 'l':
crm_debug_2("Option %c => %s", flag, optarg);
type = optarg;
break;
case 't':
crm_debug_2("Option %c => %s", flag, optarg);
type = optarg;
break;
case 'n':
crm_debug_2("Option %c => %s", flag, optarg);
attr_name = crm_strdup(optarg);
break;
case 'i':
crm_debug_2("Option %c => %s", flag, optarg);
attr_id = crm_strdup(optarg);
break;
case 'v':
crm_debug_2("Option %c => %s", flag, optarg);
attr_value = optarg;
break;
case 'r':
crm_debug_2("Option %c => %s", flag, optarg);
rsc_id = optarg;
break;
case '!':
crm_warn("Inhibiting notifications for this update");
cib_opts |= cib_inhibit_notify;
break;
default:
printf("Argument code 0%o (%c) is not (?yet?) supported\n", flag, flag);
++argerr;
break;
}
}
if (optind < argc) {
printf("non-option ARGV-elements: ");
while (optind < argc)
printf("%s ", argv[optind++]);
printf("\n");
}
if (optind > argc) {
++argerr;
}
if (argerr) {
usage(crm_system_name, LSB_EXIT_GENERIC);
}
the_cib = cib_new();
rc = the_cib->cmds->signon(the_cib, crm_system_name, cib_command);
if(rc != cib_ok) {
fprintf(stderr, "Error signing on to the CIB service: %s\n",
cib_error2string(rc));
return rc;
}
if(safe_str_eq(crm_system_name, "crm_attribute")){
if(type == NULL && dest_uname == NULL) {
/* we're updating cluster options - dont populate dest_node */
type = XML_CIB_TAG_CRMCONFIG;
} else {
rc = determine_host(the_cib);
}
} else if(safe_str_eq(crm_system_name, "crm_master")){
int len = 0;
determine_host(the_cib);
if(safe_str_eq(type, "reboot")) {
type = XML_CIB_TAG_STATUS;
} else {
type = XML_CIB_TAG_NODES;
}
rsc_id = getenv("OCF_RESOURCE_INSTANCE");
if(rsc_id == NULL && dest_node == NULL) {
fprintf(stderr, "This program should only ever be "
"invoked from inside an OCF resource agent.\n");
fprintf(stderr, "DO NOT INVOKE MANUALLY FROM THE COMMAND LINE.\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
} else if(dest_node == NULL) {
fprintf(stderr, "Could not determine node UUID.\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
} else if(rsc_id == NULL) {
fprintf(stderr, "Could not determine resource name.\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
}
len = 8 + strlen(rsc_id);
crm_malloc0(attr_name, len);
snprintf(attr_name, len, "master-%s", rsc_id);
len = 3 + strlen(type) + strlen(attr_name) + strlen(dest_node);
crm_malloc0(attr_id, len);
snprintf(attr_id, len, "%s-%s-%s", type, attr_name, dest_node);
} else if(safe_str_eq(crm_system_name, "crm_failcount")){
rc = determine_host(the_cib);
type = XML_CIB_TAG_STATUS;
if(rsc_id == NULL) {
fprintf(stderr,"Please specify a resource with -r\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
}
if(dest_node == NULL) {
fprintf(stderr,"Please specify a node with -U or -u\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
}
if(DO_WRITE && char2score(attr_value) <= 0) {
if(safe_str_neq(attr_value, "0")) {
fprintf(stderr,"%s is an inappropriate value for a failcount.\n", attr_value);
rc = CIBRES_MISSING_FIELD;
goto bail;
}
}
set_name = NULL;
attr_name = crm_concat("fail-count", rsc_id, '-');
} else if(safe_str_eq(crm_system_name, "crm_standby")) {
determine_host(the_cib);
if(dest_node == NULL) {
fprintf(stderr,"Please specify a node with -U or -u\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
} else if(DO_DELETE) {
rc = delete_standby(the_cib, dest_node, type, attr_value);
} else if(DO_WRITE) {
if(attr_value == NULL) {
fprintf(stderr, "Please specify 'true' or 'false' with -v\n");
rc = CIBRES_MISSING_FIELD;
goto bail;
}
rc = set_standby(the_cib, dest_node, type, attr_value);
} else {
char *read_value = NULL;
char *scope = NULL;
if(type) {
scope = crm_strdup(type);
}
rc = query_standby(
the_cib, dest_node, &scope, &read_value);
if(rc == cib_NOTEXISTS) {
rc = cib_ok;
read_value = crm_strdup("off");
}
if(read_value == NULL) {
read_value = crm_strdup("unknown");
}
if(BE_QUIET == FALSE) {
if(attr_id) {
fprintf(stdout, "id=%s ", attr_id);
}
if(attr_name) {
fprintf(stdout, "name=%s ", attr_name);
}
fprintf(stdout, "scope=%s value=%s\n", scope, read_value);
} else {
fprintf(stdout, "%s\n", read_value);
}
crm_free(scope);
}
goto bail;
}
if(rc != cib_ok) {
crm_info("Error during setup of %s=%s update", attr_name, DO_DELETE?"<none>":attr_value);
} else if( (DO_WRITE || DO_DELETE)
&& safe_str_eq(type, XML_CIB_TAG_STATUS)
&& set_via_attrd(set_name, attr_id, attr_name, DO_DELETE?NULL:attr_value)) {
crm_info("Update %s=%s sent via attrd", attr_name, DO_DELETE?"<none>":attr_value);
} else if(DO_DELETE) {
rc = delete_attr(the_cib, cib_opts, type, dest_node, set_name,
attr_id, attr_name, attr_value, TRUE);
if(rc == cib_NOTEXISTS) {
/* Nothing to delete...
* which means its not there...
* which is what the admin wanted
*/
rc = cib_ok;
} else if(rc != cib_missing_data
&& safe_str_eq(crm_system_name, "crm_failcount")) {
char *now_s = NULL;
time_t now = time(NULL);
now_s = crm_itoa(now);
update_attr(the_cib, cib_sync_call,
XML_CIB_TAG_CRMCONFIG, NULL, NULL, NULL,
"last-lrm-refresh", now_s, TRUE);
crm_free(now_s);
}
} else if(DO_WRITE) {
CRM_DEV_ASSERT(type != NULL);
CRM_DEV_ASSERT(attr_name != NULL);
CRM_DEV_ASSERT(attr_value != NULL);
rc = update_attr(the_cib, cib_opts, type, dest_node, set_name,
attr_id, attr_name, attr_value, TRUE);
} else /* query */ {
char *read_value = NULL;
rc = read_attr(the_cib, type, dest_node, set_name,
attr_id, attr_name, &read_value, TRUE);
if(rc == cib_NOTEXISTS
&& safe_str_eq(crm_system_name, "crm_failcount")) {
rc = cib_ok;
read_value = crm_strdup("0");
}
crm_info("Read %s=%s %s%s",
attr_name, crm_str(read_value),
set_name?"in ":"", set_name?set_name:"");
if(rc == cib_missing_data) {
rc = cib_ok;
} else if(BE_QUIET == FALSE) {
fprintf(stdout, "%s%s %s%s value=%s\n",
attr_id?"id=":"", attr_id?attr_id:"",
attr_name?"name=":"", attr_name?attr_name:"",
read_value?read_value:"(null)");
} else if(read_value != NULL) {
fprintf(stdout, "%s\n", read_value);
}
}
bail:
the_cib->cmds->signoff(the_cib);
if(rc == cib_missing_data) {
printf("Please choose from one of the matches above and suppy the 'id' with --attr-id\n");
} else if(rc != cib_ok) {
fprintf(stderr, "Error performing operation: %s\n",
cib_error2string(rc));
}
return rc;
}
void
usage(const char *cmd, int exit_status)
{
FILE *stream;
stream = exit_status ? stderr : stdout;
if(safe_str_eq(cmd, "crm_master")) {
fprintf(stream, "%s -- Manage a master/slave resource's preference for being promoted on a given node.\n"
" Designed to be used within resource agent scripts and relies on a sane master/slave resource envioronment. Should not be used manually.\n\n",
cmd);
fprintf(stream, "usage: %s [-?VQ] -(D|G|v) [-l]\n", cmd);
} else if(safe_str_eq(cmd, "crm_standby")) {
fprintf(stream, "%s -- Manage a node's standby status.\n"
" Putting a node into standby mode prevents it from hosting cluster resources.\n\n",
cmd);
fprintf(stream, "usage: %s [-?V] -(u|U) -(D|G|v) [-l]\n", cmd);
} else if(safe_str_eq(cmd, "crm_failcount")) {
fprintf(stream, "%s -- Manage the counter recording each resource's failures.\n"
" Allows the current number of failures for a resource to be queried, modified or erased/deleted.\n\n",
cmd);
fprintf(stream, "usage: %s [-?V] -(u|U) -(D|G|v) -r\n", cmd);
} else {
fprintf(stream, "%s -- Manage node's attributes and cluster options.\n"
" Allows node attributes and cluster options to be queried, modified and deleted.\n\n",
cmd);
fprintf(stream, "usage: %s [-?V] -(D|G|v) [options]\n", cmd);
}
fprintf(stream, "Options\n");
fprintf(stream, "\t--%s (-%c)\t: this help message\n", "help", '?');
fprintf(stream, "\t--%s (-%c)\t: "
"turn on debug info. additional instances increase verbosity\n",
"verbose", 'V');
fprintf(stream, "\t--%s (-%c)\t: Print only the value on stdout"
" (use with -G)\n", "quiet", 'Q');
fprintf(stream, "\t--%s (-%c)\t: "
"Retrieve rather than set the %s\n", "get-value", 'G',
safe_str_eq(cmd, "crm_master")?"named attribute":"preference to be promoted");
fprintf(stream, "\t--%s (-%c)\t: "
"Delete rather than set the attribute\n", "delete-attr", 'D');
fprintf(stream, "\t--%s (-%c) <string>\t: "
"Value to use (ignored with -G)\n", "attr-value", 'v');
fprintf(stream, "\t--%s (-%c) <string>\t: "
"The 'id' of the attribute. Advanced use only.\n", "attr-id", 'i');
if(safe_str_eq(cmd, "crm_master")) {
fprintf(stream, "\t--%s (-%c) <string>\t: "
"How long the preference lasts (reboot|forever)\n",
"lifetime", 'l');
exit(exit_status);
} else if(safe_str_eq(cmd, "crm_standby")) {
fprintf(stream, "\t--%s (-%c) <node_uname>\t: "
"uname of the node to change\n", "node", 'N');
fprintf(stream, "\t--%s (-%c) <string>\t: "
"How long the preference lasts (reboot|forever)\n"
"\t If a forever value exists, it is ALWAYS used by the CRM\n"
"\t instead of any reboot value\n", "lifetime", 'l');
exit(exit_status);
}
fprintf(stream, "\t--%s (-%c) <node_uname>\t: "
"uname of the node to change\n", "node", 'N');
if(safe_str_eq(cmd, "crm_failcount")) {
fprintf(stream, "\t--%s (-%c) <resource name>\t: "
"name of the resource to operate on\n", "resource-id", 'r');
} else {
fprintf(stream, "\t--%s (-%c) <string>\t: "
"Set of attributes in which to read/write the attribute\n",
"set-name", 's');
fprintf(stream, "\t--%s (-%c) <string>\t: "
"Attribute to set\n", "attr-name", 'n');
fprintf(stream, "\t--%s (-%c) <string>\t: "
"Which section of the CIB to set the attribute: (%s|%s|%s)\n",
"type", 't',
XML_CIB_TAG_NODES, XML_CIB_TAG_STATUS, XML_CIB_TAG_CRMCONFIG);
fprintf(stream, "\t \"-t %s\" options: -N -n [-s]\n", XML_CIB_TAG_NODES);
fprintf(stream, "\t \"-t %s\" options: -N -n [-s]\n", XML_CIB_TAG_STATUS);
fprintf(stream, "\t \"-t %s\" options: -N [-s]\n", XML_CIB_TAG_CRMCONFIG);
}
if(safe_str_neq(crm_system_name, "crm_standby")) {
fprintf(stream, "\t--%s (-%c)\t: "
"Make a change and prevent the TE/PE from seeing it straight away.\n"
"\t You may think you want this option but you don't."
" Advanced use only - you have been warned!\n", "inhibit-policy-engine", '!');
}
fflush(stream);
exit(exit_status);
}

File Metadata

Mime Type
text/x-diff
Expires
Mon, Apr 21, 2:25 PM (1 d, 3 h)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
1664921
Default Alt Text
(16 KB)

Event Timeline