ExternalVendorCode/AFLplusplus
1
0
Fork 0
mirror of https://github.com/AFLplusplus/AFLplusplus.git synced 2025-06-11 17:51:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #125 from devnexen/libtokencap_memmem

Browse Source 
libtokencap, simple optimised memmem implementation enough for this l…
This commit is contained in:
Andrea Fioraldi 2019-12-02 13:55:51 +01:00 committed by GitHub
commit 25b435060c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
libtokencap/libtokencap.so.c
View File

@ -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) {