t/testlib.c

   1 /*
   2  *  Miscellaneous functions used by the cdecl99 test suite.
   3  *  Copyright © 2011-2012, 2021-2024 Nick Bowler
   4  *
   5  *  This program is free software: you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation, either version 3 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <config.h>
  20 #include <stdio.h>
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <stdarg.h>
  24 #include <errno.h>
  25 #include <getopt.h>
  26
  27 #include "cdecl.h"
  28
  29 #include "help.h"
  30 #include "test.h"
  31 #include "getline.h"
  32 #include "intconv.h"
  33
  34 void print_error(const char *fmt, ...)
  35 {
  36         extern const char *progname;
  37         va_list ap;
  38
  39         fprintf(stderr, "%s: ", progname);
  40
  41         va_start(ap, fmt);
  42         vfprintf(stderr, fmt, ap);
  43         va_end(ap);
  44
  45         fprintf(stderr, "\n");
  46 }
  47
  48 void *realloc_nofail(void *ptr, size_t size)
  49 {
  50         void *p;
  51
  52         p = realloc(ptr, size);
  53         if (!p) {
  54                 perror("failed to allocate memory");
  55                 abort();
  56         }
  57
  58         return p;
  59 }
  60
  61 void *malloc_nofail(size_t size)
  62 {
  63         return realloc_nofail(NULL, size);
  64 }
  65
  66 static unsigned intconv_base(const char **str)
  67 {
  68         if ((*str)[0] == '0') {
  69                 if ((*str)[1] == 'X' || (*str)[1] == 'x') {
  70                         *str += 2;
  71                         return INTCONV_HEXADECIMAL;
  72                 }
  73
  74                 return INTCONV_OCTAL;
  75         }
  76
  77         return INTCONV_DECIMAL;
  78 }
  79
  80 bool test_strtoumax(cdecl_uintmax *out, const char *s, cdecl_uintmax limit)
  81 {
  82         static const char idx[] = "0123456789abcdef0123456789ABCDEF";
  83         unsigned base = intconv_base(&s);
  84         cdecl_uintmax v;
  85         char *c, d;
  86
  87         for (v = 0; (d = *s++);) {
  88                 if (!(c = strchr(idx, d)) || (d = (c-idx) & 0xf) >= base)
  89                         return 0;
  90
  91                 if (!intconv_shift(&v, base, d) || v > limit)
  92                         return 0;
  93         }
  94
  95         *out = v;
  96         return true;
  97 }
  98
  99 void test_print_options(const struct option *lopts)
 100 {
 101         const struct option *opt;
 102
 103         puts("Options:");
 104         for (opt = lopts; opt->val; opt++) {
 105                 if (help_print_optstring(opt, "ARG", 20))
 106                         putchar('\n');
 107         }
 108 }
 109
 110 int test_getline(char **linebuf, size_t *n)
 111 {
 112         int rc;
 113
 114         if ((rc = dx_getline(linebuf, n, stdin)) < 0) {
 115                 if (rc == DX_GETLINE_ENOMEM)
 116                         print_error("%s", _("failed to allocate memory"));
 117                 else
 118                         print_error("%s", strerror(errno));
 119                 abort();
 120         }
 121
 122         return rc;
 123 }