Define String8 and String16 operator<=>

Previously, in C++20 mode, using <=> on String8 or String16 would
compare the pointer values from the implicit conversion operators
returning const char* and const char16_t*. Instead, compare the string
content. This is especially a problem with STL containers that prefer
<=> when it is available.

See https://godbolt.org/z/cc1hW17h3 for a demo of the problem.

Bug: 339775405
Test: treehugger
Change-Id: I5ee6af96dee0c9968a1bab1ad96729e45bb1ac50
This commit is contained in:
Ryan Prichard 2024-05-10 01:37:09 -07:00
parent 53748d08e5
commit f01048743a
4 changed files with 161 additions and 1 deletions

View file

@ -16,6 +16,8 @@
#include <utils/String16.h>
#include <utils/String8.h>
#include <compare>
#include <utility>
#include <gtest/gtest.h>
@ -257,3 +259,45 @@ TEST(String16Test, insert) {
EXPECT_EQ(NO_MEMORY, s.insert(3, u"", SIZE_MAX));
EXPECT_STR16EQ(u"foo!bar", s.c_str());
}
TEST(String16Test, comparisons) {
const char16_t* cstr1 = u"abc";
const char16_t* cstr2 = u"def";
// str1 and str1b will point to different blocks of memory but with equal contents.
String16 str1(cstr1);
String16 str1b(cstr1);
String16 str2(cstr2);
EXPECT_TRUE((str1 <=> str1b) == 0);
EXPECT_FALSE(str1 != str1b);
EXPECT_FALSE(str1 < str1b);
EXPECT_TRUE(str1 <= str1b);
EXPECT_TRUE(str1 == str1b);
EXPECT_TRUE(str1 >= str1b);
EXPECT_FALSE(str1 > str1b);
EXPECT_TRUE((str1 <=> str2) < 0);
EXPECT_TRUE((str2 <=> str1) > 0);
EXPECT_TRUE(str1 != str2);
EXPECT_TRUE(str1 < str2);
EXPECT_TRUE(str1 <= str2);
EXPECT_FALSE(str1 == str2);
EXPECT_FALSE(str1 >= str2);
EXPECT_FALSE(str1 > str2);
// Verify that pre-C++20 comparison operators work with a std::pair of a String8, which only
// provides <=> in C++20 and up. See b/339775405.
std::pair<String16, int> pair1(str1, 13);
std::pair<String16, int> pair1b(str1b, 13);
std::pair<String16, int> pair2(str2, 13);
EXPECT_TRUE(pair1 == pair1b);
EXPECT_FALSE(pair1 < pair1b);
EXPECT_FALSE(pair1 > pair1b);
EXPECT_TRUE(pair1 != pair2);
EXPECT_TRUE(pair1 < pair2);
EXPECT_FALSE(pair1 > pair2);
}

View file

@ -17,8 +17,10 @@
#define LOG_TAG "String8_test"
#include <log/log.h>
#include <utils/String8.h>
#include <utils/String16.h>
#include <utils/String8.h>
#include <compare>
#include <utility>
#include <gtest/gtest.h>
@ -132,3 +134,45 @@ TEST_F(String8Test, removeAll) {
EXPECT_TRUE(s.removeAll("o"));
EXPECT_STREQ("Hell, wrld!", s.c_str());
}
TEST_F(String8Test, comparisons) {
const char* cstr1 = "abc";
const char* cstr2 = "def";
// str1 and str1b will point to different blocks of memory but with equal contents.
String8 str1(cstr1);
String8 str1b(cstr1);
String8 str2(cstr2);
EXPECT_TRUE((str1 <=> str1b) == 0);
EXPECT_FALSE(str1 != str1b);
EXPECT_FALSE(str1 < str1b);
EXPECT_TRUE(str1 <= str1b);
EXPECT_TRUE(str1 == str1b);
EXPECT_TRUE(str1 >= str1b);
EXPECT_FALSE(str1 > str1b);
EXPECT_TRUE((str1 <=> str2) < 0);
EXPECT_TRUE((str2 <=> str1) > 0);
EXPECT_TRUE(str1 != str2);
EXPECT_TRUE(str1 < str2);
EXPECT_TRUE(str1 <= str2);
EXPECT_FALSE(str1 == str2);
EXPECT_FALSE(str1 >= str2);
EXPECT_FALSE(str1 > str2);
// Verify that pre-C++20 comparison operators work with a std::pair of a String8, which only
// provides <=> in C++20 and up. See b/339775405.
std::pair<String8, int> pair1(str1, 13);
std::pair<String8, int> pair1b(str1b, 13);
std::pair<String8, int> pair2(str2, 13);
EXPECT_TRUE(pair1 == pair1b);
EXPECT_FALSE(pair1 < pair1b);
EXPECT_FALSE(pair1 > pair1b);
EXPECT_TRUE(pair1 != pair2);
EXPECT_TRUE(pair1 < pair2);
EXPECT_FALSE(pair1 > pair2);
}

