Add classes for reading and writing files

1 files changed, 122 insertions, 0 deletions

diff --git a/src/io/read.cpp b/src/io/read.cpp
new file mode 100644
index 0000000..fad40bb
--- /dev/null
+++ b/src/io/read.cpp
@@ -0,0 +1,122 @@
+#include "read.hpp"
+//    io/read.cpp - Input from files
+//
+//    Copyright © 2013 Ben Longbons <b.r.longbons@gmail.com>
+//
+//    This file is part of The Mana World (Athena server)
+//
+//    This program is free software: you can redistribute it and/or modify
+//    it under the terms of the GNU General Public License as published by
+//    the Free Software Foundation, either version 3 of the License, or
+//    (at your option) any later version.
+//
+//    This program is distributed in the hope that it will be useful,
+//    but WITHOUT ANY WARRANTY; without even the implied warranty of
+//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+//    GNU General Public License for more details.
+//
+//    You should have received a copy of the GNU General Public License
+//    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+#include <fcntl.h>
+#include <unistd.h>
+
+#include "../strings/fstring.hpp"
+#include "../strings/mstring.hpp"
+#include "../strings/zstring.hpp"
+
+#include "../common/cxxstdio.hpp"
+
+#include "src/poison.hpp"
+
+
+namespace io
+{
+    ReadFile::ReadFile(int f)
+    : fd(f), start(0), end(0)
+    {
+    }
+    ReadFile::ReadFile(ZString name)
+    : fd(open(name.c_str(), O_RDONLY | O_CLOEXEC)), start(0), end(0)
+    {
+    }
+    ReadFile::~ReadFile()
+    {
+        close(fd);
+        fd = -1;
+    }
+
+    bool ReadFile::get(char& c)
+    {
+        if (start == end)
+        {
+            if (fd == -1)
+                return false;
+            ssize_t rv = read(fd, &buf, sizeof(buf));
+            if (rv == 0 || rv == -1)
+            {
+                close(fd);
+                fd = -1;
+                return false;
+            }
+            start = 0;
+            end = rv;
+        }
+        c = buf[start++];
+        return true;
+    }
+    size_t ReadFile::get(char *out, size_t len)
+    {
+        for (size_t i = 0; i < len; ++i)
+        {
+            if (!get(out[i]))
+                return i;
+        }
+        return len;
+    }
+    bool ReadFile::getline(FString& line)
+    {
+        MString tmp;
+        char c;
+        bool anything = false;
+        bool happy = false;
+        bool unhappy = false;
+        while (get(c))
+        {
+            anything = true;
+            if (c == '\r')
+            {
+                unhappy = true;
+                if (!get(c))
+                    break;
+                if (c != '\n')
+                    --start;
+                else
+                    happy = true;
+                break;
+            }
+            if (c == '\n')
+            {
+                happy = true;
+                break;
+            }
+            tmp += c;
+        }
+        if (unhappy)
+        {
+            if (happy)
+                PRINTF("warning: file contains CR\n");
+            else
+                PRINTF("warning: file contains bare CR\n");
+        }
+        else if (!happy && anything)
+            PRINTF("warning: file does not contain a trailing newline\n");
+        line = FString(tmp);
+        return anything;
+    }
+
+    bool ReadFile::is_open()
+    {
+        return fd != -1;
+    }
+} // namespace io