AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Parse ELF format for rocprofiler_configure symbol (#970)

Browse Source 
* Parse ELF format to search for rocprofiler_configure

* Use ELF parsing in registration
This commit is contained in:
Jonathan R. Madsen
2024-07-11 20:22:26 -05:00
committed by GitHub
parent 8b1b074b2a
commit 2be3543c7b
9 changed files with 403 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/lib/rocprofiler-sdk/registration.cpp
+28 -3
View File
@@ -23,7 +23,9 @@
#define _GNU_SOURCE 1
#include "lib/rocprofiler-sdk/registration.hpp"
#include "lib/common/elf_utils.hpp"
#include "lib/common/environment.hpp"
#include "lib/common/filesystem.hpp"
#include "lib/common/logging.hpp"
#include "lib/common/static_object.hpp"
#include "lib/rocprofiler-sdk/agent.hpp"
@@ -88,6 +90,8 @@ namespace registration
{
namespace
{
namespace fs = ::rocprofiler::common::filesystem;
// invoke all rocprofiler_configure symbols
bool
invoke_client_configures();
@@ -129,8 +133,7 @@ std::vector<std::string>
get_link_map()
{
auto chain = std::vector<std::string>{};
void* handle = nullptr;
handle = dlopen(nullptr, RTLD_LAZY | RTLD_NOLOAD);
void* handle = dlopen(nullptr, RTLD_LAZY | RTLD_NOLOAD);
if(handle)
{
@@ -245,6 +248,16 @@ find_clients()
{
ROCP_INFO << "[env] searching " << itr << " for rocprofiler_configure";
if(fs::exists(itr))
{
auto elfinfo = common::elf_utils::read(itr);
if(!elfinfo.has_symbol(std::regex{"^rocprofiler_configure$"}))
{
ROCP_FATAL << "rocprofiler tool library " << itr
<< " did not contain rocprofiler_configure symbol";
}
}
void* handle = dlopen(itr.c_str(), RTLD_NOLOAD | RTLD_LAZY);
if(!handle)
@@ -294,12 +307,24 @@ find_clients()
// if there are two "rocprofiler_configures", we need to trigger a search of all the shared
// libraries
if(_next_configure)
if(_default_configure)
{
for(const auto& itr : get_link_map())
{
ROCP_INFO << "searching " << itr << " for rocprofiler_configure";
if(fs::exists(itr))
{
auto elfinfo = common::elf_utils::read(itr);
if(!elfinfo.has_symbol(std::regex{"^rocprofiler_configure$"})) continue;
}
else
{
continue;
}
ROCP_INFO << "dlopening " << itr << " for rocprofiler_configure";
void* handle = dlopen(itr.c_str(), RTLD_LAZY | RTLD_NOLOAD);
ROCP_ERROR_IF(handle == nullptr) << "error dlopening " << itr;