1) Use errno to indicate faulure reason.
2) Move incomplete check for / in locale name from env section to loadlocale(), add check for "." and ".." too. It allows to check any argument, not env only. 3) Redesing LOAD_CATEGORY macro to eliminate code duplication. 4) Try harder in fallback code: if old locale can't be restored, load "C" locale 5) White space formatting, long lines, etc.
This commit is contained in:
parent
0cee6519c3
commit
a17eafe2a8
@ -43,6 +43,7 @@ __FBSDID("$FreeBSD$");
|
||||
|
||||
#include <sys/types.h>
|
||||
#include <sys/stat.h>
|
||||
#include <errno.h>
|
||||
#include <limits.h>
|
||||
#include <locale.h>
|
||||
#include <rune.h>
|
||||
@ -91,6 +92,7 @@ static char saved_categories[_LC_LAST][ENCODING_LEN + 1];
|
||||
static char current_locale_string[_LC_LAST * (ENCODING_LEN + 1/*"/"*/ + 1)];
|
||||
|
||||
static char *currentlocale(void);
|
||||
static int wrap_setrunelocale(char *);
|
||||
static char *loadlocale(int);
|
||||
|
||||
char *
|
||||
@ -101,8 +103,10 @@ setlocale(category, locale)
|
||||
int i, j, len;
|
||||
char *env, *r;
|
||||
|
||||
if (category < LC_ALL || category >= _LC_LAST)
|
||||
if (category < LC_ALL || category >= _LC_LAST) {
|
||||
errno = EINVAL;
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
if (!locale)
|
||||
return (category != LC_ALL ?
|
||||
@ -126,7 +130,7 @@ setlocale(category, locale)
|
||||
if (!env || !*env)
|
||||
env = getenv("LANG");
|
||||
|
||||
if (!env || !*env || strchr(env, '/'))
|
||||
if (!env || !*env)
|
||||
env = "C";
|
||||
|
||||
(void)strlcpy(new_categories[category], env, ENCODING_LEN + 1);
|
||||
@ -134,33 +138,42 @@ setlocale(category, locale)
|
||||
for (i = 1; i < _LC_LAST; ++i) {
|
||||
if (!(env = getenv(categories[i])) || !*env)
|
||||
env = new_categories[LC_ALL];
|
||||
(void)strlcpy(new_categories[i], env, ENCODING_LEN + 1);
|
||||
(void)strlcpy(new_categories[i], env,
|
||||
ENCODING_LEN + 1);
|
||||
}
|
||||
}
|
||||
} else if (category != LC_ALL)
|
||||
(void)strlcpy(new_categories[category], locale, ENCODING_LEN + 1);
|
||||
(void)strlcpy(new_categories[category], locale,
|
||||
ENCODING_LEN + 1);
|
||||
else {
|
||||
if ((r = strchr(locale, '/')) == NULL) {
|
||||
for (i = 1; i < _LC_LAST; ++i)
|
||||
(void)strlcpy(new_categories[i], locale, ENCODING_LEN + 1);
|
||||
(void)strlcpy(new_categories[i], locale,
|
||||
ENCODING_LEN + 1);
|
||||
} else {
|
||||
for (i = 1; r[1] == '/'; ++r);
|
||||
if (!r[1])
|
||||
for (i = 1; r[1] == '/'; ++r)
|
||||
;
|
||||
if (!r[1]) {
|
||||
errno = EINVAL;
|
||||
return (NULL); /* Hmm, just slashes... */
|
||||
}
|
||||
do {
|
||||
if (i == _LC_LAST)
|
||||
break; /* Too many slashes... */
|
||||
len = r - locale > ENCODING_LEN ? ENCODING_LEN : r - locale;
|
||||
(void)strlcpy(new_categories[i], locale, len + 1);
|
||||
len = r - locale > ENCODING_LEN ?
|
||||
ENCODING_LEN : r - locale;
|
||||
(void)strlcpy(new_categories[i], locale,
|
||||
len + 1);
|
||||
i++;
|
||||
locale = r;
|
||||
while (*locale == '/')
|
||||
++locale;
|
||||
while (*++r && *r != '/');
|
||||
++locale;
|
||||
while (*++r && *r != '/')
|
||||
;
|
||||
} while (*locale);
|
||||
while (i < _LC_LAST) {
|
||||
(void)strcpy(new_categories[i],
|
||||
new_categories[i-1]);
|
||||
new_categories[i-1]);
|
||||
i++;
|
||||
}
|
||||
}
|
||||
@ -172,12 +185,14 @@ setlocale(category, locale)
|
||||
for (i = 1; i < _LC_LAST; ++i) {
|
||||
(void)strcpy(saved_categories[i], current_categories[i]);
|
||||
if (loadlocale(i) == NULL) {
|
||||
int saverr = errno;
|
||||
|
||||
for (j = 1; j < i; j++) {
|
||||
(void)strcpy(new_categories[j],
|
||||
saved_categories[j]);
|
||||
/* XXX can fail too */
|
||||
saved_categories[j]);
|
||||
(void)loadlocale(j);
|
||||
}
|
||||
errno = saverr;
|
||||
return (NULL);
|
||||
}
|
||||
}
|
||||
@ -194,14 +209,29 @@ currentlocale()
|
||||
for (i = 2; i < _LC_LAST; ++i)
|
||||
if (strcmp(current_categories[1], current_categories[i])) {
|
||||
for (i = 2; i < _LC_LAST; ++i) {
|
||||
(void) strcat(current_locale_string, "/");
|
||||
(void) strcat(current_locale_string, current_categories[i]);
|
||||
(void)strcat(current_locale_string, "/");
|
||||
(void)strcat(current_locale_string,
|
||||
current_categories[i]);
|
||||
}
|
||||
break;
|
||||
}
|
||||
return (current_locale_string);
|
||||
}
|
||||
|
||||
|
||||
static int
|
||||
wrap_setrunelocale(locale)
|
||||
char *locale;
|
||||
{
|
||||
int ret = setrunelocale(locale);
|
||||
|
||||
if (ret != 0) {
|
||||
errno = ret;
|
||||
return (-1);
|
||||
}
|
||||
return (0);
|
||||
}
|
||||
|
||||
static char *
|
||||
loadlocale(category)
|
||||
int category;
|
||||
@ -210,6 +240,13 @@ loadlocale(category)
|
||||
char *new = new_categories[category];
|
||||
char *old = current_categories[category];
|
||||
|
||||
if ((new[0] == '.' &&
|
||||
(new[1] == '\0' || (new[1] == '.' && new[2] == '\0'))) ||
|
||||
strchr(new, '/') != NULL) {
|
||||
errno = EINVAL;
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
if (_PathLocale == NULL) {
|
||||
char *p = getenv("PATH_LOCALE");
|
||||
|
||||
@ -219,8 +256,10 @@ loadlocale(category)
|
||||
#endif
|
||||
) {
|
||||
if (strlen(p) + 1/*"/"*/ + ENCODING_LEN +
|
||||
1/*"/"*/ + CATEGORY_LEN >= PATH_MAX)
|
||||
1/*"/"*/ + CATEGORY_LEN >= PATH_MAX) {
|
||||
errno = ENAMETOOLONG;
|
||||
return (NULL);
|
||||
}
|
||||
_PathLocale = strdup(p);
|
||||
if (_PathLocale == NULL)
|
||||
return (NULL);
|
||||
@ -228,35 +267,41 @@ loadlocale(category)
|
||||
_PathLocale = _PATH_LOCALE;
|
||||
}
|
||||
|
||||
if (strcmp(new, old) == 0)
|
||||
return (old);
|
||||
|
||||
if (category == LC_CTYPE) {
|
||||
ret = setrunelocale(new) ? NULL : new;
|
||||
if (!ret)
|
||||
(void)setrunelocale(old);
|
||||
else
|
||||
(void)strcpy(old, new);
|
||||
return (ret);
|
||||
#define LOAD_CATEGORY(FUNC) \
|
||||
{ \
|
||||
if (strcmp(new, old) == 0) \
|
||||
return (old); \
|
||||
ret = FUNC(new) != 0 ? NULL : new; \
|
||||
if (ret == NULL) { \
|
||||
if (FUNC(old) != 0 && FUNC("C") == 0) \
|
||||
(void)strcpy(old, "C"); \
|
||||
} else \
|
||||
(void)strcpy(old, new); \
|
||||
return (ret); \
|
||||
}
|
||||
|
||||
#define LOAD_CATEGORY(CAT, FUNC) \
|
||||
if (category == CAT) { \
|
||||
ret = (FUNC(new) < 0) ? NULL : new; \
|
||||
if (!ret) \
|
||||
(void)FUNC(old); \
|
||||
else \
|
||||
(void)strcpy(old, new); \
|
||||
return (ret); \
|
||||
switch (category) {
|
||||
case LC_CTYPE:
|
||||
LOAD_CATEGORY(wrap_setrunelocale);
|
||||
/* NOTREACHED */
|
||||
case LC_COLLATE:
|
||||
LOAD_CATEGORY(__collate_load_tables);
|
||||
/* NOTREACHED */
|
||||
case LC_TIME:
|
||||
LOAD_CATEGORY(__time_load_locale);
|
||||
/* NOTREACHED */
|
||||
case LC_NUMERIC:
|
||||
LOAD_CATEGORY(__numeric_load_locale);
|
||||
/* NOTREACHED */
|
||||
case LC_MONETARY:
|
||||
LOAD_CATEGORY(__monetary_load_locale);
|
||||
/* NOTREACHED */
|
||||
case LC_MESSAGES:
|
||||
LOAD_CATEGORY(__messages_load_locale);
|
||||
/* NOTREACHED */
|
||||
default:
|
||||
errno = EINVAL;
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
LOAD_CATEGORY(LC_COLLATE, __collate_load_tables);
|
||||
LOAD_CATEGORY(LC_TIME, __time_load_locale);
|
||||
LOAD_CATEGORY(LC_NUMERIC, __numeric_load_locale);
|
||||
LOAD_CATEGORY(LC_MONETARY, __monetary_load_locale);
|
||||
LOAD_CATEGORY(LC_MESSAGES, __messages_load_locale);
|
||||
|
||||
/* Just in case...*/
|
||||
return (NULL);
|
||||
}
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user