1 files changed, 126 insertions, 0 deletions
diff --git a/libktorrent/torrent/serverauthenticate.cpp b/libktorrent/torrent/serverauthenticate.cpp
new file mode 100644
index 0000000..479f0ce
--- /dev/null
+++ b/libktorrent/torrent/serverauthenticate.cpp
@@ -0,0 +1,126 @@
+/***************************************************************************
+ *   Copyright (C) 2005 by Joris Guisson                                   *
+ *   joris.guisson@gmail.com                                               *
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License as published by  *
+ *   the Free Software Foundation; either version 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ *   This program is distributed in the hope that it will be useful,       *
+ *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
+ *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
+ *   GNU General Public License for more details.                          *
+ *                                                                         *
+ *   You should have received a copy of the GNU General Public License     *
+ *   along with this program; if not, write to the                         *
+ *   Free Software Foundation, Inc.,                                       *
+ *   51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.             *
+ ***************************************************************************/
+#include <mse/streamsocket.h>
+#include <util/sha1hash.h>
+#include <util/log.h>
+#include <util/log.h>
+#include "globals.h"
+#include "server.h"
+#include "peermanager.h"
+#include "serverauthenticate.h"
+#include "peerid.h"
+#include "torrent.h"
+#include "ipblocklist.h"
+
+
+namespace bt
+{
+	bool ServerAuthenticate::s_firewalled = true;
+
+
+	ServerAuthenticate::ServerAuthenticate(mse::StreamSocket* sock,Server* server)
+	: AuthenticateBase(sock),server(server)
+	{
+	}
+
+
+	ServerAuthenticate::~ServerAuthenticate()
+	{
+	}
+	
+	
+	void ServerAuthenticate::onFinish(bool succes)
+	{		
+		Out(SYS_CON|LOG_NOTICE) << "Authentication(S) to " << sock->getRemoteIPAddress() 
+				<< " : " << (succes ? "ok" : "failure") << endl;
+		finished = true;
+		setFirewalled(false);
+		
+		if (!succes)
+		{
+			sock->deleteLater();
+			sock = 0;
+		}
+		
+		timer.stop();
+	}
+
+	void ServerAuthenticate::handshakeRecieved(bool full)
+	{
+		Uint8* hs = handshake;
+		IPBlocklist& ipfilter = IPBlocklist::instance();
+
+		QString IP = sock->getRemoteIPAddress();
+
+		if (ipfilter.isBlocked( IP ))
+		{
+			onFinish(false);
+			return;
+		}
+		
+		// try to find a PeerManager which has te right info hash
+		SHA1Hash rh(hs+28);
+		PeerManager* pman = server->findPeerManager(rh);
+		if (!pman)
+		{
+			Out(SYS_GEN|LOG_DEBUG) << "Cannot find PeerManager for hash : " << rh.toString() << endl;
+			onFinish(false);
+			return;
+		}
+
+		if (full)
+		{
+			// check if we aren't connecting to ourself
+			char tmp[21];
+			tmp[20] = '\0';
+			memcpy(tmp,hs+48,20);
+			PeerID peer_id = PeerID(tmp);
+			if (pman->getTorrent().getPeerID() == peer_id)
+			{
+				Out(SYS_CON|LOG_NOTICE) << "Lets not connect to our self" << endl;
+				onFinish(false);
+				return;
+			}
+			
+			// check if we aren't already connected to the client
+			if (pman->connectedTo(peer_id))
+			{
+				Out(SYS_CON|LOG_NOTICE) << "Already connected to " << peer_id.toString() << endl;
+				onFinish(false);
+				return;
+			}
+			
+					
+			// send handshake and finish off
+			sendHandshake(rh,pman->getTorrent().getPeerID());
+			onFinish(true);
+			// hand over connection
+			pman->newConnection(sock,peer_id,supportedExtensions());
+			sock = 0;
+		}
+		else
+		{
+			// if the handshake wasn't fully received just send our handshake
+			sendHandshake(rh,pman->getTorrent().getPeerID());
+		}
+	}
+}
+
+#include "serverauthenticate.moc"