src/ClientCreditsList.h

   1 //
   2 // This file is part of the aMule Project.
   3 //
   4 // Copyright (c) 2003-2011 aMule Team ( admin@amule.org / http://www.amule.org )
   5 // Copyright (c) 2002-2011 Merkur ( devs@emule-project.net / http://www.emule-project.net )
   6 //
   7 // Any parts of this program derived from the xMule, lMule or eMule project,
   8 // or contributed by third-party developers are copyrighted by their
   9 // respective authors.
  10 //
  11 // This program is free software; you can redistribute it and/or modify
  12 // it under the terms of the GNU General Public License as published by
  13 // the Free Software Foundation; either version 2 of the License, or
  14 // (at your option) any later version.
  15 //
  16 // This program is distributed in the hope that it will be useful,
  17 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  18 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19 // GNU General Public License for more details.
  20 //
  21 // You should have received a copy of the GNU General Public License
  22 // along with this program; if not, write to the Free Software
  23 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301, USA
  24 //
  25
  26 #ifndef CLIENTCREDITSLIST_H
  27 #define CLIENTCREDITSLIST_H
  28
  29 #include "MD4Hash.h"    // Needed for CMD4Hash
  30
  31 #include <map>
  32
  33 class CClientCredits;
  34
  35 class CClientCreditsList
  36 {
  37 public:
  38         CClientCreditsList();
  39         ~CClientCreditsList();
  40
  41         // return signature size, 0 = Failed | use sigkey param for debug only
  42         uint8   CreateSignature(CClientCredits* pTarget, uint8_t* pachOutput, uint8 nMaxSize, uint32 ChallengeIP, uint8 byChaIPKind, void* sigkey = NULL);
  43         bool    VerifyIdent(CClientCredits* pTarget, const uint8_t* pachSignature, uint8 nInputSize, uint32 dwForIP, uint8 byChaIPKind);
  44
  45         CClientCredits* GetCredit(const CMD4Hash& key);
  46         void    Process();
  47         uint8   GetPubKeyLen() const                    {return m_nMyPublicKeyLen;}
  48         const uint8_t*  GetPublicKey() const    {return m_abyMyPublicKey;}
  49         bool    CryptoAvailable() const;
  50         void    SaveList();
  51 protected:
  52         void    LoadList();
  53         void    InitalizeCrypting();
  54         bool    CreateKeyPair();
  55 #ifdef _DEBUG
  56         bool    Debug_CheckCrypting();
  57 #endif
  58 private:
  59         typedef std::map<CMD4Hash, CClientCredits*> ClientMap;
  60         ClientMap       m_mapClients;
  61         uint32          m_nLastSaved;
  62         // A void* to avoid having to include the large CryptoPP.h file
  63         void*           m_pSignkey;
  64         uint8_t         m_abyMyPublicKey[80];
  65         uint8           m_nMyPublicKeyLen;
  66 };
  67
  68 #endif // CLIENTCREDITSLIST_H
  69 // File_checked_for_headers