1 // Copyright 2019 Google LLC.
2 // Use of this source code is governed by a BSD-style license that can be found in the LICENSE file.
3
4 #include "SkLoadICU.h"
5
6 #if defined(_WIN32) && defined(SK_USING_THIRD_PARTY_ICU)
7
8 #ifndef WIN32_LEAN_AND_MEAN
9 #define WIN32_LEAN_AND_MEAN
10 #endif
11 #include <windows.h>
12 #include <io.h>
13
14 #include <cstdio>
15 #include <cstring>
16 #include <mutex>
17 #include <string>
18
19 #include "unicode/udata.h"
20
win_mmap(const char * dataFile)21 static void* win_mmap(const char* dataFile) {
22 if (!dataFile) {
23 return nullptr;
24 }
25 struct FCloseWrapper { void operator()(FILE* f) { fclose(f); } };
26 std::unique_ptr<FILE, FCloseWrapper> stream(fopen(dataFile, "rb"));
27 if (!stream) {
28 fprintf(stderr, "SkIcuLoader: datafile missing: %s.\n", dataFile);
29 return nullptr;
30 }
31 int fileno = _fileno(stream.get());
32 if (fileno < 0) {
33 fprintf(stderr, "SkIcuLoader: datafile fileno error.\n");
34 return nullptr;
35 }
36 HANDLE file = (HANDLE)_get_osfhandle(fileno);
37 if ((HANDLE)INVALID_HANDLE_VALUE == file) {
38 fprintf(stderr, "SkIcuLoader: datafile handle error.\n");
39 return nullptr;
40 }
41 struct CloseHandleWrapper { void operator()(HANDLE h) { CloseHandle(h); } };
42 std::unique_ptr<void, CloseHandleWrapper> mmapHandle(
43 CreateFileMapping(file, nullptr, PAGE_READONLY, 0, 0, nullptr));
44 if (!mmapHandle) {
45 fprintf(stderr, "SkIcuLoader: datafile mmap error.\n");
46 return nullptr;
47 }
48 void* addr = MapViewOfFile(mmapHandle.get(), FILE_MAP_READ, 0, 0, 0);
49 if (nullptr == addr) {
50 fprintf(stderr, "SkIcuLoader: datafile view error.\n");
51 return nullptr;
52 }
53 return addr;
54 }
55
init_icu(void * addr)56 static bool init_icu(void* addr) {
57 UErrorCode err = U_ZERO_ERROR;
58 udata_setCommonData(addr, &err);
59 if (err != U_ZERO_ERROR) {
60 fprintf(stderr, "udata_setCommonData() returned %d.\n", (int)err);
61 return false;
62 }
63 udata_setFileAccess(UDATA_ONLY_PACKAGES, &err);
64 if (err != U_ZERO_ERROR) {
65 fprintf(stderr, "udata_setFileAccess() returned %d.\n", (int)err);
66 return false;
67 }
68 return true;
69 }
70
library_directory()71 static std::string library_directory() {
72 HMODULE hModule = NULL;
73 GetModuleHandleExA(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS,
74 reinterpret_cast<LPCSTR>(&library_directory), &hModule);
75 char path[MAX_PATH];
76 GetModuleFileNameA(hModule, path, MAX_PATH);
77 const char* end = strrchr(path, '\\');
78 return end ? std::string(path, end - path) : std::string();
79 }
80
executable_directory()81 static std::string executable_directory() {
82 HMODULE hModule = GetModuleHandleA(NULL);
83 char path[MAX_PATH];
84 GetModuleFileNameA(hModule, path, MAX_PATH);
85 const char* end = strrchr(path, '\\');
86 return end ? std::string(path, end - path) : std::string();
87 }
88
load_from(const std::string & dir)89 static bool load_from(const std::string& dir) {
90 auto sPath = dir + "\\icudtl.dat";
91 if (void* addr = win_mmap(sPath.c_str())) {
92 if (init_icu(addr)) {
93 return true;
94 }
95 }
96 return false;
97 }
98
SkLoadICU()99 bool SkLoadICU() {
100 static bool good = false;
101 static std::once_flag flag;
102 std::call_once(flag, []() {
103 good = load_from(executable_directory()) || load_from(library_directory());
104 });
105 return good;
106 }
107
108 #endif // defined(_WIN32) && defined(SK_USING_THIRD_PARTY_ICU)
109