msvcrt: Handle negative file->_cnt value in _flsbuf function.
[wine] / dlls / msvcrt / printf.h
1 /*
2  * Copyright 2011 Piotr Caban for CodeWeavers
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Lesser General Public License for more details.
13  *
14  * You should have received a copy of the GNU Lesser General Public
15  * License along with this library; if not, write to the Free Software
16  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
17  */
18
19 #ifdef PRINTF_WIDE
20 #define APICHAR MSVCRT_wchar_t
21 #define CONVCHAR char
22 #define FUNC_NAME(func) func ## _w
23 #else
24 #define APICHAR char
25 #define CONVCHAR MSVCRT_wchar_t
26 #define FUNC_NAME(func) func ## _a
27 #endif
28
29 typedef struct FUNC_NAME(pf_flags_t)
30 {
31     APICHAR Sign, LeftAlign, Alternate, PadZero;
32     int FieldLength, Precision;
33     APICHAR IntegerLength, IntegerDouble;
34     APICHAR WideString;
35     APICHAR Format;
36 } FUNC_NAME(pf_flags);
37
38 struct FUNC_NAME(_str_ctx) {
39     MSVCRT_size_t len;
40     APICHAR *buf;
41 };
42
43 static int FUNC_NAME(puts_clbk_str)(void *ctx, int len, const APICHAR *str)
44 {
45     struct FUNC_NAME(_str_ctx) *out = ctx;
46
47     if(!out->buf)
48         return len;
49
50     if(out->len < len) {
51         memcpy(out->buf, str, out->len);
52         out->buf += out->len;
53         out->len = 0;
54         return -1;
55     }
56
57     memcpy(out->buf, str, len*sizeof(APICHAR));
58     out->buf += len;
59     out->len -= len;
60     return len;
61 }
62
63 static inline const APICHAR* FUNC_NAME(pf_parse_int)(const APICHAR *fmt, int *val)
64 {
65     *val = 0;
66
67     while(isdigit(*fmt)) {
68         *val *= 10;
69         *val += *fmt++ - '0';
70     }
71
72     return fmt;
73 }
74
75 /* pf_fill: takes care of signs, alignment, zero and field padding */
76 static inline int FUNC_NAME(pf_fill)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx,
77         int len, FUNC_NAME(pf_flags) *flags, BOOL left)
78 {
79     int i, r = 0, written;
80
81     if(flags->Sign && !(flags->Format=='d' || flags->Format=='i'))
82         flags->Sign = 0;
83
84     if(left && flags->Sign) {
85         flags->FieldLength--;
86         if(flags->PadZero)
87             r = pf_puts(puts_ctx, 1, &flags->Sign);
88     }
89     written = r;
90
91     if((!left && flags->LeftAlign) || (left && !flags->LeftAlign)) {
92         APICHAR ch;
93
94         if(left && flags->PadZero)
95             ch = '0';
96         else
97             ch = ' ';
98
99         for(i=0; i<flags->FieldLength-len && r>=0; i++) {
100             r = pf_puts(puts_ctx, 1, &ch);
101             written += r;
102         }
103     }
104
105
106     if(r>=0 && left && flags->Sign && !flags->PadZero) {
107         r = pf_puts(puts_ctx, 1, &flags->Sign);
108         written += r;
109     }
110
111     return r>=0 ? written : r;
112 }
113
114 static inline int FUNC_NAME(pf_output_wstr)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx,
115         const MSVCRT_wchar_t *str, int len, MSVCRT_pthreadlocinfo locinfo)
116 {
117 #ifdef PRINTF_WIDE
118     return pf_puts(puts_ctx, len, str);
119 #else
120     LPSTR out;
121     int len_a = WideCharToMultiByte(locinfo->lc_codepage, 0, str, len, NULL, 0, NULL, NULL);
122
123     out = HeapAlloc(GetProcessHeap(), 0, len_a);
124     if(!out)
125         return -1;
126
127     WideCharToMultiByte(locinfo->lc_codepage, 0, str, len, out, len_a, NULL, NULL);
128     len = pf_puts(puts_ctx, len_a, out);
129     HeapFree(GetProcessHeap(), 0, out);
130     return len;
131 #endif
132 }
133
134 static inline int FUNC_NAME(pf_output_str)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx,
135         const char *str, int len, MSVCRT_pthreadlocinfo locinfo)
136 {
137 #ifdef PRINTF_WIDE
138     LPWSTR out;
139     int len_w = MultiByteToWideChar(locinfo->lc_codepage, 0, str, len, NULL, 0);
140
141     out = HeapAlloc(GetProcessHeap(), 0, len_w*sizeof(WCHAR));
142     if(!out)
143         return -1;
144
145     MultiByteToWideChar(locinfo->lc_codepage, 0, str, len, out, len_w);
146     len = pf_puts(puts_ctx, len_w, out);
147     HeapFree(GetProcessHeap(), 0, out);
148     return len;
149 #else
150     return pf_puts(puts_ctx, len, str);
151 #endif
152 }
153
154 static inline int FUNC_NAME(pf_output_format_wstr)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx,
155         const MSVCRT_wchar_t *str, int len, FUNC_NAME(pf_flags) *flags, MSVCRT_pthreadlocinfo locinfo)
156 {
157     int r, ret;
158
159     if(len < 0)
160         len = strlenW(str);
161
162     if(flags->Precision>=0 && flags->Precision<len)
163         len = flags->Precision;
164
165     r = FUNC_NAME(pf_fill)(pf_puts, puts_ctx, len, flags, TRUE);
166     ret = r;
167     if(r >= 0) {
168         r = FUNC_NAME(pf_output_wstr)(pf_puts, puts_ctx, str, len, locinfo);
169         ret += r;
170     }
171     if(r >= 0) {
172         r = FUNC_NAME(pf_fill)(pf_puts, puts_ctx, len, flags, FALSE);
173         ret += r;
174     }
175
176     return r>=0 ? ret : r;
177 }
178
179 static inline int FUNC_NAME(pf_output_format_str)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx,
180         const char *str, int len, FUNC_NAME(pf_flags) *flags, MSVCRT_pthreadlocinfo locinfo)
181 {
182     int r, ret;
183
184     if(len < 0)
185         len = strlen(str);
186
187     if(flags->Precision>=0 && flags->Precision<len)
188         len = flags->Precision;
189
190     r = FUNC_NAME(pf_fill)(pf_puts, puts_ctx, len, flags, TRUE);
191     ret = r;
192     if(r >= 0) {
193         r = FUNC_NAME(pf_output_str)(pf_puts, puts_ctx, str, len, locinfo);
194         ret += r;
195     }
196     if(r >= 0) {
197         r = FUNC_NAME(pf_fill)(pf_puts, puts_ctx, len, flags, FALSE);
198         ret += r;
199     }
200
201     return r>=0 ? ret : r;
202 }
203
204 static inline int FUNC_NAME(pf_handle_string)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx,
205         const void *str, int len, FUNC_NAME(pf_flags) *flags, MSVCRT_pthreadlocinfo locinfo)
206 {
207 #ifdef PRINTF_WIDE
208     static const MSVCRT_wchar_t nullW[] = {'(','n','u','l','l',')',0};
209
210     if(!str)
211         return FUNC_NAME(pf_output_format_wstr)(pf_puts, puts_ctx, nullW, 6, flags, locinfo);
212 #else
213     if(!str)
214         return FUNC_NAME(pf_output_format_str)(pf_puts, puts_ctx, "(null)", 6, flags, locinfo);
215 #endif
216
217     if(flags->WideString || flags->IntegerLength=='l')
218         return FUNC_NAME(pf_output_format_wstr)(pf_puts, puts_ctx, str, len, flags, locinfo);
219     if(flags->IntegerLength == 'h')
220         return FUNC_NAME(pf_output_format_str)(pf_puts, puts_ctx, str, len, flags, locinfo);
221
222     if((flags->Format=='S' || flags->Format=='C') == (sizeof(APICHAR)==sizeof(MSVCRT_wchar_t)))
223         return FUNC_NAME(pf_output_format_str)(pf_puts, puts_ctx, str, len, flags, locinfo);
224     else
225         return FUNC_NAME(pf_output_format_wstr)(pf_puts, puts_ctx, str, len, flags, locinfo);
226 }
227
228 static inline void FUNC_NAME(pf_rebuild_format_string)(char *p, FUNC_NAME(pf_flags) *flags)
229 {
230     *p++ = '%';
231     if(flags->Sign)
232         *p++ = flags->Sign;
233     if(flags->LeftAlign)
234         *p++ = flags->LeftAlign;
235     if(flags->Alternate)
236         *p++ = flags->Alternate;
237     if(flags->PadZero)
238         *p++ = flags->PadZero;
239     if(flags->FieldLength) {
240         sprintf(p, "%d", flags->FieldLength);
241         p += strlen(p);
242     }
243     if(flags->Precision >= 0) {
244         sprintf(p, ".%d", flags->Precision);
245         p += strlen(p);
246     }
247     *p++ = flags->Format;
248     *p++ = 0;
249 }
250
251 /* pf_integer_conv:  prints x to buf, including alternate formats and
252    additional precision digits, but not field characters or the sign */
253 static inline void FUNC_NAME(pf_integer_conv)(APICHAR *buf, int buf_len,
254         FUNC_NAME(pf_flags) *flags, LONGLONG x)
255 {
256     unsigned int base;
257     const char *digits;
258     int i, j, k;
259
260     if(flags->Format == 'o')
261         base = 8;
262     else if(flags->Format=='x' || flags->Format=='X')
263         base = 16;
264     else
265         base = 10;
266
267     if(flags->Format == 'X')
268         digits = "0123456789ABCDEFX";
269     else
270         digits = "0123456789abcdefx";
271
272     if(x<0 && (flags->Format=='d' || flags->Format=='i')) {
273         x = -x;
274         flags->Sign = '-';
275     }
276
277     i = 0;
278     if(x==0 && flags->Precision)
279         buf[i++] = '0';
280     else {
281         while(x != 0) {
282             j = (ULONGLONG)x%base;
283             x = (ULONGLONG)x/base;
284             buf[i++] = digits[j];
285         }
286     }
287     k = flags->Precision-i;
288     while(k-- > 0)
289         buf[i++] = '0';
290     if(flags->Alternate) {
291         if(base == 16) {
292             buf[i++] = digits[16];
293             buf[i++] = '0';
294         } else if(base==8 && buf[i-1]!='0')
295             buf[i++] = '0';
296     }
297
298     /* Adjust precision so pf_fill won't truncate the number later */
299     flags->Precision = i;
300
301     buf[i] = '\0';
302     j = 0;
303     while(--i > j) {
304         APICHAR tmp = buf[j];
305         buf[j] = buf[i];
306         buf[i] = tmp;
307         j++;
308     }
309 }
310
311 static inline void FUNC_NAME(pf_fixup_exponent)(char *buf)
312 {
313     char* tmp = buf;
314
315     while(tmp[0] && toupper(tmp[0])!='E')
316         tmp++;
317
318     if(tmp[0] && (tmp[1]=='+' || tmp[1]=='-') &&
319             isdigit(tmp[2]) && isdigit(tmp[3])) {
320         char final;
321
322         if (isdigit(tmp[4]))
323             return; /* Exponent already 3 digits */
324
325         tmp += 2;
326         final = tmp[2];
327         tmp[2] = tmp[1];
328         tmp[1] = tmp[0];
329         tmp[0] = '0';
330
331         if(final == '\0') {
332             tmp[3] = '\0';
333             if(buf[0] == ' ')
334                 memmove(buf, buf + 1, (tmp - buf) + 3);
335         }
336     }
337 }
338
339 int FUNC_NAME(pf_printf)(FUNC_NAME(puts_clbk) pf_puts, void *puts_ctx, const APICHAR *fmt,
340         MSVCRT__locale_t locale, BOOL positional_params, BOOL invoke_invalid_param_handler,
341         args_clbk pf_args, void *args_ctx, __ms_va_list *valist)
342 {
343     MSVCRT_pthreadlocinfo locinfo;
344     const APICHAR *q, *p = fmt;
345     APICHAR buf[32];
346     int written = 0, pos, i;
347     FUNC_NAME(pf_flags) flags;
348
349     TRACE("Format is: %s\n", FUNC_NAME(debugstr)(fmt));
350
351     if(!locale)
352         locinfo = get_locinfo();
353     else
354         locinfo = locale->locinfo;
355
356     while(*p) {
357         /* output characters before '%' */
358         for(q=p; *q && *q!='%'; q++);
359         if(p != q) {
360             i = pf_puts(puts_ctx, q-p, p);
361             if(i < 0)
362                 return i;
363
364             written += i;
365             p = q;
366             continue;
367         }
368
369         /* *p == '%' here */
370         p++;
371
372         /* output a single '%' character */
373         if(*p == '%') {
374             i = pf_puts(puts_ctx, 1, p++);
375             if(i < 0)
376                 return i;
377
378             written += i;
379             continue;
380         }
381
382         /* check parameter position */
383         if(positional_params && (q = FUNC_NAME(pf_parse_int)(p, &pos)) && *q=='$')
384             p = q+1;
385         else
386             pos = -1;
387
388         /* parse the flags */
389         memset(&flags, 0, sizeof(flags));
390         while(*p) {
391             if(*p=='+' || *p==' ') {
392                 if(flags.Sign != '+')
393                     flags.Sign = *p;
394             } else if(*p == '-')
395                 flags.LeftAlign = *p;
396             else if(*p == '0')
397                 flags.PadZero = *p;
398             else if(*p == '#')
399                 flags.Alternate = *p;
400             else
401                 break;
402
403             p++;
404         }
405
406         /* parse the widh */
407         if(*p == '*') {
408             p++;
409             if(positional_params && (q = FUNC_NAME(pf_parse_int)(p, &i)) && *q=='$')
410                 p = q+1;
411             else
412                 i = -1;
413
414             flags.FieldLength = pf_args(args_ctx, i, VT_INT, valist).get_int;
415             if(flags.FieldLength < 0) {
416                 flags.LeftAlign = '-';
417                 flags.FieldLength = -flags.FieldLength;
418             }
419         } else while(isdigit(*p)) {
420             flags.FieldLength *= 10;
421             flags.FieldLength += *p++ - '0';
422         }
423
424         /* parse the precision */
425         flags.Precision = -1;
426         if(*p == '.') {
427             flags.Precision = 0;
428             p++;
429             if(*p == '*') {
430                 p++;
431                 if(positional_params && (q = FUNC_NAME(pf_parse_int)(p, &i)) && *q=='$')
432                     p = q+1;
433                 else
434                     i = -1;
435
436                 flags.Precision = pf_args(args_ctx, i, VT_INT, valist).get_int;
437             } else while(isdigit(*p)) {
438                 flags.Precision *= 10;
439                 flags.Precision += *p++ - '0';
440             }
441         }
442
443         /* parse argument size modifier */
444         while(*p) {
445             if(*p=='l' && *(p+1)=='l') {
446                 flags.IntegerDouble++;
447                 p += 2;
448             } else if(*p=='h' || *p=='l' || *p=='L') {
449                 flags.IntegerLength = *p;
450                 p++;
451             } else if(*p == 'I') {
452                 if(*(p+1)=='6' && *(p+2)=='4') {
453                     flags.IntegerDouble++;
454                     p += 3;
455                 } else if(*(p+1)=='3' && *(p+2)=='2')
456                     p += 3;
457                 else if(isdigit(*(p+1)) || !*(p+1))
458                     break;
459                 else
460                     p++;
461             } else if(*p == 'w')
462                 flags.WideString = *p++;
463             else if(*p == 'F')
464                 p++; /* ignore */
465             else
466                 break;
467         }
468
469         flags.Format = *p;
470
471         if(flags.Format == 's' || flags.Format == 'S') {
472             i = FUNC_NAME(pf_handle_string)(pf_puts, puts_ctx,
473                     pf_args(args_ctx, pos, VT_PTR, valist).get_ptr,
474                     -1,  &flags, locinfo);
475         } else if(flags.Format == 'c' || flags.Format == 'C') {
476             int ch = pf_args(args_ctx, pos, VT_INT, valist).get_int;
477
478             if((ch&0xff) != ch)
479                 FIXME("multibyte characters printing not supported\n");
480
481             i = FUNC_NAME(pf_handle_string)(pf_puts, puts_ctx, &ch, 1, &flags, locinfo);
482         } else if(flags.Format == 'p') {
483             flags.Format = 'X';
484             flags.PadZero = '0';
485             i = flags.Precision;
486             flags.Precision = 2*sizeof(void*);
487             FUNC_NAME(pf_integer_conv)(buf, sizeof(buf)/sizeof(APICHAR), &flags,
488                                        (ULONG_PTR)pf_args(args_ctx, pos, VT_PTR, valist).get_ptr);
489             flags.PadZero = 0;
490             flags.Precision = i;
491
492 #ifdef PRINTF_WIDE
493             i = FUNC_NAME(pf_output_format_wstr)(pf_puts, puts_ctx, buf, -1, &flags, locinfo);
494 #else
495             i = FUNC_NAME(pf_output_format_str)(pf_puts, puts_ctx, buf, -1, &flags, locinfo);
496 #endif
497         } else if(flags.Format == 'n') {
498             int *used;
499
500             if(!n_format_enabled) {
501                 MSVCRT_INVALID_PMT("\'n\' format specifier disabled");
502                 *MSVCRT__errno() = MSVCRT_EINVAL;
503                 return -1;
504             }
505
506             used = pf_args(args_ctx, pos, VT_PTR, valist).get_ptr;
507             *used = written;
508             i = 0;
509         } else if(flags.Format && strchr("diouxX", flags.Format)) {
510             APICHAR *tmp = buf;
511             int max_len = (flags.FieldLength>flags.Precision ? flags.FieldLength : flags.Precision) + 10;
512
513             if(max_len > sizeof(buf)/sizeof(APICHAR))
514                 tmp = HeapAlloc(GetProcessHeap(), 0, max_len);
515             if(!tmp)
516                 return -1;
517
518             if(flags.IntegerDouble)
519                 FUNC_NAME(pf_integer_conv)(tmp, max_len, &flags, pf_args(args_ctx, pos,
520                             VT_I8, valist).get_longlong);
521             else if(flags.Format=='d' || flags.Format=='i')
522                 FUNC_NAME(pf_integer_conv)(tmp, max_len, &flags, pf_args(args_ctx, pos,
523                             VT_INT, valist).get_int);
524             else
525                 FUNC_NAME(pf_integer_conv)(tmp, max_len, &flags, (unsigned)pf_args(
526                             args_ctx, pos, VT_INT, valist).get_int);
527
528 #ifdef PRINTF_WIDE
529             i = FUNC_NAME(pf_output_format_wstr)(pf_puts, puts_ctx, tmp, -1, &flags, locinfo);
530 #else
531             i = FUNC_NAME(pf_output_format_str)(pf_puts, puts_ctx, tmp, -1, &flags, locinfo);
532 #endif
533             if(tmp != buf)
534                 HeapFree(GetProcessHeap(), 0, tmp);
535         } else if(flags.Format && strchr("aeEfgG", flags.Format)) {
536             char float_fmt[20], buf_a[32], *tmp = buf_a, *decimal_point;
537             int max_len = (flags.FieldLength>flags.Precision ? flags.FieldLength : flags.Precision) + 10;
538             double val = pf_args(args_ctx, pos, VT_R8, valist).get_double;
539
540             if(flags.Format=='f') {
541                 if(val>-10.0 && val<10.0)
542                     i = 1;
543                 else
544                     i = 1 + log10(val<0 ? -val : val);
545                 /* Default precision is 6, additional space for sign, separator and nullbyte is required */
546                 i += (flags.Precision==-1 ? 6 : flags.Precision) + 3;
547
548                 if(i > max_len)
549                     max_len = i;
550             }
551
552             if(max_len > sizeof(buf_a))
553                 tmp = HeapAlloc(GetProcessHeap(), 0, max_len);
554             if(!tmp)
555                 return -1;
556
557             FUNC_NAME(pf_rebuild_format_string)(float_fmt, &flags);
558
559             sprintf(tmp, float_fmt, val);
560             if(toupper(flags.Format)=='E' || toupper(flags.Format)=='G')
561                 FUNC_NAME(pf_fixup_exponent)(tmp);
562
563             decimal_point = strchr(tmp, '.');
564             if(decimal_point)
565                 *decimal_point = *locinfo->lconv->decimal_point;
566
567             i = FUNC_NAME(pf_output_str)(pf_puts, puts_ctx, tmp, strlen(tmp), locinfo);
568             if(tmp != buf_a)
569                 HeapFree(GetProcessHeap(), 0, tmp);
570         } else {
571             if(invoke_invalid_param_handler) {
572                 MSVCRT__invalid_parameter(NULL, NULL, NULL, 0, 0);
573                 *MSVCRT__errno() = MSVCRT_EINVAL;
574                 return -1;
575             }
576
577             continue;
578         }
579
580         if(i < 0)
581             return i;
582         written += i;
583         p++;
584     }
585
586     return written;
587 }
588
589 #ifndef PRINTF_WIDE
590 enum types_clbk_flags {
591     TYPE_CLBK_VA_LIST = 1,
592     TYPE_CLBK_POSITIONAL = 2,
593     TYPE_CLBK_ERROR_POS = 4,
594     TYPE_CLBK_ERROR_TYPE = 8
595 };
596
597 /* This functions stores types of arguments. It uses args[0] internally */
598 static printf_arg arg_clbk_type(void *ctx, int pos, int type, __ms_va_list *valist)
599 {
600     static const printf_arg ret;
601     printf_arg *args = ctx;
602
603     if(pos == -1) {
604         args[0].get_int |= TYPE_CLBK_VA_LIST;
605         return ret;
606     } else
607         args[0].get_int |= TYPE_CLBK_POSITIONAL;
608
609     if(pos<1 || pos>MSVCRT__ARGMAX)
610         args[0].get_int |= TYPE_CLBK_ERROR_POS;
611     else if(args[pos].get_int && args[pos].get_int!=type)
612         args[0].get_int |= TYPE_CLBK_ERROR_TYPE;
613     else
614         args[pos].get_int = type;
615
616     return ret;
617 }
618 #endif
619
620 static int FUNC_NAME(create_positional_ctx)(void *args_ctx, const APICHAR *format, __ms_va_list valist)
621 {
622     struct FUNC_NAME(_str_ctx) puts_ctx = {INT_MAX, NULL};
623     printf_arg *args = args_ctx;
624     int i, j;
625
626     i = FUNC_NAME(pf_printf)(FUNC_NAME(puts_clbk_str), &puts_ctx, format, NULL, TRUE, FALSE,
627             arg_clbk_type, args_ctx, NULL);
628     if(i < 0)
629         return i;
630
631     if(args[0].get_int==0 || args[0].get_int==TYPE_CLBK_VA_LIST)
632         return 0;
633     if(args[0].get_int != TYPE_CLBK_POSITIONAL)
634         return -1;
635
636     for(i=MSVCRT__ARGMAX; i>0; i--)
637         if(args[i].get_int)
638             break;
639
640     for(j=1; j<=i; j++) {
641         switch(args[j].get_int) {
642         case VT_I8:
643             args[j].get_longlong = va_arg(valist, LONGLONG);
644             break;
645         case VT_INT:
646             args[j].get_int = va_arg(valist, int);
647             break;
648         case VT_R8:
649             args[j].get_double = va_arg(valist, double);
650             break;
651         case VT_PTR:
652             args[j].get_ptr = va_arg(valist, void*);
653             break;
654         default:
655             return -1;
656         }
657     }
658
659     return j;
660 }
661
662 #undef APICHAR
663 #undef CONVCHAR
664 #undef FUNC_NAME