blob: eca216fedfcecadb09d17c5da3ad0ab38b878690 [file] [log] [blame]
Alex Deymoaea4c1c2015-08-19 20:24:43 -07001//
2// Copyright (C) 2012 The Android Open Source Project
3//
4// Licensed under the Apache License, Version 2.0 (the "License");
5// you may not use this file except in compliance with the License.
6// You may obtain a copy of the License at
7//
8// http://www.apache.org/licenses/LICENSE-2.0
9//
10// Unless required by applicable law or agreed to in writing, software
11// distributed under the License is distributed on an "AS IS" BASIS,
12// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13// See the License for the specific language governing permissions and
14// limitations under the License.
15//
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070016
Alex Deymo8427b4a2014-11-05 14:00:32 -080017#include "update_engine/delta_performer.h"
18
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070019#include <inttypes.h>
20
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070021#include <string>
22#include <vector>
23
Allie Wood78750a42015-02-11 15:42:11 -080024#include <base/files/file_path.h>
Ben Chan06c76a42014-09-05 08:21:06 -070025#include <base/files/file_util.h>
Alex Vakulenko75039d72014-03-25 12:36:28 -070026#include <base/strings/stringprintf.h>
Alex Deymof1cbe172015-03-05 15:58:37 -080027#include <base/strings/string_util.h>
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070028#include <google/protobuf/repeated_field.h>
29#include <gtest/gtest.h>
30
Sen Jiang2d528b42015-09-25 11:18:12 -070031#include "update_engine/bzip.h"
Jay Srinivasand29695d2013-04-08 15:08:05 -070032#include "update_engine/constants.h"
David Zeuthene7f89172013-10-31 10:21:04 -070033#include "update_engine/fake_hardware.h"
Sen Jiang2d528b42015-09-25 11:18:12 -070034#include "update_engine/fake_prefs.h"
Gilad Arnold5bb4c902014-04-10 12:32:13 -070035#include "update_engine/fake_system_state.h"
Alex Deymo161c4a12014-05-16 15:56:21 -070036#include "update_engine/payload_constants.h"
Sen Jiang2d528b42015-09-25 11:18:12 -070037#include "update_engine/payload_generator/extent_ranges.h"
Sen Jiangd78b3892015-09-25 15:19:31 -070038#include "update_engine/payload_generator/payload_file.h"
Alex Deymo923d8fa2014-07-15 17:58:51 -070039#include "update_engine/payload_generator/payload_signer.h"
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070040#include "update_engine/test_utils.h"
41#include "update_engine/update_metadata.pb.h"
42#include "update_engine/utils.h"
43
44namespace chromeos_update_engine {
45
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070046using std::string;
47using std::vector;
Darin Petkov73058b42010-10-06 16:32:19 -070048using testing::Return;
Alex Deymo161c4a12014-05-16 15:56:21 -070049using testing::_;
Alex Deymo10875d92014-11-10 21:52:57 -080050using test_utils::kRandomString;
Alex Deymo10875d92014-11-10 21:52:57 -080051using test_utils::System;
Andrew de los Reyes09e56d62010-04-23 13:45:53 -070052
Andrew de los Reyes932bc4c2010-08-23 18:14:09 -070053extern const char* kUnittestPrivateKeyPath;
Darin Petkovd7061ab2010-10-06 14:37:09 -070054extern const char* kUnittestPublicKeyPath;
Andrew de los Reyes932bc4c2010-08-23 18:14:09 -070055
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -070056static const char* kBogusMetadataSignature1 =
57 "awSFIUdUZz2VWFiR+ku0Pj00V7bPQPQFYQSXjEXr3vaw3TE4xHV5CraY3/YrZpBv"
58 "J5z4dSBskoeuaO1TNC/S6E05t+yt36tE4Fh79tMnJ/z9fogBDXWgXLEUyG78IEQr"
59 "YH6/eBsQGT2RJtBgXIXbZ9W+5G9KmGDoPOoiaeNsDuqHiBc/58OFsrxskH8E6vMS"
60 "BmMGGk82mvgzic7ApcoURbCGey1b3Mwne/hPZ/bb9CIyky8Og9IfFMdL2uAweOIR"
61 "fjoTeLYZpt+WN65Vu7jJ0cQN8e1y+2yka5112wpRf/LLtPgiAjEZnsoYpLUd7CoV"
62 "pLRtClp97kN2+tXGNBQqkA==";
Jay Srinivasan738fdf32012-12-07 17:40:54 -080063
Andrew de los Reyes27f7d372010-10-07 11:26:07 -070064namespace {
Jay Srinivasan738fdf32012-12-07 17:40:54 -080065// Different options that determine what we should fill into the
66// install_plan.metadata_signature to simulate the contents received in the
67// Omaha response.
68enum MetadataSignatureTest {
69 kEmptyMetadataSignature,
70 kInvalidMetadataSignature,
71 kValidMetadataSignature,
72};
73
Alex Vakulenkod2779df2014-06-16 13:19:00 -070074} // namespace
Jay Srinivasan738fdf32012-12-07 17:40:54 -080075
Allie Woodfdf00512015-03-02 13:34:55 -080076class DeltaPerformerTest : public ::testing::Test {
Sen Jiang2d528b42015-09-25 11:18:12 -070077 protected:
Allie Woodfdf00512015-03-02 13:34:55 -080078
Sen Jiang2d528b42015-09-25 11:18:12 -070079 // Test helper placed where it can easily be friended from DeltaPerformer.
80 void RunManifestValidation(const DeltaArchiveManifest& manifest,
81 bool full_payload,
82 ErrorCode expected) {
Allie Woodfdf00512015-03-02 13:34:55 -080083 // The install plan is for Full or Delta.
Sen Jiang2d528b42015-09-25 11:18:12 -070084 install_plan_.is_full_update = full_payload;
Allie Woodfdf00512015-03-02 13:34:55 -080085
86 // The Manifest we are validating.
Sen Jiang2d528b42015-09-25 11:18:12 -070087 performer_.manifest_.CopyFrom(manifest);
Allie Woodfdf00512015-03-02 13:34:55 -080088
Sen Jiang2d528b42015-09-25 11:18:12 -070089 EXPECT_EQ(expected, performer_.ValidateManifest());
Allie Woodfdf00512015-03-02 13:34:55 -080090 }
91
Sen Jiang2d528b42015-09-25 11:18:12 -070092 chromeos::Blob GeneratePayload(const chromeos::Blob& blob_data,
93 const vector<AnnotatedOperation>& aops,
94 bool sign_payload,
95 int32_t minor_version) {
Sen Jiangd78b3892015-09-25 15:19:31 -070096 string blob_path;
97 EXPECT_TRUE(utils::MakeTempFile("Blob-XXXXXX", &blob_path, nullptr));
98 ScopedPathUnlinker blob_unlinker(blob_path);
99 EXPECT_TRUE(utils::WriteFile(blob_path.c_str(),
100 blob_data.data(),
101 blob_data.size()));
102
103 PayloadGenerationConfig config;
104 config.major_version = kChromeOSMajorPayloadVersion;
105 config.minor_version = minor_version;
106 config.target.rootfs.path = blob_path;
107 config.target.rootfs.size = blob_data.size();
108 config.target.kernel.path = blob_path;
109 config.target.kernel.size = blob_data.size();
110
111 PayloadFile payload;
112 EXPECT_TRUE(payload.Init(config));
113
114 payload.AddPartition(config.source.rootfs, config.target.rootfs, aops);
115
116 string payload_path;
117 EXPECT_TRUE(utils::MakeTempFile("Payload-XXXXXX", &payload_path, nullptr));
118 ScopedPathUnlinker payload_unlinker(payload_path);
119 EXPECT_TRUE(payload.WritePayload(payload_path, blob_path,
120 sign_payload ? kUnittestPrivateKeyPath : "",
Sen Jiang2d528b42015-09-25 11:18:12 -0700121 &install_plan_.metadata_size));
Sen Jiangd78b3892015-09-25 15:19:31 -0700122
123 chromeos::Blob payload_data;
124 EXPECT_TRUE(utils::ReadFile(payload_path, &payload_data));
125 return payload_data;
126 }
Allie Woodfdf00512015-03-02 13:34:55 -0800127
Sen Jiang2d528b42015-09-25 11:18:12 -0700128 // Apply |payload_data| on partition specified in |source_path|.
129 chromeos::Blob ApplyPayload(const chromeos::Blob& payload_data,
130 const string& source_path) {
131 install_plan_.source_path = source_path;
132 install_plan_.kernel_source_path = "/dev/null";
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800133
Sen Jiang2d528b42015-09-25 11:18:12 -0700134 string new_part;
135 EXPECT_TRUE(utils::MakeTempFile("Partition-XXXXXX", &new_part, nullptr));
136 ScopedPathUnlinker partition_unlinker(new_part);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800137
Sen Jiang2d528b42015-09-25 11:18:12 -0700138 EXPECT_EQ(0, performer_.Open(new_part.c_str(), 0, 0));
139 EXPECT_TRUE(performer_.Write(payload_data.data(), payload_data.size()));
140 EXPECT_EQ(0, performer_.Close());
141
142 chromeos::Blob partition_data;
143 EXPECT_TRUE(utils::ReadFile(new_part, &partition_data));
144 return partition_data;
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800145 }
146
Sen Jiang2d528b42015-09-25 11:18:12 -0700147 // Calls delta performer's Write method by pretending to pass in bytes from a
148 // delta file whose metadata size is actual_metadata_size and tests if all
149 // checks are correctly performed if the install plan contains
150 // expected_metadata_size and that the result of the parsing are as per
151 // hash_checks_mandatory flag.
152 void DoMetadataSizeTest(uint64_t expected_metadata_size,
153 uint64_t actual_metadata_size,
154 bool hash_checks_mandatory) {
155 install_plan_.hash_checks_mandatory = hash_checks_mandatory;
156 EXPECT_EQ(0, performer_.Open("/dev/null", 0, 0));
157 EXPECT_TRUE(performer_.OpenKernel("/dev/null"));
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800158
Sen Jiang2d528b42015-09-25 11:18:12 -0700159 // Set a valid magic string and version number 1.
160 EXPECT_TRUE(performer_.Write("CrAU", 4));
161 uint64_t version = htobe64(kChromeOSMajorPayloadVersion);
162 EXPECT_TRUE(performer_.Write(&version, 8));
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800163
Sen Jiang2d528b42015-09-25 11:18:12 -0700164 install_plan_.metadata_size = expected_metadata_size;
165 ErrorCode error_code;
166 // When filling in size in manifest, exclude the size of the 20-byte header.
167 uint64_t size_in_manifest = htobe64(actual_metadata_size - 20);
168 bool result = performer_.Write(&size_in_manifest, 8, &error_code);
169 if (expected_metadata_size == actual_metadata_size ||
170 !hash_checks_mandatory) {
171 EXPECT_TRUE(result);
172 } else {
173 EXPECT_FALSE(result);
174 EXPECT_EQ(ErrorCode::kDownloadInvalidMetadataSize, error_code);
175 }
Sen Jiangd78b3892015-09-25 15:19:31 -0700176
Sen Jiang2d528b42015-09-25 11:18:12 -0700177 EXPECT_LT(performer_.Close(), 0);
178 }
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800179
Sen Jiang2d528b42015-09-25 11:18:12 -0700180 // Generates a valid delta file but tests the delta performer by suppling
181 // different metadata signatures as per metadata_signature_test flag and
182 // sees if the result of the parsing are as per hash_checks_mandatory flag.
183 void DoMetadataSignatureTest(MetadataSignatureTest metadata_signature_test,
184 bool sign_payload,
185 bool hash_checks_mandatory) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800186
Sen Jiang2d528b42015-09-25 11:18:12 -0700187 // Loads the payload and parses the manifest.
188 chromeos::Blob payload = GeneratePayload(chromeos::Blob(),
189 vector<AnnotatedOperation>(), sign_payload,
190 DeltaPerformer::kFullPayloadMinorVersion);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800191
Sen Jiang2d528b42015-09-25 11:18:12 -0700192 LOG(INFO) << "Payload size: " << payload.size();
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800193
Sen Jiang2d528b42015-09-25 11:18:12 -0700194 install_plan_.hash_checks_mandatory = hash_checks_mandatory;
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800195
Sen Jiang2d528b42015-09-25 11:18:12 -0700196 DeltaPerformer::MetadataParseResult expected_result, actual_result;
197 ErrorCode expected_error, actual_error;
198
199 // Fill up the metadata signature in install plan according to the test.
200 switch (metadata_signature_test) {
201 case kEmptyMetadataSignature:
202 install_plan_.metadata_signature.clear();
203 expected_result = DeltaPerformer::kMetadataParseError;
204 expected_error = ErrorCode::kDownloadMetadataSignatureMissingError;
205 break;
206
207 case kInvalidMetadataSignature:
208 install_plan_.metadata_signature = kBogusMetadataSignature1;
209 expected_result = DeltaPerformer::kMetadataParseError;
210 expected_error = ErrorCode::kDownloadMetadataSignatureMismatch;
211 break;
212
213 case kValidMetadataSignature:
214 default:
215 // Set the install plan's metadata size to be the same as the one
216 // in the manifest so that we pass the metadata size checks. Only
217 // then we can get to manifest signature checks.
218 ASSERT_TRUE(PayloadSigner::GetMetadataSignature(
219 payload.data(),
220 install_plan_.metadata_size,
221 kUnittestPrivateKeyPath,
222 &install_plan_.metadata_signature));
223 EXPECT_FALSE(install_plan_.metadata_signature.empty());
224 expected_result = DeltaPerformer::kMetadataParseSuccess;
225 expected_error = ErrorCode::kSuccess;
226 break;
227 }
228
229 // Ignore the expected result/error if hash checks are not mandatory.
230 if (!hash_checks_mandatory) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800231 expected_result = DeltaPerformer::kMetadataParseSuccess;
Gilad Arnoldd1c4d2d2014-06-05 14:07:53 -0700232 expected_error = ErrorCode::kSuccess;
Sen Jiang2d528b42015-09-25 11:18:12 -0700233 }
234
235 // Use the public key corresponding to the private key used above to
236 // sign the metadata.
237 EXPECT_TRUE(utils::FileExists(kUnittestPublicKeyPath));
238 performer_.set_public_key_path(kUnittestPublicKeyPath);
239
240 // Init actual_error with an invalid value so that we make sure
241 // ParsePayloadMetadata properly populates it in all cases.
242 actual_error = ErrorCode::kUmaReportedMax;
243 actual_result = performer_.ParsePayloadMetadata(payload, &actual_error);
244
245 EXPECT_EQ(expected_result, actual_result);
246 EXPECT_EQ(expected_error, actual_error);
247
248 // Check that the parsed metadata size is what's expected. This test
249 // implicitly confirms that the metadata signature is valid, if required.
250 EXPECT_EQ(install_plan_.metadata_size, performer_.GetMetadataSize());
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800251 }
252
Sen Jiang2d528b42015-09-25 11:18:12 -0700253 FakePrefs prefs_;
254 InstallPlan install_plan_;
255 FakeSystemState fake_system_state_;
256 DeltaPerformer performer_{&prefs_, &fake_system_state_, &install_plan_};
257};
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800258
Sen Jiang2d528b42015-09-25 11:18:12 -0700259TEST_F(DeltaPerformerTest, FullPayloadWriteTest) {
260 install_plan_.is_full_update = true;
261 chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString),
262 std::end(kRandomString));
263 expected_data.resize(4096); // block size
264 vector<AnnotatedOperation> aops;
265 AnnotatedOperation aop;
266 *(aop.op.add_dst_extents()) = ExtentForRange(0, 1);
267 aop.op.set_data_offset(0);
268 aop.op.set_data_length(expected_data.size());
269 aop.op.set_type(InstallOperation::REPLACE);
270 aops.push_back(aop);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800271
Sen Jiang2d528b42015-09-25 11:18:12 -0700272 chromeos::Blob payload_data = GeneratePayload(expected_data, aops, false,
273 DeltaPerformer::kFullPayloadMinorVersion);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800274
Sen Jiang2d528b42015-09-25 11:18:12 -0700275 EXPECT_EQ(expected_data, ApplyPayload(payload_data, ""));
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800276}
277
Sen Jiang2d528b42015-09-25 11:18:12 -0700278TEST_F(DeltaPerformerTest, ReplaceOperationTest) {
279 chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString),
280 std::end(kRandomString));
281 expected_data.resize(4096); // block size
282 vector<AnnotatedOperation> aops;
283 AnnotatedOperation aop;
284 *(aop.op.add_dst_extents()) = ExtentForRange(0, 1);
285 aop.op.set_data_offset(0);
286 aop.op.set_data_length(expected_data.size());
287 aop.op.set_type(InstallOperation::REPLACE);
288 aops.push_back(aop);
289
290 chromeos::Blob payload_data = GeneratePayload(expected_data, aops, false,
291 kSourceMinorPayloadVersion);
292
293 EXPECT_EQ(expected_data, ApplyPayload(payload_data, "/dev/null"));
294}
295
296TEST_F(DeltaPerformerTest, ReplaceBzOperationTest) {
297 chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString),
298 std::end(kRandomString));
299 expected_data.resize(4096); // block size
300 chromeos::Blob bz_data;
301 EXPECT_TRUE(BzipCompress(expected_data, &bz_data));
302
303 vector<AnnotatedOperation> aops;
304 AnnotatedOperation aop;
305 *(aop.op.add_dst_extents()) = ExtentForRange(0, 1);
306 aop.op.set_data_offset(0);
307 aop.op.set_data_length(bz_data.size());
308 aop.op.set_type(InstallOperation::REPLACE_BZ);
309 aops.push_back(aop);
310
311 chromeos::Blob payload_data = GeneratePayload(bz_data, aops, false,
312 kSourceMinorPayloadVersion);
313
314 EXPECT_EQ(expected_data, ApplyPayload(payload_data, "/dev/null"));
315}
316
317TEST_F(DeltaPerformerTest, SourceCopyOperationTest) {
318 chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString),
319 std::end(kRandomString));
320 expected_data.resize(4096); // block size
321 vector<AnnotatedOperation> aops;
322 AnnotatedOperation aop;
323 *(aop.op.add_src_extents()) = ExtentForRange(0, 1);
324 *(aop.op.add_dst_extents()) = ExtentForRange(0, 1);
325 aop.op.set_type(InstallOperation::SOURCE_COPY);
326 aops.push_back(aop);
327
328 chromeos::Blob payload_data = GeneratePayload(chromeos::Blob(), aops, false,
329 kSourceMinorPayloadVersion);
330 string source_path;
331 EXPECT_TRUE(utils::MakeTempFile("Source-XXXXXX",
332 &source_path, nullptr));
333 ScopedPathUnlinker path_unlinker(source_path);
334 EXPECT_TRUE(utils::WriteFile(source_path.c_str(),
335 expected_data.data(),
336 expected_data.size()));
337
338 EXPECT_EQ(expected_data, ApplyPayload(payload_data, source_path));
339}
340
341TEST_F(DeltaPerformerTest, ExtentsToByteStringTest) {
Allie Wood56873452015-03-27 17:48:40 -0700342 uint64_t test[] = {1, 1, 4, 2, 0, 1};
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800343 COMPILE_ASSERT(arraysize(test) % 2 == 0, array_size_uneven);
344 const uint64_t block_size = 4096;
Allie Wood56873452015-03-27 17:48:40 -0700345 const uint64_t file_length = 4 * block_size - 13;
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800346
347 google::protobuf::RepeatedPtrField<Extent> extents;
348 for (size_t i = 0; i < arraysize(test); i += 2) {
Sen Jiang2d528b42015-09-25 11:18:12 -0700349 *(extents.Add()) = ExtentForRange(test[i], test[i + 1]);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800350 }
351
Allie Wood56873452015-03-27 17:48:40 -0700352 string expected_output = "4096:4096,16384:8192,0:4083";
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800353 string actual_output;
354 EXPECT_TRUE(DeltaPerformer::ExtentsToBsdiffPositionsString(extents,
355 block_size,
356 file_length,
357 &actual_output));
358 EXPECT_EQ(expected_output, actual_output);
359}
Darin Petkov68c10d12010-10-14 09:24:37 -0700360
Sen Jiang2d528b42015-09-25 11:18:12 -0700361TEST_F(DeltaPerformerTest, ValidateManifestFullGoodTest) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800362 // The Manifest we are validating.
363 DeltaArchiveManifest manifest;
364 manifest.mutable_new_kernel_info();
365 manifest.mutable_new_rootfs_info();
366 manifest.set_minor_version(DeltaPerformer::kFullPayloadMinorVersion);
367
Sen Jiang2d528b42015-09-25 11:18:12 -0700368 RunManifestValidation(manifest, true, ErrorCode::kSuccess);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800369}
370
Sen Jiang2d528b42015-09-25 11:18:12 -0700371TEST_F(DeltaPerformerTest, ValidateManifestDeltaGoodTest) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800372 // The Manifest we are validating.
373 DeltaArchiveManifest manifest;
374 manifest.mutable_old_kernel_info();
375 manifest.mutable_old_rootfs_info();
376 manifest.mutable_new_kernel_info();
377 manifest.mutable_new_rootfs_info();
378 manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion);
379
Sen Jiang2d528b42015-09-25 11:18:12 -0700380 RunManifestValidation(manifest, false, ErrorCode::kSuccess);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800381}
382
Sen Jiang2d528b42015-09-25 11:18:12 -0700383TEST_F(DeltaPerformerTest, ValidateManifestFullUnsetMinorVersion) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800384 // The Manifest we are validating.
385 DeltaArchiveManifest manifest;
386
Sen Jiang2d528b42015-09-25 11:18:12 -0700387 RunManifestValidation(manifest, true, ErrorCode::kSuccess);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800388}
389
Sen Jiang2d528b42015-09-25 11:18:12 -0700390TEST_F(DeltaPerformerTest, ValidateManifestDeltaUnsetMinorVersion) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800391 // The Manifest we are validating.
392 DeltaArchiveManifest manifest;
393
Sen Jiang2d528b42015-09-25 11:18:12 -0700394 RunManifestValidation(manifest, false,
395 ErrorCode::kUnsupportedMinorPayloadVersion);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800396}
397
Sen Jiang2d528b42015-09-25 11:18:12 -0700398TEST_F(DeltaPerformerTest, ValidateManifestFullOldKernelTest) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800399 // The Manifest we are validating.
400 DeltaArchiveManifest manifest;
401 manifest.mutable_old_kernel_info();
402 manifest.mutable_new_kernel_info();
403 manifest.mutable_new_rootfs_info();
404 manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion);
405
Sen Jiang2d528b42015-09-25 11:18:12 -0700406 RunManifestValidation(manifest, true, ErrorCode::kPayloadMismatchedType);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800407}
408
Sen Jiang2d528b42015-09-25 11:18:12 -0700409TEST_F(DeltaPerformerTest, ValidateManifestFullOldRootfsTest) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800410 // The Manifest we are validating.
411 DeltaArchiveManifest manifest;
412 manifest.mutable_old_rootfs_info();
413 manifest.mutable_new_kernel_info();
414 manifest.mutable_new_rootfs_info();
415 manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion);
416
Sen Jiang2d528b42015-09-25 11:18:12 -0700417 RunManifestValidation(manifest, true, ErrorCode::kPayloadMismatchedType);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800418}
419
Sen Jiang2d528b42015-09-25 11:18:12 -0700420TEST_F(DeltaPerformerTest, ValidateManifestBadMinorVersion) {
Don Garrettb8dd1d92013-11-22 17:40:02 -0800421 // The Manifest we are validating.
422 DeltaArchiveManifest manifest;
423
424 // Generate a bad version number.
425 manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion +
426 10000);
427
Sen Jiang2d528b42015-09-25 11:18:12 -0700428 RunManifestValidation(manifest, false,
429 ErrorCode::kUnsupportedMinorPayloadVersion);
Don Garrettb8dd1d92013-11-22 17:40:02 -0800430}
431
Sen Jiang2d528b42015-09-25 11:18:12 -0700432TEST_F(DeltaPerformerTest, BadDeltaMagicTest) {
433 EXPECT_EQ(0, performer_.Open("/dev/null", 0, 0));
434 EXPECT_TRUE(performer_.OpenKernel("/dev/null"));
435 EXPECT_TRUE(performer_.Write("junk", 4));
436 EXPECT_TRUE(performer_.Write("morejunk", 8));
437 EXPECT_FALSE(performer_.Write("morejunk", 8));
438 EXPECT_LT(performer_.Close(), 0);
Darin Petkov934bb412010-11-18 11:21:35 -0800439}
440
Sen Jiang2d528b42015-09-25 11:18:12 -0700441TEST_F(DeltaPerformerTest, WriteUpdatesPayloadState) {
442 EXPECT_EQ(0, performer_.Open("/dev/null", 0, 0));
443 EXPECT_TRUE(performer_.OpenKernel("/dev/null"));
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800444
Sen Jiang2d528b42015-09-25 11:18:12 -0700445 EXPECT_CALL(*(fake_system_state_.mock_payload_state()),
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800446 DownloadProgress(4)).Times(1);
Sen Jiang2d528b42015-09-25 11:18:12 -0700447 EXPECT_CALL(*(fake_system_state_.mock_payload_state()),
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800448 DownloadProgress(8)).Times(2);
449
Sen Jiang2d528b42015-09-25 11:18:12 -0700450 EXPECT_TRUE(performer_.Write("junk", 4));
451 EXPECT_TRUE(performer_.Write("morejunk", 8));
452 EXPECT_FALSE(performer_.Write("morejunk", 8));
453 EXPECT_LT(performer_.Close(), 0);
Jay Srinivasan2b5a0f02012-12-19 17:25:56 -0800454}
455
Sen Jiang2d528b42015-09-25 11:18:12 -0700456TEST_F(DeltaPerformerTest, MissingMandatoryMetadataSizeTest) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800457 DoMetadataSizeTest(0, 75456, true);
Jay Srinivasan51dcf262012-09-13 17:24:32 -0700458}
459
Sen Jiang2d528b42015-09-25 11:18:12 -0700460TEST_F(DeltaPerformerTest, MissingNonMandatoryMetadataSizeTest) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800461 DoMetadataSizeTest(0, 123456, false);
462}
463
Sen Jiang2d528b42015-09-25 11:18:12 -0700464TEST_F(DeltaPerformerTest, InvalidMandatoryMetadataSizeTest) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800465 DoMetadataSizeTest(13000, 140000, true);
466}
467
Sen Jiang2d528b42015-09-25 11:18:12 -0700468TEST_F(DeltaPerformerTest, InvalidNonMandatoryMetadataSizeTest) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800469 DoMetadataSizeTest(40000, 50000, false);
470}
471
Sen Jiang2d528b42015-09-25 11:18:12 -0700472TEST_F(DeltaPerformerTest, ValidMandatoryMetadataSizeTest) {
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800473 DoMetadataSizeTest(85376, 85376, true);
474}
475
Sen Jiang2d528b42015-09-25 11:18:12 -0700476TEST_F(DeltaPerformerTest, MandatoryEmptyMetadataSignatureTest) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700477 DoMetadataSignatureTest(kEmptyMetadataSignature, true, true);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800478}
479
Sen Jiang2d528b42015-09-25 11:18:12 -0700480TEST_F(DeltaPerformerTest, NonMandatoryEmptyMetadataSignatureTest) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700481 DoMetadataSignatureTest(kEmptyMetadataSignature, true, false);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800482}
483
Sen Jiang2d528b42015-09-25 11:18:12 -0700484TEST_F(DeltaPerformerTest, MandatoryInvalidMetadataSignatureTest) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700485 DoMetadataSignatureTest(kInvalidMetadataSignature, true, true);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800486}
487
Sen Jiang2d528b42015-09-25 11:18:12 -0700488TEST_F(DeltaPerformerTest, NonMandatoryInvalidMetadataSignatureTest) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700489 DoMetadataSignatureTest(kInvalidMetadataSignature, true, false);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800490}
491
Sen Jiang2d528b42015-09-25 11:18:12 -0700492TEST_F(DeltaPerformerTest, MandatoryValidMetadataSignature1Test) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700493 DoMetadataSignatureTest(kValidMetadataSignature, false, true);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800494}
495
Sen Jiang2d528b42015-09-25 11:18:12 -0700496TEST_F(DeltaPerformerTest, MandatoryValidMetadataSignature2Test) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700497 DoMetadataSignatureTest(kValidMetadataSignature, true, true);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800498}
499
Sen Jiang2d528b42015-09-25 11:18:12 -0700500TEST_F(DeltaPerformerTest, NonMandatoryValidMetadataSignatureTest) {
Sen Jiangd78b3892015-09-25 15:19:31 -0700501 DoMetadataSignatureTest(kValidMetadataSignature, true, false);
Jay Srinivasan738fdf32012-12-07 17:40:54 -0800502}
503
Sen Jiang2d528b42015-09-25 11:18:12 -0700504TEST_F(DeltaPerformerTest, UsePublicKeyFromResponse) {
David Zeuthene7f89172013-10-31 10:21:04 -0700505 base::FilePath key_path;
506
507 // The result of the GetPublicKeyResponse() method is based on three things
508 //
509 // 1. Whether it's an official build; and
510 // 2. Whether the Public RSA key to be used is in the root filesystem; and
Alex Vakulenko072359c2014-07-18 11:41:07 -0700511 // 3. Whether the response has a public key
David Zeuthene7f89172013-10-31 10:21:04 -0700512 //
513 // We test all eight combinations to ensure that we only use the
514 // public key in the response if
515 //
516 // a. it's not an official build; and
517 // b. there is no key in the root filesystem.
518
Sen Jiang2d528b42015-09-25 11:18:12 -0700519 FakeHardware* fake_hardware = fake_system_state_.fake_hardware();
David Zeuthene7f89172013-10-31 10:21:04 -0700520
521 string temp_dir;
Gilad Arnolda6742b32014-01-11 00:18:34 -0800522 EXPECT_TRUE(utils::MakeTempDirectory("PublicKeyFromResponseTests.XXXXXX",
David Zeuthene7f89172013-10-31 10:21:04 -0700523 &temp_dir));
524 string non_existing_file = temp_dir + "/non-existing";
525 string existing_file = temp_dir + "/existing";
Alex Vakulenko75039d72014-03-25 12:36:28 -0700526 EXPECT_EQ(0, System(base::StringPrintf("touch %s", existing_file.c_str())));
David Zeuthene7f89172013-10-31 10:21:04 -0700527
528 // Non-official build, non-existing public-key, key in response -> true
Don Garrett6646b442013-11-13 15:29:11 -0800529 fake_hardware->SetIsOfficialBuild(false);
Sen Jiang2d528b42015-09-25 11:18:12 -0700530 performer_.public_key_path_ = non_existing_file;
531 install_plan_.public_key_rsa = "VGVzdAo="; // result of 'echo "Test" | base64'
532 EXPECT_TRUE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700533 EXPECT_FALSE(key_path.empty());
534 EXPECT_EQ(unlink(key_path.value().c_str()), 0);
535 // Same with official build -> false
Don Garrett6646b442013-11-13 15:29:11 -0800536 fake_hardware->SetIsOfficialBuild(true);
Sen Jiang2d528b42015-09-25 11:18:12 -0700537 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700538
539 // Non-official build, existing public-key, key in response -> false
Don Garrett6646b442013-11-13 15:29:11 -0800540 fake_hardware->SetIsOfficialBuild(false);
Sen Jiang2d528b42015-09-25 11:18:12 -0700541 performer_.public_key_path_ = existing_file;
542 install_plan_.public_key_rsa = "VGVzdAo="; // result of 'echo "Test" | base64'
543 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700544 // Same with official build -> false
Don Garrett6646b442013-11-13 15:29:11 -0800545 fake_hardware->SetIsOfficialBuild(true);
Sen Jiang2d528b42015-09-25 11:18:12 -0700546 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700547
548 // Non-official build, non-existing public-key, no key in response -> false
Don Garrett6646b442013-11-13 15:29:11 -0800549 fake_hardware->SetIsOfficialBuild(false);
Sen Jiang2d528b42015-09-25 11:18:12 -0700550 performer_.public_key_path_ = non_existing_file;
551 install_plan_.public_key_rsa = "";
552 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700553 // Same with official build -> false
Don Garrett6646b442013-11-13 15:29:11 -0800554 fake_hardware->SetIsOfficialBuild(true);
Sen Jiang2d528b42015-09-25 11:18:12 -0700555 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700556
557 // Non-official build, existing public-key, no key in response -> false
Don Garrett6646b442013-11-13 15:29:11 -0800558 fake_hardware->SetIsOfficialBuild(false);
Sen Jiang2d528b42015-09-25 11:18:12 -0700559 performer_.public_key_path_ = existing_file;
560 install_plan_.public_key_rsa = "";
561 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700562 // Same with official build -> false
Don Garrett6646b442013-11-13 15:29:11 -0800563 fake_hardware->SetIsOfficialBuild(true);
Sen Jiang2d528b42015-09-25 11:18:12 -0700564 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700565
566 // Non-official build, non-existing public-key, key in response
567 // but invalid base64 -> false
Don Garrett6646b442013-11-13 15:29:11 -0800568 fake_hardware->SetIsOfficialBuild(false);
Sen Jiang2d528b42015-09-25 11:18:12 -0700569 performer_.public_key_path_ = non_existing_file;
570 install_plan_.public_key_rsa = "not-valid-base64";
571 EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path));
David Zeuthene7f89172013-10-31 10:21:04 -0700572
Alex Deymo10875d92014-11-10 21:52:57 -0800573 EXPECT_TRUE(test_utils::RecursiveUnlinkDir(temp_dir));
David Zeuthene7f89172013-10-31 10:21:04 -0700574}
575
Sen Jiang2d528b42015-09-25 11:18:12 -0700576TEST_F(DeltaPerformerTest, MinorVersionsMatch) {
Allie Wood78750a42015-02-11 15:42:11 -0800577 // Test that the minor version in update_engine.conf that is installed to
578 // the image matches the supported delta minor version in the update engine.
579 uint32_t minor_version;
Alex Deymob42b98d2015-07-06 17:42:38 -0700580 chromeos::KeyValueStore store;
581 EXPECT_TRUE(store.Load(base::FilePath("update_engine.conf")));
582 EXPECT_TRUE(utils::GetMinorVersion(store, &minor_version));
583 EXPECT_EQ(DeltaPerformer::kSupportedMinorPayloadVersion, minor_version);
Allie Wood78750a42015-02-11 15:42:11 -0800584}
585
Andrew de los Reyes09e56d62010-04-23 13:45:53 -0700586} // namespace chromeos_update_engine