| //===----------------------------------------------------------------------===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "lldb/Utility/RegisterTypeFlags.h" |
| #include "lldb/Utility/StreamString.h" |
| #include "gmock/gmock.h" |
| #include "gtest/gtest.h" |
| |
| #include "llvm/Support/Casting.h" |
| |
| using namespace lldb_private; |
| using namespace lldb; |
| |
| TEST(RegisterTypeTest, Field) { |
| // We assume that start <= end is always true, so that is not tested here. |
| |
| RegisterTypeFlags::Field f1("abc", 0); |
| ASSERT_EQ(f1.GetName(), "abc"); |
| // start == end means a 1 bit field. |
| ASSERT_EQ(f1.GetSizeInBits(), (unsigned)1); |
| ASSERT_EQ(f1.GetMask(), (uint64_t)1); |
| ASSERT_EQ(f1.GetValue(0), (uint64_t)0); |
| ASSERT_EQ(f1.GetValue(3), (uint64_t)1); |
| |
| // End is inclusive meaning that start 0 to end 1 includes bit 1 |
| // to make a 2 bit field. |
| RegisterTypeFlags::Field f2("", 0, 1); |
| ASSERT_EQ(f2.GetSizeInBits(), (unsigned)2); |
| ASSERT_EQ(f2.GetMask(), (uint64_t)3); |
| ASSERT_EQ(f2.GetValue(UINT64_MAX), (uint64_t)3); |
| ASSERT_EQ(f2.GetValue(UINT64_MAX & ~(uint64_t)3), (uint64_t)0); |
| |
| // If the field doesn't start at 0 we need to shift up/down |
| // to account for it. |
| RegisterTypeFlags::Field f3("", 2, 5); |
| ASSERT_EQ(f3.GetSizeInBits(), (unsigned)4); |
| ASSERT_EQ(f3.GetMask(), (uint64_t)0x3c); |
| ASSERT_EQ(f3.GetValue(UINT64_MAX), (uint64_t)0xf); |
| ASSERT_EQ(f3.GetValue(UINT64_MAX & ~(uint64_t)0x3c), (uint64_t)0); |
| |
| // Fields are sorted lowest starting bit first. |
| ASSERT_TRUE(f2 < f3); |
| ASSERT_FALSE(f3 < f1); |
| ASSERT_FALSE(f1 < f2); |
| ASSERT_FALSE(f1 < f1); |
| } |
| |
| static RegisterTypeFlags::Field make_field(unsigned start, unsigned end) { |
| return RegisterTypeFlags::Field("", start, end); |
| } |
| |
| static RegisterTypeFlags::Field make_field(unsigned bit) { |
| return RegisterTypeFlags::Field("", bit); |
| } |
| |
| TEST(RegisterTypeTest, FieldOverlaps) { |
| // Single bit fields |
| ASSERT_FALSE(make_field(0, 0).Overlaps(make_field(1))); |
| ASSERT_TRUE(make_field(1, 1).Overlaps(make_field(1))); |
| ASSERT_FALSE(make_field(1, 1).Overlaps(make_field(3))); |
| |
| ASSERT_TRUE(make_field(0, 1).Overlaps(make_field(1, 2))); |
| ASSERT_TRUE(make_field(1, 2).Overlaps(make_field(0, 1))); |
| ASSERT_FALSE(make_field(0, 1).Overlaps(make_field(2, 3))); |
| ASSERT_FALSE(make_field(2, 3).Overlaps(make_field(0, 1))); |
| |
| ASSERT_FALSE(make_field(1, 5).Overlaps(make_field(10, 20))); |
| ASSERT_FALSE(make_field(15, 30).Overlaps(make_field(7, 12))); |
| } |
| |
| TEST(RegisterTypeTest, PaddingDistance) { |
| // We assume that this method is always called with a more significant |
| // (start bit is higher) field first and that they do not overlap. |
| |
| // [field 1][field 2] |
| ASSERT_EQ(make_field(1, 1).PaddingDistance(make_field(0)), 0ULL); |
| // [field 1][..][field 2] |
| ASSERT_EQ(make_field(2, 2).PaddingDistance(make_field(0)), 1ULL); |
| // [field 1][field 1][field 2] |
| ASSERT_EQ(make_field(1, 2).PaddingDistance(make_field(0)), 0ULL); |
| // [field 1][30 bits free][field 2] |
| ASSERT_EQ(make_field(31, 31).PaddingDistance(make_field(0)), 30ULL); |
| } |
| |
| static void |
| test_padding(const std::vector<RegisterTypeFlags::Field> &fields, |
| const std::vector<RegisterTypeFlags::Field> &expected) { |
| RegisterTypeFlags rf("", 4, fields); |
| EXPECT_THAT(expected, ::testing::ContainerEq(rf.GetFields())); |
| } |
| |
| TEST(RegisterTypeFlagsTest, RegisterTypeFlagsPadding) { |
| // When creating a set of flags we assume that: |
| // * There are >= 1 fields. |
| // * They are sorted in descending order. |
| // * There may be gaps between each field. |
| |
| // Needs no padding |
| auto fields = std::vector<RegisterTypeFlags::Field>{make_field(16, 31), |
| make_field(0, 15)}; |
| test_padding(fields, fields); |
| |
| // Needs padding in between the fields, single bit. |
| test_padding({make_field(17, 31), make_field(0, 15)}, |
| {make_field(17, 31), make_field(16), make_field(0, 15)}); |
| // Multiple bits of padding. |
| test_padding({make_field(17, 31), make_field(0, 14)}, |
| {make_field(17, 31), make_field(15, 16), make_field(0, 14)}); |
| |
| // Padding before first field, single bit. |
| test_padding({make_field(0, 30)}, {make_field(31), make_field(0, 30)}); |
| // Multiple bits. |
| test_padding({make_field(0, 15)}, {make_field(16, 31), make_field(0, 15)}); |
| |
| // Padding after last field, single bit. |
| test_padding({make_field(1, 31)}, {make_field(1, 31), make_field(0)}); |
| // Multiple bits. |
| test_padding({make_field(2, 31)}, {make_field(2, 31), make_field(0, 1)}); |
| |
| // Fields need padding before, in between and after. |
| // [31-28][field 27-24][23-22][field 21-20][19-12][field 11-8][7-0] |
| test_padding({make_field(24, 27), make_field(20, 21), make_field(8, 11)}, |
| {make_field(28, 31), make_field(24, 27), make_field(22, 23), |
| make_field(20, 21), make_field(12, 19), make_field(8, 11), |
| make_field(0, 7)}); |
| } |
| |
| TEST(RegisterFieldsTest, ReverseFieldOrder) { |
| // Unchanged |
| RegisterTypeFlags rf("", 4, {make_field(0, 31)}); |
| ASSERT_EQ(0x12345678ULL, |
| (unsigned long long)rf.ReverseFieldOrder(0x12345678)); |
| |
| // Swap the two halves around. |
| RegisterTypeFlags rf2("", 4, {make_field(16, 31), make_field(0, 15)}); |
| ASSERT_EQ(0x56781234ULL, |
| (unsigned long long)rf2.ReverseFieldOrder(0x12345678)); |
| |
| // Many small fields. |
| RegisterTypeFlags rf3( |
| "", 4, {make_field(31), make_field(30), make_field(29), make_field(28)}); |
| ASSERT_EQ(0x00000005ULL, rf3.ReverseFieldOrder(0xA0000000)); |
| } |
| |
| TEST(RegisterTypeFlagsTest, AsTable) { |
| // Anonymous fields are shown with an empty name cell, |
| // whether they are known up front or added during construction. |
| RegisterTypeFlags anon_field("", 4, {make_field(0, 31)}); |
| ASSERT_EQ("| 31-0 |\n" |
| "|------|\n" |
| "| |", |
| anon_field.AsTable(100)); |
| |
| RegisterTypeFlags anon_with_pad("", 4, {make_field(16, 31)}); |
| ASSERT_EQ("| 31-16 | 15-0 |\n" |
| "|-------|------|\n" |
| "| | |", |
| anon_with_pad.AsTable(100)); |
| |
| // Use the wider of position and name to set the column width. |
| RegisterTypeFlags name_wider("", 4, |
| {RegisterTypeFlags::Field("aardvark", 0, 31)}); |
| ASSERT_EQ("| 31-0 |\n" |
| "|----------|\n" |
| "| aardvark |", |
| name_wider.AsTable(100)); |
| // When the padding is an odd number, put the remaining 1 on the right. |
| RegisterTypeFlags pos_wider("", 4, {RegisterTypeFlags::Field("?", 0, 31)}); |
| ASSERT_EQ("| 31-0 |\n" |
| "|------|\n" |
| "| ? |", |
| pos_wider.AsTable(100)); |
| |
| // Single bit fields don't need to show start and end, just one of them. |
| RegisterTypeFlags single_bit("", 4, {make_field(31)}); |
| ASSERT_EQ("| 31 | 30-0 |\n" |
| "|----|------|\n" |
| "| | |", |
| single_bit.AsTable(100)); |
| |
| // Columns are printed horizontally if max width allows. |
| RegisterTypeFlags many_fields("", 4, |
| {RegisterTypeFlags::Field("cat", 28, 31), |
| RegisterTypeFlags::Field("pigeon", 20, 23), |
| RegisterTypeFlags::Field("wolf", 12), |
| RegisterTypeFlags::Field("x", 0, 4)}); |
| ASSERT_EQ("| 31-28 | 27-24 | 23-20 | 19-13 | 12 | 11-5 | 4-0 |\n" |
| "|-------|-------|--------|-------|------|------|-----|\n" |
| "| cat | | pigeon | | wolf | | x |", |
| many_fields.AsTable(100)); |
| |
| // max_width tells us when we need to split into further tables. |
| // Here no split is needed. |
| RegisterTypeFlags exact_max_single_col( |
| "", 4, {RegisterTypeFlags::Field("?", 0, 31)}); |
| ASSERT_EQ("| 31-0 |\n" |
| "|------|\n" |
| "| ? |", |
| exact_max_single_col.AsTable(9)); |
| RegisterTypeFlags exact_max_two_col("", 4, |
| {RegisterTypeFlags::Field("?", 16, 31), |
| RegisterTypeFlags::Field("#", 0, 15)}); |
| ASSERT_EQ("| 31-16 | 15-0 |\n" |
| "|-------|------|\n" |
| "| ? | # |", |
| exact_max_two_col.AsTable(16)); |
| |
| // If max is less than a single column, just print the single column. The user |
| // will have to put up with some wrapping in this niche case. |
| RegisterTypeFlags zero_max_single_col("", 4, |
| {RegisterTypeFlags::Field("?", 0, 31)}); |
| ASSERT_EQ("| 31-0 |\n" |
| "|------|\n" |
| "| ? |", |
| zero_max_single_col.AsTable(0)); |
| // Same logic for any following columns. Effectively making a "vertical" |
| // table, just with more grid lines. |
| RegisterTypeFlags zero_max_two_col("", 4, |
| {RegisterTypeFlags::Field("?", 16, 31), |
| RegisterTypeFlags::Field("#", 0, 15)}); |
| ASSERT_EQ("| 31-16 |\n" |
| "|-------|\n" |
| "| ? |\n" |
| "\n" |
| "| 15-0 |\n" |
| "|------|\n" |
| "| # |", |
| zero_max_two_col.AsTable(0)); |
| |
| RegisterTypeFlags max_less_than_single_col( |
| "", 4, {RegisterTypeFlags::Field("?", 0, 31)}); |
| ASSERT_EQ("| 31-0 |\n" |
| "|------|\n" |
| "| ? |", |
| max_less_than_single_col.AsTable(3)); |
| RegisterTypeFlags max_less_than_two_col( |
| "", 4, |
| {RegisterTypeFlags::Field("?", 16, 31), |
| RegisterTypeFlags::Field("#", 0, 15)}); |
| ASSERT_EQ("| 31-16 |\n" |
| "|-------|\n" |
| "| ? |\n" |
| "\n" |
| "| 15-0 |\n" |
| "|------|\n" |
| "| # |", |
| max_less_than_two_col.AsTable(9)); |
| RegisterTypeFlags max_many_columns( |
| "", 4, |
| {RegisterTypeFlags::Field("A", 24, 31), |
| RegisterTypeFlags::Field("B", 16, 23), |
| RegisterTypeFlags::Field("C", 8, 15), |
| RegisterTypeFlags::Field("really long name", 0, 7)}); |
| ASSERT_EQ("| 31-24 | 23-16 |\n" |
| "|-------|-------|\n" |
| "| A | B |\n" |
| "\n" |
| "| 15-8 |\n" |
| "|------|\n" |
| "| C |\n" |
| "\n" |
| "| 7-0 |\n" |
| "|------------------|\n" |
| "| really long name |", |
| max_many_columns.AsTable(23)); |
| } |
| |
| TEST(RegisterTypeTest, DumpEnums) { |
| ASSERT_EQ(RegisterTypeFlags("", 8, {RegisterTypeFlags::Field{"A", 0}}) |
| .DumpEnums(80), |
| ""); |
| |
| RegisterTypeEnum basic_enum("test", {{0, "an_enumerator"}}); |
| ASSERT_EQ(RegisterTypeFlags( |
| "", 8, {RegisterTypeFlags::Field{"A", 0, 0, &basic_enum}}) |
| .DumpEnums(80), |
| "A: 0 = an_enumerator"); |
| |
| // If width is smaller than the enumerator name, print it anyway. |
| ASSERT_EQ(RegisterTypeFlags( |
| "", 8, {RegisterTypeFlags::Field{"A", 0, 0, &basic_enum}}) |
| .DumpEnums(5), |
| "A: 0 = an_enumerator"); |
| |
| // Mutliple values can go on the same line, up to the width. |
| RegisterTypeEnum more_enum( |
| "long_enum", {{0, "an_enumerator"}, |
| {1, "another_enumerator"}, |
| {2, "a_very_very_long_enumerator_has_its_own_line"}, |
| {3, "small"}, |
| {4, "small2"}}); |
| ASSERT_EQ(RegisterTypeFlags("", 8, |
| {RegisterTypeFlags::Field{"A", 0, 2, &more_enum}}) |
| // Width is chosen to be exactly enough to allow 0 and 1 |
| // enumerators on the first line. |
| .DumpEnums(45), |
| "A: 0 = an_enumerator, 1 = another_enumerator,\n" |
| " 2 = a_very_very_long_enumerator_has_its_own_line,\n" |
| " 3 = small, 4 = small2"); |
| |
| // If they all exceed width, one per line. |
| RegisterTypeEnum another_enum("another_enum", {{0, "an_enumerator"}, |
| {1, "another_enumerator"}, |
| {2, "a_longer_enumerator"}}); |
| ASSERT_EQ(RegisterTypeFlags( |
| "", 8, {RegisterTypeFlags::Field{"A", 0, 1, &another_enum}}) |
| .DumpEnums(5), |
| "A: 0 = an_enumerator,\n" |
| " 1 = another_enumerator,\n" |
| " 2 = a_longer_enumerator"); |
| |
| // If the name is already > the width, put one value per line. |
| RegisterTypeEnum short_enum("short_enum", {{0, "a"}, {1, "b"}, {2, "c"}}); |
| ASSERT_EQ(RegisterTypeFlags("", 8, |
| {RegisterTypeFlags::Field{"AReallyLongFieldName", |
| 0, 1, &short_enum}}) |
| .DumpEnums(10), |
| "AReallyLongFieldName: 0 = a,\n" |
| " 1 = b,\n" |
| " 2 = c"); |
| |
| // Fields are separated by a blank line. Indentation of lines split by width |
| // is set by the size of the fields name (as opposed to some max of all field |
| // names). |
| RegisterTypeEnum enum_1("enum_1", |
| {{0, "an_enumerator"}, {1, "another_enumerator"}}); |
| RegisterTypeEnum enum_2("enum_2", |
| {{0, "Cdef_enumerator_1"}, {1, "Cdef_enumerator_2"}}); |
| ASSERT_EQ(RegisterTypeFlags("", 8, |
| {RegisterTypeFlags::Field{"Ab", 1, 1, &enum_1}, |
| RegisterTypeFlags::Field{"Cdef", 0, 0, &enum_2}}) |
| .DumpEnums(10), |
| "Ab: 0 = an_enumerator,\n" |
| " 1 = another_enumerator\n" |
| "\n" |
| "Cdef: 0 = Cdef_enumerator_1,\n" |
| " 1 = Cdef_enumerator_2"); |
| |
| // Having fields without enumerators shouldn't produce any extra newlines. |
| ASSERT_EQ(RegisterTypeFlags("", 8, |
| { |
| RegisterTypeFlags::Field{"A", 4, 4}, |
| RegisterTypeFlags::Field{"B", 3, 3, &enum_1}, |
| RegisterTypeFlags::Field{"C", 2, 2}, |
| RegisterTypeFlags::Field{"D", 1, 1, &enum_1}, |
| RegisterTypeFlags::Field{"E", 0, 0}, |
| }) |
| .DumpEnums(80), |
| "B, D: 0 = an_enumerator, 1 = another_enumerator"); |
| |
| // Fields using the same enum should be grouped together. |
| RegisterTypeEnum repeated_enum("repeated_enum", |
| {{0, "zero"}, {1, "one"}, {2, "two"}}); |
| ASSERT_EQ( |
| RegisterTypeFlags("", 8, |
| { |
| RegisterTypeFlags::Field{"A", 6, 7, &repeated_enum}, |
| RegisterTypeFlags::Field{"B", 4, 5, &repeated_enum}, |
| RegisterTypeFlags::Field{"C", 2, 3, &enum_2}, |
| RegisterTypeFlags::Field{"D", 0, 1, &repeated_enum}, |
| }) |
| .DumpEnums(80), |
| "A, B, D: 0 = zero, 1 = one, 2 = two\n" |
| "\n" |
| "C: 0 = Cdef_enumerator_1, 1 = Cdef_enumerator_2"); |
| } |
| |
| TEST(RegisterFieldsTest, FlagsToXMLElementElement) { |
| StreamString strm; |
| |
| // RegisterTypeFlags requires that some fields be given, so no testing of |
| // empty input. |
| |
| // Unnamed fields are padding that are ignored. This applies to fields passed |
| // in, and those generated to fill the other bits (31-1 here). |
| RegisterTypeFlags("Foo", 4, {RegisterTypeFlags::Field("", 0, 0)}) |
| .ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), "<flags id=\"Foo\" size=\"4\">\n" |
| "</flags>\n"); |
| |
| strm.Clear(); |
| RegisterTypeFlags("Foo", 4, {RegisterTypeFlags::Field("abc", 0, 0)}) |
| .ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), "<flags id=\"Foo\" size=\"4\">\n" |
| " <field name=\"abc\" start=\"0\" end=\"0\"/>\n" |
| "</flags>\n"); |
| |
| strm.Clear(); |
| // Should use the current indentation level as a starting point. |
| strm.IndentMore(); |
| RegisterTypeFlags("Bar", 5, |
| {RegisterTypeFlags::Field("f1", 25, 32), |
| RegisterTypeFlags::Field("f2", 10, 24)}) |
| .ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), |
| " <flags id=\"Bar\" size=\"5\">\n" |
| " <field name=\"f1\" start=\"25\" end=\"32\"/>\n" |
| " <field name=\"f2\" start=\"10\" end=\"24\"/>\n" |
| " </flags>\n"); |
| |
| strm.Clear(); |
| strm.IndentLess(); |
| // Should replace any XML unsafe characters in field names. |
| RegisterTypeFlags( |
| "Safe", 8, |
| {RegisterTypeFlags::Field("A<", 4), RegisterTypeFlags::Field("B>", 3), |
| RegisterTypeFlags::Field("C'", 2), RegisterTypeFlags::Field("D\"", 1), |
| RegisterTypeFlags::Field("E&", 0)}) |
| .ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), |
| "<flags id=\"Safe\" size=\"8\">\n" |
| " <field name=\"A<\" start=\"4\" end=\"4\"/>\n" |
| " <field name=\"B>\" start=\"3\" end=\"3\"/>\n" |
| " <field name=\"C'\" start=\"2\" end=\"2\"/>\n" |
| " <field name=\"D"\" start=\"1\" end=\"1\"/>\n" |
| " <field name=\"E&\" start=\"0\" end=\"0\"/>\n" |
| "</flags>\n"); |
| |
| // Should include enumerators as the "type". |
| strm.Clear(); |
| RegisterTypeEnum enum_single("enum_single", {{0, "a"}}); |
| RegisterTypeFlags( |
| "Enumerators", 8, |
| {RegisterTypeFlags::Field("NoEnumerators", 4), |
| RegisterTypeFlags::Field("OneEnumerator", 3, 3, &enum_single)}) |
| .ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), |
| "<flags id=\"Enumerators\" size=\"8\">\n" |
| " <field name=\"NoEnumerators\" start=\"4\" end=\"4\"/>\n" |
| " <field name=\"OneEnumerator\" start=\"3\" end=\"3\" " |
| "type=\"enum_single\"/>\n" |
| "</flags>\n"); |
| } |
| |
| TEST(RegisterTypeTest, EnumeratorToXMLElement) { |
| StreamString strm; |
| |
| RegisterTypeEnum::Enumerator(1234, "test").ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), "<evalue name=\"test\" value=\"1234\"/>"); |
| |
| // Special XML chars in names must be escaped. |
| std::array special_names = { |
| std::make_pair(RegisterTypeEnum::Enumerator(0, "A<"), |
| "<evalue name=\"A<\" value=\"0\"/>"), |
| std::make_pair(RegisterTypeEnum::Enumerator(1, "B>"), |
| "<evalue name=\"B>\" value=\"1\"/>"), |
| std::make_pair(RegisterTypeEnum::Enumerator(2, "C'"), |
| "<evalue name=\"C'\" value=\"2\"/>"), |
| std::make_pair(RegisterTypeEnum::Enumerator(3, "D\""), |
| "<evalue name=\"D"\" value=\"3\"/>"), |
| std::make_pair(RegisterTypeEnum::Enumerator(4, "E&"), |
| "<evalue name=\"E&\" value=\"4\"/>"), |
| }; |
| |
| for (const auto &[enumerator, expected] : special_names) { |
| strm.Clear(); |
| enumerator.ToXMLElement(strm); |
| ASSERT_EQ(strm.GetString(), expected); |
| } |
| } |
| |
| TEST(RegisterTypeTest, EnumToXMLElement) { |
| StreamString strm; |
| |
| RegisterTypeFlags user_4("Foo", 4, {RegisterTypeFlags::Field("", 0, 0)}); |
| RegisterTypeEnum("empty_enum", {}) |
| .ToXMLElement(strm, llvm::dyn_cast<const RegisterType>(&user_4)); |
| ASSERT_EQ(strm.GetString(), "<enum id=\"empty_enum\" size=\"4\"/>\n"); |
| |
| strm.Clear(); |
| RegisterTypeFlags user_5("Foo", 5, {RegisterTypeFlags::Field("", 0, 0)}); |
| RegisterTypeEnum("single_enumerator", |
| {RegisterTypeEnum::Enumerator(0, "zero")}) |
| .ToXMLElement(strm, llvm::dyn_cast<const RegisterType>(&user_5)); |
| ASSERT_EQ(strm.GetString(), "<enum id=\"single_enumerator\" size=\"5\">\n" |
| " <evalue name=\"zero\" value=\"0\"/>\n" |
| "</enum>\n"); |
| |
| // Currently we don't emit size if the user of this type is not a flags. |
| // We don't expect to see this situation in real use. |
| strm.Clear(); |
| RegisterTypeEnum("multiple_enumerator", |
| {RegisterTypeEnum::Enumerator(0, "zero"), |
| RegisterTypeEnum::Enumerator(1, "one")}) |
| .ToXMLElement(strm, nullptr); |
| ASSERT_EQ(strm.GetString(), "<enum id=\"multiple_enumerator\">\n" |
| " <evalue name=\"zero\" value=\"0\"/>\n" |
| " <evalue name=\"one\" value=\"1\"/>\n" |
| "</enum>\n"); |
| } |
| |
| TEST(RegisterTypeTest, RegisterTypeFlagsToXML) { |
| // This method should output all the enums used by the register flag set, |
| // then the flags set itself. There should only be one definition of each |
| // enum, even if it is used by multiple fields. |
| |
| StreamString strm; |
| auto enum_a = std::make_shared<RegisterTypeEnum>( |
| "enum_a", |
| RegisterTypeEnum::Enumerators{RegisterTypeEnum::Enumerator(0, "zero")}); |
| auto enum_b = std::make_shared<RegisterTypeEnum>( |
| "enum_b", |
| RegisterTypeEnum::Enumerators{RegisterTypeEnum::Enumerator(1, "one")}); |
| auto enum_c = std::make_shared<RegisterTypeEnum>( |
| "enum_c", |
| RegisterTypeEnum::Enumerators{RegisterTypeEnum::Enumerator(2, "two")}); |
| |
| std::unordered_set<std::string> previously_emitted; |
| // Pretend that enum_c was already emitted for a different flag set. |
| previously_emitted.insert(enum_c->GetID()); |
| |
| std::vector<RegisterTypeFlags::Field> fields{ |
| RegisterTypeFlags::Field("f1", 31, 31, enum_a.get()), |
| RegisterTypeFlags::Field("f2", 30, 30, enum_a.get()), |
| RegisterTypeFlags::Field("f3", 29, 29, enum_b.get()), |
| RegisterTypeFlags::Field("f4", 27, 28, enum_c.get()), |
| }; |
| |
| auto TestFlags = std::make_shared<RegisterTypeFlags>("Test", 4, fields); |
| TestFlags->ToXML(strm, previously_emitted); |
| ASSERT_EQ(strm.GetString(), |
| "<enum id=\"enum_a\" size=\"4\">\n" |
| " <evalue name=\"zero\" value=\"0\"/>\n" |
| "</enum>\n" |
| "<enum id=\"enum_b\" size=\"4\">\n" |
| " <evalue name=\"one\" value=\"1\"/>\n" |
| "</enum>\n" |
| "<flags id=\"Test\" size=\"4\">\n" |
| " <field name=\"f1\" start=\"31\" end=\"31\" type=\"enum_a\"/>\n" |
| " <field name=\"f2\" start=\"30\" end=\"30\" type=\"enum_a\"/>\n" |
| " <field name=\"f3\" start=\"29\" end=\"29\" type=\"enum_b\"/>\n" |
| " <field name=\"f4\" start=\"27\" end=\"28\" type=\"enum_c\"/>\n" |
| "</flags>\n"); |
| |
| // If another flag set were to use the same enums we should not output them |
| // again. Only output new things. |
| auto enum_d = std::make_shared<RegisterTypeEnum>( |
| "enum_d", |
| RegisterTypeEnum::Enumerators{RegisterTypeEnum::Enumerator(3, "three")}); |
| fields.push_back(RegisterTypeFlags::Field("f5", 25, 26, enum_d.get())); |
| auto TestFlags2 = std::make_shared<RegisterTypeFlags>("Test2", 4, fields); |
| |
| strm.Clear(); |
| TestFlags2->ToXML(strm, previously_emitted); |
| ASSERT_EQ(strm.GetString(), |
| "<enum id=\"enum_d\" size=\"4\">\n" |
| " <evalue name=\"three\" value=\"3\"/>\n" |
| "</enum>\n" |
| "<flags id=\"Test2\" size=\"4\">\n" |
| " <field name=\"f1\" start=\"31\" end=\"31\" type=\"enum_a\"/>\n" |
| " <field name=\"f2\" start=\"30\" end=\"30\" type=\"enum_a\"/>\n" |
| " <field name=\"f3\" start=\"29\" end=\"29\" type=\"enum_b\"/>\n" |
| " <field name=\"f4\" start=\"27\" end=\"28\" type=\"enum_c\"/>\n" |
| " <field name=\"f5\" start=\"25\" end=\"26\" type=\"enum_d\"/>\n" |
| "</flags>\n"); |
| |
| // If we have already emitted this set of flags, don't emit it again. |
| strm.Clear(); |
| TestFlags2->ToXML(strm, previously_emitted); |
| ASSERT_EQ(strm.GetString(), ""); |
| } |
| |
| TEST(RegisterTypeTest, XMLAttributeEscaping) { |
| RegisterTypeEnum enum_type("enum<>&\"'", |
| {RegisterTypeEnum::Enumerator(0, "zero")}); |
| RegisterTypeFlags flags_type( |
| "flags<>&\"'", 1, {RegisterTypeFlags::Field("field", 0, 0, &enum_type)}); |
| |
| StreamString strm; |
| std::unordered_set<std::string> previously_emitted; |
| flags_type.ToXML(strm, previously_emitted); |
| EXPECT_EQ(strm.GetString(), |
| "<enum id=\"enum<>&"'\" size=\"1\">\n" |
| " <evalue name=\"zero\" value=\"0\"/>\n" |
| "</enum>\n" |
| "<flags id=\"flags<>&"'\" size=\"1\">\n" |
| " <field name=\"field\" start=\"0\" end=\"0\" " |
| "type=\"enum<>&"'\"/>\n" |
| "</flags>\n"); |
| } |
| |
| TEST(RegisterTypeTest, XMLDefinitionsAreDeduplicatedByID) { |
| RegisterTypeEnum first("same_id", {RegisterTypeEnum::Enumerator(0, "first")}); |
| RegisterTypeEnum second("same_id", |
| {RegisterTypeEnum::Enumerator(1, "second")}); |
| |
| StreamString strm; |
| std::unordered_set<std::string> previously_emitted; |
| first.ToXML(strm, previously_emitted); |
| second.ToXML(strm, previously_emitted); |
| EXPECT_EQ(strm.GetString(), "<enum id=\"same_id\">\n" |
| " <evalue name=\"first\" value=\"0\"/>\n" |
| "</enum>\n"); |
| } |
