393 lines
11 KiB
C++
393 lines
11 KiB
C++
|
/*
|
||
|
|
* Copyright (C) 2016 The Android Open Source Project
|
||
|
|
*
|
||
|
|
* Licensed under the Apache License, version 2.0 (the "License");
|
||
|
|
* you may not use this file except in compliance with the License.
|
||
|
|
* You may obtain a copy of the License at
|
||
|
|
*
|
||
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
*
|
||
|
|
* Unless required by applicable law or agree to in writing, software
|
||
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
* See the License for the specific language governing permissions and
|
||
|
|
* limitations under the License.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <gtest/gtest.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <sys/statvfs.h>
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <time.h>
|
||
|
|
|
||
|
|
#include <string>
|
||
|
|
|
||
|
|
#include <android-base/file.h>
|
||
|
|
#include <android-base/stringprintf.h>
|
||
|
|
#include <android-base/test_utils.h>
|
||
|
|
|
||
|
|
#include "applypatch/applypatch.h"
|
||
|
|
#include "common/test_constants.h"
|
||
|
|
#include "openssl/sha.h"
|
||
|
|
#include "print_sha1.h"
|
||
|
|
|
||
|
|
static const std::string DATA_PATH = getenv("ANDROID_DATA");
|
||
|
|
static const std::string TESTDATA_PATH = "/recovery/testdata";
|
||
|
|
static const std::string WORK_FS = "/data";
|
||
|
|
|
||
|
|
static std::string sha1sum(const std::string& fname) {
|
||
|
|
uint8_t digest[SHA_DIGEST_LENGTH];
|
||
|
|
std::string data;
|
||
|
|
android::base::ReadFileToString(fname, &data);
|
||
|
|
|
||
|
|
SHA1((const uint8_t*)data.c_str(), data.size(), digest);
|
||
|
|
return print_sha1(digest);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void mangle_file(const std::string& fname) {
|
||
|
|
FILE* fh = fopen(&fname[0], "w");
|
||
|
|
int r;
|
||
|
|
for (int i=0; i < 1024; i++) {
|
||
|
|
r = rand();
|
||
|
|
fwrite(&r, sizeof(short), 1, fh);
|
||
|
|
}
|
||
|
|
fclose(fh);
|
||
|
|
}
|
||
|
|
|
||
|
|
static bool file_cmp(std::string& f1, std::string& f2) {
|
||
|
|
std::string c1;
|
||
|
|
std::string c2;
|
||
|
|
android::base::ReadFileToString(f1, &c1);
|
||
|
|
android::base::ReadFileToString(f2, &c2);
|
||
|
|
return c1 == c2;
|
||
|
|
}
|
||
|
|
|
||
|
|
static std::string from_testdata_base(const std::string fname) {
|
||
|
|
return android::base::StringPrintf("%s%s%s/%s",
|
||
|
|
&DATA_PATH[0],
|
||
|
|
&NATIVE_TEST_PATH[0],
|
||
|
|
&TESTDATA_PATH[0],
|
||
|
|
&fname[0]);
|
||
|
|
}
|
||
|
|
|
||
|
|
class ApplyPatchTest : public ::testing::Test {
|
||
|
|
public:
|
||
|
|
static void SetUpTestCase() {
|
||
|
|
// set up files
|
||
|
|
old_file = from_testdata_base("old.file");
|
||
|
|
new_file = from_testdata_base("new.file");
|
||
|
|
patch_file = from_testdata_base("patch.bsdiff");
|
||
|
|
rand_file = "/cache/applypatch_test_rand.file";
|
||
|
|
cache_file = "/cache/saved.file";
|
||
|
|
|
||
|
|
// write stuff to rand_file
|
||
|
|
android::base::WriteStringToFile("hello", rand_file);
|
||
|
|
|
||
|
|
// set up SHA constants
|
||
|
|
old_sha1 = sha1sum(old_file);
|
||
|
|
new_sha1 = sha1sum(new_file);
|
||
|
|
srand(time(NULL));
|
||
|
|
bad_sha1_a = android::base::StringPrintf("%040x", rand());
|
||
|
|
bad_sha1_b = android::base::StringPrintf("%040x", rand());
|
||
|
|
|
||
|
|
struct stat st;
|
||
|
|
stat(&new_file[0], &st);
|
||
|
|
new_size = st.st_size;
|
||
|
|
}
|
||
|
|
|
||
|
|
static std::string old_file;
|
||
|
|
static std::string new_file;
|
||
|
|
static std::string rand_file;
|
||
|
|
static std::string cache_file;
|
||
|
|
static std::string patch_file;
|
||
|
|
|
||
|
|
static std::string old_sha1;
|
||
|
|
static std::string new_sha1;
|
||
|
|
static std::string bad_sha1_a;
|
||
|
|
static std::string bad_sha1_b;
|
||
|
|
|
||
|
|
static size_t new_size;
|
||
|
|
};
|
||
|
|
|
||
|
|
std::string ApplyPatchTest::old_file;
|
||
|
|
std::string ApplyPatchTest::new_file;
|
||
|
|
|
||
|
|
static void cp(std::string src, std::string tgt) {
|
||
|
|
std::string cmd = android::base::StringPrintf("cp %s %s",
|
||
|
|
&src[0],
|
||
|
|
&tgt[0]);
|
||
|
|
system(&cmd[0]);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void backup_old() {
|
||
|
|
cp(ApplyPatchTest::old_file, ApplyPatchTest::cache_file);
|
||
|
|
}
|
||
|
|
|
||
|
|
static void restore_old() {
|
||
|
|
cp(ApplyPatchTest::cache_file, ApplyPatchTest::old_file);
|
||
|
|
}
|
||
|
|
|
||
|
|
class ApplyPatchCacheTest : public ApplyPatchTest {
|
||
|
|
public:
|
||
|
|
virtual void SetUp() {
|
||
|
|
backup_old();
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void TearDown() {
|
||
|
|
restore_old();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
class ApplyPatchFullTest : public ApplyPatchCacheTest {
|
||
|
|
public:
|
||
|
|
static void SetUpTestCase() {
|
||
|
|
ApplyPatchTest::SetUpTestCase();
|
||
|
|
unsigned long free_kb = FreeSpaceForFile(&WORK_FS[0]);
|
||
|
|
ASSERT_GE(free_kb * 1024, new_size * 3 / 2);
|
||
|
|
output_f = new TemporaryFile();
|
||
|
|
output_loc = std::string(output_f->path);
|
||
|
|
|
||
|
|
struct FileContents fc;
|
||
|
|
|
||
|
|
ASSERT_EQ(0, LoadFileContents(&rand_file[0], &fc));
|
||
|
|
Value* patch1 = new Value();
|
||
|
|
patch1->type = VAL_BLOB;
|
||
|
|
patch1->size = fc.data.size();
|
||
|
|
patch1->data = static_cast<char*>(malloc(fc.data.size()));
|
||
|
|
memcpy(patch1->data, fc.data.data(), fc.data.size());
|
||
|
|
patches.push_back(patch1);
|
||
|
|
|
||
|
|
ASSERT_EQ(0, LoadFileContents(&patch_file[0], &fc));
|
||
|
|
Value* patch2 = new Value();
|
||
|
|
patch2->type = VAL_BLOB;
|
||
|
|
patch2->size = fc.st.st_size;
|
||
|
|
patch2->data = static_cast<char*>(malloc(fc.data.size()));
|
||
|
|
memcpy(patch2->data, fc.data.data(), fc.data.size());
|
||
|
|
patches.push_back(patch2);
|
||
|
|
}
|
||
|
|
static void TearDownTestCase() {
|
||
|
|
delete output_f;
|
||
|
|
for (auto it = patches.begin(); it != patches.end(); ++it) {
|
||
|
|
free((*it)->data);
|
||
|
|
delete *it;
|
||
|
|
}
|
||
|
|
patches.clear();
|
||
|
|
}
|
||
|
|
|
||
|
|
static std::vector<Value*> patches;
|
||
|
|
static TemporaryFile* output_f;
|
||
|
|
static std::string output_loc;
|
||
|
|
};
|
||
|
|
|
||
|
|
class ApplyPatchDoubleCacheTest : public ApplyPatchFullTest {
|
||
|
|
public:
|
||
|
|
virtual void SetUp() {
|
||
|
|
ApplyPatchCacheTest::SetUp();
|
||
|
|
cp(cache_file, "/cache/reallysaved.file");
|
||
|
|
}
|
||
|
|
|
||
|
|
virtual void TearDown() {
|
||
|
|
cp("/cache/reallysaved.file", cache_file);
|
||
|
|
ApplyPatchCacheTest::TearDown();
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
std::string ApplyPatchTest::rand_file;
|
||
|
|
std::string ApplyPatchTest::patch_file;
|
||
|
|
std::string ApplyPatchTest::cache_file;
|
||
|
|
std::string ApplyPatchTest::old_sha1;
|
||
|
|
std::string ApplyPatchTest::new_sha1;
|
||
|
|
std::string ApplyPatchTest::bad_sha1_a;
|
||
|
|
std::string ApplyPatchTest::bad_sha1_b;
|
||
|
|
|
||
|
|
size_t ApplyPatchTest::new_size;
|
||
|
|
|
||
|
|
std::vector<Value*> ApplyPatchFullTest::patches;
|
||
|
|
TemporaryFile* ApplyPatchFullTest::output_f;
|
||
|
|
std::string ApplyPatchFullTest::output_loc;
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchTest, CheckModeSingle) {
|
||
|
|
char* s = &old_sha1[0];
|
||
|
|
ASSERT_EQ(0, applypatch_check(&old_file[0], 1, &s));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchTest, CheckModeMultiple) {
|
||
|
|
char* argv[3] = {
|
||
|
|
&bad_sha1_a[0],
|
||
|
|
&old_sha1[0],
|
||
|
|
&bad_sha1_b[0]
|
||
|
|
};
|
||
|
|
ASSERT_EQ(0, applypatch_check(&old_file[0], 3, argv));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchTest, CheckModeFailure) {
|
||
|
|
char* argv[2] = {
|
||
|
|
&bad_sha1_a[0],
|
||
|
|
&bad_sha1_b[0]
|
||
|
|
};
|
||
|
|
ASSERT_NE(0, applypatch_check(&old_file[0], 2, argv));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchCacheTest, CheckCacheCorruptedSingle) {
|
||
|
|
mangle_file(old_file);
|
||
|
|
char* s = &old_sha1[0];
|
||
|
|
ASSERT_EQ(0, applypatch_check(&old_file[0], 1, &s));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchCacheTest, CheckCacheCorruptedMultiple) {
|
||
|
|
mangle_file(old_file);
|
||
|
|
char* argv[3] = {
|
||
|
|
&bad_sha1_a[0],
|
||
|
|
&old_sha1[0],
|
||
|
|
&bad_sha1_b[0]
|
||
|
|
};
|
||
|
|
ASSERT_EQ(0, applypatch_check(&old_file[0], 3, argv));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchCacheTest, CheckCacheCorruptedFailure) {
|
||
|
|
mangle_file(old_file);
|
||
|
|
char* argv[2] = {
|
||
|
|
&bad_sha1_a[0],
|
||
|
|
&bad_sha1_b[0]
|
||
|
|
};
|
||
|
|
ASSERT_NE(0, applypatch_check(&old_file[0], 2, argv));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchCacheTest, CheckCacheMissingSingle) {
|
||
|
|
unlink(&old_file[0]);
|
||
|
|
char* s = &old_sha1[0];
|
||
|
|
ASSERT_EQ(0, applypatch_check(&old_file[0], 1, &s));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchCacheTest, CheckCacheMissingMultiple) {
|
||
|
|
unlink(&old_file[0]);
|
||
|
|
char* argv[3] = {
|
||
|
|
&bad_sha1_a[0],
|
||
|
|
&old_sha1[0],
|
||
|
|
&bad_sha1_b[0]
|
||
|
|
};
|
||
|
|
ASSERT_EQ(0, applypatch_check(&old_file[0], 3, argv));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchCacheTest, CheckCacheMissingFailure) {
|
||
|
|
unlink(&old_file[0]);
|
||
|
|
char* argv[2] = {
|
||
|
|
&bad_sha1_a[0],
|
||
|
|
&bad_sha1_b[0]
|
||
|
|
};
|
||
|
|
ASSERT_NE(0, applypatch_check(&old_file[0], 2, argv));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchFullTest, ApplyInPlace) {
|
||
|
|
std::vector<char*> sha1s;
|
||
|
|
sha1s.push_back(&bad_sha1_a[0]);
|
||
|
|
sha1s.push_back(&old_sha1[0]);
|
||
|
|
|
||
|
|
int ap_result = applypatch(&old_file[0],
|
||
|
|
"-",
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_EQ(0, ap_result);
|
||
|
|
ASSERT_TRUE(file_cmp(old_file, new_file));
|
||
|
|
// reapply, applypatch is idempotent so it should succeed
|
||
|
|
ap_result = applypatch(&old_file[0],
|
||
|
|
"-",
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_EQ(0, ap_result);
|
||
|
|
ASSERT_TRUE(file_cmp(old_file, new_file));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchFullTest, ApplyInNewLocation) {
|
||
|
|
std::vector<char*> sha1s;
|
||
|
|
sha1s.push_back(&bad_sha1_a[0]);
|
||
|
|
sha1s.push_back(&old_sha1[0]);
|
||
|
|
int ap_result = applypatch(&old_file[0],
|
||
|
|
&output_loc[0],
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_EQ(0, ap_result);
|
||
|
|
ASSERT_TRUE(file_cmp(output_loc, new_file));
|
||
|
|
ap_result = applypatch(&old_file[0],
|
||
|
|
&output_loc[0],
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_EQ(0, ap_result);
|
||
|
|
ASSERT_TRUE(file_cmp(output_loc, new_file));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchFullTest, ApplyCorruptedInNewLocation) {
|
||
|
|
mangle_file(old_file);
|
||
|
|
std::vector<char*> sha1s;
|
||
|
|
sha1s.push_back(&bad_sha1_a[0]);
|
||
|
|
sha1s.push_back(&old_sha1[0]);
|
||
|
|
int ap_result = applypatch(&old_file[0],
|
||
|
|
&output_loc[0],
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_EQ(0, ap_result);
|
||
|
|
ASSERT_TRUE(file_cmp(output_loc, new_file));
|
||
|
|
ap_result = applypatch(&old_file[0],
|
||
|
|
&output_loc[0],
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_EQ(0, ap_result);
|
||
|
|
ASSERT_TRUE(file_cmp(output_loc, new_file));
|
||
|
|
}
|
||
|
|
|
||
|
|
TEST_F(ApplyPatchDoubleCacheTest, ApplyDoubleCorruptedInNewLocation) {
|
||
|
|
mangle_file(old_file);
|
||
|
|
mangle_file(cache_file);
|
||
|
|
|
||
|
|
std::vector<char*> sha1s;
|
||
|
|
sha1s.push_back(&bad_sha1_a[0]);
|
||
|
|
sha1s.push_back(&old_sha1[0]);
|
||
|
|
int ap_result = applypatch(&old_file[0],
|
||
|
|
&output_loc[0],
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_NE(0, ap_result);
|
||
|
|
ASSERT_FALSE(file_cmp(output_loc, new_file));
|
||
|
|
ap_result = applypatch(&old_file[0],
|
||
|
|
&output_loc[0],
|
||
|
|
&new_sha1[0],
|
||
|
|
new_size,
|
||
|
|
2,
|
||
|
|
sha1s.data(),
|
||
|
|
patches.data(),
|
||
|
|
nullptr);
|
||
|
|
ASSERT_NE(0, ap_result);
|
||
|
|
ASSERT_FALSE(file_cmp(output_loc, new_file));
|
||
|
|
}
|