/*
 * This is a part of the BugTrap package.
 * Copyright (c) 2005-2009 IntelleSoft.
 * All rights reserved.
 *
 * Description: Machine State dialog.
 * Author: Maksim Pyatkovskiy.
 *
 * This source code is only intended as a supplement to the
 * BugTrap package reference and related electronic documentation
 * provided with the product. See these sources for detailed
 * information regarding the BugTrap package.
 */

#include "StdAfx.h"
#include "resource.h"
#include "BugTrapUI.h"
#include "WaitCursor.h"
#include "BugTrapUtils.h"
#include "MachineStateDlg.h"
#include "LayoutManager.h"
#include "TextView.h"
#include "Globals.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

/**
 * @addtogroup BugTrapUI BugTrap Graphical User Interface
 * @{
 */

/// Process list column identifiers.
enum PROCESS_LIST_COLUMN_ID
{
	/// PID.
	CID_PROCESS_ID,
	/// Process name.
	CID_PROCESS_NAME
};

/// Process list column identifiers.
enum PROCESS_MODULES_LIST_COLUMN_ID
{
	/// Module name.
	CID_MODULE_NAME,
	/// Module version.
	CID_MODULE_VERSION,
	/// Base address.
	CID_MODULE_BASE
};


/// Control layouts for Machine State dialog.
static LAYOUT_INFO g_arrMachineStateLayout[] =
{
	LAYOUT_INFO(IDC_PROCESS_LIST_FRAME,        ALIGN_LEFT,  ALIGN_TOP,    ALIGN_RIGHT, ALIGN_CENTER),
	LAYOUT_INFO(IDC_PROCESS_LIST,              ALIGN_LEFT,  ALIGN_TOP,    ALIGN_RIGHT, ALIGN_CENTER),
	LAYOUT_INFO(IDC_PROCESS_MODULES_LIST_FRAME, ALIGN_LEFT,  ALIGN_CENTER, ALIGN_RIGHT, ALIGN_BOTTOM),
	LAYOUT_INFO(IDC_PROCESS_MODULES_LIST,       ALIGN_LEFT,  ALIGN_CENTER, ALIGN_RIGHT, ALIGN_BOTTOM),
	LAYOUT_INFO(IDCANCEL,                      ALIGN_RIGHT, ALIGN_TOP,    ALIGN_RIGHT, ALIGN_TOP)
};

/// Dialog layout manager.
static CLayoutManager g_LayoutMgr;
/// Process list sort order.
static CListViewOrder g_ProcessListOrder;
/// Modules list sort order.
static CListViewOrder g_ModulesListOrder;

/**
 * @brief WM_COMMAND handler of Machine State dialog.
 * @param hwnd - window handle.
 * @param id - control ID.
 * @param hwndCtl - control handle.
 * @param codeNotify - notification code.
 */
static void MachineStateDlg_OnCommand(HWND hwnd, int id, HWND hwndCtl, UINT codeNotify)
{
	hwndCtl;
	switch (codeNotify)
	{
	case BN_CLICKED:
		switch (id)
		{
		case IDOK:
		case IDCANCEL:
			EndDialog(hwnd, FALSE);
			break;
		}
		break;
	}
}

/**
 * @brief WM_INITDIALOG handler of Machine State dialog.
 * @param hwnd - window handle.
 * @param hwndFocus - system-defined focus window.
 * @param lParam - user-defined parameter.
 * @return true to setup focus to system-defined control.
 */
