3 files changed, 106 insertions, 10 deletions

diff --git a/src/client.cpp b/src/client.cpp
index 43d1a08..24f9bfb 100644
--- a/src/client.cpp
+++ b/src/client.cpp
@@ -17,18 +17,23 @@
 #define RBS             (1024*2)
 
 Bu::Client::Client( Bu::Socket *pSocket, class Bu::ClientLinkFactory *pfLink ) :
+        pTopStream( pSocket ),
         pSocket( pSocket ),
         pProto( NULL ),
         nRBOffset( 0 ),
         bWantsDisconnect( false ),
         pfLink( pfLink )
 {
+        lFilts.prepend( pSocket );
 }
 
 Bu::Client::~Client()
 {
-        delete pSocket;
-        pSocket = NULL;
+        for( FilterList::iterator i = lFilts.begin(); i; i++ )
+        {
+                delete *i;
+        }
+        pTopStream = pSocket = NULL;
 }
 
 void Bu::Client::processInput()
@@ -40,7 +45,7 @@ void Bu::Client::processInput()
         {
                 try
                 {
-                        nRead = pSocket->read( buf, RBS );
+                        nRead = pTopStream->read( buf, RBS );
 
                         if( nRead == 0 )
                         {
@@ -50,13 +55,13 @@ void Bu::Client::processInput()
                         {
                                 nTotal += nRead;
                                 sReadBuf.append( buf, nRead );
-                                if( !pSocket->canRead() )
+                                if( !pTopStream->canRead() )
                                         break;
                         }
                 }
                 catch( Bu::SocketException &e )
                 {
-                        pSocket->close();
+                        pTopStream->close();
                         bWantsDisconnect = true;
                         break;
                 }
@@ -64,7 +69,7 @@ void Bu::Client::processInput()
 
         if( nTotal == 0 )
         {
-                pSocket->close();
+                pTopStream->close();
                 bWantsDisconnect = true;
         }
 
@@ -79,7 +84,7 @@ void Bu::Client::processOutput()
         if( sWriteBuf.getSize() > 0 )
         {
                 int nAmnt = (sWriteBuf.getSize()<2048)?(sWriteBuf.getSize()):(2048);
-                int nReal = pSocket->write( sWriteBuf.getStr(), nAmnt );
+                int nReal = pTopStream->write( sWriteBuf.getStr(), nAmnt );
                 sWriteBuf.trimFront( nReal );
                 //sWriteBuf.clear();
         }
@@ -113,8 +118,8 @@ Bu::FString &Bu::Client::getOutput()
 
 bool Bu::Client::isOpen()
 {
-        if( !pSocket ) return false;
-        return pSocket->isOpen();
+        if( !pTopStream ) return false;
+        return pTopStream->isOpen();
 }
 
 void Bu::Client::write( const Bu::FString &sData )
@@ -235,7 +240,7 @@ bool Bu::Client::wantsDisconnect()
 
 void Bu::Client::close()
 {
-        pSocket->close();
+        pTopStream->close();
 }
 
 Bu::ClientLink *Bu::Client::getLink()
diff --git a/src/client.h b/src/client.h
index b11774e..7ba1ac5 100644
--- a/src/client.h
+++ b/src/client.h
@@ -66,7 +66,18 @@ namespace Bu
 
                 bool hasOutput() { return !sWriteBuf.isEmpty(); }
 
+                template<typename filter>
+                void pushFilter()
+                {
+                        filter *pFlt = new filter( *pTopStream );
+                        pTopStream = pFlt;
+                        lFilts.prepend( pFlt );
+                }
+
         private:
+                typedef Bu::List<Bu::Stream *> FilterList;
+                FilterList lFilts;
+                Bu::Stream *pTopStream;
                 Bu::Socket *pSocket;
                 Bu::Protocol *pProto;
                 Bu::FString sReadBuf;
diff --git a/src/tests/rot13.cpp b/src/tests/rot13.cpp
new file mode 100644
index 0000000..e677440
--- /dev/null
+++ b/src/tests/rot13.cpp
@@ -0,0 +1,80 @@
+#include "bu/protocol.h"
+#include "bu/multiserver.h"
+#include "bu/client.h"
+#include "bu/filter.h"
+
+using namespace Bu;
+
+class Rot13Filter : public Filter
+{
+public:
+        Rot13Filter( Stream &next ) :
+                Filter( next )
+        {
+        }
+
+        virtual ~Rot13Filter()
+        {
+        }
+
+        virtual void start()
+        {
+        }
+
+        virtual size_t stop()
+        {
+                return 0;
+        }
+
+        virtual size_t read( void *pBuf, size_t nBytes )
+        {
+                return rNext.read( pBuf, nBytes );
+        }
+        
+        virtual size_t write( const void *pBuf, size_t nBytes )
+        {
+                const char *cBuf = (const char *)pBuf;
+                char *buf = new char[nBytes];
+                for( size_t j = 0; j < nBytes; j++ )
+                {
+                        if( cBuf[j] >= 'a' && cBuf[j] <= 'z' )
+                                buf[j] = (cBuf[j]-'a'+13)%26+'a';
+                        else if( cBuf[j] >= 'A' && cBuf[j] <= 'Z' )
+                                buf[j] = (cBuf[j]-'A'+13)%26+'A';
+                        else
+                                buf[j] = cBuf[j];
+                }
+                rNext.write( buf, nBytes );
+                delete[] buf;
+                return nBytes;
+        }
+};
+
+class Rot13Protocol : public Protocol
+{
+public:
+        void onNewConnection( Bu::Client *pClient )
+        {
+                pClient->pushFilter<Rot13Filter>();
+        }
+
+        void onNewData( Bu::Client *pClient )
+        {
+                pClient->write( pClient->getInput().getStr(), pClient->getInputSize() );
+                pClient->seek( pClient->getInputSize() );
+        }
+};
+
+int main()
+{
+        MultiServer xSrv;
+
+        xSrv.setTimeout( 5 );
+        xSrv.addProtocol( genProtocol<Rot13Protocol>, 9999 );
+
+        for(;;)
+                xSrv.scan();
+
+        return 0;
+}
+