Loading updater/blockimg.cpp +144 −133 Original line number Original line Diff line number Diff line Loading @@ -65,16 +65,9 @@ struct RangeSet { std::vector<size_t> pos; // Actual limit is INT_MAX. std::vector<size_t> pos; // Actual limit is INT_MAX. }; }; static void parse_range(const char* range_text, RangeSet& rs) { static void parse_range(const std::string& range_text, RangeSet& rs) { if (range_text == nullptr) { fprintf(stderr, "failed to parse range: null range\n"); exit(1); } std::vector<std::string> pieces = android::base::Split(std::string(range_text), ","); long int val; std::vector<std::string> pieces = android::base::Split(range_text, ","); if (pieces.size() < 3) { if (pieces.size() < 3) { goto err; goto err; } } Loading Loading @@ -120,7 +113,7 @@ static void parse_range(const char* range_text, RangeSet& rs) { return; return; err: err: fprintf(stderr, "failed to parse range '%s'\n", range_text); fprintf(stderr, "failed to parse range '%s'\n", range_text.c_str()); exit(1); exit(1); } } Loading Loading @@ -360,25 +353,49 @@ static int WriteBlocks(const RangeSet& tgt, const std::vector<uint8_t>& buffer, return 0; return 0; } } // Parameters for transfer list command functions struct CommandParameters { std::vector<std::string> tokens; size_t cpos; const char* cmdname; const char* cmdline; std::string freestash; std::string stashbase; bool canwrite; int createdstash; int fd; bool foundwrites; bool isunresumable; int version; size_t written; NewThreadInfo nti; pthread_t thread; std::vector<uint8_t> buffer; uint8_t* patch_start; }; // Do a source/target load for move/bsdiff/imgdiff in version 1. // Do a source/target load for move/bsdiff/imgdiff in version 1. // 'wordsave' is the save_ptr of a strtok_r()-in-progress. We expect // We expect to parse the remainder of the parameter tokens as: // to parse the remainder of the string as: // // // <src_range> <tgt_range> // <src_range> <tgt_range> // // // The source range is loaded into the provided buffer, reallocating // The source range is loaded into the provided buffer, reallocating // it to make it larger if necessary. // it to make it larger if necessary. static int LoadSrcTgtVersion1(char** wordsave, RangeSet& tgt, size_t& src_blocks, static int LoadSrcTgtVersion1(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, std::vector<uint8_t>& buffer, int fd) { std::vector<uint8_t>& buffer, int fd) { if (params.cpos + 1 >= params.tokens.size()) { fprintf(stderr, "invalid parameters\n"); return -1; } // <src_range> // <src_range> char* word = strtok_r(nullptr, " ", wordsave); RangeSet src; RangeSet src; parse_range(word, src); parse_range(params.tokens[params.cpos++], src); // <tgt_range> // <tgt_range> word = strtok_r(nullptr, " ", wordsave); parse_range(params.tokens[params.cpos++], tgt); parse_range(word, tgt); allocate(src.size * BLOCKSIZE, buffer); allocate(src.size * BLOCKSIZE, buffer); int rc = ReadBlocks(src, buffer, fd); int rc = ReadBlocks(src, buffer, fd); Loading Loading @@ -694,18 +711,15 @@ static int CreateStash(State* state, int maxblocks, const char* blockdev, std::s return 0; // Using existing directory return 0; // Using existing directory } } static int SaveStash(const std::string& base, char** wordsave, std::vector<uint8_t>& buffer, static int SaveStash(CommandParameters& params, const std::string& base, int fd, bool usehash) { std::vector<uint8_t>& buffer, int fd, bool usehash) { if (!wordsave) { return -1; } char *id_tok = strtok_r(nullptr, " ", wordsave); // <stash_id> <src_range> if (id_tok == nullptr) { if (params.cpos + 1 >= params.tokens.size()) { fprintf(stderr, "missing id field in stash command\n"); fprintf(stderr, "missing id and/or src range fields in stash command\n"); return -1; return -1; } } std::string id(id_tok); const std::string& id = params.tokens[params.cpos++]; size_t blocks = 0; size_t blocks = 0; if (usehash && LoadStash(base, id, true, &blocks, buffer, false) == 0) { if (usehash && LoadStash(base, id, true, &blocks, buffer, false) == 0) { Loading @@ -715,9 +729,8 @@ static int SaveStash(const std::string& base, char** wordsave, std::vector<uint8 return 0; return 0; } } char* word = strtok_r(nullptr, " ", wordsave); RangeSet src; RangeSet src; parse_range(word, src); 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) { Loading @@ -738,13 +751,12 @@ static int SaveStash(const std::string& base, char** wordsave, std::vector<uint8 return WriteStash(base, id, blocks, buffer, false, nullptr); return WriteStash(base, id, blocks, buffer, false, nullptr); } } static int FreeStash(const std::string& base, const char* id) { static int FreeStash(const std::string& base, const std::string& id) { if (base.empty() || id == nullptr) { if (base.empty() || id.empty()) { return -1; return -1; } } std::string fn = GetStashFileName(base, std::string(id), ""); std::string fn = GetStashFileName(base, id, ""); DeleteFile(fn, nullptr); DeleteFile(fn, nullptr); return 0; return 0; Loading @@ -768,8 +780,7 @@ static void MoveRange(std::vector<uint8_t>& dest, const RangeSet& locs, } } // Do a source/target load for move/bsdiff/imgdiff in version 2. // Do a source/target load for move/bsdiff/imgdiff in version 2. // 'wordsave' is the save_ptr of a strtok_r()-in-progress. We expect // We expect to parse the remainder of the parameter tokens as one of: // to parse the remainder of the string as one of: // // // <tgt_range> <src_block_count> <src_range> // <tgt_range> <src_block_count> <src_range> // (loads data from source image only) // (loads data from source image only) Loading @@ -785,25 +796,35 @@ static void MoveRange(std::vector<uint8_t>& dest, const RangeSet& locs, // reallocated if needed to accommodate the source data. *tgt is the // reallocated if needed to accommodate the source data. *tgt is the // target RangeSet. Any stashes required are loaded using LoadStash. // target RangeSet. Any stashes required are loaded using LoadStash. static int LoadSrcTgtVersion2(char** wordsave, RangeSet& tgt, size_t& src_blocks, static int LoadSrcTgtVersion2(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, std::vector<uint8_t>& buffer, int fd, const std::string& stashbase, bool* overlap) { std::vector<uint8_t>& buffer, int fd, const std::string& stashbase, bool* overlap) { // At least it needs to provide three parameters: <tgt_range>, // <src_block_count> and "-"/<src_range>. if (params.cpos + 2 >= params.tokens.size()) { fprintf(stderr, "invalid parameters\n"); return -1; } // <tgt_range> // <tgt_range> char* word = strtok_r(nullptr, " ", wordsave); parse_range(params.tokens[params.cpos++], tgt); parse_range(word, tgt); // <src_block_count> // <src_block_count> word = strtok_r(nullptr, " ", wordsave); const std::string& token = params.tokens[params.cpos++]; android::base::ParseUint(word, &src_blocks); if (!android::base::ParseUint(token.c_str(), &src_blocks)) { fprintf(stderr, "invalid src_block_count \"%s\"\n", token.c_str()); return -1; } allocate(src_blocks * BLOCKSIZE, buffer); allocate(src_blocks * BLOCKSIZE, buffer); // "-" or <src_range> [<src_loc>] // "-" or <src_range> [<src_loc>] word = strtok_r(nullptr, " ", wordsave); if (params.tokens[params.cpos] == "-") { if (word[0] == '-' && word[1] == '\0') { // no source ranges, only stashes // no source ranges, only stashes params.cpos++; } else { } else { RangeSet src; RangeSet src; parse_range(word, src); parse_range(params.tokens[params.cpos++], src); int res = ReadBlocks(src, buffer, fd); int res = ReadBlocks(src, buffer, fd); if (overlap) { if (overlap) { Loading @@ -814,39 +835,39 @@ static int LoadSrcTgtVersion2(char** wordsave, RangeSet& tgt, size_t& src_blocks return -1; return -1; } } word = strtok_r(nullptr, " ", wordsave); if (params.cpos >= params.tokens.size()) { if (word == nullptr) { // no stashes, only source range // no stashes, only source range return 0; return 0; } } RangeSet locs; RangeSet locs; parse_range(word, locs); parse_range(params.tokens[params.cpos++], locs); MoveRange(buffer, locs, buffer); MoveRange(buffer, locs, buffer); } } // <[stash_id:stash-range]> // <[stash_id:stash_range]> char* colonsave; while (params.cpos < params.tokens.size()) { while ((word = strtok_r(nullptr, " ", wordsave)) != nullptr) { // Each word is a an index into the stash table, a colon, and // Each word is a an index into the stash table, a colon, and // then a rangeset describing where in the source block that // then a rangeset describing where in the source block that // stashed data should go. // stashed data should go. colonsave = nullptr; std::vector<std::string> tokens = android::base::Split(params.tokens[params.cpos++], ":"); char* colon = strtok_r(word, ":", &colonsave); if (tokens.size() != 2) { fprintf(stderr, "invalid parameter\n"); return -1; } std::vector<uint8_t> stash; std::vector<uint8_t> stash; int res = LoadStash(stashbase, std::string(colon), false, nullptr, stash, true); int res = LoadStash(stashbase, tokens[0], false, nullptr, stash, true); if (res == -1) { if (res == -1) { // These source blocks will fail verification if used later, but we // These source blocks will fail verification if used later, but we // will let the caller decide if this is a fatal failure // will let the caller decide if this is a fatal failure fprintf(stderr, "failed to load stash %s\n", colon); fprintf(stderr, "failed to load stash %s\n", tokens[0].c_str()); continue; continue; } } colon = strtok_r(nullptr, ":", &colonsave); RangeSet locs; RangeSet locs; parse_range(colon, locs); parse_range(tokens[1], locs); MoveRange(buffer, locs, stash); MoveRange(buffer, locs, stash); } } Loading @@ -854,25 +875,6 @@ static int LoadSrcTgtVersion2(char** wordsave, RangeSet& tgt, size_t& src_blocks return 0; return 0; } } // Parameters for transfer list command functions struct CommandParameters { char* cmdname; char* cpos; char* freestash; std::string stashbase; bool canwrite; int createdstash; int fd; bool foundwrites; bool isunresumable; int version; size_t written; NewThreadInfo nti; pthread_t thread; std::vector<uint8_t> buffer; uint8_t* patch_start; }; // Do a source/target load for move/bsdiff/imgdiff in version 3. // Do a source/target load for move/bsdiff/imgdiff in version 3. // // // Parameters are the same as for LoadSrcTgtVersion2, except for 'onehash', which // Parameters are the same as for LoadSrcTgtVersion2, except for 'onehash', which Loading @@ -893,26 +895,26 @@ struct CommandParameters { static int LoadSrcTgtVersion3(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, static int LoadSrcTgtVersion3(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, bool onehash, bool& overlap) { bool onehash, bool& overlap) { char* srchash = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos >= params.tokens.size()) { if (srchash == nullptr) { fprintf(stderr, "missing source hash\n"); fprintf(stderr, "missing source hash\n"); return -1; return -1; } } char* tgthash = nullptr; std::string srchash = params.tokens[params.cpos++]; std::string tgthash; if (onehash) { if (onehash) { tgthash = srchash; tgthash = srchash; } else { } else { tgthash = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos >= params.tokens.size()) { if (tgthash == nullptr) { fprintf(stderr, "missing target hash\n"); fprintf(stderr, "missing target hash\n"); return -1; return -1; } } tgthash = params.tokens[params.cpos++]; } } if (LoadSrcTgtVersion2(¶ms.cpos, tgt, src_blocks, params.buffer, params.fd, if (LoadSrcTgtVersion2(params, tgt, src_blocks, params.buffer, params.fd, params.stashbase, params.stashbase, &overlap) == -1) { &overlap) == -1) { return -1; return -1; } } Loading @@ -931,7 +933,8 @@ static int LoadSrcTgtVersion3(CommandParameters& params, RangeSet& tgt, size_t& // If source and target blocks overlap, stash the source blocks so we can // If source and target blocks overlap, stash the source blocks so we can // resume from possible write errors // resume from possible write errors if (overlap) { if (overlap) { fprintf(stderr, "stashing %zu overlapping blocks to %s\n", src_blocks, srchash); fprintf(stderr, "stashing %zu overlapping blocks to %s\n", src_blocks, srchash.c_str()); bool stash_exists = false; bool stash_exists = false; if (WriteStash(params.stashbase, srchash, src_blocks, params.buffer, true, if (WriteStash(params.stashbase, srchash, src_blocks, params.buffer, true, Loading Loading @@ -971,9 +974,9 @@ static int PerformCommandMove(CommandParameters& params) { RangeSet tgt; RangeSet tgt; if (params.version == 1) { if (params.version == 1) { status = LoadSrcTgtVersion1(¶ms.cpos, tgt, blocks, params.buffer, params.fd); status = LoadSrcTgtVersion1(params, tgt, blocks, params.buffer, params.fd); } else if (params.version == 2) { } else if (params.version == 2) { status = LoadSrcTgtVersion2(¶ms.cpos, tgt, blocks, params.buffer, params.fd, status = LoadSrcTgtVersion2(params, tgt, blocks, params.buffer, params.fd, params.stashbase, nullptr); params.stashbase, nullptr); } else if (params.version >= 3) { } else if (params.version >= 3) { status = LoadSrcTgtVersion3(params, tgt, blocks, true, overlap); status = LoadSrcTgtVersion3(params, tgt, blocks, true, overlap); Loading Loading @@ -1003,9 +1006,9 @@ static int PerformCommandMove(CommandParameters& params) { } } if (params.freestash) { if (!params.freestash.empty()) { FreeStash(params.stashbase, params.freestash); FreeStash(params.stashbase, params.freestash); params.freestash = nullptr; params.freestash.clear(); } } params.written += tgt.size; params.written += tgt.size; Loading @@ -1014,28 +1017,33 @@ static int PerformCommandMove(CommandParameters& params) { } } static int PerformCommandStash(CommandParameters& params) { static int PerformCommandStash(CommandParameters& params) { return SaveStash(params.stashbase, ¶ms.cpos, params.buffer, params.fd, return SaveStash(params, params.stashbase, params.buffer, params.fd, (params.version >= 3)); (params.version >= 3)); } } static int PerformCommandFree(CommandParameters& params) { static int PerformCommandFree(CommandParameters& params) { // <stash_id> if (params.cpos >= params.tokens.size()) { fprintf(stderr, "missing stash id in free command\n"); return -1; } if (params.createdstash || params.canwrite) { if (params.createdstash || params.canwrite) { return FreeStash(params.stashbase, params.cpos); return FreeStash(params.stashbase, params.tokens[params.cpos++]); } } return 0; return 0; } } static int PerformCommandZero(CommandParameters& params) { static int PerformCommandZero(CommandParameters& params) { char* range = strtok_r(nullptr, " ", ¶ms.cpos); if (range == nullptr) { if (params.cpos >= params.tokens.size()) { fprintf(stderr, "missing target blocks for zero\n"); fprintf(stderr, "missing target blocks for zero\n"); return -1; return -1; } } RangeSet tgt; RangeSet tgt; parse_range(range, tgt); parse_range(params.tokens[params.cpos++], tgt); fprintf(stderr, " zeroing %zu blocks\n", tgt.size); fprintf(stderr, " zeroing %zu blocks\n", tgt.size); Loading Loading @@ -1066,14 +1074,14 @@ static int PerformCommandZero(CommandParameters& params) { } } static int PerformCommandNew(CommandParameters& params) { static int PerformCommandNew(CommandParameters& params) { char* range = strtok_r(nullptr, " ", ¶ms.cpos); if (range == nullptr) { if (params.cpos >= params.tokens.size()) { fprintf(stderr, "missing target blocks for new\n"); return -1; return -1; } } RangeSet tgt; RangeSet tgt; parse_range(range, tgt); parse_range(params.tokens[params.cpos++], tgt); if (params.canwrite) { if (params.canwrite) { fprintf(stderr, " writing %zu blocks of new data\n", tgt.size); fprintf(stderr, " writing %zu blocks of new data\n", tgt.size); Loading Loading @@ -1105,33 +1113,32 @@ static int PerformCommandNew(CommandParameters& params) { static int PerformCommandDiff(CommandParameters& params) { static int PerformCommandDiff(CommandParameters& params) { const std::string logparams(params.cpos); // <offset> <length> char* value = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos + 1 >= params.tokens.size()) { fprintf(stderr, "missing patch offset or length for %s\n", params.cmdname); if (value == nullptr) { fprintf(stderr, "missing patch offset for %s\n", params.cmdname); return -1; return -1; } } size_t offset = strtoul(value, nullptr, 0); size_t offset; if (!android::base::ParseUint(params.tokens[params.cpos++].c_str(), &offset)) { value = strtok_r(nullptr, " ", ¶ms.cpos); fprintf(stderr, "invalid patch offset\n"); if (value == nullptr) { fprintf(stderr, "missing patch length for %s\n", params.cmdname); return -1; return -1; } } size_t len = strtoul(value, nullptr, 0); size_t len; if (!android::base::ParseUint(params.tokens[params.cpos++].c_str(), &len)) { fprintf(stderr, "invalid patch offset\n"); return -1; } RangeSet tgt; RangeSet tgt; size_t blocks = 0; size_t blocks = 0; bool overlap = false; bool overlap = false; int status = 0; int status = 0; if (params.version == 1) { if (params.version == 1) { status = LoadSrcTgtVersion1(¶ms.cpos, tgt, blocks, params.buffer, params.fd); status = LoadSrcTgtVersion1(params, tgt, blocks, params.buffer, params.fd); } else if (params.version == 2) { } else if (params.version == 2) { status = LoadSrcTgtVersion2(¶ms.cpos, tgt, blocks, params.buffer, params.fd, status = LoadSrcTgtVersion2(params, tgt, blocks, params.buffer, params.fd, params.stashbase, nullptr); params.stashbase, nullptr); } else if (params.version >= 3) { } else if (params.version >= 3) { status = LoadSrcTgtVersion3(params, tgt, blocks, false, overlap); status = LoadSrcTgtVersion3(params, tgt, blocks, false, overlap); Loading Loading @@ -1180,13 +1187,13 @@ static int PerformCommandDiff(CommandParameters& params) { } } } else { } else { fprintf(stderr, "skipping %zu blocks already patched to %zu [%s]\n", fprintf(stderr, "skipping %zu blocks already patched to %zu [%s]\n", blocks, tgt.size, logparams.c_str()); blocks, tgt.size, params.cmdline); } } } } if (params.freestash) { if (!params.freestash.empty()) { FreeStash(params.stashbase, params.freestash); FreeStash(params.stashbase, params.freestash); params.freestash = nullptr; params.freestash.clear(); } } params.written += tgt.size; params.written += tgt.size; Loading @@ -1210,15 +1217,13 @@ static int PerformCommandErase(CommandParameters& params) { return -1; return -1; } } char* range = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos >= params.tokens.size()) { if (range == nullptr) { fprintf(stderr, "missing target blocks for erase\n"); fprintf(stderr, "missing target blocks for erase\n"); return -1; return -1; } } RangeSet tgt; RangeSet tgt; parse_range(range, tgt); parse_range(params.tokens[params.cpos++], tgt); if (params.canwrite) { if (params.canwrite) { fprintf(stderr, " erasing %zu blocks\n", tgt.size); fprintf(stderr, " erasing %zu blocks\n", tgt.size); Loading Loading @@ -1278,7 +1283,6 @@ static unsigned int HashString(const char *s) { static Value* PerformBlockImageUpdate(const char* name, State* state, int /* argc */, Expr* argv[], static Value* PerformBlockImageUpdate(const char* name, State* state, int /* argc */, Expr* argv[], const Command* commands, size_t cmdcount, bool dryrun) { const Command* commands, size_t cmdcount, bool dryrun) { CommandParameters params; CommandParameters params; memset(¶ms, 0, sizeof(params)); memset(¶ms, 0, sizeof(params)); params.canwrite = !dryrun; params.canwrite = !dryrun; Loading Loading @@ -1368,10 +1372,14 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg } } } } // The data in transfer_list_value is not necessarily null-terminated, so we need // Copy all the lines in transfer_list_value into std::string for // to copy it to a new buffer and add the null that strtok_r will need. // processing. const std::string transfer_list(transfer_list_value->data, transfer_list_value->size); const std::string transfer_list(transfer_list_value->data, transfer_list_value->size); std::vector<std::string> lines = android::base::Split(transfer_list, "\n"); std::vector<std::string> lines = android::base::Split(transfer_list, "\n"); if (lines.size() < 2) { ErrorAbort(state, "too few lines in the transfer list [%zd]\n", lines.size()); return StringValue(strdup("")); } // First line in transfer list is the version number // First line in transfer list is the version number if (!android::base::ParseInt(lines[0].c_str(), ¶ms.version, 1, 4)) { if (!android::base::ParseInt(lines[0].c_str(), ¶ms.version, 1, 4)) { Loading @@ -1394,6 +1402,11 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg size_t start = 2; size_t start = 2; if (params.version >= 2) { if (params.version >= 2) { if (lines.size() < 4) { ErrorAbort(state, "too few lines in the transfer list [%zu]\n", lines.size()); return StringValue(strdup("")); } // Third line is how many stash entries are needed simultaneously // Third line is how many stash entries are needed simultaneously fprintf(stderr, "maximum stash entries %s\n", lines[2].c_str()); fprintf(stderr, "maximum stash entries %s\n", lines[2].c_str()); Loading @@ -1405,7 +1418,6 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg } } int res = CreateStash(state, stash_max_blocks, blockdev_filename->data, params.stashbase); int res = CreateStash(state, stash_max_blocks, blockdev_filename->data, params.stashbase); if (res == -1) { if (res == -1) { return StringValue(strdup("")); return StringValue(strdup("")); } } Loading Loading @@ -1433,17 +1445,15 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg continue; continue; } } char* line = strdup(line_str.c_str()); params.tokens = android::base::Split(line_str, " "); params.cmdname = strtok_r(line, " ", ¶ms.cpos); params.cpos = 0; params.cmdname = params.tokens[params.cpos++].c_str(); if (params.cmdname == nullptr) { params.cmdline = line_str.c_str(); fprintf(stderr, "missing command [%s]\n", line); goto pbiudone; } unsigned int cmdhash = HashString(params.cmdname); unsigned int cmdhash = HashString(params.cmdname); const Command* cmd = reinterpret_cast<const Command*>(mzHashTableLookup(cmdht, cmdhash, const Command* cmd = reinterpret_cast<const Command*>(mzHashTableLookup(cmdht, cmdhash, params.cmdname, CompareCommandNames, false)); const_cast<char*>(params.cmdname), CompareCommandNames, false)); if (cmd == nullptr) { if (cmd == nullptr) { fprintf(stderr, "unexpected command [%s]\n", params.cmdname); fprintf(stderr, "unexpected command [%s]\n", params.cmdname); Loading Loading @@ -1481,12 +1491,10 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg rc = 0; rc = 0; pbiudone: pbiudone: if (params.fd != -1) { if (fsync(params.fd) == -1) { if (fsync(params.fd) == -1) { fprintf(stderr, "fsync failed: %s\n", strerror(errno)); fprintf(stderr, "fsync failed: %s\n", strerror(errno)); } } close(params.fd); // params.fd will be automatically closed because of the fd_holder above. } // Only delete the stash if the update cannot be resumed, or it's // Only delete the stash if the update cannot be resumed, or it's // a verification run and we created the stash. // a verification run and we created the stash. Loading Loading @@ -1526,6 +1534,9 @@ pbiudone: // - (version 2+ only) load the given source range and stash // - (version 2+ only) load the given source range and stash // the data in the given slot of the stash table. // the data in the given slot of the stash table. // // // free <stash_id> // - (version 3+ only) free the given stash data. // // The creator of the transfer list will guarantee that no block // The creator of the transfer list will guarantee that no block // is read (ie, used as the source for a patch or move) after it // is read (ie, used as the source for a patch or move) after it // has been written. // has been written. Loading Loading @@ -1692,7 +1703,7 @@ Value* BlockImageRecoverFn(const char* name, State* state, int argc, Expr* argv[ } } if (fh.pread(buffer, BLOCKSIZE, (off64_t)j * BLOCKSIZE) != BLOCKSIZE) { if (fh.pread(buffer, BLOCKSIZE, (off64_t)j * BLOCKSIZE) != BLOCKSIZE) { ErrorAbort(state, "failed to recover %s (block %d): %s", filename->data, ErrorAbort(state, "failed to recover %s (block %zu): %s", filename->data, j, strerror(errno)); j, strerror(errno)); return StringValue(strdup("")); return StringValue(strdup("")); } } Loading Loading
updater/blockimg.cpp +144 −133 Original line number Original line Diff line number Diff line Loading @@ -65,16 +65,9 @@ struct RangeSet { std::vector<size_t> pos; // Actual limit is INT_MAX. std::vector<size_t> pos; // Actual limit is INT_MAX. }; }; static void parse_range(const char* range_text, RangeSet& rs) { static void parse_range(const std::string& range_text, RangeSet& rs) { if (range_text == nullptr) { fprintf(stderr, "failed to parse range: null range\n"); exit(1); } std::vector<std::string> pieces = android::base::Split(std::string(range_text), ","); long int val; std::vector<std::string> pieces = android::base::Split(range_text, ","); if (pieces.size() < 3) { if (pieces.size() < 3) { goto err; goto err; } } Loading Loading @@ -120,7 +113,7 @@ static void parse_range(const char* range_text, RangeSet& rs) { return; return; err: err: fprintf(stderr, "failed to parse range '%s'\n", range_text); fprintf(stderr, "failed to parse range '%s'\n", range_text.c_str()); exit(1); exit(1); } } Loading Loading @@ -360,25 +353,49 @@ static int WriteBlocks(const RangeSet& tgt, const std::vector<uint8_t>& buffer, return 0; return 0; } } // Parameters for transfer list command functions struct CommandParameters { std::vector<std::string> tokens; size_t cpos; const char* cmdname; const char* cmdline; std::string freestash; std::string stashbase; bool canwrite; int createdstash; int fd; bool foundwrites; bool isunresumable; int version; size_t written; NewThreadInfo nti; pthread_t thread; std::vector<uint8_t> buffer; uint8_t* patch_start; }; // Do a source/target load for move/bsdiff/imgdiff in version 1. // Do a source/target load for move/bsdiff/imgdiff in version 1. // 'wordsave' is the save_ptr of a strtok_r()-in-progress. We expect // We expect to parse the remainder of the parameter tokens as: // to parse the remainder of the string as: // // // <src_range> <tgt_range> // <src_range> <tgt_range> // // // The source range is loaded into the provided buffer, reallocating // The source range is loaded into the provided buffer, reallocating // it to make it larger if necessary. // it to make it larger if necessary. static int LoadSrcTgtVersion1(char** wordsave, RangeSet& tgt, size_t& src_blocks, static int LoadSrcTgtVersion1(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, std::vector<uint8_t>& buffer, int fd) { std::vector<uint8_t>& buffer, int fd) { if (params.cpos + 1 >= params.tokens.size()) { fprintf(stderr, "invalid parameters\n"); return -1; } // <src_range> // <src_range> char* word = strtok_r(nullptr, " ", wordsave); RangeSet src; RangeSet src; parse_range(word, src); parse_range(params.tokens[params.cpos++], src); // <tgt_range> // <tgt_range> word = strtok_r(nullptr, " ", wordsave); parse_range(params.tokens[params.cpos++], tgt); parse_range(word, tgt); allocate(src.size * BLOCKSIZE, buffer); allocate(src.size * BLOCKSIZE, buffer); int rc = ReadBlocks(src, buffer, fd); int rc = ReadBlocks(src, buffer, fd); Loading Loading @@ -694,18 +711,15 @@ static int CreateStash(State* state, int maxblocks, const char* blockdev, std::s return 0; // Using existing directory return 0; // Using existing directory } } static int SaveStash(const std::string& base, char** wordsave, std::vector<uint8_t>& buffer, static int SaveStash(CommandParameters& params, const std::string& base, int fd, bool usehash) { std::vector<uint8_t>& buffer, int fd, bool usehash) { if (!wordsave) { return -1; } char *id_tok = strtok_r(nullptr, " ", wordsave); // <stash_id> <src_range> if (id_tok == nullptr) { if (params.cpos + 1 >= params.tokens.size()) { fprintf(stderr, "missing id field in stash command\n"); fprintf(stderr, "missing id and/or src range fields in stash command\n"); return -1; return -1; } } std::string id(id_tok); const std::string& id = params.tokens[params.cpos++]; size_t blocks = 0; size_t blocks = 0; if (usehash && LoadStash(base, id, true, &blocks, buffer, false) == 0) { if (usehash && LoadStash(base, id, true, &blocks, buffer, false) == 0) { Loading @@ -715,9 +729,8 @@ static int SaveStash(const std::string& base, char** wordsave, std::vector<uint8 return 0; return 0; } } char* word = strtok_r(nullptr, " ", wordsave); RangeSet src; RangeSet src; parse_range(word, src); 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) { Loading @@ -738,13 +751,12 @@ static int SaveStash(const std::string& base, char** wordsave, std::vector<uint8 return WriteStash(base, id, blocks, buffer, false, nullptr); return WriteStash(base, id, blocks, buffer, false, nullptr); } } static int FreeStash(const std::string& base, const char* id) { static int FreeStash(const std::string& base, const std::string& id) { if (base.empty() || id == nullptr) { if (base.empty() || id.empty()) { return -1; return -1; } } std::string fn = GetStashFileName(base, std::string(id), ""); std::string fn = GetStashFileName(base, id, ""); DeleteFile(fn, nullptr); DeleteFile(fn, nullptr); return 0; return 0; Loading @@ -768,8 +780,7 @@ static void MoveRange(std::vector<uint8_t>& dest, const RangeSet& locs, } } // Do a source/target load for move/bsdiff/imgdiff in version 2. // Do a source/target load for move/bsdiff/imgdiff in version 2. // 'wordsave' is the save_ptr of a strtok_r()-in-progress. We expect // We expect to parse the remainder of the parameter tokens as one of: // to parse the remainder of the string as one of: // // // <tgt_range> <src_block_count> <src_range> // <tgt_range> <src_block_count> <src_range> // (loads data from source image only) // (loads data from source image only) Loading @@ -785,25 +796,35 @@ static void MoveRange(std::vector<uint8_t>& dest, const RangeSet& locs, // reallocated if needed to accommodate the source data. *tgt is the // reallocated if needed to accommodate the source data. *tgt is the // target RangeSet. Any stashes required are loaded using LoadStash. // target RangeSet. Any stashes required are loaded using LoadStash. static int LoadSrcTgtVersion2(char** wordsave, RangeSet& tgt, size_t& src_blocks, static int LoadSrcTgtVersion2(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, std::vector<uint8_t>& buffer, int fd, const std::string& stashbase, bool* overlap) { std::vector<uint8_t>& buffer, int fd, const std::string& stashbase, bool* overlap) { // At least it needs to provide three parameters: <tgt_range>, // <src_block_count> and "-"/<src_range>. if (params.cpos + 2 >= params.tokens.size()) { fprintf(stderr, "invalid parameters\n"); return -1; } // <tgt_range> // <tgt_range> char* word = strtok_r(nullptr, " ", wordsave); parse_range(params.tokens[params.cpos++], tgt); parse_range(word, tgt); // <src_block_count> // <src_block_count> word = strtok_r(nullptr, " ", wordsave); const std::string& token = params.tokens[params.cpos++]; android::base::ParseUint(word, &src_blocks); if (!android::base::ParseUint(token.c_str(), &src_blocks)) { fprintf(stderr, "invalid src_block_count \"%s\"\n", token.c_str()); return -1; } allocate(src_blocks * BLOCKSIZE, buffer); allocate(src_blocks * BLOCKSIZE, buffer); // "-" or <src_range> [<src_loc>] // "-" or <src_range> [<src_loc>] word = strtok_r(nullptr, " ", wordsave); if (params.tokens[params.cpos] == "-") { if (word[0] == '-' && word[1] == '\0') { // no source ranges, only stashes // no source ranges, only stashes params.cpos++; } else { } else { RangeSet src; RangeSet src; parse_range(word, src); parse_range(params.tokens[params.cpos++], src); int res = ReadBlocks(src, buffer, fd); int res = ReadBlocks(src, buffer, fd); if (overlap) { if (overlap) { Loading @@ -814,39 +835,39 @@ static int LoadSrcTgtVersion2(char** wordsave, RangeSet& tgt, size_t& src_blocks return -1; return -1; } } word = strtok_r(nullptr, " ", wordsave); if (params.cpos >= params.tokens.size()) { if (word == nullptr) { // no stashes, only source range // no stashes, only source range return 0; return 0; } } RangeSet locs; RangeSet locs; parse_range(word, locs); parse_range(params.tokens[params.cpos++], locs); MoveRange(buffer, locs, buffer); MoveRange(buffer, locs, buffer); } } // <[stash_id:stash-range]> // <[stash_id:stash_range]> char* colonsave; while (params.cpos < params.tokens.size()) { while ((word = strtok_r(nullptr, " ", wordsave)) != nullptr) { // Each word is a an index into the stash table, a colon, and // Each word is a an index into the stash table, a colon, and // then a rangeset describing where in the source block that // then a rangeset describing where in the source block that // stashed data should go. // stashed data should go. colonsave = nullptr; std::vector<std::string> tokens = android::base::Split(params.tokens[params.cpos++], ":"); char* colon = strtok_r(word, ":", &colonsave); if (tokens.size() != 2) { fprintf(stderr, "invalid parameter\n"); return -1; } std::vector<uint8_t> stash; std::vector<uint8_t> stash; int res = LoadStash(stashbase, std::string(colon), false, nullptr, stash, true); int res = LoadStash(stashbase, tokens[0], false, nullptr, stash, true); if (res == -1) { if (res == -1) { // These source blocks will fail verification if used later, but we // These source blocks will fail verification if used later, but we // will let the caller decide if this is a fatal failure // will let the caller decide if this is a fatal failure fprintf(stderr, "failed to load stash %s\n", colon); fprintf(stderr, "failed to load stash %s\n", tokens[0].c_str()); continue; continue; } } colon = strtok_r(nullptr, ":", &colonsave); RangeSet locs; RangeSet locs; parse_range(colon, locs); parse_range(tokens[1], locs); MoveRange(buffer, locs, stash); MoveRange(buffer, locs, stash); } } Loading @@ -854,25 +875,6 @@ static int LoadSrcTgtVersion2(char** wordsave, RangeSet& tgt, size_t& src_blocks return 0; return 0; } } // Parameters for transfer list command functions struct CommandParameters { char* cmdname; char* cpos; char* freestash; std::string stashbase; bool canwrite; int createdstash; int fd; bool foundwrites; bool isunresumable; int version; size_t written; NewThreadInfo nti; pthread_t thread; std::vector<uint8_t> buffer; uint8_t* patch_start; }; // Do a source/target load for move/bsdiff/imgdiff in version 3. // Do a source/target load for move/bsdiff/imgdiff in version 3. // // // Parameters are the same as for LoadSrcTgtVersion2, except for 'onehash', which // Parameters are the same as for LoadSrcTgtVersion2, except for 'onehash', which Loading @@ -893,26 +895,26 @@ struct CommandParameters { static int LoadSrcTgtVersion3(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, static int LoadSrcTgtVersion3(CommandParameters& params, RangeSet& tgt, size_t& src_blocks, bool onehash, bool& overlap) { bool onehash, bool& overlap) { char* srchash = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos >= params.tokens.size()) { if (srchash == nullptr) { fprintf(stderr, "missing source hash\n"); fprintf(stderr, "missing source hash\n"); return -1; return -1; } } char* tgthash = nullptr; std::string srchash = params.tokens[params.cpos++]; std::string tgthash; if (onehash) { if (onehash) { tgthash = srchash; tgthash = srchash; } else { } else { tgthash = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos >= params.tokens.size()) { if (tgthash == nullptr) { fprintf(stderr, "missing target hash\n"); fprintf(stderr, "missing target hash\n"); return -1; return -1; } } tgthash = params.tokens[params.cpos++]; } } if (LoadSrcTgtVersion2(¶ms.cpos, tgt, src_blocks, params.buffer, params.fd, if (LoadSrcTgtVersion2(params, tgt, src_blocks, params.buffer, params.fd, params.stashbase, params.stashbase, &overlap) == -1) { &overlap) == -1) { return -1; return -1; } } Loading @@ -931,7 +933,8 @@ static int LoadSrcTgtVersion3(CommandParameters& params, RangeSet& tgt, size_t& // If source and target blocks overlap, stash the source blocks so we can // If source and target blocks overlap, stash the source blocks so we can // resume from possible write errors // resume from possible write errors if (overlap) { if (overlap) { fprintf(stderr, "stashing %zu overlapping blocks to %s\n", src_blocks, srchash); fprintf(stderr, "stashing %zu overlapping blocks to %s\n", src_blocks, srchash.c_str()); bool stash_exists = false; bool stash_exists = false; if (WriteStash(params.stashbase, srchash, src_blocks, params.buffer, true, if (WriteStash(params.stashbase, srchash, src_blocks, params.buffer, true, Loading Loading @@ -971,9 +974,9 @@ static int PerformCommandMove(CommandParameters& params) { RangeSet tgt; RangeSet tgt; if (params.version == 1) { if (params.version == 1) { status = LoadSrcTgtVersion1(¶ms.cpos, tgt, blocks, params.buffer, params.fd); status = LoadSrcTgtVersion1(params, tgt, blocks, params.buffer, params.fd); } else if (params.version == 2) { } else if (params.version == 2) { status = LoadSrcTgtVersion2(¶ms.cpos, tgt, blocks, params.buffer, params.fd, status = LoadSrcTgtVersion2(params, tgt, blocks, params.buffer, params.fd, params.stashbase, nullptr); params.stashbase, nullptr); } else if (params.version >= 3) { } else if (params.version >= 3) { status = LoadSrcTgtVersion3(params, tgt, blocks, true, overlap); status = LoadSrcTgtVersion3(params, tgt, blocks, true, overlap); Loading Loading @@ -1003,9 +1006,9 @@ static int PerformCommandMove(CommandParameters& params) { } } if (params.freestash) { if (!params.freestash.empty()) { FreeStash(params.stashbase, params.freestash); FreeStash(params.stashbase, params.freestash); params.freestash = nullptr; params.freestash.clear(); } } params.written += tgt.size; params.written += tgt.size; Loading @@ -1014,28 +1017,33 @@ static int PerformCommandMove(CommandParameters& params) { } } static int PerformCommandStash(CommandParameters& params) { static int PerformCommandStash(CommandParameters& params) { return SaveStash(params.stashbase, ¶ms.cpos, params.buffer, params.fd, return SaveStash(params, params.stashbase, params.buffer, params.fd, (params.version >= 3)); (params.version >= 3)); } } static int PerformCommandFree(CommandParameters& params) { static int PerformCommandFree(CommandParameters& params) { // <stash_id> if (params.cpos >= params.tokens.size()) { fprintf(stderr, "missing stash id in free command\n"); return -1; } if (params.createdstash || params.canwrite) { if (params.createdstash || params.canwrite) { return FreeStash(params.stashbase, params.cpos); return FreeStash(params.stashbase, params.tokens[params.cpos++]); } } return 0; return 0; } } static int PerformCommandZero(CommandParameters& params) { static int PerformCommandZero(CommandParameters& params) { char* range = strtok_r(nullptr, " ", ¶ms.cpos); if (range == nullptr) { if (params.cpos >= params.tokens.size()) { fprintf(stderr, "missing target blocks for zero\n"); fprintf(stderr, "missing target blocks for zero\n"); return -1; return -1; } } RangeSet tgt; RangeSet tgt; parse_range(range, tgt); parse_range(params.tokens[params.cpos++], tgt); fprintf(stderr, " zeroing %zu blocks\n", tgt.size); fprintf(stderr, " zeroing %zu blocks\n", tgt.size); Loading Loading @@ -1066,14 +1074,14 @@ static int PerformCommandZero(CommandParameters& params) { } } static int PerformCommandNew(CommandParameters& params) { static int PerformCommandNew(CommandParameters& params) { char* range = strtok_r(nullptr, " ", ¶ms.cpos); if (range == nullptr) { if (params.cpos >= params.tokens.size()) { fprintf(stderr, "missing target blocks for new\n"); return -1; return -1; } } RangeSet tgt; RangeSet tgt; parse_range(range, tgt); parse_range(params.tokens[params.cpos++], tgt); if (params.canwrite) { if (params.canwrite) { fprintf(stderr, " writing %zu blocks of new data\n", tgt.size); fprintf(stderr, " writing %zu blocks of new data\n", tgt.size); Loading Loading @@ -1105,33 +1113,32 @@ static int PerformCommandNew(CommandParameters& params) { static int PerformCommandDiff(CommandParameters& params) { static int PerformCommandDiff(CommandParameters& params) { const std::string logparams(params.cpos); // <offset> <length> char* value = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos + 1 >= params.tokens.size()) { fprintf(stderr, "missing patch offset or length for %s\n", params.cmdname); if (value == nullptr) { fprintf(stderr, "missing patch offset for %s\n", params.cmdname); return -1; return -1; } } size_t offset = strtoul(value, nullptr, 0); size_t offset; if (!android::base::ParseUint(params.tokens[params.cpos++].c_str(), &offset)) { value = strtok_r(nullptr, " ", ¶ms.cpos); fprintf(stderr, "invalid patch offset\n"); if (value == nullptr) { fprintf(stderr, "missing patch length for %s\n", params.cmdname); return -1; return -1; } } size_t len = strtoul(value, nullptr, 0); size_t len; if (!android::base::ParseUint(params.tokens[params.cpos++].c_str(), &len)) { fprintf(stderr, "invalid patch offset\n"); return -1; } RangeSet tgt; RangeSet tgt; size_t blocks = 0; size_t blocks = 0; bool overlap = false; bool overlap = false; int status = 0; int status = 0; if (params.version == 1) { if (params.version == 1) { status = LoadSrcTgtVersion1(¶ms.cpos, tgt, blocks, params.buffer, params.fd); status = LoadSrcTgtVersion1(params, tgt, blocks, params.buffer, params.fd); } else if (params.version == 2) { } else if (params.version == 2) { status = LoadSrcTgtVersion2(¶ms.cpos, tgt, blocks, params.buffer, params.fd, status = LoadSrcTgtVersion2(params, tgt, blocks, params.buffer, params.fd, params.stashbase, nullptr); params.stashbase, nullptr); } else if (params.version >= 3) { } else if (params.version >= 3) { status = LoadSrcTgtVersion3(params, tgt, blocks, false, overlap); status = LoadSrcTgtVersion3(params, tgt, blocks, false, overlap); Loading Loading @@ -1180,13 +1187,13 @@ static int PerformCommandDiff(CommandParameters& params) { } } } else { } else { fprintf(stderr, "skipping %zu blocks already patched to %zu [%s]\n", fprintf(stderr, "skipping %zu blocks already patched to %zu [%s]\n", blocks, tgt.size, logparams.c_str()); blocks, tgt.size, params.cmdline); } } } } if (params.freestash) { if (!params.freestash.empty()) { FreeStash(params.stashbase, params.freestash); FreeStash(params.stashbase, params.freestash); params.freestash = nullptr; params.freestash.clear(); } } params.written += tgt.size; params.written += tgt.size; Loading @@ -1210,15 +1217,13 @@ static int PerformCommandErase(CommandParameters& params) { return -1; return -1; } } char* range = strtok_r(nullptr, " ", ¶ms.cpos); if (params.cpos >= params.tokens.size()) { if (range == nullptr) { fprintf(stderr, "missing target blocks for erase\n"); fprintf(stderr, "missing target blocks for erase\n"); return -1; return -1; } } RangeSet tgt; RangeSet tgt; parse_range(range, tgt); parse_range(params.tokens[params.cpos++], tgt); if (params.canwrite) { if (params.canwrite) { fprintf(stderr, " erasing %zu blocks\n", tgt.size); fprintf(stderr, " erasing %zu blocks\n", tgt.size); Loading Loading @@ -1278,7 +1283,6 @@ static unsigned int HashString(const char *s) { static Value* PerformBlockImageUpdate(const char* name, State* state, int /* argc */, Expr* argv[], static Value* PerformBlockImageUpdate(const char* name, State* state, int /* argc */, Expr* argv[], const Command* commands, size_t cmdcount, bool dryrun) { const Command* commands, size_t cmdcount, bool dryrun) { CommandParameters params; CommandParameters params; memset(¶ms, 0, sizeof(params)); memset(¶ms, 0, sizeof(params)); params.canwrite = !dryrun; params.canwrite = !dryrun; Loading Loading @@ -1368,10 +1372,14 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg } } } } // The data in transfer_list_value is not necessarily null-terminated, so we need // Copy all the lines in transfer_list_value into std::string for // to copy it to a new buffer and add the null that strtok_r will need. // processing. const std::string transfer_list(transfer_list_value->data, transfer_list_value->size); const std::string transfer_list(transfer_list_value->data, transfer_list_value->size); std::vector<std::string> lines = android::base::Split(transfer_list, "\n"); std::vector<std::string> lines = android::base::Split(transfer_list, "\n"); if (lines.size() < 2) { ErrorAbort(state, "too few lines in the transfer list [%zd]\n", lines.size()); return StringValue(strdup("")); } // First line in transfer list is the version number // First line in transfer list is the version number if (!android::base::ParseInt(lines[0].c_str(), ¶ms.version, 1, 4)) { if (!android::base::ParseInt(lines[0].c_str(), ¶ms.version, 1, 4)) { Loading @@ -1394,6 +1402,11 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg size_t start = 2; size_t start = 2; if (params.version >= 2) { if (params.version >= 2) { if (lines.size() < 4) { ErrorAbort(state, "too few lines in the transfer list [%zu]\n", lines.size()); return StringValue(strdup("")); } // Third line is how many stash entries are needed simultaneously // Third line is how many stash entries are needed simultaneously fprintf(stderr, "maximum stash entries %s\n", lines[2].c_str()); fprintf(stderr, "maximum stash entries %s\n", lines[2].c_str()); Loading @@ -1405,7 +1418,6 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg } } int res = CreateStash(state, stash_max_blocks, blockdev_filename->data, params.stashbase); int res = CreateStash(state, stash_max_blocks, blockdev_filename->data, params.stashbase); if (res == -1) { if (res == -1) { return StringValue(strdup("")); return StringValue(strdup("")); } } Loading Loading @@ -1433,17 +1445,15 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg continue; continue; } } char* line = strdup(line_str.c_str()); params.tokens = android::base::Split(line_str, " "); params.cmdname = strtok_r(line, " ", ¶ms.cpos); params.cpos = 0; params.cmdname = params.tokens[params.cpos++].c_str(); if (params.cmdname == nullptr) { params.cmdline = line_str.c_str(); fprintf(stderr, "missing command [%s]\n", line); goto pbiudone; } unsigned int cmdhash = HashString(params.cmdname); unsigned int cmdhash = HashString(params.cmdname); const Command* cmd = reinterpret_cast<const Command*>(mzHashTableLookup(cmdht, cmdhash, const Command* cmd = reinterpret_cast<const Command*>(mzHashTableLookup(cmdht, cmdhash, params.cmdname, CompareCommandNames, false)); const_cast<char*>(params.cmdname), CompareCommandNames, false)); if (cmd == nullptr) { if (cmd == nullptr) { fprintf(stderr, "unexpected command [%s]\n", params.cmdname); fprintf(stderr, "unexpected command [%s]\n", params.cmdname); Loading Loading @@ -1481,12 +1491,10 @@ static Value* PerformBlockImageUpdate(const char* name, State* state, int /* arg rc = 0; rc = 0; pbiudone: pbiudone: if (params.fd != -1) { if (fsync(params.fd) == -1) { if (fsync(params.fd) == -1) { fprintf(stderr, "fsync failed: %s\n", strerror(errno)); fprintf(stderr, "fsync failed: %s\n", strerror(errno)); } } close(params.fd); // params.fd will be automatically closed because of the fd_holder above. } // Only delete the stash if the update cannot be resumed, or it's // Only delete the stash if the update cannot be resumed, or it's // a verification run and we created the stash. // a verification run and we created the stash. Loading Loading @@ -1526,6 +1534,9 @@ pbiudone: // - (version 2+ only) load the given source range and stash // - (version 2+ only) load the given source range and stash // the data in the given slot of the stash table. // the data in the given slot of the stash table. // // // free <stash_id> // - (version 3+ only) free the given stash data. // // The creator of the transfer list will guarantee that no block // The creator of the transfer list will guarantee that no block // is read (ie, used as the source for a patch or move) after it // is read (ie, used as the source for a patch or move) after it // has been written. // has been written. Loading Loading @@ -1692,7 +1703,7 @@ Value* BlockImageRecoverFn(const char* name, State* state, int argc, Expr* argv[ } } if (fh.pread(buffer, BLOCKSIZE, (off64_t)j * BLOCKSIZE) != BLOCKSIZE) { if (fh.pread(buffer, BLOCKSIZE, (off64_t)j * BLOCKSIZE) != BLOCKSIZE) { ErrorAbort(state, "failed to recover %s (block %d): %s", filename->data, ErrorAbort(state, "failed to recover %s (block %zu): %s", filename->data, j, strerror(errno)); j, strerror(errno)); return StringValue(strdup("")); return StringValue(strdup("")); } } Loading