New module 'mbspbrk'.
[gnulib.git] / lib / mbspbrk.c
1 /* Searching a string for a character among 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 2, or (at your option)
8    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, write to the Free Software Foundation,
17    Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.  */
18
19 #include <config.h>
20
21 /* Specification.  */
22 #include <string.h>
23
24 #if HAVE_MBRTOWC
25 # include "mbuiter.h"
26 #endif
27
28 /* Find the first occurrence in the character string STRING of any character
29    in the character string ACCEPT.  Return the pointer to it, or NULL if none
30    exists.  */
31 char *
32 mbspbrk (const char *string, const char *accept)
33 {
34   /* Optimize two cases.  */
35   if (accept[0] == '\0')
36     return NULL;
37   if (accept[1] == '\0')
38     return mbschr (string, accept[0]);
39   /* General case.  */
40 #if HAVE_MBRTOWC
41   if (MB_CUR_MAX > 1)
42     {
43       mbui_iterator_t iter;
44
45       for (mbui_init (iter, string); mbui_avail (iter); mbui_advance (iter))
46         {
47           if (mb_len (mbui_cur (iter)) == 1)
48             {
49               if (mbschr (accept, (unsigned char) * mbui_cur_ptr (iter)))
50                 return (char *) mbui_cur_ptr (iter);
51             }
52           else
53             {
54               mbui_iterator_t aiter;
55
56               for (mbui_init (aiter, accept);
57                    mbui_avail (aiter);
58                    mbui_advance (aiter))
59                 if (mb_equal (mbui_cur (aiter), mbui_cur (iter)))
60                   return (char *) mbui_cur_ptr (iter);
61             }
62         }
63       return NULL;
64     }
65   else
66 #endif
67     return strpbrk (string, accept);
68 }