408a78785a
Signed-off-by: Kirill Moizik <kirill@daynix.com> Signed-off-by: Dmitry Fleytman <dfleytma@redhat.com>
34 lines
1.5 KiB
C++
34 lines
1.5 KiB
C++
#pragma once
|
|
#define RULE_MANAGER_EXCEPTION_STRING TEXT("RuleManager exception: ")
|
|
|
|
class UsbDkRuleManagerException : public UsbDkW32ErrorException
|
|
{
|
|
public:
|
|
UsbDkRuleManagerException() : UsbDkW32ErrorException(RULE_MANAGER_EXCEPTION_STRING){}
|
|
UsbDkRuleManagerException(LPCTSTR lpzMessage) : UsbDkW32ErrorException(tstring(RULE_MANAGER_EXCEPTION_STRING) + lpzMessage){}
|
|
UsbDkRuleManagerException(LPCTSTR lpzMessage, DWORD dwErrorCode) : UsbDkW32ErrorException(tstring(RULE_MANAGER_EXCEPTION_STRING) + lpzMessage, dwErrorCode){}
|
|
UsbDkRuleManagerException(tstring errMsg) : UsbDkW32ErrorException(tstring(RULE_MANAGER_EXCEPTION_STRING) + errMsg){}
|
|
UsbDkRuleManagerException(tstring errMsg, DWORD dwErrorCode) : UsbDkW32ErrorException(tstring(RULE_MANAGER_EXCEPTION_STRING) + errMsg, dwErrorCode){}
|
|
};
|
|
|
|
class CRulesManager
|
|
{
|
|
public:
|
|
CRulesManager();
|
|
|
|
void AddRule(const USB_DK_HIDE_RULE &Rule);
|
|
void DeleteRule(const USB_DK_HIDE_RULE &Rule);
|
|
private:
|
|
template <typename TFunctor>
|
|
bool FindRule(const USB_DK_HIDE_RULE &Rule, TFunctor Functor);
|
|
bool RuleExists(const USB_DK_HIDE_RULE &Rule);
|
|
|
|
DWORD ReadDword(LPCTSTR RuleName, LPCTSTR ValueName) const;
|
|
ULONG64 ReadDwordMask(LPCTSTR RuleName, LPCTSTR ValueName) const;
|
|
ULONG64 ReadBool(LPCTSTR RuleName, LPCTSTR ValueName) const;
|
|
void WriteDword(const tstring &RuleName, LPCTSTR ValueName, ULONG Value);
|
|
|
|
void ReadRule(LPCTSTR RuleName, USB_DK_HIDE_RULE &Rule) const;
|
|
|
|
UsbDkRegAccess m_RegAccess;
|
|
};
|