Joshua Haberman
3 years ago
parent
732a21bb59
commit
9bff7b82d1
5 changed files with 402 additions and 28 deletions
@ -0,0 +1,255 @@ |
||||
/*
|
||||
* Copyright (c) 2009-2021, Google LLC |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are met: |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above copyright |
||||
* notice, this list of conditions and the following disclaimer in the |
||||
* documentation and/or other materials provided with the distribution. |
||||
* * Neither the name of Google LLC nor the |
||||
* names of its contributors may be used to endorse or promote products |
||||
* derived from this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" |
||||
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||
* ARE DISCLAIMED. IN NO EVENT SHALL Google LLC BE LIABLE FOR ANY DIRECT, |
||||
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES |
||||
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; |
||||
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND |
||||
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
||||
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
*/ |
||||
|
||||
#include "upb/util/compare.h" |
||||
|
||||
#include "absl/strings/string_view.h" |
||||
#include "gmock/gmock.h" |
||||
#include "gtest/gtest.h" |
||||
|
||||
#include <stdint.h> |
||||
#include <vector> |
||||
#include <string_view> |
||||
|
||||
struct UnknownField; |
||||
|
||||
using UnknownFields = std::vector<UnknownField>; |
||||
|
||||
enum class UnknownFieldType { |
||||
kVarint, |
||||
kLongVarint, // Over-encoded to have distinct wire format.
|
||||
kDelimited, |
||||
kFixed64, |
||||
kFixed32, |
||||
kGroup, |
||||
}; |
||||
|
||||
union UnknownFieldValue { |
||||
uint64_t varint; |
||||
uint64_t fixed64; |
||||
uint32_t fixed32; |
||||
// NULL-terminated (strings must not have embedded NULL).
|
||||
const char* delimited; |
||||
UnknownFields* group; |
||||
}; |
||||
|
||||
struct TypeAndValue { |
||||
UnknownFieldType type; |
||||
UnknownFieldValue value; |
||||
}; |
||||
|
||||
struct UnknownField { |
||||
uint32_t field_number; |
||||
TypeAndValue value; |
||||
}; |
||||
|
||||
|
||||
TypeAndValue Varint(uint64_t val) { |
||||
TypeAndValue ret{UnknownFieldType::kVarint}; |
||||
ret.value.varint = val; |
||||
return ret; |
||||
} |
||||
|
||||
TypeAndValue LongVarint(uint64_t val) { |
||||
TypeAndValue ret{UnknownFieldType::kLongVarint}; |
||||
ret.value.varint = val; |
||||
return ret; |
||||
} |
||||
|
||||
TypeAndValue Fixed64(uint64_t val) { |
||||
TypeAndValue ret{UnknownFieldType::kFixed64}; |
||||
ret.value.fixed64 = val; |
||||
return ret; |
||||
} |
||||
|
||||
TypeAndValue Fixed32(uint32_t val) { |
||||
TypeAndValue ret{UnknownFieldType::kFixed32}; |
||||
ret.value.fixed32 = val; |
||||
return ret; |
||||
} |
||||
|
||||
TypeAndValue Delimited(const char* val) { |
||||
TypeAndValue ret{UnknownFieldType::kDelimited}; |
||||
ret.value.delimited = val; |
||||
return ret; |
||||
} |
||||
|
||||
TypeAndValue Group(UnknownFields nested) { |
||||
TypeAndValue ret{UnknownFieldType::kGroup}; |
||||
ret.value.group = &nested; |
||||
return ret; |
||||
} |
||||
|
||||
void f() { |
||||
auto fields = UnknownFields{ |
||||
{1, Varint(1)}, |
||||
{2, Fixed64(2)}, |
||||
}; |
||||
|
||||
auto fields2 = UnknownFields{ |
||||
{1, Group({ |
||||
{2, Group({ |
||||
{3, Varint(1)}, |
||||
})}, |
||||
})}, |
||||
}; |
||||
} |
||||
|
||||
void EncodeVarint(uint64_t val, std::string* str) { |
||||
do { |
||||
char byte = val & 0x7fU; |
||||
val >>= 7; |
||||
if (val) byte |= 0x80U; |
||||
str->push_back(byte); |
||||
} while (val); |
||||
} |
||||
|
||||
std::string ToBinaryPayload(const UnknownFields& fields) { |
||||
static const upb_wiretype_t wire_types[] = { |
||||
UPB_WIRE_TYPE_VARINT, |
||||
UPB_WIRE_TYPE_VARINT, |
||||
UPB_WIRE_TYPE_DELIMITED, |
||||
UPB_WIRE_TYPE_64BIT, |
||||
UPB_WIRE_TYPE_32BIT, |
||||
UPB_WIRE_TYPE_START_GROUP, |
||||
}; |
||||
std::string ret; |
||||
|
||||
for (const auto& field : fields) { |
||||
uint32_t tag = field.field_number << 3 | |
||||
(wire_types[static_cast<int>(field.value.type)]); |
||||
EncodeVarint(tag, &ret); |
||||
switch (field.value.type) { |
||||
case UnknownFieldType::kVarint: |
||||
EncodeVarint(field.value.value.varint, &ret); |
||||
break; |
||||
case UnknownFieldType::kLongVarint: |
||||
EncodeVarint(field.value.value.varint, &ret); |
||||
ret.back() |= 0x80; |
||||
ret.push_back(0); |
||||
break; |
||||
case UnknownFieldType::kDelimited: |
||||
EncodeVarint(strlen(field.value.value.delimited), &ret); |
||||
ret.append(field.value.value.delimited); |
||||
break; |
||||
case UnknownFieldType::kFixed64: { |
||||
uint64_t val = _upb_be_swap64(field.value.value.fixed64); |
||||
ret.append(reinterpret_cast<const char*>(&val), sizeof(val)); |
||||
break; |
||||
} |
||||
case UnknownFieldType::kFixed32: { |
||||
uint32_t val = _upb_be_swap32(field.value.value.fixed32); |
||||
ret.append(reinterpret_cast<const char*>(&val), sizeof(val)); |
||||
break; |
||||
} |
||||
case UnknownFieldType::kGroup: { |
||||
uint32_t end_tag = field.field_number << 3 | UPB_WIRE_TYPE_END_GROUP; |
||||
ret.append(ToBinaryPayload(*field.value.value.group)); |
||||
EncodeVarint(end_tag, &ret); |
||||
break; |
||||
} |
||||
} |
||||
} |
||||
|
||||
return ret; |
||||
} |
||||
|
||||
upb_UnknownCompareResult CompareUnknownWithMaxDepth(UnknownFields uf1, |
||||
UnknownFields uf2, |
||||
int max_depth) { |
||||
std::string buf1 = ToBinaryPayload(uf1); |
||||
std::string buf2 = ToBinaryPayload(uf2); |
||||
return upb_Message_UnknownFieldsAreEqual(buf1.data(), buf1.size(), |
||||
buf2.data(), buf2.size(), max_depth); |
||||
} |
||||
|
||||
upb_UnknownCompareResult CompareUnknown(UnknownFields uf1, UnknownFields uf2) { |
||||
return CompareUnknownWithMaxDepth(uf1, uf2, 64); |
||||
} |
||||
|
||||
TEST(CompareTest, UnknownFieldsReflexive) { |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_Equal, CompareUnknown({}, {})); |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown({{1, Varint(123)}, {2, Fixed32(456)}}, |
||||
{{1, Varint(123)}, {2, Fixed32(456)}})); |
||||
EXPECT_EQ( |
||||
kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown( |
||||
{{1, Group({{2, Group({{3, Fixed32(456)}, {4, Fixed64(123)}})}})}}, |
||||
{{1, Group({{2, Group({{3, Fixed32(456)}, {4, Fixed64(123)}})}})}})); |
||||
} |
||||
|
||||
TEST(CompareTest, UnknownFieldsOrdering) { |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown({{1, Varint(111)}, |
||||
{2, Delimited("ABC")}, |
||||
{3, Fixed32(456)}, |
||||
{4, Fixed64(123)}, |
||||
{5, Group({})}}, |
||||
{{5, Group({})}, |
||||
{4, Fixed64(123)}, |
||||
{3, Fixed32(456)}, |
||||
{2, Delimited("ABC")}, |
||||
{1, Varint(111)}})); |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_NotEqual, |
||||
CompareUnknown({{1, Varint(111)}, |
||||
{2, Delimited("ABC")}, |
||||
{3, Fixed32(456)}, |
||||
{4, Fixed64(123)}, |
||||
{5, Group({})}}, |
||||
{{5, Group({})}, |
||||
{4, Fixed64(123)}, |
||||
{3, Fixed32(455)}, // Small difference.
|
||||
{2, Delimited("ABC")}, |
||||
{1, Varint(111)}})); |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown({{3, Fixed32(456)}, {4, Fixed64(123)}}, |
||||
{{4, Fixed64(123)}, {3, Fixed32(456)}})); |
||||
EXPECT_EQ( |
||||
kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown( |
||||
{{1, Group({{2, Group({{3, Fixed32(456)}, {4, Fixed64(123)}})}})}}, |
||||
{{1, Group({{2, Group({{4, Fixed64(123)}, {3, Fixed32(456)}})}})}})); |
||||
} |
||||
|
||||
TEST(CompareTest, LongVarint) { |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown({{1, LongVarint(123)}, {2, LongVarint(456)}}, |
||||
{{1, Varint(123)}, {2, Varint(456)}})); |
||||
EXPECT_EQ(kUpb_UnknownCompareResult_Equal, |
||||
CompareUnknown({{2, LongVarint(456)}, {1, LongVarint(123)}}, |
||||
{{1, Varint(123)}, {2, Varint(456)}})); |
||||
} |
||||
|
||||
TEST(CompareTest, MaxDepth) { |
||||
EXPECT_EQ( |
||||
kUpb_UnknownCompareResult_MaxDepthExceeded, |
||||
CompareUnknownWithMaxDepth( |
||||
{{1, Group({{2, Group({{3, Fixed32(456)}, {4, Fixed64(123)}})}})}}, |
||||
{{1, Group({{2, Group({{4, Fixed64(123)}, {3, Fixed32(456)}})}})}}, |
||||
2)); |
||||
} |
||||
Loading…
Reference in new issue