C++擷取工作列開啟程式視窗樣本_C 語言

複製代碼 代碼如下:

// File: OpenProgramOnTaskbar.h

#pragma once
#include <vector>

struct TaskInfo
{
    // 程式名稱
    CString strProgramName;

    // 視窗控制代碼
    HWND hWnd;
};

class TaskbarInfo
{
public:
    // 回呼函數
    static BOOL CALLBACK EnumWindowProc(HWND hWnd, LPARAM lParam);
public:

    // 記錄在工作列開啟的程式
    std::vector<TaskInfo> m_TaskbarInfoArr;

    // 頂層視窗控制代碼
    HWND m_hWnd;

public:
    TaskbarInfo();
    ~TaskbarInfo();

    // 獲得在工作列開啟程式資訊
    void GetTaskbarInof (void);

    // 獲得頂層視窗控制代碼，失敗返回NULL
    HWND GetTopWnd (void);
};

複製代碼 代碼如下:

// File: OpenProgramOnTaskbar.cpp

#include "stdafx.h"        // 如果編譯出錯，請刪除這句
#include "OpenProgramOnTaskbar.h"

#include "Psapi.h"
#pragma comment(lib, "Psapi.lib") 

TaskbarInfo::TaskbarInfo()
{

}

TaskbarInfo::~TaskbarInfo()
{

}

void TaskbarInfo::GetTaskbarInof (void)
{
    EnumWindows(EnumWindowProc, (LPARAM)this);
}

BOOL CALLBACK TaskbarInfo::EnumWindowProc(HWND hWnd, LPARAM lParam)
{
    TaskbarInfo * pTaskbarInfo = (TaskbarInfo*)lParam;

    TCHAR szWindow[256]={0};
    TaskInfo taskInfo;

    ::GetWindowText(hWnd, szWindow, 255); //擷取視窗標題
    if ( ::IsWindow(hWnd) &&
        ::IsWindowVisible(hWnd) &&
        (::GetWindowLong(hWnd, GWL_EXSTYLE)&WS_EX_TOOLWINDOW)!=WS_EX_TOOLWINDOW &&
        ::GetWindowLong(hWnd, GWL_HWNDPARENT)==0 )
    {
        DWORD dwPID;  //儲存進程標識符

        GetWindowThreadProcessId(hWnd, &dwPID);  //接受一個視窗控制代碼。dwPID儲存視窗的建立者的進程標識符，GetWindowThreadProcessId傳回值是該建立者的線程標識符
        HANDLE hBrowser = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPID);  //開啟一個已存在的進程對象,並返回進程的控制代碼，這就是我們要的進程控制代碼了

        WCHAR exePath[256];
        memset(exePath, 0, 256);
        //擷取程式的path，並儲存到exePath
        GetModuleFileNameEx(hBrowser, NULL, exePath, MAX_PATH);
        CString strTemp = exePath;
        strTemp.MakeLower();

        // 過濾在案頭開啟的檔案夾等
        if (!strTemp.IsEmpty() && strTemp.Find(_T("explorer.exe")) == -1)
        {
            taskInfo.strProgramName = szWindow;
            taskInfo.hWnd = hWnd;
            pTaskbarInfo->m_TaskbarInfoArr.push_back(taskInfo);
            memset(szWindow, 0, 256);
        }
    }
    return TRUE;
}

HWND TaskbarInfo::GetTopWnd (void)
{
    DWORD dwPID;  //儲存進程標識符
    HWND hWnd = NULL;
    hWnd = ::GetForegroundWindow();

    GetWindowThreadProcessId(hWnd, &dwPID);  //接受一個視窗控制代碼。dwPID儲存視窗的建立者的進程標識符，GetWindowThreadProcessId傳回值是該建立者的線程標識符
    HANDLE hBrowser = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPID);  //開啟一個已存在的進程對象,並返回進程的控制代碼，這就是我們要的進程控制代碼了

    WCHAR exePath[256];
    memset(exePath, 0, 256);
    //擷取程式的path，並儲存到exePath
    GetModuleFileNameEx(hBrowser, NULL, exePath, MAX_PATH);
    CString strTemp = exePath;
    strTemp.MakeLower();

    if (!strTemp.IsEmpty() && strTemp.Find(_T("explorer.exe")) == -1)
    {
        m_hWnd = hWnd;
        if (m_hWnd != NULL)
        {
            return m_hWnd;
        }
    }

    return NULL;
}