mdw@git.distorted.org.uk Git - mLib/blob - test/tvec-core.c

   1 /* -*-c-*-
   2  *
   3  * Main test vector driver
   4  *
   5  * (c) 2023 Straylight/Edgeware
   6  */
   7
   8 /*----- Licensing notice --------------------------------------------------*
   9  *
  10  * This file is part of the mLib utilities library.
  11  *
  12  * mLib is free software: you can redistribute it and/or modify it under
  13  * the terms of the GNU Library General Public License as published by
  14  * the Free Software Foundation; either version 2 of the License, or (at
  15  * your option) any later version.
  16  *
  17  * mLib is distributed in the hope that it will be useful, but WITHOUT
  18  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  19  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
  20  * License for more details.
  21  *
  22  * You should have received a copy of the GNU Library General Public
  23  * License along with mLib.  If not, write to the Free Software
  24  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307,
  25  * USA.
  26  */
  27
  28 /*----- Header files ------------------------------------------------------*/
  29
  30 #include <assert.h>
  31 #include <ctype.h>
  32 #include <errno.h>
  33 #include <string.h>
  34
  35 #include "alloc.h"
  36 #include "bench.h"
  37 #include "tvec.h"
  38
  39 /*----- Output ------------------------------------------------------------*/
  40
  41 int tvec_error(struct tvec_state *tv, const char *msg, ...)
  42 {
  43   va_list ap;
  44
  45   va_start(ap, msg); tvec_error_v(tv, msg, &ap); va_end(ap);
  46   tv->f |= TVSF_ERROR; return (-1);
  47 }
  48 int tvec_error_v(struct tvec_state *tv, const char *msg, va_list *ap)
  49   { tv->output->ops->error(tv->output, msg, ap); return (-1); }
  50
  51 void tvec_notice(struct tvec_state *tv, const char *msg, ...)
  52 {
  53   va_list ap;
  54   va_start(ap, msg); tvec_notice_v(tv, msg, &ap); va_end(ap);
  55 }
  56 void tvec_notice_v(struct tvec_state *tv, const char *msg, va_list *ap)
  57   { tv->output->ops->notice(tv->output, msg, ap); }
  58
  59 int tvec_syntax(struct tvec_state *tv, int ch, const char *expect, ...)
  60 {
  61   va_list ap;
  62
  63   va_start(ap, expect); tvec_syntax_v(tv, ch, expect, &ap); va_end(ap);
  64   return (-1);
  65 }
  66 int tvec_syntax_v(struct tvec_state *tv, int ch,
  67                    const char *expect, va_list *ap)
  68 {
  69   dstr d = DSTR_INIT;
  70   char found[8];
  71
  72   switch (ch) {
  73     case EOF: strcpy(found, "<eof>"); break;
  74     case '\n': strcpy(found, "<eol>"); ungetc(ch, tv->fp); break;
  75     default:
  76       if (isprint(ch)) sprintf(found, "`%c'", ch);
  77       else sprintf(found, "<#x%02x>", ch);
  78       break;
  79   }
  80   dstr_vputf(&d, expect, ap);
  81   tvec_error(tv, "syntax error: expected %s but found %s", expect, found);
  82   return (-1);
  83 }
  84
  85 void tvec_skipgroup(struct tvec_state *tv, const char *excuse, ...)
  86 {
  87   va_list ap;
  88   va_start(ap, excuse); tvec_skipgroup_v(tv, excuse, &ap); va_end(ap);
  89 }
  90 void tvec_skipgroup_v(struct tvec_state *tv, const char *excuse, va_list *ap)
  91 {
  92   tv->f |= TVSF_SKIP; tv->grps[TVOUT_SKIP]++;
  93   tv->output->ops->skipgroup(tv->output, excuse, ap);
  94 }
  95
  96 static void set_outcome(struct tvec_state *tv, unsigned out)
  97 {
  98   tv->f &= ~(TVSF_ACTIVE | TVSF_OUTMASK);
  99   tv->f |= out << TVSF_OUTSHIFT;
 100 }
 101
 102 void tvec_skip(struct tvec_state *tv, const char *excuse, ...)
 103 {
 104   va_list ap;
 105   va_start(ap, excuse); tvec_skip_v(tv, excuse, &ap); va_end(ap);
 106 }
 107 void tvec_skip_v(struct tvec_state *tv, const char *excuse, va_list *ap)
 108 {
 109   assert(tv->f&TVSF_ACTIVE);
 110   set_outcome(tv, TVOUT_SKIP);
 111   tv->output->ops->skip(tv->output, excuse, ap);
 112 }
 113
 114 void tvec_fail(struct tvec_state *tv, const char *detail, ...)
 115 {
 116   va_list ap;
 117   va_start(ap, detail); tvec_fail_v(tv, detail, &ap); va_end(ap);
 118 }
 119 void tvec_fail_v(struct tvec_state *tv, const char *detail, va_list *ap)
 120 {
 121   assert((tv->f&TVSF_ACTIVE) ||
 122          (tv->f&TVSF_OUTMASK) == (TVOUT_LOSE << TVSF_OUTSHIFT));
 123   set_outcome(tv, TVOUT_LOSE); tv->output->ops->fail(tv->output, detail, ap);
 124 }
 125
 126 void tvec_mismatch(struct tvec_state *tv)
 127   { tv->output->ops->mismatch(tv->output); }
 128
 129 void tvec_write(struct tvec_state *tv, const char *p, ...)
 130 {
 131   va_list ap;
 132   va_start(ap, p); tvec_write_v(tv, p, &ap); va_end(ap);
 133 }
 134 void tvec_write_v(struct tvec_state *tv, const char *p, va_list *ap)
 135 {
 136   dstr d = DSTR_INIT;
 137
 138   dstr_vputf(&d, p, ap); tv->output->ops->write(tv->output, d.buf, d.len);
 139   DDESTROY(&d);
 140 }
 141
 142 /*----- Serialization and deserialization ---------------------------------*/
 143
 144 int tvec_serialize(const struct tvec_reg *rv,
 145                    const struct tvec_regdef *regs,
 146                    unsigned nr, size_t regsz,
 147                    void **p_out, size_t *sz_out)
 148 {
 149   void *p = 0; buf b;
 150   unsigned char *bitmap;
 151   size_t i, nbits, bitsz, sz;
 152   const struct tvec_regdef *rd;
 153   const struct tvec_reg *r;
 154   int rc;
 155
 156   for (rd = regs, nbits = 0, sz = 0; rd->name; rd++, nbits++) {
 157     if (rd->i >= nr) continue;
 158     r = TVEC_GREG(rv, rd->i, regsz); if (!(r->f&TVRF_LIVE)) continue;
 159     sz += rd->ty->measure(&r->v, rd);
 160   }
 161   bitsz = (nbits + 7)/8; sz += bitsz;
 162
 163   p = xmalloc(sz); buf_init(&b, p, sz);
 164   bitmap = buf_get(&b, bitsz); assert(bitmap); memset(bitmap, 0, bitsz);
 165   for (rd = regs, i = 0; rd->name; rd++, i++) {
 166     if (rd->i >= nr) continue;
 167     r = TVEC_GREG(rv, rd->i, regsz); if (!(r->f&TVRF_LIVE)) continue;
 168     bitmap[rd->i/8] |= 1 << rd->i%8;
 169     if (rd->ty->tobuf(&b, &r->v, rd)) { rc = -1; goto end; }
 170   }
 171
 172   if (BBAD(&b)) { rc = -1; goto end; }
 173   *p_out = p; *sz_out = BLEN(&b); p = 0; rc = 0;
 174 end:
 175   xfree(p);
 176   return (rc);
 177 }
 178
 179 int tvec_deserialize(struct tvec_reg *rv,
 180                      const struct tvec_regdef *regs,
 181                      unsigned nr, size_t regsz,
 182                      const void *p, size_t sz)
 183 {
 184   buf b;
 185   const unsigned char *bitmap;
 186   size_t i, nbits, bitsz;
 187   const struct tvec_regdef *rd;
 188   struct tvec_reg *r;
 189   int rc;
 190
 191   for (rd = regs, nbits = 0; rd->name; rd++, nbits++);
 192   bitsz = (nbits + 7)/8; sz += bitsz;
 193
 194   buf_init(&b, (/*unconst*/ void *)p, sz);
 195   bitmap = buf_get(&b, bitsz); if (!bitmap) { rc = -1; goto end; }
 196   for (rd = regs, i = 0; rd->name; rd++, i++) {
 197     if (rd->i >= nr) continue;
 198     if (!(bitmap[rd->i/8]&(1 << rd->i%8))) continue;
 199     r = TVEC_GREG(rv, rd->i, regsz);
 200     if (rd->ty->frombuf(&b, &r->v, rd)) { rc = -1; goto end; }
 201     r->f |= TVRF_LIVE;
 202   }
 203
 204   if (BBAD(&b)) { rc = -1; goto end; }
 205   rc = 0;
 206 end:
 207   return (rc);
 208 }
 209
 210 /*----- Main machinery ----------------------------------------------------*/
 211
 212 void tvec_skipspc(struct tvec_state *tv)
 213 {
 214   int ch;
 215
 216   for (;;) {
 217     ch = getc(tv->fp);
 218     if (ch == EOF) break;
 219     else if (ch == '\n' || !isspace(ch)) { ungetc(ch, tv->fp); break; }
 220   }
 221 }
 222
 223 int tvec_flushtoeol(struct tvec_state *tv, unsigned f)
 224 {
 225   int ch, rc = 0;
 226
 227   for (;;) {
 228     ch = getc(tv->fp);
 229     switch (ch) {
 230       case '\n': tv->lno++; return (rc);
 231       case EOF: return (rc);
 232       case ';': f |= TVFF_ALLOWANY; break;
 233       default:
 234         if (!(f&TVFF_ALLOWANY) && !isspace(ch)) {
 235           tvec_syntax(tv, ch, "end-of-line");
 236           rc = -1; f |= TVFF_ALLOWANY;
 237         }
 238         break;
 239     }
 240   }
 241 }
 242
 243 int tvec_nexttoken(struct tvec_state *tv)
 244 {
 245   enum { TAIL, NEWLINE, INDENT, COMMENT };
 246   int ch;
 247   unsigned s = TAIL;
 248
 249   for (;;) {
 250     ch = getc(tv->fp);
 251     switch (ch) {
 252       case EOF:
 253         return (-1);
 254
 255       case ';':
 256         s = COMMENT;
 257         break;
 258
 259       case '\n':
 260         if (s == NEWLINE || s == INDENT) { ungetc(ch, tv->fp); return (-1); }
 261         else { tv->lno++; s = NEWLINE; }
 262         break;
 263
 264       default:
 265         if (isspace(ch))
 266           { if (s == NEWLINE) s = INDENT; }
 267         else if (s != COMMENT) {
 268           ungetc(ch, tv->fp);
 269           if (s == NEWLINE) return (-1);
 270           else return (0);
 271         }
 272         break;
 273     }
 274   }
 275 }
 276
 277 int tvec_readword(struct tvec_state *tv, dstr *d, const char *delims,
 278                   const char *expect, ...)
 279 {
 280   va_list ap;
 281   int rc;
 282
 283   va_start(ap, expect);
 284   rc = tvec_readword_v(tv, d, delims, expect, &ap);
 285   va_end(ap);
 286   return (rc);
 287 }
 288 int tvec_readword_v(struct tvec_state *tv, dstr *d, const char *delims,
 289                     const char *expect, va_list *ap)
 290 {
 291   int ch;
 292
 293   ch = getc(tv->fp);
 294   if (ch == '\n' || ch == EOF || ch == ';' ||
 295       (delims && strchr(delims, ch))) {
 296     if (expect) return (tvec_syntax(tv, ch, expect, ap));
 297     else { ungetc(ch, tv->fp); return (-1); }
 298   }
 299   if (d->len) DPUTC(d, ' ');
 300   do {
 301     DPUTC(d, ch);
 302     ch = getc(tv->fp);
 303   } while (ch != EOF && !isspace(ch) && (!delims || !strchr(delims, ch)));
 304   DPUTZ(d); if (ch != EOF) ungetc(ch, tv->fp);
 305   return (0);
 306 }
 307
 308 static void init_registers(struct tvec_state *tv)
 309 {
 310   const struct tvec_regdef *rd;
 311   struct tvec_reg *r;
 312
 313   for (rd = tv->test->regs; rd->name; rd++) {
 314     assert(rd->i < tv->nreg); r = TVEC_REG(tv, in, rd->i);
 315     rd->ty->init(&r->v, rd); r->f = 0;
 316     if (rd->i < tv->nrout)
 317       { r = TVEC_REG(tv, out, rd->i); rd->ty->init(&r->v, rd); r->f = 0; }
 318   }
 319   tv->expst = '.';
 320 }
 321
 322 static void release_registers(struct tvec_state *tv)
 323 {
 324   const struct tvec_regdef *rd;
 325   struct tvec_reg *r;
 326
 327   for (rd = tv->test->regs; rd->name; rd++) {
 328     assert(rd->i < tv->nreg); r = TVEC_REG(tv, in, rd->i);
 329     rd->ty->release(&r->v, rd); r->f = 0;
 330     if (rd->i < tv->nrout)
 331       { r = TVEC_REG(tv, out, rd->i); rd->ty->release(&r->v, rd); r->f = 0; }
 332   }
 333 }
 334
 335 void tvec_check(struct tvec_state *tv, const char *detail, ...)
 336 {
 337   va_list ap;
 338   va_start(ap, detail); tvec_check_v(tv, detail, &ap); va_end(ap);
 339 }
 340 void tvec_check_v(struct tvec_state *tv, const char *detail, va_list *ap)
 341 {
 342   const struct tvec_regdef *rd;
 343   const struct tvec_reg *rin, *rout;
 344   unsigned f = 0;
 345 #define f_mismatch 1u
 346
 347   if (tv->expst != tv->st) f |= f_mismatch;
 348   for (rd = tv->test->regs; rd->name; rd++) {
 349     if (rd->i >= tv->nrout) continue;
 350     rin = TVEC_REG(tv, in, rd->i); rout = TVEC_REG(tv, out, rd->i);
 351     if (!rin->f&TVRF_LIVE) continue;
 352     if (!rd->ty->eq(&rin->v, &rout->v, rd)) f |= f_mismatch;
 353   }
 354   if (!(f&f_mismatch)) return;
 355
 356   tvec_fail_v(tv, detail, ap);
 357   tvec_mismatch(tv);
 358
 359 #undef f_mismatch
 360 }
 361
 362 int tvec_runtest(struct tvec_state *tv)
 363 {
 364   tv->test->fn(tv->in, tv->out, (/*unconst*/ void *)tv->test->arg.p);
 365   tvec_check(tv, 0); return (0);
 366 }
 367
 368 static void begin_test(struct tvec_state *tv)
 369 {
 370   tv->f |= TVSF_ACTIVE; tv->f &= ~TVSF_OUTMASK; tv->st = '.';
 371   tv->output->ops->btest(tv->output);
 372 }
 373
 374 void tvec_endtest(struct tvec_state *tv)
 375 {
 376   unsigned out;
 377
 378   if (tv->f&TVSF_ACTIVE) out = TVOUT_WIN;
 379   else out = (tv->f&TVSF_OUTMASK) >> TVSF_OUTSHIFT;
 380   assert(out < TVOUT_LIMIT); tv->curr[out]++;
 381   tv->output->ops->etest(tv->output, out);
 382   tv->f &= ~TVSF_OPEN;
 383 }
 384
 385 static void check(struct tvec_state *tv)
 386 {
 387   const struct tvec_regdef *rd;
 388
 389   if (!(tv->f&TVSF_OPEN)) return;
 390
 391   for (rd = tv->test->regs; rd->name; rd++) {
 392     if (TVEC_REG(tv, in, rd->i)->f&TVRF_LIVE)
 393       { if (rd->i < tv->nrout) TVEC_REG(tv, out, rd->i)->f |= TVRF_LIVE; }
 394     else if (!(rd->f&TVRF_OPT)) {
 395       tvec_error(tv, "required register `%s' not set in test `%s'",
 396                  rd->name, tv->test->name);
 397       goto end;
 398     }
 399   }
 400
 401   if (!(tv->f&TVSF_SKIP))
 402     { begin_test(tv); tv->test->run(tv); tvec_endtest(tv); }
 403
 404 end:
 405   tv->f &= ~TVSF_OPEN; release_registers(tv); init_registers(tv);
 406 }
 407
 408 static void begin_test_group(struct tvec_state *tv)
 409 {
 410   unsigned i;
 411
 412   tv->output->ops->bgroup(tv->output);
 413   tv->f &= ~TVSF_SKIP;
 414   init_registers(tv);
 415   for (i = 0; i < TVOUT_LIMIT; i++) tv->curr[i] = 0;
 416   if (tv->test->preflight) tv->test->preflight(tv);
 417 }
 418
 419 void tvec_reportgroup(struct tvec_state *tv)
 420 {
 421   unsigned i, out, nrun;
 422
 423   for (i = 0, nrun = 0; i < TVOUT_LIMIT; i++)
 424     { nrun += tv->curr[i]; tv->all[i] += tv->curr[i]; }
 425
 426   if (tv->curr[TVOUT_SKIP] == nrun)
 427     { out = TVOUT_SKIP; tvec_skipgroup(tv, nrun ? 0 : "no tests to run"); }
 428   else {
 429     if (tv->curr[TVOUT_LOSE]) out = TVOUT_LOSE;
 430     else out = TVOUT_WIN;
 431     tv->grps[out]++; tv->output->ops->egroup(tv->output, out);
 432   }
 433 }
 434
 435 static void end_test_group(struct tvec_state *tv)
 436 {
 437   if (!tv->test) return;
 438   if (tv->f&TVSF_OPEN) check(tv);
 439   if (!(tv->f&TVSF_SKIP)) tvec_reportgroup(tv);
 440   release_registers(tv); tv->test = 0;
 441 }
 442
 443 int tvec_read(struct tvec_state *tv, const char *infile, FILE *fp)
 444 {
 445   dstr d = DSTR_INIT;
 446   const struct tvec_test *test;
 447   const struct tvec_regdef *rd;
 448   struct tvec_reg *r;
 449   int ch;
 450   int rc = 0;
 451
 452   tv->infile = infile; tv->lno = 1; tv->fp = fp;
 453
 454   for (;;) {
 455     ch = getc(tv->fp);
 456     switch (ch) {
 457
 458       case EOF:
 459         goto end;
 460
 461       case '[':
 462         end_test_group(tv);
 463         tvec_skipspc(tv);
 464         DRESET(&d); tvec_readword(tv, &d, "];", "group name");
 465         tvec_skipspc(tv);
 466         ch = getc(tv->fp); if (ch != ']') tvec_syntax(tv, ch, "`]'");
 467         for (test = tv->tests; test->name; test++)
 468           if (STRCMP(d.buf, ==, test->name)) goto found_test;
 469         tvec_error(tv, "unknown test group `%s'", d.buf); goto flush_line;
 470       found_test:
 471         tvec_flushtoeol(tv, 0); tv->test = test; begin_test_group(tv);
 472         break;
 473
 474       case '\n':
 475         tv->lno++;
 476         if (tv->f&TVSF_OPEN) check(tv);
 477         break;
 478
 479       default:
 480         if (isspace(ch)) {
 481           tvec_skipspc(tv);
 482           ch = getc(tv->fp);
 483           if (ch == EOF) goto end;
 484           else if (ch == ';') tvec_flushtoeol(tv, TVFF_ALLOWANY);
 485           else if (tvec_flushtoeol(tv, 0)) rc = -1;
 486           else check(tv);
 487         } else if (ch == ';')
 488           tvec_flushtoeol(tv, TVFF_ALLOWANY);
 489         else {
 490           ungetc(ch, tv->fp);
 491           DRESET(&d);
 492           if (tvec_readword(tv, &d, "=:;", "register name")) goto flush_line;
 493           tvec_skipspc(tv); ch = getc(tv->fp);
 494           if (ch != '=' && ch != ':')
 495             { tvec_syntax(tv, ch, "`=' or `:'"); goto flush_line; }
 496           tvec_skipspc(tv);
 497           if (d.buf[0] == '@') {
 498             if (STRCMP(d.buf, ==, "@status")) {
 499               if (!(tv->f&TVSF_OPEN))
 500                 { tv->test_lno = tv->lno; tv->f |= TVSF_OPEN; }
 501               ch = getc(tv->fp);
 502               if (ch == EOF || ch == '\n' || ch == ';')
 503                 { tvec_syntax(tv, ch, "status character"); goto flush_line; }
 504               else if (ch == '\\') {
 505                 ch = getc(tv->fp);
 506                 if (ch == EOF || ch == '\n') {
 507                   tvec_syntax(tv, ch, "escaped status character");
 508                   goto flush_line;
 509                 }
 510               }
 511               tv->expst = ch;
 512               tvec_flushtoeol(tv, 0);
 513             } else {
 514               tvec_error(tv, "unknown special register `%s'", d.buf);
 515               goto flush_line;
 516             }
 517           } else {
 518             if (!tv->test)
 519               { tvec_error(tv, "no current test"); goto flush_line; }
 520             for (rd = tv->test->regs; rd->name; rd++)
 521               if (STRCMP(rd->name, ==, d.buf)) goto found_reg;
 522             tvec_error(tv, "unknown register `%s' for test `%s'",
 523                        d.buf, tv->test->name);
 524             goto flush_line;
 525           found_reg:
 526             if (!(tv->f&TVSF_OPEN))
 527               { tv->test_lno = tv->lno; tv->f |= TVSF_OPEN; }
 528             tvec_skipspc(tv);
 529             r = TVEC_REG(tv, in, rd->i);
 530             if (r->f&TVRF_LIVE) {
 531               tvec_error(tv, "register `%s' already set", rd->name);
 532               goto flush_line;
 533             }
 534             if (rd->ty->parse(&r->v, rd, tv)) goto flush_line;
 535             r->f |= TVRF_LIVE;
 536           }
 537         }
 538         break;
 539     }
 540     continue;
 541
 542   flush_line:
 543     tvec_flushtoeol(tv, TVFF_ALLOWANY); rc = -1;
 544   }
 545   if (ferror(tv->fp))
 546     { tvec_error(tv, "error reading input: %s", strerror(errno)); rc = -1; }
 547 end:
 548   end_test_group(tv);
 549   tv->infile = 0; tv->fp = 0;
 550   dstr_destroy(&d);
 551   return (rc);
 552 }
 553
 554 /*----- Benchmarking ------------------------------------------------------*/
 555
 556 struct bench_state *tvec_benchstate;
 557
 558 struct benchrun {
 559   unsigned long *n;
 560   tvec_testfn *fn;
 561   const struct tvec_reg *in; struct tvec_reg *out;
 562   void *ctx;
 563 };
 564
 565 static void benchloop_outer(unsigned long n, void *p)
 566   { struct benchrun *r = p; while (n--) r->fn(r->in, r->out, r->ctx); }
 567
 568 static void benchloop_inner(unsigned long n, void *p)
 569   { struct benchrun *r = p; *r->n = n; r->fn(r->in, r->out, r->ctx); }
 570
 571 int tvec_ensurebench(struct tvec_state *tv, struct bench_state **b_out)
 572 {
 573   const struct tvec_bench *tvb = tv->test->arg.p;
 574   struct bench_state **bb;
 575   struct bench_timer *bt;
 576
 577   if (tvb->b) bb = tvb->b;
 578   else bb = &tvec_benchstate;
 579
 580   if (!*bb) {
 581     bt = bench_createtimer();
 582     if (!bt) { tvec_skip(tv, "failed to create timer"); return (-1); }
 583     *bb = xmalloc(sizeof(**bb)); bench_init(*bb, bt);
 584   } else if (!(*bb)->tm)
 585     { tvec_skip(tv, "failed to create timer"); return (-1); }
 586
 587   *b_out = *bb;
 588   return (0);
 589 }
 590
 591 int tvec_bench(struct tvec_state *tv)
 592 {
 593   const struct tvec_bench *tvb = tv->test->arg.p;
 594   struct bench_state *b;
 595   struct bench_timing tm;
 596   struct benchrun r;
 597   bench_fn *loopfn;
 598
 599   if (tvec_ensurebench(tv, &b)) goto end_0;
 600
 601   r.in = tv->in; r.out = tv->out; r.fn = tv->test->fn;
 602   if (tvb->ctxsz) r.ctx = xmalloc(tvb->ctxsz);
 603   else r.ctx = 0;
 604   if (tvb->setup && tvb->setup(tv->in, tv->out, &tvb->arg, r.ctx))
 605     { tvec_skip(tv, "benchmark setup failed"); goto end_1; }
 606
 607   if (tvb->riter < 0)
 608     { r.n = 0; loopfn = benchloop_outer; }
 609   else
 610     { r.n = &TVEC_REG(tv, in, tvb->riter)->v.u; loopfn = benchloop_inner; }
 611
 612   tv->output->ops->bbench(tv->output);
 613   if (bench_measure(&tm, b, loopfn, &r))
 614     { tv->output->ops->ebench(tv->output, 0); goto end_2; }
 615   tv->output->ops->ebench(tv->output, &tm);
 616
 617 end_2:
 618   if (tvb->teardown) tvb->teardown(r.ctx);
 619 end_1:
 620   if (r.ctx) xfree(r.ctx);
 621 end_0:
 622   return (0);
 623 }
 624
 625 /*----- Ad-hoc testing ----------------------------------------------------*/
 626
 627 static const struct tvec_regdef no_regs = { 0, 0, 0, 0, { 0 } };
 628
 629 static int fakerun(struct tvec_state *tv)
 630   { assert(!"fake run function"); }
 631 static void fakefn(const struct tvec_reg *in, struct tvec_reg *out, void *p)
 632   { assert(!"fake test function"); }
 633
 634 void tvec_adhoc(struct tvec_state *tv, struct tvec_test *t)
 635 {
 636   t->name = "<unset>"; t->regs = &no_regs;
 637   t->preflight = 0; t->run = fakerun; t->fn = fakefn; t->arg.p = 0;
 638   tv->tests = t;
 639 }
 640
 641 void tvec_begingroup(struct tvec_state *tv, const char *name,
 642                      const char *file, unsigned lno)
 643 {
 644   struct tvec_test *t = (/*unconst*/ struct tvec_test *)tv->tests;
 645
 646   t->name = name; tv->test = t;
 647   tv->infile = file; tv->lno = tv->test_lno = lno;
 648   begin_test_group(tv);
 649 }
 650
 651 void tvec_endgroup(struct tvec_state *tv)
 652 {
 653   if (!(tv->f&TVSF_SKIP)) tvec_reportgroup(tv);
 654   tv->test = 0;
 655 }
 656
 657 void tvec_begintest(struct tvec_state *tv, const char *file, unsigned lno)
 658 {
 659   tv->infile = file; tv->lno = tv->test_lno = lno;
 660   begin_test(tv); tv->f |= TVSF_OPEN;
 661 }
 662
 663 struct adhoc_claim {
 664   unsigned f;
 665 #define ACF_FRESH 1u
 666   const char *saved_file; unsigned saved_lno;
 667 };
 668
 669 static void adhoc_claim_setup(struct tvec_state *tv,
 670                               struct adhoc_claim *ck,
 671                               const struct tvec_regdef *regs,
 672                               const char *file, unsigned lno)
 673 {
 674   struct tvec_test *t = (/*unconst*/ struct tvec_test *)tv->test;
 675
 676   ck->f = 0;
 677
 678   if (!(tv->f&TVSF_OPEN))
 679     { ck->f |= ACF_FRESH; tvec_begintest(tv, file, lno); }
 680
 681   ck->saved_file = tv->infile; if (file) tv->infile = file;
 682   ck->saved_lno = tv->test_lno; if (file) tv->test_lno = lno;
 683   t->regs = regs ? regs : &no_regs;
 684
 685   tv->st = '.';
 686 }
 687
 688 static void adhoc_claim_teardown(struct tvec_state *tv,
 689                                  struct adhoc_claim *ck)
 690 {
 691   struct tvec_test *t = (/*unconst*/ struct tvec_test *)tv->test;
 692
 693   t->regs = &no_regs;
 694   tv->infile = ck->saved_file; tv->test_lno = ck->saved_lno;
 695
 696   if (ck->f&ACF_FRESH) tvec_endtest(tv);
 697 }
 698
 699 int tvec_claim(struct tvec_state *tv, int ok,
 700                const char *file, unsigned lno, const char *expr, ...)
 701 {
 702   struct adhoc_claim ck;
 703   va_list ap;
 704
 705   adhoc_claim_setup(tv, &ck, 0, file, lno);
 706   if (!ok)
 707     { va_start(ap, expr); tvec_fail_v(tv, expr, &ap); va_end(ap); }
 708   adhoc_claim_teardown(tv, &ck);
 709   return (ok);
 710 }
 711
 712 int tvec_claimeq(struct tvec_state *tv,
 713                  const struct tvec_regty *ty, const union tvec_misc *arg,
 714                  const char *file, unsigned lno, const char *expr)
 715 {
 716   struct tvec_regdef regs[2];
 717   struct adhoc_claim ck;
 718   int ok;
 719
 720   tv->in[0].f = tv->out[0].f = TVRF_LIVE;
 721
 722   regs[0].name = "value"; regs[0].i = 0;
 723   regs[0].ty = ty; regs[0].f = 0;
 724   if (arg) regs[0].arg = *arg;
 725   else regs[0].arg.p = 0;
 726
 727   regs[1].name = 0;
 728
 729   adhoc_claim_setup(tv, &ck, regs, file, lno);
 730   ok = ty->eq(&tv->in[0].v, &tv->out[0].v, &regs[0]);
 731   if (!ok) { tvec_fail(tv, "%s", expr ); tvec_mismatch(tv); }
 732   adhoc_claim_teardown(tv, &ck);
 733   return (ok);
 734 }
 735
 736 /*----- Session lifecycle -------------------------------------------------*/
 737
 738 void tvec_begin(struct tvec_state *tv_out,
 739                 const struct tvec_info *info,
 740                 struct tvec_output *o)
 741 {
 742   unsigned i;
 743
 744   tv_out->f = 0;
 745
 746   assert(info->nrout <= info->nreg);
 747   tv_out->nrout = info->nrout; tv_out->nreg = info->nreg;
 748   tv_out->regsz = info->regsz;
 749   tv_out->in = xmalloc(tv_out->nreg*tv_out->regsz);
 750   tv_out->out = xmalloc(tv_out->nrout*tv_out->regsz);
 751   for (i = 0; i < tv_out->nreg; i++) {
 752     TVEC_REG(tv_out, in, i)->f = 0;
 753     if (i < tv_out->nrout) TVEC_REG(tv_out, out, i)->f = 0;
 754   }
 755
 756   for (i = 0; i < TVOUT_LIMIT; i++)
 757     tv_out->curr[i] = tv_out->all[i] = tv_out->grps[i] = 0;
 758
 759   tv_out->tests = info->tests; tv_out->test = 0;
 760   tv_out->infile = 0; tv_out->lno = 0; tv_out->fp = 0;
 761   o->tv = tv_out; tv_out->output = o;
 762
 763   tv_out->output->ops->bsession(tv_out->output);
 764 }
 765
 766 int tvec_end(struct tvec_state *tv)
 767 {
 768   int rc = tv->output->ops->esession(tv->output);
 769
 770   tv->output->ops->destroy(tv->output);
 771   xfree(tv->in); xfree(tv->out);
 772   return (rc);
 773 }
 774
 775 /*----- That's all, folks -------------------------------------------------*/