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