lib/libc/string/wcscspn.c

   1 /*      $NetBSD: wcscspn.c,v 1.4 2011/11/24 18:44:25 joerg Exp $        */
   2
   3 /*-
   4  * Copyright (c) 1999 Citrus Project,
   5  * Copyright (c) 2011 Joerg Sonnenberger,
   6  * All rights reserved.
   7  *
   8  * Redistribution and use in source and binary forms, with or without
   9  * modification, are permitted provided that the following conditions
  10  * are met:
  11  * 1. Redistributions of source code must retain the above copyright
  12  *    notice, this list of conditions and the following disclaimer.
  13  * 2. Redistributions in binary form must reproduce the above copyright
  14  *    notice, this list of conditions and the following disclaimer in the
  15  *    documentation and/or other materials provided with the distribution.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
  18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
  21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  27  * SUCH DAMAGE.
  28  *
  29  *      citrus Id: wcscspn.c,v 1.1 1999/12/29 21:47:45 tshiozak Exp
  30  */
  31
  32 #include <sys/cdefs.h>
  33 __RCSID("$NetBSD: wcscspn.c,v 1.4 2011/11/24 18:44:25 joerg Exp $");
  34
  35 #include <assert.h>
  36 #include <inttypes.h>
  37 #include <string.h>
  38 #include <wchar.h>
  39
  40 #include "wcscspn_bloom.h"
  41
  42 size_t
  43 wcscspn(const wchar_t *s, const wchar_t *set)
  44 {
  45         size_t bloom[BLOOM_ARRAY_SIZE];
  46         const wchar_t *p;
  47         const wchar_t *q;
  48
  49         _DIAGASSERT(s != NULL);
  50         _DIAGASSERT(set != NULL);
  51
  52         if (set[0] == '\0')
  53                 return wcslen(s);
  54         if (set[1] == '\0') {
  55                 for (p = s; *p; ++p)
  56                         if (*p == set[0])
  57                                 break;
  58                 return p - s;
  59         }
  60
  61         wcsspn_bloom_init(bloom, set);
  62
  63         for (p = s; *p; ++p) {
  64                 if (!wcsspn_in_bloom(bloom, *p))
  65                         continue;
  66
  67                 q = set;
  68                 do {
  69                         if (*p == *q)
  70                                 goto done;
  71                 } while (*++q);
  72         }
  73
  74 done:
  75         return (p - s);
  76 }