icu_util.cc 2.87 KB
Newer Older
1
// Copyright 2013 the V8 project authors. All rights reserved.
2 3
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
4

5
#include "src/icu_util.h"
6

7
#if defined(_WIN32)
8
#include <windows.h>
9 10 11 12
#endif

#if defined(V8_I18N_SUPPORT)
#include <stdio.h>
13
#include <stdlib.h>
14 15 16 17

#include "unicode/putil.h"
#include "unicode/udata.h"

18
#include "src/base/build_config.h"
19 20
#include "src/base/file-utils.h"

21 22 23 24
#define ICU_UTIL_DATA_FILE   0
#define ICU_UTIL_DATA_SHARED 1
#define ICU_UTIL_DATA_STATIC 2

25 26 27 28 29 30
#define ICU_UTIL_DATA_SYMBOL "icudt" U_ICU_VERSION_SHORT "_dat"
#define ICU_UTIL_DATA_SHARED_MODULE_NAME "icudt.dll"
#endif

namespace v8 {

31 32
namespace internal {

33
#if defined(V8_I18N_SUPPORT) && (ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_FILE)
34 35 36 37 38 39 40 41 42 43
namespace {
char* g_icu_data_ptr = NULL;

void free_icu_data_ptr() {
  delete[] g_icu_data_ptr;
}

}  // namespace
#endif

44 45 46 47 48 49 50 51 52 53
bool InitializeICUDefaultLocation(const char* exec_path,
                                  const char* icu_data_file) {
#if !defined(V8_I18N_SUPPORT)
  return true;
#else
#if ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_FILE
  if (icu_data_file) {
    return InitializeICU(icu_data_file);
  }
  char* icu_data_file_default;
54
#if defined(V8_TARGET_LITTLE_ENDIAN)
55
  base::RelativePath(&icu_data_file_default, exec_path, "icudtl.dat");
56
#elif defined(V8_TARGET_BIG_ENDIAN)
57
  base::RelativePath(&icu_data_file_default, exec_path, "icudtb.dat");
58 59 60
#else
#error Unknown byte ordering
#endif
61 62 63 64 65 66 67 68 69
  bool result = InitializeICU(icu_data_file_default);
  free(icu_data_file_default);
  return result;
#else
  return InitializeICU(NULL);
#endif
#endif
}

70 71 72 73 74
bool InitializeICU(const char* icu_data_file) {
#if !defined(V8_I18N_SUPPORT)
  return true;
#else
#if ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_SHARED
75 76 77 78 79 80 81 82 83 84
  // We expect to find the ICU data module alongside the current module.
  HMODULE module = LoadLibraryA(ICU_UTIL_DATA_SHARED_MODULE_NAME);
  if (!module) return false;

  FARPROC addr = GetProcAddress(module, ICU_UTIL_DATA_SYMBOL);
  if (!addr) return false;

  UErrorCode err = U_ZERO_ERROR;
  udata_setCommonData(reinterpret_cast<void*>(addr), &err);
  return err == U_ZERO_ERROR;
85
#elif ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_STATIC
86 87
  // Mac/Linux bundle the ICU data in.
  return true;
88 89 90
#elif ICU_UTIL_DATA_IMPL == ICU_UTIL_DATA_FILE
  if (!icu_data_file) return false;

91 92
  if (g_icu_data_ptr) return true;

93 94 95 96 97 98 99
  FILE* inf = fopen(icu_data_file, "rb");
  if (!inf) return false;

  fseek(inf, 0, SEEK_END);
  size_t size = ftell(inf);
  rewind(inf);

100 101 102 103
  g_icu_data_ptr = new char[size];
  if (fread(g_icu_data_ptr, 1, size, inf) != size) {
    delete[] g_icu_data_ptr;
    g_icu_data_ptr = NULL;
104 105 106 107
    fclose(inf);
    return false;
  }
  fclose(inf);
108 109 110

  atexit(free_icu_data_ptr);

111
  UErrorCode err = U_ZERO_ERROR;
112
  udata_setCommonData(reinterpret_cast<void*>(g_icu_data_ptr), &err);
113 114
  return err == U_ZERO_ERROR;
#endif
115 116 117
#endif
}

118 119
}  // namespace internal
}  // namespace v8