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.
 
 
 
 
 
 

810 lines
19 KiB

/*++
Copyright (C) 1996-1999 Microsoft Corporation
Module Name:
cntrtree.cpp
Abstract:
Implements internal counter management.
--*/
#include "polyline.h"
#include <strsafe.h>
#include <assert.h>
#include <pdhmsg.h>
#include "smonmsg.h"
#include "appmema.h"
#include "grphitem.h"
#include "cntrtree.h"
CCounterTree::CCounterTree()
: m_nItems (0)
{
}
HRESULT
CCounterTree::AddCounterItem(
LPWSTR pszPath,
PCGraphItem pItem,
BOOL bMonitorDuplicateInstances
)
{
HRESULT hr;
PPDH_COUNTER_PATH_ELEMENTS pPathInfo = NULL;
ULONG ulBufSize;
PDH_STATUS stat;
CMachineNode *pMachine;
CObjectNode *pObject;
CCounterNode *pCounter;
CInstanceNode *pInstance;
if ( NULL != pszPath && NULL != pItem ) {
// Record whether machine is explicit or defaults to local
pItem->m_fLocalMachine = !(pszPath[0] == L'\\' && pszPath[1] == L'\\');
pPathInfo = NULL;
do {
if (pPathInfo) {
delete [] ((char*)pPathInfo);
pPathInfo = NULL;
}
else {
ulBufSize = sizeof(PDH_COUNTER_PATH_ELEMENTS) + sizeof(WCHAR) * PDH_MAX_COUNTER_PATH;
}
pPathInfo = (PPDH_COUNTER_PATH_ELEMENTS) new char [ ulBufSize ];
if (pPathInfo == NULL) {
return E_OUTOFMEMORY;
}
stat = PdhParseCounterPath( pszPath, pPathInfo, & ulBufSize, 0);
} while (stat == PDH_INSUFFICIENT_BUFFER || stat == PDH_MORE_DATA);
//
// We use do {} while (0) here to act like a switch statement
//
do {
if (stat != ERROR_SUCCESS) {
hr = E_FAIL;
break;
}
// Find or create each level of hierarchy
hr = GetMachine( pPathInfo->szMachineName, &pMachine);
if (FAILED(hr) || NULL == pMachine ) {
break;
}
hr = pMachine->GetCounterObject(pPathInfo->szObjectName, &pObject);
if (FAILED(hr) || NULL == pObject ) {
break;
}
hr = pObject->GetCounter(pPathInfo->szCounterName, &pCounter);
if (FAILED(hr) || NULL == pCounter ) {
break;
}
hr = pObject->GetInstance(
pPathInfo->szParentInstance,
pPathInfo->szInstanceName,
pPathInfo->dwInstanceIndex,
bMonitorDuplicateInstances,
&pInstance);
if (FAILED(hr) || NULL == pInstance ) {
break;
}
hr = pInstance->AddItem(pCounter, pItem);
if (SUCCEEDED(hr)) {
m_nItems++;
UpdateAppPerfDwordData (DD_ITEM_COUNT, m_nItems);
}
} while (0);
} else {
hr = E_INVALIDARG;
}
if (pPathInfo) {
delete [] ((char*)pPathInfo);
}
return hr;
}
HRESULT
CCounterTree::GetMachine (
IN LPWSTR pszName,
OUT PCMachineNode *ppMachineRet
)
{
PCMachineNode pMachine = NULL;
PCMachineNode pMachineNew = NULL;
HRESULT hr = NOERROR;
if ( NULL == ppMachineRet || NULL == pszName ) {
hr = E_POINTER;
} else {
*ppMachineRet = NULL;
if (m_listMachines.FindByName(pszName, FIELD_OFFSET(CMachineNode, m_szName), (PCNamedNode*)&pMachine)) {
*ppMachineRet = pMachine;
hr = NOERROR;
} else {
pMachineNew = new(lstrlen(pszName) * sizeof(WCHAR)) CMachineNode;
if (!pMachineNew) {
hr = E_OUTOFMEMORY;
} else {
pMachineNew->m_pCounterTree = this;
StringCchCopy(pMachineNew->m_szName, lstrlen(pszName) + 1, pszName);
m_listMachines.Add(pMachineNew, pMachine);
*ppMachineRet = pMachineNew;
hr = NOERROR;
}
}
}
return hr;
}
void
CCounterTree::RemoveMachine (
IN PCMachineNode pMachine
)
{
// Remove machine from list and delete it
m_listMachines.Remove(pMachine);
delete pMachine ;
}
PCGraphItem
CCounterTree::FirstCounter (
void
)
{
if (!FirstMachine())
return NULL;
else
return FirstMachine()->FirstObject()->FirstInstance()->FirstItem();
}
HRESULT
CCounterTree::IndexFromCounter (
IN const CGraphItem* pItem,
OUT INT* pIndex )
{
HRESULT hr = E_POINTER;
CGraphItem* pLocalItem;
INT iLocalIndex = 0;
if ( NULL != pItem && NULL != pIndex ) {
*pIndex = 0;
hr = E_INVALIDARG;
pLocalItem = FirstCounter();
while ( NULL != pLocalItem ) {
iLocalIndex++;
if ( pLocalItem != pItem ) {
pLocalItem = pLocalItem->Next();
} else {
*pIndex = iLocalIndex;
hr = S_OK;
break;
}
}
}
return hr;
}
HRESULT
CMachineNode::GetCounterObject (
IN LPWSTR pszName,
OUT PCObjectNode *ppObjectRet
)
{
PCObjectNode pObject;
PCObjectNode pObjectNew;
if (m_listObjects.FindByName(pszName, FIELD_OFFSET(CObjectNode, m_szName), (PCNamedNode*)&pObject)) {
*ppObjectRet = pObject;
return NOERROR;
}
pObjectNew = new(lstrlen(pszName) * sizeof(WCHAR)) CObjectNode;
if (!pObjectNew)
return E_OUTOFMEMORY;
pObjectNew->m_pMachine = this;
StringCchCopy(pObjectNew->m_szName, lstrlen(pszName) + 1, pszName);
m_listObjects.Add(pObjectNew, pObject);
*ppObjectRet = pObjectNew;
return NOERROR;
}
void
CMachineNode::RemoveObject (
IN PCObjectNode pObject
)
{
// Remove object from list and delete it
m_listObjects.Remove(pObject);
delete pObject;
// If this was the last one, remove ourself
if (m_listObjects.IsEmpty())
m_pCounterTree->RemoveMachine(this);
}
void
CMachineNode::DeleteNode (
BOOL bPropagateUp
)
{
PCObjectNode pObject;
PCObjectNode pNextObject;
// Delete all object nodes
pObject = FirstObject();
while ( NULL != pObject ) {
pNextObject = pObject->Next();
pObject->DeleteNode(FALSE);
m_listObjects.Remove(pObject);
delete pObject;
pObject = pNextObject;
}
assert(m_listObjects.IsEmpty());
// Notify parent if requested
if (bPropagateUp) {
m_pCounterTree->RemoveMachine(this);
}
}
HRESULT
CObjectNode::GetCounter (
IN LPWSTR pszName,
OUT PCCounterNode *ppCounterRet
)
{
PCCounterNode pCounter;
PCCounterNode pCounterNew;
if (m_listCounters.FindByName(pszName, FIELD_OFFSET(CCounterNode, m_szName), (PCNamedNode*)&pCounter)) {
*ppCounterRet = pCounter;
return NOERROR;
}
pCounterNew = new(lstrlen(pszName) * sizeof(WCHAR)) CCounterNode;
if (!pCounterNew)
return E_OUTOFMEMORY;
pCounterNew->m_pObject = this;
StringCchCopy(pCounterNew->m_szName, lstrlen(pszName) + 1, pszName);
m_listCounters.Add(pCounterNew, pCounter);
*ppCounterRet = pCounterNew;
return NOERROR;
}
//
// The minimum of length of instance name is
// 12. 10 characters for index number(4G)
// 1 for '#' and 1 for terminating 0.
//
#define MIN_INSTANCE_NAME_LEN 12
HRESULT
CObjectNode::GetInstance (
IN LPWSTR pszParent,
IN LPWSTR pszInstance,
IN DWORD dwIndex,
IN BOOL bMonitorDuplicateInstances,
OUT PCInstanceNode *ppInstanceRet
)
{
HRESULT hr = NOERROR;
PCInstanceNode pInstance;
PCInstanceNode pInstanceNew;
LPWSTR szInstName = NULL;
LONG lSize = MIN_INSTANCE_NAME_LEN;
LONG lInstanceLen = 0;
LONG lParentLen = 0;
//
// Calculate the length of the buffer to
// hold the instance and parent names.
//
if (pszInstance) {
lSize += lstrlen(pszInstance);
}
if (pszParent) {
lParentLen = lstrlen(pszParent);
}
lSize += lParentLen + 1;
szInstName = new WCHAR [lSize];
if (szInstName == NULL) {
return E_OUTOFMEMORY;
}
//
// Initialize the string
//
szInstName[0] = L'\0';
if (pszInstance) {
//
// If the parent name exists, copy it
//
if (pszParent) {
StringCchCopy(szInstName, lSize, pszParent);
szInstName[lParentLen] = L'/';
szInstName[lParentLen+1] = L'\0';
}
//
// Copy the instance name.
//
StringCchCat(szInstName, lSize, pszInstance);
//
// Append instance index
//
// "#n" is only appended to the stored name if the index is > 0.
//
if ( dwIndex > 0 && bMonitorDuplicateInstances ) {
StringCchPrintf(&szInstName[lstrlen(szInstName)],
lSize - lstrlen(szInstName),
L"#%d",
dwIndex);
}
}
//
// We use do {} while (0) to act like a switch statement
//
do {
if (m_listInstances.FindByName(szInstName,
FIELD_OFFSET(CInstanceNode, m_szName),
(PCNamedNode*)&pInstance)) {
*ppInstanceRet = pInstance;
break;
}
//
// Create a new one if can not find it
//
lInstanceLen = lstrlen(szInstName);
pInstanceNew = new(lInstanceLen * sizeof(WCHAR)) CInstanceNode;
if (!pInstanceNew) {
hr = E_OUTOFMEMORY;
break;
}
pInstanceNew->m_pObject = this;
pInstanceNew->m_nParentLen = lParentLen;
StringCchCopy(pInstanceNew->m_szName, lInstanceLen + 1, szInstName);
m_listInstances.Add(pInstanceNew, pInstance);
*ppInstanceRet = pInstanceNew;
} while (0);
if (szInstName) {
delete [] szInstName;
}
return hr;
}
void
CObjectNode::RemoveInstance (
IN PCInstanceNode pInstance
)
{
// Remove instance from list and delete it
m_listInstances.Remove(pInstance);
if (pInstance->m_pCachedParentName) {
delete [] pInstance->m_pCachedParentName;
}
if (pInstance->m_pCachedInstName) {
delete [] pInstance->m_pCachedInstName;
}
delete pInstance ;
// if that was the last instance, remove ourself
if (m_listInstances.IsEmpty())
m_pMachine->RemoveObject(this);
}
void
CObjectNode::RemoveCounter (
IN PCCounterNode pCounter
)
{
// Remove counter from list and delete it
m_listCounters.Remove(pCounter);
delete pCounter;
// Don't propagate removal up to object.
// It will go away when the last instance is removed.
}
void
CObjectNode::DeleteNode (
BOOL bPropagateUp
)
{
PCInstanceNode pInstance;
PCInstanceNode pNextInstance;
// Delete all instance nodes
pInstance = FirstInstance();
while ( NULL != pInstance ) {
pNextInstance = pInstance->Next();
pInstance->DeleteNode(FALSE);
m_listInstances.Remove(pInstance);
delete pInstance;
pInstance = pNextInstance;
}
// No need to delete counters nodes as they get
// deleted as their last paired instance does
// Notify parent if requested
if (bPropagateUp)
m_pMachine->RemoveObject(this);
}
HRESULT
CInstanceNode::AddItem (
IN PCCounterNode pCounter,
IN PCGraphItem pItemNew
)
{
PCGraphItem pItemPrev = NULL;
PCGraphItem pItem = m_pItems;
INT iStat = 1;
// Check for existing item for specified counter, stopping at insertion point
while ( pItem != NULL && (iStat = lstrcmp(pCounter->Name(), pItem->m_pCounter->Name())) > 0) {
pItemPrev = pItem;
pItem = pItem->m_pNextItem;
}
// if item exists, return duplicate error status
if (iStat == 0) {
return SMON_STATUS_DUPL_COUNTER_PATH;
}
// else insert the new item
else {
if (pItemPrev != NULL) {
pItemNew->m_pNextItem = pItemPrev->m_pNextItem;
pItemPrev->m_pNextItem = pItemNew;
}
else if (m_pItems != NULL) {
pItemNew->m_pNextItem = m_pItems;
m_pItems = pItemNew;
}
else {
m_pItems = pItemNew;
}
}
// Set back links
pItemNew->m_pInstance = this;
pItemNew->m_pCounter = pCounter;
pCounter->AddItem(pItem);
return NOERROR;
}
void
CInstanceNode::RemoveItem (
IN PCGraphItem pitem
)
{
PCGraphItem pitemPrev = NULL;
PCGraphItem pitemTemp = m_pItems;
// Locate item in list
while (pitemTemp != NULL && pitemTemp != pitem) {
pitemPrev = pitemTemp;
pitemTemp = pitemTemp->m_pNextItem;
}
if (pitemTemp == NULL)
return;
// Remove from list
if (pitemPrev)
pitemPrev->m_pNextItem = pitem->m_pNextItem;
else
m_pItems = pitem->m_pNextItem;
// Remove item from Counter set
pitem->Counter()->RemoveItem(pitem);
// Decrement the total item count
pitem->Tree()->m_nItems--;
UpdateAppPerfDwordData (DD_ITEM_COUNT, pitem->Tree()->m_nItems);
// Release the item
pitem->Release();
// if last item under this instance, remove the instance
if (m_pItems == NULL)
m_pObject->RemoveInstance(this);
}
void
CInstanceNode::DeleteNode (
BOOL bPropagateUp
)
{
PCGraphItem pItem;
pItem = m_pItems;
while ( NULL != pItem ) {
m_pItems = pItem->m_pNextItem;
pItem->Delete(FALSE);
pItem->Counter()->RemoveItem(pItem);
pItem->Release();
pItem = m_pItems;
}
if (bPropagateUp)
m_pObject->RemoveInstance(this);
}
LPWSTR
CInstanceNode::GetParentName()
{
if (m_pCachedParentName == NULL) {
m_pCachedParentName = new WCHAR [m_nParentLen + 1];
if (m_pCachedParentName == NULL) {
return L"";
}
if (m_nParentLen) {
StringCchCopy(m_pCachedParentName, m_nParentLen + 1, m_szName);
}
else {
m_pCachedParentName[0] = 0;
}
}
return m_pCachedParentName;
}
LPWSTR
CInstanceNode::GetInstanceName()
{
LPWSTR pszInst = m_nParentLen ? (m_szName + m_nParentLen + 1) : m_szName;
if (m_pCachedInstName == NULL) {
m_pCachedInstName = new WCHAR [lstrlen(pszInst) + 1];
if (m_pCachedInstName == NULL) {
return L"";
}
StringCchCopy(m_pCachedInstName, lstrlen(pszInst) + 1, pszInst);
}
return m_pCachedInstName;
}
void
CCounterNode::DeleteNode (
BOOL bPropagateUp
)
{
PCInstanceNode pInstance, pInstNext;
PCGraphItem pItem, pItemNext;
if (!bPropagateUp)
return;
// We have to delete the counters item via the instances
// because they maintain the linked list of items
pInstance = m_pObject->FirstInstance();
while (pInstance) {
pInstNext = pInstance->Next();
pItem = pInstance->FirstItem();
while (pItem) {
if (pItem->Counter() == this) {
// Delete all UI associated with the item
pItem->Delete(FALSE);
pItemNext = pItem->m_pNextItem;
// Note that Instance->RemoveItem() will
// also remove counters that have no more items
pItem->Instance()->RemoveItem(pItem);
pItem = pItemNext;
}
else {
pItem = pItem->m_pNextItem;
}
}
pInstance = pInstNext;
}
}
/*******************************
CCounterNode::~CCounterNode (
IN PCGraphItem pItem
)
{
PCGraphItem pItemPrev = NULL;
PCGraphItem pItemFind = m_pItems;
// Find item in list
while (pItemFind != NULL && pItemFind != pItem) {
pItemPrev = pItem;
pItem = pItem->m_pNextItem;
}
if (pItemFind != pItem)
return E_FAIL;
// Unlink from counter item list
if (pItemPrev)
pItemPrev->m_pNextItem = pItem->m_pNextItem;
else
m_pItems = pItem->m_pNextItem;
// Unlink from instance
pItem->m_pInstance->RemoveCounter(pItem);
// if no more items, remove self from parnet object
if (m_pItems == NULL) {
m_pObject->RemoveCounter(this);
return NOERROR;
}
*******************************/
/*
void*
CMachineNode::operator new( size_t stBlock, LPWSTR pszName )
{ return malloc(stBlock + lstrlen(pszName) * sizeof(WCHAR)); }
void
CMachineNode::operator delete ( void * pObject, LPWSTR )
{ free(pObject); }
void*
CObjectNode::operator new( size_t stBlock, LPWSTR pszName )
{ return malloc(stBlock + lstrlen(pszName) * sizeof(WCHAR)); }
void
CObjectNode::operator delete ( void * pObject, LPWSTR )
{ free(pObject); }
void*
CInstanceNode::operator new( size_t stBlock, LPWSTR pszName )
{ return malloc(stBlock + lstrlen(pszName) * sizeof(WCHAR)); }
void
CInstanceNode::operator delete ( void * pObject, LPWSTR )
{ free(pObject); }
void*
CCounterNode::operator new( size_t stBlock, LPWSTR pszName )
{ return malloc(stBlock + lstrlen(pszName) * sizeof(WCHAR)); }
void
CCounterNode::operator delete ( void * pObject, LPWSTR )
{ free(pObject); }CMachineNode::operator new( size_t stBlock, INT iLength )
*/
void *
CMachineNode::operator new( size_t stBlock, UINT iLength )
{
return malloc(stBlock + iLength);
}
void
CMachineNode::operator delete ( void * pObject, UINT )
{
free(pObject);
}
void*
CObjectNode::operator new( size_t stBlock, UINT iLength )
{
return malloc(stBlock + iLength);
}
void
CObjectNode::operator delete ( void * pObject, UINT )
{
free(pObject);
}
void*
CInstanceNode::operator new( size_t stBlock, UINT iLength )
{
return malloc(stBlock + iLength);
}
void
CInstanceNode::operator delete ( void * pObject, UINT )
{
free(pObject);
}
void*
CCounterNode::operator new( size_t stBlock, UINT iLength )
{
return malloc(stBlock + iLength);
}
void
CCounterNode::operator delete ( void * pObject, UINT )
{
free(pObject);
}
#if _MSC_VER >= 1300
void
CMachineNode::operator delete ( void * pObject )
{ free(pObject); }
void
CObjectNode::operator delete ( void * pObject )
{ free(pObject); }
void
CInstanceNode::operator delete ( void * pObject )
{
free(pObject);
}
void
CCounterNode::operator delete ( void * pObject )
{ free(pObject); }
#endif