// KeyBoardFilterCtl.cpp : Implementation of the CKeyBoardFilterCtrl ActiveX Control class.

#include "stdafx.h"
#include "KeyBoardFilter.h"
#include "KeyBoardFilterCtl.h"
#include "KeyBoardFilterPpg.h"
#include "KeyFilter.h"


#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

CKeyFilter  MyKeyFilter;

IMPLEMENT_DYNCREATE(CKeyBoardFilterCtrl, COleControl)


/////////////////////////////////////////////////////////////////////////////
// Message map

BEGIN_MESSAGE_MAP(CKeyBoardFilterCtrl, COleControl)
	//{{AFX_MSG_MAP(CKeyBoardFilterCtrl)
	ON_WM_SHOWWINDOW()
	//}}AFX_MSG_MAP
	ON_MESSAGE(OCM_COMMAND, OnOcmCommand)
	ON_OLEVERB(AFX_IDS_VERB_PROPERTIES, OnProperties)
END_MESSAGE_MAP()


/////////////////////////////////////////////////////////////////////////////
// Dispatch map

BEGIN_DISPATCH_MAP(CKeyBoardFilterCtrl, COleControl)
	//{{AFX_DISPATCH_MAP(CKeyBoardFilterCtrl)
	DISP_PROPERTY_NOTIFY(CKeyBoardFilterCtrl, "Visible", m_visible, OnVisibleChanged, VT_BOOL)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "EnableAllKeys", EnableAllKeys, VT_BOOL, VTS_NONE)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "EnableKey", EnableKey, VT_BOOL, VTS_I2 VTS_BOOL VTS_BOOL VTS_BOOL)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "DisableKey", DisableKey, VT_BOOL, VTS_I2 VTS_BOOL VTS_BOOL VTS_BOOL)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "ReplaceKey", ReplaceKey, VT_BOOL, VTS_I2 VTS_BOOL VTS_BOOL VTS_BOOL VTS_I2)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "IsKeyDisabled", IsKeyDisabled, VT_BOOL, VTS_I2 VTS_BOOL VTS_BOOL VTS_BOOL)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "SetScreenSaverSecure", SetScreenSaverSecure, VT_BOOL, VTS_BOOL)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "LockWorkStation", LockMyWorkStation, VT_EMPTY, VTS_NONE)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "DisableLockWorkstation", DisableLockWorkstation, VT_BOOL, VTS_BOOL)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "StartScreenSaver", StartScreenSaver, VT_EMPTY, VTS_NONE)
	DISP_FUNCTION(CKeyBoardFilterCtrl, "DisableTaskManager", DisableTaskManager, VT_BOOL, VTS_BOOL)
	//}}AFX_DISPATCH_MAP
	DISP_FUNCTION_ID(CKeyBoardFilterCtrl, "AboutBox", DISPID_ABOUTBOX, AboutBox, VT_EMPTY, VTS_NONE)
END_DISPATCH_MAP()


/////////////////////////////////////////////////////////////////////////////
// Event map

BEGIN_EVENT_MAP(CKeyBoardFilterCtrl, COleControl)
	//{{AFX_EVENT_MAP(CKeyBoardFilterCtrl)
	// NOTE - ClassWizard will add and remove event map entries
	//    DO NOT EDIT what you see in these blocks of generated code !
	//}}AFX_EVENT_MAP
END_EVENT_MAP()


/////////////////////////////////////////////////////////////////////////////
// Property pages

// TODO: Add more property pages as needed.  Remember to increase the count!
BEGIN_PROPPAGEIDS(CKeyBoardFilterCtrl, 1)
	PROPPAGEID(CKeyBoardFilterPropPage::guid)
END_PROPPAGEIDS(CKeyBoardFilterCtrl)


/////////////////////////////////////////////////////////////////////////////
// Initialize class factory and guid

