Tolerate empty final block

hj424 · Sep 30, 2020 · d19def3 · d19def3
1 parent 0b4ae8c
commit d19def3
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/examples/cfg.py b/examples/cfg.py
@@ -51,12 +51,15 @@ def add_terminators(blocks):
     """
     for i, block in enumerate(blocks.values()):
         if not block:
-            dest = list(blocks.keys())[i + 1]
-            block.append({'op': 'jmp', 'labels': [dest]})
-        elif block[-1]['op'] not in TERMINATORS:
             if i == len(blocks) - 1:
                 # In the last block, return.
                 block.append({'op': 'ret', 'args': []})
+            else:
+                dest = list(blocks.keys())[i + 1]
+                block.append({'op': 'jmp', 'labels': [dest]})
+        elif block[-1]['op'] not in TERMINATORS:
+            if i == len(blocks) - 1:
+                block.append({'op': 'ret', 'args': []})
             else:
                 # Otherwise, jump to the next block.
                 dest = list(blocks.keys())[i + 1]