diff options
author | Mattias Andrée <maandree@kth.se> | 2018-10-20 20:48:13 +0200 |
---|---|---|
committer | Mattias Andrée <maandree@kth.se> | 2018-10-20 20:48:13 +0200 |
commit | 8f5bd0f85282e044da867bf12c38144b5131e915 (patch) | |
tree | f0bf62c380688942e4dedab69016bac1edc5c3ee /memelem.c | |
parent | Add man pages for memmem and memrmem (diff) | |
download | libsimple-8f5bd0f85282e044da867bf12c38144b5131e915.tar.gz libsimple-8f5bd0f85282e044da867bf12c38144b5131e915.tar.bz2 libsimple-8f5bd0f85282e044da867bf12c38144b5131e915.tar.xz |
Add memelem and memrelem
Signed-off-by: Mattias Andrée <maandree@kth.se>
Diffstat (limited to 'memelem.c')
-rw-r--r-- | memelem.c | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/memelem.c b/memelem.c new file mode 100644 index 0000000..2561456 --- /dev/null +++ b/memelem.c @@ -0,0 +1,73 @@ +/* See LICENSE file for copyright and license details. */ +#include "libsimple.h" +#ifndef TEST + + +void * +libsimple_memelem(const void *hay_, size_t hayn, const void *sub_, size_t subn) +{ + if (!subn) + return (void *)hay_; + + switch (subn) { + case 1: + return memchr(hay_, hayn, *(char *)sub_); + case 2: + { + uint16_t *hay = (void *)hay_; + uint16_t sub = *(uint16_t *)sub_; + for (; hayn--; hay++) + if (*hay == sub) + return hay; + break; + } + case 4: + { + uint32_t *hay = (void *)hay_; + uint32_t sub = *(uint32_t *)sub_; + for (; hayn--; hay++) + if (*hay == sub) + return hay; + break; + } + case 8: + { + uint64_t *hay = (void *)hay_; + uint64_t sub = *(uint64_t *)sub_; + for (; hayn--; hay++) + if (*hay == sub) + return hay; + break; + } + default: + { + char *hay = (void *)hay_; + const char *sub = sub_; + for (; hayn--; hay += subn) + if (!memcmp(hay, sub, subn)) + return hay; + break; + + } + } + + return NULL; +} + + +#else +#include "test.h" + +int +main(void) +{ + assert(!strcmpnul(libsimple_memelem("12345634", 4, "", 0), "12345634")); + assert(!strcmpnul(libsimple_memelem("12345634", 4, "34", 2), "345634")); + assert(!libsimple_memelem("12345634", 4, "xx", 2)); + assert(!libsimple_memelem("13456342", 4, "34", 2)); + assert(!libsimple_memelem("12345634", 0, "34", 2)); + assert(!strcmpnul(libsimple_memelem("12345634", 0, "34", 0), "12345634")); + return 0; +} + +#endif |