From e30122ef0af6958df0b760ad702334f0bf2c03a3 Mon Sep 17 00:00:00 2001
From: batrick <batrick@e0a8ed71-7df4-0310-8962-fdc924857419>
Date: Fri, 23 Nov 2012 03:50:17 +0000
Subject: [PATCH] Make the catch function yieldable.

---
 nse_nmaplib.cc | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/nse_nmaplib.cc b/nse_nmaplib.cc
index f31877313..fbb0b7c43 100644
--- a/nse_nmaplib.cc
+++ b/nse_nmaplib.cc
@@ -604,6 +604,12 @@ static int l_log_write (lua_State *L)
   return 0;
 }
 
+static int finalize_cleanup (lua_State *L)
+{
+  lua_settop(L, 2);
+  return lua_error(L);
+}
+
 static int new_try_finalize (lua_State *L)
 {
   if (!(lua_isboolean(L, 1) || lua_isnoneornil(L, 1)))
@@ -614,10 +620,9 @@ static int new_try_finalize (lua_State *L)
     if (!lua_isnil(L, lua_upvalueindex(1)))
     {
       lua_pushvalue(L, lua_upvalueindex(1));
-      lua_call(L, 0, 0);
+      lua_callk(L, 0, 0, 0, finalize_cleanup);
     }
-    lua_settop(L, 2);
-    lua_error(L);
+    return finalize_cleanup(L);
   }
   return lua_gettop(L)-1; /* omit first boolean argument */
 }