2002-09-02 Jakub Jelinek <jakub@redhat.com>
[kopensolaris-gnu/glibc.git] / wcsmbs / mbsrtowcs.c
1 /* Copyright (C) 1996,1997,1998,1999,2000,2002 Free Software Foundation, Inc.
2    This file is part of the GNU C Library.
3    Contributed by Ulrich Drepper <drepper@gnu.org>, 1996.
4
5    The GNU C Library is free software; you can redistribute it and/or
6    modify it under the terms of the GNU Lesser General Public
7    License as published by the Free Software Foundation; either
8    version 2.1 of the License, or (at your option) any later version.
9
10    The GNU C Library is distributed in the hope that it will be useful,
11    but WITHOUT ANY WARRANTY; without even the implied warranty of
12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13    Lesser General Public License for more details.
14
15    You should have received a copy of the GNU Lesser General Public
16    License along with the GNU C Library; if not, write to the Free
17    Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
18    02111-1307 USA.  */
19
20 #include <dlfcn.h>
21 #include <errno.h>
22 #include <gconv.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <wchar.h>
26 #include <wcsmbsload.h>
27
28 #include <assert.h>
29
30 #ifndef EILSEQ
31 # define EILSEQ EINVAL
32 #endif
33
34
35 #ifdef USE_IN_EXTENDED_LOCALE_MODEL
36 size_t
37 attribute_hidden
38 __mbsrtowcs_l (dst, src, len, ps, l)
39      wchar_t *dst;
40      const char **src;
41      size_t len;
42      mbstate_t *ps;
43      __locale_t l;
44 #else
45 /* This is the private state used if PS is NULL.  */
46 static mbstate_t state;
47
48 size_t
49 __mbsrtowcs (dst, src, len, ps)
50      wchar_t *dst;
51      const char **src;
52      size_t len;
53      mbstate_t *ps;
54 #endif
55 {
56   struct __gconv_step_data data;
57   size_t result;
58   int status;
59   struct __gconv_step *towc;
60   size_t non_reversible;
61   const struct gconv_fcts *fcts;
62
63   /* Tell where we want the result.  */
64   data.__invocation_counter = 0;
65   data.__internal_use = 1;
66   data.__flags = __GCONV_IS_LAST;
67 #ifdef USE_IN_EXTENDED_LOCALE_MODEL
68   data.__statep = ps;
69 #else
70   data.__statep = ps ?: &state;
71 #endif
72   data.__trans = NULL;
73
74   /* Get the conversion functions.  */
75 #ifdef USE_IN_EXTENDED_LOCALE_MODEL
76   fcts = get_gconv_fcts (l->__locales[LC_CTYPE]);
77 #else
78   fcts = get_gconv_fcts (_NL_CURRENT_DATA (LC_CTYPE));
79 #endif
80
81   /* Get the structure with the function pointers.  */
82   towc = fcts->towc;
83
84   /* We have to handle DST == NULL special.  */
85   if (dst == NULL)
86     {
87       mbstate_t temp_state;
88       wchar_t buf[64];          /* Just an arbitrary size.  */
89       const unsigned char *inbuf = (const unsigned char *) *src;
90       const unsigned char *srcend = inbuf + strlen (inbuf) + 1;
91
92       temp_state = *data.__statep;
93       data.__statep = &temp_state;
94
95       result = 0;
96       data.__outbufend = (char *) buf + sizeof (buf);
97       do
98         {
99           data.__outbuf = (char *) buf;
100
101           status = DL_CALL_FCT (towc->__fct,
102                                 (towc, &data, &inbuf, srcend, NULL,
103                                  &non_reversible, 0, 1));
104
105           result += (wchar_t *) data.__outbuf - buf;
106         }
107       while (status == __GCONV_FULL_OUTPUT);
108
109       if (status == __GCONV_OK || status == __GCONV_EMPTY_INPUT)
110         {
111           /* There better should be a NUL wide char at the end.  */
112           assert (((wchar_t *) data.__outbuf)[-1] == L'\0');
113           /* Don't count the NUL character in.  */
114           --result;
115         }
116     }
117   else
118     {
119       /* This code is based on the safe assumption that all internal
120          multi-byte encodings use the NUL byte only to mark the end
121          of the string.  */
122       const unsigned char *srcend;
123
124       srcend = (const unsigned char *) (*src
125                                         + __strnlen (*src, len * MB_CUR_MAX)
126                                         + 1);
127
128       data.__outbuf = (unsigned char *) dst;
129       data.__outbufend = data.__outbuf + len * sizeof (wchar_t);
130
131       status = DL_CALL_FCT (towc->__fct,
132                             (towc, &data, (const unsigned char **) src, srcend,
133                              NULL, &non_reversible, 0, 1));
134
135       result = (wchar_t *) data.__outbuf - dst;
136
137       /* We have to determine whether the last character converted
138          is the NUL character.  */
139       if ((status == __GCONV_OK || status == __GCONV_EMPTY_INPUT)
140           && ((wchar_t *) dst)[result - 1] == L'\0')
141         {
142           assert (result > 0);
143           assert (__mbsinit (data.__statep));
144           *src = NULL;
145           --result;
146         }
147     }
148
149   /* There must not be any problems with the conversion but illegal input
150      characters.  */
151   assert (status == __GCONV_OK || status == __GCONV_EMPTY_INPUT
152           || status == __GCONV_ILLEGAL_INPUT
153           || status == __GCONV_INCOMPLETE_INPUT
154           || status == __GCONV_FULL_OUTPUT);
155
156   if (status != __GCONV_OK && status != __GCONV_FULL_OUTPUT
157       && status != __GCONV_EMPTY_INPUT && status != __GCONV_INCOMPLETE_INPUT)
158     {
159       result = (size_t) -1;
160       __set_errno (EILSEQ);
161     }
162
163   return result;
164 }
165 #ifndef USE_IN_EXTENDED_LOCALE_MODEL
166 weak_alias (__mbsrtowcs, mbsrtowcs)
167 #endif