4776ca9dd1
* fix errors when automation scripts being loaded and os.execute not working * Update unicode-monkeypatch.lua
133 lines
2.6 KiB
Lua
133 lines
2.6 KiB
Lua
local ffi = require("ffi")
|
|
|
|
if ffi.os ~= "Windows" then
|
|
return
|
|
end
|
|
|
|
-- Safety first!
|
|
do
|
|
if pcall(string.dump, io.open) then
|
|
error("io.open is already patched!")
|
|
end
|
|
|
|
local function index(t, k)
|
|
return t[k]
|
|
end
|
|
|
|
if pcall(index, ffi.C, "GetACP") == false then
|
|
ffi.cdef[[
|
|
uint32_t __stdcall GetACP();
|
|
]]
|
|
end
|
|
end
|
|
|
|
local CP_UTF8 = 65001
|
|
local MB_ERR_INVALID_CHARS = 8
|
|
|
|
if ffi.C.GetACP() == CP_UTF8 then
|
|
-- "Use Unicode UTF-8 for worldwide language support" is ticked.
|
|
-- Don't bother patching it.
|
|
return
|
|
end
|
|
|
|
ffi.cdef[[
|
|
int32_t __stdcall MultiByteToWideChar(
|
|
uint32_t CodePage,
|
|
uint32_t dwFlags,
|
|
const char *lpMultiByteStr,
|
|
int32_t cbMultiByte,
|
|
wchar_t *lpWideCharStr,
|
|
int32_t cchWideChar
|
|
);
|
|
void *_wfreopen(wchar_t *path, wchar_t *mode, void *file);
|
|
int32_t _wrename(wchar_t *oldname, wchar_t *newname);
|
|
int32_t _wremove(wchar_t *path);
|
|
int32_t _wsystem(wchar_t *command);
|
|
char *strerror(int errnum);
|
|
]]
|
|
|
|
local function widen(ch)
|
|
local size = ffi.C.MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, ch, #ch, nil, 0)
|
|
if size == 0 then
|
|
error(fname .. ": invalid character sequence")
|
|
end
|
|
|
|
local buf = ffi.new("wchar_t[?]", size + 1)
|
|
if ffi.C.MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, ch, #ch, buf, size) == 0 then
|
|
error(fname .. ": char conversion error")
|
|
end
|
|
|
|
return buf
|
|
end
|
|
|
|
local function fileresult(stat, fname)
|
|
if stat == 0 then
|
|
return true
|
|
end
|
|
|
|
local errno = ffi.errno()
|
|
local msg = ffi.C.strerror(errno)
|
|
|
|
if fname then
|
|
return nil, fname .. ": " .. ffi.string(msg), errno
|
|
end
|
|
return nil, msg, errno
|
|
end
|
|
|
|
local function execresult(stat)
|
|
if stat == -1 then
|
|
return fileresult(0, nil)
|
|
end
|
|
|
|
if stat == 0 then
|
|
return true, "exit", stat
|
|
end
|
|
return nil, "exit", stat
|
|
end
|
|
|
|
local orig_open = io.open
|
|
local orig_rename = os.rename
|
|
local orig_remove = os.remove
|
|
local orig_execute = os.execute
|
|
|
|
function io.open(fname, mode)
|
|
local wfname = widen(fname)
|
|
if not mode then
|
|
mode = "r"
|
|
end
|
|
local wmode = widen(mode)
|
|
|
|
local file = assert(orig_open("nul", "rb"))
|
|
if ffi.C._wfreopen(wfname, wmode, file) == nil then
|
|
local msg, errno = select(2, file:close())
|
|
return nil, fname .. ": " .. msg, errno
|
|
end
|
|
|
|
return file
|
|
end
|
|
|
|
function os.rename(oldname, newname)
|
|
local woldname = widen(oldname)
|
|
local wnewname = widen(newname)
|
|
|
|
local stat = ffi.C._wrename(woldname, wnewname)
|
|
return fileresult(stat, oldname)
|
|
end
|
|
|
|
function os.remove(fname)
|
|
local wfname = widen(fname)
|
|
|
|
local stat = ffi.C._wremove(wfname)
|
|
return fileresult(stat, fname)
|
|
end
|
|
|
|
function os.execute(command)
|
|
local wcommand = command
|
|
if command then
|
|
wcommand = widen(command)
|
|
local stat = ffi.C._wsystem(wcommand)
|
|
return execresult(stat)
|
|
end
|
|
|
|
return true
|
|
end
|