stdio/FreeBSD/xprintf_str.c

   1 /*-
   2  * Copyright (c) 2005 Poul-Henning Kamp
   3  * Copyright (c) 1990, 1993
   4  *      The Regents of the University of California.  All rights reserved.
   5  *
   6  * This code is derived from software contributed to Berkeley by
   7  * Chris Torek.
   8  *
   9  * Redistribution and use in source and binary forms, with or without
  10  * modification, are permitted provided that the following conditions
  11  * are met:
  12  * 1. Redistributions of source code must retain the above copyright
  13  *    notice, this list of conditions and the following disclaimer.
  14  * 2. Redistributions in binary form must reproduce the above copyright
  15  *    notice, this list of conditions and the following disclaimer in the
  16  *    documentation and/or other materials provided with the distribution.
  17  * 3. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  *
  33  * $FreeBSD: src/lib/libc/stdio/xprintf_str.c,v 1.1 2005/12/16 18:56:38 phk Exp $
  34  */
  35
  36 #include <namespace.h>
  37 #include <stdio.h>
  38 #include <stdlib.h>
  39 #include <string.h>
  40 #include <limits.h>
  41 #include <stdint.h>
  42 #include <assert.h>
  43 #include <wchar.h>
  44 #include "printf.h"
  45 #include "xprintf_private.h"
  46
  47 /*
  48  * Convert a wide character string argument for the %ls format to a multibyte
  49  * string representation. If not -1, prec specifies the maximum number of
  50  * bytes to output, and also means that we can't assume that the wide char.
  51  * string ends is null-terminated.
  52  */
  53 static char *
  54 __wcsconv(wchar_t *wcsarg, int prec, locale_t loc)
  55 {
  56         static const mbstate_t initial;
  57         mbstate_t mbs;
  58         char buf[MB_LEN_MAX];
  59         wchar_t *p;
  60         char *convbuf;
  61         size_t clen, nbytes;
  62
  63         /* Allocate space for the maximum number of bytes we could output. */
  64         if (prec < 0) {
  65                 p = wcsarg;
  66                 mbs = initial;
  67                 nbytes = wcsrtombs_l(NULL, (const wchar_t **)&p, 0, &mbs, loc);
  68                 if (nbytes == (size_t)-1)
  69                         return (NULL);
  70         } else {
  71                 /*
  72                  * Optimisation: if the output precision is small enough,
  73                  * just allocate enough memory for the maximum instead of
  74                  * scanning the string.
  75                  */
  76                 if (prec < 128)
  77                         nbytes = prec;
  78                 else {
  79                         nbytes = 0;
  80                         p = wcsarg;
  81                         mbs = initial;
  82                         for (;;) {
  83                                 clen = wcrtomb_l(buf, *p++, &mbs, loc);
  84                                 if (clen == 0 || clen == (size_t)-1 ||
  85                                     (int)(nbytes + clen) > prec)
  86                                         break;
  87                                 nbytes += clen;
  88                         }
  89                 }
  90         }
  91         if ((convbuf = MALLOC(nbytes + 1)) == NULL)
  92                 return (NULL);
  93
  94         /* Fill the output buffer. */
  95         p = wcsarg;
  96         mbs = initial;
  97         if ((nbytes = wcsrtombs_l(convbuf, (const wchar_t **)&p,
  98             nbytes, &mbs, loc)) == (size_t)-1) {
  99                 free(convbuf);
 100                 return (NULL);
 101         }
 102         convbuf[nbytes] = '\0';
 103         return (convbuf);
 104 }
 105
 106
 107 /* 's' ---------------------------------------------------------------*/
 108
 109 __private_extern__ int
 110 __printf_arginfo_str(const struct printf_info *pi, size_t n, int *argt)
 111 {
 112
 113         assert (n > 0);
 114         if (pi->is_long || pi->spec == 'C')
 115                 argt[0] = PA_WSTRING;
 116         else
 117                 argt[0] = PA_STRING;
 118         return (1);
 119 }
 120
 121 __private_extern__ int
 122 __printf_render_str(struct __printf_io *io, const struct printf_info *pi, const void *const *arg)
 123 {
 124         const char *p;
 125         wchar_t *wcp;
 126         char *convbuf;
 127         int l;
 128
 129         if (pi->is_long || pi->spec == 'S') {
 130                 wcp = *((wint_t **)arg[0]);
 131                 if (wcp == NULL)
 132                         return (__printf_out(io, pi, "(null)", 6));
 133                 convbuf = __wcsconv(wcp, pi->prec, pi->loc);
 134                 if (convbuf == NULL)
 135                         return (-1);
 136                 l = __printf_out(io, pi, convbuf, strlen(convbuf));
 137                 __printf_flush(io);
 138                 free(convbuf);
 139                 return (l);
 140         }
 141         p = *((char **)arg[0]);
 142         if (p == NULL)
 143                 return (__printf_out(io, pi, "(null)", 6));
 144         l = strlen(p);
 145         if (pi->prec >= 0 && pi->prec < l)
 146                 l = pi->prec;
 147         return (__printf_out(io, pi, p, l));
 148 }
 149
 150 /* 'c' ---------------------------------------------------------------*/
 151
 152 __private_extern__ int
 153 __printf_arginfo_chr(const struct printf_info *pi, size_t n, int *argt)
 154 {
 155
 156         assert (n > 0);
 157 #ifdef VECTORS
 158         if (pi->is_vec)
 159                 argt[0] = PA_VECTOR;
 160         else
 161 #endif /* VECTORS */
 162         if (pi->is_long || pi->spec == 'C')
 163                 argt[0] = PA_WCHAR;
 164         else
 165                 argt[0] = PA_INT;
 166         return (1);
 167 }
 168
 169 __private_extern__ int
 170 __printf_render_chr(struct __printf_io *io, const struct printf_info *pi, const void *const *arg)
 171 {
 172         int i;
 173         wint_t ii;
 174         unsigned char c;
 175         static const mbstate_t initial;         /* XXX: this is bogus! */
 176         mbstate_t mbs;
 177         size_t mbseqlen;
 178         char buf[MB_CUR_MAX_L(pi->loc)];
 179
 180 #ifdef VECTORS
 181         if (pi->is_vec) return __xprintf_vector(io, pi, arg);
 182 #endif /* VECTORS */
 183
 184         if (pi->is_long || pi->spec == 'C') {
 185                 int ret;
 186                 ii = *((wint_t *)arg[0]);
 187
 188                 mbs = initial;
 189                 mbseqlen = wcrtomb_l(buf, (wchar_t)ii, &mbs, pi->loc);
 190                 if (mbseqlen == (size_t) -1)
 191                         return (-1);
 192                 ret = __printf_out(io, pi, buf, mbseqlen);
 193                 __printf_flush(io);
 194                 return ret;
 195         }
 196         i = *((int *)arg[0]);
 197         c = i;
 198         i = __printf_out(io, pi, &c, 1);
 199         __printf_flush(io);
 200         return (i);
 201 }