mirror of
https://github.com/JKornev/hidden
synced 2024-06-29 18:32:00 +00:00
97 lines
3.4 KiB
C++
97 lines
3.4 KiB
C++
#include "Query.h"
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
|
|
CommandQuery::CommandQuery() : m_command(L"/query")
|
|
{
|
|
}
|
|
|
|
CommandQuery::~CommandQuery()
|
|
{
|
|
}
|
|
|
|
bool CommandQuery::CompareCommand(std::wstring& command)
|
|
{
|
|
return (command == m_command);
|
|
}
|
|
|
|
void CommandQuery::LoadArgs(Arguments& args, CommandModeType mode)
|
|
{
|
|
wstring object, target;
|
|
|
|
if (!args.GetNext(object))
|
|
throw WException(ERROR_INVALID_PARAMETER, L"Error, mismatched argument #1 for command 'query'");
|
|
|
|
if (object == L"process")
|
|
{
|
|
m_queryType = EQueryType::QueryProcess;
|
|
|
|
if (!args.GetNext(target))
|
|
throw WException(ERROR_INVALID_PARAMETER, L"Error, mismatched argument #2 for command 'query'");
|
|
|
|
m_targetProcId = _wtol(target.c_str());
|
|
if (!m_targetProcId)
|
|
throw WException(ERROR_INVALID_PARAMETER, L"Error, invalid target pid for command 'query'");
|
|
}
|
|
else if (object == L"state")
|
|
{
|
|
m_queryType = EQueryType::QueryState;
|
|
}
|
|
else
|
|
{
|
|
|
|
throw WException(ERROR_INVALID_PARAMETER, L"Error, invalid object type for command 'query'");
|
|
}
|
|
}
|
|
|
|
void CommandQuery::PerformCommand(Connection& connection)
|
|
{
|
|
HidStatus status;
|
|
|
|
if (m_queryType == EQueryType::QueryState)
|
|
{
|
|
HidActiveState state;
|
|
|
|
status = Hid_GetState(connection.GetContext(), &state);
|
|
if (!HID_STATUS_SUCCESSFUL(status))
|
|
throw WException(HID_STATUS_CODE(status), L"Error, query state rejected");
|
|
|
|
g_stderr << L"Driver state:" << (state == HidActiveState::StateEnabled ? L"enabled" : L"disabled") << endl;
|
|
g_stdout << L"state:" << (state == HidActiveState::StateEnabled ? 1 : 0) << endl;
|
|
}
|
|
else if (m_queryType == EQueryType::QueryProcess)
|
|
{
|
|
HidActiveState excludeState, protectedState, hiddenState;
|
|
HidPsInheritTypes excludedInherit, protectedInherit, hiddenInherit;
|
|
|
|
status = Hid_GetExcludedState(connection.GetContext(), m_targetProcId, &excludeState, &excludedInherit);
|
|
if (!HID_STATUS_SUCCESSFUL(status))
|
|
throw WException(HID_STATUS_CODE(status), L"Error, query ignored state rejected");
|
|
|
|
status = Hid_GetProtectedState(connection.GetContext(), m_targetProcId, &protectedState, &protectedInherit);
|
|
if (!HID_STATUS_SUCCESSFUL(status))
|
|
throw WException(HID_STATUS_CODE(status), L"Error, query protected state rejected");
|
|
|
|
status = Hid_GetHiddenState(connection.GetContext(), m_targetProcId, &hiddenState, &hiddenInherit);
|
|
if (!HID_STATUS_SUCCESSFUL(status))
|
|
throw WException(HID_STATUS_CODE(status), L"Error, query hidden state rejected");
|
|
|
|
g_stderr << L"Ignored state:" << (excludeState == HidActiveState::StateEnabled ? L"true" : L"false")
|
|
<< L", inherit:" << ConvertInheritTypeToUnicode(excludedInherit) << endl;
|
|
g_stderr << L"Protected state:" << (protectedState == HidActiveState::StateEnabled ? L"true" : L"false")
|
|
<< L", inherit:" << ConvertInheritTypeToUnicode(protectedInherit) << endl;
|
|
g_stderr << L"Hidden state:" << (hiddenState == HidActiveState::StateEnabled ? L"true" : L"false")
|
|
<< L", inherit:" << ConvertInheritTypeToUnicode(hiddenInherit) << endl;
|
|
|
|
g_stdout << L"ignored:" << static_cast<unsigned short>(excludeState) << L"," << static_cast<unsigned short>(excludedInherit)
|
|
<< L";protected:" << static_cast<unsigned short>(protectedState) << L"," << static_cast<unsigned short>(protectedInherit)
|
|
<< L";hidden:" << static_cast<unsigned short>(hiddenState) << L"," << static_cast<unsigned short>(hiddenInherit) << endl;
|
|
}
|
|
}
|
|
|
|
CommandPtr CommandQuery::CreateInstance()
|
|
{
|
|
return CommandPtr(new CommandQuery());
|
|
}
|