]> CyberLeo.Net >> Repos - FreeBSD/FreeBSD.git/blob - usr.bin/localedef/localedef.c
localedef(1): Fix for memory leaks reported by coverity.
[FreeBSD/FreeBSD.git] / usr.bin / localedef / localedef.c
1 /*
2  * Copyright 2017 Nexenta Systems, Inc.
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$");
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
57 static int bsd = 0;
58 int verbose = 0;
59 int undefok = 0;
60 int warnok = 0;
61 static char *locname = NULL;
62 static char locpath[PATH_MAX];
63
64 const char *
65 category_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
90 static char *
91 category_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
102 FILE *
103 open_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
128 void
129 delete_category(FILE *f)
130 {
131         (void) fclose(f);
132         (void) unlink(category_file());
133 }
134
135 void
136 close_category(FILE *f)
137 {
138         if (fchmod(fileno(f), 0644) < 0 ||
139             fclose(f) != 0) {
140                 (void) fclose(f);
141                 (void) unlink(category_file());
142                 errf(strerror(errno));
143                 delete_category(f);
144         }
145         if (fclose(f) < 0) {
146                 (void) unlink(category_file());
147                 errf(strerror(errno));
148         }
149         if (verbose) {
150                 (void) fprintf(stdout, "done.\n");
151                 (void) fflush(stdout);
152         }
153 }
154
155 /*
156  * This function is used when copying the category from another
157  * locale.  Note that the copy is actually performed using a hard
158  * link for efficiency.
159  */
160 void
161 copy_category(char *src)
162 {
163         char    srcpath[PATH_MAX];
164         int     rv;
165
166         (void) snprintf(srcpath, sizeof (srcpath), "%s/%s",
167             src, category_name());
168         rv = access(srcpath, R_OK);
169         if ((rv != 0) && (strchr(srcpath, '/') == NULL)) {
170                 /* Maybe we should try the system locale */
171                 (void) snprintf(srcpath, sizeof (srcpath),
172                     "/usr/lib/locale/%s/%s", src, category_name());
173                 rv = access(srcpath, R_OK);
174         }
175
176         if (rv != 0) {
177                 fprintf(stderr,"source locale data unavailable: %s", src);
178                 return;
179         }
180
181         if (verbose > 1) {
182                 (void) printf("Copying category %s from %s: ",
183                     category_name(), src);
184                 (void) fflush(stdout);
185         }
186
187         /* make the parent directory */
188         if (!bsd)
189                 (void) mkdir(dirname(category_file()), 0755);
190
191         if (link(srcpath, category_file()) != 0) {
192                 fprintf(stderr,"unable to copy locale data: %s",
193                         strerror(errno));
194                 return;
195         }
196         if (verbose > 1) {
197                 (void) printf("done.\n");
198         }
199 }
200
201 int
202 putl_category(const char *s, FILE *f)
203 {
204         if (s && fputs(s, f) == EOF) {
205                 errf(strerror(errno));
206                 delete_category(f);
207                 return (EOF);
208         }
209         if (fputc('\n', f) == EOF) {
210                 errf(strerror(errno));
211                 delete_category(f);
212                 return (EOF);
213         }
214         return (0);
215 }
216
217 int
218 wr_category(void *buf, size_t sz, FILE *f)
219 {
220         if (!sz) {
221                 return (0);
222         }
223         if (fwrite(buf, sz, 1, f) < 1) {
224                 errf(strerror(errno));
225                 delete_category(f);
226                 return (EOF);
227         }
228         return (0);
229 }
230
231 int yyparse(void);
232
233 static void
234 usage(void)
235 {
236         (void) fprintf(stderr, "Usage: localedef [options] localename\n");
237         (void) fprintf(stderr, "[options] are:\n");
238         (void) fprintf(stderr, "  -D          : BSD-style output\n");
239         (void) fprintf(stderr, "  -c          : ignore warnings\n");
240         (void) fprintf(stderr, "  -v          : verbose output\n");
241         (void) fprintf(stderr, "  -U          : ignore undefined symbols\n");
242         (void) fprintf(stderr, "  -f charmap  : use given charmap file\n");
243         (void) fprintf(stderr, "  -u encoding : assume encoding\n");
244         (void) fprintf(stderr, "  -w widths   : use screen widths file\n");
245         (void) fprintf(stderr, "  -i locsrc   : source file for locale\n");
246         exit(4);
247 }
248
249 int
250 main(int argc, char **argv)
251 {
252         int c;
253         char *lfname = NULL;
254         char *cfname = NULL;
255         char *wfname = NULL;
256         DIR *dir;
257
258         init_charmap();
259         init_collate();
260         init_ctype();
261         init_messages();
262         init_monetary();
263         init_numeric();
264         init_time();
265
266         yydebug = 0;
267
268         (void) setlocale(LC_ALL, "");
269
270         while ((c = getopt(argc, argv, "w:i:cf:u:vUD")) != -1) {
271                 switch (c) {
272                 case 'D':
273                         bsd = 1;
274                         break;
275                 case 'v':
276                         verbose++;
277                         break;
278                 case 'i':
279                         lfname = optarg;
280                         break;
281                 case 'u':
282                         set_wide_encoding(optarg);
283                         break;
284                 case 'f':
285                         cfname = optarg;
286                         break;
287                 case 'U':
288                         undefok++;
289                         break;
290                 case 'c':
291                         warnok++;
292                         break;
293                 case 'w':
294                         wfname = optarg;
295                         break;
296                 case '?':
297                         usage();
298                         break;
299                 }
300         }
301
302         if ((argc - 1) != (optind)) {
303                 usage();
304         }
305         locname = argv[argc - 1];
306         if (verbose) {
307                 (void) printf("Processing locale %s.\n", locname);
308         }
309
310         if (cfname) {
311                 if (verbose)
312                         (void) printf("Loading charmap %s.\n", cfname);
313                 reset_scanner(cfname);
314                 (void) yyparse();
315         }
316
317         if (wfname) {
318                 if (verbose)
319                         (void) printf("Loading widths %s.\n", wfname);
320                 reset_scanner(wfname);
321                 (void) yyparse();
322         }
323
324         if (verbose) {
325                 (void) printf("Loading POSIX portable characters.\n");
326         }
327         add_charmap_posix();
328
329         if (lfname) {
330                 reset_scanner(lfname);
331         } else {
332                 reset_scanner(NULL);
333         }
334
335         /* make the directory for the locale if not already present */
336         if (!bsd) {
337                 while ((dir = opendir(locname)) == NULL) {
338                         if ((errno != ENOENT) ||
339                             (mkdir(locname, 0755) <  0)) {
340                                 errf(strerror(errno));
341                         }
342                 }
343                 (void) closedir(dir);
344                 (void) mkdir(dirname(category_file()), 0755);
345         }
346
347         (void) yyparse();
348         if (verbose) {
349                 (void) printf("All done.\n");
350         }
351         return (warnings ? 1 : 0);
352 }