From c2a18ff7903205165e6eab5c619a25e6cd2216cd Mon Sep 17 00:00:00 2001
From: dshaligram <dshaligram@c06c8d41-db1a-0410-9941-cceddc491573>
Date: Tue, 19 Jun 2007 20:02:05 +0000
Subject: Set debug hooks in the Lua VM and interrupt scripts that are running
 too long. (Necessary to make Lua usable on public servers.)

git-svn-id: https://crawl-ref.svn.sourceforge.net/svnroot/crawl-ref/trunk@1605 c06c8d41-db1a-0410-9941-cceddc491573
---
 crawl-ref/source/clua.h | 40 +++++++++++++++++++++++++++++++++++++++-
 1 file changed, 39 insertions(+), 1 deletion(-)

(limited to 'crawl-ref/source/clua.h')

diff --git a/crawl-ref/source/clua.h b/crawl-ref/source/clua.h
index ab4ec6878f..af6196f826 100644
--- a/crawl-ref/source/clua.h
+++ b/crawl-ref/source/clua.h
@@ -12,15 +12,32 @@ extern "C" {
 #include <cstdio>
 #include <cstdarg>
 #include <string>
+#include <map>
 #include <set>
 
 #include "libutil.h"
 #include "externs.h"
 
+class CLua;
+class lua_call_throttle
+{
+public:
+    lua_call_throttle(CLua *handle);
+    ~lua_call_throttle();
+
+    static CLua *find_clua(lua_State *ls);
+    
+private:
+    CLua *lua;
+
+    typedef std::map<lua_State *, CLua *> lua_clua_map;
+    static lua_clua_map lua_map;
+};
+
 class CLua
 {
 public:
-    CLua();
+    CLua(bool managed);
     ~CLua();
 
     lua_State *state();
@@ -53,17 +70,36 @@ public:
 
     static int file_write(lua_State *ls);
 
+public:
     std::string error;
+
+    // If managed_vm is set, we have to take steps to control badly-behaved
+    // scripts.
+    bool managed_vm;
+    int throttle_unit_lines;
+    int throttle_sleep_ms;
+    int throttle_sleep_start, throttle_sleep_end;
+    int n_throttle_sleeps;
+    int mixed_call_depth;
+    int lua_call_depth;
+    int max_mixed_call_depth;
+    int max_lua_call_depth;
+
+    static const int MAX_THROTTLE_SLEEPS = 100;
+
 private:
     lua_State *_state;
     typedef std::set<std::string> sfset;
     sfset sourced_files;
     unsigned long uniqindex;
 
+private:
     void init_lua();
     void set_error(int err, lua_State *ls = NULL);
     void load_cmacro();
     void load_chooks();
+    void init_throttle();
+    void guard_pcall();
 
     void vfnreturns(const char *par, va_list va);
 
@@ -83,6 +119,8 @@ private:
 
         FILE *get_file();
     };
+
+    friend class lua_call_throttle;
 };
 
 class lua_text_pattern : public base_pattern
-- 
cgit v1.2.3-54-g00ecf