创建虚拟桌面
#include<afxwin.h>
HINSTANCE hInst; //当前实例
TCHAR szAppName[] = TEXT ("VirtualDesk") ; //程序名称
HWND hwnd ; //主窗体句柄
HDESK hVirtualDesk; //虚拟桌面句柄
HDESK hCurrent; //当前桌面句柄
PROCESS_INFORMATION piExplor; //Explorer进程的PROCESS_INFORMATION结构
LRESULT CALLBACK WndProc (HWND, UINT, WPARAM, LPARAM);
//每次运行首先检测注册表相关项，如果未发现，则设置开机自动运行
void SetStartup(HINSTANCE hInst)
{
HKEY hKey;
DWORD DataType = REG_SZ;
PCSTR data_run = "Software\\Microsoft\\Windows\\CurrentVersion\\Run";
long ret = RegOpenKeyEx(HKEY_LOCAL_MACHINE,data_run,0,KEY_ALL_ACCESS,&hKey);
if (ret != ERROR_SUCCESS)
{
MessageBox(NULL,"无法打开注册表键","Error",0);
return;
}
CString ProcessName;
int len = GetModuleFileName(hInst,ProcessName.GetBuffer(256),256);
if (len == 0)
{
MessageBox(NULL,"无法获取进程的当前目录","Error",0);
return;
}
ProcessName.ReleaseBuffer(len);
DWORD direcLen = ProcessName.GetLength() + 1;
LPBYTE direc = new BYTE[direcLen];
ZeroMemory(direc,direcLen);
ret = RegQueryValueEx(hKey,"VirtualDesk",0,0,direc,&direcLen);
//如果键不存在或者键长度与目前的值长度不匹配则添加新键
if ((ret != ERROR_SUCCESS )|| (direcLen != ProcessName.GetLength()+1))
{
SetValue:
DWORD KeyLength = ProcessName.GetLength() + 1;
LPBYTE KeyValue = new BYTE[KeyLength];
ZeroMemory(KeyValue,KeyLength);
for (int i=0;i<ProcessName.GetLength();i++)
KeyValue[i] = ProcessName[i];
KeyValue[ProcessName.GetLength()] = 0;
ret = RegSetValueEx(hKey,"VirtualDesk",0,REG_SZ,KeyValue,KeyLength);
delete []KeyValue;
}
else
{
//如果键的内容与当前值不匹配同样进行更改
for (int i=0;i<ProcessName.GetLength();i++)
{
if (direc[i] != ProcessName[i])
goto SetValue;
}
}
delete []direc;
return;
}
//创建explorer进程的函数，如果创建了一个虚拟桌面，那么最好调用该函数
void StartExplorer()
{
STARTUPINFO si;
ZeroMemory( &si, sizeof(si) );
si.cb = sizeof(si);
si.lpDesktop = "Virtual";
ZeroMemory( &piExplor, sizeof(piExplor) );
if ( !CreateProcess( NULL, // No module name (use command line).
"explorer", // Command line.
NULL, // Process handle not inheritable.
NULL, // Thread handle not inheritable.
FALSE, // Set handle inheritance to FALSE.
0, // No creation flags.
NULL, // Use parent's environment block.
NULL, // Use parent's starting directory.
&si, // Pointer to STARTUPINFO structure.
&piExplor ) // Pointer to PROCESS_INFORMATION structure.
)
{
MessageBox(NULL,"无法初始化Explorer","Error",0);
ExitProcess(1);
}
}
ATOM MyRegisterClass(HINSTANCE hInstance)
{
WNDCLASSEX wcex;
wcex.cbSize = sizeof(WNDCLASSEX);
wcex.style = CS_HREDRAW | CS_VREDRAW;
wcex.lpfnWndProc = (WNDPROC)WndProc;
wcex.cbClsExtra = 0;
wcex.cbWndExtra = 0;
wcex.hInstance = hInstance;
wcex.hIcon = LoadIcon(hInstance, (LPCTSTR)IDI_APPLICATION);
wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
wcex.hbrBackground = (HBRUSH)(COLOR_WINDOW+1);
wcex.lpszMenuName = NULL;
wcex.lpszClassName = szAppName;
wcex.hIconSm = LoadIcon(wcex.hInstance, (LPCTSTR)IDI_APPLICATION);
return RegisterClassEx(&wcex);
}
BOOL InitInstance(HINSTANCE hInstance, int nCmdShow)
{
hInst = hInstance; // 将实例句柄存储在全局变量中
hwnd = CreateWindow(szAppName, TEXT ("SunBear"), WS_OVERLAPPEDWINDOW,
CW_USEDEFAULT, 0, CW_USEDEFAULT, 0, NULL, NULL, hInst, NULL);
if (!hwnd)
{
return FALSE;
}
return TRUE;
}
int WINAPI WinMain (HINSTANCE hInstance, HINSTANCE hPrevInstance,PSTR szCmdLine, int iCmdShow)
{
static TCHAR szAppName[] = TEXT ("SunBear") ;
MSG msg ;
SetStartup(hInstance);
hVirtualDesk = CreateDesktop(
"Virtual",
NULL,
NULL,
DF_ALLOWOTHERACCOUNTHOOK,
GENERIC_ALL,
NULL);
hCurrent = GetThreadDesktop(GetCurrentThreadId());
MyRegisterClass(hInstance);
if (!InitInstance (hInstance, iCmdShow))
{
return FALSE;
}
StartExplorer(); //启动Explorer进程
if (!RegisterHotKey(hwnd, 0x0001,MOD_ALT ,'Q'))
{
//处理切换虚拟桌面
return TRUE;
}
if (!RegisterHotKey(hwnd, 0x0002, 0,VK_F8))
{
//处理退出进程
return TRUE;
}
while (GetMessage (&msg, NULL, 0, 0))
{
TranslateMessage (&msg) ;
DispatchMessage (&msg) ;
}
return 0;
}
LRESULT CALLBACK WndProc (HWND hwnd, UINT message, WPARAM wParam, LPARAM lParam)
{
HDC hdc ;
PAINTSTRUCT ps ;
static HDESK hNow = hCurrent;
switch (message)
{
case WM_PAINT:
hdc = BeginPaint (hwnd, &ps) ;
EndPaint (hwnd, &ps) ;
return 0 ;
case WM_DESTROY:
//在关闭虚拟桌面前要切换回当前桌面
SwitchDesktop(hCurrent);
PostQuitMessage (0) ;
return 0 ;
case WM_HOTKEY:
if (0x0001 == wParam)
{
if (hNow == hCurrent)
{
SwitchDesktop(hVirtualDesk);
hNow = hVirtualDesk;
}
else
{
SwitchDesktop(hCurrent);
hNow = hCurrent;
}
}
if (0x0002 == wParam)
{
//用TerminateProcess终止explorer进程的时候，如果传递第二个参数为1
//那么操作系统不会在终止后自动唤醒explorer,如果为0,会重新启动
//explorer
TerminateProcess(piExplor.hProcess,1);
//关闭虚拟桌面
CloseDesktop(hVirtualDesk);
SendMessage(hwnd,WM_CLOSE,0,0);
}
return 0;
}
return DefWindowProc (hwnd, message, wParam, lParam) ;
}
