虚拟桌面：一个简单的桌面管理工具

 

转载请标明是引用于 http://blog.csdn.net/chenyujing1234 

欢迎大家拍砖!

参考英文文章： <<Virtual Desktop: A Simple Desktop Management Tool>>

源代码请到原文链接下载。

一、引言

几个月前，我的一个大学同学正在用一个工具来管理混合桌面。我很好奇有这种的工具。我想它应该仅用来隐藏显示不同桌面的应用程序。作为好奇，我开始研究混合桌面，并得到了这篇文章。我发现之前认为这个工具仅是显示隐藏用的想法是错的。让我们看它如何工作的。

二、Window Station and Window Desktop

其实Windows用Window Station and Desktop  结构来提供可增加的安全。

暂时，我们提到的“Desktop“是指逻辑显示表面，它包含用户接口对象（GDI对象）和用户对象。

一个Window Station 是一个安全的内核对象，它包含剪切板，一个原子表，一个或更多的桌面对象。这些存在默认的窗口站“WinSta0“。

它是一个交互的窗口站，“WinSta0“是仅有的交互窗口站，它能显示用户接口或接收用户输入，而其它的窗口站是非交互的。

默认在交互的容器站中有三个桌面：默认的、WinLogon和ScreenSaver。

（）1）通过按ATL+CTL+DEL弹出的窗口是WinLogon桌面。

（2）无论用户在什么时候登陆，默认窗口被创建并可见。这里的“默认”是指对所有登陆用户的默认的活动桌面。当你按了ATL+CTL+DEL，你将会切换回到WinLogon桌面。

（3当屏保被激活时，ScreenSave桌面会被可见并获得激活。

因为这些桌面名字是case-insensitive，这些仅能是一个在容器站指定名字的桌面，但是一个桌面名字能在不同的窗口站中反复出现。

一个桌面也是一个安全的内核对象，当一个新的桌面被创建，它和调用它的进程的当前的窗口站绑定在一起。那里将一直仅有一个桌面是可见并准备接收用户输入，

被提供的桌面和交互窗口站绑定在一起。激活的窗口被调用。

以下的桌面和窗口站的工程是为了提供在不同进程中的顶层解决。

三、关于虚拟桌面（一个简单的应用程序）

虚拟桌面帮助我们创建和在不同的桌面中转换。它甚至给你一个从一个桌面转到另一个桌面的感觉，但它不是真的，因为任何运行的应用程序不能被从一个

桌面移到另一个桌面。如我所说，它只是一个假象，这个应用程序在当前的桌面会被关闭，且在特殊的桌面会被打开。这个可能导致应用程序未保存的数据的丢失，

且可能不能保存状态，因为它可能重启。

这个应用程序用hooks来显示应用程序控制面板中的“Move To"菜单选项。对于hooks更多细节，在文章中我会提到，我将让你大约了解一下工程中用到的APIs和类。

Different APIs Used and Classes Built

1、初始化工作

主要是获得当前桌面的名字、根据名字设置应用程序标题并生成右下角图标、遍历系统中的桌面数量和名字并写注册表、

载入Evnent Hooker.dll，调用接口安装窗口HOOK和消息HOOK。

//Checks whether the specified desktop is current active desktop(deskop of the current thread).
bool CDesktopManager::GetCurrentDesktopName(TCHAR szDesktopName[ARRAY_SIZE])
{
	bool bReturn = false;

	try
	{
		DWORD iOutCount = 0;
		HDESK hCurrentDesktop = GetThreadDesktop(GetCurrentThreadId());
		if(!GetUserObjectInformation(hCurrentDesktop, UOI_NAME, szDesktopName, ARRAY_SIZE - 1, &iOutCount))
			throw _T("\nCDesktopManager::GetCurrentDesktopName:\t GetUserObjectInformation() failed.");

		bReturn = true;
	}
	catch(TCHAR *pszErrorString)
	{
		DebugPrintErrorMessage(pszErrorString);
	}
	catch(...)
	{
		DebugPrintErrorMessage(_T("\nCDesktopManager::IsCurrentDesktop\tException caught in CDesktopManager::IsCurrentDesktop."));
		bReturn = false;
	}

	return bReturn;
}

