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 <signal.h> 16#include <stdio.h> 17#include <stdlib.h> 18 19#include "begetctl.h" 20#include "shell.h" 21#include "shell_utils.h" 22 23static BShellHandle g_handle = NULL; 24BShellHandle GetShellHandle(void) 25{ 26 if (g_handle == NULL) { 27 BShellInfo info = {PARAM_SHELL_DEFAULT_PROMPT, NULL}; 28 BShellEnvInit(&g_handle, &info); 29 } 30 return g_handle; 31} 32 33static void signalHandler(int signal) 34{ 35 demoExit(); 36 exit(0); 37} 38 39int main(int argc, char *argv[]) 40{ 41 (void)signal(SIGINT, signalHandler); 42 const char *last = strrchr(argv[0], '/'); 43 // Get the first ending command name 44 if (last != NULL) { 45 last = last + 1; 46 } else { 47 last = argv[0]; 48 } 49 50 // If it is begetctl with subcommand name, try to do subcommand first 51 int number = argc; 52 char **args = argv; 53 if ((argc > 1) && (strcmp(last, "begetctl") == 0)) { 54 number = argc - 1; 55 args = argv + 1; 56 } 57 if (number >= 1 && strcmp(args[0], "devctl") == 0) { 58 if (memcpy_s(args[0], strlen(args[0]), "reboot", strlen("reboot")) != 0) { 59 printf("Failed to copy\n"); 60 } 61 } 62 BShellHandle handle = GetShellHandle(); 63 if (handle == NULL) { 64 printf("Failed to get shell handle \n"); 65 return 0; 66 } 67 68 BShellParamCmdRegister(handle, 0); 69#ifdef INIT_TEST 70 BShellCmdRegister(handle, 0); 71#endif 72 BShellEnvDirectExecute(handle, number, args); 73 demoExit(); 74 return 0; 75} 76