2011-06-12 15:17:40 -06:00
|
|
|
/*
|
2011-06-16 15:53:39 -06:00
|
|
|
https://github.com/peterix/dfhack
|
|
|
|
Copyright (c) 2009-2011 Petr Mrázek (peterix@gmail.com)
|
2011-06-12 15:17:40 -06:00
|
|
|
|
|
|
|
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.
|
|
|
|
*/
|
2011-06-16 15:53:39 -06:00
|
|
|
|
2011-06-12 15:17:40 -06:00
|
|
|
#include "Internal.h"
|
2011-06-19 17:12:07 -06:00
|
|
|
#include <dirent.h>
|
|
|
|
#include <errno.h>
|
2011-06-12 15:17:40 -06:00
|
|
|
|
|
|
|
#include <string>
|
|
|
|
#include <vector>
|
|
|
|
#include <map>
|
|
|
|
#include <set>
|
|
|
|
#include <cstdio>
|
|
|
|
#include <cstring>
|
|
|
|
using namespace std;
|
|
|
|
|
2011-06-19 17:12:07 -06:00
|
|
|
#include <md5wrapper.h>
|
2011-06-12 15:17:40 -06:00
|
|
|
#include "dfhack/Process.h"
|
|
|
|
#include "dfhack/VersionInfoFactory.h"
|
|
|
|
#include "dfhack/VersionInfo.h"
|
|
|
|
#include "dfhack/Error.h"
|
|
|
|
using namespace DFHack;
|
|
|
|
|
|
|
|
Process::Process(VersionInfoFactory * known_versions)
|
|
|
|
{
|
|
|
|
const char * dir_name = "/proc/self/";
|
|
|
|
const char * exe_link_name = "/proc/self/exe";
|
|
|
|
const char * cwd_name = "/proc/self/cwd";
|
|
|
|
const char * cmdline_name = "/proc/self/cmdline";
|
|
|
|
char target_name[1024];
|
|
|
|
int target_result;
|
|
|
|
|
|
|
|
identified = false;
|
|
|
|
my_descriptor = 0;
|
|
|
|
|
|
|
|
// resolve /proc/self/exe link
|
|
|
|
target_result = readlink(exe_link_name, target_name, sizeof(target_name)-1);
|
|
|
|
if (target_result == -1)
|
|
|
|
{
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
// make sure we have a null terminated string...
|
|
|
|
target_name[target_result] = 0;
|
|
|
|
|
|
|
|
// is this the regular linux DF?
|
|
|
|
if (strstr(target_name, "dwarfort.exe") != 0 || strstr(target_name,"Dwarf_Fortress") != 0)
|
|
|
|
{
|
|
|
|
md5wrapper md5;
|
|
|
|
// get hash of the running DF process
|
|
|
|
string hash = md5.getHashFromFile(target_name);
|
|
|
|
// create linux process, add it to the vector
|
|
|
|
VersionInfo * vinfo = known_versions->getVersionInfoByMD5(hash);
|
|
|
|
if(vinfo)
|
|
|
|
{
|
|
|
|
my_descriptor = new VersionInfo(*vinfo);
|
|
|
|
my_descriptor->setParentProcess(this);
|
|
|
|
identified = true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
string Process::doReadClassName (uint32_t vptr)
|
|
|
|
{
|
|
|
|
int typeinfo = Process::readDWord(vptr - 0x4);
|
|
|
|
int typestring = Process::readDWord(typeinfo + 0x4);
|
|
|
|
string raw = readCString(typestring);
|
|
|
|
size_t start = raw.find_first_of("abcdefghijklmnopqrstuvwxyz");// trim numbers
|
|
|
|
size_t end = raw.length();
|
|
|
|
return raw.substr(start,end-start);
|
|
|
|
}
|
|
|
|
|
|
|
|
//FIXME: cross-reference with ELF segment entries?
|
|
|
|
void Process::getMemRanges( vector<t_memrange> & ranges )
|
|
|
|
{
|
|
|
|
char buffer[1024];
|
|
|
|
char permissions[5]; // r/-, w/-, x/-, p/s, 0
|
|
|
|
|
|
|
|
FILE *mapFile = ::fopen("/proc/self/maps", "r");
|
|
|
|
size_t start, end, offset, device1, device2, node;
|
|
|
|
|
|
|
|
while (fgets(buffer, 1024, mapFile))
|
|
|
|
{
|
|
|
|
t_memrange temp;
|
|
|
|
temp.name[0] = 0;
|
|
|
|
sscanf(buffer, "%zx-%zx %s %zx %2zu:%2zu %zu %[^\n]s",
|
|
|
|
&start,
|
|
|
|
&end,
|
|
|
|
(char*)&permissions,
|
|
|
|
&offset, &device1, &device2, &node,
|
|
|
|
(char*)&temp.name);
|
|
|
|
temp.start = start;
|
|
|
|
temp.end = end;
|
|
|
|
temp.read = permissions[0] == 'r';
|
|
|
|
temp.write = permissions[1] == 'w';
|
|
|
|
temp.execute = permissions[2] == 'x';
|
|
|
|
temp.shared = permissions[3] == 's';
|
|
|
|
temp.valid = true;
|
|
|
|
ranges.push_back(temp);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-06-19 20:29:38 -06:00
|
|
|
static int getdir (string dir, vector<string> &files)
|
2011-06-12 15:17:40 -06:00
|
|
|
{
|
|
|
|
DIR *dp;
|
|
|
|
struct dirent *dirp;
|
|
|
|
if((dp = opendir(dir.c_str())) == NULL)
|
|
|
|
{
|
|
|
|
cout << "Error(" << errno << ") opening " << dir << endl;
|
|
|
|
return errno;
|
|
|
|
}
|
|
|
|
while ((dirp = readdir(dp)) != NULL) {
|
|
|
|
files.push_back(string(dirp->d_name));
|
|
|
|
}
|
|
|
|
closedir(dp);
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Process::getThreadIDs(vector<uint32_t> & threads )
|
|
|
|
{
|
|
|
|
stringstream ss;
|
|
|
|
vector<string> subdirs;
|
|
|
|
if(getdir("/proc/self/task/",subdirs) != 0)
|
|
|
|
{
|
|
|
|
//FIXME: needs exceptions. this is a fatal error
|
|
|
|
cerr << "unable to enumerate threads. This is BAD!" << endl;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
threads.clear();
|
|
|
|
for(size_t i = 0; i < subdirs.size();i++)
|
|
|
|
{
|
|
|
|
uint32_t tid;
|
|
|
|
if(sscanf(subdirs[i].c_str(),"%d", &tid))
|
|
|
|
{
|
|
|
|
threads.push_back(tid);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
string Process::getPath()
|
|
|
|
{
|
|
|
|
const char * cwd_name = "/proc/self/cwd";
|
|
|
|
char target_name[1024];
|
|
|
|
int target_result;
|
|
|
|
target_result = readlink(cwd_name, target_name, sizeof(target_name));
|
|
|
|
target_name[target_result] = '\0';
|
|
|
|
return(string(target_name));
|
|
|
|
}
|
|
|
|
|
|
|
|
int Process::getPID()
|
|
|
|
{
|
|
|
|
return getpid();
|
|
|
|
}
|