1 files changed, 8 insertions, 3 deletions

diff --git a/src/lj_jit.h b/src/lj_jit.h
index 1a1e407a..35595cd5 100644
--- a/src/lj_jit.h
+++ b/src/lj_jit.h
@@ -123,9 +123,14 @@ typedef struct SnapShot {
 /* Compressed snapshot entry. */
 typedef uint32_t SnapEntry;
 
-#define SNAP_FRAME              0x010000        /* Slot has frame link. */
-
-#define SNAP(slot, flags, ref)  ((SnapEntry)((slot) << 24) + (flags) + (ref))
+#define SNAP_FRAME              0x010000        /* Frame slot. */
+#define SNAP_CONT               0x020000        /* Continuation slot. */
+LJ_STATIC_ASSERT(SNAP_FRAME == TREF_FRAME);
+LJ_STATIC_ASSERT(SNAP_CONT == TREF_CONT);
+
+#define SNAP(slot, flags, ref)  (((SnapEntry)(slot) << 24) + (flags) + (ref))
+#define SNAP_TR(slot, tr) \
+  (((SnapEntry)(slot) << 24) + ((tr) & (TREF_CONT|TREF_FRAME|TREF_REFMASK)))
 #define SNAP_MKPC(pc)           ((SnapEntry)u32ptr(pc))
 #define SNAP_MKFTSZ(ftsz)       ((SnapEntry)(ftsz))
 #define snap_ref(sn)            ((sn) & 0xffff)