dlls/ntdll/printf.h

   1 /*
   2  * ntdll printf functions
   3  *
   4  * Copyright 1999, 2009 Alexandre Julliard
   5  * Copyright 2000 Jon Griffiths
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #ifdef PRINTF_WIDE
  23 #define APICHAR wchar_t
  24 #define APISTR(str) L"" str
  25 #define FUNC_NAME(func) func ## _w
  26 #else
  27 #define APICHAR char
  28 #define APISTR(str) str
  29 #define FUNC_NAME(func) func ## _a
  30 #endif
  31
  32 typedef struct
  33 {
  34     APICHAR *buf;
  35     SIZE_T   len;
  36     SIZE_T   used;
  37 } FUNC_NAME(pf_output);
  38
  39 /*
  40  * writes a string of characters to the output
  41  * returns -1 if the string doesn't fit in the output buffer
  42  * return the length of the string if all characters were written
  43  */
  44 static int FUNC_NAME(pf_output_wstr)( FUNC_NAME(pf_output) *out, const WCHAR *str, int len )
  45 {
  46     SIZE_T space = out->len - out->used;
  47     APICHAR *p = out->buf + out->used;
  48     ULONG n;
  49
  50     if (len < 0) len = wcslen( str );
  51
  52 #ifdef PRINTF_WIDE
  53     n = len;
  54     if (out->buf) memcpy( p, str, min( n, space ) * sizeof(WCHAR) );
  55 #else
  56     RtlUnicodeToMultiByteSize( &n, str, len * sizeof(WCHAR) );
  57     if (out->buf) RtlUnicodeToMultiByteN( p, min( n, space ), NULL, str, len * sizeof(WCHAR) );
  58 #endif
  59     if (out->buf && space < n)  /* overflow */
  60     {
  61         out->used = out->len;
  62         return -1;
  63     }
  64     out->used += n;
  65     return len;
  66 }
  67
  68 static int FUNC_NAME(pf_output_str)( FUNC_NAME(pf_output) *out, const char *str, int len )
  69 {
  70     SIZE_T space = out->len - out->used;
  71     APICHAR *p = out->buf + out->used;
  72     ULONG n;
  73
  74     if (len < 0) len = strlen( str );
  75
  76 #ifdef PRINTF_WIDE
  77     RtlMultiByteToUnicodeSize( &n, str, len );
  78     n /= sizeof(WCHAR);
  79     if (out->buf) RtlMultiByteToUnicodeN( p, min( n, space ) * sizeof(WCHAR), NULL, str, len );
  80 #else
  81     n = len;
  82     if (out->buf) memcpy( p, str, min( n, space ));
  83 #endif
  84     if (out->buf && space < n)  /* overflow */
  85     {
  86         out->used = out->len;
  87         return -1;
  88     }
  89     out->used += n;
  90     return len;
  91 }
  92
  93 static int FUNC_NAME(pf_output_string)( FUNC_NAME(pf_output) *out, const APICHAR *str, int len )
  94 {
  95 #ifdef PRINTF_WIDE
  96     return FUNC_NAME(pf_output_wstr)( out, str, len );
  97 #else
  98     return FUNC_NAME(pf_output_str)( out, str, len );
  99 #endif
 100 }
 101
 102 /* pf_fill: takes care of signs, alignment, zero and field padding */
 103 static int FUNC_NAME(pf_fill_left)( FUNC_NAME(pf_output) *out, int len, pf_flags *flags )
 104 {
 105     int i, r = 0;
 106
 107     if (flags->Sign && !(flags->Format == 'd' || flags->Format == 'i')) flags->Sign = 0;
 108
 109     if (flags->Sign)
 110     {
 111         APICHAR ch = flags->Sign;
 112         flags->FieldLength--;
 113         if (flags->PadZero) r = FUNC_NAME(pf_output_string)( out, &ch, 1 );
 114     }
 115     if (!flags->LeftAlign)
 116     {
 117         APICHAR ch = flags->PadZero ? '0' : ' ';
 118         for (i = 0; i < flags->FieldLength - len && r >= 0; i++)
 119             r = FUNC_NAME(pf_output_string)( out, &ch, 1 );
 120     }
 121     if (flags->Sign && !flags->PadZero && r >= 0)
 122     {
 123         APICHAR ch = flags->Sign;
 124         r = FUNC_NAME(pf_output_string)( out, &ch, 1 );
 125     }
 126     return r;
 127 }
 128
 129 static int FUNC_NAME(pf_fill_right)( FUNC_NAME(pf_output) *out, int len, pf_flags *flags )
 130 {
 131     int i, r = 0;
 132     APICHAR ch = ' ';
 133
 134     if (!flags->LeftAlign) return 0;
 135
 136     for (i = 0; i < flags->FieldLength - len && r >= 0; i++)
 137         r = FUNC_NAME(pf_output_string)( out, &ch, 1 );
 138     return r;
 139 }
 140
 141 static int FUNC_NAME(pf_output_format_wstr)( FUNC_NAME(pf_output) *out, const wchar_t *str, int len,
 142                                              pf_flags *flags )
 143 {
 144     int r = 0;
 145
 146     if (len < 0)
 147     {
 148         /* Do not search past the length specified by the precision. */
 149         if (flags->Precision >= 0)
 150         {
 151             for (len = 0; len < flags->Precision; len++) if (!str[len]) break;
 152         }
 153         else len = wcslen( str );
 154     }
 155
 156     if (flags->Precision >= 0 && flags->Precision < len) len = flags->Precision;
 157
 158     r = FUNC_NAME(pf_fill_left)( out, len, flags );
 159     if (r >= 0) r = FUNC_NAME(pf_output_wstr)( out, str, len );
 160     if (r >= 0) r = FUNC_NAME(pf_fill_right)( out, len, flags );
 161     return r;
 162 }
 163
 164 static int FUNC_NAME(pf_output_format_str)( FUNC_NAME(pf_output) *out, const char *str, int len,
 165                                             pf_flags *flags )
 166 {
 167     int r = 0;
 168
 169     if (len < 0)
 170     {
 171         /* Do not search past the length specified by the precision. */
 172         if (flags->Precision >= 0)
 173         {
 174             for (len = 0; len < flags->Precision; len++) if (!str[len]) break;
 175         }
 176         else len = strlen(str);
 177     }
 178
 179     if (flags->Precision >= 0 && flags->Precision < len) len = flags->Precision;
 180
 181     r = FUNC_NAME(pf_fill_left)( out, len, flags );
 182     if (r >= 0) r = FUNC_NAME(pf_output_str)( out, str, len );
 183     if (r >= 0) r = FUNC_NAME(pf_fill_right)( out, len, flags );
 184     return r;
 185 }
 186
 187 static int FUNC_NAME(pf_output_format)( FUNC_NAME(pf_output) *out, const APICHAR *str, int len,
 188                                         pf_flags *flags )
 189 {
 190 #ifdef PRINTF_WIDE
 191     return FUNC_NAME(pf_output_format_wstr)( out, str, len, flags );
 192 #else
 193     return FUNC_NAME(pf_output_format_str)( out, str, len, flags );
 194 #endif
 195 }
 196
 197 static int FUNC_NAME(pf_handle_string_format)( FUNC_NAME(pf_output) *out, const void* str, int len,
 198                                                pf_flags *flags, BOOL inverted )
 199 {
 200      if (str == NULL)  /* catch NULL pointer */
 201          return FUNC_NAME(pf_output_format)( out, APISTR("(null)"), -1, flags );
 202
 203      /* prefixes take priority over %c,%s vs. %C,%S, so we handle them first */
 204     if (flags->WideString || flags->IntegerLength == LEN_LONG)
 205         return FUNC_NAME(pf_output_format_wstr)( out, str, len, flags );
 206     if (flags->IntegerLength == LEN_SHORT)
 207         return FUNC_NAME(pf_output_format_str)( out, str, len, flags );
 208
 209     /* %s,%c ->  chars in ansi functions & wchars in unicode
 210      * %S,%C -> wchars in ansi functions &  chars in unicode */
 211     if (!inverted) return FUNC_NAME(pf_output_format)( out, str, len, flags);
 212 #ifdef PRINTF_WIDE
 213     return FUNC_NAME(pf_output_format_str)( out, str, len, flags );
 214 #else
 215     return FUNC_NAME(pf_output_format_wstr)( out, str, len, flags );
 216 #endif
 217 }
 218
 219 /* pf_integer_conv:  prints x to buf, including alternate formats and
 220    additional precision digits, but not field characters or the sign */
 221 static void FUNC_NAME(pf_integer_conv)( APICHAR *buf, pf_flags *flags, LONGLONG x )
 222 {
 223     unsigned int base;
 224     const APICHAR *digits;
 225     int i, j, k;
 226
 227     if( flags->Format == 'o' )
 228         base = 8;
 229     else if( flags->Format == 'x' || flags->Format == 'X' )
 230         base = 16;
 231     else
 232         base = 10;
 233
 234     if( flags->Format == 'X' )
 235         digits = APISTR("0123456789ABCDEFX");
 236     else
 237         digits = APISTR("0123456789abcdefx");
 238
 239     if( x < 0 && ( flags->Format == 'd' || flags->Format == 'i' ) )
 240     {
 241         x = -x;
 242         flags->Sign = '-';
 243     }
 244
 245     i = 0;
 246     if( x == 0 )
 247     {
 248         flags->Alternate = FALSE;
 249         if( flags->Precision )
 250             buf[i++] = '0';
 251     }
 252     else
 253         while( x != 0 )
 254         {
 255             j = (ULONGLONG) x % base;
 256             x = (ULONGLONG) x / base;
 257             buf[i++] = digits[j];
 258         }
 259     k = flags->Precision - i;
 260     while( k-- > 0 )
 261         buf[i++] = '0';
 262     if( flags->Alternate )
 263     {
 264         if( base == 16 )
 265         {
 266             buf[i++] = digits[16];
 267             buf[i++] = '0';
 268         }
 269         else if( base == 8 && buf[i-1] != '0' )
 270             buf[i++] = '0';
 271     }
 272
 273     /* Adjust precision so pf_fill won't truncate the number later */
 274     flags->Precision = i;
 275
 276     buf[i] = '\0';
 277     j = 0;
 278     while(--i > j) {
 279         APICHAR tmp = buf[j];
 280         buf[j] = buf[i];
 281         buf[i] = tmp;
 282         j++;
 283     }
 284 }
 285
 286 static int FUNC_NAME(pf_vsnprintf)( FUNC_NAME(pf_output) *out, const APICHAR *format, __ms_va_list valist )
 287 {
 288     int r;
 289     const APICHAR *q, *p = format;
 290     pf_flags flags;
 291
 292     while (*p)
 293     {
 294         for (q = p; *q; q++) if (*q == '%') break;
 295
 296         if (q != p)
 297         {
 298             r = FUNC_NAME(pf_output_string)(out, p, q - p);
 299             if (r < 0) return r;
 300             p = q;
 301             if (!*p) break;
 302         }
 303         p++;
 304
 305         /* output a single % character */
 306         if( *p == '%' )
 307         {
 308             r = FUNC_NAME(pf_output_string)(out, p++, 1);
 309             if( r<0 )
 310                 return r;
 311             continue;
 312         }
 313
 314         /* parse the flags */
 315         memset( &flags, 0, sizeof flags );
 316         while (*p)
 317         {
 318             if( *p == '+' || *p == ' ' )
 319             {
 320                 if ( flags.Sign != '+' )
 321                     flags.Sign = *p;
 322             }
 323             else if( *p == '-' )
 324                 flags.LeftAlign = TRUE;
 325             else if( *p == '0' )
 326                 flags.PadZero = TRUE;
 327             else if( *p == '#' )
 328                 flags.Alternate = TRUE;
 329             else
 330                 break;
 331             p++;
 332         }
 333
 334         /* deal with the field width specifier */
 335         flags.FieldLength = 0;
 336         if( *p == '*' )
 337         {
 338             flags.FieldLength = va_arg( valist, int );
 339             if (flags.FieldLength < 0)
 340             {
 341                 flags.LeftAlign = TRUE;
 342                 flags.FieldLength = -flags.FieldLength;
 343             }
 344             p++;
 345         }
 346         else while (*p >= '0' && *p <= '9')
 347         {
 348             flags.FieldLength *= 10;
 349             flags.FieldLength += *p++ - '0';
 350         }
 351
 352         /* deal with precision */
 353         flags.Precision = -1;
 354         if( *p == '.' )
 355         {
 356             flags.Precision = 0;
 357             p++;
 358             if( *p == '*' )
 359             {
 360                 flags.Precision = va_arg( valist, int );
 361                 p++;
 362             }
 363             else while (*p >= '0' && *p <= '9')
 364             {
 365                 flags.Precision *= 10;
 366                 flags.Precision += *p++ - '0';
 367             }
 368         }
 369
 370         /* deal with integer width modifier */
 371         while( *p )
 372         {
 373             if (*p == 'l' && *(p+1) == 'l')
 374             {
 375                 flags.IntegerDouble = TRUE;
 376                 p += 2;
 377             }
 378             else if( *p == 'l' )
 379             {
 380                 flags.IntegerLength = LEN_LONG;
 381                 p++;
 382             }
 383             else if( *p == 'L' )
 384             {
 385                 p++;
 386             }
 387             else if( *p == 'h')
 388             {
 389                 flags.IntegerLength = LEN_SHORT;
 390                 p++;
 391             }
 392             else if( *p == 'I' )
 393             {
 394                 if( *(p+1) == '6' && *(p+2) == '4' )
 395                 {
 396                     flags.IntegerDouble = TRUE;
 397                     p += 3;
 398                 }
 399                 else if( *(p+1) == '3' && *(p+2) == '2' )
 400                     p += 3;
 401                 else if( p[1] && strchr( "diouxX", p[1] ) )
 402                 {
 403                     if( sizeof(void *) == 8 )
 404                         flags.IntegerDouble = TRUE;
 405                     p++;
 406                 }
 407                 else if ((p[1] >= '0' && p[1] <= '9') || !p[1])
 408                     break;
 409                 else
 410                     p++;
 411             }
 412             else if( *p == 'w' )
 413             {
 414                 flags.WideString = TRUE;
 415                 p++;
 416             }
 417             else if ((*p == 'z' || *p == 't') && p[1] && strchr("diouxX", p[1]))
 418             {
 419                 flags.IntegerNative = TRUE;
 420                 p++;
 421             }
 422             else if (*p == 'j')
 423             {
 424                 flags.IntegerDouble = TRUE;
 425                 p++;
 426             }
 427             else if( *p == 'F' )
 428                 p++; /* ignore */
 429             else
 430                 break;
 431         }
 432
 433         flags.Format = *p;
 434         r = 0;
 435
 436         if (flags.Format == '$')
 437         {
 438             FIXME("Positional parameters are not supported (%s)\n", FUNC_NAME(wine_dbgstr)(format));
 439             return -1;
 440         }
 441         /* output a string */
 442         if(  flags.Format == 's' || flags.Format == 'S' )
 443             r = FUNC_NAME(pf_handle_string_format)( out, va_arg(valist, const void*), -1,
 444                                                     &flags, (flags.Format == 'S') );
 445
 446         /* output a single character */
 447         else if( flags.Format == 'c' || flags.Format == 'C' )
 448         {
 449             APICHAR ch = (APICHAR)va_arg( valist, int );
 450             r = FUNC_NAME(pf_handle_string_format)( out, &ch, 1, &flags, (flags.Format == 'C') );
 451         }
 452
 453         /* output a pointer */
 454         else if( flags.Format == 'p' )
 455         {
 456             APICHAR pointer[32];
 457             void *ptr = va_arg( valist, void * );
 458             int prec = flags.Precision;
 459             flags.Format = 'X';
 460             flags.PadZero = TRUE;
 461             flags.Precision = 2*sizeof(void*);
 462             FUNC_NAME(pf_integer_conv)( pointer, &flags, (ULONG_PTR)ptr );
 463             flags.PadZero = FALSE;
 464             flags.Precision = prec;
 465             r = FUNC_NAME(pf_output_format)( out, pointer, -1, &flags );
 466         }
 467
 468         /* deal with %n */
 469         else if( flags.Format == 'n' )
 470         {
 471             int *x = va_arg(valist, int *);
 472             *x = out->used;
 473         }
 474         else if( flags.Format && strchr("diouxX", flags.Format ))
 475         {
 476             APICHAR number[40], *x = number;
 477             int max_len;
 478
 479             /* 0 padding is added after '0x' if Alternate flag is in use */
 480             if((flags.Format=='x' || flags.Format=='X') && flags.PadZero && flags.Alternate
 481                     && !flags.LeftAlign && flags.Precision<flags.FieldLength-2)
 482                 flags.Precision = flags.FieldLength - 2;
 483
 484             max_len = (flags.FieldLength>flags.Precision ? flags.FieldLength : flags.Precision) + 10;
 485             if(max_len > ARRAY_SIZE(number))
 486                 if (!(x = RtlAllocateHeap( GetProcessHeap(), 0, max_len * sizeof(*x) ))) return -1;
 487
 488             if(flags.IntegerDouble || (flags.IntegerNative && sizeof(void*) == 8))
 489                 FUNC_NAME(pf_integer_conv)( x, &flags, va_arg(valist, LONGLONG) );
 490             else if(flags.Format=='d' || flags.Format=='i')
 491                 FUNC_NAME(pf_integer_conv)( x, &flags, flags.IntegerLength != LEN_SHORT ?
 492                                             va_arg(valist, int) : (short)va_arg(valist, int) );
 493             else
 494                 FUNC_NAME(pf_integer_conv)( x, &flags, flags.IntegerLength != LEN_SHORT ?
 495                                  (unsigned int)va_arg(valist, int) : (unsigned short)va_arg(valist, int) );
 496
 497             r = FUNC_NAME(pf_output_format)( out, x, -1, &flags );
 498             if( x != number )
 499                 RtlFreeHeap( GetProcessHeap(), 0, x );
 500         }
 501         else
 502             continue;
 503
 504         if( r<0 )
 505             return r;
 506         p++;
 507     }
 508     return out->used;
 509 }
 510
 511 #undef APICHAR
 512 #undef APISTR
 513 #undef FUNC_NAME