1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/installer/setup/archive_patch_helper.h"
#include "base/file_util.h"
#include "base/logging.h"
#include "chrome/installer/util/lzma_util.h"
#include "courgette/courgette.h"
#include "third_party/bspatch/mbspatch.h"
namespace installer {
ArchivePatchHelper::ArchivePatchHelper(const base::FilePath& working_directory,
const base::FilePath& compressed_archive,
const base::FilePath& patch_source,
const base::FilePath& target)
: working_directory_(working_directory),
compressed_archive_(compressed_archive),
patch_source_(patch_source),
target_(target) {}
ArchivePatchHelper::~ArchivePatchHelper() {}
// static
bool ArchivePatchHelper::UncompressAndPatch(
const base::FilePath& working_directory,
const base::FilePath& compressed_archive,
const base::FilePath& patch_source,
const base::FilePath& target) {
ArchivePatchHelper instance(working_directory, compressed_archive,
patch_source, target);
return (instance.Uncompress(NULL) &&
(instance.EnsemblePatch() || instance.BinaryPatch()));
}
bool ArchivePatchHelper::Uncompress(base::FilePath* last_uncompressed_file) {
// The target shouldn't already exist.
DCHECK(!base::PathExists(target_));
// UnPackArchive takes care of logging.
base::string16 output_file;
int32 lzma_result = LzmaUtil::UnPackArchive(compressed_archive_.value(),
working_directory_.value(),
&output_file);
if (lzma_result != NO_ERROR)
return false;
last_uncompressed_file_ = base::FilePath(output_file);
if (last_uncompressed_file)
*last_uncompressed_file = last_uncompressed_file_;
return true;
}
bool ArchivePatchHelper::EnsemblePatch() {
if (last_uncompressed_file_.empty()) {
LOG(ERROR) << "No patch file found in compressed archive.";
return false;
}
courgette::Status result =
courgette::ApplyEnsemblePatch(patch_source_.value().c_str(),
last_uncompressed_file_.value().c_str(),
target_.value().c_str());
if (result == courgette::C_OK)
return true;
LOG(ERROR)
<< "Failed to apply patch " << last_uncompressed_file_.value()
<< " to file " << patch_source_.value()
<< " and generating file " << target_.value()
<< " using courgette. err=" << result;
// Ensure a partial output is not left behind.
base::DeleteFile(target_, false);
return false;
}
bool ArchivePatchHelper::BinaryPatch() {
if (last_uncompressed_file_.empty()) {
LOG(ERROR) << "No patch file found in compressed archive.";
return false;
}
int result = ApplyBinaryPatch(patch_source_.value().c_str(),
last_uncompressed_file_.value().c_str(),
target_.value().c_str());
if (result == OK)
return true;
LOG(ERROR)
<< "Failed to apply patch " << last_uncompressed_file_.value()
<< " to file " << patch_source_.value()
<< " and generating file " << target_.value()
<< " using bsdiff. err=" << result;
// Ensure a partial output is not left behind.
base::DeleteFile(target_, false);
return false;
}
} // namespace installer
|
