downloadwithsources.h

p2p软件

//
// DownloadWithSources.h
//
// Copyright (c) Shareaza Development Team, 2002-2004.
// This file is part of SHAREAZA (www.shareaza.com)
//
// Shareaza 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.
//
// Shareaza 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 Shareaza; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
//

#if !defined(AFX_DOWNLOADWITHSOURCES_H__D6932F45_0557_4098_B2F3_AE35BC43ECC0__INCLUDED_)
#define AFX_DOWNLOADWITHSOURCES_H__D6932F45_0557_4098_B2F3_AE35BC43ECC0__INCLUDED_

#pragma once

#include "DownloadBase.h"

class CDownloadSource;
class CQueryHit;
class CXMLElement;


class CDownloadWithSources : public CDownloadBase
{
// Construction
public:
	CDownloadWithSources();
	virtual ~CDownloadWithSources();
	
// Attributes
protected:
	CDownloadSource*	m_pSourceFirst;
	CDownloadSource*	m_pSourceLast;
	int					m_nSourceCount;
public:
	CStringList			m_pFailedSources;
	CXMLElement*		m_pXML;

// Operations
public:
	CString				GetSourceURLs(CStringList* pState, int nMaximum, BOOL bHTTP, CDownloadSource* pExcept);
	int					GetSourceCount(BOOL bNoPush = FALSE, BOOL bSane = FALSE) const;
	BOOL				CheckSource(CDownloadSource* pSource) const;
	void				ClearSources();
public:
	BOOL				AddSourceHit(CQueryHit* pHit, BOOL bForce = FALSE);
	BOOL				AddSourceED2K(DWORD nClientID, WORD nClientPort, DWORD nServerIP, WORD nServerPort, GGUID* pGUID = NULL);
	BOOL				AddSourceBT(SHA1* pGUID, IN_ADDR* pAddress, WORD nPort);
	BOOL				AddSourceURL(LPCTSTR pszURL, BOOL bURN = FALSE, FILETIME* pLastSeen = NULL);
	int					AddSourceURLs(LPCTSTR pszURLs, BOOL bURN = FALSE);
	virtual BOOL		OnQueryHits(CQueryHit* pHits);
	virtual void		Serialize(CArchive& ar, int nVersion);

// Implementation
protected:
	BOOL		AddSourceInternal(CDownloadSource* pSource);
	void		RemoveSource(CDownloadSource* pSource, BOOL bBan);
	void		SortSource(CDownloadSource* pSource, BOOL bTop);
protected:
	void		RemoveOverlappingSources(QWORD nOffset, QWORD nLength);
	int			GetSourceColour();

// Inlines
public:
	inline CDownloadSource* GetFirstSource() const
	{
		return m_pSourceFirst;
	}
	
	friend class CDownloadSource;
	friend class CDownloadTransfer;
};

#endif // !defined(AFX_DOWNLOADWITHSOURCES_H__D6932F45_0557_4098_B2F3_AE35BC43ECC0__INCLUDED_)