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