Home | History | Annotate | Line # | Download | only in slave
slave.c revision 1.15
      1  1.15    rillig /*	$NetBSD: slave.c,v 1.15 2021/02/13 09:28:27 rillig Exp $	*/
      2   1.1     blymn 
      3   1.1     blymn /*-
      4   1.1     blymn  * Copyright 2009 Brett Lymn <blymn (at) NetBSD.org>
      5  1.13    rillig  * Copyright 2021 Roland Illig <rillig (at) NetBSD.org>
      6   1.1     blymn  *
      7   1.1     blymn  * All rights reserved.
      8   1.1     blymn  *
      9   1.1     blymn  * This code has been donated to The NetBSD Foundation by the Author.
     10   1.1     blymn  *
     11   1.1     blymn  * Redistribution and use in source and binary forms, with or without
     12   1.1     blymn  * modification, are permitted provided that the following conditions
     13   1.1     blymn  * are met:
     14   1.1     blymn  * 1. Redistributions of source code must retain the above copyright
     15   1.1     blymn  *    notice, this list of conditions and the following disclaimer.
     16   1.1     blymn  * 2. The name of the author may not be used to endorse or promote products
     17   1.9    rillig  *    derived from this software without specific prior written permission
     18   1.1     blymn  *
     19   1.1     blymn  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
     20   1.1     blymn  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
     21   1.1     blymn  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     22   1.1     blymn  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
     23   1.1     blymn  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
     24   1.1     blymn  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     25   1.1     blymn  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
     26   1.1     blymn  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     27   1.1     blymn  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     28   1.1     blymn  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     29   1.1     blymn  */
     30  1.13    rillig 
     31   1.1     blymn #include <fcntl.h>
     32   1.1     blymn #include <sys/ioctl.h>
     33   1.3  christos #include <unistd.h>
     34   1.3  christos #include <err.h>
     35  1.15    rillig #include <stdbool.h>
     36   1.1     blymn #include <stdio.h>
     37   1.1     blymn #include <stdlib.h>
     38   1.1     blymn #include <string.h>
     39   1.1     blymn #include <curses.h>
     40   1.1     blymn #include "returns.h"
     41   1.1     blymn #include "slave.h"
     42   1.1     blymn 
     43  1.12    rillig int from_director;
     44  1.12    rillig int to_director;
     45   1.8     blymn int initdone = 0;
     46   1.1     blymn 
     47   1.4  christos #if 0
     48   1.4  christos static const char *returns_enum_names[] = {
     49   1.1     blymn 	"unused", "numeric", "string", "byte", "ERR", "OK", "NULL", "not NULL",
     50   1.1     blymn 	"variable"
     51   1.1     blymn };
     52   1.4  christos #endif
     53   1.1     blymn 
     54  1.15    rillig static bool
     55  1.15    rillig try_read_from_director(void *data, size_t n)
     56  1.15    rillig {
     57  1.15    rillig 	ssize_t nread = read(from_director, data, n);
     58  1.15    rillig 	if (nread < 0)
     59  1.15    rillig 		err(2, "error reading from command pipe");
     60  1.15    rillig 	if (nread == 0)
     61  1.15    rillig 		return false;
     62  1.15    rillig 	if ((size_t)nread != n)
     63  1.15    rillig 		errx(2, "short read from command pipe: expected %zu, got %zu",
     64  1.15    rillig 		    n, (size_t)nread);
     65  1.15    rillig 	return true;
     66  1.15    rillig }
     67  1.15    rillig 
     68  1.15    rillig static void
     69  1.15    rillig read_from_director(void *data, size_t n)
     70  1.15    rillig {
     71  1.15    rillig 	ssize_t nread = read(from_director, data, n);
     72  1.15    rillig 	if (nread < 0)
     73  1.15    rillig 		err(2, "error reading from command pipe");
     74  1.15    rillig 	if ((size_t)nread != n)
     75  1.15    rillig 		errx(2, "short read from command pipe: expected %zu, got %zu",
     76  1.15    rillig 		    n, (size_t)nread);
     77  1.15    rillig }
     78  1.15    rillig 
     79   1.1     blymn /*
     80   1.1     blymn  * Read the command pipe for the function to execute, gather the args
     81   1.1     blymn  * and then process the command.
     82   1.1     blymn  */
     83   1.4  christos static void
     84   1.8     blymn process_commands(void)
     85   1.1     blymn {
     86   1.1     blymn 	int len, maxlen, argslen, i, ret, type;
     87   1.1     blymn 	char *cmdbuf, *tmpbuf, **args, **tmpargs;
     88   1.1     blymn 
     89   1.1     blymn 	len = maxlen = 30;
     90   1.1     blymn 	if ((cmdbuf = malloc(maxlen)) == NULL)
     91   1.1     blymn 		err(1, "slave cmdbuf malloc failed");
     92   1.1     blymn 
     93  1.15    rillig 	while (try_read_from_director(&type, sizeof type)) {
     94   1.7     blymn 		if (type != data_string)
     95   1.5  christos 			errx(1, "Unexpected type for command, got %d", type);
     96   1.1     blymn 
     97  1.15    rillig 		read_from_director(&len, sizeof len);
     98   1.1     blymn 
     99   1.1     blymn 		if ((len + 1) > maxlen) {
    100   1.1     blymn 			maxlen = len + 1;
    101   1.1     blymn 			if ((tmpbuf = realloc(cmdbuf, maxlen)) == NULL)
    102   1.1     blymn 				err(1, "slave cmdbuf realloc to %d "
    103   1.1     blymn 				    "bytes failed", maxlen);
    104   1.1     blymn 			cmdbuf = tmpbuf;
    105   1.1     blymn 		}
    106   1.1     blymn 
    107  1.15    rillig 		read_from_director(cmdbuf, len);
    108   1.1     blymn 		cmdbuf[len] = '\0';
    109   1.1     blymn 		argslen = 0;
    110   1.1     blymn 		args = NULL;
    111   1.1     blymn 
    112   1.1     blymn 		do {
    113  1.15    rillig 			read_from_director(&type, sizeof type);
    114  1.15    rillig 			read_from_director(&len, sizeof len);
    115   1.1     blymn 
    116   1.1     blymn 			if (len >= 0) {
    117   1.1     blymn 				tmpargs = realloc(args,
    118   1.1     blymn 				    (argslen + 1) * sizeof(char *));
    119   1.1     blymn 				if (tmpargs == NULL)
    120   1.1     blymn 					err(1, "slave realloc of args array "
    121   1.1     blymn 					    "failed");
    122   1.1     blymn 
    123   1.1     blymn 				args = tmpargs;
    124   1.7     blymn 				if (type != data_null) {
    125   1.2     blymn 					args[argslen] = malloc(len + 1);
    126   1.1     blymn 
    127   1.6     blymn 					if (args[argslen] == NULL)
    128   1.6     blymn 						err(1, "slave alloc of %d bytes"
    129   1.6     blymn 						    " for args failed", len);
    130   1.6     blymn 				}
    131   1.1     blymn 
    132   1.2     blymn 				if (len == 0) {
    133   1.7     blymn 					if (type == data_null)
    134   1.2     blymn 						args[argslen] = NULL;
    135   1.2     blymn 					else
    136   1.2     blymn 						args[argslen][0] = '\0';
    137   1.2     blymn 				} else {
    138  1.15    rillig 					read_from_director(args[argslen], len);
    139   1.7     blymn 					if (type != data_byte)
    140   1.1     blymn 						args[argslen][len] = '\0';
    141   1.1     blymn 
    142   1.1     blymn 					if (len == 6) {
    143   1.1     blymn 						if (strcmp(args[argslen],
    144   1.1     blymn 							   "STDSCR") == 0) {
    145   1.1     blymn 							ret = asprintf(&tmpbuf,
    146   1.4  christos 								 "%p",
    147   1.1     blymn 								 stdscr);
    148   1.1     blymn 							if (ret < 0)
    149   1.1     blymn 								err(2,
    150   1.1     blymn 								    "asprintf of stdscr failed");
    151   1.1     blymn 							free(args[argslen]);
    152   1.1     blymn 							args[argslen] = tmpbuf;
    153   1.1     blymn 						}
    154   1.1     blymn 					}
    155   1.1     blymn 				}
    156   1.1     blymn 
    157   1.1     blymn 				argslen++;
    158   1.1     blymn 			}
    159  1.11    rillig 		} while (len >= 0);
    160   1.1     blymn 
    161   1.1     blymn 		command_execute(cmdbuf, argslen, args);
    162   1.1     blymn 
    163   1.1     blymn 		if (args != NULL) {
    164   1.1     blymn 			for (i = 0; i < argslen; i++)
    165   1.1     blymn 				free(args[i]);
    166   1.1     blymn 
    167   1.1     blymn 			free(args);
    168   1.1     blymn 		}
    169   1.1     blymn 	}
    170   1.1     blymn }
    171   1.1     blymn 
    172   1.1     blymn int
    173   1.1     blymn main(int argc, char *argv[])
    174   1.1     blymn {
    175  1.12    rillig 	if (argc != 3) {
    176  1.12    rillig 		fprintf(stderr, "usage: %s <in_fd> <out_fd>\n", getprogname());
    177   1.5  christos 		return 0;
    178   1.5  christos 	}
    179  1.12    rillig 	sscanf(argv[1], "%d", &from_director);
    180  1.12    rillig 	sscanf(argv[2], "%d", &to_director);
    181   1.1     blymn 
    182   1.8     blymn 	process_commands();
    183   1.1     blymn 
    184   1.5  christos 	return 0;
    185   1.1     blymn }
    186