rawaccel/common/utility-install.hpp
2020-12-02 05:00:57 -05:00

46 lines
1.1 KiB
C++

#pragma once
#include <filesystem>
#include <Windows.h>
#include "external/WinReg.hpp"
namespace fs = std::filesystem;
namespace wr = winreg;
inline const std::wstring DRIVER_NAME = L"rawaccel";
inline const std::wstring DRIVER_FILE_NAME = DRIVER_NAME + L".sys";
fs::path get_sys_path() {
std::wstring path;
path.resize(MAX_PATH);
UINT chars_copied = GetSystemDirectoryW(path.data(), MAX_PATH);
if (chars_copied == 0) throw std::runtime_error("GetSystemDirectory failed");
path.resize(chars_copied);
return path;
}
fs::path get_target_path() {
return get_sys_path() / L"drivers" / DRIVER_FILE_NAME;
}
fs::path make_temp_path(const fs::path& p) {
auto tmp_path = p;
tmp_path.concat(".tmp");
return tmp_path;
}
template<typename Func>
void modify_upper_filters(Func fn) {
const std::wstring FILTERS_NAME = L"UpperFilters";
wr::RegKey key(
HKEY_LOCAL_MACHINE,
L"SYSTEM\\CurrentControlSet\\Control\\Class\\{4d36e96f-e325-11ce-bfc1-08002be10318}"
);
std::vector<std::wstring> filters = key.GetMultiStringValue(FILTERS_NAME);
fn(filters);
key.SetMultiStringValue(FILTERS_NAME, filters);
}