Home | History | Annotate | Line # | Download | only in slave
commands.c revision 1.4
      1  1.4     blymn /*	$NetBSD: commands.c,v 1.4 2011/09/15 11:46:19 blymn Exp $	*/
      2  1.1     blymn 
      3  1.1     blymn /*-
      4  1.1     blymn  * Copyright 2009 Brett Lymn <blymn (at) NetBSD.org>
      5  1.1     blymn  *
      6  1.1     blymn  * All rights reserved.
      7  1.1     blymn  *
      8  1.1     blymn  * This code has been donated to The NetBSD Foundation by the Author.
      9  1.1     blymn  *
     10  1.1     blymn  * Redistribution and use in source and binary forms, with or without
     11  1.1     blymn  * modification, are permitted provided that the following conditions
     12  1.1     blymn  * are met:
     13  1.1     blymn  * 1. Redistributions of source code must retain the above copyright
     14  1.1     blymn  *    notice, this list of conditions and the following disclaimer.
     15  1.1     blymn  * 2. The name of the author may not be used to endorse or promote products
     16  1.1     blymn  *    derived from this software withough specific prior written permission
     17  1.1     blymn  *
     18  1.1     blymn  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     19  1.1     blymn  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     20  1.1     blymn  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     21  1.1     blymn  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     22  1.1     blymn  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     23  1.1     blymn  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     24  1.1     blymn  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     25  1.1     blymn  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     26  1.1     blymn  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     27  1.1     blymn  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     28  1.1     blymn  *
     29  1.1     blymn  *
     30  1.1     blymn  */
     31  1.1     blymn 
     32  1.1     blymn #include <curses.h>
     33  1.1     blymn #include <string.h>
     34  1.1     blymn #include <stdlib.h>
     35  1.1     blymn #include <stdio.h>
     36  1.2  christos #include <unistd.h>
     37  1.3  christos #include <err.h>
     38  1.1     blymn #include <sys/types.h>
     39  1.1     blymn #include "returns.h"
     40  1.1     blymn #include "slave.h"
     41  1.1     blymn #include "command_table.h"
     42  1.1     blymn 
     43  1.1     blymn extern int cmdpipe[2];
     44  1.1     blymn extern int slvpipe[2];
     45  1.1     blymn 
     46  1.1     blymn static void report_type(returns_enum_t);
     47  1.3  christos static void report_message(int, const char *);
     48  1.1     blymn 
     49  1.1     blymn /*
     50  1.1     blymn  * Match the passed command string and execute the associated test
     51  1.1     blymn  * function.
     52  1.1     blymn  */
     53  1.1     blymn void
     54  1.1     blymn command_execute(char *func, int nargs, char **args)
     55  1.1     blymn {
     56  1.1     blymn 	size_t i;
     57  1.1     blymn 
     58  1.1     blymn 	i = 0;
     59  1.1     blymn 	while (i < ncmds) {
     60  1.1     blymn 		if (strcasecmp(func, commands[i].name) == 0) {
     61  1.1     blymn 			/* matched function */
     62  1.1     blymn 			commands[i].func(nargs, args);
     63  1.1     blymn 			return;
     64  1.1     blymn 		}
     65  1.1     blymn 		i++;
     66  1.1     blymn 	}
     67  1.1     blymn 
     68  1.1     blymn 	report_status("UNKNOWN_FUNCTION");
     69  1.1     blymn }
     70  1.1     blymn 
     71  1.1     blymn /*
     72  1.1     blymn  * Report an pointer value back to the director
     73  1.1     blymn  */
     74  1.1     blymn void
     75  1.1     blymn report_ptr(void *ptr)
     76  1.1     blymn {
     77  1.1     blymn 	char *string;
     78  1.1     blymn 
     79  1.4     blymn 	if (ptr == NULL)
     80  1.4     blymn 		asprintf(&string, "NULL");
     81  1.4     blymn 	else
     82  1.4     blymn 		asprintf(&string, "%p", ptr);
     83  1.1     blymn 	report_status(string);
     84  1.1     blymn 	free(string);
     85  1.1     blymn }
     86  1.1     blymn 
     87  1.1     blymn /*
     88  1.1     blymn  * Report an integer value back to the director
     89  1.1     blymn  */
     90  1.1     blymn void
     91  1.1     blymn report_int(int value)
     92  1.1     blymn {
     93  1.1     blymn 	char *string;
     94  1.1     blymn 
     95  1.1     blymn 	asprintf(&string, "%d", value);
     96  1.1     blymn 	report_status(string);
     97  1.1     blymn 	free(string);
     98  1.1     blymn }
     99  1.1     blymn 
    100  1.1     blymn /*
    101  1.1     blymn  * Report either an ERR or OK back to the director
    102  1.1     blymn  */
    103  1.1     blymn void
    104  1.1     blymn report_return(int status)
    105  1.1     blymn {
    106  1.1     blymn 	if (status == ERR)
    107  1.1     blymn 		report_type(ret_err);
    108  1.1     blymn 	else if (status == OK)
    109  1.1     blymn 		report_type(ret_ok);
    110  1.1     blymn 	else
    111  1.1     blymn 		report_status("INVALID_RETURN");
    112  1.1     blymn }
    113  1.1     blymn 
    114  1.1     blymn /*
    115  1.1     blymn  * Report the type back to the director via the command pipe
    116  1.1     blymn  */
    117  1.1     blymn static void
    118  1.1     blymn report_type(returns_enum_t return_type)
    119  1.1     blymn {
    120  1.1     blymn 	int type;
    121  1.1     blymn 
    122  1.1     blymn 	type = return_type;
    123  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
    124  1.3  christos 		err(1, "command pipe write for status type failed");
    125  1.1     blymn 
    126  1.1     blymn }
    127  1.1     blymn 
    128  1.1     blymn /*
    129  1.1     blymn  * Report the number of returns back to the director via the command pipe
    130  1.1     blymn  */
    131  1.1     blymn void
    132  1.1     blymn report_count(int count)
    133  1.1     blymn {
    134  1.1     blymn 	int type;
    135  1.1     blymn 
    136  1.1     blymn 	type = ret_count;
    137  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
    138  1.3  christos 		err(1, "command pipe write for count type failed");
    139  1.1     blymn 
    140  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &count, sizeof(int)) < 0)
    141  1.3  christos 		err(1, "command pipe write for count");
    142  1.1     blymn }
    143  1.1     blymn 
    144  1.1     blymn /*
    145  1.1     blymn  * Report the status back to the director via the command pipe
    146  1.1     blymn  */
    147  1.1     blymn void
    148  1.3  christos report_status(const char *status)
    149  1.1     blymn {
    150  1.1     blymn 	report_message(ret_string, status);
    151  1.1     blymn }
    152  1.1     blymn 
    153  1.1     blymn /*
    154  1.1     blymn  * Report an error message back to the director via the command pipe.
    155  1.1     blymn  */
    156  1.1     blymn void
    157  1.3  christos report_error(const char *status)
    158  1.1     blymn {
    159  1.1     blymn 	report_message(ret_slave_error, status);
    160  1.1     blymn }
    161  1.1     blymn 
    162  1.1     blymn /*
    163  1.1     blymn  * Report the message with the given type back to the director via the
    164  1.1     blymn  * command pipe.
    165  1.1     blymn  */
    166  1.1     blymn static void
    167  1.3  christos report_message(int type, const char *status)
    168  1.1     blymn {
    169  1.1     blymn 	int len;
    170  1.1     blymn 
    171  1.1     blymn 	len = strlen(status);
    172  1.1     blymn 
    173  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
    174  1.3  christos 		err(1, "command pipe write for message type failed");
    175  1.1     blymn 
    176  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &len, sizeof(int)) < 0)
    177  1.3  christos 		err(1, "command pipe write for message length failed");
    178  1.1     blymn 
    179  1.3  christos 	if (write(slvpipe[WRITE_PIPE], status, len) < 0)
    180  1.3  christos 		err(1, "command pipe write of message data failed");
    181  1.1     blymn }
    182  1.1     blymn 
    183  1.1     blymn /*
    184  1.1     blymn  * Report a string of chtype back to the director via the command pipe.
    185  1.1     blymn  */
    186  1.1     blymn void
    187  1.4     blymn report_byte(chtype c)
    188  1.4     blymn {
    189  1.4     blymn 	chtype string[2];
    190  1.4     blymn 
    191  1.4     blymn 	string[0] = c;
    192  1.4     blymn 	string[1] = A_NORMAL | '\0';
    193  1.4     blymn 	report_nstr(string);
    194  1.4     blymn }
    195  1.4     blymn 
    196  1.4     blymn /*
    197  1.4     blymn  * Report a string of chtype back to the director via the command pipe.
    198  1.4     blymn  */
    199  1.4     blymn void
    200  1.1     blymn report_nstr(chtype *string)
    201  1.1     blymn {
    202  1.1     blymn 	int len, type;
    203  1.1     blymn 	chtype *p;
    204  1.1     blymn 
    205  1.1     blymn 	len = 0;
    206  1.1     blymn 	p = string;
    207  1.1     blymn 
    208  1.1     blymn 	while ((*p++ & __CHARTEXT) != 0) {
    209  1.1     blymn 		len++;
    210  1.1     blymn 	}
    211  1.1     blymn 
    212  1.4     blymn 	len++; /* add in the termination chtype */
    213  1.4     blymn 	len *= sizeof(chtype);
    214  1.4     blymn 
    215  1.1     blymn 	type = ret_byte;
    216  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &type, sizeof(int)) < 0)
    217  1.3  christos 		err(1, "%s: command pipe write for status type failed",
    218  1.3  christos 		    __func__);
    219  1.3  christos 
    220  1.3  christos 	if (write(slvpipe[WRITE_PIPE], &len, sizeof(int)) < 0)
    221  1.3  christos 		err(1, "%s: command pipe write for status length failed",
    222  1.3  christos 		    __func__);
    223  1.3  christos 
    224  1.3  christos 	if (write(slvpipe[WRITE_PIPE], string, len) < 0)
    225  1.3  christos 		err(1, "%s: command pipe write of status data failed",
    226  1.3  christos 		    __func__);
    227  1.1     blymn }
    228  1.1     blymn 
    229  1.1     blymn /*
    230  1.1     blymn  * Check the number of args we received are what we expect.  Return an
    231  1.1     blymn  * error if they do not match.
    232  1.1     blymn  */
    233  1.1     blymn int
    234  1.1     blymn check_arg_count(int nargs, int expected)
    235  1.1     blymn {
    236  1.1     blymn 	if (nargs != expected) {
    237  1.1     blymn 		report_count(1);
    238  1.1     blymn 		report_error("INCORRECT_ARGUMENT_NUMBER");
    239  1.1     blymn 		return(1);
    240  1.1     blymn 	}
    241  1.1     blymn 
    242  1.1     blymn 	return(0);
    243  1.1     blymn }
    244