From 670dc5ad5527f8f614938d17e33785ebad7339c7 Mon Sep 17 00:00:00 2001
From: Leonardo de Moura <leonardo@microsoft.com>
Date: Fri, 23 Aug 2013 09:16:54 -0700
Subject: [PATCH] Add option to control maximum recursion depth in the
 expression normalizer

Signed-off-by: Leonardo de Moura <leonardo@microsoft.com>

Conflicts:
	src/tests/kernel/normalize.cpp
---
 src/kernel/normalize.cpp       | 24 ++++++++++++++++++++----
 src/kernel/normalize.h         |  3 +++
 src/tests/kernel/normalize.cpp | 14 ++++++++++++++
 3 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/src/kernel/normalize.cpp b/src/kernel/normalize.cpp
index 5f31f8186..e0520f3c3 100644
--- a/src/kernel/normalize.cpp
+++ b/src/kernel/normalize.cpp
@@ -15,9 +15,20 @@ Author: Leonardo de Moura
 #include "list.h"
 #include "flet.h"
 #include "buffer.h"
-#include "exception.h"
+#include "kernel_exception.h"
+#include "options.h"
+
+#ifndef LEAN_KERNEL_NORMALIZER_MAX_DEPTH
+#define LEAN_KERNEL_NORMALIZER_MAX_DEPTH std::numeric_limits<unsigned>::max()
+#endif
 
 namespace lean {
+static name g_kernel_normalizer_max_depth       {"kernel", "normalizer", "max_depth"};
+RegisterUnsignedOption(g_kernel_normalizer_max_depth, LEAN_KERNEL_NORMALIZER_MAX_DEPTH, "(kernel) maximum recursion depth for expression normalizer");
+unsigned get_normalizer_max_depth(options const & opts) {
+    return opts.get_unsigned(g_kernel_normalizer_max_depth, LEAN_KERNEL_NORMALIZER_MAX_DEPTH);
+}
+
 class svalue;
 typedef list<svalue> value_stack; //!< Normalization stack
 enum class svalue_kind { Expr, Closure, BoundedVar };
@@ -58,6 +69,7 @@ class normalizer::imp {
     environment   m_env;
     context       m_ctx;
     cache         m_cache;
+    unsigned      m_max_depth;
     unsigned      m_depth;
     volatile bool m_interrupted;
 
@@ -118,7 +130,8 @@ class normalizer::imp {
         flet<unsigned> l(m_depth, m_depth+1);
         if (m_interrupted)
             throw interrupted();
-
+        if (m_depth > m_max_depth)
+            throw kernel_exception(m_env, "normalizer maximum recursion depth exceeded");
         bool shared = false;
         if (is_shared(a)) {
             shared = true;
@@ -250,9 +263,10 @@ class normalizer::imp {
     }
 
 public:
-    imp(environment const & env):
+    imp(environment const & env, unsigned max_depth):
         m_env(env) {
         m_interrupted = false;
+        m_max_depth   = max_depth;
         m_depth       = 0;
     }
 
@@ -276,7 +290,9 @@ public:
     void set_interrupt(bool flag) { m_interrupted = flag; }
 };
 
-normalizer::normalizer(environment const & env):m_ptr(new imp(env)) {}
+normalizer::normalizer(environment const & env, unsigned max_depth):m_ptr(new imp(env, max_depth)) {}
+normalizer::normalizer(environment const & env):normalizer(env, std::numeric_limits<unsigned>::max()) {}
+normalizer::normalizer(environment const & env, options const & opts):normalizer(env, get_normalizer_max_depth(opts)) {}
 normalizer::~normalizer() {}
 expr normalizer::operator()(expr const & e, context const & ctx) { return (*m_ptr)(e, ctx); }
 bool normalizer::is_convertible(expr const & t1, expr const & t2, context const & ctx) { return m_ptr->is_convertible(t1, t2, ctx); }
diff --git a/src/kernel/normalize.h b/src/kernel/normalize.h
index 1dfa29ddc..0f1d0badb 100644
--- a/src/kernel/normalize.h
+++ b/src/kernel/normalize.h
@@ -12,12 +12,15 @@ Author: Leonardo de Moura
 
 namespace lean {
 class environment;
+class options;
 /** \brief Functional object for normalizing expressions */
 class normalizer {
     class imp;
     std::unique_ptr<imp> m_ptr;
 public:
     normalizer(environment const & env);
+    normalizer(environment const & env, unsigned max_depth);
+    normalizer(environment const & env, options const & opts);
     ~normalizer();
 
     expr operator()(expr const & e, context const & ctx = context());
diff --git a/src/tests/kernel/normalize.cpp b/src/tests/kernel/normalize.cpp
index 65ba87859..2f2f02788 100644
--- a/src/tests/kernel/normalize.cpp
+++ b/src/tests/kernel/normalize.cpp
@@ -13,6 +13,8 @@ Author: Leonardo de Moura
 #include "trace.h"
 #include "test.h"
 #include "expr_sets.h"
+#include "abstract.h"
+#include "kernel_exception.h"
 #include "printer.h"
 using namespace lean;
 
@@ -231,6 +233,18 @@ static void tst5() {
 #endif
 }
 
+void tst6() {
+    environment env;
+    expr x = Const("x");
+    expr t = Fun({x, Type()}, mk_app(x, x));
+    expr omega = mk_app(t, t);
+    normalizer proc(env, 512);
+    try {
+        proc(omega);
+    } catch (kernel_exception & ex) {
+        std::cout << ex.what() << "\n";
+    }
+}
 
 int main() {
     tst_church_numbers();