threaddlg.cpp

VisualC高级编程技术精粹.rar

// ThreadDlg.cpp : implementation file
//

#include "stdafx.h"
#include "processmg.h"
#include "ThreadDlg.h"
#include "ProcessMgDlg.h"
#include "tlhelp32.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CThreadDlg dialog


CThreadDlg::CThreadDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CThreadDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CThreadDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
}


void CThreadDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CThreadDlg)
	DDX_Control(pDX, IDC_LIST_THREAD, m_listctrlThread);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CThreadDlg, CDialog)
	//{{AFX_MSG_MAP(CThreadDlg)
	ON_BN_CLICKED(IDC_BUTTON_OK, OnButtonOk)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CThreadDlg message handlers

BOOL CThreadDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// 初始化列表控件
	m_listctrlThread.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES);
	m_listctrlThread.InsertColumn(1,_T("序号"),LVCFMT_CENTER,50,1);

	m_listctrlThread.InsertColumn(2,_T("线程ID"),LVCFMT_CENTER,100,1);
	m_listctrlThread.InsertColumn(3,_T("优先级"),LVCFMT_CENTER,80,1);
	CString csPID = ((CProcessMgDlg *)m_parent)->m_csPID;

	DWORD dwTID = atol(csPID);
	GetThreadInfo(dwTID);

    m_listctrlThread.SetTextBkColor(RGB(255, 191, 127));
    m_listctrlThread.SetTextColor( RGB(35,91,217) );
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

BOOL CThreadDlg::GetThreadInfo(DWORD dwProcessId)
{
	char Buffer[256];
    HANDLE hThreadSnap = NULL; 
    BOOL bRet = FALSE; 
    THREADENTRY32 te32 = {0}; 	
    hThreadSnap = CreateToolhelp32Snapshot(TH32CS_SNAPTHREAD, 0); 

    if (hThreadSnap == INVALID_HANDLE_VALUE) 
	{
        return (FALSE); 	
	}
    te32.dwSize = sizeof(THREADENTRY32); 

	int i = 0 ;
    if (Thread32First(hThreadSnap, &te32)) 
    { 
        do 
        { 
            if (te32.th32OwnerProcessID == dwProcessId) 
            { 
				itoa (i, Buffer,10);
			
				m_listctrlThread.InsertItem(1,Buffer);
				ltoa(te32.th32ThreadID,Buffer,10);

				m_listctrlThread.SetItemText(1,1,Buffer);
				ltoa(te32.tpBasePri,Buffer,10);
				m_listctrlThread.SetItemText(1,2,Buffer);

			    i++;
            } 			
        } 
        while (Thread32Next(hThreadSnap, &te32)); 
        bRet = TRUE; 
    } 
    else 
	{
        bRet = FALSE;      
	}
    CloseHandle (hThreadSnap); 	

    return (bRet); 
}

void CThreadDlg::OnButtonOk() 
{
	// TODO: Add your control notification handler code here
	OnOK();	
}