| /* | |
| * ntdll printf functions | |
| * | |
| * Copyright 1999, 2009 Alexandre Julliard | |
| * Copyright 2000 Jon Griffiths | |
| * | |
| * This library is free software; you can redistribute it and/or | |
| * modify it under the terms of the GNU Lesser General Public | |
| * License as published by the Free Software Foundation; either | |
| * version 2.1 of the License, or (at your option) any later version. | |
| * | |
| * This library is distributed in the hope that it will be useful, | |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
| * Lesser General Public License for more details. | |
| * | |
| * You should have received a copy of the GNU Lesser General Public | |
| * License along with this library; if not, write to the Free Software | |
| * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA | |
| */ | |
| #include "config.h" | |
| #include "wine/port.h" | |
| #include <assert.h> | |
| #include <ctype.h> | |
| #include <stdarg.h> | |
| #include <stdlib.h> | |
| #include <stdio.h> | |
| #include <string.h> | |
| #include "windef.h" | |
| #include "winternl.h" | |
| #include "ntdll_misc.h" | |
| #include "wine/unicode.h" | |
| #include "wine/debug.h" | |
| WINE_DEFAULT_DEBUG_CHANNEL(ntdll); | |
| static const SIZE_T size_max = ~(SIZE_T)0 >> 1; | |
| /* FIXME: convert sizes to SIZE_T etc. */ | |
| typedef struct pf_output_t | |
| { | |
| int used; | |
| int len; | |
| BOOL unicode; | |
| union { | |
| LPWSTR W; | |
| LPSTR A; | |
| } buf; | |
| } pf_output; | |
| typedef struct pf_flags_t | |
| { | |
| char Sign, LeftAlign, Alternate, PadZero; | |
| int FieldLength, Precision; | |
| char IntegerLength, IntegerDouble; | |
| char WideString; | |
| char Format; | |
| } pf_flags; | |
| /* | |
| * writes a string of characters to the output | |
| * returns -1 if the string doesn't fit in the output buffer | |
| * return the length of the string if all characters were written | |
| */ | |
| static inline int pf_output_stringW( pf_output *out, LPCWSTR str, int len ) | |
| { | |
| int space = out->len - out->used; | |
| if( len < 0 ) | |
| len = strlenW( str ); | |
| if( out->unicode ) | |
| { | |
| LPWSTR p = out->buf.W + out->used; | |
| out->used += len; | |
| if (!out->buf.W) | |
| return len; | |
| if( space >= len ) | |
| { | |
| memcpy( p, str, len*sizeof(WCHAR) ); | |
| return len; | |
| } | |
| if( space > 0 ) | |
| memcpy( p, str, space*sizeof(WCHAR) ); | |
| } | |
| else | |
| { | |
| LPSTR p = out->buf.A + out->used; | |
| ULONG n; | |
| RtlUnicodeToMultiByteSize( &n, str, len * sizeof(WCHAR) ); | |
| out->used += n; | |
| if (!out->buf.A) | |
| return len; | |
| if( space >= n ) | |
| { | |
| RtlUnicodeToMultiByteN( p, n, NULL, str, len * sizeof(WCHAR) ); | |
| return len; | |
| } | |
| if (space > 0) RtlUnicodeToMultiByteN( p, space, NULL, str, len * sizeof(WCHAR) ); | |
| } | |
| return -1; | |
| } | |
| static inline int pf_output_stringA( pf_output *out, LPCSTR str, int len ) | |
| { | |
| int space = out->len - out->used; | |
| if( len < 0 ) | |
| len = strlen( str ); | |
| if( !out->unicode ) | |
| { | |
| LPSTR p = out->buf.A + out->used; | |
| out->used += len; | |
| if (!out->buf.A) | |
| return len; | |
| if( space >= len ) | |
| { | |
| memcpy( p, str, len ); | |
| return len; | |
| } | |
| if( space > 0 ) | |
| memcpy( p, str, space ); | |
| } | |
| else | |
| { | |
| LPWSTR p = out->buf.W + out->used; | |
| ULONG n; | |
| RtlMultiByteToUnicodeSize( &n, str, len ); | |
| out->used += n / sizeof(WCHAR); | |
| if (!out->buf.W) | |
| return len; | |
| if (space >= n / sizeof(WCHAR)) | |
| { | |
| RtlMultiByteToUnicodeN( p, n, NULL, str, len ); | |
| return len; | |
| } | |
| if (space > 0) RtlMultiByteToUnicodeN( p, space * sizeof(WCHAR), NULL, str, len ); | |
| } | |
| return -1; | |
| } | |
| /* pf_fill: takes care of signs, alignment, zero and field padding */ | |
| static inline int pf_fill( pf_output *out, int len, pf_flags *flags, char left ) | |
| { | |
| int i, r = 0; | |
| if( flags->Sign && !( flags->Format == 'd' || flags->Format == 'i' ) ) | |
| flags->Sign = 0; | |
| if( left && flags->Sign ) | |
| { | |
| flags->FieldLength--; | |
| if( flags->PadZero ) | |
| r = pf_output_stringA( out, &flags->Sign, 1 ); | |
| } | |
| if( ( !left && flags->LeftAlign ) || | |
| ( left && !flags->LeftAlign )) | |
| { | |
| for( i=0; (i<(flags->FieldLength-len)) && (r>=0); i++ ) | |
| { | |
| if( left && flags->PadZero ) | |
| r = pf_output_stringA( out, "0", 1 ); | |
| else | |
| r = pf_output_stringA( out, " ", 1 ); | |
| } | |
| } | |
| if (left && flags->Sign && !flags->PadZero && r >= 0) | |
| r = pf_output_stringA( out, &flags->Sign, 1 ); | |
| return r; | |
| } | |
| static inline int pf_output_format_W( pf_output *out, LPCWSTR str, | |
| int len, pf_flags *flags ) | |
| { | |
| int r = 0; | |
| if( len < 0 ) | |
| len = strlenW( str ); | |
| if (flags->Precision >= 0 && flags->Precision < len) | |
| len = flags->Precision; | |
| r = pf_fill( out, len, flags, 1 ); | |
| if( r>=0 ) | |
| r = pf_output_stringW( out, str, len ); | |
| if( r>=0 ) | |
| r = pf_fill( out, len, flags, 0 ); | |
| return r; | |
| } | |
| static inline int pf_output_format_A( pf_output *out, LPCSTR str, | |
| int len, pf_flags *flags ) | |
| { | |
| int r = 0; | |
| if( len < 0 ) | |
| len = strlen( str ); | |
| if (flags->Precision >= 0 && flags->Precision < len) | |
| len = flags->Precision; | |
| r = pf_fill( out, len, flags, 1 ); | |
| if( r>=0 ) | |
| r = pf_output_stringA( out, str, len ); | |
| if( r>=0 ) | |
| r = pf_fill( out, len, flags, 0 ); | |
| return r; | |
| } | |
| static int pf_handle_string_format( pf_output *out, const void* str, int len, | |
| pf_flags *flags, BOOL capital_letter) | |
| { | |
| if(str == NULL) /* catch NULL pointer */ | |
| return pf_output_format_A( out, "(null)", -1, flags); | |
| /* prefixes take priority over %c,%s vs. %C,%S, so we handle them first */ | |
| if(flags->WideString || flags->IntegerLength == 'l') | |
| return pf_output_format_W( out, str, len, flags); | |
| if(flags->IntegerLength == 'h') | |
| return pf_output_format_A( out, str, len, flags); | |
| /* %s,%c -> chars in ansi functions & wchars in unicode | |
| * %S,%C -> wchars in ansi functions & chars in unicode */ | |
| if( capital_letter == out->unicode) /* either both TRUE or both FALSE */ | |
| return pf_output_format_A( out, str, len, flags); | |
| else | |
| return pf_output_format_W( out, str, len, flags); | |
| } | |
| static inline BOOL pf_is_integer_format( char fmt ) | |
| { | |
| static const char float_fmts[] = "diouxX"; | |
| if (!fmt) | |
| return FALSE; | |
| return strchr( float_fmts, fmt ) != 0; | |
| } | |
| static inline BOOL pf_is_double_format( char fmt ) | |
| { | |
| static const char float_fmts[] = "aeEfgG"; | |
| if (!fmt) | |
| return FALSE; | |
| return strchr( float_fmts, fmt ) != 0; | |
| } | |
| static inline BOOL pf_is_valid_format( char fmt ) | |
| { | |
| static const char float_fmts[] = "acCdeEfgGinouxX"; | |
| if (!fmt) | |
| return FALSE; | |
| return strchr( float_fmts, fmt ) != 0; | |
| } | |
| static void pf_rebuild_format_string( char *p, pf_flags *flags ) | |
| { | |
| *p++ = '%'; | |
| if( flags->Sign ) | |
| *p++ = flags->Sign; | |
| if( flags->LeftAlign ) | |
| *p++ = flags->LeftAlign; | |
| if( flags->Alternate ) | |
| *p++ = flags->Alternate; | |
| if( flags->PadZero ) | |
| *p++ = flags->PadZero; | |
| if( flags->FieldLength ) | |
| { | |
| sprintf(p, "%d", flags->FieldLength); | |
| p += strlen(p); | |
| } | |
| if( flags->Precision >= 0 ) | |
| { | |
| sprintf(p, ".%d", flags->Precision); | |
| p += strlen(p); | |
| } | |
| *p++ = flags->Format; | |
| *p++ = 0; | |
| } | |
| /* pf_integer_conv: prints x to buf, including alternate formats and | |
| additional precision digits, but not field characters or the sign */ | |
| static void pf_integer_conv( char *buf, int buf_len, pf_flags *flags, | |
| LONGLONG x ) | |
| { | |
| unsigned int base; | |
| const char *digits; | |
| int i, j, k; | |
| char number[40], *tmp = number; | |
| if( buf_len > sizeof number ) | |
| { | |
| if (!(tmp = RtlAllocateHeap( GetProcessHeap(), 0, buf_len ))) | |
| { | |
| buf[0] = '\0'; | |
| return; | |
| } | |
| } | |
| base = 10; | |
| if( flags->Format == 'o' ) | |
| base = 8; | |
| else if( flags->Format == 'x' || flags->Format == 'X' ) | |
| base = 16; | |
| if( flags->Format == 'X' ) | |
| digits = "0123456789ABCDEFX"; | |
| else | |
| digits = "0123456789abcdefx"; | |
| if( x < 0 && ( flags->Format == 'd' || flags->Format == 'i' ) ) | |
| { | |
| x = -x; | |
| flags->Sign = '-'; | |
| } | |
| /* Do conversion (backwards) */ | |
| i = 0; | |
| if( x == 0 && flags->Precision ) | |
| tmp[i++] = '0'; | |
| else | |
| while( x != 0 ) | |
| { | |
| j = (ULONGLONG) x % base; | |
| x = (ULONGLONG) x / base; | |
| tmp[i++] = digits[j]; | |
| } | |
| k = flags->Precision - i; | |
| while( k-- > 0 ) | |
| tmp[i++] = '0'; | |
| if( flags->Alternate ) | |
| { | |
| if( base == 16 ) | |
| { | |
| tmp[i++] = digits[16]; | |
| tmp[i++] = '0'; | |
| } | |
| else if( base == 8 && tmp[i-1] != '0' ) | |
| tmp[i++] = '0'; | |
| } | |
| /* Reverse for buf */ | |
| j = 0; | |
| while( i-- > 0 ) | |
| buf[j++] = tmp[i]; | |
| buf[j] = '\0'; | |
| /* Adjust precision so pf_fill won't truncate the number later */ | |
| flags->Precision = strlen( buf ); | |
| if( tmp != number ) | |
| RtlFreeHeap( GetProcessHeap(), 0, tmp ); | |
| return; | |
| } | |
| /* pf_fixup_exponent: convert a string containing a 2 digit exponent | |
| to 3 digits, accounting for padding, in place. Needed to match | |
| the native printf's which always use 3 digits. */ | |
| static void pf_fixup_exponent( char *buf ) | |
| { | |
| char* tmp = buf; | |
| while (tmp[0] && toupper(tmp[0]) != 'E') | |
| tmp++; | |
| if (tmp[0] && (tmp[1] == '+' || tmp[1] == '-') && | |
| isdigit(tmp[2]) && isdigit(tmp[3])) | |
| { | |
| char final; | |
| if (isdigit(tmp[4])) | |
| return; /* Exponent already 3 digits */ | |
| /* We have a 2 digit exponent. Prepend '0' to make it 3 */ | |
| tmp += 2; | |
| final = tmp[2]; | |
| tmp[2] = tmp[1]; | |
| tmp[1] = tmp[0]; | |
| tmp[0] = '0'; | |
| if (final == '\0') | |
| { | |
| /* We didn't expand into trailing space, so this string isn't left | |
| * justified. Terminate the string and strip a ' ' at the start of | |
| * the string if there is one (as there may be if the string is | |
| * right justified). | |
| */ | |
| tmp[3] = '\0'; | |
| if (buf[0] == ' ') | |
| memmove(buf, buf + 1, (tmp - buf) + 3); | |
| } | |
| /* Otherwise, we expanded into trailing space -> nothing to do */ | |
| } | |
| } | |
| static inline BOOL isDigit(WCHAR c) | |
| { | |
| return c >= '0' && c <= '9'; | |
| } | |
| /********************************************************************* | |
| * pf_vsnprintf (INTERNAL) | |
| * | |
| * implements both A and W vsnprintf functions | |
| */ | |
| static int pf_vsnprintf( pf_output *out, const WCHAR *format, __ms_va_list valist ) | |
| { | |
| int r; | |
| LPCWSTR q, p = format; | |
| pf_flags flags; | |
| TRACE("format is %s\n",debugstr_w(format)); | |
| while (*p) | |
| { | |
| q = strchrW( p, '%' ); | |
| /* there are no % characters left: output the rest of the string */ | |
| if( !q ) | |
| { | |
| r = pf_output_stringW(out, p, -1); | |
| if( r<0 ) | |
| return r; | |
| p += r; | |
| continue; | |
| } | |
| /* there are characters before the %: output them */ | |
| if( q != p ) | |
| { | |
| r = pf_output_stringW(out, p, q - p); | |
| if( r<0 ) | |
| return r; | |
| p = q; | |
| } | |
| /* we must be at a % now, skip over it */ | |
| assert( *p == '%' ); | |
| p++; | |
| /* output a single % character */ | |
| if( *p == '%' ) | |
| { | |
| r = pf_output_stringW(out, p++, 1); | |
| if( r<0 ) | |
| return r; | |
| continue; | |
| } | |
| /* parse the flags */ | |
| memset( &flags, 0, sizeof flags ); | |
| while (*p) | |
| { | |
| if( *p == '+' || *p == ' ' ) | |
| { | |
| if ( flags.Sign != '+' ) | |
| flags.Sign = *p; | |
| } | |
| else if( *p == '-' ) | |
| flags.LeftAlign = *p; | |
| else if( *p == '0' ) | |
| flags.PadZero = *p; | |
| else if( *p == '#' ) | |
| flags.Alternate = *p; | |
| else | |
| break; | |
| p++; | |
| } | |
| /* deal with the field width specifier */ | |
| flags.FieldLength = 0; | |
| if( *p == '*' ) | |
| { | |
| flags.FieldLength = va_arg( valist, int ); | |
| if (flags.FieldLength < 0) | |
| { | |
| flags.LeftAlign = '-'; | |
| flags.FieldLength = -flags.FieldLength; | |
| } | |
| p++; | |
| } | |
| else while( isDigit(*p) ) | |
| { | |
| flags.FieldLength *= 10; | |
| flags.FieldLength += *p++ - '0'; | |
| } | |
| /* deal with precision */ | |
| flags.Precision = -1; | |
| if( *p == '.' ) | |
| { | |
| flags.Precision = 0; | |
| p++; | |
| if( *p == '*' ) | |
| { | |
| flags.Precision = va_arg( valist, int ); | |
| p++; | |
| } | |
| else while( isDigit(*p) ) | |
| { | |
| flags.Precision *= 10; | |
| flags.Precision += *p++ - '0'; | |
| } | |
| } | |
| /* deal with integer width modifier */ | |
| while( *p ) | |
| { | |
| if( *p == 'h' || *p == 'l' || *p == 'L' ) | |
| { | |
| flags.IntegerLength = *p; | |
| p++; | |
| } | |
| else if( *p == 'I' ) | |
| { | |
| if( *(p+1) == '6' && *(p+2) == '4' ) | |
| { | |
| flags.IntegerDouble++; | |
| p += 3; | |
| } | |
| else if( *(p+1) == '3' && *(p+2) == '2' ) | |
| p += 3; | |
| else if( isDigit(*(p+1)) || *(p+1) == 0 ) | |
| break; | |
| else | |
| p++; | |
| } | |
| else if( *p == 'w' ) | |
| flags.WideString = *p++; | |
| else if( *p == 'F' ) | |
| p++; /* ignore */ | |
| else | |
| break; | |
| } | |
| flags.Format = *p; | |
| r = 0; | |
| if (flags.Format == '$') | |
| { | |
| FIXME("Positional parameters are not supported (%s)\n", wine_dbgstr_w(format)); | |
| return -1; | |
| } | |
| /* output a string */ | |
| if( flags.Format == 's' || flags.Format == 'S' ) | |
| r = pf_handle_string_format( out, va_arg(valist, const void*), -1, | |
| &flags, (flags.Format == 'S') ); | |
| /* output a single character */ | |
| else if( flags.Format == 'c' || flags.Format == 'C' ) | |
| { | |
| INT ch = va_arg( valist, int ); | |
| r = pf_handle_string_format( out, &ch, 1, &flags, (flags.Format == 'C') ); | |
| } | |
| /* output a pointer */ | |
| else if( flags.Format == 'p' ) | |
| { | |
| char pointer[32]; | |
| void *ptr = va_arg( valist, void * ); | |
| flags.PadZero = 0; | |
| if( flags.Alternate ) | |
| sprintf(pointer, "0X%0*lX", 2 * (int)sizeof(ptr), (ULONG_PTR)ptr); | |
| else | |
| sprintf(pointer, "%0*lX", 2 * (int)sizeof(ptr), (ULONG_PTR)ptr); | |
| r = pf_output_format_A( out, pointer, -1, &flags ); | |
| } | |
| /* deal with %n */ | |
| else if( flags.Format == 'n' ) | |
| { | |
| int *x = va_arg(valist, int *); | |
| *x = out->used; | |
| } | |
| /* deal with 64-bit integers */ | |
| else if( pf_is_integer_format( flags.Format ) && flags.IntegerDouble ) | |
| { | |
| char number[40], *x = number; | |
| /* Estimate largest possible required buffer size: | |
| * Chooses the larger of the field or precision | |
| * Includes extra bytes: 1 byte for null, 1 byte for sign, | |
| 4 bytes for exponent, 2 bytes for alternate formats, 1 byte | |
| for a decimal, and 1 byte for an additional float digit. */ | |
| int x_len = ((flags.FieldLength > flags.Precision) ? | |
| flags.FieldLength : flags.Precision) + 10; | |
| if( x_len >= sizeof number) | |
| if (!(x = RtlAllocateHeap( GetProcessHeap(), 0, x_len ))) | |
| return -1; | |
| pf_integer_conv( x, x_len, &flags, va_arg(valist, LONGLONG) ); | |
| r = pf_output_format_A( out, x, -1, &flags ); | |
| if( x != number ) | |
| RtlFreeHeap( GetProcessHeap(), 0, x ); | |
| } | |
| /* deal with integers and floats using libc's printf */ | |
| else if( pf_is_valid_format( flags.Format ) ) | |
| { | |
| char fmt[20], number[40], *x = number; | |
| /* Estimate largest possible required buffer size: | |
| * Chooses the larger of the field or precision | |
| * Includes extra bytes: 1 byte for null, 1 byte for sign, | |
| 4 bytes for exponent, 2 bytes for alternate formats, 1 byte | |
| for a decimal, and 1 byte for an additional float digit. */ | |
| int x_len = ((flags.FieldLength > flags.Precision) ? | |
| flags.FieldLength : flags.Precision) + 10; | |
| if( x_len >= sizeof number) | |
| if (!(x = RtlAllocateHeap( GetProcessHeap(), 0, x_len ))) | |
| return -1; | |
| pf_rebuild_format_string( fmt, &flags ); | |
| if( pf_is_double_format( flags.Format ) ) | |
| { | |
| sprintf( x, fmt, va_arg(valist, double) ); | |
| if (toupper(flags.Format) == 'E' || toupper(flags.Format) == 'G') | |
| pf_fixup_exponent( x ); | |
| } | |
| else | |
| sprintf( x, fmt, va_arg(valist, int) ); | |
| r = pf_output_stringA( out, x, -1 ); | |
| if( x != number ) | |
| RtlFreeHeap( GetProcessHeap(), 0, x ); | |
| } | |
| else | |
| continue; | |
| if( r<0 ) | |
| return r; | |
| p++; | |
| } | |
| /* check we reached the end, and null terminate the string */ | |
| assert( *p == 0 ); | |
| pf_output_stringW( out, p, 1 ); | |
| return out->used - 1; | |
| } | |
| /********************************************************************* | |
| * _vsnprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL__vsnprintf( char *str, SIZE_T len, const char *format, __ms_va_list args ) | |
| { | |
| DWORD sz; | |
| LPWSTR formatW = NULL; | |
| pf_output out; | |
| int r; | |
| out.unicode = FALSE; | |
| out.buf.A = str; | |
| out.used = 0; | |
| out.len = len; | |
| if (format) | |
| { | |
| RtlMultiByteToUnicodeSize( &sz, format, strlen(format) + 1 ); | |
| if (!(formatW = RtlAllocateHeap( GetProcessHeap(), 0, sz ))) return -1; | |
| RtlMultiByteToUnicodeN( formatW, sz, NULL, format, strlen(format) + 1 ); | |
| } | |
| r = pf_vsnprintf( &out, formatW, args ); | |
| RtlFreeHeap( GetProcessHeap(), 0, formatW ); | |
| return r; | |
| } | |
| /*********************************************************************** | |
| * _vsnwprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL__vsnwprintf( WCHAR *str, SIZE_T len, const WCHAR *format, __ms_va_list args ) | |
| { | |
| pf_output out; | |
| out.unicode = TRUE; | |
| out.buf.W = str; | |
| out.used = 0; | |
| out.len = len; | |
| return pf_vsnprintf( &out, format, args ); | |
| } | |
| /********************************************************************* | |
| * _snprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL__snprintf( char *str, SIZE_T len, const char *format, ... ) | |
| { | |
| int ret; | |
| __ms_va_list valist; | |
| __ms_va_start( valist, format ); | |
| ret = NTDLL__vsnprintf( str, len, format, valist ); | |
| __ms_va_end( valist ); | |
| return ret; | |
| } | |
| /*********************************************************************** | |
| * _snwprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL__snwprintf( WCHAR *str, SIZE_T len, const WCHAR *format, ... ) | |
| { | |
| int ret; | |
| __ms_va_list valist; | |
| __ms_va_start(valist, format); | |
| ret = NTDLL__vsnwprintf( str, len, format, valist ); | |
| __ms_va_end(valist); | |
| return ret; | |
| } | |
| /********************************************************************* | |
| * vsprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL_vsprintf( char *str, const char *format, __ms_va_list args ) | |
| { | |
| return NTDLL__vsnprintf( str, size_max, format, args ); | |
| } | |
| /********************************************************************* | |
| * sprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL_sprintf( char *str, const char *format, ... ) | |
| { | |
| int ret; | |
| __ms_va_list valist; | |
| __ms_va_start( valist, format ); | |
| ret = NTDLL__vsnprintf( str, size_max, format, valist ); | |
| __ms_va_end( valist ); | |
| return ret; | |
| } | |
| /*********************************************************************** | |
| * swprintf (NTDLL.@) | |
| */ | |
| int CDECL NTDLL_swprintf( WCHAR *str, const WCHAR *format, ... ) | |
| { | |
| int ret; | |
| __ms_va_list valist; | |
| __ms_va_start(valist, format); | |
| ret = NTDLL__vsnwprintf( str, size_max, format, valist ); | |
| __ms_va_end(valist); | |
| return ret; | |
| } |