btpeermessage.h

模拟P2P各种网络环境的,适合新手们的学习,不错的源码.

#ifndef __BTPEERMESSAGE_H__
#define __BTPEERMESSAGE_H__
#include <string>
#include <vector>
using namespace std;
typedef enum{
	CHOKE,
	UNCHOKE,
	INTERESTED,
	NOT_INTERESTED,
	HAVE,
	BITFIELD,
	REQUEST,
	PIECE,
	CANCEL,
	HAND_SHAKE
}PeerEventType;
class BTSocket;
class BTPeerMessage{
public:
	string mDownloadHash;
	BTSocket* mConnection;
	vector<bool>* mBitfield;
	int mInsterested;
	unsigned  int mIndex;
	unsigned int mBegin;
	unsigned int mLength;
	PeerEventType mType;
 	
public:
	BTPeerMessage(PeerEventType type){
		mType = type;
	};
	~BTPeerMessage(void){};
	void setType(PeerEventType type){
		mType = type;
	}
	void setConnection(BTSocket* connection){
		mConnection = connection;
	}
	void setLength( unsigned int length){
		mLength = length;
	};
	void setIndex(  unsigned int index){
		mIndex = index;
	}
	void setBegin(unsigned int begin){
		mBegin = begin;
	}

	void setBitField(vector<bool> * bitfield){
		mBitfield = bitfield;
	};
	void setKey(string key){
		mDownloadHash= key;
	}
	string toString(void);
};


#endif