| /* |
| * Copyright (C) 2019 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include "misc_writer/misc_writer.h" |
| |
| #include <android-base/file.h> |
| #include <android-base/logging.h> |
| #include <android-base/properties.h> |
| #include <android-base/stringprintf.h> |
| #include <bootloader_message/bootloader_message.h> |
| #include <string.h> |
| |
| namespace android { |
| namespace hardware { |
| namespace google { |
| namespace pixel { |
| |
| bool MiscWriter::OffsetAndSizeInVendorSpace(size_t offset, size_t size) { |
| auto total_size = WIPE_PACKAGE_OFFSET_IN_MISC - VENDOR_SPACE_OFFSET_IN_MISC; |
| return size <= total_size && offset <= total_size - size; |
| } |
| |
| bool MiscWriter::WriteMiscPartitionVendorSpace(const void* data, size_t size, size_t offset, |
| std::string* err) { |
| if (!OffsetAndSizeInVendorSpace(offset, size)) { |
| *err = android::base::StringPrintf("Out of bound write (offset %zu size %zu)", offset, size); |
| return false; |
| } |
| auto misc_blk_device = get_misc_blk_device(err); |
| if (misc_blk_device.empty()) { |
| return false; |
| } |
| return write_misc_partition(data, size, misc_blk_device, VENDOR_SPACE_OFFSET_IN_MISC + offset, |
| err); |
| } |
| |
| bool MiscWriter::PerformAction(std::optional<size_t> override_offset) { |
| size_t offset = 0; |
| std::string content; |
| switch (action_) { |
| case MiscWriterActions::kSetDarkThemeFlag: |
| case MiscWriterActions::kClearDarkThemeFlag: |
| offset = override_offset.value_or(kThemeFlagOffsetInVendorSpace); |
| content = (action_ == MiscWriterActions::kSetDarkThemeFlag) |
| ? kDarkThemeFlag |
| : std::string(strlen(kDarkThemeFlag), 0); |
| break; |
| case MiscWriterActions::kSetSotaFlag: |
| case MiscWriterActions::kClearSotaFlag: |
| offset = override_offset.value_or(kSotaFlagOffsetInVendorSpace); |
| content = (action_ == MiscWriterActions::kSetSotaFlag) ? kSotaFlag |
| : std::string(strlen(kSotaFlag), 0); |
| break; |
| case MiscWriterActions::kSetEnablePkvmFlag: |
| case MiscWriterActions::kSetDisablePkvmFlag: |
| offset = override_offset.value_or(kPkvmFlagOffsetInVendorSpace); |
| content = (action_ == MiscWriterActions::kSetEnablePkvmFlag) ? kEnablePkvmFlag |
| : kDisablePkvmFlag; |
| break; |
| case MiscWriterActions::kSetWristOrientationFlag: |
| case MiscWriterActions::kClearWristOrientationFlag: |
| offset = override_offset.value_or(kWristOrientationFlagOffsetInVendorSpace); |
| content = (action_ == MiscWriterActions::kSetWristOrientationFlag) |
| ? std::string(kWristOrientationFlag) + chardata_ |
| : std::string(strlen(kWristOrientationFlag) + sizeof(chardata_), 0); |
| break; |
| case MiscWriterActions::kWriteTimeFormat: |
| offset = override_offset.value_or(kTimeFormatValOffsetInVendorSpace); |
| content = std::string(kTimeFormat) + chardata_; |
| break; |
| case MiscWriterActions::kWriteTimeOffset: |
| offset = override_offset.value_or(kTimeOffsetValOffsetInVendorSpace); |
| content = std::string(kTimeOffset) + stringdata_; |
| content.resize(strlen(kTimeOffset) + std::to_string(kMinTimeOffset).size(), 0); |
| break; |
| case MiscWriterActions::kSetMaxRamSize: |
| case MiscWriterActions::kClearMaxRamSize: |
| offset = override_offset.value_or(kMaxRamSizeOffsetInVendorSpace); |
| content = (action_ == MiscWriterActions::kSetMaxRamSize) |
| ? std::string(kMaxRamSize).append(stringdata_).append("\n") |
| : std::string(32, 0); |
| break; |
| case MiscWriterActions::kWriteTimeRtcOffset: |
| offset = override_offset.value_or(kRTimeRtcOffsetValOffsetInVendorSpace); |
| content = std::string(kTimeRtcOffset) + stringdata_; |
| content.resize(32); |
| break; |
| case MiscWriterActions::kWriteTimeMinRtc: |
| offset = override_offset.value_or(kRTimeMinRtcValOffsetInVendorSpace); |
| content = std::string(kTimeMinRtc) + stringdata_; |
| content.resize(32); |
| break; |
| case MiscWriterActions::kSetSotaConfig: |
| goto sota_config; |
| case MiscWriterActions::kWriteDstTransition: |
| offset = override_offset.value_or(kDstTransitionOffsetInVendorSpace); |
| content = std::string(kDstTransition) + stringdata_; |
| content.resize(32); |
| break; |
| case MiscWriterActions::kWriteDstOffset: |
| offset = override_offset.value_or(kDstOffsetOffsetInVendorSpace); |
| content = std::string(kDstOffset) + stringdata_; |
| content.resize(32); |
| break; |
| case MiscWriterActions::kUnset: |
| LOG(ERROR) << "The misc writer action must be set"; |
| return false; |
| } |
| |
| if (std::string err; |
| !WriteMiscPartitionVendorSpace(content.data(), content.size(), offset, &err)) { |
| LOG(ERROR) << "Failed to write " << content << " at offset " << offset << " : " << err; |
| return false; |
| } |
| |
| sota_config: |
| if (action_ == MiscWriterActions::kSetSotaFlag || action_ == MiscWriterActions::kSetSotaConfig) { |
| content = ::android::base::GetProperty("persist.vendor.nfc.factoryota.state", ""); |
| if (content.size() != 0 && content.size() <= 40) { |
| offset = kSotaStateOffsetInVendorSpace; |
| if (std::string err; |
| !WriteMiscPartitionVendorSpace(content.data(), content.size(), offset, &err)) { |
| LOG(ERROR) << "Failed to write " << content << " at offset " << offset << " : " << err; |
| return false; |
| } |
| } |
| content = ::android::base::GetProperty("persist.vendor.nfc.factoryota.schedule_shipmode", ""); |
| if (content.size() != 0 && content.size() <= 32) { |
| offset = kSotaScheduleShipmodeOffsetInVendorSpace; |
| if (std::string err; |
| !WriteMiscPartitionVendorSpace(content.data(), content.size(), offset, &err)) { |
| LOG(ERROR) << "Failed to write " << content << " at offset " << offset << " : " << err; |
| return false; |
| } |
| } |
| } |
| |
| return true; |
| } |
| |
| } // namespace pixel |
| } // namespace google |
| } // namespace hardware |
| } // namespace android |