1/*
2 * Copyright (c) 2022 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
16#include <stdio.h>
17#include <string.h>
18#include <unistd.h>
19#include "init_reboot.h"
20#include "begetctl.h"
21
22#define REBOOT_CMD_NUMBER 2
23static int main_cmd(BShellHandle shell, int argc, char* argv[])
24{
25    if (argc > REBOOT_CMD_NUMBER) {
26        BShellCmdHelp(shell, argc, argv);
27        return 0;
28    }
29    int ret;
30    if (argc == REBOOT_CMD_NUMBER) {
31        ret = DoReboot(argv[1]);
32    } else {
33        ret = DoReboot(NULL);
34    }
35    if (ret != 0) {
36        printf("[reboot command] DoReboot Api return error\n");
37    } else {
38        printf("[reboot command] DoReboot Api return ok\n");
39    }
40#ifndef STARTUP_INIT_TEST
41    while (1) {
42        pause();
43    }
44#endif
45    return 0;
46}
47
48MODULE_CONSTRUCTOR(void)
49{
50    CmdInfo infos[] = {
51        {"reboot", main_cmd, "reboot system", "reboot", ""},
52        {"reboot", main_cmd, "shutdown system", "reboot shutdown[:options]", ""},
53        {"reboot", main_cmd, "suspend system", "reboot suspend", ""},
54        {"reboot", main_cmd, "reboot and boot into updater", "reboot updater", ""},
55        {"reboot", main_cmd, "reboot and boot into updater", "reboot updater[:options]", ""},
56        {"reboot", main_cmd, "reboot and boot into flashd", "reboot flashd", ""},
57        {"reboot", main_cmd, "reboot and boot into flashd", "reboot flashd[:options]", ""},
58        {"reboot", main_cmd, "reboot and boot into charge", "reboot charge", ""},
59    };
60    for (size_t i = sizeof(infos) / sizeof(infos[0]); i > 0; i--) {
61        BShellEnvRegisterCmd(GetShellHandle(), &infos[i - 1]);
62    }
63}
64