152 lines
3.1 KiB
C++
152 lines
3.1 KiB
C++
/*++
|
|
|
|
Copyright (C) 1998-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
Abstract:
|
|
|
|
History:
|
|
|
|
--*/
|
|
|
|
|
|
///////////////////////////////////////////////////////////////////
|
|
//
|
|
// StressOps.cpp
|
|
//
|
|
///////////////////////////////////////////////////////////////////
|
|
|
|
#include "Refresher.h"
|
|
#include "Agents.h"
|
|
|
|
|
|
CBasicRefreshAgent::CBasicRefreshAgent()
|
|
{
|
|
m_pRef = 0;
|
|
m_dwIterations = 0;
|
|
m_dwPeriod = 0;
|
|
m_hThread = 0;
|
|
}
|
|
|
|
CBasicRefreshAgent::~CBasicRefreshAgent()
|
|
{
|
|
}
|
|
|
|
BOOL CBasicRefreshAgent::Create(CRefresher *pRef, DWORD dwIterations, DWORD dwPeriod, HANDLE hRefreshEvent)
|
|
{
|
|
m_hRefreshEvent = hRefreshEvent;
|
|
|
|
m_pRef = pRef;
|
|
m_dwPeriod = dwPeriod;
|
|
m_dwIterations = dwIterations;
|
|
m_bInfinite = (0 == dwIterations);
|
|
|
|
if (m_dwIterations)
|
|
{
|
|
DWORD dwTID;
|
|
m_hThread = CreateThread(NULL, 0, CBasicRefreshAgent::StressThreadEntry,
|
|
(LPVOID)this, CREATE_SUSPENDED,
|
|
&dwTID);
|
|
InterlockedIncrement(&g_lRefThreadCount);
|
|
}
|
|
return (NULL != m_hThread);
|
|
}
|
|
|
|
void CBasicRefreshAgent::BeginStress()
|
|
{
|
|
if (m_hThread)
|
|
ResumeThread(m_hThread);
|
|
}
|
|
|
|
void CBasicRefreshAgent::StressLoop()
|
|
{
|
|
DWORD dw = 0;
|
|
while ((dw < m_dwIterations) || m_bInfinite)
|
|
{
|
|
if (!m_bInfinite)
|
|
dw++;
|
|
|
|
if (!m_pRef->Refresh())
|
|
break;
|
|
|
|
if (m_dwPeriod)
|
|
Sleep(m_dwPeriod);
|
|
}
|
|
}
|
|
|
|
DWORD WINAPI CBasicRefreshAgent::StressThreadEntry(LPVOID lpParameter)
|
|
{
|
|
CBasicRefreshAgent *pStress = (CBasicRefreshAgent*)lpParameter;
|
|
|
|
pStress->StressLoop();
|
|
|
|
if (InterlockedDecrement(&g_lRefThreadCount) == 0)
|
|
if (pStress->m_hRefreshEvent)
|
|
SetEvent(pStress->m_hRefreshEvent);
|
|
|
|
return 0;
|
|
}
|
|
|
|
BOOL CRandomOpRefreshAgent::Create(CRefresher *pRef, DWORD dwPeriod)
|
|
{
|
|
CBasicRefreshAgent::Create(pRef, 0, dwPeriod, NULL);
|
|
|
|
DWORD dwTID;
|
|
m_hThreadAddRefs = CreateThread(NULL, 0, CRandomOpRefreshAgent::AddRefThreadEntry,
|
|
(LPVOID)this, CREATE_SUSPENDED,
|
|
&dwTID);
|
|
m_hThreadRemoveRefs = CreateThread(NULL, 0, CRandomOpRefreshAgent::RemRefThreadEntry,
|
|
(LPVOID)this, CREATE_SUSPENDED,
|
|
&dwTID);
|
|
return TRUE;
|
|
}
|
|
|
|
void CRandomOpRefreshAgent::BeginStress()
|
|
{
|
|
CBasicRefreshAgent::BeginStress();
|
|
}
|
|
|
|
void CRandomOpRefreshAgent::StressLoop()
|
|
{
|
|
DWORD dw = 0;
|
|
while ((dw < m_dwIterations) || m_bInfinite)
|
|
{
|
|
if (!m_bInfinite)
|
|
dw++;
|
|
|
|
if (!m_pRef->Refresh())
|
|
break;
|
|
|
|
if (m_dwPeriod)
|
|
Sleep(m_dwPeriod);
|
|
}
|
|
}
|
|
|
|
DWORD WINAPI CRandomOpRefreshAgent::AddRefThreadEntry(LPVOID lpParameter)
|
|
{
|
|
CRandomOpRefreshAgent *pAgent = (CRandomOpRefreshAgent*)lpParameter;
|
|
|
|
// If number of instances are lower than MAX_INST, then add instance then sleep for random period
|
|
if (pAgent->m_pRef->GetNumObjects() < MAX_INST)
|
|
pAgent->m_pRef->AddObject(L"root\\default", L"Win32_HiPerfCounter.Name=\"Inst_1\"");
|
|
|
|
Sleep(0);
|
|
|
|
return 0;
|
|
}
|
|
|
|
DWORD WINAPI CRandomOpRefreshAgent::RemRefThreadEntry(LPVOID lpParameter)
|
|
{
|
|
CRandomOpRefreshAgent *pAgent = (CRandomOpRefreshAgent*)lpParameter;
|
|
|
|
// If number of instances are greater than MIN_INST, then remove instance then sleep for random period
|
|
int nNumObj = pAgent->m_pRef->GetNumObjects();
|
|
if (nNumObj > MIN_INST)
|
|
pAgent->m_pRef->RemoveObject(GetTickCount() % nNumObj);
|
|
|
|
Sleep(0);
|
|
|
|
return 0;
|
|
}
|