1 /*
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15 #include <iostream>
16 #include "fs_manager/mount.h"
17 #include "misc_info/misc_info.h"
18 #include "securec.h"
19 #include "updater/updater_const.h"
20 #include "parameter.h"
21
22 using namespace std;
23 using namespace Updater;
24
PrintPrompts()25 static void PrintPrompts()
26 {
27 cout << "Please input correct command, examples :" << endl;
28 cout << "updater : write_updater updater /data/updater/updater.zip" << endl;
29 cout << "factory_reset : write_updater user_factory_reset" << endl;
30 cout << "sdcard_update : write_updater sdcard_update" << endl;
31 cout << "clear command : write_updater clear" << endl;
32 cout << "updater_para : write_updater updater_para" << endl;
33 }
34
ExceptionUpdater(int argc,char ** argv,UpdateMessage & boot)35 static int ExceptionUpdater(int argc, char **argv, UpdateMessage &boot)
36 {
37 if (argc < BINARY_MAX_ARGS) {
38 cout << "Please input correct updater command!" << endl;
39 return -1;
40 }
41 if (argv[WRITE_SECOND_CMD] != nullptr) {
42 if (snprintf_s(boot.update, sizeof(boot.update), sizeof(boot.update) - 1, "--update_package=%s",
43 argv[WRITE_SECOND_CMD]) == -1) {
44 cout << "WriteUpdaterMessage snprintf_s failed!" << endl;
45 return -1;
46 }
47 }
48 return 0;
49 }
50
WriteUpdaterPara(int argc,UpdaterPara & para)51 static int WriteUpdaterPara(int argc, UpdaterPara ¶)
52 {
53 if (argc != 2) { // 2 : Not 2 is an invalid input
54 cout << "please input correct updater command!" << endl;
55 return -1;
56 }
57 int res = GetParameter("persist.global.language", "", para.language, MAX_PARA_SIZE);
58 if (res <= 0) {
59 cout << "Get language parameter failed" << endl;
60 return -1;
61 }
62 if (!WriteUpdaterParaMisc(para)) {
63 cout << "WriteUpdaterParaMisc failed!" << endl;
64 return -1;
65 }
66 return 0;
67 }
68
main(int argc,char ** argv)69 int main(int argc, char **argv)
70 {
71 if (argc == 1) {
72 PrintPrompts();
73 return -1;
74 }
75
76 const std::string miscFile = "/dev/block/by-name/misc";
77 struct UpdateMessage boot {};
78 struct UpdaterPara para {};
79 if (strcmp(argv[1], "updater") == 0) {
80 if (ExceptionUpdater(argc, argv, boot) == -1) {
81 return -1;
82 }
83 } else if (strcmp(argv[1], "user_factory_reset") == 0) {
84 if (strncpy_s(boot.update, sizeof(boot.update), "--user_wipe_data", sizeof(boot.update) - 1) != 0) {
85 cout << "strncpy_s failed!" << endl;
86 return -1;
87 }
88 } else if (strcmp(argv[1], "boot_flash") == 0) {
89 if (strncpy_s(boot.update, sizeof(boot.update), "boot_flash", sizeof(boot.update) - 1) != 0) {
90 cout << "strncpy_s failed!" << endl;
91 return -1;
92 }
93 } else if (strcmp(argv[1], "clear") == 0) {
94 cout << "clear misc" << endl;
95 } else if (strcmp(argv[1], "sdcard_update") == 0) {
96 if (strncpy_s(boot.update, sizeof(boot.update), "--sdcard_update", sizeof(boot.update) - 1) != 0) {
97 cout << "strncpy_s failed!" << endl;
98 return -1;
99 }
100 } else if (strcmp(argv[1], "updater_para") == 0) {
101 if (WriteUpdaterPara(argc, para) != 0) {
102 return -1;
103 }
104 return 0;
105 } else {
106 cout << "Please input correct command!" << endl;
107 return -1;
108 }
109 bool ret = WriteUpdaterMessage(miscFile, boot);
110 if (!ret) {
111 cout << "WriteUpdaterMessage failed!" << endl;
112 return -1;
113 }
114 return 0;
115 }
116