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