2009-09-13 18:02:46 -06:00
|
|
|
/*
|
|
|
|
www.sourceforge.net/projects/dfhack
|
|
|
|
Copyright (c) 2009 Petr Mrázek (peterix), Kenneth Ferland (Impaler[WrG]), dorf
|
|
|
|
|
|
|
|
This software is provided 'as-is', without any express or implied
|
|
|
|
warranty. In no event will the authors be held liable for any
|
|
|
|
damages arising from the use of this software.
|
|
|
|
|
|
|
|
Permission is granted to anyone to use this software for any
|
|
|
|
purpose, including commercial applications, and to alter it and
|
|
|
|
redistribute it freely, subject to the following restrictions:
|
|
|
|
|
|
|
|
1. The origin of this software must not be misrepresented; you must
|
|
|
|
not claim that you wrote the original software. If you use this
|
|
|
|
software in a product, an acknowledgment in the product documentation
|
|
|
|
would be appreciated but is not required.
|
|
|
|
|
|
|
|
2. Altered source versions must be plainly marked as such, and
|
|
|
|
must not be misrepresented as being the original software.
|
|
|
|
|
|
|
|
3. This notice may not be removed or altered from any source
|
|
|
|
distribution.
|
|
|
|
*/
|
|
|
|
|
2009-10-04 16:32:31 -06:00
|
|
|
#ifndef BUILD_DFHACK_LIB
|
|
|
|
# define BUILD_DFHACK_LIB
|
|
|
|
#endif
|
2009-10-04 07:08:20 -06:00
|
|
|
|
2009-09-13 18:02:46 -06:00
|
|
|
#include "DFCommon.h"
|
|
|
|
#ifdef LINUX_BUILD
|
|
|
|
#include <sys/wait.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
|
|
|
Process::Process(DataModel * dm, memory_info* mi, ProcessHandle ph, uint32_t pid)
|
|
|
|
{
|
|
|
|
my_datamodel = dm;
|
|
|
|
my_descriptor = mi;
|
2009-10-04 07:08:20 -06:00
|
|
|
my_handle = ph;
|
2009-09-13 18:02:46 -06:00
|
|
|
my_pid = pid;
|
|
|
|
attached = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
Process::~Process()
|
|
|
|
{
|
|
|
|
if(attached)
|
|
|
|
{
|
|
|
|
detach();
|
|
|
|
}
|
|
|
|
// destroy data model. this is assigned by processmanager
|
|
|
|
delete my_datamodel;
|
|
|
|
freeResources();
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
DataModel *Process::getDataModel()
|
|
|
|
{
|
|
|
|
return my_datamodel;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
memory_info * Process::getDescriptor()
|
|
|
|
{
|
|
|
|
return my_descriptor;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void Process::setMemFile(const string & memf)
|
|
|
|
{
|
|
|
|
assert(!attached);
|
|
|
|
memFile = memf;
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef LINUX_BUILD
|
|
|
|
/*
|
|
|
|
* LINUX PART
|
|
|
|
*/
|
2009-10-29 08:39:40 -06:00
|
|
|
|
2009-10-31 12:09:13 -06:00
|
|
|
|
|
|
|
//TODO: rewrite this. really. It's ugly as hell.
|
|
|
|
bool isStopped(pid_t pid)
|
|
|
|
{
|
|
|
|
char filename[256];
|
|
|
|
sprintf(filename, "/proc/%d/status", pid);
|
|
|
|
|
|
|
|
// evil mess, that's a fitting name for this thing
|
|
|
|
FILE* evil = fopen(filename,"rb");
|
|
|
|
if(evil)
|
|
|
|
{
|
|
|
|
// zlo means evil in czech.
|
2009-11-01 02:32:47 -07:00
|
|
|
char zlo[64];
|
2009-10-31 12:09:13 -06:00
|
|
|
char test;
|
|
|
|
// read first line, ignore
|
2009-11-01 02:32:47 -07:00
|
|
|
fgets(zlo,64,evil);
|
2009-10-31 12:09:13 -06:00
|
|
|
// read second line
|
2009-11-01 02:32:47 -07:00
|
|
|
fgets(zlo,64,evil);
|
|
|
|
sscanf(zlo,"State: %c",&test);
|
2009-10-31 12:09:13 -06:00
|
|
|
fclose(evil);
|
|
|
|
if(test == 'T')
|
|
|
|
{
|
2009-11-01 02:32:47 -07:00
|
|
|
return true;
|
2009-10-31 12:09:13 -06:00
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
cerr << "couldn't open file " << filename << "assuming process stopped" << endl;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
2009-10-29 08:39:40 -06:00
|
|
|
// shamelessly stolen from dwarf therapist code. credit should go to the author there :)
|
2009-09-13 18:02:46 -06:00
|
|
|
bool Process::attach()
|
|
|
|
{
|
2009-10-31 12:09:13 -06:00
|
|
|
int status;
|
|
|
|
cout << "Attach: start" << endl;
|
2009-10-29 08:39:40 -06:00
|
|
|
// check if another process is attached
|
2009-09-13 18:02:46 -06:00
|
|
|
if(g_pProcess != NULL)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2009-10-31 12:09:13 -06:00
|
|
|
if(!isStopped(my_handle))
|
|
|
|
{
|
|
|
|
kill(my_handle,SIGSTOP);
|
2009-11-01 02:58:49 -07:00
|
|
|
cout << "sent SIGSTOP" << endl;
|
|
|
|
while (!isStopped(my_handle))
|
|
|
|
{
|
|
|
|
cout << "wait step" << endl;
|
|
|
|
}
|
2009-10-31 12:09:13 -06:00
|
|
|
}
|
2009-11-01 02:32:47 -07:00
|
|
|
usleep(10000);
|
2009-10-31 12:09:13 -06:00
|
|
|
cout << "Attach: after conditional stop" << endl;
|
2009-10-29 08:39:40 -06:00
|
|
|
// can we attach?
|
|
|
|
if (ptrace(PTRACE_ATTACH , my_handle, NULL, NULL) == -1)
|
2009-10-29 09:37:06 -06:00
|
|
|
{
|
2009-10-29 08:39:40 -06:00
|
|
|
// no, we got an error
|
|
|
|
perror("ptrace attach error");
|
2009-10-31 12:09:13 -06:00
|
|
|
cerr << "attach failed on pid " << my_handle << endl;
|
2009-10-29 08:39:40 -06:00
|
|
|
return false;
|
|
|
|
}
|
2009-11-01 02:32:47 -07:00
|
|
|
usleep(10000);
|
2009-10-31 12:09:13 -06:00
|
|
|
cout << "Attach: after ptrace" << endl;
|
|
|
|
/*
|
2009-10-29 08:39:40 -06:00
|
|
|
while(true)
|
|
|
|
{
|
|
|
|
// we wait on the pid
|
|
|
|
pid_t w = waitpid(my_handle, &status, 0);
|
|
|
|
if (w == -1)
|
|
|
|
{
|
|
|
|
// child died
|
|
|
|
perror("wait inside attach()");
|
|
|
|
//LOGC << "child died?";
|
|
|
|
//exit(-1);
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
// stopped -> let's continue
|
|
|
|
if (WIFSTOPPED(status))
|
|
|
|
{
|
|
|
|
break;
|
|
|
|
}
|
2009-10-31 12:09:13 -06:00
|
|
|
}*/
|
|
|
|
// cout << "Managed to attach to pid " << my_handle << endl;
|
2009-10-29 08:39:40 -06:00
|
|
|
|
2009-10-29 11:40:39 -06:00
|
|
|
int proc_pid_mem = open(memFile.c_str(),O_RDONLY);
|
|
|
|
if(proc_pid_mem == -1)
|
|
|
|
{
|
|
|
|
ptrace(PTRACE_DETACH, my_handle, NULL, NULL);
|
|
|
|
cerr << "couldn't open /proc/" << my_handle << "/mem" << endl;
|
|
|
|
perror("open(memFile.c_str(),O_RDONLY)");
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
attached = true;
|
|
|
|
g_pProcess = this;
|
|
|
|
g_ProcessHandle = my_handle;
|
|
|
|
|
|
|
|
g_ProcessMemFile = proc_pid_mem;
|
2009-10-31 12:09:13 -06:00
|
|
|
cout << "Attach: after opening /proc/"<< my_handle <<"/mem" << endl;
|
2009-10-29 11:40:39 -06:00
|
|
|
return true; // we are attached
|
|
|
|
}
|
2009-09-13 18:02:46 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
bool Process::detach()
|
|
|
|
{
|
2009-11-01 02:32:47 -07:00
|
|
|
if(!attached) return false;
|
2009-10-29 11:52:40 -06:00
|
|
|
int result = 0;
|
|
|
|
cout << "detach: start" << endl;
|
|
|
|
result = close(g_ProcessMemFile);// close /proc/PID/mem
|
|
|
|
if(result == -1)
|
|
|
|
{
|
|
|
|
cerr << "couldn't close /proc/"<< my_handle <<"/mem" << endl;
|
2009-10-29 11:54:02 -06:00
|
|
|
perror("mem file close");
|
2009-10-29 11:52:40 -06:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
cout << "detach: after closing /proc/"<< my_handle <<"/mem" << endl;
|
|
|
|
g_ProcessMemFile = -1;
|
|
|
|
result = ptrace(PTRACE_DETACH, my_handle, NULL, NULL);
|
|
|
|
if(result == -1)
|
|
|
|
{
|
|
|
|
cerr << "couldn't detach from process pid" << my_handle << endl;
|
2009-10-29 11:54:02 -06:00
|
|
|
perror("ptrace detach");
|
2009-10-29 11:52:40 -06:00
|
|
|
return false;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
2009-10-31 12:09:13 -06:00
|
|
|
// cout << "detach: after detaching from "<< my_handle << endl;
|
2009-10-29 11:52:40 -06:00
|
|
|
attached = false;
|
|
|
|
g_pProcess = NULL;
|
|
|
|
g_ProcessHandle = 0;
|
2009-10-31 12:09:13 -06:00
|
|
|
// continue, wait for it to recover
|
|
|
|
kill(my_handle,SIGCONT);
|
|
|
|
while (isStopped(my_handle));
|
2009-11-01 02:32:47 -07:00
|
|
|
usleep(10000);
|
2009-10-31 12:09:13 -06:00
|
|
|
// we finish
|
2009-10-29 11:52:40 -06:00
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
2009-09-13 18:02:46 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void Process::freeResources()
|
|
|
|
{
|
|
|
|
// nil
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
#else
|
|
|
|
/*
|
|
|
|
* WINDOWS PART
|
|
|
|
*/
|
|
|
|
|
|
|
|
//FIXME: should support stopping and resuming the process
|
|
|
|
|
|
|
|
bool Process::attach()
|
2009-10-04 07:08:20 -06:00
|
|
|
{
|
|
|
|
if(attached)
|
|
|
|
{
|
|
|
|
return false;
|
|
|
|
}
|
2009-09-14 14:47:20 -06:00
|
|
|
/*
|
2009-09-13 18:02:46 -06:00
|
|
|
if(DebugActiveProcess(my_pid))
|
2009-10-04 07:08:20 -06:00
|
|
|
{
|
2009-09-14 14:47:20 -06:00
|
|
|
*/
|
2009-09-13 18:02:46 -06:00
|
|
|
attached = true;
|
|
|
|
g_pProcess = this;
|
|
|
|
g_ProcessHandle = my_handle;
|
|
|
|
|
|
|
|
return true;
|
2009-09-14 14:47:20 -06:00
|
|
|
/*}
|
|
|
|
return false;*/
|
2009-09-13 18:02:46 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
bool Process::detach()
|
|
|
|
{
|
|
|
|
if(!attached)
|
|
|
|
{
|
|
|
|
return false;
|
2009-10-04 07:08:20 -06:00
|
|
|
}
|
2009-09-14 14:47:20 -06:00
|
|
|
//TODO: find a safer way to suspend processes
|
|
|
|
/*if(DebugActiveProcessStop(my_pid))
|
|
|
|
{*/
|
2009-09-13 18:02:46 -06:00
|
|
|
attached = false;
|
|
|
|
g_pProcess = NULL;
|
|
|
|
g_ProcessHandle = 0;
|
|
|
|
return true;
|
2009-09-14 14:47:20 -06:00
|
|
|
/*}
|
|
|
|
return false;*/
|
2009-09-13 18:02:46 -06:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void Process::freeResources()
|
|
|
|
{
|
|
|
|
// opened by process manager
|
|
|
|
CloseHandle(my_handle);
|
|
|
|
};
|
|
|
|
#endif
|
|
|
|
|