1 files changed, 9 insertions, 1 deletions
diff --git a/dte-asm.S b/dte-asm.S
index 2491826..0b7eea2 100644
--- a/dte-asm.S
+++ b/dte-asm.S
@@ -66,6 +66,7 @@ innerjump:
 
     andl $0xffff, %edx
     testb $0xff, (%edi, %edx)
+    jp   twice
     jnz  invalid
     incl (%ebx, %edx, 4)
     movl (%ebx, %edx, 4), %eax
@@ -79,7 +80,8 @@ invalid:
     
     decl %ecx
     jnz  loop
-    
+
+quit:    
     pop %edi
     pop %esi
     pop %edx
@@ -87,3 +89,9 @@ invalid:
     pop %ebx
     pop %eax
     ret
+
+twice:
+    decl %ecx
+    jnz  invalid
+    pop  %edx
+    jmp  quit