tequilaOS/platform_bootable_recovery
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge "Move sysMapFile and sysReleaseMap into MemMapping class." am: fe7eecff4c am: 08d9ede8f2

Browse source 
am: b3eeb817ab

Change-Id: I1bcf86cbbc495bf5df92dc53e01be39e7c623255
This commit is contained in:
Tao Bao 2017-05-02 23:51:32 +00:00 committed by android-build-merger
commit 586377d1f3
8 changed files with 113 additions and 141 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
install.cpp
View file

@ -569,7 +569,7 @@ static int really_install_package(const std::string& path, bool* wipe_cache, boo
} }
MemMapping map; MemMapping map;
if (sysMapFile(path.c_str(), &map) != 0) { if (!map.MapFile(path)) {
LOG(ERROR) << "failed to map file"; LOG(ERROR) << "failed to map file";
return INSTALL_CORRUPT; return INSTALL_CORRUPT;
} }
@ -577,7 +577,6 @@ static int really_install_package(const std::string& path, bool* wipe_cache, boo
// Verify package. // Verify package.
if (!verify_package(map.addr, map.length)) { if (!verify_package(map.addr, map.length)) {
log_buffer->push_back(android::base::StringPrintf("error: %d", kZipVerificationFailure)); log_buffer->push_back(android::base::StringPrintf("error: %d", kZipVerificationFailure));
sysReleaseMap(&map);
return INSTALL_CORRUPT; return INSTALL_CORRUPT;
} }
@ -588,7 +587,6 @@ static int really_install_package(const std::string& path, bool* wipe_cache, boo
LOG(ERROR) << "Can't open " << path << " : " << ErrorCodeString(err); LOG(ERROR) << "Can't open " << path << " : " << ErrorCodeString(err);
log_buffer->push_back(android::base::StringPrintf("error: %d", kZipOpenFailure)); log_buffer->push_back(android::base::StringPrintf("error: %d", kZipOpenFailure));
sysReleaseMap(&map);
CloseArchive(zip); CloseArchive(zip);
return INSTALL_CORRUPT; return INSTALL_CORRUPT;
} }
@ -596,7 +594,6 @@ static int really_install_package(const std::string& path, bool* wipe_cache, boo
// Additionally verify the compatibility of the package. // Additionally verify the compatibility of the package.
if (!verify_package_compatibility(zip)) { if (!verify_package_compatibility(zip)) {
log_buffer->push_back(android::base::StringPrintf("error: %d", kPackageCompatibilityFailure)); log_buffer->push_back(android::base::StringPrintf("error: %d", kPackageCompatibilityFailure));
sysReleaseMap(&map);
CloseArchive(zip); CloseArchive(zip);
return INSTALL_CORRUPT; return INSTALL_CORRUPT;
} }
@ -611,7 +608,6 @@ static int really_install_package(const std::string& path, bool* wipe_cache, boo
ui->SetEnableReboot(true); ui->SetEnableReboot(true);
ui->Print("\n"); ui->Print("\n");
sysReleaseMap(&map);
CloseArchive(zip); CloseArchive(zip);
return result; return result;
} }

126
otautil/SysUtil.cpp
View file

