Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

688 lines
23 KiB

//+-------------------------------------------------------------------------
//
// Microsoft Windows
//
// Copyright (C) Microsoft Corporation, 1996 - 1999
//
// File: svrappdlg.cpp
//
//--------------------------------------------------------------------------
// SvrAppDlg.cpp : implementation file
//
#include "stdafx.h"
#include <winsvc.h>
#include <dbt.h>
#include "SvrApp.h"
#include "SvrAppDlg.h"
#include <CalServe.h>
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
#define RSP_SIMPLE_SERVICE 0x00000001 // Registers the process as a
// simple service process.
#define RSP_UNREGISTER_SERVICE 0x00000000 // Unregisters the process as a
// service process.
typedef BOOL (WINAPI *LPREGISTER_SERIVCE)(DWORD, int);
static BOOL g_fStarted = FALSE;
static SERVICE_STATUS_HANDLE l_hService = NULL;
static const TCHAR l_szEventSource[] = TEXT("SYSTEM\\CurrentControlSet\\Services\\EventLog\\Application\\SCardApp");
static const TCHAR l_szServiceName[] = TEXT("SCardApp");
static const GUID l_guidSmartcards
= { // 50DD5230-BA8A-11D1-BF5D-0000F805F530
0x50DD5230,
0xBA8A,
0x11D1,
{ 0xBF, 0x5D, 0x00, 0x00, 0xF8, 0x05, 0xF5, 0x30 } };
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
//{{AFX_DATA(CAboutDlg)
enum { IDD = IDD_ABOUTBOX };
//}}AFX_DATA
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CAboutDlg)
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
//}}AFX_VIRTUAL
// Implementation
protected:
//{{AFX_MSG(CAboutDlg)
//}}AFX_MSG
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
//{{AFX_DATA_INIT(CAboutDlg)
//}}AFX_DATA_INIT
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CAboutDlg)
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
//{{AFX_MSG_MAP(CAboutDlg)
// No message handlers
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CSvrAppDlg dialog
CSvrAppDlg::CSvrAppDlg(CWnd* pParent /*=NULL*/)
: CDialog(CSvrAppDlg::IDD, pParent)
{
//{{AFX_DATA_INIT(CSvrAppDlg)
// NOTE: the ClassWizard will add member initialization here
//}}AFX_DATA_INIT
// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
m_hIfDev = NULL;
}
void CSvrAppDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CSvrAppDlg)
// NOTE: the ClassWizard will add DDX and DDV calls here
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CSvrAppDlg, CDialog)
//{{AFX_MSG_MAP(CSvrAppDlg)
ON_WM_SYSCOMMAND()
ON_WM_PAINT()
ON_WM_QUERYDRAGICON()
ON_BN_CLICKED(IDC_START, OnStart)
ON_BN_CLICKED(IDC_STOP, OnStop)
//}}AFX_MSG_MAP
ON_WM_DEVICECHANGE()
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CSvrAppDlg message handlers
BOOL CSvrAppDlg::OnInitDialog()
{
CDialog::OnInitDialog();
// Add "About..." menu item to system menu.
// IDM_ABOUTBOX must be in the system command range.
ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
ASSERT(IDM_ABOUTBOX < 0xF000);
CMenu* pSysMenu = GetSystemMenu(FALSE);
CString strAboutMenu;
strAboutMenu.LoadString(IDS_ABOUTBOX);
if (!strAboutMenu.IsEmpty())
{
pSysMenu->AppendMenu(MF_SEPARATOR);
pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
}
// Set the icon for this dialog. The framework does this automatically
// when the application's main window is not a dialog
SetIcon(m_hIcon, TRUE); // Set big icon
SetIcon(m_hIcon, FALSE); // Set small icon
// TODO: Add extra initialization here
InitializeCriticalSection(&m_csMessageLock);
return TRUE; // return TRUE unless you set the focus to a control
}
void CSvrAppDlg::OnSysCommand(UINT nID, LPARAM lParam)
{
if ((nID & 0xFFF0) == IDM_ABOUTBOX)
{
CAboutDlg dlgAbout;
dlgAbout.DoModal();
}
else
{
CDialog::OnSysCommand(nID, lParam);
}
}
// If you add a minimize button to your dialog, you will need the code below
// to draw the icon. For MFC applications using the document/view model,
// this is automatically done for you by the framework.
void CSvrAppDlg::OnPaint()
{
if (IsIconic())
{
CPaintDC dc(this); // device context for painting
SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
// Center icon in client rectangle
int cxIcon = GetSystemMetrics(SM_CXICON);
int cyIcon = GetSystemMetrics(SM_CYICON);
CRect rect;
GetClientRect(&rect);
int x = (rect.Width() - cxIcon + 1) / 2;
int y = (rect.Height() - cyIcon + 1) / 2;
// Draw the icon
dc.DrawIcon(x, y, m_hIcon);
}
else
{
CDialog::OnPaint();
}
}
// The system calls this to obtain the cursor to display while the user drags
// the minimized window.
HCURSOR CSvrAppDlg::OnQueryDragIcon()
{
return (HCURSOR) m_hIcon;
}
void CSvrAppDlg::OnStart()
{
ASSERT(!g_fStarted);
try
{
//
// Initialize Event Logging.
//
DWORD dwStatus;
TCHAR szModulePath[MAX_PATH];
CRegistry
rgSCardSvr(
HKEY_LOCAL_MACHINE,
l_szEventSource,
KEY_ALL_ACCESS,
REG_OPTION_NON_VOLATILE);
//
// Add our source name as a subkey under the Application
// key in the EventLog service portion of the registry.
//
dwStatus = GetModuleFileName(
NULL,
szModulePath,
sizeof(szModulePath));
if (0 == dwStatus)
{
dwStatus = GetLastError();
CalaisWarning(
DBGT("Smart Card Resource Manager cannot determine its module name: %1"),
dwStatus);
lstrcpy(
szModulePath,
#ifdef DBG
TEXT("D:\\NT\\ISPU\\Calais\\bin\\objd\\i386\\SvrApp.exe"));
#else
TEXT("D:\\NT\\ISPU\\Calais\\bin\\obj\\i386\\SvrApp.exe"));
#endif
}
rgSCardSvr.SetValue(
TEXT("EventMessageFile"),
szModulePath,
REG_EXPAND_SZ);
rgSCardSvr.SetValue(
TEXT("TypesSupported"),
(DWORD)(EVENTLOG_ERROR_TYPE
| EVENTLOG_WARNING_TYPE
| EVENTLOG_INFORMATION_TYPE));
CalaisMessageInit(
l_szServiceName,
RegisterEventSource(NULL, l_szServiceName));
}
catch (...)
{
// No error logging!
}
AppInitializeDeviceRegistration(
GetSafeHwnd(),
DEVICE_NOTIFY_WINDOW_HANDLE);
g_fStarted = (ERROR_SUCCESS == CalaisStart());
if (!g_fStarted)
AppTerminateDeviceRegistration();
GetDlgItem(IDC_START)->EnableWindow(!g_fStarted);
GetDlgItem(IDC_STOP)->EnableWindow(g_fStarted);
}
void CSvrAppDlg::OnStop()
{
ASSERT(g_fStarted);
CalaisStop();
AppTerminateDeviceRegistration();
CalaisMessageClose();
g_fStarted = FALSE;
GetDlgItem(IDC_START)->EnableWindow(!g_fStarted);
GetDlgItem(IDC_STOP)->EnableWindow(g_fStarted);
}
void CSvrAppDlg::OnOK()
{
if (g_fStarted)
OnStop();
DeleteCriticalSection(&m_csMessageLock);
CDialog::OnOK();
}
afx_msg CSvrAppDlg::OnDeviceChange(UINT nEventType, DWORD_PTR EventData)
{
int nRetVal = CDialog::OnDeviceChange(nEventType, EventData);
try
{
CCritSect csLock(&m_csMessageLock);
DWORD dwSts;
LPCTSTR szReader = NULL;
DEV_BROADCAST_HDR *pDevHdr = (DEV_BROADCAST_HDR *)EventData;
switch (nEventType)
{
//
// A device has been inserted and is now available.
case DBT_DEVICEARRIVAL:
{
DEV_BROADCAST_DEVICEINTERFACE *pDev
= (DEV_BROADCAST_DEVICEINTERFACE *)EventData;
try
{
if (DBT_DEVTYP_DEVICEINTERFACE == pDev->dbcc_devicetype)
{
CTextString tzReader;
ASSERT(sizeof(DEV_BROADCAST_DEVICEINTERFACE) < pDev->dbcc_size);
ASSERT(0 == memcmp(
&pDev->dbcc_classguid,
&l_guidSmartcards,
sizeof(GUID)));
ASSERT(0 != pDev->dbcc_name[0]);
if (0 == pDev->dbcc_name[1])
tzReader = (LPCWSTR)pDev->dbcc_name;
else
tzReader = (LPCTSTR)pDev->dbcc_name;
szReader = tzReader;
dwSts = CalaisAddReader(szReader);
if (ERROR_SUCCESS != dwSts)
throw dwSts;
CalaisWarning(
DBGT("New device '%1' added."),
szReader);
}
else
CalaisWarning(
DBGT("Spurious device arrival event."));
}
catch (DWORD dwError)
{
CalaisError(514, dwError, szReader);
}
catch (...)
{
CalaisError(517, szReader);
}
}
//
// Permission to remove a device is requested. Any application can deny
// this request and cancel the removal.
case DBT_DEVICEQUERYREMOVE:
{
DEV_BROADCAST_HANDLE *pDev = (DEV_BROADCAST_HANDLE *)EventData;
try
{
if (DBT_DEVTYP_HANDLE == pDev->dbch_devicetype)
{
ASSERT(FIELD_OFFSET(DEV_BROADCAST_HANDLE, dbch_eventguid)
<= pDev->dbch_size);
ASSERT(NULL != pDev->dbch_handle);
ASSERT(NULL != pDev->dbch_hdevnotify);
if (NULL != pDev->dbch_handle)
{
if (!CalaisQueryReader(pDev->dbch_handle))
{
CalaisError(
520,
TEXT("DBT_DEVICEQUERYREMOVE/dbch_handle"));
nRetVal = BROADCAST_QUERY_DENY;
}
else
{
szReader = CalaisDisableReader(
(LPVOID)pDev->dbch_handle);
CalaisWarning(
DBGT("Device '%1' removal pending."),
szReader);
}
}
else
{
CalaisError(
519,
TEXT("DBT_DEVICEQUERYREMOVE/dbch_handle"));
nRetVal = BROADCAST_QUERY_DENY;
}
}
else
{
CalaisWarning(
DBGT("Spurious device removal query event."));
nRetVal = TRUE;
}
}
catch (DWORD dwError)
{
CalaisWarning(
DBGT("Error querying device busy state on reader %2: %1"),
dwError,
szReader);
nRetVal = BROADCAST_QUERY_DENY;
}
catch (...)
{
CalaisWarning(
DBGT("Exception querying device busy state on reader %1"),
szReader);
CalaisError(
518,
TEXT("DBT_DEVICEQUERYREMOVE"));
nRetVal = BROADCAST_QUERY_DENY;
}
break;
}
//
// Request to remove a device has been canceled.
case DBT_DEVICEQUERYREMOVEFAILED:
{
CBuffer bfDevice;
DEV_BROADCAST_HANDLE *pDev = (DEV_BROADCAST_HANDLE *)EventData;
try
{
if (DBT_DEVTYP_HANDLE == pDev->dbch_devicetype)
{
ASSERT(FIELD_OFFSET(DEV_BROADCAST_HANDLE, dbch_eventguid)
<= pDev->dbch_size);
ASSERT(NULL != pDev->dbch_handle);
ASSERT(NULL != pDev->dbch_hdevnotify);
if (NULL != pDev->dbch_handle)
{
szReader = CalaisConfirmClosingReader(pDev->dbch_handle);
if (NULL != szReader)
{
bfDevice.Set(
(LPBYTE)szReader,
(lstrlen(szReader) + 1) * sizeof(TCHAR));
szReader = (LPCTSTR)bfDevice.Access();
CalaisWarning(
DBGT("Smart Card Resource Manager asked to cancel release of reader %1"),
szReader);
if (NULL != pDev->dbch_hdevnotify)
{
CalaisRemoveReader((LPVOID)pDev->dbch_hdevnotify);
if (NULL != szReader)
dwSts = CalaisAddReader(szReader);
}
}
else
CalaisWarning(
DBGT("Smart Card Resource Manager asked to cancel release on unreleased reader"));
}
else
CalaisError(
519,
TEXT("DBT_DEVICEQUERYREMOVEFAILED/dbch_handle"));
}
else
{
CalaisWarning(
DBGT("Spurious device removal query failure event."));
}
}
catch (DWORD dwError)
{
CalaisWarning(
DBGT("Error cancelling removal on reader %2: %1"),
dwError,
szReader);
}
catch (...)
{
CalaisWarning(
DBGT("Exception cancelling removal on reader %1"),
szReader);
CalaisError(
518,
TEXT("DBT_DEVICEQUERYREMOVEFAILED"));
}
break;
}
//
// Device is about to be removed. Cannot be denied.
case DBT_DEVICEREMOVEPENDING:
{
DEV_BROADCAST_HANDLE *pDev = (DEV_BROADCAST_HANDLE *)EventData;
try
{
if (DBT_DEVTYP_HANDLE == pDev->dbch_devicetype)
{
ASSERT(FIELD_OFFSET(DEV_BROADCAST_HANDLE, dbch_eventguid)
<= pDev->dbch_size);
ASSERT(NULL != pDev->dbch_handle);
ASSERT(NULL != pDev->dbch_hdevnotify);
if (NULL != pDev->dbch_handle)
{
szReader = CalaisDisableReader(pDev->dbch_handle);
CalaisWarning(
DBGT("Device '%1' being removed."),
szReader);
}
else
CalaisError(
519,
TEXT("DBT_DEVICEREMOVEPENDING/dbch_handle"));
}
else
{
CalaisWarning(
DBGT("Spurious device removal pending event."));
}
}
catch (DWORD dwError)
{
CalaisWarning(
DBGT("Error removing reader %2: %1"),
dwError,
szReader);
}
catch (...)
{
CalaisWarning(
DBGT("Exception removing reader %1"),
szReader);
CalaisError(
518,
TEXT("DBT_DEVICEREMOVEPENDING"));
}
break;
}
//
// Device has been removed.
case DBT_DEVICEREMOVECOMPLETE:
{
try
{
switch (pDevHdr->dbch_devicetype)
{
case DBT_DEVTYP_HANDLE:
{
DEV_BROADCAST_HANDLE *pDev =
(DEV_BROADCAST_HANDLE *)EventData;
try
{
ASSERT(FIELD_OFFSET(DEV_BROADCAST_HANDLE, dbch_eventguid) <= pDev->dbch_size);
ASSERT(DBT_DEVTYP_HANDLE == pDev->dbch_devicetype);
ASSERT(NULL != pDev->dbch_handle);
ASSERT(NULL != pDev->dbch_hdevnotify);
if ((NULL != pDev->dbch_handle)
&& (NULL != pDev->dbch_hdevnotify))
{
szReader = CalaisDisableReader(
pDev->dbch_handle);
CalaisRemoveReader(
(LPVOID)pDev->dbch_hdevnotify);
CalaisWarning(
DBGT("Device '%1' removed."),
szReader);
}
else
{
if (NULL == pDev->dbch_handle)
CalaisError(
519,
TEXT("DBT_DEVICEREMOVECOMPLETE/DBT_DEVTYP_HANDLE/dbch_handle"));
if (NULL == pDev->dbch_hdevnotify)
CalaisError(
519,
TEXT("DBT_DEVICEREMOVECOMPLETE/DBT_DEVTYP_HANDLE/dbch_hdevnotify"));
}
}
catch (DWORD dwError)
{
CalaisWarning(
DBGT("Error completing removal of reader %2: %1"),
dwError,
szReader);
}
catch (...)
{
CalaisWarning(
DBGT("Exception completing removal of reader %1"),
szReader);
CalaisError(
518,
TEXT("DBT_DEVICEREMOVECOMPLETE/DBT_DEVTYP_HANDLE"));
}
break;
}
case DBT_DEVTYP_DEVICEINTERFACE:
{
DEV_BROADCAST_DEVICEINTERFACE *pDev
= (DEV_BROADCAST_DEVICEINTERFACE *)EventData;
try
{
CTextString tzReader;
ASSERT(sizeof(DEV_BROADCAST_DEVICEINTERFACE) < pDev->dbcc_size);
ASSERT(DBT_DEVTYP_DEVICEINTERFACE == pDev->dbcc_devicetype);
ASSERT(0 == memcmp(
&pDev->dbcc_classguid,
&l_guidSmartcards,
sizeof(GUID)));
ASSERT(0 != pDev->dbcc_name[0]);
if (0 == pDev->dbcc_name[1])
tzReader = (LPCWSTR)pDev->dbcc_name;
else
tzReader = (LPCTSTR)pDev->dbcc_name;
szReader = tzReader;
dwSts = CalaisRemoveDevice(szReader);
if (ERROR_SUCCESS == dwSts)
CalaisWarning(
DBGT("Device '%1' Removed."),
szReader);
else
CalaisWarning(
DBGT("Error removing device '%2': %1"),
dwSts,
szReader);
}
catch (DWORD dwError)
{
CalaisWarning(
DBGT("Error completing removal of reader %2: %1"),
dwError,
szReader);
}
catch (...)
{
CalaisWarning(
DBGT("Exception completing removal of reader %1"),
szReader);
CalaisError(
518,
TEXT("DBT_DEVICEREMOVECOMPLETE/DBT_DEVTYP_DEVICEINTERFACE"));
}
break;
}
default:
CalaisWarning(
DBGT("Unrecognized PnP Device Removal Type"));
break;
}
}
catch (...)
{
CalaisError(
518,
TEXT("DBT_DEVICEREMOVECOMPLETE"));
}
break;
}
default:
CalaisWarning(
DBGT("Unrecognized PnP Event"));
break;
}
}
catch (DWORD dwError)
{
CalaisWarning(
DBGT("Smart Card Resource Manager recieved error on device action: %1"),
dwError);
}
catch (...)
{
CalaisWarning(
DBGT("Smart Card Resource Manager recieved exception on device action"));
}
return nRetVal;
}