src/base/i18n/icu_util.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/i18n/icu_util.h"
   6
   7 #include "build/build_config.h"
   8
   9 #if defined(OS_WIN)
  10 #include <windows.h>
  11 #endif
  12
  13 #include <string>
  14
  15 #include "base/files/file_path.h"
  16 #include "base/files/memory_mapped_file.h"
  17 #include "base/logging.h"
  18 #include "base/path_service.h"
  19 #include "base/strings/string_util.h"
  20 #include "base/strings/sys_string_conversions.h"
  21 #include "third_party/icu/source/common/unicode/putil.h"
  22 #include "third_party/icu/source/common/unicode/udata.h"
  23
  24 #if defined(OS_MACOSX)
  25 #include "base/mac/foundation_util.h"
  26 #endif
  27
  28 #define ICU_UTIL_DATA_FILE   0
  29 #define ICU_UTIL_DATA_SHARED 1
  30 #define ICU_UTIL_DATA_STATIC 2
  31
  32 #ifndef ICU_UTIL_DATA_IMPL
  33
  34 #if defined(OS_WIN)
  35 #define ICU_UTIL_DATA_IMPL ICU_UTIL_DATA_SHARED
  36 #elif defined(OS_IOS)
  37 #define ICU_UTIL_DATA_IMPL ICU_UTIL_DATA_FILE
  38 #else
  39 #define ICU_UTIL_DATA_IMPL ICU_UTIL_DATA_STATIC
  40 #endif
  41
  42 #endif  // ICU_UTIL_DATA_IMPL
  43
  44 #if ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_FILE
  45 #define ICU_UTIL_DATA_FILE_NAME "icudt" U_ICU_VERSION_SHORT "l.dat"
  46 #elif ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_SHARED
  47 #define ICU_UTIL_DATA_SYMBOL "icudt" U_ICU_VERSION_SHORT "_dat"
  48 #if defined(OS_WIN)
  49 #define ICU_UTIL_DATA_SHARED_MODULE_NAME "icudt.dll"
  50 #endif
  51 #endif
  52
  53 namespace base {
  54 namespace i18n {
  55
  56 bool InitializeICU() {
  57 #ifndef NDEBUG
  58   // Assert that we are not called more than once.  Even though calling this
  59   // function isn't harmful (ICU can handle it), being called twice probably
  60   // indicates a programming error.
  61   static bool called_once = false;
  62   DCHECK(!called_once);
  63   called_once = true;
  64 #endif
  65
  66 #if (ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_SHARED)
  67   // We expect to find the ICU data module alongside the current module.
  68   FilePath data_path;
  69   PathService::Get(base::DIR_MODULE, &data_path);
  70   data_path = data_path.AppendASCII(ICU_UTIL_DATA_SHARED_MODULE_NAME);
  71
  72   HMODULE module = LoadLibrary(data_path.value().c_str());
  73   if (!module) {
  74     DLOG(ERROR) << "Failed to load " << ICU_UTIL_DATA_SHARED_MODULE_NAME;
  75     return false;
  76   }
  77
  78   FARPROC addr = GetProcAddress(module, ICU_UTIL_DATA_SYMBOL);
  79   if (!addr) {
  80     DLOG(ERROR) << ICU_UTIL_DATA_SYMBOL << ": not found in "
  81                << ICU_UTIL_DATA_SHARED_MODULE_NAME;
  82     return false;
  83   }
  84
  85   UErrorCode err = U_ZERO_ERROR;
  86   udata_setCommonData(reinterpret_cast<void*>(addr), &err);
  87   return err == U_ZERO_ERROR;
  88 #elif (ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_STATIC)
  89   // Mac/Linux bundle the ICU data in.
  90   return true;
  91 #elif (ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_FILE)
  92 #if !defined(OS_MACOSX)
  93   // For now, expect the data file to be alongside the executable.
  94   // This is sufficient while we work on unit tests, but will eventually
  95   // likely live in a data directory.
  96   FilePath data_path;
  97   bool path_ok = PathService::Get(base::DIR_EXE, &data_path);
  98   DCHECK(path_ok);
  99   u_setDataDirectory(data_path.value().c_str());
 100   // Only look for the packaged data file;
 101   // the default behavior is to look for individual files.
 102   UErrorCode err = U_ZERO_ERROR;
 103   udata_setFileAccess(UDATA_ONLY_PACKAGES, &err);
 104   return err == U_ZERO_ERROR;
 105 #else
 106   // If the ICU data directory is set, ICU won't actually load the data until
 107   // it is needed.  This can fail if the process is sandboxed at that time.
 108   // Instead, Mac maps the file in and hands off the data so the sandbox won't
 109   // cause any problems.
 110
 111   // Chrome doesn't normally shut down ICU, so the mapped data shouldn't ever
 112   // be released.
 113   CR_DEFINE_STATIC_LOCAL(base::MemoryMappedFile, mapped_file, ());
 114   if (!mapped_file.IsValid()) {
 115     // Assume it is in the framework bundle's Resources directory.
 116     FilePath data_path =
 117       base::mac::PathForFrameworkBundleResource(CFSTR(ICU_UTIL_DATA_FILE_NAME));
 118     if (data_path.empty()) {
 119       DLOG(ERROR) << ICU_UTIL_DATA_FILE_NAME << " not found in bundle";
 120       return false;
 121     }
 122     if (!mapped_file.Initialize(data_path)) {
 123       DLOG(ERROR) << "Couldn't mmap " << data_path.value();
 124       return false;
 125     }
 126   }
 127   UErrorCode err = U_ZERO_ERROR;
 128   udata_setCommonData(const_cast<uint8*>(mapped_file.data()), &err);
 129   return err == U_ZERO_ERROR;
 130 #endif  // OS check
 131 #endif
 132 }
 133
 134 }  // namespace i18n
 135 }  // namespace base