Skip to content

Commit a12f44e

Browse files
committed
[BOLT][BTI] rename addBTItoBBStart to insertBTI
1 parent 5ab3f19 commit a12f44e

File tree

3 files changed

+18
-18
lines changed

3 files changed

+18
-18
lines changed

bolt/include/bolt/Core/MCPlusBuilder.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1901,9 +1901,9 @@ class MCPlusBuilder {
19011901
return false;
19021902
}
19031903

1904-
/// Adds a BTI landing pad to the start of the BB, that matches the indirect
1905-
/// call/jump inst.
1906-
virtual void addBTItoBBStart(BinaryBasicBlock &BB, MCInst &Call) const {
1904+
/// Inserts a BTI landing pad to the start of the BB, that matches the
1905+
/// indirect call inst used to call the BB.
1906+
virtual void insertBTI(BinaryBasicBlock &BB, MCInst &Call) const {
19071907
llvm_unreachable("not implemented");
19081908
}
19091909

bolt/lib/Target/AArch64/AArch64MCPlusBuilder.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2836,7 +2836,7 @@ class AArch64MCPlusBuilder : public MCPlusBuilder {
28362836
return false;
28372837
}
28382838

2839-
void addBTItoBBStart(BinaryBasicBlock &BB, MCInst &Call) const override {
2839+
void insertBTI(BinaryBasicBlock &BB, MCInst &Call) const override {
28402840
auto II = BB.getFirstNonPseudo();
28412841
if (II != BB.end()) {
28422842
if (isBTIVariantCoveringCall(Call, *II))

bolt/unittests/Core/MCPlusBuilder.cpp

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ TEST_P(MCPlusBuilderTester, AArch64_BTI) {
198198
ASSERT_TRUE(BC->MIB->isImplicitBTIC(*II));
199199
}
200200

201-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_0) {
201+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_0) {
202202
if (GetParam() != Triple::aarch64)
203203
GTEST_SKIP();
204204
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -207,12 +207,12 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_0) {
207207
BB->addInstruction(Inst);
208208
// BR x16 needs BTI c or BTI j. We prefer adding a BTI c.
209209
MCInst CallInst = MCInstBuilder(AArch64::BR).addReg(AArch64::X16);
210-
BC->MIB->addBTItoBBStart(*BB, CallInst);
210+
BC->MIB->insertBTI(*BB, CallInst);
211211
auto II = BB->begin();
212212
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, true, false));
213213
}
214214

215-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_1) {
215+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_1) {
216216
if (GetParam() != Triple::aarch64)
217217
GTEST_SKIP();
218218
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -222,12 +222,12 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_1) {
222222
BB->addInstruction(BTIc);
223223
// BR x16 needs BTI c or BTI j. We have a BTI c, no change is needed.
224224
MCInst CallInst = MCInstBuilder(AArch64::BR).addReg(AArch64::X16);
225-
BC->MIB->addBTItoBBStart(*BB, CallInst);
225+
BC->MIB->insertBTI(*BB, CallInst);
226226
auto II = BB->begin();
227227
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, true, false));
228228
}
229229

230-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_2) {
230+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_2) {
231231
if (GetParam() != Triple::aarch64)
232232
GTEST_SKIP();
233233
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -238,12 +238,12 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_2) {
238238
// BR x5 needs BTI j
239239
// we have BTI c -> extend it to BTI jc.
240240
MCInst CallInst = MCInstBuilder(AArch64::BR).addReg(AArch64::X5);
241-
BC->MIB->addBTItoBBStart(*BB, CallInst);
241+
BC->MIB->insertBTI(*BB, CallInst);
242242
auto II = BB->begin();
243243
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, true, true));
244244
}
245245

246-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_3) {
246+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_3) {
247247
if (GetParam() != Triple::aarch64)
248248
GTEST_SKIP();
249249
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -252,12 +252,12 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_3) {
252252
BB->addInstruction(Inst);
253253
// BR x5 needs BTI j
254254
MCInst CallInst = MCInstBuilder(AArch64::BR).addReg(AArch64::X5);
255-
BC->MIB->addBTItoBBStart(*BB, CallInst);
255+
BC->MIB->insertBTI(*BB, CallInst);
256256
auto II = BB->begin();
257257
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, false, true));
258258
}
259259

260-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_4) {
260+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_4) {
261261
if (GetParam() != Triple::aarch64)
262262
GTEST_SKIP();
263263
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -266,12 +266,12 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_4) {
266266
BB->addInstruction(Inst);
267267
// BLR needs BTI c, regardless of the register used.
268268
MCInst CallInst = MCInstBuilder(AArch64::BLR).addReg(AArch64::X5);
269-
BC->MIB->addBTItoBBStart(*BB, CallInst);
269+
BC->MIB->insertBTI(*BB, CallInst);
270270
auto II = BB->begin();
271271
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, true, false));
272272
}
273273

274-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_5) {
274+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_5) {
275275
if (GetParam() != Triple::aarch64)
276276
GTEST_SKIP();
277277
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -282,12 +282,12 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_5) {
282282
// BLR needs BTI c, regardless of the register used.
283283
// We have a BTI j -> extend it to BTI jc.
284284
MCInst CallInst = MCInstBuilder(AArch64::BLR).addReg(AArch64::X5);
285-
BC->MIB->addBTItoBBStart(*BB, CallInst);
285+
BC->MIB->insertBTI(*BB, CallInst);
286286
auto II = BB->begin();
287287
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, true, true));
288288
}
289289

290-
TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_6) {
290+
TEST_P(MCPlusBuilderTester, AArch64_insertBTI_6) {
291291
if (GetParam() != Triple::aarch64)
292292
GTEST_SKIP();
293293
BinaryFunction *BF = BC->createInjectedBinaryFunction("BF", true);
@@ -297,7 +297,7 @@ TEST_P(MCPlusBuilderTester, AArch64_addBTItoBBStart_6) {
297297
BB->addInstruction(Paciasp);
298298
// PACI(AB)SP are implicit BTI c, no change needed.
299299
MCInst CallInst = MCInstBuilder(AArch64::BR).addReg(AArch64::X17);
300-
BC->MIB->addBTItoBBStart(*BB, CallInst);
300+
BC->MIB->insertBTI(*BB, CallInst);
301301
auto II = BB->begin();
302302
ASSERT_TRUE(BC->MIB->isBTILandingPad(*II, true, false));
303303
ASSERT_TRUE(BC->MIB->isPSignOnLR(*II));

0 commit comments

Comments
 (0)