Merge tag 'dm-pull-6feb20' of https://gitlab.denx.de/u-boot/custodians/u-boot-dm
[pandora-u-boot.git] / common / console.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2000
4  * Paolo Scaffardi, AIRVENT SAM s.p.a - RIMINI(ITALY), arsenio@tin.it
5  */
6
7 #include <common.h>
8 #include <console.h>
9 #include <debug_uart.h>
10 #include <dm.h>
11 #include <env.h>
12 #include <stdarg.h>
13 #include <iomux.h>
14 #include <malloc.h>
15 #include <mapmem.h>
16 #include <os.h>
17 #include <serial.h>
18 #include <stdio_dev.h>
19 #include <exports.h>
20 #include <env_internal.h>
21 #include <watchdog.h>
22
23 DECLARE_GLOBAL_DATA_PTR;
24
25 static int on_console(const char *name, const char *value, enum env_op op,
26         int flags)
27 {
28         int console = -1;
29
30         /* Check for console redirection */
31         if (strcmp(name, "stdin") == 0)
32                 console = stdin;
33         else if (strcmp(name, "stdout") == 0)
34                 console = stdout;
35         else if (strcmp(name, "stderr") == 0)
36                 console = stderr;
37
38         /* if not actually setting a console variable, we don't care */
39         if (console == -1 || (gd->flags & GD_FLG_DEVINIT) == 0)
40                 return 0;
41
42         switch (op) {
43         case env_op_create:
44         case env_op_overwrite:
45
46 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
47                 if (iomux_doenv(console, value))
48                         return 1;
49 #else
50                 /* Try assigning specified device */
51                 if (console_assign(console, value) < 0)
52                         return 1;
53 #endif
54                 return 0;
55
56         case env_op_delete:
57                 if ((flags & H_FORCE) == 0)
58                         printf("Can't delete \"%s\"\n", name);
59                 return 1;
60
61         default:
62                 return 0;
63         }
64 }
65 U_BOOT_ENV_CALLBACK(console, on_console);
66
67 #ifdef CONFIG_SILENT_CONSOLE
68 static int on_silent(const char *name, const char *value, enum env_op op,
69         int flags)
70 {
71 #if !CONFIG_IS_ENABLED(SILENT_CONSOLE_UPDATE_ON_SET)
72         if (flags & H_INTERACTIVE)
73                 return 0;
74 #endif
75 #if !CONFIG_IS_ENABLED(SILENT_CONSOLE_UPDATE_ON_RELOC)
76         if ((flags & H_INTERACTIVE) == 0)
77                 return 0;
78 #endif
79
80         if (value != NULL)
81                 gd->flags |= GD_FLG_SILENT;
82         else
83                 gd->flags &= ~GD_FLG_SILENT;
84
85         return 0;
86 }
87 U_BOOT_ENV_CALLBACK(silent, on_silent);
88 #endif
89
90 #if CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV)
91 /*
92  * if overwrite_console returns 1, the stdin, stderr and stdout
93  * are switched to the serial port, else the settings in the
94  * environment are used
95  */
96 #ifdef CONFIG_SYS_CONSOLE_OVERWRITE_ROUTINE
97 extern int overwrite_console(void);
98 #define OVERWRITE_CONSOLE overwrite_console()
99 #else
100 #define OVERWRITE_CONSOLE 0
101 #endif /* CONFIG_SYS_CONSOLE_OVERWRITE_ROUTINE */
102
103 #endif /* CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV) */
104
105 static int console_setfile(int file, struct stdio_dev * dev)
106 {
107         int error = 0;
108
109         if (dev == NULL)
110                 return -1;
111
112         switch (file) {
113         case stdin:
114         case stdout:
115         case stderr:
116                 /* Start new device */
117                 if (dev->start) {
118                         error = dev->start(dev);
119                         /* If it's not started dont use it */
120                         if (error < 0)
121                                 break;
122                 }
123
124                 /* Assign the new device (leaving the existing one started) */
125                 stdio_devices[file] = dev;
126
127                 /*
128                  * Update monitor functions
129                  * (to use the console stuff by other applications)
130                  */
131                 switch (file) {
132                 case stdin:
133                         gd->jt->getc = getc;
134                         gd->jt->tstc = tstc;
135                         break;
136                 case stdout:
137                         gd->jt->putc  = putc;
138                         gd->jt->puts  = puts;
139                         gd->jt->printf = printf;
140                         break;
141                 }
142                 break;
143
144         default:                /* Invalid file ID */
145                 error = -1;
146         }
147         return error;
148 }
149
150 /**
151  * console_dev_is_serial() - Check if a stdio device is a serial device
152  *
153  * @sdev: Device to check
154  * @return true if this device is in the serial uclass (or for pre-driver-model,
155  * whether it is called "serial".
156  */
157 static bool console_dev_is_serial(struct stdio_dev *sdev)
158 {
159         bool is_serial;
160
161 #ifdef CONFIG_DM_SERIAL
162         if (sdev->flags & DEV_FLAGS_DM) {
163                 struct udevice *dev = sdev->priv;
164
165                 is_serial = device_get_uclass_id(dev) == UCLASS_SERIAL;
166         } else
167 #endif
168         is_serial = !strcmp(sdev->name, "serial");
169
170         return is_serial;
171 }
172
173 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
174 /** Console I/O multiplexing *******************************************/
175
176 static struct stdio_dev *tstcdev;
177 struct stdio_dev **console_devices[MAX_FILES];
178 int cd_count[MAX_FILES];
179
180 /*
181  * This depends on tstc() always being called before getc().
182  * This is guaranteed to be true because this routine is called
183  * only from fgetc() which assures it.
184  * No attempt is made to demultiplex multiple input sources.
185  */
186 static int console_getc(int file)
187 {
188         unsigned char ret;
189
190         /* This is never called with testcdev == NULL */
191         ret = tstcdev->getc(tstcdev);
192         tstcdev = NULL;
193         return ret;
194 }
195
196 static int console_tstc(int file)
197 {
198         int i, ret;
199         struct stdio_dev *dev;
200         int prev;
201
202         prev = disable_ctrlc(1);
203         for (i = 0; i < cd_count[file]; i++) {
204                 dev = console_devices[file][i];
205                 if (dev->tstc != NULL) {
206                         ret = dev->tstc(dev);
207                         if (ret > 0) {
208                                 tstcdev = dev;
209                                 disable_ctrlc(prev);
210                                 return ret;
211                         }
212                 }
213         }
214         disable_ctrlc(prev);
215
216         return 0;
217 }
218
219 static void console_putc(int file, const char c)
220 {
221         int i;
222         struct stdio_dev *dev;
223
224         for (i = 0; i < cd_count[file]; i++) {
225                 dev = console_devices[file][i];
226                 if (dev->putc != NULL)
227                         dev->putc(dev, c);
228         }
229 }
230
231 static void console_puts_noserial(int file, const char *s)
232 {
233         int i;
234         struct stdio_dev *dev;
235
236         for (i = 0; i < cd_count[file]; i++) {
237                 dev = console_devices[file][i];
238                 if (dev->puts != NULL && !console_dev_is_serial(dev))
239                         dev->puts(dev, s);
240         }
241 }
242
243 static void console_puts(int file, const char *s)
244 {
245         int i;
246         struct stdio_dev *dev;
247
248         for (i = 0; i < cd_count[file]; i++) {
249                 dev = console_devices[file][i];
250                 if (dev->puts != NULL)
251                         dev->puts(dev, s);
252         }
253 }
254
255 #if CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV)
256 static inline void console_doenv(int file, struct stdio_dev *dev)
257 {
258         iomux_doenv(file, dev->name);
259 }
260 #endif
261 #else
262 static inline int console_getc(int file)
263 {
264         return stdio_devices[file]->getc(stdio_devices[file]);
265 }
266
267 static inline int console_tstc(int file)
268 {
269         return stdio_devices[file]->tstc(stdio_devices[file]);
270 }
271
272 static inline void console_putc(int file, const char c)
273 {
274         stdio_devices[file]->putc(stdio_devices[file], c);
275 }
276
277 static inline void console_puts_noserial(int file, const char *s)
278 {
279         if (!console_dev_is_serial(stdio_devices[file]))
280                 stdio_devices[file]->puts(stdio_devices[file], s);
281 }
282
283 static inline void console_puts(int file, const char *s)
284 {
285         stdio_devices[file]->puts(stdio_devices[file], s);
286 }
287
288 #if CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV)
289 static inline void console_doenv(int file, struct stdio_dev *dev)
290 {
291         console_setfile(file, dev);
292 }
293 #endif
294 #endif /* CONIFIG_IS_ENABLED(CONSOLE_MUX) */
295
296 /** U-Boot INITIAL CONSOLE-NOT COMPATIBLE FUNCTIONS *************************/
297
298 int serial_printf(const char *fmt, ...)
299 {
300         va_list args;
301         uint i;
302         char printbuffer[CONFIG_SYS_PBSIZE];
303
304         va_start(args, fmt);
305
306         /* For this to work, printbuffer must be larger than
307          * anything we ever want to print.
308          */
309         i = vscnprintf(printbuffer, sizeof(printbuffer), fmt, args);
310         va_end(args);
311
312         serial_puts(printbuffer);
313         return i;
314 }
315
316 int fgetc(int file)
317 {
318         if (file < MAX_FILES) {
319                 /*
320                  * Effectively poll for input wherever it may be available.
321                  */
322                 for (;;) {
323                         WATCHDOG_RESET();
324 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
325                         /*
326                          * Upper layer may have already called tstc() so
327                          * check for that first.
328                          */
329                         if (tstcdev != NULL)
330                                 return console_getc(file);
331                         console_tstc(file);
332 #else
333                         if (console_tstc(file))
334                                 return console_getc(file);
335 #endif
336 #ifdef CONFIG_WATCHDOG
337                         /*
338                          * If the watchdog must be rate-limited then it should
339                          * already be handled in board-specific code.
340                          */
341                          udelay(1);
342 #endif
343                 }
344         }
345
346         return -1;
347 }
348
349 int ftstc(int file)
350 {
351         if (file < MAX_FILES)
352                 return console_tstc(file);
353
354         return -1;
355 }
356
357 void fputc(int file, const char c)
358 {
359         if (file < MAX_FILES)
360                 console_putc(file, c);
361 }
362
363 void fputs(int file, const char *s)
364 {
365         if (file < MAX_FILES)
366                 console_puts(file, s);
367 }
368
369 int fprintf(int file, const char *fmt, ...)
370 {
371         va_list args;
372         uint i;
373         char printbuffer[CONFIG_SYS_PBSIZE];
374
375         va_start(args, fmt);
376
377         /* For this to work, printbuffer must be larger than
378          * anything we ever want to print.
379          */
380         i = vscnprintf(printbuffer, sizeof(printbuffer), fmt, args);
381         va_end(args);
382
383         /* Send to desired file */
384         fputs(file, printbuffer);
385         return i;
386 }
387
388 /** U-Boot INITIAL CONSOLE-COMPATIBLE FUNCTION *****************************/
389
390 int getc(void)
391 {
392 #ifdef CONFIG_DISABLE_CONSOLE
393         if (gd->flags & GD_FLG_DISABLE_CONSOLE)
394                 return 0;
395 #endif
396
397         if (!gd->have_console)
398                 return 0;
399
400 #ifdef CONFIG_CONSOLE_RECORD
401         if (gd->console_in.start) {
402                 int ch;
403
404                 ch = membuff_getbyte(&gd->console_in);
405                 if (ch != -1)
406                         return 1;
407         }
408 #endif
409         if (gd->flags & GD_FLG_DEVINIT) {
410                 /* Get from the standard input */
411                 return fgetc(stdin);
412         }
413
414         /* Send directly to the handler */
415         return serial_getc();
416 }
417
418 int tstc(void)
419 {
420 #ifdef CONFIG_DISABLE_CONSOLE
421         if (gd->flags & GD_FLG_DISABLE_CONSOLE)
422                 return 0;
423 #endif
424
425         if (!gd->have_console)
426                 return 0;
427 #ifdef CONFIG_CONSOLE_RECORD
428         if (gd->console_in.start) {
429                 if (membuff_peekbyte(&gd->console_in) != -1)
430                         return 1;
431         }
432 #endif
433         if (gd->flags & GD_FLG_DEVINIT) {
434                 /* Test the standard input */
435                 return ftstc(stdin);
436         }
437
438         /* Send directly to the handler */
439         return serial_tstc();
440 }
441
442 #define PRE_CONSOLE_FLUSHPOINT1_SERIAL                  0
443 #define PRE_CONSOLE_FLUSHPOINT2_EVERYTHING_BUT_SERIAL   1
444
445 #if CONFIG_IS_ENABLED(PRE_CONSOLE_BUFFER)
446 #define CIRC_BUF_IDX(idx) ((idx) % (unsigned long)CONFIG_PRE_CON_BUF_SZ)
447
448 static void pre_console_putc(const char c)
449 {
450         char *buffer;
451
452         buffer = map_sysmem(CONFIG_PRE_CON_BUF_ADDR, CONFIG_PRE_CON_BUF_SZ);
453
454         buffer[CIRC_BUF_IDX(gd->precon_buf_idx++)] = c;
455
456         unmap_sysmem(buffer);
457 }
458
459 static void pre_console_puts(const char *s)
460 {
461         while (*s)
462                 pre_console_putc(*s++);
463 }
464
465 static void print_pre_console_buffer(int flushpoint)
466 {
467         unsigned long in = 0, out = 0;
468         char buf_out[CONFIG_PRE_CON_BUF_SZ + 1];
469         char *buf_in;
470
471 #ifdef CONFIG_SILENT_CONSOLE
472         if (gd->flags & GD_FLG_SILENT)
473                 return;
474 #endif
475
476         buf_in = map_sysmem(CONFIG_PRE_CON_BUF_ADDR, CONFIG_PRE_CON_BUF_SZ);
477         if (gd->precon_buf_idx > CONFIG_PRE_CON_BUF_SZ)
478                 in = gd->precon_buf_idx - CONFIG_PRE_CON_BUF_SZ;
479
480         while (in < gd->precon_buf_idx)
481                 buf_out[out++] = buf_in[CIRC_BUF_IDX(in++)];
482         unmap_sysmem(buf_in);
483
484         buf_out[out] = 0;
485
486         switch (flushpoint) {
487         case PRE_CONSOLE_FLUSHPOINT1_SERIAL:
488                 puts(buf_out);
489                 break;
490         case PRE_CONSOLE_FLUSHPOINT2_EVERYTHING_BUT_SERIAL:
491                 console_puts_noserial(stdout, buf_out);
492                 break;
493         }
494 }
495 #else
496 static inline void pre_console_putc(const char c) {}
497 static inline void pre_console_puts(const char *s) {}
498 static inline void print_pre_console_buffer(int flushpoint) {}
499 #endif
500
501 void putc(const char c)
502 {
503 #ifdef CONFIG_SANDBOX
504         /* sandbox can send characters to stdout before it has a console */
505         if (!gd || !(gd->flags & GD_FLG_SERIAL_READY)) {
506                 os_putc(c);
507                 return;
508         }
509 #endif
510 #ifdef CONFIG_DEBUG_UART
511         /* if we don't have a console yet, use the debug UART */
512         if (!gd || !(gd->flags & GD_FLG_SERIAL_READY)) {
513                 printch(c);
514                 return;
515         }
516 #endif
517         if (!gd)
518                 return;
519 #ifdef CONFIG_CONSOLE_RECORD
520         if ((gd->flags & GD_FLG_RECORD) && gd->console_out.start)
521                 membuff_putbyte(&gd->console_out, c);
522 #endif
523 #ifdef CONFIG_SILENT_CONSOLE
524         if (gd->flags & GD_FLG_SILENT) {
525                 if (!(gd->flags & GD_FLG_DEVINIT))
526                         pre_console_putc(c);
527                 return;
528         }
529 #endif
530
531 #ifdef CONFIG_DISABLE_CONSOLE
532         if (gd->flags & GD_FLG_DISABLE_CONSOLE)
533                 return;
534 #endif
535
536         if (!gd->have_console)
537                 return pre_console_putc(c);
538
539         if (gd->flags & GD_FLG_DEVINIT) {
540                 /* Send to the standard output */
541                 fputc(stdout, c);
542         } else {
543                 /* Send directly to the handler */
544                 pre_console_putc(c);
545                 serial_putc(c);
546         }
547 }
548
549 void puts(const char *s)
550 {
551 #ifdef CONFIG_SANDBOX
552         /* sandbox can send characters to stdout before it has a console */
553         if (!gd || !(gd->flags & GD_FLG_SERIAL_READY)) {
554                 os_puts(s);
555                 return;
556         }
557 #endif
558 #ifdef CONFIG_DEBUG_UART
559         if (!gd || !(gd->flags & GD_FLG_SERIAL_READY)) {
560                 while (*s) {
561                         int ch = *s++;
562
563                         printch(ch);
564                 }
565                 return;
566         }
567 #endif
568         if (!gd)
569                 return;
570 #ifdef CONFIG_CONSOLE_RECORD
571         if ((gd->flags & GD_FLG_RECORD) && gd->console_out.start)
572                 membuff_put(&gd->console_out, s, strlen(s));
573 #endif
574 #ifdef CONFIG_SILENT_CONSOLE
575         if (gd->flags & GD_FLG_SILENT) {
576                 if (!(gd->flags & GD_FLG_DEVINIT))
577                         pre_console_puts(s);
578                 return;
579         }
580 #endif
581
582 #ifdef CONFIG_DISABLE_CONSOLE
583         if (gd->flags & GD_FLG_DISABLE_CONSOLE)
584                 return;
585 #endif
586
587         if (!gd->have_console)
588                 return pre_console_puts(s);
589
590         if (gd->flags & GD_FLG_DEVINIT) {
591                 /* Send to the standard output */
592                 fputs(stdout, s);
593         } else {
594                 /* Send directly to the handler */
595                 pre_console_puts(s);
596                 serial_puts(s);
597         }
598 }
599
600 #ifdef CONFIG_CONSOLE_RECORD
601 int console_record_init(void)
602 {
603         int ret;
604
605         ret = membuff_new(&gd->console_out, CONFIG_CONSOLE_RECORD_OUT_SIZE);
606         if (ret)
607                 return ret;
608         ret = membuff_new(&gd->console_in, CONFIG_CONSOLE_RECORD_IN_SIZE);
609
610         return ret;
611 }
612
613 void console_record_reset(void)
614 {
615         membuff_purge(&gd->console_out);
616         membuff_purge(&gd->console_in);
617 }
618
619 void console_record_reset_enable(void)
620 {
621         console_record_reset();
622         gd->flags |= GD_FLG_RECORD;
623 }
624
625 int console_record_readline(char *str, int maxlen)
626 {
627         return membuff_readline(&gd->console_out, str, maxlen, ' ');
628 }
629
630 int console_record_avail(void)
631 {
632         return membuff_avail(&gd->console_out);
633 }
634
635 #endif
636
637 /* test if ctrl-c was pressed */
638 static int ctrlc_disabled = 0;  /* see disable_ctrl() */
639 static int ctrlc_was_pressed = 0;
640 int ctrlc(void)
641 {
642         if (!ctrlc_disabled && gd->have_console) {
643                 if (tstc()) {
644                         switch (getc()) {
645                         case 0x03:              /* ^C - Control C */
646                                 ctrlc_was_pressed = 1;
647                                 return 1;
648                         default:
649                                 break;
650                         }
651                 }
652         }
653
654         return 0;
655 }
656 /* Reads user's confirmation.
657    Returns 1 if user's input is "y", "Y", "yes" or "YES"
658 */
659 int confirm_yesno(void)
660 {
661         int i;
662         char str_input[5];
663
664         /* Flush input */
665         while (tstc())
666                 getc();
667         i = 0;
668         while (i < sizeof(str_input)) {
669                 str_input[i] = getc();
670                 putc(str_input[i]);
671                 if (str_input[i] == '\r')
672                         break;
673                 i++;
674         }
675         putc('\n');
676         if (strncmp(str_input, "y\r", 2) == 0 ||
677             strncmp(str_input, "Y\r", 2) == 0 ||
678             strncmp(str_input, "yes\r", 4) == 0 ||
679             strncmp(str_input, "YES\r", 4) == 0)
680                 return 1;
681         return 0;
682 }
683 /* pass 1 to disable ctrlc() checking, 0 to enable.
684  * returns previous state
685  */
686 int disable_ctrlc(int disable)
687 {
688         int prev = ctrlc_disabled;      /* save previous state */
689
690         ctrlc_disabled = disable;
691         return prev;
692 }
693
694 int had_ctrlc (void)
695 {
696         return ctrlc_was_pressed;
697 }
698
699 void clear_ctrlc(void)
700 {
701         ctrlc_was_pressed = 0;
702 }
703
704 /** U-Boot INIT FUNCTIONS *************************************************/
705
706 struct stdio_dev *search_device(int flags, const char *name)
707 {
708         struct stdio_dev *dev;
709
710         dev = stdio_get_by_name(name);
711 #ifdef CONFIG_VIDCONSOLE_AS_LCD
712         if (!dev && !strcmp(name, "lcd"))
713                 dev = stdio_get_by_name("vidconsole");
714 #endif
715
716         if (dev && (dev->flags & flags))
717                 return dev;
718
719         return NULL;
720 }
721
722 int console_assign(int file, const char *devname)
723 {
724         int flag;
725         struct stdio_dev *dev;
726
727         /* Check for valid file */
728         switch (file) {
729         case stdin:
730                 flag = DEV_FLAGS_INPUT;
731                 break;
732         case stdout:
733         case stderr:
734                 flag = DEV_FLAGS_OUTPUT;
735                 break;
736         default:
737                 return -1;
738         }
739
740         /* Check for valid device name */
741
742         dev = search_device(flag, devname);
743
744         if (dev)
745                 return console_setfile(file, dev);
746
747         return -1;
748 }
749
750 /* return true if the 'silent' flag is removed */
751 static bool console_update_silent(void)
752 {
753 #ifdef CONFIG_SILENT_CONSOLE
754         if (env_get("silent")) {
755                 gd->flags |= GD_FLG_SILENT;
756         } else {
757                 unsigned long flags = gd->flags;
758
759                 gd->flags &= ~GD_FLG_SILENT;
760
761                 return !!(flags & GD_FLG_SILENT);
762         }
763 #endif
764
765         return false;
766 }
767
768 int console_announce_r(void)
769 {
770 #if !CONFIG_IS_ENABLED(PRE_CONSOLE_BUFFER)
771         char buf[DISPLAY_OPTIONS_BANNER_LENGTH];
772
773         display_options_get_banner(false, buf, sizeof(buf));
774
775         console_puts_noserial(stdout, buf);
776 #endif
777
778         return 0;
779 }
780
781 /* Called before relocation - use serial functions */
782 int console_init_f(void)
783 {
784         gd->have_console = 1;
785
786         console_update_silent();
787
788         print_pre_console_buffer(PRE_CONSOLE_FLUSHPOINT1_SERIAL);
789
790         return 0;
791 }
792
793 void stdio_print_current_devices(void)
794 {
795         /* Print information */
796         puts("In:    ");
797         if (stdio_devices[stdin] == NULL) {
798                 puts("No input devices available!\n");
799         } else {
800                 printf ("%s\n", stdio_devices[stdin]->name);
801         }
802
803         puts("Out:   ");
804         if (stdio_devices[stdout] == NULL) {
805                 puts("No output devices available!\n");
806         } else {
807                 printf ("%s\n", stdio_devices[stdout]->name);
808         }
809
810         puts("Err:   ");
811         if (stdio_devices[stderr] == NULL) {
812                 puts("No error devices available!\n");
813         } else {
814                 printf ("%s\n", stdio_devices[stderr]->name);
815         }
816 }
817
818 #if CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV)
819 /* Called after the relocation - use desired console functions */
820 int console_init_r(void)
821 {
822         char *stdinname, *stdoutname, *stderrname;
823         struct stdio_dev *inputdev = NULL, *outputdev = NULL, *errdev = NULL;
824 #ifdef CONFIG_SYS_CONSOLE_ENV_OVERWRITE
825         int i;
826 #endif /* CONFIG_SYS_CONSOLE_ENV_OVERWRITE */
827 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
828         int iomux_err = 0;
829 #endif
830         int flushpoint;
831
832         /* update silent for env loaded from flash (initr_env) */
833         if (console_update_silent())
834                 flushpoint = PRE_CONSOLE_FLUSHPOINT1_SERIAL;
835         else
836                 flushpoint = PRE_CONSOLE_FLUSHPOINT2_EVERYTHING_BUT_SERIAL;
837
838         /* set default handlers at first */
839         gd->jt->getc  = serial_getc;
840         gd->jt->tstc  = serial_tstc;
841         gd->jt->putc  = serial_putc;
842         gd->jt->puts  = serial_puts;
843         gd->jt->printf = serial_printf;
844
845         /* stdin stdout and stderr are in environment */
846         /* scan for it */
847         stdinname  = env_get("stdin");
848         stdoutname = env_get("stdout");
849         stderrname = env_get("stderr");
850
851         if (OVERWRITE_CONSOLE == 0) {   /* if not overwritten by config switch */
852                 inputdev  = search_device(DEV_FLAGS_INPUT,  stdinname);
853                 outputdev = search_device(DEV_FLAGS_OUTPUT, stdoutname);
854                 errdev    = search_device(DEV_FLAGS_OUTPUT, stderrname);
855 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
856                 iomux_err = iomux_doenv(stdin, stdinname);
857                 iomux_err += iomux_doenv(stdout, stdoutname);
858                 iomux_err += iomux_doenv(stderr, stderrname);
859                 if (!iomux_err)
860                         /* Successful, so skip all the code below. */
861                         goto done;
862 #endif
863         }
864         /* if the devices are overwritten or not found, use default device */
865         if (inputdev == NULL) {
866                 inputdev  = search_device(DEV_FLAGS_INPUT,  "serial");
867         }
868         if (outputdev == NULL) {
869                 outputdev = search_device(DEV_FLAGS_OUTPUT, "serial");
870         }
871         if (errdev == NULL) {
872                 errdev    = search_device(DEV_FLAGS_OUTPUT, "serial");
873         }
874         /* Initializes output console first */
875         if (outputdev != NULL) {
876                 /* need to set a console if not done above. */
877                 console_doenv(stdout, outputdev);
878         }
879         if (errdev != NULL) {
880                 /* need to set a console if not done above. */
881                 console_doenv(stderr, errdev);
882         }
883         if (inputdev != NULL) {
884                 /* need to set a console if not done above. */
885                 console_doenv(stdin, inputdev);
886         }
887
888 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
889 done:
890 #endif
891
892 #ifndef CONFIG_SYS_CONSOLE_INFO_QUIET
893         stdio_print_current_devices();
894 #endif /* CONFIG_SYS_CONSOLE_INFO_QUIET */
895 #ifdef CONFIG_VIDCONSOLE_AS_LCD
896         if (strstr(stdoutname, "lcd"))
897                 printf("Warning: Please change 'lcd' to 'vidconsole' in stdout/stderr environment vars\n");
898 #endif
899
900 #ifdef CONFIG_SYS_CONSOLE_ENV_OVERWRITE
901         /* set the environment variables (will overwrite previous env settings) */
902         for (i = 0; i < MAX_FILES; i++) {
903                 env_set(stdio_names[i], stdio_devices[i]->name);
904         }
905 #endif /* CONFIG_SYS_CONSOLE_ENV_OVERWRITE */
906
907         gd->flags |= GD_FLG_DEVINIT;    /* device initialization completed */
908
909 #if 0
910         /* If nothing usable installed, use only the initial console */
911         if ((stdio_devices[stdin] == NULL) && (stdio_devices[stdout] == NULL))
912                 return 0;
913 #endif
914         print_pre_console_buffer(flushpoint);
915         return 0;
916 }
917
918 #else /* !CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV) */
919
920 /* Called after the relocation - use desired console functions */
921 int console_init_r(void)
922 {
923         struct stdio_dev *inputdev = NULL, *outputdev = NULL;
924         int i;
925         struct list_head *list = stdio_get_list();
926         struct list_head *pos;
927         struct stdio_dev *dev;
928         int flushpoint;
929
930         /* update silent for env loaded from flash (initr_env) */
931         if (console_update_silent())
932                 flushpoint = PRE_CONSOLE_FLUSHPOINT1_SERIAL;
933         else
934                 flushpoint = PRE_CONSOLE_FLUSHPOINT2_EVERYTHING_BUT_SERIAL;
935
936 #ifdef CONFIG_SPLASH_SCREEN
937         /*
938          * suppress all output if splash screen is enabled and we have
939          * a bmp to display. We redirect the output from frame buffer
940          * console to serial console in this case or suppress it if
941          * "silent" mode was requested.
942          */
943         if (env_get("splashimage") != NULL) {
944                 if (!(gd->flags & GD_FLG_SILENT))
945                         outputdev = search_device (DEV_FLAGS_OUTPUT, "serial");
946         }
947 #endif
948
949         /* Scan devices looking for input and output devices */
950         list_for_each(pos, list) {
951                 dev = list_entry(pos, struct stdio_dev, list);
952
953                 if ((dev->flags & DEV_FLAGS_INPUT) && (inputdev == NULL)) {
954                         inputdev = dev;
955                 }
956                 if ((dev->flags & DEV_FLAGS_OUTPUT) && (outputdev == NULL)) {
957                         outputdev = dev;
958                 }
959                 if(inputdev && outputdev)
960                         break;
961         }
962
963         /* Initializes output console first */
964         if (outputdev != NULL) {
965                 console_setfile(stdout, outputdev);
966                 console_setfile(stderr, outputdev);
967 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
968                 console_devices[stdout][0] = outputdev;
969                 console_devices[stderr][0] = outputdev;
970 #endif
971         }
972
973         /* Initializes input console */
974         if (inputdev != NULL) {
975                 console_setfile(stdin, inputdev);
976 #if CONFIG_IS_ENABLED(CONSOLE_MUX)
977                 console_devices[stdin][0] = inputdev;
978 #endif
979         }
980
981 #ifndef CONFIG_SYS_CONSOLE_INFO_QUIET
982         stdio_print_current_devices();
983 #endif /* CONFIG_SYS_CONSOLE_INFO_QUIET */
984
985         /* Setting environment variables */
986         for (i = 0; i < MAX_FILES; i++) {
987                 env_set(stdio_names[i], stdio_devices[i]->name);
988         }
989
990         gd->flags |= GD_FLG_DEVINIT;    /* device initialization completed */
991
992 #if 0
993         /* If nothing usable installed, use only the initial console */
994         if ((stdio_devices[stdin] == NULL) && (stdio_devices[stdout] == NULL))
995                 return 0;
996 #endif
997         print_pre_console_buffer(flushpoint);
998         return 0;
999 }
1000
1001 #endif /* CONFIG_IS_ENABLED(SYS_CONSOLE_IS_IN_ENV) */