309 lines
6.8 KiB
C++
309 lines
6.8 KiB
C++
//| Copyright: (C) 2020-2024 Kevin Larke <contact AT larke DOT org>
|
|
//| License: GNU GPL version 3.0 or above. See the accompanying LICENSE file.
|
|
#include "cwCommon.h"
|
|
#include "cwLog.h"
|
|
#include "cwCommonImpl.h"
|
|
#include "cwTest.h"
|
|
#include "cwObject.h"
|
|
#include "cwMem.h"
|
|
#include "cwFileSys.h"
|
|
#include "cwLib.h"
|
|
|
|
#ifdef OS_LINUX
|
|
|
|
#include <dlfcn.h>
|
|
|
|
typedef void* dynLibH_t;
|
|
|
|
bool _cmLibIsNull( dynLibH_t lh )
|
|
{ return lh == nullptr; };
|
|
|
|
const char* _cmLibSysError()
|
|
{
|
|
const char* msg = dlerror();
|
|
if( msg == nullptr )
|
|
msg = "<none>";
|
|
return msg;
|
|
}
|
|
|
|
dynLibH_t _cmLibOpen( const char* libFn )
|
|
{ return dlopen(libFn,RTLD_LAZY); }
|
|
|
|
const char* _cmLibClose( dynLibH_t* lH )
|
|
{
|
|
if( *lH != nullptr )
|
|
{
|
|
if( dlclose(*lH) == 0 )
|
|
*lH = nullptr;
|
|
else
|
|
return dlerror();
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
void* _cmLibSym( dynLibH_t h, const char* symLabel )
|
|
{ return dlsym(h,symLabel); }
|
|
|
|
|
|
#endif
|
|
|
|
|
|
namespace cw
|
|
{
|
|
namespace lib
|
|
{
|
|
typedef void* dynLibH_t;
|
|
|
|
typedef struct node_str
|
|
{
|
|
char* fn; // NULL for available nodes
|
|
unsigned id; // kInvalidId for available nodes
|
|
dynLibH_t dynLibH; // The platform dependent library handle
|
|
struct node_str* link;
|
|
} node_t;
|
|
|
|
typedef struct lib_str
|
|
{
|
|
node_t* list; // list of open libraries
|
|
unsigned id; // next library id
|
|
} lib_t;
|
|
|
|
lib_t* _handleToPtr( handle_t h )
|
|
{ return handleToPtr<handle_t,lib_t>(h); }
|
|
|
|
// Given a library id return the associated node
|
|
node_t* _libIdToNode( lib_t* p, unsigned libId )
|
|
{
|
|
node_t* np = p->list;
|
|
|
|
while( np != nullptr )
|
|
{
|
|
if( np->id == libId )
|
|
return np;
|
|
|
|
np = np->link;
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
// Close a library and make it's node available for reuse
|
|
rc_t _closeLib( node_t* np )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
const char* errMsg = nullptr;
|
|
|
|
// tell the system to close the library
|
|
if((errMsg = _cmLibClose( &np->dynLibH )) != nullptr )
|
|
rc = cwLogError(kInvalidOpRC,"Library %s close failed. Error:%s.", np->fn, errMsg );
|
|
else
|
|
{
|
|
// mark the node as available
|
|
mem::release(np->fn);
|
|
np->id = kInvalidId;
|
|
}
|
|
|
|
return rc;
|
|
}
|
|
|
|
// Finalize the library manager
|
|
rc_t _finalize( lib_t* p )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
node_t* np = p->list;
|
|
while( np!=nullptr )
|
|
{
|
|
node_t* n0p = np->link;
|
|
|
|
// close the node's library
|
|
rc_t rc0 = _closeLib( np );
|
|
|
|
// store the error code
|
|
if( rc0 != kOkRC )
|
|
rc = rc0;
|
|
|
|
// release the node
|
|
mem::release(np);
|
|
|
|
np = n0p;
|
|
}
|
|
|
|
mem::release(p);
|
|
return rc;
|
|
}
|
|
}
|
|
}
|
|
|
|
cw::rc_t cw::lib::initialize( handle_t& h, const char* dirStr )
|
|
{
|
|
rc_t rc;
|
|
if((rc = finalize(h)) != kOkRC )
|
|
return rc;
|
|
|
|
lib_t* p = mem::allocZ<lib_t>();
|
|
|
|
h.set(p);
|
|
|
|
return rc;
|
|
}
|
|
|
|
cw::rc_t cw::lib::finalize( handle_t& h )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
if( !h.isValid() )
|
|
return rc;
|
|
|
|
lib_t* p = _handleToPtr(h);
|
|
if((rc = _finalize(p)) != kOkRC )
|
|
return rc;
|
|
|
|
h.clear();
|
|
|
|
return rc;
|
|
}
|
|
|
|
cw::rc_t cw::lib::open( handle_t h, const char* fn, unsigned& libIdRef )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
lib_t* p = _handleToPtr(h);
|
|
dynLibH_t lH = _cmLibOpen(fn);
|
|
node_t* np = p->list;
|
|
unsigned idx = 0;
|
|
|
|
libIdRef = kInvalidId;
|
|
|
|
if( _cmLibIsNull(lH) )
|
|
{
|
|
// There is apparently no way to get an error code which indicates that the
|
|
// file load attempt failed because the file was not a shared library -
|
|
// which should not generate an error message - therefore
|
|
// we must match the end of the the error string returned by dlerror() with
|
|
// 'invalid ELF header'.
|
|
const char* errMsg = _cmLibSysError();
|
|
const char* s = "invalid ELF header";
|
|
unsigned sn = strlen(s);
|
|
unsigned mn = strlen(errMsg);
|
|
|
|
// Did this load fail because the file was not a shared library?
|
|
if( errMsg!=nullptr && mn>sn && strcmp(errMsg+mn-sn,s)==0 )
|
|
rc = cwLogError(kOpenFailRC,"Library load failed. No error message." ); // signal error but no error message
|
|
else
|
|
rc = cwLogError(kOpenFailRC,"Library load failed. System Message: %s", errMsg );
|
|
|
|
return rc;
|
|
}
|
|
|
|
// find an available node
|
|
while( np != nullptr )
|
|
{
|
|
if( np->fn == nullptr )
|
|
break;
|
|
|
|
np = np->link;
|
|
++idx;
|
|
}
|
|
|
|
// no available node was found - allocate a new one
|
|
if( np == nullptr )
|
|
{
|
|
np = mem::allocZ<node_t>(1);
|
|
np->link = p->list;
|
|
p->list = np;
|
|
}
|
|
|
|
// initialize the node
|
|
np->fn = mem::duplStr(fn);
|
|
np->dynLibH = lH;
|
|
np->id = p->id++;
|
|
libIdRef = np->id;
|
|
|
|
return idx;
|
|
}
|
|
|
|
cw::rc_t cw::lib::close( handle_t h, unsigned libId )
|
|
{
|
|
lib_t* p = _handleToPtr(h);
|
|
node_t* np;
|
|
|
|
// locate the library to close
|
|
if((np = _libIdToNode(p,libId)) == nullptr )
|
|
return cwLogError(kInvalidIdRC,"Library close failed. The library with id:%i not found.",libId);
|
|
|
|
return _closeLib(np);
|
|
}
|
|
|
|
void* cw::lib::symbol( handle_t h, unsigned libId, const char* symName )
|
|
{
|
|
void* f;
|
|
lib_t* p = _handleToPtr(h);
|
|
node_t* np = _libIdToNode(p,libId);
|
|
|
|
if( (np == NULL) || _cmLibIsNull(np->dynLibH) )
|
|
{
|
|
cwLogError(kInvalidArgRC,"The library id %i is not valid or the library is closed.",libId);
|
|
return NULL;
|
|
}
|
|
|
|
if((f = _cmLibSym(np->dynLibH,symName)) == NULL)
|
|
{
|
|
cwLogError(kInvalidArgRC,"The dynamic symbol '%s' was not found. System Message: %s", cwStringNullGuard(symName), _cmLibSysError());
|
|
return NULL;
|
|
}
|
|
|
|
return f;
|
|
}
|
|
|
|
cw::rc_t cw::lib::scan( handle_t h, const char* dirStr )
|
|
{
|
|
rc_t rc = kOkRC;
|
|
unsigned dirEntryN = 0;
|
|
filesys::dirEntry_t* de = dirEntries( dirStr, filesys::kFileFsFl, &dirEntryN );
|
|
unsigned libId;
|
|
|
|
for(unsigned i=0; i<dirEntryN; ++i)
|
|
if( de[i].name != nullptr )
|
|
if((rc = open( h, de[i].name, libId )) != kOkRC )
|
|
cwLogWarning("The file '%s' is not a shared library.",de[i].name);
|
|
|
|
mem::release(de);
|
|
return rc;
|
|
}
|
|
|
|
unsigned cw::lib::count( handle_t h )
|
|
{
|
|
lib_t* p = _handleToPtr(h);
|
|
unsigned n = 0;
|
|
node_t* np = p->list;
|
|
for(; np!=nullptr; np=np->link)
|
|
++n;
|
|
return n;
|
|
}
|
|
|
|
unsigned cw::lib::indexToId( handle_t h, unsigned idx )
|
|
{
|
|
lib_t* p = _handleToPtr(h);
|
|
unsigned n = 0;
|
|
node_t* np = p->list;
|
|
|
|
for(; np!=nullptr; np=np->link)
|
|
if( n == idx )
|
|
return np->id;
|
|
|
|
cwLogError(kInvalidArgRC, "The library index %i is not valid.",idx);
|
|
return kInvalidId;
|
|
}
|
|
|
|
|
|
const char* cw::lib::name( handle_t h, unsigned id )
|
|
{
|
|
lib_t* p = _handleToPtr(h);
|
|
node_t* np;
|
|
if((np = _libIdToNode(p,id)) == nullptr )
|
|
{
|
|
cwLogError(kInvalidArgRC,"The library associated with the the id:%i could not be found.",id);
|
|
return nullptr;
|
|
}
|
|
return np->fn;
|
|
}
|