313 lines
7.4 KiB
C++
313 lines
7.4 KiB
C++
|
|
|
|
// TaskMan - NT TaskManager
|
|
// Copyright (C) Microsoft
|
|
|
|
// File: TaskMan.H
|
|
|
|
// History: Nov-10-95 DavePl Created
|
|
// Jun-30-98 Alhen Adding code for TerminalServer
|
|
|
|
|
|
|
|
#define WM_FINDPROC (WM_USER + 1)
|
|
#define PM_NOTIFYWAITING (WM_USER + 2)
|
|
#define PM_QUITTRAYTHREAD (WM_USER + 3)
|
|
|
|
extern CRITICAL_SECTION g_CSTrayThread;
|
|
extern DWORD g_idTrayThread;
|
|
extern LONG g_minWidth;
|
|
extern LONG g_minHeight;
|
|
|
|
DWORD TrayThreadMessageLoop(LPVOID);
|
|
|
|
#define TASK_PAGE 0
|
|
#define PROC_PAGE 1
|
|
#define PERF_PAGE 2
|
|
#define NUM_PAGES 3
|
|
|
|
|
|
#define MIN_DLG_SIZE_X 203
|
|
#define MIN_DLG_SIZE_Y 224
|
|
#define DLG_SCALE_X 4
|
|
#define DLG_SCALE_Y 8
|
|
|
|
|
|
// Process Page Column ID enumeration
|
|
|
|
|
|
typedef enum COLUMNID
|
|
{
|
|
COL_IMAGENAME = 0,
|
|
COL_PID = 1,
|
|
|
|
// _HYDRA
|
|
|
|
COL_USERNAME = 2,
|
|
|
|
COL_SESSIONID = 3,
|
|
|
|
|
|
COL_CPU = 4,
|
|
COL_CPUTIME = 5,
|
|
COL_MEMUSAGE = 6,
|
|
COL_MEMPEAK = 7,
|
|
COL_MEMUSAGEDIFF = 8,
|
|
COL_PAGEFAULTS = 9,
|
|
COL_PAGEFAULTSDIFF = 10,
|
|
COL_COMMITCHARGE = 11,
|
|
COL_PAGEDPOOL = 12,
|
|
COL_NONPAGEDPOOL = 13,
|
|
COL_BASEPRIORITY = 14,
|
|
COL_HANDLECOUNT = 15,
|
|
COL_THREADCOUNT = 16,
|
|
COL_USEROBJECTS = 17,
|
|
COL_GDIOBJECTS = 18,
|
|
COL_READOPERCOUNT = 19,
|
|
COL_WRITEOPERCOUNT = 20,
|
|
COL_OTHEROPERCOUNT = 21,
|
|
COL_READXFERCOUNT = 22,
|
|
COL_WRITEXFERCOUNT = 23,
|
|
COL_OTHERXFERCOUNT = 24
|
|
};
|
|
|
|
#define MAX_COLUMN 24
|
|
#define NUM_COLUMN (MAX_COLUMN + 1)
|
|
|
|
#define IDS_FIRSTCOL 20001 // 20000 is first column name ID in rc file
|
|
|
|
// GetLastHRESULT
|
|
|
|
// Little wrapper func that returns the GetLastError value as an HRESULT
|
|
|
|
inline HRESULT GetLastHRESULT()
|
|
{
|
|
return HRESULT_FROM_WIN32(GetLastError());
|
|
}
|
|
|
|
// Possible values for the viewmode
|
|
|
|
typedef enum
|
|
{
|
|
VM_LARGEICON,
|
|
VM_SMALLICON,
|
|
VM_DETAILS,
|
|
VM_INVALID
|
|
} VIEWMODE;
|
|
#define VM_FIRST IDM_LARGEICONS
|
|
#define VM_LAST IDM_DETAILS
|
|
|
|
// Possible values for the cpu history mode
|
|
|
|
typedef enum
|
|
{
|
|
CM_SUM,
|
|
CM_PANES
|
|
} CPUHISTMODE;
|
|
#define CM_FIRST IDM_ALLCPUS
|
|
#define CM_LAST IDM_MULTIGRAPH
|
|
|
|
// Possible values for the update speed option
|
|
|
|
typedef enum
|
|
{
|
|
US_HIGH,
|
|
US_NORMAL,
|
|
US_LOW,
|
|
US_PAUSED
|
|
} UPDATESPEED;
|
|
#define US_FIRST IDM_HIGH
|
|
#define US_LAST IDM_PAUSED
|
|
|
|
|
|
|
|
|
|
//BOOL ValidateProductSuite( LPTSTR lpszSuiteName);
|
|
BOOL ValidateProductSuite( );
|
|
|
|
// This function determines if the machine is running hydra.
|
|
BOOL IsTerminalServer ();
|
|
|
|
// PtrToFns for RPC calls
|
|
|
|
typedef BOOLEAN ( WINAPI *pfnWinStationGetProcessSid )( HANDLE hServer, DWORD ProcessId , FILETIME ProcessStartTime , PBYTE pProcessUserSid , PDWORD dwSidSize );
|
|
|
|
typedef void ( WINAPI *pfnCachedGetUserFromSid )( PSID pSid , PWCHAR pUserName , PULONG cbUserName );
|
|
|
|
typedef BOOLEAN (WINAPI *pfnWinStationTerminateProcess)( HANDLE hServer, ULONG ProcessId, ULONG ExitCode);
|
|
|
|
|
|
extern pfnWinStationGetProcessSid gpfnWinStationGetProcessSid;
|
|
|
|
extern pfnCachedGetUserFromSid gpfnCachedGetUserFromSid;
|
|
|
|
extern pfnWinStationTerminateProcess gpfnWinStationTerminateProcess;
|
|
|
|
|
|
// COptions
|
|
|
|
// App's persistent state across sessions, saved in the registry
|
|
|
|
class COptions
|
|
{
|
|
public:
|
|
|
|
DWORD m_cbSize;
|
|
DWORD m_dwTimerInterval;
|
|
VIEWMODE m_vmViewMode;
|
|
CPUHISTMODE m_cmHistMode;
|
|
UPDATESPEED m_usUpdateSpeed;
|
|
RECT m_rcWindow;
|
|
INT m_iCurrentPage;
|
|
|
|
COLUMNID m_ActiveProcCol[NUM_COLUMN + 1];
|
|
INT m_ColumnWidths[NUM_COLUMN + 1];
|
|
INT m_ColumnPositions[NUM_COLUMN + 1];
|
|
|
|
BOOL m_fMinimizeOnUse : 1;
|
|
BOOL m_fConfirmations : 1;
|
|
BOOL m_fAlwaysOnTop : 1;
|
|
BOOL m_fKernelTimes : 1;
|
|
BOOL m_fNoTitle : 1;
|
|
BOOL m_fHideWhenMin : 1;
|
|
BOOL m_fShow16Bit : 1;
|
|
BOOL m_fShowSandboxNames : 1;
|
|
BOOL bUnused;
|
|
BOOL bUnused2;
|
|
BOOL m_bShowAllProcess;
|
|
|
|
HRESULT Load();
|
|
HRESULT Save();
|
|
|
|
// SetDefaultValues
|
|
|
|
// Uses to init the options to a default state when the saved copy
|
|
// cannot be found, is damaged, or is not the correct version
|
|
|
|
void SetDefaultValues()
|
|
{
|
|
ZeroMemory(this, sizeof(COptions));
|
|
|
|
m_cbSize = sizeof(COptions);
|
|
|
|
BOOL bScreenReader = FALSE;
|
|
if (SystemParametersInfo(SPI_GETSCREENREADER, 0, (PVOID) &bScreenReader, 0) && bScreenReader)
|
|
{
|
|
// No automatic updates for machines with screen readers
|
|
m_dwTimerInterval = 0;
|
|
}
|
|
else
|
|
{
|
|
m_dwTimerInterval = 1000;
|
|
}
|
|
|
|
m_vmViewMode = VM_DETAILS;
|
|
m_cmHistMode = CM_PANES;
|
|
m_usUpdateSpeed = US_NORMAL;
|
|
m_fMinimizeOnUse = TRUE;
|
|
m_fConfirmations = TRUE;
|
|
m_fAlwaysOnTop = TRUE;
|
|
m_fShow16Bit = TRUE;
|
|
m_fShowSandboxNames= TRUE;
|
|
|
|
m_iCurrentPage = -1;
|
|
m_rcWindow.top = 10;
|
|
m_rcWindow.left = 10;
|
|
m_rcWindow.bottom = 10 + g_minHeight;
|
|
m_rcWindow.right = 10 + g_minWidth;
|
|
|
|
// Columns which are visible, by default, in the process view
|
|
//#ifdef _HYDRA_
|
|
|
|
m_bShowAllProcess = FALSE;
|
|
|
|
if( IsTerminalServer( ) )
|
|
{
|
|
m_ActiveProcCol[ 0 ] = COL_IMAGENAME;
|
|
|
|
m_ActiveProcCol[ 1 ] = COL_USERNAME;
|
|
|
|
m_ActiveProcCol[ 2 ] = COL_SESSIONID;
|
|
|
|
m_ActiveProcCol[ 3 ] = COL_PID;
|
|
|
|
m_ActiveProcCol[ 4 ] = COL_CPU;
|
|
|
|
m_ActiveProcCol[ 5 ] = COL_CPUTIME;
|
|
|
|
m_ActiveProcCol[ 6 ] = COL_MEMUSAGE;
|
|
|
|
m_ActiveProcCol[ 7 ] = (COLUMNID) -1;
|
|
|
|
}
|
|
else
|
|
{
|
|
m_ActiveProcCol[ 0 ] = COL_IMAGENAME;
|
|
|
|
m_ActiveProcCol[ 1 ] = COL_PID;
|
|
|
|
m_ActiveProcCol[ 2 ] = COL_CPU;
|
|
|
|
m_ActiveProcCol[ 3 ] = COL_CPUTIME;
|
|
|
|
m_ActiveProcCol[ 4 ] = COL_MEMUSAGE;
|
|
|
|
m_ActiveProcCol[ 5 ] = (COLUMNID) -1;
|
|
}
|
|
//#endif _HYDRA_
|
|
|
|
|
|
// Set all of the columns widths to -1
|
|
|
|
FillMemory(m_ColumnWidths, sizeof(m_ColumnWidths), 0xFF);
|
|
FillMemory(m_ColumnPositions, sizeof(m_ColumnPositions), 0xFF);
|
|
}
|
|
|
|
COptions()
|
|
{
|
|
SetDefaultValues();
|
|
}
|
|
};
|
|
|
|
// CTrayNotification
|
|
|
|
// Class to encapsulate all of the info needed to do a tray notification
|
|
|
|
class CTrayNotification
|
|
{
|
|
public:
|
|
|
|
CTrayNotification()
|
|
{
|
|
ASSERT(0 && "Some bozo is using the default constuctor for CTrayNotification");
|
|
ZeroMemory(this, sizeof(*this));
|
|
}
|
|
|
|
CTrayNotification(HWND hWnd,
|
|
UINT uCallbackMessage,
|
|
DWORD Message,
|
|
HICON hIcon,
|
|
LPTSTR pszTip)
|
|
{
|
|
m_hWnd = hWnd;
|
|
m_uCallbackMessage = uCallbackMessage;
|
|
m_Message = Message;
|
|
m_hIcon = hIcon;
|
|
|
|
if (pszTip)
|
|
lstrcpyn(m_szTip, pszTip, ARRAYSIZE(m_szTip));
|
|
else
|
|
m_szTip[0] = TEXT('\0');
|
|
}
|
|
|
|
HWND m_hWnd;
|
|
UINT m_uCallbackMessage;
|
|
DWORD m_Message;
|
|
HICON m_hIcon;
|
|
TCHAR m_szTip[MAX_PATH];
|
|
};
|
|
|
|
BOOL DeliverTrayNotification(CTrayNotification * pNot);
|
|
void SubclassListView(HWND hwnd);
|
|
int Compare64(unsigned __int64 First, unsigned __int64 Second);
|