1 //
2 // Copyright (C) 2013 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 //
16
17 #include "update_engine/payload_consumer/install_plan.h"
18
19 #include <algorithm>
20 #include <utility>
21
22 #include <base/format_macros.h>
23 #include <base/logging.h>
24 #include <base/strings/string_number_conversions.h>
25 #include <base/strings/string_util.h>
26 #include <base/strings/stringprintf.h>
27
28 #include "update_engine/common/utils.h"
29 #include "update_engine/payload_consumer/payload_constants.h"
30
31 using std::string;
32 using std::vector;
33
34 namespace chromeos_update_engine {
35
36 namespace {
PayloadUrlsToString(const decltype(InstallPlan::Payload::payload_urls) & payload_urls)37 string PayloadUrlsToString(
38 const decltype(InstallPlan::Payload::payload_urls)& payload_urls) {
39 return "(" + base::JoinString(payload_urls, ",") + ")";
40 }
41
VectorToString(const vector<std::pair<string,string>> & input,const string & separator)42 string VectorToString(const vector<std::pair<string, string>>& input,
43 const string& separator) {
44 vector<string> vec;
45 std::transform(input.begin(),
46 input.end(),
47 std::back_inserter(vec),
48 [](const auto& pair) {
49 return base::JoinString({pair.first, pair.second}, ": ");
50 });
51 return base::JoinString(vec, separator);
52 }
53 } // namespace
54
InstallPayloadTypeToString(InstallPayloadType type)55 string InstallPayloadTypeToString(InstallPayloadType type) {
56 switch (type) {
57 case InstallPayloadType::kUnknown:
58 return "unknown";
59 case InstallPayloadType::kFull:
60 return "full";
61 case InstallPayloadType::kDelta:
62 return "delta";
63 }
64 return "invalid type";
65 }
66
operator ==(const InstallPlan & that) const67 bool InstallPlan::operator==(const InstallPlan& that) const {
68 return ((is_resume == that.is_resume) &&
69 (download_url == that.download_url) && (payloads == that.payloads) &&
70 (source_slot == that.source_slot) &&
71 (target_slot == that.target_slot) && (partitions == that.partitions));
72 }
73
operator !=(const InstallPlan & that) const74 bool InstallPlan::operator!=(const InstallPlan& that) const {
75 return !((*this) == that);
76 }
77
Dump() const78 void InstallPlan::Dump() const {
79 LOG(INFO) << "InstallPlan: \n" << ToString();
80 }
81
ToString() const82 string InstallPlan::ToString() const {
83 string url_str = download_url;
84 if (base::StartsWith(
85 url_str, "fd://", base::CompareCase::INSENSITIVE_ASCII)) {
86 int fd = std::stoi(url_str.substr(strlen("fd://")));
87 url_str = utils::GetFilePath(fd);
88 }
89
90 vector<string> result_str;
91 result_str.emplace_back(VectorToString(
92 {
93 {"type", (is_resume ? "resume" : "new_update")},
94 {"version", version},
95 {"source_slot", BootControlInterface::SlotName(source_slot)},
96 {"target_slot", BootControlInterface::SlotName(target_slot)},
97 {"initial url", url_str},
98 {"hash_checks_mandatory", utils::ToString(hash_checks_mandatory)},
99 {"powerwash_required", utils::ToString(powerwash_required)},
100 {"switch_slot_on_reboot", utils::ToString(switch_slot_on_reboot)},
101 {"run_post_install", utils::ToString(run_post_install)},
102 {"is_rollback", utils::ToString(is_rollback)},
103 {"rollback_data_save_requested",
104 utils::ToString(rollback_data_save_requested)},
105 {"write_verity", utils::ToString(write_verity)},
106 },
107 "\n"));
108
109 for (const auto& partition : partitions) {
110 result_str.emplace_back(VectorToString(
111 {
112 {"Partition", partition.name},
113 {"source_size", base::NumberToString(partition.source_size)},
114 {"source_path", partition.source_path},
115 {"source_hash",
116 base::HexEncode(partition.source_hash.data(),
117 partition.source_hash.size())},
118 {"target_size", base::NumberToString(partition.target_size)},
119 {"target_path", partition.target_path},
120 {"target_hash",
121 base::HexEncode(partition.target_hash.data(),
122 partition.target_hash.size())},
123 {"run_postinstall", utils::ToString(partition.run_postinstall)},
124 {"postinstall_path", partition.postinstall_path},
125 {"readonly_target_path", partition.readonly_target_path},
126 {"filesystem_type", partition.filesystem_type},
127 },
128 "\n "));
129 }
130
131 for (unsigned int i = 0; i < payloads.size(); ++i) {
132 const auto& payload = payloads[i];
133 result_str.emplace_back(VectorToString(
134 {
135 {"Payload", base::NumberToString(i)},
136 {"urls", PayloadUrlsToString(payload.payload_urls)},
137 {"size", base::NumberToString(payload.size)},
138 {"metadata_size", base::NumberToString(payload.metadata_size)},
139 {"metadata_signature", payload.metadata_signature},
140 {"hash", base::HexEncode(payload.hash.data(), payload.hash.size())},
141 {"type", InstallPayloadTypeToString(payload.type)},
142 {"fingerprint", payload.fp},
143 {"app_id", payload.app_id},
144 {"already_applied", utils::ToString(payload.already_applied)},
145 },
146 "\n "));
147 }
148
149 return base::JoinString(result_str, "\n");
150 }
151
LoadPartitionsFromSlots(BootControlInterface * boot_control)152 bool InstallPlan::LoadPartitionsFromSlots(BootControlInterface* boot_control) {
153 bool result = true;
154 for (Partition& partition : partitions) {
155 if (source_slot != BootControlInterface::kInvalidSlot &&
156 partition.source_size > 0) {
157 TEST_AND_RETURN_FALSE(boot_control->GetPartitionDevice(
158 partition.name, source_slot, &partition.source_path));
159 } else {
160 partition.source_path.clear();
161 }
162
163 if (target_slot != BootControlInterface::kInvalidSlot &&
164 partition.target_size > 0) {
165 auto device = boot_control->GetPartitionDevice(
166 partition.name, target_slot, source_slot);
167 TEST_AND_RETURN_FALSE(device.has_value());
168 partition.target_path = device->rw_device_path;
169 partition.readonly_target_path = device->readonly_device_path;
170 } else {
171 partition.target_path.clear();
172 }
173 }
174 return result;
175 }
176
operator ==(const InstallPlan::Partition & that) const177 bool InstallPlan::Partition::operator==(
178 const InstallPlan::Partition& that) const {
179 return (name == that.name && source_path == that.source_path &&
180 source_size == that.source_size && source_hash == that.source_hash &&
181 target_path == that.target_path && target_size == that.target_size &&
182 target_hash == that.target_hash &&
183 run_postinstall == that.run_postinstall &&
184 postinstall_path == that.postinstall_path &&
185 filesystem_type == that.filesystem_type &&
186 postinstall_optional == that.postinstall_optional);
187 }
188
189 } // namespace chromeos_update_engine
190