You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
3.2 KiB
109 lines
3.2 KiB
// Copyright 2019 Google LLC.
|
|
// Use of this source code is governed by a BSD-style license that can be found in the LICENSE file.
|
|
|
|
#include "SkLoadICU.h"
|
|
|
|
#if defined(_WIN32) && defined(SK_USING_THIRD_PARTY_ICU)
|
|
|
|
#ifndef WIN32_LEAN_AND_MEAN
|
|
#define WIN32_LEAN_AND_MEAN
|
|
#endif
|
|
#include <windows.h>
|
|
#include <io.h>
|
|
|
|
#include <cstdio>
|
|
#include <cstring>
|
|
#include <mutex>
|
|
#include <string>
|
|
|
|
#include "unicode/udata.h"
|
|
|
|
static void* win_mmap(const char* dataFile) {
|
|
if (!dataFile) {
|
|
return nullptr;
|
|
}
|
|
struct FCloseWrapper { void operator()(FILE* f) { fclose(f); } };
|
|
std::unique_ptr<FILE, FCloseWrapper> stream(fopen(dataFile, "rb"));
|
|
if (!stream) {
|
|
fprintf(stderr, "SkIcuLoader: datafile missing: %s.\n", dataFile);
|
|
return nullptr;
|
|
}
|
|
int fileno = _fileno(stream.get());
|
|
if (fileno < 0) {
|
|
fprintf(stderr, "SkIcuLoader: datafile fileno error.\n");
|
|
return nullptr;
|
|
}
|
|
HANDLE file = (HANDLE)_get_osfhandle(fileno);
|
|
if ((HANDLE)INVALID_HANDLE_VALUE == file) {
|
|
fprintf(stderr, "SkIcuLoader: datafile handle error.\n");
|
|
return nullptr;
|
|
}
|
|
struct CloseHandleWrapper { void operator()(HANDLE h) { CloseHandle(h); } };
|
|
std::unique_ptr<void, CloseHandleWrapper> mmapHandle(
|
|
CreateFileMapping(file, nullptr, PAGE_READONLY, 0, 0, nullptr));
|
|
if (!mmapHandle) {
|
|
fprintf(stderr, "SkIcuLoader: datafile mmap error.\n");
|
|
return nullptr;
|
|
}
|
|
void* addr = MapViewOfFile(mmapHandle.get(), FILE_MAP_READ, 0, 0, 0);
|
|
if (nullptr == addr) {
|
|
fprintf(stderr, "SkIcuLoader: datafile view error.\n");
|
|
return nullptr;
|
|
}
|
|
return addr;
|
|
}
|
|
|
|
static bool init_icu(void* addr) {
|
|
UErrorCode err = U_ZERO_ERROR;
|
|
udata_setCommonData(addr, &err);
|
|
if (err != U_ZERO_ERROR) {
|
|
fprintf(stderr, "udata_setCommonData() returned %d.\n", (int)err);
|
|
return false;
|
|
}
|
|
udata_setFileAccess(UDATA_ONLY_PACKAGES, &err);
|
|
if (err != U_ZERO_ERROR) {
|
|
fprintf(stderr, "udata_setFileAccess() returned %d.\n", (int)err);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
static std::string library_directory() {
|
|
HMODULE hModule = NULL;
|
|
GetModuleHandleExA(GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS,
|
|
reinterpret_cast<LPCSTR>(&library_directory), &hModule);
|
|
char path[MAX_PATH];
|
|
GetModuleFileNameA(hModule, path, MAX_PATH);
|
|
const char* end = strrchr(path, '\\');
|
|
return end ? std::string(path, end - path) : std::string();
|
|
}
|
|
|
|
static std::string executable_directory() {
|
|
HMODULE hModule = GetModuleHandleA(NULL);
|
|
char path[MAX_PATH];
|
|
GetModuleFileNameA(hModule, path, MAX_PATH);
|
|
const char* end = strrchr(path, '\\');
|
|
return end ? std::string(path, end - path) : std::string();
|
|
}
|
|
|
|
static bool load_from(const std::string& dir) {
|
|
auto sPath = dir + "\\icudtl.dat";
|
|
if (void* addr = win_mmap(sPath.c_str())) {
|
|
if (init_icu(addr)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool SkLoadICU() {
|
|
static bool good = false;
|
|
static std::once_flag flag;
|
|
std::call_once(flag, []() {
|
|
good = load_from(executable_directory()) || load_from(library_directory());
|
|
});
|
|
return good;
|
|
}
|
|
|
|
#endif // defined(_WIN32) && defined(SK_USING_THIRD_PARTY_ICU)
|