4f9e4174c1000a6ffec096bf19ed497a2d15e5c4
[musl] / src / locale / dcngettext.c
1 #include <libintl.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <errno.h>
5 #include <limits.h>
6 #include <sys/stat.h>
7 #include "locale_impl.h"
8 #include "libc.h"
9 #include "atomic.h"
10
11 struct binding {
12         struct binding *next;
13         int dirlen;
14         int active;
15         char *domainname;
16         char *dirname;
17         char buf[];
18 };
19
20 static void *volatile bindings;
21
22 static char *gettextdir(const char *domainname, size_t *dirlen)
23 {
24         struct binding *p;
25         for (p=bindings; p; p=p->next) {
26                 if (!strcmp(p->domainname, domainname) && p->active) {
27                         *dirlen = p->dirlen;
28                         return (char *)p->dirname;
29                 }
30         }
31         return 0;
32 }
33
34 char *bindtextdomain(const char *domainname, const char *dirname)
35 {
36         static int lock[2];
37         struct binding *p, *q;
38
39         if (!domainname) return 0;
40         if (!dirname) return gettextdir(domainname, &(size_t){0});
41
42         size_t domlen = strlen(domainname);
43         size_t dirlen = strlen(dirname);
44         if (domlen > NAME_MAX || dirlen >= PATH_MAX) {
45                 errno = EINVAL;
46                 return 0;
47         }
48
49         LOCK(lock);
50
51         for (p=bindings; p; p=p->next) {
52                 if (!strcmp(p->domainname, domainname) &&
53                     !strcmp(p->dirname, dirname)) {
54                         break;
55                 }
56         }
57
58         if (!p) {
59                 p = malloc(sizeof *p + domlen + dirlen + 2);
60                 if (!p) {
61                         UNLOCK(lock);
62                         return 0;
63                 }
64                 p->next = bindings;
65                 p->dirlen = dirlen;
66                 p->domainname = p->buf;
67                 p->dirname = p->buf + domlen + 1;
68                 memcpy(p->domainname, domainname, domlen+1);
69                 memcpy(p->dirname, dirname, dirlen+1);
70                 a_cas_p(&bindings, bindings, p);
71         }
72
73         a_store(&p->active, 1);
74
75         for (q=bindings; q; q=q->next) {
76                 if (!strcmp(p->domainname, domainname) && q != p)
77                         a_store(&q->active, 0);
78         }
79
80         UNLOCK(lock);
81         
82         return (char *)p->dirname;
83 }
84
85 static const char catnames[][12] = {
86         "LC_TIME",
87         "LC_COLLATE",
88         "LC_MONETARY",
89         "LC_MESSAGES",
90 };
91
92 static const char catlens[] = { 7, 10, 11, 11 };
93
94 struct msgcat {
95         struct msgcat *next;
96         const void *map;
97         size_t map_size;
98         char name[];
99 };
100
101 static char *dummy_gettextdomain()
102 {
103         return "messages";
104 }
105
106 weak_alias(dummy_gettextdomain, __gettextdomain);
107
108 const unsigned char *__map_file(const char *, size_t *);
109 int __munmap(void *, size_t);
110
111 char *dcngettext(const char *domainname, const char *msgid1, const char *msgid2, unsigned long int n, int category)
112 {
113         static struct msgcat *volatile cats;
114         struct msgcat *p;
115         struct __locale_struct *loc = CURRENT_LOCALE;
116         struct __locale_map *lm;
117         const char *dirname, *locname, *catname;
118         size_t dirlen, loclen, catlen, domlen;
119
120         if (!domainname) domainname = __gettextdomain();
121
122         domlen = strlen(domainname);
123         if (domlen > NAME_MAX) goto notrans;
124
125         dirname = gettextdir(domainname, &dirlen);
126         if (!dirname) goto notrans;
127
128         switch (category) {
129         case LC_MESSAGES:
130                 locname = loc->messages_name;
131                 if (!*locname) goto notrans;
132                 break;
133         case LC_TIME:
134         case LC_MONETARY:
135         case LC_COLLATE:
136                 lm = loc->cat[category-2];
137                 if (!lm) goto notrans;
138                 locname = lm->name;
139                 break;
140         default:
141 notrans:
142                 return (char *) ((n == 1) ? msgid1 : msgid2);
143         }
144
145         catname = catnames[category-2];
146         catlen = catlens[category-2];
147         loclen = strlen(locname);
148
149         size_t namelen = dirlen+1 + loclen+1 + catlen+1 + domlen+3;
150         char name[namelen+1], *s = name;
151
152         memcpy(s, dirname, dirlen);
153         s[dirlen] = '/';
154         s += dirlen + 1;
155         memcpy(s, locname, loclen);
156         s[loclen] = '/';
157         s += loclen + 1;
158         memcpy(s, catname, catlen);
159         s[catlen] = '/';
160         s += catlen + 1;
161         memcpy(s, domainname, domlen);
162         s[domlen] = '.';
163         s[domlen+1] = 'm';
164         s[domlen+2] = 'o';
165         s[domlen+3] = 0;
166
167         for (p=cats; p; p=p->next)
168                 if (!strcmp(p->name, name))
169                         break;
170
171         if (!p) {
172                 void *old_cats;
173                 size_t map_size;
174                 const void *map = __map_file(name, &map_size);
175                 if (!map) goto notrans;
176                 p = malloc(sizeof *p + namelen + 1);
177                 if (!p) {
178                         __munmap((void *)map, map_size);
179                         goto notrans;
180                 }
181                 p->map = map;
182                 p->map_size = map_size;
183                 memcpy(p->name, name, namelen+1);
184                 do {
185                         old_cats = cats;
186                         p->next = old_cats;
187                 } while (a_cas_p(&cats, old_cats, p) != old_cats);
188         }
189
190         const char *trans = __mo_lookup(p->map, p->map_size, msgid1);
191         if (!trans) goto notrans;
192
193         /* FIXME: support alternate plural rules */
194         if (n != 1) {
195                 size_t l = strlen(trans);
196                 if (l+1 >= p->map_size - (trans - (char *)p->map))
197                         goto notrans;
198                 trans += l+1;
199         }
200         return (char *)trans;
201 }
202
203 char *dcgettext(const char *domainname, const char *msgid, int category)
204 {
205         return dcngettext(domainname, msgid, msgid, 1, category);
206 }
207
208 char *dngettext(const char *domainname, const char *msgid1, const char *msgid2, unsigned long int n)
209 {
210         return dcngettext(domainname, msgid1, msgid2, n, LC_MESSAGES);
211 }
212
213 char *dgettext(const char *domainname, const char *msgid)
214 {
215         return dcngettext(domainname, msgid, msgid, 1, LC_MESSAGES);
216 }