376 lines
12 KiB
C
376 lines
12 KiB
C
/*
|
|
* ReactOS Task Manager
|
|
*
|
|
* dbgchnl.c
|
|
*
|
|
* Copyright (C) 2003 - 2004 Eric Pouech
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#define WIN32_LEAN_AND_MEAN /* Exclude rarely-used stuff from Windows headers */
|
|
#include <windows.h>
|
|
#include <commctrl.h>
|
|
#include <stdlib.h>
|
|
#include <malloc.h>
|
|
#include <memory.h>
|
|
#include <tchar.h>
|
|
#include <stdio.h>
|
|
#include <winnt.h>
|
|
#include <dbghelp.h>
|
|
|
|
#include "taskmgr.h"
|
|
#include "perfdata.h"
|
|
#include "column.h"
|
|
#include <ctype.h>
|
|
|
|
/* TODO:
|
|
* - the dialog box could be non modal
|
|
* - in that case,
|
|
* + could refresh channels from time to time
|
|
* + set the name of exec (and perhaps its pid) in dialog title
|
|
* - get a better UI (replace the 'x' by real tick boxes in list view)
|
|
* - enhance visual feedback: the list is large, and it's hard to get the
|
|
* right line when clicking on rightmost column (trace for example)
|
|
* - get rid of printfs (error reporting) and use real message boxes
|
|
* - include the column width settings in the full column management scheme
|
|
* - use more global settings (like having a temporary on/off
|
|
* setting for a fixme:s or err:s
|
|
*/
|
|
|
|
static DWORD (WINAPI *pSymSetOptions)(DWORD);
|
|
static BOOL (WINAPI *pSymInitialize)(HANDLE, PSTR, BOOL);
|
|
static DWORD (WINAPI *pSymLoadModule)(HANDLE, HANDLE, PSTR, PSTR, DWORD, DWORD);
|
|
static BOOL (WINAPI *pSymCleanup)(HANDLE);
|
|
static BOOL (WINAPI *pSymFromName)(HANDLE, LPSTR, PSYMBOL_INFO);
|
|
|
|
BOOL AreDebugChannelsSupported(void)
|
|
{
|
|
static HANDLE hDbgHelp /* = NULL */;
|
|
|
|
if (hDbgHelp) return TRUE;
|
|
|
|
if (!(hDbgHelp = LoadLibrary("dbghelp.dll"))) return FALSE;
|
|
pSymSetOptions = (void*)GetProcAddress(hDbgHelp, "SymSetOptions");
|
|
pSymInitialize = (void*)GetProcAddress(hDbgHelp, "SymInitialize");
|
|
pSymLoadModule = (void*)GetProcAddress(hDbgHelp, "SymLoadModule");
|
|
pSymFromName = (void*)GetProcAddress(hDbgHelp, "SymFromName");
|
|
pSymCleanup = (void*)GetProcAddress(hDbgHelp, "SymCleanup");
|
|
if (!pSymSetOptions || !pSymInitialize || !pSymLoadModule || !pSymCleanup || !pSymFromName)
|
|
{
|
|
FreeLibrary(hDbgHelp);
|
|
hDbgHelp = NULL;
|
|
return FALSE;
|
|
}
|
|
return TRUE;
|
|
}
|
|
|
|
static DWORD get_selected_pid(void)
|
|
{
|
|
LVITEM lvitem;
|
|
ULONG Index;
|
|
DWORD dwProcessId;
|
|
|
|
for (Index = 0; Index < (ULONG)ListView_GetItemCount(hProcessPageListCtrl); Index++)
|
|
{
|
|
memset(&lvitem, 0, sizeof(LVITEM));
|
|
|
|
lvitem.mask = LVIF_STATE;
|
|
lvitem.stateMask = LVIS_SELECTED;
|
|
lvitem.iItem = Index;
|
|
|
|
ListView_GetItem(hProcessPageListCtrl, &lvitem);
|
|
|
|
if (lvitem.state & LVIS_SELECTED)
|
|
break;
|
|
}
|
|
|
|
dwProcessId = PerfDataGetProcessId(Index);
|
|
|
|
if ((ListView_GetSelectedCount(hProcessPageListCtrl) != 1) || (dwProcessId == 0))
|
|
return 0;
|
|
return dwProcessId;
|
|
}
|
|
|
|
static int list_channel_CB(HANDLE hProcess, void* addr, char* buffer, void* user)
|
|
{
|
|
int j;
|
|
char val[2];
|
|
LVITEMA lvi;
|
|
int index;
|
|
HWND hChannelLV = (HWND)user;
|
|
|
|
memset(&lvi, 0, sizeof(lvi));
|
|
|
|
lvi.mask = LVIF_TEXT;
|
|
lvi.pszText = buffer + 1;
|
|
|
|
index = ListView_InsertItem(hChannelLV, &lvi);
|
|
if (index == -1) return 0;
|
|
|
|
val[1] = '\0';
|
|
for (j = 0; j < 4; j++)
|
|
{
|
|
val[0] = (buffer[0] & (1 << j)) ? 'x' : ' ';
|
|
ListView_SetItemText(hChannelLV, index, j + 1, val);
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
struct cce_user
|
|
{
|
|
const char* name; /* channel to look for */
|
|
unsigned value, mask; /* how to change channel */
|
|
unsigned done; /* number of successful changes */
|
|
unsigned notdone; /* number of unsuccessful changes */
|
|
};
|
|
|
|
/******************************************************************
|
|
* change_channel_CB
|
|
*
|
|
* Callback used for changing a given channel attributes
|
|
*/
|
|
static int change_channel_CB(HANDLE hProcess, void* addr, char* buffer, void* pmt)
|
|
{
|
|
struct cce_user* user = (struct cce_user*)pmt;
|
|
|
|
if (!user->name || !strcmp(buffer + 1, user->name))
|
|
{
|
|
buffer[0] = (buffer[0] & ~user->mask) | (user->value & user->mask);
|
|
if (WriteProcessMemory(hProcess, addr, buffer, 1, NULL))
|
|
user->done++;
|
|
else
|
|
user->notdone++;
|
|
}
|
|
return 1;
|
|
}
|
|
|
|
void* get_symbol(HANDLE hProcess, char* name, char* lib)
|
|
{
|
|
char buffer[sizeof(IMAGEHLP_SYMBOL) + 256];
|
|
SYMBOL_INFO* si = (SYMBOL_INFO*)buffer;
|
|
void* ret = NULL;
|
|
|
|
pSymSetOptions(SYMOPT_DEFERRED_LOADS | SYMOPT_PUBLICS_ONLY);
|
|
/* FIXME: the TRUE option is due to the face that dbghelp requires it
|
|
* when loading an ELF module
|
|
*/
|
|
if (pSymInitialize(hProcess, NULL, TRUE))
|
|
{
|
|
si->SizeOfStruct = sizeof(*si);
|
|
si->MaxNameLen = sizeof(buffer) - sizeof(IMAGEHLP_SYMBOL);
|
|
if (pSymLoadModule(hProcess, NULL, lib, NULL, 0, 0) &&
|
|
pSymFromName(hProcess, name, si))
|
|
ret = (void*)(ULONG_PTR)si->Address;
|
|
pSymCleanup(hProcess);
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
struct dll_option_layout
|
|
{
|
|
void* next;
|
|
void* prev;
|
|
char* const* channels;
|
|
int nb_channels;
|
|
};
|
|
|
|
typedef int (*EnumChannelCB)(HANDLE, void*, char*, void*);
|
|
|
|
/******************************************************************
|
|
* enum_channel
|
|
*
|
|
* Enumerates all known channels on process hProcess through callback
|
|
* ce.
|
|
*/
|
|
static int enum_channel(HANDLE hProcess, EnumChannelCB ce, void* user, unsigned unique)
|
|
{
|
|
struct dll_option_layout dol;
|
|
int i, ret = 1;
|
|
unsigned int j;
|
|
void* buf_addr;
|
|
unsigned char buffer[32];
|
|
void* addr;
|
|
const char** cache = NULL;
|
|
unsigned num_cache, used_cache;
|
|
|
|
if (!(addr = get_symbol(hProcess, "first_dll", "libwine.so"))) return -1;
|
|
if (unique)
|
|
cache = HeapAlloc(GetProcessHeap(), 0, (num_cache = 32) * sizeof(char*));
|
|
else
|
|
num_cache = 0;
|
|
used_cache = 0;
|
|
|
|
for (;
|
|
ret && addr && ReadProcessMemory(hProcess, addr, &dol, sizeof(dol), NULL);
|
|
addr = dol.next)
|
|
{
|
|
for (i = 0; i < dol.nb_channels; i++)
|
|
{
|
|
if (ReadProcessMemory(hProcess, (void*)(dol.channels + i), &buf_addr, sizeof(buf_addr), NULL) &&
|
|
ReadProcessMemory(hProcess, buf_addr, buffer, sizeof(buffer), NULL))
|
|
{
|
|
if (unique)
|
|
{
|
|
/* since some channels are defined in multiple compilation units,
|
|
* they will appear several times...
|
|
* so cache the channel's names we already reported and don't report
|
|
* them again
|
|
*/
|
|
for (j = 0; j < used_cache; j++)
|
|
if (!strcmp(cache[j], buffer + 1)) break;
|
|
if (j != used_cache) continue;
|
|
if (used_cache == num_cache)
|
|
cache = HeapReAlloc(GetProcessHeap(), 0, cache, (num_cache *= 2) * sizeof(char*));
|
|
cache[used_cache++] = strcpy(HeapAlloc(GetProcessHeap(), 0, strlen(buffer + 1) + 1),
|
|
buffer + 1);
|
|
}
|
|
ret = ce(hProcess, buf_addr, buffer, user);
|
|
}
|
|
}
|
|
}
|
|
if (unique)
|
|
{
|
|
for (j = 0; j < used_cache; j++) HeapFree(GetProcessHeap(), 0, (char*)cache[j]);
|
|
HeapFree(GetProcessHeap(), 0, cache);
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
static void DebugChannels_FillList(HWND hChannelLV)
|
|
{
|
|
HANDLE hProcess;
|
|
|
|
ListView_DeleteAllItems(hChannelLV);
|
|
|
|
hProcess = OpenProcess(PROCESS_VM_OPERATION | PROCESS_VM_READ, FALSE, get_selected_pid());
|
|
if (!hProcess) return; /* FIXME messagebox */
|
|
SendMessage(hChannelLV, WM_SETREDRAW, FALSE, 0);
|
|
enum_channel(hProcess, list_channel_CB, (void*)hChannelLV, TRUE);
|
|
SendMessage(hChannelLV, WM_SETREDRAW, TRUE, 0);
|
|
CloseHandle(hProcess);
|
|
}
|
|
|
|
static void DebugChannels_OnCreate(HWND hwndDlg)
|
|
{
|
|
HWND hLV = GetDlgItem(hwndDlg, IDC_DEBUG_CHANNELS_LIST);
|
|
LVCOLUMN lvc;
|
|
|
|
lvc.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
|
|
lvc.fmt = LVCFMT_LEFT;
|
|
lvc.pszText = _T("Debug Channel");
|
|
lvc.cx = 100;
|
|
ListView_InsertColumn(hLV, 0, &lvc);
|
|
|
|
lvc.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
|
|
lvc.fmt = LVCFMT_CENTER;
|
|
lvc.pszText = _T("Fixme");
|
|
lvc.cx = 55;
|
|
ListView_InsertColumn(hLV, 1, &lvc);
|
|
|
|
lvc.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
|
|
lvc.fmt = LVCFMT_CENTER;
|
|
lvc.pszText = _T("Err");
|
|
lvc.cx = 55;
|
|
ListView_InsertColumn(hLV, 2, &lvc);
|
|
|
|
lvc.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
|
|
lvc.fmt = LVCFMT_CENTER;
|
|
lvc.pszText = _T("Warn");
|
|
lvc.cx = 55;
|
|
ListView_InsertColumn(hLV, 3, &lvc);
|
|
|
|
lvc.mask = LVCF_FMT | LVCF_TEXT | LVCF_WIDTH;
|
|
lvc.fmt = LVCFMT_CENTER;
|
|
lvc.pszText = _T("Trace");
|
|
lvc.cx = 55;
|
|
ListView_InsertColumn(hLV, 4, &lvc);
|
|
|
|
DebugChannels_FillList(hLV);
|
|
}
|
|
|
|
static void DebugChannels_OnNotify(HWND hDlg, LPARAM lParam)
|
|
{
|
|
NMHDR* nmh = (NMHDR*)lParam;
|
|
|
|
switch (nmh->code)
|
|
{
|
|
case NM_CLICK:
|
|
if (nmh->idFrom == IDC_DEBUG_CHANNELS_LIST)
|
|
{
|
|
LVHITTESTINFO lhti;
|
|
HWND hChannelLV;
|
|
HANDLE hProcess;
|
|
NMITEMACTIVATE* nmia = (NMITEMACTIVATE*)lParam;
|
|
|
|
hProcess = OpenProcess(PROCESS_VM_OPERATION | PROCESS_VM_READ | PROCESS_VM_WRITE, FALSE, get_selected_pid());
|
|
if (!hProcess) return; /* FIXME message box */
|
|
lhti.pt = nmia->ptAction;
|
|
hChannelLV = GetDlgItem(hDlg, IDC_DEBUG_CHANNELS_LIST);
|
|
SendMessage(hChannelLV, LVM_SUBITEMHITTEST, 0, (LPARAM)&lhti);
|
|
if (nmia->iSubItem >= 1 && nmia->iSubItem <= 4)
|
|
{
|
|
TCHAR val[2];
|
|
TCHAR name[32];
|
|
unsigned bitmask = 1 << (lhti.iSubItem - 1);
|
|
struct cce_user user;
|
|
|
|
ListView_GetItemText(hChannelLV, lhti.iItem, 0, name, sizeof(name) / sizeof(name[0]));
|
|
ListView_GetItemText(hChannelLV, lhti.iItem, lhti.iSubItem, val, sizeof(val) / sizeof(val[0]));
|
|
user.name = name;
|
|
user.value = (val[0] == 'x') ? 0 : bitmask;
|
|
user.mask = bitmask;
|
|
user.done = user.notdone = 0;
|
|
enum_channel(hProcess, change_channel_CB, &user, FALSE);
|
|
if (user.done)
|
|
{
|
|
val[0] ^= ('x' ^ ' ');
|
|
ListView_SetItemText(hChannelLV, lhti.iItem, lhti.iSubItem, val);
|
|
}
|
|
if (user.notdone)
|
|
printf("Some channel instance weren't correctly set\n");
|
|
}
|
|
CloseHandle(hProcess);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
static LRESULT CALLBACK DebugChannelsDlgProc(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
|
|
{
|
|
switch (message)
|
|
{
|
|
case WM_INITDIALOG:
|
|
DebugChannels_OnCreate(hDlg);
|
|
return TRUE;
|
|
case WM_COMMAND:
|
|
if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
|
|
{
|
|
EndDialog(hDlg, LOWORD(wParam));
|
|
return TRUE;
|
|
}
|
|
break;
|
|
case WM_NOTIFY:
|
|
DebugChannels_OnNotify(hDlg, lParam);
|
|
break;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
void ProcessPage_OnDebugChannels(void)
|
|
{
|
|
DialogBox(hInst, (LPCTSTR)IDD_DEBUG_CHANNELS_DIALOG, hMainWnd, (DLGPROC)DebugChannelsDlgProc);
|
|
}
|