From 19294c43397d48e43f3fa7a07f3d0214f7af5a5c Mon Sep 17 00:00:00 2001
From: spx <spx268@gmail.com>
Date: Sun, 28 Nov 2010 22:20:27 +0000
Subject: [PATCH] Minor change.

---
 Library/MeasureDiskSpace.cpp | 41 ++++++++++++++++++------------------
 1 file changed, 20 insertions(+), 21 deletions(-)

diff --git a/Library/MeasureDiskSpace.cpp b/Library/MeasureDiskSpace.cpp
index 01579a38..5d8aeae3 100644
--- a/Library/MeasureDiskSpace.cpp
+++ b/Library/MeasureDiskSpace.cpp
@@ -58,9 +58,8 @@ bool CMeasureDiskSpace::Update()
 
 	if (!m_Drive.empty())
 	{
-		BOOL sizeResult = FALSE, labelResult = FALSE;
+		BOOL sizeResult = FALSE;
 		ULARGE_INTEGER i64TotalBytes, i64FreeBytes;
-		WCHAR volumeName[MAX_PATH] = {0};
 
 		UINT type = GetDriveType(m_Drive.c_str());
 		if (type != DRIVE_NO_ROOT_DIR)
@@ -75,22 +74,11 @@ bool CMeasureDiskSpace::Update()
 				sizeResult = GetDiskFreeSpaceEx(m_Drive.c_str(), &i64FreeBytesToCaller, &i64TotalBytes, &i64FreeBytes);
 				SetErrorMode(oldMode);  // Reset
 			}
-
-			if (m_Label)
-			{
-				if (!m_IgnoreRemovable || type != DRIVE_REMOVABLE)	// Ignore removable drives
-				{
-					UINT oldMode = SetErrorMode(0);
-					SetErrorMode(oldMode | SEM_FAILCRITICALERRORS);  // Prevent the system from displaying message box
-					labelResult = GetVolumeInformation(m_Drive.c_str(), volumeName, MAX_PATH, NULL, NULL, NULL, NULL, 0);
-					SetErrorMode(oldMode);  // Reset
-				}
-			}
 		}
 
 		if (sizeResult)
 		{
-			m_Value = (double)((m_Total) ? i64TotalBytes.QuadPart : i64FreeBytes.QuadPart);
+			m_Value = (double)((m_Total) ? i64TotalBytes : i64FreeBytes).QuadPart;
 
 			if (i64TotalBytes.QuadPart != m_OldTotalBytes)
 			{
@@ -106,16 +94,27 @@ bool CMeasureDiskSpace::Update()
 			m_OldTotalBytes = 0;
 		}
 
-		if (labelResult)
+		if (m_Label)
 		{
-			m_LabelName = volumeName;
-		}
-		else
-		{
-			if (m_Label || !m_LabelName.empty())
+			BOOL labelResult = FALSE;
+			WCHAR volumeName[MAX_PATH] = {0};
+
+			if (type != DRIVE_NO_ROOT_DIR)
 			{
-				m_LabelName = L"";
+				if (!m_IgnoreRemovable || type != DRIVE_REMOVABLE)	// Ignore removable drives
+				{
+					UINT oldMode = SetErrorMode(0);
+					SetErrorMode(oldMode | SEM_FAILCRITICALERRORS);  // Prevent the system from displaying message box
+					labelResult = GetVolumeInformation(m_Drive.c_str(), volumeName, MAX_PATH, NULL, NULL, NULL, NULL, 0);
+					SetErrorMode(oldMode);  // Reset
+				}
 			}
+
+			m_LabelName = (labelResult) ? volumeName : L"";
+		}
+		else if (!m_LabelName.empty())
+		{
+			m_LabelName = L"";
 		}
 	}