Index: compiler-rt/trunk/lib/fuzzer/FuzzerCorpus.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerCorpus.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerCorpus.h @@ -35,7 +35,7 @@ size_t NumSuccessfullMutations = 0; bool MayDeleteFile = false; bool Reduced = false; - std::vector UniqFeatureSet; + Vector UniqFeatureSet; }; class InputCorpus { @@ -71,7 +71,7 @@ bool empty() const { return Inputs.empty(); } const Unit &operator[] (size_t Idx) const { return Inputs[Idx]->U; } void AddToCorpus(const Unit &U, size_t NumFeatures, bool MayDeleteFile, - const std::vector &FeatureSet) { + const Vector &FeatureSet) { assert(!U.empty()); if (FeatureDebug) Printf("ADD_TO_CORPUS %zd NF %zd\n", Inputs.size(), NumFeatures); @@ -100,7 +100,7 @@ } // Debug-only - void PrintFeatureSet(const std::vector &FeatureSet) { + void PrintFeatureSet(const Vector &FeatureSet) { if (!FeatureDebug) return; Printf("{"); for (uint32_t Feature: FeatureSet) @@ -256,11 +256,11 @@ } std::piecewise_constant_distribution CorpusDistribution; - std::vector Intervals; - std::vector Weights; + Vector Intervals; + Vector Weights; std::unordered_set Hashes; - std::vector Inputs; + Vector Inputs; size_t NumAddedFeatures = 0; size_t NumUpdatedFeatures = 0; Index: compiler-rt/trunk/lib/fuzzer/FuzzerDefs.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerDefs.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerDefs.h @@ -18,6 +18,8 @@ #include #include #include +#include +#include // Platform detection. #ifdef __linux__ @@ -102,8 +104,23 @@ // Global interface to functions that may or may not be available. extern ExternalFunctions *EF; -typedef std::vector Unit; -typedef std::vector UnitVector; +// We are using a custom allocator to give a different symbol name to STL +// containers in order to avoid ODR violations. +template + class fuzzer_allocator: public std::allocator { + public: + template + struct rebind { typedef fuzzer_allocator other; }; + }; + +template +using Vector = std::vector>; + +template +using Set = std::set, fuzzer_allocator>; + +typedef Vector Unit; +typedef Vector UnitVector; typedef int (*UserCallback)(const uint8_t *Data, size_t Size); int FuzzerDriver(int *argc, char ***argv, UserCallback Callback); Index: compiler-rt/trunk/lib/fuzzer/FuzzerDictionary.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerDictionary.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerDictionary.h @@ -120,7 +120,7 @@ bool ParseOneDictionaryEntry(const std::string &Str, Unit *U); // Parses the dictionary file, fills Units, returns true iff all lines // were parsed succesfully. -bool ParseDictionaryFile(const std::string &Text, std::vector *Units); +bool ParseDictionaryFile(const std::string &Text, Vector *Units); } // namespace fuzzer Index: compiler-rt/trunk/lib/fuzzer/FuzzerDriver.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerDriver.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerDriver.cpp @@ -74,7 +74,7 @@ static const size_t kNumFlags = sizeof(FlagDescriptions) / sizeof(FlagDescriptions[0]); -static std::vector *Inputs; +static Vector *Inputs; static std::string *ProgName; static void PrintHelp() { @@ -175,7 +175,7 @@ } // We don't use any library to minimize dependencies. -static void ParseFlags(const std::vector &Args) { +static void ParseFlags(const Vector &Args) { for (size_t F = 0; F < kNumFlags; F++) { if (FlagDescriptions[F].IntFlag) *FlagDescriptions[F].IntFlag = FlagDescriptions[F].Default; @@ -185,7 +185,7 @@ if (FlagDescriptions[F].StrFlag) *FlagDescriptions[F].StrFlag = nullptr; } - Inputs = new std::vector; + Inputs = new Vector; for (size_t A = 1; A < Args.size(); A++) { if (ParseOneFlag(Args[A].c_str())) { if (Flags.ignore_remaining_args) @@ -225,7 +225,7 @@ } } -std::string CloneArgsWithoutX(const std::vector &Args, +std::string CloneArgsWithoutX(const Vector &Args, const char *X1, const char *X2) { std::string Cmd; for (auto &S : Args) { @@ -236,12 +236,12 @@ return Cmd; } -static int RunInMultipleProcesses(const std::vector &Args, +static int RunInMultipleProcesses(const Vector &Args, unsigned NumWorkers, unsigned NumJobs) { std::atomic Counter(0); std::atomic HasErrors(false); std::string Cmd = CloneArgsWithoutX(Args, "jobs", "workers"); - std::vector V; + Vector V; std::thread Pulse(PulseThread); Pulse.detach(); for (unsigned i = 0; i < NumWorkers; i++) @@ -294,7 +294,7 @@ return S.substr(Beg, End - Beg); } -int CleanseCrashInput(const std::vector &Args, +int CleanseCrashInput(const Vector &Args, const FuzzingOptions &Options) { if (Inputs->size() != 1 || !Flags.exact_artifact_path) { Printf("ERROR: -cleanse_crash should be given one input file and" @@ -322,7 +322,7 @@ auto U = FileToVector(CurrentFilePath); size_t Size = U.size(); - const std::vector ReplacementBytes = {' ', 0xff}; + const Vector ReplacementBytes = {' ', 0xff}; for (int NumAttempts = 0; NumAttempts < 5; NumAttempts++) { bool Changed = false; for (size_t Idx = 0; Idx < Size; Idx++) { @@ -354,7 +354,7 @@ return 0; } -int MinimizeCrashInput(const std::vector &Args, +int MinimizeCrashInput(const Vector &Args, const FuzzingOptions &Options) { if (Inputs->size() != 1) { Printf("ERROR: -minimize_crash should be given one input file\n"); @@ -456,17 +456,17 @@ return 0; } -int AnalyzeDictionary(Fuzzer *F, const std::vector& Dict, +int AnalyzeDictionary(Fuzzer *F, const Vector& Dict, UnitVector& Corpus) { Printf("Started dictionary minimization (up to %d tests)\n", Dict.size() * Corpus.size() * 2); // Scores and usage count for each dictionary unit. - std::vector Scores(Dict.size()); - std::vector Usages(Dict.size()); + Vector Scores(Dict.size()); + Vector Usages(Dict.size()); - std::vector InitialFeatures; - std::vector ModifiedFeatures; + Vector InitialFeatures; + Vector ModifiedFeatures; for (auto &C : Corpus) { // Get coverage for the testcase without modifications. F->ExecuteCallback(C.data(), C.size()); @@ -477,7 +477,7 @@ }); for (size_t i = 0; i < Dict.size(); ++i) { - auto Data = C; + Vector Data = C; auto StartPos = std::search(Data.begin(), Data.end(), Dict[i].begin(), Dict[i].end()); // Skip dictionary unit, if the testcase does not contain it. @@ -531,7 +531,7 @@ EF = new ExternalFunctions(); if (EF->LLVMFuzzerInitialize) EF->LLVMFuzzerInitialize(argc, argv); - const std::vector Args(*argv, *argv + *argc); + const Vector Args(*argv, *argv + *argc); assert(!Args.empty()); ProgName = new std::string(Args[0]); if (Argv0 != *ProgName) { @@ -593,7 +593,7 @@ Options.ArtifactPrefix = Flags.artifact_prefix; if (Flags.exact_artifact_path) Options.ExactArtifactPath = Flags.exact_artifact_path; - std::vector Dictionary; + Vector Dictionary; if (Flags.dict) if (!ParseDictionaryFile(FileToString(Flags.dict), &Dictionary)) return 1; Index: compiler-rt/trunk/lib/fuzzer/FuzzerIO.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerIO.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerIO.h @@ -27,7 +27,7 @@ void WriteToFile(const Unit &U, const std::string &Path); -void ReadDirToVectorOfUnits(const char *Path, std::vector *V, +void ReadDirToVectorOfUnits(const char *Path, Vector *V, long *Epoch, size_t MaxSize, bool ExitOnError); // Returns "Dir/FileName" or equivalent for the current OS. @@ -55,7 +55,7 @@ bool IsFile(const std::string &Path); void ListFilesInDirRecursive(const std::string &Dir, long *Epoch, - std::vector *V, bool TopDir); + Vector *V, bool TopDir); char GetSeparator(); Index: compiler-rt/trunk/lib/fuzzer/FuzzerIO.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerIO.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerIO.cpp @@ -68,10 +68,10 @@ fclose(Out); } -void ReadDirToVectorOfUnits(const char *Path, std::vector *V, +void ReadDirToVectorOfUnits(const char *Path, Vector *V, long *Epoch, size_t MaxSize, bool ExitOnError) { long E = Epoch ? *Epoch : 0; - std::vector Files; + Vector Files; ListFilesInDirRecursive(Path, Epoch, &Files, /*TopDir*/true); size_t NumLoaded = 0; for (size_t i = 0; i < Files.size(); i++) { Index: compiler-rt/trunk/lib/fuzzer/FuzzerIOPosix.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerIOPosix.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerIOPosix.cpp @@ -33,7 +33,7 @@ } void ListFilesInDirRecursive(const std::string &Dir, long *Epoch, - std::vector *V, bool TopDir) { + Vector *V, bool TopDir) { auto E = GetEpoch(Dir); if (Epoch) if (E && *Epoch >= E) return; Index: compiler-rt/trunk/lib/fuzzer/FuzzerIOWindows.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerIOWindows.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerIOWindows.cpp @@ -73,7 +73,7 @@ } void ListFilesInDirRecursive(const std::string &Dir, long *Epoch, - std::vector *V, bool TopDir) { + Vector *V, bool TopDir) { auto E = GetEpoch(Dir); if (Epoch) if (E && *Epoch >= E) return; Index: compiler-rt/trunk/lib/fuzzer/FuzzerInternal.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerInternal.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerInternal.h @@ -69,9 +69,9 @@ InputInfo *II = nullptr); // Merge Corpora[1:] into Corpora[0]. - void Merge(const std::vector &Corpora); - void CrashResistantMerge(const std::vector &Args, - const std::vector &Corpora, + void Merge(const Vector &Corpora); + void CrashResistantMerge(const Vector &Args, + const Vector &Corpora, const char *CoverageSummaryInputPathOrNull, const char *CoverageSummaryOutputPathOrNull); void CrashResistantMergeInternalStep(const std::string &ControlFilePath); @@ -139,7 +139,7 @@ size_t MaxMutationLen = 0; size_t TmpMaxMutationLen = 0; - std::vector UniqFeatureSetTmp; + Vector UniqFeatureSetTmp; // Need to know our own thread. static thread_local bool IsMyThread; Index: compiler-rt/trunk/lib/fuzzer/FuzzerLoop.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerLoop.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerLoop.cpp @@ -327,7 +327,7 @@ void Fuzzer::CheckExitOnSrcPosOrItem() { if (!Options.ExitOnSrcPos.empty()) { - static auto *PCsSet = new std::set; + static auto *PCsSet = new Set; auto HandlePC = [&](uintptr_t PC) { if (!PCsSet->insert(PC).second) return; std::string Descr = DescribePC("%F %L", PC + 1); @@ -350,7 +350,7 @@ void Fuzzer::RereadOutputCorpus(size_t MaxSize) { if (Options.OutputCorpus.empty() || !Options.ReloadIntervalSec) return; - std::vector AdditionalCorpus; + Vector AdditionalCorpus; ReadDirToVectorOfUnits(Options.OutputCorpus.c_str(), &AdditionalCorpus, &EpochOfLastReadOfOutputCorpus, MaxSize, /*ExitOnError*/ false); Index: compiler-rt/trunk/lib/fuzzer/FuzzerMerge.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerMerge.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerMerge.h @@ -52,11 +52,11 @@ struct MergeFileInfo { std::string Name; size_t Size = 0; - std::vector Features; + Vector Features; }; struct Merger { - std::vector Files; + Vector Files; size_t NumFilesInFirstCorpus = 0; size_t FirstNotProcessedFile = 0; std::string LastFailure; @@ -65,14 +65,14 @@ bool Parse(const std::string &Str, bool ParseCoverage); void ParseOrExit(std::istream &IS, bool ParseCoverage); void PrintSummary(std::ostream &OS); - std::set ParseSummary(std::istream &IS); - size_t Merge(const std::set &InitialFeatures, - std::vector *NewFiles); - size_t Merge(std::vector *NewFiles) { - return Merge(std::set{}, NewFiles); + Set ParseSummary(std::istream &IS); + size_t Merge(const Set &InitialFeatures, + Vector *NewFiles); + size_t Merge(Vector *NewFiles) { + return Merge(Set{}, NewFiles); } size_t ApproximateMemoryConsumption() const; - std::set AllFeatures() const; + Set AllFeatures() const; }; } // namespace fuzzer Index: compiler-rt/trunk/lib/fuzzer/FuzzerMerge.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerMerge.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerMerge.cpp @@ -74,7 +74,7 @@ size_t ExpectedStartMarker = 0; const size_t kInvalidStartMarker = -1; size_t LastSeenStartMarker = kInvalidStartMarker; - std::vector TmpFeatures; + Vector TmpFeatures; while (std::getline(IS, Line, '\n')) { std::istringstream ISS1(Line); std::string Marker; @@ -122,11 +122,11 @@ // Decides which files need to be merged (add thost to NewFiles). // Returns the number of new features added. -size_t Merger::Merge(const std::set &InitialFeatures, - std::vector *NewFiles) { +size_t Merger::Merge(const Set &InitialFeatures, + Vector *NewFiles) { NewFiles->clear(); assert(NumFilesInFirstCorpus <= Files.size()); - std::set AllFeatures(InitialFeatures); + Set AllFeatures(InitialFeatures); // What features are in the initial corpus? for (size_t i = 0; i < NumFilesInFirstCorpus; i++) { @@ -138,7 +138,7 @@ // Remove all features that we already know from all other inputs. for (size_t i = NumFilesInFirstCorpus; i < Files.size(); i++) { auto &Cur = Files[i].Features; - std::vector Tmp; + Vector Tmp; std::set_difference(Cur.begin(), Cur.end(), AllFeatures.begin(), AllFeatures.end(), std::inserter(Tmp, Tmp.begin())); Cur.swap(Tmp); @@ -178,16 +178,16 @@ } } -std::set Merger::AllFeatures() const { - std::set S; +Set Merger::AllFeatures() const { + Set S; for (auto &File : Files) S.insert(File.Features.begin(), File.Features.end()); return S; } -std::set Merger::ParseSummary(std::istream &IS) { +Set Merger::ParseSummary(std::istream &IS) { std::string Line, Tmp; - std::set Res; + Set Res; while (std::getline(IS, Line, '\n')) { size_t N; std::istringstream ISS1(Line); @@ -235,7 +235,7 @@ TPC.ResetMaps(); ExecuteCallback(U.data(), U.size()); // Collect coverage. - std::set Features; + Set Features; TPC.CollectFeatures([&](size_t Feature) -> bool { Features.insert(Feature); return true; @@ -252,15 +252,15 @@ } // Outer process. Does not call the target code and thus sohuld not fail. -void Fuzzer::CrashResistantMerge(const std::vector &Args, - const std::vector &Corpora, +void Fuzzer::CrashResistantMerge(const Vector &Args, + const Vector &Corpora, const char *CoverageSummaryInputPathOrNull, const char *CoverageSummaryOutputPathOrNull) { if (Corpora.size() <= 1) { Printf("Merge requires two or more corpus dirs\n"); return; } - std::vector AllFiles; + Vector AllFiles; ListFilesInDirRecursive(Corpora[0], nullptr, &AllFiles, /*TopDir*/true); size_t NumFilesInFirstCorpus = AllFiles.size(); for (size_t i = 1; i < Corpora.size(); i++) @@ -318,8 +318,8 @@ std::ofstream SummaryOut(CoverageSummaryOutputPathOrNull); M.PrintSummary(SummaryOut); } - std::vector NewFiles; - std::set InitialFeatures; + Vector NewFiles; + Set InitialFeatures; if (CoverageSummaryInputPathOrNull) { std::ifstream SummaryIn(CoverageSummaryInputPathOrNull); InitialFeatures = M.ParseSummary(SummaryIn); Index: compiler-rt/trunk/lib/fuzzer/FuzzerMutate.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerMutate.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerMutate.h @@ -96,7 +96,7 @@ size_t AddWordFromDictionary(Dictionary &D, uint8_t *Data, size_t Size, size_t MaxSize); size_t MutateImpl(uint8_t *Data, size_t Size, size_t MaxSize, - const std::vector &Mutators); + Vector &Mutators); size_t InsertPartOf(const uint8_t *From, size_t FromSize, uint8_t *To, size_t ToSize, size_t MaxToSize); @@ -128,21 +128,21 @@ // entries that led to successfull discoveries in the past mutations. Dictionary PersistentAutoDictionary; - std::vector CurrentMutatorSequence; - std::vector CurrentDictionaryEntrySequence; + Vector CurrentMutatorSequence; + Vector CurrentDictionaryEntrySequence; static const size_t kCmpDictionaryEntriesDequeSize = 16; DictionaryEntry CmpDictionaryEntriesDeque[kCmpDictionaryEntriesDequeSize]; size_t CmpDictionaryEntriesDequeIdx = 0; const InputCorpus *Corpus = nullptr; - std::vector MutateInPlaceHere; + Vector MutateInPlaceHere; // CustomCrossOver needs its own buffer as a custom implementation may call // LLVMFuzzerMutate, which in turn may resize MutateInPlaceHere. - std::vector CustomCrossOverInPlaceHere; + Vector CustomCrossOverInPlaceHere; - std::vector Mutators; - std::vector DefaultMutators; + Vector Mutators; + Vector DefaultMutators; }; } // namespace fuzzer Index: compiler-rt/trunk/lib/fuzzer/FuzzerMutate.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerMutate.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerMutate.cpp @@ -466,7 +466,7 @@ } void MutationDispatcher::PrintRecommendedDictionary() { - std::vector V; + Vector V; for (auto &DE : PersistentAutoDictionary) if (!ManualDictionary.ContainsWord(DE.GetW())) V.push_back(DE); @@ -506,7 +506,7 @@ // Mutates Data in place, returns new size. size_t MutationDispatcher::MutateImpl(uint8_t *Data, size_t Size, size_t MaxSize, - const std::vector &Mutators) { + Vector &Mutators) { assert(MaxSize > 0); // Some mutations may fail (e.g. can't insert more bytes if Size == MaxSize), // in which case they will return 0. Index: compiler-rt/trunk/lib/fuzzer/FuzzerTracePC.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerTracePC.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerTracePC.h @@ -159,8 +159,8 @@ uint8_t *Counters() const; uintptr_t *PCs() const; - std::set ObservedPCs; - std::set ObservedFuncs; + Set ObservedPCs; + Set ObservedFuncs; ValueBitMap ValueProfileMap; uintptr_t InitialStack; Index: compiler-rt/trunk/lib/fuzzer/FuzzerTracePC.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerTracePC.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerTracePC.cpp @@ -221,8 +221,8 @@ Printf("COVERAGE:\n"); std::string LastFunctionName = ""; std::string LastFileStr = ""; - std::set UncoveredLines; - std::set CoveredLines; + Set UncoveredLines; + Set CoveredLines; auto FunctionEndCallback = [&](const std::string &CurrentFunc, const std::string &CurrentFile) { @@ -270,7 +270,7 @@ void TracePC::DumpCoverage() { if (EF->__sanitizer_dump_coverage) { - std::vector PCsCopy(GetNumPCs()); + Vector PCsCopy(GetNumPCs()); for (size_t i = 0; i < GetNumPCs(); i++) PCsCopy[i] = PCs()[i] ? GetPreviousInstructionPc(PCs()[i]) : 0; EF->__sanitizer_dump_coverage(PCsCopy.data(), PCsCopy.size()); Index: compiler-rt/trunk/lib/fuzzer/FuzzerUtil.h =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerUtil.h +++ compiler-rt/trunk/lib/fuzzer/FuzzerUtil.h @@ -57,10 +57,10 @@ const void *SearchMemory(const void *haystack, size_t haystacklen, const void *needle, size_t needlelen); -std::string CloneArgsWithoutX(const std::vector &Args, +std::string CloneArgsWithoutX(const Vector &Args, const char *X1, const char *X2); -inline std::string CloneArgsWithoutX(const std::vector &Args, +inline std::string CloneArgsWithoutX(const Vector &Args, const char *X) { return CloneArgsWithoutX(Args, X, X); } Index: compiler-rt/trunk/lib/fuzzer/FuzzerUtil.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/FuzzerUtil.cpp +++ compiler-rt/trunk/lib/fuzzer/FuzzerUtil.cpp @@ -124,7 +124,7 @@ return true; } -bool ParseDictionaryFile(const std::string &Text, std::vector *Units) { +bool ParseDictionaryFile(const std::string &Text, Vector *Units) { if (Text.empty()) { Printf("ParseDictionaryFile: file does not exist or is empty\n"); return false; Index: compiler-rt/trunk/lib/fuzzer/tests/FuzzerUnittest.cpp =================================================================== --- compiler-rt/trunk/lib/fuzzer/tests/FuzzerUnittest.cpp +++ compiler-rt/trunk/lib/fuzzer/tests/FuzzerUnittest.cpp @@ -72,7 +72,7 @@ { 0, 5, 6, 7, 1, 2 } }; for (size_t Len = 1; Len < 8; Len++) { - std::set FoundUnits, ExpectedUnitsWitThisLength; + Set FoundUnits, ExpectedUnitsWitThisLength; for (int Iter = 0; Iter < 3000; Iter++) { C.resize(Len); size_t NewSize = MD.CrossOver(A.data(), A.size(), B.data(), B.size(), @@ -526,7 +526,7 @@ } TEST(FuzzerDictionary, ParseDictionaryFile) { - std::vector Units; + Vector Units; EXPECT_FALSE(ParseDictionaryFile("zzz\n", &Units)); EXPECT_FALSE(ParseDictionaryFile("", &Units)); EXPECT_TRUE(ParseDictionaryFile("\n", &Units)); @@ -538,11 +538,11 @@ EXPECT_TRUE(ParseDictionaryFile(" #zzzz\n", &Units)); EXPECT_EQ(Units.size(), 0U); EXPECT_TRUE(ParseDictionaryFile(" #zzzz\naaa=\"aa\"", &Units)); - EXPECT_EQ(Units, std::vector({Unit({'a', 'a'})})); + EXPECT_EQ(Units, Vector({Unit({'a', 'a'})})); EXPECT_TRUE( ParseDictionaryFile(" #zzzz\naaa=\"aa\"\n\nabc=\"abc\"", &Units)); EXPECT_EQ(Units, - std::vector({Unit({'a', 'a'}), Unit({'a', 'b', 'c'})})); + Vector({Unit({'a', 'a'}), Unit({'a', 'b', 'c'})})); } TEST(FuzzerUtil, Base64) { @@ -566,7 +566,7 @@ for (size_t i = 0; i < N; i++) C->AddToCorpus(Unit{ static_cast(i) }, 1, false, {}); - std::vector Hist(N); + Vector Hist(N); for (size_t i = 0; i < N * TriesPerUnit; i++) { Hist[C->ChooseUnitIdxToMutate(Rand)]++; } @@ -596,21 +596,21 @@ } } -void EQ(const std::vector &A, const std::vector &B) { +void EQ(const Vector &A, const Vector &B) { EXPECT_EQ(A, B); } -void EQ(const std::vector &A, const std::vector &B) { - std::set a(A.begin(), A.end()); - std::set b(B.begin(), B.end()); +void EQ(const Vector &A, const Vector &B) { + Set a(A.begin(), A.end()); + Set b(B.begin(), B.end()); EXPECT_EQ(a, b); } static void Merge(const std::string &Input, - const std::vector Result, + const Vector Result, size_t NumNewFeatures) { Merger M; - std::vector NewFiles; + Vector NewFiles; EXPECT_TRUE(M.Parse(Input, true)); std::stringstream SS; M.PrintSummary(SS); @@ -658,7 +658,7 @@ EQ(M.Files[1].Features, {4, 5, 6}); - std::vector NewFiles; + Vector NewFiles; EXPECT_TRUE(M.Parse("3\n2\nAA\nBB\nC\n" "STARTED 0 1000\nDONE 0 1 2 3\n" @@ -693,7 +693,11 @@ "", true)); EQ(M.Files[0].Features, {4, 5, 6}); EQ(M.Files[1].Features, {1, 3, 6}); - EXPECT_EQ(3U, M.Merge({1, 2, 3}, &NewFiles)); + Set InitialFeatures; + InitialFeatures.insert(1); + InitialFeatures.insert(2); + InitialFeatures.insert(3); + EXPECT_EQ(3U, M.Merge(InitialFeatures, &NewFiles)); EQ(NewFiles, {"B"}); } @@ -739,7 +743,7 @@ 0, 0, 0, 0, 0, 0, 0, 8, 9, 9, 9, 9, 9, 9, 9, 9, }; - typedef std::vector > Vec; + typedef Vector > Vec; Vec Res, Expected; auto CB = [&](size_t FirstFeature, size_t Idx, uint8_t V) { Res.push_back({FirstFeature + Idx, V});