blob: 5b3b1533360953af4b83dc38743fc55878da2739 [file] [log] [blame]
// 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)