strnstr.c 1.59 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
/*****************************************************************************
 * strnstr.c: BSD strnstr() replacement
 *****************************************************************************
 * Copyright © 2015 VLC authors and VideoLAN
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
 *****************************************************************************/

#ifdef HAVE_CONFIG_H
# include <config.h>
#endif

#include <string.h>
26
#include <assert.h>
27 28 29

char * strnstr (const char *haystack, const char *needle, size_t len)
{
30
    assert(needle != NULL);
31

32
    const size_t i = strlen(needle);
33 34 35
    if (i == 0) /* corner case (if haystack is NULL, memcmp not allowed) */
        return (char *)haystack;

36
    if( len < i )
37
      return NULL;
38

39 40
    size_t count = len - i;

41
    do
42
    {
43
      if( memcmp(haystack, needle, i) == 0 )
44 45 46
        return (char*) haystack;
      haystack++;
    }
47 48
    while(count--);

49 50
    return NULL;
}