source: dev/trunk/ab5.0/abdev/abdev/Attach.cpp @ 681

#include "stdafx.h"

#ifndef THETEXT

#include "common.h"
#include <psapi.h>
#include <tlhelp32.h>

#pragma comment(lib, "psapi.lib")

using namespace ActiveBasic::IDE;


int GetSelectingProcessId(HWND hListView,DWORD *lpdwPlatform){
    int nCount;
    nCount=ListView_GetItemCount(hListView);

    char temporary[255];
    DWORD dwProcessId;
    int i;
    for(i=0;i<nCount;i++){
        if(ListView_GetItemState(hListView,i,LVIS_SELECTED)){
            ListView_GetItemText(hListView,i,1,temporary,255);
            dwProcessId=atoi(temporary);

            ListView_GetItemText(hListView,i,2,temporary,255);
            if(lstrcmpi(temporary,"Win32")==0) *lpdwPlatform=IMAGE_FILE_MACHINE_I386;
            else if(lstrcmpi(temporary,"Win64")==0) *lpdwPlatform=IMAGE_FILE_MACHINE_AMD64;
            break;
        }
    }
    if(i==nCount) return 0;

    return dwProcessId;
}

void InsertProcess_ToListView(HWND hListView,char *lpszAppPath,int id,char *lpszPlatform,char *lpszTitle){
    char szAppName[MAX_PATH],temp2[MAX_PATH];
    _splitpath(lpszAppPath,NULL,NULL,szAppName,temp2);
    lstrcat(szAppName,temp2);

    //イメージリストを取得
    HIMAGELIST hImageList;
    hImageList=ListView_GetImageList(hListView,LVSIL_SMALL );

    //アイコンを取得
    int iImage;
    SHFILEINFO shfi;
    SHGetFileInfo(lpszAppPath, FILE_ATTRIBUTE_ARCHIVE, &shfi, sizeof(SHFILEINFO), SHGFI_ICON | SHGFI_SMALLICON );
    if(shfi.hIcon){
        iImage=ImageList_AddIcon(hImageList,shfi.hIcon);
        DestroyIcon(shfi.hIcon);
    }
    else iImage=-1;

    char szId[255];
    sprintf(szId,"%d",id);

    //リストビューに追加
    LV_ITEM item;
    if(iImage==-1) item.mask=LVIF_TEXT;
    else item.mask=LVIF_TEXT|LVIF_IMAGE;
    item.iItem=0;

    item.pszText=szAppName;
    item.iSubItem=0;
    item.iImage=iImage;
    ListView_InsertItem(hListView,&item);

    item.mask=LVIF_TEXT ;
    item.pszText=szId;
    item.iSubItem++;
    ListView_SetItem(hListView,&item);

    item.pszText=lpszPlatform;
    item.iSubItem++;
    ListView_SetItem(hListView,&item);

    item.pszText=lpszTitle;
    item.iSubItem++;
    ListView_SetItem(hListView,&item);
}

void ListupProcesses64(HWND hListView)
{
    ListView_DeleteAllItems(hListView);


    /////////////////////////////////////////////////////
    // 64ビット版プロセス列挙モジュールを呼び出す
    /////////////////////////////////////////////////////

    std::string enumProcess64ExePath = ActiveBasic::Common::Environment::GetAbdevSystemDirPath() + "\\enum_process64\\enum_process64.exe";

    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    memset(&si,0,sizeof(STARTUPINFO));
    si.cb=sizeof(STARTUPINFO);
    CreateProcess(NULL,const_cast<char *>(enumProcess64ExePath.c_str()),NULL,NULL,0,NORMAL_PRIORITY_CLASS,NULL,NULL,&si,&pi);

    WaitForSingleObject(pi.hProcess,INFINITE);


    //////////////////////////////////
    // 生成されたファイルの読み込み
    //////////////////////////////////

    std::string listPath = ActiveBasic::Common::Environment::GetAbdevSystemDirPath() + "\\enum_process64\\list.dat";

    char *buffer;
    buffer=ReadBuffer_NonErrMsg( listPath );
    if(!buffer) return;

    int i=0;
    char temporary[MAX_PATH];
    while(1){
        //ファイルパス
        char szAppPath[MAX_PATH];
        i=GetOneParameter(buffer,i,szAppPath);
        if(szAppPath[0]=='\0') break;
        RemoveStringQuotes(szAppPath);

        //プロセスID
        DWORD dwProcessId;
        i=GetOneParameter(buffer,i,temporary);
        if(temporary[0]=='\0') break;
        dwProcessId=(DWORD)atoi(temporary);

        //プラットフォーム
        char szPlatform[255];
        i=GetOneParameter(buffer,i,szPlatform);
        if(temporary[0]=='\0') break;

        //タイトル
        char szTitle[255];
        int i2;
        for(i2=0;;i++,i2++){
            if(IS_RETURN(buffer,i)||buffer[i]=='\0'){
                szTitle[i2]=0;
                break;
            }
            szTitle[i2]=buffer[i];
        }


        //リストビューへ追加
        InsertProcess_ToListView(hListView,szAppPath,dwProcessId,szPlatform,szTitle);

        if(buffer[i]=='\r'&&buffer[i+1]=='\n') i+=2;
        if(buffer[i]=='\0') break;
    }

    HeapDefaultFree(buffer);
}

