Sunday, January 8, 2012

Create system tray icon using window api

Continue previous entry about winapi. I will create system tray icon
First. include shell.api
Then:
Register varible:
NOTIFYICONDATA niData;
Create niData struct:
ZeroMemory(&niData,sizeof(NOTIFYICONDATA));
niData.cbSize = sizeof(NOTIFYICONDATA);
niData.hWnd = hWnd;
niData.uID = ID_TRAY;
niData.uFlags = NIF_ICON+NIF_MESSAGE+NIF_TIP;
niData.uCallbackMessage = WM_SHELLNOTIFY;
niData.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_HELLOWORLD));
lstrcpyn(niData.szTip, _T("ProjectZero tips!"),sizeof(niData.szTip)/sizeof(TCHAR));

To add icon to taskbar:
Shell_NotifyIcon(NIM_ADD, &niData);
Remove it:
Shell_NotifyIcon(NIM_DELETE, &niData);
Now you can see one icon at your taskbar.
Next, we will create popup menu. When you click ( one- not double) icon will show menu:
Create Contextmenu:
void ShowContextMenu(HWND hWnd)
{
POINT curPoint;
GetCursorPos(&curPoint);
HMENU pMenu = CreatePopupMenu();
if(pMenu)
{
AppendMenu(pMenu, MF_STRING, ID_TRAY_SHOW, L"Show");
AppendMenu(pMenu, MF_STRING, ID_TRAY_HIDE, L"Hide");
AppendMenu(pMenu, MF_STRING, ID_TRAY_EXIT, L"Exit");
SetForegroundWindow(hWnd);
TrackPopupMenu(pMenu, TPM_BOTTOMALIGN,curPoint.x, curPoint.y, 0, hWnd, NULL );
DestroyMenu(pMenu);
}
}
Add event:
switch (message)
{
case WM_COMMAND:
wmId = LOWORD(wParam);
wmEvent = HIWORD(wParam);
// Parse the menu selections:
switch (wmId)
{
case IDM_ABOUT:
DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
break;
case IDM_EXIT:
DestroyWindow(hWnd);
break;
case ID_TRAY_ABOUT:
DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
break;
case ID_TRAY_EXIT:
DestroyWindow(hWnd);
break;
case ID_TRAY_SHOW:
ShowWindow(hWnd, SW_RESTORE);
break;
case ID_TRAY_HIDE:
ShowWindow(hWnd, SW_HIDE);
break;
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
Finish. Full sourcode here:
// Helloworld.cpp : Defines the entry point for the application.
//
#include "stdafx.h"
#include "Helloworld.h"

#define MAX_LOADSTRING 100

// Global Variables:
HINSTANCE hInst; // current instance
TCHAR szTitle[MAX_LOADSTRING]; // The title bar text
TCHAR szWindowClass[MAX_LOADSTRING]; // the main window class name
NOTIFYICONDATA niData;

// Forward declarations of functions included in this code module:
ATOM MyRegisterClass(HINSTANCE hInstance);
BOOL InitInstance(HINSTANCE, int);
LRESULT CALLBACK WndProc(HWND, UINT, WPARAM, LPARAM);
INT_PTR CALLBACK About(HWND, UINT, WPARAM, LPARAM);
void ShowContextMenu(HWND hWnd);
void ShowAboutMenu(HWND hWnd);
int APIENTRY _tWinMain(HINSTANCE hInstance,
HINSTANCE hPrevInstance,
LPTSTR lpCmdLine,
int nCmdShow)
{
UNREFERENCED_PARAMETER(hPrevInstance);
UNREFERENCED_PARAMETER(lpCmdLine);

// TODO: Place code here.
MSG msg;
HACCEL hAccelTable;

// Initialize global strings
LoadString(hInstance, IDS_APP_TITLE, szTitle, MAX_LOADSTRING);
LoadString(hInstance, IDC_HELLOWORLD, szWindowClass, MAX_LOADSTRING);
MyRegisterClass(hInstance);

// Perform application initialization:
if (!InitInstance (hInstance, nCmdShow))
{
return FALSE;
}

hAccelTable = LoadAccelerators(hInstance, MAKEINTRESOURCE(IDC_HELLOWORLD));

// Main message loop:
while (GetMessage(&msg, NULL, 0, 0))
{
if (!TranslateAccelerator(msg.hwnd, hAccelTable, &msg))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
}

return (int) msg.wParam;
}



//
// FUNCTION: MyRegisterClass()
//
// PURPOSE: Registers the window class.
//
// COMMENTS:
//
// This function and its usage are only necessary if you want this code
// to be compatible with Win32 systems prior to the 'RegisterClassEx'
// function that was added to Windows 95. It is important to call this function
// so that the application will get 'well formed' small icons associated
// with it.
//
ATOM MyRegisterClass(HINSTANCE hInstance)
{
WNDCLASSEX wcex;

wcex.cbSize = sizeof(WNDCLASSEX);

wcex.style = CS_HREDRAW | CS_VREDRAW;
wcex.lpfnWndProc = WndProc;
wcex.cbClsExtra = 0;
wcex.cbWndExtra = 0;
wcex.hInstance = hInstance;
wcex.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_HELLOWORLD));
wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
wcex.hbrBackground = (HBRUSH)(COLOR_BTNFACE+1);
wcex.lpszMenuName = MAKEINTRESOURCE(IDC_HELLOWORLD);
wcex.lpszClassName = szWindowClass;
wcex.hIconSm = LoadIcon(wcex.hInstance, MAKEINTRESOURCE(IDI_HELLOWORLD));

return RegisterClassEx(&wcex);
}