@ -16,14 +16,12 @@
#include "SysUtil.h" #include "SysUtil.h"
#include <errno.h>
#include <fcntl.h> #include <fcntl.h>
#include <stdint.h> #include <stdint.h> // SIZE_MAX
#include <sys/mman.h> #include <sys/mman.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <sys/types.h> #include <sys/types.h>
#include <algorithm>
#include <string> #include <string>
#include <vector> #include <vector>
@ -32,9 +30,7 @@
#include <android-base/strings.h> #include <android-base/strings.h>
#include <android-base/unique_fd.h> #include <android-base/unique_fd.h>
static bool sysMapFD(int fd, MemMapping* pMap) { bool MemMapping::MapFD(int fd) {
CHECK(pMap != nullptr);
struct stat sb; struct stat sb;
if (fstat(fd, &sb) == -1) { if (fstat(fd, &sb) == -1) {
PLOG(ERROR) << "fstat(" << fd << ") failed"; PLOG(ERROR) << "fstat(" << fd << ") failed";
@ -47,50 +43,49 @@ static bool sysMapFD(int fd, MemMapping* pMap) {
return false; return false;
} }
pMap->addr = static_cast<unsigned char*>(memPtr); addr = static_cast<unsigned char*>(memPtr);
pMap->length = sb.st_size; length = sb.st_size;
pMap->ranges.push_back({ memPtr, static_cast<size_t>(sb.st_size) }); ranges_.clear();
ranges_.emplace_back(MappedRange{ memPtr, static_cast<size_t>(sb.st_size) });
return true; return true;
} }
// A "block map" which looks like this (from uncrypt/uncrypt.cpp): // A "block map" which looks like this (from uncrypt/uncrypt.cpp):
// //
// /dev/block/platform/msm_sdcc.1/by-name/userdata # block device // /dev/block/platform/msm_sdcc.1/by-name/userdata # block device
// 49652 4096 # file size in bytes, block size // 49652 4096 # file size in bytes, block size
// 3 # count of block ranges // 3 # count of block ranges
// 1000 1008 # block range 0 // 1000 1008 # block range 0
// 2100 2102 # ... block range 1 // 2100 2102 # ... block range 1
// 30 33 # ... block range 2 // 30 33 # ... block range 2
// //
// Each block range represents a half-open interval; the line "30 33" // Each block range represents a half-open interval; the line "30 33" reprents the blocks
// reprents the blocks [30, 31, 32]. // [30, 31, 32].
static int sysMapBlockFile(const char* filename, MemMapping* pMap) { bool MemMapping::MapBlockFile(const std::string& filename) {
CHECK(pMap != nullptr);
std::string content; std::string content;
if (!android::base::ReadFileToString(filename, &content)) { if (!android::base::ReadFileToString(filename, &content)) {
PLOG(ERROR) << "Failed to read " << filename; PLOG(ERROR) << "Failed to read " << filename;
return -1; return false;
} }
std::vector<std::string> lines = android::base::Split(android::base::Trim(content), "\n"); std::vector<std::string> lines = android::base::Split(android::base::Trim(content), "\n");
if (lines.size() < 4) { if (lines.size() < 4) {
LOG(ERROR) << "Block map file is too short: " << lines.size(); LOG(ERROR) << "Block map file is too short: " << lines.size();
return -1; return false;
} }
size_t size; size_t size;
unsigned int blksize; size_t blksize;
if (sscanf(lines[1].c_str(), "%zu %u", &size, &blksize) != 2) { if (sscanf(lines[1].c_str(), "%zu %zu", &size, &blksize) != 2) {
LOG(ERROR) << "Failed to parse file size and block size: " << lines[1]; LOG(ERROR) << "Failed to parse file size and block size: " << lines[1];
return -1; return false;
} }
size_t range_count; size_t range_count;
if (sscanf(lines[2].c_str(), "%zu", &range_count) != 1) { if (sscanf(lines[2].c_str(), "%zu", &range_count) != 1) {
LOG(ERROR) << "Failed to parse block map header: " << lines[2]; LOG(ERROR) << "Failed to parse block map header: " << lines[2];
return -1; return false;
} }
size_t blocks; size_t blocks;
@ -101,14 +96,14 @@ static int sysMapBlockFile(const char* filename, MemMapping* pMap) {
lines.size() != 3 + range_count) { lines.size() != 3 + range_count) {
LOG(ERROR) << "Invalid data in block map file: size " << size << ", blksize " << blksize LOG(ERROR) << "Invalid data in block map file: size " << size << ", blksize " << blksize
<< ", range_count " << range_count << ", lines " << lines.size(); << ", range_count " << range_count << ", lines " << lines.size();
return -1; return false;
} }
// Reserve enough contiguous address space for the whole file. // Reserve enough contiguous address space for the whole file.
void* reserve = mmap64(nullptr, blocks * blksize, PROT_NONE, MAP_PRIVATE | MAP_ANON, -1, 0); void* reserve = mmap64(nullptr, blocks * blksize, PROT_NONE, MAP_PRIVATE | MAP_ANON, -1, 0);
if (reserve == MAP_FAILED) { if (reserve == MAP_FAILED) {
PLOG(ERROR) << "failed to reserve address space"; PLOG(ERROR) << "failed to reserve address space";
return -1; return false;
} }
const std::string& block_dev = lines[0]; const std::string& block_dev = lines[0];
@ -116,10 +111,10 @@ static int sysMapBlockFile(const char* filename, MemMapping* pMap) {
if (fd == -1) { if (fd == -1) {
PLOG(ERROR) << "failed to open block device " << block_dev; PLOG(ERROR) << "failed to open block device " << block_dev;
munmap(reserve, blocks * blksize); munmap(reserve, blocks * blksize);
return -1; return false;
} }
pMap->ranges.resize(range_count); ranges_.clear();
unsigned char* next = static_cast<unsigned char*>(reserve); unsigned char* next = static_cast<unsigned char*>(reserve);
size_t remaining_size = blocks * blksize; size_t remaining_size = blocks * blksize;
@ -129,84 +124,79 @@ static int sysMapBlockFile(const char* filename, MemMapping* pMap) {
size_t start, end; size_t start, end;
if (sscanf(line.c_str(), "%zu %zu\n", &start, &end) != 2) { if (sscanf(line.c_str(), "%zu %zu\n", &start, &end) != 2) {
LOG(ERROR) << "failed to parse range " << i << " in block map: " << line; LOG(ERROR) << "failed to parse range " << i << ": " << line;
success = false; success = false;
break; break;
} }
size_t length = (end - start) * blksize; size_t range_size = (end - start) * blksize;
if (end <= start || (end - start) > SIZE_MAX / blksize || length > remaining_size) { if (end <= start || (end - start) > SIZE_MAX / blksize || range_size > remaining_size) {
LOG(ERROR) << "unexpected range in block map: " << start << " " << end; LOG(ERROR) << "Invalid range: " << start << " " << end;
success = false; success = false;
break; break;
} }
void* addr = mmap64(next, length, PROT_READ, MAP_PRIVATE | MAP_FIXED, fd, void* range_start = mmap64(next, range_size, PROT_READ, MAP_PRIVATE | MAP_FIXED, fd,
static_cast<off64_t>(start) * blksize); static_cast<off64_t>(start) * blksize);
if (addr == MAP_FAILED) { if (range_start == MAP_FAILED) {
PLOG(ERROR) << "failed to map block " << i; PLOG(ERROR) << "failed to map range " << i << ": " << line;
success = false; success = false;
break; break;
} }
pMap->ranges[i].addr = addr; ranges_.emplace_back(MappedRange{ range_start, range_size });
pMap->ranges[i].length = length;
next += length; next += range_size;
remaining_size -= length; remaining_size -= range_size;
} }
if (success && remaining_size != 0) { if (success && remaining_size != 0) {
LOG(ERROR) << "ranges in block map are invalid: remaining_size = " << remaining_size; LOG(ERROR) << "Invalid ranges: remaining_size " << remaining_size;
success = false; success = false;
} }
if (!success) { if (!success) {
munmap(reserve, blocks * blksize); munmap(reserve, blocks * blksize);
return -1; return false;
} }
pMap->addr = static_cast<unsigned char*>(reserve); addr = static_cast<unsigned char*>(reserve);
pMap->length = size; length = size;
LOG(INFO) << "mmapped " << range_count << " ranges"; LOG(INFO) << "mmapped " << range_count << " ranges";
return 0; return true;
} }
int sysMapFile(const char* fn, MemMapping* pMap) { bool MemMapping::MapFile(const std::string& fn) {
if (fn == nullptr || pMap == nullptr) { if (fn.empty()) {
LOG(ERROR) << "Invalid argument(s)"; LOG(ERROR) << "Empty filename";
return -1; return false;
} }
*pMap = {};
if (fn[0] == '@') { if (fn[0] == '@') {
if (sysMapBlockFile(fn + 1, pMap) != 0) { // Block map file "@/cache/recovery/block.map".
if (!MapBlockFile(fn.substr(1))) {
LOG(ERROR) << "Map of '" << fn << "' failed"; LOG(ERROR) << "Map of '" << fn << "' failed";
return -1; return false;
} }
} else { } else {
// This is a regular file. // This is a regular file.
android::base::unique_fd fd(TEMP_FAILURE_RETRY(open(fn, O_RDONLY))); android::base::unique_fd fd(TEMP_FAILURE_RETRY(open(fn.c_str(), O_RDONLY)));
if (fd == -1) { if (fd == -1) {
PLOG(ERROR) << "Unable to open '" << fn << "'"; PLOG(ERROR) << "Unable to open '" << fn << "'";
return -1; return false;
} }
if (!sysMapFD(fd, pMap)) { if (!MapFD(fd)) {
LOG(ERROR) << "Map of '" << fn << "' failed"; LOG(ERROR) << "Map of '" << fn << "' failed";
return -1; return false;
} }
} }
return 0; return true;
} }
/* MemMapping::~MemMapping() {
* Release a memory mapping. for (const auto& range : ranges_) {
*/
void sysReleaseMap(MemMapping* pMap) {
std::for_each(pMap->ranges.cbegin(), pMap->ranges.cend(), [](const MappedRange& range) {
if (munmap(range.addr, range.length) == -1) { if (munmap(range.addr, range.length) == -1) {
PLOG(ERROR) << "munmap(" << range.addr << ", " << range.length << ") failed"; PLOG(ERROR) << "Failed to munmap(" << range.addr << ", " << range.length << ")";
} }
}); };
pMap->ranges.clear(); ranges_.clear();
} }

