ConfigRecoder.cpp

// ConfigRecoder.cpp : 实现文件
//

#include "stdafx.h"
#include "ResourceUsageRecoder.h"
#include "ConfigRecoder.h"
#include "TipsDlg.h"
#include "afxdialogex.h"
#include "psapi.h"
#include <TlHelp32.h>
#pragma comment(lib,"psapi.lib")

// CConfigRecoder 对话框

int CConfigRecoder::m_iSortType = false;

IMPLEMENT_DYNAMIC(CConfigRecoder, CDialogEx)

CConfigRecoder::CConfigRecoder(CWnd* pParent /*=NULL*/)
	: CDialogEx(CConfigRecoder::IDD, pParent),
    m_listctrl_index(0)
{
}

CConfigRecoder::~CConfigRecoder()
{
}

void CConfigRecoder::GetConfig(RecoderInfo* info)
{
    //进程号
    int pos = m_listctrl.GetSelectionMark();
    int pid = _ttoi(m_listctrl.GetItemText(pos, 1));
    info->iPID = pid;
    info->strName = m_listctrl.GetItemText(pos, 0).GetString();

    //监视的资源消耗指标
    info->iCheck = 0x0;
    if (m_ck_mem.GetCheck())
    {
        info->iCheck = info->iCheck | RESTYPE_MEM;
    }

    if (m_ck_cpu.GetCheck())
    {
        info->iCheck = info->iCheck | RESTYPE_CPU;
    }

    if (m_ck_handle.GetCheck())
    {
        info->iCheck = info->iCheck | RESTYPE_HANDLE;
    }

    if (m_ck_thread.GetCheck())
    {
        info->iCheck = info->iCheck | RESTYPE_THREAD;
    }

    if (m_ck_io.GetCheck())
    {
        info->iCheck = info->iCheck | RESTYPE_IO;
    }

    if (m_ck_Nonpaged.GetCheck())
    {
        info->iCheck = info->iCheck | RESTYPE_NONP;
    }

    //
    CString strSampRate;
    m_edit_samprate.GetWindowText(strSampRate);
    info->iSampRate = _wtoi(strSampRate.GetBuffer());
    int index = m_cmb_sampunit.GetCurSel();
    if (index == 0)//秒
    {
        info->iSampRateUnit = 0;
    }
    else if (index == 1)//分
    {
        info->iSampRateUnit = 1;
    }

    //数据是否需要保存到本地
    if (m_ck_save.GetCheck())
    {
        info->bSave = true;
    }
    else
    {
        info->bSave = false;
    }
}

void CConfigRecoder::DoDataExchange(CDataExchange* pDX)
{
	CDialogEx::DoDataExchange(pDX);
	DDX_Control(pDX, IDC_LIST_PROCESS, m_listctrl);
	DDX_Control(pDX, IDC_CK_MEM, m_ck_mem);
	DDX_Control(pDX, IDC_CK_CPU, m_ck_cpu);
	DDX_Control(pDX, IDC_CK_HANDLE, m_ck_handle);
	DDX_Control(pDX, IDC_CK_THREAD, m_ck_thread);
	DDX_Control(pDX, IDC_CK_IO, m_ck_io);
	DDX_Control(pDX, IDC_CK_NOD, m_ck_Nonpaged);
	DDX_Control(pDX, IDC_CK_SAVE, m_ck_save);
	DDX_Control(pDX, IDC_EDIT_SAMPRATE, m_edit_samprate);
	DDX_Control(pDX, IDC_CMB_SAMPUNIT, m_cmb_sampunit);
	DDX_Control(pDX, IDC_CK_NETUP, m_ck_netup);
	DDX_Control(pDX, IDC_CK_NETDOWN, m_ck_netdown);
}


