2011-12-24 03:51:58 -07:00
|
|
|
/*
|
|
|
|
https://github.com/peterix/dfhack
|
|
|
|
Copyright (c) 2009-2011 Petr Mrázek (peterix@gmail.com)
|
|
|
|
|
|
|
|
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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "Internal.h"
|
|
|
|
|
|
|
|
#include <string>
|
|
|
|
#include <vector>
|
|
|
|
#include <map>
|
|
|
|
|
2011-12-31 04:48:42 -07:00
|
|
|
#include "MemAccess.h"
|
|
|
|
#include "Core.h"
|
|
|
|
#include "VersionInfo.h"
|
2011-12-24 03:51:58 -07:00
|
|
|
#include "tinythread.h"
|
2011-12-24 06:27:35 -07:00
|
|
|
// must be last due to MS stupidity
|
2011-12-31 04:48:42 -07:00
|
|
|
#include "DataDefs.h"
|
2011-12-24 03:51:58 -07:00
|
|
|
|
|
|
|
using namespace DFHack;
|
|
|
|
|
|
|
|
/* The order of global object constructor calls is
|
|
|
|
* undefined between compilation units. Therefore,
|
|
|
|
* this list has to be plain data, so that it gets
|
|
|
|
* initialized by the loader in the initial mmap.
|
|
|
|
*/
|
|
|
|
virtual_identity *virtual_identity::list = NULL;
|
|
|
|
|
|
|
|
virtual_identity::virtual_identity(const char *dfhack_name, const char *original_name, virtual_identity *parent)
|
|
|
|
: dfhack_name(dfhack_name), original_name(original_name), parent(parent),
|
|
|
|
prev(NULL), vtable_ptr(NULL), has_children(true)
|
|
|
|
{
|
|
|
|
// Link into the static list. Nothing else can be safely done at this point.
|
|
|
|
next = list; list = this;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Vtable to identity lookup. */
|
|
|
|
static tthread::mutex *known_mutex = NULL;
|
|
|
|
std::map<void*, virtual_identity*> virtual_identity::known;
|
|
|
|
|
2011-12-24 06:17:01 -07:00
|
|
|
virtual_identity *virtual_identity::get(virtual_ptr instance_ptr)
|
2011-12-24 03:51:58 -07:00
|
|
|
{
|
|
|
|
if (!instance_ptr) return NULL;
|
|
|
|
|
|
|
|
// Actually, a reader/writer lock would be sufficient,
|
|
|
|
// since the table is only written once per class.
|
|
|
|
tthread::lock_guard<tthread::mutex> lock(*known_mutex);
|
|
|
|
|
|
|
|
void *vtable = get_vtable(instance_ptr);
|
|
|
|
std::map<void*, virtual_identity*>::iterator it = known.find(vtable);
|
|
|
|
|
|
|
|
if (it != known.end())
|
|
|
|
return it->second;
|
|
|
|
|
|
|
|
// If using a reader/writer lock, re-grab as write here, and recheck
|
|
|
|
Core &core = Core::getInstance();
|
|
|
|
std::string name = core.p->doReadClassName(vtable);
|
|
|
|
|
|
|
|
virtual_identity *actual = NULL;
|
|
|
|
|
|
|
|
for (virtual_identity *p = list; p; p = p->next) {
|
|
|
|
if (strcmp(name.c_str(), p->getOriginalName()) != 0) continue;
|
2011-12-29 05:30:55 -07:00
|
|
|
|
|
|
|
if (p->vtable_ptr && p->vtable_ptr != vtable) {
|
|
|
|
std::cerr << "Conflicting vtable ptr for class '" << p->getName()
|
|
|
|
<< "': found 0x" << std::hex << unsigned(vtable)
|
|
|
|
<< ", previous 0x" << unsigned(p->vtable_ptr) << std::dec << std::endl;
|
|
|
|
abort();
|
|
|
|
} else if (!p->vtable_ptr) {
|
|
|
|
std::cerr << "class '" << p->getName() << "': vtable = 0x"
|
|
|
|
<< std::hex << unsigned(vtable) << std::dec << std::endl;
|
|
|
|
}
|
|
|
|
|
2011-12-24 03:51:58 -07:00
|
|
|
known[vtable] = p;
|
|
|
|
p->vtable_ptr = vtable;
|
|
|
|
return p;
|
|
|
|
}
|
|
|
|
|
2011-12-29 05:30:55 -07:00
|
|
|
std::cerr << "UNKNOWN CLASS '" << name << "': vtable = 0x"
|
|
|
|
<< std::hex << unsigned(vtable) << std::dec << std::endl;
|
|
|
|
|
2011-12-24 03:51:58 -07:00
|
|
|
known[vtable] = NULL;
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2011-12-29 05:30:55 -07:00
|
|
|
bool virtual_identity::is_subclass(virtual_identity *actual)
|
2011-12-24 03:51:58 -07:00
|
|
|
{
|
2011-12-29 05:30:55 -07:00
|
|
|
for (; actual; actual = actual->parent)
|
2011-12-24 03:51:58 -07:00
|
|
|
if (actual == this) return true;
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2011-12-29 05:30:55 -07:00
|
|
|
void virtual_identity::adjust_vtable(virtual_ptr obj, virtual_identity *main)
|
|
|
|
{
|
|
|
|
if (vtable_ptr) {
|
|
|
|
*(void**)obj = vtable_ptr;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (main && main != this && is_subclass(main))
|
|
|
|
return;
|
|
|
|
|
|
|
|
std::cerr << "Attempt to create class '" << getName() << "' without known vtable." << std::endl;
|
|
|
|
abort();
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual_ptr virtual_identity::clone(virtual_ptr obj)
|
|
|
|
{
|
|
|
|
virtual_identity *id = get(obj);
|
|
|
|
if (!id) return NULL;
|
|
|
|
virtual_ptr copy = id->instantiate();
|
|
|
|
if (!copy) return NULL;
|
|
|
|
id->do_copy(copy, obj);
|
|
|
|
return copy;
|
|
|
|
}
|
|
|
|
|
|
|
|
void virtual_identity::Init(Core *core)
|
2011-12-24 03:51:58 -07:00
|
|
|
{
|
|
|
|
if (!known_mutex)
|
|
|
|
known_mutex = new tthread::mutex();
|
|
|
|
|
|
|
|
// This cannot be done in the constructors, because
|
|
|
|
// they are called in an undefined order.
|
|
|
|
for (virtual_identity *p = list; p; p = p->next) {
|
|
|
|
p->has_children = false;
|
|
|
|
p->children.clear();
|
|
|
|
}
|
|
|
|
for (virtual_identity *p = list; p; p = p->next) {
|
|
|
|
if (p->parent) {
|
|
|
|
p->parent->children.push_back(p);
|
|
|
|
p->parent->has_children = true;
|
|
|
|
}
|
|
|
|
}
|
2011-12-29 05:30:55 -07:00
|
|
|
|
|
|
|
// Read pre-filled vtable ptrs
|
|
|
|
OffsetGroup *ptr_table = core->vinfo->getGroup("vtable");
|
|
|
|
for (virtual_identity *p = list; p; p = p->next) {
|
2012-01-03 17:45:11 -07:00
|
|
|
void * tmp;
|
2011-12-29 05:30:55 -07:00
|
|
|
if (ptr_table->getSafeAddress(p->getName(),tmp))
|
2012-01-03 17:45:11 -07:00
|
|
|
p->vtable_ptr = tmp;
|
2011-12-29 05:30:55 -07:00
|
|
|
}
|
2011-12-24 03:51:58 -07:00
|
|
|
}
|
|
|
|
|
2012-01-01 12:05:45 -07:00
|
|
|
#define SIMPLE_GLOBAL(name,tname) \
|
|
|
|
tname *df::global::name = NULL;
|
|
|
|
#define GLOBAL(name,tname) SIMPLE_GLOBAL(name,df::tname)
|
2011-12-24 03:51:58 -07:00
|
|
|
DF_KNOWN_GLOBALS
|
|
|
|
#undef GLOBAL
|
2012-01-01 12:05:45 -07:00
|
|
|
#undef SIMPLE_GLOBAL
|
2011-12-24 03:51:58 -07:00
|
|
|
|
|
|
|
void DFHack::InitDataDefGlobals(Core *core) {
|
|
|
|
OffsetGroup *global_table = core->vinfo->getGroup("global");
|
2012-01-03 17:45:11 -07:00
|
|
|
void * tmp;
|
2011-12-24 03:51:58 -07:00
|
|
|
|
2012-01-01 12:05:45 -07:00
|
|
|
#define SIMPLE_GLOBAL(name,tname) \
|
|
|
|
if (global_table->getSafeAddress(#name,tmp)) df::global::name = (tname*)tmp;
|
|
|
|
#define GLOBAL(name,tname) SIMPLE_GLOBAL(name,df::tname)
|
2011-12-24 03:51:58 -07:00
|
|
|
DF_KNOWN_GLOBALS
|
|
|
|
#undef GLOBAL
|
2012-01-01 12:05:45 -07:00
|
|
|
#undef SIMPLE_GLOBAL
|
2011-12-24 03:51:58 -07:00
|
|
|
}
|