mirror of https://github.com/tongzx/nt5src
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.
449 lines
14 KiB
449 lines
14 KiB
/*======================================================================================//
|
|
| Process Control //
|
|
| //
|
|
|Copyright (c) 1998 Sequent Computer Systems, Incorporated. All rights reserved. //
|
|
| //
|
|
|File Name: JobFolder.cpp //
|
|
| //
|
|
|Description: Implementation of Job List node //
|
|
| //
|
|
|Created: Paul Skoglund 08-1998 //
|
|
| //
|
|
|Rev History: //
|
|
| //
|
|
|=======================================================================================*/
|
|
|
|
#include "StdAfx.h"
|
|
#include "BaseNode.h"
|
|
|
|
#include "JobPages.h"
|
|
#include "ManagementPages.h"
|
|
#include "ManagementRuleWizards.h"
|
|
|
|
#pragma warning(push)
|
|
#include <algorithm>
|
|
#pragma warning(pop)
|
|
|
|
using std::list<PCJobListItem*>;
|
|
using std::list<CBaseNode *>;
|
|
using std::find;
|
|
|
|
const GUID CJobFolder::m_GUID = {0xff9baf62,0x064e,0x11d2,{0x80, 0x14,0x00,0x10,0x4b,0x9a,0x31,0x06} };
|
|
const TCHAR *const CJobFolder::m_szGUID = _T("{ff9baf62-064e-11d2-8014-00104b9a3106}");
|
|
|
|
const CONTEXTMENUITEMBYID CJobFolder::ViewMenuItems[] =
|
|
{
|
|
{ IDS_JOBVIEW_ALL, ID_JOBVIEW_ALL, ID_JOBVIEW_ALL, CCM_INSERTIONPOINTID_PRIMARY_VIEW },
|
|
{ IDS_JOBVIEW_RUN, ID_JOBVIEW_RUN, ID_JOBVIEW_RUN, CCM_INSERTIONPOINTID_PRIMARY_VIEW },
|
|
{ IDS_JOBVIEW_MANAGED, ID_JOBVIEW_MANAGED, ID_JOBVIEW_MANAGED, CCM_INSERTIONPOINTID_PRIMARY_VIEW },
|
|
{ 0, 0, 0, 0 }
|
|
};
|
|
|
|
|
|
CJobFolder::CJobFolder(CBaseNode *pParent) : CBaseNode(JOB_NODE, pParent), m_ID(0), m_fViewOption(ID_JOBVIEW_ALL)
|
|
{
|
|
LoadStringHelper(m_name, IDS_JOBS_FOLDER);
|
|
}
|
|
|
|
CJobFolder::~CJobFolder()
|
|
{
|
|
FreeNodes();
|
|
ATLTRACE( _T("~CJobFolder end\n"));
|
|
}
|
|
|
|
void CJobFolder::FreeNodes()
|
|
{
|
|
ATLTRACE( _T("CJobFolder::FreeNodes()\n"));
|
|
for (list<CBaseNode *>::iterator i = m_NodeList.begin(); i != m_NodeList.end(); ++i)
|
|
{
|
|
(*i)->Release();
|
|
}
|
|
m_NodeList.clear();
|
|
}
|
|
|
|
LPCTSTR CJobFolder::GetNodeName()
|
|
{
|
|
return m_name;
|
|
}
|
|
|
|
HRESULT CJobFolder::GetDisplayInfo(RESULTDATAITEM &ResultItem)
|
|
{
|
|
if (ResultItem.bScopeItem)
|
|
{
|
|
if( ResultItem.mask & RDI_STR )
|
|
{
|
|
if (0 == ResultItem.nCol)
|
|
ResultItem.str = const_cast<LPOLESTR>(GetNodeName());
|
|
else
|
|
ResultItem.str = _T("");
|
|
}
|
|
if (ResultItem.mask & RDI_IMAGE)
|
|
ResultItem.nImage = sImage();
|
|
return S_OK;
|
|
}
|
|
ASSERT(FALSE);
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnExpand(BOOL bExpand, HSCOPEITEM hItem, IConsoleNameSpace2 *ipConsoleNameSpace2)
|
|
{
|
|
if (bExpand)
|
|
{
|
|
ASSERT(m_ID == hItem);
|
|
|
|
RePopulateScopePane(ipConsoleNameSpace2);
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::AddNode(IConsoleNameSpace2 *ipConsoleNameSpace2, CBaseNode *pSubNode)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
SCOPEDATAITEM sdi = {0};
|
|
|
|
if (!pSubNode)
|
|
return E_OUTOFMEMORY;
|
|
|
|
// Place items into the scope pane
|
|
sdi.mask = SDI_STR | // Displayname is valid
|
|
SDI_PARAM | // lParam is valid
|
|
SDI_IMAGE | // nImage is valid
|
|
SDI_OPENIMAGE | // nOpenImage is valid
|
|
SDI_CHILDREN | // cChildren is valid
|
|
SDI_PARENT;
|
|
|
|
sdi.displayname = (LPOLESTR)MMC_CALLBACK;
|
|
sdi.nImage = pSubNode->sImage();
|
|
sdi.nOpenImage = pSubNode->sOpenImage();
|
|
//sdi.nState = ???
|
|
sdi.cChildren = pSubNode->GetChildrenCount();
|
|
sdi.lParam = reinterpret_cast <LPARAM> (pSubNode);
|
|
sdi.relativeID = m_ID;
|
|
|
|
|
|
hr = ipConsoleNameSpace2->InsertItem( &sdi );
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
pSubNode->SetID(sdi.ID);
|
|
m_NodeList.push_front( pSubNode );
|
|
}
|
|
else
|
|
{
|
|
pSubNode->Release();
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnShow(BOOL bSelecting, HSCOPEITEM hItem, IHeaderCtrl2* ipHeaderCtrl,
|
|
IConsole2 *ipConsole2, IConsoleNameSpace2 *ipConsoleNameSpace2)
|
|
{
|
|
ASSERT(hItem == GetID());
|
|
|
|
if (!bSelecting)
|
|
return S_OK;
|
|
|
|
ITEM_STR str;
|
|
|
|
LoadStringHelper(str, IDS_JOB_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( JOB_COLUMN, str, 0, JOB_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_STATUS_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( STATUS_COLUMN, str, 0, STATUS_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_ACTIVE_PROCESS_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( ACTIVE_PROCESS_COUNT_COLUMN, str, 0, ACTIVE_PROCESS_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_AFFINITY_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( AFFINITY_COLUMN, str, 0, AFFINITY_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PRIORITY_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PRIORITY_COLUMN, str, 0, PRIORITY_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_SCHEDULING_CLASS_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( SCHEDULING_CLASS_COLUMN, str, 0, SCHEDULING_CLASS_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_USER_TIME_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( USER_TIME_COLUMN, str, 0, USER_TIME_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_KERNEL_TIME_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( KERNEL_TIME_COLUMN, str, 0, KERNEL_TIME_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PERIOD_USER_TIME_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PERIOD_USER_TIME_COLUMN, str, 0, PERIOD_USER_TIME_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PERIOD_KERNEL_TIME_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PERIOD_KERNEL_TIME_COLUMN, str, 0, PERIOD_KERNEL_TIME_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PAGE_FAULT_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PAGE_FAULT_COUNT_COLUMN, str, 0, PAGE_FAULT_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PROCESS_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PROCESS_COUNT_COLUMN, str, 0, PROCESS_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_TERMINATED_PROCESS_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( TERMINATED_PROCESS_COUNT_COLUMN, str, 0, TERMINATED_PROCESS_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_READOP_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( READOP_COUNT_COLUMN, str, 0, READOP_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_WRITEOP_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( WRITEOP_COUNT_COLUMN, str, 0, WRITEOP_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_OTHEROP_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( OTHEROP_COUNT_COLUMN, str, 0, OTHEROP_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_READTRANS_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( READTRANS_COUNT_COLUMN, str, 0, READTRANS_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_WRITETRANS_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( WRITETRANS_COUNT_COLUMN, str, 0, WRITETRANS_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_OTHERTRANS_COUNT_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( OTHERTRANS_COUNT_COLUMN, str, 0, OTHERTRANS_COUNT_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PEAK_PROC_MEM_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PEAK_PROC_MEM_COLUMN, str, 0, PEAK_PROC_MEM_COLUMN_WIDTH ));
|
|
|
|
LoadStringHelper(str, IDS_PEAK_JOB_MEM_HDR);
|
|
VERIFY(S_OK == ipHeaderCtrl->InsertColumn( PEAK_JOB_MEM_COLUMN, str, 0, PEAK_JOB_MEM_COLUMN_WIDTH ));
|
|
|
|
return OnRefresh(ipConsole2, ipConsoleNameSpace2);
|
|
}
|
|
|
|
HRESULT CJobFolder::AddMenuItems(LPCONTEXTMENUCALLBACK piCallback, long * pInsertionAllowed)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
ITEM_STR name;
|
|
ITEM_STR status;
|
|
|
|
if (*pInsertionAllowed & CCM_INSERTIONALLOWED_VIEW)
|
|
{
|
|
CONTEXTMENUITEM m = { 0 };
|
|
for (const CONTEXTMENUITEMBYID *M = ViewMenuItems; M->lCommandID; M++)
|
|
{
|
|
m.strName = const_cast<TCHAR *>(LoadStringHelper(name, M->strNameID));
|
|
m.strStatusBarText = const_cast<TCHAR *>(LoadStringHelper(status, M->strStatusBarTextID));
|
|
m.lCommandID = M->lCommandID;
|
|
m.lInsertionPointID = M->lInsertionPointID;
|
|
//m.fSpecialFlags = 0; // currently always 0, initialized to 0
|
|
|
|
if (m.lCommandID == m_fViewOption)
|
|
m.fFlags = MF_CHECKED | MF_ENABLED;
|
|
else
|
|
m.fFlags = MF_UNCHECKED | MF_ENABLED;
|
|
|
|
hr = piCallback->AddItem(&m);
|
|
|
|
if (FAILED(hr))
|
|
break;
|
|
}
|
|
*pInsertionAllowed ^= CCM_INSERTIONALLOWED_VIEW;
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CJobFolder::AddMenuItems(LPCONTEXTMENUCALLBACK piCallback, long * pInsertionAllowed, LPARAM Cookie)
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnMenuCommand(IConsole2 *ipConsole2, IConsoleNameSpace2 *ipConsoleNameSpace2, long nCommandID )
|
|
{
|
|
switch(nCommandID)
|
|
{
|
|
case ID_JOBVIEW_RUN:
|
|
case ID_JOBVIEW_MANAGED:
|
|
case ID_JOBVIEW_ALL:
|
|
m_fViewOption = nCommandID;
|
|
return OnRefresh(ipConsole2, ipConsoleNameSpace2);
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnMenuCommand(IConsole2 *ipConsole2, long nCommandID, LPARAM Cookie )
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnHelpCmd(IDisplayHelp *ipDisplayHelp)
|
|
{
|
|
if (!ipDisplayHelp)
|
|
return E_UNEXPECTED;
|
|
|
|
ipDisplayHelp->ShowTopic(const_cast<TCHAR *>(HELP_jo_overview));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnSelect(BOOL bScope, BOOL bSelect, IConsoleVerb* ipConsoleVerb)
|
|
{
|
|
ASSERT(bScope);
|
|
|
|
if ( bSelect && bScope )
|
|
{
|
|
ATLTRACE(_T("CJobFolder::OnSelect Scope=%s, Select=%s\n"), bScope ? _T("true") : _T("false"), bSelect ? _T("true") : _T("false"));
|
|
|
|
VERIFY( ipConsoleVerb->SetVerbState( MMC_VERB_REFRESH, ENABLED, TRUE ) == S_OK);
|
|
VERIFY( ipConsoleVerb->SetDefaultVerb( MMC_VERB_OPEN ) == S_OK );
|
|
}
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnSelect(BOOL bScope, BOOL bSelect, IConsoleVerb* ipConsoleVerb, LPARAM Cookie)
|
|
{
|
|
ASSERT(FALSE);
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnViewChange(IResultData *ipResultData, LPARAM thing, LONG_PTR hint)
|
|
{
|
|
ASSERT(ipResultData);
|
|
if (!ipResultData)
|
|
return E_UNEXPECTED;
|
|
|
|
VERIFY(ipResultData->ModifyViewStyle(MMC_SINGLESEL, MMC_NOSORTHEADER) == S_OK);
|
|
|
|
HRESULT hr = E_UNEXPECTED;
|
|
switch (hint)
|
|
{
|
|
case PC_VIEW_UPDATEALL:
|
|
hr = ShowAllItems(ipResultData, FALSE);
|
|
break;
|
|
case PC_VIEW_REDRAWALL:
|
|
case PC_VIEW_ADDITEM:
|
|
case PC_VIEW_UPDATEITEM:
|
|
case PC_VIEW_SETITEM:
|
|
case PC_VIEW_DELETEITEM:
|
|
default:
|
|
hr = E_UNEXPECTED;
|
|
break;
|
|
}
|
|
ASSERT(hr == S_OK);
|
|
return hr;
|
|
}
|
|
|
|
HRESULT CJobFolder::ShowAllItems(IResultData* ipResultData, BOOL bCacheValid)
|
|
{
|
|
VERIFY( S_OK == ipResultData->DeleteAllRsltItems() );
|
|
|
|
ITEM_STR str;
|
|
if (ID_JOBVIEW_RUN == m_fViewOption)
|
|
ipResultData->SetDescBarText(const_cast<TCHAR *>(LoadStringHelper(str, IDS_RUNNING_JOBS)));
|
|
else if (ID_JOBVIEW_MANAGED == m_fViewOption)
|
|
ipResultData->SetDescBarText(const_cast<TCHAR *>(LoadStringHelper(str, IDS_MANAGED_JOBS)));
|
|
else
|
|
ipResultData->SetDescBarText(_T(""));
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HRESULT CJobFolder::OnRefresh(IConsole2 *ipConsole2, IConsoleNameSpace2 *ipConsoleNameSpace2)
|
|
{
|
|
RePopulateScopePane(ipConsoleNameSpace2);
|
|
return SendViewChange(ipConsole2, NULL, PC_VIEW_UPDATEALL);
|
|
}
|
|
|
|
HRESULT CJobFolder::RePopulateScopePane(IConsoleNameSpace2 *ipConsoleNameSpace2)
|
|
{
|
|
int counter = 0;
|
|
|
|
ATLTRACE(_T("CJobFolder:: RePopulateScopePane\n"));
|
|
|
|
ASSERT( 0 <= (counter = ScopeCount(m_ID, ipConsoleNameSpace2)) ); // Debugging aid only
|
|
|
|
// delete the MMC scope pane items for all the jobs - NOT the
|
|
// process groups folder itself but delete it's content
|
|
VERIFY(S_OK == ipConsoleNameSpace2->DeleteItem(m_ID, FALSE));
|
|
// free 'our' objects for the job items nodes
|
|
FreeNodes();
|
|
|
|
// debug purposes only
|
|
ASSERT(0 == ScopeCount(m_ID, ipConsoleNameSpace2)); // Debugging aid only
|
|
|
|
PCINT32 res = 0;
|
|
PCULONG32 err = 0;
|
|
PCINT32 last = 0;
|
|
|
|
const int MINIMUM_ALLOCATION = min((COM_BUFFER_SIZE/sizeof(PCJobListItem)), 100);
|
|
|
|
PCJobListItem tempjoblist[MINIMUM_ALLOCATION];
|
|
|
|
PCid hID = GetPCid();
|
|
if (!hID)
|
|
{
|
|
ReportPCError();
|
|
return false;
|
|
}
|
|
|
|
do
|
|
{
|
|
if (last)
|
|
memcpy(&tempjoblist[0], &tempjoblist[last], sizeof(PCJobListItem));
|
|
else
|
|
memset(&tempjoblist[0], 0, sizeof(PCJobListItem));
|
|
|
|
res = PCGetJobList( hID, &tempjoblist[0], MINIMUM_ALLOCATION * sizeof(PCJobListItem));
|
|
if (res < 0 )
|
|
{
|
|
err = GetLastPCError();
|
|
break;
|
|
}
|
|
|
|
if (res > MINIMUM_ALLOCATION)
|
|
{
|
|
err = PCERROR_INVALID_RESPONSE;
|
|
break;
|
|
}
|
|
|
|
if (res > 0)
|
|
{
|
|
last = res - 1 ;
|
|
|
|
for (PCINT32 i = 0; i < res; i++)
|
|
{
|
|
if (ID_JOBVIEW_RUN == m_fViewOption && !(tempjoblist[i].lFlags & PCLFLAG_IS_RUNNING))
|
|
continue;
|
|
else if (ID_JOBVIEW_MANAGED == m_fViewOption && !(tempjoblist[i].lFlags & PCLFLAG_IS_MANAGED))
|
|
continue;
|
|
|
|
VERIFY( S_OK == AddNode(ipConsoleNameSpace2, new CJobItemFolder(this, tempjoblist[i]) ) );
|
|
}
|
|
}
|
|
} while (res > 0 && PCERROR_MORE_DATA == GetLastPCError() );
|
|
|
|
if (err)
|
|
ReportPCError(err);
|
|
|
|
ASSERT( 0 <= (counter = ScopeCount(m_ID, ipConsoleNameSpace2)) ); // Debugging aid only
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
int CJobFolder::ScopeCount(HSCOPEITEM ID, IConsoleNameSpace2 *ipConsoleNameSpace2)
|
|
{
|
|
HSCOPEITEM curID, prevID;
|
|
MMC_COOKIE Cookie;
|
|
|
|
int count = 0;
|
|
|
|
HRESULT hr = ipConsoleNameSpace2->GetChildItem(ID, &curID, &Cookie);
|
|
while(hr == S_OK && curID)
|
|
{
|
|
prevID = curID;
|
|
count++;
|
|
|
|
hr = ipConsoleNameSpace2->GetNextItem(prevID, &curID, &Cookie);
|
|
}
|
|
|
|
return count;
|
|
}
|