| Index: unittests/Bitcode/NaClMungedBitcodeTest.cpp
|
| diff --git a/unittests/Bitcode/NaClMungedBitcodeTest.cpp b/unittests/Bitcode/NaClMungedBitcodeTest.cpp
|
| index f20e01a97489a69215105d41ea0bcad05ccf40ff..f7b8f46e7269bd0739c3bf4a6692a3f9a1b923c4 100644
|
| --- a/unittests/Bitcode/NaClMungedBitcodeTest.cpp
|
| +++ b/unittests/Bitcode/NaClMungedBitcodeTest.cpp
|
| @@ -10,30 +10,13 @@
|
|
|
| // Tests munging NaCl bitcode records.
|
|
|
| -#include "llvm/ADT/STLExtras.h"
|
| -#include "llvm/Bitcode/NaCl/NaClBitcodeMungeUtils.h"
|
| +#include "NaClMungeTest.h"
|
|
|
| -#include "gtest/gtest.h"
|
| #include <limits>
|
|
|
| using namespace llvm;
|
|
|
| -namespace {
|
| -
|
| -static const uint64_t Terminator = 0x5768798008978675LL;
|
| -
|
| -std::string stringify(NaClMungedBitcode &MungedBitcode) {
|
| - std::string Buffer;
|
| - raw_string_ostream StrBuf(Buffer);
|
| - MungedBitcode.print(StrBuf);
|
| - return StrBuf.str();
|
| -}
|
| -
|
| -} // end of anonymous namespace
|
| -
|
| -#define ARRAY_ARGS(name) name, array_lengthof(name), Terminator
|
| -
|
| -namespace NaClMungedBc {
|
| +namespace naclmungetest {
|
|
|
| TEST(NaClMungedBitcodeTest, TestInsertBefore) {
|
| const uint64_t Records[] = {
|
| @@ -42,7 +25,7 @@ TEST(NaClMungedBitcodeTest, TestInsertBefore) {
|
| 6, 7, 8 , 9, Terminator,
|
| 10, 11, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -54,7 +37,7 @@ TEST(NaClMungedBitcodeTest, TestInsertBefore) {
|
| const uint64_t BeforeSecond[] = {
|
| 1, NaClMungedBitcode::AddBefore, 12, 13, 14, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(BeforeSecond));
|
| + MungedRecords.munge(ARRAY_TERM(BeforeSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 12: [13, 14]\n"
|
| @@ -68,7 +51,7 @@ TEST(NaClMungedBitcodeTest, TestInsertBefore) {
|
| 1, NaClMungedBitcode::AddBefore, 15, 16, 17, Terminator,
|
| 1, NaClMungedBitcode::AddBefore, 18, 19, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(BeforeSecondMore));
|
| + MungedRecords.munge(ARRAY_TERM(BeforeSecondMore));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 12: [13, 14]\n"
|
| @@ -84,7 +67,7 @@ TEST(NaClMungedBitcodeTest, TestInsertBefore) {
|
| 3, NaClMungedBitcode::AddBefore, 21, 22, 23, Terminator,
|
| 3, NaClMungedBitcode::AddBefore, 24, 25, 26, 27, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(BeforeLast));
|
| + MungedRecords.munge(ARRAY_TERM(BeforeLast));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 12: [13, 14]\n"
|
| @@ -105,7 +88,7 @@ TEST(NaClMungedBitcodeTest, TestInsertAfter) {
|
| 6, 7, 8 , 9, Terminator,
|
| 10, 11, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -117,7 +100,7 @@ TEST(NaClMungedBitcodeTest, TestInsertAfter) {
|
| const uint64_t AfterSecond[] = {
|
| 1, NaClMungedBitcode::AddAfter, 12, 13, 14, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AfterSecond));
|
| + MungedRecords.munge(ARRAY_TERM(AfterSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -131,7 +114,7 @@ TEST(NaClMungedBitcodeTest, TestInsertAfter) {
|
| 1, NaClMungedBitcode::AddAfter, 15, 16, 17, Terminator,
|
| 1, NaClMungedBitcode::AddAfter, 18, 19, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AfterSecondMore));
|
| + MungedRecords.munge(ARRAY_TERM(AfterSecondMore));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -147,7 +130,7 @@ TEST(NaClMungedBitcodeTest, TestInsertAfter) {
|
| 3, NaClMungedBitcode::AddAfter, 21, 22, 23, Terminator,
|
| 3, NaClMungedBitcode::AddAfter, 24, 25, 26, 27, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AfterLast));
|
| + MungedRecords.munge(ARRAY_TERM(AfterLast));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -168,7 +151,7 @@ TEST(NaClMungedBitcodeTest, TestRemove) {
|
| 6, 7, 8 , 9, Terminator,
|
| 10, 11, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -180,7 +163,7 @@ TEST(NaClMungedBitcodeTest, TestRemove) {
|
| const uint64_t RemoveSecond[] = {
|
| 1, NaClMungedBitcode::Remove
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveSecond));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 6: [7, 8, 9]\n"
|
| @@ -192,7 +175,7 @@ TEST(NaClMungedBitcodeTest, TestRemove) {
|
| 0, NaClMungedBitcode::Remove,
|
| 3, NaClMungedBitcode::Remove
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveEnds));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveEnds));
|
| EXPECT_EQ(
|
| " 6: [7, 8, 9]\n",
|
| stringify(MungedRecords));
|
| @@ -201,7 +184,7 @@ TEST(NaClMungedBitcodeTest, TestRemove) {
|
| const uint64_t RemoveOther[] = {
|
| 2, NaClMungedBitcode::Remove
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveOther));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveOther));
|
| EXPECT_EQ(
|
| "",
|
| stringify(MungedRecords));
|
| @@ -214,7 +197,7 @@ TEST(NaClMungedBitcodeTest, TestReplace) {
|
| 6, 7, 8 , 9, Terminator,
|
| 10, 11, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -226,7 +209,7 @@ TEST(NaClMungedBitcodeTest, TestReplace) {
|
| const uint64_t ReplaceSecond[] = {
|
| 1, NaClMungedBitcode::Replace, 12, 13, 14, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ReplaceSecond));
|
| + MungedRecords.munge(ARRAY_TERM(ReplaceSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 12: [13, 14]\n"
|
| @@ -239,7 +222,7 @@ TEST(NaClMungedBitcodeTest, TestReplace) {
|
| 0, NaClMungedBitcode::Replace, 15, 16, 17, 18, Terminator,
|
| 3, NaClMungedBitcode::Replace, 19, 20, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ReplaceEnds));
|
| + MungedRecords.munge(ARRAY_TERM(ReplaceEnds));
|
| EXPECT_EQ(
|
| " 15: [16, 17, 18]\n"
|
| " 12: [13, 14]\n"
|
| @@ -254,7 +237,7 @@ TEST(NaClMungedBitcodeTest, TestReplace) {
|
| 1, NaClMungedBitcode::Replace, 24, 25, Terminator,
|
| 2, NaClMungedBitcode::Replace, 26, 27, 28, 29, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ReplaceFirst3));
|
| + MungedRecords.munge(ARRAY_TERM(ReplaceFirst3));
|
| EXPECT_EQ(
|
| " 21: [22, 23]\n"
|
| " 24: [25]\n"
|
| @@ -267,7 +250,7 @@ TEST(NaClMungedBitcodeTest, TestReplace) {
|
| 1, NaClMungedBitcode::Remove,
|
| 3, NaClMungedBitcode::Remove
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveReplaced));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveReplaced));
|
| EXPECT_EQ(
|
| " 21: [22, 23]\n"
|
| " 26: [27, 28, 29]\n",
|
| @@ -287,7 +270,7 @@ TEST(NaClMungedBitcodeTest, TestBlockStructure) {
|
| 19, 20, 21, Terminator,
|
| 22, naclbitc::BLK_CODE_EXIT, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3, 4]\n"
|
| " 5: [65535, 6]\n"
|
| @@ -307,7 +290,7 @@ TEST(NaClMungedBitcodeTest, TestBlockStructure) {
|
| 4, NaClMungedBitcode::AddAfter, 0, naclbitc::BLK_CODE_EXIT, Terminator,
|
| 2, NaClMungedBitcode::Replace, 0, naclbitc::BLK_CODE_EXIT, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ExitEdits));
|
| + MungedRecords.munge(ARRAY_TERM(ExitEdits));
|
| EXPECT_EQ(
|
| " 1: [2, 3, 4]\n"
|
| " 5: [65535, 6]\n"
|
| @@ -333,7 +316,7 @@ TEST(NaClMungedBitcodeTest, TestReplaceRemoveEffects) {
|
| 6, 7, 8 , 9, Terminator,
|
| 10, 11, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -345,7 +328,7 @@ TEST(NaClMungedBitcodeTest, TestReplaceRemoveEffects) {
|
| const uint64_t RemoveSecond[] = {
|
| 1, NaClMungedBitcode::Remove
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveSecond));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 6: [7, 8, 9]\n"
|
| @@ -353,7 +336,7 @@ TEST(NaClMungedBitcodeTest, TestReplaceRemoveEffects) {
|
| stringify(MungedRecords));
|
|
|
| // Try it again. Should have no effect.
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveSecond));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 6: [7, 8, 9]\n"
|
| @@ -364,7 +347,7 @@ TEST(NaClMungedBitcodeTest, TestReplaceRemoveEffects) {
|
| const uint64_t ReplaceSecond[] = {
|
| 1, NaClMungedBitcode::Replace, 12, 12, 14, 15, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ReplaceSecond));
|
| + MungedRecords.munge(ARRAY_TERM(ReplaceSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 12: [12, 14, 15]\n"
|
| @@ -376,7 +359,7 @@ TEST(NaClMungedBitcodeTest, TestReplaceRemoveEffects) {
|
| const uint64_t ReplaceSecondAgain[] = {
|
| 1, NaClMungedBitcode::Replace, 16, 17, 18, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ReplaceSecondAgain));
|
| + MungedRecords.munge(ARRAY_TERM(ReplaceSecondAgain));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 16: [17, 18]\n"
|
| @@ -385,7 +368,7 @@ TEST(NaClMungedBitcodeTest, TestReplaceRemoveEffects) {
|
| stringify(MungedRecords));
|
|
|
| // Override replacement with a remove.
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveSecond));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 6: [7, 8, 9]\n"
|
| @@ -401,7 +384,7 @@ TEST(NaClMungedBitcodeTest, TestBeforeAfterInteraction) {
|
| 6, 7, 8 , 9, Terminator,
|
| 10, 11, Terminator
|
| };
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -413,7 +396,7 @@ TEST(NaClMungedBitcodeTest, TestBeforeAfterInteraction) {
|
| const uint64_t AddBeforeThird[] = {
|
| 2, NaClMungedBitcode::AddBefore, 12, 13, 14, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddBeforeThird));
|
| + MungedRecords.munge(ARRAY_TERM(AddBeforeThird));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -426,7 +409,7 @@ TEST(NaClMungedBitcodeTest, TestBeforeAfterInteraction) {
|
| const uint64_t AddAfterSecond[] = {
|
| 1, NaClMungedBitcode::AddAfter, 15, 16, 17, 18, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddAfterSecond));
|
| + MungedRecords.munge(ARRAY_TERM(AddAfterSecond));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -441,7 +424,7 @@ TEST(NaClMungedBitcodeTest, TestBeforeAfterInteraction) {
|
| 2, NaClMungedBitcode::AddBefore, 19, 20, Terminator,
|
| 2, NaClMungedBitcode::AddBefore, 21, 22, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddBeforeThirdMore));
|
| + MungedRecords.munge(ARRAY_TERM(AddBeforeThirdMore));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -458,7 +441,7 @@ TEST(NaClMungedBitcodeTest, TestBeforeAfterInteraction) {
|
| 1, NaClMungedBitcode::AddAfter, 23, 24, 25, Terminator,
|
| 1, NaClMungedBitcode::AddAfter, 26, 27, 28, 29, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddAfterSecondMore));
|
| + MungedRecords.munge(ARRAY_TERM(AddAfterSecondMore));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -482,7 +465,7 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| 10, 11, Terminator
|
| };
|
|
|
| - NaClMungedBitcode MungedRecords(ARRAY_ARGS(Records));
|
| + NaClMungedBitcode MungedRecords(ARRAY_TERM(Records));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 4: [5]\n"
|
| @@ -494,7 +477,7 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| const uint64_t RemoveFirst[] = {
|
| 0, NaClMungedBitcode::Remove
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(RemoveFirst));
|
| + MungedRecords.munge(ARRAY_TERM(RemoveFirst));
|
| EXPECT_EQ(
|
| " 4: [5]\n"
|
| " 6: [7, 8, 9]\n"
|
| @@ -507,7 +490,7 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| 0, NaClMungedBitcode::AddAfter, 12, 13, 14, Terminator,
|
| 0, NaClMungedBitcode::AddAfter, 15, 16, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddAfterFirst));
|
| + MungedRecords.munge(ARRAY_TERM(AddAfterFirst));
|
| EXPECT_EQ(
|
| " 12: [13, 14]\n"
|
| " 15: [16]\n"
|
| @@ -522,7 +505,7 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| 1, NaClMungedBitcode::AddBefore, 17, 18, 19, 20, Terminator,
|
| 1, NaClMungedBitcode::AddBefore, 21, 22, 23, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddBeforeSecond));
|
| + MungedRecords.munge(ARRAY_TERM(AddBeforeSecond));
|
| EXPECT_EQ(
|
| " 12: [13, 14]\n"
|
| " 15: [16]\n"
|
| @@ -538,7 +521,7 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| const uint64_t ReplaceFirst[] = {
|
| 0, NaClMungedBitcode::Replace, 1, 2, 3, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(ReplaceFirst));
|
| + MungedRecords.munge(ARRAY_TERM(ReplaceFirst));
|
| EXPECT_EQ(
|
| " 1: [2, 3]\n"
|
| " 12: [13, 14]\n"
|
| @@ -557,7 +540,7 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| 0, NaClMungedBitcode::AddBefore, 28, 29, Terminator,
|
| 0, NaClMungedBitcode::AddBefore, 30, 31, 32, Terminator
|
| };
|
| - MungedRecords.munge(ARRAY_ARGS(AddBeforeFirst));
|
| + MungedRecords.munge(ARRAY_TERM(AddBeforeFirst));
|
| EXPECT_EQ(
|
| " 24: [25, 26, 27]\n"
|
| " 28: [29]\n"
|
| @@ -573,4 +556,4 @@ TEST(NaClMungedBitcodeTest, CombinationEdits) {
|
| stringify(MungedRecords));
|
| }
|
|
|
| -} // end of namespace NaClMungedBc
|
| +} // end of namespace naclmungetest
|
|
|