Alex Deymo | aea4c1c | 2015-08-19 20:24:43 -0700 | [diff] [blame] | 1 | // |
| 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 Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 16 | |
Alex Deymo | 8427b4a | 2014-11-05 14:00:32 -0800 | [diff] [blame] | 17 | #include "update_engine/delta_performer.h" |
| 18 | |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 19 | #include <inttypes.h> |
| 20 | |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 21 | #include <string> |
| 22 | #include <vector> |
| 23 | |
Allie Wood | 78750a4 | 2015-02-11 15:42:11 -0800 | [diff] [blame] | 24 | #include <base/files/file_path.h> |
Ben Chan | 06c76a4 | 2014-09-05 08:21:06 -0700 | [diff] [blame] | 25 | #include <base/files/file_util.h> |
Alex Vakulenko | 75039d7 | 2014-03-25 12:36:28 -0700 | [diff] [blame] | 26 | #include <base/strings/stringprintf.h> |
Alex Deymo | cbf0989 | 2015-09-11 16:13:16 -0700 | [diff] [blame^] | 27 | #include <base/strings/string_number_conversions.h> |
Alex Deymo | f1cbe17 | 2015-03-05 15:58:37 -0800 | [diff] [blame] | 28 | #include <base/strings/string_util.h> |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 29 | #include <google/protobuf/repeated_field.h> |
| 30 | #include <gtest/gtest.h> |
| 31 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 32 | #include "update_engine/bzip.h" |
Jay Srinivasan | d29695d | 2013-04-08 15:08:05 -0700 | [diff] [blame] | 33 | #include "update_engine/constants.h" |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 34 | #include "update_engine/fake_hardware.h" |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 35 | #include "update_engine/fake_prefs.h" |
Gilad Arnold | 5bb4c90 | 2014-04-10 12:32:13 -0700 | [diff] [blame] | 36 | #include "update_engine/fake_system_state.h" |
Alex Deymo | 161c4a1 | 2014-05-16 15:56:21 -0700 | [diff] [blame] | 37 | #include "update_engine/payload_constants.h" |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 38 | #include "update_engine/payload_generator/extent_ranges.h" |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 39 | #include "update_engine/payload_generator/payload_file.h" |
Alex Deymo | 923d8fa | 2014-07-15 17:58:51 -0700 | [diff] [blame] | 40 | #include "update_engine/payload_generator/payload_signer.h" |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 41 | #include "update_engine/test_utils.h" |
| 42 | #include "update_engine/update_metadata.pb.h" |
| 43 | #include "update_engine/utils.h" |
| 44 | |
| 45 | namespace chromeos_update_engine { |
| 46 | |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 47 | using std::string; |
| 48 | using std::vector; |
Darin Petkov | 73058b4 | 2010-10-06 16:32:19 -0700 | [diff] [blame] | 49 | using testing::Return; |
Alex Deymo | 161c4a1 | 2014-05-16 15:56:21 -0700 | [diff] [blame] | 50 | using testing::_; |
Alex Deymo | 10875d9 | 2014-11-10 21:52:57 -0800 | [diff] [blame] | 51 | using test_utils::kRandomString; |
Alex Deymo | 10875d9 | 2014-11-10 21:52:57 -0800 | [diff] [blame] | 52 | using test_utils::System; |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 53 | |
Andrew de los Reyes | 932bc4c | 2010-08-23 18:14:09 -0700 | [diff] [blame] | 54 | extern const char* kUnittestPrivateKeyPath; |
Darin Petkov | d7061ab | 2010-10-06 14:37:09 -0700 | [diff] [blame] | 55 | extern const char* kUnittestPublicKeyPath; |
Andrew de los Reyes | 932bc4c | 2010-08-23 18:14:09 -0700 | [diff] [blame] | 56 | |
Gilad Arnold | d1c4d2d | 2014-06-05 14:07:53 -0700 | [diff] [blame] | 57 | static const char* kBogusMetadataSignature1 = |
| 58 | "awSFIUdUZz2VWFiR+ku0Pj00V7bPQPQFYQSXjEXr3vaw3TE4xHV5CraY3/YrZpBv" |
| 59 | "J5z4dSBskoeuaO1TNC/S6E05t+yt36tE4Fh79tMnJ/z9fogBDXWgXLEUyG78IEQr" |
| 60 | "YH6/eBsQGT2RJtBgXIXbZ9W+5G9KmGDoPOoiaeNsDuqHiBc/58OFsrxskH8E6vMS" |
| 61 | "BmMGGk82mvgzic7ApcoURbCGey1b3Mwne/hPZ/bb9CIyky8Og9IfFMdL2uAweOIR" |
| 62 | "fjoTeLYZpt+WN65Vu7jJ0cQN8e1y+2yka5112wpRf/LLtPgiAjEZnsoYpLUd7CoV" |
| 63 | "pLRtClp97kN2+tXGNBQqkA=="; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 64 | |
Andrew de los Reyes | 27f7d37 | 2010-10-07 11:26:07 -0700 | [diff] [blame] | 65 | namespace { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 66 | // Different options that determine what we should fill into the |
| 67 | // install_plan.metadata_signature to simulate the contents received in the |
| 68 | // Omaha response. |
| 69 | enum MetadataSignatureTest { |
| 70 | kEmptyMetadataSignature, |
| 71 | kInvalidMetadataSignature, |
| 72 | kValidMetadataSignature, |
| 73 | }; |
| 74 | |
Alex Vakulenko | d2779df | 2014-06-16 13:19:00 -0700 | [diff] [blame] | 75 | } // namespace |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 76 | |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 77 | class DeltaPerformerTest : public ::testing::Test { |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 78 | protected: |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 79 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 80 | // Test helper placed where it can easily be friended from DeltaPerformer. |
| 81 | void RunManifestValidation(const DeltaArchiveManifest& manifest, |
| 82 | bool full_payload, |
| 83 | ErrorCode expected) { |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 84 | // The install plan is for Full or Delta. |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 85 | install_plan_.is_full_update = full_payload; |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 86 | |
| 87 | // The Manifest we are validating. |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 88 | performer_.manifest_.CopyFrom(manifest); |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 89 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 90 | EXPECT_EQ(expected, performer_.ValidateManifest()); |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 91 | } |
| 92 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 93 | chromeos::Blob GeneratePayload(const chromeos::Blob& blob_data, |
| 94 | const vector<AnnotatedOperation>& aops, |
| 95 | bool sign_payload, |
| 96 | int32_t minor_version) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 97 | string blob_path; |
| 98 | EXPECT_TRUE(utils::MakeTempFile("Blob-XXXXXX", &blob_path, nullptr)); |
| 99 | ScopedPathUnlinker blob_unlinker(blob_path); |
| 100 | EXPECT_TRUE(utils::WriteFile(blob_path.c_str(), |
| 101 | blob_data.data(), |
| 102 | blob_data.size())); |
| 103 | |
| 104 | PayloadGenerationConfig config; |
| 105 | config.major_version = kChromeOSMajorPayloadVersion; |
| 106 | config.minor_version = minor_version; |
| 107 | config.target.rootfs.path = blob_path; |
| 108 | config.target.rootfs.size = blob_data.size(); |
| 109 | config.target.kernel.path = blob_path; |
| 110 | config.target.kernel.size = blob_data.size(); |
| 111 | |
| 112 | PayloadFile payload; |
| 113 | EXPECT_TRUE(payload.Init(config)); |
| 114 | |
| 115 | payload.AddPartition(config.source.rootfs, config.target.rootfs, aops); |
| 116 | |
| 117 | string payload_path; |
| 118 | EXPECT_TRUE(utils::MakeTempFile("Payload-XXXXXX", &payload_path, nullptr)); |
| 119 | ScopedPathUnlinker payload_unlinker(payload_path); |
| 120 | EXPECT_TRUE(payload.WritePayload(payload_path, blob_path, |
| 121 | sign_payload ? kUnittestPrivateKeyPath : "", |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 122 | &install_plan_.metadata_size)); |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 123 | |
| 124 | chromeos::Blob payload_data; |
| 125 | EXPECT_TRUE(utils::ReadFile(payload_path, &payload_data)); |
| 126 | return payload_data; |
| 127 | } |
Allie Wood | fdf0051 | 2015-03-02 13:34:55 -0800 | [diff] [blame] | 128 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 129 | // Apply |payload_data| on partition specified in |source_path|. |
| 130 | chromeos::Blob ApplyPayload(const chromeos::Blob& payload_data, |
| 131 | const string& source_path) { |
| 132 | install_plan_.source_path = source_path; |
| 133 | install_plan_.kernel_source_path = "/dev/null"; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 134 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 135 | string new_part; |
| 136 | EXPECT_TRUE(utils::MakeTempFile("Partition-XXXXXX", &new_part, nullptr)); |
| 137 | ScopedPathUnlinker partition_unlinker(new_part); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 138 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 139 | EXPECT_EQ(0, performer_.Open(new_part.c_str(), 0, 0)); |
| 140 | EXPECT_TRUE(performer_.Write(payload_data.data(), payload_data.size())); |
| 141 | EXPECT_EQ(0, performer_.Close()); |
| 142 | |
| 143 | chromeos::Blob partition_data; |
| 144 | EXPECT_TRUE(utils::ReadFile(new_part, &partition_data)); |
| 145 | return partition_data; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 146 | } |
| 147 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 148 | // Calls delta performer's Write method by pretending to pass in bytes from a |
| 149 | // delta file whose metadata size is actual_metadata_size and tests if all |
| 150 | // checks are correctly performed if the install plan contains |
| 151 | // expected_metadata_size and that the result of the parsing are as per |
| 152 | // hash_checks_mandatory flag. |
| 153 | void DoMetadataSizeTest(uint64_t expected_metadata_size, |
| 154 | uint64_t actual_metadata_size, |
| 155 | bool hash_checks_mandatory) { |
| 156 | install_plan_.hash_checks_mandatory = hash_checks_mandatory; |
| 157 | EXPECT_EQ(0, performer_.Open("/dev/null", 0, 0)); |
| 158 | EXPECT_TRUE(performer_.OpenKernel("/dev/null")); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 159 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 160 | // Set a valid magic string and version number 1. |
| 161 | EXPECT_TRUE(performer_.Write("CrAU", 4)); |
| 162 | uint64_t version = htobe64(kChromeOSMajorPayloadVersion); |
| 163 | EXPECT_TRUE(performer_.Write(&version, 8)); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 164 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 165 | install_plan_.metadata_size = expected_metadata_size; |
| 166 | ErrorCode error_code; |
| 167 | // When filling in size in manifest, exclude the size of the 20-byte header. |
| 168 | uint64_t size_in_manifest = htobe64(actual_metadata_size - 20); |
| 169 | bool result = performer_.Write(&size_in_manifest, 8, &error_code); |
| 170 | if (expected_metadata_size == actual_metadata_size || |
| 171 | !hash_checks_mandatory) { |
| 172 | EXPECT_TRUE(result); |
| 173 | } else { |
| 174 | EXPECT_FALSE(result); |
| 175 | EXPECT_EQ(ErrorCode::kDownloadInvalidMetadataSize, error_code); |
| 176 | } |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 177 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 178 | EXPECT_LT(performer_.Close(), 0); |
| 179 | } |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 180 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 181 | // Generates a valid delta file but tests the delta performer by suppling |
| 182 | // different metadata signatures as per metadata_signature_test flag and |
| 183 | // sees if the result of the parsing are as per hash_checks_mandatory flag. |
| 184 | void DoMetadataSignatureTest(MetadataSignatureTest metadata_signature_test, |
| 185 | bool sign_payload, |
| 186 | bool hash_checks_mandatory) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 187 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 188 | // Loads the payload and parses the manifest. |
| 189 | chromeos::Blob payload = GeneratePayload(chromeos::Blob(), |
| 190 | vector<AnnotatedOperation>(), sign_payload, |
Alex Deymo | cbf0989 | 2015-09-11 16:13:16 -0700 | [diff] [blame^] | 191 | kFullPayloadMinorVersion); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 192 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 193 | LOG(INFO) << "Payload size: " << payload.size(); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 194 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 195 | install_plan_.hash_checks_mandatory = hash_checks_mandatory; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 196 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 197 | DeltaPerformer::MetadataParseResult expected_result, actual_result; |
| 198 | ErrorCode expected_error, actual_error; |
| 199 | |
| 200 | // Fill up the metadata signature in install plan according to the test. |
| 201 | switch (metadata_signature_test) { |
| 202 | case kEmptyMetadataSignature: |
| 203 | install_plan_.metadata_signature.clear(); |
| 204 | expected_result = DeltaPerformer::kMetadataParseError; |
| 205 | expected_error = ErrorCode::kDownloadMetadataSignatureMissingError; |
| 206 | break; |
| 207 | |
| 208 | case kInvalidMetadataSignature: |
| 209 | install_plan_.metadata_signature = kBogusMetadataSignature1; |
| 210 | expected_result = DeltaPerformer::kMetadataParseError; |
| 211 | expected_error = ErrorCode::kDownloadMetadataSignatureMismatch; |
| 212 | break; |
| 213 | |
| 214 | case kValidMetadataSignature: |
| 215 | default: |
| 216 | // Set the install plan's metadata size to be the same as the one |
| 217 | // in the manifest so that we pass the metadata size checks. Only |
| 218 | // then we can get to manifest signature checks. |
| 219 | ASSERT_TRUE(PayloadSigner::GetMetadataSignature( |
| 220 | payload.data(), |
| 221 | install_plan_.metadata_size, |
| 222 | kUnittestPrivateKeyPath, |
| 223 | &install_plan_.metadata_signature)); |
| 224 | EXPECT_FALSE(install_plan_.metadata_signature.empty()); |
| 225 | expected_result = DeltaPerformer::kMetadataParseSuccess; |
| 226 | expected_error = ErrorCode::kSuccess; |
| 227 | break; |
| 228 | } |
| 229 | |
| 230 | // Ignore the expected result/error if hash checks are not mandatory. |
| 231 | if (!hash_checks_mandatory) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 232 | expected_result = DeltaPerformer::kMetadataParseSuccess; |
Gilad Arnold | d1c4d2d | 2014-06-05 14:07:53 -0700 | [diff] [blame] | 233 | expected_error = ErrorCode::kSuccess; |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 234 | } |
| 235 | |
| 236 | // Use the public key corresponding to the private key used above to |
| 237 | // sign the metadata. |
| 238 | EXPECT_TRUE(utils::FileExists(kUnittestPublicKeyPath)); |
| 239 | performer_.set_public_key_path(kUnittestPublicKeyPath); |
| 240 | |
| 241 | // Init actual_error with an invalid value so that we make sure |
| 242 | // ParsePayloadMetadata properly populates it in all cases. |
| 243 | actual_error = ErrorCode::kUmaReportedMax; |
| 244 | actual_result = performer_.ParsePayloadMetadata(payload, &actual_error); |
| 245 | |
| 246 | EXPECT_EQ(expected_result, actual_result); |
| 247 | EXPECT_EQ(expected_error, actual_error); |
| 248 | |
| 249 | // Check that the parsed metadata size is what's expected. This test |
| 250 | // implicitly confirms that the metadata signature is valid, if required. |
| 251 | EXPECT_EQ(install_plan_.metadata_size, performer_.GetMetadataSize()); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 252 | } |
| 253 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 254 | FakePrefs prefs_; |
| 255 | InstallPlan install_plan_; |
| 256 | FakeSystemState fake_system_state_; |
| 257 | DeltaPerformer performer_{&prefs_, &fake_system_state_, &install_plan_}; |
| 258 | }; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 259 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 260 | TEST_F(DeltaPerformerTest, FullPayloadWriteTest) { |
| 261 | install_plan_.is_full_update = true; |
| 262 | chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString), |
| 263 | std::end(kRandomString)); |
| 264 | expected_data.resize(4096); // block size |
| 265 | vector<AnnotatedOperation> aops; |
| 266 | AnnotatedOperation aop; |
| 267 | *(aop.op.add_dst_extents()) = ExtentForRange(0, 1); |
| 268 | aop.op.set_data_offset(0); |
| 269 | aop.op.set_data_length(expected_data.size()); |
| 270 | aop.op.set_type(InstallOperation::REPLACE); |
| 271 | aops.push_back(aop); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 272 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 273 | chromeos::Blob payload_data = GeneratePayload(expected_data, aops, false, |
Alex Deymo | cbf0989 | 2015-09-11 16:13:16 -0700 | [diff] [blame^] | 274 | kFullPayloadMinorVersion); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 275 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 276 | EXPECT_EQ(expected_data, ApplyPayload(payload_data, "")); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 277 | } |
| 278 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 279 | TEST_F(DeltaPerformerTest, ReplaceOperationTest) { |
| 280 | chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString), |
| 281 | std::end(kRandomString)); |
| 282 | expected_data.resize(4096); // block size |
| 283 | vector<AnnotatedOperation> aops; |
| 284 | AnnotatedOperation aop; |
| 285 | *(aop.op.add_dst_extents()) = ExtentForRange(0, 1); |
| 286 | aop.op.set_data_offset(0); |
| 287 | aop.op.set_data_length(expected_data.size()); |
| 288 | aop.op.set_type(InstallOperation::REPLACE); |
| 289 | aops.push_back(aop); |
| 290 | |
| 291 | chromeos::Blob payload_data = GeneratePayload(expected_data, aops, false, |
| 292 | kSourceMinorPayloadVersion); |
| 293 | |
| 294 | EXPECT_EQ(expected_data, ApplyPayload(payload_data, "/dev/null")); |
| 295 | } |
| 296 | |
| 297 | TEST_F(DeltaPerformerTest, ReplaceBzOperationTest) { |
| 298 | chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString), |
| 299 | std::end(kRandomString)); |
| 300 | expected_data.resize(4096); // block size |
| 301 | chromeos::Blob bz_data; |
| 302 | EXPECT_TRUE(BzipCompress(expected_data, &bz_data)); |
| 303 | |
| 304 | vector<AnnotatedOperation> aops; |
| 305 | AnnotatedOperation aop; |
| 306 | *(aop.op.add_dst_extents()) = ExtentForRange(0, 1); |
| 307 | aop.op.set_data_offset(0); |
| 308 | aop.op.set_data_length(bz_data.size()); |
| 309 | aop.op.set_type(InstallOperation::REPLACE_BZ); |
| 310 | aops.push_back(aop); |
| 311 | |
| 312 | chromeos::Blob payload_data = GeneratePayload(bz_data, aops, false, |
| 313 | kSourceMinorPayloadVersion); |
| 314 | |
| 315 | EXPECT_EQ(expected_data, ApplyPayload(payload_data, "/dev/null")); |
| 316 | } |
| 317 | |
| 318 | TEST_F(DeltaPerformerTest, SourceCopyOperationTest) { |
| 319 | chromeos::Blob expected_data = chromeos::Blob(std::begin(kRandomString), |
| 320 | std::end(kRandomString)); |
| 321 | expected_data.resize(4096); // block size |
| 322 | vector<AnnotatedOperation> aops; |
| 323 | AnnotatedOperation aop; |
| 324 | *(aop.op.add_src_extents()) = ExtentForRange(0, 1); |
| 325 | *(aop.op.add_dst_extents()) = ExtentForRange(0, 1); |
| 326 | aop.op.set_type(InstallOperation::SOURCE_COPY); |
| 327 | aops.push_back(aop); |
| 328 | |
| 329 | chromeos::Blob payload_data = GeneratePayload(chromeos::Blob(), aops, false, |
| 330 | kSourceMinorPayloadVersion); |
| 331 | string source_path; |
| 332 | EXPECT_TRUE(utils::MakeTempFile("Source-XXXXXX", |
| 333 | &source_path, nullptr)); |
| 334 | ScopedPathUnlinker path_unlinker(source_path); |
| 335 | EXPECT_TRUE(utils::WriteFile(source_path.c_str(), |
| 336 | expected_data.data(), |
| 337 | expected_data.size())); |
| 338 | |
| 339 | EXPECT_EQ(expected_data, ApplyPayload(payload_data, source_path)); |
| 340 | } |
| 341 | |
| 342 | TEST_F(DeltaPerformerTest, ExtentsToByteStringTest) { |
Allie Wood | 5687345 | 2015-03-27 17:48:40 -0700 | [diff] [blame] | 343 | uint64_t test[] = {1, 1, 4, 2, 0, 1}; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 344 | COMPILE_ASSERT(arraysize(test) % 2 == 0, array_size_uneven); |
| 345 | const uint64_t block_size = 4096; |
Allie Wood | 5687345 | 2015-03-27 17:48:40 -0700 | [diff] [blame] | 346 | const uint64_t file_length = 4 * block_size - 13; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 347 | |
| 348 | google::protobuf::RepeatedPtrField<Extent> extents; |
| 349 | for (size_t i = 0; i < arraysize(test); i += 2) { |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 350 | *(extents.Add()) = ExtentForRange(test[i], test[i + 1]); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 351 | } |
| 352 | |
Allie Wood | 5687345 | 2015-03-27 17:48:40 -0700 | [diff] [blame] | 353 | string expected_output = "4096:4096,16384:8192,0:4083"; |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 354 | string actual_output; |
| 355 | EXPECT_TRUE(DeltaPerformer::ExtentsToBsdiffPositionsString(extents, |
| 356 | block_size, |
| 357 | file_length, |
| 358 | &actual_output)); |
| 359 | EXPECT_EQ(expected_output, actual_output); |
| 360 | } |
Darin Petkov | 68c10d1 | 2010-10-14 09:24:37 -0700 | [diff] [blame] | 361 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 362 | TEST_F(DeltaPerformerTest, ValidateManifestFullGoodTest) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 363 | // The Manifest we are validating. |
| 364 | DeltaArchiveManifest manifest; |
| 365 | manifest.mutable_new_kernel_info(); |
| 366 | manifest.mutable_new_rootfs_info(); |
Alex Deymo | cbf0989 | 2015-09-11 16:13:16 -0700 | [diff] [blame^] | 367 | manifest.set_minor_version(kFullPayloadMinorVersion); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 368 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 369 | RunManifestValidation(manifest, true, ErrorCode::kSuccess); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 370 | } |
| 371 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 372 | TEST_F(DeltaPerformerTest, ValidateManifestDeltaGoodTest) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 373 | // The Manifest we are validating. |
| 374 | DeltaArchiveManifest manifest; |
| 375 | manifest.mutable_old_kernel_info(); |
| 376 | manifest.mutable_old_rootfs_info(); |
| 377 | manifest.mutable_new_kernel_info(); |
| 378 | manifest.mutable_new_rootfs_info(); |
| 379 | manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion); |
| 380 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 381 | RunManifestValidation(manifest, false, ErrorCode::kSuccess); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 382 | } |
| 383 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 384 | TEST_F(DeltaPerformerTest, ValidateManifestFullUnsetMinorVersion) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 385 | // The Manifest we are validating. |
| 386 | DeltaArchiveManifest manifest; |
| 387 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 388 | RunManifestValidation(manifest, true, ErrorCode::kSuccess); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 389 | } |
| 390 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 391 | TEST_F(DeltaPerformerTest, ValidateManifestDeltaUnsetMinorVersion) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 392 | // The Manifest we are validating. |
| 393 | DeltaArchiveManifest manifest; |
| 394 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 395 | RunManifestValidation(manifest, false, |
| 396 | ErrorCode::kUnsupportedMinorPayloadVersion); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 397 | } |
| 398 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 399 | TEST_F(DeltaPerformerTest, ValidateManifestFullOldKernelTest) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 400 | // The Manifest we are validating. |
| 401 | DeltaArchiveManifest manifest; |
| 402 | manifest.mutable_old_kernel_info(); |
| 403 | manifest.mutable_new_kernel_info(); |
| 404 | manifest.mutable_new_rootfs_info(); |
| 405 | manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion); |
| 406 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 407 | RunManifestValidation(manifest, true, ErrorCode::kPayloadMismatchedType); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 408 | } |
| 409 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 410 | TEST_F(DeltaPerformerTest, ValidateManifestFullOldRootfsTest) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 411 | // The Manifest we are validating. |
| 412 | DeltaArchiveManifest manifest; |
| 413 | manifest.mutable_old_rootfs_info(); |
| 414 | manifest.mutable_new_kernel_info(); |
| 415 | manifest.mutable_new_rootfs_info(); |
| 416 | manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion); |
| 417 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 418 | RunManifestValidation(manifest, true, ErrorCode::kPayloadMismatchedType); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 419 | } |
| 420 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 421 | TEST_F(DeltaPerformerTest, ValidateManifestBadMinorVersion) { |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 422 | // The Manifest we are validating. |
| 423 | DeltaArchiveManifest manifest; |
| 424 | |
| 425 | // Generate a bad version number. |
| 426 | manifest.set_minor_version(DeltaPerformer::kSupportedMinorPayloadVersion + |
| 427 | 10000); |
| 428 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 429 | RunManifestValidation(manifest, false, |
| 430 | ErrorCode::kUnsupportedMinorPayloadVersion); |
Don Garrett | b8dd1d9 | 2013-11-22 17:40:02 -0800 | [diff] [blame] | 431 | } |
| 432 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 433 | TEST_F(DeltaPerformerTest, BadDeltaMagicTest) { |
| 434 | EXPECT_EQ(0, performer_.Open("/dev/null", 0, 0)); |
| 435 | EXPECT_TRUE(performer_.OpenKernel("/dev/null")); |
| 436 | EXPECT_TRUE(performer_.Write("junk", 4)); |
| 437 | EXPECT_TRUE(performer_.Write("morejunk", 8)); |
| 438 | EXPECT_FALSE(performer_.Write("morejunk", 8)); |
| 439 | EXPECT_LT(performer_.Close(), 0); |
Darin Petkov | 934bb41 | 2010-11-18 11:21:35 -0800 | [diff] [blame] | 440 | } |
| 441 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 442 | TEST_F(DeltaPerformerTest, WriteUpdatesPayloadState) { |
| 443 | EXPECT_EQ(0, performer_.Open("/dev/null", 0, 0)); |
| 444 | EXPECT_TRUE(performer_.OpenKernel("/dev/null")); |
Jay Srinivasan | 2b5a0f0 | 2012-12-19 17:25:56 -0800 | [diff] [blame] | 445 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 446 | EXPECT_CALL(*(fake_system_state_.mock_payload_state()), |
Jay Srinivasan | 2b5a0f0 | 2012-12-19 17:25:56 -0800 | [diff] [blame] | 447 | DownloadProgress(4)).Times(1); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 448 | EXPECT_CALL(*(fake_system_state_.mock_payload_state()), |
Jay Srinivasan | 2b5a0f0 | 2012-12-19 17:25:56 -0800 | [diff] [blame] | 449 | DownloadProgress(8)).Times(2); |
| 450 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 451 | EXPECT_TRUE(performer_.Write("junk", 4)); |
| 452 | EXPECT_TRUE(performer_.Write("morejunk", 8)); |
| 453 | EXPECT_FALSE(performer_.Write("morejunk", 8)); |
| 454 | EXPECT_LT(performer_.Close(), 0); |
Jay Srinivasan | 2b5a0f0 | 2012-12-19 17:25:56 -0800 | [diff] [blame] | 455 | } |
| 456 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 457 | TEST_F(DeltaPerformerTest, MissingMandatoryMetadataSizeTest) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 458 | DoMetadataSizeTest(0, 75456, true); |
Jay Srinivasan | 51dcf26 | 2012-09-13 17:24:32 -0700 | [diff] [blame] | 459 | } |
| 460 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 461 | TEST_F(DeltaPerformerTest, MissingNonMandatoryMetadataSizeTest) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 462 | DoMetadataSizeTest(0, 123456, false); |
| 463 | } |
| 464 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 465 | TEST_F(DeltaPerformerTest, InvalidMandatoryMetadataSizeTest) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 466 | DoMetadataSizeTest(13000, 140000, true); |
| 467 | } |
| 468 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 469 | TEST_F(DeltaPerformerTest, InvalidNonMandatoryMetadataSizeTest) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 470 | DoMetadataSizeTest(40000, 50000, false); |
| 471 | } |
| 472 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 473 | TEST_F(DeltaPerformerTest, ValidMandatoryMetadataSizeTest) { |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 474 | DoMetadataSizeTest(85376, 85376, true); |
| 475 | } |
| 476 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 477 | TEST_F(DeltaPerformerTest, MandatoryEmptyMetadataSignatureTest) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 478 | DoMetadataSignatureTest(kEmptyMetadataSignature, true, true); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 479 | } |
| 480 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 481 | TEST_F(DeltaPerformerTest, NonMandatoryEmptyMetadataSignatureTest) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 482 | DoMetadataSignatureTest(kEmptyMetadataSignature, true, false); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 483 | } |
| 484 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 485 | TEST_F(DeltaPerformerTest, MandatoryInvalidMetadataSignatureTest) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 486 | DoMetadataSignatureTest(kInvalidMetadataSignature, true, true); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 487 | } |
| 488 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 489 | TEST_F(DeltaPerformerTest, NonMandatoryInvalidMetadataSignatureTest) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 490 | DoMetadataSignatureTest(kInvalidMetadataSignature, true, false); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 491 | } |
| 492 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 493 | TEST_F(DeltaPerformerTest, MandatoryValidMetadataSignature1Test) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 494 | DoMetadataSignatureTest(kValidMetadataSignature, false, true); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 495 | } |
| 496 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 497 | TEST_F(DeltaPerformerTest, MandatoryValidMetadataSignature2Test) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 498 | DoMetadataSignatureTest(kValidMetadataSignature, true, true); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 499 | } |
| 500 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 501 | TEST_F(DeltaPerformerTest, NonMandatoryValidMetadataSignatureTest) { |
Sen Jiang | d78b389 | 2015-09-25 15:19:31 -0700 | [diff] [blame] | 502 | DoMetadataSignatureTest(kValidMetadataSignature, true, false); |
Jay Srinivasan | 738fdf3 | 2012-12-07 17:40:54 -0800 | [diff] [blame] | 503 | } |
| 504 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 505 | TEST_F(DeltaPerformerTest, UsePublicKeyFromResponse) { |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 506 | base::FilePath key_path; |
| 507 | |
| 508 | // The result of the GetPublicKeyResponse() method is based on three things |
| 509 | // |
| 510 | // 1. Whether it's an official build; and |
| 511 | // 2. Whether the Public RSA key to be used is in the root filesystem; and |
Alex Vakulenko | 072359c | 2014-07-18 11:41:07 -0700 | [diff] [blame] | 512 | // 3. Whether the response has a public key |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 513 | // |
| 514 | // We test all eight combinations to ensure that we only use the |
| 515 | // public key in the response if |
| 516 | // |
| 517 | // a. it's not an official build; and |
| 518 | // b. there is no key in the root filesystem. |
| 519 | |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 520 | FakeHardware* fake_hardware = fake_system_state_.fake_hardware(); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 521 | |
| 522 | string temp_dir; |
Gilad Arnold | a6742b3 | 2014-01-11 00:18:34 -0800 | [diff] [blame] | 523 | EXPECT_TRUE(utils::MakeTempDirectory("PublicKeyFromResponseTests.XXXXXX", |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 524 | &temp_dir)); |
| 525 | string non_existing_file = temp_dir + "/non-existing"; |
| 526 | string existing_file = temp_dir + "/existing"; |
Alex Vakulenko | 75039d7 | 2014-03-25 12:36:28 -0700 | [diff] [blame] | 527 | EXPECT_EQ(0, System(base::StringPrintf("touch %s", existing_file.c_str()))); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 528 | |
| 529 | // Non-official build, non-existing public-key, key in response -> true |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 530 | fake_hardware->SetIsOfficialBuild(false); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 531 | performer_.public_key_path_ = non_existing_file; |
| 532 | install_plan_.public_key_rsa = "VGVzdAo="; // result of 'echo "Test" | base64' |
| 533 | EXPECT_TRUE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 534 | EXPECT_FALSE(key_path.empty()); |
| 535 | EXPECT_EQ(unlink(key_path.value().c_str()), 0); |
| 536 | // Same with official build -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 537 | fake_hardware->SetIsOfficialBuild(true); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 538 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 539 | |
| 540 | // Non-official build, existing public-key, key in response -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 541 | fake_hardware->SetIsOfficialBuild(false); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 542 | performer_.public_key_path_ = existing_file; |
| 543 | install_plan_.public_key_rsa = "VGVzdAo="; // result of 'echo "Test" | base64' |
| 544 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 545 | // Same with official build -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 546 | fake_hardware->SetIsOfficialBuild(true); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 547 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 548 | |
| 549 | // Non-official build, non-existing public-key, no key in response -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 550 | fake_hardware->SetIsOfficialBuild(false); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 551 | performer_.public_key_path_ = non_existing_file; |
| 552 | install_plan_.public_key_rsa = ""; |
| 553 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 554 | // Same with official build -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 555 | fake_hardware->SetIsOfficialBuild(true); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 556 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 557 | |
| 558 | // Non-official build, existing public-key, no key in response -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 559 | fake_hardware->SetIsOfficialBuild(false); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 560 | performer_.public_key_path_ = existing_file; |
| 561 | install_plan_.public_key_rsa = ""; |
| 562 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 563 | // Same with official build -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 564 | fake_hardware->SetIsOfficialBuild(true); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 565 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 566 | |
| 567 | // Non-official build, non-existing public-key, key in response |
| 568 | // but invalid base64 -> false |
Don Garrett | 6646b44 | 2013-11-13 15:29:11 -0800 | [diff] [blame] | 569 | fake_hardware->SetIsOfficialBuild(false); |
Sen Jiang | 2d528b4 | 2015-09-25 11:18:12 -0700 | [diff] [blame] | 570 | performer_.public_key_path_ = non_existing_file; |
| 571 | install_plan_.public_key_rsa = "not-valid-base64"; |
| 572 | EXPECT_FALSE(performer_.GetPublicKeyFromResponse(&key_path)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 573 | |
Alex Deymo | 10875d9 | 2014-11-10 21:52:57 -0800 | [diff] [blame] | 574 | EXPECT_TRUE(test_utils::RecursiveUnlinkDir(temp_dir)); |
David Zeuthen | e7f8917 | 2013-10-31 10:21:04 -0700 | [diff] [blame] | 575 | } |
| 576 | |
Alex Deymo | cbf0989 | 2015-09-11 16:13:16 -0700 | [diff] [blame^] | 577 | TEST_F(DeltaPerformerTest, ConfVersionsMatch) { |
| 578 | // Test that the versions in update_engine.conf that is installed to the |
| 579 | // image match the supported delta versions in the update engine. |
Allie Wood | 78750a4 | 2015-02-11 15:42:11 -0800 | [diff] [blame] | 580 | uint32_t minor_version; |
Alex Deymo | b42b98d | 2015-07-06 17:42:38 -0700 | [diff] [blame] | 581 | chromeos::KeyValueStore store; |
| 582 | EXPECT_TRUE(store.Load(base::FilePath("update_engine.conf"))); |
| 583 | EXPECT_TRUE(utils::GetMinorVersion(store, &minor_version)); |
| 584 | EXPECT_EQ(DeltaPerformer::kSupportedMinorPayloadVersion, minor_version); |
Alex Deymo | cbf0989 | 2015-09-11 16:13:16 -0700 | [diff] [blame^] | 585 | |
| 586 | string major_version_str; |
| 587 | uint64_t major_version; |
| 588 | EXPECT_TRUE(store.GetString("PAYLOAD_MAJOR_VERSION", &major_version_str)); |
| 589 | EXPECT_TRUE(base::StringToUint64(major_version_str, &major_version)); |
| 590 | EXPECT_EQ(DeltaPerformer::kSupportedMajorPayloadVersion, major_version); |
Allie Wood | 78750a4 | 2015-02-11 15:42:11 -0800 | [diff] [blame] | 591 | } |
| 592 | |
Andrew de los Reyes | 09e56d6 | 2010-04-23 13:45:53 -0700 | [diff] [blame] | 593 | } // namespace chromeos_update_engine |