src/uniqstr.c

   1 /* Keep a unique copy of strings.
   2
   3    Copyright (C) 2002-2005, 2009-2013 Free Software Foundation, Inc.
   4
   5    This file is part of Bison, the GNU Compiler Compiler.
   6
   7    This program is free software: you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation, either version 3 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program.  If not, see <http://www.gnu.org/licenses/>.  */
  19
  20 #include <config.h>
  21 #include "system.h"
  22
  23 #include <error.h>
  24 #include <hash.h>
  25 #include <quotearg.h>
  26 #include <stdarg.h>
  27
  28 #include "uniqstr.h"
  29
  30 /*-----------------------.
  31 | A uniqstr hash table.  |
  32 `-----------------------*/
  33
  34 /* Initial capacity of uniqstr hash table.  */
  35 #define HT_INITIAL_CAPACITY 257
  36
  37 static struct hash_table *uniqstrs_table = NULL;
  38
  39 /*-------------------------------------.
  40 | Create the uniqstr for S if needed.  |
  41 `-------------------------------------*/
  42
  43 uniqstr
  44 uniqstr_new (char const *str)
  45 {
  46   uniqstr res = hash_lookup (uniqstrs_table, str);
  47   if (!res)
  48     {
  49       /* First insertion in the hash. */
  50       res = xstrdup (str);
  51       if (!hash_insert (uniqstrs_table, res))
  52         xalloc_die ();
  53     }
  54   return res;
  55 }
  56
  57 uniqstr
  58 uniqstr_vsprintf (char const *format, ...)
  59 {
  60   va_list args;
  61   size_t length;
  62   va_start (args, format);
  63   length = vsnprintf (NULL, 0, format, args);
  64   va_end (args);
  65
  66   char res[length + 1];
  67   va_start (args, format);
  68   vsprintf (res, format, args);
  69   va_end (args);
  70   return uniqstr_new (res);
  71 }
  72
  73 /*------------------------------.
  74 | Abort if S is not a uniqstr.  |
  75 `------------------------------*/
  76
  77 void
  78 uniqstr_assert (char const *str)
  79 {
  80   uniqstr *s = hash_lookup (uniqstrs_table, str);
  81   if (!s || s != (uniqstr *)str)
  82     {
  83       error (0, 0,
  84              "not a uniqstr: %s", quotearg (str));
  85       abort ();
  86     }
  87 }
  88
  89
  90 /*--------------------.
  91 | Print the uniqstr.  |
  92 `--------------------*/
  93
  94 static inline bool
  95 uniqstr_print (uniqstr ustr)
  96 {
  97   fprintf (stderr, "%s\n", ustr);
  98   return true;
  99 }
 100
 101 static bool
 102 uniqstr_print_processor (void *ustr, void *null ATTRIBUTE_UNUSED)
 103 {
 104   return uniqstr_print (ustr);
 105 }
 106
 107
 108 int
 109 uniqstr_cmp (uniqstr l, uniqstr r)
 110 {
 111   return (l == r ? 0
 112           : !l ? -1
 113           : !r ? +1
 114           : strcmp (l, r));
 115 }
 116
 117
 118 /*-----------------------.
 119 | A uniqstr hash table.  |
 120 `-----------------------*/
 121
 122 static bool
 123 hash_compare_uniqstr (void const *m1, void const *m2)
 124 {
 125   return STREQ (m1, m2);
 126 }
 127
 128 static size_t
 129 hash_uniqstr (void const *m, size_t tablesize)
 130 {
 131   return hash_string (m, tablesize);
 132 }
 133
 134
 135 /*----------------------------.
 136 | Create the uniqstrs table.  |
 137 `----------------------------*/
 138
 139 void
 140 uniqstrs_new (void)
 141 {
 142   uniqstrs_table = hash_initialize (HT_INITIAL_CAPACITY,
 143                                     NULL,
 144                                     hash_uniqstr,
 145                                     hash_compare_uniqstr,
 146                                     free);
 147 }
 148
 149
 150 /*-------------------------------------.
 151 | Perform a task on all the uniqstrs.  |
 152 `-------------------------------------*/
 153
 154 static void
 155 uniqstrs_do (Hash_processor processor, void *processor_data)
 156 {
 157   hash_do_for_each (uniqstrs_table, processor, processor_data);
 158 }
 159
 160
 161 /*-----------------.
 162 | Print them all.  |
 163 `-----------------*/
 164
 165 void
 166 uniqstrs_print (void)
 167 {
 168   uniqstrs_do (uniqstr_print_processor, NULL);
 169 }
 170
 171
 172 /*--------------------.
 173 | Free the uniqstrs.  |
 174 `--------------------*/
 175
 176 void
 177 uniqstrs_free (void)
 178 {
 179   hash_free (uniqstrs_table);
 180 }