2019-03-03 20:08:39 +00:00
|
|
|
#pragma once
|
|
|
|
|
|
|
|
#include <algorithm>
|
2022-07-31 14:34:05 +00:00
|
|
|
#include <bit>
|
2022-06-15 13:19:29 +00:00
|
|
|
#include <cstdint>
|
2019-03-03 20:08:39 +00:00
|
|
|
|
2019-07-24 22:31:39 +00:00
|
|
|
#include <Core/Defines.h>
|
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
namespace detail
|
|
|
|
{
|
|
|
|
|
|
|
|
template <typename T>
|
|
|
|
inline int cmp(T a, T b)
|
|
|
|
{
|
|
|
|
if (a < b)
|
|
|
|
return -1;
|
|
|
|
if (a > b)
|
|
|
|
return 1;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2019-07-25 00:28:27 +00:00
|
|
|
|
|
|
|
/// We can process uninitialized memory in the functions below.
|
|
|
|
/// Results don't depend on the values inside uninitialized memory but Memory Sanitizer cannot see it.
|
|
|
|
/// Disable optimized functions if compile with Memory Sanitizer.
|
2022-01-14 12:21:22 +00:00
|
|
|
#if defined(__AVX512BW__) && defined(__AVX512VL__) && !defined(MEMORY_SANITIZER)
|
2022-06-15 13:19:29 +00:00
|
|
|
# include <immintrin.h>
|
2019-07-25 00:28:27 +00:00
|
|
|
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
/** All functions works under the following assumptions:
|
|
|
|
* - it's possible to read up to 15 excessive bytes after end of 'a' and 'b' region;
|
|
|
|
* - memory regions are relatively small and extra loop unrolling is not worth to do.
|
|
|
|
*/
|
|
|
|
|
|
|
|
/** Variant when memory regions may have different sizes.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size = std::min(a_size, b_size);
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < min_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
2022-06-15 13:19:29 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset)),
|
|
|
|
_MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (offset >= min_size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return detail::cmp(a_size, b_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when memory regions may have different sizes.
|
|
|
|
* But compare the regions as the smaller one is padded with zero bytes up to the size of the larger.
|
|
|
|
* It's needed to hold that: toFixedString('abc', 5) = 'abc'
|
|
|
|
* for compatibility with SQL standard.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallLikeZeroPaddedAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size = std::min(a_size, b_size);
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < min_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
2022-06-15 13:19:29 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset)),
|
|
|
|
_MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (offset >= min_size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The strings are equal up to min_size.
|
|
|
|
/// If the rest of the larger string is zero bytes then the strings are considered equal.
|
|
|
|
|
|
|
|
size_t max_size;
|
|
|
|
const Char * longest;
|
|
|
|
int cmp;
|
|
|
|
|
|
|
|
if (a_size == b_size)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
else if (a_size > b_size)
|
|
|
|
{
|
|
|
|
max_size = a_size;
|
|
|
|
longest = a;
|
|
|
|
cmp = 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
max_size = b_size;
|
|
|
|
longest = b;
|
|
|
|
cmp = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
const __m128i zero16 = _mm_setzero_si128();
|
|
|
|
|
|
|
|
for (size_t offset = min_size; offset < max_size; offset += 16)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
uint16_t mask = _mm_cmpneq_epi8_mask(_mm_loadu_si128(reinterpret_cast<const __m128i *>(longest + offset)), zero16);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (offset >= max_size)
|
|
|
|
return 0;
|
|
|
|
return cmp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when memory regions have same size.
|
|
|
|
* TODO Check if the compiler can optimize previous function when the caller pass identical sizes.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
2022-06-15 13:19:29 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset)),
|
|
|
|
_MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (offset >= size)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Compare memory regions for equality.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline bool memequalSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
if (a_size != b_size)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < a_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
2022-06-15 13:19:29 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset)),
|
|
|
|
_MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
return offset >= a_size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when the caller know in advance that the size is a multiple of 16.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallMultipleOf16(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
2022-06-15 13:19:29 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset)),
|
|
|
|
_MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when the size is 16 exactly.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmp16(const Char * a, const Char * b)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
2022-06-15 13:19:29 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a)), _mm_loadu_si128(reinterpret_cast<const __m128i *>(b)), _MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
auto offset = std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when the size is 16 exactly.
|
|
|
|
*/
|
|
|
|
inline bool memequal16(const void * a, const void * b)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
return 0xFFFF
|
|
|
|
== _mm_cmp_epi8_mask(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a)), _mm_loadu_si128(reinterpret_cast<const __m128i *>(b)), _MM_CMPINT_EQ);
|
2022-01-14 12:21:22 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Compare memory region to zero */
|
|
|
|
inline bool memoryIsZeroSmallAllowOverflow15(const void * data, size_t size)
|
|
|
|
{
|
|
|
|
const __m128i zero16 = _mm_setzero_si128();
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
uint16_t mask = _mm_cmp_epi8_mask(
|
|
|
|
zero16, _mm_loadu_si128(reinterpret_cast<const __m128i *>(reinterpret_cast<const char *>(data) + offset)), _MM_CMPINT_NE);
|
2022-01-14 12:21:22 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2022-01-14 12:21:22 +00:00
|
|
|
return offset >= size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
#elif defined(__SSE2__) && !defined(MEMORY_SANITIZER)
|
2022-06-15 13:19:29 +00:00
|
|
|
# include <emmintrin.h>
|
2019-07-25 00:28:27 +00:00
|
|
|
|
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
/** All functions works under the following assumptions:
|
|
|
|
* - it's possible to read up to 15 excessive bytes after end of 'a' and 'b' region;
|
|
|
|
* - memory regions are relatively small and extra loop unrolling is not worth to do.
|
|
|
|
*/
|
|
|
|
|
|
|
|
/** Variant when memory regions may have different sizes.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size = std::min(a_size, b_size);
|
|
|
|
|
2019-03-03 23:12:50 +00:00
|
|
|
for (size_t offset = 0; offset < min_size; offset += 16)
|
2019-03-03 20:08:39 +00:00
|
|
|
{
|
2019-03-04 15:50:23 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(_mm_cmpeq_epi8(
|
2019-03-03 20:08:39 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset))));
|
2019-03-04 15:50:23 +00:00
|
|
|
mask = ~mask;
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (offset >= min_size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return detail::cmp(a_size, b_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2020-04-20 01:01:31 +00:00
|
|
|
/** Variant when memory regions may have different sizes.
|
|
|
|
* But compare the regions as the smaller one is padded with zero bytes up to the size of the larger.
|
|
|
|
* It's needed to hold that: toFixedString('abc', 5) = 'abc'
|
|
|
|
* for compatibility with SQL standard.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallLikeZeroPaddedAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size = std::min(a_size, b_size);
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < min_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint16_t mask = _mm_movemask_epi8(_mm_cmpeq_epi8(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2020-04-20 01:01:31 +00:00
|
|
|
|
|
|
|
if (offset >= min_size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The strings are equal up to min_size.
|
|
|
|
/// If the rest of the larger string is zero bytes then the strings are considered equal.
|
|
|
|
|
|
|
|
size_t max_size;
|
|
|
|
const Char * longest;
|
|
|
|
int cmp;
|
|
|
|
|
|
|
|
if (a_size == b_size)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
else if (a_size > b_size)
|
|
|
|
{
|
|
|
|
max_size = a_size;
|
|
|
|
longest = a;
|
|
|
|
cmp = 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
max_size = b_size;
|
|
|
|
longest = b;
|
|
|
|
cmp = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
const __m128i zero16 = _mm_setzero_si128();
|
|
|
|
|
|
|
|
for (size_t offset = min_size; offset < max_size; offset += 16)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(_mm_cmpeq_epi8(_mm_loadu_si128(reinterpret_cast<const __m128i *>(longest + offset)), zero16));
|
2021-01-29 07:37:57 +00:00
|
|
|
mask = ~mask;
|
2020-04-20 01:01:31 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2020-04-20 01:01:31 +00:00
|
|
|
|
|
|
|
if (offset >= max_size)
|
|
|
|
return 0;
|
|
|
|
return cmp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
/** Variant when memory regions have same size.
|
|
|
|
* TODO Check if the compiler can optimize previous function when the caller pass identical sizes.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
2019-03-03 23:12:50 +00:00
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
2019-03-03 20:08:39 +00:00
|
|
|
{
|
2019-03-04 15:50:23 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(_mm_cmpeq_epi8(
|
2019-03-03 20:08:39 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset))));
|
2019-03-04 15:50:23 +00:00
|
|
|
mask = ~mask;
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (offset >= size)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Compare memory regions for equality.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline bool memequalSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
if (a_size != b_size)
|
|
|
|
return false;
|
|
|
|
|
2019-03-03 23:12:50 +00:00
|
|
|
for (size_t offset = 0; offset < a_size; offset += 16)
|
2019-03-03 20:08:39 +00:00
|
|
|
{
|
2019-03-04 15:50:23 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(_mm_cmpeq_epi8(
|
2019-03-03 20:08:39 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset))));
|
2019-03-04 15:50:23 +00:00
|
|
|
mask = ~mask;
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2019-03-03 20:08:39 +00:00
|
|
|
return offset >= a_size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when the caller know in advance that the size is a multiple of 16.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallMultipleOf16(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
2019-03-04 15:50:23 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(_mm_cmpeq_epi8(
|
2019-03-03 20:08:39 +00:00
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a + offset)),
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(b + offset))));
|
2019-03-04 15:50:23 +00:00
|
|
|
mask = ~mask;
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2019-03-03 20:08:39 +00:00
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when the size is 16 exactly.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmp16(const Char * a, const Char * b)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(
|
|
|
|
_mm_cmpeq_epi8(_mm_loadu_si128(reinterpret_cast<const __m128i *>(a)), _mm_loadu_si128(reinterpret_cast<const __m128i *>(b))));
|
2019-03-04 15:50:23 +00:00
|
|
|
mask = ~mask;
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
auto offset = std::countr_zero(mask);
|
2019-03-03 20:08:39 +00:00
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Variant when the size is 16 exactly.
|
|
|
|
*/
|
|
|
|
inline bool memequal16(const void * a, const void * b)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
return 0xFFFF
|
|
|
|
== _mm_movemask_epi8(_mm_cmpeq_epi8(
|
|
|
|
_mm_loadu_si128(reinterpret_cast<const __m128i *>(a)), _mm_loadu_si128(reinterpret_cast<const __m128i *>(b))));
|
2019-03-03 20:08:39 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Compare memory region to zero */
|
|
|
|
inline bool memoryIsZeroSmallAllowOverflow15(const void * data, size_t size)
|
|
|
|
{
|
2019-03-03 21:07:52 +00:00
|
|
|
const __m128i zero16 = _mm_setzero_si128();
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
2022-06-15 13:19:29 +00:00
|
|
|
uint16_t mask = _mm_movemask_epi8(
|
|
|
|
_mm_cmpeq_epi8(zero16, _mm_loadu_si128(reinterpret_cast<const __m128i *>(reinterpret_cast<const char *>(data) + offset))));
|
2019-03-04 15:50:23 +00:00
|
|
|
mask = ~mask;
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask);
|
2019-03-03 20:08:39 +00:00
|
|
|
return offset >= size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2022-06-15 13:19:29 +00:00
|
|
|
#elif defined(__aarch64__) && defined(__ARM_NEON)
|
|
|
|
|
|
|
|
# include <arm_neon.h>
|
2023-03-17 13:27:48 +00:00
|
|
|
# pragma clang diagnostic ignored "-Wreserved-identifier"
|
2022-06-15 13:19:29 +00:00
|
|
|
|
|
|
|
inline uint64_t getNibbleMask(uint8x16_t res)
|
|
|
|
{
|
|
|
|
return vget_lane_u64(vreinterpret_u64_u8(vshrn_n_u16(vreinterpretq_u16_u8(res), 4)), 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size = std::min(a_size, b_size);
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < min_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqq_u8(
|
|
|
|
vld1q_u8(reinterpret_cast<const unsigned char *>(a + offset)), vld1q_u8(reinterpret_cast<const unsigned char *>(b + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
|
|
|
|
if (offset >= min_size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return detail::cmp(a_size, b_size);
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallLikeZeroPaddedAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size = std::min(a_size, b_size);
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < min_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqq_u8(
|
|
|
|
vld1q_u8(reinterpret_cast<const unsigned char *>(a + offset)), vld1q_u8(reinterpret_cast<const unsigned char *>(b + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
|
|
|
|
if (offset >= min_size)
|
|
|
|
break;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/// The strings are equal up to min_size.
|
|
|
|
/// If the rest of the larger string is zero bytes then the strings are
|
|
|
|
/// considered equal.
|
|
|
|
|
|
|
|
size_t max_size;
|
|
|
|
const Char * longest;
|
|
|
|
int cmp;
|
|
|
|
|
|
|
|
if (a_size == b_size)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
else if (a_size > b_size)
|
|
|
|
{
|
|
|
|
max_size = a_size;
|
|
|
|
longest = a;
|
|
|
|
cmp = 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
max_size = b_size;
|
|
|
|
longest = b;
|
|
|
|
cmp = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (size_t offset = min_size; offset < max_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqzq_u8(vld1q_u8(reinterpret_cast<const unsigned char *>(longest + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
|
|
|
|
if (offset >= max_size)
|
|
|
|
return 0;
|
|
|
|
return cmp;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqq_u8(
|
|
|
|
vld1q_u8(reinterpret_cast<const unsigned char *>(a + offset)), vld1q_u8(reinterpret_cast<const unsigned char *>(b + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
|
|
|
|
if (offset >= size)
|
|
|
|
return 0;
|
|
|
|
|
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Char>
|
|
|
|
inline bool memequalSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
if (a_size != b_size)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
for (size_t offset = 0; offset < a_size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqq_u8(
|
|
|
|
vld1q_u8(reinterpret_cast<const unsigned char *>(a + offset)), vld1q_u8(reinterpret_cast<const unsigned char *>(b + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
return offset >= a_size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallMultipleOf16(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqq_u8(
|
|
|
|
vld1q_u8(reinterpret_cast<const unsigned char *>(a + offset)), vld1q_u8(reinterpret_cast<const unsigned char *>(b + offset))));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
template <typename Char>
|
|
|
|
inline int memcmp16(const Char * a, const Char * b)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(
|
|
|
|
vceqq_u8(vld1q_u8(reinterpret_cast<const unsigned char *>(a)), vld1q_u8(reinterpret_cast<const unsigned char *>(b))));
|
|
|
|
mask = ~mask;
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
auto offset = std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
return detail::cmp(a[offset], b[offset]);
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
inline bool memequal16(const void * a, const void * b)
|
|
|
|
{
|
|
|
|
return 0xFFFFFFFFFFFFFFFFull
|
|
|
|
== getNibbleMask(
|
|
|
|
vceqq_u8(vld1q_u8(reinterpret_cast<const unsigned char *>(a)), vld1q_u8(reinterpret_cast<const unsigned char *>(b))));
|
|
|
|
}
|
|
|
|
|
|
|
|
inline bool memoryIsZeroSmallAllowOverflow15(const void * data, size_t size)
|
|
|
|
{
|
|
|
|
for (size_t offset = 0; offset < size; offset += 16)
|
|
|
|
{
|
|
|
|
uint64_t mask = getNibbleMask(vceqzq_u8(vld1q_u8(reinterpret_cast<const unsigned char *>(data) + offset)));
|
|
|
|
mask = ~mask;
|
|
|
|
|
|
|
|
if (mask)
|
|
|
|
{
|
2022-07-31 14:34:05 +00:00
|
|
|
offset += std::countr_zero(mask) >> 2;
|
2022-06-15 13:19:29 +00:00
|
|
|
return offset >= size;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
2019-03-03 20:08:39 +00:00
|
|
|
|
|
|
|
#else
|
|
|
|
|
2022-06-15 14:40:21 +00:00
|
|
|
#include <cstring>
|
2019-03-03 22:46:46 +00:00
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
2019-07-25 00:28:27 +00:00
|
|
|
if (auto res = memcmp(a, b, std::min(a_size, b_size)))
|
|
|
|
return res;
|
|
|
|
else
|
|
|
|
return detail::cmp(a_size, b_size);
|
2019-03-03 20:08:39 +00:00
|
|
|
}
|
|
|
|
|
2020-04-20 01:01:31 +00:00
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallLikeZeroPaddedAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
size_t min_size;
|
|
|
|
size_t max_size;
|
|
|
|
const Char * longest;
|
|
|
|
int size_cmp;
|
|
|
|
|
|
|
|
if (a_size == b_size)
|
|
|
|
{
|
|
|
|
min_size = a_size;
|
|
|
|
max_size = a_size;
|
|
|
|
longest = a;
|
|
|
|
size_cmp = 0;
|
|
|
|
}
|
|
|
|
else if (a_size > b_size)
|
|
|
|
{
|
|
|
|
min_size = b_size;
|
|
|
|
max_size = a_size;
|
|
|
|
longest = a;
|
|
|
|
size_cmp = 1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
min_size = a_size;
|
|
|
|
max_size = b_size;
|
|
|
|
longest = b;
|
|
|
|
size_cmp = -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (auto res = memcmp(a, b, min_size))
|
|
|
|
return res;
|
|
|
|
|
|
|
|
for (size_t i = min_size; i < max_size; ++i)
|
|
|
|
if (longest[i] != 0)
|
|
|
|
return size_cmp;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallAllowOverflow15(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
return memcmp(a, b, size);
|
|
|
|
}
|
|
|
|
|
2019-03-03 23:16:38 +00:00
|
|
|
template <typename Char>
|
|
|
|
inline bool memequalSmallAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
return a_size == b_size && 0 == memcmp(a, b, a_size);
|
|
|
|
}
|
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
template <typename Char>
|
|
|
|
inline int memcmpSmallMultipleOf16(const Char * a, const Char * b, size_t size)
|
|
|
|
{
|
|
|
|
return memcmp(a, b, size);
|
|
|
|
}
|
|
|
|
|
2019-03-03 23:16:38 +00:00
|
|
|
template <typename Char>
|
|
|
|
inline int memcmp16(const Char * a, const Char * b)
|
|
|
|
{
|
|
|
|
return memcmp(a, b, 16);
|
|
|
|
}
|
|
|
|
|
|
|
|
inline bool memequal16(const void * a, const void * b)
|
|
|
|
{
|
|
|
|
return 0 == memcmp(a, b, 16);
|
|
|
|
}
|
|
|
|
|
2019-03-03 20:08:39 +00:00
|
|
|
inline bool memoryIsZeroSmallAllowOverflow15(const void * data, size_t size)
|
|
|
|
{
|
|
|
|
const char * pos = reinterpret_cast<const char *>(data);
|
|
|
|
const char * end = pos + size;
|
|
|
|
|
|
|
|
for (; pos < end; ++pos)
|
|
|
|
if (*pos)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|
2020-04-20 01:01:31 +00:00
|
|
|
|
|
|
|
|
|
|
|
/** Compare memory regions for equality.
|
|
|
|
* But if the sizes are different, compare the regions as the smaller one is padded with zero bytes up to the size of the larger.
|
|
|
|
*/
|
|
|
|
template <typename Char>
|
|
|
|
inline bool memequalSmallLikeZeroPaddedAllowOverflow15(const Char * a, size_t a_size, const Char * b, size_t b_size)
|
|
|
|
{
|
|
|
|
return 0 == memcmpSmallLikeZeroPaddedAllowOverflow15(a, a_size, b, b_size);
|
|
|
|
}
|