strspn.c

linux内核

/*
 * strspn, strcspn
 */

#include <string.h>
#include <stddef.h>
#include <inttypes.h>
#include <limits.h>

#ifndef LONG_BIT
#define LONG_BIT (CHAR_BIT*sizeof(long))
#endif

static inline void
set_bit(unsigned long *bitmap, unsigned int bit)
{
  bitmap[bit/LONG_BIT] |= 1UL << (bit%LONG_BIT);
}

static inline int
test_bit(unsigned long *bitmap, unsigned int bit)
{
  return (int)(bitmap[bit/LONG_BIT] >> (bit%LONG_BIT)) & 1;
}

static size_t
strxspn(const char *s, const char *map, int parity)
{
  unsigned long matchmap[((1 << CHAR_BIT)+LONG_BIT-1)/LONG_BIT];
  size_t n = 0;

  /* Create bitmap */
  memset(matchmap, 0, sizeof matchmap);
  while ( *map )
    set_bit(matchmap, (unsigned char) *map++);

  /* Make sure the null character never matches */
  if ( parity )
    set_bit(matchmap, 0);

  /* Calculate span length */
  while ( test_bit(matchmap, (unsigned char) *s++)^parity )
    n++;

  return n;
}

size_t
strspn(const char *s, const char *accept)
{
  return strxspn(s, accept, 0);
}

size_t
strcspn(const char *s, const char *reject)
{
  return strxspn(s, reject, 1);
}

char *
strpbrk(const char *s, const char *accept)
{
  const char *ss = s+strxspn(s, accept, 1);
  
  return *ss ? (char *)ss : NULL;
}