36 lines
947 B
C++
36 lines
947 B
C++
|
#include "kdmapper.hpp"
|
||
|
|
|
||
|
|
int main(const int argc, char** argv)
|
||
|
|
{
|
||
|
|
if (argc != 2 || std::filesystem::path(argv[1]).extension().string().compare(".sys"))
|
||
|
|
{
|
||
|
|
std::cout << "[-] Incorrect usage" << std::endl;
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
const std::string driver_path = argv[1];
|
||
|
|
|
||
|
|
if (!std::filesystem::exists(driver_path))
|
||
|
|
{
|
||
|
|
std::cout << "[-] File " << driver_path << " doesn't exist" << std::endl;
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
HANDLE iqvw64e_device_handle = intel_driver::Load();
|
||
|
|
|
||
|
|
if (!iqvw64e_device_handle || iqvw64e_device_handle == INVALID_HANDLE_VALUE)
|
||
|
|
{
|
||
|
|
std::cout << "[-] Failed to load driver iqvw64e.sys" << std::endl;
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (!kdmapper::MapDriver(iqvw64e_device_handle, driver_path))
|
||
|
|
{
|
||
|
|
std::cout << "[-] Failed to map " << driver_path << std::endl;
|
||
|
|
intel_driver::Unload(iqvw64e_device_handle);
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
|
||
|
|
intel_driver::Unload(iqvw64e_device_handle);
|
||
|
|
std::cout << "[+] success" << std::endl;
|
||
|
|
}
|