lib/mbuiter.h

   1 /* Iterating through multibyte strings: macros for multi-byte encodings.
   2    Copyright (C) 2001, 2005, 2007, 2009-2024 Free Software Foundation, Inc.
   3
   4    This file is free software: you can redistribute it and/or modify
   5    it under the terms of the GNU Lesser General Public License as
   6    published by the Free Software Foundation, either version 3 of the
   7    License, or (at your option) any later version.
   8
   9    This file 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 Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public License
  15    along with this program.  If not, see <https://www.gnu.org/licenses/>.  */
  16
  17 /* Written by Bruno Haible <bruno@clisp.org>.  */
  18
  19 /* The macros in this file implement forward iteration through a
  20    multi-byte string, without knowing its length a-priori.
  21
  22    With these macros, an iteration loop that looks like
  23
  24       char *iter;
  25       for (iter = buf; *iter != '\0'; iter++)
  26         {
  27           do_something (*iter);
  28         }
  29
  30    becomes
  31
  32       mbui_iterator_t iter;
  33       for (mbui_init (iter, buf); mbui_avail (iter); mbui_advance (iter))
  34         {
  35           do_something (mbui_cur_ptr (iter), mb_len (mbui_cur (iter)));
  36         }
  37
  38    The benefit of these macros over plain use of mbrtowc is:
  39    - Handling of invalid multibyte sequences is possible without
  40      making the code more complicated, while still preserving the
  41      invalid multibyte sequences.
  42
  43    Compared to mbiter.h, the macros here don't need to know the string's
  44    length a-priori.  The downside is that at each step, the look-ahead
  45    that guards against overrunning the terminating '\0' is more expensive.
  46    The mbui_* macros are therefore suitable when there is a high probability
  47    that only the first few multibyte characters need to be inspected.
  48    Whereas the mbi_* macros are better if usually the iteration runs
  49    through the entire string.
  50
  51    mbui_iterator_t
  52      is a type usable for variable declarations.
  53
  54    mbui_init (iter, startptr)
  55      initializes the iterator, starting at startptr.
  56
  57    mbui_avail (iter)
  58      returns true if there are more multibyte characters available before
  59      the end of string is reached. In this case, mbui_cur (iter) is
  60      initialized to the next multibyte character.
  61
  62    mbui_advance (iter)
  63      advances the iterator by one multibyte character.
  64
  65    mbui_cur (iter)
  66      returns the current multibyte character, of type mbchar_t.  All the
  67      macros defined in mbchar.h can be used on it.
  68
  69    mbui_cur_ptr (iter)
  70      return a pointer to the beginning of the current multibyte character.
  71
  72    mbui_reloc (iter, ptrdiff)
  73      relocates iterator when the string is moved by ptrdiff bytes.
  74
  75    mbui_copy (&destiter, &srciter)
  76      copies srciter to destiter.
  77
  78    Here are the function prototypes of the macros.
  79
  80    extern void          mbui_init (mbui_iterator_t iter, const char *startptr);
  81    extern bool          mbui_avail (mbui_iterator_t iter);
  82    extern void          mbui_advance (mbui_iterator_t iter);
  83    extern mbchar_t      mbui_cur (mbui_iterator_t iter);
  84    extern const char *  mbui_cur_ptr (mbui_iterator_t iter);
  85    extern void          mbui_reloc (mbui_iterator_t iter, ptrdiff_t ptrdiff);
  86    extern void          mbui_copy (mbui_iterator_t *new, const mbui_iterator_t *old);
  87  */
  88
  89 #ifndef _MBUITER_H
  90 #define _MBUITER_H 1
  91
  92 /* This file uses _GL_INLINE_HEADER_BEGIN, _GL_INLINE,
  93    _GL_ATTRIBUTE_ALWAYS_INLINE.  */
  94 #if !_GL_CONFIG_H_INCLUDED
  95  #error "Please include config.h first."
  96 #endif
  97
  98 #include <assert.h>
  99 #include <stddef.h>
 100 #include <stdlib.h>
 101 #include <string.h>
 102 #include <uchar.h>
 103 #include <wchar.h>
 104
 105 #include "mbchar.h"
 106 #include "strnlen1.h"
 107
 108 _GL_INLINE_HEADER_BEGIN
 109 #ifndef MBUITER_INLINE
 110 # define MBUITER_INLINE _GL_INLINE _GL_ATTRIBUTE_ALWAYS_INLINE
 111 #endif
 112
 113 #ifdef __cplusplus
 114 extern "C" {
 115 #endif
 116
 117
 118 struct mbuiter_multi
 119 {
 120   #if !GNULIB_MBRTOC32_REGULAR
 121   bool in_shift;        /* true if next byte may not be interpreted as ASCII */
 122                         /* If GNULIB_MBRTOC32_REGULAR, it is always false,
 123                            so optimize it away.  */
 124   #endif
 125   mbstate_t state;      /* if in_shift: current shift state */
 126                         /* If GNULIB_MBRTOC32_REGULAR, it is in an initial state
 127                            before and after every mbuiter_multi_next invocation.
 128                          */
 129   bool next_done;       /* true if mbui_avail has already filled the following */
 130   unsigned int cur_max; /* A cache of MB_CUR_MAX.  */
 131   struct mbchar cur;    /* the current character:
 132         const char *cur.ptr          pointer to current character
 133         The following are only valid after mbui_avail.
 134         size_t cur.bytes             number of bytes of current character
 135         bool cur.wc_valid            true if wc is a valid 32-bit wide character
 136         char32_t cur.wc              if wc_valid: the current character
 137         */
 138 };
 139
 140 MBUITER_INLINE void
 141 mbuiter_multi_next (struct mbuiter_multi *iter)
 142 {
 143   if (iter->next_done)
 144     return;
 145   #if !GNULIB_MBRTOC32_REGULAR
 146   if (iter->in_shift)
 147     goto with_shift;
 148   #endif
 149   /* Handle most ASCII characters quickly, without calling mbrtowc().  */
 150   if (is_basic (*iter->cur.ptr))
 151     {
 152       /* These characters are part of the POSIX portable character set.
 153          For most of them, namely those in the ISO C basic character set,
 154          ISO C 99 guarantees that their wide character code is identical to
 155          their char code.  For the few other ones, this is the case as well,
 156          in all locale encodings that are in use.  The 32-bit wide character
 157          code is the same as well.  */
 158       iter->cur.bytes = 1;
 159       iter->cur.wc = *iter->cur.ptr;
 160       iter->cur.wc_valid = true;
 161     }
 162   else
 163     {
 164       assert (mbsinit (&iter->state));
 165       #if !GNULIB_MBRTOC32_REGULAR
 166       iter->in_shift = true;
 167     with_shift:
 168       #endif
 169       iter->cur.bytes = mbrtoc32 (&iter->cur.wc, iter->cur.ptr,
 170                                   strnlen1 (iter->cur.ptr, iter->cur_max),
 171                                   &iter->state);
 172       if (iter->cur.bytes == (size_t) -1)
 173         {
 174           /* An invalid multibyte sequence was encountered.  */
 175           iter->cur.bytes = 1;
 176           iter->cur.wc_valid = false;
 177           /* Allow the next invocation to continue from a sane state.  */
 178           #if !GNULIB_MBRTOC32_REGULAR
 179           iter->in_shift = false;
 180           #endif
 181           mbszero (&iter->state);
 182         }
 183       else if (iter->cur.bytes == (size_t) -2)
 184         {
 185           /* An incomplete multibyte character at the end.  */
 186           iter->cur.bytes = strlen (iter->cur.ptr);
 187           iter->cur.wc_valid = false;
 188           /* Whether to set iter->in_shift = false and reset iter->state
 189              or not is not important; the string end is reached anyway.  */
 190         }
 191       else
 192         {
 193           if (iter->cur.bytes == 0)
 194             {
 195               /* A null wide character was encountered.  */
 196               iter->cur.bytes = 1;
 197               assert (*iter->cur.ptr == '\0');
 198               assert (iter->cur.wc == 0);
 199             }
 200           #if !GNULIB_MBRTOC32_REGULAR
 201           else if (iter->cur.bytes == (size_t) -3)
 202             /* The previous multibyte sequence produced an additional 32-bit
 203                wide character.  */
 204             iter->cur.bytes = 0;
 205           #endif
 206           iter->cur.wc_valid = true;
 207
 208           /* When in an initial state, we can go back treating ASCII
 209              characters more quickly.  */
 210           #if !GNULIB_MBRTOC32_REGULAR
 211           if (mbsinit (&iter->state))
 212             iter->in_shift = false;
 213           #endif
 214         }
 215     }
 216   iter->next_done = true;
 217 }
 218
 219 MBUITER_INLINE void
 220 mbuiter_multi_reloc (struct mbuiter_multi *iter, ptrdiff_t ptrdiff)
 221 {
 222   iter->cur.ptr += ptrdiff;
 223 }
 224
 225 MBUITER_INLINE void
 226 mbuiter_multi_copy (struct mbuiter_multi *new_iter, const struct mbuiter_multi *old_iter)
 227 {
 228   #if !GNULIB_MBRTOC32_REGULAR
 229   if ((new_iter->in_shift = old_iter->in_shift))
 230     memcpy (&new_iter->state, &old_iter->state, sizeof (mbstate_t));
 231   else
 232   #endif
 233     mbszero (&new_iter->state);
 234   new_iter->next_done = old_iter->next_done;
 235   new_iter->cur_max = old_iter->cur_max;
 236   mb_copy (&new_iter->cur, &old_iter->cur);
 237 }
 238
 239 /* Iteration macros.  */
 240 typedef struct mbuiter_multi mbui_iterator_t;
 241 #if !GNULIB_MBRTOC32_REGULAR
 242 #define mbui_init(iter, startptr) \
 243   ((iter).cur.ptr = (startptr), \
 244    (iter).in_shift = false, mbszero (&(iter).state), \
 245    (iter).next_done = false, \
 246    (iter).cur_max = MB_CUR_MAX)
 247 #else
 248 /* Optimized: no in_shift.  */
 249 #define mbui_init(iter, startptr) \
 250   ((iter).cur.ptr = (startptr), \
 251    mbszero (&(iter).state), \
 252    (iter).next_done = false, \
 253    (iter).cur_max = MB_CUR_MAX)
 254 #endif
 255 #define mbui_avail(iter) \
 256   (mbuiter_multi_next (&(iter)), !mb_isnul ((iter).cur))
 257 #define mbui_advance(iter) \
 258   ((iter).cur.ptr += (iter).cur.bytes, (iter).next_done = false)
 259
 260 /* Access to the current character.  */
 261 #define mbui_cur(iter) (iter).cur
 262 #define mbui_cur_ptr(iter) (iter).cur.ptr
 263
 264 /* Relocation.  */
 265 #define mbui_reloc(iter, ptrdiff) mbuiter_multi_reloc (&iter, ptrdiff)
 266
 267 /* Copying an iterator.  */
 268 #define mbui_copy mbuiter_multi_copy
 269
 270
 271 #ifdef __cplusplus
 272 }
 273 #endif
 274
 275 _GL_INLINE_HEADER_END
 276
 277 #endif /* _MBUITER_H */