usbtransdlg.cpp

wince5.0下动态切换Antisync和mass storage的源代码

// USBTransDlg.cpp : implementation file
//

#include "stdafx.h"
#include "USBTrans.h"
#include "USBTransDlg.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CUSBTransDlg dialog

CUSBTransDlg::CUSBTransDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CUSBTransDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CUSBTransDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
	// Note that LoadIcon does not require a subsequent DestroyIcon in Win32
	m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CUSBTransDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CUSBTransDlg)
		// NOTE: the ClassWizard will add DDX and DDV calls here
	//}}AFX_DATA_MAP
}

BEGIN_MESSAGE_MAP(CUSBTransDlg, CDialog)
	//{{AFX_MSG_MAP(CUSBTransDlg)
	ON_BN_CLICKED(IDC_BTN_ASYNC_MASS, OnBtnAsyncMass)
	ON_BN_CLICKED(IDC_BTN_DEFAULT_ASYNC_MASS, OnBtnDefaultAsyncMass)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CUSBTransDlg message handlers
#include <devload.h>
#include <usbfnioctl.h>

BOOL CUSBTransDlg::OnInitDialog()
{
	CDialog::OnInitDialog();

	// Set the icon for this dialog.  The framework does this automatically
	//  when the application's main window is not a dialog
	SetIcon(m_hIcon, TRUE);			// Set big icon
	SetIcon(m_hIcon, FALSE);		// Set small icon
	
	CenterWindow(GetDesktopWindow());	// center to the hpc screen

	// TODO: Add extra initialization here
	TCHAR szName[UFN_CLIENT_NAME_MAX_CHARS];
	if (Client_IOControl(IOCTL_UFN_GET_CURRENT_CLIENT, (DWORD)szName))
	{
		if (!_tcscmp(szName, _T("Serial_Class")))
			SetDlgItemText(IDC_BTN_ASYNC_MASS, _T("Change Current Client to Mass Storage"));
		else
			SetDlgItemText(IDC_BTN_ASYNC_MASS, _T("Change Current Client to ActiveSync"));
	}
	if (Client_IOControl(IOCTL_UFN_GET_DEFAULT_CLIENT, (DWORD)szName))
	{
		if (!_tcscmp(szName, _T("Serial_Class")))
			SetDlgItemText(IDC_BTN_DEFAULT_ASYNC_MASS, _T("Change Default Client to Mass Storage"));
		else
			SetDlgItemText(IDC_BTN_DEFAULT_ASYNC_MASS, _T("Change Default Client to ActiveSync"));
	}

	return TRUE;  // return TRUE  unless you set the focus to a control
}

void CUSBTransDlg::OnBtnAsyncMass() 
{
	TCHAR szName[UFN_CLIENT_NAME_MAX_CHARS];

	if (Client_IOControl(IOCTL_UFN_GET_CURRENT_CLIENT, (DWORD)szName))
	{
		if (!_tcscmp(szName, _T("Serial_Class")))
		{
			if (Client_IOControl(IOCTL_UFN_CHANGE_CURRENT_CLIENT, (DWORD)_T("Mass_Storage_Class")))
				SetDlgItemText(IDC_BTN_ASYNC_MASS, _T("Change Current Client to ActiveSync"));
		}
		else
		{
			if (Client_IOControl(IOCTL_UFN_CHANGE_CURRENT_CLIENT, (DWORD)_T("Serial_Class")))
				SetDlgItemText(IDC_BTN_ASYNC_MASS, _T("Change Current Client to Mass Storage"));
		}
	}
}

void CUSBTransDlg::OnBtnDefaultAsyncMass() 
{
	TCHAR szName[UFN_CLIENT_NAME_MAX_CHARS];

	if (Client_IOControl(IOCTL_UFN_GET_DEFAULT_CLIENT, (DWORD)szName))
	{
		if (!_tcscmp(szName, _T("Serial_Class")))
		{
			if (Client_IOControl(IOCTL_UFN_CHANGE_DEFAULT_CLIENT, (DWORD)_T("Mass_Storage_Class")))
				SetDlgItemText(IDC_BTN_ASYNC_MASS, _T("Change Default Client to ActiveSync"));
		}
		else
		{
			if (Client_IOControl(IOCTL_UFN_CHANGE_DEFAULT_CLIENT, (DWORD)_T("Serial_Class")))
				SetDlgItemText(IDC_BTN_ASYNC_MASS, _T("Change Default Client to Mass Storage"));
		}

		AfxMessageBox(_T("reboot can become effective !"));
	}	
}

BOOL CUSBTransDlg::Client_IOControl(DWORD dwIoControlCode, DWORD lParams)
{
	HANDLE hUfn = CreateFile(_T("UFN1:"), GENERIC_READ|GENERIC_WRITE|DEVACCESS_BUSNAMESPACE, 0, NULL, OPEN_EXISTING, 0, 0);
	if (hUfn == INVALID_HANDLE_VALUE)
	{
		AfxMessageBox(_T("Invalid UFN handle"));
		return FALSE;
	}
	else 
	{
		UFN_CLIENT_INFO ufnClientInfo;
		BOOL bRet=FALSE;
		TCHAR *pszName = (TCHAR *)lParams;
		UFN_CLIENT_NAME ufnClientName;

		switch (dwIoControlCode)
		{
		case IOCTL_UFN_GET_CURRENT_CLIENT :
			bRet = ::DeviceIoControl(hUfn, dwIoControlCode, NULL, NULL, &ufnClientInfo, sizeof(ufnClientInfo), NULL, NULL);
			if (bRet)
			{
				TRACE(_T("Name = %s\nDesc = %s\n"), ufnClientInfo.szName, ufnClientInfo.szDescription);
				_tcscpy(pszName, ufnClientInfo.szName);
			}
			break;

		case IOCTL_UFN_GET_DEFAULT_CLIENT :
			bRet = ::DeviceIoControl(hUfn, dwIoControlCode, NULL, NULL, &ufnClientInfo, sizeof(ufnClientInfo), NULL, NULL);
			if (bRet)
			{
				TRACE(_T("Name = %s\nDesc = %s\n"), ufnClientInfo.szName, ufnClientInfo.szDescription);
				_tcscpy(pszName, ufnClientInfo.szName);
			}
			break;

		case IOCTL_UFN_CHANGE_CURRENT_CLIENT :
			_tcscpy(ufnClientName.szName, pszName);
			bRet = ::DeviceIoControl(hUfn, dwIoControlCode, &ufnClientName, sizeof(ufnClientName), NULL, NULL, NULL, NULL);
			break;

		case IOCTL_UFN_CHANGE_DEFAULT_CLIENT :
			_tcscpy(ufnClientName.szName, pszName);
			bRet = ::DeviceIoControl(hUfn, dwIoControlCode, &ufnClientName, sizeof(ufnClientName), NULL, NULL, NULL, NULL);
			break;
		}

		CloseHandle(hUfn);
		return bRet;
	}
}