* dll_init.cc (dll_list::find_by_modname): New function to search the dll list
for a module name only (no path). (dll_list::alloc): Initialize newly-added members of struct dll. (dll_list::append): New function to factor out the append operation (used by dll_list::topsort). (dll_list::populate_deps): New function to identify dll dependencies. (dll_list::topsort): New function to sort the dll list topologically by dependencies. (dll_list::topsort_visit): New helper function for the above. * dll_init.h (dll::ndeps): New class member. (dll::deps): Ditto. (dll::modname): Ditto. (dll_list::find_by_modname): New function related to topsort. (dll_list::populate_all_deps): Ditto. (dll_list::populate_deps): Ditto. (dll_list::topsort): Ditto. (dll_list::topsort_visit): Ditto. (dll_list::append): Ditto. (pefile): New struct allowing simple introspection of dll images. * fork.cc (fork): Topologically sort the dll list before forking.
This commit is contained in:
@@ -52,6 +52,9 @@ struct dll
|
||||
int count;
|
||||
bool has_dtors;
|
||||
dll_type type;
|
||||
long ndeps;
|
||||
dll** deps;
|
||||
PWCHAR modname;
|
||||
WCHAR name[1];
|
||||
void detach ();
|
||||
int init ();
|
||||
@@ -84,6 +87,13 @@ public:
|
||||
void detach (void *);
|
||||
void init ();
|
||||
void load_after_fork (HANDLE);
|
||||
dll *find_by_modname (const PWCHAR name);
|
||||
void populate_all_deps ();
|
||||
void populate_deps (dll* d);
|
||||
void topsort ();
|
||||
void topsort_visit (dll* d, bool goto_tail);
|
||||
void append (dll* d);
|
||||
|
||||
dll *inext ()
|
||||
{
|
||||
while ((hold = hold->next))
|
||||
@@ -109,6 +119,25 @@ public:
|
||||
dll_list () { protect.init ("dll_list"); }
|
||||
};
|
||||
|
||||
/* References:
|
||||
http://msdn.microsoft.com/en-us/windows/hardware/gg463125
|
||||
http://msdn.microsoft.com/en-us/library/ms809762.aspx
|
||||
*/
|
||||
/* FIXME: Integrate with other similar uses in source. */
|
||||
struct pefile
|
||||
{
|
||||
IMAGE_DOS_HEADER dos_hdr;
|
||||
|
||||
char* rva (long offset) { return (char*) this + offset; }
|
||||
PIMAGE_NT_HEADERS32 pe_hdr () { return (PIMAGE_NT_HEADERS32) rva (dos_hdr.e_lfanew); }
|
||||
PIMAGE_OPTIONAL_HEADER32 optional_hdr () { return &pe_hdr ()->OptionalHeader; }
|
||||
PIMAGE_DATA_DIRECTORY idata_dir (DWORD which)
|
||||
{
|
||||
PIMAGE_OPTIONAL_HEADER32 oh = optional_hdr ();
|
||||
return (which < oh->NumberOfRvaAndSizes)? oh->DataDirectory + which : 0;
|
||||
}
|
||||
};
|
||||
|
||||
extern dll_list dlls;
|
||||
void dll_global_dtors ();
|
||||
|
||||
|
Reference in New Issue
Block a user