external/bsd/bind/dist/bin/tests/sym_test.c

   1 /*      $NetBSD: sym_test.c,v 1.6 2014/12/10 04:37:53 christos Exp $    */
   2
   3 /*
   4  * Copyright (C) 2004, 2005, 2007  Internet Systems Consortium, Inc. ("ISC")
   5  * Copyright (C) 1998-2001  Internet Software Consortium.
   6  *
   7  * Permission to use, copy, modify, and/or distribute this software for any
   8  * purpose with or without fee is hereby granted, provided that the above
   9  * copyright notice and this permission notice appear in all copies.
  10  *
  11  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
  12  * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
  13  * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
  14  * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
  15  * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
  16  * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
  17  * PERFORMANCE OF THIS SOFTWARE.
  18  */
  19
  20 /* Id: sym_test.c,v 1.28 2007/06/19 23:46:59 tbox Exp  */
  21
  22 #include <config.h>
  23
  24 #include <string.h>
  25
  26 #include <isc/commandline.h>
  27 #include <isc/mem.h>
  28 #include <isc/symtab.h>
  29 #include <isc/util.h>
  30
  31 isc_mem_t *mctx;
  32 isc_symtab_t *st;
  33
  34 static void
  35 undefine_action(char *key, unsigned int type, isc_symvalue_t value, void *arg)
  36 {
  37         UNUSED(arg);
  38
  39         INSIST(type == 1);
  40         isc_mem_free(mctx, key);
  41         isc_mem_free(mctx, value.as_pointer);
  42 }
  43
  44 int
  45 main(int argc, char *argv[]) {
  46         char s[1000], *cp, *key;
  47         size_t len;
  48         isc_result_t result;
  49         isc_symvalue_t value;
  50         int trace = 0;
  51         int c;
  52         isc_symexists_t exists_policy = isc_symexists_reject;
  53         isc_boolean_t case_sensitive = ISC_FALSE;
  54
  55         while ((c = isc_commandline_parse(argc, argv, "tarc")) != -1) {
  56                 switch (c) {
  57                 case 't':
  58                         trace = 1;
  59                         break;
  60                 case 'a':
  61                         exists_policy = isc_symexists_add;
  62                         break;
  63                 case 'r':
  64                         exists_policy = isc_symexists_replace;
  65                         break;
  66                 case 'c':
  67                         case_sensitive = ISC_TRUE;
  68                         break;
  69                 }
  70         }
  71
  72         RUNTIME_CHECK(isc_mem_create(0, 0, &mctx) == ISC_R_SUCCESS);
  73         RUNTIME_CHECK(isc_symtab_create(mctx, 691, undefine_action, NULL,
  74                                         case_sensitive, &st) == ISC_R_SUCCESS);
  75
  76         while (fgets(s, sizeof(s), stdin) != NULL) {
  77                 len = strlen(s);
  78                 if (len > 0U && s[len - 1] == '\n') {
  79                         s[len - 1] = '\0';
  80                         len--;
  81                 }
  82
  83                 cp = s;
  84
  85                 if (cp[0] == '!') {
  86                         cp++;
  87                         result = isc_symtab_undefine(st, cp, 1);
  88                         if (trace || result != ISC_R_SUCCESS)
  89                                 printf("undefine('%s'): %s\n", cp,
  90                                        isc_result_totext(result));
  91                 } else {
  92                         key = cp;
  93                         while (*cp != '\0' && *cp != ' ' && *cp != '\t')
  94                                 cp++;
  95                         if (*cp == '\0') {
  96                                 result = isc_symtab_lookup(st, key, 0, &value);
  97                                 if (trace || result != ISC_R_SUCCESS) {
  98                                         printf("lookup('%s'): %s", key,
  99                                                isc_result_totext(result));
 100                                         if (result == ISC_R_SUCCESS) {
 101                                                 cp = value.as_pointer;
 102                                                 printf(", value == '%s'", cp);
 103                                         }
 104                                         printf("\n");
 105                                 }
 106                         } else {
 107                                 *cp++ = '\0';
 108                                 key = isc_mem_strdup(mctx, key);
 109                                 value.as_pointer = isc_mem_strdup(mctx, cp);
 110                                 result = isc_symtab_define(st, key, 1, value,
 111                                                            exists_policy);
 112                                 if (trace || result != ISC_R_SUCCESS) {
 113                                         printf("define('%s', '%s'): %s\n",
 114                                                key, cp,
 115                                                isc_result_totext(result));
 116                                         if (result != ISC_R_SUCCESS)
 117                                                 undefine_action(key, 1,
 118                                                         value, NULL);
 119                                 }
 120                         }
 121                 }
 122         }
 123
 124         isc_symtab_destroy(&st);
 125         isc_mem_stats(mctx, stdout);
 126         isc_mem_destroy(&mctx);
 127
 128         return (0);
 129 }