oneechanhax/crackem
Archived
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed broken lib

Browse Source
This commit is contained in:
Rebekah 2019-06-21 13:47:34 -04:00
parent 637175c4d3
commit 5f88edcc5b
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/sharedlibrary.cpp
View File

@ -43,7 +43,7 @@ void SharedLibrary::Init(std::string_view _name){
if (!init_flag) { if (!init_flag) {
this->name = _name; this->name = _name;
if (this->name.empty()) if (this->name.empty())
throw std::logic_error("SharedLibrary: no name avaiable."); throw std::runtime_error("SharedLibrary: no name avaiable.");
Passthrough pass; Passthrough pass;
pass.name = '/' + std::string(_name) + '.'; pass.name = '/' + std::string(_name) + '.';
if (!dl_iterate_phdr([](struct dl_phdr_info* info, size_t, void* _pass) { if (!dl_iterate_phdr([](struct dl_phdr_info* info, size_t, void* _pass) {
@ -58,21 +58,23 @@ void SharedLibrary::Init(std::string_view _name){
}, reinterpret_cast<void*>(&pass))) }, reinterpret_cast<void*>(&pass)))
throw std::runtime_error("SharedLibrary: Unable to find loaded library"); throw std::runtime_error("SharedLibrary: Unable to find loaded library");
this->lmap = static_cast<LMap>(
dlopen(path.c_str(), RTLD_NOLOAD | RTLD_NOW | RTLD_LOCAL));
if (const char* error = dlerror())
throw std::runtime_error("SharedLibrary recieved dlerror: + error");
this->path = pass.path; this->path = pass.path;
this->_begin = pass.begin; this->_begin = pass.begin;
this->_end = mem::Offset<void*>(pass.begin, pass.size); this->_end = mem::Offset<void*>(pass.begin, pass.size);
this->_size = pass.size; this->_size = pass.size;
this->lmap = static_cast<LMap>(dlopen(this->path.c_str(), RTLD_NOLOAD));
if (const char* error = dlerror())
throw std::runtime_error("SharedLibrary recieved dlerror: " + std::string(error));
this->init_flag = true; this->init_flag = true;
} }
} }
void SharedLibrary::ForceInit(std::string_view _name) { void SharedLibrary::ForceInit(std::string_view _name) {
while(!this->init_flag) { while(!this->init_flag) {
this->Init(); try {
this->Init();
} catch(...) {}
} }
} }
std::string_view SharedLibrary::GetName() { std::string_view SharedLibrary::GetName() {
@ -100,7 +102,10 @@ void SharedLibrary::Clear(){
} }
void* SharedLibrary::GetSymInternal(SymStr name) { void* SharedLibrary::GetSymInternal(SymStr name) {
#if defined(__linux__) #if defined(__linux__)
return dlsym(this->GetLMap(), name); void* symbol = dlsym(this->GetLMap(), name);
if (const char* error = dlerror())
throw std::runtime_error("SharedLibrary recieved dlerror: " + std::string(error));
return symbol;
#elif defined(_WIN32) #elif defined(_WIN32)
return GetProcAddress(this->GetLMap(), name); return GetProcAddress(this->GetLMap(), name);
#endif #endif