client/WebServerManager.h - master - makani/strongdc

client/WebServerManager.h
220 строк5 KB
bigmuscle
updated core
24 июл 2011, 22:28
24 июл 2011, 22:28172ec44
/* 
 * Copyright (C) 2003 Twink, spm7@waikato.ac.nz
 *
 * 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.
 */

#pragma once
#include "DCPlusPlus.h"
#include "ServerSocket.h"
#include "SettingsManager.h"
#include "SearchManager.h"
#include "Singleton.h"
#include "Thread.h"
#include "Speaker.h"

namespace dcpp {

class WebServerListener{
public:
	template<int I>	struct X { enum { TYPE = I };  };

	typedef X<0> Setup;
	typedef X<1> ShutdownPC;

	virtual void on(Setup) noexcept = 0;
	virtual void on(ShutdownPC, int action) noexcept = 0;

};

class WebServerManager :  public Singleton<WebServerManager>, public ServerSocketListener, public Speaker<WebServerListener>, private SettingsManagerListener, private SearchManagerListener
{
public:
	ServerSocket& getServerSocket() {
		return socket;
	}

	string getPage(const string& file, const string& IP);
	string getLoginPage();


	// SettingsManagerListener
	void on(SettingsManagerListener::Save, SimpleXML& /*xml*/) noexcept {
		if(BOOLSETTING(WEBSERVER)){
			Restart();
		} else {
			Stop();
		}
	}
	// SearchManagerListener
	void on(SearchManagerListener::SR, const SearchResultPtr& aResult) noexcept;
	
	void Start();
	void Restart(){		
		Stop();
		Start();
	}

	string getResults() {
		return results;
	}

private:
	friend Singleton<WebServerManager>;
	WebServerManager(void);
	~WebServerManager(void);

	
	void Stop(); 


	enum PageID{
		INDEX,
		DOWNLOAD_QUEUE,
		DOWNLOAD_FINISHED,
		UPLOAD_QUEUE,
		UPLOAD_FINISHED,
		SEARCH,
		LOG,
		SYSLOG,
		LOGOUT,
		PAGE_404
	};

	struct WebPageInfo {
		WebPageInfo(PageID n,string t) : id(n), title(t) {}
		PageID id;
		string title;
	};

	WebPageInfo *page404;

	struct eqstr{
		bool operator()(const char* s1, const char* s2) const {  
			return _stricmp(s1, s2) == 0;
		}
	};

	typedef unordered_map<string, WebPageInfo*> WebPages;
	WebPages pages;

	string getDLQueue();
	string getULQueue();
	string getFinished(bool);
	string getSearch();	
	string getLogs();
	string getSysLogs();
	string results;
	unsigned int results_size;
	bool sended_search;

	bool started;
	CriticalSection cs;
	// ServerSocketListener
	void on(ServerSocketListener::IncomingConnection) noexcept;

	ServerSocket socket;
	HWND m_hWnd;

	std::string token;
	
	map<string, uint64_t> loggedin;
	int row;
public:
	void login(const string& ip){
		loggedin[ip] = GET_TICK();
	}
	void search(string search_str, int search_type) {
		if(sended_search == false) {
			size_t i = 0;
			while( (i = search_str.find("+", i)) != string::npos) {
				search_str.replace(i, 1, " ");
				i++;
			}
			if((SearchManager::TypeModes)search_type == SearchManager::TYPE_TTH) {
				search_str = "TTH:" + search_str;
			}
			
			token = Util::toString(Util::rand());
			
			SearchManager::getInstance()->addListener(this);

			// Get ADC searchtype extensions if any is selected
			StringList extList;
			try {
				if(search_type == SearchManager::TYPE_ANY) {
					// Custom searchtype
					// disabled with current GUI extList = SettingsManager::getInstance()->getExtensions(Text::fromT(fileType->getText()));
				} else if(search_type > SearchManager::TYPE_ANY && search_type < SearchManager::TYPE_DIRECTORY) {
					// Predefined searchtype
					extList = SettingsManager::getInstance()->getExtensions(string(1, '0' + search_type));
				}
			} catch(const SearchTypeException&) {
				search_type = SearchManager::TYPE_ANY;
			}
			
			searchInterval = SearchManager::getInstance()->search(WebServerManager::getInstance()->sClients, search_str, 0, (SearchManager::TypeModes)search_type, SearchManager::SIZE_DONTCARE, token, extList, (void*)this);
			results = Util::emptyString;
			row = 0;
			sended_search = true;
		}
	}
	
	void reset() {
		row = 0; /* Counter to permit FireFox correct item clicks */
		SearchManager::getInstance()->removeListener(this);
	}

	bool isloggedin(const string& ip) {
		map<string, uint64_t>::iterator i;
		if((i = loggedin.find(ip)) != loggedin.end()) {
            uint64_t elapsed = (GET_TICK() - loggedin[ip]) / 1000;
			if(elapsed > 600) {
				loggedin.erase(i);
				return false;
			}

			return true;
		}

		return false;
	}

	StringList sClients;
	uint64_t searchInterval;
};

class WebServerSocket : public Thread {
public:

	void accept(ServerSocket *s){
		int fromlen=sizeof(from);

		printf("accepting\n");
		sock = ::accept(s->getSock(), (struct sockaddr*)&from,&fromlen);
		u_long b = 1;
		ioctlsocket(sock, FIONBIO, &b);		
	}

	StringMap getArgs(const string& arguments); 
	int run();

private:
	sockaddr_in from;
	SOCKET sock;
	HANDLE thread;
};

} // namespace dcpp