C / C++ / MFC

HANDLE disk;
DWORD dwNeeded;
DISK_GEOMETRY *geometry;

disk=CreateFile("\\\\.\\PHYSICALDRIVE0",
     GENERIC_READ,
     0,
     NULL,
     OPEN_EXISTING,
     0,
     NULL);
if (INVALID_HANDLE_VALUE != disk)
{
    geometry = new DISK_GEOMETRY;

    DeviceIoControl(disk,
         IOCTL_DISK_GET_DRIVE_GEOMETRY,
         NULL,
         0,
         geometry,
         sizeof(DISK_GEOMETRY),
         &dwNeeded,
         NULL);

    TRACE("      Cylinders: %I64u\n", geometry->Cylinders);
    TRACE("     Media Type: %d\n",    geometry->MediaType);
    TRACE("Tracks/Cylinder: %lu\n",   geometry->TracksPerCylinder);
    TRACE("  Sectors/Track: %lu\n",   geometry->SectorsPerTrack);
    TRACE("   Bytes/Sector: %lu\n",   geometry->BytesPerSector);

    delete [] geometry;

    CloseHandle(disk);
}

if( !IsWindow( m_YourTreeCtrl ) )
   return;

// Here, m_YourTreeCtrl can be accessed safely.

CRect treeRect;
treeRect.NormalizeRect();
CImageList *pImageList=new CImageList;
	
	
if(this->m_ctrlSetupTree.Create(WS_VISIBLE,treeRect,this,IDC_SETUP_TREE))
	{
		Beep(999,99);
		int i, icons=0;
		for (icons=0; RCanalTree::m_icons[icons]; icons++) {}
		pImageList->Create(16, 16, TRUE, icons,6);
		for (i=0; RCanalTree::m_icons[i]; i++)
		{
			pImageList->Add(AfxGetApp()->LoadIcon(RCanalTree::m_icons[i]));
		}
		// Assign the image list to all list and tree controls
		// so that icons can be used.	
		m_ctrlSetupTree.SetImageList(pImageList,TVSIL_NORMAL);
		
		// Instruct the Canal Manager to create a new system based on the information
		// in the Canal Structures database.
		
		
		if(this->m_oCanalManager.CreateNewDataStructures())
		{
			// Based on the data structures created in memory by the Canal Manager,
			// populate the setup tree.
			//this->m_ctrlSetupTree.Populate();
		}
		
	}
}