Use the correct buffer - reported by Jesse Allen.
[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         (  left &&  flags->PadZero   ) )
291     {
292         for( i=0; (i<(flags->FieldLength-len)) && (r>=0); i++ )
293         {
294             if( flags->PadZero )
295                 r = pf_output_stringA( out, "0", 1 );
296             else
297                 r = pf_output_stringA( out, " ", 1 );
298         }
299     }
300
301     return r;
302 }
303
304 static inline int pf_output_format_W( pf_output *out, LPCWSTR str,
305                                       int len, pf_flags *flags )
306 {
307     int r = 0;
308
309     if( len < 0 )
310         len = strlenW( str );
311
312     if (flags->Precision >= 0 && flags->Precision < len)
313         len = flags->Precision;
314
315     r = pf_fill( out, len, flags, 1 );
316
317     if( r>=0 )
318         r = pf_output_stringW( out, str, len );
319
320     if( r>=0 )
321         r = pf_fill( out, len, flags, 0 );
322
323     return r;
324 }
325
326 static inline int pf_output_format_A( pf_output *out, LPCSTR str,
327                                       int len, pf_flags *flags )
328 {
329     int r = 0;
330
331     if( len < 0 )
332         len = strlen( str );
333
334     if (flags->Precision >= 0 && flags->Precision < len)
335         len = flags->Precision;
336
337     r = pf_fill( out, len, flags, 1 );
338
339     if( r>=0 )
340         r = pf_output_stringA( out, str, len );
341
342     if( r>=0 )
343         r = pf_fill( out, len, flags, 0 );
344
345     return r;
346 }
347
348 static inline BOOL pf_is_double_format( char fmt )
349 {
350     char float_fmts[] = "aefg";
351     if (!fmt)
352         return FALSE;
353     fmt = tolower( fmt );
354     return strchr( float_fmts, fmt ) ? TRUE : FALSE;
355 }
356
357 static inline BOOL pf_is_valid_format( char fmt )
358 {
359     char float_fmts[] = "acdefginoux";
360     if (!fmt)
361         return FALSE;
362     fmt = tolower( fmt );
363     return strchr( float_fmts, fmt ) ? TRUE : FALSE;
364 }
365
366 static void pf_rebuild_format_string( char *p, pf_flags *flags )
367 {
368     *p++ = '%';
369     if( flags->Sign )
370         *p++ = flags->Sign;
371     if( flags->LeftAlign )
372         *p++ = flags->LeftAlign;
373     if( flags->Alternate )
374         *p++ = flags->Alternate;
375     if( flags->PadZero )
376         *p++ = flags->PadZero;
377     if( flags->FieldLength )
378     {
379         sprintf(p, "%d", flags->FieldLength);
380         p += strlen(p);
381     }
382     if( flags->Precision >= 0 )
383     {
384         sprintf(p, ".%d", flags->Precision);
385         p += strlen(p);
386     }
387     *p++ = flags->Format;
388     *p++ = 0;
389 }
390
391 /*********************************************************************
392  *  pf_vsnprintf  (INTERNAL)
393  *
394  *  implements both A and W vsnprintf functions
395  */
396 static int pf_vsnprintf( pf_output *out, const WCHAR *format, va_list valist )
397 {
398     int r;
399     LPCWSTR q, p = format;
400     pf_flags flags;
401
402     TRACE("format is %s\n",debugstr_w(format));
403     while (*p)
404     {
405         q = strchrW( p, '%' );
406
407         /* there's no % characters left, output the rest of the string */
408         if( !q )
409         {
410             r = pf_output_stringW(out, p, -1);
411             if( r<0 )
412                 return r;
413             p += r;
414             continue;
415         }
416
417         /* there's characters before the %, output them */
418         if( q != p )
419         {
420             r = pf_output_stringW(out, p, q - p);
421             if( r<0 )
422                 return r;
423             p = q;
424         }
425
426         /* we must be at a % now, skip over it */
427         assert( *p == '%' );
428         p++;
429
430         /* output a single % character */
431         if( *p == '%' )
432         {
433             r = pf_output_stringW(out, p++, 1);
434             if( r<0 )
435                 return r;
436             continue;
437         }
438
439         /* parse the flags */
440         memset( &flags, 0, sizeof flags );
441         while (*p)
442         {
443             if( *p == '+' || *p == ' ' )
444                 flags.Sign = '+';
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                 if( flags.IntegerLength == *p )  /* FIXME: this is wrong */
493                     flags.IntegerDouble++;
494                 else
495                     flags.IntegerLength = *p;
496                 p++;
497             }
498             else if( *p == 'w' )
499                 flags.WideString = *p++;
500             else if( *p == 'F' )
501                 p++; /* ignore */
502             else
503                 break;
504         }
505
506         flags.Format = *p;
507         r = 0;
508
509         /* output a unicode string */
510         if( ( flags.Format == 's' && (flags.WideString || flags.IntegerLength == 'l' )) ||
511             ( !out->unicode && flags.Format == 'S' ) ||
512             ( out->unicode && flags.Format == 's' ) )
513         {
514             LPCWSTR str = va_arg( valist, const WCHAR * );
515
516             if( str )
517                 r = pf_output_format_W( out, str, -1, &flags );
518             else
519                 r = pf_output_format_A( out, "(null)", -1, &flags );
520         }
521
522         /* output a ASCII string */
523         else if( ( flags.Format == 's' && flags.IntegerLength == 'h' ) ||
524             ( out->unicode && flags.Format == 'S' ) ||
525             ( !out->unicode && flags.Format == 's' ) )
526         {
527             LPCSTR str = va_arg( valist, const CHAR * );
528
529             if( str )
530                 r = pf_output_format_A( out, str, -1, &flags );
531             else
532                 r = pf_output_format_A( out, "(null)", -1, &flags );
533         }
534
535         /* output a single wide character */
536         else if( ( flags.Format == 'c' && flags.IntegerLength == 'w' ) ||
537             ( out->unicode && flags.Format == 'c' ) ||
538             ( !out->unicode && flags.Format == 'C' ) )
539         {
540             WCHAR ch = va_arg( valist, int );
541
542             r = pf_output_format_W( out, &ch, 1, &flags );
543         }
544
545         /* output a single ascii character */
546         else if( ( flags.Format == 'c' && flags.IntegerLength == 'h' ) ||
547             ( out->unicode && flags.Format == 'C' ) ||
548             ( !out->unicode && flags.Format == 'c' ) )
549         {
550             CHAR ch = va_arg( valist, int );
551
552             r = pf_output_format_A( out, &ch, 1, &flags );
553         }
554
555         /* output a pointer */
556         else if( flags.Format == 'p' )
557         {
558             char pointer[11];
559
560             flags.PadZero = 0;
561             if( flags.Alternate )
562                 sprintf(pointer, "0X%08lX", va_arg(valist, long));
563             else
564                 sprintf(pointer, "%08lX", va_arg(valist, long));
565             r = pf_output_format_A( out, pointer, -1, &flags );
566         }
567
568         /* deal with %n */
569         else if( flags.Format == 'n' )
570         {
571             int *x = va_arg(valist, int *);
572             *x = out->used;
573         }
574
575         /* deal with integers and floats using libc's printf */
576         else if( pf_is_valid_format( flags.Format ) )
577         {
578             char fmt[20], number[40], *x = number;
579
580             if( flags.FieldLength >= sizeof number )
581                 x = HeapAlloc( GetProcessHeap(), 0, flags.FieldLength+1 );
582
583             pf_rebuild_format_string( fmt, &flags );
584
585             if( pf_is_double_format( flags.Format ) )
586                 sprintf( x, fmt, va_arg(valist, double) );
587             else
588                 sprintf( x, fmt, va_arg(valist, int) );
589
590             r = pf_output_stringA( out, x, -1 );
591             if( x != number )
592                 HeapFree( GetProcessHeap(), 0, number );
593         }
594         else
595             continue;
596
597         if( r<0 )
598             return r;
599         p++;
600     }
601
602     /* check we reached the end, and null terminate the string */
603     assert( *p == 0 );
604     r = pf_output_stringW( out, p, 1 );
605     if( r<0 )
606         return r;
607
608     return out->used - 1;
609 }
610
611 /*********************************************************************
612  *              _vsnprintf (MSVCRT.@)
613  */
614 int MSVCRT_vsnprintf( char *str, unsigned int len,
615                 const char *format, va_list valist )
616 {
617     DWORD sz;
618     LPWSTR formatW = NULL;
619     pf_output out;
620     int r;
621
622     out.unicode = FALSE;
623     out.buf.A = str;
624     out.used = 0;
625     out.len = len;
626
627     if( format )
628     {
629         sz = MultiByteToWideChar( CP_ACP, 0, format, -1, NULL, 0 );
630         formatW = HeapAlloc( GetProcessHeap(), 0, sz*sizeof(WCHAR) );
631         MultiByteToWideChar( CP_ACP, 0, format, -1, formatW, sz );
632     }
633
634     r = pf_vsnprintf( &out, formatW, valist );
635
636     HeapFree( GetProcessHeap(), 0, formatW );
637
638     return r;
639 }
640
641 /*********************************************************************
642  *              _vsnwsprintf (MSVCRT.@)
643  */
644 int MSVCRT_vsnwprintf( MSVCRT_wchar_t *str, unsigned int len,
645                               const WCHAR *format, va_list valist )
646 {
647     pf_output out;
648
649     out.unicode = TRUE;
650     out.buf.W = str;
651     out.used = 0;
652     out.len = len;
653
654     return pf_vsnprintf( &out, format, valist );
655 }
656
657 /*********************************************************************
658  *              sprintf (MSVCRT.@)
659  */
660 int MSVCRT_sprintf( char *str, const char *format, ... )
661 {
662     va_list ap;
663     int r;
664
665     va_start( ap, format );
666     r = MSVCRT_vsnprintf( str, INT_MAX, format, ap );
667     va_end( ap );
668     return r;
669 }
670
671 /*********************************************************************
672  *              swprintf (MSVCRT.@)
673  */
674 int MSVCRT_swprintf( MSVCRT_wchar_t *str, const MSVCRT_wchar_t *format, ... )
675 {
676     va_list ap;
677     int r;
678
679     va_start( ap, format );
680     r = MSVCRT_vsnwprintf( str, INT_MAX, format, ap );
681     va_end( ap );
682     return r;
683 }
684
685 /*********************************************************************
686  *              vswprintf (MSVCRT.@)
687  */
688 int MSVCRT_vswprintf( MSVCRT_wchar_t* str, const MSVCRT_wchar_t* format, va_list args )
689 {
690     return MSVCRT_vsnwprintf( str, INT_MAX, format, args );
691 }
692
693 /*********************************************************************
694  *              wcscoll (MSVCRT.@)
695  */
696 int MSVCRT_wcscoll( const MSVCRT_wchar_t* str1, const MSVCRT_wchar_t* str2 )
697 {
698   /* FIXME: handle collates */
699   return strcmpW( str1, str2 );
700 }
701
702 /*********************************************************************
703  *              wcspbrk (MSVCRT.@)
704  */
705 MSVCRT_wchar_t* MSVCRT_wcspbrk( const MSVCRT_wchar_t* str, const MSVCRT_wchar_t* accept )
706 {
707   const MSVCRT_wchar_t* p;
708   while (*str)
709   {
710     for (p = accept; *p; p++) if (*p == *str) return (MSVCRT_wchar_t*)str;
711       str++;
712   }
713   return NULL;
714 }
715
716 /*********************************************************************
717  *              wctomb (MSVCRT.@)
718  */
719 INT MSVCRT_wctomb( char *dst, MSVCRT_wchar_t ch )
720 {
721   return WideCharToMultiByte( CP_ACP, 0, &ch, 1, dst, 6, NULL, NULL );
722 }
723
724 /*********************************************************************
725  *              iswalnum (MSVCRT.@)
726  */
727 INT MSVCRT_iswalnum( MSVCRT_wchar_t wc )
728 {
729     return isalnumW( wc );
730 }
731
732 /*********************************************************************
733  *              iswalpha (MSVCRT.@)
734  */
735 INT MSVCRT_iswalpha( MSVCRT_wchar_t wc )
736 {
737     return isalphaW( wc );
738 }
739
740 /*********************************************************************
741  *              iswcntrl (MSVCRT.@)
742  */
743 INT MSVCRT_iswcntrl( MSVCRT_wchar_t wc )
744 {
745     return iscntrlW( wc );
746 }
747
748 /*********************************************************************
749  *              iswdigit (MSVCRT.@)
750  */
751 INT MSVCRT_iswdigit( MSVCRT_wchar_t wc )
752 {
753     return isdigitW( wc );
754 }
755
756 /*********************************************************************
757  *              iswgraph (MSVCRT.@)
758  */
759 INT MSVCRT_iswgraph( MSVCRT_wchar_t wc )
760 {
761     return isgraphW( wc );
762 }
763
764 /*********************************************************************
765  *              iswlower (MSVCRT.@)
766  */
767 INT MSVCRT_iswlower( MSVCRT_wchar_t wc )
768 {
769     return islowerW( wc );
770 }
771
772 /*********************************************************************
773  *              iswprint (MSVCRT.@)
774  */
775 INT MSVCRT_iswprint( MSVCRT_wchar_t wc )
776 {
777     return isprintW( wc );
778 }
779
780 /*********************************************************************
781  *              iswpunct (MSVCRT.@)
782  */
783 INT MSVCRT_iswpunct( MSVCRT_wchar_t wc )
784 {
785     return ispunctW( wc );
786 }
787
788 /*********************************************************************
789  *              iswspace (MSVCRT.@)
790  */
791 INT MSVCRT_iswspace( MSVCRT_wchar_t wc )
792 {
793     return isspaceW( wc );
794 }
795
796 /*********************************************************************
797  *              iswupper (MSVCRT.@)
798  */
799 INT MSVCRT_iswupper( MSVCRT_wchar_t wc )
800 {
801     return isupperW( wc );
802 }
803
804 /*********************************************************************
805  *              iswxdigit (MSVCRT.@)
806  */
807 INT MSVCRT_iswxdigit( MSVCRT_wchar_t wc )
808 {
809     return isxdigitW( wc );
810 }