From d41e055d1cbf17c1eb2ae03c773bbf5e319dc398 Mon Sep 17 00:00:00 2001
From: Leonardo de Moura <leonardo@microsoft.com>
Date: Mon, 27 Apr 2015 12:58:11 -0700
Subject: [PATCH] feat(frontends/lean/migrate_cmd): catch memory error in the
 migrate command

---
 src/frontends/lean/migrate_cmd.cpp | 48 ++++++++++++++++++++++--------
 1 file changed, 36 insertions(+), 12 deletions(-)

diff --git a/src/frontends/lean/migrate_cmd.cpp b/src/frontends/lean/migrate_cmd.cpp
index 7b2ccb572..b45462365 100644
--- a/src/frontends/lean/migrate_cmd.cpp
+++ b/src/frontends/lean/migrate_cmd.cpp
@@ -20,6 +20,7 @@ Author: Leonardo de Moura
 #include "library/util.h"
 #include "library/occurs.h"
 #include "library/module.h"
+#include "library/flycheck.h"
 #include "library/constants.h"
 #include "library/replace_visitor.h"
 #include "library/error_handling/error_handling.h"
@@ -76,12 +77,15 @@ struct migrate_cmd_fn {
     buffer<pair<name, name>> m_renames;
     buffer<name>             m_hiding;
 
+    bool                     m_throwable; // true if a throwable exception (which is not a lean::exception) was thrown
+
     list<expr>               m_ctx;
     buffer<expr>             m_S_params;
 
     migrate_cmd_fn(parser & p):
         m_p(p), m_env(p.env()), m_ngen(p.mk_ngen()) {
-        m_trace = get_migrate_trace(p.get_options());
+        m_throwable = false;
+        m_trace     = get_migrate_trace(p.get_options());
     }
 
     io_state_stream out() const { return regular(m_env, m_p.ios()); }
@@ -156,6 +160,22 @@ struct migrate_cmd_fn {
             m_main(m), m_num_params(m_main.m_S_params.size()) {}
     };
 
+    void report_fail_to_migrate(declaration const & d, throwable & ex) {
+        if (m_trace) {
+            out() << "failed to migrate '" << d.get_name() << "', kernel rejected new declaration\n";
+            ::lean::display_error(out(), nullptr, ex);
+            out() << "\n";
+        }
+    }
+
+    void report_skip(declaration const & d, throwable & ex) {
+        if (m_trace) {
+            out() << "skipped '" << d.get_name() << "': failed to be elaborated\n";
+            ::lean::display_error(out(), nullptr, ex);
+            out() << "\n";
+        }
+    }
+
     void migrate_decl(declaration const & d) {
         if (!check_sufficient_args(d.get_type())) {
             if (m_trace)
@@ -217,24 +237,23 @@ struct migrate_cmd_fn {
                 if (m_trace)
                     out() << "migrated: " << full_name << " : " << new_type << "\n";
             } catch (exception & ex) {
-                if (m_trace) {
-                    out() << "failed to migrate '" << d.get_name() << "', kernel rejected new declaration\n";
-                    ::lean::display_error(out(), nullptr, ex);
-                    out() << "\n";
-                }
+                report_fail_to_migrate(d, ex);
+                return;
+            } catch (throwable & ex) {
+                m_throwable = true;
+                report_fail_to_migrate(d, ex);
                 return;
             }
         } catch (exception & ex) {
-            if (m_trace) {
-                out() << "skipped '" << d.get_name() << "': failed to be elaborated\n";
-                ::lean::display_error(out(), nullptr, ex);
-                out() << "\n";
-            }
+            report_skip(d, ex);
+            return;
+        } catch (throwable & ex) {
+            m_throwable = true;
+            report_skip(d, ex);
             return;
         }
     }
 
-
     void parse_params() {
         if (!m_p.curr_is_token(get_from_tk())) {
             unsigned rbp = 0;
@@ -423,6 +442,11 @@ struct migrate_cmd_fn {
                 if (is_prefix_of(m_from, d.get_name()))
                     migrate_decl(d);
             });
+        if (m_throwable && !m_trace) {
+            flycheck_warning wrn(m_p.regular_stream());
+            m_p.display_warning_pos(m_pos);
+            m_p.regular_stream() << " some declarations were not migrated due to resource constraints (use 'set_option migrate.trace true' for additional details)" << endl;
+        }
         return m_env;
     }
 };