commands.c revision 1.5 1 1.5 blymn /* $NetBSD: commands.c,v 1.5 2019/09/19 11:31:58 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.5 blymn static void report_type(data_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.5 blymn report_type(data_err);
108 1.1 blymn else if (status == OK)
109 1.5 blymn report_type(data_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.5 blymn report_type(data_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.5 blymn type = data_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.5 blymn report_message(data_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.5 blymn report_message(data_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.5 blymn type = data_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