IMPLEMENT_OLECREATE_EX(CKeyBoardFilterCtrl, "KEYBOARDFILTER.KeyBoardFilterCtrl.1",
	0xe872d000, 0xb8aa, 0x476f, 0xa0, 0x7e, 0xd, 0xa9, 0xbf, 0xda, 0x2c, 0x7d)


/////////////////////////////////////////////////////////////////////////////
// Type library ID and version

IMPLEMENT_OLETYPELIB(CKeyBoardFilterCtrl, _tlid, _wVerMajor, _wVerMinor)


/////////////////////////////////////////////////////////////////////////////
// Interface IDs

const IID BASED_CODE IID_DKeyBoardFilter =
		{ 0xc52126d0, 0x996e, 0x4470, { 0x9b, 0xae, 0xa1, 0x27, 0xb, 0x52, 0x8c, 0x8f } };
const IID BASED_CODE IID_DKeyBoardFilterEvents =
		{ 0x21e520de, 0xfe4d, 0x42af, { 0x91, 0x5b, 0xc4, 0x46, 0x2b, 0x30, 0xd3, 0xeb } };


/////////////////////////////////////////////////////////////////////////////
// Control type information

static const DWORD BASED_CODE _dwKeyBoardFilterOleMisc =
	OLEMISC_ACTIVATEWHENVISIBLE |
	OLEMISC_SETCLIENTSITEFIRST |
	OLEMISC_INSIDEOUT |
	OLEMISC_CANTLINKINSIDE |
	OLEMISC_RECOMPOSEONRESIZE;

IMPLEMENT_OLECTLTYPE(CKeyBoardFilterCtrl, IDS_KEYBOARDFILTER, _dwKeyBoardFilterOleMisc)


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::CKeyBoardFilterCtrlFactory::UpdateRegistry -
// Adds or removes system registry entries for CKeyBoardFilterCtrl