48
otautil/SysUtil.h
View file

@ -19,37 +19,35 @@
#include <sys/types.h> #include <sys/types.h>
#include <string>
#include <vector> #include <vector>
struct MappedRange {
void* addr;
size_t length;
};
/* /*
* Use this to keep track of mapped segments. * Use this to keep track of mapped segments.
*/ */
struct MemMapping { class MemMapping {
unsigned char* addr; /* start of data */ public:
size_t length; /* length of data */ ~MemMapping();
// Map a file into a private, read-only memory segment. If 'filename' begins with an '@'
// character, it is a map of blocks to be mapped, otherwise it is treated as an ordinary file.
bool MapFile(const std::string& filename);
size_t ranges() const {
return ranges_.size();
};
std::vector<MappedRange> ranges; unsigned char* addr; // start of data
size_t length; // length of data
private:
struct MappedRange {
void* addr;
size_t length;
};
bool MapBlockFile(const std::string& filename);
bool MapFD(int fd);
std::vector<MappedRange> ranges_;
}; };
/*
* Map a file into a private, read-only memory segment. If 'fn'
* begins with an '@' character, it is a map of blocks to be mapped,
* otherwise it is treated as an ordinary file.
*
* On success, "pMap" is filled in, and zero is returned.
*/
int sysMapFile(const char* fn, MemMapping* pMap);
/*
* Release the pages associated with a shared memory segment.
*
* This does not free "pMap"; it just releases the memory.
*/
void sysReleaseMap(MemMapping* pMap);
#endif // _OTAUTIL_SYSUTIL #endif // _OTAUTIL_SYSUTIL

