From 1e4f706ace7d84f8edcfd74ae444047fab836f77 Mon Sep 17 00:00:00 2001
From: gatecat <gatecat@ds0.me>
Date: Thu, 16 Sep 2021 20:56:00 +0100
Subject: [PATCH] command: Allow running Python on failure for state
 introspection

Signed-off-by: gatecat <gatecat@ds0.me>
---
 common/command.cc | 15 +++++++++++++++
 common/command.h  |  2 +-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/common/command.cc b/common/command.cc
index 34bb93cb..954a3442 100644
--- a/common/command.cc
+++ b/common/command.cc
@@ -122,6 +122,8 @@ po::options_description CommandHandler::getGeneralOptions()
     general.add_options()("pre-place", po::value<std::vector<std::string>>(), "python file to run before placement");
     general.add_options()("pre-route", po::value<std::vector<std::string>>(), "python file to run before routing");
     general.add_options()("post-route", po::value<std::vector<std::string>>(), "python file to run after routing");
+    general.add_options()("on-failure", po::value<std::vector<std::string>>(),
+                          "python file to run in event of crash for design introspection");
 
 #endif
     general.add_options()("json", po::value<std::string>(), "JSON design file to ingest");
@@ -184,6 +186,15 @@ po::options_description CommandHandler::getGeneralOptions()
     return general;
 }
 
+namespace {
+static CommandHandler *global_command_handler = nullptr;
+void script_terminate_handler()
+{
+    if (global_command_handler != nullptr)
+        global_command_handler->run_script_hook("on-failure");
+}
+}; // namespace
+
 void CommandHandler::setupContext(Context *ctx)
 {
     if (ctx->settings.find(ctx->id("seed")) != ctx->settings.end())
@@ -321,6 +332,10 @@ void CommandHandler::setupContext(Context *ctx)
 
 int CommandHandler::executeMain(std::unique_ptr<Context> ctx)
 {
+    if (vm.count("on-failure")) {
+        global_command_handler = this;
+        std::set_terminate(script_terminate_handler);
+    }
     if (vm.count("test")) {
         ctx->archcheck();
         return 0;
diff --git a/common/command.h b/common/command.h
index 092ab215..6cce8c61 100644
--- a/common/command.h
+++ b/common/command.h
@@ -39,6 +39,7 @@ class CommandHandler
     int exec();
     void load_json(Context *ctx, std::string filename);
     void clear();
+    void run_script_hook(const std::string &name);
 
   protected:
     virtual void setupArchContext(Context *ctx) = 0;
@@ -55,7 +56,6 @@ class CommandHandler
     void setupContext(Context *ctx);
     int executeMain(std::unique_ptr<Context> ctx);
     po::options_description getGeneralOptions();
-    void run_script_hook(const std::string &name);
     void printFooter();
 
   protected: