Fixed a minor bug, connections should poll properly now.

2 files changed, 6 insertions, 4 deletions

diff --git a/src/connectionmanager.cpp b/src/connectionmanager.cpp
index 367b9d7..a927092 100644
--- a/src/connectionmanager.cpp
+++ b/src/connectionmanager.cpp
@@ -195,7 +195,7 @@ bool ConnectionManager::scanConnections( int nTimeout, bool bForceTimeout )
                         lActive.erase( l );
                         continue;
                 }
-                (*i)->poll();
+                (*i)->getProtocol()->poll();
                 if( (*i)->hasOutput() )
                 {
                         (*i)->writeOutput();
@@ -328,7 +328,8 @@ bool ConnectionManager::addConnection( int nSocket )
 void ConnectionManager::connect(
         const char *lpAddress,
         int nPort,
-        int nProtocolPort
+        int nProtocolPort,
+        Protocol *pNewProto
         )
 {
         Connection *pCon = getInactiveConnection();
@@ -336,8 +337,9 @@ void ConnectionManager::connect(
         int nSocket = pCon->getSocket();
         FD_SET( nSocket, &fdActive );
 
+        pCon->setProtocol( pNewProto );
         pMonitor->onNewClientConnection( pCon, nProtocolPort );
-        pCon->getProtocol()->onNewClientConnection();
+        //pCon->getProtocol()->onNewClientConnection();
 
         lActive.insert( lActive.end(), pCon );
 }
diff --git a/src/connectionmanager.h b/src/connectionmanager.h
index 23d059a..e80119b 100644
--- a/src/connectionmanager.h
+++ b/src/connectionmanager.h
@@ -93,7 +93,7 @@ public:
           */
         void setConnectionMonitor( ConnectionMonitor *pNewMonitor );
 
-        void connect( const char *lpAddress, int nPort, int nProtocolPort );
+        void connect( const char *lpAddress, int nPort, int nProtocolPort, Protocol *pNewProto );
 
 private:
         /**