archive
/
windows-nt-4.0
mirror of https://github.com/lianthony/NT4.0
2
0
Fork 0
Code Issues Projects Releases Wiki Activity
Windows NT 4.0 source code leak
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.
2 Commits
1 Branch
0 Tags
184 MiB
 
 
 
 
 
 
Branch: master
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from 'master'
${ noResults }
windows-nt-4.0/private/windows/shell/sysmon/procpage.cpp

3423 lines
87 KiB
Raw Permalink Blame History

//+-------------------------------------------------------------------------
//
// TaskMan - NT TaskManager
// Copyright (C) Microsoft
//
// File: procpage.cpp
//
// History: Nov-16-95 DavePl Created
//
//--------------------------------------------------------------------------
#include "precomp.h"
//
// Project-scope globals
//
DWORD g_cProcesses = 0;
//
// File-scope globals
//
SYSTEM_BASIC_INFORMATION g_BasicInfo;
//
// Table of which resource IDs in the column selection dialog
// correspond to which columns
//
const int g_aDlgColIDs[] =
{
IDC_IMAGENAME,
IDC_PID,
IDC_CPU,
IDC_CPUTIME,
IDC_MEMUSAGE,
IDC_MEMUSAGEDIFF,
IDC_PAGEFAULTS,
IDC_PAGEFAULTSDIFF,
IDC_COMMITCHARGE,
IDC_PAGEDPOOL,
IDC_NONPAGEDPOOL,
IDC_BASEPRIORITY,
IDC_HANDLECOUNT,
IDC_THREADCOUNT
};
//
// Column ID on which to sort in the listview, and for
// compares in general
//
COLUMNID g_iProcSortColumnID = COL_PID;
INT g_iProcSortDirection = 1; // 1 = asc, -1 = desc
//
// Column Default Info
//
struct
{
INT Format;
INT Width;
} ColumnDefaults[NUM_COLUMN] =
{
{ LVCFMT_LEFT, 0x6B }, // COL_IMAGENAME
{ LVCFMT_RIGHT, 50 }, // COL_PID
{ LVCFMT_RIGHT, 35}, // COL_CPU
{ LVCFMT_RIGHT, 70 }, // COL_CPUTIME
{ LVCFMT_RIGHT, 70 }, // COL_MEMUSAGE
{ LVCFMT_RIGHT, 70 }, // COL_MEMUSAGEDIFF
{ LVCFMT_RIGHT, 70 }, // COL_PAGEFAULTS
{ LVCFMT_RIGHT, 70 }, // COL_PAGEFAULTSDIFF
{ LVCFMT_RIGHT, 70 }, // COL_COMMITCHARGE
{ LVCFMT_RIGHT, 70 }, // COL_PAGEDPOOL
{ LVCFMT_RIGHT, 70 }, // COL_NONPAGEDPOOL
{ LVCFMT_RIGHT, 60 }, // COL_BASEPRIORITY
{ LVCFMT_RIGHT, 60 }, // COL_HANDLECOUNT
{ LVCFMT_RIGHT, 60 }, // COL_THREADCOUNT
};
/*++ class CProcInfo
Class Description:
Represents the last known information about a running process
Arguments:
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
class CProcInfo
{
public:
LARGE_INTEGER m_uPassCount;
HANDLE m_UniqueProcessId;
BYTE m_CPU;
BYTE m_DisplayCPU;
LARGE_INTEGER m_CPUTime;
LARGE_INTEGER m_DisplayCPUTime;
ULONG m_MemUsage;
ULONG m_MemDiff;
ULONG m_PageFaults;
ULONG m_PageFaultsDiff;
ULONG m_CommitCharge;
ULONG m_PagedPool;
ULONG m_NonPagedPool;
KPRIORITY m_PriClass;
ULONG m_HandleCount;
ULONG m_ThreadCount;
LPTSTR m_pszImageName;
CProcInfo * m_pWowParentProcInfo; // non-NULL for WOW tasks
WORD m_htaskWow; // non-zero for WOW tasks
BOOL m_fWowProcess:1; // TRUE for real WOW process
BOOL m_fWowProcessTested:1; // TRUE once fWowProcess is valid
//
// This is a union of who (which column) is dirty. You can look at
// or set any particular column's bit, or just inspect m_fDirty
// to see if anyone at all is dirty. Used to optimize listview
// painting
//
union
{
DWORD m_fDirty;
struct
{
DWORD m_fDirty_COL_CPU :1;
DWORD m_fDirty_COL_CPUTIME :1;
DWORD m_fDirty_COL_MEMUSAGE :1;
DWORD m_fDirty_COL_MEMUSAGEDIFF :1;
DWORD m_fDirty_COL_PAGEFAULTS :1;
DWORD m_fDirty_COL_PAGEFAULTSDIFF :1;
DWORD m_fDirty_COL_COMMITCHARGE :1;
DWORD m_fDirty_COL_PAGEDPOOL :1;
DWORD m_fDirty_COL_NONPAGEDPOOL :1;
DWORD m_fDirty_COL_BASEPRIORITY :1;
DWORD m_fDirty_COL_HANDLECOUNT :1;
DWORD m_fDirty_COL_IMAGENAME :1;
DWORD m_fDirty_COL_PID :1;
DWORD m_fDirty_COL_THREADCOUNT :1;
};
};
HRESULT SetData(LARGE_INTEGER TotalTime,
PSYSTEM_PROCESS_INFORMATION pInfo,
LARGE_INTEGER uPassCount,
CProcPage * pProcPage,
BOOL fUpdateOnly);
HRESULT SetDataWowTask(LARGE_INTEGER TotalTime,
DWORD dwThreadId,
CHAR * pszFilePath,
LARGE_INTEGER uPassCount,
CProcInfo * pParentProcInfo,
LARGE_INTEGER *pTimeLeft,
WORD htask,
BOOL fUpdateOnly);
CProcInfo()
{
ZeroMemory(this, sizeof(*this));
}
~CProcInfo()
{
if (m_pszImageName)
{
delete [] m_pszImageName;
}
}
// Invalidate() marks this proc with a bogus pid so that it is removed
// on the next cleanup pass
void Invalidate()
{
m_UniqueProcessId = INVALID_HANDLE_VALUE;
}
LONGLONG GetCPUTime() const
{
return m_CPUTime.QuadPart;
}
INT Compare(CProcInfo * pOther);
//
// Is this a WOW task psuedo-process?
//
BOOL IsWowTask(void) const
{
return (BOOL) m_pWowParentProcInfo;
}
//
// Get the Win32 PID for this task
//
DWORD GetRealPID(void) const
{
return m_pWowParentProcInfo
? (DWORD) m_pWowParentProcInfo->m_UniqueProcessId
: (DWORD) m_UniqueProcessId;
}
void SetCPU(LARGE_INTEGER CPUTimeDelta,
LARGE_INTEGER TotalTime,
BOOL fDisplayOnly);
};
/*++ ColSelectDlgProc
Function Description:
Dialog Procedure for the column selection dialog
Arguments:
Standard wndproc stuff
Revision History:
Jan-05-96 Davepl Created
--*/
BOOL CALLBACK ColSelectDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
static CProcPage * pPage = NULL;
switch(uMsg)
{
case WM_INITDIALOG:
{
// Looks scary, but we're single threaded
pPage = (CProcPage *) lParam;
// Start with none of the boxes checked
for (int i = 0; i < NUM_COLUMN; i++)
{
CheckDlgButton(hwndDlg, g_aDlgColIDs[i], BST_UNCHECKED);
}
// Then turn on the ones for the columns we have active
for (i = 0; i < NUM_COLUMN + 1; i++)
{
if (g_Options.m_ActiveProcCol[i] == -1)
{
break;
}
CheckDlgButton(hwndDlg, g_aDlgColIDs[g_Options.m_ActiveProcCol[i]], BST_CHECKED);
}
return TRUE;
}
case WM_COMMAND:
{
// If user clicked OK, add the columns to the array and reset the listview
if (LOWORD(wParam) == IDOK)
{
// First, make sure the column width array is up to date
pPage->SaveColumnWidths();
INT iCol = 1;
g_Options.m_ActiveProcCol[0] = COL_IMAGENAME;
for (int i = 1; i < NUM_COLUMN && g_aDlgColIDs[i] >= 0; i++)
{
if (BST_CHECKED == IsDlgButtonChecked(hwndDlg, g_aDlgColIDs[i]))
{
// It is checked
if (g_Options.m_ActiveProcCol[iCol] != (COLUMNID) i)
{
// If the column wasn't already there, insert its column
// width into the column width array
ShiftArray(g_Options.m_ColumnWidths, iCol, SHIFT_UP);
ShiftArray(g_Options.m_ActiveProcCol, iCol, SHIFT_UP);
g_Options.m_ColumnWidths[iCol] = ColumnDefaults[ i ].Width;
g_Options.m_ActiveProcCol[iCol] = (COLUMNID) i;
}
iCol++;
}
else
{
// Not checked, column not active. If it used to be active,
// remove its column width from the column width array
if (g_Options.m_ActiveProcCol[iCol] == (COLUMNID) i)
{
ShiftArray(g_Options.m_ColumnWidths, iCol, SHIFT_DOWN);
ShiftArray(g_Options.m_ActiveProcCol, iCol, SHIFT_DOWN);
}
}
}
// Terminate the column list
g_Options.m_ActiveProcCol[iCol] = (COLUMNID) -1;
pPage->SetupColumns();
pPage->TimerEvent();
EndDialog(hwndDlg, IDOK);
}
else if (LOWORD(wParam) == IDCANCEL)
{
EndDialog(hwndDlg, IDCANCEL);
}
}
}
return FALSE;
}
/*++ CProcPage::~CProcPage()
- Destructor
*/
CProcPage::~CProcPage()
{
if (m_pProcArray)
{
INT c = m_pProcArray->GetSize();
while (c)
{
delete (CProcInfo *) (m_pProcArray->GetAt(c - 1));
c--;
}
delete m_pProcArray;
}
}
/*++ CProcPage::PickColumns()
Function Description:
Puts up UI that lets the user select what columns to display in the
process page, and then resets the listview with the new column list
Arguments:
none
Return Value:
none
Revision History:
Jan-05-96 Davepl Created
--*/
void CProcPage::PickColumns()
{
DialogBoxParam(g_hInstance,
MAKEINTRESOURCE(IDD_SELECTPROCCOLS),
g_hMainWnd,
ColSelectDlgProc,
(LPARAM) this);
}
/*++ GetPriRanking
Function Description:
Since the priority class defines aren't in order, this helper
exists to make comparisons between pri classes easier. It returns
a larger number for "higher" priority classes
Arguments:
Return Value:
rank of priority (0 to 5)
Revision History:
Nov-27-95 Davepl Created
--*/
DWORD GetPriRanking(DWORD dwClass)
{
switch(dwClass)
{
case REALTIME_PRIORITY_CLASS:
return 3;
case HIGH_PRIORITY_CLASS:
return 2;
case NORMAL_PRIORITY_CLASS:
return 1;
default:
return 0;
}
}
/*++ QuickConfirm
Function Description:
Gets a confirmation for things like terminating/debugging processes
Arguments:
idtitle - string ID of title for message box
idmsg - string ID of message body
Return Value:
IDNO/IDYES, whatever comes back from MessageBox
Revision History:
Nov-28-95 Davepl Created
--*/
UINT CProcPage::QuickConfirm(UINT idTitle, UINT idBody)
{
if (FALSE == g_Options.m_fConfirmations)
{
return IDYES;
}
// Get confirmation before we dust the process, or something similar
TCHAR szTitle[MAX_PATH];
TCHAR szBody[MAX_PATH];
if (0 == LoadString(g_hInstance, idTitle, szTitle, ARRAYSIZE(szTitle)) ||
0 == LoadString(g_hInstance, idBody, szBody, ARRAYSIZE(szBody)))
{
return IDNO;
}
if (IDYES == MessageBox(m_hPage, szBody, szTitle, MB_ICONEXCLAMATION | MB_YESNO))
{
return IDYES;
}
return IDNO;
}
/*++ class CProcPage::SetupColumns
Class Description:
Removes any existing columns from the process listview and
adds all of the columns listed in the g_Options.m_ActiveProcCol array.
Arguments:
Return Value:
HRESULT
Revision History:
Nov-16-95 Davepl Created
--*/
HRESULT CProcPage::SetupColumns()
{
HWND hwndList = GetDlgItem(m_hPage, IDC_PROCLIST);
if (NULL == hwndList)
{
return E_UNEXPECTED;
}
ListView_DeleteAllItems(hwndList);
// Remove all existing columns
LV_COLUMN lvcolumn;
while(ListView_DeleteColumn(hwndList, 0))
{
NULL;
}
// Add all of the new columns
INT iColumn = 0;
while (g_Options.m_ActiveProcCol[iColumn] >= 0)
{
INT idColumn = g_Options.m_ActiveProcCol[iColumn];
TCHAR szTitle[MAX_PATH];
LoadString(g_hInstance, IDS_FIRSTCOL + idColumn, szTitle, ARRAYSIZE(szTitle));
lvcolumn.mask = LVCF_FMT | LVCF_TEXT | LVCF_TEXT | LVCF_WIDTH;
lvcolumn.fmt = ColumnDefaults[ idColumn ].Format;
// If no width preference has been recorded for this column, use the
// default
if (-1 == g_Options.m_ColumnWidths[iColumn])
{
lvcolumn.cx = ColumnDefaults[ idColumn ].Width;
}
else
{
lvcolumn.cx = g_Options.m_ColumnWidths[iColumn];
}
lvcolumn.pszText = szTitle;
lvcolumn.iSubItem = iColumn;
if (-1 == ListView_InsertColumn(hwndList, iColumn, &lvcolumn))
{
return E_FAIL;
}
iColumn++;
}
return S_OK;
}
/*++ class CProcInfo::Compare
Class Description:
Compares this CProcInfo object to another, and returns its ranking
based on the g_iProcSortColumnID field.
Note that if the objects are equal based on the current sort column,
the PID is used as a secondary sort key to prevent items from
jumping around in the listview
WOW psuedo-processes always sort directly after their parent
ntvdm.exe process. So really the sort order is:
1. WOW task psuedo-processes under parent in alpha order
2. User's selected order.
3. PID
Arguments:
pOther - the CProcInfo object to compare this to
Return Value:
< 0 - This CProcInfo is "less" than the other
0 - Equal (Can't happen, since PID is used to sort)
> 0 - This CProcInfo is "greater" than the other
Revision History:
Nov-20-95 Davepl Created
--*/
INT CProcInfo::Compare(CProcInfo * pOther)
{
CProcInfo * pMyThis;
CProcInfo * pMyOther;
INT iRet = 0;
//
// Wow psuedo-processes don't have any performance information,
// so use the parent "real" ntvdm.exe CProcInfo for sorting.
//
ASSERT(this != pOther);
pMyThis = this->IsWowTask()
? this->m_pWowParentProcInfo
: this;
pMyOther = pOther->IsWowTask()
? pOther->m_pWowParentProcInfo
: pOther;
if (pMyThis == pMyOther) {
//
// This implies one or the other or both this and pOther
// are WOW tasks, and they're in the same WOW VDM. Sort
// the "real" process entry first, followed by its associated
// WOW task entries alphabetical.
//
if (this->IsWowTask()) {
if (pOther->IsWowTask()) {
//
// They are siblings and we sort by
// image name.
//
ASSERT(this->m_pWowParentProcInfo ==
pOther->m_pWowParentProcInfo);
iRet = lstrcmpi(this->m_pszImageName, pOther->m_pszImageName);
} else {
//
// pOther is not a Wow task, it must be ntvdm.exe
// the parent of this. this sorts after pOther.
//
ASSERT(pOther == this->m_pWowParentProcInfo);
iRet = 1;
}
} else {
//
// this is not a Wow task, pOther must be and
// this must be pOther's parent.
//
ASSERT(pOther->IsWowTask());
iRet = -1;
}
}
if (0 == iRet) {
switch (g_iProcSortColumnID)
{
case COL_CPU:
iRet = (pMyThis->m_CPU - pMyOther->m_CPU);
break;
case COL_CPUTIME:
iRet = (INT) ((pMyThis->m_CPUTime.QuadPart - pMyOther->m_CPUTime.QuadPart) / 100000);
break;
case COL_MEMUSAGE:
iRet = (pMyThis->m_MemUsage - pMyOther->m_MemUsage);
break;
case COL_MEMUSAGEDIFF:
iRet = (pMyThis->m_MemDiff - pMyOther->m_MemDiff);
break;
case COL_PAGEFAULTS:
iRet = (pMyThis->m_PageFaults - pMyOther->m_PageFaults);
break;
case COL_PAGEFAULTSDIFF:
iRet = (pMyThis->m_PageFaultsDiff - pMyOther->m_PageFaultsDiff);
break;
case COL_COMMITCHARGE:
iRet = (pMyThis->m_CommitCharge - pMyOther->m_CommitCharge);
break;
case COL_PAGEDPOOL:
iRet = (pMyThis->m_PagedPool - pMyOther->m_PagedPool);
break;
case COL_NONPAGEDPOOL:
iRet = (pMyThis->m_NonPagedPool - pMyOther->m_NonPagedPool);
break;
case COL_BASEPRIORITY:
iRet = (GetPriRanking(pMyThis->m_PriClass) - GetPriRanking(pMyOther->m_PriClass));
break;
case COL_HANDLECOUNT:
iRet = (pMyThis->m_HandleCount - pMyOther->m_HandleCount);
break;
case COL_THREADCOUNT:
iRet = (pMyThis->m_ThreadCount - pMyOther->m_ThreadCount);
break;
case COL_PID:
iRet = (INT)((DWORD)pMyThis->m_UniqueProcessId - (DWORD)pMyOther->m_UniqueProcessId);
break;
case COL_IMAGENAME:
iRet = lstrcmpi(pMyThis->m_pszImageName, pMyOther->m_pszImageName);
break;
default:
#ifdef DEBUG
DebugBreak();
#endif
iRet = 0;
}
iRet *= g_iProcSortDirection;
}
// If objects look equal, compare on PID as secondary sort column
// so that items don't jump around in the listview
if (0 == iRet)
{
iRet = (INT) ((DWORD)pMyThis->m_UniqueProcessId -
(DWORD)pMyOther->m_UniqueProcessId) *
g_iProcSortDirection;
}
return iRet;
}
/*++ class CProcInfo::SetCPU
Method Description:
Sets the CPU percentage.
Arguments:
CPUTime - Time for this process
TotalTime - Total elapsed time, used as the denominator in calculations
Return Value:
Revision History:
19-Feb-96 DaveHart Created
--*/
void CProcInfo::SetCPU(LARGE_INTEGER CPUTimeDelta,
LARGE_INTEGER TotalTime,
BOOL fDisplayOnly)
{
// Calc CPU time based on this process's ratio of the total process time used
INT cpu = (BYTE) (((CPUTimeDelta.QuadPart / ((TotalTime.QuadPart / 1000) ?
(TotalTime.QuadPart / 1000) : 1)) + 5)
/ 10);
if (100 == cpu)
{
cpu = 99;
}
ASSERT( cpu <= 100);
if (m_DisplayCPU != cpu)
{
m_fDirty_COL_CPU = TRUE;
m_DisplayCPU = (BYTE) cpu;
if ( ! fDisplayOnly )
{
m_CPU = (BYTE) cpu;
}
}
}
/*++ CProcPage::GetProcessInfo
Class Description:
Reads the process info table into a virtual alloc'd buffer, resizing
the buffer if needed
Arguments:
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
static const int PROCBUF_GROWSIZE = 4096;
HRESULT CProcPage::GetProcessInfo()
{
HRESULT hr = S_OK;
NTSTATUS status;
while(hr == S_OK)
{
if (m_pvBuffer)
{
status = NtQuerySystemInformation(SystemProcessInformation,
m_pvBuffer,
m_cbBuffer,
NULL);
//
// If we succeeded, great, get outta here. If not, any error other
// than "buffer too small" is fatal, in which case we bail
//
if (NT_SUCCESS(status))
{
break;
}
if (status != STATUS_INFO_LENGTH_MISMATCH)
{
hr = E_FAIL;
break;
}
}
//
// Buffer wasn't large enough to hold the process info table, so resize it
// to be larger, then retry.
//
if (m_pvBuffer)
{
VirtualFree(m_pvBuffer, 0, MEM_RELEASE);
m_pvBuffer = NULL;
}
m_cbBuffer += PROCBUF_GROWSIZE;
m_pvBuffer = VirtualAlloc (NULL,
m_cbBuffer,
MEM_COMMIT,
PAGE_READWRITE);
if (m_pvBuffer == NULL)
{
hr = E_OUTOFMEMORY;
break;
}
}
return hr;
}
/*++ FindProcInArrayByPID
Class Description:
Walks the ptrarray given and looks for the CProcInfo object
that has the PID supplied. If not found, returns NULL
Arguments:
pArray - The CPtrArray where the CProcInfos could live
pid - The pid to search for
Return Value:
CProcInfo * in the array, if found, or NULL if not
Revision History:
Nov-20-95 Davepl Created
--*/
// REVIEW (DavePl) could provide a static search hint here so
// that it doesn't always need to start back at zero, or could
// do a binary search
CProcInfo * FindProcInArrayByPID(CPtrArray * pArray, HANDLE pid)
{
for (int i = 0; i < pArray->GetSize(); i++)
{
CProcInfo * pTmp = (CProcInfo *) (pArray->GetAt(i));
if (pTmp->m_UniqueProcessId == pid)
{
// Found it
return pTmp;
}
}
// Not found
return NULL;
}
/*++ InsertIntoSortedArray
Class Description:
Sticks a CProcInfo ptr into the ptrarray supplied at the
appropriate location based on the current sort column (which
is used by the Compare member function)
Arguments:
pArray - The CPtrArray to add to
pProc - The CProcInfo object to add to the array
Return Value:
TRUE if successful, FALSE if fails
Revision History:
Nov-20-95 Davepl Created
--*/
// REVIEW (davepl) Use binary insert here, not linear
BOOL InsertIntoSortedArray(CPtrArray * pArray, CProcInfo * pProc)
{
INT cItems = pArray->GetSize();
for (INT iIndex = 0; iIndex < cItems; iIndex++)
{
CProcInfo * pTmp = (CProcInfo *) pArray->GetAt(iIndex);
if (pProc->Compare(pTmp) < 0)
{
return pArray->InsertAt(iIndex, pProc);
}
}
return pArray->Add(pProc);
}
/*++ ResortArray
Function Description:
Creates a new ptr array sorted in the current sort order based
on the old array, and then replaces the old with the new
Arguments:
ppArray - The CPtrArray to resort
Return Value:
TRUE if successful, FALSE if fails
Revision History:
Nov-21-95 Davepl Created
--*/
BOOL ResortArray(CPtrArray ** ppArray)
{
// Create a new array which will be sorted in the new
// order and used to replace the existing array
CPtrArray * pNew = new CPtrArray(GetProcessHeap());
if (NULL == pNew)
{
return FALSE;
}
// Insert each of the existing items in the old array into
// the new array in the correct spot
INT cItems = (*ppArray)->GetSize();
for (int i = 0; i < cItems; i++)
{
CProcInfo * pItem = (CProcInfo *) (*ppArray)->GetAt(i);
if (FALSE == InsertIntoSortedArray(pNew, pItem))
{
delete pNew;
return FALSE;
}
}
// Kill off the old array, replace it with the new
delete (*ppArray);
(*ppArray) = pNew;
return TRUE;
}
typedef struct
{
LARGE_INTEGER uPassCount;
CProcPage * pProcPage;
CProcInfo * pParentProcInfo;
LARGE_INTEGER TotalTime;
LARGE_INTEGER TimeLeft;
} WOWTASKCALLBACKPARMS, *PWOWTASKCALLBACKPARMS;
BOOL WINAPI WowTaskCallback(
DWORD dwThreadId,
WORD hMod16,
WORD hTask16,
CHAR *pszModName,
CHAR *pszFileName,
LPARAM lparam
)
{
PWOWTASKCALLBACKPARMS pParms = (PWOWTASKCALLBACKPARMS)lparam;
HRESULT hr;
//
// See if this task is already in the list.
//
CProcInfo * pOldProcInfo;
pOldProcInfo = FindProcInArrayByPID(
pParms->pProcPage->m_pProcArray,
(HANDLE) dwThreadId);
if (NULL == pOldProcInfo)
{
//
// We don't already have this process in our array, so create a new one
// and add it to the array
//
CProcInfo * pNewProcInfo = new CProcInfo;
if (NULL == pNewProcInfo)
{
goto done;
}
hr = pNewProcInfo->SetDataWowTask(pParms->TotalTime,
dwThreadId,
pszFileName,
pParms->uPassCount,
pParms->pParentProcInfo,
&pParms->TimeLeft,
hTask16,
FALSE);
if (FAILED(hr) ||
FALSE == pParms->pProcPage->m_pProcArray->Add(pNewProcInfo))
{
delete pNewProcInfo;
goto done;
}
}
else
{
//
// This process already existed in our array, so update its info
//
pOldProcInfo->SetDataWowTask(pParms->TotalTime,
dwThreadId,
pszFileName,
pParms->uPassCount,
pParms->pParentProcInfo,
&pParms->TimeLeft,
hTask16,
TRUE);
}
done:
return FALSE; // continue enumeration
}
/*++ class CProcInfo::SetDataWowTask
Method Description:
Sets up a single CProcInfo object based on the parameters.
This is a WOW task pseudo-process entry.
Arguments:
dwThreadId
pszFilePath Fully-qualified path from VDMEnumTaskWOWEx.
Return Value:
Revision History:
18-Feb-96 DaveHart created
--*/
HRESULT CProcInfo::SetDataWowTask(LARGE_INTEGER TotalTime,
DWORD dwThreadId,
CHAR * pszFilePath,
LARGE_INTEGER uPassCount,
CProcInfo * pParentProcInfo,
LARGE_INTEGER *pTimeLeft,
WORD htask,
BOOL fUpdateOnly)
{
CHAR *pchExe;
//
// Touch this CProcInfo to indicate the process is still alive
//
m_uPassCount.QuadPart = uPassCount.QuadPart;
//
// Update the thread's execution times.
//
HANDLE hThread;
NTSTATUS Status;
OBJECT_ATTRIBUTES obja;
CLIENT_ID cid;
InitializeObjectAttributes(
&obja,
NULL,
0,
NULL,
0 );
cid.UniqueProcess = 0; // 0 means any process
cid.UniqueThread = (HANDLE) dwThreadId;
Status = NtOpenThread(
&hThread,
THREAD_QUERY_INFORMATION,
&obja,
&cid );
if ( NT_SUCCESS(Status) )
{
ULONGLONG ullCreation, ullExit, ullKernel, ullUser;
LARGE_INTEGER TimeDelta, Time;
if (GetThreadTimes(
hThread,
(LPFILETIME) &ullCreation,
(LPFILETIME) &ullExit,
(LPFILETIME) &ullKernel,
(LPFILETIME) &ullUser
) )
{
Time.QuadPart = (LONGLONG)(ullUser + ullKernel);
TimeDelta.QuadPart = Time.QuadPart - m_CPUTime.QuadPart;
if (TimeDelta.QuadPart < 0)
{
ASSERT(0 && "WOW tasks's cpu total usage went DOWN since last refresh.");
Invalidate();
return E_FAIL;
}
if (TimeDelta.QuadPart)
{
m_fDirty_COL_CPUTIME = TRUE;
m_CPUTime.QuadPart = Time.QuadPart;
}
//
// Don't allow sum of WOW child task times to
// exceed ntvdm.exe total. We call GetThreadTimes
// substantially after we get process times, so
// this can happen.
//
if (TimeDelta.QuadPart > pTimeLeft->QuadPart)
{
TimeDelta.QuadPart = pTimeLeft->QuadPart;
pTimeLeft->QuadPart = 0;
}
else
{
pTimeLeft->QuadPart -= TimeDelta.QuadPart;
}
SetCPU( TimeDelta, TotalTime, FALSE );
//
// When WOW tasks are being displayed, the line for ntvdm.exe
// should show times only for overhead or historic threads,
// not including any active task threads.
//
if (pParentProcInfo->m_DisplayCPUTime.QuadPart > m_CPUTime.QuadPart)
{
pParentProcInfo->m_DisplayCPUTime.QuadPart -= m_CPUTime.QuadPart;
}
else
{
pParentProcInfo->m_DisplayCPUTime.QuadPart = 0;
}
m_DisplayCPUTime.QuadPart = m_CPUTime.QuadPart;
}
NtClose(hThread);
}
if (m_PriClass != pParentProcInfo->m_PriClass) {
m_fDirty_COL_BASEPRIORITY = TRUE;
m_PriClass = pParentProcInfo->m_PriClass;
}
if (FALSE == fUpdateOnly)
{
UINT uLen;
//
// Set the task's image name, thread ID, thread count,
// htask, and parent CProcInfo which do not change over
// time.
//
m_htaskWow = htask;
m_fDirty_COL_PID = TRUE;
m_fDirty_COL_IMAGENAME = TRUE;
m_fDirty_COL_THREADCOUNT = TRUE;
m_UniqueProcessId = (HANDLE) dwThreadId;
m_ThreadCount = 1;
//
// We're only interested in the filename of the EXE
// with the path stripped.
//
pchExe = strrchr(pszFilePath, '\\');
if (NULL == pchExe) {
pchExe = pszFilePath;
}
else
{
// skip backslash
pchExe++;
}
uLen = strlen(pchExe);
//
// Indent the EXE name by two spaces
// so WOW tasks look subordinate to
// their ntvdm.exe
//
m_pszImageName = new TCHAR[uLen + 3];
if (NULL == m_pszImageName)
{
return E_OUTOFMEMORY;
}
m_pszImageName[0] = m_pszImageName[1] = TEXT(' ');
MultiByteToWideChar(
CP_ACP,
0,
pchExe,
uLen,
&m_pszImageName[2],
uLen
);
m_pszImageName[uLen + 2] = 0;
//
// WOW EXE filenames are always uppercase, so lowercase it.
//
CharLowerBuff(&m_pszImageName[2], uLen);
m_pWowParentProcInfo = pParentProcInfo;
}
return S_OK;
}
/*++ class CProcInfo::SetData
Class Description:
Sets up a single CProcInfo object based on the data contained in a
SYSTEM_PROCESS_INFORMATION block.
If fUpdate is set, the imagename and icon fields are not processed,
since they do not change throughout the lifetime of the process
Arguments:
TotalTime - Total elapsed time, used as the denominator in calculations
for the process' CPU usage, etc
pInfo - The SYSTEM_PROCESS_INFORMATION block for this process
uPassCount- Current passcount, used to timestamp the last update of
this objectg
fUpdate - See synopsis
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
HRESULT CProcInfo::SetData(LARGE_INTEGER TotalTime,
PSYSTEM_PROCESS_INFORMATION pInfo,
LARGE_INTEGER uPassCount,
CProcPage * pProcPage,
BOOL fUpdateOnly)
{
HRESULT hr = S_OK;
// Touch this CProcInfo to indicate the process is still alive
m_uPassCount.QuadPart = uPassCount.QuadPart;
// Calc this process's total time as the sum of its user and kernel time
LARGE_INTEGER TimeDelta;
LARGE_INTEGER Time;
if (pInfo->UserTime.QuadPart + pInfo->KernelTime.QuadPart < m_CPUTime.QuadPart)
{
ASSERT(0 && "Proc's cpu total usage went DOWN since last refresh.");
Invalidate();
return hr = E_FAIL;
}
Time.QuadPart = pInfo->UserTime.QuadPart +
pInfo->KernelTime.QuadPart;
TimeDelta.QuadPart = Time.QuadPart - m_CPUTime.QuadPart;
if (TimeDelta.QuadPart)
{
m_CPUTime.QuadPart = m_DisplayCPUTime.QuadPart = Time.QuadPart;
m_fDirty_COL_CPUTIME = TRUE;
}
SetCPU( TimeDelta, TotalTime, FALSE );
//
// For each of the fields, we check to see if anything has changed, and if
// so, we mark that particular column as having changed, and update the value.
// This allows me to opimize which fields of the listview to repaint, since
// repainting an entire listview column causes flicker and looks bad in
// general
//
// Miscellaneous fields
if (m_UniqueProcessId != pInfo->UniqueProcessId)
{
m_fDirty_COL_PID = TRUE;
m_UniqueProcessId = pInfo->UniqueProcessId;
}
if (m_MemDiff != (pInfo->WorkingSetSize / 1024) - m_MemUsage)
{
m_fDirty_COL_MEMUSAGEDIFF = TRUE;
m_MemDiff = (pInfo->WorkingSetSize / 1024) - m_MemUsage;
}
if (m_MemUsage != pInfo->WorkingSetSize / 1024)
{
m_fDirty_COL_MEMUSAGE = TRUE;
m_MemUsage = pInfo->WorkingSetSize / 1024;
}
if (m_PageFaultsDiff != (pInfo->PageFaultCount) - m_PageFaults)
{
m_fDirty_COL_PAGEFAULTSDIFF = TRUE;
m_PageFaultsDiff = (pInfo->PageFaultCount) - m_PageFaults;
}
if (m_PageFaults != (pInfo->PageFaultCount))
{
m_fDirty_COL_PAGEFAULTS = TRUE;
m_PageFaults = (pInfo->PageFaultCount);
}
if (m_CommitCharge != pInfo->PrivatePageCount / 1024)
{
m_fDirty_COL_COMMITCHARGE = TRUE;
m_CommitCharge = pInfo->PrivatePageCount / 1024;
}
if (m_PagedPool != pInfo->QuotaPagedPoolUsage / 1024)
{
m_fDirty_COL_PAGEDPOOL = TRUE;
m_PagedPool = pInfo->QuotaPagedPoolUsage / 1024;
}
if (m_NonPagedPool != pInfo->QuotaNonPagedPoolUsage / 1024)
{
m_fDirty_COL_NONPAGEDPOOL = TRUE;
m_NonPagedPool = pInfo->QuotaNonPagedPoolUsage / 1024;
}
if (m_PriClass != pInfo->BasePriority)
{
m_fDirty_COL_BASEPRIORITY = TRUE;
m_PriClass = pInfo->BasePriority;
}
if (m_HandleCount != pInfo->HandleCount)
{
m_fDirty_COL_HANDLECOUNT = TRUE;
m_HandleCount = pInfo->HandleCount;
}
if (m_ThreadCount != pInfo->NumberOfThreads)
{
m_fDirty_COL_HANDLECOUNT = TRUE;
m_ThreadCount = pInfo->NumberOfThreads;
}
if (FALSE == fUpdateOnly)
{
//
// Set the process' image name. If its NULL it could be the "Idle Process" or simply
// a process whose image name is unknown. In both cases we load a string resource
// with an appropriate replacement name.
//
m_fDirty_COL_IMAGENAME = TRUE;
if (pInfo->ImageName.Buffer == NULL)
{
// No image name, so replace it with "Unknown"
TCHAR szTmp[MAX_PATH];
UINT uLen = LoadString(g_hInstance, IDS_SYSPROC, szTmp, MAX_PATH);
m_pszImageName = new TCHAR[uLen + 1];
if (NULL == m_pszImageName)
{
return hr = E_OUTOFMEMORY;
}
lstrcpy(m_pszImageName, szTmp);
}
else
{
//
// We have a valid image name, so allocate enough space and then
// make a copy of it
//
m_pszImageName = new TCHAR[pInfo->ImageName.Length + 1];
if (NULL == m_pszImageName)
{
return hr = E_OUTOFMEMORY;
}
lstrcpy(m_pszImageName, pInfo->ImageName.Buffer);
}
}
//
// Check if this process is a WOW process. There is some latency
// between the time a WOW process is created and the time
// the shared memory used by VDMEnumTaskWOWEx reflects the new
// process and tasks. However, once a process becomes a WOW
// process, it is always a WOW process until it dies.
//
if (g_Options.m_fShow16Bit)
{
if ( m_fWowProcess ||
! m_fWowProcessTested)
{
if ( ! _wcsicmp(m_pszImageName, TEXT("ntvdm.exe")))
{
WOWTASKCALLBACKPARMS WowTaskCallbackParms;
WowTaskCallbackParms.uPassCount = uPassCount;
WowTaskCallbackParms.pProcPage = pProcPage;
WowTaskCallbackParms.pParentProcInfo = this;
WowTaskCallbackParms.TotalTime.QuadPart = TotalTime.QuadPart;
WowTaskCallbackParms.TimeLeft.QuadPart = TimeDelta.QuadPart;
if (VDMEnumTaskWOWEx((DWORD) m_UniqueProcessId,
WowTaskCallback,
(LPARAM) &WowTaskCallbackParms))
{
if ( ! m_fWowProcess )
{
m_fWowProcessTested =
m_fWowProcess = TRUE;
}
SetCPU( WowTaskCallbackParms.TimeLeft, TotalTime, TRUE );
}
else
{
//
// We avoid calling VDMEnumTaskWOWEx if the process has an
// execution time of more than 10 seconds and has not so
// far been seen as a WOW process.
//
if (GetCPUTime() > (10 * 10 * 1000 * 1000))
{
m_fWowProcessTested = TRUE;
}
}
}
else
{
m_fWowProcessTested = TRUE;
}
}
}
return S_OK;
}
/*++ CProcPage::UpdateProcListview
Class Description:
Walks the listview and checks to see if each line in the
listview matches the corresponding entry in our process
array. Those which differe by PID are replaced, and those
that need updating are updated.
Items are also added and removed to/from the tail of the
listview as required.
Arguments:
Return Value:
HRESULT
Revision History:
Nov-20-95 Davepl Created
--*/
HRESULT CProcPage::UpdateProcListview()
{
HWND hListView = GetDlgItem(m_hPage, IDC_PROCLIST);
// Stop repaints while we party on the listview
SendMessage(hListView, WM_SETREDRAW, FALSE, 0);
INT cListViewItems = ListView_GetItemCount(hListView);
INT cProcArrayItems = m_pProcArray->GetSize();
//
// Walk the existing lines in the listview and replace/update
// them as needed
//
CProcInfo * pSelected = GetSelectedProcess();
for (INT iCurrent = 0;
iCurrent < cListViewItems && iCurrent < cProcArrayItems;
iCurrent++)
{
LV_ITEM lvitem = { 0 };
lvitem.mask = LVIF_PARAM | LVIF_TEXT | LVIF_STATE;
lvitem.iItem = iCurrent;
if (FALSE == ListView_GetItem(hListView, &lvitem))
{
SendMessage(hListView, WM_SETREDRAW, TRUE, 0);
return E_FAIL;
}
CProcInfo * pTmp = (CProcInfo *) lvitem.lParam;
CProcInfo * pProc = (CProcInfo *) m_pProcArray->GetAt(iCurrent);
if (pTmp != pProc)
{
// If the objects aren't the same, we need to replace this line
lvitem.pszText = pProc->m_pszImageName;
lvitem.lParam = (LPARAM) pProc;
if (pProc == pSelected)
{
lvitem.state |= LVIS_SELECTED | LVIS_FOCUSED;
}
else
{
lvitem.state &= ~(LVIS_SELECTED | LVIS_FOCUSED);
}
lvitem.stateMask |= LVIS_SELECTED | LVIS_FOCUSED;
ListView_SetItem(hListView, &lvitem);
ListView_RedrawItems(hListView, iCurrent, iCurrent);
}
else if (pProc->m_fDirty)
{
// Same PID, but item needs updating
ListView_RedrawItems(hListView, iCurrent, iCurrent);
pProc->m_fDirty = 0;
}
}
//
// We've either run out of listview items or run out of proc array
// entries, so remove/add to the listview as appropriate
//
while (iCurrent < cListViewItems)
{
// Extra items in the listview (processes gone away), so remove them
ListView_DeleteItem(hListView, iCurrent);
cListViewItems--;
}
while (iCurrent < cProcArrayItems)
{
// Need to add new items to the listview (new processes appeared)
CProcInfo * pProc = (CProcInfo *)m_pProcArray->GetAt(iCurrent);
LV_ITEM lvitem = { 0 };
lvitem.mask = LVIF_PARAM | LVIF_TEXT;
lvitem.iItem = iCurrent;
lvitem.pszText = pProc->m_pszImageName;
lvitem.lParam = (LPARAM) pProc;
ListView_InsertItem(hListView, &lvitem);
iCurrent++;
}
// Let the listview paint again
SendMessage(hListView, WM_SETREDRAW, TRUE, 0);
return S_OK;
}
/*++ class CProcPage::UpdateProcInfoArray
Class Description:
Retrieves the list of process info blocks from the system,
and runs through our array of CProcInfo items. Items which
already exist are updated, and those that do not are added.
At the end, any process which has not been touched by this
itteration of the function are considered to have completed
and are removed from the array.
Arguments:
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
// See comments near the usage of this table below for info on why it exists
static struct
{
size_t cbOffset;
UINT idString;
}
g_OffsetMap[] =
{
{ FIELD_OFFSET(CSysInfo, m_cHandles), IDC_TOTAL_HANDLES },
{ FIELD_OFFSET(CSysInfo, m_cThreads), IDC_TOTAL_THREADS },
{ FIELD_OFFSET(CSysInfo, m_cProcesses), IDC_TOTAL_PROCESSES },
{ FIELD_OFFSET(CSysInfo, m_dwPhysicalMemory), IDC_TOTAL_PHYSICAL },
{ FIELD_OFFSET(CSysInfo, m_dwPhysAvail), IDC_AVAIL_PHYSICAL },
{ FIELD_OFFSET(CSysInfo, m_dwFileCache), IDC_FILE_CACHE },
{ FIELD_OFFSET(CSysInfo, m_dwCommitTotal), IDC_COMMIT_TOTAL },
{ FIELD_OFFSET(CSysInfo, m_dwCommitLimit), IDC_COMMIT_LIMIT },
{ FIELD_OFFSET(CSysInfo, m_dwCommitPeak), IDC_COMMIT_PEAK },
{ FIELD_OFFSET(CSysInfo, m_dwKernelPaged), IDC_KERNEL_PAGED },
{ FIELD_OFFSET(CSysInfo, m_dwKernelNP), IDC_KERNEL_NONPAGED },
{ FIELD_OFFSET(CSysInfo, m_dwKernelTotal), IDC_KERNEL_TOTAL },
};
HRESULT CProcPage::UpdateProcInfoArray()
{
HRESULT hr;
INT i;
INT iField;
ULONG cbOffset = 0;
CSysInfo SysInfoTemp;
NTSTATUS Status;
PSYSTEM_PROCESS_INFORMATION pCurrent;
SYSTEM_PERFORMANCE_INFORMATION PerfInfo;
SYSTEM_FILECACHE_INFORMATION FileCache;
LARGE_INTEGER TotalTime = {0,0};
LARGE_INTEGER LastTotalTime = {0,0};
//
// Pass-count for this function. It ain't thread-safe, of course, but I
// can't imagine a scenario where we'll have mode than one thread running
// through this (the app is currently single threaded anyway). If we
// overflow LARGE_INTEGER updates, I'll already be long gone, so don't bug me.
//
static LARGE_INTEGER uPassCount = {0,0};
//
// Get some non-process specific info, like memory status
//
SysInfoTemp.m_dwPhysicalMemory = g_BasicInfo.NumberOfPhysicalPages *
(g_BasicInfo.PageSize / 1024);
Status = NtQuerySystemInformation(
SystemPerformanceInformation,
&PerfInfo,
sizeof(PerfInfo),
NULL
);
if (!NT_SUCCESS(Status))
{
return E_FAIL;
}
SysInfoTemp.m_dwPhysAvail = PerfInfo.AvailablePages * (g_BasicInfo.PageSize / 1024);
SysInfoTemp.m_dwCommitTotal = PerfInfo.CommittedPages * (g_BasicInfo.PageSize / 1024);
SysInfoTemp.m_dwCommitLimit = PerfInfo.CommitLimit * (g_BasicInfo.PageSize / 1024);
SysInfoTemp.m_dwCommitPeak = PerfInfo.PeakCommitment * (g_BasicInfo.PageSize / 1024);
SysInfoTemp.m_dwKernelPaged = PerfInfo.PagedPoolPages * (g_BasicInfo.PageSize / 1024);
SysInfoTemp.m_dwKernelNP = PerfInfo.NonPagedPoolPages * (g_BasicInfo.PageSize / 1024);
SysInfoTemp.m_dwKernelTotal = SysInfoTemp.m_dwKernelNP + SysInfoTemp.m_dwKernelPaged;
g_MEMMax = SysInfoTemp.m_dwCommitLimit;
Status = NtQuerySystemInformation(
SystemFileCacheInformation,
&FileCache,
sizeof(FileCache),
NULL
);
if (!NT_SUCCESS(Status))
{
return E_FAIL;
}
SysInfoTemp.m_dwFileCache = FileCache.CurrentSize / 1024;
//
// Read the process info structures into the flat buffer
//
hr = GetProcessInfo();
if (FAILED(hr))
{
goto done;
}
//
// First walk all of the process info blocks and sum their times, so that we can
// calculate a CPU usage ratio (%) for each individual process
//
cbOffset = 0;
do
{
CProcInfo * pOldProcInfo;
pCurrent = (PSYSTEM_PROCESS_INFORMATION)&((LPBYTE)m_pvBuffer)[cbOffset];
ASSERT( FALSE == IsBadReadPtr((LPVOID)pCurrent, sizeof(PSYSTEM_PROCESS_INFORMATION)));
if (pCurrent->UniqueProcessId == NULL && pCurrent->NumberOfThreads == 0)
{
// Zombie process, just skip it
goto next;
}
pOldProcInfo = FindProcInArrayByPID(m_pProcArray, pCurrent->UniqueProcessId);
if (pOldProcInfo)
{
if (pOldProcInfo->GetCPUTime() > pCurrent->KernelTime.QuadPart + pCurrent->UserTime.QuadPart)
{
// If CPU has gone DOWN, its because the PID has been reused, so invalidate this
// CProcInfo such that it is removed and the new one added
pOldProcInfo->Invalidate();
dprintf(TEXT("Invalidating %08x\n"), pOldProcInfo);
goto next;
}
else if (pCurrent->UniqueProcessId == (HANDLE) 0 &&
pCurrent->KernelTime.QuadPart == 0 &&
pCurrent->UserTime.QuadPart == 0)
{
dprintf(TEXT("System idle process has 0 times\n"));
pOldProcInfo->Invalidate();
goto next;
}
else
{
LastTotalTime.QuadPart += pOldProcInfo->GetCPUTime();
}
}
TotalTime.QuadPart += pCurrent->KernelTime.QuadPart + pCurrent->UserTime.QuadPart;
SysInfoTemp.m_cHandles += pCurrent->HandleCount;
SysInfoTemp.m_cThreads += pCurrent->NumberOfThreads;
SysInfoTemp.m_cProcesses++;
next:
cbOffset += pCurrent->NextEntryOffset;
} while (pCurrent->NextEntryOffset);
LARGE_INTEGER TimeDelta;
TimeDelta.QuadPart = TotalTime.QuadPart - LastTotalTime.QuadPart;
ASSERT(TimeDelta.QuadPart >= 0);
// Update the global count (visible to the status bar)
g_cProcesses = SysInfoTemp.m_cProcesses;
TotalTime.QuadPart - LastTotalTime.QuadPart;
//
// We have a number of text fields in the dialog that are based on counts we accumulate
// here. Rather than painting all of the time, we only change the ones whose values have
// really changed. We have a table up above of the offsets into the CSysInfo object
// where these values live (the same offset in the real g_SysInfo object and the temp
// working copy, of course), and what control ID they correspond to. We then loop through
// and compare each real one to the temp working copy, updating as needed. Hard to
// read, but smaller than a dozen if() statements.
//
// BUGBUG Gross and dirty hack
//
// At the last moment we moved all these controls to the performance page. They're
// sort of inextricably tied to the data collected on this page, so as a temp measure,
// I have this (the process page) update the controls on the performance page.
extern CPage * g_pPages[];
if (g_pPages[2])
{
for (iField = 0; iField < ARRAYSIZE(g_OffsetMap); iField++)
{
DWORD * pdwRealCopy = (DWORD *)(((LPBYTE)&m_SysInfo) + g_OffsetMap[iField].cbOffset);
DWORD * pdwTempCopy = (DWORD *)(((LPBYTE)&SysInfoTemp) + g_OffsetMap[iField].cbOffset);
// if (*pdwRealCopy != *pdwTempCopy)
{
*pdwRealCopy = *pdwTempCopy;
TCHAR szText[32];
wsprintf(szText, TEXT("%d"), *pdwRealCopy);
HWND hPage = g_pPages[2]->GetPageWindow();
// Updates can come through before page is created, so verify
// that it exists before we party on its children
if (hPage)
{
SetWindowText(GetDlgItem(hPage, g_OffsetMap[iField].idString), szText);
}
}
}
}
//
// Now walk the process info blocks again and refresh the CProcInfo array for each
// individual process
//
cbOffset = 0;
do
{
//
// Grab a PROCESS_INFORMATION struct from the buffer
//
pCurrent = (PSYSTEM_PROCESS_INFORMATION)&((LPBYTE)m_pvBuffer)[cbOffset];
ASSERT( FALSE == IsBadReadPtr((LPVOID)pCurrent, sizeof(PSYSTEM_PROCESS_INFORMATION)));
if (pCurrent->UniqueProcessId == NULL && pCurrent->NumberOfThreads == 0)
{
// Zombie process, just skip it
goto nextprocinfo;
}
//
// This is really ugly, but... NtQuerySystemInfo has too much latency, and if you
// change a process' priority, you don't see it reflected right away. And, if you
// don't have autoupdate on, you never do. So, we use GetPriorityClass() to get
// the value instead. This means BasePriority is now the pri class, not the pri value.
//
if (pCurrent->UniqueProcessId)
{
HANDLE hProcess;
hProcess = OpenProcess( PROCESS_QUERY_INFORMATION, FALSE, (DWORD) pCurrent->UniqueProcessId );
DWORD dwPriClass;
dwPriClass = 0;
if (hProcess)
{
dwPriClass = GetPriorityClass(hProcess);
if (dwPriClass)
{
pCurrent->BasePriority = dwPriClass;
}
CloseHandle( hProcess );
}
if (NULL == hProcess || dwPriClass == 0)
{
// We're not allowed to open this process, so convert what NtQuerySystemInfo
// gave us into a priority class... its the next bexthing
if (pCurrent->BasePriority <= 4)
{
pCurrent->BasePriority = IDLE_PRIORITY_CLASS;
}
else if (pCurrent->BasePriority <= 9)
{
pCurrent->BasePriority = NORMAL_PRIORITY_CLASS;
}
else if (pCurrent->BasePriority <= 13)
{
pCurrent->BasePriority = HIGH_PRIORITY_CLASS;
}
else
{
pCurrent->BasePriority = REALTIME_PRIORITY_CLASS;
}
}
}
//
// Try to find an existing CProcInfo instance which corresponds to this process
//
CProcInfo * pProcInfo;
pProcInfo = FindProcInArrayByPID(m_pProcArray, pCurrent->UniqueProcessId);
if (NULL == pProcInfo)
{
//
// We don't already have this process in our array, so create a new one
// and add it to the array
//
pProcInfo = new CProcInfo;
if (NULL == pProcInfo)
{
hr = E_OUTOFMEMORY;
goto done;
}
hr = pProcInfo->SetData(TimeDelta,
pCurrent,
uPassCount,
this,
FALSE);
if (FAILED(hr) || FALSE == m_pProcArray->Add(pProcInfo))
{
delete pProcInfo;
goto done;
}
}
else
{
//
// This process already existed in our array, so update its info
//
hr = pProcInfo->SetData(TimeDelta,
pCurrent,
uPassCount,
this,
TRUE);
if (FAILED(hr))
{
goto done;
}
}
nextprocinfo:
cbOffset += pCurrent->NextEntryOffset;
} while (pCurrent->NextEntryOffset);
//
// Run through the CProcInfo array and remove anyone that hasn't been touched
// by this pass through this function (which indicates the process is no
// longer alive)
//
i = 0;
while (i < m_pProcArray->GetSize())
{
CProcInfo * pProcInfo = (CProcInfo *)(m_pProcArray->GetAt(i));
ASSERT(pProcInfo);
//
// If passcount doesn't match, delete the CProcInfo instance and remove
// its pointer from the array. Note that we _don't_ increment the index
// if we remove an element, since the next element would now live at
// the current index after the deletion
//
if (pProcInfo->m_uPassCount.QuadPart != uPassCount.QuadPart)
{
delete pProcInfo;
m_pProcArray->RemoveAt(i, 1);
}
else
{
i++;
}
}
done:
ResortArray(&m_pProcArray);
uPassCount.QuadPart++;
return hr;
}
/*++ CPerfPage::SizeProcPage
Routine Description:
Sizes its children based on the size of the
tab control on which it appears.
Arguments:
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
void CProcPage::SizeProcPage()
{
// Get the coords of the outer dialog
RECT rcParent;
GetClientRect(m_hPage, &rcParent);
HDWP hdwp = BeginDeferWindowPos(10);
// Calc the deltas in the x and y positions that we need to
// move each of the child controls
RECT rcTerminate;
HWND hwndTerminate = GetDlgItem(m_hPage, IDC_TERMINATE);
GetWindowRect(hwndTerminate, &rcTerminate);
MapWindowPoints(HWND_DESKTOP, m_hPage, (LPPOINT) &rcTerminate, 2);
INT dx = ((rcParent.right - g_DefSpacing * 2) - rcTerminate.right);
INT dy = ((rcParent.bottom - g_DefSpacing * 2) - rcTerminate.bottom);
// Move the EndProcess button
DeferWindowPos(hdwp, hwndTerminate, NULL,
rcTerminate.left + dx,
rcTerminate.top + dy,
0, 0,
SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE);
// Size the listbox
HWND hwndListbox = GetDlgItem(m_hPage, IDC_PROCLIST);
RECT rcListbox;
GetWindowRect(hwndListbox, &rcListbox);
MapWindowPoints(HWND_DESKTOP, m_hPage, (LPPOINT) &rcListbox, 2);
DeferWindowPos(hdwp, hwndListbox, NULL,
0, 0,
rcTerminate.right - rcListbox.left + dx,
rcTerminate.top - rcListbox.top + dy - g_DefSpacing,
SWP_NOMOVE | SWP_NOZORDER | SWP_NOACTIVATE);
EndDeferWindowPos(hdwp);
}
/*++ CProcPage::HandleTaskManNotify
Routine Description:
Processes WM_NOTIFY messages received by the procpage dialog
Arguments:
hWnd - Control that generated the WM_NOTIFY
pnmhdr - Ptr to the NMHDR notification stucture
Return Value:
BOOL "did we handle it" code
Revision History:
Nov-20-95 Davepl Created
--*/
INT CProcPage::HandleProcPageNotify(HWND hWnd, LPNMHDR pnmhdr)
{
switch(pnmhdr->code)
{
case LVN_COLUMNCLICK:
{
// User clicked a header control, so set the sort column. If its the
// same as the current sort column, just invert the sort direction in
// the column. Then resort the task array
const NM_LISTVIEW * pnmv = (const NM_LISTVIEW *) pnmhdr;
if (g_iProcSortColumnID == g_Options.m_ActiveProcCol[pnmv->iSubItem])
{
g_iProcSortDirection *= -1;
}
else
{
g_iProcSortColumnID = g_Options.m_ActiveProcCol[pnmv->iSubItem];
g_iProcSortDirection = -1;
}
ResortArray(&m_pProcArray);
TimerEvent();
break;
}
case LVN_GETDISPINFO:
{
LV_ITEM * plvitem = &(((LV_DISPINFO *) pnmhdr)->item);
// Listview needs a text string
if (plvitem->mask & LVIF_TEXT)
{
COLUMNID columnid = (COLUMNID) g_Options.m_ActiveProcCol[plvitem->iSubItem];
const CProcInfo * pProcInfo = (const CProcInfo *) plvitem->lParam;
//
// Most columns are blank for WOW tasks.
//
if (pProcInfo->IsWowTask() &&
columnid != COL_IMAGENAME &&
columnid != COL_BASEPRIORITY &&
columnid != COL_THREADCOUNT &&
columnid != COL_CPUTIME &&
columnid != COL_CPU) {
plvitem->pszText[0] = 0;
goto done;
}
switch(columnid)
{
case COL_PID:
wsprintf(plvitem->pszText, TEXT("%d"), (ULONG) (pProcInfo->m_UniqueProcessId));
break;
case COL_CPU:
wsprintf(plvitem->pszText, TEXT("%02d %"), pProcInfo->m_DisplayCPU);
break;
case COL_IMAGENAME:
lstrcpyn(plvitem->pszText, pProcInfo->m_pszImageName, plvitem->cchTextMax);
//plvitem->mask |= LVIF_DI_SETITEM;
break;
case COL_CPUTIME:
TIME_FIELDS TimeOut;
// BUGBUG This Rtl should take a const...
RtlTimeToElapsedTimeFields ( (LARGE_INTEGER *)&(pProcInfo->m_DisplayCPUTime), &TimeOut);
TimeOut.Hour += (SHORT)(TimeOut.Day * 24);
wsprintf(plvitem->pszText, TEXT("%2d:%02d:%02d"), TimeOut.Hour, TimeOut.Minute, TimeOut.Second);
break;
case COL_MEMUSAGE:
wsprintf(plvitem->pszText, TEXT("%d %s"), pProcInfo->m_MemUsage, g_szK);
break;
case COL_MEMUSAGEDIFF:
wsprintf(plvitem->pszText, TEXT("%d %s"), pProcInfo->m_MemDiff, g_szK);
break;
case COL_PAGEFAULTS:
wsprintf(plvitem->pszText, TEXT("%d"), pProcInfo->m_PageFaults);
break;
case COL_PAGEFAULTSDIFF:
wsprintf(plvitem->pszText, TEXT("%d"), pProcInfo->m_PageFaultsDiff);
break;
case COL_COMMITCHARGE:
wsprintf(plvitem->pszText, TEXT("%d %s"), pProcInfo->m_CommitCharge, g_szK);
break;
case COL_PAGEDPOOL:
wsprintf(plvitem->pszText, TEXT("%d %s"), pProcInfo->m_PagedPool, g_szK);
break;
case COL_NONPAGEDPOOL:
wsprintf(plvitem->pszText, TEXT("%d %s"), pProcInfo->m_NonPagedPool, g_szK);
break;
case COL_BASEPRIORITY:
{
LPCTSTR pszClass = NULL;
switch(pProcInfo->m_PriClass)
{
case REALTIME_PRIORITY_CLASS:
pszClass = g_szRealtime;
break;
case HIGH_PRIORITY_CLASS:
pszClass = g_szHigh;
break;
case NORMAL_PRIORITY_CLASS:
pszClass = g_szNormal;
break;
case IDLE_PRIORITY_CLASS:
pszClass = g_szLow;
break;
default:
pszClass = g_szUnknown;
break;
}
lstrcpyn(plvitem->pszText, pszClass, plvitem->cchTextMax);
break;
}
case COL_HANDLECOUNT:
wsprintf(plvitem->pszText, TEXT("%d"), pProcInfo->m_HandleCount);
break;
case COL_THREADCOUNT:
wsprintf(plvitem->pszText, TEXT("%d"), pProcInfo->m_ThreadCount);
break;
default:
Assert( 0 && "Unknown listview subitem" );
break;
} // end switch(columnid)
} // end LVIF_TEXT case
} // end LVN_GETDISPINFO case
} // end switch(pnmhdr->code)
done:
return 1;
}
/*++ CProcPage::TimerEvent
Routine Description:
Called by main app when the update time fires
Arguments:
Return Value:
Revision History:
Nov-20-95 Davepl Created
--*/
void CProcPage::TimerEvent()
{
// REVIEW (DavePl)
// We might want to optimize the amount of calculation we do when
// we are iconic, but we still need to track the deltas (mem usage,
// faults, etc) so might as well just calc it all. Listview won't
// repaint anyway until its visible, which is the real work
if (FALSE == m_fPaused)
{
// We only process updates when the display is not paused, ie:
// not during trackpopupmenu loop
UpdateProcInfoArray();
UpdateProcListview();
}
}
/*++ CProcPage::HandleProcListContextMenu
Routine Description:
Handles right-clicks (context menu) in the proc list
Arguments:
xPos, yPos - coords of where the click occurred
Return Value:
Revision History:
Nov-22-95 Davepl Created
--*/
void CProcPage::HandleProcListContextMenu(INT xPos, INT yPos)
{
HWND hTaskList = GetDlgItem(m_hPage, IDC_PROCLIST);
INT iItem = ListView_GetNextItem(hTaskList, -1, LVNI_SELECTED);
if (-1 != iItem)
{
HMENU hPopup = LoadPopupMenu(g_hInstance, IDR_PROC_CONTEXT);
if (hPopup)
{
if (hPopup && SHRestricted(REST_NORUN))
{
DeleteMenu(hPopup, IDM_RUN, MF_BYCOMMAND);
}
CProcInfo * pProc = GetSelectedProcess();
if (NULL == pProc)
{
return;
}
//
// If no debugger is installed or it's a 16-bit app
// ghost the debug menu item
//
if (NULL == m_pszDebugger || pProc->IsWowTask())
{
EnableMenuItem(hPopup, IDM_PROC_DEBUG, MF_GRAYED | MF_DISABLED | MF_BYCOMMAND);
}
//
// If it's a 16-bit task grey the priority choices
//
if (pProc->IsWowTask())
{
EnableMenuItem(hPopup, IDM_PROC_REALTIME, MF_GRAYED | MF_DISABLED | MF_BYCOMMAND);
EnableMenuItem(hPopup, IDM_PROC_NORMAL, MF_GRAYED | MF_DISABLED | MF_BYCOMMAND);
EnableMenuItem(hPopup, IDM_PROC_HIGH, MF_GRAYED | MF_DISABLED | MF_BYCOMMAND);
EnableMenuItem(hPopup, IDM_PROC_LOW, MF_GRAYED | MF_DISABLED | MF_BYCOMMAND);
}
//
// If not an MP machine, remove the affinity option
//
if (1 == g_cProcessors || pProc->IsWowTask())
{
DeleteMenu(hPopup, IDM_AFFINITY, MF_BYCOMMAND);
}
DWORD dwPri = pProc->m_PriClass;
INT idCheck = 0;
// These constants are listed in the SDK
if (dwPri == IDLE_PRIORITY_CLASS)
{
idCheck = IDM_PROC_LOW;
}
else if (dwPri == NORMAL_PRIORITY_CLASS)
{
idCheck = IDM_PROC_NORMAL;
}
else if (dwPri == HIGH_PRIORITY_CLASS)
{
idCheck = IDM_PROC_HIGH;
}
else
{
Assert(dwPri == REALTIME_PRIORITY_CLASS);
idCheck = IDM_PROC_REALTIME;
}
// Check the appropriate radio menu for this process' priority class
CheckMenuRadioItem(hPopup, IDM_PROC_REALTIME, IDM_PROC_LOW, idCheck, MF_BYCOMMAND);
m_fPaused = TRUE;
g_fInPopup = TRUE;
TrackPopupMenuEx(hPopup, 0, xPos, yPos, m_hPage, NULL);
g_fInPopup = FALSE;
m_fPaused = FALSE;
//
// If one of the context menu actions (ie: Kill) requires that the display
// get updated, do it now
//
DestroyMenu(hPopup);
}
}
}
/*++ AffinityDlgProc
Routine Description:
Dialog procedure for the affinity mask dialog. Basically just tracks 32 check
boxes that represent the processors
Arguments:
standard dlgproc fare 0 - initial lParam is pointer to affinity mask
Revision History:
Jan-17-96 Davepl Created
--*/
BOOL CALLBACK AffinityDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
static DWORD * pdwAffinity = NULL; // One of the joys of single threadedness
switch (uMsg)
{
case WM_INITDIALOG:
{
pdwAffinity = (DWORD *) lParam;
for (int i = 0; i < MAX_PROCESSOR; i++)
{
EnableWindow(GetDlgItem(hwndDlg, IDC_CPU0 + i), i < g_cProcessors);
CheckDlgButton(hwndDlg, IDC_CPU0 + i, i < g_cProcessors && ((*pdwAffinity & (1 << i)) != 0));
}
return TRUE;
}
case WM_COMMAND:
{
switch (LOWORD(wParam))
{
case IDCANCEL:
EndDialog(hwndDlg, IDCANCEL);
break;
case IDOK:
{
*pdwAffinity = 0;
for (int i = 0; i < g_cProcessors; i++)
{
if (IsDlgButtonChecked(hwndDlg, IDC_CPU0 + i))
{
*pdwAffinity |= 1 << i;
}
}
if (*pdwAffinity == 0)
{
// Can't set affinity to "none"
TCHAR szTitle[MAX_PATH];
TCHAR szBody[MAX_PATH];
if (0 == LoadString(g_hInstance, IDS_INVALIDOPTION, szTitle, ARRAYSIZE(szTitle)) ||
0 == LoadString(g_hInstance, IDS_NOAFFINITYMASK, szBody, ARRAYSIZE(szBody)))
{
break;
}
MessageBox(hwndDlg, szBody, szTitle, MB_ICONERROR);
break;
}
EndDialog(hwndDlg, IDOK);
}
}
}
}
return FALSE;
}
/*++ SetAffinity
Routine Description:
Puts up a dialog that lets the user adjust the processor affinity
for a process
Arguments:
pid - process Id of process to modify
Return Value:
boolean success
Revision History:
Jan-17-96 Davepl Created
--*/
BOOL CProcPage::SetAffinity(DWORD pid)
{
BOOL fSuccess = FALSE;
// REVIEW (Davepl) may only need get/set info access here
HANDLE hProcess = OpenProcess( PROCESS_ALL_ACCESS, FALSE, pid );
if (hProcess)
{
DWORD dwAffinity;
DWORD dwUnusedSysAfin;
if (GetProcessAffinityMask(hProcess, &dwAffinity, &dwUnusedSysAfin))
{
if (IDOK == DialogBoxParam(g_hInstance,
MAKEINTRESOURCE(IDD_AFFINITY),
m_hPage,
AffinityDlgProc,
(LPARAM) &dwAffinity))
{
if (FALSE == SetProcessAffinityMask(hProcess, dwAffinity))
{
DWORD dwError = GetLastError();
DisplayFailureMsg(m_hPage, IDS_CANTSETAFFINITY, dwError);
}
else
{
fSuccess = TRUE;
}
}
}
CloseHandle(hProcess);
}
return fSuccess;
}
/*++ KillProcess
Routine Description:
Kills a process
Arguments:
pid - process Id of process to kill
Return Value:
Revision History:
Nov-22-95 Davepl Created
--*/
BOOL CProcPage::KillProcess(DWORD pid)
{
DWORD dwError = ERROR_SUCCESS;
if (IDYES != QuickConfirm(IDS_WARNING, IDS_KILL))
{
return FALSE;
}
//
// Special-case killing WOW tasks
//
CProcInfo * pProcInfo;
pProcInfo = FindProcInArrayByPID(m_pProcArray, (HANDLE)pid);
if (NULL == pProcInfo)
return FALSE;
if (pProcInfo->IsWowTask()) {
HWND hwnd = NULL;
BOOL fDone = FALSE;
DWORD pid;
DWORD pidTarget;
return VDMTerminateTaskWOW(pProcInfo->GetRealPID(), pProcInfo->m_htaskWow);
}
HANDLE hProcess = OpenProcess( PROCESS_ALL_ACCESS, FALSE, pid );
if (hProcess)
{
if (FALSE == TerminateProcess( hProcess, 1 ))
{
dwError = GetLastError();
dprintf(TEXT("Can't terminate process: %08x\n"), dwError);
}
else
{
TimerEvent();
}
CloseHandle( hProcess );
}
else
{
dwError = GetLastError();
}
if (ERROR_SUCCESS != dwError)
{
DisplayFailureMsg(m_hPage, IDS_CANTKILL, dwError);
return FALSE;
}
else
{
return TRUE;
}
}
/*++ AttachDebugger
Routine Description:
Attaches the debugger listed in the AeDebug reg key to the specified
running process
Arguments:
pid - process Id of process to debug
Return Value:
Revision History:
Nov-27-95 Davepl Created
--*/
BOOL CProcPage::AttachDebugger(DWORD pid)
{
DWORD dwError = ERROR_SUCCESS;
if (IDYES != QuickConfirm(IDS_WARNING, IDS_DEBUG))
{
return FALSE;
}
TCHAR szCmdline[MAX_PATH * 2];
wsprintf(szCmdline, TEXT("%s -p %ld"), m_pszDebugger, pid);
STARTUPINFO sinfo =
{
sizeof(STARTUPINFO),
};
PROCESS_INFORMATION pinfo;
if (FALSE == CreateProcess(NULL, //m_pszDebugger,
szCmdline,
NULL,
NULL,
FALSE,
CREATE_NEW_CONSOLE,
NULL,
NULL,
&sinfo,
&pinfo))
{
dwError = GetLastError();
}
else
{
CloseHandle(pinfo.hThread);
CloseHandle(pinfo.hProcess);
}
if (ERROR_SUCCESS != dwError)
{
DisplayFailureMsg(m_hPage, IDS_CANTDEBUG, dwError);
return FALSE;
}
else
{
return TRUE;
}
}
/*++ SetPriority
Routine Description:
Sets a process' priority class
Arguments:
pid - process Id of process to change
pri - ID_CMD_XXXXXX menu choice of priority
Return Value:
Revision History:
Nov-27-95 Davepl Created
--*/
BOOL CProcPage::SetPriority(DWORD pid, DWORD idCmd)
{
DWORD dwError = ERROR_SUCCESS;
DWORD pri;
// Determine which priority class we need to use based
// on the menu selection
switch (idCmd)
{
case IDM_PROC_LOW:
pri = IDLE_PRIORITY_CLASS;
break;
case IDM_PROC_HIGH:
pri = HIGH_PRIORITY_CLASS;
break;
case IDM_PROC_REALTIME:
pri = REALTIME_PRIORITY_CLASS;
break;
default:
Assert(idCmd == IDM_PROC_NORMAL);
pri = NORMAL_PRIORITY_CLASS;
}
// Get confirmation before we change the priority
if (IDYES != QuickConfirm(IDS_WARNING, IDS_PRICHANGE))
{
return FALSE;
}
HANDLE hProcess = OpenProcess( PROCESS_SET_INFORMATION, FALSE, pid );
if (hProcess)
{
if (FALSE == SetPriorityClass( hProcess, pri ))
{
dwError = GetLastError();
dprintf(TEXT("Cant open process for pri change: %08x\n"), dwError);
}
else
{
TimerEvent();
}
CloseHandle( hProcess );
}
else
{
dwError = GetLastError();
}
if (ERROR_SUCCESS != dwError)
{
DisplayFailureMsg(m_hPage, IDS_CANTCHANGEPRI, dwError);
return FALSE;
}
else
{
return TRUE;
}
}
/*++ CProcPage::GetSelectedProcess
Routine Description:
Returns the CProcInfo * of the currently selected process
Arguments:
Return Value:
CProcInfo * on success, NULL on error or nothing selected
Revision History:
Nov-22-95 Davepl Created
--*/
CProcInfo * CProcPage::GetSelectedProcess()
{
HWND hTaskList = GetDlgItem(m_hPage, IDC_PROCLIST);
INT iItem = ListView_GetNextItem(hTaskList, -1, LVNI_SELECTED);
CProcInfo * pProc;
if (-1 != iItem)
{
LV_ITEM lvitem = { LVIF_PARAM };
lvitem.iItem = iItem;
if (ListView_GetItem(hTaskList, &lvitem))
{
pProc = (CProcInfo *) (lvitem.lParam);
}
else
{
pProc = NULL;
}
}
else
{
pProc = NULL;
}
return pProc;
}
/*++ CProcPage::HandleWMCOMMAND
Routine Description:
Handles WM_COMMANDS received at the main page dialog
Arguments:
id - Command id of command received
Return Value:
Revision History:
Nov-22-95 Davepl Created
--*/
void CProcPage::HandleWMCOMMAND(INT id)
{
CProcInfo * pProc = GetSelectedProcess();
switch(id)
{
case IDC_DEBUG:
case IDM_PROC_DEBUG:
{
if (pProc && m_pszDebugger)
{
AttachDebugger( (DWORD)(pProc->m_UniqueProcessId));
}
break;
}
case IDC_TERMINATE:
case IDM_PROC_TERMINATE:
{
if (pProc)
{
KillProcess( (DWORD)(pProc->m_UniqueProcessId));
}
break;
}
case IDM_AFFINITY:
{
if (pProc)
{
SetAffinity( (DWORD)(pProc->m_UniqueProcessId));
}
break;
}
case IDM_PROC_REALTIME:
case IDM_PROC_HIGH:
case IDM_PROC_NORMAL:
case IDM_PROC_LOW:
{
if (pProc)
{
SetPriority( (DWORD)(pProc->m_UniqueProcessId), id);
}
break;
}
}
}
/*++ ProcPageProc
Routine Description:
Dialogproc for the process page.
Arguments:
hwnd - handle to dialog box
uMsg - message
wParam - first message parameter
lParam - second message parameter
Return Value:
For WM_INITDIALOG, TRUE == success
For others, TRUE == this proc handles the message
Revision History:
Nov-16-95 Davepl Created
--*/
BOOL CALLBACK ProcPageProc(
HWND hwnd, // handle to dialog box
UINT uMsg, // message
WPARAM wParam, // first message parameter
LPARAM lParam // second message parameter
)
{
CProcPage * thispage = (CProcPage *) GetWindowLong(hwnd, GWL_USERDATA);
// See if the parent wants this message
if (TRUE == CheckParentDeferrals(uMsg, wParam, lParam))
{
return TRUE;
}
switch(uMsg)
{
case WM_INITDIALOG:
{
SetWindowLong(hwnd, GWL_USERDATA, lParam);
CProcPage * thispage = (CProcPage *) lParam;
thispage->m_hPage = hwnd;
// Turn on SHOWSELALWAYS so that the selection is still highlighted even
// when focus is lost to one of the buttons (for example)
HWND hTaskList = GetDlgItem(hwnd, IDC_PROCLIST);
SetWindowLong(hTaskList, GWL_STYLE, GetWindowLong(hTaskList, GWL_STYLE) | LVS_SHOWSELALWAYS);
return TRUE;
}
// We need to fake client mouse clicks in this child to appear as nonclient
// (caption) clicks in the parent so that the user can drag the entire app
// when the title bar is hidden by dragging the client area of this child
case WM_LBUTTONUP:
case WM_LBUTTONDOWN:
{
if (g_Options.m_fNoTitle)
{
SendMessage(g_hMainWnd,
uMsg == WM_LBUTTONUP ? WM_NCLBUTTONUP : WM_NCLBUTTONDOWN,
HTCAPTION,
lParam);
}
break;
}
case WM_NCLBUTTONDBLCLK:
case WM_LBUTTONDBLCLK:
{
SendMessage(g_hMainWnd, uMsg, wParam, lParam);
break;
}
// We have been asked to find and select a process
case WM_FINDPROC:
{
DWORD cProcs = thispage->m_pProcArray->GetSize();
DWORD dwProcessId;
for (INT iPass = 0; iPass < 2; iPass++)
{
//
// On the first pass we try to find a WOW
// task with a thread ID which matches the
// one given in wParam. If we don't find
// such a task, we look for a process which
// matches the PID in lParam.
//
for (UINT i = 0; i < cProcs; i++)
{
dwProcessId = (DWORD) ((CProcInfo *)(thispage->m_pProcArray->GetAt(i)))->m_UniqueProcessId;
if ((!iPass && wParam == (WPARAM) dwProcessId) ||
( iPass && lParam == (LPARAM) dwProcessId))
{
ListView_SetItemState (GetDlgItem(hwnd, IDC_PROCLIST),
i,
LVIS_FOCUSED | LVIS_SELECTED,
0x000F);
ListView_EnsureVisible(GetDlgItem(hwnd, IDC_PROCLIST), i, FALSE);
goto FoundProc;
}
}
}
FoundProc:
break;
}
case WM_COMMAND:
{
thispage->HandleWMCOMMAND(LOWORD(wParam));
break;
}
case WM_CONTEXTMENU:
{
CProcInfo * pProc = thispage->GetSelectedProcess();
if (pProc && pProc->m_UniqueProcessId)
{
if ((HWND) wParam == GetDlgItem(hwnd, IDC_PROCLIST))
{
thispage->HandleProcListContextMenu(LOWORD(lParam), HIWORD(lParam));
return TRUE;
}
}
break;
}
case WM_NOTIFY:
{
return thispage->HandleProcPageNotify((HWND) wParam, (LPNMHDR) lParam);
}
// Size our kids
case WM_SIZE:
{
thispage->SizeProcPage();
return TRUE;
}
default:
return FALSE;
}
return FALSE;
}
/*++ CProcPage::GetTitle
Routine Description:
Copies the title of this page to the caller-supplied buffer
Arguments:
pszText - the buffer to copy to
bufsize - size of buffer, in characters
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
void CProcPage::GetTitle(LPTSTR pszText, size_t bufsize)
{
LoadString(g_hInstance, IDS_PROCPAGETITLE, pszText, bufsize);
}
/*++ CProcPage::Activate
Routine Description:
Brings this page to the front, sets its initial position,
and shows it
Arguments:
Return Value:
HRESULT (S_OK on success)
Revision History:
Nov-16-95 Davepl Created
--*/
HRESULT CProcPage::Activate()
{
// Make this page visible
ShowWindow(m_hPage, SW_SHOW);
SetWindowPos(m_hPage,
HWND_TOP,
0, 0, 0, 0,
SWP_NOMOVE | SWP_NOSIZE);
// Newly created dialogs seem to steal the focus, so give it back to the
// tab control (which must have had it if we got here in the first place)
SetFocus(m_hwndTabs);
// Change the menu bar to be the menu for this page
HMENU hMenuOld = GetMenu(g_hMainWnd);
HMENU hMenuNew = LoadMenu(g_hInstance, MAKEINTRESOURCE(IDR_MAINMENU_PROC));
if (hMenuNew && SHRestricted(REST_NORUN))
{
DeleteMenu(hMenuNew, IDM_RUN, MF_BYCOMMAND);
}
g_hMenu = hMenuNew;
if (g_Options.m_fNoTitle == FALSE)
{
SetMenu(g_hMainWnd, hMenuNew);
}
if (hMenuOld)
{
DestroyMenu(hMenuOld);
}
return S_OK;
}
/*++ CProcPage::Initialize
Routine Description:
Initializes the process page
Arguments:
hwndParent - Parent on which to base sizing on: not used for creation,
since the main app window is always used as the parent in
order to keep tab order correct
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
HRESULT CProcPage::Initialize(HWND hwndParent)
{
//
// Find out what debbuger is configured on this system
//
HKEY hkDebug;
if (ERROR_SUCCESS == RegOpenKeyEx(HKEY_LOCAL_MACHINE,
TEXT("SOFTWARE\\Microsoft\\Windows NT\\CurrentVersion\\AeDebug"),
0, KEY_READ, &hkDebug))
{
TCHAR szDebugger[MAX_PATH * 2];
DWORD cbString = sizeof(szDebugger);
if (ERROR_SUCCESS == RegQueryValueEx(hkDebug, TEXT("Debugger"), NULL,
NULL, (LPBYTE) szDebugger, &cbString))
{
// Find the first token (which is the debugger exe name/path)
LPTSTR pszCmdLine = szDebugger;
if ( *pszCmdLine == TEXT('\"') )
{
//
// Scan, and skip over, subsequent characters until
// another double-quote or a null is encountered.
//
while ( *++pszCmdLine && (*pszCmdLine != TEXT('\"')) )
{
NULL;
}
//
// If we stopped on a double-quote (usual case), skip
// over it.
//
if ( *pszCmdLine == TEXT('\"') )
{
pszCmdLine++;
}
}
else
{
while (*pszCmdLine > TEXT(' '))
{
pszCmdLine++;
}
}
*pszCmdLine = TEXT('\0'); // Don't need the rest of the args, etc
// If the doctor is in, we don't allow the Debug action
if (lstrlen(szDebugger) && lstrcmpi(szDebugger, TEXT("drwtsn32")) && lstrcmpi(szDebugger, TEXT("drwtsn32.exe")))
{
m_pszDebugger = (LPTSTR) LocalAlloc( 0, (lstrlen(szDebugger) + 1) * sizeof(TCHAR));
if (NULL == m_pszDebugger)
{
return E_OUTOFMEMORY;
}
lstrcpy(m_pszDebugger, szDebugger);
}
}
RegCloseKey(hkDebug);
}
//
// Get basic info like page size, etc.
//
NTSTATUS Status = NtQuerySystemInformation(
SystemBasicInformation,
&g_BasicInfo,
sizeof(g_BasicInfo),
NULL
);
if (!NT_SUCCESS(Status))
{
return E_FAIL;
}
//
// Create the ptr array used to hold the info on running processes
//
m_pProcArray = new CPtrArray(GetProcessHeap());
if (NULL == m_pProcArray)
{
return E_OUTOFMEMORY;
}
// Our pseudo-parent is the tab contrl, and is what we base our
// sizing on. However, in order to keep tab order right among
// the controls, we actually create ourselves with the main
// window as the parent
m_hwndTabs = hwndParent;
//
// Create the dialog which represents the body of this page
//
m_hPage = CreateDialogParam(
g_hInstance, // handle to application instance
MAKEINTRESOURCE(IDD_PROCPAGE), // identifies dialog box template name
g_hMainWnd, // handle to owner window
ProcPageProc, // pointer to dialog box procedure
(LPARAM) this ); // User data (our this pointer)
if (NULL == m_hPage)
{
return GetLastHRESULT();
}
// Set up the columns in the listview
if (FAILED(SetupColumns()))
{
DestroyWindow(m_hPage);
m_hPage = NULL;
return E_FAIL;
}
// Do one initial calculation
TimerEvent();
return S_OK;
}
/*++ CProcPage::Destroy
Routine Description:
Frees whatever has been allocated by the Initialize call
Arguments:
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
// BUGBUG cleanup ptr array, etc
HRESULT CProcPage::Destroy()
{
if (m_hPage)
{
DestroyWindow(m_hPage);
m_hPage = NULL;
}
if (m_pszDebugger)
{
LocalFree(m_pszDebugger);
m_pszDebugger = NULL;
}
return S_OK;
}
/*++ CProcPage::SaveColumnWidths
Routine Description:
Saves the widths of all of the columns in the global options structure
Revision History:
Jan-26-95 Davepl Created
--*/
void CProcPage::SaveColumnWidths()
{
UINT i = 0;
LV_COLUMN col = { 0 };
while (g_Options.m_ActiveProcCol[i] != (COLUMNID) -1)
{
col.mask = LVCF_WIDTH;
if (ListView_GetColumn(GetDlgItem(m_hPage, IDC_PROCLIST), i, &col) )
{
g_Options.m_ColumnWidths[i] = col.cx;
}
else
{
ASSERT(0 && "Couldn't get the column width");
}
i++;
}
}
/*++ CProcPage::Deactivate
Routine Description:
Called when this page is losing its place up front
Arguments:
Return Value:
Revision History:
Nov-16-95 Davepl Created
--*/
void CProcPage::Deactivate()
{
SaveColumnWidths();
if (m_hPage)
{
ShowWindow(m_hPage, SW_HIDE);
}
}