static BOOL MachineStateDlg_OnInitDialog(HWND hwnd, HWND hwndFocus, LPARAM lParam)
{
	lParam; hwndFocus;

	_ASSERTE(g_pResManager != NULL);
	if (g_pResManager->m_hBigAppIcon)
		SendMessage(hwnd, WM_SETICON, ICON_BIG, (LPARAM)g_pResManager->m_hBigAppIcon);
	if (g_pResManager->m_hSmallAppIcon)
		SendMessage(hwnd, WM_SETICON, ICON_SMALL, (LPARAM)g_pResManager->m_hSmallAppIcon);
	CenterWindow(hwnd, GetParent(hwnd));
	g_LayoutMgr.InitLayout(hwnd, g_arrMachineStateLayout, countof(g_arrMachineStateLayout));

	HWND hwndProcessList = GetDlgItem(hwnd, IDC_PROCESS_LIST);
	ListView_SetExtendedListViewStyle(hwndProcessList, LVS_EX_FULLROWSELECT | LVS_EX_LABELTIP);

	HWND hwndModuleList = GetDlgItem(hwnd, IDC_PROCESS_MODULES_LIST);
	ListView_SetExtendedListViewStyle(hwndModuleList, LVS_EX_FULLROWSELECT | LVS_EX_LABELTIP);

	RECT rcList;
	GetClientRect(hwndProcessList, &rcList);
	rcList.right -= GetSystemMetrics(SM_CXHSCROLL);

	TCHAR szColumnTitle[64];
	LVCOLUMN lvc;
	ZeroMemory(&lvc, sizeof(lvc));
	lvc.mask = LVCF_TEXT | LVCF_WIDTH;
	lvc.pszText = szColumnTitle;

	lvc.cx = rcList.right / 5;
	LoadString(g_hInstance, IDS_COLUMN_PID, szColumnTitle, countof(szColumnTitle));
	ListView_InsertColumn(hwndProcessList, CID_PROCESS_ID, &lvc);

	lvc.cx = rcList.right * 4 / 5;
	LoadString(g_hInstance, IDS_COLUMN_PROCESS, szColumnTitle, countof(szColumnTitle));
	ListView_InsertColumn(hwndProcessList, CID_PROCESS_NAME, &lvc);

	lvc.cx = rcList.right / 2;
	LoadString(g_hInstance, IDS_COLUMN_MODULE, szColumnTitle, countof(szColumnTitle));
	ListView_InsertColumn(hwndModuleList, CID_MODULE_NAME, &lvc);

	lvc.cx = rcList.right / 4;
	LoadString(g_hInstance, IDS_COLUMN_VERSION, szColumnTitle, countof(szColumnTitle));
	ListView_InsertColumn(hwndModuleList, CID_MODULE_VERSION, &lvc);

	lvc.cx = rcList.right / 4;
	LoadString(g_hInstance, IDS_COLUMN_BASE, szColumnTitle, countof(szColumnTitle));
	ListView_InsertColumn(hwndModuleList, CID_MODULE_BASE, &lvc);

	CEnumProcess::CProcessEntry ProcEntry;
	if (g_pEnumProc->GetProcessFirst(ProcEntry))
	{
		LVITEM lvi;
		ZeroMemory(&lvi, sizeof(lvi));
		lvi.mask = LVIF_TEXT | LVIF_PARAM;
		int iItemPos = 0;
		do
		{
			TCHAR szProcessID[64];
			_ultot_s(ProcEntry.m_dwProcessID, szProcessID, countof(szProcessID), 10);
			lvi.iItem = iItemPos;
			lvi.pszText = szProcessID;
			lvi.lParam = ProcEntry.m_dwProcessID;
			ListView_InsertItem(hwndProcessList, &lvi);
			ListView_SetItemText(hwndProcessList, iItemPos, CID_PROCESS_NAME, ProcEntry.m_szProcessName);
			++iItemPos;
		}
		while (g_pEnumProc->GetProcessNext(ProcEntry));
	}

	// LVM_SETIMAGELIST resets header control image list
	g_ProcessListOrder.InitList(hwndProcessList);
	g_ModulesListOrder.InitList(hwndModuleList);
	return TRUE;
}

/**
 * @brief WM_DESTROY handler of Machine State dialog.
 * @param hwnd - window handle.
 */
static void MachineStateDlg_OnDestroy(HWND hwnd)
{
	hwnd;
	g_pEnumProc->Close();
}

/**
 * @brief WM_SIZE handler of Machine State dialog.
 * @param hwnd - window handle.
 * @param state - window state.
 * @param cx - window width.
 * @param cy - window height.
 */
static void MachineStateDlg_OnSize(HWND hwnd, UINT state, int cx, int cy)
{
	hwnd; cx; cy; state;
	g_LayoutMgr.ApplyLayout();
}

/**
 * @brief WM_GETMINMAXINFO handler of Machine State dialog.
 * @param hwnd - window handle.
 * @param pMinMaxInfo - window min-max info.
 */
static void MachineStateDlg_OnGetMinMaxInfo(HWND hwnd, PMINMAXINFO pMinMaxInfo)
{
	hwnd;
	pMinMaxInfo->ptMinTrackSize = g_LayoutMgr.GetMinTrackSize();
}

/**
 * @brief WM_NOTIFY handler of Machine State dialog.
 * @param hwnd - window handle.
 * @param idCtrl - identifier of the common control sending the message.
 * @param pnmh - pointer to an NMHDR structure that contains the notification code and additional information.
 */