BEGIN_MESSAGE_MAP(CConfigRecoder, CDialogEx)
	ON_WM_PAINT()
	ON_WM_CTLCOLOR()
    ON_NOTIFY(LVN_COLUMNCLICK, IDC_LIST_PROCESS, &CConfigRecoder::OnLvnColumnclickListProcess)
    ON_EN_CHANGE(IDC_EDIT_SAMPRATE, &CConfigRecoder::OnEnChangeEditSamprate)
    ON_CBN_SELCHANGE(IDC_CMB_SAMPUNIT, &CConfigRecoder::OnCbnSelchangeCmbSampunit)
	ON_BN_CLICKED(IDC_CK_NETUP, &CConfigRecoder::OnBnClickedCkNetup)
	ON_BN_CLICKED(IDC_CK_NETDOWN, &CConfigRecoder::OnBnClickedCkNetdown)
END_MESSAGE_MAP()


// CConfigRecoder 消息处理程序


BOOL CConfigRecoder::OnInitDialog()
{
	CDialogEx::OnInitDialog();

	// TODO:  在此添加额外的初始化
	m_brush.CreateSolidBrush(RGB(255, 255, 255));

    //初始化进程列表
    m_listctrl.SetExtendedStyle(LVS_EX_FULLROWSELECT | LVS_EX_FULLROWSELECT);
    m_listctrl.SetBkColor(RGB(255, 255, 255));
    m_listctrl.InsertColumn(0, _T("Name"), LVCFMT_LEFT, 150);
    m_listctrl.InsertColumn(1, _T("PID"), LVCFMT_LEFT, 60);
    m_listctrl.InsertColumn(2, _T("File Path"), LVCFMT_LEFT, 600);

    //列表表头排序图标
    m_pListctrlHeader = m_listctrl.GetHeaderCtrl();
    m_ImageList.Create(16, 16, ILC_COLOR16 | ILC_MASK, 0, 4);
    m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_ICO_SORT_AZ));
    m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_ICO_SORT_ZA));
    m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_ICO_SORT_19));
    m_ImageList.Add(AfxGetApp()->LoadIcon(IDI_ICO_SORT_91));
    m_pListctrlHeader->SetImageList(&m_ImageList);

    GetAllProcess();

    //初始化编辑框
    m_edit_samprate.SetWindowText(_T("10"));

    //初始化combobox
    m_cmb_sampunit.InsertString(0,_T("秒"));
    m_cmb_sampunit.InsertString(1,_T("分"));
    m_cmb_sampunit.SetCurSel(1);

	return TRUE;  // return TRUE unless you set the focus to a control
	// 异常:  OCX 属性页应返回 FALSE
}

BOOL CConfigRecoder::PreTranslateMessage(MSG* pMsg)
{
	//屏蔽ESC Enter关闭窗体
	if (pMsg->message == WM_KEYDOWN && (pMsg->wParam == VK_ESCAPE || pMsg->wParam == VK_RETURN))
		return TRUE;
	else
		return CDialogEx::PreTranslateMessage(pMsg);
}

void CConfigRecoder::OnPaint()
{
	CPaintDC dc(this); // device context for painting
	// TODO:  在此处添加消息处理程序代码
	// 不为绘图消息调用 CDialogEx::OnPaint()
	CRect rect;
	GetClientRect(rect);
	dc.FillSolidRect(rect, RGB(255, 255, 255));
}


HBRUSH CConfigRecoder::OnCtlColor(CDC* pDC, CWnd* pWnd, UINT nCtlColor)
{
	HBRUSH hbr = CDialogEx::OnCtlColor(pDC, pWnd, nCtlColor);

	// TODO:  在此更改 DC 的任何特性

	if (pWnd->GetDlgCtrlID() == IDC_STATIC_SAMPRATE)
	{
		pDC->SetBkMode(TRANSPARENT);
		pDC->SetTextColor(RGB(0, 0, 0));
		return (HBRUSH)GetStockObject(HOLLOW_BRUSH);
	}

	switch (nCtlColor)
	{
	case CTLCOLOR_STATIC:
		return m_brush;
	default:
		break;
	}
	// TODO:  如果默认的不是所需画笔，则返回另一个画笔
	return hbr;
}

