WinDLLTool.cpp

Go to the documentation of this file.

 
#include  "..\stdafx.h"
 
#include  "tools++.h"
#include  "WinDLLTool.h"
 
 
using namespace jbxl;
using namespace jbxwl;
 
 
//
// DLLModule
//
//        pgetinfo: char* (*pgetinfo)(int n) の関数へのポインタ．モジュールの情報を獲得するための関数．
//                  モジュールには (*pgetinfo)(0) で情報（タイトル）を返す関数が必ず必要．
//
 
void  DLLModule::init(LPCTSTR fname, int n)
{
    status = 0;
    fnum   = n;
    
    title    = NULL;
    dllname  = _T("");
    filename = _T("");
 
    if (fname!=NULL) funcname = fname;
    else             funcname = _T("");
    
    hmod     = NULL;
    pgetinfo = NULL;
    pfunc    = NULL;
    if (fnum>0) pfunc = (void**)malloc(sizeof(void*)*fnum);
}
 
 
 
void  DLLModule::free()
{
    status = 0;
 
    hmod = NULL;
    pgetinfo = NULL;
    if (fnum>0 && pfunc!=NULL) ::free(pfunc);
    fnum = 0;
    pfunc = NULL;
}
 
 
 
BOOL  DLLModule::set_funcname(LPCTSTR fname, int n)
{
    if (fname!=NULL) funcname = fname;
    else             funcname = _T("");
    
    if (fnum>0 && pfunc!=NULL) ::free(pfunc);
    pfunc = NULL;
 
    fnum = n;
    if (fnum>0) pfunc = (void**)malloc(sizeof(void*)*fnum);
 
    if (fname==NULL || n<=0) return FALSE;
    return TRUE;
}
 
 
 
BOOL  DLLModule::get_module_info(tList* lp)
{
    if (lp==NULL) return FALSE;
    if (funcname==_T("")) return FALSE;
 
    status   = 0;
    hmod = (HMODULE)(lp->ldat.ptr);
    char* mbstr = ts2mbs(funcname);
    pgetinfo = (FUNC_STRING)GetProcAddress(hmod, mbstr);
    ::free(mbstr);
    if (pgetinfo==NULL) return FALSE;
    
    dllname  = mbs2ts((char*)(lp->ldat.key.buf));
    filename = mbs2ts((char*)(lp->ldat.val.buf));
    title    = (*pgetinfo)(0);
 
    pfunc[0] = (void*)pgetinfo;
    for (int i=1; i<fnum; i++) {
        char* nfn = (*pgetinfo)(i);
        void* pfn = (void*)GetProcAddress(hmod, nfn);
        pfunc[i] = pfn;
        //if (pfn==NULL) {
        //  status = -1;
        //  return FALSE;
        //}
    }
 
    status = 1;
    return TRUE;
}
 
 
 
void*  DLLModule::get_funcptr(int n)
{   
    if (status<=0) return NULL;
 
    if (n>=fnum) n = 0;
    return pfunc[n];
}
 
 
 
 
// DLLModuleTBL
//
 
void  DLLModuleTBL::init(LPCTSTR dirn)
{   
    tnum = vnum = 0;
    loaded  = false;
    lp      = NULL;
    dllmod  = NULL;
    dirname = _T("");
    
    if (dirn!=NULL) load_module(dirn);
}
 
 
 
void DLLModuleTBL::free()
{
    
    if (dllmod!=NULL) {
        for (int i=0; i<vnum; i++) {
            if (dllmod[i]!=NULL) {
                dllmod[i]->free();
                delete(dllmod[i]);
                dllmod[i] = NULL;
            }
        }
    }
 
    tList* lt = lp;
    while (lt!=NULL) {
        if (lt->ldat.ptr!=NULL) {
            FreeLibrary((HMODULE)(lt->ldat.ptr));       // 関数へのハンドル
            lt->ldat.ptr = NULL;
            lt->ldat.sz  = 0;
        }
        lt = lt->next;
    }
    del_all_tList(&lp); 
 
    tnum = vnum = 0;
    if (dllmod!=NULL) ::free(dllmod);
    dllmod = NULL;
    loaded = false;
}
 
 
 
BOOL  DLLModuleTBL::load_module(LPCTSTR dirn)
{
    if (dirn!=NULL) dirname = dirn;
    else            dirname = _T(".");
 
    lp = get_dir_module((LPCTSTR)dirname);
    if (lp==NULL) return FALSE;
 
    tnum = 0;
    tList* lt = lp;
    while (lt!=NULL) {
        tnum++;
        lt = lt->next;
    }
    loaded = true;
 
    return TRUE;
}
 
 
 
tList*  DLLModuleTBL::get_dir_module(LPCTSTR dirn)
{
    tList* lm;
    tList* lt;
 
    if (dirn==NULL) return NULL;
    lm = lt = get_dir_files_t(dirn);
    if (lm==NULL) return NULL;
 
    while (lt!=NULL) {
        HMODULE hmod = LoadLibrary((LPCTSTR)mbs2ts((char*)(lt->ldat.val.buf)));
        lt->ldat.ptr = (void*)hmod;
        char* fn = get_file_name((char*)(lt->ldat.val.buf));
        lt->ldat.key = make_Buffer_bystr(fn);
        lt = lt->next;
    }
 
    // ハンドラが NULLの場合は削除
    lt = lm;
    while (lt!=NULL) {
        if (lt->ldat.ptr==NULL) {
            tList* ln = lt;
            if (lm==lt) lm = lt->next;
            lt = lt->next;
            del_tList_node(&ln);
        }
        else lt = lt->next;
    }
 
    return lm;
}
 
 
 
void  DLLModuleTBL::make_module_tbl(LPCTSTR fname, int m)
{
    tList* lm = lp;
 
    vnum = 0;
    if (fname==NULL) return;
 
    dllmod = (DLLModule**)malloc(sizeof(DLLModule*)*tnum);
    if (dllmod==NULL) return;
 
    int n = 0;
    if (m<=0) m = 1;
    while (lm!=NULL) {
        dllmod[n] = this->new_module(fname, m); 
        int ret = dllmod[n]->get_module_info(lm);
        if (ret) {
            n++;
            if (n>=tnum) break;
        }
        else {
            delete(dllmod[n]);
            dllmod[n] = NULL;
        }
        lm = lm->next;
    }
    vnum = n;
 
    this->set_module_function();
    return;
}