Merge pull request #125 from devnexen/libtokencap_memmem

libtokencap, simple optimised memmem implementation enough for this l…

1 files changed, 30 insertions, 0 deletions

diff --git a/libtokencap/libtokencap.so.c b/libtokencap/libtokencap.so.c
index 467be05b..eea6d29f 100644
--- a/libtokencap/libtokencap.so.c
+++ b/libtokencap/libtokencap.so.c
@@ -425,6 +425,36 @@ char* strcasestr(const char* haystack, const char* needle) {
 
 }
 
+#undef memmem
+
+void* memmem(const void *haystack, size_t haystack_len, const void *needle, size_t needle_len)
+{
+  if (__tokencap_is_ro(haystack))
+    __tokencap_dump(haystack, haystack_len, 1);
+
+  if (__tokencap_is_ro(needle)) __tokencap_dump(needle, needle_len, 1);
+
+  const char *n = (const char *)needle;
+  const char *h = (const char *)haystack;
+  if (haystack_len < needle_len) return 0;
+  if (needle_len == 0) return (void *)haystack;
+  if (needle_len == 1) return memchr(haystack, *n, haystack_len);
+
+  const char *end = h + (haystack_len - needle_len);
+
+  do {
+
+    if (*h == *n) {
+
+	    if (memcmp(h, n, needle_len) == 0) return (void *)h;
+    }
+
+  } while (h++ <= end);
+
+  return 0;
+
+}
+
 /* Init code to open the output file (or default to stderr). */
 
 __attribute__((constructor)) void __tokencap_init(void) {