+++ /dev/null
-/*
- * 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