244 lines
5.3 KiB
C
244 lines
5.3 KiB
C
|
/* $NetBSD: commands.c,v 1.4 2011/09/15 11:46:19 blymn Exp $ */
|
||
|
|
||
|
/*-
|
||
|
* Copyright 2009 Brett Lymn <blymn@NetBSD.org>
|
||
|
*
|
||
|
* All rights reserved.
|
||
|
*
|
||
|
* This code has been donated to The NetBSD Foundation by the Author.
|
||
|
*
|
||
|
* Redistribution and use in source and binary forms, with or without
|
||
|
* modification, are permitted provided that the following conditions
|
||
|
* are met:
|
||
|
* 1. Redistributions of source code must retain the above copyright
|
||
|
* notice, this list of conditions and the following disclaimer.
|
||
|
* 2. The name of the author may not be used to endorse or promote products
|
||
|
* derived from this software withough specific prior written permission
|
||
|
*
|
||
|
* THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
|
||
|
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
||
|
* OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
||
|
* IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
|
||
|
* INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
||
|
* NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
||
|
* THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||
|
*
|
||
|
*
|
||
|
*/
|
||
|
|
||
|
#include <curses.h>
|
||
|
#include <string.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <stdio.h>
|
||
|
#include <unistd.h>
|
||
|
#include <err.h>
|
||
|
#include <sys/types.h>
|
||
|
#include "returns.h"
|
||
|
#include "slave.h"
|
||
|
#include "command_table.h"
|
||
|
|
||
|
extern int cmdpipe[2];
|
||
|
extern int slvpipe[2];
|
||
|
|
||
|
static void report_type(returns_enum_t);
|
||
|
static void report_message(int, const char *);
|
||
|
|
||
|
/*
|
||
|
* Match the passed command string and execute the associated test
|
||
|
* function.
|
||
|
*/
|
||
|
void
|
||
|
command_execute(char *func, int nargs, char **args)
|
||
|
{
|
||
|
size_t i;
|
||
|
|
||
|
i = 0;
|
||
|
while (i < ncmds) {
|
||
|
if (strcasecmp(func, commands[i].name) == 0) {
|
||
|
/* matched function */
|
||
|
commands[i].func(nargs, args);
|
||
|
return;
|
||
|
}
|
||
|
i++;
|
||
|
}
|
||
|
|
||
|
report_status("UNKNOWN_FUNCTION");
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report an pointer value back to the director
|
||
|
*/
|
||
|
void
|
||
|
report_ptr(void *ptr)
|
||
|
{
|
||
|
char *string;
|
||
|
|
||
|
if (ptr == NULL)
|
||
|
asprintf(&string, "NULL");
|
||
|
else
|
||
|
asprintf(&string, "%p", ptr);
|
||
|
report_status(string);
|
||
|
free(string);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report an integer value back to the director
|
||
|
*/
|
||
|
void
|
||
|
report_int(int value)
|
||
|
{
|
||
|
char *string;
|
||
|
|
||
|
asprintf(&string, "%d", value);
|
||
|
report_status(string);
|
||
|
free(string);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report either an ERR or OK back to the director
|
||
|
*/
|
||
|
void
|
||
|
report_return(int status)
|
||
|
{
|
||
|
if (status == ERR)
|
||
|
report_type(ret_err);
|
||
|
else if (status == OK)
|
||
|
report_type(ret_ok);
|
||
|
else
|
||
|
report_status("INVALID_RETURN");
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report the type back to the director via the command pipe
|
||
|
*/
|
||
|
static void
|
||
|
report_type(returns_enum_t return_type)
|
||
|
{
|
||
|
int type;
|
||
|
|
||
|
type = return_type;
|
||
|
if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
|
||
|
err(1, "command pipe write for status type failed");
|
||
|
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report the number of returns back to the director via the command pipe
|
||
|
*/
|
||
|
void
|
||
|
report_count(int count)
|
||
|
{
|
||
|
int type;
|
||
|
|
||
|
type = ret_count;
|
||
|
if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
|
||
|
err(1, "command pipe write for count type failed");
|
||
|
|
||
|
if (write(slvpipe[WRITE_PIPE], &count, sizeof(int)) < 0)
|
||
|
err(1, "command pipe write for count");
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report the status back to the director via the command pipe
|
||
|
*/
|
||
|
void
|
||
|
report_status(const char *status)
|
||
|
{
|
||
|
report_message(ret_string, status);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report an error message back to the director via the command pipe.
|
||
|
*/
|
||
|
void
|
||
|
report_error(const char *status)
|
||
|
{
|
||
|
report_message(ret_slave_error, status);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report the message with the given type back to the director via the
|
||
|
* command pipe.
|
||
|
*/
|
||
|
static void
|
||
|
report_message(int type, const char *status)
|
||
|
{
|
||
|
int len;
|
||
|
|
||
|
len = strlen(status);
|
||
|
|
||
|
if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
|
||
|
err(1, "command pipe write for message type failed");
|
||
|
|
||
|
if (write(slvpipe[WRITE_PIPE], &len, sizeof(int)) < 0)
|
||
|
err(1, "command pipe write for message length failed");
|
||
|
|
||
|
if (write(slvpipe[WRITE_PIPE], status, len) < 0)
|
||
|
err(1, "command pipe write of message data failed");
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report a string of chtype back to the director via the command pipe.
|
||
|
*/
|
||
|
void
|
||
|
report_byte(chtype c)
|
||
|
{
|
||
|
chtype string[2];
|
||
|
|
||
|
string[0] = c;
|
||
|
string[1] = A_NORMAL | '\0';
|
||
|
report_nstr(string);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Report a string of chtype back to the director via the command pipe.
|
||
|
*/
|
||
|
void
|
||
|
report_nstr(chtype *string)
|
||
|
{
|
||
|
int len, type;
|
||
|
chtype *p;
|
||
|
|
||
|
len = 0;
|
||
|
p = string;
|
||
|
|
||
|
while ((*p++ & __CHARTEXT) != 0) {
|
||
|
len++;
|
||
|
}
|
||
|
|
||
|
len++; /* add in the termination chtype */
|
||
|
len *= sizeof(chtype);
|
||
|
|
||
|
type = ret_byte;
|
||
|
if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
|
||
|
err(1, "%s: command pipe write for status type failed",
|
||
|
__func__);
|
||
|
|
||
|
if (write(slvpipe[WRITE_PIPE], &len, sizeof(int)) < 0)
|
||
|
err(1, "%s: command pipe write for status length failed",
|
||
|
__func__);
|
||
|
|
||
|
if (write(slvpipe[WRITE_PIPE], string, len) < 0)
|
||
|
err(1, "%s: command pipe write of status data failed",
|
||
|
__func__);
|
||
|
}
|
||
|
|
||
|
/*
|
||
|
* Check the number of args we received are what we expect. Return an
|
||
|
* error if they do not match.
|
||
|
*/
|
||
|
int
|
||
|
check_arg_count(int nargs, int expected)
|
||
|
{
|
||
|
if (nargs != expected) {
|
||
|
report_count(1);
|
||
|
report_error("INCORRECT_ARGUMENT_NUMBER");
|
||
|
return(1);
|
||
|
}
|
||
|
|
||
|
return(0);
|
||
|
}
|