From 1a5dacb93f2c3f2cfc8f4a9da6599beaebcc6845 Mon Sep 17 00:00:00 2001
From: Pixel <pixel@nobis-crew.org>
Date: Tue, 15 Nov 2011 11:53:58 -0800
Subject: Adding the forceRead/Write methods, to circumvent the potential
 problem that a file descriptor can do less than being asked.

---
 src/Handle.cc | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

(limited to 'src/Handle.cc')

diff --git a/src/Handle.cc b/src/Handle.cc
index 0ec39be..8fa0417 100644
--- a/src/Handle.cc
+++ b/src/Handle.cc
@@ -83,6 +83,56 @@ ssize_t Balau::Handle::write(const void * buf, size_t count) throw (GeneralExcep
     return -1;
 }
 
+ssize_t Balau::Handle::forceRead(void * _buf, size_t count) throw (GeneralException) {
+    ssize_t total;
+    uint8_t * buf = (uint8_t *) _buf;
+    if (!canRead())
+        throw GeneralException("Handle can't read");
+
+    while (count && !isClosed()) {
+        ssize_t r;
+        try {
+            r = read(buf, count);
+        }
+        catch (EAgain e) {
+            Task::yield(e.getEvent());
+            continue;
+        }
+        if (r < 0)
+            return r;
+        total += r;
+        count -= r;
+        buf += r;
+    }
+
+    return total;
+}
+
+ssize_t Balau::Handle::forceWrite(const void * _buf, size_t count) throw (GeneralException) {
+    ssize_t total;
+    const uint8_t * buf = (const uint8_t *) _buf;
+    if (!canWrite())
+        throw GeneralException("Handle can't write");
+
+    while (count && !isClosed()) {
+        ssize_t r;
+        try {
+            r = write(buf, count);
+        }
+        catch (EAgain e) {
+            Task::yield(e.getEvent());
+            continue;
+        }
+        if (r < 0)
+            return r;
+        total += r;
+        count -= r;
+        buf += r;
+    }
+
+    return total;
+}
+
 void Balau::Handle::rseek(off_t offset, int whence) throw (GeneralException) {
     if (canSeek())
         throw GeneralException(String("Handle ") + getName() + " can seek, but rseek() not implemented (missing in class " + ClassName(this).c_str() + ")");
-- 
cgit v1.2.3