Initial checkin1.5.6

1 files changed, 284 insertions, 0 deletions

diff --git a/src/pipe.cpp b/src/pipe.cpp
new file mode 100644
index 000000000..8e1f46f00
--- /dev/null
+++ b/src/pipe.cpp
@@ -0,0 +1,284 @@
+/*************************************************
+* Pipe Source File                               *
+* (C) 1999-2006 The Botan Project                *
+*************************************************/
+
+#include <botan/pipe.h>
+#include <botan/out_buf.h>
+#include <botan/secqueue.h>
+
+namespace Botan {
+
+namespace {
+
+/*************************************************
+* A Filter that does nothing                     *
+*************************************************/
+class Null_Filter : public Filter
+   {
+   public:
+      void write(const byte input[], u32bit length)
+         { send(input, length); }
+   };
+
+}
+
+/*************************************************
+* Pipe Constructor                               *
+*************************************************/
+Pipe::Pipe(Filter* f1, Filter* f2, Filter* f3, Filter* f4)
+   {
+   init();
+   append(f1);
+   append(f2);
+   append(f3);
+   append(f4);
+   }
+
+/*************************************************
+* Pipe Constructor                               *
+*************************************************/
+Pipe::Pipe(Filter* filter_array[], u32bit count)
+   {
+   init();
+   for(u32bit j = 0; j != count; ++j)
+      append(filter_array[j]);
+   }
+
+/*************************************************
+* Pipe Destructor                                *
+*************************************************/
+Pipe::~Pipe()
+   {
+   destruct(pipe);
+   delete outputs;
+   }
+
+/*************************************************
+* Initialize the Pipe                            *
+*************************************************/
+void Pipe::init()
+   {
+   outputs = new Output_Buffers;
+   pipe = 0;
+   default_read = 0;
+   inside_msg = false;
+   }
+
+/*************************************************
+* Reset the Pipe                                 *
+*************************************************/
+void Pipe::reset()
+   {
+   if(inside_msg)
+      throw Invalid_State("Pipe cannot be reset while it is processing");
+   destruct(pipe);
+   pipe = 0;
+   inside_msg = false;
+   }
+
+/*************************************************
+* Destroy the Pipe                               *
+*************************************************/
+void Pipe::destruct(Filter* to_kill)
+   {
+   if(!to_kill || dynamic_cast<SecureQueue*>(to_kill))
+      return;
+   for(u32bit j = 0; j != to_kill->total_ports(); ++j)
+      destruct(to_kill->next[j]);
+   delete to_kill;
+   }
+
+/*************************************************
+* Test if the Pipe has any data in it            *
+*************************************************/
+bool Pipe::end_of_data() const
+   {
+   return (remaining() == 0);
+   }
+
+/*************************************************
+* Set the default read message                   *
+*************************************************/
+void Pipe::set_default_msg(u32bit msg)
+   {
+   if(msg >= message_count())
+      throw Invalid_Argument("Pipe::set_default_msg: msg number is too high");
+   default_read = msg;
+   }
+
+/*************************************************
+* Process a full message at once                 *
+*************************************************/
+void Pipe::process_msg(const byte input[], u32bit length)
+   {
+   start_msg();
+   write(input, length);
+   end_msg();
+   }
+
+/*************************************************
+* Process a full message at once                 *
+*************************************************/
+void Pipe::process_msg(const MemoryRegion<byte>& input)
+   {
+   process_msg(input.begin(), input.size());
+   }
+
+/*************************************************
+* Process a full message at once                 *
+*************************************************/
+void Pipe::process_msg(const std::string& input)
+   {
+   process_msg((const byte*)input.c_str(), input.length());
+   }
+
+/*************************************************
+* Process a full message at once                 *
+*************************************************/
+void Pipe::process_msg(DataSource& input)
+   {
+   start_msg();
+   write(input);
+   end_msg();
+   }
+
+/*************************************************
+* Start a new message                            *
+*************************************************/
+void Pipe::start_msg()
+   {
+   if(inside_msg)
+      throw Invalid_State("Pipe::start_msg: Message was already started");
+   if(pipe == 0)
+      pipe = new Null_Filter;
+   find_endpoints(pipe);
+   pipe->new_msg();
+   inside_msg = true;
+   }
+
+/*************************************************
+* End the current message                        *
+*************************************************/
+void Pipe::end_msg()
+   {
+   if(!inside_msg)
+      throw Invalid_State("Pipe::end_msg: Message was already ended");
+   pipe->finish_msg();
+   clear_endpoints(pipe);
+   if(dynamic_cast<Null_Filter*>(pipe))
+      {
+      delete pipe;
+      pipe = 0;
+      }
+   inside_msg = false;
+
+   outputs->retire();
+   }
+
+/*************************************************
+* Find the endpoints of the Pipe                 *
+*************************************************/
+void Pipe::find_endpoints(Filter* f)
+   {
+   for(u32bit j = 0; j != f->total_ports(); ++j)
+      if(f->next[j] && !dynamic_cast<SecureQueue*>(f->next[j]))
+         find_endpoints(f->next[j]);
+      else
+         {
+         SecureQueue* q = new SecureQueue;
+         f->next[j] = q;
+         outputs->add(q);
+         }
+   }
+
+/*************************************************
+* Remove the SecureQueues attached to the Filter *
+*************************************************/
+void Pipe::clear_endpoints(Filter* f)
+   {
+   if(!f) return;
+   for(u32bit j = 0; j != f->total_ports(); ++j)
+      {
+      if(f->next[j] && dynamic_cast<SecureQueue*>(f->next[j]))
+         f->next[j] = 0;
+      clear_endpoints(f->next[j]);
+      }
+   }
+
+/*************************************************
+* Append a Filter to the Pipe                    *
+*************************************************/
+void Pipe::append(Filter* filter)
+   {
+   if(inside_msg)
+      throw Invalid_State("Cannot append to a Pipe while it is processing");
+   if(!filter)
+      return;
+   if(dynamic_cast<SecureQueue*>(filter))
+      throw Invalid_Argument("Pipe::append: SecureQueue cannot be used");
+   if(filter->owned)
+      throw Invalid_Argument("Filters cannot be shared among multiple Pipes");
+
+   filter->owned = true;
+
+   if(!pipe) pipe = filter;
+   else      pipe->attach(filter);
+   }
+
+/*************************************************
+* Prepend a Filter to the Pipe                   *
+*************************************************/
+void Pipe::prepend(Filter* filter)
+   {
+   if(inside_msg)
+      throw Invalid_State("Cannot prepend to a Pipe while it is processing");
+   if(!filter)
+      return;
+   if(dynamic_cast<SecureQueue*>(filter))
+      throw Invalid_Argument("Pipe::prepend: SecureQueue cannot be used");
+   if(filter->owned)
+      throw Invalid_Argument("Filters cannot be shared among multiple Pipes");
+
+   filter->owned = true;
+
+   if(pipe) filter->attach(pipe);
+   pipe = filter;
+   }
+
+/*************************************************
+* Pop a Filter off the Pipe                      *
+*************************************************/
+void Pipe::pop()
+   {
+   if(inside_msg)
+      throw Invalid_State("Cannot pop off a Pipe while it is processing");
+
+   if(!pipe)
+      return;
+
+   if(pipe->total_ports() > 1)
+      throw Invalid_State("Cannot pop off a Filter with multiple ports");
+
+   Filter* f = pipe;
+   u32bit owns = f->owns();
+   pipe = pipe->next[0];
+   delete f;
+
+   while(owns--)
+      {
+      f = pipe;
+      pipe = pipe->next[0];
+      delete f;
+      }
+   }
+
+/*************************************************
+* Return the number of messages in this Pipe     *
+*************************************************/
+u32bit Pipe::message_count() const
+   {
+   return outputs->message_count();
+   }
+
+}