MTSSPY.CPP
// MtsSpy.cpp : Implementation of CMtsSpy
#include "stdafx.h"
#include "MtsSpyCtl.h"
#include "MtsAudit.h"
#include "MtsSpy.h"
#include "mtsevents.h"
#include "MtsSink.h"
#include "objsink.h"
#include "PackageSink.h"
#include "PackageInfo.h"
#include "TxSink.h"
#include "resourcesink.h"
#include "selecteventsdlg.h"
#include "commdlg.h"
#include <time.h>
#include "shellapi.h"
#include "MtsGrp.h"
LONGLONG PerformanceFrequency = 0;
//
//for some reason this define isn't being included
//
const GUID CLSID_StockFontPage = {0x7ebdaae0, 0x8120, 0x11cf, { 0x89, 0x9f, 0x0, 0xaa, 0x0, 0x68, 0x8b, 0x10}};
//
// given a GUID, return its string representation
//
CComBSTR GuidToBstr(REFGUID guid)
{
CComBSTR b(40, L"");
int nRet = StringFromGUID2(guid, b.m_str, 40);
_ASSERTE(nRet);
return b;
}
/////////////////////////////////////////////////////////////////////////////
// CMtsSpy
#define MAX_MSG_LENGTH (1024*sizeof(TCHAR))
static LPTSTR sColNames[] = {_TEXT("Count"),_TEXT("Event"),_TEXT("Tick Count"),
_TEXT("Package"),_TEXT("Parameter"),TEXT("Value")};
HRESULT CMtsSpy::OnDrawAdvanced(ATL_DRAWINFO& di)
{
return S_OK;
}
LRESULT CMtsSpy::OnRButtonDown(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
HMENU hMenu = LoadMenu(_Module.GetResourceInstance(), MAKEINTRESOURCE(IDR_SPYMENU));
if (!hMenu)
{
_ASSERTE(0);
return 0;
}
POINT local;
local.x = LOWORD(lParam);
local.y = HIWORD(lParam);
ClientToScreen(&local);
HMENU hPopup = GetSubMenu(hMenu, 0);
_ASSERTE(hPopup != NULL);
TrackPopupMenu(hPopup, TPM_LEFTALIGN | TPM_RIGHTBUTTON, local.x, local.y, NULL, m_hWndList, NULL);
return 0;
}
LRESULT CMtsSpy::OnCreate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
RECT rc;
GetWindowRect(&rc);
rc.right -= rc.left;
rc.bottom -= rc.top;
rc.top = rc.left = 0;
DWORD dwStyle = WS_VISIBLE|WS_CHILD|LVS_REPORT|LVS_SINGLESEL|WS_BORDER;
if (!m_hWndList)
m_hWndList = m_ctlSysListView32.Create(m_hWnd, rc, _T(""), dwStyle);
_ASSERTE(m_hWndList);
put_ShowGridLines(m_bShowGridLines);
//
//we always want the list to be Full Row Select --
//note: this only works if the user has IE 3.0 installed
//
dwStyle = ListView_GetExtendedListViewStyle(m_hWndList);
dwStyle |= LVS_EX_FULLROWSELECT;
ListView_SetExtendedListViewStyle(m_hWndList, dwStyle);
//
//create the columns in our list
//
LV_COLUMN col;
memset(&col,NULL, sizeof(col));
col.mask = LVCF_TEXT|LVCF_WIDTH;
col.fmt = LVCFMT_LEFT;
int i;
for (i=0;i<NUMBER_COLUMNS;i++)
{
col.cx = m_nWidth[i];
col.pszText = sColNames[i];
col.cchTextMax = lstrlen(col.pszText);
ListView_InsertColumn(m_hWndList, i, &col);
}
QueryPerformanceFrequency((LARGE_INTEGER *)&PerformanceFrequency);
if (m_hFont)
{
// this is set if the container app has set our properties using the property bag
::SendMessage(m_hWndList, WM_SETFONT, (WPARAM)m_hFont, MAKELPARAM(TRUE,0));
}
return 0;
}
//
//update the UI in menus
//
LRESULT CMtsSpy::OnInitMenu(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
HMENU hMenu = (HMENU)wParam;
CComBSTR temp = "Log to File ";
if (m_sLogFile.Length() > 5)
temp += m_sLogFile;
else
temp += "...";
USES_CONVERSION;
ModifyMenu(hMenu, ID_LOG, MF_BYCOMMAND|MF_STRING, ID_LOG, OLE2T(temp));
CheckMenuItem(hMenu, ID_LOG, m_bLogToFile ? MF_BYCOMMAND|MF_CHECKED : MF_BYCOMMAND|MF_UNCHECKED);
CheckMenuItem(hMenu, ID_OPTIONS_GRID_LINES, m_bShowGridLines ? MF_BYCOMMAND|MF_CHECKED : MF_BYCOMMAND|MF_UNCHECKED);
CheckMenuItem(hMenu, ID_SHOW_ON_SCREEN, m_bShowOnScreen ? MF_BYCOMMAND|MF_CHECKED : MF_BYCOMMAND|MF_UNCHECKED);
CheckMenuItem(hMenu, ID_AUDIT, m_bAudit ? MF_BYCOMMAND|MF_CHECKED : MF_BYCOMMAND|MF_UNCHECKED);
m_hMenuDebug = GetSubMenu(hMenu, 5);
_ASSERTE(m_hMenuDebug);
AddRunningAspsToDebugMenu(m_hMenuDebug);
return 0;
}
//
//sets a glog to log to file
//if the user hasn't chosen a log file, show the common dialog to allow them to select one
//
LRESULT CMtsSpy::OnLogToFile(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
m_bLogToFile = 1 - m_bLogToFile;
if (m_bLogToFile)
{
if (m_sLogFile.Length() == 0)
{
BOOL bHandled = TRUE;
OnChooseLogFile(0,0,0,bHandled);
if (m_sLogFile.Length() == 0)
return 0;
}
m_hFile = CreateFile(m_sLogFile, GENERIC_WRITE,
0, (LPSECURITY_ATTRIBUTES) NULL,
OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL,
(HANDLE) NULL);
if (!m_hFile)
{
m_bLogToFile = FALSE;
return 0;
}
}
else
{
if (m_hFile)
{
CloseHandle(m_hFile);
m_hFile = NULL;
}
}
return 0;
}
//
//User wants to choose a log file -- show the common dialog
//
LRESULT CMtsSpy::OnChooseLogFile(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
USES_CONVERSION;
OPENFILENAME OpenFileName;
TCHAR szFile[MAX_PATH];
memset(szFile, NULL, MAX_PATH * sizeof(TCHAR));
if (m_sLogFile.Length() > 2)
lstrcpy(szFile, OLE2T(m_sLogFile.m_str));
else
lstrcpy(szFile, _T("MTSSpy.Log"));
// Fill in the OPENFILENAME structure to support a template and hook.
OpenFileName.lStructSize = sizeof(OPENFILENAME);
OpenFileName.hwndOwner = m_hWndList;
OpenFileName.hInstance = _Module.GetModuleInstance( );
OpenFileName.lpstrFilter = _T("TAB seperated Files (*.log)\0*.log\0CSV Files (*.csv)\0*.csv\0");
OpenFileName.lpstrCustomFilter = NULL;
OpenFileName.nMaxCustFilter = 0;
OpenFileName.nFilterIndex = 0;
OpenFileName.lpstrFile = szFile;
OpenFileName.nMaxFile = sizeof(szFile);
OpenFileName.lpstrFileTitle = NULL;
OpenFileName.nMaxFileTitle = 0;
OpenFileName.lpstrInitialDir = NULL;
OpenFileName.lpstrTitle = _T("Choose Log File");
OpenFileName.nFileOffset = 0;
OpenFileName.nFileExtension = 0;
OpenFileName.lpstrDefExt = NULL;
OpenFileName.lCustData = 0;
OpenFileName.lpfnHook = NULL;
OpenFileName.lpTemplateName = NULL;
OpenFileName.Flags = OFN_EXPLORER | OFN_HIDEREADONLY |OFN_NOREADONLYRETURN |OFN_CREATEPROMPT;
// Call the common dialog function.
if (GetSaveFileName(&OpenFileName))
{
m_sLogFile = OpenFileName.lpstrFile;
if (lstrcmpi(OpenFileName.lpstrFile + OpenFileName.nFileExtension, _T("csv")) == 0)
m_bCSV = TRUE;
}
return 0;
}
//
//show/hide gridlines in the list
//
//NOTE: This will only work if the user has installed IE 3.0 or greater
//
LRESULT CMtsSpy::OnToggleGridLines(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
m_bShowGridLines = 1 - m_bShowGridLines;
DWORD dwStyle = ListView_GetExtendedListViewStyle(m_hWndList);
(m_bShowGridLines) ? dwStyle |= LVS_EX_GRIDLINES :
dwStyle &= ~LVS_EX_GRIDLINES;
ListView_SetExtendedListViewStyle(m_hWndList, dwStyle);
return 0;
}
//
//enable/disable auditing
//
//
LRESULT CMtsSpy::OnToggleAudit(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
m_bAudit = 1 - m_bAudit;
HRESULT hr = EnableAudit(m_bAudit);
if (hr != S_OK)
m_bAudit = FALSE;
return 0;
}
LRESULT CMtsSpy::OnToggleShowOnScreen(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
m_bShowOnScreen = 1-m_bShowOnScreen;
return 0;
}
//
//show the standard Shell About box
//
LRESULT CMtsSpy::OnAbout(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
HICON hIcon = LoadIcon(_Module.GetModuleInstance(), MAKEINTRESOURCE(IDI_MAIN));
ShellAbout(m_hWnd, _T("MTS Spy"), _T("Version 1.00"), hIcon);
return 0;
}
//
//clear the list box
//
LRESULT CMtsSpy::OnClear(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
ListView_DeleteAllItems(m_hWndList);
m_cEvents = 0;
return 0;
}
//
//save the items in the list box to the log file
//
LRESULT CMtsSpy::OnSave(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
USES_CONVERSION;
OPENFILENAME OpenFileName;
TCHAR szFile[MAX_PATH];
memset(szFile, NULL, MAX_PATH * sizeof(TCHAR));
if (m_sLogFile.Length() > 2)
lstrcpy(szFile, OLE2T(m_sLogFile.m_str));
else
lstrcpy(szFile, _T("MTSSpy.Log"));
OpenFileName.lStructSize = sizeof(OPENFILENAME);
OpenFileName.hwndOwner = m_hWndList;
OpenFileName.hInstance = _Module.GetModuleInstance( );
OpenFileName.lpstrFilter = NULL;
OpenFileName.lpstrCustomFilter = NULL;
OpenFileName.nMaxCustFilter = 0;
OpenFileName.nFilterIndex = 0;
OpenFileName.lpstrFile = szFile;
OpenFileName.nMaxFile = sizeof(szFile);
OpenFileName.lpstrFileTitle = NULL;
OpenFileName.nMaxFileTitle = 0;
OpenFileName.lpstrInitialDir = NULL;
OpenFileName.lpstrTitle = _T("Log ToFile");
OpenFileName.nFileOffset = 0;
OpenFileName.nFileExtension = 0;
OpenFileName.lpstrDefExt = NULL;
OpenFileName.lCustData = 0;
OpenFileName.lpfnHook = NULL;
OpenFileName.lpTemplateName = NULL;
OpenFileName.Flags = OFN_EXPLORER | OFN_HIDEREADONLY |OFN_NOREADONLYRETURN;
// Call the common dialog function.
if (GetSaveFileName(&OpenFileName))
{
FILE* f = _tfopen(OpenFileName.lpstrFile, _T("a+"));
if (f)
{
int nCount = ListView_GetItemCount(m_hWndList);
int i, j;
TCHAR buf[MAX_MSG_LENGTH];
for (i=0; i<nCount; i++)
{
for (j=0; j < 4; j++)
{
ListView_GetItemText(m_hWndList, i, j, buf, MAX_MSG_LENGTH);
_ftprintf(f, _T("%s\t"),buf);
}
while ( i < nCount - 1)
{
i++; // move to next column
// now get the name/value pairs
ListView_GetItemText(m_hWndList, i, 4, buf, MAX_MSG_LENGTH);
if (*buf)
{
_ftprintf(f, _T("\n\t\t\t\t%s\t"), buf);
ListView_GetItemText(m_hWndList, i, 5, buf, MAX_MSG_LENGTH);
_ftprintf(f, _T("%s"),buf);
}
else
{
i--;
_ftprintf(f, _T("\n"), NULL);
break;
}
}
}
fclose(f);
}
}
return 0;
}
//
//given the index to the list, we return TRUE if the index
//should scroll -- this occurs when the user is looking
//at the last element in the list. If they have scrolled
//up to look at a previous element in the list, then we
//don't scroll when an event is added
//
BOOL CMtsSpy::ShouldScroll(int nIndex)
{
BOOL bRet = FALSE;
int nCountPage = ListView_GetCountPerPage(m_hWndList);
int nTopIndex = ListView_GetTopIndex(m_hWndList);
if (nTopIndex + nCountPage>= nIndex)
bRet = TRUE;
return bRet;
}
//
//disconnect & destroy all sinks associated with the given package
//
HRESULT CMtsSpy::ShutdownPackage(LPCOLESTR sPackageName)
{
USES_CONVERSION;
CPackageInfo * pInfo = m_map[W2A(sPackageName)];
if (!pInfo)
{
_ASSERTE(0); // we should never get this unless we have a sink for (at least) the Package events
return E_FAIL;
}
int nElementsRemoved = m_map.erase(W2A(sPackageName));
_ASSERTE(nElementsRemoved == 1);
delete pInfo;
return S_OK;
}
//
//this method is called by all of the sinks to add an item to the list box
//
BOOL CMtsSpy::AddEventToList(LONGLONG perfCount, LPCOLESTR sEvent, LPCOLESTR sPackage)
{
long TickCount = 0;
if (PerformanceFrequency != 0)
TickCount = (long)((1000 * perfCount) / PerformanceFrequency);
BOOL bRet;
if (m_bLogToFile)
{
TCHAR sSep[2];
ZeroMemory(sSep,2 * sizeof(TCHAR));
(m_bCSV) ? lstrcat(sSep, _T(",")) : lstrcat(sSep, _T("\t"));
sSep[1] = NULL;
_ASSERTE(m_hFile);
TCHAR sBuf[MAX_MSG_LENGTH];
ZeroMemory(sBuf, MAX_MSG_LENGTH*sizeof(TCHAR));
wsprintf(sBuf, _T("%s%s%d%s%s\n"), sEvent, sSep, TickCount, sSep, sPackage);
if (m_hFile)
{
DWORD cbWritten;
bRet = WriteFile(m_hFile, sBuf, lstrlen(sBuf) * sizeof(TCHAR), &cbWritten, NULL);
}
}
if (m_bShowOnScreen)
{
int nCount = ListView_GetItemCount(m_hWndList);
LV_ITEM item;
TCHAR sz[8];
wsprintf(sz,_TEXT("%ld"), m_cEvents++);
memset(&item, NULL, sizeof(item));
item.mask = LVIF_TEXT;
item.iItem = nCount;
item.pszText = sz;
item.cchTextMax = lstrlen(sz);
ListView_InsertItem(m_hWndList, &item);
ListView_SetItemText(m_hWndList, nCount, 1, (LPWSTR)sEvent);
TCHAR szTick[16];
wsprintf(szTick, _T("%ld"), TickCount);
ListView_SetItemText(m_hWndList, nCount, 2, szTick);
ListView_SetItemText(m_hWndList, nCount, 3, (LPWSTR)sPackage);
if (ShouldScroll(nCount))
ListView_EnsureVisible(m_hWndList, nCount, FALSE);
}
return TRUE;
}
//
//this method is called by all the sinks to add a Named/Value pair to the
//list
//
BOOL CMtsSpy::AddParamValueToList(LPCOLESTR sParamName, LPCOLESTR sValue)
{
BOOL bRet;
if (m_bLogToFile)
{
TCHAR sSep[2];
ZeroMemory(sSep,2 * sizeof(TCHAR));
(m_bCSV) ? lstrcat(sSep, _T(",")) : lstrcat(sSep, _T("\t"));
sSep[1] = NULL;
_ASSERTE(m_hFile);
TCHAR sBuf[MAX_MSG_LENGTH];
ZeroMemory(sBuf, MAX_MSG_LENGTH*sizeof(TCHAR));
wsprintf(sBuf, _T("%s%s%s%s%s%s\n"), sSep, sSep, sSep, sParamName,sSep, sValue);
if (m_hFile)
{
DWORD cbWritten;
bRet = WriteFile(m_hFile, sBuf, lstrlen(sBuf) * sizeof(TCHAR), &cbWritten, NULL);
}
}
if (m_bShowOnScreen)
{
int nCount = ListView_GetItemCount(m_hWndList);
LV_ITEM item;
item.mask = LVIF_TEXT;
item.iItem = nCount;
item.pszText = _T("");
item.cchTextMax = 1;
item.iSubItem = 0;
int nItem;
nItem = ListView_InsertItem(m_hWndList, &item);
ListView_SetItemText(m_hWndList, nItem, 4, (LPWSTR)sParamName);
ListView_SetItemText(m_hWndList, nItem, 5, (LPWSTR)sValue);
if (ShouldScroll(nCount))
ListView_EnsureVisible(m_hWndList, nItem, FALSE);
}
return TRUE;
}
//
//show the select packages dialog
//
LRESULT CMtsSpy::OnSelectPackages(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
CSelectEventsDlg dlg(&m_map, this);
dlg.DoModal();
return 0;
}
LRESULT CMtsSpy::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
bHandled = FALSE; // continue processing
//
//clean up our package information
//
MapStringToPackageInfo::iterator iter;
CPackageInfo * pInfo;
for (iter = m_map.begin(); iter != m_map.end(); ++iter)
{
pInfo = (*iter).second;
delete pInfo;
}
m_map.clear();
if (m_hFont)
{
DeleteObject(m_hFont);
m_hFont = NULL;
}
m_bLogToFile = FALSE;
if (m_hFile)
{
CloseHandle(m_hFile);
m_hFile = NULL;
}
m_cEvents = 0;
m_bShowGridLines = TRUE;
return 0;
}
//
//show only the ChooseFont property page
//
LRESULT CMtsSpy::OnChooseFont(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
HRESULT hr = S_OK;
CComPtr<IUnknown> pUnk;
ControlQueryInterface(IID_IUnknown, (void**)&pUnk);
_ASSERTE(pUnk != NULL);
LPOLESTR szTitle = _T("MTS Spy - Choose Font");
hr = OleCreatePropertyFrame(m_hWnd, m_rcPos.top, m_rcPos.left, szTitle,
1, &pUnk.p, 1, (GUID *)&CLSID_StockFontPage, LOCALE_USER_DEFAULT, 0, 0);
return 0;
}
STDMETHODIMP CMtsSpy::get_LogFile(BSTR * pVal)
{
*pVal = m_sLogFile.Copy();
return S_OK;
}
STDMETHODIMP CMtsSpy::put_LogFile(BSTR newVal)
{
HANDLE hTemp;
hTemp = CreateFile(newVal, GENERIC_WRITE,
0, (LPSECURITY_ATTRIBUTES) NULL,
OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL,
(HANDLE) NULL);
if (hTemp)
{
if (m_hFile)
{
CloseHandle(m_hFile);
}
m_hFile = hTemp;
}
else
return E_INVALIDARG;
m_sLogFile = newVal;
return S_OK;
}
STDMETHODIMP CMtsSpy::get_ShowGridLines(BOOL * pVal)
{
*pVal = m_bShowGridLines;
return S_OK;
}
STDMETHODIMP CMtsSpy::put_ShowGridLines(BOOL newVal)
{
m_bShowGridLines = newVal;
DWORD dwStyle = ListView_GetExtendedListViewStyle(m_hWndList);
(m_bShowGridLines) ? dwStyle |= LVS_EX_GRIDLINES :
dwStyle &= ~LVS_EX_GRIDLINES;
ListView_SetExtendedListViewStyle(m_hWndList, dwStyle);
FireViewChange();
return S_OK;
}
STDMETHODIMP CMtsSpy::get_Audit(BOOL * pVal)
{
*pVal = m_bAudit;
return S_OK;
}
STDMETHODIMP CMtsSpy::put_Audit(BOOL newVal)
{
m_bAudit = newVal;
EnableAudit(m_bAudit);
return S_OK;
}
STDMETHODIMP CMtsSpy::get_ColWidth(short nColumn, long * pVal)
{
_ASSERTE(m_hWndList);
_ASSERTE(nColumn >= 0 && nColumn <= NUMBER_COLUMNS);
*pVal = ListView_GetColumnWidth(m_hWndList, nColumn);
return S_OK;
}
STDMETHODIMP CMtsSpy::put_ColWidth(short nColumn, long newVal)
{
if (nColumn >= NUMBER_COLUMNS)
return E_INVALIDARG;
m_nWidth[nColumn] = newVal;
ListView_SetColumnWidth(m_hWndList, nColumn, newVal);
FireViewChange();
return S_OK;
}
HRESULT CMtsSpy::IPersistStreamInit_Load(LPSTREAM pStm, ATL_PROPMAP_ENTRY* pMap)
{
ULONG cbRead;
// read in the version
UINT nVersion;
pStm -> Read((void *)&nVersion, sizeof(UINT), &cbRead);
if (nVersion != 3)
{
::MessageBox(m_hWndList, _T("The saved information is from a different version of the control.\n\nThe saved state will have to be discarded."), _T("MTS Spy"), MB_ICONINFORMATION);
return S_OK;
}
// get log file size
long lLen;
pStm -> Read((void *)&lLen, sizeof(long), &cbRead);
// read in the chars for the log name
char * pLog = new char[lLen+1];
ZeroMemory(pLog, lLen+1);
pStm -> Read((void*)pLog, lLen, &cbRead);
m_sLogFile = pLog;
delete [] pLog;
// gridlines
pStm -> Read((void *)&m_bShowGridLines, sizeof(BOOL),&cbRead);
put_ShowGridLines(m_bShowGridLines);
// show on screen
pStm -> Read((void *)&m_bShowOnScreen, sizeof(BOOL),&cbRead);
put_ShowOnScreen(m_bShowOnScreen);
// log to file
pStm -> Read((void *)&m_bLogToFile, sizeof(BOOL),&cbRead);
put_LogToFile(m_bLogToFile);
// csv files
pStm -> Read((void *)&m_bCSV, sizeof(BOOL), &cbRead);
// csv files
pStm -> Read((void *)&m_bAudit, sizeof(BOOL), &cbRead);
EnableAudit(m_bAudit);
// read the col widths
int i;
long nWidth;
for (i=0;i<NUMBER_COLUMNS;i++)
{
pStm -> Read((void *)&nWidth, sizeof(long), &cbRead);
put_ColWidth(i, nWidth);
}
BOOL bCustomFont = FALSE;
pStm -> Read((void *)&bCustomFont, sizeof(BOOL),&cbRead);
if (bCustomFont)
{
if (!m_pFont)
{
//create an IFontDisp
HRESULT hr= CoCreateInstance(CLSID_StdFont, NULL, CLSCTX_ALL, IID_IFontDisp, (void **)&m_pFont);
_ASSERTE(SUCCEEDED(hr));
}
if (m_pFont)
{
IPersistStream * pFontSt = NULL;
m_pFont -> QueryInterface(IID_IPersistStream, (void **)&pFontSt);
pFontSt -> Load(pStm);
pFontSt -> Release();
if (m_hFont) // should I tell the list that I deleted its font?
DeleteObject(m_hFont);
IFont * pFont = NULL;
m_pFont -> QueryInterface(IID_IFont, (void **)&pFont);
_ASSERTE(pFont);
m_hFont = CreateHFontFromIFont(pFont);
pFont -> Release();
}
}
return S_OK;
}
HRESULT CMtsSpy::IPersistStreamInit_Save(LPSTREAM pStm, BOOL /* fClearDirty */, ATL_PROPMAP_ENTRY* pMap)
{
_ASSERTE(this);
_ASSERTE(m_hWndList);
ULONG cbWritten = 0;
// write a version
UINT nVersion = 3;
pStm -> Write((void *)&nVersion, sizeof(UINT), &cbWritten);
// write the log file size
long nChars = m_sLogFile.Length();
pStm -> Write((void *)&nChars, sizeof(long), &cbWritten);
_ASSERTE(cbWritten == sizeof(long));
// write the log file name (in ANSI!)
USES_CONVERSION;
char * sFile = W2A((BSTR)m_sLogFile);
pStm -> Write((void *)sFile, m_sLogFile.Length(), &cbWritten);
// write the BOOL if gridlines should be shown
pStm -> Write((void *)&m_bShowGridLines, sizeof(BOOL), &cbWritten);
// write the BOOL if the user wants to see the messages
pStm -> Write((void *)&m_bShowOnScreen, sizeof(BOOL), &cbWritten);
// write the BOOL if the user wants to log the messages
pStm -> Write((void *)&m_bLogToFile, sizeof(BOOL), &cbWritten);
// write the BOOL if we have CSV files
pStm -> Write((void *)&m_bCSV, sizeof(BOOL), &cbWritten);
// write the BOOL if Auditing is enabled
pStm -> Write((void *)&m_bAudit, sizeof(BOOL), &cbWritten);
// write the col widths
int i;
long nWidth;
for (i=0;i<NUMBER_COLUMNS;i++)
{
get_ColWidth(i, &nWidth);
_ASSERTE(nWidth);
pStm -> Write((void *)&nWidth, sizeof(long), &cbWritten);
}
// ask the font to persist itself in the stream
BOOL bCustomFont;
if (m_pFont)
{
bCustomFont = TRUE;
pStm -> Write((void *)&bCustomFont, sizeof(BOOL), &cbWritten);
IPersistStream * pSt = NULL;
m_pFont -> QueryInterface(IID_IPersistStream, (void **)&pSt);
_ASSERTE(pSt);
pSt -> Save(pStm, FALSE);
pSt -> Release();
}
else
{
bCustomFont = FALSE;
pStm -> Write((void *)&bCustomFont, sizeof(BOOL), &cbWritten);
}
return S_OK;
}
HRESULT CMtsSpy::Close( DWORD dwSaveOption )
{
return IOleObject_Close(dwSaveOption);
}
HFONT CMtsSpy::CreateHFontFromIFont(IFont * pFont)
{
_ASSERTE(pFont);
HFONT hFont;
LOGFONT lf;
ZeroMemory(&lf, sizeof(lf));
CComBSTR bstr;
pFont -> get_Name(&bstr);
lstrcpy(lf.lfFaceName, bstr.m_str);
CY cy;
pFont -> get_Size(&cy);
HDC hDC = ::GetDC(NULL);
_ASSERTE(hDC);
long lCaps = GetDeviceCaps(hDC, LOGPIXELSY);
lf.lfHeight = -MulDiv(cy.Hi, lCaps, 72);
lf.lfHeight += -MulDiv(cy.Lo, lCaps, 720000);
::ReleaseDC(NULL, hDC);
short sVal;
pFont -> get_Weight(&sVal);
lf.lfWeight = sVal;
BOOL bFlag;
pFont -> get_Italic(&bFlag);
lf.lfItalic = bFlag;
pFont -> get_Underline(&bFlag);
lf.lfUnderline = bFlag;
pFont -> get_Strikethrough(&bFlag);
lf.lfStrikeOut = bFlag;
pFont -> get_Charset(&sVal);
lf.lfCharSet = sVal;
hFont = CreateFontIndirect(&lf);
return hFont;
}
STDMETHODIMP CMtsSpy::SelectPackages()
{
BOOL bHandled;
OnSelectPackages(0,0,0,bHandled);
return S_OK;
}
STDMETHODIMP CMtsSpy::SaveToFile()
{
BOOL bHandled;
OnSave(0,0,0,bHandled);
return S_OK;
}
STDMETHODIMP CMtsSpy::ClearAllEvents()
{
BOOL bHandled;
OnClear(0,0,0,bHandled);
return S_OK;
}
STDMETHODIMP CMtsSpy::About()
{
BOOL bHandled;
OnAbout(0,0,0,bHandled);
return S_OK;
}
STDMETHODIMP CMtsSpy::get_LogToFile(BOOL * pVal)
{
*pVal = m_bLogToFile;
return S_OK;
}
STDMETHODIMP CMtsSpy::put_LogToFile(BOOL newVal)
{
m_bLogToFile = newVal;
if (m_bLogToFile)
{
if (m_sLogFile.Length() == 0)
{
BOOL bHandled = TRUE;
OnChooseLogFile(0,0,0,bHandled);
if (m_sLogFile.Length() == 0)
return 0;
}
if (m_hFile)
{
CloseHandle(m_hFile);
m_hFile = NULL;
}
m_hFile = CreateFile(m_sLogFile, GENERIC_WRITE,
0, (LPSECURITY_ATTRIBUTES) NULL,
OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL,
(HANDLE) NULL);
if (!m_hFile)
{
m_bLogToFile = FALSE;
return E_FAIL;
}
}
else
{
if (m_hFile)
{
CloseHandle(m_hFile);
m_hFile = NULL;
}
}
return S_OK;
}
STDMETHODIMP CMtsSpy::ChooseFont()
{
BOOL bHandled;
OnChooseFont(0,0,0,bHandled);
return S_OK;
}
STDMETHODIMP CMtsSpy::get_ShowOnScreen(BOOL * pVal)
{
*pVal = m_bShowOnScreen;
return S_OK;
}
STDMETHODIMP CMtsSpy::put_ShowOnScreen(BOOL newVal)
{
m_bShowOnScreen = newVal;
return S_OK;
}
STDMETHODIMP CMtsSpy::ChooseLogFile(BSTR * sLogFileName, BOOL * bCanceled)
{
*bCanceled = TRUE;
BOOL bHandled = TRUE;
OnChooseLogFile(0,0,0,bHandled);
if (m_sLogFile.Length() == 0)
{
return S_OK;
}
*sLogFileName = ::SysAllocString(m_sLogFile.m_str);
*bCanceled = FALSE;
return S_OK;
}
//
//loop through the ROT and add package names
//to the debug menu
//
BOOL CMtsSpy::AddRunningAspsToDebugMenu(HMENU hMenu)
{
//
//remove the <No Running Packages> menu item
//
RemoveMenu(hMenu, 0, MF_BYPOSITION);
HRESULT hr = E_FAIL;
IMtsGrp * pMtsGrp = NULL;
hr = CoCreateInstance (CLSID_MtsGrp, NULL, CLSCTX_ALL, IID_IMtsGrp, (void **)&pMtsGrp);
if (!pMtsGrp)
{
_ASSERTE(0);
return FALSE;
}
long lPackages;
pMtsGrp -> Refresh(); // its important to call this!
pMtsGrp -> get_Count(&lPackages);
IUnknown * pUnk = NULL;
IMtsEvents * pEvents = NULL;
USES_CONVERSION;
long lPID;
TCHAR szBuf[128];
for (int i=0; i<lPackages; i++)
{
pMtsGrp -> Item(i, &pUnk);
_ASSERTE(pUnk);
pUnk -> QueryInterface(IID_IMtsEvents, (void **)&pEvents);
_ASSERTE(pEvents);
if (SUCCEEDED(hr))
{
CComBSTR sName;
pEvents -> get_PackageName(&sName.m_str);
pEvents -> GetProcessID(&lPID);
ZeroMemory(szBuf, 128*sizeof(TCHAR));
if (sName.m_str)
{
wsprintf(szBuf, _T("%s - PID: %ld"), sName.m_str, lPID);
int nRet = InsertMenu(hMenu, -1, MF_BYPOSITION|MF_STRING, ID_DEBUG_BEGIN + i, szBuf);
}
SafeRelease(pEvents);
}
SafeRelease(pEvents);
SafeRelease(pUnk);
}
SafeRelease(pMtsGrp);
return TRUE;
}
//
//user wants to debug a package --
//
//NOTE: This is designed to work with Microsoft Developer Studio
//if you want to add support for another tool, modify the
//ShellExecute() line below. This is alson non-robust in
//the sense that it requires MSDEV.EXE to be in the path
//and set up correctly.
//
LRESULT CMtsSpy::OnDebugPackage(WORD wNotifyCode, WORD wID, HWND hWndCtl, BOOL& bHandled)
{
HCURSOR hBusy = LoadCursor(NULL, IDC_WAIT);
HCURSOR hOldCur = SetCursor(hBusy);
TCHAR szBuf[128];
ZeroMemory(szBuf, 128 * sizeof(TCHAR));
int nChars;
_ASSERTE(m_hMenuDebug);
nChars = GetMenuString(m_hMenuDebug, wID, szBuf, 128, MF_BYCOMMAND);
if (nChars)
{
int nCount = lstrlen(szBuf);
LPTSTR p = szBuf + nCount;
while (*p-- != ':' && nCount)
{
nCount --;
}
if (nCount == 0)
{
SetCursor(hOldCur);
return 0;
}
p+=3;
TCHAR buf[MAX_PATH];
wsprintf(buf, _T("-p %s -e %s"), p, p);
HINSTANCE hInst = ShellExecute( NULL, _T("open"), _T("msdev"), buf, NULL, SW_SHOWNORMAL);
}
SetCursor(hOldCur);
return 0;
}
//
//Enable/Disable auditing
//
HRESULT CMtsSpy::EnableAudit(BOOL bEnable)
{
if (!bEnable)
{
SafeRelease(m_pSqlAudit);
return S_OK;
}
if (m_pSqlAudit)// make EnableAudit() idempotent -- we are already enabled
return S_OK;
HRESULT hr;
hr = CoCreateInstance(CLSID_MtsSqlAudit, NULL, CLSCTX_ALL, IID_IMtsSqlAudit, (void **)&m_pSqlAudit);
if (SUCCEEDED(hr))
{
//
//these hard coded strings (user and pw) should be looked up in the registry for a production
//app
hr = m_pSqlAudit -> Init(L"MtsAudit", L"sa", L"");
if (FAILED(hr))
{
LPTSTR sMsg = _T("Unable to connect to DSN 'MtsAudit'. Make sure the DSN\nexists, that has it \
the schema contained in MtsAudit.SQL,\nand that you have permission to the database.");
::MessageBox(m_hWndList, sMsg, _T("MtsSpy"), MB_ICONSTOP);
SafeRelease(m_pSqlAudit);
}
}
else
{
::MessageBox(m_hWndList, _T("Unable to create the MtsAudit component!"), _T("MtsSpy"), MB_ICONSTOP);
}
return hr;
}