Initial

remotepsapp/RemoteProcessInfo.cpp

@@ -0,0 +1,165 @@
+#include <remotepsapp/RemoteProcessInfo.hpp>
+#include <remotepsapp/windowview.hpp>
+#include <com/bstring.hpp>
+#include <com/variant.hpp>
+#include <com/vtime.hpp>
+#include <com/SafeArray.hpp>
+#include <remoteps/remoteps_i.c>
+#include <remoteps/remoteps.h>
+#include <common/purehdc.hpp>
+#include <common/openfile.hpp>
+#include <common/pathfnd.hpp>
+#include <common/errormsg.hpp>
+
+RemoteProcessInfo::RemoteProcessInfo(void)
+: mpIRemoteProcess(0)
+{
+}
+
+RemoteProcessInfo::~RemoteProcessInfo()
+{
+  disconnect();
+}
+
+bool RemoteProcessInfo::connect(const String &serverName)
+{
+  ServerInfo serverInfo(serverName);
+	MultiQuery multiQuery;
+	ComResult comResult;
+	ComObj comObj;
+
+  if(isOkay())return false;
+  mServerName=serverName;
+  multiQuery.size(1);					
+	multiQuery[0].pIID=&IID_IRemoteProcess;
+	comResult=comObj.createInstance(CLSID_CoRemoteProcess,0,ClassContext(RemoteServer|LocalServer),serverInfo,multiQuery);
+	if(!comResult.success())
+	{
+	  ErrorMessage::message(comResult.error());
+	  return false;
+	}
+	if(!ComResult(multiQuery[0].hr).success())return false;
+	mpIRemoteProcess=(IRemoteProcess*)multiQuery[0].pItf;
+	return isOkay();
+}
+
+void RemoteProcessInfo::disconnect(void)
+{
+  if(!isOkay())return;
+	mpIRemoteProcess->Release();
+	mpIRemoteProcess=0;
+}
+
+bool RemoteProcessInfo::getListItems(void)
+{
+  if(!isOkay())return false;
+	return getListItems(mpIRemoteProcess);
+}
+
+bool RemoteProcessInfo::getDesktopWindow(WindowView &windowView)
+{
+  if(!isOkay())return false;
+	return getDesktopWindow(mpIRemoteProcess,windowView);
+}
+
+bool RemoteProcessInfo::getListItems(IRemoteProcess *pIRemoteProcess)
+{
+  Variant variant;
+	Variant processVariant;
+	String ansiString;
+	String strParent;
+	DWORD processID;
+	int processCount;
+
+  if(!pIRemoteProcess)return false;
+  processCount=0;
+	remove();
+	computerName(mServerName);
+  try{pIRemoteProcess->Snapshot(&variant.getVARIANT());}
+	catch(...){return false;}
+	variant.getData(VTInt4,(void*)&processCount);
+	variant.clear();
+	if(!processCount)return false;
+	pIRemoteProcess->GetProcessFirst(&variant.getVARIANT());
+	variant.getData(VTInt4,&processID);
+	variant.clear();
+	for(int index=0;index<processCount;index++)
+	{
+    pIRemoteProcess->GetModuleFirst(&variant.getVARIANT());
+		if(VTInt4==variant.getType())variant.clear();
+		else
+		{
+		  BString string;
+			variant.getData(string);
+			ansiString=string.toString();
+			ansiString.upper();
+			strParent=ansiString;
+			insert(&ProcessItem());
+      ProcessItem &processItem=operator[](size()-1);
+			processItem.processName(ansiString);
+			processItem.processID(processID);
+			variant.clear();
+		  while(true)
+		  {
+        pIRemoteProcess->GetModuleNext(&variant.getVARIANT());
+		    if(VTInt4==variant.getType())
+		    {
+		      variant.clear();
+			    break;
+		    }
+		    else
+		    {
+		      BString string;
+			    variant.getData(string);
+					ansiString=string.toString();
+					ansiString.upper();
+          processItem.insert(&ansiString);
+			    variant.clear();
+		    }
+		  }
+		}
+		pIRemoteProcess->GetProcessNext(&variant.getVARIANT());
+		variant.getData(VTInt4,&processID);
+		variant.clear();
+	}
+	return size()?true:false;
+}
+
+bool RemoteProcessInfo::getDesktopWindow(IRemoteProcess *pIRemoteProcess,WindowView &windowView)
+{
+	String strFileName("image000.tmp");
+	String strPathFileName;
+  FileHandle outFile;
+  Variant variant;
+	SafeArray safeArray;
+	int upperBound;
+	BYTE *pData;
+	GlobalData<BYTE> imageBytes;
+	PathFind pathFind;
+
+  pathFind.getWindowsTempDirectory(strPathFileName);
+	strPathFileName+=strFileName;
+  pIRemoteProcess->GetDesktopWindow(&variant.getVARIANT());
+	if(VTArray!=variant.getType()){variant.clear();return false;}
+	if(!variant.getData(safeArray)){variant.clear();return false;}
+	upperBound=safeArray.getUpperBound();
+	safeArray.accessData((void**)&pData);
+	imageBytes.size(*((DWORD*)pData));
+	::memcpy(&imageBytes[0],pData+(sizeof(DWORD)),imageBytes.size());
+	outFile.open(strPathFileName,FileHandle::Write,FileHandle::ShareRead,FileHandle::Overwrite);
+	if(!outFile.isOkay())
+	{
+    safeArray.unaccessData();
+	  safeArray.destroy();
+	  variant.clear();
+		return false;
+	}
+	outFile.write(&imageBytes[0],imageBytes.size());
+	outFile.close();
+	windowView.createImage(strPathFileName);
+	::unlink(strPathFileName);
+  safeArray.unaccessData();
+	safeArray.destroy();
+	variant.clear();
+	return true;
+}