107 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			C
		
	
	
	
			
		
		
	
	
			107 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			C
		
	
	
	
| /*
 | |
|  * Copyright (C) 2012 Boundary Devices Inc.
 | |
|  *
 | |
|  * SPDX-License-Identifier:	GPL-2.0+
 | |
|  */
 | |
| #include <common.h>
 | |
| #include <linux/errno.h>
 | |
| #include <asm/io.h>
 | |
| #include <asm/mach-imx/boot_mode.h>
 | |
| #include <malloc.h>
 | |
| #include <command.h>
 | |
| 
 | |
| static const struct boot_mode *modes[2];
 | |
| 
 | |
| static const struct boot_mode *search_modes(char *arg)
 | |
| {
 | |
| 	int i;
 | |
| 
 | |
| 	for (i = 0; i < ARRAY_SIZE(modes); i++) {
 | |
| 		const struct boot_mode *p = modes[i];
 | |
| 		if (p) {
 | |
| 			while (p->name) {
 | |
| 				if (!strcmp(p->name, arg))
 | |
| 					return p;
 | |
| 				p++;
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	return NULL;
 | |
| }
 | |
| 
 | |
| static int create_usage(char *dest)
 | |
| {
 | |
| 	int i;
 | |
| 	int size = 0;
 | |
| 
 | |
| 	for (i = 0; i < ARRAY_SIZE(modes); i++) {
 | |
| 		const struct boot_mode *p = modes[i];
 | |
| 		if (p) {
 | |
| 			while (p->name) {
 | |
| 				int len = strlen(p->name);
 | |
| 				if (dest) {
 | |
| 					memcpy(dest, p->name, len);
 | |
| 					dest += len;
 | |
| 					*dest++ = '|';
 | |
| 				}
 | |
| 				size += len + 1;
 | |
| 				p++;
 | |
| 			}
 | |
| 		}
 | |
| 	}
 | |
| 	if (dest)
 | |
| 		memcpy(dest - 1, " [noreset]", 11);	/* include trailing 0 */
 | |
| 	size += 10;
 | |
| 	return size;
 | |
| }
 | |
| 
 | |
| static int do_boot_mode(cmd_tbl_t *cmdtp, int flag, int argc,
 | |
| 		char * const argv[])
 | |
| {
 | |
| 	const struct boot_mode *p;
 | |
| 	int reset_requested = 1;
 | |
| 
 | |
| 	if (argc < 2)
 | |
| 		return CMD_RET_USAGE;
 | |
| 	p = search_modes(argv[1]);
 | |
| 	if (!p)
 | |
| 		return CMD_RET_USAGE;
 | |
| 	if (argc == 3) {
 | |
| 		if (strcmp(argv[2], "noreset"))
 | |
| 			return CMD_RET_USAGE;
 | |
| 		reset_requested = 0;
 | |
| 	}
 | |
| 
 | |
| 	boot_mode_apply(p->cfg_val);
 | |
| 	if (reset_requested && p->cfg_val)
 | |
| 		do_reset(NULL, 0, 0, NULL);
 | |
| 	return 0;
 | |
| }
 | |
| 
 | |
| U_BOOT_CMD(
 | |
| 	bmode, 3, 0, do_boot_mode,
 | |
| 	NULL,
 | |
| 	"");
 | |
| 
 | |
| void add_board_boot_modes(const struct boot_mode *p)
 | |
| {
 | |
| 	int size;
 | |
| 	char *dest;
 | |
| 
 | |
| 	cmd_tbl_t *entry = ll_entry_get(cmd_tbl_t, bmode, cmd);
 | |
| 
 | |
| 	if (entry->usage) {
 | |
| 		free(entry->usage);
 | |
| 		entry->usage = NULL;
 | |
| 	}
 | |
| 
 | |
| 	modes[0] = p;
 | |
| 	modes[1] = soc_boot_modes;
 | |
| 	size = create_usage(NULL);
 | |
| 	dest = malloc(size);
 | |
| 	if (dest) {
 | |
| 		create_usage(dest);
 | |
| 		entry->usage = dest;
 | |
| 	}
 | |
| }
 |