1 
2 /*  Copyright (C) 2002, 2003, 2004     Manuel Novoa III
3  *
4  *  This library is free software; you can redistribute it and/or
5  *  modify it under the terms of the GNU Library General Public
6  *  License as published by the Free Software Foundation; either
7  *  version 2 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  *  Library General Public License for more details.
13  *
14  *  You should have received a copy of the GNU Library General Public
15  *  License along with this library; if not, see
16  *  <http://www.gnu.org/licenses/>.
17  */
18 
19 /*  ATTENTION!   ATTENTION!   ATTENTION!   ATTENTION!   ATTENTION!
20  *
21  *  Besides uClibc, I'm using this code in my libc for elks, which is
22  *  a 16-bit environment with a fairly limited compiler.  It would make
23  *  things much easier for me if this file isn't modified unnecessarily.
24  *  In particular, please put any new or replacement functions somewhere
25  *  else, and modify the makefile to use your version instead.
26  *  Thanks.  Manuel
27  *
28  *  ATTENTION!   ATTENTION!   ATTENTION!   ATTENTION!   ATTENTION! */
29 
30 
31 /* May 23, 2002     Initial Notes:
32  *
33  * I'm still tweaking this stuff, but it passes the tests I've thrown
34  * at it, and Erik needs it for the gcc port.  The glibc extension
35  * __wcsnrtombs() hasn't been tested, as I didn't find a test for it
36  * in the glibc source.  I also need to fix the behavior of
37  * _wchar_utf8sntowcs() if the max number of wchars to convert is 0.
38  *
39  * UTF-8 -> wchar -> UTF-8 conversion tests on Markus Kuhn's UTF-8-demo.txt
40  * file on my platform (x86) show about 5-10% faster conversion speed than
41  * glibc with mbsrtowcs()/wcsrtombs() and almost twice as fast as glibc with
42  * individual mbrtowc()/wcrtomb() calls.
43  *
44  * If 'DECODER' is defined, then _wchar_utf8sntowcs() will be compiled
45  * as a fail-safe UTF-8 decoder appropriate for a terminal, etc.  which
46  * needs to deal gracefully with whatever is sent to it.  In that mode,
47  * it passes Markus Kuhn's UTF-8-test.txt stress test.  I plan to add
48  * an arg to force that behavior, so the interface will be changing.
49  *
50  * I need to fix the error checking for 16-bit wide chars.  This isn't
51  * an issue for uClibc, but may be for ELKS.  I'm currently not sure
52  * if I'll use 16-bit, 32-bit, or configureable wchars in ELKS.
53  *
54  * July 1, 2002
55  *
56  * Fixed _wchar_utf8sntowcs() for the max number of wchars == 0 case.
57  * Fixed nul-char bug in btowc(), and another in __mbsnrtowcs() for 8-bit
58  *    locales.
59  * Enabled building of a C/POSIX-locale-only version, so full locale support
60  *    no longer needs to be enabled.
61  *
62  * Nov 4, 2002
63  *
64  * Fixed a bug in _wchar_wcsntoutf8s().  Don't store wcs position if dst is NULL.
65  * Also, introduce an awful hack into _wchar_wcsntoutf8s() and wcsrtombs() in
66  *   order to support %ls in printf.  See comments below for details.
67  * Change behaviour of wc<->mb functions when in the C locale.  Now they do
68  *   a 1-1 map for the range 0x80-UCHAR_MAX.  This is for backwards compatibility
69  *   and consistency with the stds requirements that a printf format string by
70  *   a valid multibyte string beginning and ending in it's initial shift state.
71  *
72  * Nov 5, 2002
73  *
74  * Forgot to change btowc and wctob when I changed the wc<->mb functions yesterday.
75  *
76  * Nov 7, 2002
77  *
78  * Add wcwidth and wcswidth, based on Markus Kuhn's wcwidth of 2002-05-08.
79  *   Added some size/speed optimizations and integrated it into my locale
80  *   framework.  Minimally tested at the moment, but the stub C-locale
81  *   version (which most people would probably be using) should be fine.
82  *
83  * Nov 21, 2002
84  *
85  * Revert the wc<->mb changes from earlier this month involving the C-locale.
86  * Add a couple of ugly hacks to support *wprintf.
87  * Add a mini iconv() and iconv implementation (requires locale support).
88  *
89  * Aug 1, 2003
90  * Bug fix for mbrtowc.
91  *
92  * Aug 18, 2003
93  * Bug fix: _wchar_utf8sntowcs and _wchar_wcsntoutf8s now set errno if EILSEQ.
94  *
95  * Feb 11, 2004
96  * Bug fix: Fix size check for remaining output space in iconv().
97  *
98  * Manuel
99  */
100 
101 /* keep libgen before string.h - and porting.h to use the
102  * XPG version of basename */
103 #include <libgen.h>
104 #include "porting.h"
105 #include <string.h>
106 #include <iconv.h>
107 #include <stdarg.h>
108 #include <wchar.h>
109 #include "wchar.c" /* for _UC_iconv_t and __iconv_codesets */
110 
111 #ifdef L_iconv_main
112 static
113 #else
114 extern
115 #endif
116 const unsigned char __iconv_codesets[];
117 
118 #define IBUF BUFSIZ
119 #define OBUF BUFSIZ
120 
121 static char *progname;
122 static int hide_errors;
123 
124 static void error_msg(const char *fmt, ...)
125 	 __attribute__ ((noreturn, format (printf, 1, 2)));
126 
error_msg(const char * fmt,...)127 static void error_msg(const char *fmt, ...)
128 {
129 	va_list arg;
130 
131 	if (!hide_errors) {
132 		fprintf(stderr, "%s: ", progname);
133 		va_start(arg, fmt);
134 		vfprintf(stderr, fmt, arg);
135 		va_end(arg);
136 	}
137 
138 	exit(EXIT_FAILURE);
139 }
140 
main(int argc,char ** argv)141 int main(int argc, char **argv)
142 {
143 	FILE *ifile;
144 	FILE *ofile = stdout;
145 	const char *p;
146 	const char *s;
147 	static const char opt_chars[] = "tfocsl";
148 	                              /* 012345 */
149 	const char *opts[sizeof(opt_chars)]; /* last is infile name */
150 	iconv_t ic;
151 	char ibuf[IBUF];
152 	char obuf[OBUF];
153 	char *pi;
154 	char *po;
155 	size_t ni, no, r, pos;
156 
157 	hide_errors = 0;
158 
159 	for (s = opt_chars ; *s ; s++) {
160 		opts[ s - opt_chars ] = NULL;
161 	}
162 
163 	progname = *argv;
164 	while (--argc) {
165 		p = *++argv;
166 		if ((*p != '-') || (*++p == 0)) {
167 			break;
168 		}
169 		do {
170 			if ((s = strchr(opt_chars,*p)) == NULL) {
171 			USAGE:
172 				s = basename(progname);
173 				fprintf(stderr,
174 						"%s [-cs] -f fromcode -t tocode [-o outputfile] [inputfile ...]\n"
175 						"  or\n%s -l\n", s, s);
176 				return EXIT_FAILURE;
177 			}
178 			if ((s - opt_chars) < 3) {
179 				if ((--argc == 0) || opts[s - opt_chars]) {
180 					goto USAGE;
181 				}
182 				opts[s - opt_chars] = *++argv;
183 			} else {
184 				opts[s - opt_chars] = p;
185 			}
186 		} while (*++p);
187 	}
188 
189 	if (opts[5]) {				/* -l */
190 		fprintf(stderr, "Recognized codesets:\n");
191 		for (s = (char *)__iconv_codesets ; *s ; s += *s) {
192 			fprintf(stderr,"  %s\n", s+2);
193 		}
194 		s = __LOCALE_DATA_CODESET_LIST;
195 		do {
196 			fprintf(stderr,"  %s\n", __LOCALE_DATA_CODESET_LIST+ (unsigned char)(*s));
197 		} while (*++s);
198 
199 		return EXIT_SUCCESS;
200 	}
201 
202 	if (opts[4]) {
203 		hide_errors = 1;
204 	}
205 
206 	if (!opts[0] || !opts[1]) {
207 		goto USAGE;
208 	}
209 	if ((ic = iconv_open(opts[0],opts[1])) == ((iconv_t)(-1))) {
210 		error_msg( "unsupported codeset in %s -> %s conversion\n", opts[1], opts[0]);
211 	}
212 	if (opts[3]) {				/* -c */
213 		((_UC_iconv_t *) ic)->skip_invalid_input = 1;
214 	}
215 
216 	if ((s = opts[2]) != NULL) {
217 		if (!(ofile = fopen(s, "w"))) {
218 			error_msg( "couldn't open %s for writing\n", s);
219 		}
220 	}
221 
222 	pos = ni = 0;
223 	do {
224 		if (!argc || ((**argv == '-') && !((*argv)[1]))) {
225 			ifile = stdin;		/* we don't check for duplicates */
226 		} else if (!(ifile = fopen(*argv, "r"))) {
227 			error_msg( "couldn't open %s for reading\n", *argv);
228 		}
229 
230 		while ((r = fread(ibuf + ni, 1, IBUF - ni, ifile)) > 0) {
231 			pos += r;
232 			ni += r;
233 			no = OBUF;
234 			pi = ibuf;
235 			po = obuf;
236 			if ((r = iconv(ic, &pi, &ni, &po, &no)) == ((size_t)(-1))) {
237 				if ((errno != EINVAL) && (errno != E2BIG)) {
238 					error_msg( "iconv failed at pos %lu : %m\n", (unsigned long) (pos - ni));
239 				}
240 			}
241 			if ((r = OBUF - no) > 0) {
242 				if (fwrite(obuf, 1, OBUF - no, ofile) < r) {
243 					error_msg( "write error\n");
244 				}
245 			}
246 			if (ni) {			/* still bytes in buffer! */
247 				memmove(ibuf, pi, ni);
248 			}
249 		}
250 
251 		if (ferror(ifile)) {
252 			error_msg( "read error\n");
253 		}
254 
255 		++argv;
256 
257 		if (ifile != stdin) {
258 			fclose(ifile);
259 		}
260 
261 	} while (--argc > 0);
262 
263 	iconv_close(ic);
264 
265 	if (ni) {
266 		error_msg( "incomplete sequence\n");
267 	}
268 
269 	return (((_UC_iconv_t *) ic)->skip_invalid_input < 2)
270 		? EXIT_SUCCESS : EXIT_FAILURE;
271 }
272