GatsCon now supports proxying, and way better than qtbenc ever did. Not only

does it proxy, but you can inject your own messages going to the client or host.

Pretty slick, really.  Next up, reading and creating files.

1 files changed, 108 insertions, 0 deletions

diff --git a/src/gatscon/proxythread.cpp b/src/gatscon/proxythread.cpp
new file mode 100644
index 0000000..574b56b
--- /dev/null
+++ b/src/gatscon/proxythread.cpp
@@ -0,0 +1,108 @@
+#include "proxythread.h"
+
+#include <gats/types.h>
+
+#include <bu/tcpsocket.h>
+#include <bu/tcpserversocket.h>
+#include <bu/sio.h>
+#include <bu/membuf.h>
+
+using namespace Bu;
+
+ProxyThread::ProxyThread( QObject *pParent, int iPortIn,
+                const QByteArray &baHostOut, int iPortOut ) :
+        QThread( pParent ),
+        pHost( NULL ),
+        iPortIn( iPortIn ),
+        baHostOut( baHostOut ),
+        iPortOut( iPortOut ),
+        gsCli( ssCli )
+{
+        pHost = new ProxyHostThread( pParent, this );
+}
+
+ProxyThread::~ProxyThread()
+{
+}
+
+void ProxyThread::send( Gats::Object *pObj )
+{
+        MemBuf bg;
+        Gats::GatsStream gs( bg );
+        gs.writeObject( pObj );
+        ssCli.write( bg.getString().getStr(), bg.getString().getSize() );
+}
+
+void ProxyThread::run()
+{
+        int iSockIn;
+
+        {
+                TcpServerSocket tsIn( iPortIn );
+                do
+                {
+                        iSockIn = tsIn.accept( 5 );
+                } while( iSockIn < 0 );
+        }
+
+        emit gotConnection();
+
+        ssCli.setStream( new TcpSocket( iSockIn ) );
+        ssCli.setBlocking( true );
+
+        pHost->setStream(
+                new TcpSocket( baHostOut.constData(), iPortOut )
+                );
+
+        pHost->start();
+
+        while( !ssCli.isEos() )
+        {
+                Gats::Object *pObj = gsCli.readObject();
+                if( pObj == NULL )
+                        continue;
+
+                pHost->send( pObj );
+                emit recv( pObj );
+        }
+
+}
+
+ProxyHostThread::ProxyHostThread( QObject *pParent, ProxyThread *pClient ) :
+        QThread( pParent ),
+        pClient( pClient ),
+        ssHst(),
+        gsHst( ssHst )
+{
+}
+
+ProxyHostThread::~ProxyHostThread()
+{
+}
+
+void ProxyHostThread::send( Gats::Object *pObj )
+{
+        MemBuf bg;
+        Gats::GatsStream gs( bg );
+        gs.writeObject( pObj );
+        ssHst.write( bg.getString().getStr(), bg.getString().getSize() );
+}
+
+void ProxyHostThread::setStream( Bu::Stream *pStr )
+{
+        ssHst.setStream( pStr );
+}
+
+void ProxyHostThread::run()
+{
+        while( !ssHst.isEos() )
+        {
+                Gats::Object *pObj = gsHst.readObject();
+                if( pObj == NULL )
+                        continue;
+
+                pClient->send( pObj );
+                emit recv( pObj );
+        }
+}
+