diff --git a/libc/src/string/memcmp.cpp b/libc/src/string/memcmp.cpp index 5569ab29f76531..2bb9552d9361a5 100644 --- a/libc/src/string/memcmp.cpp +++ b/libc/src/string/memcmp.cpp @@ -8,20 +8,44 @@ #include "src/string/memcmp.h" #include "src/__support/common.h" +#include "src/string/memory_utils/elements.h" + #include // size_t namespace __llvm_libc { -// TODO: It is a simple implementation, an optimized version is preparing. +static int memcmp_impl(const char *lhs, const char *rhs, size_t count) { +#if defined(__i386__) || defined(__x86_64__) + using namespace ::__llvm_libc::x86; +#else + using namespace ::__llvm_libc::scalar; +#endif + + if (count == 0) + return 0; + if (count == 1) + return ThreeWayCompare<_1>(lhs, rhs); + if (count == 2) + return ThreeWayCompare<_2>(lhs, rhs); + if (count == 3) + return ThreeWayCompare<_3>(lhs, rhs); + if (count <= 8) + return ThreeWayCompare>(lhs, rhs, count); + if (count <= 16) + return ThreeWayCompare>(lhs, rhs, count); + if (count <= 32) + return ThreeWayCompare>(lhs, rhs, count); + if (count <= 64) + return ThreeWayCompare>(lhs, rhs, count); + if (count <= 128) + return ThreeWayCompare>(lhs, rhs, count); + return ThreeWayCompare::Then>>(lhs, rhs, count); +} + LLVM_LIBC_FUNCTION(int, memcmp, (const void *lhs, const void *rhs, size_t count)) { - const unsigned char *_lhs = reinterpret_cast(lhs); - const unsigned char *_rhs = reinterpret_cast(rhs); - for (size_t i = 0; i < count; ++i) - if (_lhs[i] != _rhs[i]) - return _lhs[i] - _rhs[i]; - // count is 0 or _lhs and _rhs are the same. - return 0; + return memcmp_impl(static_cast(lhs), + static_cast(rhs), count); } } // namespace __llvm_libc