* Copyright (c) 2021 Huawei Device Co., Ltd.
* 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 "applypatch/partition_record.h"
#include <cerrno>
#include <fcntl.h>
#include <unistd.h>
#include "fs_manager/mount.h"
#include "log/log.h"
#include "securec.h"
namespace Updater {
PartitionRecord &PartitionRecord::GetInstance()
{
static PartitionRecord partitionRecord;
return partitionRecord;
}
bool PartitionRecord::IsPartitionUpdated(const std::string &partitionName)
{
auto miscBlockDevice = GetMiscPartitionPath();
uint8_t buffer[PARTITION_UPDATER_RECORD_MSG_SIZE];
if (!miscBlockDevice.empty()) {
char *realPath = realpath(miscBlockDevice.c_str(), NULL);
if (realPath == nullptr) {
LOG(ERROR) << "realPath is NULL" << " : " << strerror(errno);
return false;
}
int fd = open(realPath, O_RDONLY | O_EXCL | O_CLOEXEC | O_BINARY);
free(realPath);
if (fd < 0) {
LOG(ERROR) << "PartitionRecord: Open misc to recording partition failed" << " : " << strerror(errno);
return false;
}
fdsan_exchange_owner_tag(fd, 0, FDSAN_UPDATER_TAG);
if (lseek(fd, PARTITION_RECORD_START, SEEK_CUR) < 0) {
LOG(ERROR) << "PartitionRecord: Seek misc to specific offset failed" << " : " << strerror(errno);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
if (read(fd, buffer, PARTITION_UPDATER_RECORD_MSG_SIZE) != PARTITION_UPDATER_RECORD_MSG_SIZE) {
LOG(ERROR) << "PartitionRecord: Read from misc partition failed" << " : " << strerror(errno);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
for (uint8_t *p = buffer; p < buffer + PARTITION_UPDATER_RECORD_MSG_SIZE; p += sizeof(PartitionRecordInfo)) {
PartitionRecordInfo *pri = reinterpret_cast<PartitionRecordInfo*>(p);
if (strcmp(pri->partitionName, partitionName.c_str()) == 0) {
LOG(DEBUG) << "PartitionRecord: Found " << partitionName << " record in misc partition";
LOG(DEBUG) << "PartitionRecord: update status: " << pri->updated;
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return pri->updated;
}
}
fsync(fd);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
LOG(INFO) << "PartitionRecord: Cannot found " << partitionName << " record in misc partition";
}
return false;
}
bool PartitionRecord::RecordPartitionSetOffset(int fd)
{
off_t newOffset = 0;
if (lseek(fd, PARTITION_RECORD_OFFSET, SEEK_SET) < 0) {
LOG(ERROR) << "PartitionRecord: Seek misc to record offset failed: " << strerror(errno);
return false;
}
if (read(fd, &newOffset, sizeof(off_t)) != static_cast<ssize_t>(sizeof(off_t))) {
LOG(ERROR) << "PartitionRecord: Read offset failed: " << strerror(errno);
return false;
}
offset_ = newOffset;
if (lseek(fd, PARTITION_RECORD_START + offset_, SEEK_SET) < 0) {
LOG(ERROR) << "PartitionRecord: Seek misc to specific offset failed: " << strerror(errno);
return false;
}
return true;
}
bool PartitionRecord::RecordPartitionSetInfo(const std::string &partitionName, bool updated, int fd)
{
(void)memset_s(&info_, sizeof(info_), 0, sizeof(info_));
if (strncpy_s(info_.partitionName, PARTITION_NAME_LEN, partitionName.c_str(), PARTITION_NAME_LEN - 1) != EOK) {
LOG(ERROR) << "PartitionRecord: strncpy_s failed: " << strerror(errno);
return false;
}
info_.updated = updated;
if (write(fd, &info_, sizeof(PartitionRecordInfo)) != static_cast<ssize_t>(sizeof(PartitionRecordInfo))) {
LOG(ERROR) << "PartitionRecord: write failed: " << strerror(errno);
return false;
}
offset_ += static_cast<off_t>(sizeof(PartitionRecordInfo));
if (lseek(fd, PARTITION_RECORD_OFFSET, SEEK_SET) < 0) {
LOG(ERROR) << "PartitionRecord: Seek misc to record offset failed: " << strerror(errno);
return false;
}
if (write(fd, &offset_, sizeof(off_t)) != static_cast<ssize_t>(sizeof(off_t))) {
LOG(ERROR) << "PartitionRecord: write misc to record offset failed: " << strerror(errno);
return false;
}
return true;
}
bool PartitionRecord::RecordPartitionUpdateStatus(const std::string &partitionName, bool updated)
{
auto miscBlockDevice = GetMiscPartitionPath();
if (!miscBlockDevice.empty()) {
char *realPath = realpath(miscBlockDevice.c_str(), NULL);
if (realPath == nullptr) {
LOG(ERROR) << "realPath is NULL" << " : " << strerror(errno);
return false;
}
int fd = open(realPath, O_RDWR | O_EXCL | O_CLOEXEC | O_BINARY);
free(realPath);
if (fd < 0) {
LOG(ERROR) << "PartitionRecord: Open misc to recording partition failed" << " : " << strerror(errno);
return false;
}
fdsan_exchange_owner_tag(fd, 0, FDSAN_UPDATER_TAG);
if (!RecordPartitionSetOffset(fd)) {
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
if (offset_ + static_cast<off_t>(sizeof(PartitionRecordInfo)) < PARTITION_UPDATER_RECORD_SIZE) {
if (!RecordPartitionSetInfo(partitionName, updated, fd)) {
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
LOG(DEBUG) << "PartitionRecord: offset is " << offset_;
} else {
LOG(WARNING) << "PartitionRecord: partition record overflow, offset = " << offset_;
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
LOG(DEBUG) << "PartitionRecord: record " << partitionName << " successfully.";
fsync(fd);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
}
return true;
}
bool PartitionRecord::ClearRecordPartitionOffset()
{
auto miscBlockDevice = GetMiscPartitionPath();
if (!miscBlockDevice.empty()) {
int fd = open(miscBlockDevice.c_str(), O_RDWR | O_EXCL | O_CLOEXEC | O_BINARY);
if (fd < 0) {
LOG(ERROR) << "Open misc to recording partition failed" << " : " << strerror(errno);
return false;
}
fdsan_exchange_owner_tag(fd, 0, FDSAN_UPDATER_TAG);
if (lseek(fd, PARTITION_RECORD_OFFSET, SEEK_SET) < 0) {
LOG(ERROR) << "Seek misc to specific offset failed" << " : " << strerror(errno);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
off_t initOffset = 0;
if (write(fd, &initOffset, sizeof(off_t)) != static_cast<ssize_t>(sizeof(off_t))) {
LOG(ERROR) << "StartUpdater: Write misc initOffset 0 failed" << " : " << strerror(errno);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
return false;
}
fsync(fd);
fdsan_close_with_tag(fd, FDSAN_UPDATER_TAG);
}
return true;
}
std::string PartitionRecord::GetMiscPartitionPath(const std::string &misc)
{
auto miscBlockDevice = GetBlockDeviceByMountPoint(misc);
if (miscBlockDevice.empty()) {
LOG(WARNING) << "Can not find misc partition";
}
return miscBlockDevice;
}
}