bool CConfigRecoder::raisePrivilege()
{
    HANDLE hToken;
    bool fOk = false;
    if (OpenProcessToken(GetCurrentProcess(), TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY, &hToken))
    {
        TOKEN_PRIVILEGES tp;
        tp.PrivilegeCount = 1;
        tp.Privileges[0].Attributes = SE_PRIVILEGE_ENABLED;
        if (!LookupPrivilegeValue(NULL, SE_DEBUG_NAME, &tp.Privileges[0].Luid))
        {
            CloseHandle(hToken);
            return fOk;
        }
        if (!AdjustTokenPrivileges(hToken, FALSE, &tp, sizeof(tp), NULL, NULL))
        {
            CloseHandle(hToken);
            return fOk;
        }

        fOk = (GetLastError() == ERROR_SUCCESS);
        CloseHandle(hToken);
    }
    return fOk;
}

void CConfigRecoder::GetAllProcess()
{
	if (!raisePrivilege())
	{
		CTipsDlg tips;
		tips.SetTips(_T("提升权限失败"));
		tips.DoModal();
	}

    DWORD aProcesses[1024], cbNeeded;
    if (!EnumProcesses(aProcesses, sizeof(aProcesses), &cbNeeded))
        return;

    for (unsigned int i = 0; (i<cbNeeded)&&(i<1024); i++)
    {
        HMODULE hMods[512] = { 0 };
        DWORD cbNeeded = 0;
        TCHAR szModName[MAX_PATH] = { 0 };
        TCHAR szPathName[MAX_PATH] = { 0 };
        //PROCESS_ALL_ACCESS
        HANDLE hProcess = ::OpenProcess(PROCESS_ALL_ACCESS, FALSE, aProcesses[i]);
        if (hProcess != NULL)
        {
            BOOL Wow64Process;
            IsWow64Process(hProcess, &Wow64Process);//判断是32位还是64位进程
            if (EnumProcessModulesEx(hProcess, hMods, sizeof(hMods), &cbNeeded, Wow64Process ? LIST_MODULES_32BIT : LIST_MODULES_64BIT))
                if (GetModuleBaseName(hProcess, hMods[0], szModName, _countof(szModName)) && GetModuleFileNameEx(hProcess, hMods[0], szPathName, _countof(szPathName)))
                    AddProcess2List(szModName, aProcesses[i], szPathName);
            CloseHandle(hProcess);
        }
    }

	// 排序
    m_iCol = 0;//第0列
    m_iCurrSortType = 1;//从上到下降序排列
	SortListCtrl(m_iCol, m_iCurrSortType);

    //设置列表头排序字段的图标
    HDITEM hDi = { HDI_FORMAT | HDI_IMAGE };
    m_pListctrlHeader->GetItem(0, &hDi);
    hDi.fmt |= HDF_IMAGE | HDF_BITMAP_ON_RIGHT;
    hDi.iImage = 0;
    m_pListctrlHeader->SetItem(0, &hDi);
}

