take loops into account in register allocatorHEADmaster

1 files changed, 8 insertions, 0 deletions

diff --git a/src/ejit.c b/src/ejit.c
index 2341890..1bb272d 100644
--- a/src/ejit.c
+++ b/src/ejit.c
@@ -341,6 +341,7 @@ struct ejit_func *ejit_create_func(enum ejit_type rtype, size_t argc,
 	f->sign = types_create(0);
 	f->insns = insns_create(0);
 	f->labels = labels_create(0);
+	f->barriers = barriers_create(0);
 	f->gpr = gpr_stats_create(0);
 	f->fpr = fpr_stats_create(0);
 	f->arena = NULL;
@@ -434,6 +435,7 @@ void ejit_destroy_func(struct ejit_func *f)
 	types_destroy(&f->sign);
 	insns_destroy(&f->insns);
 	labels_destroy(&f->labels);
+	barriers_destroy(&f->barriers);
 	gpr_stats_destroy(&f->gpr);
 	fpr_stats_destroy(&f->fpr);
 	free(f);
@@ -452,6 +454,12 @@ void ejit_patch(struct ejit_func *f, struct ejit_reloc r, struct ejit_label l)
 	/** @todo some assert that checks the opcode? */
 	i.r0 = l.addr;
 	*insns_at(&f->insns, r.insn) = i;
+
+	struct barrier_tuple tuple = {
+		.start = r.insn > l.addr ? l.addr : r.insn,
+		.end = r.insn > l.addr ? r.insn : l.addr
+	};
+	barriers_append(&f->barriers, tuple);
 }
 
 void ejit_taili(struct ejit_func *s, struct ejit_func *f,