xref: /llvm-project/llvm/unittests/TextAPI/TextStubHelpers.h (revision d9a9872ec4760762fdc467ef283cea302a3742e5)
1 //===-- TextStubHelpers.cpp -------------------------------------*- C++ -*-===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===-----------------------------------------------------------------------===/
8 
9 #include "llvm/Support/MemoryBuffer.h"
10 #include "llvm/TextAPI/InterfaceFile.h"
11 #include <algorithm>
12 #include <string>
13 
14 #ifndef TEXT_STUB_HELPERS_H
15 #define TEXT_STUB_HELPERS_H
16 
17 namespace llvm {
18 struct ExportedSymbol {
19   MachO::EncodeKind Kind = MachO::EncodeKind::GlobalSymbol;
20   std::string Name = {};
21   bool Weak = false;
22   bool ThreadLocalValue = false;
23   bool isData = false;
24   MachO::TargetList Targets = {};
25 };
26 
27 using ExportedSymbolSeq = std::vector<ExportedSymbol>;
28 using TargetToAttr = std::vector<std::pair<llvm::MachO::Target, std::string>>;
29 using TBDFile = std::unique_ptr<MachO::InterfaceFile>;
30 using TBDReexportFile = std::shared_ptr<MachO::InterfaceFile>;
31 
32 inline bool operator<(const ExportedSymbol &LHS, const ExportedSymbol &RHS) {
33   return std::tie(LHS.Kind, LHS.Name) < std::tie(RHS.Kind, RHS.Name);
34 }
35 
36 inline bool operator==(const ExportedSymbol &LHS, const ExportedSymbol &RHS) {
37   return std::tie(LHS.Kind, LHS.Name, LHS.Weak, LHS.ThreadLocalValue) ==
38          std::tie(RHS.Kind, RHS.Name, RHS.Weak, RHS.ThreadLocalValue);
39 }
40 
stripWhitespace(std::string S)41 inline std::string stripWhitespace(std::string S) {
42   S.erase(std::remove_if(S.begin(), S.end(), ::isspace), S.end());
43   return S;
44 }
45 
46 // This will transform a single InterfaceFile then compare against the other
47 // InterfaceFile then transform the second InterfaceFile in the same way to
48 // regain equality.
49 inline bool
checkEqualityOnTransform(MachO::InterfaceFile & FileA,MachO::InterfaceFile & FileB,void (* Transform)(MachO::InterfaceFile *))50 checkEqualityOnTransform(MachO::InterfaceFile &FileA,
51                          MachO::InterfaceFile &FileB,
52                          void (*Transform)(MachO::InterfaceFile *)) {
53   Transform(&FileA);
54   // Files should not be equal.
55   if (FileA == FileB)
56     return false;
57   Transform(&FileB);
58   // Files should be equal.
59   if (FileA != FileB)
60     return false;
61   return true;
62 }
63 
64 } // namespace llvm
65 #endif
66