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, byte* pachOutput, uint8 nMaxSize, uint32 ChallengeIP, uint8 byChaIPKind, void* sigkey = NULL);
43 	bool	VerifyIdent(CClientCredits* pTarget, const byte* pachSignature, uint8 nInputSize, uint32 dwForIP, uint8 byChaIPKind);
44 
45 	CClientCredits* GetCredit(const CMD4Hash& key);
46 	void	Process();
GetPubKeyLen()47 	uint8	GetPubKeyLen() const			{return m_nMyPublicKeyLen;}
GetPublicKey()48 	const byte*	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 	byte		m_abyMyPublicKey[80];
65 	uint8		m_nMyPublicKeyLen;
66 };
67 
68 #endif // CLIENTCREDITSLIST_H
69 // File_checked_for_headers
70