1 files changed, 78 insertions, 0 deletions
diff --git a/src/core/data_src.h b/src/core/data_src.h
new file mode 100644
index 000000000..f7285e8fb
--- /dev/null
+++ b/src/core/data_src.h
@@ -0,0 +1,78 @@
+/*************************************************
+* DataSource Header File                         *
+* (C) 1999-2007 Jack Lloyd                       *
+*************************************************/
+
+#ifndef BOTAN_DATA_SRC_H__
+#define BOTAN_DATA_SRC_H__
+
+#include <botan/base.h>
+#include <iosfwd>
+
+namespace Botan {
+
+/*************************************************
+* Generic DataSource Interface                   *
+*************************************************/
+class BOTAN_DLL DataSource
+   {
+   public:
+      virtual u32bit read(byte[], u32bit) = 0;
+      virtual u32bit peek(byte[], u32bit, u32bit) const = 0;
+      virtual bool end_of_data() const = 0;
+      virtual std::string id() const { return ""; }
+
+      u32bit read_byte(byte&);
+      u32bit peek_byte(byte&) const;
+      u32bit discard_next(u32bit);
+
+      DataSource() {}
+      virtual ~DataSource() {}
+   private:
+      DataSource& operator=(const DataSource&) { return (*this); }
+      DataSource(const DataSource&);
+   };
+
+/*************************************************
+* Memory-Based DataSource                        *
+*************************************************/
+class BOTAN_DLL DataSource_Memory : public DataSource
+   {
+   public:
+      u32bit read(byte[], u32bit);
+      u32bit peek(byte[], u32bit, u32bit) const;
+      bool end_of_data() const;
+
+      DataSource_Memory(const std::string&);
+      DataSource_Memory(const byte[], u32bit);
+      DataSource_Memory(const MemoryRegion<byte>&);
+   private:
+      SecureVector<byte> source;
+      u32bit offset;
+   };
+
+/*************************************************
+* Stream-Based DataSource                        *
+*************************************************/
+class BOTAN_DLL DataSource_Stream : public DataSource
+   {
+   public:
+      u32bit read(byte[], u32bit);
+      u32bit peek(byte[], u32bit, u32bit) const;
+      bool end_of_data() const;
+      std::string id() const;
+
+      DataSource_Stream(std::istream&, const std::string& id = "");
+      DataSource_Stream(const std::string&, bool = false);
+      ~DataSource_Stream();
+   private:
+      const std::string identifier;
+      const bool owner;
+
+      std::istream* source;
+      u32bit total_read;
+   };
+
+}
+
+#endif