blob: 3a344386a127047dfa26777037a201ee512f4362 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
// license:BSD-3-Clause
// copyright-holders:Aaron Giles
//============================================================
//
// winutil.cpp - Win32 OSD core utility functions
//
//============================================================
#include "winutil.h"
// lib/util
#include "timeconv.h"
// osd
#include "strconv.h"
// standard windows headers
#include <windows.h>
#include <direct.h>
//============================================================
// win_attributes_to_entry_type
//============================================================
osd::directory::entry::entry_type win_attributes_to_entry_type(DWORD attributes)
{
if (attributes == 0xFFFFFFFF)
return osd::directory::entry::entry_type::NONE;
else if (attributes & FILE_ATTRIBUTE_DIRECTORY)
return osd::directory::entry::entry_type::DIR;
else
return osd::directory::entry::entry_type::FILE;
}
//============================================================
// win_time_point_from_filetime
//============================================================
std::chrono::system_clock::time_point win_time_point_from_filetime(LPFILETIME file_time)
{
auto converted_file_time = util::ntfs_duration_from_filetime(file_time->dwHighDateTime, file_time->dwLowDateTime);
return util::system_clock_time_point_from_ntfs_duration(converted_file_time);
}
//============================================================
// win_is_gui_application
//============================================================
BOOL win_is_gui_application()
{
static BOOL is_gui_frontend;
static BOOL is_first_time = TRUE;
HMODULE module;
BYTE *image_ptr;
IMAGE_DOS_HEADER *dos_header;
IMAGE_NT_HEADERS *nt_headers;
IMAGE_OPTIONAL_HEADER *opt_header;
// is this the first time we've been ran?
if (is_first_time)
{
is_first_time = FALSE;
// get the current module
module = GetModuleHandleUni();
if (!module)
return FALSE;
image_ptr = (BYTE*) module;
// access the DOS header
dos_header = (IMAGE_DOS_HEADER *) image_ptr;
if (dos_header->e_magic != IMAGE_DOS_SIGNATURE)
return FALSE;
// access the NT headers
nt_headers = (IMAGE_NT_HEADERS *) ((BYTE*)(dos_header) + (DWORD)(dos_header->e_lfanew));
if (nt_headers->Signature != IMAGE_NT_SIGNATURE)
return FALSE;
// access the optional header
opt_header = &nt_headers->OptionalHeader;
switch (opt_header->Subsystem)
{
case IMAGE_SUBSYSTEM_WINDOWS_GUI:
is_gui_frontend = TRUE;
break;
case IMAGE_SUBSYSTEM_WINDOWS_CUI:
is_gui_frontend = FALSE;
break;
}
}
return is_gui_frontend;
}
//============================================================
// osd_subst_env
//============================================================
std::string osd_subst_env(std::string_view src)
{
std::wstring const w_src = osd::text::to_wstring(src);
std::vector<wchar_t> buffer(w_src.size() + 2);
DWORD length(ExpandEnvironmentStringsW(w_src.c_str(), &buffer[0], buffer.size()));
while (length && (buffer.size() < length))
{
buffer.clear();
buffer.resize(length + 1);
length = ExpandEnvironmentStringsW(w_src.c_str(), &buffer[0], buffer.size());
}
std::string dst;
if (length)
osd::text::from_wstring(dst, &buffer[0]);
return dst;
}
//-------------------------------------------------
// Universal way to get module handle
//-------------------------------------------------
HMODULE WINAPI GetModuleHandleUni()
{
MEMORY_BASIC_INFORMATION mbi;
VirtualQuery((LPCVOID)GetModuleHandleUni, &mbi, sizeof(mbi));
return (HMODULE)mbi.AllocationBase;
}
|