1/*
2 * Copyright 2010 Nexenta Systems, Inc.  All rights reserved.
3 * Copyright 2015 John Marino <draco@marino.st>
4 *
5 * This source code is derived from the illumos localedef command, and
6 * provided under BSD-style license terms by Nexenta Systems, Inc.
7 *
8 * Redistribution and use in source and binary forms, with or without
9 * modification, are permitted provided that the following conditions
10 * are met:
11 *
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 *
18 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
19 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21 * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
22 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
23 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
24 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
25 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
26 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
28 * POSSIBILITY OF SUCH DAMAGE.
29 */
30
31/*
32 * POSIX localedef.
33 */
34#include <sys/cdefs.h>
35__FBSDID("$FreeBSD: releng/11.0/usr.bin/localedef/localedef.c 290510 2015-11-07 20:27:31Z bapt $");
36
37#include <stdio.h>
38#include <stdlib.h>
39#include <errno.h>
40#include <sys/types.h>
41#include <sys/stat.h>
42#include <string.h>
43#include <unistd.h>
44#include <libgen.h>
45#include <stddef.h>
46#include <unistd.h>
47#include <limits.h>
48#include <locale.h>
49#include <dirent.h>
50#include "localedef.h"
51#include "parser.h"
52
53#ifndef	TEXT_DOMAIN
54#define	TEXT_DOMAIN	"SYS_TEST"
55#endif
56
57static int bsd = 0;
58int verbose = 0;
59int undefok = 0;
60int warnok = 0;
61static char *locname = NULL;
62static char locpath[PATH_MAX];
63
64const char *
65category_name(void)
66{
67	switch (get_category()) {
68	case T_CHARMAP:
69		return ("CHARMAP");
70	case T_WIDTH:
71		return ("WIDTH");
72	case T_COLLATE:
73		return ("LC_COLLATE");
74	case T_CTYPE:
75		return ("LC_CTYPE");
76	case T_MESSAGES:
77		return ("LC_MESSAGES");
78	case T_MONETARY:
79		return ("LC_MONETARY");
80	case T_NUMERIC:
81		return ("LC_NUMERIC");
82	case T_TIME:
83		return ("LC_TIME");
84	default:
85		INTERR;
86		return (NULL);
87	}
88}
89
90static char *
91category_file(void)
92{
93	if (bsd)
94		(void) snprintf(locpath, sizeof (locpath), "%s.%s",
95		    locname, category_name());
96	else
97		(void) snprintf(locpath, sizeof (locpath), "%s/%s",
98		    locname, category_name());
99	return (locpath);
100}
101
102FILE *
103open_category(void)
104{
105	FILE *file;
106
107	if (verbose) {
108		(void) printf("Writing category %s: ", category_name());
109		(void) fflush(stdout);
110	}
111
112	/* make the parent directory */
113	if (!bsd)
114		(void) mkdir(dirname(category_file()), 0755);
115
116	/*
117	 * note that we have to regenerate the file name, as dirname
118	 * clobbered it.
119	 */
120	file = fopen(category_file(), "w");
121	if (file == NULL) {
122		errf(strerror(errno));
123		return (NULL);
124	}
125	return (file);
126}
127
128void
129close_category(FILE *f)
130{
131	if (fchmod(fileno(f), 0644) < 0) {
132		(void) fclose(f);
133		(void) unlink(category_file());
134		errf(strerror(errno));
135	}
136	if (fclose(f) < 0) {
137		(void) unlink(category_file());
138		errf(strerror(errno));
139	}
140	if (verbose) {
141		(void) fprintf(stdout, "done.\n");
142		(void) fflush(stdout);
143	}
144}
145
146/*
147 * This function is used when copying the category from another
148 * locale.  Note that the copy is actually performed using a hard
149 * link for efficiency.
150 */
151void
152copy_category(char *src)
153{
154	char	srcpath[PATH_MAX];
155	int	rv;
156
157	(void) snprintf(srcpath, sizeof (srcpath), "%s/%s",
158	    src, category_name());
159	rv = access(srcpath, R_OK);
160	if ((rv != 0) && (strchr(srcpath, '/') == NULL)) {
161		/* Maybe we should try the system locale */
162		(void) snprintf(srcpath, sizeof (srcpath),
163		    "/usr/lib/locale/%s/%s", src, category_name());
164		rv = access(srcpath, R_OK);
165	}
166
167	if (rv != 0) {
168		fprintf(stderr,"source locale data unavailable: %s", src);
169		return;
170	}
171
172	if (verbose > 1) {
173		(void) printf("Copying category %s from %s: ",
174		    category_name(), src);
175		(void) fflush(stdout);
176	}
177
178	/* make the parent directory */
179	if (!bsd)
180		(void) mkdir(dirname(category_file()), 0755);
181
182	if (link(srcpath, category_file()) != 0) {
183		fprintf(stderr,"unable to copy locale data: %s",
184			strerror(errno));
185		return;
186	}
187	if (verbose > 1) {
188		(void) printf("done.\n");
189	}
190}
191
192int
193putl_category(const char *s, FILE *f)
194{
195	if (s && fputs(s, f) == EOF) {
196		(void) fclose(f);
197		(void) unlink(category_file());
198		errf(strerror(errno));
199		return (EOF);
200	}
201	if (fputc('\n', f) == EOF) {
202		(void) fclose(f);
203		(void) unlink(category_file());
204		errf(strerror(errno));
205		return (EOF);
206	}
207	return (0);
208}
209
210int
211wr_category(void *buf, size_t sz, FILE *f)
212{
213	if (!sz) {
214		return (0);
215	}
216	if (fwrite(buf, sz, 1, f) < 1) {
217		(void) fclose(f);
218		(void) unlink(category_file());
219		errf(strerror(errno));
220		return (EOF);
221	}
222	return (0);
223}
224
225int yyparse(void);
226
227static void
228usage(void)
229{
230	(void) fprintf(stderr, "Usage: localedef [options] localename\n");
231	(void) fprintf(stderr, "[options] are:\n");
232	(void) fprintf(stderr, "  -D          : BSD-style output\n");
233	(void) fprintf(stderr, "  -c          : ignore warnings\n");
234	(void) fprintf(stderr, "  -v          : verbose output\n");
235	(void) fprintf(stderr, "  -U          : ignore undefined symbols\n");
236	(void) fprintf(stderr, "  -f charmap  : use given charmap file\n");
237	(void) fprintf(stderr, "  -u encoding : assume encoding\n");
238	(void) fprintf(stderr, "  -w widths   : use screen widths file\n");
239	(void) fprintf(stderr, "  -i locsrc   : source file for locale\n");
240	exit(4);
241}
242
243int
244main(int argc, char **argv)
245{
246	int c;
247	char *lfname = NULL;
248	char *cfname = NULL;
249	char *wfname = NULL;
250	DIR *dir;
251
252	init_charmap();
253	init_collate();
254	init_ctype();
255	init_messages();
256	init_monetary();
257	init_numeric();
258	init_time();
259
260	yydebug = 0;
261
262	(void) setlocale(LC_ALL, "");
263
264	while ((c = getopt(argc, argv, "w:i:cf:u:vUD")) != -1) {
265		switch (c) {
266		case 'D':
267			bsd = 1;
268			break;
269		case 'v':
270			verbose++;
271			break;
272		case 'i':
273			lfname = optarg;
274			break;
275		case 'u':
276			set_wide_encoding(optarg);
277			break;
278		case 'f':
279			cfname = optarg;
280			break;
281		case 'U':
282			undefok++;
283			break;
284		case 'c':
285			warnok++;
286			break;
287		case 'w':
288			wfname = optarg;
289			break;
290		case '?':
291			usage();
292			break;
293		}
294	}
295
296	if ((argc - 1) != (optind)) {
297		usage();
298	}
299	locname = argv[argc - 1];
300	if (verbose) {
301		(void) printf("Processing locale %s.\n", locname);
302	}
303
304	if (cfname) {
305		if (verbose)
306			(void) printf("Loading charmap %s.\n", cfname);
307		reset_scanner(cfname);
308		(void) yyparse();
309	}
310
311	if (wfname) {
312		if (verbose)
313			(void) printf("Loading widths %s.\n", wfname);
314		reset_scanner(wfname);
315		(void) yyparse();
316	}
317
318	if (verbose) {
319		(void) printf("Loading POSIX portable characters.\n");
320	}
321	add_charmap_posix();
322
323	if (lfname) {
324		reset_scanner(lfname);
325	} else {
326		reset_scanner(NULL);
327	}
328
329	/* make the directory for the locale if not already present */
330	if (!bsd) {
331		while ((dir = opendir(locname)) == NULL) {
332			if ((errno != ENOENT) ||
333			    (mkdir(locname, 0755) <  0)) {
334				errf(strerror(errno));
335			}
336		}
337		(void) closedir(dir);
338		(void) mkdir(dirname(category_file()), 0755);
339	}
340
341	(void) yyparse();
342	if (verbose) {
343		(void) printf("All done.\n");
344	}
345	return (warnings ? 1 : 0);
346}
347