BTInfo.h

Go to the documentation of this file.

//
// BTInfo.h
//
// Copyright (c) Shareaza Development Team, 2002-2005.
// 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_BTINFO_H__AA44CA36_464F_4FB8_9D79_884D8092ADA0__INCLUDED_)
#define AFX_BTINFO_H__AA44CA36_464F_4FB8_9D79_884D8092ADA0__INCLUDED_

#pragma once

#include "SHA.h"
#include "ED2K.h"
#include "Buffer.h"
#include "TigerTree.h"

class CBuffer;
class CBENode;


class CBTInfo  
{
// Construction
public:
        CBTInfo();
        ~CBTInfo();
        
// Subclass
public:
        class CBTFile
        {
        public:
                CBTFile();
                void            Copy(CBTFile* pSource);
                void            Serialize(CArchive& ar, int nVersion);
        public:
                CString         m_sPath;
                QWORD           m_nSize;
                BOOL            m_bSHA1;
                SHA1            m_pSHA1;
                BOOL            m_bED2K;
                MD4                     m_pED2K;
                BOOL            m_bTiger;
                TIGEROOT        m_pTiger;
        };

// Subclass
public:
        class CBTTracker
        {
        public:
                CBTTracker();
                ~CBTTracker();
                void            Copy(CBTTracker* pSource);
                void            Serialize(CArchive& ar, int nVersion);
        public:
                CString         m_sAddress;
                DWORD           m_tLastAccess;
                DWORD           m_tLastSuccess;
                DWORD           m_tLastFail;
                DWORD           m_nFailures;
                int                     m_nTier;
                int                     m_nType;
        };
        
// Attributes
public:
        BOOL            m_bValid;
        BOOL            m_bEncodingError;
        SHA1            m_pInfoSHA1;
        BOOL            m_bDataSHA1;
        SHA1            m_pDataSHA1;
        BOOL            m_bDataED2K;
        MD4                     m_pDataED2K;
        BOOL            m_bDataTiger;
        TIGEROOT        m_pDataTiger;
public:
        QWORD           m_nTotalSize;
        DWORD           m_nBlockSize;
        DWORD           m_nBlockCount;
        SHA1*           m_pBlockSHA1;
public:
        CString         m_sName;                                                // Name of the torrent
        int                     m_nFiles;                                               // Number of files
        CBTFile*        m_pFiles;                                               // List of files
public:
        CString         m_sTracker;                                             // Address of tracker we are using

        CBTTracker*     m_pAnnounceTracker;                             // Tracker in the announce key
        CPtrArray       m_pTrackerList;                                 // Multi-tracker list
        int                     m_nTrackerIndex;                                // The tracker we are currently using
public:
        UINT            m_nEncoding;
        CString         m_sComment;
        DWORD           m_tCreationDate;
        CString         m_sCreatedBy;
private:
        CSHA            m_pTestSHA1;
        DWORD           m_nTestByte;
        CBuffer         m_pSource;
        
// Operations
public:
        void            Clear();
        void            Copy(CBTInfo* pSource);
        void            Serialize(CArchive& ar);
public:
        BOOL            LoadTorrentFile(LPCTSTR pszFile);
        BOOL            LoadTorrentBuffer(CBuffer* pBuffer);
        BOOL            LoadTorrentTree(CBENode* pRoot);
        BOOL            SaveTorrentFile(LPCTSTR pszPath);
public:
        void            BeginBlockTest();
        void            AddToTest(LPCVOID pInput, DWORD nLength);
        BOOL            FinishBlockTest(DWORD nBlock);
protected:
        BOOL            CheckFiles();

// Inlines
public:
        inline BOOL IsAvailable() const { return m_bValid; }
        inline BOOL HasEncodingError() const { return m_bEncodingError; }

        // Check if a string is a valid path/file name.
        inline BOOL IsValid(LPCTSTR psz) const
        {
                if ( _tcsclen( psz ) == 0 ) return FALSE;
                if ( _tcschr( psz, '?' ) != NULL ) return FALSE;
                if ( _tcsicmp( psz , _T("#ERROR#") ) == 0 ) return FALSE;
                
                return TRUE;
        }

};


#endif // !defined(AFX_BTINFO_H__AA44CA36_464F_4FB8_9D79_884D8092ADA0__INCLUDED_)

---