src/CorruptionBlackBox.h

   1 //
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2008-2009 aMule Team ( admin@amule.org / http://www.amule.org )
   5 // Copyright (c) 2008-2009 Stu Redman ( sturedman@amule.org )
   6 // Copyright (C) 2002-2008 Merkur ( strEmail.Format("%s@%s", "devteam", "emule-project.net") / http://www.emule-project.net )
   7 //
   8 // Any parts of this program derived from the xMule, lMule or eMule project,
   9 // or contributed by third-party developers are copyrighted by their
  10 // respective authors.
  11 //
  12 // This program is free software; you can redistribute it and/or modify
  13 // it under the terms of the GNU General Public License as published by
  14 // the Free Software Foundation; either version 2 of the License, or
  15 // (at your option) any later version.
  16 //
  17 // This program is distributed in the hope that it will be useful,
  18 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  19 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20 // GNU General Public License for more details.
  21 //
  22 // You should have received a copy of the GNU General Public License
  23 // along with this program; if not, write to the Free Software
  24 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
  25 //
  26
  27 #ifndef CORRUPTIONBLACKBOX_H
  28 #define CORRUPTIONBLACKBOX_H
  29
  30 #include "Types.h"
  31 #include <list>
  32 #include <map>
  33
  34 class CCorruptionBlackBox
  35 {
  36 public:
  37         void Free();
  38         void TransferredData(uint64 nStartPos, uint64 nEndPos, uint32 senderIP);
  39         void VerifiedData(bool ok, uint16 nPart, uint32 nRelStartPos, uint32 nRelEndPos);
  40         void EvaluateData();
  41         void SetPartFileInfo(const wxString& name, const wxString& nr) { m_fileName = name; m_partNumber = nr; }
  42         void DumpAll();
  43
  44 private:
  45         // downloaded data for each part
  46         class CCBBRecord
  47         {
  48         public:
  49                 CCBBRecord(uint32 nStartPos, uint32 nEndPos, uint32 dwIP);
  50                 bool Merge(uint32 nStartPos, uint32 nEndPos, uint32 dwIP);
  51
  52                 // Startpos / Endpos relative to part
  53                 uint32  m_nStartPos;
  54                 uint32  m_nEndPos;
  55                 // IP of client
  56                 uint32  m_dwIP;
  57         };
  58         typedef std::list<CCBBRecord> CRecordList;
  59         std::map<uint16, CRecordList> m_Records;
  60
  61         // good/bad data for each client
  62         class CCBBClient
  63         {
  64         public:
  65                 CCBBClient() { m_downloaded = 0; }
  66                 uint64 m_downloaded;
  67         };
  68         typedef std::map<uint32, CCBBClient> CCBBClientMap;
  69         CCBBClientMap m_goodClients, m_badClients;
  70
  71         // for debug prints
  72         wxString m_fileName;
  73         wxString m_partNumber;
  74 };
  75
  76 #endif