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, 115 insertions, 0 deletions

diff --git a/src/gatscon/proxywidget.cpp b/src/gatscon/proxywidget.cpp
new file mode 100644
index 0000000..28bd1fc
--- /dev/null
+++ b/src/gatscon/proxywidget.cpp
@@ -0,0 +1,115 @@
+#include "proxywidget.h"
+#include "proxythread.h"
+
+#include "gatstotree.h"
+
+#include <QMessageBox>
+
+#include <gats/types.h>
+#include <bu/sio.h>
+
+using namespace Bu;
+
+ProxyWidget::ProxyWidget( QWidget *pParent, int iPortIn,
+                const QByteArray baHost, int iPortOut ) :
+        QWidget( pParent ),
+        pPrx( NULL )
+{
+        setupUi( this );
+
+        pPrx = new ProxyThread( this, iPortIn, baHost, iPortOut );
+
+        connect( pPrx, SIGNAL(gotConnection()),
+                this, SLOT(gotConnection()), Qt::QueuedConnection );
+        connect( pPrx, SIGNAL(recv( Gats::Object *)),
+                this, SLOT(clientRecv(Gats::Object *)), Qt::QueuedConnection );
+        connect( pPrx->pHost, SIGNAL(recv( Gats::Object *)),
+                this, SLOT(hostRecv(Gats::Object *)), Qt::QueuedConnection );
+
+        pPrx->start();
+}
+
+ProxyWidget::~ProxyWidget()
+{
+}
+
+void ProxyWidget::sendToClient()
+{
+        try
+        {
+                Gats::Object *pObj = Gats::Object::strToGats(
+                        leGats->text().toAscii().constData()
+                        );
+                sio << "Send: " << *pObj << sio.nl;
+                QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                        twHistory->invisibleRootItem()
+                        );
+                pIt->setText( 0, "proxy -> client" );
+                gatsToTree( pIt, pObj );
+                pPrx->send( pObj );
+                delete pObj;
+
+                leGats->setText("");
+                leGats->setFocus();
+        }
+        catch( Bu::ExceptionBase &e )
+        {
+                QMessageBox::critical( this, "GatsCon - Error", e.what() );
+        }
+}
+
+void ProxyWidget::sendToServer()
+{
+        try
+        {
+                Gats::Object *pObj = Gats::Object::strToGats(
+                        leGats->text().toAscii().constData()
+                        );
+                sio << "Send: " << *pObj << sio.nl;
+                QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                        twHistory->invisibleRootItem()
+                        );
+                pIt->setText( 0, "proxy -> host" );
+                gatsToTree( pIt, pObj );
+                pPrx->pHost->send( pObj );
+                delete pObj;
+
+                leGats->setText("");
+                leGats->setFocus();
+        }
+        catch( Bu::ExceptionBase &e )
+        {
+                QMessageBox::critical( this, "GatsCon - Error", e.what() );
+        }
+}
+
+void ProxyWidget::clientRecv( Gats::Object *pObj )
+{
+        sio << "Recv: " << *pObj << sio.nl;
+
+        QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                twHistory->invisibleRootItem()
+                );
+        pIt->setText( 0, "client -> host" );
+        gatsToTree( pIt, pObj );
+        delete pObj;
+}
+
+void ProxyWidget::hostRecv( Gats::Object *pObj )
+{
+        sio << "Recv: " << *pObj << sio.nl;
+
+        QTreeWidgetItem *pIt = new QTreeWidgetItem(
+                twHistory->invisibleRootItem()
+                );
+        pIt->setText( 0, "host -> client" );
+        gatsToTree( pIt, pObj );
+        delete pObj;
+}
+
+void ProxyWidget::gotConnection()
+{
+        lwConnect->stop();
+        swRoot->setCurrentIndex( 1 );
+}
+