Commit | Line | Data |
---|---|---|
2874c5fd | 1 | // SPDX-License-Identifier: GPL-2.0-or-later |
66a45dd3 PM |
2 | /* |
3 | * Copyright (C) Paul Mackerras 1997. | |
66a45dd3 PM |
4 | */ |
5 | #include <stdarg.h> | |
6 | #include <stddef.h> | |
7 | #include "string.h" | |
8 | #include "stdio.h" | |
b2c5f619 | 9 | #include "ops.h" |
66a45dd3 PM |
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 | ||
a54b81ea RH |
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 | ||
f4bce2f7 CLG |
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 | ||
66a45dd3 PM |
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 | ||
f4bce2f7 CLG |
60 | #endif /* __powerpc64__ */ |
61 | ||
66a45dd3 PM |
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(&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(&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; | |
0aa97d6e GL |
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') { | |
66a45dd3 PM |
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; | |
0aa97d6e GL |
308 | } else if (qualifier == 'q') { |
309 | num = va_arg(args, unsigned long long); | |
310 | if (flags & SIGN) | |
311 | num = (signed long long) num; | |
66a45dd3 PM |
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, field_width, precision, 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); | |
c888554b MG |
351 | if (console_ops.write) |
352 | console_ops.write(sprint_buf, n); | |
66a45dd3 PM |
353 | return n; |
354 | } |