static LRESULT MachineStateDlg_OnNotify(HWND hwnd, int idCtrl, LPNMHDR pnmh)
{
	switch (pnmh->code)
	{
	case LVN_ITEMCHANGED:
		{
			if (idCtrl != IDC_PROCESS_LIST)
				break;
			LPNMLISTVIEW pnmv = (LPNMLISTVIEW)pnmh;
			if ((pnmv->uNewState & LVIS_SELECTED) != (pnmv->uOldState & LVIS_SELECTED))
			{
				if (pnmv->uNewState & LVIS_SELECTED)
				{
					CWaitCursor wait(true);
					HWND hwndModuleList = GetDlgItem(hwnd, IDC_PROCESS_MODULES_LIST);
					SendMessage(hwndModuleList, WM_SETREDRAW, FALSE, 0);
					DisplayWaitBanner(hwndModuleList);

					LVITEM lvi;
					ZeroMemory(&lvi, sizeof(lvi));
					lvi.mask = LVIF_TEXT;
					CEnumProcess::CModuleEntry module;
					if (g_pEnumProc->GetModuleFirst((DWORD)pnmv->lParam, module))
					{
						int iItemPos = 0;
						do
						{
							lvi.iItem = iItemPos;
							lvi.pszText = module.m_szModuleName;
							ListView_InsertItem(hwndModuleList, &lvi);
							TCHAR szVersionString[64];
							if (CSymEngine::GetVersionString(module.m_szModuleName, szVersionString, countof(szVersionString)))
								ListView_SetItemText(hwndModuleList, iItemPos, CID_MODULE_VERSION, szVersionString);
							TCHAR szTempBuf[32];
#if defined _WIN64
							_stprintf_s(szTempBuf, countof(szTempBuf), _T("%016X"), (DWORD_PTR)module.m_pLoadBase);
#elif defined _WIN32
							_stprintf_s(szTempBuf, countof(szTempBuf), _T("%08X"), (DWORD_PTR)module.m_pLoadBase);
#endif
							ListView_SetItemText(hwndModuleList, iItemPos, CID_MODULE_BASE, szTempBuf);
							++iItemPos;
						}
						while (g_pEnumProc->GetModuleNext((DWORD)pnmv->lParam, module));
					}
					else
					{
						TCHAR szMessage[100];
						LoadString(g_hInstance, IDS_ERROR_NOT_AVAILABLE, szMessage, countof(szMessage));
						lvi.iItem = 0;
						lvi.pszText = szMessage;
						ListView_InsertItem(hwndModuleList, &lvi);
					}

					SendMessage(hwndModuleList, WM_SETREDRAW, TRUE, 0);
					InvalidateRect(hwndModuleList, NULL, TRUE);
				}
				else
				{
					HWND hwndModuleList = GetDlgItem(hwnd, IDC_PROCESS_MODULES_LIST);
					ListView_DeleteAllItems(hwndModuleList);
				}
			}
		}
		break;
	case LVN_COLUMNCLICK:
		{
			LPNMLISTVIEW pnmv = (LPNMLISTVIEW)pnmh;
			if (idCtrl == IDC_PROCESS_LIST)
			{
				HWND hwndProcessList = GetDlgItem(hwnd, IDC_PROCESS_LIST);
				g_ProcessListOrder.ToggleSortParams(hwndProcessList, pnmv->iSubItem);
				LISTVIEW_SORT_PARAMS lvSortParams;
				lvSortParams.hwndList = hwndProcessList;
				lvSortParams.iColumnNumber = pnmv->iSubItem;
				lvSortParams.bAscending = g_ProcessListOrder.GetSortOrder();
				switch (pnmv->iSubItem)
				{
				case CID_PROCESS_ID:
					lvSortParams.eCompareType = LISTVIEW_SORT_PARAMS::ICT_INTEGER;
					break;
				case CID_PROCESS_NAME:
					lvSortParams.eCompareType = LISTVIEW_SORT_PARAMS::ICT_STRING;
					break;
				}
				ListView_SortItemsEx(hwndProcessList, &ListViewCompareFunc, (LPARAM)&lvSortParams);
			}
			else if (idCtrl == IDC_PROCESS_MODULES_LIST)
			{
				HWND hwndModuleList = GetDlgItem(hwnd, IDC_PROCESS_MODULES_LIST);
				g_ModulesListOrder.ToggleSortParams(hwndModuleList, pnmv->iSubItem);
				LISTVIEW_SORT_PARAMS lvSortParams;
				lvSortParams.hwndList = hwndModuleList;
				lvSortParams.iColumnNumber = pnmv->iSubItem;
				lvSortParams.bAscending = g_ModulesListOrder.GetSortOrder();
				switch (pnmv->iSubItem)
				{
				case CID_MODULE_NAME:
				case CID_MODULE_VERSION:
					lvSortParams.eCompareType = LISTVIEW_SORT_PARAMS::ICT_STRING;
					break;
				case CID_MODULE_BASE:
					lvSortParams.eCompareType = LISTVIEW_SORT_PARAMS::ICT_HEXADECIMAL;
					break;
				}
				ListView_SortItemsEx(hwndModuleList, &ListViewCompareFunc, (LPARAM)&lvSortParams);
			}
		}
		break;
	}
	return FALSE;
}

/**
 * @brief Dialog procedure of Machine State dialog.
 * @param hwndDlg - window handle.
 * @param uMsg - message identifier.
 * @param wParam - 1st message parameter.
 * @param lParam - 2nd message parameter.
 * @return message result.
 */
INT_PTR CALLBACK MachineStateDlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	switch (uMsg)
	{
	HANDLE_MSG(hwndDlg, WM_INITDIALOG, MachineStateDlg_OnInitDialog);
	HANDLE_MSG(hwndDlg, WM_DESTROY, MachineStateDlg_OnDestroy);
	HANDLE_MSG(hwndDlg, WM_COMMAND, MachineStateDlg_OnCommand);
	HANDLE_MSG(hwndDlg, WM_SIZE, MachineStateDlg_OnSize);
	HANDLE_MSG(hwndDlg, WM_GETMINMAXINFO, MachineStateDlg_OnGetMinMaxInfo);
	HANDLE_MSG(hwndDlg, WM_NOTIFY, MachineStateDlg_OnNotify);
	default: return FALSE;
	}
}

/** @} */