BOOL CKeyBoardFilterCtrl::CKeyBoardFilterCtrlFactory::UpdateRegistry(BOOL bRegister)
{
	// TODO: Verify that your control follows apartment-model threading rules.
	// Refer to MFC TechNote 64 for more information.
	// If your control does not conform to the apartment-model rules, then
	// you must modify the code below, changing the 6th parameter from
	// afxRegApartmentThreading to 0.

	if (bRegister)
		return AfxOleRegisterControlClass(
			AfxGetInstanceHandle(),
			m_clsid,
			m_lpszProgID,
			IDS_KEYBOARDFILTER,
			IDB_KEYBOARDFILTER,
			afxRegApartmentThreading,
			_dwKeyBoardFilterOleMisc,
			_tlid,
			_wVerMajor,
			_wVerMinor);
	else
		return AfxOleUnregisterClass(m_clsid, m_lpszProgID);
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::CKeyBoardFilterCtrl - Constructor

CKeyBoardFilterCtrl::CKeyBoardFilterCtrl()
{


	InitializeIIDs(&IID_DKeyBoardFilter, &IID_DKeyBoardFilterEvents);

  m_iCounter = 0;
  m_bIsRegistered = MyRegistration.IsRegistered();
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::~CKeyBoardFilterCtrl - Destructor

CKeyBoardFilterCtrl::~CKeyBoardFilterCtrl()
{

	// TODO: Cleanup your control's instance data here.
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::OnDraw - Drawing function

void CKeyBoardFilterCtrl::OnDraw(
			CDC* pdc, const CRect& rcBounds, const CRect& rcInvalid)
{

  HICON hIcon = LoadIcon( AfxGetInstanceHandle(), MAKEINTRESOURCE(IDI_ABOUTDLL));


	DoSuperclassPaint(pdc, rcBounds);
  if (m_visible)
  DrawIcon( pdc->m_hDC, 4, 4, hIcon);
  if (m_visible)
    ShowWindow( SW_SHOWNA);
  else
    ShowWindow( SW_HIDE);
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::DoPropExchange - Persistence support

void CKeyBoardFilterCtrl::DoPropExchange(CPropExchange* pPX)
{
	ExchangeVersion(pPX, MAKELONG(_wVerMinor, _wVerMajor));
	COleControl::DoPropExchange(pPX);

	// TODO: Call PX_ functions for each persistent custom property.
  PX_Bool( pPX, _T("Visible"), m_visible, false);
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::GetControlFlags -
// Flags to customize MFC's implementation of ActiveX controls.
//
// For information on using these flags, please see MFC technical note
// #nnn, "Optimizing an ActiveX Control".
DWORD CKeyBoardFilterCtrl::GetControlFlags()
{
	DWORD dwFlags = COleControl::GetControlFlags();


	// The control will not be redrawn when making the transition
	// between the active and inactivate state.
	dwFlags |= noFlickerActivate;

	// The control can optimize its OnDraw method, by not restoring
	// the original GDI objects in the device context.
	//dwFlags |= canOptimizeDraw;
	return dwFlags;
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::OnResetState - Reset control to default state

void CKeyBoardFilterCtrl::OnResetState()
{
	COleControl::OnResetState();  // Resets defaults found in DoPropExchange

	// TODO: Reset any other control state here.
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::AboutBox - Display an "About" box to the user

void CKeyBoardFilterCtrl::AboutBox()
{
	CDialog dlgAbout(IDD_ABOUTBOX_KEYBOARDFILTER);
	dlgAbout.DoModal();
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::PreCreateWindow - Modify parameters for CreateWindowEx

BOOL CKeyBoardFilterCtrl::PreCreateWindow(CREATESTRUCT& cs)
{
	cs.lpszClass = _T("STATIC");
	return COleControl::PreCreateWindow(cs);
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::IsSubclassedControl - This is a subclassed control

BOOL CKeyBoardFilterCtrl::IsSubclassedControl()
{
	return TRUE;
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl::OnOcmCommand - Handle command messages

LRESULT CKeyBoardFilterCtrl::OnOcmCommand(WPARAM wParam, LPARAM lParam)
{
#ifdef _WIN32
	WORD wNotifyCode = HIWORD(wParam);
#else
	WORD wNotifyCode = HIWORD(lParam);
#endif

	// TODO: Switch on wNotifyCode here.

	return 0;
}


/////////////////////////////////////////////////////////////////////////////
// CKeyBoardFilterCtrl message handlers

void CKeyBoardFilterCtrl::Registration()
{


  //check if applicaiton is registrated
//  if (!m_bIsRegistered) 
  {

    //show registration dialog
    MyRegistration.DoModal();

    //update internal variable
    m_bIsRegistered = MyRegistration.IsRegistered();
  }
}


void CKeyBoardFilterCtrl::OnVisibleChanged() 
{

  if (m_visible)
    ShowWindow( SW_SHOWNA);
  else
    ShowWindow( SW_HIDE);

  InvalidateControl();
	SetModifiedFlag();
}

//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::EnableAllKeys() 
// 
// Enable all keystrokes. Keystrokes which are disabled, will be enabled 
// after the function call.
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::EnableAllKeys() 
{
	

  //check registration code
  //Registration();

  MyKeyFilter.EnableAllKeys();
	return TRUE;
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::EnableKey(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::EnableKey(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
{
	

  //check registration code
  //Registration();

	return MyKeyFilter.EnableKey( ScanCode, Shift, Alt, Control);
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::DisableKey(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::DisableKey(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
{
	

  //check registration code
  //Registration();

	return MyKeyFilter.DisableKey( ScanCode, Shift, Alt, Control);
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::ReplaceKey(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::ReplaceKey(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control, short ReplaceCode) 
{
	

  //check registration code
  //Registration();

	return MyKeyFilter.ReplaceKey( ScanCode, Shift, Alt, Control, ReplaceCode);
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::IsKeyDisabled(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::IsKeyDisabled(short ScanCode, BOOL Shift, BOOL Alt, BOOL Control) 
{

  BOOL ret = false;


  //check registration code
  //Registration();

	MyKeyFilter.IsKeyDisabled( &ret, ScanCode, Shift, Alt, Control);
  return ret;
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::SetScreenSaverSecure(BOOL Secure) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::SetScreenSaverSecure(BOOL Secure) 
{

  //check registration code
  //Registration();

  HKEY m_hkResult;
  BOOL ret = true;

  if (RegOpenKeyEx( HKEY_CURRENT_USER, 
                    REGSTR_PATH_SCREENSAVE,
                    0,
                    KEY_ALL_ACCESS,
                    &m_hkResult) == ERROR_SUCCESS)
  {

    CString pw = Secure ? _T("1"): _T("0");

    if (RegSetValueEx( m_hkResult, REGSTR_VALUE_USESCRPASSWORD,
                       0, REG_SZ, (unsigned char *)((LPCTSTR)pw), 2))
      ret = false;

    RegCloseKey( m_hkResult);
  }
  else ret = false;

  return ret;
}


//////////////////////////////////////////////////////////////////////
// void CKeyBoardFilterCtrl::LockMyWorkStation() 
// 
// 
//////////////////////////////////////////////////////////////////////
void CKeyBoardFilterCtrl::LockMyWorkStation() 
{

	
  //check registration code
  //Registration();

  //lock the current desktop
  LockWorkStation();
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::DisableLockWorkstation(BOOL Disable) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::DisableLockWorkstation(BOOL Disable) 
{
	
  BOOL ret = true;
  HKEY m_hkResult;

  //check registration code
  //Registration();

  if (RegOpenKeyEx( HKEY_CURRENT_USER, 
                    REGSTR_VALUE_POLSYSTEM,
                    0,
                    KEY_ALL_ACCESS,
                    &m_hkResult) == ERROR_SUCCESS)
  {

    DWORD dWord = Disable ? 1: 0;

    if (RegSetValueEx( m_hkResult, _T("DisableLockWorkstation"),
                       0, REG_DWORD, (unsigned char *)(&dWord), sizeof(DWORD)2))
      ret = false;

    RegCloseKey( m_hkResult);
  }
  else
    ret = false;

	return ret;
}


//////////////////////////////////////////////////////////////////////
// void CKeyBoardFilterCtrl::StartScreenSaver() 
// 
// 
//////////////////////////////////////////////////////////////////////
void CKeyBoardFilterCtrl::StartScreenSaver() 
{


  //check registration code
  //Registration();

  SendMessage( WM_SYSCOMMAND, SC_SCREENSAVE,0);
  //GetDesktopWindow(), 
}


//////////////////////////////////////////////////////////////////////
// BOOL CKeyBoardFilterCtrl::DisableTaskManager(BOOL Disable) 
// 
// 
//////////////////////////////////////////////////////////////////////
BOOL CKeyBoardFilterCtrl::DisableTaskManager(BOOL Disable) 
{
	
  BOOL ret = true;
  HKEY m_hkResult;

  //check registration code
  //Registration();

  if (RegOpenKeyEx( HKEY_CURRENT_USER, 
                    REGSTR_VALUE_POLSYSTEM,
                    0,
                    KEY_ALL_ACCESS,
                    &m_hkResult) == ERROR_SUCCESS)
  {

    DWORD dWord = Disable ? 1: 0;

    if (RegSetValueEx( m_hkResult, _T("DisableTaskMgr"),
                       0, REG_DWORD, (unsigned char *)(&dWord), sizeof(DWORD)))
      ret = false;

    RegCloseKey( m_hkResult);
  }
  else
    ret = false;

	return ret;
}


void CKeyBoardFilterCtrl::OnShowWindow(BOOL bShow, UINT nStatus) 
{
	COleControl::OnShowWindow(bShow, nStatus);
	
	// TODO: Add your message handler code here
  if (bShow && (m_iCounter == 1)) Registration();	
  m_iCounter++;
}