stdio.c 7.0 KB

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