The client now downloads a zipped server list file from pokerth.net. This is parsed and the first server entry is used for now. Error handling still missing. boost::iostreams now required.

2008-04-19 11:48:09 +00:00
parent 17a7a19298
commit 7a157b682a
10 changed files with 380 additions and 11 deletions
@@ -27,17 +27,27 @@
 #include <net/clientexception.h>
 #include <net/socket_helper.h>
 #include <net/socket_msg.h>
+#include <net/downloadhelper.h>
 #include <core/avatarmanager.h>
 #include <qttoolsinterface.h>

 #include <game.h>
 #include <playerinterface.h>

+#include "tinyxml.h"
+#include "zlib.h"
 #include <boost/bind.hpp>
+#include <boost/iostreams/filtering_streambuf.hpp>
+#include <boost/iostreams/copy.hpp>
+#include <boost/iostreams/filter/zlib.hpp>
+#include <boost/filesystem.hpp>

+#include <iostream>
+#include <fstream>
 #include <sstream>

 using namespace std;
+using namespace boost::filesystem;

 #define CLIENT_WAIT_TIMEOUT_MSEC	50
 #define CLIENT_CONNECT_TIMEOUT_SEC	10
@@ -92,7 +102,9 @@ ClientStateInit::Process(ClientThread &client)
 	setsockopt(context.GetSocket(), SOL_SOCKET, SO_NOSIGPIPE, (char *)&nosigpipe, sizeof(nosigpipe));
 #endif

-	client.SetState(ClientStateStartResolve::Instance());
+	// TODO
+	//client.SetState(ClientStateStartResolve::Instance());
+	client.SetState(ClientStateStartServerListDownload::Instance());

 	return MSG_SOCK_INIT_DONE;
 }
@@ -117,7 +129,7 @@ ClientStateStartResolve::~ClientStateStartResolve()
 int
 ClientStateStartResolve::Process(ClientThread &client)
 {
-	int retVal;
+	int retVal = MSG_SOCK_INTERNAL_PENDING;

 	ClientContext &context = client.GetContext();

@@ -148,8 +160,6 @@ ClientStateStartResolve::Process(ClientThread &client)

 		ClientStateResolving::Instance().SetResolver(resolver.release());
 		client.SetState(ClientStateResolving::Instance());
-
-		retVal = MSG_SOCK_INTERNAL_PENDING;
 	}

 	return retVal;
@@ -223,6 +233,127 @@ ClientStateResolving::Cleanup()

 //-----------------------------------------------------------------------------

