spicetools/util/utils.cpp

127 lines
3.5 KiB
C++
Raw Normal View History

2024-08-28 15:10:34 +00:00
#include <winsock2.h>
#include <ws2tcpip.h>
#include <random>
#include "utils.h"
const char *inet_ntop(short af, const void *src, char *dst, DWORD size) {
// prepare storage
struct sockaddr_storage ss {};
ZeroMemory(&ss, sizeof(ss));
ss.ss_family = af;
// IPv6 compatibility
switch (af) {
case AF_INET:
((struct sockaddr_in *) &ss)->sin_addr = *(struct in_addr *) src;
break;
case AF_INET6:
((struct sockaddr_in6 *) &ss)->sin6_addr = *(struct in6_addr *) src;
break;
default:
return nullptr;
}
// convert to string
int result = WSAAddressToStringA((struct sockaddr *) &ss, sizeof(ss), nullptr, dst, &size);
// return on success
return (result == 0) ? dst : nullptr;
}
BOOL CALLBACK _find_window_begins_with_cb(HWND wnd, LPARAM lParam) {
auto windows = reinterpret_cast<std::vector<HWND> *>(lParam);
windows->push_back(wnd);
return TRUE;
}
std::wstring s2ws(const std::string &str) {
if (str.empty()) {
return std::wstring();
}
int length = MultiByteToWideChar(CP_ACP, 0, str.data(), -1, nullptr, 0);
if (length == 0) {
log_fatal("utils", "failed to get length of wide string: {}", get_last_error_string());
}
std::wstring buffer;
buffer.resize(length - 1);
length = MultiByteToWideChar(CP_ACP, 0, str.data(), -1, buffer.data(), length);
if (length == 0) {
log_fatal("utils", "failed to convert string to wide string: {}", get_last_error_string());
}
return buffer;
}
std::string ws2s(const std::wstring &wstr) {
if (wstr.empty()) {
return std::string();
}
int length = WideCharToMultiByte(CP_ACP, 0, wstr.data(), -1, nullptr, 0, nullptr, nullptr);
if (length == 0) {
log_fatal("utils", "failed to get length of wide string: {}", get_last_error_string());
}
std::string buffer;
buffer.resize(length - 1);
length = WideCharToMultiByte(CP_ACP, 0, wstr.data(), -1, buffer.data(), length, nullptr, nullptr);
if (length == 0) {
log_fatal("utils", "failed to convert string to wide string: {}", get_last_error_string());
}
return buffer;
}
bool acquire_shutdown_privs() {
// check if already acquired
static bool acquired = false;
if (acquired)
return true;
// get process token
HANDLE hToken;
if (!OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
return false;
// get the LUID for the shutdown privilege
TOKEN_PRIVILEGES tkp;
LookupPrivilegeValue(NULL, SE_SHUTDOWN_NAME, &tkp.Privileges[0].Luid);
tkp.PrivilegeCount = 1;
tkp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
// get the shutdown privilege for this process
AdjustTokenPrivileges(hToken, FALSE, &tkp, 0, (PTOKEN_PRIVILEGES) NULL, 0);
// check for error
bool success = GetLastError() == ERROR_SUCCESS;
if (success)
acquired = true;
return success;
}
void generate_ea_card(char card[17]) {
// don't ask why the existing codebase uses 18 char array when there are only 16+1 chars
// create random
std::random_device rd;
std::mt19937 generator(rd());
std::uniform_int_distribution<> uniform(0, 15);
// randomize card
strcpy(card, "E00401");
char hex[] = {'0','1','2','3','4','5','6','7','8','9','A','B','C','D','E','F'};
for (int i = 6; i < 16; i++) {
card[i] = hex[uniform(generator)];
}
// terminate and flush
card[16] = 0;
}