Skip to content

Commit 848258c

Browse files
atrosinenkoatrosinenko
committed
[BOLT] Gadget scanner: make use of C++17 features and LLVM helpers
Perform trivial syntactical cleanups: * make use of structured binding declarations * use LLVM utility functions when appropriate * omit braces around single expression inside single-line LLVM_DEBUG() This patch is NFC aside from minor debug output changes.
1 parent e62d4e4 commit 848258c

File tree

2 files changed

+37
-41
lines changed

2 files changed

+37
-41
lines changed

‎bolt/lib/Passes/PAuthGadgetScanner.cpp‎

Lines changed: 30 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,8 @@ class TrackedRegisters {
8888
TrackedRegisters(ArrayRef<MCPhysReg> RegsToTrack)
8989
: Registers(RegsToTrack),
9090
RegToIndexMapping(getMappingSize(RegsToTrack), NoIndex) {
91-
for (unsigned I = 0; I < RegsToTrack.size(); ++I)
92-
RegToIndexMapping[RegsToTrack[I]] = I;
91+
for (auto [MappedIndex, Reg] : llvm::enumerate(RegsToTrack))
92+
RegToIndexMapping[Reg] = MappedIndex;
9393
}
9494

9595
ArrayRef<MCPhysReg> getRegisters() const { return Registers; }
@@ -203,9 +203,9 @@ struct SrcState {
203203

204204
SafeToDerefRegs &= StateIn.SafeToDerefRegs;
205205
TrustedRegs &= StateIn.TrustedRegs;
206-
for (unsigned I = 0; I < LastInstWritingReg.size(); ++I)
207-
for (const MCInst *J : StateIn.LastInstWritingReg[I])
208-
LastInstWritingReg[I].insert(J);
206+
for (auto [ThisSet, OtherSet] :
207+
llvm::zip_equal(LastInstWritingReg, StateIn.LastInstWritingReg))
208+
ThisSet.insert_range(OtherSet);
209209
return *this;
210210
}
211211

@@ -224,11 +224,9 @@ struct SrcState {
224224
static void printInstsShort(raw_ostream &OS,
225225
ArrayRef<SetOfRelatedInsts> Insts) {
226226
OS << "Insts: ";
227-
for (unsigned I = 0; I < Insts.size(); ++I) {
228-
auto &Set = Insts[I];
227+
for (auto [I, PtrSet] : llvm::enumerate(Insts)) {
229228
OS << "[" << I << "](";
230-
for (const MCInst *MCInstP : Set)
231-
OS << MCInstP << " ";
229+
interleave(PtrSet, OS, " ");
232230
OS << ")";
233231
}
234232
}
@@ -416,8 +414,9 @@ class SrcSafetyAnalysis {
416414
// ... an address can be updated in a safe manner, producing the result
417415
// which is as trusted as the input address.
418416
if (auto DstAndSrc = BC.MIB->analyzeAddressArithmeticsForPtrAuth(Point)) {
419-
if (Cur.SafeToDerefRegs[DstAndSrc->second])
420-
Regs.push_back(DstAndSrc->first);
417+
auto [DstReg, SrcReg] = *DstAndSrc;
418+
if (Cur.SafeToDerefRegs[SrcReg])
419+
Regs.push_back(DstReg);
421420
}
422421

423422
// Make sure explicit checker sequence keeps register safe-to-dereference
@@ -469,8 +468,9 @@ class SrcSafetyAnalysis {
469468
// ... an address can be updated in a safe manner, producing the result
470469
// which is as trusted as the input address.
471470
if (auto DstAndSrc = BC.MIB->analyzeAddressArithmeticsForPtrAuth(Point)) {
472-
if (Cur.TrustedRegs[DstAndSrc->second])
473-
Regs.push_back(DstAndSrc->first);
471+
auto [DstReg, SrcReg] = *DstAndSrc;
472+
if (Cur.TrustedRegs[SrcReg])
473+
Regs.push_back(DstReg);
474474
}
475475

476476
return Regs;
@@ -865,9 +865,9 @@ struct DstState {
865865
return (*this = StateIn);
866866

867867
CannotEscapeUnchecked &= StateIn.CannotEscapeUnchecked;
868-
for (unsigned I = 0; I < FirstInstLeakingReg.size(); ++I)
869-
for (const MCInst *J : StateIn.FirstInstLeakingReg[I])
870-
FirstInstLeakingReg[I].insert(J);
868+
for (auto [ThisSet, OtherSet] :
869+
llvm::zip_equal(FirstInstLeakingReg, StateIn.FirstInstLeakingReg))
870+
ThisSet.insert_range(OtherSet);
871871
return *this;
872872
}
873873

@@ -1033,8 +1033,7 @@ class DstSafetyAnalysis {
10331033

10341034
// ... an address can be updated in a safe manner, or
10351035
if (auto DstAndSrc = BC.MIB->analyzeAddressArithmeticsForPtrAuth(Inst)) {
1036-
MCPhysReg DstReg, SrcReg;
1037-
std::tie(DstReg, SrcReg) = *DstAndSrc;
1036+
auto [DstReg, SrcReg] = *DstAndSrc;
10381037
// Note that *all* registers containing the derived values must be safe,
10391038
// both source and destination ones. No temporaries are supported at now.
10401039
if (Cur.CannotEscapeUnchecked[SrcReg] &&
@@ -1074,7 +1073,7 @@ class DstSafetyAnalysis {
10741073
// If this instruction terminates the program immediately, no
10751074
// authentication oracles are possible past this point.
10761075
if (BC.MIB->isTrap(Point)) {
1077-
LLVM_DEBUG({ traceInst(BC, "Trap instruction found", Point); });
1076+
LLVM_DEBUG(traceInst(BC, "Trap instruction found", Point));
10781077
DstState Next(NumRegs, RegsToTrackInstsFor.getNumTrackedRegisters());
10791078
Next.CannotEscapeUnchecked.set();
10801079
return Next;
@@ -1249,7 +1248,7 @@ class CFGUnawareDstSafetyAnalysis : public DstSafetyAnalysis,
12491248
// starting to analyze Inst.
12501249
if (BC.MIB->isCall(Inst) || BC.MIB->isBranch(Inst) ||
12511250
BC.MIB->isReturn(Inst)) {
1252-
LLVM_DEBUG({ traceInst(BC, "Control flow instruction", Inst); });
1251+
LLVM_DEBUG(traceInst(BC, "Control flow instruction", Inst));
12531252
S = createUnsafeState();
12541253
}
12551254

@@ -1394,12 +1393,12 @@ shouldReportUnsafeTailCall(const BinaryContext &BC, const BinaryFunction &BF,
13941393
// such libc, ignore tail calls performed by ELF entry function.
13951394
if (BC.StartFunctionAddress &&
13961395
*BC.StartFunctionAddress == Inst.getFunction()->getAddress()) {
1397-
LLVM_DEBUG({ dbgs() << " Skipping tail call in ELF entry function.\n"; });
1396+
LLVM_DEBUG(dbgs() << " Skipping tail call in ELF entry function.\n");
13981397
return std::nullopt;
13991398
}
14001399

14011400
if (BC.MIB->isSafeJumpTableBranchForPtrAuth(Inst)) {
1402-
LLVM_DEBUG({ dbgs() << " Safe jump table detected, skipping.\n"; });
1401+
LLVM_DEBUG(dbgs() << " Safe jump table detected, skipping.\n");
14031402
return std::nullopt;
14041403
}
14051404

@@ -1434,7 +1433,7 @@ shouldReportCallGadget(const BinaryContext &BC, const MCInstReference &Inst,
14341433
return std::nullopt;
14351434

14361435
if (BC.MIB->isSafeJumpTableBranchForPtrAuth(Inst)) {
1437-
LLVM_DEBUG({ dbgs() << " Safe jump table detected, skipping.\n"; });
1436+
LLVM_DEBUG(dbgs() << " Safe jump table detected, skipping.\n");
14381437
return std::nullopt;
14391438
}
14401439

@@ -1478,7 +1477,7 @@ shouldReportAuthOracle(const BinaryContext &BC, const MCInstReference &Inst,
14781477
});
14791478

14801479
if (S.empty()) {
1481-
LLVM_DEBUG({ dbgs() << " DstState is empty!\n"; });
1480+
LLVM_DEBUG(dbgs() << " DstState is empty!\n");
14821481
return make_generic_report(
14831482
Inst, "Warning: no state computed for an authentication instruction "
14841483
"(possibly unreachable)");
@@ -1505,7 +1504,7 @@ collectRegsToTrack(ArrayRef<PartialReport<MCPhysReg>> Reports) {
15051504
void FunctionAnalysisContext::findUnsafeUses(
15061505
SmallVector<PartialReport<MCPhysReg>> &Reports) {
15071506
auto Analysis = SrcSafetyAnalysis::create(BF, AllocatorId, {});
1508-
LLVM_DEBUG({ dbgs() << "Running src register safety analysis...\n"; });
1507+
LLVM_DEBUG(dbgs() << "Running src register safety analysis...\n");
15091508
Analysis->run();
15101509
LLVM_DEBUG({
15111510
dbgs() << "After src register safety analysis:\n";
@@ -1562,8 +1561,7 @@ void FunctionAnalysisContext::findUnsafeUses(
15621561

15631562
const SrcState &S = Analysis->getStateBefore(Inst);
15641563
if (S.empty()) {
1565-
LLVM_DEBUG(
1566-
{ traceInst(BC, "Instruction has no state, skipping", Inst); });
1564+
LLVM_DEBUG(traceInst(BC, "Instruction has no state, skipping", Inst));
15671565
assert(UnreachableBBReported && "Should be reported at least once");
15681566
(void)UnreachableBBReported;
15691567
return;
@@ -1590,8 +1588,7 @@ void FunctionAnalysisContext::augmentUnsafeUseReports(
15901588
SmallVector<MCPhysReg> RegsToTrack = collectRegsToTrack(Reports);
15911589
// Re-compute the analysis with register tracking.
15921590
auto Analysis = SrcSafetyAnalysis::create(BF, AllocatorId, RegsToTrack);
1593-
LLVM_DEBUG(
1594-
{ dbgs() << "\nRunning detailed src register safety analysis...\n"; });
1591+
LLVM_DEBUG(dbgs() << "\nRunning detailed src register safety analysis...\n");
15951592
Analysis->run();
15961593
LLVM_DEBUG({
15971594
dbgs() << "After detailed src register safety analysis:\n";
@@ -1601,7 +1598,7 @@ void FunctionAnalysisContext::augmentUnsafeUseReports(
16011598
// Augment gadget reports.
16021599
for (auto &Report : Reports) {
16031600
MCInstReference Location = Report.Issue->Location;
1604-
LLVM_DEBUG({ traceInst(BC, "Attaching clobbering info to", Location); });
1601+
LLVM_DEBUG(traceInst(BC, "Attaching clobbering info to", Location));
16051602
assert(Report.RequestedDetails &&
16061603
"Should be removed by handleSimpleReports");
16071604
auto DetailedInfo =
@@ -1619,7 +1616,7 @@ void FunctionAnalysisContext::findUnsafeDefs(
16191616
return;
16201617

16211618
auto Analysis = DstSafetyAnalysis::create(BF, AllocatorId, {});
1622-
LLVM_DEBUG({ dbgs() << "Running dst register safety analysis...\n"; });
1619+
LLVM_DEBUG(dbgs() << "Running dst register safety analysis...\n");
16231620
Analysis->run();
16241621
LLVM_DEBUG({
16251622
dbgs() << "After dst register safety analysis:\n";
@@ -1642,8 +1639,7 @@ void FunctionAnalysisContext::augmentUnsafeDefReports(
16421639
SmallVector<MCPhysReg> RegsToTrack = collectRegsToTrack(Reports);
16431640
// Re-compute the analysis with register tracking.
16441641
auto Analysis = DstSafetyAnalysis::create(BF, AllocatorId, RegsToTrack);
1645-
LLVM_DEBUG(
1646-
{ dbgs() << "\nRunning detailed dst register safety analysis...\n"; });
1642+
LLVM_DEBUG(dbgs() << "\nRunning detailed dst register safety analysis...\n");
16471643
Analysis->run();
16481644
LLVM_DEBUG({
16491645
dbgs() << "After detailed dst register safety analysis:\n";
@@ -1653,7 +1649,7 @@ void FunctionAnalysisContext::augmentUnsafeDefReports(
16531649
// Augment gadget reports.
16541650
for (auto &Report : Reports) {
16551651
MCInstReference Location = Report.Issue->Location;
1656-
LLVM_DEBUG({ traceInst(BC, "Attaching leakage info to", Location); });
1652+
LLVM_DEBUG(traceInst(BC, "Attaching leakage info to", Location));
16571653
assert(Report.RequestedDetails &&
16581654
"Should be removed by handleSimpleReports");
16591655
auto DetailedInfo = std::make_shared<LeakageInfo>(

‎bolt/test/binary-analysis/AArch64/gs-pauth-debug-output.s‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -177,9 +177,9 @@ clobber:
177177
// CHECK-EMPTY:
178178
// CHECK-NEXT: Running detailed src register safety analysis...
179179
// CHECK-NEXT: SrcSafetyAnalysis::ComputeNext( mov w30, #0x0, src-state<SafeToDerefRegs: LR W30 W30_HI , TrustedRegs: LR W30 W30_HI , Insts: [0]()>)
180-
// CHECK-NEXT: .. result: (src-state<SafeToDerefRegs: W30_HI , TrustedRegs: W30_HI , Insts: [0](0x{{[0-9a-f]+}} )>)
181-
// CHECK-NEXT: SrcSafetyAnalysis::ComputeNext( ret x30, src-state<SafeToDerefRegs: W30_HI , TrustedRegs: W30_HI , Insts: [0](0x{{[0-9a-f]+}} )>)
182-
// CHECK-NEXT: .. result: (src-state<SafeToDerefRegs: W30_HI , TrustedRegs: W30_HI , Insts: [0](0x{{[0-9a-f]+}} )>)
180+
// CHECK-NEXT: .. result: (src-state<SafeToDerefRegs: W30_HI , TrustedRegs: W30_HI , Insts: [0](0x{{[0-9a-f]+}})>)
181+
// CHECK-NEXT: SrcSafetyAnalysis::ComputeNext( ret x30, src-state<SafeToDerefRegs: W30_HI , TrustedRegs: W30_HI , Insts: [0](0x{{[0-9a-f]+}})>)
182+
// CHECK-NEXT: .. result: (src-state<SafeToDerefRegs: W30_HI , TrustedRegs: W30_HI , Insts: [0](0x{{[0-9a-f]+}})>)
183183
// CHECK-NEXT: After detailed src register safety analysis:
184184
// CHECK-NEXT: Binary Function "clobber" {
185185
// ...
@@ -189,7 +189,7 @@ clobber:
189189
// Iterating over the reports and attaching clobbering info:
190190

191191
// CHECK-EMPTY:
192-
// CHECK-NEXT: Attaching clobbering info to: 00000000: ret # DataflowSrcSafetyAnalysis: src-state<SafeToDerefRegs: BitVector, TrustedRegs: BitVector, Insts: [0](0x{{[0-9a-f]+}} )>
192+
// CHECK-NEXT: Attaching clobbering info to: 00000000: ret # DataflowSrcSafetyAnalysis: src-state<SafeToDerefRegs: BitVector, TrustedRegs: BitVector, Insts: [0](0x{{[0-9a-f]+}})>
193193

194194
.globl nocfg
195195
.type nocfg,@function
@@ -315,7 +315,7 @@ auth_oracle:
315315
// AUTH-ORACLES-NEXT: DstSafetyAnalysis::ComputeNext( ret x30, dst-state<CannotEscapeUnchecked: , Insts: [0]()>)
316316
// AUTH-ORACLES-NEXT: .. result: (dst-state<CannotEscapeUnchecked: LR W30 W30_HI , Insts: [0]()>)
317317
// AUTH-ORACLES-NEXT: DstSafetyAnalysis::ComputeNext( autia x0, x1, dst-state<CannotEscapeUnchecked: LR W30 W30_HI , Insts: [0]()>)
318-
// AUTH-ORACLES-NEXT: .. result: (dst-state<CannotEscapeUnchecked: LR W30 W30_HI , Insts: [0](0x{{[0-9a-f]+}} )>)
318+
// AUTH-ORACLES-NEXT: .. result: (dst-state<CannotEscapeUnchecked: LR W30 W30_HI , Insts: [0](0x{{[0-9a-f]+}})>)
319319
// AUTH-ORACLES-NEXT: After detailed dst register safety analysis:
320320
// AUTH-ORACLES-NEXT: Binary Function "auth_oracle" {
321321
// AUTH-ORACLES-NEXT: Number : 4
@@ -325,14 +325,14 @@ auth_oracle:
325325
// AUTH-ORACLES-NEXT: }
326326
// AUTH-ORACLES-NEXT: [[BB0]] (2 instructions, align : 1)
327327
// AUTH-ORACLES-NEXT: Entry Point
328-
// AUTH-ORACLES-NEXT: 00000000: autia x0, x1 # DataflowDstSafetyAnalysis: dst-state<CannotEscapeUnchecked: BitVector, Insts: [0](0x{{[0-9a-f]+}} )>
328+
// AUTH-ORACLES-NEXT: 00000000: autia x0, x1 # DataflowDstSafetyAnalysis: dst-state<CannotEscapeUnchecked: BitVector, Insts: [0](0x{{[0-9a-f]+}})>
329329
// AUTH-ORACLES-NEXT: 00000004: ret # DataflowDstSafetyAnalysis: dst-state<CannotEscapeUnchecked: BitVector, Insts: [0]()>
330330
// AUTH-ORACLES-EMPTY:
331331
// AUTH-ORACLES-NEXT: DWARF CFI Instructions:
332332
// AUTH-ORACLES-NEXT: <empty>
333333
// AUTH-ORACLES-NEXT: End of Function "auth_oracle"
334334
// AUTH-ORACLES-EMPTY:
335-
// AUTH-ORACLES-NEXT: Attaching leakage info to: 00000000: autia x0, x1 # DataflowDstSafetyAnalysis: dst-state<CannotEscapeUnchecked: BitVector, Insts: [0](0x{{[0-9a-f]+}} )>
335+
// AUTH-ORACLES-NEXT: Attaching leakage info to: 00000000: autia x0, x1 # DataflowDstSafetyAnalysis: dst-state<CannotEscapeUnchecked: BitVector, Insts: [0](0x{{[0-9a-f]+}})>
336336

337337
// Gadget scanner should not crash on CFI instructions, including when debug-printing them.
338338
// Note that the particular debug output is not checked, but BOLT should be

0 commit comments

Comments
 (0)