msvcrt: Do not treat %ll as an IntegerDouble in printf.
[wine] / dlls / msvcrt / wcs.c
1 /*
2  * msvcrt.dll wide-char functions
3  *
4  * Copyright 1999 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20  */
21 #include <limits.h>
22 #include <stdio.h>
23 #include <math.h>
24 #include <assert.h>
25 #include "msvcrt.h"
26 #include "winnls.h"
27 #include "wine/unicode.h"
28 #include "wine/debug.h"
29
30 WINE_DEFAULT_DEBUG_CHANNEL(msvcrt);
31
32
33 /* INTERNAL: MSVCRT_malloc() based wstrndup */
34 MSVCRT_wchar_t* msvcrt_wstrndup(const MSVCRT_wchar_t *buf, unsigned int size)
35 {
36   MSVCRT_wchar_t* ret;
37   unsigned int len = strlenW(buf), max_len;
38
39   max_len = size <= len? size : len + 1;
40
41   ret = MSVCRT_malloc(max_len * sizeof (MSVCRT_wchar_t));
42   if (ret)
43   {
44     memcpy(ret,buf,max_len * sizeof (MSVCRT_wchar_t));
45     ret[max_len] = 0;
46   }
47   return ret;
48 }
49
50 /*********************************************************************
51  *              _wcsdup (MSVCRT.@)
52  */
53 MSVCRT_wchar_t* _wcsdup( const MSVCRT_wchar_t* str )
54 {
55   MSVCRT_wchar_t* ret = NULL;
56   if (str)
57   {
58     int size = (strlenW(str) + 1) * sizeof(MSVCRT_wchar_t);
59     ret = MSVCRT_malloc( size );
60     if (ret) memcpy( ret, str, size );
61   }
62   return ret;
63 }
64
65 /*********************************************************************
66  *              _wcsicoll (MSVCRT.@)
67  */
68 INT _wcsicoll( const MSVCRT_wchar_t* str1, const MSVCRT_wchar_t* str2 )
69 {
70   /* FIXME: handle collates */
71   return strcmpiW( str1, str2 );
72 }
73
74 /*********************************************************************
75  *              _wcsnset (MSVCRT.@)
76  */
77 MSVCRT_wchar_t* _wcsnset( MSVCRT_wchar_t* str, MSVCRT_wchar_t c, MSVCRT_size_t n )
78 {
79   MSVCRT_wchar_t* ret = str;
80   while ((n-- > 0) && *str) *str++ = c;
81   return ret;
82 }
83
84 /*********************************************************************
85  *              _wcsrev (MSVCRT.@)
86  */
87 MSVCRT_wchar_t* _wcsrev( MSVCRT_wchar_t* str )
88 {
89   MSVCRT_wchar_t* ret = str;
90   MSVCRT_wchar_t* end = str + strlenW(str) - 1;
91   while (end > str)
92   {
93     MSVCRT_wchar_t t = *end;
94     *end--  = *str;
95     *str++  = t;
96   }
97   return ret;
98 }
99
100 /*********************************************************************
101  *              _wcsset (MSVCRT.@)
102  */
103 MSVCRT_wchar_t* _wcsset( MSVCRT_wchar_t* str, MSVCRT_wchar_t c )
104 {
105   MSVCRT_wchar_t* ret = str;
106   while (*str) *str++ = c;
107   return ret;
108 }
109
110 /*********************************************************************
111  *              wcstod (MSVCRT.@)
112  */
113 double MSVCRT_wcstod(const MSVCRT_wchar_t* lpszStr, MSVCRT_wchar_t** end)
114 {
115   const MSVCRT_wchar_t* str = lpszStr;
116   int negative = 0;
117   double ret = 0, divisor = 10.0;
118
119   TRACE("(%s,%p) semi-stub\n", debugstr_w(lpszStr), end);
120
121   /* FIXME:
122    * - Should set errno on failure
123    * - Should fail on overflow
124    * - Need to check which input formats are allowed
125    */
126   while (isspaceW(*str))
127     str++;
128
129   if (*str == '-')
130   {
131     negative = 1;
132     str++;
133   }
134
135   while (isdigitW(*str))
136   {
137     ret = ret * 10.0 + (*str - '0');
138     str++;
139   }
140   if (*str == '.')
141     str++;
142   while (isdigitW(*str))
143   {
144     ret = ret + (*str - '0') / divisor;
145     divisor *= 10;
146     str++;
147   }
148
149   if (*str == 'E' || *str == 'e' || *str == 'D' || *str == 'd')
150   {
151     int negativeExponent = 0;
152     int exponent = 0;
153     if (*(++str) == '-')
154     {
155       negativeExponent = 1;
156       str++;
157     }
158     while (isdigitW(*str))
159     {
160       exponent = exponent * 10 + (*str - '0');
161       str++;
162     }
163     if (exponent != 0)
164     {
165       if (negativeExponent)
166         ret = ret / pow(10.0, exponent);
167       else
168         ret = ret * pow(10.0, exponent);
169     }
170   }
171
172   if (negative)
173     ret = -ret;
174
175   if (end)
176     *end = (MSVCRT_wchar_t*)str;
177
178   TRACE("returning %g\n", ret);
179   return ret;
180 }
181
182
183 typedef struct pf_output_t
184 {
185     int used;
186     int len;
187     BOOL unicode;
188     union {
189         LPWSTR W;
190         LPSTR  A;
191     } buf;
192 } pf_output;
193
194 typedef struct pf_flags_t
195 {
196     char Sign, LeftAlign, Alternate, PadZero;
197     int FieldLength, Precision;
198     char IntegerLength, IntegerDouble;
199     char WideString;
200     char Format;
201 } pf_flags;
202
203 /*
204  * writes a string of characters to the output
205  * returns -1 if the string doesn't fit in the output buffer
206  * return the length of the string if all characters were written
207  */
208 static inline int pf_output_stringW( pf_output *out, LPCWSTR str, int len )
209 {
210     int space = out->len - out->used;
211
212     if( len < 0 )
213         len = strlenW( str );
214     if( out->unicode )
215     {
216         LPWSTR p = out->buf.W + out->used;
217
218         if( space >= len )
219         {
220             memcpy( p, str, len*sizeof(WCHAR) );
221             out->used += len;
222             return len;
223         }
224         if( space > 0 )
225             memcpy( p, str, space*sizeof(WCHAR) );
226         out->used += len;
227     }
228     else
229     {
230         int n = WideCharToMultiByte( CP_ACP, 0, str, len, NULL, 0, NULL, NULL );
231         LPSTR p = out->buf.A + out->used;
232
233         if( space >= n )
234         {
235             WideCharToMultiByte( CP_ACP, 0, str, len, p, n, NULL, NULL );
236             out->used += n;
237             return len;
238         }
239         if( space > 0 )
240             WideCharToMultiByte( CP_ACP, 0, str, len, p, space, NULL, NULL );
241         out->used += n;
242     }
243     return -1;
244 }
245
246 static inline int pf_output_stringA( pf_output *out, LPCSTR str, int len )
247 {
248     int space = out->len - out->used;
249
250     if( len < 0 )
251         len = strlen( str );
252     if( !out->unicode )
253     {
254         LPSTR p = out->buf.A + out->used;
255
256         if( space >= len )
257         {
258             memcpy( p, str, len );
259             out->used += len;
260             return len;
261         }
262         if( space > 0 )
263             memcpy( p, str, space );
264         out->used += len;
265     }
266     else
267     {
268         int n = MultiByteToWideChar( CP_ACP, 0, str, len, NULL, 0 );
269         LPWSTR p = out->buf.W + out->used;
270
271         if( space >= n )
272         {
273             MultiByteToWideChar( CP_ACP, 0, str, len, p, n );
274             out->used += n;
275             return len;
276         }
277         if( space > 0 )
278             MultiByteToWideChar( CP_ACP, 0, str, len, p, space );
279         out->used += n;
280     }
281     return -1;
282 }
283
284 static inline int pf_fill( pf_output *out, int len, pf_flags *flags, char left )
285 {
286     int i, r = 0;
287
288     if( ( !left &&  flags->LeftAlign ) || 
289         (  left && !flags->LeftAlign ))
290     {
291         for( i=0; (i<(flags->FieldLength-len)) && (r>=0); i++ )
292         {
293             if( left && flags->PadZero )
294                 r = pf_output_stringA( out, "0", 1 );
295             else
296                 r = pf_output_stringA( out, " ", 1 );
297         }
298     }
299
300     return r;
301 }
302
303 static inline int pf_output_format_W( pf_output *out, LPCWSTR str,
304                                       int len, pf_flags *flags )
305 {
306     int r = 0;
307
308     if( len < 0 )
309         len = strlenW( str );
310
311     if (flags->Precision >= 0 && flags->Precision < len)
312         len = flags->Precision;
313
314     r = pf_fill( out, len, flags, 1 );
315
316     if( r>=0 )
317         r = pf_output_stringW( out, str, len );
318
319     if( r>=0 )
320         r = pf_fill( out, len, flags, 0 );
321
322     return r;
323 }
324
325 static inline int pf_output_format_A( pf_output *out, LPCSTR str,
326                                       int len, pf_flags *flags )
327 {
328     int r = 0;
329
330     if( len < 0 )
331         len = strlen( str );
332
333     if (flags->Precision >= 0 && flags->Precision < len)
334         len = flags->Precision;
335
336     r = pf_fill( out, len, flags, 1 );
337
338     if( r>=0 )
339         r = pf_output_stringA( out, str, len );
340
341     if( r>=0 )
342         r = pf_fill( out, len, flags, 0 );
343
344     return r;
345 }
346
347 static inline BOOL pf_is_double_format( char fmt )
348 {
349     static const char float_fmts[] = "aeEfgG";
350     if (!fmt)
351         return FALSE;
352     return strchr( float_fmts, fmt ) ? TRUE : FALSE;
353 }
354
355 static inline BOOL pf_is_valid_format( char fmt )
356 {
357     static const char float_fmts[] = "acCdeEfgGinouxX";
358     if (!fmt)
359         return FALSE;
360     return strchr( float_fmts, fmt ) ? TRUE : FALSE;
361 }
362
363 static void pf_rebuild_format_string( char *p, pf_flags *flags )
364 {
365     *p++ = '%';
366     if( flags->Sign )
367         *p++ = flags->Sign;
368     if( flags->LeftAlign )
369         *p++ = flags->LeftAlign;
370     if( flags->Alternate )
371         *p++ = flags->Alternate;
372     if( flags->PadZero )
373         *p++ = flags->PadZero;
374     if( flags->FieldLength )
375     {
376         sprintf(p, "%d", flags->FieldLength);
377         p += strlen(p);
378     }
379     if( flags->Precision >= 0 )
380     {
381         sprintf(p, ".%d", flags->Precision);
382         p += strlen(p);
383     }
384     *p++ = flags->Format;
385     *p++ = 0;
386 }
387
388 /*********************************************************************
389  *  pf_vsnprintf  (INTERNAL)
390  *
391  *  implements both A and W vsnprintf functions
392  */
393 static int pf_vsnprintf( pf_output *out, const WCHAR *format, va_list valist )
394 {
395     int r;
396     LPCWSTR q, p = format;
397     pf_flags flags;
398
399     TRACE("format is %s\n",debugstr_w(format));
400     while (*p)
401     {
402         q = strchrW( p, '%' );
403
404         /* there's no % characters left, output the rest of the string */
405         if( !q )
406         {
407             r = pf_output_stringW(out, p, -1);
408             if( r<0 )
409                 return r;
410             p += r;
411             continue;
412         }
413
414         /* there's characters before the %, output them */
415         if( q != p )
416         {
417             r = pf_output_stringW(out, p, q - p);
418             if( r<0 )
419                 return r;
420             p = q;
421         }
422
423         /* we must be at a % now, skip over it */
424         assert( *p == '%' );
425         p++;
426
427         /* output a single % character */
428         if( *p == '%' )
429         {
430             r = pf_output_stringW(out, p++, 1);
431             if( r<0 )
432                 return r;
433             continue;
434         }
435
436         /* parse the flags */
437         memset( &flags, 0, sizeof flags );
438         while (*p)
439         {
440             if( *p == '+' || *p == ' ' )
441             {
442                 if ( flags.Sign != '+' )
443                     flags.Sign = *p;
444             }
445             else if( *p == '-' )
446                 flags.LeftAlign = *p;
447             else if( *p == '0' )
448                 flags.PadZero = *p;
449             else if( *p == '#' )
450                 flags.Alternate = *p;
451             else
452                 break;
453             p++;
454         }
455
456         /* deal with the field width specifier */
457         flags.FieldLength = 0;
458         if( *p == '*' ) 
459         {
460             flags.FieldLength = va_arg( valist, int );
461             p++;
462         }
463         else while( isdigit(*p) )
464         {
465             flags.FieldLength *= 10;
466             flags.FieldLength += *p++ - '0';
467         }
468
469         /* deal with precision */
470         flags.Precision = -1;
471         if( *p == '.' )
472         {
473             flags.Precision = 0;
474             p++;
475             if( *p == '*' )
476             {
477                 flags.Precision = va_arg( valist, int );
478                 p++;
479             }
480             else while( isdigit(*p) )
481             {
482                 flags.Precision *= 10;
483                 flags.Precision += *p++ - '0';
484             }
485         }
486
487         /* deal with integer width modifier */
488         while( *p )
489         {
490             if( *p == 'h' || *p == 'l' || *p == 'L' )
491             {
492                 flags.IntegerLength = *p;
493                 p++;
494             }
495             else if( *p == 'I' )
496             {
497                 if( *(p+1) == '6' && *(p+2) == '4' )
498                 {
499                     flags.IntegerDouble++;
500                     p += 3;
501                 }
502                 else if( *(p+1) == '3' && *(p+2) == '2' )
503                     p += 3;
504                 else if( isdigit(*(p+1)) || *(p+1) == 0 )
505                     break;
506                 else
507                     p++;
508             }
509             else if( *p == 'w' )
510                 flags.WideString = *p++;
511             else if( *p == 'F' )
512                 p++; /* ignore */
513             else
514                 break;
515         }
516
517         flags.Format = *p;
518         r = 0;
519
520         /* output a unicode string */
521         if( ( flags.Format == 's' && (flags.WideString || flags.IntegerLength == 'l' )) ||
522             ( !out->unicode && flags.Format == 'S' ) ||
523             ( out->unicode && flags.Format == 's' ) )
524         {
525             LPCWSTR str = va_arg( valist, const WCHAR * );
526
527             if( str )
528                 r = pf_output_format_W( out, str, -1, &flags );
529             else
530                 r = pf_output_format_A( out, "(null)", -1, &flags );
531         }
532
533         /* output a ASCII string */
534         else if( ( flags.Format == 's' && flags.IntegerLength == 'h' ) ||
535             ( out->unicode && flags.Format == 'S' ) ||
536             ( !out->unicode && flags.Format == 's' ) )
537         {
538             LPCSTR str = va_arg( valist, const CHAR * );
539
540             if( str )
541                 r = pf_output_format_A( out, str, -1, &flags );
542             else
543                 r = pf_output_format_A( out, "(null)", -1, &flags );
544         }
545
546         /* output a single wide character */
547         else if( ( flags.Format == 'c' && flags.IntegerLength == 'w' ) ||
548             ( out->unicode && flags.Format == 'c' ) ||
549             ( !out->unicode && flags.Format == 'C' ) )
550         {
551             WCHAR ch = va_arg( valist, int );
552
553             r = pf_output_format_W( out, &ch, 1, &flags );
554         }
555
556         /* output a single ascii character */
557         else if( ( flags.Format == 'c' && flags.IntegerLength == 'h' ) ||
558             ( out->unicode && flags.Format == 'C' ) ||
559             ( !out->unicode && flags.Format == 'c' ) )
560         {
561             CHAR ch = va_arg( valist, int );
562
563             r = pf_output_format_A( out, &ch, 1, &flags );
564         }
565
566         /* output a pointer */
567         else if( flags.Format == 'p' )
568         {
569             char pointer[11];
570
571             flags.PadZero = 0;
572             if( flags.Alternate )
573                 sprintf(pointer, "0X%08lX", va_arg(valist, long));
574             else
575                 sprintf(pointer, "%08lX", va_arg(valist, long));
576             r = pf_output_format_A( out, pointer, -1, &flags );
577         }
578
579         /* deal with %n */
580         else if( flags.Format == 'n' )
581         {
582             int *x = va_arg(valist, int *);
583             *x = out->used;
584         }
585
586         /* deal with integers and floats using libc's printf */
587         else if( pf_is_valid_format( flags.Format ) )
588         {
589             char fmt[20], number[40], *x = number;
590
591             if( flags.FieldLength >= sizeof number )
592                 x = HeapAlloc( GetProcessHeap(), 0, flags.FieldLength+1 );
593
594             pf_rebuild_format_string( fmt, &flags );
595
596             if( pf_is_double_format( flags.Format ) )
597                 sprintf( x, fmt, va_arg(valist, double) );
598             else
599                 sprintf( x, fmt, va_arg(valist, int) );
600
601             r = pf_output_stringA( out, x, -1 );
602             if( x != number )
603                 HeapFree( GetProcessHeap(), 0, number );
604         }
605         else
606             continue;
607
608         if( r<0 )
609             return r;
610         p++;
611     }
612
613     /* check we reached the end, and null terminate the string */
614     assert( *p == 0 );
615     r = pf_output_stringW( out, p, 1 );
616     if( r<0 )
617         return r;
618
619     return out->used - 1;
620 }
621
622 /*********************************************************************
623  *              _vsnprintf (MSVCRT.@)
624  */
625 int MSVCRT_vsnprintf( char *str, unsigned int len,
626                 const char *format, va_list valist )
627 {
628     DWORD sz;
629     LPWSTR formatW = NULL;
630     pf_output out;
631     int r;
632
633     out.unicode = FALSE;
634     out.buf.A = str;
635     out.used = 0;
636     out.len = len;
637
638     if( format )
639     {
640         sz = MultiByteToWideChar( CP_ACP, 0, format, -1, NULL, 0 );
641         formatW = HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
642         MultiByteToWideChar( CP_ACP, 0, format, -1, formatW, sz );
643     }
644
645     r = pf_vsnprintf( &out, formatW, valist );
646
647     HeapFree( GetProcessHeap(), 0, formatW );
648
649     return r;
650 }
651
652 /*********************************************************************
653  *              _vsnwsprintf (MSVCRT.@)
654  */
655 int MSVCRT_vsnwprintf( MSVCRT_wchar_t *str, unsigned int len,
656                               const WCHAR *format, va_list valist )
657 {
658     pf_output out;
659
660     out.unicode = TRUE;
661     out.buf.W = str;
662     out.used = 0;
663     out.len = len;
664
665     return pf_vsnprintf( &out, format, valist );
666 }
667
668 /*********************************************************************
669  *              sprintf (MSVCRT.@)
670  */
671 int MSVCRT_sprintf( char *str, const char *format, ... )
672 {
673     va_list ap;
674     int r;
675
676     va_start( ap, format );
677     r = MSVCRT_vsnprintf( str, INT_MAX, format, ap );
678     va_end( ap );
679     return r;
680 }
681
682 /*********************************************************************
683  *              swprintf (MSVCRT.@)
684  */
685 int MSVCRT_swprintf( MSVCRT_wchar_t *str, const MSVCRT_wchar_t *format, ... )
686 {
687     va_list ap;
688     int r;
689
690     va_start( ap, format );
691     r = MSVCRT_vsnwprintf( str, INT_MAX, format, ap );
692     va_end( ap );
693     return r;
694 }
695
696 /*********************************************************************
697  *              vswprintf (MSVCRT.@)
698  */
699 int MSVCRT_vswprintf( MSVCRT_wchar_t* str, const MSVCRT_wchar_t* format, va_list args )
700 {
701     return MSVCRT_vsnwprintf( str, INT_MAX, format, args );
702 }
703
704 /*********************************************************************
705  *              wcscoll (MSVCRT.@)
706  */
707 int MSVCRT_wcscoll( const MSVCRT_wchar_t* str1, const MSVCRT_wchar_t* str2 )
708 {
709   /* FIXME: handle collates */
710   return strcmpW( str1, str2 );
711 }
712
713 /*********************************************************************
714  *              wcspbrk (MSVCRT.@)
715  */
716 MSVCRT_wchar_t* MSVCRT_wcspbrk( const MSVCRT_wchar_t* str, const MSVCRT_wchar_t* accept )
717 {
718   const MSVCRT_wchar_t* p;
719   while (*str)
720   {
721     for (p = accept; *p; p++) if (*p == *str) return (MSVCRT_wchar_t*)str;
722       str++;
723   }
724   return NULL;
725 }
726
727 /*********************************************************************
728  *              wctomb (MSVCRT.@)
729  */
730 INT MSVCRT_wctomb( char *dst, MSVCRT_wchar_t ch )
731 {
732   return WideCharToMultiByte( CP_ACP, 0, &ch, 1, dst, 6, NULL, NULL );
733 }
734
735 /*********************************************************************
736  *              iswalnum (MSVCRT.@)
737  */
738 INT MSVCRT_iswalnum( MSVCRT_wchar_t wc )
739 {
740     return isalnumW( wc );
741 }
742
743 /*********************************************************************
744  *              iswalpha (MSVCRT.@)
745  */
746 INT MSVCRT_iswalpha( MSVCRT_wchar_t wc )
747 {
748     return isalphaW( wc );
749 }
750
751 /*********************************************************************
752  *              iswcntrl (MSVCRT.@)
753  */
754 INT MSVCRT_iswcntrl( MSVCRT_wchar_t wc )
755 {
756     return iscntrlW( wc );
757 }
758
759 /*********************************************************************
760  *              iswdigit (MSVCRT.@)
761  */
762 INT MSVCRT_iswdigit( MSVCRT_wchar_t wc )
763 {
764     return isdigitW( wc );
765 }
766
767 /*********************************************************************
768  *              iswgraph (MSVCRT.@)
769  */
770 INT MSVCRT_iswgraph( MSVCRT_wchar_t wc )
771 {
772     return isgraphW( wc );
773 }
774
775 /*********************************************************************
776  *              iswlower (MSVCRT.@)
777  */
778 INT MSVCRT_iswlower( MSVCRT_wchar_t wc )
779 {
780     return islowerW( wc );
781 }
782
783 /*********************************************************************
784  *              iswprint (MSVCRT.@)
785  */
786 INT MSVCRT_iswprint( MSVCRT_wchar_t wc )
787 {
788     return isprintW( wc );
789 }
790
791 /*********************************************************************
792  *              iswpunct (MSVCRT.@)
793  */
794 INT MSVCRT_iswpunct( MSVCRT_wchar_t wc )
795 {
796     return ispunctW( wc );
797 }
798
799 /*********************************************************************
800  *              iswspace (MSVCRT.@)
801  */
802 INT MSVCRT_iswspace( MSVCRT_wchar_t wc )
803 {
804     return isspaceW( wc );
805 }
806
807 /*********************************************************************
808  *              iswupper (MSVCRT.@)
809  */
810 INT MSVCRT_iswupper( MSVCRT_wchar_t wc )
811 {
812     return isupperW( wc );
813 }
814
815 /*********************************************************************
816  *              iswxdigit (MSVCRT.@)
817  */
818 INT MSVCRT_iswxdigit( MSVCRT_wchar_t wc )
819 {
820     return isxdigitW( wc );
821 }