From 9af0e9c91a94218d35a5e773019ec98552052f7b Mon Sep 17 00:00:00 2001
From: Sebastien Bourdeauducq <sb@m-labs.hk>
Date: Wed, 17 Sep 2014 18:03:54 +0800
Subject: [PATCH] devices/runtime: support dumping of ELF object via
 ARTIQ_DUMP_OBJECT environment variable

---
 artiq/devices/runtime.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/artiq/devices/runtime.py b/artiq/devices/runtime.py
index c0ad64fa7..4a2c8e9c3 100644
--- a/artiq/devices/runtime.py
+++ b/artiq/devices/runtime.py
@@ -1,3 +1,5 @@
+import os
+
 from llvm import core as lc
 from llvm import target as lt
 
@@ -72,6 +74,25 @@ class LinkInterface:
         return r
 
 
+def _debug_dump_obj(obj):
+    try:
+        env = os.environ["ARTIQ_DUMP_OBJECT"]
+    except KeyError:
+        return
+
+    for i in range(1000):
+        filename = "{}_{:03d}.elf".format(env, i)
+        try:
+            f = open(filename, "xb")
+        except FileExistsError:
+            pass
+        else:
+            f.write(obj)
+            f.close()
+            return
+    raise IOError
+
+
 class Environment(LinkInterface):
     def __init__(self, ref_period):
         self.ref_period = ref_period
@@ -80,4 +101,5 @@ class Environment(LinkInterface):
     def emit_object(self):
         tm = lt.TargetMachine.new(triple="or1k", cpu="generic")
         obj = tm.emit_object(self.llvm_module)
+        _debug_dump_obj(obj)
         return obj