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

Merge "updater: Refactor parse_range()."

Browse source
This commit is contained in:
Tao Bao 2017-01-04 06:06:17 +00:00 committed by Gerrit Code Review
commit 466e1b396a
1 changed files with 64 additions and 75 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

139
updater/blockimg.cpp
View file

@ -64,83 +64,82 @@ static constexpr size_t BLOCKSIZE = 4096;
#define STASH_FILE_MODE 0600 #define STASH_FILE_MODE 0600
struct RangeSet { struct RangeSet {
size_t count; // Limit is INT_MAX. size_t count; // Limit is INT_MAX.
size_t size; size_t size;
std::vector<size_t> pos; // Actual limit is INT_MAX. std::vector<size_t> pos; // Actual limit is INT_MAX.
}; };
static CauseCode failure_type = kNoCause; static CauseCode failure_type = kNoCause;
static bool is_retry = false; static bool is_retry = false;
static std::unordered_map<std::string, RangeSet> stash_map; static std::unordered_map<std::string, RangeSet> stash_map;
static void parse_range(const std::string& range_text, RangeSet& rs) { static RangeSet parse_range(const std::string& range_text) {
RangeSet rs;
std::vector<std::string> pieces = android::base::Split(range_text, ","); std::vector<std::string> pieces = android::base::Split(range_text, ",");
if (pieces.size() < 3) { if (pieces.size() < 3) {
goto err; goto err;
}
size_t num;
if (!android::base::ParseUint(pieces[0], &num, static_cast<size_t>(INT_MAX))) {
goto err;
}
if (num == 0 || num % 2) {
goto err; // must be even
} else if (num != pieces.size() - 1) {
goto err;
}
rs.pos.resize(num);
rs.count = num / 2;
rs.size = 0;
for (size_t i = 0; i < num; i += 2) {
if (!android::base::ParseUint(pieces[i + 1], &rs.pos[i], static_cast<size_t>(INT_MAX))) {
goto err;
} }
size_t num; if (!android::base::ParseUint(pieces[i + 2], &rs.pos[i + 1], static_cast<size_t>(INT_MAX))) {
if (!android::base::ParseUint(pieces[0].c_str(), &num, static_cast<size_t>(INT_MAX))) { goto err;
goto err;
} }
if (num == 0 || num % 2) { if (rs.pos[i] >= rs.pos[i + 1]) {
goto err; // must be even goto err; // empty or negative range
} else if (num != pieces.size() - 1) {
goto err;
} }
rs.pos.resize(num); size_t sz = rs.pos[i + 1] - rs.pos[i];
rs.count = num / 2; if (rs.size > SIZE_MAX - sz) {
rs.size = 0; goto err; // overflow
for (size_t i = 0; i < num; i += 2) {
if (!android::base::ParseUint(pieces[i+1].c_str(), &rs.pos[i],
static_cast<size_t>(INT_MAX))) {
goto err;
}
if (!android::base::ParseUint(pieces[i+2].c_str(), &rs.pos[i+1],
static_cast<size_t>(INT_MAX))) {
goto err;
}
if (rs.pos[i] >= rs.pos[i+1]) {
goto err; // empty or negative range
}
size_t sz = rs.pos[i+1] - rs.pos[i];
if (rs.size > SIZE_MAX - sz) {
goto err; // overflow
}
rs.size += sz;
} }
return; rs.size += sz;
}
return rs;
err: err:
LOG(ERROR) << "failed to parse range '" << range_text << "'"; LOG(ERROR) << "failed to parse range '" << range_text << "'";
exit(1); exit(1);
} }
static bool range_overlaps(const RangeSet& r1, const RangeSet& r2) { static bool range_overlaps(const RangeSet& r1, const RangeSet& r2) {
for (size_t i = 0; i < r1.count; ++i) { for (size_t i = 0; i < r1.count; ++i) {
size_t r1_0 = r1.pos[i * 2]; size_t r1_0 = r1.pos[i * 2];
size_t r1_1 = r1.pos[i * 2 + 1]; size_t r1_1 = r1.pos[i * 2 + 1];
for (size_t j = 0; j < r2.count; ++j) { for (size_t j = 0; j < r2.count; ++j) {
size_t r2_0 = r2.pos[j * 2]; size_t r2_0 = r2.pos[j * 2];
size_t r2_1 = r2.pos[j * 2 + 1]; size_t r2_1 = r2.pos[j * 2 + 1];
if (!(r2_0 >= r1_1 || r1_0 >= r2_1)) { if (!(r2_0 >= r1_1 || r1_0 >= r2_1)) {
return true; return true;
} }
}
} }
}
return false; return false;
} }
static int read_all(int fd, uint8_t* data, size_t size) { static int read_all(int fd, uint8_t* data, size_t size) {
@ -431,11 +430,10 @@ static int LoadSrcTgtVersion1(CommandParameters& params, RangeSet& tgt, size_t&
} }
// <src_range> // <src_range>
RangeSet src; RangeSet src = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], src);
// <tgt_range> // <tgt_range>
parse_range(params.tokens[params.cpos++], tgt); tgt = parse_range(params.tokens[params.cpos++]);
allocate(src.size * BLOCKSIZE, buffer); allocate(src.size * BLOCKSIZE, buffer);
int rc = ReadBlocks(src, buffer, fd); int rc = ReadBlocks(src, buffer, fd);
@ -787,8 +785,7 @@ static int SaveStash(CommandParameters& params, const std::string& base,
return 0; return 0;
} }
RangeSet src; RangeSet src = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], src);
allocate(src.size * BLOCKSIZE, buffer); allocate(src.size * BLOCKSIZE, buffer);
if (ReadBlocks(src, buffer, fd) == -1) { if (ReadBlocks(src, buffer, fd) == -1) {
@ -872,7 +869,7 @@ static int LoadSrcTgtVersion2(CommandParameters& params, RangeSet& tgt, size_t&
} }
// <tgt_range> // <tgt_range>
parse_range(params.tokens[params.cpos++], tgt); tgt = parse_range(params.tokens[params.cpos++]);
// <src_block_count> // <src_block_count>
const std::string& token = params.tokens[params.cpos++]; const std::string& token = params.tokens[params.cpos++];
@ -888,8 +885,7 @@ static int LoadSrcTgtVersion2(CommandParameters& params, RangeSet& tgt, size_t&
// no source ranges, only stashes // no source ranges, only stashes
params.cpos++; params.cpos++;
} else { } else {
RangeSet src; RangeSet src = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], src);
int res = ReadBlocks(src, buffer, fd); int res = ReadBlocks(src, buffer, fd);
if (overlap) { if (overlap) {
@ -905,8 +901,7 @@ static int LoadSrcTgtVersion2(CommandParameters& params, RangeSet& tgt, size_t&
return 0; return 0;
} }
RangeSet locs; RangeSet locs = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], locs);
MoveRange(buffer, locs, buffer); MoveRange(buffer, locs, buffer);
} }
@ -931,8 +926,7 @@ static int LoadSrcTgtVersion2(CommandParameters& params, RangeSet& tgt, size_t&
continue; continue;
} }
RangeSet locs; RangeSet locs = parse_range(tokens[1]);
parse_range(tokens[1], locs);
MoveRange(buffer, locs, stash); MoveRange(buffer, locs, stash);
} }
@ -1116,8 +1110,7 @@ static int PerformCommandZero(CommandParameters& params) {
return -1; return -1;
} }
RangeSet tgt; RangeSet tgt = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], tgt);
LOG(INFO) << " zeroing " << tgt.size << " blocks"; LOG(INFO) << " zeroing " << tgt.size << " blocks";
@ -1160,8 +1153,7 @@ static int PerformCommandNew(CommandParameters& params) {
return -1; return -1;
} }
RangeSet tgt; RangeSet tgt = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], tgt);
if (params.canwrite) { if (params.canwrite) {
LOG(INFO) << " writing " << tgt.size << " blocks of new data"; LOG(INFO) << " writing " << tgt.size << " blocks of new data";
@ -1316,8 +1308,7 @@ static int PerformCommandErase(CommandParameters& params) {
return -1; return -1;
} }
RangeSet tgt; RangeSet tgt = parse_range(params.tokens[params.cpos++]);
parse_range(params.tokens[params.cpos++], tgt);
if (params.canwrite) { if (params.canwrite) {
LOG(INFO) << " erasing " << tgt.size << " blocks"; LOG(INFO) << " erasing " << tgt.size << " blocks";
@ -1707,8 +1698,7 @@ Value* RangeSha1Fn(const char* name, State* state, int /* argc */, Expr* argv[])
return StringValue(""); return StringValue("");
} }
RangeSet rs; RangeSet rs = parse_range(ranges->data);
parse_range(ranges->data, rs);
SHA_CTX ctx; SHA_CTX ctx;
SHA1_Init(&ctx); SHA1_Init(&ctx);
@ -1832,8 +1822,7 @@ Value* BlockImageRecoverFn(const char* name, State* state, int argc, Expr* argv[
return StringValue(""); return StringValue("");
} }
RangeSet rs; RangeSet rs = parse_range(ranges->data);
parse_range(ranges->data, rs);
uint8_t buffer[BLOCKSIZE]; uint8_t buffer[BLOCKSIZE];