summary refs log tree commit diff
diff options
context:
space:
mode:
authorTobias Heider <me@tobhe.de>2023-12-03 17:32:20 +0100
committerQuentin Carbonneaux <quentin@c9x.me>2023-12-30 15:59:25 +0100
commit5af33410f6faa6c0f7d97af5b802b2608b228bc2 (patch)
treed1cc0a83543d529898949b85e54003d867234722
parent00501eeb6ee3a59cc2aa5f78c6569b8d1a3dae49 (diff)
downloadroux-5af33410f6faa6c0f7d97af5b802b2608b228bc2.tar.gz
Fix IBT/BTI by instrumenting function calls
-rw-r--r--amd64/emit.c2
-rw-r--r--arm64/emit.c1
2 files changed, 2 insertions, 1 deletions
diff --git a/amd64/emit.c b/amd64/emit.c
index 297cc76..c949589 100644
--- a/amd64/emit.c
+++ b/amd64/emit.c
@@ -581,7 +581,7 @@ amd64_emitfn(Fn *fn, FILE *f)
 	uint64_t fs;
 
 	emitfnlnk(fn->name, &fn->lnk, f);
-	fputs("\tpushq %rbp\n\tmovq %rsp, %rbp\n", f);
+	fputs("\tendbr64\n\tpushq %rbp\n\tmovq %rsp, %rbp\n", f);
 	fs = framesz(fn);
 	if (fs)
 		fprintf(f, "\tsubq $%"PRIu64", %%rsp\n", fs);
diff --git a/arm64/emit.c b/arm64/emit.c
index 78a0358..85b5f3d 100644
--- a/arm64/emit.c
+++ b/arm64/emit.c
@@ -514,6 +514,7 @@ arm64_emitfn(Fn *fn, FILE *out)
 	if (T.apple)
 		e->fn->lnk.align = 4;
 	emitfnlnk(e->fn->name, &e->fn->lnk, e->f);
+	fputs("\thint\t#34\n", e->f);
 	framelayout(e);
 
 	if (e->fn->vararg && !T.apple) {