4
tests/component/updater_test.cpp
View file

@ -570,7 +570,7 @@ TEST_F(UpdaterTest, block_image_update) {
ASSERT_EQ(0, fclose(zip_file_ptr)); ASSERT_EQ(0, fclose(zip_file_ptr));
MemMapping map; MemMapping map;
ASSERT_EQ(0, sysMapFile(zip_file.path, &map)); ASSERT_TRUE(map.MapFile(zip_file.path));
ZipArchiveHandle handle; ZipArchiveHandle handle;
ASSERT_EQ(0, OpenArchiveFromMemory(map.addr, map.length, zip_file.path, &handle)); ASSERT_EQ(0, OpenArchiveFromMemory(map.addr, map.length, zip_file.path, &handle));
@ -646,7 +646,7 @@ TEST_F(UpdaterTest, new_data_short_write) {
ASSERT_EQ(0, fclose(zip_file_ptr)); ASSERT_EQ(0, fclose(zip_file_ptr));
MemMapping map; MemMapping map;
ASSERT_EQ(0, sysMapFile(zip_file.path, &map)); ASSERT_TRUE(map.MapFile(zip_file.path));
ZipArchiveHandle handle; ZipArchiveHandle handle;
ASSERT_EQ(0, OpenArchiveFromMemory(map.addr, map.length, zip_file.path, &handle)); ASSERT_EQ(0, OpenArchiveFromMemory(map.addr, map.length, zip_file.path, &handle));

2
tests/component/verifier_test.cpp
View file

@ -40,7 +40,7 @@ class VerifierTest : public testing::TestWithParam<std::vector<std::string>> {
void SetUp() override { void SetUp() override {
std::vector<std::string> args = GetParam(); std::vector<std::string> args = GetParam();
std::string package = from_testdata_base(args[0]); std::string package = from_testdata_base(args[0]);
if (sysMapFile(package.c_str(), &memmap) != 0) { if (!memmap.MapFile(package)) {
FAIL() << "Failed to mmap " << package << ": " << strerror(errno) << "\n"; FAIL() << "Failed to mmap " << package << ": " << strerror(errno) << "\n";
} }

60
tests/unit/sysutil_test.cpp
View file

@ -27,27 +27,23 @@ TEST(SysUtilTest, InvalidArgs) {
MemMapping mapping; MemMapping mapping;
// Invalid argument. // Invalid argument.
ASSERT_EQ(-1, sysMapFile(nullptr, &mapping)); ASSERT_FALSE(mapping.MapFile(""));
ASSERT_EQ(-1, sysMapFile("/somefile", nullptr));
} }
TEST(SysUtilTest, sysMapFileRegularFile) { TEST(SysUtilTest, MapFileRegularFile) {
TemporaryFile temp_file1; TemporaryFile temp_file1;
std::string content = "abc"; std::string content = "abc";
ASSERT_TRUE(android::base::WriteStringToFile(content, temp_file1.path)); ASSERT_TRUE(android::base::WriteStringToFile(content, temp_file1.path));
// sysMapFile() should map the file to one range. // MemMapping::MapFile() should map the file to one range.
MemMapping mapping; MemMapping mapping;
ASSERT_EQ(0, sysMapFile(temp_file1.path, &mapping)); ASSERT_TRUE(mapping.MapFile(temp_file1.path));
ASSERT_NE(nullptr, mapping.addr); ASSERT_NE(nullptr, mapping.addr);
ASSERT_EQ(content.size(), mapping.length); ASSERT_EQ(content.size(), mapping.length);
ASSERT_EQ(1U, mapping.ranges.size()); ASSERT_EQ(1U, mapping.ranges());
sysReleaseMap(&mapping);
ASSERT_EQ(0U, mapping.ranges.size());
} }
TEST(SysUtilTest, sysMapFileBlockMap) { TEST(SysUtilTest, MapFileBlockMap) {
// Create a file that has 10 blocks. // Create a file that has 10 blocks.
TemporaryFile package; TemporaryFile package;
std::string content; std::string content;
@ -63,78 +59,72 @@ TEST(SysUtilTest, sysMapFileBlockMap) {
std::string block_map_content = std::string(package.path) + "\n40960 4096\n1\n0 10\n"; std::string block_map_content = std::string(package.path) + "\n40960 4096\n1\n0 10\n";
ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path)); ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path));
ASSERT_EQ(0, sysMapFile(filename.c_str(), &mapping)); ASSERT_TRUE(mapping.MapFile(filename));
ASSERT_EQ(file_size, mapping.length); ASSERT_EQ(file_size, mapping.length);
ASSERT_EQ(1U, mapping.ranges.size()); ASSERT_EQ(1U, mapping.ranges());
// It's okay to not have the trailing '\n'. // It's okay to not have the trailing '\n'.
block_map_content = std::string(package.path) + "\n40960 4096\n1\n0 10"; block_map_content = std::string(package.path) + "\n40960 4096\n1\n0 10";
ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path)); ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path));
ASSERT_EQ(0, sysMapFile(filename.c_str(), &mapping)); ASSERT_TRUE(mapping.MapFile(filename));
ASSERT_EQ(file_size, mapping.length); ASSERT_EQ(file_size, mapping.length);
ASSERT_EQ(1U, mapping.ranges.size()); ASSERT_EQ(1U, mapping.ranges());
// Or having multiple trailing '\n's. // Or having multiple trailing '\n's.
block_map_content = std::string(package.path) + "\n40960 4096\n1\n0 10\n\n\n"; block_map_content = std::string(package.path) + "\n40960 4096\n1\n0 10\n\n\n";
ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path)); ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path));
ASSERT_EQ(0, sysMapFile(filename.c_str(), &mapping)); ASSERT_TRUE(mapping.MapFile(filename));
ASSERT_EQ(file_size, mapping.length); ASSERT_EQ(file_size, mapping.length);
ASSERT_EQ(1U, mapping.ranges.size()); ASSERT_EQ(1U, mapping.ranges());
// Multiple ranges. // Multiple ranges.
block_map_content = std::string(package.path) + "\n40960 4096\n3\n0 3\n3 5\n5 10\n"; block_map_content = std::string(package.path) + "\n40960 4096\n3\n0 3\n3 5\n5 10\n";
ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path)); ASSERT_TRUE(android::base::WriteStringToFile(block_map_content, block_map_file.path));
ASSERT_EQ(0, sysMapFile(filename.c_str(), &mapping)); ASSERT_TRUE(mapping.MapFile(filename));
ASSERT_EQ(file_size, mapping.length); ASSERT_EQ(file_size, mapping.length);
ASSERT_EQ(3U, mapping.ranges.size()); ASSERT_EQ(3U, mapping.ranges());
sysReleaseMap(&mapping);
ASSERT_EQ(0U, mapping.ranges.size());
} }
TEST(SysUtilTest, sysMapFileBlockMapInvalidBlockMap) { TEST(SysUtilTest, MapFileBlockMapInvalidBlockMap) {
MemMapping mapping; MemMapping mapping;
TemporaryFile temp_file; TemporaryFile temp_file;
std::string filename = std::string("@") + temp_file.path; std::string filename = std::string("@") + temp_file.path;
// Block map file is too short. // Block map file is too short.
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n0\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n0\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
// Block map file has unexpected number of lines. // Block map file has unexpected number of lines.
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n2\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n2\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
// Invalid size/blksize/range_count. // Invalid size/blksize/range_count.
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\nabc 4096\n1\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\nabc 4096\n1\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
// size/blksize/range_count don't match. // size/blksize/range_count don't match.
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n0 4096\n1\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n0 4096\n1\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 0\n1\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 0\n1\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n0\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/somefile\n4096 4096\n0\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
// Invalid block dev path. // Invalid block dev path.
ASSERT_TRUE(android::base::WriteStringToFile("/doesntexist\n4096 4096\n1\n0 1\n", temp_file.path)); ASSERT_TRUE(android::base::WriteStringToFile("/doesntexist\n4096 4096\n1\n0 1\n", temp_file.path));
ASSERT_EQ(-1, sysMapFile(filename.c_str(), &mapping)); ASSERT_FALSE(mapping.MapFile(filename));
sysReleaseMap(&mapping);
ASSERT_EQ(0U, mapping.ranges.size());
} }

