MS Com Spy的源代码

// ---------------------------------------------------------------------------- // // This file is part of the Microsoft COM+ Samples. // // Copyright (C) 1995-2000 Microsoft Corporation. All rights reserved. // // This source code is intended only as a supplement to Microsoft // Development Tools and/or on-line documentation. See these other // materials for detailed information regarding Microsoft code samples. // // THIS CODE AND INFORMATION ARE PROVIDED "AS IS" WITHOUT WARRANTY OF ANY // KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE // IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A // PARTICULAR PURPOSE. // // ---------------------------------------------------------------------------- #include "stdafx.h" #include "ComSpyAudit.h" #ifndef DBNTWIN32 #define DBNTWIN32 #include <SQL.h> #include <SQLEXT.h> #include <ODBCINST.h> #endif // DBNTWIN32 #include <ODBCINST.h> #include "ComSpySqlAudit.h" #define SQL_EXEC(x) \ { \ RETCODE rc; \ rc = SQLExecDirect(m_hstmt, x, lstrlen(x)); \ if( rc != SQL_SUCCESS && rc != SQL_SUCCESS_WITH_INFO ) \ { \ CComBSTR msg; \ GetError( m_henv, m_hdbc, m_hstmt, &msg.m_str); \ ATLTRACE(L"Warning! SQlExecDirec failed! Msg: "); \ ATLTRACE(msg.m_str); \ ATLTRACE(L"\n"); \ } \ } long CComSpySqlAudit::PerfCountToTickCount(LONGLONG perfCount) { long TickCount = 0; if (m_PerformanceFrequency != 0) TickCount = (long)((1000 * perfCount) / m_PerformanceFrequency); return TickCount; } //////////////////////////Thread Events/////////////////////////////////// STDMETHODIMP CComSpySqlAudit::OnThreadStart( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR pszW2KThreadID, DWORD dwThreadCnt) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, W2KThreadID, ThreadCnt) values ('%s',%d,'%s','%s','%s', %d)", L"OnThreadStart", TickCount, pszGuidApp, pszThreadID, pszW2KThreadID, dwThreadCnt); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadTerminate( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR pszW2KThreadID, DWORD dwThreadCnt ) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, W2KThreadID, ThreadCnt) values ('%s',%d,'%s','%s','%s', %d)", L"OnThreadTerminate", TickCount, pszGuidApp, pszThreadID, pszW2KThreadID, dwThreadCnt); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadBindToApartment( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR pszAptID, DWORD dwActivityCnt) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, ApartmentID, ActivityCnt) values ('%s', %d,'%s','%s','%s', %d)", L"OnThreadBindToApartment", TickCount, pszGuidApp, pszThreadID, pszAptID, dwActivityCnt); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadUnBind( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR pszAptID, DWORD dwActivityCnt ) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, Threadid, ApartmentID, ActivityCnt) values ('%s',%d,'%s','%s','%s', %d)", L"OnThreadUnBind", TickCount, pszGuidApp, pszThreadID, pszAptID, dwActivityCnt); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadAssignApartment( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszGuidActivityID, LPCOLESTR pszAptID ) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ActivityID, ApartmentID) values ('%s',%d,'%s','%s','%s')", L"OnThreadAssignApartment", TickCount, pszGuidApp, pszGuidActivityID, pszAptID); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadUnassignApartment( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszAptID ) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ApartmentID) values ('%s', %d,'%s','%s')", L"OnThreadUnAssignApartment", TickCount, pszGuidApp, pszAptID); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadWorkEnque( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR MsgWorkID, DWORD dwQueueLen) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, MsgWorkID,QueueLen) values ('%s', %d,'%s','%s','%s', %d)", L"OnThreadWorkEnque", TickCount, pszGuidApp, pszThreadID, MsgWorkID, dwQueueLen); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadWorkPrivate( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR MsgWorkID) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, MsgWorkID) values ('%s', %d, '%s','%s','%s')", L"OnThreadWorkPrivate", TickCount, pszGuidApp, pszThreadID, MsgWorkID); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadWorkPublic( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR MsgWorkID, DWORD dwQueueLen) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, MsgWorkID,QueueLen) values ('%s', %d,'%s','%s','%s', %d)", L"OnThreadWorkPublic", TickCount, pszGuidApp, pszThreadID, MsgWorkID, dwQueueLen); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadWorkRedirect( LONGLONG perfCount, LPCOLESTR pszGuidApp, LPCOLESTR pszThreadID, LPCOLESTR MsgWorkID, DWORD dwQueueLen, LPCOLESTR pszThreadNum) { long TickCount = PerfCountToTickCount(perfCount); ZeroMemory(m_sSQL, SQL_BUF_SIZE * sizeof(WCHAR)); wsprintf(m_sSQL, L"insert into ThreadEvents (Event, TickCount, Application, ThreadID, MsgWorkID,QueueLen, ThreadNum) values ('%s',%d,'%s','%s','%s', %d, '%s')", L"OnThreadWorkPublic", TickCount, pszGuidApp, pszThreadID, MsgWorkID, dwQueueLen, pszThreadNum); SQL_EXEC(m_sSQL); return S_OK; } STDMETHODIMP CComSpySqlAudit::OnThreadWorkReject(