1 /* 2 * (C) Copyright 2008 3 * Gary Jennejohn, DENX Software Engineering GmbH, garyj@denx.de. 4 * 5 * SPDX-License-Identifier: GPL-2.0+ 6 */ 7 8 #include <common.h> 9 #include <console.h> 10 #include <serial.h> 11 #include <malloc.h> 12 13 #if CONFIG_IS_ENABLED(CONSOLE_MUX) 14 void iomux_printdevs(const int console) 15 { 16 int i; 17 struct stdio_dev *dev; 18 19 for (i = 0; i < cd_count[console]; i++) { 20 dev = console_devices[console][i]; 21 printf("%s ", dev->name); 22 } 23 printf("\n"); 24 } 25 26 /* This tries to preserve the old list if an error occurs. */ 27 int iomux_doenv(const int console, const char *arg) 28 { 29 char *console_args, *temp, **start; 30 int i, j, k, io_flag, cs_idx, repeat; 31 struct stdio_dev *dev; 32 struct stdio_dev **cons_set; 33 34 console_args = strdup(arg); 35 if (console_args == NULL) 36 return 1; 37 /* 38 * Check whether a comma separated list of devices was 39 * entered and count how many devices were entered. 40 * The array start[] has pointers to the beginning of 41 * each device name (up to MAX_CONSARGS devices). 42 * 43 * Have to do this twice - once to count the number of 44 * commas and then again to populate start. 45 */ 46 i = 0; 47 temp = console_args; 48 for (;;) { 49 temp = strchr(temp, ','); 50 if (temp != NULL) { 51 i++; 52 temp++; 53 continue; 54 } 55 /* There's always one entry more than the number of commas. */ 56 i++; 57 break; 58 } 59 start = (char **)malloc(i * sizeof(char *)); 60 if (start == NULL) { 61 free(console_args); 62 return 1; 63 } 64 i = 0; 65 start[0] = console_args; 66 for (;;) { 67 temp = strchr(start[i++], ','); 68 if (temp == NULL) 69 break; 70 *temp = '\0'; 71 start[i] = temp + 1; 72 } 73 cons_set = (struct stdio_dev **)calloc(i, sizeof(struct stdio_dev *)); 74 if (cons_set == NULL) { 75 free(start); 76 free(console_args); 77 return 1; 78 } 79 80 switch (console) { 81 case stdin: 82 io_flag = DEV_FLAGS_INPUT; 83 break; 84 case stdout: 85 case stderr: 86 io_flag = DEV_FLAGS_OUTPUT; 87 break; 88 default: 89 free(start); 90 free(console_args); 91 free(cons_set); 92 return 1; 93 } 94 95 cs_idx = 0; 96 for (j = 0; j < i; j++) { 97 /* 98 * Check whether the device exists and is valid. 99 * console_assign() also calls search_device(), 100 * but I need the pointer to the device. 101 */ 102 dev = search_device(io_flag, start[j]); 103 if (dev == NULL) 104 continue; 105 /* 106 * Prevent multiple entries for a device. 107 */ 108 repeat = 0; 109 for (k = 0; k < cs_idx; k++) { 110 if (dev == cons_set[k]) { 111 repeat++; 112 break; 113 } 114 } 115 if (repeat) 116 continue; 117 /* 118 * Try assigning the specified device. 119 * This could screw up the console settings for apps. 120 */ 121 if (console_assign(console, start[j]) < 0) 122 continue; 123 cons_set[cs_idx++] = dev; 124 } 125 free(console_args); 126 free(start); 127 /* failed to set any console */ 128 if (cs_idx == 0) { 129 free(cons_set); 130 return 1; 131 } else { 132 /* Works even if console_devices[console] is NULL. */ 133 console_devices[console] = 134 (struct stdio_dev **)realloc(console_devices[console], 135 cs_idx * sizeof(struct stdio_dev *)); 136 if (console_devices[console] == NULL) { 137 free(cons_set); 138 return 1; 139 } 140 memcpy(console_devices[console], cons_set, cs_idx * 141 sizeof(struct stdio_dev *)); 142 143 cd_count[console] = cs_idx; 144 } 145 free(cons_set); 146 return 0; 147 } 148 #endif /* CONSOLE_MUX */ 149