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