PawletOS/android_bootable_recovery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge changes from topic 'sinkfn' am: 34df98ee6d

Browse Source 
am: 862b7e8523

Change-Id: I859296b62fcd7b43775885f4d2ece3272a9bff11
This commit is contained in:
Tao Bao
2017-03-28 19:16:34 +00:00
committed by android-build-merger
parent 52569ce2ed 862b7e8523
commit a2015e0d86
7 changed files with 127 additions and 160 deletions
Download Patch File Download Diff File Expand all files Collapse all files
applypatch/applypatch.cpp
+10 -14
View File
@@ -27,6 +27,7 @@
#include <sys/types.h> #include <sys/types.h>
#include <unistd.h> #include <unistd.h>
#include <functional>
#include <memory> #include <memory>
#include <string> #include <string>
#include <utility> #include <utility>
@@ -42,7 +43,7 @@
#include "print_sha1.h" #include "print_sha1.h"
static int LoadPartitionContents(const std::string& filename, FileContents* file); static int LoadPartitionContents(const std::string& filename, FileContents* file);
static size_t FileSink(const unsigned char* data, size_t len, void* token); static size_t FileSink(const unsigned char* data, size_t len, int fd);
static int GenerateTarget(const FileContents& source_file, const std::unique_ptr<Value>& patch, static int GenerateTarget(const FileContents& source_file, const std::unique_ptr<Value>& patch,
const std::string& target_filename, const std::string& target_filename,
const uint8_t target_sha1[SHA_DIGEST_LENGTH], const Value* bonus_data); const uint8_t target_sha1[SHA_DIGEST_LENGTH], const Value* bonus_data);
@@ -194,7 +195,7 @@ int SaveFileContents(const char* filename, const FileContents* file) {
return -1; return -1;
} }
size_t bytes_written = FileSink(file->data.data(), file->data.size(), &fd); size_t bytes_written = FileSink(file->data.data(), file->data.size(), fd);
if (bytes_written != file->data.size()) { if (bytes_written != file->data.size()) {
printf("short write of \"%s\" (%zd bytes of %zu): %s\n", filename, bytes_written, printf("short write of \"%s\" (%zd bytes of %zu): %s\n", filename, bytes_written,
file->data.size(), strerror(errno)); file->data.size(), strerror(errno));
@@ -433,8 +434,7 @@ int ShowLicenses() {
return 0; return 0;
} }
static size_t FileSink(const unsigned char* data, size_t len, void* token) { static size_t FileSink(const unsigned char* data, size_t len, int fd) {
int fd = *static_cast<int*>(token);
size_t done = 0; size_t done = 0;
while (done < len) { while (done < len) {
ssize_t wrote = TEMP_FAILURE_RETRY(ota_write(fd, data + done, len - done)); ssize_t wrote = TEMP_FAILURE_RETRY(ota_write(fd, data + done, len - done));
@@ -447,12 +447,6 @@ static size_t FileSink(const unsigned char* data, size_t len, void* token) {
return done; return done;
} }
size_t MemorySink(const unsigned char* data, size_t len, void* token) {
std::string* s = static_cast<std::string*>(token);
s->append(reinterpret_cast<const char*>(data), len);
return len;
}
// Return the amount of free space (in bytes) on the filesystem // Return the amount of free space (in bytes) on the filesystem
// containing filename. filename must exist. Return -1 on error. // containing filename. filename must exist. Return -1 on error.
size_t FreeSpaceForFile(const char* filename) { size_t FreeSpaceForFile(const char* filename) {
@@ -646,9 +640,11 @@ static int GenerateTarget(const FileContents& source_file, const std::unique_ptr
} }
// We store the decoded output in memory. // We store the decoded output in memory.
SinkFn sink = MemorySink;
std::string memory_sink_str; // Don't need to reserve space. std::string memory_sink_str; // Don't need to reserve space.
void* token = &memory_sink_str; SinkFn sink = [&memory_sink_str](const unsigned char* data, size_t len) {
memory_sink_str.append(reinterpret_cast<const char*>(data), len);
return len;
};
SHA_CTX ctx; SHA_CTX ctx;
SHA1_Init(&ctx); SHA1_Init(&ctx);
@@ -656,10 +652,10 @@ static int GenerateTarget(const FileContents& source_file, const std::unique_ptr
int result; int result;
if (use_bsdiff) { if (use_bsdiff) {
result = ApplyBSDiffPatch(source_file.data.data(), source_file.data.size(), patch.get(), 0, result = ApplyBSDiffPatch(source_file.data.data(), source_file.data.size(), patch.get(), 0,
sink, token, &ctx); sink, &ctx);
} else { } else {
result = ApplyImagePatch(source_file.data.data(), source_file.data.size(), patch.get(), sink, result = ApplyImagePatch(source_file.data.data(), source_file.data.size(), patch.get(), sink,
token, &ctx, bonus_data); &ctx, bonus_data);
} }
if (result != 0) { if (result != 0) {
applypatch/bspatch.cpp
+4 -4
View File
@@ -24,9 +24,9 @@
#include <sys/types.h> #include <sys/types.h>
#include <bspatch.h> #include <bspatch.h>
#include <openssl/sha.h>
#include "applypatch/applypatch.h" #include "applypatch/applypatch.h"
#include "openssl/sha.h"
void ShowBSDiffLicense() { void ShowBSDiffLicense() {
puts("The bsdiff library used herein is:\n" puts("The bsdiff library used herein is:\n"
@@ -61,9 +61,9 @@ void ShowBSDiffLicense() {
} }
int ApplyBSDiffPatch(const unsigned char* old_data, size_t old_size, const Value* patch, int ApplyBSDiffPatch(const unsigned char* old_data, size_t old_size, const Value* patch,
size_t patch_offset, SinkFn sink, void* token, SHA_CTX* ctx) { size_t patch_offset, SinkFn sink, SHA_CTX* ctx) {
auto sha_sink = [&](const uint8_t* data, size_t len) { auto sha_sink = [&sink, &ctx](const uint8_t* data, size_t len) {
len = sink(data, len, token); len = sink(data, len);
if (ctx) SHA1_Update(ctx, data, len); if (ctx) SHA1_Update(ctx, data, len);
return len; return len;
}; };
applypatch/imgpatch.cpp
+6 -7
View File
@@ -44,10 +44,10 @@ static inline int32_t Read4(const void *address) {
} }
int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const unsigned char* patch_data, int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const unsigned char* patch_data,
size_t patch_size, SinkFn sink, void* token) { size_t patch_size, SinkFn sink) {
Value patch(VAL_BLOB, std::string(reinterpret_cast<const char*>(patch_data), patch_size)); Value patch(VAL_BLOB, std::string(reinterpret_cast<const char*>(patch_data), patch_size));
return ApplyImagePatch(old_data, old_size, &patch, sink, token, nullptr, nullptr); return ApplyImagePatch(old_data, old_size, &patch, sink, nullptr, nullptr);
} }
/* /*
@@ -57,7 +57,7 @@ int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const unsign
* Return 0 on success. * Return 0 on success.
*/ */
int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value* patch, SinkFn sink, int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value* patch, SinkFn sink,
void* token, SHA_CTX* ctx, const Value* bonus_data) { SHA_CTX* ctx, const Value* bonus_data) {
if (patch->data.size() < 12) { if (patch->data.size() < 12) {
printf("patch too short to contain header\n"); printf("patch too short to contain header\n");
return -1; return -1;
@@ -100,7 +100,7 @@ int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value*
printf("source data too short\n"); printf("source data too short\n");
return -1; return -1;
} }
ApplyBSDiffPatch(old_data + src_start, src_len, patch, patch_offset, sink, token, ctx); ApplyBSDiffPatch(old_data + src_start, src_len, patch, patch_offset, sink, ctx);
} else if (type == CHUNK_RAW) { } else if (type == CHUNK_RAW) {
const char* raw_header = &patch->data[pos]; const char* raw_header = &patch->data[pos];
pos += 4; pos += 4;
@@ -116,8 +116,7 @@ int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value*
return -1; return -1;
} }
if (ctx) SHA1_Update(ctx, &patch->data[pos], data_len); if (ctx) SHA1_Update(ctx, &patch->data[pos], data_len);
if (sink(reinterpret_cast<const unsigned char*>(&patch->data[pos]), data_len, token) != if (sink(reinterpret_cast<const unsigned char*>(&patch->data[pos]), data_len) != data_len) {
data_len) {
printf("failed to write chunk %d raw data\n", i); printf("failed to write chunk %d raw data\n", i);
return -1; return -1;
} }
@@ -241,7 +240,7 @@ int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value*
ret = deflate(&strm, Z_FINISH); ret = deflate(&strm, Z_FINISH);
size_t have = temp_data.size() - strm.avail_out; size_t have = temp_data.size() - strm.avail_out;
if (sink(temp_data.data(), have, token) != have) { if (sink(temp_data.data(), have) != have) {
printf("failed to write %zd compressed bytes to output\n", have); printf("failed to write %zd compressed bytes to output\n", have);
return -1; return -1;
} }
applypatch/include/applypatch/applypatch.h
+4 -3
View File
@@ -20,6 +20,7 @@
#include <stdint.h> #include <stdint.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <functional>
#include <memory> #include <memory>
#include <string> #include <string>
#include <vector> #include <vector>
@@ -41,7 +42,7 @@ struct FileContents {
// and use it as the source instead. // and use it as the source instead.
#define CACHE_TEMP_SOURCE "/cache/saved.file" #define CACHE_TEMP_SOURCE "/cache/saved.file"
using SinkFn = size_t (*)(const unsigned char*, size_t, void*); using SinkFn = std::function<size_t(const unsigned char*, size_t)>;
// applypatch.cpp // applypatch.cpp
int ShowLicenses(); int ShowLicenses();
@@ -67,13 +68,13 @@ int SaveFileContents(const char* filename, const FileContents* file);
// bspatch.cpp // bspatch.cpp
void ShowBSDiffLicense(); void ShowBSDiffLicense();
int ApplyBSDiffPatch(const unsigned char* old_data, size_t old_size, const Value* patch, int ApplyBSDiffPatch(const unsigned char* old_data, size_t old_size, const Value* patch,
size_t patch_offset, SinkFn sink, void* token, SHA_CTX* ctx); size_t patch_offset, SinkFn sink, SHA_CTX* ctx);
int ApplyBSDiffPatchMem(const unsigned char* old_data, size_t old_size, const Value* patch, int ApplyBSDiffPatchMem(const unsigned char* old_data, size_t old_size, const Value* patch,
size_t patch_offset, std::vector<unsigned char>* new_data); size_t patch_offset, std::vector<unsigned char>* new_data);
// imgpatch.cpp // imgpatch.cpp
int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value* patch, SinkFn sink, int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const Value* patch, SinkFn sink,
void* token, SHA_CTX* ctx, const Value* bonus_data); SHA_CTX* ctx, const Value* bonus_data);
// freecache.cpp // freecache.cpp
int MakeFreeSpaceOnCache(size_t bytes_needed); int MakeFreeSpaceOnCache(size_t bytes_needed);
applypatch/include/applypatch/imgpatch.h
+4 -2
View File
@@ -19,9 +19,11 @@
#include <sys/types.h> #include <sys/types.h>
using SinkFn = size_t (*)(const unsigned char*, size_t, void*); #include <functional>
using SinkFn = std::function<size_t(const unsigned char*, size_t)>;
int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const unsigned char* patch_data, int ApplyImagePatch(const unsigned char* old_data, size_t old_size, const unsigned char* patch_data,
size_t patch_size, SinkFn sink, void* token); size_t patch_size, SinkFn sink);
#endif // _APPLYPATCH_IMGPATCH_H #endif // _APPLYPATCH_IMGPATCH_H
tests/component/imgdiff_test.cpp
+24 -56
View File
@@ -14,6 +14,8 @@
* limitations under the License. * limitations under the License.
*/ */
#include <stdio.h>
#include <string> #include <string>
#include <vector> #include <vector>
@@ -27,12 +29,6 @@
using android::base::get_unaligned; using android::base::get_unaligned;
static size_t MemorySink(const unsigned char* data, size_t len, void* token) {
std::string* s = static_cast<std::string*>(token);
s->append(reinterpret_cast<const char*>(data), len);
return len;
}
// Sanity check for the given imgdiff patch header. // Sanity check for the given imgdiff patch header.
static void verify_patch_header(const std::string& patch, size_t* num_normal, size_t* num_raw, static void verify_patch_header(const std::string& patch, size_t* num_normal, size_t* num_raw,
size_t* num_deflate) { size_t* num_deflate) {
@@ -79,6 +75,18 @@ static void verify_patch_header(const std::string& patch, size_t* num_normal, si
if (num_deflate != nullptr) *num_deflate = deflate; if (num_deflate != nullptr) *num_deflate = deflate;
} }
static void verify_patched_image(const std::string& src, const std::string& patch,
const std::string& tgt) {
std::string patched;
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
[&patched](const unsigned char* data, size_t len) {
patched.append(reinterpret_cast<const char*>(data), len);
return len;
}));
ASSERT_EQ(tgt, patched);
}
TEST(ImgdiffTest, invalid_args) { TEST(ImgdiffTest, invalid_args) {
// Insufficient inputs. // Insufficient inputs.
ASSERT_EQ(2, imgdiff(1, (const char* []){ "imgdiff" })); ASSERT_EQ(2, imgdiff(1, (const char* []){ "imgdiff" }));
@@ -124,11 +132,7 @@ TEST(ImgdiffTest, image_mode_smoke) {
ASSERT_EQ(0U, num_deflate); ASSERT_EQ(0U, num_deflate);
ASSERT_EQ(1U, num_raw); ASSERT_EQ(1U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, zip_mode_smoke_store) { TEST(ImgdiffTest, zip_mode_smoke_store) {
@@ -177,11 +181,7 @@ TEST(ImgdiffTest, zip_mode_smoke_store) {
ASSERT_EQ(0U, num_deflate); ASSERT_EQ(0U, num_deflate);
ASSERT_EQ(1U, num_raw); ASSERT_EQ(1U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, zip_mode_smoke_compressed) { TEST(ImgdiffTest, zip_mode_smoke_compressed) {
@@ -230,11 +230,7 @@ TEST(ImgdiffTest, zip_mode_smoke_compressed) {
ASSERT_EQ(1U, num_deflate); ASSERT_EQ(1U, num_deflate);
ASSERT_EQ(2U, num_raw); ASSERT_EQ(2U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, zip_mode_smoke_trailer_zeros) { TEST(ImgdiffTest, zip_mode_smoke_trailer_zeros) {
@@ -286,11 +282,7 @@ TEST(ImgdiffTest, zip_mode_smoke_trailer_zeros) {
ASSERT_EQ(1U, num_deflate); ASSERT_EQ(1U, num_deflate);
ASSERT_EQ(2U, num_raw); ASSERT_EQ(2U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, image_mode_simple) { TEST(ImgdiffTest, image_mode_simple) {
@@ -333,11 +325,7 @@ TEST(ImgdiffTest, image_mode_simple) {
ASSERT_EQ(1U, num_deflate); ASSERT_EQ(1U, num_deflate);
ASSERT_EQ(2U, num_raw); ASSERT_EQ(2U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, image_mode_different_num_chunks) { TEST(ImgdiffTest, image_mode_different_num_chunks) {
@@ -413,11 +401,7 @@ TEST(ImgdiffTest, image_mode_merge_chunks) {
ASSERT_EQ(1U, num_deflate); ASSERT_EQ(1U, num_deflate);
ASSERT_EQ(2U, num_raw); ASSERT_EQ(2U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, image_mode_spurious_magic) { TEST(ImgdiffTest, image_mode_spurious_magic) {
@@ -454,11 +438,7 @@ TEST(ImgdiffTest, image_mode_spurious_magic) {
ASSERT_EQ(0U, num_deflate); ASSERT_EQ(0U, num_deflate);
ASSERT_EQ(1U, num_raw); ASSERT_EQ(1U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, image_mode_short_input1) { TEST(ImgdiffTest, image_mode_short_input1) {
@@ -494,11 +474,7 @@ TEST(ImgdiffTest, image_mode_short_input1) {
ASSERT_EQ(0U, num_deflate); ASSERT_EQ(0U, num_deflate);
ASSERT_EQ(1U, num_raw); ASSERT_EQ(1U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, image_mode_short_input2) { TEST(ImgdiffTest, image_mode_short_input2) {
@@ -534,11 +510,7 @@ TEST(ImgdiffTest, image_mode_short_input2) {
ASSERT_EQ(0U, num_deflate); ASSERT_EQ(0U, num_deflate);
ASSERT_EQ(1U, num_raw); ASSERT_EQ(1U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
TEST(ImgdiffTest, image_mode_single_entry_long) { TEST(ImgdiffTest, image_mode_single_entry_long) {
@@ -577,9 +549,5 @@ TEST(ImgdiffTest, image_mode_single_entry_long) {
ASSERT_EQ(0U, num_deflate); ASSERT_EQ(0U, num_deflate);
ASSERT_EQ(0U, num_raw); ASSERT_EQ(0U, num_raw);
std::string patched; verify_patched_image(src, patch, tgt);
ASSERT_EQ(0, ApplyImagePatch(reinterpret_cast<const unsigned char*>(src.data()), src.size(),
reinterpret_cast<const unsigned char*>(patch.data()), patch.size(),
MemorySink, &patched));
ASSERT_EQ(tgt, patched);
} }
updater/blockimg.cpp
+75 -74
View File
@@ -240,9 +240,7 @@ struct RangeSinkState {
size_t p_remain; size_t p_remain;
}; };
static size_t RangeSinkWrite(const uint8_t* data, size_t size, void* token) { static size_t RangeSinkWrite(const uint8_t* data, size_t size, RangeSinkState* rss) {
RangeSinkState* rss = static_cast<RangeSinkState*>(token);
if (rss->p_remain == 0) { if (rss->p_remain == 0) {
LOG(ERROR) << "range sink write overrun"; LOG(ERROR) << "range sink write overrun";
return 0; return 0;
@@ -1258,92 +1256,95 @@ static int PerformCommandNew(CommandParameters& params) {
} }
static int PerformCommandDiff(CommandParameters& params) { static int PerformCommandDiff(CommandParameters& params) {
// <offset> <length>
if (params.cpos + 1 >= params.tokens.size()) {
LOG(ERROR) << "missing patch offset or length for " << params.cmdname;
return -1;
}
// <offset> <length> size_t offset;
if (params.cpos + 1 >= params.tokens.size()) { if (!android::base::ParseUint(params.tokens[params.cpos++], &offset)) {
LOG(ERROR) << "missing patch offset or length for " << params.cmdname; LOG(ERROR) << "invalid patch offset";
return -1; return -1;
} }
size_t offset; size_t len;
if (!android::base::ParseUint(params.tokens[params.cpos++], &offset)) { if (!android::base::ParseUint(params.tokens[params.cpos++], &len)) {
LOG(ERROR) << "invalid patch offset"; LOG(ERROR) << "invalid patch len";
return -1; return -1;
} }
size_t len; RangeSet tgt;
if (!android::base::ParseUint(params.tokens[params.cpos++], &len)) { size_t blocks = 0;
LOG(ERROR) << "invalid patch len"; bool overlap = false;
return -1; int status = LoadSrcTgtVersion3(params, tgt, &blocks, false, &overlap);
}
RangeSet tgt; if (status == -1) {
size_t blocks = 0; LOG(ERROR) << "failed to read blocks for diff";
bool overlap = false; return -1;
int status = LoadSrcTgtVersion3(params, tgt, &blocks, false, &overlap); }
if (status == -1) { if (status == 0) {
LOG(ERROR) << "failed to read blocks for diff"; params.foundwrites = true;
return -1; } else if (params.foundwrites) {
} LOG(WARNING) << "warning: commands executed out of order [" << params.cmdname << "]";
}
if (params.canwrite) {
if (status == 0) { if (status == 0) {
params.foundwrites = true; LOG(INFO) << "patching " << blocks << " blocks to " << tgt.size;
} else if (params.foundwrites) { Value patch_value(
LOG(WARNING) << "warning: commands executed out of order [" << params.cmdname << "]"; VAL_BLOB, std::string(reinterpret_cast<const char*>(params.patch_start + offset), len));
} RangeSinkState rss(tgt);
rss.fd = params.fd;
rss.p_block = 0;
rss.p_remain = (tgt.pos[1] - tgt.pos[0]) * BLOCKSIZE;
if (params.canwrite) { off64_t offset = static_cast<off64_t>(tgt.pos[0]) * BLOCKSIZE;
if (status == 0) { if (!discard_blocks(params.fd, offset, rss.p_remain)) {
LOG(INFO) << "patching " << blocks << " blocks to " << tgt.size; return -1;
Value patch_value(VAL_BLOB, }
std::string(reinterpret_cast<const char*>(params.patch_start + offset), len));
RangeSinkState rss(tgt);
rss.fd = params.fd;
rss.p_block = 0;
rss.p_remain = (tgt.pos[1] - tgt.pos[0]) * BLOCKSIZE;
off64_t offset = static_cast<off64_t>(tgt.pos[0]) * BLOCKSIZE; if (!check_lseek(params.fd, offset, SEEK_SET)) {
if (!discard_blocks(params.fd, offset, rss.p_remain)) { return -1;
return -1; }
}
if (!check_lseek(params.fd, offset, SEEK_SET)) { if (params.cmdname[0] == 'i') { // imgdiff
return -1; if (ApplyImagePatch(
} params.buffer.data(), blocks * BLOCKSIZE, &patch_value,
std::bind(&RangeSinkWrite, std::placeholders::_1, std::placeholders::_2, &rss),
if (params.cmdname[0] == 'i') { // imgdiff nullptr, nullptr) != 0) {
if (ApplyImagePatch(params.buffer.data(), blocks * BLOCKSIZE, &patch_value, LOG(ERROR) << "Failed to apply image patch.";
&RangeSinkWrite, &rss, nullptr, nullptr) != 0) { return -1;
LOG(ERROR) << "Failed to apply image patch.";
return -1;
}
} else {
if (ApplyBSDiffPatch(params.buffer.data(), blocks * BLOCKSIZE, &patch_value,
0, &RangeSinkWrite, &rss, nullptr) != 0) {
LOG(ERROR) << "Failed to apply bsdiff patch.";
return -1;
}
}
// We expect the output of the patcher to fill the tgt ranges exactly.
if (rss.p_block != tgt.count || rss.p_remain != 0) {
LOG(ERROR) << "range sink underrun?";
}
} else {
LOG(INFO) << "skipping " << blocks << " blocks already patched to " << tgt.size
<< " [" << params.cmdline << "]";
} }
} else {
if (ApplyBSDiffPatch(
params.buffer.data(), blocks * BLOCKSIZE, &patch_value, 0,
std::bind(&RangeSinkWrite, std::placeholders::_1, std::placeholders::_2, &rss),
nullptr) != 0) {
LOG(ERROR) << "Failed to apply bsdiff patch.";
return -1;
}
}
// We expect the output of the patcher to fill the tgt ranges exactly.
if (rss.p_block != tgt.count || rss.p_remain != 0) {
LOG(ERROR) << "range sink underrun?";
}
} else {
LOG(INFO) << "skipping " << blocks << " blocks already patched to " << tgt.size << " ["
<< params.cmdline << "]";
} }
}
if (!params.freestash.empty()) { if (!params.freestash.empty()) {
FreeStash(params.stashbase, params.freestash); FreeStash(params.stashbase, params.freestash);
params.freestash.clear(); params.freestash.clear();
} }
params.written += tgt.size; params.written += tgt.size;
return 0; return 0;
} }
static int PerformCommandErase(CommandParameters& params) { static int PerformCommandErase(CommandParameters& params) {