hanliang.xiong | cd3ad4c | 2024-12-23 11:07:00 +0800 | [diff] [blame] | 1 | // SPDX-License-Identifier: Apache2.0 |
| 2 | /* |
| 3 | * Copyright (C) 2024 Amlogic Inc. |
| 4 | */ |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 5 | #include <stdio.h> |
| 6 | #include <stdlib.h> |
| 7 | |
| 8 | #include "bootloader_message.h" |
| 9 | |
Bing Jiang | a8fad17 | 2022-10-19 13:51:09 +0800 | [diff] [blame] | 10 | void usage(char *program_name) { |
| 11 | printf("ab system select!!!\n"); |
| 12 | printf("%s get ---get current slot\n", program_name); |
| 13 | printf("%s set ---set slot\n", program_name); |
| 14 | printf("%s success_boot ---set boot success\n", program_name); |
| 15 | printf("example: \n"); |
| 16 | printf("%s get\n", program_name); |
| 17 | printf("%s set _a\n", program_name); |
| 18 | printf("%s set _b\n", program_name); |
| 19 | printf("%s success_boot\n", program_name); |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 20 | } |
| 21 | |
| 22 | int main(int argc, char **argv) { |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 23 | int ret = 0; |
Bing Jiang | a8fad17 | 2022-10-19 13:51:09 +0800 | [diff] [blame] | 24 | |
| 25 | if ((argc != 2) && (argc != 3)) { |
| 26 | usage(argv[0]); |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 27 | return -1; |
| 28 | } |
Bing Jiang | a8fad17 | 2022-10-19 13:51:09 +0800 | [diff] [blame] | 29 | |
| 30 | if (argc == 3) { |
| 31 | int slot; |
| 32 | |
| 33 | if (!strcmp(argv[2],"_a")) { |
| 34 | slot = 0; |
| 35 | } else if (!strcmp(argv[2],"_b")) { |
| 36 | slot = 1; |
| 37 | } else { |
| 38 | printf("invalid slot: %s\n", argv[2]); |
| 39 | usage(argv[0]); |
| 40 | return -1; |
| 41 | } |
| 42 | |
| 43 | if (!strcmp(argv[1], "set")) { |
| 44 | ret = set_active_slot(slot); |
| 45 | if (ret != 0) { |
| 46 | printf("set active slot: %s failed!\n", argv[2]); |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 47 | return -1; |
| 48 | } |
Bing Jiang | a8fad17 | 2022-10-19 13:51:09 +0800 | [diff] [blame] | 49 | } else { |
| 50 | usage(argv[0]); |
| 51 | return -1; |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 52 | } |
| 53 | } |
Bing Jiang | a8fad17 | 2022-10-19 13:51:09 +0800 | [diff] [blame] | 54 | |
| 55 | if (!strcmp(argv[1], "get")) { |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 56 | int active = 0; |
| 57 | get_active_slot_from_misc(&active); |
| 58 | printf("active is %d\n", active); |
Bing Jiang | a8fad17 | 2022-10-19 13:51:09 +0800 | [diff] [blame] | 59 | } else if (!strcmp(argv[1], "success_boot")) { |
| 60 | ret = set_successful_boot(); |
| 61 | if (ret != 0) { |
| 62 | printf("set successful boot slot failed!\n"); |
| 63 | return -1; |
| 64 | } |
Zhizhong Zhang | c44e077 | 2021-06-10 21:25:57 +0800 | [diff] [blame] | 65 | } |
| 66 | |
| 67 | return 0; |
| 68 | } |