System Information Utility

// Disclaimer and Copyright Information
// HDiskInfoPage.cpp : Implementation file
//
// All rights reserved.
//
// Written by Naveen K Kohli (naveenkohli@netzero.net)
// Version 1.0
//
// Distribute freely, except: don't remove my name from the source or
// documentation (don't take credit for my work), mark your changes (don't
// get me blamed for your possible bugs), don't alter or remove this
// notice.
// No warrantee of any kind, express or implied, is included with this
// software; use at your own risk, responsibility for damages (if any) to
// anyone resulting from the use of this software rests entirely with the
// user.
//
// Send bug reports, bug fixes, enhancements, requests, flames, etc. to
// naveenkohli@netzero.net
///////////////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "SystemApplication.h"
#include "HDiskInfoPage.h"

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

/////////////////////////////////////////////////////////////////////////////
// HDiskInfoPage property page

IMPLEMENT_DYNCREATE(HDiskInfoPage, CPropertyPage)

HDiskInfoPage::HDiskInfoPage() : CPropertyPage(HDiskInfoPage::IDD)
{
	//{{AFX_DATA_INIT(HDiskInfoPage)
	m_pbstrDriveLetterArr = NULL;
	m_pbBootableArr = NULL;
	m_pbstrTypeArr = NULL;
	m_plPartitionNumberArr = NULL;
	m_plLengthArr = NULL;
	m_plHiddenSectorsArr = NULL;
	//}}AFX_DATA_INIT
}

HDiskInfoPage::~HDiskInfoPage()
{
	if (m_pbstrDriveLetterArr != NULL) {
		SafeArrayDestroy (m_pbstrDriveLetterArr);
		m_pbstrDriveLetterArr = NULL;
	}

	if (m_pbBootableArr != NULL) {
		SafeArrayDestroy (m_pbBootableArr);
		m_pbBootableArr = NULL;
	}

	if (m_pbstrTypeArr != NULL) {
		SafeArrayDestroy (m_pbstrTypeArr);
		m_pbstrTypeArr = NULL;
	}

	if (m_plPartitionNumberArr != NULL) {
		SafeArrayDestroy (m_plPartitionNumberArr);
		m_plPartitionNumberArr = NULL;
	}

	if (m_plLengthArr != NULL) {
		SafeArrayDestroy (m_plLengthArr);
		m_plLengthArr = NULL;
	}

	if (m_plHiddenSectorsArr != NULL) {
		SafeArrayDestroy (m_plHiddenSectorsArr);
		m_plHiddenSectorsArr = NULL;
	}
}

