/*
Copyright 2013 to 2017 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/>.
*/
#include <fcntl.h>
#include <string>
#include <unistd.h>
#include "data.hpp"
#include "partitions.hpp"
#include "set_metadata.h"
#include "twrpDigestDriver.hpp"
#include "twrp-functions.hpp"
#include "twcommon.h"
#include "variables.h"
#include "gui/gui.hpp"
#include "twrpDigest/twrpDigest.hpp"
#include "twrpDigest/twrpMD5.hpp"
#include "twrpDigest/twrpSHA.hpp"
bool twrpDigestDriver::Check_File_Digest(const string& Filename) {
twrpDigest *digest;
string digestfile = Filename, file_name = Filename;
string digest_str;
bool use_sha2 = false;
#ifndef TW_NO_SHA2_LIBRARY
digestfile += ".sha2";
if (TWFunc::Path_Exists(digestfile)) {
digest = new twrpSHA256();
use_sha2 = true;
}
else {
digestfile = Filename + ".sha256";
if (TWFunc::Path_Exists(digestfile)) {
digest = new twrpSHA256();
use_sha2 = true;
} else {
digest = new twrpMD5();
digestfile = Filename + ".md5";
if (!TWFunc::Path_Exists(digestfile)) {
digestfile = Filename + ".md5sum";
}
}
}
#else
digest = new twrpMD5();
digestfile = Filename + ".md5";
if (!TWFunc::Path_Exists(digestfile)) {
digestfile = Filename + ".md5sum";
}
#endif
if (!TWFunc::Path_Exists(digestfile)) {
if (Filename.find(".zip") == std::string::npos) {
gui_msg(Msg(msg::kError, "no_digest_found=No digest file found for '{1}'. Please unselect Enable Digest verification to restore.")(Filename));
}
delete digest;
return false;
}
if (TWFunc::read_file(digestfile, digest_str) != 0) {
gui_msg("digest_error=Digest Error!");
delete digest;
return false;
}
if (!stream_file_to_digest(file_name, digest)) {
delete digest;
return false;
}
string digest_check = digest->return_digest_string();
if (digest_check == digest_str) {
if (use_sha2)
LOGINFO("SHA2 Digest: %s %s\n", digest_str.c_str(), TWFunc::Get_Filename(Filename).c_str());
else
LOGINFO("MD5 Digest: %s %s\n", digest_str.c_str(), TWFunc::Get_Filename(Filename).c_str());
gui_msg(Msg("digest_matched=Digest matched for '{1}'.")(Filename));
delete digest;
return true;
}
gui_msg(Msg(msg::kError, "digest_fail_match=Digest failed to match on '{1}'.")(Filename));
delete digest;
return false;
}
bool twrpDigestDriver::Check_Digest(string Full_Filename) {
char split_filename[512];
int index = 0;
sync();
if (!TWFunc::Path_Exists(Full_Filename)) {
// This is a split archive, we presume
memset(split_filename, 0, sizeof(split_filename));
while (index < 1000) {
sprintf(split_filename, "%s%03i", Full_Filename.c_str(), index);
if (!TWFunc::Path_Exists(split_filename))
break;
LOGINFO("split_filename: %s\n", split_filename);
if (!Check_File_Digest(split_filename))
return false;
index++;
}
return true;
}
return Check_File_Digest(Full_Filename); // Single file archive
}
bool twrpDigestDriver::Write_Digest(string Full_Filename) {
twrpDigest *digest;
string digest_filename, digest_str;
int use_sha2;
#ifdef TW_NO_SHA2_LIBRARY
use_sha2 = 0;
#else
DataManager::GetValue(TW_USE_SHA2, use_sha2);
#endif
if (use_sha2) {
#ifndef TW_NO_SHA2_LIBRARY
digest = new twrpSHA256();
digest_filename = Full_Filename + ".sha2";
if (!stream_file_to_digest(Full_Filename, digest)) {
delete digest;
return false;
}
digest_str = digest->return_digest_string();
if (digest_str.empty()) {
delete digest;
return false;
}
LOGINFO("SHA2 Digest: %s %s\n", digest_str.c_str(), TWFunc::Get_Filename(Full_Filename).c_str());
#endif
}
else {
digest = new twrpMD5();
digest_filename = Full_Filename + ".md5";
if (!stream_file_to_digest(Full_Filename, digest)) {
delete digest;
return false;
}
digest_str = digest->return_digest_string();
if (digest_str.empty()) {
delete digest;
return false;
}
LOGINFO("MD5 Digest: %s %s\n", digest_str.c_str(), TWFunc::Get_Filename(Full_Filename).c_str());
}
digest_str = digest_str + " " + TWFunc::Get_Filename(Full_Filename) + "\n";
LOGINFO("digest_filename: %s\n", digest_filename.c_str());
if (TWFunc::write_to_file(digest_filename, digest_str) == 0) {
tw_set_default_metadata(digest_filename.c_str());
gui_msg("digest_created= * Digest Created.");
}
else {
gui_err("digest_error= * Digest Error!");
delete digest;
return false;
}
delete digest;
return true;
}
bool twrpDigestDriver::Make_Digest(string Full_Filename) {
string command, result;
TWFunc::GUI_Operation_Text(TW_GENERATE_DIGEST_TEXT, gui_parse_text("{@generating_digest1}"));
gui_msg("generating_digest2= * Generating digest...");
if (TWFunc::Path_Exists(Full_Filename)) {
if (!Write_Digest(Full_Filename))
return false;
} else {
char filename[512];
int index = 0;
sprintf(filename, "%s%03i", Full_Filename.c_str(), index);
while (index < 1000) {
string digest_src(filename);
if (TWFunc::Path_Exists(filename)) {
if (!Write_Digest(filename))
return false;
}
else
break;
index++;
sprintf(filename, "%s%03i", Full_Filename.c_str(), index);
}
if (index == 0) {
LOGERR("Backup file: '%s' not found!\n", filename);
return false;
}
gui_msg("digest_created= * Digest Created.");
}
return true;
}
bool twrpDigestDriver::stream_file_to_digest(string filename, twrpDigest* digest) {
char buf[4096];
int bytes;
int fd = open(filename.c_str(), O_RDONLY);
if (fd < 0) {
return false;
}
while ((bytes = read(fd, &buf, sizeof(buf))) != 0) {
digest->update((unsigned char*)buf, bytes);
}
close(fd);
return true;
}
