CFormatDriveDialog - A wrapper class for the undocumented SHFormatDrive API function

/*
	Author - Nishant S
	EMail - nishforever@vsnl.com
*/

#include "StdAfx.h"
#include "formatdrive.h"

HHOOK CFormatDriveDialog::m_hHook=NULL;
LPCTSTR CFormatDriveDialog::m_vol_label = NULL;

CFormatDriveDialog::CFormatDriveDialog(void)
{
}

CFormatDriveDialog::~CFormatDriveDialog(void)
{
}

int CFormatDriveDialog::DoModal(HWND hWnd, UINT Drive, bool bQuickFormat,
								LPCTSTR vol_label)
{
	ASSERT(hWnd);
	m_vol_label = vol_label;
	int rv = 0;
	UINT Options = bQuickFormat ? SHFMT_OPT_QUICK : 0;
	FMTDRIVEFUNC* pshfd;
	HMODULE  hMod = LoadLibrary("shell32.dll");
	if(hMod)
	{
		pshfd = reinterpret_cast<FMTDRIVEFUNC*>(GetProcAddress(hMod,"SHFormatDrive"));
		m_hHook = SetWindowsHookEx(WH_CBT,CBTProc,
			AfxGetApp()->m_hInstance,AfxGetApp()->m_nThreadID);
		rv = ((*pshfd)(hWnd,Drive,SHFMT_ID_DEFAULT,Options)==SHFMT_FMTSUCCESS) ? IDOK : IDCANCEL;
		FreeLibrary(hMod);
	}
	return rv;
}

LRESULT CALLBACK CFormatDriveDialog::CBTProc(int nCode,WPARAM wParam, LPARAM lParam)
{
	if (nCode == HCBT_ACTIVATE )
	{	
		HWND hWnd = reinterpret_cast<HWND>(wParam);
		
		EnumChildWindows(hWnd,EnumChildProc,NULL);	
		
		UnhookWindowsHookEx(m_hHook);
		CFormatDriveDialog::m_hHook = NULL;		
	}
	return FALSE;
}

BOOL CALLBACK CFormatDriveDialog::EnumChildProc( HWND hwnd, LPARAM lParam )
{	
	char buff[256];
	GetClassName(hwnd,buff,255);
	if(strcmp(buff,"Edit")==0)
	{
		SetWindowText(hwnd,m_vol_label);
		return FALSE;
	}
	return TRUE;
}