]>
Commit | Line | Data |
---|---|---|
63b1700f AL |
1 | // -*- mode: cpp; mode: fold -*- |
2 | // Description /*{{{*/ | |
678bc33e | 3 | // $Id: hashes.h,v 1.2 2001/03/11 05:30:20 jgg Exp $ |
63b1700f AL |
4 | /* ###################################################################### |
5 | ||
6 | Hashes - Simple wrapper around the hash functions | |
7 | ||
8 | This is just used to make building the methods simpler, this is the | |
9 | only interface required.. | |
10 | ||
11 | ##################################################################### */ | |
12 | /*}}}*/ | |
13 | #ifndef APTPKG_HASHES_H | |
14 | #define APTPKG_HASHES_H | |
15 | ||
63b1700f AL |
16 | |
17 | #include <apt-pkg/md5.h> | |
18 | #include <apt-pkg/sha1.h> | |
84a0890e | 19 | #include <apt-pkg/sha2.h> |
109eb151 | 20 | #include <apt-pkg/fileutl.h> |
63b1700f | 21 | |
42ab8223 | 22 | #include <algorithm> |
495e5cb2 | 23 | #include <vector> |
b16c2617 | 24 | #include <cstring> |
42ab8223 | 25 | |
a4f6bdc8 DK |
26 | |
27 | #ifndef APT_8_CLEANER_HEADERS | |
28 | using std::min; | |
29 | using std::vector; | |
30 | #endif | |
31 | ||
495e5cb2 MV |
32 | // helper class that contains hash function name |
33 | // and hash | |
34 | class HashString | |
35 | { | |
36 | protected: | |
8f3ba4e8 DK |
37 | std::string Type; |
38 | std::string Hash; | |
e6645b9f MV |
39 | static const char* _SupportedHashes[10]; |
40 | ||
41 | // internal helper | |
42 | std::string GetHashForFile(std::string filename) const; | |
495e5cb2 MV |
43 | |
44 | public: | |
8f3ba4e8 DK |
45 | HashString(std::string Type, std::string Hash); |
46 | HashString(std::string StringedHashString); // init from str as "type:hash" | |
495e5cb2 MV |
47 | HashString(); |
48 | ||
8a8feb29 | 49 | // get hash type used |
8f3ba4e8 | 50 | std::string HashType() { return Type; }; |
8a8feb29 | 51 | |
495e5cb2 | 52 | // verify the given filename against the currently loaded hash |
8f3ba4e8 | 53 | bool VerifyFile(std::string filename) const; |
495e5cb2 | 54 | |
e6645b9f MV |
55 | // generate a hash string from the given filename |
56 | bool FromFile(std::string filename); | |
57 | ||
58 | ||
495e5cb2 | 59 | // helper |
8f3ba4e8 | 60 | std::string toStr() const; // convert to str as "type:hash" |
495e5cb2 MV |
61 | bool empty() const; |
62 | ||
63 | // return the list of hashes we support | |
64 | static const char** SupportedHashes(); | |
65 | }; | |
42ab8223 | 66 | |
63b1700f AL |
67 | class Hashes |
68 | { | |
69 | public: | |
70 | ||
71 | MD5Summation MD5; | |
72 | SHA1Summation SHA1; | |
cde41ae8 | 73 | SHA256Summation SHA256; |
d9b9e9e2 | 74 | SHA512Summation SHA512; |
63b1700f | 75 | |
650faab0 | 76 | inline bool Add(const unsigned char *Data,unsigned long long Size) |
63b1700f | 77 | { |
d9b9e9e2 | 78 | return MD5.Add(Data,Size) && SHA1.Add(Data,Size) && SHA256.Add(Data,Size) && SHA512.Add(Data,Size); |
63b1700f AL |
79 | }; |
80 | inline bool Add(const char *Data) {return Add((unsigned char *)Data,strlen(Data));}; | |
650faab0 | 81 | inline bool AddFD(int const Fd,unsigned long long Size = 0) |
1dab797c | 82 | { return AddFD(Fd, Size, true, true, true, true); }; |
650faab0 | 83 | bool AddFD(int const Fd, unsigned long long Size, bool const addMD5, |
1dab797c | 84 | bool const addSHA1, bool const addSHA256, bool const addSHA512); |
109eb151 DK |
85 | inline bool AddFD(FileFd &Fd,unsigned long long Size = 0) |
86 | { return AddFD(Fd, Size, true, true, true, true); }; | |
87 | bool AddFD(FileFd &Fd, unsigned long long Size, bool const addMD5, | |
88 | bool const addSHA1, bool const addSHA256, bool const addSHA512); | |
63b1700f AL |
89 | inline bool Add(const unsigned char *Beg,const unsigned char *End) |
90 | {return Add(Beg,End-Beg);}; | |
91 | }; | |
92 | ||
93 | #endif |