source3/intl/lang_tdb.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    tdb based replacement for gettext
   4    Copyright (C) Andrew Tridgell 2001
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19
  20 #include "includes.h"
  21 #include "system/filesys.h"
  22 #include "intl/lang_tdb.h"
  23 #include "util_tdb.h"
  24
  25 static TDB_CONTEXT *tdb;
  26
  27 /* the currently selected language */
  28 static char *current_lang;
  29
  30
  31 /* load a msg file into the tdb */
  32 static bool load_msg(const char *msg_file)
  33 {
  34         char **lines;
  35         int num_lines, i;
  36         char *msgid, *msgstr;
  37         TDB_DATA data;
  38
  39         lines = file_lines_load(msg_file, &num_lines, 0, NULL);
  40
  41         if (!lines) {
  42                 return False;
  43         }
  44
  45         if (tdb_lockall(tdb) != 0) {
  46                 TALLOC_FREE(lines);
  47                 return False;
  48         }
  49
  50         /* wipe the db */
  51         tdb_wipe_all(tdb);
  52
  53         msgid = NULL;
  54
  55         for (i=0;i<num_lines;i++) {
  56                 if (strncmp(lines[i], "msgid \"", 7) == 0) {
  57                         msgid = lines[i] + 7;
  58                 }
  59                 if (msgid && strncmp(lines[i], "msgstr \"", 8) == 0) {
  60                         msgstr = lines[i] + 8;
  61                         trim_char(msgid, '\0', '\"');
  62                         trim_char(msgstr, '\0', '\"');
  63                         if (*msgstr == 0) {
  64                                 msgstr = msgid;
  65                         }
  66                         all_string_sub(msgid, "\\n", "\n", 0);
  67                         all_string_sub(msgstr, "\\n", "\n", 0);
  68                         data = string_term_tdb_data(msgstr);
  69                         tdb_store_bystring(tdb, msgid, data, 0);
  70                         msgid = NULL;
  71                 }
  72         }
  73
  74         TALLOC_FREE(lines);
  75         tdb_unlockall(tdb);
  76
  77         return True;
  78 }
  79
  80
  81 /* work out what language to use from locale variables */
  82 static const char *get_lang(void)
  83 {
  84         const char *vars[] = {"LANGUAGE", "LC_ALL", "LC_LANG", "LANG", NULL};
  85         int i;
  86         char *p;
  87
  88         for (i=0; vars[i]; i++) {
  89                 if ((p = getenv(vars[i]))) {
  90                         return p;
  91                 }
  92         }
  93
  94         return NULL;
  95 }
  96
  97 /* initialise the message translation subsystem. If the "lang" argument
  98    is NULL then get the language from the normal environment variables */
  99 bool lang_tdb_init(const char *lang)
 100 {
 101         char *path = NULL;
 102         char *msg_path = NULL;
 103         struct stat st;
 104         static int initialised;
 105         time_t loadtime;
 106         bool result = false;
 107         char *dpath = NULL;
 108         char *lpath = NULL;
 109
 110         /* we only want to init once per process, unless given
 111            an override */
 112         if (initialised && !lang)
 113                 return True;
 114
 115         if (initialised) {
 116                 /* we are re-initialising, free up any old init */
 117                 if (tdb) {
 118                         tdb_close(tdb);
 119                         tdb = NULL;
 120                 }
 121                 SAFE_FREE(current_lang);
 122         }
 123
 124         initialised = 1;
 125
 126         if (!lang) {
 127                 /* no lang given, use environment */
 128                 lang = get_lang();
 129         }
 130
 131         /* if no lang then we don't translate */
 132         if (!lang)
 133                 return True;
 134
 135         dpath = data_path(talloc_tos(), (const char *)lang);
 136         if (dpath == NULL) {
 137                 goto done;
 138         }
 139
 140         if (asprintf(&msg_path, "%s.msg", dpath) == -1) {
 141                 DEBUG(0, ("asprintf failed\n"));
 142                 goto done;
 143         }
 144         if (stat(msg_path, &st) != 0) {
 145                 /* the msg file isn't available */
 146                 DEBUG(10, ("lang_tdb_init: %s: %s\n", msg_path,
 147                            strerror(errno)));
 148                 goto done;
 149         }
 150
 151         lpath = lock_path("lang_");
 152         if (lpath == NULL) {
 153                 goto done;
 154         }
 155
 156         if (asprintf(&path, "%s%s.tdb", lpath, lang) == -1) {
 157                 DEBUG(0, ("asprintf failed\n"));
 158                 goto done;
 159         }
 160
 161         DEBUG(10, ("lang_tdb_init: loading %s\n", path));
 162
 163         tdb = tdb_open_log(path, 0, TDB_DEFAULT, O_RDWR|O_CREAT, 0644);
 164         if (!tdb) {
 165                 tdb = tdb_open_log(path, 0, TDB_DEFAULT, O_RDONLY, 0);
 166                 if (!tdb) {
 167                         DEBUG(10, ("lang_tdb_init: %s: %s\n", path,
 168                                    strerror(errno)));
 169                         goto done;
 170                 }
 171                 current_lang = SMB_STRDUP(lang);
 172                 result = True;
 173                 goto done;
 174         }
 175
 176         loadtime = tdb_fetch_int32(tdb, "/LOADTIME/");
 177
 178         if (loadtime == -1 || loadtime < st.st_mtime) {
 179                 load_msg(msg_path);
 180                 tdb_store_int32(tdb, "/LOADTIME/", (int)time(NULL));
 181         }
 182
 183         current_lang = SMB_STRDUP(lang);
 184         result = True;
 185
 186  done:
 187         SAFE_FREE(msg_path);
 188         SAFE_FREE(path);
 189         TALLOC_FREE(lpath);
 190         TALLOC_FREE(dpath);
 191
 192         return result;
 193 }
 194
 195 /* translate a msgid to a message string in the current language
 196    returns a string that must be freed by calling lang_msg_free()
 197 */
 198 const char *lang_msg(const char *msgid)
 199 {
 200         TDB_DATA data;
 201         const char *p;
 202         char *q, *msgid_quoted;
 203         int count;
 204
 205         lang_tdb_init(NULL);
 206
 207         if (!tdb) return msgid;
 208
 209         /* Due to the way quotes in msgids are escaped in the msg file we
 210            must replace " with \" before doing a lookup in the tdb. */
 211
 212         count = 0;
 213
 214         for(p = msgid; *p; p++) {
 215                 if (*p == '\"')
 216                         count++;
 217         }
 218
 219         if (!(msgid_quoted = (char *)SMB_MALLOC(strlen(msgid) + count + 1)))
 220                 return msgid;
 221
 222         /* string_sub() is unsuitable here as it replaces some punctuation
 223            chars with underscores. */
 224
 225         for(p = msgid, q = msgid_quoted; *p; p++) {
 226                 if (*p == '\"') {
 227                         *q = '\\';
 228                         q++;
 229                 }
 230                 *q = *p;
 231                 q++;
 232         }
 233
 234         *q = 0;
 235
 236         data = tdb_fetch_bystring(tdb, msgid_quoted);
 237
 238         free(msgid_quoted);
 239
 240         /* if the message isn't found then we still need to return a pointer
 241            that can be freed. Pity. */
 242         if (!data.dptr)
 243                 return SMB_STRDUP(msgid);
 244
 245         return (const char *)data.dptr;
 246 }
 247
 248
 249 /* free up a string from lang_msg() */
 250 void lang_msg_free(const char *msgstr)
 251 {
 252         if (!tdb) return;
 253         free(discard_const_p(void, msgstr));
 254 }
 255
 256 /*
 257    return the current language - needed for language file mappings
 258 */
 259 char *lang_tdb_current(void)
 260 {
 261         return current_lang;
 262 }