summaryrefslogtreecommitdiffstats
path: root/twrpDigest.hpp
blob: ec4a65f7580f90a7d54a02e18854eef928e6cbdc (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
/*
        Copyright 2012 to 2016 bigbiff/Dees_Troy TeamWin
        This file is part of TWRP/TeamWin Recovery Project.

        TWRP 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 3 of the License, or
        (at your option) any later version.

        TWRP 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 TWRP.  If not, see <http://www.gnu.org/licenses/>.
*/

extern "C" {
	#include "digest/md5.h"
}

/* verify_md5digest return codes */
enum {
	MD5_MATCH_FAIL = -2, // -2: md5 did not match
	MD5_NOT_FOUND,       // -1: no md5 file found
	MD5_OK,              //  0: md5 matches
	MD5_FILE_UNREADABLE  //  1: md5 file unreadable
};

using namespace std;

class twrpDigest
{
public:
	void setfn(const string& fn);
	int computeMD5(void);
	int verify_md5digest(void);
	int write_md5digest(void);
	int updateMD5stream(unsigned char* stream, int len);
	void finalizeMD5stream(void);
	string createMD5string(void);
	void initMD5(void);

private:
	int read_md5digest(void);
	struct MD5Context md5c;
	string md5fn;
	string line;
	unsigned char md5sum[MD5LENGTH];
	string md5string;
};