1 | // SPDX-License-Identifier: GPL-2.0-or-later |
2 | /* |
3 | * Copyright (C) Paul Mackerras 1997. |
4 | */ |
5 | #include <stdarg.h> |
6 | #include <stddef.h> |
7 | #include "string.h" |
8 | #include "stdio.h" |
9 | #include "ops.h" |
10 | |
11 | size_t strnlen(const char * s, size_t count) |
12 | { |
13 | const char *sc; |
14 | |
15 | for (sc = s; count-- && *sc != '\0'; ++sc) |
16 | /* nothing */; |
17 | return sc - s; |
18 | } |
19 | |
20 | char *strrchr(const char *s, int c) |
21 | { |
22 | const char *last = NULL; |
23 | do { |
24 | if (*s == (char)c) |
25 | last = s; |
26 | } while (*s++); |
27 | return (char *)last; |
28 | } |
29 | |
30 | #ifdef __powerpc64__ |
31 | |
32 | # define do_div(n, base) ({ \ |
33 | unsigned int __base = (base); \ |
34 | unsigned int __rem; \ |
35 | __rem = ((unsigned long long)(n)) % __base; \ |
36 | (n) = ((unsigned long long)(n)) / __base; \ |
37 | __rem; \ |
38 | }) |
39 | |
40 | #else |
41 | |
42 | extern unsigned int __div64_32(unsigned long long *dividend, |
43 | unsigned int divisor); |
44 | |
45 | /* The unnecessary pointer compare is there |
46 | * to check for type safety (n must be 64bit) |
47 | */ |
48 | # define do_div(n,base) ({ \ |
49 | unsigned int __base = (base); \ |
50 | unsigned int __rem; \ |
51 | (void)(((typeof((n)) *)0) == ((unsigned long long *)0)); \ |
52 | if (((n) >> 32) == 0) { \ |
53 | __rem = (unsigned int)(n) % __base; \ |
54 | (n) = (unsigned int)(n) / __base; \ |
55 | } else \ |
56 | __rem = __div64_32(&(n), __base); \ |
57 | __rem; \ |
58 | }) |
59 | |
60 | #endif /* __powerpc64__ */ |
61 | |
62 | static int skip_atoi(const char **s) |
63 | { |
64 | int i, c; |
65 | |
66 | for (i = 0; '0' <= (c = **s) && c <= '9'; ++*s) |
67 | i = i*10 + c - '0'; |
68 | return i; |
69 | } |
70 | |
71 | #define ZEROPAD 1 /* pad with zero */ |
72 | #define SIGN 2 /* unsigned/signed long */ |
73 | #define PLUS 4 /* show plus */ |
74 | #define SPACE 8 /* space if plus */ |
75 | #define LEFT 16 /* left justified */ |
76 | #define SPECIAL 32 /* 0x */ |
77 | #define LARGE 64 /* use 'ABCDEF' instead of 'abcdef' */ |
78 | |
79 | static char * number(char * str, unsigned long long num, int base, int size, int precision, int type) |
80 | { |
81 | char c,sign,tmp[66]; |
82 | const char *digits="0123456789abcdefghijklmnopqrstuvwxyz" ; |
83 | int i; |
84 | |
85 | if (type & LARGE) |
86 | digits = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ" ; |
87 | if (type & LEFT) |
88 | type &= ~ZEROPAD; |
89 | if (base < 2 || base > 36) |
90 | return 0; |
91 | c = (type & ZEROPAD) ? '0' : ' '; |
92 | sign = 0; |
93 | if (type & SIGN) { |
94 | if ((signed long long)num < 0) { |
95 | sign = '-'; |
96 | num = - (signed long long)num; |
97 | size--; |
98 | } else if (type & PLUS) { |
99 | sign = '+'; |
100 | size--; |
101 | } else if (type & SPACE) { |
102 | sign = ' '; |
103 | size--; |
104 | } |
105 | } |
106 | if (type & SPECIAL) { |
107 | if (base == 16) |
108 | size -= 2; |
109 | else if (base == 8) |
110 | size--; |
111 | } |
112 | i = 0; |
113 | if (num == 0) |
114 | tmp[i++]='0'; |
115 | else while (num != 0) { |
116 | tmp[i++] = digits[do_div(num, base)]; |
117 | } |
118 | if (i > precision) |
119 | precision = i; |
120 | size -= precision; |
121 | if (!(type&(ZEROPAD+LEFT))) |
122 | while(size-->0) |
123 | *str++ = ' '; |
124 | if (sign) |
125 | *str++ = sign; |
126 | if (type & SPECIAL) { |
127 | if (base==8) |
128 | *str++ = '0'; |
129 | else if (base==16) { |
130 | *str++ = '0'; |
131 | *str++ = digits[33]; |
132 | } |
133 | } |
134 | if (!(type & LEFT)) |
135 | while (size-- > 0) |
136 | *str++ = c; |
137 | while (i < precision--) |
138 | *str++ = '0'; |
139 | while (i-- > 0) |
140 | *str++ = tmp[i]; |
141 | while (size-- > 0) |
142 | *str++ = ' '; |
143 | return str; |
144 | } |
145 | |
146 | int vsprintf(char *buf, const char *fmt, va_list args) |
147 | { |
148 | int len; |
149 | unsigned long long num; |
150 | int i, base; |
151 | char * str; |
152 | const char *s; |
153 | |
154 | int flags; /* flags to number() */ |
155 | |
156 | int field_width; /* width of output field */ |
157 | int precision; /* min. # of digits for integers; max |
158 | number of chars for from string */ |
159 | int qualifier; /* 'h', 'l', or 'L' for integer fields */ |
160 | /* 'z' support added 23/7/1999 S.H. */ |
161 | /* 'z' changed to 'Z' --davidm 1/25/99 */ |
162 | |
163 | |
164 | for (str=buf ; *fmt ; ++fmt) { |
165 | if (*fmt != '%') { |
166 | *str++ = *fmt; |
167 | continue; |
168 | } |
169 | |
170 | /* process flags */ |
171 | flags = 0; |
172 | repeat: |
173 | ++fmt; /* this also skips first '%' */ |
174 | switch (*fmt) { |
175 | case '-': flags |= LEFT; goto repeat; |
176 | case '+': flags |= PLUS; goto repeat; |
177 | case ' ': flags |= SPACE; goto repeat; |
178 | case '#': flags |= SPECIAL; goto repeat; |
179 | case '0': flags |= ZEROPAD; goto repeat; |
180 | } |
181 | |
182 | /* get field width */ |
183 | field_width = -1; |
184 | if ('0' <= *fmt && *fmt <= '9') |
185 | field_width = skip_atoi(s: &fmt); |
186 | else if (*fmt == '*') { |
187 | ++fmt; |
188 | /* it's the next argument */ |
189 | field_width = va_arg(args, int); |
190 | if (field_width < 0) { |
191 | field_width = -field_width; |
192 | flags |= LEFT; |
193 | } |
194 | } |
195 | |
196 | /* get the precision */ |
197 | precision = -1; |
198 | if (*fmt == '.') { |
199 | ++fmt; |
200 | if ('0' <= *fmt && *fmt <= '9') |
201 | precision = skip_atoi(s: &fmt); |
202 | else if (*fmt == '*') { |
203 | ++fmt; |
204 | /* it's the next argument */ |
205 | precision = va_arg(args, int); |
206 | } |
207 | if (precision < 0) |
208 | precision = 0; |
209 | } |
210 | |
211 | /* get the conversion qualifier */ |
212 | qualifier = -1; |
213 | if (*fmt == 'l' && *(fmt + 1) == 'l') { |
214 | qualifier = 'q'; |
215 | fmt += 2; |
216 | } else if (*fmt == 'h' || *fmt == 'l' || *fmt == 'L' |
217 | || *fmt == 'Z') { |
218 | qualifier = *fmt; |
219 | ++fmt; |
220 | } |
221 | |
222 | /* default base */ |
223 | base = 10; |
224 | |
225 | switch (*fmt) { |
226 | case 'c': |
227 | if (!(flags & LEFT)) |
228 | while (--field_width > 0) |
229 | *str++ = ' '; |
230 | *str++ = (unsigned char) va_arg(args, int); |
231 | while (--field_width > 0) |
232 | *str++ = ' '; |
233 | continue; |
234 | |
235 | case 's': |
236 | s = va_arg(args, char *); |
237 | if (!s) |
238 | s = "<NULL>" ; |
239 | |
240 | len = strnlen(s, precision); |
241 | |
242 | if (!(flags & LEFT)) |
243 | while (len < field_width--) |
244 | *str++ = ' '; |
245 | for (i = 0; i < len; ++i) |
246 | *str++ = *s++; |
247 | while (len < field_width--) |
248 | *str++ = ' '; |
249 | continue; |
250 | |
251 | case 'p': |
252 | if (field_width == -1) { |
253 | field_width = 2*sizeof(void *); |
254 | flags |= ZEROPAD; |
255 | } |
256 | str = number(str, |
257 | (unsigned long) va_arg(args, void *), 16, |
258 | field_width, precision, flags); |
259 | continue; |
260 | |
261 | |
262 | case 'n': |
263 | if (qualifier == 'l') { |
264 | long * ip = va_arg(args, long *); |
265 | *ip = (str - buf); |
266 | } else if (qualifier == 'Z') { |
267 | size_t * ip = va_arg(args, size_t *); |
268 | *ip = (str - buf); |
269 | } else { |
270 | int * ip = va_arg(args, int *); |
271 | *ip = (str - buf); |
272 | } |
273 | continue; |
274 | |
275 | case '%': |
276 | *str++ = '%'; |
277 | continue; |
278 | |
279 | /* integer number formats - set up the flags and "break" */ |
280 | case 'o': |
281 | base = 8; |
282 | break; |
283 | |
284 | case 'X': |
285 | flags |= LARGE; |
286 | case 'x': |
287 | base = 16; |
288 | break; |
289 | |
290 | case 'd': |
291 | case 'i': |
292 | flags |= SIGN; |
293 | case 'u': |
294 | break; |
295 | |
296 | default: |
297 | *str++ = '%'; |
298 | if (*fmt) |
299 | *str++ = *fmt; |
300 | else |
301 | --fmt; |
302 | continue; |
303 | } |
304 | if (qualifier == 'l') { |
305 | num = va_arg(args, unsigned long); |
306 | if (flags & SIGN) |
307 | num = (signed long) num; |
308 | } else if (qualifier == 'q') { |
309 | num = va_arg(args, unsigned long long); |
310 | if (flags & SIGN) |
311 | num = (signed long long) num; |
312 | } else if (qualifier == 'Z') { |
313 | num = va_arg(args, size_t); |
314 | } else if (qualifier == 'h') { |
315 | num = (unsigned short) va_arg(args, int); |
316 | if (flags & SIGN) |
317 | num = (signed short) num; |
318 | } else { |
319 | num = va_arg(args, unsigned int); |
320 | if (flags & SIGN) |
321 | num = (signed int) num; |
322 | } |
323 | str = number(str, num, base, size: field_width, precision, type: flags); |
324 | } |
325 | *str = '\0'; |
326 | return str-buf; |
327 | } |
328 | |
329 | int sprintf(char * buf, const char *fmt, ...) |
330 | { |
331 | va_list args; |
332 | int i; |
333 | |
334 | va_start(args, fmt); |
335 | i=vsprintf(buf,fmt,args); |
336 | va_end(args); |
337 | return i; |
338 | } |
339 | |
340 | static char sprint_buf[1024]; |
341 | |
342 | int |
343 | printf(const char *fmt, ...) |
344 | { |
345 | va_list args; |
346 | int n; |
347 | |
348 | va_start(args, fmt); |
349 | n = vsprintf(sprint_buf, fmt, args); |
350 | va_end(args); |
351 | if (console_ops.write) |
352 | console_ops.write(sprint_buf, n); |
353 | return n; |
354 | } |
355 | |