source: dev/trunk/ab5.0/abdev/BasicCompiler_Common/CDebugThreadInfo.cpp

#include "stdafx.h"

#include "common.h"
#include "DebugSection.h"

#include <ProcessAndModule.h>

//デバッグ用
#include "../BasicCompiler_Common/debug.h"


extern HANDLE hDebugProcess;
extern DWORD processIdDebug;
extern DebugSectionCollection debugSectionCollection;

CDebugThreadInfo *pobj_dti;

CDebugThreadInfo::CDebugThreadInfo(){
    memset(this,0,sizeof(CDebugThreadInfo));
}
CDebugThreadInfo::~CDebugThreadInfo(){
    Free();
}

BOOL CDebugThreadInfo::Reflesh(int ThreadNum){
    Free();

    ULONG_PTR lpData;
    ULONG_PTR lpAccBytes;

    DWORD dwRWSectionPos;
    dwRWSectionPos=debugSectionCollection.debugSections[0]->dwImageBase+
                        debugSectionCollection.debugSections[0]->dwRVA_RWSection;

    ReadProcessMemory(hDebugProcess,
        (void *)(dwRWSectionPos+sizeof(DWORD)*256+ThreadNum*sizeof(DWORD)),
        &iProcLevel,
        sizeof(DWORD),&lpAccBytes);
    lplpObp=(ULONG_PTR *)HeapAlloc(hHeap,0,(iProcLevel+1)*sizeof(ULONG_PTR));
    lplpSpBase=(ULONG_PTR *)HeapAlloc(hHeap,0,(iProcLevel+1)*sizeof(ULONG_PTR));
    lpdwCp=(DWORD *)HeapAlloc(hHeap,0,(iProcLevel+1)*sizeof(DWORD));
    this->relationalObjectModuleIndexes.resize( iProcLevel + 1 );

    //lplpObp
    ReadProcessMemory(hDebugProcess,
        (void *)(dwRWSectionPos+sizeof(DWORD)*256*2+ThreadNum*sizeof(DWORD)),
        &lpData,
        sizeof(ULONG_PTR),&lpAccBytes);
    ReadProcessMemory(hDebugProcess,
        (void *)lpData,
        lplpObp,
        (iProcLevel+1)*sizeof(ULONG_PTR),&lpAccBytes);

    //lplpSpBase
    ReadProcessMemory(hDebugProcess,
        (void *)(dwRWSectionPos+sizeof(DWORD)*256*2+sizeof(ULONG_PTR)*256+ThreadNum*sizeof(ULONG_PTR)),
        &lpData,
        sizeof(ULONG_PTR),&lpAccBytes);
    ReadProcessMemory(hDebugProcess,
        (void *)lpData,
        lplpSpBase,
        (iProcLevel+1)*sizeof(ULONG_PTR),&lpAccBytes);

    //現在の実行情報
    CONTEXT Context;
    Context.ContextFlags=CONTEXT_CONTROL;
    GetThreadContext(array_hDebugThread[ThreadNum],&Context);


    //マッピングされているモジュールのイメージベースを取得
    //HMODULE array_hModule[1024];
    //DWORD cbReturned;
    //if(!EnumProcessModules( hDebugProcess, array_hModule, sizeof(HMODULE)*1024, &cbReturned )) return 0;
    auto modules = ActiveBasic::Common::GetModules(processIdDebug);

    HMODULE hTargetModule=0;
    for (std::size_t i = 0; i < modules.size(); ++i)
    {
        if ((LONG_PTR)modules[i]<=(LONG_PTR)EIP_RIP(Context))
        {
            if (hTargetModule<modules[i])
            {
                hTargetModule=modules[i];
            }
        }
    }

    extern DWORD ImageBase;
    if((HMODULE)(ULONG_PTR)ImageBase!=hTargetModule){
        for(std::size_t i=0;i<debugSectionCollection.debugSections.size();i++){
            if((HMODULE)(ULONG_PTR)debugSectionCollection.debugSections[i]->dwImageBase==hTargetModule){
                debugSectionCollection.choice(i);
                break;
            }
        }
    }



    extern int MemPos_CodeSection;
    extern int FileSize_CodeSection;

    if(ImageBase+MemPos_CodeSection <= EIP_RIP(Context) &&
        EIP_RIP(Context) < ImageBase+MemPos_CodeSection+FileSize_CodeSection){
        lplpObp[iProcLevel]=EIP_RIP(Context);
        lplpSpBase[iProcLevel]=SP_BASE(Context);
    }
    //else API関数内などの場合は既存の lplpObp[iProcLevel] を利用する

    return 1;
}

void CDebugThreadInfo::Free(void){
    if(lplpObp){
        HeapDefaultFree(lplpObp);
        HeapDefaultFree(lplpSpBase);
        HeapDefaultFree(lpdwCp);
        relationalObjectModuleIndexes.clear();
    }

    memset(this,0,sizeof(CDebugThreadInfo));
}