mirror of
https://github.com/JKornev/hidden
synced 2024-06-16 03:58:04 +00:00
Added tests for Reg filter
This commit is contained in:
JKornev
parent
d325a8d91a
commit
22fdb1d00b
@ -18,14 +18,14 @@ private:
|
||||
HANDLE m_handle;
|
||||
|
||||
public:
|
||||
CHandle(HANDLE handle) : m_handle(handle), m_error(GetLastError()) { }
|
||||
~CHandle() { if (m_handle != INVALID_HANDLE_VALUE) CloseHandle(m_handle); }
|
||||
CHandle(HANDLE handle) : m_handle(handle), m_error(::GetLastError()) { }
|
||||
~CHandle() { if (m_handle != INVALID_HANDLE_VALUE) ::CloseHandle(m_handle); }
|
||||
|
||||
HANDLE get() { return m_handle; }
|
||||
DWORD error() { return m_error; }
|
||||
};
|
||||
|
||||
void GenTempPath(wstring& path)
|
||||
void gen_temp_path(wstring& path)
|
||||
{
|
||||
wchar_t temp_file[MAX_PATH];
|
||||
wchar_t temp_dir[MAX_PATH];
|
||||
@ -65,7 +65,7 @@ void do_fsmon_tests(HidContext context)
|
||||
// Test 1
|
||||
wcout << L"Test 1: create single file, hide it, unhide it" << endl;
|
||||
|
||||
GenTempPath(file_path);
|
||||
gen_temp_path(file_path);
|
||||
|
||||
CHandle hfile(
|
||||
::CreateFileW(
|
||||
@ -87,7 +87,7 @@ void do_fsmon_tests(HidContext context)
|
||||
hid_status = Hid_AddHiddenFile(context, file_path.c_str(), &objId[0]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_AddHiddenFile failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
wcout << L"Error, Hid_AddHiddenFile() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
@ -100,7 +100,7 @@ void do_fsmon_tests(HidContext context)
|
||||
hid_status = Hid_RemoveHiddenFile(context, objId[0]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_RemoveHiddenFile failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
wcout << L"Error, Hid_RemoveHiddenFile() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
@ -115,7 +115,7 @@ void do_fsmon_tests(HidContext context)
|
||||
// Test 2
|
||||
wcout << L"Test 2: create single directory, hide it, unhide it" << endl;
|
||||
|
||||
GenTempPath(dir_path);
|
||||
gen_temp_path(dir_path);
|
||||
|
||||
if (::CreateDirectoryW(dir_path.c_str(), NULL) == 0)
|
||||
{
|
||||
@ -144,25 +144,25 @@ void do_fsmon_tests(HidContext context)
|
||||
hid_status = Hid_AddHiddenDir(context, dir_path.c_str(), &objId[1]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_AddHiddenDir failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
wcout << L"Error, Hid_AddHiddenDir() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
if (::GetFileAttributesW(dir_path.c_str()) != INVALID_FILE_ATTRIBUTES)
|
||||
{
|
||||
wcout << L"Error, hidden file has been found" << hfile.error() << endl;
|
||||
wcout << L"Error, hidden file has been found " << hfile.error() << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
hid_status = Hid_RemoveHiddenDir(context, objId[1]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_RemoveHiddenDir failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
wcout << L"Error, Hid_RemoveHiddenDir() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
if (::GetFileAttributesW(dir_path.c_str()) == INVALID_FILE_ATTRIBUTES)
|
||||
{
|
||||
wcout << L"Error, unhidden dir hasn't been found" << hfile.error() << endl;
|
||||
wcout << L"Error, unhidden dir hasn't been found " << hfile.error() << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
@ -171,8 +171,8 @@ void do_fsmon_tests(HidContext context)
|
||||
// Test 3
|
||||
wcout << L"Test 3: create two files, hide them, unhide using unhide all feature" << endl;
|
||||
|
||||
GenTempPath(file_paths[0]);
|
||||
GenTempPath(file_paths[1]);
|
||||
gen_temp_path(file_paths[0]);
|
||||
gen_temp_path(file_paths[1]);
|
||||
|
||||
CHandle hfile2(
|
||||
::CreateFileW(
|
||||
@ -211,14 +211,14 @@ void do_fsmon_tests(HidContext context)
|
||||
hid_status = Hid_AddHiddenFile(context, file_paths[0].c_str(), &objId[0]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_AddHiddenFile failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
wcout << L"Error, Hid_AddHiddenFile() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
hid_status = Hid_AddHiddenFile(context, file_paths[1].c_str(), &objId[0]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_AddHiddenFile failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
wcout << L"Error, Hid_AddHiddenFile() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
@ -249,20 +249,195 @@ void do_fsmon_tests(HidContext context)
|
||||
}
|
||||
|
||||
wcout << L" successful!" << endl;
|
||||
|
||||
// Test 4
|
||||
// TODO: repeat test 3 but with directories
|
||||
|
||||
}
|
||||
catch (exception&)
|
||||
{
|
||||
wcout << L" failed!" << endl;
|
||||
return;
|
||||
}
|
||||
|
||||
Hid_RemoveAllHiddenFiles(context);
|
||||
Hid_RemoveAllHiddenDirs(context);
|
||||
}
|
||||
|
||||
void gen_random_string(wstring& path, const wchar_t* prefix)
|
||||
{
|
||||
unsigned int value = (rand() << 16) + rand();
|
||||
wchar_t buff[32];
|
||||
|
||||
wsprintf(buff, L"%d", value);
|
||||
|
||||
path.clear();
|
||||
path += prefix;
|
||||
path += buff;
|
||||
}
|
||||
|
||||
void do_regmon_tests(HidContext context)
|
||||
{
|
||||
//HidStatus hid_status;
|
||||
HidStatus hid_status;
|
||||
HKEY hkey = 0, hkey2;
|
||||
wstring temp, reg_key, reg_value;
|
||||
DWORD disposition, value, type, size;
|
||||
unsigned int error_code;
|
||||
HidObjId objId[3];
|
||||
VALENT valList;
|
||||
|
||||
wcout << L"--------------------------------" << endl;
|
||||
wcout << L"Registry monitor tests result:" << endl;
|
||||
wcout << L"--------------------------------" << endl;
|
||||
|
||||
try
|
||||
{
|
||||
// Test 1
|
||||
wcout << L"Test 1: create single reg key, hide it, unhide it" << endl;
|
||||
|
||||
gen_random_string(temp, L"Hid_");
|
||||
reg_key = L"Software\\";
|
||||
reg_key += temp;
|
||||
|
||||
error_code = RegCreateKeyExW(HKEY_CURRENT_USER, reg_key.c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &hkey, &disposition);
|
||||
if (error_code != ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, RegCreateKeyExW() failed with code: " << error_code << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
if (disposition != REG_CREATED_NEW_KEY)
|
||||
wcout << L"Warning, existing key is used: " << reg_key.c_str() << endl;
|
||||
|
||||
hid_status = Hid_AddHiddenRegKey(context, HidRegRootTypes::RegHKCU, reg_key.c_str(), &objId[0]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_AddHiddenRegKey() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
error_code = RegOpenKeyExW(HKEY_CURRENT_USER, reg_key.c_str(), 0, KEY_ALL_ACCESS, &hkey2);
|
||||
if (error_code == ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, hidden reg key has been found " << endl;
|
||||
RegCloseKey(hkey2);
|
||||
throw exception();
|
||||
}
|
||||
|
||||
hid_status = Hid_RemoveHiddenRegKey(context, objId[0]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_RemoveHiddenRegKey() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
error_code = RegOpenKeyExW(HKEY_CURRENT_USER, reg_key.c_str(), 0, KEY_ALL_ACCESS, &hkey2);
|
||||
if (error_code != ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, unhidden reg key hasn't been found, code: " << error_code << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
RegCloseKey(hkey2);
|
||||
|
||||
wcout << L" successful!" << endl;
|
||||
|
||||
// Test 2
|
||||
wcout << L"Test 2: create single reg value, hide it, unhide it" << endl;
|
||||
|
||||
gen_random_string(temp, L"value");
|
||||
reg_value = reg_key;
|
||||
reg_value += L"\\";
|
||||
reg_value += temp;
|
||||
|
||||
value = 0;
|
||||
|
||||
error_code = RegSetKeyValueW(HKEY_CURRENT_USER, reg_key.c_str(), temp.c_str(), REG_DWORD, &value, sizeof(value));
|
||||
if (error_code != ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, RegSetKeyValueW() failed with code: " << error_code << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
hid_status = Hid_AddHiddenRegValue(context, HidRegRootTypes::RegHKCU, reg_value.c_str(), &objId[1]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, Hid_AddHiddenRegValue() failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
error_code = RegSetKeyValueW(HKEY_CURRENT_USER, reg_key.c_str(), temp.c_str(), REG_DWORD, &value, sizeof(value));
|
||||
if (error_code == ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, hidden reg value has been found " << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
error_code = RegDeleteValueW(hkey, temp.c_str());
|
||||
if (error_code == ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, hidden reg value has been deleted " << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
error_code = RegQueryValueExW(hkey, temp.c_str(), NULL, &type, NULL, NULL);
|
||||
if (error_code == ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, hidden reg value query has been performed " << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
memset(&valList, 0, sizeof(valList));
|
||||
valList.ve_valuename = (LPWSTR)temp.c_str();
|
||||
|
||||
size = sizeof(value);
|
||||
error_code = RegQueryMultipleValuesW(hkey, &valList, 1, (LPWSTR)&value, &size);
|
||||
if (error_code == ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, hidden reg multiple value query has been performed " << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
hid_status = Hid_RemoveHiddenRegValue(context, objId[1]);
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
wcout << L"Error, unhidden reg value hasn't been found, code: " << error_code << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
memset(&valList, 0, sizeof(valList));
|
||||
valList.ve_valuename = (LPWSTR)temp.c_str();
|
||||
|
||||
size = sizeof(value);
|
||||
error_code = RegQueryMultipleValuesW(hkey, &valList, 1, (LPWSTR)&value, &size);
|
||||
if (error_code != ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, unhidden reg value query hasn't been performed, code: " << error_code << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
error_code = RegDeleteValueW(hkey, temp.c_str());
|
||||
if (error_code != ERROR_SUCCESS)
|
||||
{
|
||||
wcout << L"Error, unhidden reg value hasn't been removed, code: " << error_code << endl;
|
||||
throw exception();
|
||||
}
|
||||
|
||||
wcout << L" successful!" << endl;
|
||||
|
||||
}
|
||||
catch (exception&)
|
||||
{
|
||||
wcout << L" failed!" << endl;
|
||||
}
|
||||
|
||||
if (hkey)
|
||||
{
|
||||
RegCloseKey(hkey);
|
||||
RegDeleteKeyW(HKEY_CURRENT_USER, reg_key.c_str());
|
||||
}
|
||||
|
||||
Hid_RemoveAllHiddenRegKeys(context);
|
||||
Hid_RemoveAllHiddenRegValues(context);
|
||||
}
|
||||
|
||||
void do_psmon_tests(HidContext context)
|
||||
@ -284,7 +459,7 @@ int wmain(int argc, wchar_t* argv[])
|
||||
if (!HID_STATUS_SUCCESSFUL(hid_status))
|
||||
{
|
||||
cout << "Error, HiddenLib initialization failed with code: " << HID_STATUS_CODE(hid_status) << endl;
|
||||
return 1;
|
||||
// return 1;
|
||||
}
|
||||
|
||||
do_fsmon_tests(hid_context);
|
||||
|
||||
Loading…
Reference in New Issue
Block a user