Changeset 798 in dev for branches/egtra/ab5.0/abdev/BasicCompiler_Common

Timestamp:

Feb 5, 2011, 11:43:01 PM (14 years ago)

Author:

イグトランス (egtra)

Message:

PSAPIを使用している箇所について、Toolhelp32と選択するよう修正

Location:

branches/egtra/ab5.0/abdev/BasicCompiler_Common

Files:

• BasicCompiler.cpp (modified) (3 diffs)
• CDebugThreadInfo.cpp (modified) (4 diffs)
• Debug.cpp (modified) (7 diffs)

branches/egtra/ab5.0/abdev/BasicCompiler_Common/BasicCompiler.cpp

@@ -377 +377 @@
 BOOL CALLBACK DlgCompile(HWND hwnd,UINT message,WPARAM wParam,LPARAM lParam){
     extern HANDLE hDebugProcess;
+    extern DWORD processIdDebug;
     char temporary[MAX_PATH];
     RECT rect;
@@ -444 +445 @@
                             TerminateProcess(hDebugProcess,0);
                             hDebugProcess=0;
+                            processIdDebug = 0;
 
                             //デバッグダイアログを終了
@@ -475 +477 @@
                         TerminateProcess(hDebugProcess,0);
                         hDebugProcess=0;
+                        processIdDebug = 0;
 
                         //デバッグダイアログを終了

branches/egtra/ab5.0/abdev/BasicCompiler_Common/CDebugThreadInfo.cpp

@@ -4 +4 @@
 #include "DebugSection.h"
 
+#include <ProcessAndModule.h>
 
 //デバッグ用
@@ -10 +11 @@
 
 extern HANDLE hDebugProcess;
+extern DWORD processIdDebug;
 extern DebugSectionCollection debugSectionCollection;
 
@@ -67 +69 @@
 
     //マッピングされているモジュールのイメージベースを取得
-    HMODULE array_hModule[1024];
-    DWORD cbReturned;
-    if(!EnumProcessModules( hDebugProcess, array_hModule, sizeof(HMODULE)*1024, &cbReturned )) return 0;
+    //HMODULE array_hModule[1024];
+    //DWORD cbReturned;
+    //if(!EnumProcessModules( hDebugProcess, array_hModule, sizeof(HMODULE)*1024, &cbReturned )) return 0;
+    auto modules = ActiveBasic::Common::GetModules(processIdDebug);
 
-    int i;
     HMODULE hTargetModule=0;
-    for(i=0;i<(int)(cbReturned/sizeof(HMODULE));i++){
-        if((LONG_PTR)array_hModule[i]<=(LONG_PTR)EIP_RIP(Context)){
-            if(hTargetModule<array_hModule[i]) hTargetModule=array_hModule[i];
+    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];
+            }
         }
     }
@@ -81 +88 @@
     extern DWORD ImageBase;
     if((HMODULE)(ULONG_PTR)ImageBase!=hTargetModule){
-        for(i=0;i<debugSectionCollection.debugSections.size();i++){
+        for(std::size_t i=0;i<debugSectionCollection.debugSections.size();i++){
             if((HMODULE)(ULONG_PTR)debugSectionCollection.debugSections[i]->dwImageBase==hTargetModule){
                 debugSectionCollection.choice(i);

branches/egtra/ab5.0/abdev/BasicCompiler_Common/Debug.cpp

@@ -9 +9 @@
 #include "../BasicCompiler_Common/debug.h"
 
+#include <Path.h>
+#include <ProcessAndModule.h>
+
 char *OpBuffer;
 
@@ -14 +17 @@
 
 HANDLE hDebugProcess;
+DWORD processIdDebug;
 
 DWORD _DebugSys_dwThreadID[MAX_DEBUG_THREAD];
@@ -155 +159 @@
     TerminateProcess(hDebugProcess,0);
     hDebugProcess=0;
+    processIdDebug = 0;
 
     //デバッグダイアログを終了
@@ -435 +440 @@
     if( program.IsAttach() )
     {
-        //プロセスIDを元にハンドルを取得
-        HANDLE hProcess;
-        hProcess=OpenProcess(PROCESS_ALL_ACCESS,0, program.GetAttachProcessId() );
-        if(!hProcess) goto AttachError;
-
-        //そのプロセスにおける実行モジュールのインスタンスハンドルを取得
-        HINSTANCE hModule;
-        DWORD cbReturned;
-        if(!EnumProcessModules( hProcess, &hModule, sizeof(HINSTANCE), &cbReturned )) goto AttachError;
+        ////プロセスIDを元にハンドルを取得
+        //HANDLE hProcess = OpenProcess(PROCESS_ALL_ACCESS,0, program.GetAttachProcessId() );
+        //if(!hProcess)
+        //{
+        //  DebugMessage("アタッチに失敗しました。");
+        //  return;
+        //}
+
+        ////そのプロセスにおける実行モジュールのインスタンスハンドルを取得
+        //HINSTANCE hModule;
+        //DWORD cbReturned;
+        //if(!EnumProcessModules( hProcess, &hModule, sizeof(HINSTANCE), &cbReturned ))
+        //{
+        //  DebugMessage("アタッチに失敗しました。");
+        //  return;
+        //}
 
         //実行ファイル名を取得
-        char tempOutputFileName[MAX_PATH];
-        GetModuleFileNameEx(hProcess,hModule,tempOutputFileName,MAX_PATH);
-        program.SetOutputFilePath( tempOutputFileName );
-
-        CloseHandle(hProcess);
+        auto tempOutputFilePath = ActiveBasic::Common::GetExecutableModulePath(program.GetAttachProcessId());
+        program.SetOutputFilePath( tempOutputFilePath.string().c_str() );
+
+        //CloseHandle(hProcess);
 
 /*
@@ -456 +467 @@
         pobj_DebugSection->load(program.GetOutputFilePath().c_str());*/
 
-        if(!DebugActiveProcess( program.GetAttachProcessId() )){
-AttachError:
+        if(!DebugActiveProcess( program.GetAttachProcessId() ))
+        {
             DebugMessage("アタッチに失敗しました。");
             return;
@@ -532 +543 @@
             else{
 Attach_DllLoad:
-                //アタッチした場合
-                GetModuleFileNameEx(hDebugProcess,(HINSTANCE)de.u.LoadDll.lpBaseOfDll,temporary,MAX_PATH);
+                {
+                    //アタッチした場合
+                    auto path = ActiveBasic::Common::GetModuleFilePath(hDebugProcess, reinterpret_cast<HINSTANCE>(de.u.LoadDll.lpBaseOfDll));
+                    strncpy_s(temporary, path.string().c_str(), _TRUNCATE);
+                }
             }
 
@@ -562 +576 @@
             hDebugProcess=de.u.CreateProcessInfo.hProcess;
             hMainThread=de.u.CreateProcessInfo.hThread;
+            processIdDebug = de.dwProcessId;
 
             if(debugSectionCollection.add((HMODULE)de.u.CreateProcessInfo.lpBaseOfImage)){