5
tests/unit/zip_test.cpp
View file

@ -66,9 +66,9 @@ TEST(ZipTest, ExtractPackageRecursive) {
} }
TEST(ZipTest, OpenFromMemory) { TEST(ZipTest, OpenFromMemory) {
MemMapping map;
std::string zip_path = from_testdata_base("ziptest_dummy-update.zip"); std::string zip_path = from_testdata_base("ziptest_dummy-update.zip");
ASSERT_EQ(0, sysMapFile(zip_path.c_str(), &map)); MemMapping map;
ASSERT_TRUE(map.MapFile(zip_path));
// Map an update package into memory and open the archive from there. // Map an update package into memory and open the archive from there.
ZipArchiveHandle handle; ZipArchiveHandle handle;
@ -85,6 +85,5 @@ TEST(ZipTest, OpenFromMemory) {
ASSERT_EQ(0, ExtractEntryToFile(handle, &binary_entry, tmp_binary.fd)); ASSERT_EQ(0, ExtractEntryToFile(handle, &binary_entry, tmp_binary.fd));
CloseArchive(handle); CloseArchive(handle);
sysReleaseMap(&map);
} }

3
updater/updater.cpp
View file

@ -89,7 +89,7 @@ int main(int argc, char** argv) {
const char* package_filename = argv[3]; const char* package_filename = argv[3];
MemMapping map; MemMapping map;
if (sysMapFile(package_filename, &map) != 0) { if (!map.MapFile(package_filename)) {
LOG(ERROR) << "failed to map package " << argv[3]; LOG(ERROR) << "failed to map package " << argv[3];
return 3; return 3;
} }
@ -213,7 +213,6 @@ int main(int argc, char** argv) {
if (updater_info.package_zip) { if (updater_info.package_zip) {
CloseArchive(updater_info.package_zip); CloseArchive(updater_info.package_zip);
} }
sysReleaseMap(&map);
return 0; return 0;
} }