From c6518eb6e1caa68cdf0dc242d52cfc172b96efdc Mon Sep 17 00:00:00 2001
From: Nicolas Noble <pixel@nobis-crew.org>
Date: Fri, 2 Aug 2013 18:36:38 -0700
Subject: Lua's dumpvars is now properly yielding if needed.

---
 src/BLua.cc        | 22 +++++++++++++++++++++-
 src/HelperTasks.cc | 42 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+), 1 deletion(-)
 create mode 100644 src/HelperTasks.cc

(limited to 'src')

diff --git a/src/BLua.cc b/src/BLua.cc
index 29f4a58..7eda510 100644
--- a/src/BLua.cc
+++ b/src/BLua.cc
@@ -3,6 +3,7 @@
 #include "Printer.h"
 #include "Input.h"
 #include "Buffer.h"
+#include "HelperTasks.h"
 
 extern "C" {
 #include <lualib.h>
@@ -90,7 +91,26 @@ int Balau::LuaStatics::dumpvars(lua_State * __L) {
 
     IO<Handle> h(L.recast<Balau::Handle>());
 
-    L.dumpvars(h, prefix);
+    if (h.isA<Buffer>()) {
+        L.dumpvars(h, prefix);
+    } else {
+        IO<Handle> s(new Buffer());
+        L.dumpvars(h, prefix);
+        Task * t = new CopyTask(s, h);
+        Events::TaskEvent * evt = new Events::TaskEvent(t);
+        L.yield(Future<int>([evt, s]() mutable {
+            for (;;) {
+                if (evt->gotSignal()) {
+                    evt->ack();
+                    delete evt;
+                    s->close();
+                } else {
+                    Task::operationYield(evt, Task::INTERRUPTIBLE);
+                }
+            }
+            return 0;
+        }));
+    }
 
     return 0;
 }
diff --git a/src/HelperTasks.cc b/src/HelperTasks.cc
new file mode 100644
index 0000000..ab1c2a9
--- /dev/null
+++ b/src/HelperTasks.cc
@@ -0,0 +1,42 @@
+#include <algorithm>
+
+#include "HelperTasks.h"
+
+Balau::CopyTask::CopyTask(IO<Handle> s, IO<Handle> d, ssize_t tocopy)
+    : m_s(s)
+    , m_d(d)
+    , m_tocopy(tocopy)
+{
+    m_name.set("CopyTask from %s to %s", s->getName(), d->getName());
+}
+
+void Balau::CopyTask::Do() {
+    ssize_t toread, w;
+
+    try {
+        for (;;) {
+            switch (m_state) {
+            case 0:
+                toread = m_tocopy >= 0 ? m_tocopy - m_current : COPY_BUFSIZE;
+                toread = std::min(toread, (ssize_t) COPY_BUFSIZE);
+                m_read = m_s->read(m_buffer, toread);
+                AAssert(m_read >= 0, "Error while reading");
+                if (m_s->isEOF() || !m_read)
+                    return;
+                m_written = 0;
+                m_state = 1;
+            case 1:
+                do {
+                    w = m_d->write(m_buffer + m_written, m_read - m_written);
+                    AAssert(w >= 0, "Error while writing");
+                    m_written += w;
+                } while (m_read != m_written);
+                m_state = 0;
+                m_current += m_read;
+            }
+        }
+    }
+    catch (EAgain & e) {
+        taskSwitch();
+    }
+}
-- 
cgit v1.2.3