xref: /rk3399_rockchip-uboot/common/console.c (revision a0ff7f2eda50de4d4096265a086c94ac640fab7c)
1 /*
2  * (C) Copyright 2000
3  * Paolo Scaffardi, AIRVENT SAM s.p.a - RIMINI(ITALY), arsenio@tin.it
4  *
5  * See file CREDITS for list of people who contributed to this
6  * project.
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License as
10  * published by the Free Software Foundation; either version 2 of
11  * the License, or (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
21  * MA 02111-1307 USA
22  */
23 
24 #include <common.h>
25 #include <stdarg.h>
26 #include <malloc.h>
27 #include <console.h>
28 #include <exports.h>
29 
30 #ifdef CONFIG_AMIGAONEG3SE
31 int console_changed = 0;
32 #endif
33 
34 #ifdef CFG_CONSOLE_IS_IN_ENV
35 /*
36  * if overwrite_console returns 1, the stdin, stderr and stdout
37  * are switched to the serial port, else the settings in the
38  * environment are used
39  */
40 #ifdef CFG_CONSOLE_OVERWRITE_ROUTINE
41 extern int overwrite_console (void);
42 #else
43 int overwrite_console (void)
44 {
45 	return (0);
46 }
47 #endif /* CFG_CONSOLE_OVERWRITE_ROUTINE */
48 
49 #endif /* CFG_CONSOLE_IS_IN_ENV */
50 
51 static int console_setfile (int file, device_t * dev)
52 {
53 	DECLARE_GLOBAL_DATA_PTR;
54 	int error = 0;
55 
56 	if (dev == NULL)
57 		return -1;
58 
59 	switch (file) {
60 	case stdin:
61 	case stdout:
62 	case stderr:
63 		/* Start new device */
64 		if (dev->start) {
65 			error = dev->start ();
66 			/* If it's not started dont use it */
67 			if (error < 0)
68 				break;
69 		}
70 
71 		/* Assign the new device (leaving the existing one started) */
72 		stdio_devices[file] = dev;
73 
74 		/*
75 		 * Update monitor functions
76 		 * (to use the console stuff by other applications)
77 		 */
78 		switch (file) {
79 		case stdin:
80 			gd->jt[XF_getc] = dev->getc;
81 			gd->jt[XF_tstc] = dev->tstc;
82 			break;
83 		case stdout:
84 			gd->jt[XF_putc] = dev->putc;
85 			gd->jt[XF_puts] = dev->puts;
86 			gd->jt[XF_printf] = printf;
87 			break;
88 		}
89 		break;
90 
91 	default:		/* Invalid file ID */
92 		error = -1;
93 	}
94 	return error;
95 }
96 
97 /** U-Boot INITIAL CONSOLE-NOT COMPATIBLE FUNCTIONS *************************/
98 
99 void serial_printf (const char *fmt, ...)
100 {
101 	va_list args;
102 	uint i;
103 	char printbuffer[CFG_PBSIZE];
104 
105 	va_start (args, fmt);
106 
107 	/* For this to work, printbuffer must be larger than
108 	 * anything we ever want to print.
109 	 */
110 	i = vsprintf (printbuffer, fmt, args);
111 	va_end (args);
112 
113 	serial_puts (printbuffer);
114 }
115 
116 int fgetc (int file)
117 {
118 	if (file < MAX_FILES)
119 		return stdio_devices[file]->getc ();
120 
121 	return -1;
122 }
123 
124 int ftstc (int file)
125 {
126 	if (file < MAX_FILES)
127 		return stdio_devices[file]->tstc ();
128 
129 	return -1;
130 }
131 
132 void fputc (int file, const char c)
133 {
134 	if (file < MAX_FILES)
135 		stdio_devices[file]->putc (c);
136 }
137 
138 void fputs (int file, const char *s)
139 {
140 	if (file < MAX_FILES)
141 		stdio_devices[file]->puts (s);
142 }
143 
144 void fprintf (int file, const char *fmt, ...)
145 {
146 	va_list args;
147 	uint i;
148 	char printbuffer[CFG_PBSIZE];
149 
150 	va_start (args, fmt);
151 
152 	/* For this to work, printbuffer must be larger than
153 	 * anything we ever want to print.
154 	 */
155 	i = vsprintf (printbuffer, fmt, args);
156 	va_end (args);
157 
158 	/* Send to desired file */
159 	fputs (file, printbuffer);
160 }
161 
162 /** U-Boot INITIAL CONSOLE-COMPATIBLE FUNCTION *****************************/
163 
164 int getc (void)
165 {
166 	DECLARE_GLOBAL_DATA_PTR;
167 
168 	if (gd->flags & GD_FLG_DEVINIT) {
169 		/* Get from the standard input */
170 		return fgetc (stdin);
171 	}
172 
173 	/* Send directly to the handler */
174 	return serial_getc ();
175 }
176 
177 int tstc (void)
178 {
179 	DECLARE_GLOBAL_DATA_PTR;
180 
181 	if (gd->flags & GD_FLG_DEVINIT) {
182 		/* Test the standard input */
183 		return ftstc (stdin);
184 	}
185 
186 	/* Send directly to the handler */
187 	return serial_tstc ();
188 }
189 
190 void putc (const char c)
191 {
192 	DECLARE_GLOBAL_DATA_PTR;
193 
194 	if (gd->flags & GD_FLG_DEVINIT) {
195 		/* Send to the standard output */
196 		fputc (stdout, c);
197 	} else {
198 		/* Send directly to the handler */
199 		serial_putc (c);
200 	}
201 }
202 
203 void puts (const char *s)
204 {
205 	DECLARE_GLOBAL_DATA_PTR;
206 
207 	if (gd->flags & GD_FLG_DEVINIT) {
208 		/* Send to the standard output */
209 		fputs (stdout, s);
210 	} else {
211 		/* Send directly to the handler */
212 		serial_puts (s);
213 	}
214 }
215 
216 void printf (const char *fmt, ...)
217 {
218 	va_list args;
219 	uint i;
220 	char printbuffer[CFG_PBSIZE];
221 
222 	va_start (args, fmt);
223 
224 	/* For this to work, printbuffer must be larger than
225 	 * anything we ever want to print.
226 	 */
227 	i = vsprintf (printbuffer, fmt, args);
228 	va_end (args);
229 
230 	/* Print the string */
231 	puts (printbuffer);
232 }
233 
234 void vprintf (const char *fmt, va_list args)
235 {
236 	uint i;
237 	char printbuffer[CFG_PBSIZE];
238 
239 	/* For this to work, printbuffer must be larger than
240 	 * anything we ever want to print.
241 	 */
242 	i = vsprintf (printbuffer, fmt, args);
243 
244 	/* Print the string */
245 	puts (printbuffer);
246 }
247 
248 /* test if ctrl-c was pressed */
249 static int ctrlc_disabled = 0;	/* see disable_ctrl() */
250 static int ctrlc_was_pressed = 0;
251 int ctrlc (void)
252 {
253 	DECLARE_GLOBAL_DATA_PTR;
254 
255 	if (!ctrlc_disabled && gd->have_console) {
256 		if (tstc ()) {
257 			switch (getc ()) {
258 			case 0x03:		/* ^C - Control C */
259 				ctrlc_was_pressed = 1;
260 				return 1;
261 			default:
262 				break;
263 			}
264 		}
265 	}
266 	return 0;
267 }
268 
269 /* pass 1 to disable ctrlc() checking, 0 to enable.
270  * returns previous state
271  */
272 int disable_ctrlc (int disable)
273 {
274 	int prev = ctrlc_disabled;	/* save previous state */
275 
276 	ctrlc_disabled = disable;
277 	return prev;
278 }
279 
280 int had_ctrlc (void)
281 {
282 	return ctrlc_was_pressed;
283 }
284 
285 void clear_ctrlc (void)
286 {
287 	ctrlc_was_pressed = 0;
288 }
289 
290 #ifdef CONFIG_MODEM_SUPPORT_DEBUG
291 char	screen[1024];
292 char *cursor = screen;
293 int once = 0;
294 inline void dbg(const char *fmt, ...)
295 {
296 	va_list	args;
297 	uint	i;
298 	char	printbuffer[CFG_PBSIZE];
299 
300 	if (!once) {
301 		memset(screen, 0, sizeof(screen));
302 		once++;
303 	}
304 
305 	va_start(args, fmt);
306 
307 	/* For this to work, printbuffer must be larger than
308 	 * anything we ever want to print.
309 	 */
310 	i = vsprintf(printbuffer, fmt, args);
311 	va_end(args);
312 
313 	if ((screen + sizeof(screen) - 1 - cursor) < strlen(printbuffer)+1) {
314 		memset(screen, 0, sizeof(screen));
315 		cursor = screen;
316 	}
317 	sprintf(cursor, printbuffer);
318 	cursor += strlen(printbuffer);
319 
320 }
321 #else
322 inline void dbg(const char *fmt, ...)
323 {
324 }
325 #endif
326 
327 /** U-Boot INIT FUNCTIONS *************************************************/
328 
329 int console_assign (int file, char *devname)
330 {
331 	int flag, i;
332 
333 	/* Check for valid file */
334 	switch (file) {
335 	case stdin:
336 		flag = DEV_FLAGS_INPUT;
337 		break;
338 	case stdout:
339 	case stderr:
340 		flag = DEV_FLAGS_OUTPUT;
341 		break;
342 	default:
343 		return -1;
344 	}
345 
346 	/* Check for valid device name */
347 
348 	for (i = 1; i <= ListNumItems (devlist); i++) {
349 		device_t *dev = ListGetPtrToItem (devlist, i);
350 
351 		if (strcmp (devname, dev->name) == 0) {
352 			if (dev->flags & flag)
353 				return console_setfile (file, dev);
354 
355 			return -1;
356 		}
357 	}
358 
359 	return -1;
360 }
361 
362 /* Called before relocation - use serial functions */
363 int console_init_f (void)
364 {
365 	DECLARE_GLOBAL_DATA_PTR;
366 
367 	gd->have_console = 1;
368 	return (0);
369 }
370 
371 #if defined(CFG_CONSOLE_IS_IN_ENV) || defined(CONFIG_SPLASH_SCREEN)
372 /* search a device */
373 device_t *search_device (int flags, char *name)
374 {
375 	int i, items;
376 	device_t *dev = NULL;
377 
378 	items = ListNumItems (devlist);
379 	if (name == NULL)
380 		return dev;
381 
382 	for (i = 1; i <= items; i++) {
383 		dev = ListGetPtrToItem (devlist, i);
384 		if ((dev->flags & flags) && (strcmp (name, dev->name) == 0)) {
385 			break;
386 		}
387 	}
388 	return dev;
389 }
390 #endif /* CFG_CONSOLE_IS_IN_ENV || CONFIG_SPLASH_SCREEN */
391 
392 #ifdef CFG_CONSOLE_IS_IN_ENV
393 /* Called after the relocation - use desired console functions */
394 int console_init_r (void)
395 {
396 	DECLARE_GLOBAL_DATA_PTR;
397 	char *stdinname, *stdoutname, *stderrname;
398 	device_t *inputdev = NULL, *outputdev = NULL, *errdev = NULL;
399 
400 	/* set default handlers at first */
401 	gd->jt[XF_getc] = serial_getc;
402 	gd->jt[XF_tstc] = serial_tstc;
403 	gd->jt[XF_putc] = serial_putc;
404 	gd->jt[XF_puts] = serial_puts;
405 	gd->jt[XF_printf] = serial_printf;
406 
407 	/* stdin stdout and stderr are in environment */
408 	/* scan for it */
409 	stdinname  = getenv ("stdin");
410 	stdoutname = getenv ("stdout");
411 	stderrname = getenv ("stderr");
412 
413 	if (overwrite_console () == 0) { /* if not overwritten by config switch */
414 		inputdev  = search_device (DEV_FLAGS_INPUT,  stdinname);
415 		outputdev = search_device (DEV_FLAGS_OUTPUT, stdoutname);
416 		errdev    = search_device (DEV_FLAGS_OUTPUT, stderrname);
417 	}
418 	/* if the devices are overwritten or not found, use default device */
419 	if (inputdev == NULL) {
420 		inputdev  = search_device (DEV_FLAGS_INPUT,  "serial");
421 	}
422 	if (outputdev == NULL) {
423 		outputdev = search_device (DEV_FLAGS_OUTPUT, "serial");
424 	}
425 	if (errdev == NULL) {
426 		errdev    = search_device (DEV_FLAGS_OUTPUT, "serial");
427 	}
428 	/* Initializes output console first */
429 	if (outputdev != NULL) {
430 		console_setfile (stdout, outputdev);
431 	}
432 	if (errdev != NULL) {
433 		console_setfile (stderr, errdev);
434 	}
435 	if (inputdev != NULL) {
436 		console_setfile (stdin, inputdev);
437 	}
438 
439 	gd->flags |= GD_FLG_DEVINIT;	/* device initialization completed */
440 
441 #ifndef CFG_CONSOLE_INFO_QUIET
442 	/* Print information */
443 	printf ("In:    ");
444 	if (stdio_devices[stdin] == NULL) {
445 		printf ("No input devices available!\n");
446 	} else {
447 		printf ("%s\n", stdio_devices[stdin]->name);
448 	}
449 
450 	printf ("Out:   ");
451 	if (stdio_devices[stdout] == NULL) {
452 		printf ("No output devices available!\n");
453 	} else {
454 		printf ("%s\n", stdio_devices[stdout]->name);
455 	}
456 
457 	printf ("Err:   ");
458 	if (stdio_devices[stderr] == NULL) {
459 		printf ("No error devices available!\n");
460 	} else {
461 		printf ("%s\n", stdio_devices[stderr]->name);
462 	}
463 #endif /* CFG_CONSOLE_INFO_QUIET */
464 
465 #ifdef CFG_CONSOLE_ENV_OVERWRITE
466 	/* set the environment variables (will overwrite previous env settings) */
467 	for (i = 0; i < 3; i++) {
468 		setenv (stdio_names[i], stdio_devices[i]->name);
469 	}
470 #endif /*  CFG_CONSOLE_ENV_OVERWRITE */
471 
472 #if 0
473 	/* If nothing usable installed, use only the initial console */
474 	if ((stdio_devices[stdin] == NULL) && (stdio_devices[stdout] == NULL))
475 		return (0);
476 #endif
477 	return (0);
478 }
479 
480 #else /* CFG_CONSOLE_IS_IN_ENV */
481 
482 /* Called after the relocation - use desired console functions */
483 int console_init_r (void)
484 {
485 	DECLARE_GLOBAL_DATA_PTR;
486 
487 	device_t *inputdev = NULL, *outputdev = NULL;
488 	int i, items = ListNumItems (devlist);
489 
490 #ifdef CONFIG_SPLASH_SCREEN
491 	/* suppress all output if splash screen is enabled and we have
492            a bmp to display                                            */
493 	if (getenv("splashimage") != NULL)
494 		outputdev = search_device (DEV_FLAGS_OUTPUT, "nulldev");
495 #endif
496 
497 	/* Scan devices looking for input and output devices */
498 	for (i = 1;
499 	     (i <= items) && ((inputdev == NULL) || (outputdev == NULL));
500 	     i++
501 	    ) {
502 		device_t *dev = ListGetPtrToItem (devlist, i);
503 
504 		if ((dev->flags & DEV_FLAGS_INPUT) && (inputdev == NULL)) {
505 			inputdev = dev;
506 		}
507 		if ((dev->flags & DEV_FLAGS_OUTPUT) && (outputdev == NULL)) {
508 			outputdev = dev;
509 		}
510 	}
511 
512 	/* Initializes output console first */
513 	if (outputdev != NULL) {
514 		console_setfile (stdout, outputdev);
515 		console_setfile (stderr, outputdev);
516 	}
517 
518 	/* Initializes input console */
519 	if (inputdev != NULL) {
520 		console_setfile (stdin, inputdev);
521 	}
522 
523 	gd->flags |= GD_FLG_DEVINIT;	/* device initialization completed */
524 
525 #ifndef CFG_CONSOLE_INFO_QUIET
526 	/* Print information */
527 	printf ("In:    ");
528 	if (stdio_devices[stdin] == NULL) {
529 		printf ("No input devices available!\n");
530 	} else {
531 		printf ("%s\n", stdio_devices[stdin]->name);
532 	}
533 
534 	printf ("Out:   ");
535 	if (stdio_devices[stdout] == NULL) {
536 		printf ("No output devices available!\n");
537 	} else {
538 		printf ("%s\n", stdio_devices[stdout]->name);
539 	}
540 
541 	printf ("Err:   ");
542 	if (stdio_devices[stderr] == NULL) {
543 		printf ("No error devices available!\n");
544 	} else {
545 		printf ("%s\n", stdio_devices[stderr]->name);
546 	}
547 #endif /* CFG_CONSOLE_INFO_QUIET */
548 
549 	/* Setting environment variables */
550 	for (i = 0; i < 3; i++) {
551 		setenv (stdio_names[i], stdio_devices[i]->name);
552 	}
553 
554 #if 0
555 	/* If nothing usable installed, use only the initial console */
556 	if ((stdio_devices[stdin] == NULL) && (stdio_devices[stdout] == NULL))
557 		return (0);
558 #endif
559 
560 	return (0);
561 }
562 
563 #endif /* CFG_CONSOLE_IS_IN_ENV */
564