1 files changed, 66 insertions, 0 deletions
diff --git a/src/tls/tls_handshake_reader.cpp b/src/tls/tls_handshake_reader.cpp
new file mode 100644
index 000000000..a3fe48f71
--- /dev/null
+++ b/src/tls/tls_handshake_reader.cpp
@@ -0,0 +1,66 @@
+/*
+* TLS Handshake Reader
+* (C) 2012 Jack Lloyd
+*
+* Released under the terms of the Botan license
+*/
+
+#include <botan/internal/tls_handshake_reader.h>
+#include <botan/exceptn.h>
+
+namespace Botan {
+
+namespace TLS {
+
+void Stream_Handshake_Reader::add_input(const byte record[],
+                                        size_t record_size)
+   {
+   m_queue.write(record, record_size);
+   }
+
+bool Stream_Handshake_Reader::empty() const
+   {
+   return m_queue.empty();
+   }
+
+bool Stream_Handshake_Reader::have_full_record() const
+   {
+   if(m_queue.size() >= 4)
+      {
+      byte head[4] = { 0 };
+      m_queue.peek(head, 4);
+
+      const size_t length = make_u32bit(0, head[1], head[2], head[3]);
+
+      return (m_queue.size() >= length + 4);
+      }
+
+   return false;
+   }
+
+std::pair<Handshake_Type, std::vector<byte> > Stream_Handshake_Reader::get_next_record()
+   {
+   if(m_queue.size() >= 4)
+      {
+      byte head[4] = { 0 };
+      m_queue.peek(head, 4);
+
+      const size_t length = make_u32bit(0, head[1], head[2], head[3]);
+
+      if(m_queue.size() >= length + 4)
+         {
+         Handshake_Type type = static_cast<Handshake_Type>(head[0]);
+         std::vector<byte> contents(length);
+         m_queue.read(head, 4); // discard
+         m_queue.read(&contents[0], contents.size());
+
+         return std::make_pair(type, contents);
+         }
+      }
+
+   throw Internal_Error("Stream_Handshake_Reader::get_next_record called without a full record");
+   }
+
+}
+
+}