//Windows procedure hook (filter) function.
LRESULT CALLBACK WinProcHookProcedure(int nCode, WPARAM wParam, LPARAM lParam)
{
	if(0 > nCode)
		return CallNextHookEx(g_hPreviousWinProcHook ,nCode,wParam,lParam);

	CWPSTRUCT *pwsStruct = (CWPSTRUCT *) lParam;
	
	switch(pwsStruct->message)
	{
		//Handle the init menu message.
	case WM_INITMENU:
		{
			HMENU hMenu = (HMENU) pwsStruct->wParam;
			OutputDebugString(_T("\nWinProcHookProcedure:\tMenu WM_INITMENUPOPUP Event Fired."));

			TCHAR szWindowText[ARRAY_SIZE] = {0};
			GetWindowText(pwsStruct->hwnd, szWindowText, ARRAY_SIZE);

			//Hook "Move To" Menu Into System Menu.
			if(0 != _tcscmp(szWindowText, _T("Virtual Desktop !")))
				InsertMenu(pwsStruct->hwnd);
		}
	}

	return CallNextHookEx(g_hPreviousWinProcHook ,nCode,wParam,lParam);
}

//Set the Windows procedure filter.
bool InstallWinProcHook(void)
{
	bool bReturn = false;
	try
	{
		//Set the Windows procedure filter.
		OutputDebugString(_T("\nInstallWinProcHook:\tWinProc Event Hooked."));
		g_hPreviousWinProcHook = SetWindowsHookEx(WH_CALLWNDPROC,&WinProcHookProcedure,g_hInstance,0);

		if(NULL == g_hPreviousWinProcHook)
		{
			TCHAR szError[ARRAY_SIZE] = {0};
			wsprintf(szError,_T("Last Error : %d"),GetLastError());
			OutputDebugString(_T("\nInstallWinProcHook:\tFailed to Hook WinProc Event.\n"));
			OutputDebugString(szError);

			bReturn = false;
		}
		else
		{
			OutputDebugString(_T("\nInstallWinProcHook:\tWinProc Event Hooked.\n"));
			bReturn = true;
		}
	}
	catch(...)
	{
		OutputDebugString(_T("\nInstallWinProcHook:\tException caught in InstallWinProcHook."));
		bReturn = false;
	}

	return bReturn;
}

在安装窗口HOOK的API函数 SetWindowsHookEx 时360安全卫士报告有异常！（这种情况怎么避免呢？知道的读者能告知我解决方法吗？）

//Set the message hook (Filter).
bool InstallMessageHook(void)
{
	bool bReturn = false;
	try
	{
		//Set the message hook (Filter).
		g_hPreviousMsgHook = SetWindowsHookEx(WH_GETMESSAGE,&MessageHookProcedure,g_hInstance,0);

		if(NULL == g_hPreviousMsgHook)
		{
			OutputDebugString(_T("\nInstallMessageHook:\tFailed to Hook Messages.\n"));
			bReturn = false;
		}
		else
		{
			OutputDebugString(_T("\nInstallMessageHook:\tMessages Hooked.\n"));
			bReturn = true;
		}
	}
	catch(...)
	{
		OutputDebugString(_T("\nInstallMessageHook:\tException caught in InstallMessageHook."));
		bReturn = false;
	}

	return bReturn;
}

2、响应窗口切换、启动应用程序、新建窗口动作

 