//
// FUNCTION: InitInstance(HINSTANCE, int)
//
// PURPOSE: Saves instance handle and creates main window
//
// COMMENTS:
//
// In this function, we save the instance handle in a global variable and
// create and display the main program window.
//
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
HWND hWnd;

hInst = hInstance; // Store instance handle in our global variable

hWnd = CreateWindow(szWindowClass, szTitle, WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT, 0, 400, 300, NULL, NULL, hInstance, NULL);

if (!hWnd)
{
return FALSE;
}
ZeroMemory(&niData,sizeof(NOTIFYICONDATA));
niData.cbSize = sizeof(NOTIFYICONDATA);
niData.hWnd = hWnd;
niData.uID = ID_TRAY;
niData.uFlags = NIF_ICON+NIF_MESSAGE+NIF_TIP;
niData.uCallbackMessage = WM_SHELLNOTIFY;
niData.hIcon = LoadIcon(hInstance, MAKEINTRESOURCE(IDI_HELLOWORLD));
lstrcpyn(niData.szTip, _T("ProjectZero tips!"), sizeof(niData.szTip)/sizeof(TCHAR));
Shell_NotifyIcon(NIM_ADD, &niData);

ShowWindow(hWnd, nCmdShow);
UpdateWindow(hWnd);

return TRUE;
}

void ShowContextMenu(HWND hWnd)
{
POINT curPoint;
GetCursorPos(&curPoint);
HMENU pMenu = CreatePopupMenu();
if(pMenu)
{
AppendMenu(pMenu, MF_STRING, ID_TRAY_SHOW, L"Show");
AppendMenu(pMenu, MF_STRING, ID_TRAY_HIDE, L"Hide");
AppendMenu(pMenu, MF_STRING, ID_TRAY_EXIT, L"Exit");
SetForegroundWindow(hWnd);
TrackPopupMenu(pMenu, TPM_BOTTOMALIGN,curPoint.x, curPoint.y, 0, hWnd, NULL );
DestroyMenu(pMenu);
}
}
void ShowAboutMenu(HWND hWnd)
{
POINT curPoint;
GetCursorPos(&curPoint);
HMENU pMenu = CreatePopupMenu();
if(pMenu)
{
AppendMenu(pMenu, MF_STRING, ID_TRAY_ABOUT, L"About");
SetForegroundWindow(hWnd);
TrackPopupMenu(pMenu, TPM_BOTTOMALIGN,curPoint.x, curPoint.y, 0, hWnd, NULL );
DestroyMenu(pMenu);
}
}
//
// FUNCTION: WndProc(HWND, UINT, WPARAM, LPARAM)
//
// PURPOSE: Processes messages for the main window.
//
// WM_COMMAND - process the application menu
// WM_PAINT - Paint the main window
// WM_DESTROY - post a quit message and return
//
//
LRESULT CALLBACK WndProc(HWND hWnd, UINT message, WPARAM wParam, LPARAM lParam)
{
int wmId, wmEvent;
PAINTSTRUCT ps;
HDC hdc;

switch (message)
{
case WM_COMMAND:
wmId = LOWORD(wParam);
wmEvent = HIWORD(wParam);
// Parse the menu selections:
switch (wmId)
{
case IDM_ABOUT:
DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
break;
case IDM_EXIT:
DestroyWindow(hWnd);
break;
case ID_TRAY_ABOUT:
DialogBox(hInst, MAKEINTRESOURCE(IDD_ABOUTBOX), hWnd, About);
break;
case ID_TRAY_EXIT:
DestroyWindow(hWnd);
break;
case ID_TRAY_SHOW:
ShowWindow(hWnd, SW_RESTORE);
break;
case ID_TRAY_HIDE:
ShowWindow(hWnd, SW_HIDE);
break;
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
break;
case WM_PAINT:
hdc = BeginPaint(hWnd, &ps);
// TODO: Add any drawing code here...
EndPaint(hWnd, &ps);
break;
case WM_DESTROY:
Shell_NotifyIcon(NIM_DELETE, &niData);
PostQuitMessage(0);
break;
case WM_SHELLNOTIFY:
switch(lParam)
{
case WM_LBUTTONDBLCLK:
MessageBox(NULL, L"Tray icon double clicked!", L"clicked", MB_OK);
break;
case WM_RBUTTONDOWN:
ShowAboutMenu(hWnd);
break;
case WM_LBUTTONUP:
ShowContextMenu(hWnd);
break;
}
break;
default:
return DefWindowProc(hWnd, message, wParam, lParam);
}
return 0;
}

// Message handler for about box.
INT_PTR CALLBACK About(HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
UNREFERENCED_PARAMETER(lParam);
switch (message)
{
case WM_INITDIALOG:
return (INT_PTR)TRUE;

case WM_COMMAND:
if (LOWORD(wParam) == IDOK || LOWORD(wParam) == IDCANCEL)
{
EndDialog(hDlg, LOWORD(wParam));
return (INT_PTR)TRUE;
}
break;
}
return (INT_PTR)FALSE;
}

Soure ( using visuall c++ 2008 express): http://www.mediafire.com/?hqnrya9hcl7mb7x
------------------------------------------------------------
Thanks for reading
--------------------------------------------------------------------------
All my Lab:
Linux Lab -- window and Cisco Lab
to be continued - I will update more.

No comments:

Install Xposed Inspector and Frida on Genymotion

Today i had some work with android. So i need trace application. I found 2 nice tool can help me: Xposed Inspector and Frida. To setup ther...