void HDiskInfoPage::DoDataExchange(CDataExchange* pDX)
{
	CPropertyPage::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(HDiskInfoPage)
	DDX_Control(pDX, IDC_HDISK_INFO_LIST, m_ListCtrl);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(HDiskInfoPage, CPropertyPage)
	//{{AFX_MSG_MAP(HDiskInfoPage)
	ON_WM_VSCROLL()
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// HDiskInfoPage message handlers

BOOL HDiskInfoPage::OnInitDialog() 
{
	CPropertyPage::OnInitDialog();
	
	if (FAILED (GetInformation ())) {
		AfxMessageBox (_T ("Application Failed To Get Hardisk Information From Interface Pointer"),
			MB_ICONSTOP);
		return TRUE;
	}

	SetListData();
	
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

HRESULT HDiskInfoPage::GetInformation ()
{
	VARIANT bstrDriveLetterArr;
	VARIANT bBootableArr;
	VARIANT bstrTypeArr;
	VARIANT lPartitionNumberArr;
	VARIANT lLengthArr;
	VARIANT lHiddenSectorsArr;

	if (FAILED (m_pSystemInfo->GetHDiskInformation (&m_lNumberOfPartitions, &bstrDriveLetterArr,
		&bBootableArr, &bstrTypeArr, &lPartitionNumberArr, &lLengthArr, &lHiddenSectorsArr))) {
		return E_FAIL;
	}

	if (!(bstrDriveLetterArr.vt & VT_BSTR)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (!(bstrDriveLetterArr.vt & VT_ARRAY)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (bstrDriveLetterArr.vt & VT_BYREF) {
		m_pbstrDriveLetterArr = * (bstrDriveLetterArr.pparray);
	}
	else {
		m_pbstrDriveLetterArr = bstrDriveLetterArr.parray;
	}

	if (!(bstrTypeArr.vt & VT_BSTR)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (!(bstrTypeArr.vt & VT_ARRAY)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (bstrDriveLetterArr.vt & VT_BYREF) {
		m_pbstrTypeArr = * (bstrTypeArr.pparray);
	}
	else {
		m_pbstrTypeArr = bstrTypeArr.parray;
	}

	if (!(bBootableArr.vt & VT_BOOL)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (!(bBootableArr.vt & VT_ARRAY)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (bBootableArr.vt & VT_BYREF) {
		m_pbBootableArr = * (bBootableArr.pparray);
	}
	else {
		m_pbBootableArr = bBootableArr.parray;
	}

	if (!(lPartitionNumberArr.vt & VT_I4)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (!(lPartitionNumberArr.vt & VT_ARRAY)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (bBootableArr.vt & VT_BYREF) {
		m_plPartitionNumberArr = * (lPartitionNumberArr.pparray);
	}
	else {
		m_plPartitionNumberArr = lPartitionNumberArr.parray;
	}

	if (!(lLengthArr.vt & VT_R8)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (!(lLengthArr.vt & VT_ARRAY)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (bBootableArr.vt & VT_BYREF) {
		m_plLengthArr = * (lLengthArr.pparray);
	}
	else {
		m_plLengthArr = lLengthArr.parray;
	}

	if (!(lHiddenSectorsArr.vt & VT_I4)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (!(lHiddenSectorsArr.vt & VT_ARRAY)) {
		return DISP_E_TYPEMISMATCH;
	}

	if (bBootableArr.vt & VT_BYREF) {
		m_plHiddenSectorsArr = * (lHiddenSectorsArr.pparray);
	}
	else {
		m_plHiddenSectorsArr = lHiddenSectorsArr.parray;
	}

	return S_OK;
}

void HDiskInfoPage::SetListData()
{

	CString tmpStr;
	BSTR tmpBstr;
	VARIANT_BOOL bBootable;
	int i = 0;
	long lLBound, lUBound ;
	long tmpNum;
	double length = 0.;
	LV_ITEM item;
	LV_COLUMN col1, col2, col3, col4, col5, col6;

	col1.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH;
	col1.fmt = LVCFMT_LEFT;
	col1.iSubItem = 0;
	col1.cx = 40;
	col1.pszText = _T ("Drive");

	col2.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH;
	col2.fmt = LVCFMT_LEFT;
	col2.iSubItem = 1;
	col2.cx = 50;
	col2.pszText = _T ("Number");

	col3.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH;
	col3.fmt = LVCFMT_LEFT;
	col3.iSubItem = 2;
	col3.cx = 55;
	col3.pszText = _T ("Bootable");

	col4.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH;
	col4.fmt = LVCFMT_LEFT;
	col4.iSubItem = 3;
	col4.cx = 155;
	col4.pszText = _T ("Type");

	col5.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH;
	col5.fmt = LVCFMT_LEFT;
	col5.iSubItem = 4;
	col5.cx = 85;
	col5.pszText = _T ("Length");

	col6.mask = LVCF_FMT | LVCF_SUBITEM | LVCF_TEXT | LVCF_WIDTH;
	col6.fmt = LVCFMT_LEFT;
	col6.iSubItem = 5;
	col6.cx = 95;
	col6.pszText = _T ("Hidden Sectors");

	m_ListCtrl.InsertColumn (0, &col1);
	m_ListCtrl.InsertColumn (1, &col2);
	m_ListCtrl.InsertColumn (2, &col3);
	m_ListCtrl.InsertColumn (3, &col4);
	m_ListCtrl.InsertColumn (4, &col5);
	m_ListCtrl.InsertColumn (5, &col6);

	// Get the uprrer and lower bound of arrays. And assume it will be same for all arrays.

	::SafeArrayGetLBound (m_pbstrDriveLetterArr, 1, &lLBound);
	::SafeArrayGetUBound (m_pbstrDriveLetterArr, 1, &lUBound);

	for (i = lLBound; i <= lUBound; i++) {
		item.mask = LVIF_TEXT;
		item.iItem = i;
		item.iSubItem = 0;
		item.pszText = _T (" ");
		m_ListCtrl.InsertItem (&item);

		::SafeArrayGetElement (m_pbstrDriveLetterArr, (long *) &i, &tmpBstr);
		m_ListCtrl.SetItemText (i, 0, CString (tmpBstr));

		::SafeArrayGetElement (m_pbBootableArr, (long *) &i, &bBootable);
		tmpStr = (bBootable) ? _T ("YES") : _T ("NO");
		m_ListCtrl.SetItemText (i, 2, tmpStr);

		::SafeArrayGetElement (m_plPartitionNumberArr, (long *) &i, &tmpNum);
		tmpStr.Format ("%d", tmpNum);
		m_ListCtrl.SetItemText (i, 1, tmpStr);

		::SafeArrayGetElement (m_pbstrTypeArr, (long *) &i, &tmpBstr);
		m_ListCtrl.SetItemText (i, 3, CString (tmpBstr));

		::SafeArrayGetElement (m_plLengthArr, (long *) &i, &length);
		tmpStr.Format ("%12.0lf", length);
		tmpStr.TrimLeft ();
		m_ListCtrl.SetItemText (i, 4, tmpStr);

		::SafeArrayGetElement (m_plHiddenSectorsArr, (long *) &i, &tmpNum);
		tmpStr.Format ("%d", tmpNum);
		m_ListCtrl.SetItemText (i, 5, tmpStr);
	}
}

void HDiskInfoPage::OnVScroll(UINT nSBCode, UINT nPos, CScrollBar* pScrollBar) 
{
	// TODO: Add your message handler code here and/or call default
	
	CPropertyPage::OnVScroll(nSBCode, nPos, pScrollBar);
}