1 /* stdout_console.c */
2 
3 /*
4  * Copyright (c) 2014 Wind River Systems, Inc.
5  *
6  * SPDX-License-Identifier: Apache-2.0
7  */
8 
9 #include <stdio.h>
10 #include <zephyr/sys/libc-hooks.h>
11 #include <zephyr/internal/syscall_handler.h>
12 #include <string.h>
13 
_stdout_hook_default(int c)14 static int _stdout_hook_default(int c)
15 {
16 	(void)(c);  /* Prevent warning about unused argument */
17 
18 	return EOF;
19 }
20 
21 static int (*_stdout_hook)(int c) = _stdout_hook_default;
22 
__stdout_hook_install(int (* hook)(int c))23 void __stdout_hook_install(int (*hook)(int c))
24 {
25 	_stdout_hook = hook;
26 }
27 
z_impl_zephyr_fputc(int c,FILE * stream)28 int z_impl_zephyr_fputc(int c, FILE *stream)
29 {
30 	return ((stream == stdout) || (stream == stderr)) ? _stdout_hook(c) : EOF;
31 }
32 
33 #ifdef CONFIG_USERSPACE
z_vrfy_zephyr_fputc(int c,FILE * stream)34 static inline int z_vrfy_zephyr_fputc(int c, FILE *stream)
35 {
36 	return z_impl_zephyr_fputc(c, stream);
37 }
38 #include <zephyr/syscalls/zephyr_fputc_mrsh.c>
39 #endif
40 
fputc(int c,FILE * stream)41 int fputc(int c, FILE *stream)
42 {
43 	return zephyr_fputc(c, stream);
44 }
45 
fputs(const char * ZRESTRICT s,FILE * ZRESTRICT stream)46 int fputs(const char *ZRESTRICT s, FILE *ZRESTRICT stream)
47 {
48 	int len = strlen(s);
49 	int ret;
50 
51 	ret = fwrite(s, 1, len, stream);
52 
53 	return (len == ret) ? 0 : EOF;
54 }
55 
56 #undef putc
putc(int c,FILE * stream)57 int putc(int c, FILE *stream)
58 {
59 	return zephyr_fputc(c, stream);
60 }
61 
62 #undef putchar
putchar(int c)63 int putchar(int c)
64 {
65 	return zephyr_fputc(c, stdout);
66 }
67 
z_impl_zephyr_fwrite(const void * ZRESTRICT ptr,size_t size,size_t nitems,FILE * ZRESTRICT stream)68 size_t z_impl_zephyr_fwrite(const void *ZRESTRICT ptr, size_t size,
69 			    size_t nitems, FILE *ZRESTRICT stream)
70 {
71 	size_t i;
72 	size_t j;
73 	const unsigned char *p;
74 
75 	if (((stream != stdout) && (stream != stderr)) ||
76 	    (nitems == 0) || (size == 0)) {
77 		return 0;
78 	}
79 
80 	p = ptr;
81 	i = nitems;
82 	do {
83 		j = size;
84 		do {
85 			if (_stdout_hook((int) *p++) == EOF) {
86 				goto done;
87 			}
88 			j--;
89 		} while (j > 0);
90 
91 		i--;
92 	} while (i > 0);
93 
94 done:
95 	return (nitems - i);
96 }
97 
98 #ifdef CONFIG_USERSPACE
z_vrfy_zephyr_fwrite(const void * ZRESTRICT ptr,size_t size,size_t nitems,FILE * ZRESTRICT stream)99 static inline size_t z_vrfy_zephyr_fwrite(const void *ZRESTRICT ptr,
100 					  size_t size, size_t nitems,
101 					  FILE *ZRESTRICT stream)
102 {
103 
104 	K_OOPS(K_SYSCALL_MEMORY_ARRAY_READ(ptr, nitems, size));
105 	return z_impl_zephyr_fwrite(ptr, size, nitems, stream);
106 }
107 #include <zephyr/syscalls/zephyr_fwrite_mrsh.c>
108 #endif
109 
fwrite(const void * ZRESTRICT ptr,size_t size,size_t nitems,FILE * ZRESTRICT stream)110 size_t fwrite(const void *ZRESTRICT ptr, size_t size, size_t nitems,
111 			  FILE *ZRESTRICT stream)
112 {
113 	return zephyr_fwrite(ptr, size, nitems, stream);
114 }
115 
116 
puts(const char * s)117 int puts(const char *s)
118 {
119 	if (fputs(s, stdout) == EOF) {
120 		return EOF;
121 	}
122 
123 	return (fputc('\n', stdout) == EOF) ? EOF : 0;
124 }
125