//Switch between desktops.
bool CDesktopManager::SwitchDesktop(TCHAR *pszDesktopName)
{
	bool bReturn = false;
	try
	{
		if (NULL == pszDesktopName)
		{
			OutputDebugString(_T("\nCDesktopManager::SwitchDesktop:\tNULL DesktopName in CDesktopManager::SwitchToDesktop"));
			throw false;
		}

		/*if( !_tcsicmp(_T("Winlogon"), szDesktopName) || !_tcsicmp(_T("Disconnect"), szDesktopName))
		{
			TCHAR szErrorMsg[ARRAY_SIZE] = {_T("You can not switch to ")};
			_tcscat_s(szErrorMsg, ARRAY_SIZE -1, szDesktopName);
			_tcscat_s(szErrorMsg, ARRAY_SIZE -1, _T(" Desktop."));
			MessageBox(NULL, szErrorMsg, TXT_MESSAGEBOX_TITLE, MB_ICONINFORMATION);
			throw false;
		}*/

		//Open desktop handle to switch to.
		HDESK hDesktopToSwitch = OpenDesktop(pszDesktopName, DF_ALLOWOTHERACCOUNTHOOK, TRUE, GENERIC_ALL);
		if(NULL == hDesktopToSwitch)
		{
			TCHAR *pszError = NULL;
			TCHAR szErrorMsg[ARRAY_SIZE] = {0};
			int iErrorCode = GetLastError();
			if(5 == iErrorCode)
			{
				
				FormatMessage(FORMAT_MESSAGE_ALLOCATE_BUFFER |FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS, 
								NULL, iErrorCode, 0, (LPWSTR) &pszError, 0, NULL);

				wsprintf(szErrorMsg, _T("Failed to switch to %s desktop.\n\t %s"), pszDesktopName, pszError);
				MessageBox(NULL, szErrorMsg, TXT_MESSAGEBOX_TITLE, MB_ICONINFORMATION | MB_TOPMOST | MB_TASKMODAL);
				OutputDebugString(pszError);
			}

			wsprintf(szErrorMsg, _T("\nCDesktopManager::SwitchDesktop:\tOpenDesktop Failed in CDesktopManager::SwitchToDesktop. Last Error : %s"), pszError);
			OutputDebugString(szErrorMsg);

			throw false;
		}

		//Switch the desktop.
		if(FALSE == ::SwitchDesktop(hDesktopToSwitch))
		{
			OutputDebugString(_T("\nCDesktopManager::SwitchDesktop:\tSwitchDesktop Failed in CDesktopManager::SwitchToDesktop"));
			throw false;
		}
 
		//Close the desktop handle.
		CloseDesktop(hDesktopToSwitch);
		bReturn = true;
	}
	catch(bool bThrownVal)
	{
		bReturn = bThrownVal;
	}
	catch(...)
	{
		bReturn = false;
		DebugPrintErrorMessage(_T("\nCDesktopManager::SwitchDesktop:\tException caught in CDesktopManager::SwitchToDesktop"));
	}
	return bReturn;
}

 

//Creates the new desktop (Adds new desktop)
void CVirtualDesktopDlg::OnBnClickedAddNewDesktop()
{
	// TODO: Add your control notification handler code here
	TCHAR szCaption[ARRAY_SIZE] = {0};
	m_AddNewDesktop.GetWindowText(szCaption, ARRAY_SIZE - 1);
	if(_tcsicmp(szCaption, _T("&New")))
	{
		m_DesktopNameControl.GetWindowText(szCaption, ARRAY_SIZE -1);

		//Left trimming the string.
		while(' ' == szCaption[0])
			_tcscpy_s(szCaption, ARRAY_SIZE - 1, szCaption + 1);
		
		int iLen = (int) _tcslen(szCaption);
		//Right trimming the string.
		while(' ' == szCaption[--iLen])
			szCaption[iLen] = '\0';

		if(_tcslen(szCaption))
		{
			//Check the desktop name in the list.
			if(-1 != m_DesktopListControl.SelectString(0, szCaption))
			{
				MessageBox(_T("Desktop already created !"), TXT_MESSAGEBOX_TITLE, MB_ICONEXCLAMATION | MB_TOPMOST | MB_TASKMODAL);
			}

			//Create the desktop
			if(CDesktopManager::CreateDesktop(szCaption))
			{
				if(IDYES == MessageBox(_T("New Desktop is been created.\nWould you like to switch to new desktop ?"), TXT_MESSAGEBOX_TITLE, MB_YESNO | MB_ICONINFORMATION | MB_TOPMOST | MB_TASKMODAL))
					SwitchDesktopTo(szCaption);

				m_DesktopListControl.AddString(szCaption);
				m_DesktopListControl.SelectString(0, szCaption);

				OnLbnSelchangeDesktopList();

				UpdateHotKeys();
			}
		}
		else
		{
			MessageBox(_T("Please enter Desktop Name"), TXT_MESSAGEBOX_TITLE, MB_ICONEXCLAMATION | MB_TOPMOST | MB_TASKMODAL);
			m_DesktopNameControl.SetWindowText(_T("")); 
			m_DesktopNameControl.SetFocus();
		}
	}
	else
	{
		m_AddNewDesktop.SetWindowText(_T("&Add"));
		m_DesktopNameControl.SetWindowText(_T(""));
		m_DesktopNameControl.EnableWindow(TRUE);
		m_SwitchToDesktop.EnableWindow(FALSE);
		m_DesktopNameControl.SetFocus();
	}
}