// EnumWindowsProcコールバック関数
BOOL CALLBACK FindWindowProc(HWND hwnd, LPARAM lParam)
{
    HWND* pphWnd = (HWND*)lParam;
    *pphWnd = hwnd;
    return FALSE; /* 列挙を中断 */
}
void ListupProcesses(HWND hListView){
    char temporary[MAX_PATH];

    ListView_DeleteAllItems(hListView);

    DWORD *pdwProcessId;
    pdwProcessId=(DWORD *)HeapAlloc(hHeap,0,8192*sizeof(DWORD));

    //プロセスを列挙
    DWORD cbNeeded;
    EnumProcesses(pdwProcessId,8192*sizeof(DWORD),&cbNeeded);

    char szMyAppPath[MAX_PATH];
    GetModuleFileName(GetModuleHandle(0),szMyAppPath,MAX_PATH);

    int i;
    for(i=0;i<(int)(cbNeeded/sizeof(DWORD));i++){

        //プロセスIDを元にハンドルを取得
        HANDLE hProcess;
        hProcess=OpenProcess(PROCESS_ALL_ACCESS,0,pdwProcessId[i]);
        if(!hProcess) continue;

        //そのプロセスにおける実行モジュールのインスタンスハンドルを取得
        HINSTANCE hModule;
        DWORD cbReturned;
        if(!EnumProcessModules( hProcess, &hModule, sizeof(HINSTANCE), &cbReturned ))
            continue;

        //実行ファイル名を取得
        GetModuleFileNameEx(hProcess,hModule,temporary,MAX_PATH);


        //////////////////////////////
        // ウィンドウタイトルを取得
        //////////////////////////////

        //スナップショットを取得
        HANDLE hSnapshot;
        hSnapshot=CreateToolhelp32Snapshot(TH32CS_SNAPALL,0);

        THREADENTRY32 te;
        memset(&te,0,sizeof(THREADENTRY32));
        te.dwSize=sizeof(THREADENTRY32);
        if(Thread32First(hSnapshot,&te)){
            do{
                if(te.th32OwnerProcessID==pdwProcessId[i]){
                    break;
                }
            }while(Thread32Next(hSnapshot,&te));
        }

        HWND hwnd;
        hwnd=0;
        EnumThreadWindows(te.th32ThreadID,FindWindowProc,(LPARAM)&hwnd);

        while(GetParent(hwnd)) hwnd=GetParent(hwnd);

        char szWndTitle[1024];
        szWndTitle[0]=0;
        if(hwnd) GetWindowText(hwnd,szWndTitle,1024);

        CloseHandle(hSnapshot);



        CloseHandle(hProcess);

        if(lstrcmpi(szMyAppPath,temporary)==0){
            //自分自身（ProjectEditor.exe）は無視
            continue;
        }

        //リストビューへ追加
        InsertProcess_ToListView(hListView,temporary,pdwProcessId[i],"Win32",szWndTitle);
    }

    HeapDefaultFree(pdwProcessId);
}

BOOL CALLBACK DlgAttach(HWND hwnd,UINT message,WPARAM wParam,LPARAM lParam){
    static HWND hListView;
    static HIMAGELIST hImageList;
    static DWORD *lpdwPlatform;
    switch(message){
        case WM_INITDIALOG:
            SetPosCenter(hwnd);

            lpdwPlatform=(DWORD *)lParam;

            hListView=GetDlgItem(hwnd,IDC_PROCESSLIST);

            //拡張スタイル
            DWORD dwStyle;
            dwStyle=ListView_GetExtendedListViewStyle(hListView);
            dwStyle|=LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES;
            ListView_SetExtendedListViewStyle(hListView,dwStyle);


            //イメージリストの生成と設定
            hImageList=ImageList_Create(16,16,ILC_COLOR24|ILC_MASK,0,0);
            ListView_SetImageList(hListView,hImageList,LVSIL_SMALL );


            ///////////////////////////
            // カラムをセット
            ///////////////////////////

            LV_COLUMN ListView_Column;
            ListView_Column.mask=LVCF_FMT | LVCF_WIDTH | LVCF_TEXT | LVCF_SUBITEM;
            ListView_Column.fmt=LVCFMT_LEFT;

            ListView_Column.cx=150;
            ListView_Column.pszText="プロセス名";
            ListView_Column.iSubItem=0;
            ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);

            ListView_Column.cx=70;
            ListView_Column.pszText="ID";
            ListView_Column.iSubItem++;
            ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);

            ListView_Column.cx=50;
            ListView_Column.pszText="種別";
            ListView_Column.iSubItem++;
            ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);

            ListView_Column.cx=200;
            ListView_Column.pszText="タイトル";
            ListView_Column.iSubItem++;
            ListView_InsertColumn(hListView,ListView_Column.iSubItem,&ListView_Column);


            //リスビューを更新
            if(IsWow64()) ListupProcesses64(hListView);
            else ListupProcesses(hListView);

            break;
        case WM_COMMAND:
            switch(LOWORD(wParam)){
                case IDOK:
                    EndDialog(hwnd,GetSelectingProcessId(hListView,lpdwPlatform));
                    return 1;
                case IDCANCEL:
                    EndDialog(hwnd,0);
                    return 1;
            }
            return 0;

        case WM_NOTIFY:
            NMLISTVIEW *nmListView;
            nmListView=(NMLISTVIEW *)lParam;
            if(nmListView->hdr.hwndFrom==hListView){
                if(nmListView->hdr.code==NM_DBLCLK){
                    if(nmListView->iItem==-1) return 1;
                    SendMessage(hwnd,WM_COMMAND,IDOK,0);
                    return 1;
                }
            }
            break;

        case WM_DESTROY:
            ImageList_Destroy(hImageList);
            return 1;
    }
    return 0;
}


#endif //THETEXT