* 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 <dlfcn.h>
#include <iostream>
#include "fs_manager/mount.h"
#include "misc_info/misc_info.h"
#include "securec.h"
#include "updater/updater_const.h"
#include "parameter.h"
#include "utils.h"
#include "utils_fs.h"
using namespace std;
using namespace Updater;
constexpr const char *NOTIFY_MISC_INFO = "NotifyWriteMiscInfo";
constexpr const char *HANDLE_MISC_LIB_PATH = "/system/lib64/libupdater_handle_misc.z.so";
static void PrintPrompts()
{
cout << "Please input correct command, examples :" << endl;
cout << "bin : write_updater bin /data/updater/update.bin" <<endl;
cout << "updater : write_updater updater /data/updater/updater.zip" << endl;
cout << "factory_reset : write_updater user_factory_reset" << endl;
cout << "sdcard_update : write_updater sdcard_update" << endl;
cout << "clear command : write_updater clear" << endl;
cout << "updater_para : write_updater updater_para" << endl;
cout << "intral_update : write_updater ota_intral_update /data/updater/updater.zip" << endl;
cout << "subpkg_update : write_updater subpkg_update" << endl;
cout << "notify_update : write_updater notify_update" << endl;
cout << "secure_erase : write_updater secure_erase" << endl;
}
static int ExceptionBin(int argc, char **argv, UpdateMessage &boot)
{
if (argc < BINARY_MAX_ARGS) {
cout << "Please input correct updater command!" << endl;
return -1;
}
if (argv[WRITE_SECOND_CMD] != nullptr) {
if (snprintf_s(boot.update, sizeof(boot.update), sizeof(boot.update) - 1, "--update_bin=%s",
argv[WRITE_SECOND_CMD]) == -1) {
cout << "WriteUpdaterMessage snprintf_s failed!" << endl;
return -1;
}
}
return 0;
}
static int WriteUpdaterLanguage(UpdaterPara ¶)
{
char language[MAX_PARA_SIZE + 1] {};
int res = GetParameter("persist.global.locale", "", language, MAX_PARA_SIZE);
if (res <= 0) {
cout << "Get persist.global.locale parameter failed" << endl;
res = GetParameter("const.global.locale", "", language, MAX_PARA_SIZE);
if (res <= 0) {
cout << "Get const.global.locale parameter failed" << endl;
return -1;
}
}
(void)memset_s(para.language, MAX_PARA_SIZE, 0, MAX_PARA_SIZE);
res = memcpy_s(para.language, MAX_PARA_SIZE, language, sizeof(language) - 1);
para.language[MAX_PARA_SIZE - 1] = '\0';
if (res != 0) {
cout << "memcpy_s para.language failed" << endl;
return -1;
}
return 0;
}
static int WriteUpdaterVersionSuffix(UpdaterPara ¶)
{
char osVersionSuffix[MAX_PARA_SIZE + 1] {};
int res = GetParameter("const.settings.os_version_suffix", "", osVersionSuffix, MAX_PARA_SIZE);
if (res < 0) {
cout << "Get const.settings.os_version_suffix parameter failed" << endl;
return -1;
}
(void)memset_s(para.osVersionSuffix, MAX_PARA_SIZE, 0, MAX_PARA_SIZE);
res = memcpy_s(para.osVersionSuffix, MAX_PARA_SIZE, osVersionSuffix, sizeof(osVersionSuffix) - 1);
para.osVersionSuffix[MAX_PARA_SIZE - 1] = '\0';
if (res != 0) {
cout << "memcpy_s para.osVersionSuffix failed" << endl;
return -1;
}
return 0;
}
static int ExceptionUpdater(int argc, char **argv, UpdateMessage &boot)
{
if (argc < BINARY_MAX_ARGS) {
cout << "Please input correct updater command!" << endl;
return -1;
}
if (argv[WRITE_SECOND_CMD] != nullptr) {
if (snprintf_s(boot.update, sizeof(boot.update), sizeof(boot.update) - 1, "--update_package=%s",
argv[WRITE_SECOND_CMD]) == -1) {
cout << "WriteUpdaterMessage snprintf_s failed!" << endl;
return -1;
}
}
return 0;
}
static int WriteUpdaterPara(int argc, UpdaterPara ¶)
{
if (argc != 2) {
cout << "please input correct updater command!" << endl;
return -1;
}
if (!ReadUpdaterParaMisc(para)) {
cout << "ReadUpdaterParaMisc failed" << endl;
return -1;
}
int resLanguage = WriteUpdaterLanguage(para);
int resVersionSuffix = WriteUpdaterVersionSuffix(para);
if (!WriteUpdaterParaMisc(para)) {
cout << "WriteUpdaterParaMisc failed!" << endl;
return -1;
}
if (resLanguage != 0 || resVersionSuffix != 0) {
cout << "WriteUpdaterLanguage or WriteUpdaterVersionSuffix fail" << endl;
return -1;
}
return 0;
}
static void HandleMiscInfo(int argc, char **argv)
{
if (!Utils::IsFileExist(HANDLE_MISC_LIB_PATH)) {
return;
}
auto handle = Utils::LoadLibrary(HANDLE_MISC_LIB_PATH);
if (handle == nullptr) {
cout << "load libupdater_handle_misc fail";
return;
}
auto getFunc = (void(*)(int, char **))Utils::GetFunction(handle, NOTIFY_MISC_INFO);
if (getFunc == nullptr) {
cout << "getFunc is nullptr";
Utils::CloseLibrary(handle);
return;
}
getFunc(argc, argv);
Utils::CloseLibrary(handle);
}
static int HandleSecureErase(struct UpdateMessage &boot)
{
if (strncpy_s(boot.update, sizeof(boot.update), SECURE_ERASE_MESSAGE_INFO,
sizeof(boot.update) - 1) != 0) {
cout << "strncpy_s failed!" << endl;
return -1;
}
return 0;
}
static int HandleCommand(int argc, char** argv, struct UpdateMessage& boot, struct UpdaterPara& para)
{
if (strcmp(argv[1], "bin") == 0) {
return ExceptionBin(argc, argv, boot);
} else if (strcmp(argv[1], "updater") == 0) {
return ExceptionUpdater(argc, argv, boot);
} else if (strcmp(argv[1], "user_factory_reset") == 0) {
if (strncpy_s(boot.update, sizeof(boot.update), "--user_wipe_data\n--reset_enter:211",
sizeof(boot.update) - 1) != 0) {
cout << "strncpy_s failed!" << endl;
return -1;
}
} else if (strcmp(argv[1], "boot_flash") == 0) {
if (strncpy_s(boot.update, sizeof(boot.update), "boot_flash", sizeof(boot.update) - 1) != 0) {
cout << "strncpy_s failed!" << endl;
return -1;
}
} else if (strcmp(argv[1], "clear") == 0) {
cout << "clear misc" << endl;
} else if (strcmp(argv[1], "sdcard_update") == 0) {
if (strncpy_s(boot.update, sizeof(boot.update), "--sdcard_update", sizeof(boot.update) - 1) != 0) {
cout << "strncpy_s failed!" << endl;
return -1;
}
} else if (strcmp(argv[1], "ota_intral_update") == 0) {
if (ExceptionUpdater(argc, argv, boot) == -1 ||
strcat_s(boot.update, sizeof(boot.update), "\n--ota_intral_update") != 0) {
return -1;
}
} else if (strcmp(argv[1], "subpkg_update") == 0) {
if (strncpy_s(boot.update, sizeof(boot.update), "--subpkg_update", sizeof(boot.update) - 1) != 0) {
cout << "strncpy_s failed!" << endl;
return -1;
}
} else if (strcmp(argv[1], "updater_para") == 0) {
return WriteUpdaterPara(argc, para) != 0 ? -1 : 0;
} else if (strcmp(argv[1], "notify_update") == 0) {
return 0;
} else if (strcmp(argv[1], "secure_erase") == 0) {
return HandleSecureErase(boot);
} else {
cout << "Please input correct command!" << endl;
return -1;
}
return 0;
}
int main(int argc, char **argv)
{
if (argc == 1) {
PrintPrompts();
return -1;
}
const std::string miscFile = "/dev/block/by-name/misc";
struct UpdateMessage boot {};
struct UpdaterPara para {};
int cmdResult = HandleCommand(argc, argv, boot, para);
if (cmdResult == -1) {
cout << "HandleCommand failed!" << endl;
return -1;
}
if (strcmp(argv[1], "notify_update") != 0) {
bool ret = WriteUpdaterMessage(miscFile, boot);
if (!ret) {
cout << "WriteUpdaterMessage failed!" << endl;
return -1;
}
}
HandleMiscInfo(argc, argv);
_exit(-1);
return 0;
}