void CConfigRecoder::GetAllProcess_V2()
{
    // 为进程的所有线程拍个快照  
    HANDLE hSnapshort = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
    if (hSnapshort == INVALID_HANDLE_VALUE)
        return;

    // 获得线程列表，里面记录了线程的详细信息，再使用Thread32First和Thread32Next遍历快照中记录的每个线程信息  
    PROCESSENTRY32 stcProcessInfo;
    stcProcessInfo.dwSize = sizeof(stcProcessInfo);
    BOOL bRet = Process32First(hSnapshort, &stcProcessInfo);
    while (bRet)
    {
        bRet = Process32Next(hSnapshort, &stcProcessInfo);

        HMODULE hMods[512] = { 0 };
        DWORD cbNeeded = 0;
        TCHAR szPathName[MAX_PATH] = { 0 };
        //PROCESS_ALL_ACCESS
        HANDLE hProcess = ::OpenProcess(PROCESS_ALL_ACCESS, FALSE, stcProcessInfo.th32ProcessID);
        if (hProcess != NULL)
        {
            BOOL Wow64Process;
            IsWow64Process(hProcess, &Wow64Process);//判断是32位还是64位进程
            if (EnumProcessModulesEx(hProcess, hMods, sizeof(hMods), &cbNeeded, Wow64Process ? LIST_MODULES_32BIT : LIST_MODULES_64BIT))
                if (GetModuleFileNameEx(hProcess, hMods[0], szPathName, _countof(szPathName)))
                    AddProcess2List(stcProcessInfo.szExeFile, stcProcessInfo.th32ProcessID, szPathName);
            CloseHandle(hProcess);
        }
    }
    CloseHandle(hSnapshort);

    // 排序
    m_iCol = 0;//第0列
    m_iCurrSortType = 1;//从上到下降序排列
    SortListCtrl(m_iCol, m_iCurrSortType);

    //设置列表头排序字段的图标
    HDITEM hDi = { HDI_FORMAT | HDI_IMAGE };
    m_pListctrlHeader->GetItem(0, &hDi);
    hDi.fmt |= HDF_IMAGE | HDF_BITMAP_ON_RIGHT;
    hDi.iImage = 0;
    m_pListctrlHeader->SetItem(0, &hDi);
}


void CConfigRecoder::AddProcess2List(TCHAR* name, DWORD pid, TCHAR* pathname)
{
    m_listctrl.InsertItem(m_listctrl_index, L"");
    m_listctrl.SetItemText(m_listctrl_index, 0, name);

    TCHAR tmp[12] = { 0 };
#ifdef UNICODE
    _itow_s(pid, tmp, 10);
#else
    _itoa_s(pid, tmp, 10);
#endif

    m_listctrl.SetItemText(m_listctrl_index, 1, tmp);
    if (pathname != NULL)
    {
        m_listctrl.SetItemText(m_listctrl_index, 2, pathname);
    }
    else
    {
        m_listctrl.SetItemText(m_listctrl_index, 2, _T(""));
    }
    
    ++m_listctrl_index;
}


int CConfigRecoder::listCompare(LPARAM lParam1, LPARAM lParam2, LPARAM lParamSort)
{
    DATA* pListCtrl = (DATA*)lParamSort;
    int col = pListCtrl->subitem;
    int colType = UNKNOW_TYPE;
    CListCtrl* plist = pListCtrl->plist;
    if (col == 1)
        colType = INT_TYPE;
    else if (col == 0)
        colType = TEXT_TYPE;
    else
        colType = UNKNOW_TYPE;

    //获取列头信息
    LVCOLUMN Vol;
    TCHAR szCol[MAX_PATH] = { 0 };
    Vol.pszText = szCol;
    Vol.mask = LVCF_TEXT;
    Vol.cchTextMax = sizeof(szCol);
    plist->GetColumn(col, &Vol);

    if (colType == INT_TYPE)//int型比较
    {
        CString strRow1 = plist->GetItemText(lParam1, col);
        CString strRow2 = plist->GetItemText(lParam2, col);
        int n1 = _ttoi(strRow1);
        int n2 = _ttoi(strRow2);
        if (m_iSortType ==1)
        {
            // 比较两个数 如果返回-1表示n1排在n2前面，如果返回1表示n1排在n2后面，如果相等返回0，这里没有判断相等的情况
            if (n1 < n2)
                return -1;
            else if (n1 > n2)
                return 1;
            else
                return 0;
        }
        else
        {
            if (n1 < n2)
                return 1;
            else if (n1 > n2)
                return -1;
            else
                return 0;
        }
    }
    else if (colType == TEXT_TYPE)
    {
        CString strItem1 = plist->GetItemText(lParam1, col);
        CString strItem2 = plist->GetItemText(lParam2, col);
        int iRet = 0;
        if (m_iSortType==1)
            iRet = _tcscmp(strItem1.MakeUpper().GetBuffer(MAX_PATH), strItem2.MakeUpper().GetBuffer(MAX_PATH));
        else
            iRet = _tcscmp(strItem2.MakeUpper().GetBuffer(MAX_PATH), strItem1.MakeUpper().GetBuffer(MAX_PATH));
        return iRet;
    }
    else
    {
        return 0;
    }
}

