_ Git - BearSSL/blob - test/test_x509.c

   1 /*
   2  * Copyright (c) 2016 Thomas Pornin <pornin@bolet.org>
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining
   5  * a copy of this software and associated documentation files (the
   6  * "Software"), to deal in the Software without restriction, including
   7  * without limitation the rights to use, copy, modify, merge, publish,
   8  * distribute, sublicense, and/or sell copies of the Software, and to
   9  * permit persons to whom the Software is furnished to do so, subject to
  10  * the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be
  13  * included in all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  16  * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  17  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  18  * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
  19  * BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
  20  * ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
  21  * CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22  * SOFTWARE.
  23  */
  24
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <stdint.h>
  29
  30 #ifdef _WIN32
  31 #include <windows.h>
  32 #else
  33 #include <unistd.h>
  34 #endif
  35
  36 #include "bearssl.h"
  37
  38 #define STR(x)    STR_(x)
  39 #define STR_(x)   #x
  40 #ifdef SRCDIRNAME
  41 #define DIRNAME        STR(SRCDIRNAME) "/test/x509"
  42 #else
  43 #define DIRNAME        "test/x509"
  44 #endif
  45 #define CONFFILE       DIRNAME "/alltests.txt"
  46 #define DEFAULT_TIME   "2016-08-30T18:00:00Z"
  47
  48 static void *
  49 xmalloc(size_t len)
  50 {
  51         void *buf;
  52
  53         if (len == 0) {
  54                 return NULL;
  55         }
  56         buf = malloc(len);
  57         if (buf == NULL) {
  58                 fprintf(stderr, "error: cannot allocate %lu byte(s)\n",
  59                         (unsigned long)len);
  60                 exit(EXIT_FAILURE);
  61         }
  62         return buf;
  63 }
  64
  65 static void
  66 xfree(void *buf)
  67 {
  68         if (buf != NULL) {
  69                 free(buf);
  70         }
  71 }
  72
  73 static char *
  74 xstrdup(const char *name)
  75 {
  76         size_t n;
  77         char *s;
  78
  79         if (name == NULL) {
  80                 return NULL;
  81         }
  82         n = strlen(name) + 1;
  83         s = xmalloc(n);
  84         memcpy(s, name, n);
  85         return s;
  86 }
  87
  88 typedef struct {
  89         char *buf;
  90         size_t ptr, len;
  91 } string_builder;
  92
  93 static string_builder *
  94 SB_new(void)
  95 {
  96         string_builder *sb;
  97
  98         sb = xmalloc(sizeof *sb);
  99         sb->len = 8;
 100         sb->buf = xmalloc(sb->len);
 101         sb->ptr = 0;
 102         return sb;
 103 }
 104
 105 static void
 106 SB_expand(string_builder *sb, size_t extra_len)
 107 {
 108         size_t nlen;
 109         char *nbuf;
 110
 111         if (extra_len < (sb->len - sb->ptr)) {
 112                 return;
 113         }
 114         nlen = sb->len << 1;
 115         if (extra_len > (nlen - sb->ptr)) {
 116                 nlen = sb->ptr + extra_len;
 117         }
 118         nbuf = xmalloc(nlen);
 119         memcpy(nbuf, sb->buf, sb->ptr);
 120         xfree(sb->buf);
 121         sb->buf = nbuf;
 122         sb->len = nlen;
 123 }
 124
 125 static void
 126 SB_append_char(string_builder *sb, int c)
 127 {
 128         SB_expand(sb, 1);
 129         sb->buf[sb->ptr ++] = c;
 130 }
 131
 132 /* unused
 133 static void
 134 SB_append_string(string_builder *sb, const char *s)
 135 {
 136         size_t n;
 137
 138         n = strlen(s);
 139         SB_expand(sb, n);
 140         memcpy(sb->buf + sb->ptr, s, n);
 141         sb->ptr += n;
 142 }
 143 */
 144
 145 /* unused
 146 static char *
 147 SB_to_string(string_builder *sb)
 148 {
 149         char *s;
 150
 151         s = xmalloc(sb->ptr + 1);
 152         memcpy(s, sb->buf, sb->ptr);
 153         s[sb->ptr] = 0;
 154         return s;
 155 }
 156 */
 157
 158 static char *
 159 SB_contents(string_builder *sb)
 160 {
 161         return sb->buf;
 162 }
 163
 164 static size_t
 165 SB_length(string_builder *sb)
 166 {
 167         return sb->ptr;
 168 }
 169
 170 static void
 171 SB_set_length(string_builder *sb, size_t len)
 172 {
 173         if (sb->ptr < len) {
 174                 SB_expand(sb, len - sb->ptr);
 175                 memset(sb->buf + sb->ptr, ' ', len - sb->ptr);
 176         }
 177         sb->ptr = len;
 178 }
 179
 180 static void
 181 SB_reset(string_builder *sb)
 182 {
 183         SB_set_length(sb, 0);
 184 }
 185
 186 static void
 187 SB_free(string_builder *sb)
 188 {
 189         xfree(sb->buf);
 190         xfree(sb);
 191 }
 192
 193 typedef struct ht_elt_ {
 194         char *name;
 195         void *value;
 196         struct ht_elt_ *next;
 197 } ht_elt;
 198
 199 typedef struct {
 200         size_t size;
 201         ht_elt **buckets;
 202         size_t num_buckets;
 203 } HT;
 204
 205 static HT *
 206 HT_new(void)
 207 {
 208         HT *ht;
 209         size_t u;
 210
 211         ht = xmalloc(sizeof *ht);
 212         ht->size = 0;
 213         ht->num_buckets = 8;
 214         ht->buckets = xmalloc(ht->num_buckets * sizeof(ht_elt *));
 215         for (u = 0; u < ht->num_buckets; u ++) {
 216                 ht->buckets[u] = NULL;
 217         }
 218         return ht;
 219 }
 220
 221 static uint32_t
 222 hash_string(const char *name)
 223 {
 224         uint32_t hc;
 225
 226         hc = 0;
 227         while (*name) {
 228                 int x;
 229
 230                 hc = (hc << 5) - hc;
 231                 x = *(const unsigned char *)name;
 232                 if (x >= 'A' && x <= 'Z') {
 233                         x += 'a' - 'A';
 234                 }
 235                 hc += (uint32_t)x;
 236                 name ++;
 237         }
 238         return hc;
 239 }
 240
 241 static int
 242 eqstring(const char *s1, const char *s2)
 243 {
 244         while (*s1 && *s2) {
 245                 int x1, x2;
 246
 247                 x1 = *(const unsigned char *)s1;
 248                 x2 = *(const unsigned char *)s2;
 249                 if (x1 >= 'A' && x1 <= 'Z') {
 250                         x1 += 'a' - 'A';
 251                 }
 252                 if (x2 >= 'A' && x2 <= 'Z') {
 253                         x2 += 'a' - 'A';
 254                 }
 255                 if (x1 != x2) {
 256                         return 0;
 257                 }
 258                 s1 ++;
 259                 s2 ++;
 260         }
 261         return !(*s1 || *s2);
 262 }
 263
 264 static void
 265 HT_expand(HT *ht)
 266 {
 267         size_t n, n2, u;
 268         ht_elt **new_buckets;
 269
 270         n = ht->num_buckets;
 271         n2 = n << 1;
 272         new_buckets = xmalloc(n2 * sizeof *new_buckets);
 273         for (u = 0; u < n2; u ++) {
 274                 new_buckets[u] = NULL;
 275         }
 276         for (u = 0; u < n; u ++) {
 277                 ht_elt *e, *f;
 278
 279                 f = NULL;
 280                 for (e = ht->buckets[u]; e != NULL; e = f) {
 281                         uint32_t hc;
 282                         size_t v;
 283
 284                         hc = hash_string(e->name);
 285                         v = (size_t)(hc & ((uint32_t)n2 - 1));
 286                         f = e->next;
 287                         e->next = new_buckets[v];
 288                         new_buckets[v] = e;
 289                 }
 290         }
 291         xfree(ht->buckets);
 292         ht->buckets = new_buckets;
 293         ht->num_buckets = n2;
 294 }
 295
 296 static void *
 297 HT_put(HT *ht, const char *name, void *value)
 298 {
 299         uint32_t hc;
 300         size_t k;
 301         ht_elt *e, **prev;
 302
 303         hc = hash_string(name);
 304         k = (size_t)(hc & ((uint32_t)ht->num_buckets - 1));
 305         prev = &ht->buckets[k];
 306         e = *prev;
 307         while (e != NULL) {
 308                 if (eqstring(name, e->name)) {
 309                         void *old_value;
 310
 311                         old_value = e->value;
 312                         if (value == NULL) {
 313                                 *prev = e->next;
 314                                 xfree(e->name);
 315                                 xfree(e);
 316                                 ht->size --;
 317                         } else {
 318                                 e->value = value;
 319                         }
 320                         return old_value;
 321                 }
 322                 prev = &e->next;
 323                 e = *prev;
 324         }
 325         if (value != NULL) {
 326                 e = xmalloc(sizeof *e);
 327                 e->name = xstrdup(name);
 328                 e->value = value;
 329                 e->next = ht->buckets[k];
 330                 ht->buckets[k] = e;
 331                 ht->size ++;
 332                 if (ht->size > ht->num_buckets) {
 333                         HT_expand(ht);
 334                 }
 335         }
 336         return NULL;
 337 }
 338
 339 /* unused
 340 static void *
 341 HT_remove(HT *ht, const char *name)
 342 {
 343         return HT_put(ht, name, NULL);
 344 }
 345 */
 346
 347 static void *
 348 HT_get(const HT *ht, const char *name)
 349 {
 350         uint32_t hc;
 351         size_t k;
 352         ht_elt *e;
 353
 354         hc = hash_string(name);
 355         k = (size_t)(hc & ((uint32_t)ht->num_buckets - 1));
 356         for (e = ht->buckets[k]; e != NULL; e = e->next) {
 357                 if (eqstring(name, e->name)) {
 358                         return e->value;
 359                 }
 360         }
 361         return NULL;
 362 }
 363
 364 static void
 365 HT_clear(HT *ht, void (*free_value)(void *value))
 366 {
 367         size_t u;
 368
 369         for (u = 0; u < ht->num_buckets; u ++) {
 370                 ht_elt *e, *f;
 371
 372                 f = NULL;
 373                 for (e = ht->buckets[u]; e != NULL; e = f) {
 374                         f = e->next;
 375                         xfree(e->name);
 376                         if (free_value != 0) {
 377                                 free_value(e->value);
 378                         }
 379                         xfree(e);
 380                 }
 381                 ht->buckets[u] = NULL;
 382         }
 383         ht->size = 0;
 384 }
 385
 386 static void
 387 HT_free(HT *ht, void (*free_value)(void *value))
 388 {
 389         HT_clear(ht, free_value);
 390         xfree(ht->buckets);
 391         xfree(ht);
 392 }
 393
 394 /* unused
 395 static size_t
 396 HT_size(HT *ht)
 397 {
 398         return ht->size;
 399 }
 400 */
 401
 402 static unsigned char *
 403 read_all(FILE *f, size_t *len)
 404 {
 405         unsigned char *buf;
 406         size_t ptr, blen;
 407
 408         blen = 1024;
 409         buf = xmalloc(blen);
 410         ptr = 0;
 411         for (;;) {
 412                 size_t rlen;
 413
 414                 if (ptr == blen) {
 415                         unsigned char *buf2;
 416
 417                         blen <<= 1;
 418                         buf2 = xmalloc(blen);
 419                         memcpy(buf2, buf, ptr);
 420                         xfree(buf);
 421                         buf = buf2;
 422                 }
 423                 rlen = fread(buf + ptr, 1, blen - ptr, f);
 424                 if (rlen == 0) {
 425                         unsigned char *buf3;
 426
 427                         buf3 = xmalloc(ptr);
 428                         memcpy(buf3, buf, ptr);
 429                         xfree(buf);
 430                         *len = ptr;
 431                         return buf3;
 432                 }
 433                 ptr += rlen;
 434         }
 435 }
 436
 437 static unsigned char *
 438 read_file(const char *name, size_t *len)
 439 {
 440         FILE *f;
 441         unsigned char *buf;
 442
 443 #ifdef DIRNAME
 444         char *dname;
 445
 446         dname = xmalloc(strlen(DIRNAME) + strlen(name) + 2);
 447         sprintf(dname, "%s/%s", DIRNAME, name);
 448         name = dname;
 449 #endif
 450         f = fopen(name, "rb");
 451         if (f == NULL) {
 452                 fprintf(stderr, "could not open file '%s'\n", name);
 453                 exit(EXIT_FAILURE);
 454         }
 455         buf = read_all(f, len);
 456         if (ferror(f)) {
 457                 fprintf(stderr, "read error on file '%s'\n", name);
 458                 exit(EXIT_FAILURE);
 459         }
 460         fclose(f);
 461 #ifdef DIRNAME
 462         xfree(dname);
 463 #endif
 464         return buf;
 465 }
 466
 467 static int
 468 parse_dec(const char *s, unsigned len, int *val)
 469 {
 470         int acc;
 471
 472         acc = 0;
 473         while (len -- > 0) {
 474                 int c;
 475
 476                 c = *s ++;
 477                 if (c >= '0' && c <= '9') {
 478                         acc = (acc * 10) + (c - '0');
 479                 } else {
 480                         return -1;
 481                 }
 482         }
 483         *val = acc;
 484         return 0;
 485 }
 486
 487 static int
 488 parse_choice(const char *s, const char *acceptable)
 489 {
 490         int c;
 491
 492         c = *s;
 493         while (*acceptable) {
 494                 if (c == *acceptable ++) {
 495                         return 0;
 496                 }
 497         }
 498         return -1;
 499 }
 500
 501 static int
 502 month_length(int year, int month)
 503 {
 504         static const int base_month_length[] = {
 505                 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
 506         };
 507
 508         int x;
 509
 510         x = base_month_length[month - 1];
 511         if (month == 2 && year % 4 == 0
 512                 && (year % 100 != 0 || year % 400 == 0))
 513         {
 514                 x ++;
 515         }
 516         return x;
 517 }
 518
 519 /*
 520  * Convert a time string to a days+seconds count. Returned value is 0
 521  * on success, -1 on error.
 522  */
 523 static int
 524 string_to_time(const char *s, uint32_t *days, uint32_t *seconds)
 525 {
 526         int year, month, day, hour, minute, second;
 527         int day_of_year, leaps, i;
 528
 529         if (parse_dec(s, 4, &year) < 0) {
 530                 return -1;
 531         }
 532         s += 4;
 533         if (parse_choice(s ++, "-:/ ") < 0) {
 534                 return -1;
 535         }
 536         if (parse_dec(s, 2, &month) < 0) {
 537                 return -1;
 538         }
 539         s += 2;
 540         if (parse_choice(s ++, "-:/ ") < 0) {
 541                 return -1;
 542         }
 543         if (parse_dec(s, 2, &day) < 0) {
 544                 return -1;
 545         }
 546         s += 2;
 547         if (parse_choice(s ++, " T") < 0) {
 548                 return -1;
 549         }
 550         if (parse_dec(s, 2, &hour) < 0) {
 551                 return -1;
 552         }
 553         s += 2;
 554         if (parse_choice(s ++, "-:/ ") < 0) {
 555                 return -1;
 556         }
 557         if (parse_dec(s, 2, &minute) < 0) {
 558                 return -1;
 559         }
 560         s += 2;
 561         if (parse_choice(s ++, "-:/ ") < 0) {
 562                 return -1;
 563         }
 564         if (parse_dec(s, 2, &second) < 0) {
 565                 return -1;
 566         }
 567         s += 2;
 568         if (*s == '.') {
 569                 while (*s && *s >= '0' && *s <= '9') {
 570                         s ++;
 571                 }
 572         }
 573         if (*s) {
 574                 if (*s ++ != 'Z') {
 575                         return -1;
 576                 }
 577                 if (*s) {
 578                         return -1;
 579                 }
 580         }
 581
 582         if (month < 1 || month > 12) {
 583                 return -1;
 584         }
 585         day_of_year = 0;
 586         for (i = 1; i < month; i ++) {
 587                 day_of_year += month_length(year, i);
 588         }
 589         if (day < 1 || day > month_length(year, month)) {
 590                 return -1;
 591         }
 592         day_of_year += (day - 1);
 593         leaps = (year + 3) / 4 - (year + 99) / 100 + (year + 399) / 400;
 594
 595         if (hour > 23 || minute > 59 || second > 60) {
 596                 return -1;
 597         }
 598         *days = (uint32_t)year * 365 + (uint32_t)leaps + day_of_year;
 599         *seconds = (uint32_t)hour * 3600 + minute * 60 + second;
 600         return 0;
 601 }
 602
 603 static FILE *conf;
 604 static int conf_delayed_char;
 605 static long conf_linenum;
 606 static string_builder *line_builder;
 607 static long current_linenum;
 608
 609 static void
 610 conf_init(const char *fname)
 611 {
 612         conf = fopen(fname, "r");
 613         if (conf == NULL) {
 614                 fprintf(stderr, "could not open file '%s'\n", fname);
 615                 exit(EXIT_FAILURE);
 616         }
 617         conf_delayed_char = -1;
 618         conf_linenum = 1;
 619         line_builder = SB_new();
 620 }
 621
 622 static void
 623 conf_close(void)
 624 {
 625         if (conf != NULL) {
 626                 if (ferror(conf)) {
 627                         fprintf(stderr, "read error on configuration file\n");
 628                         exit(EXIT_FAILURE);
 629                 }
 630                 fclose(conf);
 631                 conf = NULL;
 632         }
 633         if (line_builder != NULL) {
 634                 SB_free(line_builder);
 635                 line_builder = NULL;
 636         }
 637 }
 638
 639 /*
 640  * Get next character from the config file.
 641  */
 642 static int
 643 conf_next_low(void)
 644 {
 645         int x;
 646
 647         x = conf_delayed_char;
 648         if (x >= 0) {
 649                 conf_delayed_char = -1;
 650         } else {
 651                 x = fgetc(conf);
 652                 if (x == EOF) {
 653                         x = -1;
 654                 }
 655         }
 656         if (x == '\r') {
 657                 x = fgetc(conf);
 658                 if (x == EOF) {
 659                         x = -1;
 660                 }
 661                 if (x != '\n') {
 662                         conf_delayed_char = x;
 663                         x = '\n';
 664                 }
 665         }
 666         if (x == '\n') {
 667                 conf_linenum ++;
 668         }
 669         return x;
 670 }
 671
 672 static int
 673 is_ws(int x)
 674 {
 675         return x <= 32;
 676 }
 677
 678 static int
 679 is_name_char(int c)
 680 {
 681         return (c >= 'A' && c <= 'Z')
 682                 || (c >= 'a' && c <= 'z')
 683                 || (c >= '0' && c <= '9')
 684                 || (c == '_' || c == '-' || c == '.');
 685 }
 686
 687 /*
 688  * Read a complete line. This handles line continuation; empty lines and
 689  * comment lines are skipped; leading and trailing whitespace is removed.
 690  * Returned value is 0 (line read) or -1 (no line, EOF reached). The line
 691  * contents are accumulated in the line_builder.
 692  */
 693 static int
 694 conf_next_line(void)
 695 {
 696         for (;;) {
 697                 int c;
 698                 int lcwb;
 699
 700                 SB_reset(line_builder);
 701
 702                 /*
 703                  * Get first non-whitespace character. This skips empty
 704                  * lines. Comment lines (first non-whitespace character
 705                  * is a semicolon) are also skipped.
 706                  */
 707                 for (;;) {
 708                         c = conf_next_low();
 709                         if (c < 0) {
 710                                 return -1;
 711                         }
 712                         if (is_ws(c)) {
 713                                 continue;
 714                         }
 715                         if (c == ';') {
 716                                 for (;;) {
 717                                         c = conf_next_low();
 718                                         if (c < 0) {
 719                                                 return -1;
 720                                         }
 721                                         if (c == '\n') {
 722                                                 break;
 723                                         }
 724                                 }
 725                                 continue;
 726                         }
 727                         break;
 728                 }
 729
 730                 /*
 731                  * Read up the remaining of the line. The line continuation
 732                  * sequence (final backslash) is detected and processed.
 733                  */
 734                 current_linenum = conf_linenum;
 735                 lcwb = (c == '\\');
 736                 SB_append_char(line_builder, c);
 737                 for (;;) {
 738                         c = conf_next_low();
 739                         if (c < 0) {
 740                                 break;
 741                         }
 742                         if (lcwb) {
 743                                 if (c == '\n') {
 744                                         SB_set_length(line_builder,
 745                                                 SB_length(line_builder) - 1);
 746                                 }
 747                                 lcwb = 0;
 748                                 continue;
 749                         }
 750                         if (c == '\n') {
 751                                 break;
 752                         } else if (c == '\\') {
 753                                 lcwb = 1;
 754                         }
 755                         SB_append_char(line_builder, c);
 756                 }
 757
 758                 /*
 759                  * Remove trailing whitespace (if any).
 760                  */
 761                 for (;;) {
 762                         size_t u;
 763
 764                         u = SB_length(line_builder);
 765                         if (u == 0 || !is_ws(
 766                                 SB_contents(line_builder)[u - 1]))
 767                         {
 768                                 break;
 769                         }
 770                         SB_set_length(line_builder, u - 1);
 771                 }
 772
 773                 /*
 774                  * We might end up with a totally empty line (in case there
 775                  * was a line continuation but nothing else), in which case
 776                  * we must loop.
 777                  */
 778                 if (SB_length(line_builder) > 0) {
 779                         return 0;
 780                 }
 781         }
 782 }
 783
 784 /*
 785  * Test whether the current line is a section header. If yes, then the
 786  * header name is extracted, and returned as a newly allocated string.
 787  * Otherwise, NULL is returned.
 788  */
 789 static char *
 790 parse_header_name(void)
 791 {
 792         char *buf, *name;
 793         size_t u, v, w, len;
 794
 795         buf = SB_contents(line_builder);
 796         len = SB_length(line_builder);
 797         if (len < 2 || buf[0] != '[' || buf[len - 1] != ']') {
 798                 return NULL;
 799         }
 800         u = 1;
 801         v = len - 1;
 802         while (u < v && is_ws(buf[u])) {
 803                 u ++;
 804         }
 805         while (u < v && is_ws(buf[v - 1])) {
 806                 v --;
 807         }
 808         if (u == v) {
 809                 return NULL;
 810         }
 811         for (w = u; w < v; w ++) {
 812                 if (!is_name_char(buf[w])) {
 813                         return NULL;
 814                 }
 815         }
 816         len = v - u;
 817         name = xmalloc(len + 1);
 818         memcpy(name, buf + u, len);
 819         name[len] = 0;
 820         return name;
 821 }
 822
 823 /*
 824  * Parse the current line as a 'name = value' pair. The pair is pushed into
 825  * the provided hash table. On error (including a duplicate key name),
 826  * this function returns -1; otherwise, it returns 0.
 827  */
 828 static int
 829 parse_keyvalue(HT *d)
 830 {
 831         char *buf, *name, *value;
 832         size_t u, len;
 833
 834         buf = SB_contents(line_builder);
 835         len = SB_length(line_builder);
 836         for (u = 0; u < len; u ++) {
 837                 if (!is_name_char(buf[u])) {
 838                         break;
 839                 }
 840         }
 841         if (u == 0) {
 842                 return -1;
 843         }
 844         name = xmalloc(u + 1);
 845         memcpy(name, buf, u);
 846         name[u] = 0;
 847         if (HT_get(d, name) != NULL) {
 848                 xfree(name);
 849                 return -1;
 850         }
 851         while (u < len && is_ws(buf[u])) {
 852                 u ++;
 853         }
 854         if (u >= len || buf[u] != '=') {
 855                 xfree(name);
 856                 return -1;
 857         }
 858         u ++;
 859         while (u < len && is_ws(buf[u])) {
 860                 u ++;
 861         }
 862         value = xmalloc(len - u + 1);
 863         memcpy(value, buf + u, len - u);
 864         value[len - u] = 0;
 865         HT_put(d, name, value);
 866         xfree(name);
 867         return 0;
 868 }
 869
 870 /*
 871  * Public keys, indexed by name. Elements are pointers to br_x509_pkey
 872  * structures.
 873  */
 874 static HT *keys;
 875
 876 /*
 877  * Trust anchors, indexed by name. Elements are pointers to
 878  * test_trust_anchor structures.
 879  */
 880 static HT *trust_anchors;
 881
 882 typedef struct {
 883         unsigned char *dn;
 884         size_t dn_len;
 885         unsigned flags;
 886         char *key_name;
 887 } test_trust_anchor;
 888
 889 /*
 890  * Test case: trust anchors, certificates (file names), key type and
 891  * usage, expected status and EE public key.
 892  */
 893 typedef struct {
 894         char *name;
 895         char **ta_names;
 896         char **cert_names;
 897         char *servername;
 898         unsigned key_type_usage;
 899         unsigned status;
 900         char *ee_key_name;
 901         unsigned hashes;
 902         uint32_t days, seconds;
 903 } test_case;
 904
 905 static test_case *all_chains;
 906 static size_t all_chains_ptr, all_chains_len;
 907
 908 static void
 909 free_key(void *value)
 910 {
 911         br_x509_pkey *pk;
 912
 913         pk = value;
 914         switch (pk->key_type) {
 915         case BR_KEYTYPE_RSA:
 916                 xfree((void *)pk->key.rsa.n);
 917                 xfree((void *)pk->key.rsa.e);
 918                 break;
 919         case BR_KEYTYPE_EC:
 920                 xfree((void *)pk->key.ec.q);
 921                 break;
 922         default:
 923                 fprintf(stderr, "unknown key type: %d\n", pk->key_type);
 924                 exit(EXIT_FAILURE);
 925                 break;
 926         }
 927         xfree(pk);
 928 }
 929
 930 static void
 931 free_trust_anchor(void *value)
 932 {
 933         test_trust_anchor *ttc;
 934
 935         ttc = value;
 936         xfree(ttc->dn);
 937         xfree(ttc->key_name);
 938         xfree(ttc);
 939 }
 940
 941 static void
 942 free_test_case_contents(test_case *tc)
 943 {
 944         size_t u;
 945
 946         xfree(tc->name);
 947         for (u = 0; tc->ta_names[u]; u ++) {
 948                 xfree(tc->ta_names[u]);
 949         }
 950         xfree(tc->ta_names);
 951         for (u = 0; tc->cert_names[u]; u ++) {
 952                 xfree(tc->cert_names[u]);
 953         }
 954         xfree(tc->cert_names);
 955         xfree(tc->servername);
 956         xfree(tc->ee_key_name);
 957 }
 958
 959 static char *
 960 get_value(char *objtype, HT *objdata, long linenum, char *name)
 961 {
 962         char *value;
 963
 964         value = HT_get(objdata, name);
 965         if (value == NULL) {
 966                 fprintf(stderr,
 967                         "missing property '%s' in section '%s' (line %ld)\n",
 968                         name, objtype, linenum);
 969                 exit(EXIT_FAILURE);
 970         }
 971         return value;
 972 }
 973
 974 static unsigned char *
 975 parse_hex(const char *name, long linenum, const char *value, size_t *len)
 976 {
 977         unsigned char *buf;
 978
 979         buf = NULL;
 980         for (;;) {
 981                 size_t u, ptr;
 982                 int acc, z;
 983
 984                 ptr = 0;
 985                 acc = 0;
 986                 z = 0;
 987                 for (u = 0; value[u]; u ++) {
 988                         int c;
 989
 990                         c = value[u];
 991                         if (c >= '0' && c <= '9') {
 992                                 c -= '0';
 993                         } else if (c >= 'A' && c <= 'F') {
 994                                 c -= 'A' - 10;
 995                         } else if (c >= 'a' && c <= 'f') {
 996                                 c -= 'a' - 10;
 997                         } else if (c == ' ' || c == ':') {
 998                                 continue;
 999                         } else {
1000                                 fprintf(stderr, "invalid hexadecimal character"
1001                                         " in '%s' (line %ld)\n",
1002                                         name, linenum);
1003                                 exit(EXIT_FAILURE);
1004                         }
1005                         if (z) {
1006                                 if (buf != NULL) {
1007                                         buf[ptr] = (acc << 4) + c;
1008                                 }
1009                                 ptr ++;
1010                         } else {
1011                                 acc = c;
1012                         }
1013                         z = !z;
1014                 }
1015                 if (z) {
1016                         fprintf(stderr, "invalid hexadecimal value (partial"
1017                                 " byte) in '%s' (line %ld)\n",
1018                                 name, linenum);
1019                         exit(EXIT_FAILURE);
1020                 }
1021                 if (buf == NULL) {
1022                         buf = xmalloc(ptr);
1023                 } else {
1024                         *len = ptr;
1025                         return buf;
1026                 }
1027         }
1028 }
1029
1030 static char **
1031 split_names(const char *value)
1032 {
1033         char **names;
1034         size_t len;
1035
1036         names = NULL;
1037         len = strlen(value);
1038         for (;;) {
1039                 size_t u, ptr;
1040
1041                 ptr = 0;
1042                 u = 0;
1043                 while (u < len) {
1044                         size_t v;
1045
1046                         while (u < len && is_ws(value[u])) {
1047                                 u ++;
1048                         }
1049                         v = u;
1050                         while (v < len && !is_ws(value[v])) {
1051                                 v ++;
1052                         }
1053                         if (v > u) {
1054                                 if (names != NULL) {
1055                                         char *name;
1056
1057                                         name = xmalloc(v - u + 1);
1058                                         memcpy(name, value + u, v - u);
1059                                         name[v - u] = 0;
1060                                         names[ptr] = name;
1061                                 }
1062                                 ptr ++;
1063                         }
1064                         u = v;
1065                 }
1066                 if (names == NULL) {
1067                         names = xmalloc((ptr + 1) * sizeof *names);
1068                 } else {
1069                         names[ptr] = NULL;
1070                         return names;
1071                 }
1072         }
1073 }
1074
1075 static int
1076 string_to_hash(const char *name)
1077 {
1078         char tmp[20];
1079         size_t u, v;
1080
1081         for (u = 0, v = 0; name[u]; u ++) {
1082                 int c;
1083
1084                 c = name[u];
1085                 if ((c >= '0' && c <= '9')
1086                         || (c >= 'A' && c <= 'Z')
1087                         || (c >= 'a' && c <= 'z'))
1088                 {
1089                         tmp[v ++] = c;
1090                         if (v == sizeof tmp) {
1091                                 return -1;
1092                         }
1093                 }
1094         }
1095         tmp[v] = 0;
1096         if (eqstring(tmp, "md5")) {
1097                 return br_md5_ID;
1098         } else if (eqstring(tmp, "sha1")) {
1099                 return br_sha1_ID;
1100         } else if (eqstring(tmp, "sha224")) {
1101                 return br_sha224_ID;
1102         } else if (eqstring(tmp, "sha256")) {
1103                 return br_sha256_ID;
1104         } else if (eqstring(tmp, "sha384")) {
1105                 return br_sha384_ID;
1106         } else if (eqstring(tmp, "sha512")) {
1107                 return br_sha512_ID;
1108         } else {
1109                 return -1;
1110         }
1111 }
1112
1113 static int
1114 string_to_curve(const char *name)
1115 {
1116         char tmp[20];
1117         size_t u, v;
1118
1119         for (u = 0, v = 0; name[u]; u ++) {
1120                 int c;
1121
1122                 c = name[u];
1123                 if ((c >= '0' && c <= '9')
1124                         || (c >= 'A' && c <= 'Z')
1125                         || (c >= 'a' && c <= 'z'))
1126                 {
1127                         tmp[v ++] = c;
1128                         if (v == sizeof tmp) {
1129                                 return -1;
1130                         }
1131                 }
1132         }
1133         tmp[v] = 0;
1134         if (eqstring(tmp, "p256") || eqstring(tmp, "secp256r1")) {
1135                 return BR_EC_secp256r1;
1136         } else if (eqstring(tmp, "p384") || eqstring(tmp, "secp384r1")) {
1137                 return BR_EC_secp384r1;
1138         } else if (eqstring(tmp, "p521") || eqstring(tmp, "secp521r1")) {
1139                 return BR_EC_secp521r1;
1140         } else {
1141                 return -1;
1142         }
1143 }
1144
1145 static void
1146 parse_object(char *objtype, HT *objdata, long linenum)
1147 {
1148         char *name;
1149
1150         name = get_value(objtype, objdata, linenum, "name");
1151         if (eqstring(objtype, "key")) {
1152                 char *stype;
1153                 br_x509_pkey *pk;
1154
1155                 stype = get_value(objtype, objdata, linenum, "type");
1156                 pk = xmalloc(sizeof *pk);
1157                 if (eqstring(stype, "RSA")) {
1158                         char *sn, *se;
1159
1160                         sn = get_value(objtype, objdata, linenum, "n");
1161                         se = get_value(objtype, objdata, linenum, "e");
1162                         pk->key_type = BR_KEYTYPE_RSA;
1163                         pk->key.rsa.n = parse_hex("modulus", linenum,
1164                                 sn, &pk->key.rsa.nlen);
1165                         pk->key.rsa.e = parse_hex("exponent", linenum,
1166                                 se, &pk->key.rsa.elen);
1167                 } else if (eqstring(stype, "EC")) {
1168                         char *sc, *sq;
1169                         int curve;
1170
1171                         sc = get_value(objtype, objdata, linenum, "curve");
1172                         sq = get_value(objtype, objdata, linenum, "q");
1173                         curve = string_to_curve(sc);
1174                         if (curve < 0) {
1175                                 fprintf(stderr, "unknown curve name: '%s'"
1176                                         " (line %ld)\n", sc, linenum);
1177                                 exit(EXIT_FAILURE);
1178                         }
1179                         pk->key_type = BR_KEYTYPE_EC;
1180                         pk->key.ec.curve = curve;
1181                         pk->key.ec.q = parse_hex("public point", linenum,
1182                                 sq, &pk->key.ec.qlen);
1183                 } else {
1184                         fprintf(stderr, "unknown key type '%s' (line %ld)\n",
1185                                 stype, linenum);
1186                         exit(EXIT_FAILURE);
1187                 }
1188                 if (HT_put(keys, name, pk) != NULL) {
1189                         fprintf(stderr, "duplicate key: '%s' (line %ld)\n",
1190                                 name, linenum);
1191                         exit(EXIT_FAILURE);
1192                 }
1193         } else if (eqstring(objtype, "anchor")) {
1194                 char *dnfile, *kname, *tatype;
1195                 test_trust_anchor *tta;
1196
1197                 dnfile = get_value(objtype, objdata, linenum, "DN_file");
1198                 kname = get_value(objtype, objdata, linenum, "key");
1199                 tatype = get_value(objtype, objdata, linenum, "type");
1200                 tta = xmalloc(sizeof *tta);
1201                 tta->dn = read_file(dnfile, &tta->dn_len);
1202                 tta->key_name = xstrdup(kname);
1203                 if (eqstring(tatype, "CA")) {
1204                         tta->flags = BR_X509_TA_CA;
1205                 } else if (eqstring(tatype, "EE")) {
1206                         tta->flags = 0;
1207                 } else {
1208                         fprintf(stderr,
1209                                 "unknown trust anchor type: '%s' (line %ld)\n",
1210                                 tatype, linenum);
1211                 }
1212                 if (HT_put(trust_anchors, name, tta) != NULL) {
1213                         fprintf(stderr,
1214                                 "duplicate trust anchor: '%s' (line %ld)\n",
1215                                 name, linenum);
1216                         exit(EXIT_FAILURE);
1217                 }
1218         } else if (eqstring(objtype, "chain")) {
1219                 test_case tc;
1220                 char *ktype, *kusage, *sstatus, *shashes, *stime;
1221
1222                 ktype = get_value(objtype, objdata, linenum, "keytype");
1223                 kusage = get_value(objtype, objdata, linenum, "keyusage");
1224                 sstatus = get_value(objtype, objdata, linenum, "status");
1225                 tc.name = xstrdup(name);
1226                 tc.ta_names = split_names(
1227                         get_value(objtype, objdata, linenum, "anchors"));
1228                 tc.cert_names = split_names(
1229                         get_value(objtype, objdata, linenum, "chain"));
1230                 tc.servername = xstrdup(HT_get(objdata, "servername"));
1231                 if (eqstring(ktype, "RSA")) {
1232                         tc.key_type_usage = BR_KEYTYPE_RSA;
1233                 } else if (eqstring(ktype, "EC")) {
1234                         tc.key_type_usage = BR_KEYTYPE_EC;
1235                 } else {
1236                         fprintf(stderr,
1237                                 "unknown key type: '%s' (line %ld)\n",
1238                                 ktype, linenum);
1239                         exit(EXIT_FAILURE);
1240                 }
1241                 if (eqstring(kusage, "KEYX")) {
1242                         tc.key_type_usage |= BR_KEYTYPE_KEYX;
1243                 } else if (eqstring(kusage, "SIGN")) {
1244                         tc.key_type_usage |= BR_KEYTYPE_SIGN;
1245                 } else {
1246                         fprintf(stderr,
1247                                 "unknown key usage: '%s' (line %ld)\n",
1248                                 kusage, linenum);
1249                         exit(EXIT_FAILURE);
1250                 }
1251                 tc.status = (unsigned)atoi(sstatus);
1252                 if (tc.status == 0) {
1253                         tc.ee_key_name = xstrdup(
1254                                 get_value(objtype, objdata, linenum, "eekey"));
1255                 } else {
1256                         tc.ee_key_name = NULL;
1257                 }
1258                 shashes = HT_get(objdata, "hashes");
1259                 if (shashes == NULL) {
1260                         tc.hashes = (unsigned)-1;
1261                 } else {
1262                         char **hns;
1263                         size_t u;
1264
1265                         tc.hashes = 0;
1266                         hns = split_names(shashes);
1267                         for (u = 0;; u ++) {
1268                                 char *hn;
1269                                 int id;
1270
1271                                 hn = hns[u];
1272                                 if (hn == NULL) {
1273                                         break;
1274                                 }
1275                                 id = string_to_hash(hn);
1276                                 if (id < 0) {
1277                                         fprintf(stderr,
1278                                                 "unknown hash function '%s'"
1279                                                 " (line %ld)\n", hn, linenum);
1280                                         exit(EXIT_FAILURE);
1281                                 }
1282                                 tc.hashes |= (unsigned)1 << id;
1283                                 xfree(hn);
1284                         }
1285                         xfree(hns);
1286                 }
1287                 stime = HT_get(objdata, "time");
1288                 if (stime == NULL) {
1289                         stime = DEFAULT_TIME;
1290                 }
1291                 if (string_to_time(stime, &tc.days, &tc.seconds) < 0) {
1292                         fprintf(stderr, "invalid time string '%s' (line %ld)\n",
1293                                 stime, linenum);
1294                         exit(EXIT_FAILURE);
1295                 }
1296                 if (all_chains_ptr == all_chains_len) {
1297                         if (all_chains_len == 0) {
1298                                 all_chains_len = 8;
1299                                 all_chains = xmalloc(
1300                                         all_chains_len * sizeof *all_chains);
1301                         } else {
1302                                 test_case *ntc;
1303                                 size_t nlen;
1304
1305                                 nlen = all_chains_len << 1;
1306                                 ntc = xmalloc(nlen * sizeof *ntc);
1307                                 memcpy(ntc, all_chains,
1308                                         all_chains_len * sizeof *all_chains);
1309                                 xfree(all_chains);
1310                                 all_chains = ntc;
1311                                 all_chains_len = nlen;
1312                         }
1313                 }
1314                 all_chains[all_chains_ptr ++] = tc;
1315         } else {
1316                 fprintf(stderr, "unknown section type '%s' (line %ld)\n",
1317                         objtype, linenum);
1318                 exit(EXIT_FAILURE);
1319         }
1320 }
1321
1322 static void
1323 process_conf_file(const char *fname)
1324 {
1325         char *objtype;
1326         HT *objdata;
1327         long objlinenum;
1328
1329         keys = HT_new();
1330         trust_anchors = HT_new();
1331         all_chains = NULL;
1332         all_chains_ptr = 0;
1333         all_chains_len = 0;
1334         conf_init(fname);
1335         objtype = NULL;
1336         objdata = HT_new();
1337         objlinenum = 0;
1338         for (;;) {
1339                 char *hname;
1340
1341                 if (conf_next_line() < 0) {
1342                         break;
1343                 }
1344                 hname = parse_header_name();
1345                 if (hname != NULL) {
1346                         if (objtype != NULL) {
1347                                 parse_object(objtype, objdata, objlinenum);
1348                                 HT_clear(objdata, xfree);
1349                                 xfree(objtype);
1350                         }
1351                         objtype = hname;
1352                         objlinenum = current_linenum;
1353                         continue;
1354                 }
1355                 if (objtype == NULL) {
1356                         fprintf(stderr, "no current section (line %ld)\n",
1357                                 current_linenum);
1358                         exit(EXIT_FAILURE);
1359                 }
1360                 if (parse_keyvalue(objdata) < 0) {
1361                         fprintf(stderr, "wrong configuration, line %ld\n",
1362                                 current_linenum);
1363                         exit(EXIT_FAILURE);
1364                 }
1365         }
1366         if (objtype != NULL) {
1367                 parse_object(objtype, objdata, objlinenum);
1368                 xfree(objtype);
1369         }
1370         HT_free(objdata, xfree);
1371         conf_close();
1372 }
1373
1374 static const struct {
1375         int id;
1376         const br_hash_class *impl;
1377 } hash_impls[] = {
1378         { br_md5_ID, &br_md5_vtable },
1379         { br_sha1_ID, &br_sha1_vtable },
1380         { br_sha224_ID, &br_sha224_vtable },
1381         { br_sha256_ID, &br_sha256_vtable },
1382         { br_sha384_ID, &br_sha384_vtable },
1383         { br_sha512_ID, &br_sha512_vtable },
1384         { 0, NULL }
1385 };
1386
1387 typedef struct {
1388         unsigned char *data;
1389         size_t len;
1390 } blob;
1391
1392 static int
1393 eqbigint(const unsigned char *b1, size_t b1_len,
1394         const unsigned char *b2, size_t b2_len)
1395 {
1396         while (b1_len > 0 && *b1 == 0) {
1397                 b1 ++;
1398                 b1_len --;
1399         }
1400         while (b2_len > 0 && *b2 == 0) {
1401                 b2 ++;
1402                 b2_len --;
1403         }
1404         return b1_len == b2_len && memcmp(b1, b2, b1_len) == 0;
1405 }
1406
1407 static int
1408 eqpkey(const br_x509_pkey *pk1, const br_x509_pkey *pk2)
1409 {
1410         if (pk1 == pk2) {
1411                 return 1;
1412         }
1413         if (pk1 == NULL || pk2 == NULL) {
1414                 return 0;
1415         }
1416         if (pk1->key_type != pk2->key_type) {
1417                 return 0;
1418         }
1419         switch (pk1->key_type) {
1420         case BR_KEYTYPE_RSA:
1421                 return eqbigint(pk1->key.rsa.n, pk1->key.rsa.nlen,
1422                         pk2->key.rsa.n, pk2->key.rsa.nlen)
1423                         && eqbigint(pk1->key.rsa.e, pk1->key.rsa.elen,
1424                         pk2->key.rsa.e, pk2->key.rsa.elen);
1425         case BR_KEYTYPE_EC:
1426                 return pk1->key.ec.curve == pk2->key.ec.curve
1427                         && pk1->key.ec.qlen == pk2->key.ec.qlen
1428                         && memcmp(pk1->key.ec.q,
1429                                 pk2->key.ec.q, pk1->key.ec.qlen) == 0;
1430         default:
1431                 fprintf(stderr, "unknown key type: %d\n", pk1->key_type);
1432                 exit(EXIT_FAILURE);
1433                 break;
1434         }
1435         return 0;
1436 }
1437
1438 static size_t max_dp_usage;
1439 static size_t max_rp_usage;
1440
1441 static void
1442 run_test_case(test_case *tc)
1443 {
1444         br_x509_minimal_context ctx;
1445         br_x509_trust_anchor *anchors;
1446         size_t num_anchors;
1447         size_t u;
1448         const br_hash_class *dnhash;
1449         size_t num_certs;
1450         blob *certs;
1451         br_x509_pkey *ee_pkey_ref;
1452         const br_x509_pkey *ee_pkey;
1453         unsigned usages;
1454         unsigned status;
1455
1456         printf("%s: ", tc->name);
1457         fflush(stdout);
1458
1459         /*
1460          * Get the hash function to use for hashing DN. We can use just
1461          * any supported hash function, but for the elegance of things,
1462          * we will use one of the hash function implementations
1463          * supported for this test case (with SHA-1 as fallback).
1464          */
1465         dnhash = &br_sha1_vtable;
1466         for (u = 0; hash_impls[u].id; u ++) {
1467                 if ((tc->hashes & ((unsigned)1 << (hash_impls[u].id))) != 0) {
1468                         dnhash = hash_impls[u].impl;
1469                 }
1470         }
1471
1472         /*
1473          * Get trust anchors.
1474          */
1475         for (num_anchors = 0; tc->ta_names[num_anchors]; num_anchors ++);
1476         anchors = xmalloc(num_anchors * sizeof *anchors);
1477         for (u = 0; tc->ta_names[u]; u ++) {
1478                 test_trust_anchor *tta;
1479                 br_x509_pkey *tak;
1480
1481                 tta = HT_get(trust_anchors, tc->ta_names[u]);
1482                 if (tta == NULL) {
1483                         fprintf(stderr, "no such trust anchor: '%s'\n",
1484                                 tc->ta_names[u]);
1485                         exit(EXIT_FAILURE);
1486                 }
1487                 tak = HT_get(keys, tta->key_name);
1488                 if (tak == NULL) {
1489                         fprintf(stderr, "no such public key: '%s'\n",
1490                                 tta->key_name);
1491                         exit(EXIT_FAILURE);
1492                 }
1493                 anchors[u].dn.data = tta->dn;
1494                 anchors[u].dn.len = tta->dn_len;
1495                 anchors[u].flags = tta->flags;
1496                 anchors[u].pkey = *tak;
1497         }
1498
1499         /*
1500          * Read all relevant certificates.
1501          */
1502         for (num_certs = 0; tc->cert_names[num_certs]; num_certs ++);
1503         certs = xmalloc(num_certs * sizeof *certs);
1504         for (u = 0; u < num_certs; u ++) {
1505                 certs[u].data = read_file(tc->cert_names[u], &certs[u].len);
1506         }
1507
1508         /*
1509          * Get expected EE public key (if any).
1510          */
1511         if (tc->ee_key_name == NULL) {
1512                 ee_pkey_ref = NULL;
1513         } else {
1514                 ee_pkey_ref = HT_get(keys, tc->ee_key_name);
1515                 if (ee_pkey_ref == NULL) {
1516                         fprintf(stderr, "no such public key: '%s'\n",
1517                                 tc->ee_key_name);
1518                         exit(EXIT_FAILURE);
1519                 }
1520         }
1521
1522         /*
1523          * Initialise the engine.
1524          */
1525         br_x509_minimal_init(&ctx, dnhash, anchors, num_anchors);
1526         for (u = 0; hash_impls[u].id; u ++) {
1527                 int id;
1528
1529                 id = hash_impls[u].id;
1530                 if ((tc->hashes & ((unsigned)1 << id)) != 0) {
1531                         br_x509_minimal_set_hash(&ctx, id, hash_impls[u].impl);
1532                 }
1533         }
1534         br_x509_minimal_set_rsa(&ctx, br_rsa_pkcs1_vrfy_get_default());
1535         br_x509_minimal_set_ecdsa(&ctx,
1536                 br_ec_get_default(), br_ecdsa_vrfy_asn1_get_default());
1537
1538         /*
1539          * Set the validation date.
1540          */
1541         br_x509_minimal_set_time(&ctx, tc->days, tc->seconds);
1542
1543         /*
1544          * Put "canaries" to detect actual stack usage.
1545          */
1546         for (u = 0; u < (sizeof ctx.dp_stack) / sizeof(uint32_t); u ++) {
1547                 ctx.dp_stack[u] = 0xA7C083FE;
1548         }
1549         for (u = 0; u < (sizeof ctx.rp_stack) / sizeof(uint32_t); u ++) {
1550                 ctx.rp_stack[u] = 0xA7C083FE;
1551         }
1552
1553         /*
1554          * Run the engine. We inject certificates by chunks of 100 bytes
1555          * in order to exercise the coroutine API.
1556          */
1557         ctx.vtable->start_chain(&ctx.vtable, tc->servername);
1558         for (u = 0; u < num_certs; u ++) {
1559                 size_t v;
1560
1561                 ctx.vtable->start_cert(&ctx.vtable, certs[u].len);
1562                 v = 0;
1563                 while (v < certs[u].len) {
1564                         size_t w;
1565
1566                         w = certs[u].len - v;
1567                         if (w > 100) {
1568                                 w = 100;
1569                         }
1570                         ctx.vtable->append(&ctx.vtable, certs[u].data + v, w);
1571                         v += w;
1572                 }
1573                 ctx.vtable->end_cert(&ctx.vtable);
1574         }
1575         status = ctx.vtable->end_chain(&ctx.vtable);
1576         ee_pkey = ctx.vtable->get_pkey(&ctx.vtable, &usages);
1577
1578         /*
1579          * Check key type and usage.
1580          */
1581         if (ee_pkey != NULL) {
1582                 unsigned ktu;
1583
1584                 ktu = ee_pkey->key_type | usages;
1585                 if (tc->key_type_usage != (ktu & tc->key_type_usage)) {
1586                         fprintf(stderr, "wrong key type + usage"
1587                                 " (expected 0x%02X, got 0x%02X)\n",
1588                                 tc->key_type_usage, ktu);
1589                         exit(EXIT_FAILURE);
1590                 }
1591         }
1592
1593         /*
1594          * Check results. Note that we may still get a public key if
1595          * the path is "not trusted" (but otherwise fine).
1596          */
1597         if (status != tc->status) {
1598                 fprintf(stderr, "wrong status (got %d, expected %d)\n",
1599                         status, tc->status);
1600                 exit(EXIT_FAILURE);
1601         }
1602         if (status == BR_ERR_X509_NOT_TRUSTED) {
1603                 ee_pkey = NULL;
1604         }
1605         if (!eqpkey(ee_pkey, ee_pkey_ref)) {
1606                 fprintf(stderr, "wrong EE public key\n");
1607                 exit(EXIT_FAILURE);
1608         }
1609
1610         /*
1611          * Check stack usage.
1612          */
1613         for (u = (sizeof ctx.dp_stack) / sizeof(uint32_t); u > 0; u --) {
1614                 if (ctx.dp_stack[u - 1] != 0xA7C083FE) {
1615                         if (max_dp_usage < u) {
1616                                 max_dp_usage = u;
1617                         }
1618                         break;
1619                 }
1620         }
1621         for (u = (sizeof ctx.rp_stack) / sizeof(uint32_t); u > 0; u --) {
1622                 if (ctx.rp_stack[u - 1] != 0xA7C083FE) {
1623                         if (max_rp_usage < u) {
1624                                 max_rp_usage = u;
1625                         }
1626                         break;
1627                 }
1628         }
1629
1630         /*
1631          * Release everything.
1632          */
1633         for (u = 0; u < num_certs; u ++) {
1634                 xfree(certs[u].data);
1635         }
1636         xfree(certs);
1637         xfree(anchors);
1638         printf("OK\n");
1639 }
1640
1641 /*
1642  * A custom structure for tests, synchronised with the test certificate
1643  * names.crt.
1644  *
1645  * If num is 1 or more, then this is a DN element with OID '1.1.1.1.num'.
1646  * If num is -1 or less, then this is a SAN element of type -num.
1647  * If num is 0, then this is a SAN element of type OtherName with
1648  * OID 1.3.6.1.4.1.311.20.2.3 (Microsoft UPN).
1649  */
1650 typedef struct {
1651         int num;
1652         int status;
1653         const char *expected;
1654 } name_element_test;
1655
1656 static name_element_test names_ref[] = {
1657         /* === DN tests === */
1658         {
1659                 /* [12] 66:6f:6f */
1660                 1, 1, "foo"
1661         },
1662         {
1663                 /* [12] 62:61:72 */
1664                 1, 1, "bar"
1665         },
1666         {
1667                 /* [18] 31:32:33:34 */
1668                 2, 1, "1234"
1669         },
1670         {
1671                 /* [19] 66:6f:6f */
1672                 3, 1, "foo"
1673         },
1674         {
1675                 /* [20] 66:6f:6f */
1676                 4, 1, "foo"
1677         },
1678         {
1679                 /* [22] 66:6f:6f */
1680                 5, 1, "foo"
1681         },
1682         {
1683                 /* [30] 00:66:00:6f:00:6f */
1684                 6, 1, "foo"
1685         },
1686         {
1687                 /* [30] fe:ff:00:66:00:6f:00:6f */
1688                 7, 1, "foo"
1689         },
1690         {
1691                 /* [30] ff:fe:66:00:6f:00:6f:00 */
1692                 8, 1, "foo"
1693         },
1694         {
1695                 /* [20] 63:61:66:e9 */
1696                 9, 1, "caf\xC3\xA9"
1697         },
1698         {
1699                 /* [12] 63:61:66:c3:a9 */
1700                 10, 1, "caf\xC3\xA9"
1701         },
1702         {
1703                 /* [12] 63:61:66:e0:83:a9 */
1704                 11, -1, NULL
1705         },
1706         {
1707                 /* [12] 63:61:66:e3:90:8c */
1708                 12, 1, "caf\xE3\x90\x8C"
1709         },
1710         {
1711                 /* [30] 00:63:00:61:00:66:34:0c */
1712                 13, 1, "caf\xE3\x90\x8C"
1713         },
1714         {
1715                 /* [12] 63:61:66:c3 */
1716                 14, -1, NULL
1717         },
1718         {
1719                 /* [30] d8:42:df:f4:00:67:00:6f */
1720                 15, 1, "\xF0\xA0\xAF\xB4go"
1721         },
1722         {
1723                 /* [30] 00:66:d8:42 */
1724                 16, -1, NULL
1725         },
1726         {
1727                 /* [30] d8:42:00:66 */
1728                 17, -1, NULL
1729         },
1730         {
1731                 /* [30] df:f4:00:66 */
1732                 18, -1, NULL
1733         },
1734         {
1735                 /* [12] 66:00:6f */
1736                 19, -1, NULL
1737         },
1738         {
1739                 /* [30] 00:00:34:0c */
1740                 20, -1, NULL
1741         },
1742         {
1743                 /* [30] 34:0c:00:00:00:66 */
1744                 21, -1, NULL
1745         },
1746         {
1747                 /* [12] ef:bb:bf:66:6f:6f */
1748                 22, 1, "foo"
1749         },
1750         {
1751                 /* [30] 00:66:ff:fe:00:6f */
1752                 23, -1, NULL
1753         },
1754         {
1755                 /* [30] 00:66:ff:fd:00:6f */
1756                 24, 1, "f\xEF\xBF\xBDo"
1757         },
1758
1759         /* === Value not found in the DN === */
1760         {
1761                 127, 0, NULL
1762         },
1763
1764         /* === SAN tests === */
1765         {
1766                 /* SAN OtherName (Microsoft UPN) */
1767                 0, 1, "foo@bar.com"
1768         },
1769         {
1770                 /* SAN rfc822Name */
1771                 -1, 1, "bar@foo.com"
1772         },
1773         {
1774                 /* SAN dNSName */
1775                 -2, 1, "example.com"
1776         },
1777         {
1778                 /* SAN dNSName */
1779                 -2, 1, "www.example.com"
1780         },
1781         {
1782                 /* uniformResourceIdentifier */
1783                 -6, 1, "http://www.example.com/"
1784         }
1785 };
1786
1787 static void
1788 free_name_elements(br_name_element *elts, size_t num)
1789 {
1790         size_t u;
1791
1792         for (u = 0; u < num; u ++) {
1793                 xfree((void *)elts[u].oid);
1794                 xfree(elts[u].buf);
1795         }
1796         xfree(elts);
1797 }
1798
1799 static void
1800 test_name_extraction(void)
1801 {
1802         unsigned char *data;
1803         size_t len;
1804         br_x509_minimal_context ctx;
1805         uint32_t days, seconds;
1806         size_t u;
1807         unsigned status;
1808         br_name_element *names;
1809         size_t num_names;
1810         int good;
1811
1812         printf("Name extraction: ");
1813         fflush(stdout);
1814         data = read_file("names.crt", &len);
1815         br_x509_minimal_init(&ctx, &br_sha256_vtable, NULL, 0);
1816         for (u = 0; hash_impls[u].id; u ++) {
1817                 int id;
1818
1819                 id = hash_impls[u].id;
1820                 br_x509_minimal_set_hash(&ctx, id, hash_impls[u].impl);
1821         }
1822         br_x509_minimal_set_rsa(&ctx, br_rsa_pkcs1_vrfy_get_default());
1823         br_x509_minimal_set_ecdsa(&ctx,
1824                 br_ec_get_default(), br_ecdsa_vrfy_asn1_get_default());
1825         string_to_time(DEFAULT_TIME, &days, &seconds);
1826         br_x509_minimal_set_time(&ctx, days, seconds);
1827
1828         num_names = (sizeof names_ref) / (sizeof names_ref[0]);
1829         names = xmalloc(num_names * sizeof *names);
1830         for (u = 0; u < num_names; u ++) {
1831                 int num;
1832                 unsigned char *oid;
1833
1834                 num = names_ref[u].num;
1835                 if (num > 0) {
1836                         oid = xmalloc(5);
1837                         oid[0] = 4;
1838                         oid[1] = 0x29;
1839                         oid[2] = 0x01;
1840                         oid[3] = 0x01;
1841                         oid[4] = num;
1842                 } else if (num == 0) {
1843                         oid = xmalloc(13);
1844                         oid[0] = 0x00;
1845                         oid[1] = 0x00;
1846                         oid[2] = 0x0A;
1847                         oid[3] = 0x2B;
1848                         oid[4] = 0x06;
1849                         oid[5] = 0x01;
1850                         oid[6] = 0x04;
1851                         oid[7] = 0x01;
1852                         oid[8] = 0x82;
1853                         oid[9] = 0x37;
1854                         oid[10] = 0x14;
1855                         oid[11] = 0x02;
1856                         oid[12] = 0x03;
1857                 } else {
1858                         oid = xmalloc(2);
1859                         oid[0] = 0x00;
1860                         oid[1] = -num;
1861                 }
1862                 names[u].oid = oid;
1863                 names[u].buf = xmalloc(256);
1864                 names[u].len = 256;
1865         }
1866         br_x509_minimal_set_name_elements(&ctx, names, num_names);
1867
1868         /*
1869          * Put "canaries" to detect actual stack usage.
1870          */
1871         for (u = 0; u < (sizeof ctx.dp_stack) / sizeof(uint32_t); u ++) {
1872                 ctx.dp_stack[u] = 0xA7C083FE;
1873         }
1874         for (u = 0; u < (sizeof ctx.rp_stack) / sizeof(uint32_t); u ++) {
1875                 ctx.rp_stack[u] = 0xA7C083FE;
1876         }
1877
1878         /*
1879          * Run the engine. Since we set no trust anchor, we expect a status
1880          * of "not trusted".
1881          */
1882         ctx.vtable->start_chain(&ctx.vtable, NULL);
1883         ctx.vtable->start_cert(&ctx.vtable, len);
1884         ctx.vtable->append(&ctx.vtable, data, len);
1885         ctx.vtable->end_cert(&ctx.vtable);
1886         status = ctx.vtable->end_chain(&ctx.vtable);
1887         if (status != BR_ERR_X509_NOT_TRUSTED) {
1888                 fprintf(stderr, "wrong status: %u\n", status);
1889                 exit(EXIT_FAILURE);
1890         }
1891
1892         /*
1893          * Check stack usage.
1894          */
1895         for (u = (sizeof ctx.dp_stack) / sizeof(uint32_t); u > 0; u --) {
1896                 if (ctx.dp_stack[u - 1] != 0xA7C083FE) {
1897                         if (max_dp_usage < u) {
1898                                 max_dp_usage = u;
1899                         }
1900                         break;
1901                 }
1902         }
1903         for (u = (sizeof ctx.rp_stack) / sizeof(uint32_t); u > 0; u --) {
1904                 if (ctx.rp_stack[u - 1] != 0xA7C083FE) {
1905                         if (max_rp_usage < u) {
1906                                 max_rp_usage = u;
1907                         }
1908                         break;
1909                 }
1910         }
1911
1912         good = 1;
1913         for (u = 0; u < num_names; u ++) {
1914                 if (names[u].status != names_ref[u].status) {
1915                         printf("ERR: name %u (id=%d): status=%d, expected=%d\n",
1916                                 (unsigned)u, names_ref[u].num,
1917                                 names[u].status, names_ref[u].status);
1918                         if (names[u].status > 0) {
1919                                 unsigned char *p;
1920
1921                                 printf("  obtained:");
1922                                 p = (unsigned char *)names[u].buf;
1923                                 while (*p) {
1924                                         printf(" %02X", *p ++);
1925                                 }
1926                                 printf("\n");
1927                         }
1928                         good = 0;
1929                         continue;
1930                 }
1931                 if (names_ref[u].expected == NULL) {
1932                         if (names[u].buf[0] != 0) {
1933                                 printf("ERR: name %u not zero-terminated\n",
1934                                         (unsigned)u);
1935                                 good = 0;
1936                                 continue;
1937                         }
1938                 } else {
1939                         if (strcmp(names[u].buf, names_ref[u].expected) != 0) {
1940                                 unsigned char *p;
1941
1942                                 printf("ERR: name %u (id=%d): wrong value\n",
1943                                         (unsigned)u, names_ref[u].num);
1944                                 printf("  expected:");
1945                                 p = (unsigned char *)names_ref[u].expected;
1946                                 while (*p) {
1947                                         printf(" %02X", *p ++);
1948                                 }
1949                                 printf("\n");
1950                                 printf("  obtained:");
1951                                 p = (unsigned char *)names[u].buf;
1952                                 while (*p) {
1953                                         printf(" %02X", *p ++);
1954                                 }
1955                                 printf("\n");
1956                                 good = 0;
1957                                 continue;
1958                         }
1959                 }
1960         }
1961         if (!good) {
1962                 exit(EXIT_FAILURE);
1963         }
1964
1965         /*
1966         for (u = 0; u < num_names; u ++) {
1967                 printf("%u: (%d)", (unsigned)u, names[u].status);
1968                 if (names[u].status > 0) {
1969                         size_t v;
1970
1971                         for (v = 0; names[u].buf[v]; v ++) {
1972                                 printf(" %02x", names[u].buf[v]);
1973                         }
1974                 }
1975                 printf("\n");
1976         }
1977         */
1978
1979         xfree(data);
1980         free_name_elements(names, num_names);
1981         printf("OK\n");
1982 }
1983
1984 int
1985 main(int argc, const char *argv[])
1986 {
1987         size_t u;
1988
1989 #ifdef SRCDIRNAME
1990         /*
1991          * We want to change the current directory to that of the
1992          * executable, so that test files are reliably located. We
1993          * do that only if SRCDIRNAME is defined (old Makefile would
1994          * not do that).
1995          */
1996         if (argc >= 1) {
1997                 const char *arg, *c;
1998
1999                 arg = argv[0];
2000                 for (c = arg + strlen(arg);; c --) {
2001                         int sep, r;
2002
2003 #ifdef _WIN32
2004                         sep = (*c == '/') || (*c == '\\');
2005 #else
2006                         sep = (*c == '/');
2007 #endif
2008                         if (sep) {
2009                                 size_t len;
2010                                 char *dn;
2011
2012                                 len = 1 + (c - arg);
2013                                 dn = xmalloc(len + 1);
2014                                 memcpy(dn, arg, len);
2015                                 dn[len] = 0;
2016 #ifdef _WIN32
2017                                 r = _chdir(dn);
2018 #else
2019                                 r = chdir(dn);
2020 #endif
2021                                 if (r != 0) {
2022                                         fprintf(stderr, "warning: could not"
2023                                                 " set directory to '%s'\n", dn);
2024                                 }
2025                                 xfree(dn);
2026                                 break;
2027                         }
2028                         if (c == arg) {
2029                                 break;
2030                         }
2031                 }
2032         }
2033 #else
2034         (void)argc;
2035         (void)argv;
2036 #endif
2037
2038         process_conf_file(CONFFILE);
2039
2040         max_dp_usage = 0;
2041         max_rp_usage = 0;
2042         for (u = 0; u < all_chains_ptr; u ++) {
2043                 run_test_case(&all_chains[u]);
2044         }
2045         test_name_extraction();
2046
2047         printf("Maximum data stack usage:    %u\n", (unsigned)max_dp_usage);
2048         printf("Maximum return stack usage:  %u\n", (unsigned)max_rp_usage);
2049
2050         HT_free(keys, free_key);
2051         HT_free(trust_anchors, free_trust_anchor);
2052         for (u = 0; u < all_chains_ptr; u ++) {
2053                 free_test_case_contents(&all_chains[u]);
2054         }
2055         xfree(all_chains);
2056
2057         return 0;
2058 }