View file

@ -29,6 +29,10 @@
#define HAS_STRING_VIEW
#endif
#if __cplusplus >= 202002L
#include <compare>
#endif
// ---------------------------------------------------------------------------
namespace android {
@ -105,6 +109,9 @@ public:
inline bool operator!=(const String16& other) const;
inline bool operator>=(const String16& other) const;
inline bool operator>(const String16& other) const;
#if __cplusplus >= 202002L
inline std::strong_ordering operator<=>(const String16& other) const;
#endif
inline bool operator<(const char16_t* other) const;
inline bool operator<=(const char16_t* other) const;
@ -112,6 +119,9 @@ public:
inline bool operator!=(const char16_t* other) const;
inline bool operator>=(const char16_t* other) const;
inline bool operator>(const char16_t* other) const;
#if __cplusplus >= 202002L
inline std::strong_ordering operator<=>(const char16_t* other) const;
#endif
inline operator const char16_t*() const;
@ -334,6 +344,19 @@ inline bool String16::operator>(const String16& other) const
return strzcmp16(mString, size(), other.mString, other.size()) > 0;
}
#if __cplusplus >= 202002L
inline std::strong_ordering String16::operator<=>(const String16& other) const {
int result = strzcmp16(mString, size(), other.mString, other.size());
if (result == 0) {
return std::strong_ordering::equal;
} else if (result < 0) {
return std::strong_ordering::less;
} else {
return std::strong_ordering::greater;
}
}
#endif
inline bool String16::operator<(const char16_t* other) const
{
return strcmp16(mString, other) < 0;
@ -364,6 +387,19 @@ inline bool String16::operator>(const char16_t* other) const
return strcmp16(mString, other) > 0;
}
#if __cplusplus >= 202002L
inline std::strong_ordering String16::operator<=>(const char16_t* other) const {
int result = strcmp16(mString, other);
if (result == 0) {
return std::strong_ordering::equal;
} else if (result < 0) {
return std::strong_ordering::less;
} else {
return std::strong_ordering::greater;
}
}
#endif
inline String16::operator const char16_t*() const
{
return mString;

View file

@ -36,6 +36,10 @@
#define HAS_STRING_VIEW
#endif
#if __cplusplus >= 202002L
#include <compare>
#endif
// ---------------------------------------------------------------------------
namespace android {
@ -106,6 +110,9 @@ public:
inline bool operator!=(const String8& other) const;
inline bool operator>=(const String8& other) const;
inline bool operator>(const String8& other) const;
#if __cplusplus >= 202002L
inline std::strong_ordering operator<=>(const String8& other) const;
#endif
inline bool operator<(const char* other) const;
inline bool operator<=(const char* other) const;
@ -113,6 +120,9 @@ public:
inline bool operator!=(const char* other) const;
inline bool operator>=(const char* other) const;
inline bool operator>(const char* other) const;
#if __cplusplus >= 202002L
inline std::strong_ordering operator<=>(const char* other) const;
#endif
inline operator const char*() const;
@ -302,6 +312,19 @@ inline bool String8::operator>(const String8& other) const
return strcmp(mString, other.mString) > 0;
}
#if __cplusplus >= 202002L
inline std::strong_ordering String8::operator<=>(const String8& other) const {
int result = strcmp(mString, other.mString);
if (result == 0) {
return std::strong_ordering::equal;
} else if (result < 0) {
return std::strong_ordering::less;
} else {
return std::strong_ordering::greater;
}
}
#endif
inline bool String8::operator<(const char* other) const
{
return strcmp(mString, other) < 0;
@ -332,6 +355,19 @@ inline bool String8::operator>(const char* other) const
return strcmp(mString, other) > 0;
}
#if __cplusplus >= 202002L
inline std::strong_ordering String8::operator<=>(const char* other) const {
int result = strcmp(mString, other);
if (result == 0) {
return std::strong_ordering::equal;
} else if (result < 0) {
return std::strong_ordering::less;
} else {
return std::strong_ordering::greater;
}
}
#endif
inline String8::operator const char*() const
{
return mString;