+ClientStateStartServerListDownload &
+ClientStateStartServerListDownload::Instance()
+{
+	static ClientStateStartServerListDownload state;
+	return state;
+}
+
+ClientStateStartServerListDownload::ClientStateStartServerListDownload()
+{
+}
+
+ClientStateStartServerListDownload::~ClientStateStartServerListDownload()
+{
+}
+
+int
+ClientStateStartServerListDownload::Process(ClientThread &client)
+{
+	int retVal = MSG_SOCK_INTERNAL_PENDING;
+
+	const ClientContext &context = client.GetContext();
+	path tmpServerListPath(context.GetCacheDir());
+	tmpServerListPath /= "serverlist.xml.z";
+
+	std::auto_ptr<DownloadHelper> downloader(new DownloadHelper);
+	downloader->Init("http://pokerth.net/serverlist.xml.z", tmpServerListPath.directory_string());
+	ClientStateDownloadingServerList::Instance().SetDownloadHelper(downloader.release());
+	client.SetState(ClientStateDownloadingServerList::Instance());
+
+	return retVal;
+}
+
+//-----------------------------------------------------------------------------
+
+ClientStateDownloadingServerList &
+ClientStateDownloadingServerList::Instance()
+{
+	static ClientStateDownloadingServerList state;
+	return state;
+}
+
+ClientStateDownloadingServerList::ClientStateDownloadingServerList()
+: m_downloadHelper(NULL)
+{
+}
+
+ClientStateDownloadingServerList::~ClientStateDownloadingServerList()
+{
+	Cleanup();
+}
+
+void
+ClientStateDownloadingServerList::SetDownloadHelper(DownloadHelper *helper)
+{
+	Cleanup();
+	m_downloadHelper = helper;
+}
+
+int
+ClientStateDownloadingServerList::Process(ClientThread &client)
+{
+	int retVal = MSG_SOCK_INTERNAL_PENDING;
+
+	// TODO
+//	if (!m_resolver)
+//		throw ClientException(__FILE__, __LINE__, ERR_SOCK_RESOLVE_FAILED, 0);
+
+	if (m_downloadHelper->Process())
+	{
+		ClientContext &context = client.GetContext();
+		path zippedServerListPath(context.GetCacheDir());
+		zippedServerListPath /= "serverlist.xml.z";
+		path xmlServerListPath = change_extension(zippedServerListPath, "");
+		// Unzip the file.
+		{
+			ifstream inFile(zippedServerListPath.directory_string().c_str(), ios_base::in | ios_base::binary);
+			ofstream outFile(xmlServerListPath.directory_string().c_str(), ios_base::out);
+			boost::iostreams::filtering_streambuf<boost::iostreams::input> in;
+			in.push(boost::iostreams::zlib_decompressor());
+			in.push(inFile);
+			boost::iostreams::copy(in, outFile);
+		}
+		// Parse the server address.
+		TiXmlDocument doc(xmlServerListPath.directory_string());
+
+		if (doc.LoadFile())
+		{
+			TiXmlHandle docHandle(&doc);	
+			const TiXmlElement *firstServer = docHandle.FirstChild("ServerList" ).FirstChild("Server").ToElement();
+			if (firstServer)
+			{
+				ClientContext &context = client.GetContext();
+				const TiXmlNode *addrNode = firstServer->FirstChild("IPv4Address");
+				if (addrNode && addrNode->ToElement())
+					context.SetServerAddr(addrNode->ToElement()->Attribute("value"));
+				const TiXmlNode *portNode = firstServer->FirstChild("Port");
+				if (portNode && portNode->ToElement())
+				{
+					int tmpPort = 0;
+					portNode->ToElement()->QueryIntAttribute("value", &tmpPort);
+					context.SetServerPort((unsigned)tmpPort);
+				}
+			}
+		}
+
+		client.SetState(ClientStateStartResolve::Instance());
+	}
+
+	return retVal;
+}
+
+
+void
+ClientStateDownloadingServerList::Cleanup()
+{
+	delete m_downloadHelper;
+	m_downloadHelper = NULL;
+}
+
+//-----------------------------------------------------------------------------
+
 ClientStateStartConnect &
 ClientStateStartConnect::Instance()
 {
@@ -76,7 +76,8 @@ void
 ClientThread::Init(
 	const string &serverAddress, const string &alternateServerAddress,
 	unsigned serverPort, bool ipv6, bool sctp,
-	const string &pwd, const string &playerName, const string &avatarFile)
+	const string &pwd, const string &playerName, const string &avatarFile,
+	const string &cacheDir)
 {
 	if (IsRunning())
 	{
@@ -94,6 +95,7 @@ ClientThread::Init(
 	context.SetPassword(pwd);
 	context.SetPlayerName(playerName);
 	context.SetAvatarFile(avatarFile);
+	context.SetCacheDir(cacheDir);
 }

 void
@@ -0,0 +1,127 @@
+/***************************************************************************
+ *   Copyright (C) 2008 by Lothar May                                      *
+ *                                                                         *
+ *   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.,                                       *
+ *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
+ ***************************************************************************/
+
+#include <net/socket_helper.h>
+#include <net/downloadhelper.h>
+#include <curl/curl.h>
+
+#include <cstdio>
+
+using namespace std;
+
+struct DownloadData
+{
+	CURL *curlHandle;
+	CURLM *curlMultiHandle;
+	FILE *targetFile;
+};
+
+
+DownloadHelper::DownloadHelper()
+{
+	m_data.reset(new DownloadData);
+	m_data->curlHandle = NULL;
+	m_data->curlMultiHandle = NULL;
+	m_data->targetFile = NULL;
+}
+
+DownloadHelper::~DownloadHelper()
+{
+	Cleanup();
+}
+
+static size_t curlWriter(void *ptr, size_t size, size_t nmemb, void *stream)
+{
+	int written = fwrite(ptr, size, nmemb, (FILE *)stream);
+	return written;
+}
+
+void
+DownloadHelper::Init(const string &url, const string &targetFileName)
+{
+	m_data->targetFile = fopen(targetFileName.c_str(), "wb");
+	m_data->curlHandle = curl_easy_init();
+	m_data->curlMultiHandle = curl_multi_init();
+
+	// TODO throw exception on error
+
+	curl_easy_setopt(m_data->curlHandle, CURLOPT_URL, url.c_str());
+	curl_easy_setopt(m_data->curlHandle, CURLOPT_WRITEFUNCTION, curlWriter);
+	curl_easy_setopt(m_data->curlHandle, CURLOPT_WRITEDATA, m_data->targetFile);
+
+	curl_multi_add_handle(m_data->curlMultiHandle, m_data->curlHandle);
+}
+
+bool
+DownloadHelper::Process()
+{
+	bool retVal = false;
+	int runningHandles;
+	while(CURLM_CALL_MULTI_PERFORM ==
+		curl_multi_perform(m_data->curlMultiHandle, &runningHandles));
+
+	if (runningHandles)
+	{
+		struct timeval timeout;
+		fd_set readSet;
+		fd_set writeSet;
+		fd_set exceptSet;
+		int maxfd;
+
+		FD_ZERO(&readSet);
+		FD_ZERO(&writeSet);
+		FD_ZERO(&exceptSet);
+
+		timeout.tv_sec = 0;
+		timeout.tv_usec = RECV_TIMEOUT_MSEC * 1000;
+
+		curl_multi_fdset(m_data->curlMultiHandle, &readSet, &writeSet, &exceptSet, &maxfd);
+
+		if (maxfd >= 0)
+			int selectResult = select(maxfd+1, &readSet, &writeSet, &exceptSet, &timeout);
+		// TODO throw exception on error
+	}
+	else
+	{
+		Cleanup();
+		retVal = true;
+	}
+	return retVal;
+}
+
+void
+DownloadHelper::Cleanup()
+{
+	if (m_data->curlMultiHandle)
+	{
+		curl_multi_cleanup(m_data->curlMultiHandle);
+		m_data->curlMultiHandle = NULL;
+	}
+	if (m_data->curlHandle)
+	{
+		curl_easy_cleanup(m_data->curlHandle);
+		m_data->curlHandle = NULL;
+	}
+	if (m_data->targetFile)
+	{
+		fclose(m_data->targetFile);
+		m_data->targetFile = NULL;
+	}
+}
+