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/HelperTasks.cc | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 src/HelperTasks.cc

(limited to 'src/HelperTasks.cc')

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