2019-07-01 18:45:55 +03:00
|
|
|
module os
|
|
|
|
|
2019-07-16 21:00:41 +03:00
|
|
|
#flag -lws2_32
|
|
|
|
#include <winsock2.h>
|
|
|
|
|
2019-07-16 02:57:03 +03:00
|
|
|
const (
|
2019-07-16 02:59:28 +03:00
|
|
|
PathSeparator = '\\'
|
2019-07-16 02:57:03 +03:00
|
|
|
)
|
|
|
|
|
2019-07-01 18:45:55 +03:00
|
|
|
// Ref - https://docs.microsoft.com/en-us/windows/desktop/winprog/windows-data-types
|
|
|
|
// A handle to an object.
|
2019-07-14 23:59:25 +03:00
|
|
|
type HANDLE voidptr
|
2019-07-01 18:45:55 +03:00
|
|
|
|
2019-07-14 23:59:25 +03:00
|
|
|
// Ref - https://docs.microsoft.com/en-us/cpp/c-runtime-library/reference/get-osfhandle?view=vs-2019
|
|
|
|
// get_file_handle retrieves the operating-system file handle that is associated with the specified file descriptor.
|
2019-07-01 18:45:55 +03:00
|
|
|
pub fn get_file_handle(path string) HANDLE {
|
2019-07-01 22:31:36 +03:00
|
|
|
mode := 'rb'
|
2019-07-14 23:59:25 +03:00
|
|
|
_fd := C.fopen(path.cstr(), mode.cstr())
|
|
|
|
if _fd == 0 {
|
|
|
|
return HANDLE(INVALID_HANDLE_VALUE)
|
2019-07-01 18:45:55 +03:00
|
|
|
}
|
2019-07-14 23:59:25 +03:00
|
|
|
_handle := C._get_osfhandle(C._fileno(_fd)) // CreateFile? - hah, no -_-
|
2019-07-01 18:45:55 +03:00
|
|
|
return _handle
|
|
|
|
}
|
2019-07-14 23:59:25 +03:00
|
|
|
|
|
|
|
// Ref - https://docs.microsoft.com/en-us/windows/win32/api/libloaderapi/nf-libloaderapi-getmodulefilenamea
|
|
|
|
// get_module_filename retrieves the fully qualified path for the file that contains the specified module.
|
|
|
|
// The module must have been loaded by the current process.
|
|
|
|
pub fn get_module_filename(handle HANDLE) ?string {
|
|
|
|
mut sz := int(1024) // Optimized length
|
|
|
|
mut buf := [byte(0); sz] // Not work for GetModuleFileNameW :(
|
|
|
|
for {
|
|
|
|
status := C.GetModuleFileName(handle, &buf, sz)
|
|
|
|
switch status {
|
|
|
|
case SUCCESS:
|
|
|
|
_filename := tos(buf.data, sz)
|
|
|
|
return _filename
|
|
|
|
case ERROR_INSUFFICIENT_BUFFER:
|
|
|
|
sz += 1024 // increment buffer cluster by 1024
|
|
|
|
buf = [byte(0); sz] // clear buffer
|
|
|
|
default:
|
|
|
|
// Must handled with GetLastError and converted by FormatMessage
|
|
|
|
return error('Cannot get file name from handle.')
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2019-07-15 20:30:40 +03:00
|
|
|
|
|
|
|
// Ref - https://docs.microsoft.com/en-us/windows/win32/api/winbase/nf-winbase-formatmessagea#parameters
|
|
|
|
const (
|
|
|
|
FORMAT_MESSAGE_ALLOCATE_BUFFER = 0x00000100
|
|
|
|
FORMAT_MESSAGE_ARGUMENT_ARRAY = 0x00002000
|
|
|
|
FORMAT_MESSAGE_FROM_HMODULE = 0x00000800
|
|
|
|
FORMAT_MESSAGE_FROM_STRING = 0x00000400
|
|
|
|
FORMAT_MESSAGE_FROM_SYSTEM = 0x00001000
|
|
|
|
FORMAT_MESSAGE_IGNORE_INSERTS = 0x00000200
|
|
|
|
)
|
|
|
|
|
|
|
|
// Ref - winnt.h
|
|
|
|
const (
|
|
|
|
SUBLANG_NEUTRAL = 0x00
|
|
|
|
SUBLANG_DEFAULT = 0x01
|
|
|
|
LANG_NEUTRAL = (SUBLANG_NEUTRAL)
|
|
|
|
)
|
|
|
|
|
2019-07-16 21:38:19 +03:00
|
|
|
// Ref - https://docs.microsoft.com/en-us/windows/win32/debug/system-error-codes--12000-15999-
|
|
|
|
const (
|
|
|
|
MAX_ERROR_CODE = 15841 // ERROR_API_UNAVAILABLE
|
|
|
|
)
|
|
|
|
|
2019-07-15 20:30:40 +03:00
|
|
|
fn ptr_get_error_message(code u32) voidptr {
|
|
|
|
mut buf := voidptr(0)
|
2019-07-16 21:38:19 +03:00
|
|
|
// Check for code overflow
|
|
|
|
if code > u32(MAX_ERROR_CODE) {
|
|
|
|
return buf
|
|
|
|
}
|
2019-07-15 20:30:40 +03:00
|
|
|
C.FormatMessage(
|
|
|
|
FORMAT_MESSAGE_ALLOCATE_BUFFER
|
|
|
|
| FORMAT_MESSAGE_FROM_SYSTEM
|
|
|
|
| FORMAT_MESSAGE_IGNORE_INSERTS,
|
|
|
|
0, code, C.MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), &buf, 0, 0)
|
|
|
|
return buf
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn get_error_msg(code u32) string {
|
|
|
|
_ptrdata := ptr_get_error_message(code)
|
2019-07-16 21:38:19 +03:00
|
|
|
if _ptrdata == voidptr(0) {
|
|
|
|
return ''
|
|
|
|
}
|
2019-07-15 20:30:40 +03:00
|
|
|
return tos(_ptrdata, C.strlen(_ptrdata))
|
|
|
|
}
|