pintos-os.org Git - pspp/blob - src/libpspp/str.h

   1 /* PSPP - a program for statistical analysis.
   2    Copyright (C) 1997-9, 2000, 2009, 2010, 2011, 2012, 2014 Free Software Foundation, Inc.
   3
   4    This program is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation, either version 3 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program.  If not, see <http://www.gnu.org/licenses/>. */
  16
  17 #if !str_h
  18 #define str_h 1
  19
  20 #include <assert.h>
  21 #include <stdarg.h>
  22 #include <stdbool.h>
  23 #include <stdint.h>
  24 #include <stdio.h>
  25 #include <string.h>
  26 #include <unitypes.h>
  27
  28 #include "compiler.h"
  29 #include "memcasecmp.h"
  30 #include "xstrndup.h"
  31 #include "xvasprintf.h"
  32 \f
  33 /* Miscellaneous. */
  34
  35 void buf_reverse (char *, size_t);
  36 int buf_compare_case (const char *, const char *, size_t);
  37 int buf_compare_rpad (const char *, size_t, const char *, size_t);
  38 void buf_copy_lpad (char *, size_t, const char *, size_t, char pad);
  39 void buf_copy_rpad (char *, size_t, const char *, size_t, char pad);
  40 void buf_copy_str_lpad (char *, size_t, const char *, char pad);
  41 void buf_copy_str_rpad (char *, size_t, const char *, char pad);
  42
  43 int str_compare_rpad (const char *, const char *);
  44 void str_copy_rpad (char *, size_t, const char *);
  45 void str_copy_trunc (char *, size_t, const char *);
  46 void str_copy_buf_trunc (char *, size_t, const char *, size_t);
  47 void str_uppercase (char *);
  48 void str_lowercase (char *);
  49
  50 bool str_format_26adic (unsigned long int number, bool uppercase,
  51                         char buffer[], size_t);
  52
  53 void *mempset (void *, int, size_t);
  54 \f
  55 /* Common character classes for use with substring and string functions. */
  56
  57 #define CC_SPACES " \t\v\r\n"
  58 #define CC_DIGITS "0123456789"
  59 #define CC_XDIGITS "0123456789abcdefABCDEF"
  60 #define CC_LETTERS "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
  61 #define CC_ALNUM CC_LETTERS CC_DIGITS
  62 \f
  63 /* Substrings. */
  64 struct substring
  65   {
  66     char *string;
  67     size_t length;
  68   };
  69
  70 #define SS_EMPTY_INITIALIZER {NULL, 0}
  71 #define SS_LITERAL_INITIALIZER(LITERAL)                 \
  72         {(char *) LITERAL, (sizeof LITERAL) - 1}
  73
  74 /* Constructors.
  75    These functions do not allocate any memory, so the substrings
  76    they create should not normally be destroyed. */
  77 static inline struct substring ss_empty (void);
  78 static inline struct substring ss_cstr (const char *);
  79 static inline struct substring ss_buffer (const char *, size_t);
  80 struct substring ss_substr (struct substring, size_t start, size_t);
  81 struct substring ss_head (struct substring, size_t);
  82 struct substring ss_tail (struct substring, size_t);
  83
  84 /* Constructors and destructor that allocate and deallocate
  85    memory. */
  86 struct pool;
  87 void ss_alloc_substring (struct substring *, struct substring);
  88 void ss_alloc_uninit (struct substring *, size_t);
  89 void ss_realloc (struct substring *, size_t);
  90 void ss_alloc_substring_pool (struct substring *, struct substring,
  91                               struct pool *);
  92 void ss_alloc_uninit_pool (struct substring *, size_t, struct pool *);
  93 void ss_dealloc (struct substring *);
  94
  95 /* Mutators.
  96    Functions that advance the beginning of a string should not be
  97    used if a substring is to be deallocated. */
  98 void ss_truncate (struct substring *, size_t);
  99 size_t ss_rtrim (struct substring *, struct substring trim_set);
 100 size_t ss_ltrim (struct substring *, struct substring trim_set);
 101 void ss_trim (struct substring *, struct substring trim_set);
 102 bool ss_chomp_byte (struct substring *, char);
 103 bool ss_chomp (struct substring *, struct substring);
 104 bool ss_separate (struct substring src, struct substring delimiters,
 105                   size_t *save_idx, struct substring *token);
 106 bool ss_tokenize (struct substring src, struct substring delimiters,
 107                   size_t *save_idx, struct substring *token);
 108 void ss_advance (struct substring *, size_t);
 109 bool ss_match_byte (struct substring *, char);
 110 int ss_match_byte_in (struct substring *, struct substring);
 111 bool ss_match_string (struct substring *, const struct substring);
 112 int ss_get_byte (struct substring *);
 113 size_t ss_get_bytes (struct substring *, size_t cnt, struct substring *);
 114 bool ss_get_until (struct substring *, char delimiter, struct substring *);
 115 size_t ss_get_long (struct substring *, long *);
 116
 117 /* Inspectors. */
 118 bool ss_is_empty (struct substring);
 119 size_t ss_length (struct substring);
 120 char *ss_data (struct substring);
 121 char *ss_end (struct substring);
 122 int ss_at (struct substring, size_t idx);
 123 int ss_first (struct substring);
 124 int ss_last (struct substring);
 125 bool ss_starts_with (struct substring, struct substring prefix);
 126 bool ss_ends_with (struct substring, struct substring suffix);
 127 size_t ss_span (struct substring, struct substring skip_set);
 128 size_t ss_cspan (struct substring, struct substring stop_set);
 129 size_t ss_find_byte (struct substring, char);
 130 size_t ss_find_substring (struct substring, struct substring);
 131 int ss_compare (struct substring, struct substring);
 132 int ss_compare_case (struct substring, struct substring);
 133 int ss_equals (struct substring, struct substring);
 134 int ss_equals_case (struct substring, struct substring);
 135 size_t ss_pointer_to_position (struct substring, const char *);
 136 char *ss_xstrdup (struct substring);
 137
 138 /* UTF-8. */
 139 ucs4_t ss_first_mb (struct substring);
 140 int ss_first_mblen (struct substring);
 141 ucs4_t ss_get_mb (struct substring *);
 142 ucs4_t ss_at_mb (struct substring, size_t ofs);
 143 int ss_at_mblen (struct substring, size_t ofs);
 144 \f
 145 /* Variable length strings. */
 146
 147 struct string
 148   {
 149     struct substring ss;
 150
 151     size_t capacity;    /* Allocated capacity, not including one
 152                            extra byte allocated for null terminator. */
 153   };
 154
 155 #define DS_EMPTY_INITIALIZER {SS_EMPTY_INITIALIZER, 0}
 156
 157 /* Constructors, destructors. */
 158 void ds_init_empty (struct string *);
 159 void ds_init_string (struct string *, const struct string *);
 160 void ds_init_substring (struct string *, struct substring);
 161 void ds_init_cstr (struct string *, const char *);
 162 void ds_destroy (struct string *);
 163 void ds_swap (struct string *, struct string *);
 164
 165 /* Pools. */
 166 struct pool;
 167 void ds_register_pool (struct string *, struct pool *);
 168 void ds_unregister_pool (struct string *, struct pool *);
 169
 170 /* Replacement. */
 171 void ds_assign_string (struct string *, const struct string *);
 172 void ds_assign_substring (struct string *, struct substring);
 173 void ds_assign_cstr (struct string *, const char *);
 174
 175 /* Shrink, extend. */
 176 void ds_clear (struct string *);
 177 void ds_extend (struct string *, size_t);
 178 void ds_shrink (struct string *);
 179 void ds_truncate (struct string *, size_t);
 180
 181 /* Padding, trimming. */
 182 size_t ds_rtrim (struct string *, struct substring trim_set);
 183 size_t ds_ltrim (struct string *, struct substring trim_set);
 184 size_t ds_trim (struct string *, struct substring trim_set);
 185 bool ds_chomp_byte (struct string *, char);
 186 bool ds_chomp (struct string *, struct substring);
 187 bool ds_separate (const struct string *src, struct substring delimiters,
 188                   size_t *save_idx, struct substring *token);
 189 bool ds_tokenize (const struct string *src, struct substring delimiters,
 190                   size_t *save_idx, struct substring *token);
 191 void ds_rpad (struct string *, size_t length, char pad);
 192 void ds_set_length (struct string *, size_t new_length, char pad);
 193 void ds_remove (struct string *, size_t start, size_t n);
 194
 195 /* Extracting substrings. */
 196 struct substring ds_ss (const struct string *);
 197 struct substring ds_substr (const struct string *, size_t start, size_t);
 198 struct substring ds_head (const struct string *, size_t);
 199 struct substring ds_tail (const struct string *, size_t);
 200
 201 /* Inspectors. */
 202 bool ds_is_empty (const struct string *);
 203 size_t ds_length (const struct string *);
 204 char *ds_data (const struct string *);
 205 char *ds_end (const struct string *);
 206 int ds_at (const struct string *, size_t idx);
 207 int ds_first (const struct string *);
 208 int ds_last (const struct string *);
 209 bool ds_ends_with (const struct string *, struct substring suffix);
 210 size_t ds_span (const struct string *, struct substring skip_set);
 211 size_t ds_cspan (const struct string *, struct substring stop_set);
 212 size_t ds_find_byte (const struct string *, char);
 213 int ds_compare (const struct string *, const struct string *);
 214 size_t ds_pointer_to_position (const struct string *, const char *);
 215 char *ds_xstrdup (const struct string *);
 216
 217 size_t ds_capacity (const struct string *);
 218 char *ds_cstr (const struct string *);
 219 char *ds_steal_cstr (struct string *);
 220
 221 /* File input. */
 222 bool ds_read_line (struct string *, FILE *, size_t max_length);
 223 bool ds_read_config_line (struct string *, int *line_number, FILE *);
 224 bool ds_read_stream (struct string *, size_t size, size_t cnt, FILE *stream);
 225
 226 /* Append. */
 227 void ds_put_byte (struct string *, int ch);
 228 void ds_put_byte_multiple (struct string *, int ch, size_t);
 229 void ds_put_unichar (struct string *, ucs4_t uc);
 230 void ds_put_cstr (struct string *, const char *);
 231 void ds_put_substring (struct string *, struct substring);
 232 void ds_put_vformat (struct string *st, const char *, va_list)
 233      PRINTF_FORMAT (2, 0);
 234 void ds_put_c_vformat (struct string *st, const char *, va_list)
 235      PRINTF_FORMAT (2, 0);
 236
 237 void ds_put_format (struct string *, const char *, ...)
 238      PRINTF_FORMAT (2, 3);
 239 void ds_put_c_format (struct string *, const char *, ...)
 240      PRINTF_FORMAT (2, 3);
 241
 242 char *ds_put_uninit (struct string *st, size_t incr);
 243
 244 char *ds_splice_uninit (struct string *, size_t ofs, size_t old_len,
 245                         size_t new_len);
 246
 247 /* Other */
 248 /* calls relocate from gnulib on ST */
 249 void ds_relocate (struct string *st);
 250
 251
 252 void u8_buf_copy_rpad (uint8_t *dst, size_t dst_size,
 253                        const uint8_t *src, size_t src_size,
 254                        char pad);
 255 \f
 256 static inline struct substring
 257 ss_empty (void)
 258 {
 259   struct substring ss;
 260   ss.string = NULL;
 261   ss.length = 0;
 262   return ss;
 263 }
 264
 265 /* Returns a substring whose contents are the given C-style
 266    string CSTR. */
 267 static inline struct substring
 268 ss_cstr (const char *cstr)
 269 {
 270   return ss_buffer (cstr, strlen (cstr));
 271 }
 272
 273 /* Returns a substring whose contents are the CNT characters in
 274    BUFFER. */
 275 static inline struct substring
 276 ss_buffer (const char *buffer, size_t cnt)
 277 {
 278   struct substring ss;
 279   ss.string = (char *) buffer;
 280   ss.length = cnt;
 281   return ss;
 282 }
 283
 284 #endif /* str_h */