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.
 
 
 
 
 
 

548 lines
15 KiB

// LogUICtl.cpp : Implementation of the CLogUICtrl OLE control class.
#include "stdafx.h"
#include "cnfgprts.h"
#include "LogUICtl.h"
#include "LogUIPpg.h"
#include <iiscnfg.h>
#include "initguid.h"
#include <inetcom.h>
#include <logtype.h>
#include <ilogobj.hxx>
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
IMPLEMENT_DYNCREATE(CLogUICtrl, COleControl)
/////////////////////////////////////////////////////////////////////////////
// Message map
BEGIN_MESSAGE_MAP(CLogUICtrl, COleControl)
//{{AFX_MSG_MAP(CLogUICtrl)
//}}AFX_MSG_MAP
ON_MESSAGE(OCM_COMMAND, OnOcmCommand)
ON_OLEVERB(AFX_IDS_VERB_PROPERTIES, OnProperties)
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// Dispatch map
BEGIN_DISPATCH_MAP(CLogUICtrl, COleControl)
//{{AFX_DISPATCH_MAP(CLogUICtrl)
DISP_FUNCTION(CLogUICtrl, "SetAdminTarget", SetAdminTarget, VT_EMPTY, VTS_BSTR VTS_BSTR)
DISP_FUNCTION(CLogUICtrl, "ApplyLogSelection", ApplyLogSelection, VT_EMPTY, VTS_NONE)
DISP_FUNCTION(CLogUICtrl, "SetComboBox", SetComboBox, VT_EMPTY, VTS_HANDLE)
DISP_FUNCTION(CLogUICtrl, "Terminate", Terminate, VT_EMPTY, VTS_NONE)
DISP_FUNCTION(CLogUICtrl, "SetUserData", SetUserData, VT_EMPTY, VTS_BSTR VTS_BSTR)
DISP_STOCKFUNC_DOCLICK()
DISP_STOCKPROP_CAPTION()
DISP_STOCKPROP_FONT()
DISP_STOCKPROP_ENABLED()
DISP_STOCKPROP_BORDERSTYLE()
//}}AFX_DISPATCH_MAP
END_DISPATCH_MAP()
/////////////////////////////////////////////////////////////////////////////
// Event map
BEGIN_EVENT_MAP(CLogUICtrl, COleControl)
//{{AFX_EVENT_MAP(CLogUICtrl)
EVENT_STOCK_CLICK()
EVENT_STOCK_KEYUP()
EVENT_STOCK_KEYDOWN()
EVENT_STOCK_KEYPRESS()
//}}AFX_EVENT_MAP
END_EVENT_MAP()
/////////////////////////////////////////////////////////////////////////////
// Property pages
BEGIN_PROPPAGEIDS(CLogUICtrl, 2)
PROPPAGEID(CLogUIPropPage::guid)
PROPPAGEID(CLSID_CFontPropPage)
END_PROPPAGEIDS(CLogUICtrl)
/////////////////////////////////////////////////////////////////////////////
// Initialize class factory and guid
IMPLEMENT_OLECREATE_EX(CLogUICtrl, "CNFGPRTS.LogUICtrl.1",
0xba634603, 0xb771, 0x11d0, 0x92, 0x96, 0, 0xc0, 0x4f, 0xb6, 0x67, 0x8b)
/////////////////////////////////////////////////////////////////////////////
// Type library ID and version
IMPLEMENT_OLETYPELIB(CLogUICtrl, _tlid, _wVerMajor, _wVerMinor)
/////////////////////////////////////////////////////////////////////////////
// Interface IDs
const IID BASED_CODE IID_DLogUI =
{ 0xba634601, 0xb771, 0x11d0, { 0x92, 0x96, 0, 0xc0, 0x4f, 0xb6, 0x67, 0x8b } };
const IID BASED_CODE IID_DLogUIEvents =
{ 0xba634602, 0xb771, 0x11d0, { 0x92, 0x96, 0, 0xc0, 0x4f, 0xb6, 0x67, 0x8b } };
/////////////////////////////////////////////////////////////////////////////
// Control type information
static const DWORD BASED_CODE _dwLogUIOleMisc =
OLEMISC_ACTIVATEWHENVISIBLE |
OLEMISC_SETCLIENTSITEFIRST |
OLEMISC_INSIDEOUT |
OLEMISC_CANTLINKINSIDE |
OLEMISC_ACTSLIKEBUTTON |
OLEMISC_RECOMPOSEONRESIZE;
IMPLEMENT_OLECTLTYPE(CLogUICtrl, IDS_LOGUI, _dwLogUIOleMisc)
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::CLogUICtrlFactory::UpdateRegistry -
// Adds or removes system registry entries for CLogUICtrl
BOOL CLogUICtrl::CLogUICtrlFactory::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_LOGUI,
IDB_LOGUI,
afxRegApartmentThreading,
_dwLogUIOleMisc,
_tlid,
_wVerMajor,
_wVerMinor);
else
return AfxOleUnregisterClass(m_clsid, m_lpszProgID);
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::CLogUICtrl - Constructor
CLogUICtrl::CLogUICtrl():
m_fUpdateFont( FALSE ),
m_fComboInit( FALSE ),
m_hAccel( NULL ),
m_cAccel( 0 )
{
InitializeIIDs(&IID_DLogUI, &IID_DLogUIEvents);
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::~CLogUICtrl - Destructor
CLogUICtrl::~CLogUICtrl()
{
if ( m_hAccel )
DestroyAcceleratorTable( m_hAccel );
m_hAccel = NULL;
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::OnDraw - Drawing function
void CLogUICtrl::OnDraw(CDC* pdc, const CRect& rcBounds, const CRect& rcInvalid)
{
DoSuperclassPaint(pdc, rcBounds);
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::DoPropExchange - Persistence support
void CLogUICtrl::DoPropExchange(CPropExchange* pPX)
{
ExchangeVersion(pPX, MAKELONG(_wVerMinor, _wVerMajor));
COleControl::DoPropExchange(pPX);
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::OnResetState - Reset control to default state
void CLogUICtrl::OnResetState()
{
COleControl::OnResetState(); // Resets defaults found in DoPropExchange
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::PreCreateWindow - Modify parameters for CreateWindowEx
BOOL CLogUICtrl::PreCreateWindow(CREATESTRUCT& cs)
{
if ( cs.style & WS_CLIPSIBLINGS )
cs.style ^= WS_CLIPSIBLINGS;
cs.lpszClass = _T("BUTTON");
return COleControl::PreCreateWindow(cs);
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::IsSubclassedControl - This is a subclassed control
BOOL CLogUICtrl::IsSubclassedControl()
{
return TRUE;
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl::OnOcmCommand - Handle command messages
LRESULT CLogUICtrl::OnOcmCommand(WPARAM wParam, LPARAM lParam)
{
#ifdef _WIN32
WORD wNotifyCode = HIWORD(wParam);
#else
WORD wNotifyCode = HIWORD(lParam);
#endif
return 0;
}
/////////////////////////////////////////////////////////////////////////////
// CLogUICtrl message handlers
//---------------------------------------------------------------------------
// OLE Interfaced Routine
void CLogUICtrl::OnClick(USHORT iButton)
{
CWaitCursor wait;
CString sz;
sz.LoadString( IDS_LOG_ERR_TITLE );
free((void*)AfxGetApp()->m_pszAppName);
AfxGetApp()->m_pszAppName = _tcsdup(sz);
if (GetSelectedStringIID(sz))
{
IID iid;
HRESULT h = CLSIDFromString((LPTSTR)(LPCTSTR)sz, &iid);
ActivateLogProperties(iid);
}
COleControl::OnClick(iButton);
}
void CLogUICtrl::OnFontChanged()
{
m_fUpdateFont = TRUE;
COleControl::OnFontChanged();
}
void CLogUICtrl::SetAdminTarget(LPCTSTR szMachineName, LPCTSTR szMetaTarget)
{
m_szMachine = szMachineName;
m_szMetaObject = szMetaTarget;
}
void CLogUICtrl::SetUserData(LPCTSTR szName, LPCTSTR szPassword)
{
m_szUserName = szName;
m_szPassword = szPassword;
}
//---------------------------------------------------------------------------
void CLogUICtrl::ActivateLogProperties(REFIID clsidUI)
{
AFX_MANAGE_STATE(::AfxGetStaticModuleState());
IClassFactory * pcsfFactory = NULL;
HRESULT hresError;
ILogUIPlugin2 * pUI = NULL;
hresError = CoGetClassObject(clsidUI, CLSCTX_INPROC, NULL, IID_IClassFactory, (void **)&pcsfFactory);
if (SUCCEEDED(hresError))
{
hresError = pcsfFactory->CreateInstance(NULL, IID_LOGGINGUI2, (void **)&pUI);
if (SUCCEEDED(hresError))
{
CString csTempPassword;
m_szPassword.CopyTo(csTempPassword);
pcsfFactory->Release();
hresError = pUI->OnPropertiesEx(
(LPTSTR)(LPCTSTR)m_szMachine,
(LPTSTR)(LPCTSTR)m_szMetaObject,
(LPTSTR)(LPCTSTR)m_szUserName,
(LPTSTR)(LPCTSTR)csTempPassword);
pUI->Release();
}
}
}
void CLogUICtrl::ApplyLogSelection()
{
CString szName;
m_comboBox.GetWindowText( szName );
// if nothing is selected, fail
if (!szName.IsEmpty())
{
CString guid;
CString csTempPassword;
m_szPassword.CopyTo(csTempPassword);
CComAuthInfo auth(m_szMachine, m_szUserName, csTempPassword);
CMetaKey mk(&auth, NULL, METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE);
if (mk.Succeeded())
{
CMetabasePath path(TRUE, _T("logging"), szName);
mk.QueryValue(MD_LOG_PLUGIN_MOD_ID, guid, NULL, path);
}
if (!guid.IsEmpty())
{
mk.SetValue(MD_LOG_PLUGIN_ORDER, guid, NULL, m_szMetaObject);
}
}
}
//---------------------------------------------------------------------------
BOOL CLogUICtrl::GetSelectedStringIID( CString &szIID )
{
if (!m_fComboInit)
return FALSE;
BOOL bRes = FALSE;
CString szName;
m_comboBox.GetWindowText( szName );
if (!szName.IsEmpty())
{
CString log_path = _T("/lm/logging"), guid;
CString csTempPassword;
m_szPassword.CopyTo(csTempPassword);
CComAuthInfo auth(m_szMachine, m_szUserName, csTempPassword);
CMetaKey mk(&auth, log_path, METADATA_PERMISSION_READ | METADATA_PERMISSION_WRITE);
if (mk.Succeeded())
{
mk.QueryValue(MD_LOG_PLUGIN_UI_ID, szIID, NULL, szName);
bRes = mk.Succeeded();
}
}
return bRes;
}
//---------------------------------------------------------------------------
// OLE Interfaced Routine
void CLogUICtrl::SetComboBox(HWND hComboBox)
{
CString szAvailableList;
CString szCurrentModGuid;
// in case there are any errors, prepare the error string
// set the name of the application correctly
szAvailableList.LoadString( IDS_LOG_ERR_TITLE );
AfxGetApp()->m_pszAppName = _tcsdup(szAvailableList);
szAvailableList.Empty();
// attach the combo box
m_comboBox.Attach(hComboBox);
m_fComboInit = TRUE;
CString csTempPassword;
m_szPassword.CopyTo(csTempPassword);
CComAuthInfo auth(m_szMachine, m_szUserName, csTempPassword);
CMetaKey mk(&auth);
if (mk.Succeeded())
{
if (FAILED(mk.QueryValue(MD_LOG_PLUGIN_ORDER, szCurrentModGuid, NULL, m_szMetaObject)))
{
AfxMessageBox( IDS_ERR_LOG_PLUGIN );
return;
}
}
CString info;
CMetabasePath::GetServiceInfoPath(m_szMetaObject, info);
if (FAILED(mk.QueryValue(MD_LOG_PLUGINS_AVAILABLE, szAvailableList, NULL, info)))
{
AfxMessageBox( IDS_ERR_LOG_PLUGIN );
return;
}
CMetaEnumerator me(FALSE, &mk);
CMetabasePath log_path(TRUE, _T("logging"));
CString key, buf;
BOOL fFoundCurrent = FALSE;
while (SUCCEEDED(me.Next(key, log_path)))
{
int idx = 0;
if ((idx = szAvailableList.Find(key)) >= 0)
{
// Log plugin name could include "Custom Logging". Check if this is part of string.
// we should have comma before and after string
BOOL bCommaAfter =
szAvailableList.GetLength() == idx + key.GetLength()
|| szAvailableList.GetAt(idx + key.GetLength()) == _T(',');
BOOL bCommaBefore = idx == 0 || szAvailableList.GetAt(idx - 1) == _T(',');
if (!fFoundCurrent)
{
CMetabasePath current_path(FALSE, log_path, key);
mk.QueryValue(MD_LOG_PLUGIN_MOD_ID, buf, NULL, current_path);
fFoundCurrent = (buf == szCurrentModGuid);
if (fFoundCurrent)
{
buf = key;
}
}
if (bCommaBefore && bCommaAfter)
{
m_comboBox.AddString(key);
}
}
}
// select the current item in the combo box
m_comboBox.SelectString(-1, buf);
}
//---------------------------------------------------------------------------
// OLE Interfaced Routine
void CLogUICtrl::Terminate()
{
if ( m_fComboInit )
m_comboBox.Detach();
m_fComboInit = FALSE;
}
//------------------------------------------------------------------------
void CLogUICtrl::OnAmbientPropertyChange(DISPID dispid)
{
BOOL flag;
UINT style;
// do the right thing depending on the dispid
switch ( dispid )
{
case DISPID_AMBIENT_DISPLAYASDEFAULT:
if ( GetAmbientProperty( DISPID_AMBIENT_DISPLAYASDEFAULT, VT_BOOL, &flag ) )
{
style = GetWindowLong(
GetSafeHwnd(), // handle of window
GWL_STYLE // offset of value to retrieve
);
if ( flag )
style |= BS_DEFPUSHBUTTON;
else
style ^= BS_DEFPUSHBUTTON;
SetWindowLong(
GetSafeHwnd(), // handle of window
GWL_STYLE, // offset of value to retrieve
style
);
Invalidate(TRUE);
}
break;
};
COleControl::OnAmbientPropertyChange(dispid);
}
//------------------------------------------------------------------------
// an important method where we tell the container how to deal with us.
// pControlInfo is passed in by the container, although we are responsible
// for maintining the hAccel structure
void CLogUICtrl::OnGetControlInfo(LPCONTROLINFO pControlInfo)
{
if ( !pControlInfo || pControlInfo->cb < sizeof(CONTROLINFO) )
return;
pControlInfo->hAccel = m_hAccel;
pControlInfo->cAccel = m_cAccel;
pControlInfo->dwFlags = CTRLINFO_EATS_RETURN;
}
//------------------------------------------------------------------------
// the ole control container object specifically filters out the space
// key so we do not get it as a OnMnemonic call. Thus we need to look
// for it ourselves
void CLogUICtrl::OnKeyUpEvent(USHORT nChar, USHORT nShiftState)
{
if ( nChar == _T(' ') )
{
OnClick((USHORT)GetDlgCtrlID());
}
COleControl::OnKeyUpEvent(nChar, nShiftState);
}
//------------------------------------------------------------------------
void CLogUICtrl::OnMnemonic(LPMSG pMsg)
{
OnClick((USHORT)GetDlgCtrlID());
COleControl::OnMnemonic(pMsg);
}
//------------------------------------------------------------------------
void CLogUICtrl::OnTextChanged()
{
// get the new text
CString sz = InternalGetText();
// set the accelerator table
SetAccelTable((LPCTSTR)sz);
if ( SetAccelTable((LPCTSTR)sz) )
// make sure the new accelerator table gets loaded
ControlInfoChanged();
// finish with the default handling.
COleControl::OnTextChanged();
}
//------------------------------------------------------------------------
BOOL CLogUICtrl::SetAccelTable( LPCTSTR pszCaption )
{
BOOL fAnswer = FALSE;
ACCEL accel;
int iAccel;
// get the new text
CString sz = pszCaption;
sz.MakeLower();
// if the handle has already been allocated, free it
if ( m_hAccel )
{
DestroyAcceleratorTable( m_hAccel );
m_hAccel = NULL;
m_cAccel = 0;
}
// if there is a & character, then declare the accelerator
iAccel = sz.Find(_T('&'));
if ( iAccel >= 0 )
{
// fill in the accererator record
accel.fVirt = FALT;
accel.key = sz.GetAt(iAccel + 1);
accel.cmd = (USHORT)GetDlgCtrlID();
m_hAccel = CreateAcceleratorTable( &accel, 1 );
if ( m_hAccel )
m_cAccel = 1;
fAnswer = TRUE;
}
// return the answer
return fAnswer;
}