diff options
Diffstat (limited to 'include/linux')
-rw-r--r-- | include/linux/bsearch.h | 32 | ||||
-rw-r--r-- | include/linux/types.h | 2 |
2 files changed, 34 insertions, 0 deletions
diff --git a/include/linux/bsearch.h b/include/linux/bsearch.h new file mode 100644 index 0000000..e66b711 --- /dev/null +++ b/include/linux/bsearch.h @@ -0,0 +1,32 @@ +/* SPDX-License-Identifier: GPL-2.0 */ +#ifndef _LINUX_BSEARCH_H +#define _LINUX_BSEARCH_H + +#include <linux/types.h> + +static __always_inline +void *__inline_bsearch(const void *key, const void *base, size_t num, size_t size, cmp_func_t cmp) +{ + const char *pivot; + int result; + + while (num > 0) { + pivot = base + (num >> 1) * size; + result = cmp(key, pivot); + + if (result == 0) + return (void *)pivot; + + if (result > 0) { + base = pivot + size; + num--; + } + num >>= 1; + } + + return NULL; +} + +extern void *bsearch(const void *key, const void *base, size_t num, size_t size, cmp_func_t cmp); + +#endif /* _LINUX_BSEARCH_H */ diff --git a/include/linux/types.h b/include/linux/types.h index c9886cb..77f9673 100644 --- a/include/linux/types.h +++ b/include/linux/types.h @@ -76,4 +76,6 @@ typedef __u64 __bitwise __be64; typedef u64 sector_t; +typedef int (*cmp_func_t)(const void *a, const void *b); + #endif /* _TOOLS_LINUX_TYPES_H_ */ |