Assert that the # of bytes matches for label refs (#316)

Shopify · Jul 25, 2022 · 49e1c38 · 49e1c38
1 parent 6822d78
commit 49e1c38
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/yjit/src/asm/mod.rs b/yjit/src/asm/mod.rs
@@ -252,6 +252,10 @@ impl CodeBlock {
 
             self.set_pos(ref_pos);
             (label_ref.encode)(self, (ref_pos + label_ref.num_bytes) as i64, label_addr as i64);
+
+            // Assert that we've written the same number of bytes that we
+            // expected to have written.
+            assert!(self.write_pos == ref_pos + label_ref.num_bytes);
         }
 
         self.write_pos = orig_pos;