lib/mbsspn.c

   1 /* Searching a string for a character outside a given set of characters.
   2    Copyright (C) 1999, 2002, 2006-2007 Free Software Foundation, Inc.
   3    Written by Bruno Haible <bruno@clisp.org>, 2007.
   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 #include <config.h>
  19
  20 /* Specification.  */
  21 #include <string.h>
  22
  23 #if HAVE_MBRTOWC
  24 # include "mbuiter.h"
  25 #endif
  26
  27 /* Find the first occurrence in the character string STRING of any character
  28    not in the character string REJECT.  Return the number of bytes from the
  29    beginning of the string to this occurrence, or to the end of the string
  30    if none exists.  */
  31 size_t
  32 mbsspn (const char *string, const char *reject)
  33 {
  34   /* Optimize two cases.  */
  35   if (reject[0] == '\0')
  36     return 0;
  37   if (reject[1] == '\0')
  38     {
  39       unsigned char uc = (unsigned char) reject[0];
  40
  41 #if HAVE_MBRTOWC
  42       if (MB_CUR_MAX > 1)
  43         {
  44           mbui_iterator_t iter;
  45
  46           for (mbui_init (iter, string); mbui_avail (iter); mbui_advance (iter))
  47             if (!(mb_len (mbui_cur (iter)) == 1
  48                   && (unsigned char) * mbui_cur_ptr (iter) == uc))
  49               break;
  50           return mbui_cur_ptr (iter) - string;
  51         }
  52       else
  53 #endif
  54         {
  55           const char *ptr;
  56
  57           for (ptr = string; *ptr != '\0'; ptr++)
  58             if ((unsigned char) *ptr != uc)
  59               break;
  60           return ptr - string;
  61         }
  62     }
  63   /* General case.  */
  64 #if HAVE_MBRTOWC
  65   if (MB_CUR_MAX > 1)
  66     {
  67       mbui_iterator_t iter;
  68
  69       for (mbui_init (iter, string); mbui_avail (iter); mbui_advance (iter))
  70         {
  71           if (mb_len (mbui_cur (iter)) == 1)
  72             {
  73               if (mbschr (reject, * mbui_cur_ptr (iter)) == NULL)
  74                 goto found;
  75             }
  76           else
  77             {
  78               mbui_iterator_t aiter;
  79
  80               for (mbui_init (aiter, reject);; mbui_advance (aiter))
  81                 {
  82                   if (!mbui_avail (aiter))
  83                     goto found;
  84                   if (mb_equal (mbui_cur (aiter), mbui_cur (iter)))
  85                     break;
  86                 }
  87             }
  88         }
  89      found:
  90       return mbui_cur_ptr (iter) - string;
  91     }
  92   else
  93 #endif
  94     return strspn (string, reject);
  95 }