1 fb1a36c0 2022-01-09 op /*
   2 1e84c7b7 2022-01-10 op  * Copyright (c) 2021, 2022 Omar Polo <op@omarpolo.com>
   3 fb1a36c0 2022-01-09 op  *
   4 fb1a36c0 2022-01-09 op  * Permission to use, copy, modify, and distribute this software for any
   5 fb1a36c0 2022-01-09 op  * purpose with or without fee is hereby granted, provided that the above
   6 fb1a36c0 2022-01-09 op  * copyright notice and this permission notice appear in all copies.
   7 fb1a36c0 2022-01-09 op  *
   8 fb1a36c0 2022-01-09 op  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9 fb1a36c0 2022-01-09 op  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10 fb1a36c0 2022-01-09 op  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11 fb1a36c0 2022-01-09 op  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12 fb1a36c0 2022-01-09 op  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13 fb1a36c0 2022-01-09 op  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14 fb1a36c0 2022-01-09 op  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15 fb1a36c0 2022-01-09 op  */
  16 fb1a36c0 2022-01-09 op 
  17 bbcba3ed 2022-01-10 op #include "compat.h"
  18 bbcba3ed 2022-01-10 op 
  19 fb1a36c0 2022-01-09 op #include <sys/socket.h>
  20 fb1a36c0 2022-01-09 op #include <sys/stat.h>
  21 fb1a36c0 2022-01-09 op #include <sys/wait.h>
  22 fb1a36c0 2022-01-09 op 
  23 fb1a36c0 2022-01-09 op #include <assert.h>
  24 fb1a36c0 2022-01-09 op #include <errno.h>
  25 fb1a36c0 2022-01-09 op #include <fcntl.h>
  26 fb1a36c0 2022-01-09 op #include <inttypes.h>
  27 fb1a36c0 2022-01-09 op #include <poll.h>
  28 fb1a36c0 2022-01-09 op #include <pwd.h>
  29 fb1a36c0 2022-01-09 op #include <regex.h>
  30 fb1a36c0 2022-01-09 op #include <signal.h>
  31 fb1a36c0 2022-01-09 op #include <stdio.h>
  32 fb1a36c0 2022-01-09 op #include <stdlib.h>
  33 fb1a36c0 2022-01-09 op #include <string.h>
  34 fb1a36c0 2022-01-09 op #include <syslog.h>
  35 fb1a36c0 2022-01-09 op #include <unistd.h>
  36 fb1a36c0 2022-01-09 op 
  37 fb1a36c0 2022-01-09 op #include "client.h"
  38 fb1a36c0 2022-01-09 op #include "kami.h"
  39 fb1a36c0 2022-01-09 op #include "kamid.h"
  40 fb1a36c0 2022-01-09 op #include "log.h"
  41 fb1a36c0 2022-01-09 op #include "script.h"
  42 fb1a36c0 2022-01-09 op #include "utils.h"
  43 fb1a36c0 2022-01-09 op 
  44 fb1a36c0 2022-01-09 op #define DEBUG 0
  45 d08dbf9c 2022-01-28 op 
  46 d08dbf9c 2022-01-28 op int			 verbose;
  47 fb1a36c0 2022-01-09 op 
  48 fb1a36c0 2022-01-09 op static const char	*argv0;
  49 a0ec53a2 2022-01-09 op static const char	*dir;
  50 a0ec53a2 2022-01-09 op static uid_t		 uid;
  51 fb1a36c0 2022-01-09 op 
  52 fb1a36c0 2022-01-09 op static uint8_t		*lastmsg;
  53 fb1a36c0 2022-01-09 op 
  54 fb1a36c0 2022-01-09 op static struct imsgbuf	 ibuf;
  55 fb1a36c0 2022-01-09 op static int		 ibuf_inuse;
  56 fb1a36c0 2022-01-09 op static int		 child_out = -1;
  57 fb1a36c0 2022-01-09 op 
  58 fb1a36c0 2022-01-09 op static struct procs procs = TAILQ_HEAD_INITIALIZER(procs);
  59 fb1a36c0 2022-01-09 op static struct tests tests = TAILQ_HEAD_INITIALIZER(tests);
  60 fb1a36c0 2022-01-09 op 
  61 fb1a36c0 2022-01-09 op static int ntests;
  62 fb1a36c0 2022-01-09 op 
  63 fb1a36c0 2022-01-09 op static struct opstacks blocks = TAILQ_HEAD_INITIALIZER(blocks);
  64 fb1a36c0 2022-01-09 op static struct opstacks args   = TAILQ_HEAD_INITIALIZER(args);
  65 fb1a36c0 2022-01-09 op 
  66 fb1a36c0 2022-01-09 op #define STACK_HEIGHT 64
  67 fb1a36c0 2022-01-09 op static struct value	vstack[STACK_HEIGHT];
  68 fb1a36c0 2022-01-09 op static int		stackh;
  69 fb1a36c0 2022-01-09 op 
  70 fb1a36c0 2022-01-09 op static struct envs envs = TAILQ_HEAD_INITIALIZER(envs);
  71 fb1a36c0 2022-01-09 op 
  72 0ea030fa 2022-03-19 op static const struct value v_false = {.type = V_NUM, .v = {.num = 0}};
  73 0ea030fa 2022-03-19 op static const struct value v_true  = {.type = V_NUM, .v = {.num = 1}};
  74 fb1a36c0 2022-01-09 op 
  75 fb1a36c0 2022-01-09 op static uint8_t lasttag;
  76 fb1a36c0 2022-01-09 op 
  77 fb1a36c0 2022-01-09 op static int debug;
  78 fb1a36c0 2022-01-09 op static int syntaxcheck;
  79 fb1a36c0 2022-01-09 op 
  80 fb1a36c0 2022-01-09 op static const char	*filler;
  81 fb1a36c0 2022-01-09 op 
  82 fb1a36c0 2022-01-09 op static inline void
  83 fb1a36c0 2022-01-09 op before_printing(void)
  84 fb1a36c0 2022-01-09 op {
  85 fb1a36c0 2022-01-09 op 	if (filler != NULL) {
  86 fb1a36c0 2022-01-09 op 		printf("%s", filler);
  87 fb1a36c0 2022-01-09 op 		filler = NULL;
  88 fb1a36c0 2022-01-09 op 	}
  89 fb1a36c0 2022-01-09 op }
  90 fb1a36c0 2022-01-09 op 
  91 fb1a36c0 2022-01-09 op static inline void
  92 fb1a36c0 2022-01-09 op check_for_output(void)
  93 fb1a36c0 2022-01-09 op {
  94 fb1a36c0 2022-01-09 op 	static char	buf[BUFSIZ];
  95 fb1a36c0 2022-01-09 op 	struct pollfd	pfd;
  96 fb1a36c0 2022-01-09 op 	ssize_t		r;
  97 fb1a36c0 2022-01-09 op 
  98 fb1a36c0 2022-01-09 op 	pfd.fd = child_out;
  99 fb1a36c0 2022-01-09 op 	pfd.events = POLLIN;
 100 fb1a36c0 2022-01-09 op 	if (poll(&pfd, 1, 0) == -1)
 101 fb1a36c0 2022-01-09 op 		fatal("poll");
 102 fb1a36c0 2022-01-09 op 
 103 fb1a36c0 2022-01-09 op 	if (!(pfd.revents & POLLIN))
 104 fb1a36c0 2022-01-09 op 		return;
 105 fb1a36c0 2022-01-09 op 
 106 fb1a36c0 2022-01-09 op 	for (;;) {
 107 fb1a36c0 2022-01-09 op 		if ((r = read(child_out, buf, sizeof(buf))) == -1) {
 108 fb1a36c0 2022-01-09 op 			if (errno == EAGAIN)
 109 fb1a36c0 2022-01-09 op 				break;
 110 fb1a36c0 2022-01-09 op 			fatal("read");
 111 fb1a36c0 2022-01-09 op 		}
 112 fb1a36c0 2022-01-09 op 		if (r == 0)
 113 fb1a36c0 2022-01-09 op 			break;
 114 fb1a36c0 2022-01-09 op 		before_printing();
 115 fb1a36c0 2022-01-09 op 		fwrite(buf, 1, r, stdout);
 116 fb1a36c0 2022-01-09 op 	}
 117 fb1a36c0 2022-01-09 op }
 118 fb1a36c0 2022-01-09 op 
 119 fb1a36c0 2022-01-09 op static inline void
 120 fb1a36c0 2022-01-09 op peekn(int depth, struct value *v)
 121 fb1a36c0 2022-01-09 op {
 122 fb1a36c0 2022-01-09 op 	if (depth > stackh)
 123 fb1a36c0 2022-01-09 op 		errx(1, "can't peek the stack at %d: underflow",
 124 fb1a36c0 2022-01-09 op 		    depth);
 125 fb1a36c0 2022-01-09 op 	memcpy(v, &vstack[stackh - depth], sizeof(*v));
 126 fb1a36c0 2022-01-09 op 
 127 fb1a36c0 2022-01-09 op #if DEBUG
 128 fb1a36c0 2022-01-09 op 	printf("peeking(%d) ", depth); pp_val(v); printf("\n");
 129 fb1a36c0 2022-01-09 op #endif
 130 fb1a36c0 2022-01-09 op }
 131 fb1a36c0 2022-01-09 op 
 132 fb1a36c0 2022-01-09 op static inline void
 133 fb1a36c0 2022-01-09 op popv(struct value *v)
 134 fb1a36c0 2022-01-09 op {
 135 fb1a36c0 2022-01-09 op 	if (stackh == 0)
 136 fb1a36c0 2022-01-09 op 		errx(1, "can't pop the stack: underflow");
 137 fb1a36c0 2022-01-09 op 	memcpy(v, &vstack[--stackh], sizeof(*v));
 138 fb1a36c0 2022-01-09 op 
 139 fb1a36c0 2022-01-09 op #if DEBUG
 140 fb1a36c0 2022-01-09 op 	printf("popping "); pp_val(v); printf("\n");
 141 fb1a36c0 2022-01-09 op #endif
 142 fb1a36c0 2022-01-09 op }
 143 fb1a36c0 2022-01-09 op 
 144 fb1a36c0 2022-01-09 op static inline void
 145 fb1a36c0 2022-01-09 op popvn(int n)
 146 fb1a36c0 2022-01-09 op {
 147 fb1a36c0 2022-01-09 op 	struct value v;
 148 fb1a36c0 2022-01-09 op 
 149 fb1a36c0 2022-01-09 op 	while (n-- > 0)
 150 fb1a36c0 2022-01-09 op 		popv(&v);
 151 fb1a36c0 2022-01-09 op }
 152 fb1a36c0 2022-01-09 op 
 153 fb1a36c0 2022-01-09 op static inline void
 154 0ea030fa 2022-03-19 op pushv(const struct value *v)
 155 fb1a36c0 2022-01-09 op {
 156 fb1a36c0 2022-01-09 op 	if (stackh == STACK_HEIGHT)
 157 fb1a36c0 2022-01-09 op 		errx(1, "can't push the stack: overflow");
 158 fb1a36c0 2022-01-09 op 
 159 fb1a36c0 2022-01-09 op #if DEBUG
 160 fb1a36c0 2022-01-09 op 	printf("pushing "); pp_val(v); printf("\n");
 161 fb1a36c0 2022-01-09 op #endif
 162 fb1a36c0 2022-01-09 op 
 163 fb1a36c0 2022-01-09 op 	memcpy(&vstack[stackh++], v, sizeof(*v));
 164 fb1a36c0 2022-01-09 op }
 165 fb1a36c0 2022-01-09 op 
 166 fb1a36c0 2022-01-09 op static inline void
 167 fb1a36c0 2022-01-09 op pushbool(int n)
 168 fb1a36c0 2022-01-09 op {
 169 fb1a36c0 2022-01-09 op 	pushv(n ? &v_true : &v_false);
 170 fb1a36c0 2022-01-09 op }
 171 fb1a36c0 2022-01-09 op 
 172 fb1a36c0 2022-01-09 op static inline void
 173 fb1a36c0 2022-01-09 op pushnum(int64_t n)
 174 fb1a36c0 2022-01-09 op {
 175 fb1a36c0 2022-01-09 op 	struct value v;
 176 fb1a36c0 2022-01-09 op 
 177 fb1a36c0 2022-01-09 op 	v.type = V_NUM;
 178 fb1a36c0 2022-01-09 op 	v.v.num = n;
 179 fb1a36c0 2022-01-09 op 	pushv(&v);
 180 fb1a36c0 2022-01-09 op }
 181 fb1a36c0 2022-01-09 op 
 182 fb1a36c0 2022-01-09 op static inline struct opstack *
 183 fb1a36c0 2022-01-09 op pushstack(struct opstacks *stack)
 184 fb1a36c0 2022-01-09 op {
 185 fb1a36c0 2022-01-09 op 	struct opstack *ops;
 186 fb1a36c0 2022-01-09 op 
 187 fb1a36c0 2022-01-09 op 	ops = xcalloc(1, sizeof(*ops));
 188 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(stack, ops, entry);
 189 fb1a36c0 2022-01-09 op 	return ops;
 190 fb1a36c0 2022-01-09 op }
 191 fb1a36c0 2022-01-09 op 
 192 fb1a36c0 2022-01-09 op static inline struct opstack *
 193 fb1a36c0 2022-01-09 op peek(struct opstacks *stack)
 194 fb1a36c0 2022-01-09 op {
 195 fb1a36c0 2022-01-09 op 	if (TAILQ_EMPTY(stack))
 196 fb1a36c0 2022-01-09 op 		errx(1, "%s: args underflow", __func__);
 197 fb1a36c0 2022-01-09 op 
 198 fb1a36c0 2022-01-09 op 	return TAILQ_FIRST(stack);
 199 fb1a36c0 2022-01-09 op }
 200 fb1a36c0 2022-01-09 op 
 201 fb1a36c0 2022-01-09 op static inline struct op *
 202 fb1a36c0 2022-01-09 op finalize(struct opstacks *stack, int *argc)
 203 fb1a36c0 2022-01-09 op {
 204 fb1a36c0 2022-01-09 op 	struct opstack	*ops;
 205 fb1a36c0 2022-01-09 op 	struct op	*op;
 206 fb1a36c0 2022-01-09 op 
 207 fb1a36c0 2022-01-09 op 	if (TAILQ_EMPTY(stack))
 208 fb1a36c0 2022-01-09 op 		errx(1, "%s: args underflow", __func__);
 209 fb1a36c0 2022-01-09 op 
 210 fb1a36c0 2022-01-09 op 	ops = peek(stack);
 211 fb1a36c0 2022-01-09 op 	TAILQ_REMOVE(&args, ops, entry);
 212 fb1a36c0 2022-01-09 op 	op = ops->base.next;
 213 fb1a36c0 2022-01-09 op 
 214 fb1a36c0 2022-01-09 op 	if (argc != NULL)
 215 fb1a36c0 2022-01-09 op 		*argc = ops->counter;
 216 fb1a36c0 2022-01-09 op 
 217 fb1a36c0 2022-01-09 op 	free(ops);
 218 fb1a36c0 2022-01-09 op 	return op;
 219 fb1a36c0 2022-01-09 op }
 220 fb1a36c0 2022-01-09 op 
 221 fb1a36c0 2022-01-09 op static inline void
 222 fb1a36c0 2022-01-09 op push(struct opstacks *stack, struct op *op)
 223 fb1a36c0 2022-01-09 op {
 224 fb1a36c0 2022-01-09 op 	struct opstack *ops;
 225 fb1a36c0 2022-01-09 op 
 226 fb1a36c0 2022-01-09 op 	ops = peek(stack);
 227 fb1a36c0 2022-01-09 op 	if (ops->last == NULL) {
 228 fb1a36c0 2022-01-09 op 		ops->base.next = op;
 229 fb1a36c0 2022-01-09 op 		ops->last = op;
 230 fb1a36c0 2022-01-09 op 	} else {
 231 fb1a36c0 2022-01-09 op 		ops->last->next = op;
 232 fb1a36c0 2022-01-09 op 		ops->last = op;
 233 fb1a36c0 2022-01-09 op 	}
 234 fb1a36c0 2022-01-09 op 
 235 fb1a36c0 2022-01-09 op 	ops->counter++;
 236 fb1a36c0 2022-01-09 op }
 237 fb1a36c0 2022-01-09 op 
 238 fb1a36c0 2022-01-09 op static inline void
 239 fb1a36c0 2022-01-09 op pushenv(void)
 240 fb1a36c0 2022-01-09 op {
 241 fb1a36c0 2022-01-09 op 	struct env	*e;
 242 fb1a36c0 2022-01-09 op 
 243 fb1a36c0 2022-01-09 op 	e = xcalloc(1, sizeof(*e));
 244 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(&envs, e, entry);
 245 fb1a36c0 2022-01-09 op }
 246 fb1a36c0 2022-01-09 op 
 247 fb1a36c0 2022-01-09 op static inline struct env *
 248 fb1a36c0 2022-01-09 op currentenv(void)
 249 fb1a36c0 2022-01-09 op {
 250 fb1a36c0 2022-01-09 op 	assert(!TAILQ_EMPTY(&envs));
 251 fb1a36c0 2022-01-09 op 	return TAILQ_FIRST(&envs);
 252 fb1a36c0 2022-01-09 op }
 253 fb1a36c0 2022-01-09 op 
 254 fb1a36c0 2022-01-09 op static void
 255 fb1a36c0 2022-01-09 op popenv(void)
 256 fb1a36c0 2022-01-09 op {
 257 fb1a36c0 2022-01-09 op 	struct env	*e;
 258 fb1a36c0 2022-01-09 op 	struct binding	*b, *tb;
 259 fb1a36c0 2022-01-09 op 
 260 fb1a36c0 2022-01-09 op 	e = currentenv();
 261 fb1a36c0 2022-01-09 op 	TAILQ_REMOVE(&envs, e, entry);
 262 fb1a36c0 2022-01-09 op 
 263 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH_SAFE(b, &e->bindings, entry, tb)
 264 fb1a36c0 2022-01-09 op 		free(b);
 265 fb1a36c0 2022-01-09 op 
 266 fb1a36c0 2022-01-09 op 	free(e);
 267 fb1a36c0 2022-01-09 op }
 268 fb1a36c0 2022-01-09 op 
 269 fb1a36c0 2022-01-09 op static inline int
 270 fb1a36c0 2022-01-09 op setvar(char *sym, struct op *op)
 271 fb1a36c0 2022-01-09 op {
 272 fb1a36c0 2022-01-09 op 	struct binding	*b;
 273 fb1a36c0 2022-01-09 op 	struct env	*e;
 274 fb1a36c0 2022-01-09 op 	int		 ret, height;
 275 fb1a36c0 2022-01-09 op 
 276 fb1a36c0 2022-01-09 op 	height = stackh;
 277 fb1a36c0 2022-01-09 op 	if ((ret = eval(op)) != EVAL_OK)
 278 fb1a36c0 2022-01-09 op 		return ret;
 279 fb1a36c0 2022-01-09 op 
 280 fb1a36c0 2022-01-09 op 	if (stackh != height + 1) {
 281 fb1a36c0 2022-01-09 op 		before_printing();
 282 fb1a36c0 2022-01-09 op 		printf("trying to assign to `%s' a void value: ", sym);
 283 fb1a36c0 2022-01-09 op 		pp_op(op);
 284 fb1a36c0 2022-01-09 op 		printf("\n");
 285 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
 286 fb1a36c0 2022-01-09 op 	}
 287 fb1a36c0 2022-01-09 op 
 288 fb1a36c0 2022-01-09 op 	b = xcalloc(1, sizeof(*b));
 289 fb1a36c0 2022-01-09 op 	b->name = sym;
 290 fb1a36c0 2022-01-09 op 	popv(&b->val);
 291 fb1a36c0 2022-01-09 op 
 292 fb1a36c0 2022-01-09 op 	e = TAILQ_FIRST(&envs);
 293 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(&e->bindings, b, entry);
 294 fb1a36c0 2022-01-09 op 
 295 fb1a36c0 2022-01-09 op 	return EVAL_OK;
 296 fb1a36c0 2022-01-09 op }
 297 fb1a36c0 2022-01-09 op 
 298 fb1a36c0 2022-01-09 op static inline void
 299 fb1a36c0 2022-01-09 op setvar_raw(char *sym, struct op *op)
 300 fb1a36c0 2022-01-09 op {
 301 fb1a36c0 2022-01-09 op 	struct binding	*b;
 302 fb1a36c0 2022-01-09 op 	struct env	*e;
 303 fb1a36c0 2022-01-09 op 
 304 fb1a36c0 2022-01-09 op 	b = xcalloc(1, sizeof(*b));
 305 fb1a36c0 2022-01-09 op 	b->name = sym;
 306 fb1a36c0 2022-01-09 op 	b->raw = op;
 307 fb1a36c0 2022-01-09 op 
 308 fb1a36c0 2022-01-09 op 	e = TAILQ_FIRST(&envs);
 309 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(&e->bindings, b, entry);
 310 fb1a36c0 2022-01-09 op }
 311 fb1a36c0 2022-01-09 op 
 312 fb1a36c0 2022-01-09 op static inline int
 313 fb1a36c0 2022-01-09 op getvar(const char *sym, struct value *v)
 314 fb1a36c0 2022-01-09 op {
 315 fb1a36c0 2022-01-09 op 	struct env	*e;
 316 fb1a36c0 2022-01-09 op 	struct binding	*b;
 317 fb1a36c0 2022-01-09 op 
 318 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH(e, &envs, entry) {
 319 fb1a36c0 2022-01-09 op 		TAILQ_FOREACH(b, &e->bindings, entry) {
 320 fb1a36c0 2022-01-09 op 			if (!strcmp(sym, b->name)) {
 321 fb1a36c0 2022-01-09 op 				memcpy(v, &b->val, sizeof(*v));
 322 fb1a36c0 2022-01-09 op 				return EVAL_OK;
 323 fb1a36c0 2022-01-09 op 			}
 324 fb1a36c0 2022-01-09 op 		}
 325 fb1a36c0 2022-01-09 op 	}
 326 fb1a36c0 2022-01-09 op 
 327 fb1a36c0 2022-01-09 op 	before_printing();
 328 fb1a36c0 2022-01-09 op 	fprintf(stderr, "unbound variable %s\n", sym);
 329 fb1a36c0 2022-01-09 op 	return EVAL_ERR;
 330 fb1a36c0 2022-01-09 op }
 331 fb1a36c0 2022-01-09 op 
 332 fb1a36c0 2022-01-09 op static inline int
 333 fb1a36c0 2022-01-09 op getvar_raw(const char *sym, struct op **raw)
 334 fb1a36c0 2022-01-09 op {
 335 fb1a36c0 2022-01-09 op 	struct env	*e;
 336 fb1a36c0 2022-01-09 op 	struct binding	*b;
 337 fb1a36c0 2022-01-09 op 
 338 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH(e, &envs, entry) {
 339 fb1a36c0 2022-01-09 op 		TAILQ_FOREACH(b, &e->bindings, entry) {
 340 fb1a36c0 2022-01-09 op 			if (!strcmp(sym, b->name)) {
 341 fb1a36c0 2022-01-09 op 				*raw = b->raw;
 342 fb1a36c0 2022-01-09 op 				return EVAL_OK;
 343 fb1a36c0 2022-01-09 op 			}
 344 fb1a36c0 2022-01-09 op 		}
 345 fb1a36c0 2022-01-09 op 	}
 346 fb1a36c0 2022-01-09 op 
 347 fb1a36c0 2022-01-09 op 	return EVAL_ERR;
 348 fb1a36c0 2022-01-09 op }
 349 fb1a36c0 2022-01-09 op 
 350 fb1a36c0 2022-01-09 op int
 351 fb1a36c0 2022-01-09 op global_set(char *sym, struct op *op)
 352 fb1a36c0 2022-01-09 op {
 353 fb1a36c0 2022-01-09 op 	struct binding	*b;
 354 fb1a36c0 2022-01-09 op 	struct env	*e;
 355 fb1a36c0 2022-01-09 op 
 356 fb1a36c0 2022-01-09 op 	/* TODO: check for duplicates */
 357 fb1a36c0 2022-01-09 op 
 358 fb1a36c0 2022-01-09 op 	if (op->type != OP_LITERAL &&
 359 fb1a36c0 2022-01-09 op 	    (op->type == OP_CAST && op->v.cast.expr->type != OP_LITERAL))
 360 fb1a36c0 2022-01-09 op 		return 0;
 361 fb1a36c0 2022-01-09 op 
 362 fb1a36c0 2022-01-09 op 	b = xcalloc(1, sizeof(*b));
 363 fb1a36c0 2022-01-09 op 	b->name = sym;
 364 fb1a36c0 2022-01-09 op 
 365 fb1a36c0 2022-01-09 op 	/* it's only a cast on a literal! */
 366 fb1a36c0 2022-01-09 op 	if (op->type == OP_CAST) {
 367 fb1a36c0 2022-01-09 op 		if (eval(op) != EVAL_OK) {
 368 fb1a36c0 2022-01-09 op 			free(b);
 369 fb1a36c0 2022-01-09 op 			return 0;
 370 fb1a36c0 2022-01-09 op 		}
 371 fb1a36c0 2022-01-09 op 		popv(&b->val);
 372 fb1a36c0 2022-01-09 op 	} else
 373 fb1a36c0 2022-01-09 op 		memcpy(&b->val, &op->v.literal, sizeof(b->val));
 374 fb1a36c0 2022-01-09 op 
 375 fb1a36c0 2022-01-09 op 	e = TAILQ_LAST(&envs, envs);
 376 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(&e->bindings, b, entry);
 377 fb1a36c0 2022-01-09 op 
 378 fb1a36c0 2022-01-09 op 	return 1;
 379 fb1a36c0 2022-01-09 op }
 380 fb1a36c0 2022-01-09 op 
 381 fb1a36c0 2022-01-09 op struct op *
 382 fb1a36c0 2022-01-09 op newop(int type)
 383 fb1a36c0 2022-01-09 op {
 384 fb1a36c0 2022-01-09 op 	struct op *op;
 385 fb1a36c0 2022-01-09 op 
 386 fb1a36c0 2022-01-09 op 	op = xcalloc(1, sizeof(*op));
 387 fb1a36c0 2022-01-09 op 	op->type = type;
 388 fb1a36c0 2022-01-09 op 
 389 fb1a36c0 2022-01-09 op 	return op;
 390 fb1a36c0 2022-01-09 op }
 391 fb1a36c0 2022-01-09 op 
 392 fb1a36c0 2022-01-09 op void
 393 fb1a36c0 2022-01-09 op free_op_rec(struct op *op)
 394 fb1a36c0 2022-01-09 op {
 395 fb1a36c0 2022-01-09 op 	struct op *n;
 396 fb1a36c0 2022-01-09 op 
 397 fb1a36c0 2022-01-09 op 	while (op != NULL) {
 398 fb1a36c0 2022-01-09 op 		n = op->next;
 399 fb1a36c0 2022-01-09 op 		free_op(op);
 400 fb1a36c0 2022-01-09 op 		op = n;
 401 fb1a36c0 2022-01-09 op 	}
 402 fb1a36c0 2022-01-09 op }
 403 fb1a36c0 2022-01-09 op 
 404 fb1a36c0 2022-01-09 op void
 405 fb1a36c0 2022-01-09 op free_op(struct op *op)
 406 fb1a36c0 2022-01-09 op {
 407 fb1a36c0 2022-01-09 op 	if (op == NULL)
 408 fb1a36c0 2022-01-09 op 		return;
 409 fb1a36c0 2022-01-09 op 
 410 fb1a36c0 2022-01-09 op 	switch (op->type) {
 411 fb1a36c0 2022-01-09 op 	case OP_REST:
 412 fb1a36c0 2022-01-09 op 	case OP_LITERAL:
 413 fb1a36c0 2022-01-09 op 	case OP_VARGS:
 414 fb1a36c0 2022-01-09 op 		break;
 415 fb1a36c0 2022-01-09 op 	case OP_ASSIGN:
 416 fb1a36c0 2022-01-09 op 		free(op->v.assign.name);
 417 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.assign.expr);
 418 fb1a36c0 2022-01-09 op 		break;
 419 fb1a36c0 2022-01-09 op 	case OP_ASSERT:
 420 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.assert);
 421 fb1a36c0 2022-01-09 op 		break;
 422 fb1a36c0 2022-01-09 op 	case OP_FUNCALL:
 423 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.funcall.argv);
 424 fb1a36c0 2022-01-09 op 		break;
 425 fb1a36c0 2022-01-09 op 	case OP_VAR:
 426 fb1a36c0 2022-01-09 op 		free(op->v.var);
 427 fb1a36c0 2022-01-09 op 		break;
 428 fb1a36c0 2022-01-09 op 	case OP_CAST:
 429 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.cast.expr);
 430 fb1a36c0 2022-01-09 op 		break;
 431 fb1a36c0 2022-01-09 op 	case OP_CMP_EQ:
 432 fb1a36c0 2022-01-09 op 	case OP_CMP_LEQ:
 433 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.bin_cmp.a);
 434 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.bin_cmp.b);
 435 fb1a36c0 2022-01-09 op 		break;
 436 fb1a36c0 2022-01-09 op 	case OP_FACCESS:
 437 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.faccess.expr);
 438 fb1a36c0 2022-01-09 op 		free(op->v.faccess.field);
 439 fb1a36c0 2022-01-09 op 		break;
 440 fb1a36c0 2022-01-09 op 	case OP_SFAIL:
 441 fb1a36c0 2022-01-09 op 		free(op->v.sfail.msg);
 442 fb1a36c0 2022-01-09 op 		free_op_rec(op->v.sfail.expr);
 443 fb1a36c0 2022-01-09 op 		break;
 444 fb1a36c0 2022-01-09 op 	default:
 445 fb1a36c0 2022-01-09 op 		/* unreachable */
 446 fb1a36c0 2022-01-09 op 		abort();
 447 fb1a36c0 2022-01-09 op 	}
 448 fb1a36c0 2022-01-09 op 
 449 fb1a36c0 2022-01-09 op 	free(op);
 450 fb1a36c0 2022-01-09 op }
 451 fb1a36c0 2022-01-09 op 
 452 fb1a36c0 2022-01-09 op struct op *
 453 fb1a36c0 2022-01-09 op op_rest(void)
 454 fb1a36c0 2022-01-09 op {
 455 fb1a36c0 2022-01-09 op 	return newop(OP_REST);
 456 fb1a36c0 2022-01-09 op }
 457 fb1a36c0 2022-01-09 op 
 458 fb1a36c0 2022-01-09 op struct op *
 459 fb1a36c0 2022-01-09 op op_assign(char *sym, struct op *expr)
 460 fb1a36c0 2022-01-09 op {
 461 fb1a36c0 2022-01-09 op 	struct op *op;
 462 fb1a36c0 2022-01-09 op 
 463 fb1a36c0 2022-01-09 op 	op = newop(OP_ASSIGN);
 464 fb1a36c0 2022-01-09 op 	op->v.assign.name = sym;
 465 fb1a36c0 2022-01-09 op 	op->v.assign.expr = expr;
 466 fb1a36c0 2022-01-09 op 
 467 fb1a36c0 2022-01-09 op 	return op;
 468 fb1a36c0 2022-01-09 op }
 469 fb1a36c0 2022-01-09 op 
 470 fb1a36c0 2022-01-09 op struct op *
 471 fb1a36c0 2022-01-09 op op_assert(struct op *expr)
 472 fb1a36c0 2022-01-09 op {
 473 fb1a36c0 2022-01-09 op 	struct op *op;
 474 fb1a36c0 2022-01-09 op 
 475 fb1a36c0 2022-01-09 op 	op = newop(OP_ASSERT);
 476 fb1a36c0 2022-01-09 op 	op->v.assert = expr;
 477 fb1a36c0 2022-01-09 op 
 478 fb1a36c0 2022-01-09 op 	return op;
 479 fb1a36c0 2022-01-09 op }
 480 fb1a36c0 2022-01-09 op 
 481 fb1a36c0 2022-01-09 op struct op *
 482 fb1a36c0 2022-01-09 op op_var(char *sym)
 483 fb1a36c0 2022-01-09 op {
 484 fb1a36c0 2022-01-09 op 	struct op *op;
 485 fb1a36c0 2022-01-09 op 
 486 fb1a36c0 2022-01-09 op 	op = newop(OP_VAR);
 487 fb1a36c0 2022-01-09 op 	op->v.var = sym;
 488 fb1a36c0 2022-01-09 op 
 489 fb1a36c0 2022-01-09 op 	return op;
 490 fb1a36c0 2022-01-09 op }
 491 fb1a36c0 2022-01-09 op 
 492 fb1a36c0 2022-01-09 op struct op *
 493 fb1a36c0 2022-01-09 op op_lit_str(char *str)
 494 fb1a36c0 2022-01-09 op {
 495 fb1a36c0 2022-01-09 op 	struct op *op;
 496 fb1a36c0 2022-01-09 op 
 497 fb1a36c0 2022-01-09 op 	op = newop(OP_LITERAL);
 498 fb1a36c0 2022-01-09 op 	op->v.literal.type = V_STR;
 499 fb1a36c0 2022-01-09 op 	op->v.literal.v.str = str;
 500 fb1a36c0 2022-01-09 op 
 501 fb1a36c0 2022-01-09 op 	return op;
 502 fb1a36c0 2022-01-09 op }
 503 fb1a36c0 2022-01-09 op 
 504 fb1a36c0 2022-01-09 op struct op *
 505 fb1a36c0 2022-01-09 op op_lit_num(uint64_t n)
 506 fb1a36c0 2022-01-09 op {
 507 fb1a36c0 2022-01-09 op 	struct op *op;
 508 fb1a36c0 2022-01-09 op 
 509 fb1a36c0 2022-01-09 op 	op = newop(OP_LITERAL);
 510 fb1a36c0 2022-01-09 op 	op->v.literal.type = V_NUM;
 511 fb1a36c0 2022-01-09 op 	op->v.literal.v.num = n;
 512 fb1a36c0 2022-01-09 op 
 513 fb1a36c0 2022-01-09 op 	return op;
 514 fb1a36c0 2022-01-09 op }
 515 fb1a36c0 2022-01-09 op 
 516 fb1a36c0 2022-01-09 op struct op *
 517 fb1a36c0 2022-01-09 op op_cmp_eq(struct op *a, struct op *b)
 518 fb1a36c0 2022-01-09 op {
 519 fb1a36c0 2022-01-09 op 	struct op *op;
 520 fb1a36c0 2022-01-09 op 
 521 fb1a36c0 2022-01-09 op 	op = newop(OP_CMP_EQ);
 522 fb1a36c0 2022-01-09 op 	op->v.bin_cmp.a = a;
 523 fb1a36c0 2022-01-09 op 	op->v.bin_cmp.b = b;
 524 fb1a36c0 2022-01-09 op 
 525 fb1a36c0 2022-01-09 op 	return op;
 526 fb1a36c0 2022-01-09 op }
 527 fb1a36c0 2022-01-09 op 
 528 fb1a36c0 2022-01-09 op struct op *
 529 fb1a36c0 2022-01-09 op op_cmp_leq(struct op *a, struct op *b)
 530 fb1a36c0 2022-01-09 op {
 531 fb1a36c0 2022-01-09 op 	struct op *op;
 532 fb1a36c0 2022-01-09 op 
 533 fb1a36c0 2022-01-09 op 	op = newop(OP_CMP_LEQ);
 534 fb1a36c0 2022-01-09 op 	op->v.bin_cmp.a = a;
 535 fb1a36c0 2022-01-09 op 	op->v.bin_cmp.b = b;
 536 fb1a36c0 2022-01-09 op 
 537 fb1a36c0 2022-01-09 op 	return op;
 538 fb1a36c0 2022-01-09 op }
 539 fb1a36c0 2022-01-09 op 
 540 fb1a36c0 2022-01-09 op struct op *
 541 fb1a36c0 2022-01-09 op op_cast(struct op *expr, int totype)
 542 fb1a36c0 2022-01-09 op {
 543 fb1a36c0 2022-01-09 op 	struct op *op;
 544 fb1a36c0 2022-01-09 op 
 545 fb1a36c0 2022-01-09 op 	op = newop(OP_CAST);
 546 fb1a36c0 2022-01-09 op 	op->v.cast.expr = expr;
 547 fb1a36c0 2022-01-09 op 	op->v.cast.totype = totype;
 548 fb1a36c0 2022-01-09 op 
 549 fb1a36c0 2022-01-09 op 	return op;
 550 fb1a36c0 2022-01-09 op }
 551 fb1a36c0 2022-01-09 op 
 552 fb1a36c0 2022-01-09 op struct op *
 553 fb1a36c0 2022-01-09 op op_faccess(struct op *expr, char *field)
 554 fb1a36c0 2022-01-09 op {
 555 fb1a36c0 2022-01-09 op 	struct op *op;
 556 fb1a36c0 2022-01-09 op 
 557 fb1a36c0 2022-01-09 op 	op = newop(OP_FACCESS);
 558 fb1a36c0 2022-01-09 op 	op->v.faccess.expr = expr;
 559 fb1a36c0 2022-01-09 op 	op->v.faccess.field = field;
 560 fb1a36c0 2022-01-09 op 
 561 fb1a36c0 2022-01-09 op 	return op;
 562 fb1a36c0 2022-01-09 op }
 563 fb1a36c0 2022-01-09 op 
 564 fb1a36c0 2022-01-09 op struct op *
 565 fb1a36c0 2022-01-09 op op_sfail(struct op *expr, char *msg)
 566 fb1a36c0 2022-01-09 op {
 567 fb1a36c0 2022-01-09 op 	struct op	*op;
 568 fb1a36c0 2022-01-09 op 
 569 fb1a36c0 2022-01-09 op 	op = newop(OP_SFAIL);
 570 fb1a36c0 2022-01-09 op 	op->v.sfail.expr = expr;
 571 fb1a36c0 2022-01-09 op 	op->v.sfail.msg = msg;
 572 fb1a36c0 2022-01-09 op 
 573 fb1a36c0 2022-01-09 op 	return op;
 574 fb1a36c0 2022-01-09 op }
 575 fb1a36c0 2022-01-09 op 
 576 fb1a36c0 2022-01-09 op struct op *
 577 fb1a36c0 2022-01-09 op op_vargs(void)
 578 fb1a36c0 2022-01-09 op {
 579 fb1a36c0 2022-01-09 op 	struct op	*op;
 580 fb1a36c0 2022-01-09 op 
 581 fb1a36c0 2022-01-09 op 	op = newop(OP_VARGS);
 582 fb1a36c0 2022-01-09 op 
 583 fb1a36c0 2022-01-09 op 	return op;
 584 fb1a36c0 2022-01-09 op }
 585 fb1a36c0 2022-01-09 op 
 586 fb1a36c0 2022-01-09 op void
 587 9ded84e7 2022-05-10 op ppf_val(FILE *f, const struct value *val)
 588 fb1a36c0 2022-01-09 op {
 589 fb1a36c0 2022-01-09 op 	size_t	i;
 590 fb1a36c0 2022-01-09 op 
 591 fb1a36c0 2022-01-09 op 	switch (val->type) {
 592 fb1a36c0 2022-01-09 op 	case V_SYM:
 593 fb1a36c0 2022-01-09 op 		fprintf(f, "%s", val->v.str);
 594 fb1a36c0 2022-01-09 op 		break;
 595 fb1a36c0 2022-01-09 op 	case V_STR:
 596 fb1a36c0 2022-01-09 op 		fprintf(f, "\"%s\"", val->v.str);
 597 fb1a36c0 2022-01-09 op 		break;
 598 fb1a36c0 2022-01-09 op 	case V_NUM:
 599 fb1a36c0 2022-01-09 op 		fprintf(f, "%"PRIi64, val->v.num);
 600 fb1a36c0 2022-01-09 op 		break;
 601 fb1a36c0 2022-01-09 op 	case V_U8:
 602 fb1a36c0 2022-01-09 op 		fprintf(f, "%"PRIu8, val->v.u8);
 603 fb1a36c0 2022-01-09 op 		break;
 604 fb1a36c0 2022-01-09 op 	case V_U16:
 605 fb1a36c0 2022-01-09 op 		fprintf(f, "%"PRIu16, val->v.u16);
 606 fb1a36c0 2022-01-09 op 		break;
 607 fb1a36c0 2022-01-09 op 	case V_U32:
 608 fb1a36c0 2022-01-09 op 		fprintf(f, "%"PRIu32, val->v.u32);
 609 fb1a36c0 2022-01-09 op 		break;
 610 fb1a36c0 2022-01-09 op 	case V_MSG:
 611 fb1a36c0 2022-01-09 op 		fprintf(f, "(");
 612 fb1a36c0 2022-01-09 op 		for (i = 0; i < val->v.msg.len; ++i)
 613 fb1a36c0 2022-01-09 op 			fprintf(f, "%x%s", val->v.msg.msg[i],
 614 fb1a36c0 2022-01-09 op 			    i == val->v.msg.len-1 ? "" : " ");
 615 fb1a36c0 2022-01-09 op 		fprintf(f, ")");
 616 fb1a36c0 2022-01-09 op 		break;
 617 fb1a36c0 2022-01-09 op 	case V_QIDVEC:
 618 fb1a36c0 2022-01-09 op 		fprintf(f, "qids[n=%zu]", val->v.qidvec.len);
 619 fb1a36c0 2022-01-09 op 		break;
 620 fb1a36c0 2022-01-09 op 	default:
 621 fb1a36c0 2022-01-09 op 		fprintf(f, "<unknown value>");
 622 fb1a36c0 2022-01-09 op 		break;
 623 fb1a36c0 2022-01-09 op 	}
 624 fb1a36c0 2022-01-09 op }
 625 fb1a36c0 2022-01-09 op 
 626 fb1a36c0 2022-01-09 op void
 627 9ded84e7 2022-05-10 op pp_val(const struct value *val)
 628 fb1a36c0 2022-01-09 op {
 629 fb1a36c0 2022-01-09 op 	ppf_val(stdout, val);
 630 fb1a36c0 2022-01-09 op }
 631 fb1a36c0 2022-01-09 op 
 632 fb1a36c0 2022-01-09 op const char *
 633 fb1a36c0 2022-01-09 op val_type(struct value *v)
 634 fb1a36c0 2022-01-09 op {
 635 fb1a36c0 2022-01-09 op 	switch (v->type) {
 636 fb1a36c0 2022-01-09 op 	case V_SYM: return "symbol";
 637 fb1a36c0 2022-01-09 op 	case V_STR: return "string";
 638 fb1a36c0 2022-01-09 op 	case V_NUM: return "number";
 639 fb1a36c0 2022-01-09 op 	case V_MSG: return "message";
 640 fb1a36c0 2022-01-09 op 	case V_QID: return "qid";
 641 fb1a36c0 2022-01-09 op 	case V_U8: return "u8";
 642 fb1a36c0 2022-01-09 op 	case V_U16: return "u16";
 643 fb1a36c0 2022-01-09 op 	case V_U32: return "u32";
 644 fb1a36c0 2022-01-09 op 	default: return "unknown";
 645 fb1a36c0 2022-01-09 op 	}
 646 fb1a36c0 2022-01-09 op }
 647 fb1a36c0 2022-01-09 op 
 648 fb1a36c0 2022-01-09 op int
 649 fb1a36c0 2022-01-09 op val_trueish(struct value *a)
 650 fb1a36c0 2022-01-09 op {
 651 fb1a36c0 2022-01-09 op 	if (val_isnum(a))
 652 fb1a36c0 2022-01-09 op 		return val_tonum(a);
 653 fb1a36c0 2022-01-09 op 	return 1;
 654 fb1a36c0 2022-01-09 op }
 655 fb1a36c0 2022-01-09 op 
 656 fb1a36c0 2022-01-09 op int
 657 fb1a36c0 2022-01-09 op val_isnum(struct value *a)
 658 fb1a36c0 2022-01-09 op {
 659 fb1a36c0 2022-01-09 op 	return a->type == V_NUM
 660 fb1a36c0 2022-01-09 op 		|| a->type == V_U8
 661 fb1a36c0 2022-01-09 op 		|| a->type == V_U16
 662 fb1a36c0 2022-01-09 op 		|| a->type == V_U32;
 663 fb1a36c0 2022-01-09 op }
 664 fb1a36c0 2022-01-09 op 
 665 fb1a36c0 2022-01-09 op int64_t
 666 fb1a36c0 2022-01-09 op val_tonum(struct value *a)
 667 fb1a36c0 2022-01-09 op {
 668 fb1a36c0 2022-01-09 op 	switch (a->type) {
 669 fb1a36c0 2022-01-09 op 	case V_NUM: return a->v.num;
 670 fb1a36c0 2022-01-09 op 	case V_U8:  return a->v.u8;
 671 fb1a36c0 2022-01-09 op 	case V_U16: return a->v.u16;
 672 fb1a36c0 2022-01-09 op 	case V_U32: return a->v.u32;
 673 fb1a36c0 2022-01-09 op 	default:
 674 fb1a36c0 2022-01-09 op 		before_printing();
 675 fb1a36c0 2022-01-09 op 		fprintf(stderr, "%s: given value is not a number\n", __func__);
 676 fb1a36c0 2022-01-09 op 		abort();
 677 fb1a36c0 2022-01-09 op 	}
 678 fb1a36c0 2022-01-09 op }
 679 fb1a36c0 2022-01-09 op 
 680 fb1a36c0 2022-01-09 op int
 681 fb1a36c0 2022-01-09 op val_eq(struct value *a, struct value *b)
 682 fb1a36c0 2022-01-09 op {
 683 fb1a36c0 2022-01-09 op 	if (val_isnum(a) && val_isnum(b))
 684 fb1a36c0 2022-01-09 op 		return val_tonum(a) == val_tonum(b);
 685 fb1a36c0 2022-01-09 op 
 686 fb1a36c0 2022-01-09 op 	if (a->type != b->type)
 687 fb1a36c0 2022-01-09 op 		return 0;
 688 fb1a36c0 2022-01-09 op 
 689 fb1a36c0 2022-01-09 op 	switch (a->type) {
 690 fb1a36c0 2022-01-09 op 	case V_STR:
 691 fb1a36c0 2022-01-09 op 	case V_SYM:
 692 fb1a36c0 2022-01-09 op 		return !strcmp(a->v.str, b->v.str);
 693 fb1a36c0 2022-01-09 op 	}
 694 fb1a36c0 2022-01-09 op 
 695 fb1a36c0 2022-01-09 op 	return 0;
 696 fb1a36c0 2022-01-09 op }
 697 fb1a36c0 2022-01-09 op 
 698 fb1a36c0 2022-01-09 op int
 699 fb1a36c0 2022-01-09 op val_leq(struct value *a, struct value *b)
 700 fb1a36c0 2022-01-09 op {
 701 fb1a36c0 2022-01-09 op 	if (val_isnum(a) && val_isnum(b))
 702 fb1a36c0 2022-01-09 op 		return val_tonum(a) <= val_tonum(b);
 703 fb1a36c0 2022-01-09 op 	return 0;
 704 fb1a36c0 2022-01-09 op }
 705 fb1a36c0 2022-01-09 op 
 706 fb1a36c0 2022-01-09 op static inline const char *
 707 fb1a36c0 2022-01-09 op pp_totype(int totype)
 708 fb1a36c0 2022-01-09 op {
 709 fb1a36c0 2022-01-09 op 	/*
 710 fb1a36c0 2022-01-09 op 	 * Not all of these are valid cast type thought, including
 711 fb1a36c0 2022-01-09 op 	 * every possibility only to aid debugging.
 712 fb1a36c0 2022-01-09 op 	 */
 713 fb1a36c0 2022-01-09 op 	switch (totype) {
 714 fb1a36c0 2022-01-09 op 	case V_STR: return "str";
 715 fb1a36c0 2022-01-09 op 	case V_SYM: return "sym";
 716 fb1a36c0 2022-01-09 op 	case V_NUM: return "num";
 717 fb1a36c0 2022-01-09 op 	case V_QID: return "qid";
 718 fb1a36c0 2022-01-09 op 	case V_U8:  return "u8";
 719 fb1a36c0 2022-01-09 op 	case V_U16: return "u16";
 720 fb1a36c0 2022-01-09 op 	case V_U32: return "u32";
 721 fb1a36c0 2022-01-09 op 	default:    return "unknown";
 722 fb1a36c0 2022-01-09 op 	}
 723 fb1a36c0 2022-01-09 op }
 724 fb1a36c0 2022-01-09 op 
 725 fb1a36c0 2022-01-09 op int
 726 fb1a36c0 2022-01-09 op val_cast(struct value *a, int totype)
 727 fb1a36c0 2022-01-09 op {
 728 fb1a36c0 2022-01-09 op 	int64_t v;
 729 fb1a36c0 2022-01-09 op 
 730 fb1a36c0 2022-01-09 op #define NUMCAST(val, t, c, totype, max) do {				\
 731 fb1a36c0 2022-01-09 op 		if (val > max) {					\
 732 fb1a36c0 2022-01-09 op 			before_printing();				\
 733 fb1a36c0 2022-01-09 op 			fprintf(stderr, "can't cast %"PRIu64		\
 734 fb1a36c0 2022-01-09 op 			    " to %s\n", val, pp_totype(totype));	\
 735 fb1a36c0 2022-01-09 op 			return EVAL_ERR;				\
 736 fb1a36c0 2022-01-09 op 		}							\
 737 fb1a36c0 2022-01-09 op 		a->type = totype;					\
 738 fb1a36c0 2022-01-09 op 		a->v.t = (c)val;					\
 739 fb1a36c0 2022-01-09 op 		return EVAL_OK;						\
 740 fb1a36c0 2022-01-09 op 	} while (0)
 741 fb1a36c0 2022-01-09 op 
 742 fb1a36c0 2022-01-09 op 	if (a->type == totype)
 743 fb1a36c0 2022-01-09 op 		return EVAL_OK;
 744 fb1a36c0 2022-01-09 op 
 745 fb1a36c0 2022-01-09 op 	if (!val_isnum(a)) {
 746 fb1a36c0 2022-01-09 op 		before_printing();
 747 fb1a36c0 2022-01-09 op 		fprintf(stderr, "can't cast ");
 748 fb1a36c0 2022-01-09 op 		ppf_val(stderr, a);
 749 fb1a36c0 2022-01-09 op 		fprintf(stderr, " to type %s\n", pp_totype(totype));
 750 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
 751 fb1a36c0 2022-01-09 op 	}
 752 fb1a36c0 2022-01-09 op 
 753 fb1a36c0 2022-01-09 op 	v = a->v.num;
 754 fb1a36c0 2022-01-09 op 	switch (totype) {
 755 fb1a36c0 2022-01-09 op 	case V_U8:  NUMCAST(v, u8,  uint8_t,  totype, UINT8_MAX);
 756 fb1a36c0 2022-01-09 op 	case V_U16: NUMCAST(v, u16, uint16_t, totype, UINT16_MAX);
 757 fb1a36c0 2022-01-09 op 	case V_U32: NUMCAST(v, u32, uint32_t, totype, UINT32_MAX);
 758 fb1a36c0 2022-01-09 op 	default:
 759 fb1a36c0 2022-01-09 op 		before_printing();
 760 fb1a36c0 2022-01-09 op 		fprintf(stderr, "can't cast %"PRIu64" to %s\n",
 761 fb1a36c0 2022-01-09 op 		    v, pp_totype(totype));
 762 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
 763 fb1a36c0 2022-01-09 op 	}
 764 fb1a36c0 2022-01-09 op 
 765 fb1a36c0 2022-01-09 op #undef NUMCAST
 766 fb1a36c0 2022-01-09 op }
 767 fb1a36c0 2022-01-09 op 
 768 fb1a36c0 2022-01-09 op int
 769 fb1a36c0 2022-01-09 op val_faccess(struct value *a, const char *field, struct value *ret)
 770 fb1a36c0 2022-01-09 op {
 771 fb1a36c0 2022-01-09 op 	uint8_t		 mtype;
 772 fb1a36c0 2022-01-09 op 	uint16_t	 len;
 773 fb1a36c0 2022-01-09 op 	const char	*errstr;
 774 fb1a36c0 2022-01-09 op 
 775 fb1a36c0 2022-01-09 op #define MSGTYPE(m) *(m.msg + 4)	/* skip the length */
 776 fb1a36c0 2022-01-09 op 
 777 fb1a36c0 2022-01-09 op 	switch (a->type) {
 778 fb1a36c0 2022-01-09 op 	case V_QID:
 779 fb1a36c0 2022-01-09 op 		/* TODO: add path.  needs uint64_t values thought! */
 780 fb1a36c0 2022-01-09 op 		if (!strcmp(field, "vers")) {
 781 fb1a36c0 2022-01-09 op 			ret->type = V_U32;
 782 fb1a36c0 2022-01-09 op 			memcpy(&ret->v.u32, a->v.qid+1, 4);
 783 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 784 fb1a36c0 2022-01-09 op 		} else if (!strcmp(field, "type")) {
 785 fb1a36c0 2022-01-09 op 			ret->type = V_U8;
 786 fb1a36c0 2022-01-09 op 			ret->v.u8 = *a->v.qid;
 787 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 788 fb1a36c0 2022-01-09 op 		}
 789 fb1a36c0 2022-01-09 op 		break;
 790 fb1a36c0 2022-01-09 op 
 791 fb1a36c0 2022-01-09 op 	case V_MSG:
 792 fb1a36c0 2022-01-09 op 		mtype = MSGTYPE(a->v.msg);
 793 fb1a36c0 2022-01-09 op 		if (!strcmp(field, "type")) {
 794 fb1a36c0 2022-01-09 op 			ret->type = V_U8;
 795 fb1a36c0 2022-01-09 op 			ret->v.u8 = MSGTYPE(a->v.msg);
 796 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 797 fb1a36c0 2022-01-09 op 		} else if (!strcmp(field, "tag")) {
 798 fb1a36c0 2022-01-09 op 			ret->type = V_U16;
 799 fb1a36c0 2022-01-09 op                         memcpy(&ret->v.u16, &a->v.msg.msg[5], 2);
 800 fb1a36c0 2022-01-09 op 			ret->v.u16 = le16toh(ret->v.u16);
 801 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 802 fb1a36c0 2022-01-09 op 		} else if (!strcmp(field, "msize") && mtype == Rversion) {
 803 fb1a36c0 2022-01-09 op 			ret->type = V_U32;
 804 fb1a36c0 2022-01-09 op 			memcpy(&ret->v.u32, &a->v.msg.msg[7], 4);
 805 fb1a36c0 2022-01-09 op 			ret->v.u32 = le32toh(ret->v.u32);
 806 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 807 fb1a36c0 2022-01-09 op 		} else if (!strcmp(field, "qid") && mtype == Rattach) {
 808 fb1a36c0 2022-01-09 op 			ret->type = V_QID;
 809 fb1a36c0 2022-01-09 op 			memcpy(&ret->v.qid, &a->v.msg.msg[7], QIDSIZE);
 810 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 811 fb1a36c0 2022-01-09 op 		} else if (!strcmp(field, "nwqid") && mtype == Rwalk) {
 812 fb1a36c0 2022-01-09 op 			ret->type = V_U16;
 813 fb1a36c0 2022-01-09 op 			memcpy(&ret->v.u16, &a->v.msg.msg[7], 2);
 814 fb1a36c0 2022-01-09 op 			ret->v.u16 = le16toh(ret->v.u16);
 815 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 816 fb1a36c0 2022-01-09 op 		} else if (!strcmp(field, "wqid") && mtype == Rwalk) {
 817 fb1a36c0 2022-01-09 op 			ret->type = V_QIDVEC;
 818 fb1a36c0 2022-01-09 op 			ret->v.qidvec.start = &a->v.msg.msg[9];
 819 fb1a36c0 2022-01-09 op 			memcpy(&len, &a->v.msg.msg[7], 2);
 820 fb1a36c0 2022-01-09 op 			len = le16toh(len);
 821 fb1a36c0 2022-01-09 op 			ret->v.qidvec.len = len;
 822 fb1a36c0 2022-01-09 op 			return EVAL_OK;
 823 fb1a36c0 2022-01-09 op 		}
 824 fb1a36c0 2022-01-09 op 		break;
 825 fb1a36c0 2022-01-09 op 
 826 fb1a36c0 2022-01-09 op 	case V_QIDVEC:
 827 fb1a36c0 2022-01-09 op 		len = strtonum(field, 0, MAXWELEM, &errstr);
 828 fb1a36c0 2022-01-09 op 		if (errstr != NULL) {
 829 fb1a36c0 2022-01-09 op 			before_printing();
 830 fb1a36c0 2022-01-09 op 			printf("can't access qid #%s: %s\n", field, errstr);
 831 fb1a36c0 2022-01-09 op 			return EVAL_ERR;
 832 fb1a36c0 2022-01-09 op 		}
 833 fb1a36c0 2022-01-09 op 
 834 fb1a36c0 2022-01-09 op 		if (len >= a->v.qidvec.len) {
 835 fb1a36c0 2022-01-09 op 			before_printing();
 836 fb1a36c0 2022-01-09 op 			printf("can't access qid #%d: out-of-bound "
 837 fb1a36c0 2022-01-09 op 			    "(max %zu)\n", len, a->v.qidvec.len);
 838 fb1a36c0 2022-01-09 op 			return EVAL_ERR;
 839 fb1a36c0 2022-01-09 op 		}
 840 fb1a36c0 2022-01-09 op 
 841 fb1a36c0 2022-01-09 op 		ret->type = V_QID;
 842 fb1a36c0 2022-01-09 op 		memcpy(&ret->v.qid, a->v.qidvec.start + len * QIDSIZE,
 843 fb1a36c0 2022-01-09 op 		    QIDSIZE);
 844 fb1a36c0 2022-01-09 op 
 845 fb1a36c0 2022-01-09 op                 return EVAL_OK;
 846 fb1a36c0 2022-01-09 op 
 847 fb1a36c0 2022-01-09 op 	default:
 848 fb1a36c0 2022-01-09 op 		break;
 849 fb1a36c0 2022-01-09 op 	}
 850 fb1a36c0 2022-01-09 op 
 851 fb1a36c0 2022-01-09 op 	before_printing();
 852 fb1a36c0 2022-01-09 op 	printf("can't access field `%s' on type %s (", field, val_type(a));
 853 fb1a36c0 2022-01-09 op 	pp_val(a);
 854 fb1a36c0 2022-01-09 op 	printf(")\n");
 855 fb1a36c0 2022-01-09 op 	return EVAL_ERR;
 856 fb1a36c0 2022-01-09 op 
 857 fb1a36c0 2022-01-09 op #undef MSGTYPE
 858 fb1a36c0 2022-01-09 op }
 859 fb1a36c0 2022-01-09 op 
 860 fb1a36c0 2022-01-09 op void
 861 fb1a36c0 2022-01-09 op pp_op(struct op *op)
 862 fb1a36c0 2022-01-09 op {
 863 fb1a36c0 2022-01-09 op 	struct op	*aux;
 864 fb1a36c0 2022-01-09 op 
 865 fb1a36c0 2022-01-09 op 	switch (op->type) {
 866 fb1a36c0 2022-01-09 op 	case OP_REST:
 867 fb1a36c0 2022-01-09 op 		printf("...");
 868 fb1a36c0 2022-01-09 op 		break;
 869 fb1a36c0 2022-01-09 op 	case OP_ASSIGN:
 870 fb1a36c0 2022-01-09 op 		printf("%s = ", op->v.assign.name);
 871 fb1a36c0 2022-01-09 op                 pp_op(op->v.assign.expr);
 872 fb1a36c0 2022-01-09 op 		break;
 873 fb1a36c0 2022-01-09 op 	case OP_ASSERT:
 874 fb1a36c0 2022-01-09 op 		printf("assert ");
 875 fb1a36c0 2022-01-09 op 		pp_op(op->v.assert);
 876 fb1a36c0 2022-01-09 op 		break;
 877 fb1a36c0 2022-01-09 op 	case OP_FUNCALL:
 878 fb1a36c0 2022-01-09 op 		printf("funcall %s(", op->v.funcall.proc->name);
 879 fb1a36c0 2022-01-09 op 		for (aux = op->v.funcall.argv; aux != NULL; aux = aux->next) {
 880 fb1a36c0 2022-01-09 op 			pp_op(aux);
 881 fb1a36c0 2022-01-09 op 			if (aux->next != NULL)
 882 fb1a36c0 2022-01-09 op 				printf(", ");
 883 fb1a36c0 2022-01-09 op 		}
 884 fb1a36c0 2022-01-09 op 		printf(")");
 885 fb1a36c0 2022-01-09 op 		break;
 886 fb1a36c0 2022-01-09 op 	case OP_LITERAL:
 887 fb1a36c0 2022-01-09 op                 pp_val(&op->v.literal);
 888 fb1a36c0 2022-01-09 op 		break;
 889 fb1a36c0 2022-01-09 op 	case OP_VAR:
 890 fb1a36c0 2022-01-09 op 		printf("%s", op->v.var);
 891 fb1a36c0 2022-01-09 op 		break;
 892 fb1a36c0 2022-01-09 op 	case OP_CAST:
 893 fb1a36c0 2022-01-09 op 		pp_op(op->v.cast.expr);
 894 fb1a36c0 2022-01-09 op 		printf(":");
 895 fb1a36c0 2022-01-09 op 		switch (op->v.cast.totype) {
 896 fb1a36c0 2022-01-09 op 		case V_U8: printf("u8"); break;
 897 fb1a36c0 2022-01-09 op 		case V_U16: printf("u16"); break;
 898 fb1a36c0 2022-01-09 op 		case V_U32: printf("u32"); break;
 899 fb1a36c0 2022-01-09 op 		case V_STR: printf("str"); break;
 900 fb1a36c0 2022-01-09 op 		default: printf("???"); break;
 901 fb1a36c0 2022-01-09 op 		}
 902 fb1a36c0 2022-01-09 op 		break;
 903 fb1a36c0 2022-01-09 op 	case OP_CMP_EQ:
 904 fb1a36c0 2022-01-09 op 		pp_op(op->v.bin_cmp.a);
 905 fb1a36c0 2022-01-09 op 		printf(" == ");
 906 fb1a36c0 2022-01-09 op 		pp_op(op->v.bin_cmp.b);
 907 fb1a36c0 2022-01-09 op 		break;
 908 fb1a36c0 2022-01-09 op 	case OP_CMP_LEQ:
 909 fb1a36c0 2022-01-09 op 		pp_op(op->v.bin_cmp.a);
 910 fb1a36c0 2022-01-09 op 		printf(" <= ");
 911 fb1a36c0 2022-01-09 op 		pp_op(op->v.bin_cmp.b);
 912 fb1a36c0 2022-01-09 op 		break;
 913 fb1a36c0 2022-01-09 op 	case OP_FACCESS:
 914 fb1a36c0 2022-01-09 op 		pp_op(op->v.faccess.expr);
 915 fb1a36c0 2022-01-09 op 		printf(".%s", op->v.faccess.field);
 916 fb1a36c0 2022-01-09 op 		break;
 917 fb1a36c0 2022-01-09 op 	case OP_SFAIL:
 918 fb1a36c0 2022-01-09 op 		printf("should-fail ");
 919 fb1a36c0 2022-01-09 op 		pp_op(op->v.sfail.expr);
 920 fb1a36c0 2022-01-09 op 		if (op->v.sfail.msg != NULL)
 921 fb1a36c0 2022-01-09 op 			printf(": \"%s\"", op->v.sfail.msg);
 922 fb1a36c0 2022-01-09 op 		break;
 923 fb1a36c0 2022-01-09 op 	case OP_VARGS:
 924 fb1a36c0 2022-01-09 op 		printf("vargs");
 925 fb1a36c0 2022-01-09 op 		break;
 926 fb1a36c0 2022-01-09 op 	default:
 927 fb1a36c0 2022-01-09 op 		printf(" ???[%d] ", op->type);
 928 fb1a36c0 2022-01-09 op 	}
 929 fb1a36c0 2022-01-09 op }
 930 fb1a36c0 2022-01-09 op 
 931 fb1a36c0 2022-01-09 op void
 932 fb1a36c0 2022-01-09 op pp_block(struct op *op)
 933 fb1a36c0 2022-01-09 op {
 934 fb1a36c0 2022-01-09 op         while (op != NULL) {
 935 fb1a36c0 2022-01-09 op 		printf("> ");
 936 fb1a36c0 2022-01-09 op 		pp_op(op);
 937 fb1a36c0 2022-01-09 op 		printf("\n");
 938 fb1a36c0 2022-01-09 op 
 939 fb1a36c0 2022-01-09 op 		op = op->next;
 940 fb1a36c0 2022-01-09 op 	}
 941 fb1a36c0 2022-01-09 op }
 942 fb1a36c0 2022-01-09 op 
 943 fb1a36c0 2022-01-09 op int
 944 fb1a36c0 2022-01-09 op eval(struct op *op)
 945 fb1a36c0 2022-01-09 op {
 946 fb1a36c0 2022-01-09 op 	struct value	 a, b;
 947 fb1a36c0 2022-01-09 op 	struct proc	*proc;
 948 fb1a36c0 2022-01-09 op 	struct op	*t, *tnext;
 949 fb1a36c0 2022-01-09 op 	int		 i, ret;
 950 fb1a36c0 2022-01-09 op 
 951 fb1a36c0 2022-01-09 op #if DEBUG
 952 fb1a36c0 2022-01-09 op 	pp_op(op);
 953 fb1a36c0 2022-01-09 op 	printf("\n");
 954 fb1a36c0 2022-01-09 op #endif
 955 fb1a36c0 2022-01-09 op 
 956 fb1a36c0 2022-01-09 op 	switch (op->type) {
 957 fb1a36c0 2022-01-09 op 	case OP_REST:
 958 fb1a36c0 2022-01-09 op 		/*
 959 fb1a36c0 2022-01-09 op 		 * Try to load the rest argument.  Note that it can be
 960 fb1a36c0 2022-01-09 op 		 * empty!
 961 fb1a36c0 2022-01-09 op 		 */
 962 fb1a36c0 2022-01-09 op                 if ((ret = getvar_raw("...", &t)) == EVAL_OK)
 963 fb1a36c0 2022-01-09 op 			if ((ret = eval(t)) != EVAL_OK)
 964 fb1a36c0 2022-01-09 op 				return ret;
 965 fb1a36c0 2022-01-09 op 		break;
 966 fb1a36c0 2022-01-09 op 
 967 fb1a36c0 2022-01-09 op 	case OP_ASSIGN:
 968 fb1a36c0 2022-01-09 op 		ret = setvar(op->v.assign.name, op->v.assign.expr);
 969 fb1a36c0 2022-01-09 op 		if (ret != EVAL_OK)
 970 fb1a36c0 2022-01-09 op 			return ret;
 971 fb1a36c0 2022-01-09 op 		break;
 972 fb1a36c0 2022-01-09 op 
 973 fb1a36c0 2022-01-09 op 	case OP_ASSERT:
 974 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.assert)) != EVAL_OK)
 975 fb1a36c0 2022-01-09 op 			return ret;
 976 fb1a36c0 2022-01-09 op                 popv(&a);
 977 fb1a36c0 2022-01-09 op                 if (!val_trueish(&a)) {
 978 fb1a36c0 2022-01-09 op 			before_printing();
 979 fb1a36c0 2022-01-09 op 			printf("assertion failed: ");
 980 fb1a36c0 2022-01-09 op 			pp_op(op->v.assert);
 981 fb1a36c0 2022-01-09 op 			printf("\n");
 982 fb1a36c0 2022-01-09 op 			return EVAL_ERR;
 983 fb1a36c0 2022-01-09 op 		}
 984 fb1a36c0 2022-01-09 op 		break;
 985 fb1a36c0 2022-01-09 op 
 986 fb1a36c0 2022-01-09 op 	case OP_FUNCALL:
 987 fb1a36c0 2022-01-09 op 		/* assume airity matches */
 988 fb1a36c0 2022-01-09 op 
 989 fb1a36c0 2022-01-09 op 		proc = op->v.funcall.proc;
 990 fb1a36c0 2022-01-09 op 		if (proc->nativefn != NULL) {
 991 fb1a36c0 2022-01-09 op 			/*
 992 fb1a36c0 2022-01-09 op 			 * Push arguments on the stack for builtin
 993 fb1a36c0 2022-01-09 op 			 * functions.  Counting the height of the
 994 fb1a36c0 2022-01-09 op 			 * stack is done to compute the correct number
 995 fb1a36c0 2022-01-09 op 			 * in the vararg case.  argc only counts the
 996 fb1a36c0 2022-01-09 op 			 * "syntactical" arguments, i.e. foo(x, ...)
 997 fb1a36c0 2022-01-09 op 			 * has argc == 2, but at runtime argc may be
 998 fb1a36c0 2022-01-09 op 			 * 1, 2 or a greater number!
 999 fb1a36c0 2022-01-09 op 			 */
1000 fb1a36c0 2022-01-09 op 
1001 fb1a36c0 2022-01-09 op 			i = stackh;
1002 fb1a36c0 2022-01-09 op 			t = op->v.funcall.argv;
1003 fb1a36c0 2022-01-09 op 			if (t != NULL && (ret = eval(t)) != EVAL_OK)
1004 fb1a36c0 2022-01-09 op 				return ret;
1005 fb1a36c0 2022-01-09 op 			i = stackh - i;
1006 fb1a36c0 2022-01-09 op 
1007 fb1a36c0 2022-01-09 op 			assert(i >= 0);
1008 fb1a36c0 2022-01-09 op 
1009 fb1a36c0 2022-01-09 op 			if ((ret = proc->nativefn(i))
1010 fb1a36c0 2022-01-09 op 			    != EVAL_OK)
1011 fb1a36c0 2022-01-09 op 				return ret;
1012 fb1a36c0 2022-01-09 op 		} else {
1013 fb1a36c0 2022-01-09 op 			if (proc->body == NULL) {
1014 fb1a36c0 2022-01-09 op 				before_printing();
1015 fb1a36c0 2022-01-09 op 				printf("warn: calling the empty proc `%s'\n",
1016 fb1a36c0 2022-01-09 op 				    proc->name);
1017 fb1a36c0 2022-01-09 op 				break;
1018 fb1a36c0 2022-01-09 op 			}
1019 fb1a36c0 2022-01-09 op 
1020 fb1a36c0 2022-01-09 op 			pushenv();
1021 fb1a36c0 2022-01-09 op 
1022 fb1a36c0 2022-01-09 op 			for (t = op->v.funcall.argv, i = 0;
1023 fb1a36c0 2022-01-09 op 			     t != NULL;
1024 fb1a36c0 2022-01-09 op 			     t = t->next, i++) {
1025 fb1a36c0 2022-01-09 op 				/*
1026 fb1a36c0 2022-01-09 op 				 * Push a pseudo variable `...' (and
1027 fb1a36c0 2022-01-09 op 				 * don't evaluate it) in the vararg
1028 fb1a36c0 2022-01-09 op 				 * case.  A special case is when the
1029 fb1a36c0 2022-01-09 op 				 * variable is itself `...'.
1030 fb1a36c0 2022-01-09 op 				 */
1031 fb1a36c0 2022-01-09 op 				if (proc->vararg && i == proc->minargs) {
1032 fb1a36c0 2022-01-09 op 					if (t->type != OP_REST)
1033 fb1a36c0 2022-01-09 op 						setvar_raw(xstrdup("..."), t);
1034 fb1a36c0 2022-01-09 op 					break;
1035 fb1a36c0 2022-01-09 op 				}
1036 fb1a36c0 2022-01-09 op 
1037 fb1a36c0 2022-01-09 op 				/*
1038 fb1a36c0 2022-01-09 op 				 * The arguments are a linked list of
1039 fb1a36c0 2022-01-09 op 				 * ops.  Setvar will call eval that
1040 fb1a36c0 2022-01-09 op 				 * will evaluate *all* the arguments.
1041 fb1a36c0 2022-01-09 op 				 * The dance here that sets next to
1042 fb1a36c0 2022-01-09 op 				 * NULL and then restores it is to
1043 fb1a36c0 2022-01-09 op 				 * avoid this behaviour.
1044 fb1a36c0 2022-01-09 op 				 */
1045 fb1a36c0 2022-01-09 op 				tnext = t->next;
1046 fb1a36c0 2022-01-09 op 				t->next = NULL;
1047 fb1a36c0 2022-01-09 op 				ret = setvar(proc->args[i], t);
1048 fb1a36c0 2022-01-09 op 				t->next = tnext;
1049 fb1a36c0 2022-01-09 op 
1050 fb1a36c0 2022-01-09 op 				if (ret != EVAL_OK)
1051 fb1a36c0 2022-01-09 op 					return ret;
1052 fb1a36c0 2022-01-09 op 			}
1053 fb1a36c0 2022-01-09 op 
1054 fb1a36c0 2022-01-09 op 			if ((ret = eval(proc->body)) != EVAL_OK)
1055 fb1a36c0 2022-01-09 op 				return ret;
1056 fb1a36c0 2022-01-09 op 
1057 fb1a36c0 2022-01-09 op 			popenv();
1058 fb1a36c0 2022-01-09 op 		}
1059 fb1a36c0 2022-01-09 op 
1060 fb1a36c0 2022-01-09 op 		break;
1061 fb1a36c0 2022-01-09 op 
1062 fb1a36c0 2022-01-09 op 	case OP_LITERAL:
1063 fb1a36c0 2022-01-09 op 		pushv(&op->v.literal);
1064 fb1a36c0 2022-01-09 op 		break;
1065 fb1a36c0 2022-01-09 op 
1066 fb1a36c0 2022-01-09 op 	case OP_VAR:
1067 fb1a36c0 2022-01-09 op                 if ((ret = getvar(op->v.var, &a)) != EVAL_OK)
1068 fb1a36c0 2022-01-09 op 			return ret;
1069 fb1a36c0 2022-01-09 op 		pushv(&a);
1070 fb1a36c0 2022-01-09 op 		break;
1071 fb1a36c0 2022-01-09 op 
1072 fb1a36c0 2022-01-09 op 	case OP_CAST:
1073 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.cast.expr)) != EVAL_OK)
1074 fb1a36c0 2022-01-09 op 			return ret;
1075 fb1a36c0 2022-01-09 op 		popv(&a);
1076 fb1a36c0 2022-01-09 op 		if ((ret = val_cast(&a, op->v.cast.totype)) != EVAL_OK)
1077 fb1a36c0 2022-01-09 op 			return ret;
1078 fb1a36c0 2022-01-09 op 		pushv(&a);
1079 fb1a36c0 2022-01-09 op 		break;
1080 fb1a36c0 2022-01-09 op 
1081 fb1a36c0 2022-01-09 op 	case OP_CMP_EQ:
1082 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.bin_cmp.a)) != EVAL_OK)
1083 fb1a36c0 2022-01-09 op 			return ret;
1084 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.bin_cmp.b)) != EVAL_OK)
1085 fb1a36c0 2022-01-09 op 			return ret;
1086 fb1a36c0 2022-01-09 op 
1087 fb1a36c0 2022-01-09 op 		popv(&b);
1088 fb1a36c0 2022-01-09 op 		popv(&a);
1089 fb1a36c0 2022-01-09 op 		pushbool(val_eq(&a, &b));
1090 fb1a36c0 2022-01-09 op 		break;
1091 fb1a36c0 2022-01-09 op 
1092 fb1a36c0 2022-01-09 op 	case OP_CMP_LEQ:
1093 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.bin_cmp.a)) != EVAL_OK)
1094 fb1a36c0 2022-01-09 op 			return ret;
1095 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.bin_cmp.b)) != EVAL_OK)
1096 fb1a36c0 2022-01-09 op 			return ret;
1097 fb1a36c0 2022-01-09 op 
1098 fb1a36c0 2022-01-09 op 		popv(&b);
1099 fb1a36c0 2022-01-09 op 		popv(&a);
1100 fb1a36c0 2022-01-09 op 		pushbool(val_leq(&a, &b));
1101 fb1a36c0 2022-01-09 op 		break;
1102 fb1a36c0 2022-01-09 op 
1103 fb1a36c0 2022-01-09 op 	case OP_FACCESS:
1104 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.faccess.expr)) != EVAL_OK)
1105 fb1a36c0 2022-01-09 op 			return ret;
1106 fb1a36c0 2022-01-09 op 		popv(&a);
1107 fb1a36c0 2022-01-09 op 		if ((ret = val_faccess(&a, op->v.faccess.field, &b))
1108 fb1a36c0 2022-01-09 op 		    != EVAL_OK)
1109 fb1a36c0 2022-01-09 op 			return ret;
1110 fb1a36c0 2022-01-09 op 		pushv(&b);
1111 fb1a36c0 2022-01-09 op 		break;
1112 fb1a36c0 2022-01-09 op 
1113 fb1a36c0 2022-01-09 op 	case OP_SFAIL:
1114 fb1a36c0 2022-01-09 op 		if ((ret = eval(op->v.sfail.expr)) == EVAL_OK) {
1115 fb1a36c0 2022-01-09 op 			before_printing();
1116 fb1a36c0 2022-01-09 op 			printf("expecting failure");
1117 fb1a36c0 2022-01-09 op 			if (op->v.sfail.msg != NULL)
1118 fb1a36c0 2022-01-09 op 				printf(" \"%s\"", op->v.sfail.msg);
1119 fb1a36c0 2022-01-09 op 			printf("\n");
1120 fb1a36c0 2022-01-09 op 			printf("expression: ");
1121 fb1a36c0 2022-01-09 op 			pp_op(op->v.sfail.expr);
1122 fb1a36c0 2022-01-09 op 			printf("\n");
1123 fb1a36c0 2022-01-09 op 			return EVAL_ERR;
1124 fb1a36c0 2022-01-09 op 		}
1125 fb1a36c0 2022-01-09 op 		if (ret == EVAL_SKIP)
1126 fb1a36c0 2022-01-09 op 			return ret;
1127 fb1a36c0 2022-01-09 op 		break;
1128 fb1a36c0 2022-01-09 op 
1129 fb1a36c0 2022-01-09 op 	case OP_VARGS:
1130 fb1a36c0 2022-01-09 op 		if ((ret = getvar_raw("...", &t)) == EVAL_OK) {
1131 fb1a36c0 2022-01-09 op                         for (i = 0; t != NULL; t = t->next)
1132 fb1a36c0 2022-01-09 op 				i++;
1133 fb1a36c0 2022-01-09 op 			pushnum(i);
1134 fb1a36c0 2022-01-09 op 		} else
1135 fb1a36c0 2022-01-09 op 			pushnum(0);
1136 fb1a36c0 2022-01-09 op 		break;
1137 fb1a36c0 2022-01-09 op 
1138 fb1a36c0 2022-01-09 op 	default:
1139 fb1a36c0 2022-01-09 op 		before_printing();
1140 fb1a36c0 2022-01-09 op 		fprintf(stderr, "invalid op, aborting.\n");
1141 fb1a36c0 2022-01-09 op 		abort();
1142 fb1a36c0 2022-01-09 op 	}
1143 fb1a36c0 2022-01-09 op 
1144 fb1a36c0 2022-01-09 op 	if (op->next)
1145 fb1a36c0 2022-01-09 op 		return eval(op->next);
1146 fb1a36c0 2022-01-09 op 	return EVAL_OK;
1147 fb1a36c0 2022-01-09 op }
1148 fb1a36c0 2022-01-09 op 
1149 fb1a36c0 2022-01-09 op void
1150 fb1a36c0 2022-01-09 op prepare_funcall(void)
1151 fb1a36c0 2022-01-09 op {
1152 fb1a36c0 2022-01-09 op 	pushstack(&args);
1153 fb1a36c0 2022-01-09 op }
1154 fb1a36c0 2022-01-09 op 
1155 fb1a36c0 2022-01-09 op void
1156 fb1a36c0 2022-01-09 op push_arg(struct op *op)
1157 fb1a36c0 2022-01-09 op {
1158 fb1a36c0 2022-01-09 op 	push(&args, op);
1159 fb1a36c0 2022-01-09 op }
1160 fb1a36c0 2022-01-09 op 
1161 fb1a36c0 2022-01-09 op struct op *
1162 fb1a36c0 2022-01-09 op op_funcall(struct proc *proc)
1163 fb1a36c0 2022-01-09 op {
1164 fb1a36c0 2022-01-09 op 	struct op	*op, *argv;
1165 fb1a36c0 2022-01-09 op 	int		 argc;
1166 fb1a36c0 2022-01-09 op 
1167 fb1a36c0 2022-01-09 op 	argv = finalize(&args, &argc);
1168 fb1a36c0 2022-01-09 op 
1169 fb1a36c0 2022-01-09 op 	op = newop(OP_FUNCALL);
1170 fb1a36c0 2022-01-09 op 	op->v.funcall.proc = proc;
1171 fb1a36c0 2022-01-09 op 	op->v.funcall.argv = argv;
1172 fb1a36c0 2022-01-09 op 	op->v.funcall.argc = argc;
1173 fb1a36c0 2022-01-09 op 
1174 fb1a36c0 2022-01-09 op 	return op;
1175 fb1a36c0 2022-01-09 op }
1176 fb1a36c0 2022-01-09 op 
1177 fb1a36c0 2022-01-09 op void
1178 fb1a36c0 2022-01-09 op add_builtin_proc(const char *name, int (*fn)(int), int argc, int vararg)
1179 fb1a36c0 2022-01-09 op {
1180 fb1a36c0 2022-01-09 op 	struct proc *proc;
1181 fb1a36c0 2022-01-09 op 
1182 fb1a36c0 2022-01-09 op 	proc = xcalloc(1, sizeof(*proc));
1183 fb1a36c0 2022-01-09 op 	proc->name = xstrdup(name);
1184 fb1a36c0 2022-01-09 op 	proc->nativefn = fn;
1185 fb1a36c0 2022-01-09 op 	proc->minargs = argc;
1186 fb1a36c0 2022-01-09 op 	proc->vararg = vararg;
1187 fb1a36c0 2022-01-09 op 
1188 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(&procs, proc, entry);
1189 fb1a36c0 2022-01-09 op }
1190 fb1a36c0 2022-01-09 op 
1191 fb1a36c0 2022-01-09 op void
1192 fb1a36c0 2022-01-09 op prepare_proc(void)
1193 fb1a36c0 2022-01-09 op {
1194 fb1a36c0 2022-01-09 op 	pushstack(&args);
1195 fb1a36c0 2022-01-09 op }
1196 fb1a36c0 2022-01-09 op 
1197 fb1a36c0 2022-01-09 op int
1198 fb1a36c0 2022-01-09 op proc_setup_body(void)
1199 fb1a36c0 2022-01-09 op {
1200 fb1a36c0 2022-01-09 op 	struct opstack	*argv;
1201 fb1a36c0 2022-01-09 op 	struct op	*op;
1202 fb1a36c0 2022-01-09 op 	int		 i;
1203 fb1a36c0 2022-01-09 op 
1204 fb1a36c0 2022-01-09 op 	argv = peek(&args);
1205 fb1a36c0 2022-01-09 op 	for (i = 0, op = argv->base.next; op != NULL; i++) {
1206 fb1a36c0 2022-01-09 op 		/*
1207 fb1a36c0 2022-01-09 op 		 * TODO: should free the whole list on error but..,
1208 fb1a36c0 2022-01-09 op 		 * we're gonna exit real soon(tm)!
1209 fb1a36c0 2022-01-09 op 		 */
1210 fb1a36c0 2022-01-09 op 		if (op->type != OP_VAR && op->type != OP_REST)
1211 fb1a36c0 2022-01-09 op 			return 0;
1212 fb1a36c0 2022-01-09 op 
1213 fb1a36c0 2022-01-09 op 		op = op->next;
1214 fb1a36c0 2022-01-09 op 	}
1215 fb1a36c0 2022-01-09 op 
1216 fb1a36c0 2022-01-09 op 	assert(i == argv->counter);
1217 fb1a36c0 2022-01-09 op 	pushstack(&blocks);
1218 fb1a36c0 2022-01-09 op 	return 1;
1219 fb1a36c0 2022-01-09 op }
1220 fb1a36c0 2022-01-09 op 
1221 fb1a36c0 2022-01-09 op void
1222 fb1a36c0 2022-01-09 op proc_done(char *name)
1223 fb1a36c0 2022-01-09 op {
1224 fb1a36c0 2022-01-09 op 	struct proc	*proc;
1225 fb1a36c0 2022-01-09 op 	struct op	*op, *next, *argv, *body;
1226 fb1a36c0 2022-01-09 op 	int		 i, argc;
1227 fb1a36c0 2022-01-09 op 
1228 fb1a36c0 2022-01-09 op 	argv = finalize(&args, &argc);
1229 fb1a36c0 2022-01-09 op 	body = finalize(&blocks, NULL);
1230 fb1a36c0 2022-01-09 op 
1231 fb1a36c0 2022-01-09 op 	proc = xcalloc(1, sizeof(*proc));
1232 fb1a36c0 2022-01-09 op 	proc->name = name;
1233 fb1a36c0 2022-01-09 op 	proc->minargs = argc;
1234 fb1a36c0 2022-01-09 op 
1235 fb1a36c0 2022-01-09 op         for (i = 0, op = argv; op != NULL; ++i) {
1236 fb1a36c0 2022-01-09 op 		if (op->type == OP_REST) {
1237 fb1a36c0 2022-01-09 op 			proc->vararg = 1;
1238 fb1a36c0 2022-01-09 op 			proc->minargs = i;
1239 fb1a36c0 2022-01-09 op 			break;
1240 fb1a36c0 2022-01-09 op 		}
1241 fb1a36c0 2022-01-09 op 
1242 fb1a36c0 2022-01-09 op 		proc->args[i] = xstrdup(op->v.var);
1243 fb1a36c0 2022-01-09 op 
1244 fb1a36c0 2022-01-09 op 		next = op->next;
1245 fb1a36c0 2022-01-09 op 		free_op(op);
1246 fb1a36c0 2022-01-09 op 		op = next;
1247 fb1a36c0 2022-01-09 op 	}
1248 fb1a36c0 2022-01-09 op 	assert(i == argc || (proc->vararg && i == proc->minargs));
1249 fb1a36c0 2022-01-09 op 
1250 fb1a36c0 2022-01-09 op 	proc->body = body;
1251 fb1a36c0 2022-01-09 op 
1252 fb1a36c0 2022-01-09 op 	TAILQ_INSERT_HEAD(&procs, proc, entry);
1253 fb1a36c0 2022-01-09 op }
1254 fb1a36c0 2022-01-09 op 
1255 fb1a36c0 2022-01-09 op void
1256 fb1a36c0 2022-01-09 op block_push(struct op *op)
1257 fb1a36c0 2022-01-09 op {
1258 fb1a36c0 2022-01-09 op 	push(&blocks, op);
1259 fb1a36c0 2022-01-09 op }
1260 fb1a36c0 2022-01-09 op 
1261 fb1a36c0 2022-01-09 op struct proc *
1262 fb1a36c0 2022-01-09 op proc_by_name(const char *name)
1263 fb1a36c0 2022-01-09 op {
1264 fb1a36c0 2022-01-09 op 	struct proc *p;
1265 fb1a36c0 2022-01-09 op 
1266 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH(p, &procs, entry) {
1267 fb1a36c0 2022-01-09 op 		if (!strcmp(p->name, name))
1268 fb1a36c0 2022-01-09 op 			return p;
1269 fb1a36c0 2022-01-09 op 	}
1270 fb1a36c0 2022-01-09 op 
1271 fb1a36c0 2022-01-09 op 	return NULL;
1272 fb1a36c0 2022-01-09 op }
1273 fb1a36c0 2022-01-09 op 
1274 fb1a36c0 2022-01-09 op void
1275 fb1a36c0 2022-01-09 op prepare_test(void)
1276 fb1a36c0 2022-01-09 op {
1277 fb1a36c0 2022-01-09 op 	pushstack(&blocks);
1278 fb1a36c0 2022-01-09 op }
1279 fb1a36c0 2022-01-09 op 
1280 fb1a36c0 2022-01-09 op void
1281 a0ec53a2 2022-01-09 op test_done(int shouldfail, char *name)
1282 fb1a36c0 2022-01-09 op {
1283 fb1a36c0 2022-01-09 op 	struct test	*test;
1284 fb1a36c0 2022-01-09 op 
1285 fb1a36c0 2022-01-09 op 	test = xcalloc(1, sizeof(*test));
1286 fb1a36c0 2022-01-09 op 	test->shouldfail = shouldfail;
1287 fb1a36c0 2022-01-09 op 	test->name = name;
1288 fb1a36c0 2022-01-09 op 	test->body = finalize(&blocks, NULL);
1289 fb1a36c0 2022-01-09 op 
1290 7c94ab42 2022-01-10 op 	TAILQ_INSERT_TAIL(&tests, test, entry);
1291 fb1a36c0 2022-01-09 op 
1292 fb1a36c0 2022-01-09 op 	ntests++;
1293 fb1a36c0 2022-01-09 op }
1294 fb1a36c0 2022-01-09 op 
1295 fb1a36c0 2022-01-09 op static int
1296 fb1a36c0 2022-01-09 op builtin_print(int argc)
1297 fb1a36c0 2022-01-09 op {
1298 fb1a36c0 2022-01-09 op 	struct value	v;
1299 fb1a36c0 2022-01-09 op 	int		i;
1300 fb1a36c0 2022-01-09 op 
1301 fb1a36c0 2022-01-09 op 	before_printing();
1302 fb1a36c0 2022-01-09 op 
1303 fb1a36c0 2022-01-09 op 	for (i = argc; i > 0; --i) {
1304 fb1a36c0 2022-01-09 op 		peekn(i, &v);
1305 fb1a36c0 2022-01-09 op 		if (v.type == V_STR)
1306 fb1a36c0 2022-01-09 op 			printf("%s", v.v.str);
1307 fb1a36c0 2022-01-09 op 		else
1308 fb1a36c0 2022-01-09 op 			pp_val(&v);
1309 fb1a36c0 2022-01-09 op 		printf(" ");
1310 fb1a36c0 2022-01-09 op 	}
1311 fb1a36c0 2022-01-09 op 
1312 fb1a36c0 2022-01-09 op 	printf("\n");
1313 fb1a36c0 2022-01-09 op 
1314 fb1a36c0 2022-01-09 op 	popvn(argc);
1315 fb1a36c0 2022-01-09 op 
1316 fb1a36c0 2022-01-09 op 	return EVAL_OK;
1317 fb1a36c0 2022-01-09 op }
1318 fb1a36c0 2022-01-09 op 
1319 fb1a36c0 2022-01-09 op static int
1320 fb1a36c0 2022-01-09 op builtin_debug(int argc)
1321 fb1a36c0 2022-01-09 op {
1322 fb1a36c0 2022-01-09 op 	if (debug)
1323 fb1a36c0 2022-01-09 op 		return builtin_print(argc);
1324 fb1a36c0 2022-01-09 op 
1325 fb1a36c0 2022-01-09 op 	popvn(argc);
1326 fb1a36c0 2022-01-09 op 	return EVAL_OK;
1327 fb1a36c0 2022-01-09 op }
1328 fb1a36c0 2022-01-09 op 
1329 fb1a36c0 2022-01-09 op static int
1330 fb1a36c0 2022-01-09 op builtin_skip(int argc)
1331 fb1a36c0 2022-01-09 op {
1332 fb1a36c0 2022-01-09 op 	return EVAL_SKIP;
1333 fb1a36c0 2022-01-09 op }
1334 fb1a36c0 2022-01-09 op 
1335 fb1a36c0 2022-01-09 op static int
1336 fb1a36c0 2022-01-09 op builtin_iota(int argc)
1337 fb1a36c0 2022-01-09 op {
1338 fb1a36c0 2022-01-09 op 	struct value v;
1339 fb1a36c0 2022-01-09 op 
1340 fb1a36c0 2022-01-09 op 	v.type = V_U16;
1341 fb1a36c0 2022-01-09 op 	if ((v.v.u16 = ++lasttag) == 255)
1342 fb1a36c0 2022-01-09 op 		v.v.u16 = ++lasttag;
1343 fb1a36c0 2022-01-09 op 
1344 fb1a36c0 2022-01-09 op 	pushv(&v);
1345 fb1a36c0 2022-01-09 op 	return EVAL_OK;
1346 fb1a36c0 2022-01-09 op }
1347 fb1a36c0 2022-01-09 op 
1348 fb1a36c0 2022-01-09 op static int
1349 fb1a36c0 2022-01-09 op builtin_send(int argc)
1350 fb1a36c0 2022-01-09 op {
1351 fb1a36c0 2022-01-09 op 	struct ibuf	*buf;
1352 fb1a36c0 2022-01-09 op 	struct value	 v;
1353 fb1a36c0 2022-01-09 op 	uint32_t	 len;
1354 fb1a36c0 2022-01-09 op 	uint16_t	 slen;
1355 fb1a36c0 2022-01-09 op 	int		 i;
1356 fb1a36c0 2022-01-09 op 
1357 fb1a36c0 2022-01-09 op 	check_for_output();
1358 fb1a36c0 2022-01-09 op 
1359 fb1a36c0 2022-01-09 op 	/*
1360 fb1a36c0 2022-01-09 op 	 * Compute the length of the packet.  4 is for the initial
1361 fb1a36c0 2022-01-09 op 	 * length field
1362 fb1a36c0 2022-01-09 op 	 */
1363 fb1a36c0 2022-01-09 op 	len = 4;
1364 fb1a36c0 2022-01-09 op 
1365 fb1a36c0 2022-01-09 op 	for (i = argc; i > 0; --i) {
1366 fb1a36c0 2022-01-09 op 		peekn(i, &v);
1367 fb1a36c0 2022-01-09 op 		switch (v.type) {
1368 fb1a36c0 2022-01-09 op 		case V_STR:
1369 fb1a36c0 2022-01-09 op 			len += 2; /* count */
1370 fb1a36c0 2022-01-09 op 			len += strlen(v.v.str);
1371 fb1a36c0 2022-01-09 op 			break;
1372 fb1a36c0 2022-01-09 op 
1373 fb1a36c0 2022-01-09 op 		case V_U8:
1374 fb1a36c0 2022-01-09 op 			len += 1;
1375 fb1a36c0 2022-01-09 op 			break;
1376 fb1a36c0 2022-01-09 op 
1377 fb1a36c0 2022-01-09 op 		case V_U16:
1378 fb1a36c0 2022-01-09 op 			len += 2;
1379 fb1a36c0 2022-01-09 op 			break;
1380 fb1a36c0 2022-01-09 op 
1381 fb1a36c0 2022-01-09 op 		case V_U32:
1382 fb1a36c0 2022-01-09 op 			len += 4;
1383 fb1a36c0 2022-01-09 op 			break;
1384 fb1a36c0 2022-01-09 op 
1385 fb1a36c0 2022-01-09 op 		default:
1386 fb1a36c0 2022-01-09 op 			before_printing();
1387 fb1a36c0 2022-01-09 op 			printf("%s: can't serialize ", __func__);
1388 fb1a36c0 2022-01-09 op 			pp_val(&v);
1389 fb1a36c0 2022-01-09 op 			printf("\n");
1390 fb1a36c0 2022-01-09 op 			return EVAL_ERR;
1391 fb1a36c0 2022-01-09 op 		}
1392 fb1a36c0 2022-01-09 op 	}
1393 fb1a36c0 2022-01-09 op 
1394 fb1a36c0 2022-01-09 op 	if (len > UINT16_MAX) {
1395 fb1a36c0 2022-01-09 op 		before_printing();
1396 fb1a36c0 2022-01-09 op 		printf("%s: message size too long: got %d when max is %d\n",
1397 fb1a36c0 2022-01-09 op 		    __func__, len, UINT16_MAX);
1398 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
1399 fb1a36c0 2022-01-09 op 	}
1400 fb1a36c0 2022-01-09 op 
1401 fb1a36c0 2022-01-09 op 	if ((buf = imsg_create(&ibuf, IMSG_BUF, 0, 0, len)) == NULL)
1402 fb1a36c0 2022-01-09 op 		fatal("imsg_create(%d)", len);
1403 fb1a36c0 2022-01-09 op 
1404 fb1a36c0 2022-01-09 op 	len = htole32(len);
1405 fb1a36c0 2022-01-09 op 	imsg_add(buf, &len, sizeof(len));
1406 fb1a36c0 2022-01-09 op 
1407 fb1a36c0 2022-01-09 op 	for (i = argc; i > 0; --i) {
1408 fb1a36c0 2022-01-09 op 		peekn(i, &v);
1409 fb1a36c0 2022-01-09 op 		switch (v.type) {
1410 fb1a36c0 2022-01-09 op 		case V_STR:
1411 fb1a36c0 2022-01-09 op 			slen = strlen(v.v.str);
1412 fb1a36c0 2022-01-09 op 			slen = htole16(slen);
1413 fb1a36c0 2022-01-09 op 			imsg_add(buf, &slen, sizeof(slen));
1414 fb1a36c0 2022-01-09 op 			imsg_add(buf, v.v.str, strlen(v.v.str));
1415 fb1a36c0 2022-01-09 op 			break;
1416 fb1a36c0 2022-01-09 op 
1417 fb1a36c0 2022-01-09 op 		case V_U8:
1418 fb1a36c0 2022-01-09 op 			imsg_add(buf, &v.v.u8, 1);
1419 fb1a36c0 2022-01-09 op 			break;
1420 fb1a36c0 2022-01-09 op 
1421 fb1a36c0 2022-01-09 op 		case V_U16:
1422 fb1a36c0 2022-01-09 op 			v.v.u16 = htole16(v.v.u16);
1423 fb1a36c0 2022-01-09 op 			imsg_add(buf, &v.v.u16, 2);
1424 fb1a36c0 2022-01-09 op 			break;
1425 fb1a36c0 2022-01-09 op 
1426 fb1a36c0 2022-01-09 op 		case V_U32:
1427 fb1a36c0 2022-01-09 op 			v.v.u32 = htole32(v.v.u32);
1428 fb1a36c0 2022-01-09 op 			imsg_add(buf, &v.v.u32, 4);
1429 fb1a36c0 2022-01-09 op 			break;
1430 fb1a36c0 2022-01-09 op 		}
1431 fb1a36c0 2022-01-09 op 	}
1432 fb1a36c0 2022-01-09 op 
1433 fb1a36c0 2022-01-09 op 	imsg_close(&ibuf, buf);
1434 fb1a36c0 2022-01-09 op 
1435 fb1a36c0 2022-01-09 op 	if (imsg_flush(&ibuf) == -1) {
1436 fb1a36c0 2022-01-09 op 		i = errno;
1437 fb1a36c0 2022-01-09 op 		before_printing();
1438 fb1a36c0 2022-01-09 op 		printf("%s: imsg_flush failed: %s\n", __func__, strerror(i));
1439 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
1440 fb1a36c0 2022-01-09 op 	}
1441 fb1a36c0 2022-01-09 op 
1442 fb1a36c0 2022-01-09 op 	check_for_output();
1443 fb1a36c0 2022-01-09 op 	return EVAL_OK;
1444 fb1a36c0 2022-01-09 op }
1445 fb1a36c0 2022-01-09 op 
1446 fb1a36c0 2022-01-09 op static int
1447 fb1a36c0 2022-01-09 op builtin_recv(int argc)
1448 fb1a36c0 2022-01-09 op {
1449 fb1a36c0 2022-01-09 op 	struct pollfd	pfd;
1450 afc95614 2022-05-10 op 	struct value	v;
1451 fb1a36c0 2022-01-09 op 	struct imsg	imsg;
1452 fb1a36c0 2022-01-09 op 	ssize_t		n, datalen;
1453 afc95614 2022-05-10 op 	int		serrno, want_more = 0;
1454 fb1a36c0 2022-01-09 op 
1455 fb1a36c0 2022-01-09 op 	if (lastmsg != NULL) {
1456 fb1a36c0 2022-01-09 op 		free(lastmsg);
1457 fb1a36c0 2022-01-09 op 		lastmsg = NULL;
1458 fb1a36c0 2022-01-09 op 	}
1459 fb1a36c0 2022-01-09 op 
1460 fb1a36c0 2022-01-09 op 	pfd.fd = ibuf.fd;
1461 fb1a36c0 2022-01-09 op 	pfd.events = POLLIN;
1462 afc95614 2022-05-10 op again:
1463 fb1a36c0 2022-01-09 op 	if (poll(&pfd, 1, INFTIM) == -1) {
1464 fb1a36c0 2022-01-09 op 		serrno = errno;
1465 fb1a36c0 2022-01-09 op 		before_printing();
1466 fb1a36c0 2022-01-09 op 		printf("%s: poll failed: %s\n", __func__, strerror(serrno));
1467 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
1468 fb1a36c0 2022-01-09 op 	}
1469 fb1a36c0 2022-01-09 op 
1470 fb1a36c0 2022-01-09 op 	if ((n = imsg_read(&ibuf)) == -1) {
1471 fb1a36c0 2022-01-09 op 		if (errno == EAGAIN)
1472 fb1a36c0 2022-01-09 op 			goto again;
1473 fb1a36c0 2022-01-09 op 		fatal("imsg_read");
1474 fb1a36c0 2022-01-09 op 	}
1475 fb1a36c0 2022-01-09 op 	if (n == 0) {
1476 fb1a36c0 2022-01-09 op 		before_printing();
1477 fb1a36c0 2022-01-09 op 		printf("child disconnected\n");
1478 fb1a36c0 2022-01-09 op 		return EVAL_ERR;
1479 fb1a36c0 2022-01-09 op 	}
1480 fb1a36c0 2022-01-09 op 
1481 fb1a36c0 2022-01-09 op 	check_for_output();
1482 fb1a36c0 2022-01-09 op 
1483 afc95614 2022-05-10 op 	for (;;) {
1484 afc95614 2022-05-10 op 		if ((n = imsg_get(&ibuf, &imsg)) == -1)
1485 afc95614 2022-05-10 op 			fatal("imsg_get");
1486 afc95614 2022-05-10 op 		if (n == 0)
1487 afc95614 2022-05-10 op 			break;
1488 fb1a36c0 2022-01-09 op 
1489 afc95614 2022-05-10 op 		want_more = 0;
1490 afc95614 2022-05-10 op 		datalen = imsg.hdr.len - IMSG_HEADER_SIZE;
1491 afc95614 2022-05-10 op 		switch (imsg.hdr.type) {
1492 afc95614 2022-05-10 op 		case IMSG_BUF:
1493 afc95614 2022-05-10 op 			v.type = V_MSG;
1494 afc95614 2022-05-10 op 			if ((v.v.msg.msg = malloc(datalen)) == NULL)
1495 afc95614 2022-05-10 op 				fatal("malloc");
1496 afc95614 2022-05-10 op 			memcpy(v.v.msg.msg, imsg.data, datalen);
1497 afc95614 2022-05-10 op 			v.v.msg.len = datalen;
1498 afc95614 2022-05-10 op 			pushv(&v);
1499 afc95614 2022-05-10 op 			imsg_free(&imsg);
1500 afc95614 2022-05-10 op 			return EVAL_OK;
1501 fb1a36c0 2022-01-09 op 
1502 afc95614 2022-05-10 op 		case IMSG_CLOSE:
1503 afc95614 2022-05-10 op 			before_printing();
1504 afc95614 2022-05-10 op 			printf("subprocess closed the connection\n");
1505 afc95614 2022-05-10 op 			imsg_free(&imsg);
1506 afc95614 2022-05-10 op 			return EVAL_ERR;
1507 afc95614 2022-05-10 op 
1508 afc95614 2022-05-10 op 		case IMSG_MSIZE:
1509 afc95614 2022-05-10 op 			imsg_free(&imsg);
1510 afc95614 2022-05-10 op 			want_more = 1;
1511 afc95614 2022-05-10 op 			break;
1512 afc95614 2022-05-10 op 
1513 afc95614 2022-05-10 op 		default:
1514 afc95614 2022-05-10 op 			before_printing();
1515 afc95614 2022-05-10 op 			printf("got unknown message from subprocess: %d\n",
1516 afc95614 2022-05-10 op 			    imsg.hdr.type);
1517 afc95614 2022-05-10 op 			imsg_free(&imsg);
1518 afc95614 2022-05-10 op 			return EVAL_ERR;
1519 afc95614 2022-05-10 op 		}
1520 fb1a36c0 2022-01-09 op 	}
1521 afc95614 2022-05-10 op 
1522 afc95614 2022-05-10 op 	if (want_more)
1523 afc95614 2022-05-10 op 		goto again;
1524 afc95614 2022-05-10 op 
1525 afc95614 2022-05-10 op 	fatalx("reached the end of %s\n", __func__);
1526 fb1a36c0 2022-01-09 op }
1527 fb1a36c0 2022-01-09 op 
1528 fb1a36c0 2022-01-09 op static pid_t
1529 fb1a36c0 2022-01-09 op spawn_client_proc(void)
1530 fb1a36c0 2022-01-09 op {
1531 fb1a36c0 2022-01-09 op 	const char	*argv[4];
1532 fb1a36c0 2022-01-09 op 	int		 p[2], out[2], argc = 0;
1533 fb1a36c0 2022-01-09 op 	pid_t		 pid;
1534 fb1a36c0 2022-01-09 op 
1535 fb1a36c0 2022-01-09 op 	if (child_out != -1)
1536 fb1a36c0 2022-01-09 op 		close(child_out);
1537 fb1a36c0 2022-01-09 op 
1538 fb1a36c0 2022-01-09 op 	if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC | SOCK_NONBLOCK,
1539 fb1a36c0 2022-01-09 op 	    PF_UNSPEC, p) == -1)
1540 fb1a36c0 2022-01-09 op 		fatal("socketpair");
1541 fb1a36c0 2022-01-09 op 
1542 fb1a36c0 2022-01-09 op 	if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC | SOCK_NONBLOCK,
1543 fb1a36c0 2022-01-09 op 	    PF_UNSPEC, out) == -1)
1544 fb1a36c0 2022-01-09 op 		fatal("socketpair");
1545 fb1a36c0 2022-01-09 op 
1546 fb1a36c0 2022-01-09 op 	switch (pid = fork()) {
1547 fb1a36c0 2022-01-09 op 	case -1:
1548 fb1a36c0 2022-01-09 op 		fatal("cannot fork");
1549 fb1a36c0 2022-01-09 op 	case 0:
1550 fb1a36c0 2022-01-09 op 		break;
1551 fb1a36c0 2022-01-09 op 	default:
1552 fb1a36c0 2022-01-09 op 		close(p[1]);
1553 fb1a36c0 2022-01-09 op 		close(out[1]);
1554 fb1a36c0 2022-01-09 op 		child_out = out[0];
1555 fb1a36c0 2022-01-09 op 		if (ibuf_inuse) {
1556 fb1a36c0 2022-01-09 op 			msgbuf_clear(&ibuf.w);
1557 fb1a36c0 2022-01-09 op 			close(ibuf.fd);
1558 fb1a36c0 2022-01-09 op 		}
1559 fb1a36c0 2022-01-09 op 		imsg_init(&ibuf, p[0]);
1560 fb1a36c0 2022-01-09 op 		ibuf_inuse = 1;
1561 fb1a36c0 2022-01-09 op 		return pid;
1562 fb1a36c0 2022-01-09 op 	}
1563 fb1a36c0 2022-01-09 op 
1564 fb1a36c0 2022-01-09 op 	close(p[0]);
1565 fb1a36c0 2022-01-09 op 	close(out[0]);
1566 fb1a36c0 2022-01-09 op 
1567 fb1a36c0 2022-01-09 op 	if (dup2(out[1], 1) == -1 ||
1568 fb1a36c0 2022-01-09 op 	    dup2(out[1], 2) == -1)
1569 fb1a36c0 2022-01-09 op 		fatal("dup2");
1570 fb1a36c0 2022-01-09 op 
1571 fb1a36c0 2022-01-09 op 	if (p[1] != 3) {
1572 fb1a36c0 2022-01-09 op 		if (dup2(p[1], 3) == -1)
1573 fb1a36c0 2022-01-09 op 			fatal("cannot setup imsg fd");
1574 fb1a36c0 2022-01-09 op 	} else if (fcntl(F_SETFD, 0) == -1)
1575 fb1a36c0 2022-01-09 op 		fatal("cannot setup imsg fd");
1576 fb1a36c0 2022-01-09 op 
1577 fb1a36c0 2022-01-09 op 	argv[argc++] = argv0;
1578 fb1a36c0 2022-01-09 op 	argv[argc++] = "-Tc";
1579 fb1a36c0 2022-01-09 op 
1580 fb1a36c0 2022-01-09 op #if DEBUG
1581 fb1a36c0 2022-01-09 op 	argv[argc++] = "-v";
1582 fb1a36c0 2022-01-09 op #endif
1583 fb1a36c0 2022-01-09 op 
1584 fb1a36c0 2022-01-09 op 	argv[argc++] = NULL;
1585 fb1a36c0 2022-01-09 op 
1586 fb1a36c0 2022-01-09 op 	execvp(argv0, (char *const *)argv);
1587 fb1a36c0 2022-01-09 op 	fatal("execvp");
1588 fb1a36c0 2022-01-09 op }
1589 fb1a36c0 2022-01-09 op 
1590 fb1a36c0 2022-01-09 op static void
1591 fb1a36c0 2022-01-09 op prepare_child_for_test(struct test *t)
1592 fb1a36c0 2022-01-09 op {
1593 1e84c7b7 2022-01-10 op 	struct passwd		*pw;
1594 1e84c7b7 2022-01-10 op 	struct kd_auth_proc	 rauth;
1595 fb1a36c0 2022-01-09 op 
1596 a0ec53a2 2022-01-09 op 	if ((pw = getpwuid(uid)) == NULL)
1597 a0ec53a2 2022-01-09 op 		fatal("getpwuid(%d)", uid);
1598 fb1a36c0 2022-01-09 op 
1599 1e84c7b7 2022-01-10 op 	memset(&rauth, 0, sizeof(rauth));
1600 1e84c7b7 2022-01-10 op 	strlcpy(rauth.uname, pw->pw_name, sizeof(rauth.uname));
1601 1e84c7b7 2022-01-10 op 	strlcpy(rauth.dir, dir, sizeof(rauth.dir));
1602 1e84c7b7 2022-01-10 op 
1603 fb1a36c0 2022-01-09 op 	imsg_compose(&ibuf, IMSG_AUTH, 0, 0, -1,
1604 1e84c7b7 2022-01-10 op 	    &rauth, sizeof(rauth));
1605 fb1a36c0 2022-01-09 op 
1606 fb1a36c0 2022-01-09 op 	if (imsg_flush(&ibuf) == -1)
1607 fb1a36c0 2022-01-09 op 		fatal("imsg_flush");
1608 fb1a36c0 2022-01-09 op }
1609 fb1a36c0 2022-01-09 op 
1610 fb1a36c0 2022-01-09 op static int
1611 fb1a36c0 2022-01-09 op run_test(struct test *t)
1612 fb1a36c0 2022-01-09 op {
1613 fb1a36c0 2022-01-09 op 	pid_t	pid;
1614 fb1a36c0 2022-01-09 op 	int	ret;
1615 fb1a36c0 2022-01-09 op 
1616 fb1a36c0 2022-01-09 op #if DEBUG
1617 fb1a36c0 2022-01-09 op 	before_printing();
1618 fb1a36c0 2022-01-09 op 	puts("=====================");
1619 fb1a36c0 2022-01-09 op 	pp_block(t->body);
1620 fb1a36c0 2022-01-09 op 	puts("=====================");
1621 fb1a36c0 2022-01-09 op #endif
1622 fb1a36c0 2022-01-09 op 
1623 fb1a36c0 2022-01-09 op         if (stackh != 0)
1624 fb1a36c0 2022-01-09 op 		popvn(stackh);
1625 fb1a36c0 2022-01-09 op 
1626 fb1a36c0 2022-01-09 op 	if (t->body == NULL) {
1627 fb1a36c0 2022-01-09 op 		before_printing();
1628 fb1a36c0 2022-01-09 op 		printf("no instructions, skipping...\n");
1629 fb1a36c0 2022-01-09 op 		return EVAL_SKIP;
1630 fb1a36c0 2022-01-09 op 	}
1631 fb1a36c0 2022-01-09 op 
1632 fb1a36c0 2022-01-09 op 	pid = spawn_client_proc();
1633 fb1a36c0 2022-01-09 op         prepare_child_for_test(t);
1634 fb1a36c0 2022-01-09 op 	ret = eval(t->body);
1635 fb1a36c0 2022-01-09 op 
1636 fb1a36c0 2022-01-09 op 	imsg_compose(&ibuf, IMSG_CONN_GONE, 0, 0, -1, NULL, 0);
1637 fb1a36c0 2022-01-09 op 	imsg_flush(&ibuf);
1638 fb1a36c0 2022-01-09 op 
1639 fb1a36c0 2022-01-09 op 	while (waitpid(pid, NULL, 0) != pid)
1640 fb1a36c0 2022-01-09 op 		; /* nop */
1641 fb1a36c0 2022-01-09 op 
1642 fb1a36c0 2022-01-09 op 	check_for_output();
1643 fb1a36c0 2022-01-09 op 
1644 fb1a36c0 2022-01-09 op 	if (t->shouldfail) {
1645 fb1a36c0 2022-01-09 op 		if (ret == EVAL_OK) {
1646 fb1a36c0 2022-01-09 op 			before_printing();
1647 fb1a36c0 2022-01-09 op 			printf("test was expected to fail\n");
1648 fb1a36c0 2022-01-09 op 			return EVAL_ERR;
1649 fb1a36c0 2022-01-09 op 		} else if (ret == EVAL_ERR)
1650 fb1a36c0 2022-01-09 op 			return EVAL_OK;
1651 fb1a36c0 2022-01-09 op 	}
1652 fb1a36c0 2022-01-09 op 
1653 fb1a36c0 2022-01-09 op 	return ret;
1654 fb1a36c0 2022-01-09 op }
1655 fb1a36c0 2022-01-09 op 
1656 fb1a36c0 2022-01-09 op int
1657 fb1a36c0 2022-01-09 op main(int argc, char **argv)
1658 fb1a36c0 2022-01-09 op {
1659 a0ec53a2 2022-01-09 op 	struct stat	 sb;
1660 fb1a36c0 2022-01-09 op 	struct test	*t;
1661 fb1a36c0 2022-01-09 op 	int		 ch, i, r, passed = 0, failed = 0, skipped = 0;
1662 fb1a36c0 2022-01-09 op 	int		 runclient = 0;
1663 fb1a36c0 2022-01-09 op 	const char	*pat = NULL;
1664 fb1a36c0 2022-01-09 op 	regex_t		 reg;
1665 fb1a36c0 2022-01-09 op 
1666 676b0c99 2022-12-21 op 	if ((argv0 = argv[0]) == NULL)
1667 676b0c99 2022-12-21 op 		fatal("empty argv[0]");
1668 fb1a36c0 2022-01-09 op 
1669 fb1a36c0 2022-01-09 op 	signal(SIGPIPE, SIG_IGN);
1670 fb1a36c0 2022-01-09 op 
1671 fb1a36c0 2022-01-09 op 	log_init(1, LOG_DAEMON);
1672 fb1a36c0 2022-01-09 op 	log_setverbose(1);
1673 fb1a36c0 2022-01-09 op 
1674 fb1a36c0 2022-01-09 op 	/* prepare the global env */
1675 fb1a36c0 2022-01-09 op 	pushenv();
1676 fb1a36c0 2022-01-09 op 
1677 fb1a36c0 2022-01-09 op 	add_builtin_proc("print", builtin_print, 1, 1);
1678 fb1a36c0 2022-01-09 op 	add_builtin_proc("debug", builtin_debug, 1, 1);
1679 fb1a36c0 2022-01-09 op 	add_builtin_proc("skip", builtin_skip, 0, 0);
1680 fb1a36c0 2022-01-09 op 	add_builtin_proc("iota", builtin_iota, 0, 0);
1681 fb1a36c0 2022-01-09 op 	add_builtin_proc("send", builtin_send, 2, 1);
1682 fb1a36c0 2022-01-09 op 	add_builtin_proc("recv", builtin_recv, 0, 0);
1683 fb1a36c0 2022-01-09 op 
1684 a0ec53a2 2022-01-09 op 	while ((ch = getopt(argc, argv, "nT:r:vx:")) != -1) {
1685 fb1a36c0 2022-01-09 op 		switch (ch) {
1686 fb1a36c0 2022-01-09 op 		case 'n':
1687 fb1a36c0 2022-01-09 op 			syntaxcheck = 1;
1688 fb1a36c0 2022-01-09 op 			break;
1689 fb1a36c0 2022-01-09 op 		case 'T':
1690 fb1a36c0 2022-01-09 op 			assert(*optarg == 'c');
1691 fb1a36c0 2022-01-09 op                         runclient = 1;
1692 fb1a36c0 2022-01-09 op 			break;
1693 a0ec53a2 2022-01-09 op 		case 'r':
1694 a0ec53a2 2022-01-09 op 			dir = optarg;
1695 a0ec53a2 2022-01-09 op 			break;
1696 fb1a36c0 2022-01-09 op 		case 'v':
1697 fb1a36c0 2022-01-09 op 			debug = 1;
1698 fb1a36c0 2022-01-09 op 			break;
1699 fb1a36c0 2022-01-09 op 		case 'x':
1700 fb1a36c0 2022-01-09 op 			pat = optarg;
1701 fb1a36c0 2022-01-09 op 			break;
1702 fb1a36c0 2022-01-09 op 		default:
1703 fb1a36c0 2022-01-09 op 			fprintf(stderr, "Usage: %s [-nv] [files...]\n",
1704 fb1a36c0 2022-01-09 op 			    *argv);
1705 fb1a36c0 2022-01-09 op 			exit(1);
1706 fb1a36c0 2022-01-09 op 		}
1707 fb1a36c0 2022-01-09 op 	}
1708 fb1a36c0 2022-01-09 op 	argc -= optind;
1709 fb1a36c0 2022-01-09 op 	argv += optind;
1710 fb1a36c0 2022-01-09 op 
1711 fb1a36c0 2022-01-09 op 	if (runclient)
1712 fb1a36c0 2022-01-09 op 		client(1, debug);
1713 fb1a36c0 2022-01-09 op 
1714 a0ec53a2 2022-01-09 op 	if (dir == NULL)
1715 a0ec53a2 2022-01-09 op 		fatal("missing root test dir");
1716 1e84c7b7 2022-01-10 op 
1717 a0ec53a2 2022-01-09 op 	if (stat(dir, &sb) == -1)
1718 a0ec53a2 2022-01-09 op 		fatal("stat(\"%s\")", dir);
1719 a0ec53a2 2022-01-09 op 	uid = sb.st_uid;
1720 a0ec53a2 2022-01-09 op 
1721 fb1a36c0 2022-01-09 op 	if (pat == NULL)
1722 fb1a36c0 2022-01-09 op 		pat = ".*";
1723 fb1a36c0 2022-01-09 op 
1724 ba8b636b 2022-01-09 op 	if (regcomp(&reg, pat, REG_ICASE | REG_NOSUB) != 0)
1725 fb1a36c0 2022-01-09 op 		fatalx("invalid regexp: %s", pat);
1726 fb1a36c0 2022-01-09 op 
1727 fb1a36c0 2022-01-09 op 	for (i = 0; i < argc; ++i)
1728 fb1a36c0 2022-01-09 op 		loadfile(argv[i]);
1729 fb1a36c0 2022-01-09 op 
1730 fb1a36c0 2022-01-09 op 	if (syntaxcheck) {
1731 fb1a36c0 2022-01-09 op 		fprintf(stderr, "files OK\n");
1732 fb1a36c0 2022-01-09 op 		return 0;
1733 fb1a36c0 2022-01-09 op 	}
1734 fb1a36c0 2022-01-09 op 
1735 fb1a36c0 2022-01-09 op 	/* Check for root privileges. */
1736 fb1a36c0 2022-01-09 op         if (geteuid())
1737 fb1a36c0 2022-01-09 op                 fatalx("need root privileges");
1738 fb1a36c0 2022-01-09 op 
1739 fb1a36c0 2022-01-09 op 	i = 0;
1740 fb1a36c0 2022-01-09 op 	TAILQ_FOREACH(t, &tests, entry) {
1741 fb1a36c0 2022-01-09 op 		if (regexec(&reg, t->name, 0, NULL, 0) != 0)
1742 fb1a36c0 2022-01-09 op 			continue;
1743 fb1a36c0 2022-01-09 op 
1744 fb1a36c0 2022-01-09 op 		printf("===> [%d/%d] running test \"%s\"... ", i+1, ntests,
1745 fb1a36c0 2022-01-09 op 		    t->name);
1746 fb1a36c0 2022-01-09 op 		fflush(stdout);
1747 fb1a36c0 2022-01-09 op 
1748 fb1a36c0 2022-01-09 op 		filler = "\n";
1749 fb1a36c0 2022-01-09 op 		r = run_test(t);
1750 fb1a36c0 2022-01-09 op 		if (filler == NULL)
1751 fb1a36c0 2022-01-09 op 			printf("=> test ");
1752 fb1a36c0 2022-01-09 op 
1753 fb1a36c0 2022-01-09 op 		switch (r) {
1754 fb1a36c0 2022-01-09 op 		case EVAL_OK:
1755 fb1a36c0 2022-01-09 op 			printf("passed\n");
1756 fb1a36c0 2022-01-09 op 			passed++;
1757 fb1a36c0 2022-01-09 op 			break;
1758 fb1a36c0 2022-01-09 op 		case EVAL_ERR:
1759 fb1a36c0 2022-01-09 op 			failed++;
1760 fb1a36c0 2022-01-09 op 			printf("failed\n");
1761 fb1a36c0 2022-01-09 op 			break;
1762 fb1a36c0 2022-01-09 op 		case EVAL_SKIP:
1763 fb1a36c0 2022-01-09 op 			printf("skipped\n");
1764 fb1a36c0 2022-01-09 op 			skipped++;
1765 fb1a36c0 2022-01-09 op 			break;
1766 fb1a36c0 2022-01-09 op 		}
1767 fb1a36c0 2022-01-09 op 
1768 fb1a36c0 2022-01-09 op 		if (filler == NULL)
1769 fb1a36c0 2022-01-09 op 			printf("\n");
1770 fb1a36c0 2022-01-09 op 		i++;
1771 fb1a36c0 2022-01-09 op 	}
1772 fb1a36c0 2022-01-09 op 
1773 fb1a36c0 2022-01-09 op 	printf("\n");
1774 fb1a36c0 2022-01-09 op 	printf("%d/%d passed (%d skipped and %d failed)\n",
1775 fb1a36c0 2022-01-09 op 	    passed, i, skipped, failed);
1776 fb1a36c0 2022-01-09 op 
1777 fb1a36c0 2022-01-09 op 	popenv();
1778 fb1a36c0 2022-01-09 op 	free(lastmsg);
1779 fb1a36c0 2022-01-09 op 	regfree(&reg);
1780 fb1a36c0 2022-01-09 op 
1781 fb1a36c0 2022-01-09 op 	return failed != 0;
1782 fb1a36c0 2022-01-09 op }