void CConfigRecoder::SortListCtrl(int subitem, bool bNotChange)
{
    //给表中的每一行加上索引号
    for (int i = 0; i < m_listctrl.GetItemCount(); ++i)
        m_listctrl.SetItemData(i, i);

    DATA data;
    data.subitem = subitem;
    data.plist = &m_listctrl;
    //bNotChange ? fav = fav : fav = !fav;
    m_iSortType = bNotChange;
    m_listctrl.SortItems(listCompare, (LPARAM)&data);
}

void CConfigRecoder::OnLvnColumnclickListProcess(NMHDR *pNMHDR, LRESULT *pResult)
{
    LPNMLISTVIEW pNMListView = reinterpret_cast<LPNMLISTVIEW>(pNMHDR);
    // TODO: 在此添加控件通知处理程序代码

    //设置图标
    int nCol = pNMListView->iSubItem;
    if (nCol == 2)
        return;
    HDITEM hDi = { HDI_FORMAT | HDI_IMAGE };
    if (nCol != m_iCol)
    {
        if (m_iCol >= 0)//删除之前排序表头字段图标
        {
            m_pListctrlHeader->GetItem(m_iCol, &hDi);
            hDi.mask = HDI_FORMAT;
            hDi.fmt &= (~HDF_IMAGE);
            //hDi.fmt &= ~HDF_IMAGE & ~HDF_BITMAP_ON_RIGHT;
            m_pListctrlHeader->SetItem(m_iCol, &hDi);
        }

        //显示新图标
        m_iCol = nCol;
        m_pListctrlHeader->GetItem(m_iCol, &hDi);
        hDi.fmt |= HDF_IMAGE | HDF_BITMAP_ON_RIGHT;
        if (m_iCol == 0 || m_iCol == 2)//进程名列或者路径列
        {
            hDi.iImage = 0;//与上次排序是不同字段,则本字段排序按降序
        }
        else if (m_iCol == 1)//PID列
        {
            hDi.iImage = 2;//与上次排序是不同字段,则本字段排序按降序
        }
        m_iCurrSortType = 1;
        m_pListctrlHeader->SetItem(m_iCol, &hDi);
    }
    else
    {
        m_pListctrlHeader->GetItem(nCol, &hDi);
        hDi.fmt |= HDF_IMAGE | HDF_BITMAP_ON_RIGHT;
        if (m_iCol == 0 || m_iCol == 2)//进程名列或者路径列
        {
            if (m_iCurrSortType == 0)
            {
                hDi.iImage = 0;
                m_iCurrSortType = 1;
            }
            else
            {
                hDi.iImage = 1;
                m_iCurrSortType = 0;
            }
        }
        else if (m_iCol == 1)//PID列
        {
            if (m_iCurrSortType == 0)
            {
                hDi.iImage = 2;
                m_iCurrSortType = 1;
            }
            else
            {
                hDi.iImage = 3;
                m_iCurrSortType = 0;
            }
        }
        m_pListctrlHeader->SetItem(nCol, &hDi);
    }

    SortListCtrl(m_iCol, m_iCurrSortType);

    *pResult = 0;
}

void CConfigRecoder::OnEnChangeEditSamprate()
{
    // TODO:  If this is a RICHEDIT control, the control will not
    // send this notification unless you override the CDialogEx::OnInitDialog()
    // function and call CRichEditCtrl().SetEventMask()
    // with the ENM_CHANGE flag ORed into the mask.

    // TODO:  Add your control notification handler code here
}

void CConfigRecoder::OnCbnSelchangeCmbSampunit()
{
    // TODO: Add your control notification handler code here
}


void CConfigRecoder::OnBnClickedCkNetup()
{
	// TODO: Add your control notification handler code here
}


void CConfigRecoder::OnBnClickedCkNetdown()
{
	// TODO: Add your control notification handler code here
}