nut-debian/clients/upsrw.c

461 lines
10 KiB
C
Raw Normal View History

2010-03-26 01:20:59 +02:00
/* upsrw - simple client for read/write variable access (formerly upsct2)
Copyright (C) 1999 Russell Kroll <rkroll@exploits.org>
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 of the License, or
(at your option) any later version.
This program 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 program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
#include "common.h"
#include <pwd.h>
#include <netdb.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include "upsclient.h"
2011-01-26 11:35:08 +02:00
static char *upsname = NULL, *hostname = NULL;
static UPSCONN_t *ups = NULL;
2010-03-26 01:20:59 +02:00
struct list_t {
char *name;
struct list_t *next;
};
static void usage(const char *prog)
{
2011-01-26 11:35:08 +02:00
printf("Network UPS Tools %s %s\n\n", prog, UPS_VERSION);
2010-03-26 01:20:59 +02:00
printf("usage: %s [-h]\n", prog);
printf(" %s [-s <variable>] [-u <username>] [-p <password>] <ups>\n\n", prog);
printf("Demo program to set variables within UPS hardware.\n");
printf("\n");
printf(" -h display this help text\n");
printf(" -s <variable> specify variable to be changed\n");
printf(" use -s VAR=VALUE to avoid prompting for value\n");
printf(" -u <username> set username for command authentication\n");
printf(" -p <password> set password for command authentication\n");
printf("\n");
printf(" <ups> UPS identifier - <upsname>[@<hostname>[:<port>]]\n");
printf("\n");
printf("Call without -s to show all possible read/write variables.\n");
}
2011-01-26 11:35:08 +02:00
static void clean_exit(void)
2010-03-26 01:20:59 +02:00
{
2011-01-26 11:35:08 +02:00
if (ups) {
upscli_disconnect(ups);
}
2010-03-26 01:20:59 +02:00
free(upsname);
free(hostname);
2011-01-26 11:35:08 +02:00
free(ups);
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
static void do_set(const char *varname, const char *newval)
2010-03-26 01:20:59 +02:00
{
char buf[SMALLBUF], enc[SMALLBUF];
2011-01-26 11:35:08 +02:00
snprintf(buf, sizeof(buf), "SET VAR %s %s \"%s\"\n", upsname, varname, pconf_encode(newval, enc, sizeof(enc)));
2010-03-26 01:20:59 +02:00
if (upscli_sendline(ups, buf, strlen(buf)) < 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Can't set variable: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
if (upscli_readline(ups, buf, sizeof(buf)) < 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Set variable failed: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
/* FUTURE: status cookies will tie in here */
if (strncmp(buf, "OK", 2) != 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Unexpected response from upsd: %s", buf);
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
fprintf(stderr, "%s\n", buf);
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
static void do_setvar(const char *varname, char *uin, const char *pass)
2010-03-26 01:20:59 +02:00
{
char newval[SMALLBUF], temp[SMALLBUF], user[SMALLBUF], *ptr;
2011-01-26 11:35:08 +02:00
struct passwd *pw;
2010-03-26 01:20:59 +02:00
if (uin) {
snprintf(user, sizeof(user), "%s", uin);
} else {
memset(user, '\0', sizeof(user));
pw = getpwuid(getuid());
2011-01-26 11:35:08 +02:00
if (pw) {
2010-03-26 01:20:59 +02:00
printf("Username (%s): ", pw->pw_name);
2011-01-26 11:35:08 +02:00
} else {
2010-03-26 01:20:59 +02:00
printf("Username: ");
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
if (fgets(user, sizeof(user), stdin) == NULL) {
upsdebug_with_errno(LOG_INFO, "%s", __func__);
}
/* deal with that pesky newline */
2011-01-26 11:35:08 +02:00
if (strlen(user) > 1) {
2010-03-26 01:20:59 +02:00
user[strlen(user) - 1] = '\0';
2011-01-26 11:35:08 +02:00
} else {
if (!pw) {
2010-03-26 01:20:59 +02:00
fatalx(EXIT_FAILURE, "No username available - even tried getpwuid");
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
snprintf(user, sizeof(user), "%s", pw->pw_name);
}
}
/* leaks - use -p when running in valgrind */
if (!pass) {
pass = GETPASS("Password: " );
if (!pass) {
2011-01-26 11:35:08 +02:00
fatal_with_errno(EXIT_FAILURE, "getpass failed");
2010-03-26 01:20:59 +02:00
}
}
/* Check if varname is in VAR=VALUE form */
if ((ptr = strchr(varname, '=')) != NULL) {
*ptr++ = 0;
snprintf(newval, sizeof(newval), "%s", ptr);
} else {
printf("Enter new value for %s: ", varname);
fflush(stdout);
if (fgets(newval, sizeof(newval), stdin) == NULL) {
upsdebug_with_errno(LOG_INFO, "%s", __func__);
}
newval[strlen(newval) - 1] = '\0';
}
snprintf(temp, sizeof(temp), "USERNAME %s\n", user);
if (upscli_sendline(ups, temp, strlen(temp)) < 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Can't set username: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
if (upscli_readline(ups, temp, sizeof(temp)) < 0) {
if (upscli_upserror(ups) == UPSCLI_ERR_UNKCOMMAND) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Set username failed due to an unknown command. You probably need to upgrade upsd.");
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Set username failed: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
snprintf(temp, sizeof(temp), "PASSWORD %s\n", pass);
if (upscli_sendline(ups, temp, strlen(temp)) < 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Can't set password: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
if (upscli_readline(ups, temp, sizeof(temp)) < 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Set password failed: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
/* no upsname means die */
if (!upsname) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: a UPS name must be specified (upsname[@hostname[:port]])");
2010-03-26 01:20:59 +02:00
}
/* old variable names are no longer supported */
if (!strchr(varname, '.')) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: old variable names are not supported");
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
do_set(varname, newval);
}
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
static const char *get_data(const char *type, const char *varname)
2010-03-26 01:20:59 +02:00
{
int ret;
unsigned int numq, numa;
char **answer;
2011-01-26 11:35:08 +02:00
const char *query[4];
2010-03-26 01:20:59 +02:00
query[0] = type;
query[1] = upsname;
query[2] = varname;
2011-01-26 11:35:08 +02:00
2010-03-26 01:20:59 +02:00
numq = 3;
ret = upscli_get(ups, numq, query, &numa, &answer);
2011-01-26 11:35:08 +02:00
if ((ret < 0) || (numa < numq)) {
2010-03-26 01:20:59 +02:00
return NULL;
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
/* <type> <upsname> <varname> <desc> */
return answer[3];
}
2011-01-26 11:35:08 +02:00
static void do_string(const char *varname)
2010-03-26 01:20:59 +02:00
{
2011-01-26 11:35:08 +02:00
const char *val;
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
val = get_data("VAR", varname);
2010-03-26 01:20:59 +02:00
if (!val) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "do_string: can't get current value of %s", varname);
2010-03-26 01:20:59 +02:00
}
printf("Type: STRING\n");
printf("Value: %s\n", val);
}
2011-01-26 11:35:08 +02:00
static void do_enum(const char *varname)
2010-03-26 01:20:59 +02:00
{
int ret;
unsigned int numq, numa;
2011-01-26 11:35:08 +02:00
char **answer, buf[SMALLBUF];
const char *query[4], *val;
2010-03-26 01:20:59 +02:00
/* get current value */
2011-01-26 11:35:08 +02:00
val = get_data("VAR", varname);
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
if (!val) {
fatalx(EXIT_FAILURE, "do_enum: can't get current value of %s", varname);
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
snprintf(buf, sizeof(buf), "%s", val);
2010-03-26 01:20:59 +02:00
query[0] = "ENUM";
query[1] = upsname;
query[2] = varname;
numq = 3;
ret = upscli_list_start(ups, numq, query);
if (ret < 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
ret = upscli_list_next(ups, numq, query, &numa, &answer);
printf("Type: ENUM\n");
while (ret == 1) {
/* ENUM <upsname> <varname> <value> */
if (numa < 4) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: insufficient data (got %d args, need at least 4)", numa);
2010-03-26 01:20:59 +02:00
}
printf("Option: \"%s\"", answer[3]);
2011-01-26 11:35:08 +02:00
if (!strcmp(answer[3], buf)) {
2010-03-26 01:20:59 +02:00
printf(" SELECTED");
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
printf("\n");
ret = upscli_list_next(ups, numq, query, &numa, &answer);
}
}
2011-01-26 11:35:08 +02:00
static void do_type(const char *varname)
2010-03-26 01:20:59 +02:00
{
int ret;
unsigned int i, numq, numa;
char **answer;
2011-01-26 11:35:08 +02:00
const char *query[4];
2010-03-26 01:20:59 +02:00
query[0] = "TYPE";
query[1] = upsname;
query[2] = varname;
numq = 3;
ret = upscli_get(ups, numq, query, &numa, &answer);
if ((ret < 0) || (numa < numq)) {
printf("Unknown type\n");
return;
}
/* TYPE <upsname> <varname> <type>... */
for (i = 3; i < numa; i++) {
if (!strcasecmp(answer[i], "ENUM")) {
2011-01-26 11:35:08 +02:00
do_enum(varname);
2010-03-26 01:20:59 +02:00
return;
}
if (!strncasecmp(answer[i], "STRING:", 7)) {
2011-01-26 11:35:08 +02:00
do_string(varname);
2010-03-26 01:20:59 +02:00
return;
}
/* ignore this one */
2011-01-26 11:35:08 +02:00
if (!strcasecmp(answer[i], "RW")) {
2010-03-26 01:20:59 +02:00
continue;
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
printf("Type: %s (unrecognized)\n", answer[i]);
}
}
2011-01-26 11:35:08 +02:00
static void print_rw(const char *varname)
2010-03-26 01:20:59 +02:00
{
2011-01-26 11:35:08 +02:00
const char *tmp;
2010-03-26 01:20:59 +02:00
printf("[%s]\n", varname);
2011-01-26 11:35:08 +02:00
tmp = get_data("DESC", varname);
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
if (tmp) {
2010-03-26 01:20:59 +02:00
printf("%s\n", tmp);
2011-01-26 11:35:08 +02:00
} else {
2010-03-26 01:20:59 +02:00
printf("Description unavailable\n");
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
do_type(varname);
2010-03-26 01:20:59 +02:00
printf("\n");
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
static void print_rwlist(void)
2010-03-26 01:20:59 +02:00
{
int ret;
unsigned int numq, numa;
2011-01-26 11:35:08 +02:00
const char *query[2];
2010-03-26 01:20:59 +02:00
char **answer;
struct list_t *lhead, *llast, *ltmp, *lnext;
/* the upsname is now required */
if (!upsname) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: a UPS name must be specified (upsname[@hostname[:port]])");
2010-03-26 01:20:59 +02:00
}
llast = lhead = NULL;
query[0] = "RW";
query[1] = upsname;
numq = 2;
ret = upscli_list_start(ups, numq, query);
if (ret < 0) {
/* old upsd --> fall back on old LISTRW technique */
if (upscli_upserror(ups) == UPSCLI_ERR_UNKCOMMAND) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: upsd is too old to support this query");
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
ret = upscli_list_next(ups, numq, query, &numa, &answer);
while (ret == 1) {
/* RW <upsname> <varname> <value> */
if (numa < 4) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: insufficient data (got %d args, need at least 4)", numa);
2010-03-26 01:20:59 +02:00
}
/* sock this entry away for later */
ltmp = xmalloc(sizeof(struct list_t));
ltmp->name = xstrdup(answer[2]);
ltmp->next = NULL;
2011-01-26 11:35:08 +02:00
if (llast) {
2010-03-26 01:20:59 +02:00
llast->next = ltmp;
2011-01-26 11:35:08 +02:00
} else {
2010-03-26 01:20:59 +02:00
lhead = ltmp;
2011-01-26 11:35:08 +02:00
}
2010-03-26 01:20:59 +02:00
llast = ltmp;
ret = upscli_list_next(ups, numq, query, &numa, &answer);
}
/* use the list to get descriptions and types */
ltmp = lhead;
while (ltmp) {
lnext = ltmp->next;
2011-01-26 11:35:08 +02:00
print_rw(ltmp->name);
2010-03-26 01:20:59 +02:00
free(ltmp->name);
free(ltmp);
ltmp = lnext;
}
}
int main(int argc, char **argv)
{
2011-01-26 11:35:08 +02:00
int i, port;
const char *prog = xbasename(argv[0]);
char *password = NULL, *username = NULL, *setvar = NULL;
2010-03-26 01:20:59 +02:00
2011-06-01 23:31:49 +03:00
while ((i = getopt(argc, argv, "+hs:p:u:V")) != -1) {
2011-01-26 11:35:08 +02:00
switch (i)
{
2010-03-26 01:20:59 +02:00
case 's':
setvar = optarg;
break;
case 'p':
password = optarg;
break;
case 'u':
username = optarg;
break;
case 'V':
2011-01-26 11:35:08 +02:00
printf("Network UPS Tools %s %s\n", prog, UPS_VERSION);
2010-03-26 01:20:59 +02:00
exit(EXIT_SUCCESS);
2011-06-01 23:31:49 +03:00
case 'h':
2010-03-26 01:20:59 +02:00
default:
usage(prog);
2011-01-26 11:35:08 +02:00
exit(EXIT_SUCCESS);
2010-03-26 01:20:59 +02:00
}
}
argc -= optind;
argv += optind;
2011-01-26 11:35:08 +02:00
if (argc < 1) {
2010-03-26 01:20:59 +02:00
usage(prog);
2011-01-26 11:35:08 +02:00
exit(EXIT_SUCCESS);
}
2010-03-26 01:20:59 +02:00
2011-01-26 11:35:08 +02:00
/* be a good little client that cleans up after itself */
atexit(clean_exit);
2010-03-26 01:20:59 +02:00
if (upscli_splitname(argv[0], &upsname, &hostname, &port) != 0) {
2011-01-26 11:35:08 +02:00
fatalx(EXIT_FAILURE, "Error: invalid UPS definition. Required format: upsname[@hostname[:port]]");
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
ups = xcalloc(1, sizeof(*ups));
if (upscli_connect(ups, hostname, port, 0) < 0) {
fatalx(EXIT_FAILURE, "Error: %s", upscli_strerror(ups));
2010-03-26 01:20:59 +02:00
}
if (setvar) {
2011-01-26 11:35:08 +02:00
/* setting a variable */
do_setvar(setvar, username, password);
} else {
/* if not, get the list of supported read/write variables */
print_rwlist();
2010-03-26 01:20:59 +02:00
}
2011-01-26 11:35:08 +02:00
exit(EXIT_SUCCESS);
2010-03-26 01:20:59 +02:00
}