Home | History | Annotate | Line # | Download | only in slave
slave.c revision 1.2
      1  1.2  blymn /*	$NetBSD: slave.c,v 1.2 2011/04/21 10:23:50 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 #include <fcntl.h>
     32  1.1  blymn #include <sys/ioctl.h>
     33  1.1  blymn #include <stdio.h>
     34  1.1  blymn #include <stdlib.h>
     35  1.1  blymn #include <string.h>
     36  1.1  blymn #include <curses.h>
     37  1.1  blymn #include "returns.h"
     38  1.1  blymn #include "slave.h"
     39  1.1  blymn 
     40  1.1  blymn int cmdpipe[2];
     41  1.1  blymn int slvpipe[2];
     42  1.1  blymn 
     43  1.1  blymn char *returns_enum_names[] = {
     44  1.1  blymn 	"unused", "numeric", "string", "byte", "ERR", "OK", "NULL", "not NULL",
     45  1.1  blymn 	"variable"
     46  1.1  blymn };
     47  1.1  blymn 
     48  1.1  blymn /*
     49  1.1  blymn  * Read the command pipe for the function to execute, gather the args
     50  1.1  blymn  * and then process the command.
     51  1.1  blymn  */
     52  1.1  blymn void
     53  1.1  blymn process_commands(WINDOW *mainscr)
     54  1.1  blymn {
     55  1.1  blymn 	int len, maxlen, argslen, i, ret, type;
     56  1.1  blymn 	char *cmdbuf, *tmpbuf, **args, **tmpargs;
     57  1.1  blymn 	int farg;
     58  1.1  blymn 
     59  1.1  blymn 	len = maxlen = 30;
     60  1.1  blymn 	if ((cmdbuf = malloc(maxlen)) == NULL)
     61  1.1  blymn 		err(1, "slave cmdbuf malloc failed");
     62  1.1  blymn 
     63  1.1  blymn 	while(1) {
     64  1.1  blymn 		if (read(cmdpipe[READ_PIPE], &type, sizeof(int)) < 0)
     65  1.1  blymn 			err(1, "slave command type read failed");
     66  1.1  blymn 
     67  1.1  blymn 		if (type != ret_string)
     68  1.1  blymn 			err(1, "Unexpected type for command, got %d", type);
     69  1.1  blymn 
     70  1.1  blymn 		if (read(cmdpipe[READ_PIPE], &len, sizeof(int)) < 0)
     71  1.1  blymn 			err(1, "slave command len read failed");
     72  1.1  blymn 
     73  1.1  blymn 		if ((len + 1) > maxlen) {
     74  1.1  blymn 			maxlen = len + 1;
     75  1.1  blymn 			if ((tmpbuf = realloc(cmdbuf, maxlen)) == NULL)
     76  1.1  blymn 				err(1, "slave cmdbuf realloc to %d "
     77  1.1  blymn 				    "bytes failed", maxlen);
     78  1.1  blymn 			cmdbuf = tmpbuf;
     79  1.1  blymn 		}
     80  1.1  blymn 
     81  1.1  blymn 		if (read(cmdpipe[READ_PIPE], cmdbuf, len) < 0)
     82  1.1  blymn 			err(1, "slave command read failed");
     83  1.1  blymn 		cmdbuf[len] = '\0';
     84  1.1  blymn 
     85  1.1  blymn 		argslen = 0;
     86  1.1  blymn 		args = NULL;
     87  1.1  blymn 
     88  1.1  blymn 		do {
     89  1.1  blymn 			if (read(cmdpipe[READ_PIPE], &type, sizeof(int)) < 0)
     90  1.1  blymn 				err(1, "slave arg type read failed");
     91  1.1  blymn 
     92  1.1  blymn 			if (read(cmdpipe[READ_PIPE], &len, sizeof(int)) < 0)
     93  1.1  blymn 				err(1, "slave arg len read failed");
     94  1.1  blymn 
     95  1.1  blymn 			if (len >= 0) {
     96  1.1  blymn 				tmpargs = realloc(args,
     97  1.1  blymn 				    (argslen + 1) * sizeof(char *));
     98  1.1  blymn 				if (tmpargs == NULL)
     99  1.1  blymn 					err(1, "slave realloc of args array "
    100  1.1  blymn 					    "failed");
    101  1.1  blymn 
    102  1.1  blymn 				args = tmpargs;
    103  1.2  blymn 				if (type != ret_null)
    104  1.2  blymn 					args[argslen] = malloc(len + 1);
    105  1.1  blymn 
    106  1.1  blymn 				if (args[argslen] == NULL)
    107  1.1  blymn 					err(1, "slave alloc of %d bytes for"
    108  1.1  blymn 					    " args failed", len);
    109  1.1  blymn 
    110  1.2  blymn 				if (len == 0) {
    111  1.2  blymn 					if (type == ret_null)
    112  1.2  blymn 						args[argslen] = NULL;
    113  1.2  blymn 					else
    114  1.2  blymn 						args[argslen][0] = '\0';
    115  1.2  blymn 				} else {
    116  1.1  blymn 					read(cmdpipe[READ_PIPE], args[argslen],
    117  1.1  blymn 					     len);
    118  1.1  blymn 					if (type != ret_byte)
    119  1.1  blymn 						args[argslen][len] = '\0';
    120  1.1  blymn 
    121  1.1  blymn 					if (len == 6) {
    122  1.1  blymn 						if (strcmp(args[argslen],
    123  1.1  blymn 							   "STDSCR") == 0) {
    124  1.1  blymn 							ret = asprintf(&tmpbuf,
    125  1.1  blymn 								 "%td",
    126  1.1  blymn 								 stdscr);
    127  1.1  blymn 							if (ret < 0)
    128  1.1  blymn 								err(2,
    129  1.1  blymn 								    "asprintf of stdscr failed");
    130  1.1  blymn 							free(args[argslen]);
    131  1.1  blymn 							args[argslen] = tmpbuf;
    132  1.1  blymn 						}
    133  1.1  blymn 					}
    134  1.1  blymn 				}
    135  1.1  blymn 
    136  1.1  blymn 				argslen++;
    137  1.1  blymn 			}
    138  1.1  blymn 		}
    139  1.1  blymn 		while(len >= 0);
    140  1.1  blymn 
    141  1.1  blymn 
    142  1.1  blymn 		command_execute(cmdbuf, argslen, args);
    143  1.1  blymn 
    144  1.1  blymn 		if (args != NULL) {
    145  1.1  blymn 			for (i = 0; i < argslen; i++)
    146  1.1  blymn 				free(args[i]);
    147  1.1  blymn 
    148  1.1  blymn 			free(args);
    149  1.1  blymn 		}
    150  1.1  blymn 	}
    151  1.1  blymn }
    152  1.1  blymn 
    153  1.1  blymn int
    154  1.1  blymn main(int argc, char *argv[])
    155  1.1  blymn {
    156  1.1  blymn 	WINDOW *mainscr;
    157  1.1  blymn 
    158  1.1  blymn 	sscanf(argv[1], "%d", &cmdpipe[0]);
    159  1.1  blymn 	sscanf(argv[2], "%d", &cmdpipe[1]);
    160  1.1  blymn 	sscanf(argv[3], "%d", &slvpipe[0]);
    161  1.1  blymn 	sscanf(argv[4], "%d", &slvpipe[1]);
    162  1.1  blymn 
    163  1.1  blymn 	mainscr = initscr();
    164  1.1  blymn 	if (mainscr == NULL) {
    165  1.1  blymn 		fprintf(stderr, "initscr failed\n");
    166  1.1  blymn 		exit(1);
    167  1.1  blymn 	}
    168  1.1  blymn 
    169  1.1  blymn 	process_commands(mainscr);
    170  1.1  blymn 
    171  1.1  blymn 	exit(0);
    172  1.1  blymn }
    173