peermanager.h

来自「这是一个嵌入式linux系统下的BT下载工具包」· C头文件 代码 · 共 77 行

/***************************************************************************
 *   Copyright (C) 2005-2006 Gao Xianchao                                  *
 *                 2007 Gao Xianchao gnap_an linux_lyb ahlongxp            *
 *                                                                         *
 *   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.             *
 ***************************************************************************/

/*
 * Author:	gxc
 * Create data:	2005-10-15 10:09
 */

#ifndef PEERMANAGER_H_
#define PEERMANAGER_H_

#include <string>
#include "BitStorm.h"
#include "ThreadMutex.h"

class CPeerManager : 
	public IPeerManager,
	public ITimerCallback
{
public:
	CPeerManager();
	virtual ~CPeerManager();
private:
	IBTTask* _task;
	unsigned int _connectTimerID;
	unsigned int _chokeTimerID;
	
	TPeerInfoMap _unusedPeerList;
	CThreadMutex _unusedPeerListMutex;
	
	TPeerInfoMap _connectingPeerList;
	TPeerInfoMap _connectedPeerList;
	TPeerInfoMap _banedPeerList;
private:
	void checkPeerConnection();	
	void checkPeerChoke();
	void countPeersSpeed();
	bool peerInUnusedList(std::string id);
	bool peerInConnectingList(std::string id);
	bool peerInConnectedList(std::string id);
	bool peerInBanedList(std::string id);
public:	
 	void setBTTask(IBTTask* task);
	IBTTask* getBTTask();	
	
	bool start();
	void stop();
	
	void onTimer(unsigned int id);
	
	void addPeerInfoWithoutID(const char* ip, unsigned short port);
	bool addAcceptedPeer(int handle, const char* ip, unsigned short port);
	void broadcastHave(unsigned int pieceIndex);
	void cancelPieceRequest(unsigned int pieceIndex);
	unsigned int getConnectedPeerCount();
	void onDownloadComplete();
};

#endif /*PEERMANAGER_H_*/