/***************************************************************************
 *   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-26 20:09
 */

#ifndef UDPTRACKER_H_
#define UDPTRACKER_H_

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <string>
#include "BitStorm.h"
#include "Socket.h"

class CUDPTracker :
	public ITracker,
	public CSocket
{
public:
	CUDPTracker();
	virtual ~CUDPTracker();
private:
	ITrackerManager* _trackerManager;
	std::string _url;
	
	TTrackerState _state;
	std::string _stateStr;	
	
	unsigned int _completePeer;
	unsigned int _totalPeer;
	unsigned int _interval;
	unsigned int _nextUpdateTick;
	
	int _currentEvent;
	bool _startEventSend;
	bool _completeEventSend;
	
	struct sockaddr_in _serverAddr;
	unsigned int _transID;
	int64_t _connectionID;
private:
	void sendConnectPacket();	
	void sendAnnouncPacket();	
	void processPkg(unsigned int actionID, char* data, size_t len);
	int getCurrentEvent();
public:
	void setTrackerManager(ITrackerManager* trackerManager);
	bool isProtocolSupported(const char* protocolName);
	void setUrl(const char* url);
	void update();
	void stop();
	TTrackerState getState();
	const std::string& getStateStr();
	unsigned int getSeedCount();
	unsigned int getPeerCount();
	unsigned int getInterval();
	unsigned int getNextUpdateTick();	
	
	int handleRead();
	int handleWrite();
	void handleClose();		
};

#endif /*UDPTRACKER_H_*/