X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/65a51d11c64daa4df3d4388df60bb4efc3e67e14..1538d6276533140505fddb71c99a0bafe6ca9182:/projects/stargazer/plugins/other/radius/reader.h

diff --git a/projects/stargazer/plugins/other/radius/reader.h b/projects/stargazer/plugins/other/radius/reader.h
new file mode 100644
index 00000000..3bc1f9e6
--- /dev/null
+++ b/projects/stargazer/plugins/other/radius/reader.h
@@ -0,0 +1,170 @@
+/*
+ *    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 2 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, write to the Free Software
+ *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+/*
+ *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
+ */
+
+#ifndef __STG_SGCONFIG_READER_H__
+#define __STG_SGCONFIG_READER_H__
+
+#include <string>
+#include <vector>
+#include <utility>
+
+namespace STG
+{
+
+struct BaseReader
+{
+    virtual ~BaseReader() {}
+
+    virtual ssize_t read(TransportProto& proto) = 0;
+    virtual bool done() const = 0;
+};
+
+template <typename T>
+class Reader : public BaseReader
+{
+    public:
+        Reader(size_t size = sizeof(T)) : m_size(size), m_done(0) {}
+
+        virtual ssize_t read(TransportProto& proto)
+        {
+            char* pos = static_cast<void*>(&m_dest);
+            pos += m_done;
+            ssize_t res = proto.read(pos, m_size - m_done);
+            if (res < 0)
+                return res;
+            if (res == 0) {
+                m_done = m_size;
+                return 0;
+            }
+            m_done += res;
+            return res;
+        }
+
+        virtual bool done() const { return m_done == m_size; }
+
+        T get() const { return ntoh(m_dest); }
+
+    private:
+        T m_dest;
+        size_t m_size;
+        size_t m_done;
+
+};
+
+template <>
+class Reader<std::vector<Reader*> > : public BaseReader
+{
+    public:
+        Reader(const std::vector<Reader*>& readers) : m_size(readers.size()), m_done(0) {}
+
+        virtual ssize_t read(TransportProto& proto)
+        {
+            if (m_size == 0)
+                return 0;
+            size_t res = m_dest[m_done]->read(proto);
+            if (res < 0)
+                return res;
+            if (res == 0) {
+                m_done = m_size;
+                return 0;
+            }
+            if (m_dest[m_done].done())
+                ++m_dest;
+            return res;
+        }
+
+        virtual bool done() const { return m_done == m_size; }
+
+        const T& get() const { return m_dest; }
+
+    private:
+        T m_dest;
+        size_t m_size;
+        size_t m_done;
+
+};
+
+template <>
+class Reader<std::vector<char> > : public BaseReader
+{
+    public:
+        Reader(size_t size ) : m_dest(size), m_size(size), m_done(0) {}
+
+        virtual ssize_t read(TransportProto& proto)
+        {
+            char* pos = static_cast<void*>(m_dest.data());
+            pos += m_done;
+            ssize_t res = proto.read(pos, m_size - m_done);
+            if (res < 0)
+                return res;
+            if (res == 0) {
+                m_done = m_size;
+                return 0;
+            }
+            m_done += res;
+            return res;
+        }
+
+        virtual bool done() const { return m_done == m_size; }
+
+        const std::vector<char>& get() const { return m_dest; }
+
+    private:
+        std::vector<char> m_dest;
+        size_t m_size;
+        size_t m_done;
+
+};
+
+template <>
+class Reader<std::string>
+{
+    public:
+        Reader() : m_dest(Reader<std::string>::initDest()) {}
+
+        virtual ssize_t read(TransportProto& proto)
+        {
+            if (m_size == 0)
+                return 0;
+            size_t res = m_dest[m_done]->read(proto);
+            if (res < 0)
+                return res;
+            if (res == 0) {
+                m_done = m_size;
+                return 0;
+            }
+            if (m_dest[m_done].done())
+                ++m_dest;
+            return res;
+        }
+
+        virtual bool done() const { return m_done == m_size; }
+
+        const T& get() const { return m_dest; }
+
+    private:
+        T m_dest;
+        size_t m_size;
+        size_t m_done;
+};
+
+} // namespace STG
+
+#endif