147 lines
4.3 KiB
C++
147 lines
4.3 KiB
C++
// Demonstrates the use of ProcessEnumerator
|
|
// queries the Enumerator for all DF Processes on user input. Prints them to the terminal.
|
|
// also tracks processes that were invalidated
|
|
|
|
#include <iostream>
|
|
#include <climits>
|
|
#include <vector>
|
|
#include <ctime>
|
|
using namespace std;
|
|
|
|
#include <DFHack.h>
|
|
#include <dfhack/DFProcessEnumerator.h>
|
|
using namespace DFHack;
|
|
#ifndef LINUX_BUILD
|
|
#endif
|
|
|
|
void printhelp ()
|
|
{
|
|
cout << "enter empty line for next try." << endl;
|
|
cout << "enter 'next' or 'n' for next test." << endl;
|
|
cout << "enter 'help' to show this text again." << endl;
|
|
}
|
|
|
|
int inputwait (const char * prompt)
|
|
{
|
|
inputwait_reset:
|
|
string command = "";
|
|
cout <<"[" << prompt << "]# ";
|
|
getline(cin, command);
|
|
if(command == "help")
|
|
{
|
|
printhelp();
|
|
goto inputwait_reset;
|
|
}
|
|
else if(command == "")
|
|
{
|
|
return 1;
|
|
}
|
|
else if(command == "next")
|
|
{
|
|
return 0;
|
|
}
|
|
else
|
|
{
|
|
cout << "Command not recognized. Try 'help' for a list of valid commands." << endl;
|
|
goto inputwait_reset;
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
int main (void)
|
|
{
|
|
printhelp();
|
|
cout << endl;
|
|
// first test ProcessEnumerator and BadProcesses
|
|
{
|
|
cout << "Testing ProcessEnumerator" << endl;
|
|
ProcessEnumerator Penum("Memory.xml");
|
|
memory_info * mem;
|
|
do
|
|
{
|
|
// make the ProcessEnumerator update its list of Processes
|
|
// by passing the pointer to 'inval', we make it export expired
|
|
// processes instead of destroying them outright
|
|
// (processes expire when the OS kills them for whatever reason)
|
|
BadProcesses inval;
|
|
Penum.Refresh(&inval);
|
|
int nProc = Penum.size();
|
|
int nInval = inval.size();
|
|
|
|
cout << "Processes:" << endl;
|
|
for(int i = 0; i < nProc; i++)
|
|
{
|
|
mem = Penum[i]->getDescriptor();
|
|
cout << "DF instance: " << Penum[i]->getPID()
|
|
<< ", " << mem->getVersion() << endl;
|
|
}
|
|
|
|
cout << "Invalidated:" << endl;
|
|
for(int i = 0; i < nInval; i++)
|
|
{
|
|
mem = inval[i]->getDescriptor();
|
|
cout << "DF instance: " << inval[i]->getPID()
|
|
<< ", " << mem->getVersion() << endl;
|
|
}
|
|
}
|
|
while(inputwait("ProcessEnumerator"));
|
|
}
|
|
// next test ContextManager and BadContexts
|
|
{
|
|
cout << "Testing ContextManager" << endl;
|
|
ContextManager Cman("Memory.xml");
|
|
memory_info * mem;
|
|
do
|
|
{
|
|
// make the ContextManager update its list of Contexts
|
|
// by passing the pointer to 'inval', we make it export expired
|
|
// contexts instead of destroying them outright
|
|
// (contexts expire when the OS kills their process for whatever
|
|
// reason)
|
|
BadContexts inval;
|
|
Cman.Refresh(&inval);
|
|
int nCont = Cman.size();
|
|
int nInval = inval.size();
|
|
DFHack::Context * cont = Cman.getSingleContext();
|
|
if(cont)
|
|
{
|
|
if(cont->Attach())
|
|
{
|
|
DFHack::Maps * mapz = cont->getMaps();
|
|
cont->Suspend();
|
|
mapz->Start();
|
|
cont->Resume();
|
|
}
|
|
|
|
bool result = cont->Detach();
|
|
if(!result)
|
|
{
|
|
cerr << "Something went horribly wrong during detach" << endl;
|
|
}
|
|
}
|
|
|
|
cout << "Contexts:" << endl;
|
|
for(int i = 0; i < nCont; i++)
|
|
{
|
|
mem = Cman[i]->getMemoryInfo();
|
|
cout << "DF instance: " << Cman[i]->getProcess()->getPID()
|
|
<< ", " << mem->getVersion() << endl;
|
|
}
|
|
|
|
cout << "Invalidated:" << endl;
|
|
for(int i = 0; i < nInval; i++)
|
|
{
|
|
mem = inval[i]->getMemoryInfo();
|
|
cout << "DF instance: " << inval[i]->getProcess()->getPID()
|
|
<< ", " << mem->getVersion() << endl;
|
|
}
|
|
}
|
|
while(inputwait("ContextManager"));
|
|
}
|
|
#ifndef LINUX_BUILD
|
|
cout << "Done. Press any key to continue" << endl;
|
|
cin.ignore();
|
|
#endif
|
|
return 0;
|
|
}
|