mirror of
https://github.com/silverf0x/RpcView
synced 2026-06-08 17:26:38 +00:00
silverf0x
118 lines
3.5 KiB
C++
118 lines
3.5 KiB
C++
#include "ProcessSelectedVisitor.h"
|
|
#include "EndpointsWidget.h"
|
|
#include "InterfacesWidget.h"
|
|
#include "InterfaceInfoWidget.h"
|
|
#include "ProcessWidget.h"
|
|
#include "ProcessInfoWidget.h"
|
|
#include "ProceduresWidget.h"
|
|
#include "../RpcCommon/Misc.h"
|
|
|
|
typedef struct _EnumCtxt_T{
|
|
ProcessInfoWidget_C* pProcessInfoWidget;
|
|
}EnumCtxt_T;
|
|
|
|
//------------------------------------------------------------------------------
|
|
ProcessSelectedVisitor_C::ProcessSelectedVisitor_C(quint32 Pid,RpcCore_T* pRpcCore,void* pRpcCoreCtxt)
|
|
{
|
|
this->Pid = Pid;
|
|
this->pRpcCore = pRpcCore;
|
|
this->pRpcCoreCtxt = pRpcCoreCtxt;
|
|
|
|
pRpcProcessInfo = pRpcCore->RpcCoreGetProcessInfoFn(pRpcCoreCtxt, Pid, INVALID_PID_VALUE, RPC_PROCESS_INFO_ALL);
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
ProcessSelectedVisitor_C::~ProcessSelectedVisitor_C()
|
|
{
|
|
pRpcCore->RpcCoreFreeProcessInfoFn(pRpcCoreCtxt, pRpcProcessInfo);
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
void ProcessSelectedVisitor_C::Visit(EndpointsWidget_C* pEndpointsWidget)
|
|
{
|
|
//Apply process filter
|
|
pEndpointsWidget->ApplyProcessFilter(Pid);
|
|
this->NbOfEndpoints = pEndpointsWidget->GetEndpoints();
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
void ProcessSelectedVisitor_C::Visit(InterfacesWidget_C* pInterfacesWidget)
|
|
{
|
|
//Apply process filter
|
|
pInterfacesWidget->ApplyProcessFilter(Pid);
|
|
this->NbOfInterfaces = pInterfacesWidget->GetInterfaces();
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
void ProcessSelectedVisitor_C::Visit(InterfaceInfoWidget_C* pInterfaceInfoWidget)
|
|
{
|
|
if (this->Pid!=pInterfaceInfoWidget->GetPid())
|
|
{
|
|
pInterfaceInfoWidget->Reset();
|
|
}
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
BOOL __fastcall EnumProcessAuth(DWORD Pid, RpcAuthInfo_T* pRpcAuthInfo, EnumCtxt_T* pEnumCtxt, BOOL* pbContinue)
|
|
{
|
|
UNREFERENCED_PARAMETER(Pid);
|
|
UNREFERENCED_PARAMETER(pbContinue);
|
|
|
|
pEnumCtxt->pProcessInfoWidget->AddAuthInfo(pRpcAuthInfo);
|
|
return (TRUE);
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
void ProcessSelectedVisitor_C::Visit(ProcessInfoWidget_C* pProcessInfoWidget)
|
|
{
|
|
EnumCtxt_T EnumCtxt;
|
|
|
|
EnumCtxt.pProcessInfoWidget = pProcessInfoWidget;
|
|
|
|
pProcessInfoWidget->UpdateProcessInfo( pRpcProcessInfo );
|
|
pProcessInfoWidget->reset(); //for authinfo
|
|
pRpcCore->RpcCoreEnumProcessAuthInfoFn(pRpcCoreCtxt, pRpcProcessInfo->Pid,(RpcCoreEnumProcessAuthInfoCallbackFn_T)&EnumProcessAuth,&EnumCtxt);
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void ProcessSelectedVisitor_C::Visit(ProceduresWidget_C* pProceduresWidget)
|
|
{
|
|
if (pProceduresWidget->GetPid() != this->Pid )
|
|
{
|
|
pProceduresWidget->reset(this->Pid);
|
|
}
|
|
}
|
|
|
|
//------------------------------------------------------------------------------
|
|
void ProcessSelectedVisitor_C::Visit(ProcessWidget_C* pProcessWidget)
|
|
{
|
|
this->NbOfProcesses = pProcessWidget->GetProcesses();
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
ULONG ProcessSelectedVisitor_C::GetEndpoints()
|
|
{
|
|
return (this->NbOfEndpoints);
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
ULONG ProcessSelectedVisitor_C::GetInterfaces()
|
|
{
|
|
return (this->NbOfInterfaces);
|
|
}
|
|
|
|
|
|
//------------------------------------------------------------------------------
|
|
ULONG ProcessSelectedVisitor_C::GetProcesses()
|
|
{
|
|
return (this->NbOfProcesses);
|
|
}
|