int AddStickyStatusIcon(WPARAM wParam, LPARAM lParam) {
StatusIconData *sid = (StatusIconData *)lParam;
struct StatusIconListNode *siln = (struct StatusIconListNode *)mir_alloc(sizeof(struct StatusIconListNode));
struct StatusIconListNode *siln2 = status_icon_list;
siln->sid.cbSize = sid->cbSize;
siln->sid.szModule = mir_strdup(sid->szModule);
siln->sid.dwId = sid->dwId;
siln->sid.hIcon = DuplicateIcon(NULL, sid->hIcon);
siln->sid.hIconDisabled = DuplicateIcon(NULL, sid->hIconDisabled);
siln->sid.flags = sid->flags;
if(sid->szTooltip) siln->sid.szTooltip = mir_strdup(sid->szTooltip);
else siln->sid.szTooltip = 0;
siln->next = NULL;
while(siln2 && siln2->next) {
siln2 = siln2->next;
}
if (siln2) {
siln2->next = siln;
} else {
status_icon_list = siln;
}
status_icon_list_size++;
WindowList_Broadcast(g_dat->hParentWindowList, DM_STATUSICONCHANGE, 0, 0);
return 0;
}
static int ModifyStatusIcon(WPARAM wParam, LPARAM lParam) {
HANDLE hContact = (HANDLE)wParam;
StatusIconData *sid = (StatusIconData *)lParam;
struct StatusIconListNode *current = status_icon_list;
while(current) {
if(strcmp(current->sid.szModule, sid->szModule) == 0 && current->sid.dwId == sid->dwId) {
if(!hContact) {
current->sid.flags = sid->flags;
if(sid->hIcon) {
DestroyIcon(current->sid.hIcon);
current->sid.hIcon = DuplicateIcon(NULL, sid->hIcon);
}
if(sid->hIconDisabled) {
DestroyIcon(current->sid.hIconDisabled);
current->sid.hIconDisabled = DuplicateIcon(NULL, sid->hIconDisabled);
}
if(sid->szTooltip) {
if(current->sid.szTooltip) mir_free(current->sid.szTooltip);
current->sid.szTooltip = mir_strdup(sid->szTooltip);
}
WindowList_Broadcast(g_dat->hParentWindowList, DM_STATUSICONCHANGE, 0, 0);
} else {
char buff[256];
HWND hwnd;
sprintf(buff, "SRMMStatusIconFlags%d", (int)sid->dwId);
DBWriteContactSettingByte(hContact, sid->szModule, buff, (BYTE)sid->flags);
hwnd = WindowList_Find(g_dat->hMessageWindowList, hContact);
if (hwnd == NULL) {
hwnd = SM_FindWindowByContact(hContact);
}
if (hwnd != NULL) {
PostMessage(GetParent(hwnd), DM_STATUSICONCHANGE, 0, 0);
}
}
return 0;
}
current = current->next;
}
return 1;
}
PET_PROCESS_ICON EtProcIconCreateProcessIcon(
__in HICON Icon
)
{
PET_PROCESS_ICON processIcon;
processIcon = PhAllocate(sizeof(ET_PROCESS_ICON));
processIcon->RefCount = 1;
processIcon->Icon = DuplicateIcon(NULL, Icon);
return processIcon;
}
ICON_IMAGE load_icon_default(ICON_TYPE type) {
WORD idi;
ICON_IMAGE img;
switch(type) {
case ICON_ACTIVE:
idi=IDI_STUNNEL_ACTIVE;
break;
case ICON_ERROR:
idi=IDI_STUNNEL_ERROR;
break;
case ICON_IDLE:
idi=IDI_STUNNEL_IDLE;
break;
default:
return NULL;
}
img=LoadImage(ghInst, MAKEINTRESOURCE(idi), IMAGE_ICON,
GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), 0);
return DuplicateIcon(NULL, img);
}
ShortCutItem *ShortCutManager::MakeSystemShortCut(int systemIndex)
{
MYICON_INFO info = MyGetIconInfo(m_initIcon[systemIndex]);
ShortCutItem *p = new ShortCutItem(this, m_dockPanelDlg, m_initIconTitle[systemIndex].GetBuffer(), DuplicateIcon(NULL, m_initIcon[systemIndex]), true, systemIndex);
RECT rc;
p->Create((CWnd *)m_dockPanelDlg, CRect(10, 10, 210, 30), _T("Nice Work"));
return p;
}
static LRESULT CALLBACK TSButtonWndProc(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
MButtonCtrl *bct = (MButtonCtrl *) GetWindowLongPtr(hwndDlg, 0);
switch(msg) {
case WM_NCCREATE: {
SetWindowLong(hwndDlg, GWL_STYLE, GetWindowLong(hwndDlg, GWL_STYLE) | BS_OWNERDRAW);
bct = reinterpret_cast<MButtonCtrl *>(malloc(sizeof(MButtonCtrl)));
if(bct == NULL)
return FALSE;
bct->hwnd = hwndDlg;
bct->stateId = PBS_NORMAL;
bct->focus = 0;
bct->hFont = reinterpret_cast<HFONT>(GetStockObject(DEFAULT_GUI_FONT));
bct->arrow = NULL;
bct->defbutton = 0;
bct->hIcon = bct->hIconPrivate = 0;
bct->iIcon = 0;
bct->hIml = 0;
bct->hBitmap = NULL;
bct->pushBtn = 0;
bct->pbState = 0;
bct->hThemeButton = NULL;
bct->hThemeToolbar = NULL;
bct->cHot = 0;
bct->flatBtn = 0;
bct->bThemed = FALSE;
bct->bSkinned = bct->bSendOnDown = 0;
bct->buttonItem = NULL;
LoadTheme(bct);
SetWindowLongPtr(hwndDlg, 0, (LONG_PTR) bct);
if(((CREATESTRUCTA *) lParam)->lpszName)
SetWindowText(hwndDlg, ((CREATESTRUCT *) lParam)->lpszName);
return TRUE;
}
case WM_DESTROY: {
if(bct) {
if(hwndToolTips) {
TOOLINFO ti;
ZeroMemory(&ti, sizeof(ti));
ti.cbSize = sizeof(ti);
ti.uFlags = TTF_IDISHWND;
ti.hwnd = bct->hwnd;
ti.uId = (UINT_PTR) bct->hwnd;
if(SendMessage(hwndToolTips, TTM_GETTOOLINFO, 0, (LPARAM) &ti)) {
SendMessage(hwndToolTips, TTM_DELTOOL, 0, (LPARAM) &ti);
}
if(SendMessage(hwndToolTips, TTM_GETTOOLCOUNT, 0, (LPARAM) &ti) == 0) {
DestroyWindow(hwndToolTips);
hwndToolTips = NULL;
}
}
if(bct->hIconPrivate)
DestroyIcon(bct->hIconPrivate);
DestroyTheme(bct);
free(bct);
}
SetWindowLongPtr(hwndDlg, 0, 0);
break; // DONT! fall thru
}
case WM_SETTEXT: {
bct->cHot = 0;
if((char*) lParam) {
char *tmp = (char *) lParam;
while(*tmp) {
if(*tmp == '&' && *(tmp + 1)) {
bct->cHot = tolower(*(tmp + 1));
break;
}
tmp++;
}
InvalidateRect(bct->hwnd, NULL, TRUE);
lstrcpyn(bct->szText, (wchar_t *)lParam, 127);
bct->szText[127] = 0;
}
break;
}
case WM_SYSKEYUP:
if(bct->stateId != PBS_DISABLED && bct->cHot && bct->cHot == tolower((int) wParam)) {
if(bct->pushBtn) {
if(bct->pbState)
bct->pbState = 0;
else
bct->pbState = 1;
InvalidateRect(bct->hwnd, NULL, TRUE);
}
if(!bct->bSendOnDown)
SendMessage(GetParent(hwndDlg), WM_COMMAND, MAKELONG(GetDlgCtrlID(hwndDlg), BN_CLICKED), (LPARAM) hwndDlg);
return 0;
}
break;
case WM_THEMECHANGED: {
if(bct->bThemed)
LoadTheme(bct);
InvalidateRect(bct->hwnd, NULL, TRUE); // repaint it
break;
}
case WM_SETFONT: {
bct->hFont = (HFONT) wParam; // maybe we should redraw?
break;
}
case WM_PAINT: {
PAINTSTRUCT ps;
HDC hdcPaint;
hdcPaint = BeginPaint(hwndDlg, &ps);
if(hdcPaint) {
PaintWorker(bct, hdcPaint);
EndPaint(hwndDlg, &ps);
}
break;
}
case BM_GETIMAGE:
if(wParam == IMAGE_ICON)
return (LRESULT)(bct->hIconPrivate ? bct->hIconPrivate : bct->hIcon);
break;
case BM_SETIMAGE:
if(!lParam)
break;
bct->hIml = 0;
bct->iIcon = 0;
if(wParam == IMAGE_ICON) {
ICONINFO ii = {0};
BITMAP bm = {0};
if(bct->hIconPrivate) {
DestroyIcon(bct->hIconPrivate);
bct->hIconPrivate = 0;
}
GetIconInfo((HICON) lParam, &ii);
GetObject(ii.hbmColor, sizeof(bm), &bm);
if(bm.bmWidth > CXSMICON || bm.bmHeight > CYSMICON) {
HIMAGELIST hImageList;
hImageList = ImageList_Create(CXSMICON, CYSMICON, ILC_COLOR32 | ILC_MASK, 1, 0);
ImageList_AddIcon(hImageList, (HICON) lParam);
bct->hIconPrivate = ImageList_GetIcon(hImageList, 0, ILD_NORMAL);
ImageList_RemoveAll(hImageList);
ImageList_Destroy(hImageList);
bct->hIcon = 0;
} else {
bct->hIcon = (HICON) lParam;
bct->hIconPrivate = 0;
}
DeleteObject(ii.hbmMask);
DeleteObject(ii.hbmColor);
bct->hBitmap = NULL;
InvalidateRect(bct->hwnd, NULL, TRUE);
} else if(wParam == IMAGE_BITMAP) {
bct->hBitmap = (HBITMAP) lParam;
if(bct->hIconPrivate)
DestroyIcon(bct->hIconPrivate);
bct->hIcon = bct->hIconPrivate = NULL;
InvalidateRect(bct->hwnd, NULL, TRUE);
}
break;
case BM_SETPRIVATEICON:
bct->hIml = 0;
bct->iIcon = 0;
{
if(bct->hIconPrivate)
DestroyIcon(bct->hIconPrivate);
bct->hIconPrivate = DuplicateIcon(g_hInst, (HICON) lParam);
bct->hIcon = 0;
break;
}
case BM_SETIMLICON: {
if(bct->hIconPrivate)
DestroyIcon(bct->hIconPrivate);
bct->hIml = (HIMAGELIST) wParam;
bct->iIcon = (int) lParam;
bct->hIcon = bct->hIconPrivate = 0;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
}
case BM_SETSKINNED:
bct->bSkinned = (DWORD)lParam;
bct->bThemed = bct->bSkinned ? FALSE : bct->bThemed;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case BM_SETBTNITEM:
bct->buttonItem = (TButtonItem *)lParam;
break;
case BM_SETASMENUACTION:
bct->bSendOnDown = wParam ? TRUE : FALSE;
return 0;
case BM_SETCHECK:
if(!bct->pushBtn)
break;
if(wParam == BST_CHECKED) {
bct->pbState = 1;
bct->stateId = PBS_PRESSED;
} else if(wParam == BST_UNCHECKED) {
bct->pbState = 0;
bct->stateId = PBS_NORMAL;
}
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case BM_GETCHECK:
if(bct->pushBtn) {
return bct->pbState ? BST_CHECKED : BST_UNCHECKED;
}
return 0;
case BUTTONSETARROW:
// turn arrow on/off
if(wParam) {
if(!bct->arrow)
bct->arrow = (HICON) LoadImage(g_hInst, MAKEINTRESOURCE(IDI_MINIMIZE), IMAGE_ICON, CXSMICON, CYSMICON, 0);
} else {
if(bct->arrow) {
DestroyIcon(bct->arrow);
bct->arrow = NULL;
}
}
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case BUTTONSETDEFAULT:
bct->defbutton = wParam ? 1 : 0;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case BUTTONSETASPUSHBTN:
bct->pushBtn = 1;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case BUTTONSETASFLATBTN:
bct->flatBtn = lParam == 0 ? 1 : 0;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case BUTTONSETASFLATBTN + 10:
bct->bThemed = lParam ? TRUE : FALSE;
bct->bSkinned = bct->bThemed ? 0 : bct->bSkinned;
break;
case BUTTONADDTOOLTIP: {
TOOLINFO ti;
if(!(char*) wParam)
break;
if(!hwndToolTips) {
hwndToolTips = CreateWindowEx(WS_EX_TOPMOST, TOOLTIPS_CLASS, _T(""), WS_POPUP, 0, 0, 0, 0, NULL, NULL, GetModuleHandle(NULL), NULL);
SetWindowPos(hwndToolTips, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOSIZE | SWP_NOMOVE);
}
ZeroMemory(&ti, sizeof(ti));
ti.cbSize = sizeof(ti);
ti.uFlags = TTF_IDISHWND;
ti.hwnd = bct->hwnd;
ti.uId = (UINT_PTR) bct->hwnd;
if(SendMessage(hwndToolTips, TTM_GETTOOLINFO, 0, (LPARAM) &ti)) {
SendMessage(hwndToolTips, TTM_DELTOOL, 0, (LPARAM) &ti);
}
ti.uFlags = TTF_IDISHWND | TTF_SUBCLASS;
ti.uId = (UINT_PTR) bct->hwnd;
ti.lpszText = (wchar_t *) wParam;
SendMessage(hwndToolTips, TTM_ADDTOOL, 0, (LPARAM) &ti);
break;
}
case WM_SETFOCUS:
// set keybord focus and redraw
bct->focus = 1;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case WM_KILLFOCUS:
// kill focus and redraw
bct->focus = 0;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case WM_WINDOWPOSCHANGED:
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
case WM_ENABLE: {
bct->stateId = wParam ? PBS_NORMAL : PBS_DISABLED;
InvalidateRect(bct->hwnd, NULL, TRUE);
break;
}
case WM_MOUSELEAVE: {
if(bct->stateId != PBS_DISABLED) {
// don't change states if disabled
bct->stateId = PBS_NORMAL;
InvalidateRect(bct->hwnd, NULL, TRUE);
}
break;
}
case WM_LBUTTONDOWN: {
if(bct->stateId != PBS_DISABLED && bct->stateId != PBS_PRESSED) {
bct->stateId = PBS_PRESSED;
InvalidateRect(bct->hwnd, NULL, TRUE);
if(bct->bSendOnDown) {
SendMessage(GetParent(hwndDlg), WM_COMMAND, MAKELONG(GetDlgCtrlID(hwndDlg), BN_CLICKED), (LPARAM) hwndDlg);
bct->stateId = PBS_NORMAL;
InvalidateRect(bct->hwnd, NULL, TRUE);
}
}
break;
}
case WM_LBUTTONUP: {
if(bct->pushBtn) {
if(bct->pbState)
bct->pbState = 0;
else
bct->pbState = 1;
}
if(bct->stateId != PBS_DISABLED) {
// don't change states if disabled
if(msg == WM_LBUTTONUP)
bct->stateId = PBS_HOT;
else
bct->stateId = PBS_NORMAL;
InvalidateRect(bct->hwnd, NULL, TRUE);
}
if(!bct->bSendOnDown)
SendMessage(GetParent(hwndDlg), WM_COMMAND, MAKELONG(GetDlgCtrlID(hwndDlg), BN_CLICKED), (LPARAM) hwndDlg);
break;
}
case WM_MOUSEMOVE:
if(bct->stateId == PBS_NORMAL) {
bct->stateId = PBS_HOT;
InvalidateRect(bct->hwnd, NULL, TRUE);
}
// Call timer, used to start cheesy TrackMouseEvent faker
SetTimer(hwndDlg, BUTTON_POLLID, BUTTON_POLLDELAY, NULL);
break;
case WM_NCHITTEST: {
LRESULT lr = SendMessage(GetParent(hwndDlg), WM_NCHITTEST, wParam, lParam);
if(lr == HTLEFT || lr == HTRIGHT || lr == HTBOTTOM || lr == HTTOP || lr == HTTOPLEFT || lr == HTTOPRIGHT
|| lr == HTBOTTOMLEFT || lr == HTBOTTOMRIGHT)
return HTTRANSPARENT;
break;
}
case WM_TIMER: {
if(wParam == BUTTON_POLLID) {
RECT rc;
POINT pt;
GetWindowRect(hwndDlg, &rc);
GetCursorPos(&pt);
if(!PtInRect(&rc, pt)) {
// mouse must be gone, trigger mouse leave
PostMessage(hwndDlg, WM_MOUSELEAVE, 0, 0L);
KillTimer(hwndDlg, BUTTON_POLLID);
}
}
break;
}
case WM_ERASEBKGND:
return 1;
}
return DefWindowProc(hwndDlg, msg, wParam, lParam);
}
HICON PhGetFileShellIcon(
_In_opt_ PWSTR FileName,
_In_opt_ PWSTR DefaultExtension,
_In_ BOOLEAN LargeIcon
)
{
SHFILEINFO fileInfo;
ULONG iconFlag;
HICON icon;
if (DefaultExtension && PhEqualStringZ(DefaultExtension, L".exe", TRUE))
{
// Special case for executable files (see above for reasoning).
icon = NULL;
if (FileName)
{
ExtractIconEx(
FileName,
0,
LargeIcon ? &icon : NULL,
!LargeIcon ? &icon : NULL,
1
);
}
if (!icon)
{
PhGetStockApplicationIcon(
!LargeIcon ? &icon : NULL,
LargeIcon ? &icon : NULL
);
if (icon)
icon = DuplicateIcon(NULL, icon);
}
return icon;
}
iconFlag = LargeIcon ? SHGFI_LARGEICON : SHGFI_SMALLICON;
icon = NULL;
if (FileName && SHGetFileInfo(
FileName,
0,
&fileInfo,
sizeof(SHFILEINFO),
SHGFI_ICON | iconFlag
))
{
icon = fileInfo.hIcon;
}
if (!icon && DefaultExtension)
{
if (SHGetFileInfo(
DefaultExtension,
FILE_ATTRIBUTE_NORMAL,
&fileInfo,
sizeof(SHFILEINFO),
SHGFI_ICON | iconFlag | SHGFI_USEFILEATTRIBUTES
))
icon = fileInfo.hIcon;
}
return icon;
}
static PPH_PROCESS_ITEM PhpCreateProcessItemForHiddenProcess(
_In_ PPH_HIDDEN_PROCESS_ENTRY Entry
)
{
NTSTATUS status;
PPH_PROCESS_ITEM processItem;
PPH_PROCESS_ITEM idleProcessItem;
HANDLE processHandle;
PROCESS_BASIC_INFORMATION basicInfo;
KERNEL_USER_TIMES times;
PROCESS_PRIORITY_CLASS priorityClass;
ULONG handleCount;
HANDLE processHandle2;
if (Entry->Type == NormalProcess)
{
processItem = PhReferenceProcessItem(Entry->ProcessId);
if (processItem)
return processItem;
}
processItem = PhCreateProcessItem(Entry->ProcessId);
// Mark the process as terminated if necessary.
if (Entry->Type == TerminatedProcess)
processItem->State |= PH_PROCESS_ITEM_REMOVED;
// We need a process record. Just use the record of System Idle Process.
if (idleProcessItem = PhReferenceProcessItem(SYSTEM_IDLE_PROCESS_ID))
{
processItem->Record = idleProcessItem->Record;
PhReferenceProcessRecord(processItem->Record);
}
else
{
PhDereferenceObject(processItem);
return NULL;
}
// Set up the file name and process name.
PhSwapReference(&processItem->FileName, Entry->FileName);
if (processItem->FileName)
{
processItem->ProcessName = PhGetBaseName(processItem->FileName);
}
else
{
processItem->ProcessName = PhCreateString(L"Unknown");
}
if (ProcessesMethod == BruteForceScanMethod)
{
status = PhOpenProcess(
&processHandle,
ProcessQueryAccess,
Entry->ProcessId
);
}
else
{
status = PhOpenProcessByCsrHandles(
&processHandle,
ProcessQueryAccess,
Entry->ProcessId
);
}
if (NT_SUCCESS(status))
{
// Basic information and not-so-dynamic information
processItem->QueryHandle = processHandle;
if (NT_SUCCESS(PhGetProcessBasicInformation(processHandle, &basicInfo)))
{
processItem->ParentProcessId = basicInfo.InheritedFromUniqueProcessId;
processItem->BasePriority = basicInfo.BasePriority;
}
PhGetProcessSessionId(processHandle, &processItem->SessionId);
PhPrintUInt32(processItem->ParentProcessIdString, HandleToUlong(processItem->ParentProcessId));
PhPrintUInt32(processItem->SessionIdString, processItem->SessionId);
if (NT_SUCCESS(PhGetProcessTimes(processHandle, ×)))
{
processItem->CreateTime = times.CreateTime;
processItem->KernelTime = times.KernelTime;
processItem->UserTime = times.UserTime;
}
// TODO: Token information?
if (NT_SUCCESS(NtQueryInformationProcess(
processHandle,
ProcessPriorityClass,
&priorityClass,
sizeof(PROCESS_PRIORITY_CLASS),
NULL
)))
{
processItem->PriorityClass = priorityClass.PriorityClass;
}
if (NT_SUCCESS(NtQueryInformationProcess(
processHandle,
ProcessHandleCount,
&handleCount,
sizeof(ULONG),
NULL
)))
{
processItem->NumberOfHandles = handleCount;
}
}
// Stage 1
// Some copy and paste magic here...
if (processItem->FileName)
{
// Small icon, large icon.
ExtractIconEx(
processItem->FileName->Buffer,
0,
&processItem->LargeIcon,
&processItem->SmallIcon,
1
);
// Version info.
PhInitializeImageVersionInfo(&processItem->VersionInfo, processItem->FileName->Buffer);
}
// Use the default EXE icon if we didn't get the file's icon.
{
if (!processItem->SmallIcon || !processItem->LargeIcon)
{
if (processItem->SmallIcon)
{
DestroyIcon(processItem->SmallIcon);
processItem->SmallIcon = NULL;
}
else if (processItem->LargeIcon)
{
DestroyIcon(processItem->LargeIcon);
processItem->LargeIcon = NULL;
}
PhGetStockApplicationIcon(&processItem->SmallIcon, &processItem->LargeIcon);
processItem->SmallIcon = DuplicateIcon(NULL, processItem->SmallIcon);
processItem->LargeIcon = DuplicateIcon(NULL, processItem->LargeIcon);
}
}
// POSIX, command line
status = PhOpenProcess(
&processHandle2,
ProcessQueryAccess | PROCESS_VM_READ,
Entry->ProcessId
);
if (NT_SUCCESS(status))
{
BOOLEAN isPosix = FALSE;
PPH_STRING commandLine;
ULONG i;
status = PhGetProcessIsPosix(processHandle2, &isPosix);
processItem->IsPosix = isPosix;
if (!NT_SUCCESS(status) || !isPosix)
{
status = PhGetProcessCommandLine(processHandle2, &commandLine);
if (NT_SUCCESS(status))
{
// Some command lines (e.g. from taskeng.exe) have nulls in them.
// Since Windows can't display them, we'll replace them with
// spaces.
for (i = 0; i < (ULONG)commandLine->Length / 2; i++)
{
if (commandLine->Buffer[i] == 0)
commandLine->Buffer[i] = ' ';
}
}
}
else
{
// Get the POSIX command line.
status = PhGetProcessPosixCommandLine(processHandle2, &commandLine);
}
if (NT_SUCCESS(status))
{
processItem->CommandLine = commandLine;
}
NtClose(processHandle2);
}
// TODO: Other stage 1 tasks.
PhSetEvent(&processItem->Stage1Event);
return processItem;
}
