From fb430c403b2e0deb3f34328d3b65bd39c10b14ba Mon Sep 17 00:00:00 2001
From: Mounir IDRASSI <mounir.idrassi@idrix.fr>
Date: Sun, 10 Jan 2016 19:35:57 +0100
Subject: Windows: Don't show disconnected network drives in the list of
 available drives. Add option to make them available for mounting if needed.

---
 src/Common/Dlgcode.c    | 46 ++++++++++++++++++++++++++++++++++++++++++----
 src/Common/Dlgcode.h    |  2 ++
 src/Common/Language.xml |  1 +
 3 files changed, 45 insertions(+), 4 deletions(-)

(limited to 'src/Common')

diff --git a/src/Common/Dlgcode.c b/src/Common/Dlgcode.c
index 6d86449c..cc7f0e74 100644
--- a/src/Common/Dlgcode.c
+++ b/src/Common/Dlgcode.c
@@ -101,6 +101,7 @@ wchar_t *lpszTitle = NULL;
 
 BOOL Silent = FALSE;
 BOOL bPreserveTimestamp = TRUE;
+BOOL bShowDisconnectedNetworkDrives = FALSE;
 BOOL bStartOnLogon = FALSE;
 BOOL bMountDevicesOnLogon = FALSE;
 BOOL bMountFavoritesOnLogon = FALSE;
@@ -225,6 +226,7 @@ HMODULE hbcryptdll = NULL;
 HMODULE hbcryptprimitivesdll = NULL;
 HMODULE hMsls31 = NULL;
 HMODULE hntmartadll = NULL;
+HMODULE hwinscarddll = NULL;
 
 #define FREE_DLL(h)	if (h) { FreeLibrary (h); h = NULL;}
 
@@ -572,6 +574,7 @@ void AbortProcessDirect (wchar_t *abortMsg)
 	FREE_DLL (hbcryptprimitivesdll);
 	FREE_DLL (hMsls31);
 	FREE_DLL (hntmartadll);
+	FREE_DLL (hwinscarddll);
 
 	exit (1);
 }
@@ -620,6 +623,7 @@ void AbortProcessSilent (void)
 	FREE_DLL (hbcryptprimitivesdll);
 	FREE_DLL (hMsls31);
 	FREE_DLL (hntmartadll);
+	FREE_DLL (hwinscarddll);
 
 	// Note that this function also causes localcleanup() to be called (see atexit())
 	exit (1);
@@ -2492,6 +2496,7 @@ void InitApp (HINSTANCE hInstance, wchar_t *lpszCommandLine)
 	InitOSVersionInfo();
 
 	LoadSystemDll (L"ntmarta.dll", &hntmartadll, TRUE, SRC_POS);
+	LoadSystemDll (L"MPR.DLL", &hmprdll, TRUE, SRC_POS);
 #ifdef SETUP
 	if (IsOSAtLeast (WIN_7))
 	{
@@ -2529,7 +2534,6 @@ void InitApp (HINSTANCE hInstance, wchar_t *lpszCommandLine)
 			LoadSystemDll (L"netapi32.dll", &hnetapi32dll, TRUE, SRC_POS);
 			LoadSystemDll (L"authz.dll", &hauthzdll, TRUE, SRC_POS);
 			LoadSystemDll (L"xmllite.dll", &hxmllitedll, TRUE, SRC_POS);
-			LoadSystemDll (L"mpr.dll", &hmprdll, TRUE, SRC_POS);						
 		}
 	}
 
@@ -2555,6 +2559,8 @@ void InitApp (HINSTANCE hInstance, wchar_t *lpszCommandLine)
 			LoadSystemDll (L"bcryptprimitives.dll", &hbcryptprimitivesdll, TRUE, SRC_POS);								
 		}
 	}	
+#else
+	LoadSystemDll (L"WINSCARD.DLL", &hwinscarddll, TRUE, SRC_POS);
 #endif
 
 	LoadSystemDll (L"COMCTL32.DLL", &hComctl32Dll, FALSE, SRC_POS);
@@ -2804,6 +2810,7 @@ void InitApp (HINSTANCE hInstance, wchar_t *lpszCommandLine)
 		FREE_DLL (hbcryptprimitivesdll);
 		FREE_DLL (hMsls31);
 		FREE_DLL (hntmartadll);
+		FREE_DLL (hwinscarddll);
 		exit (1);
 	}
 #endif
@@ -2847,6 +2854,7 @@ void FinalizeApp (void)
 	FREE_DLL (hbcryptprimitivesdll);
 	FREE_DLL (hMsls31);
 	FREE_DLL (hntmartadll);
+	FREE_DLL (hwinscarddll);
 }
 
 void InitHelpFileName (void)
@@ -6506,10 +6514,40 @@ BOOL WrongPwdRetryCountOverLimit (void)
 	return (WrongPwdRetryCounter > TC_TRY_HEADER_BAK_AFTER_NBR_WRONG_PWD_TRIES);
 }
 
+DWORD GetUsedLogicalDrives (void)
+{
+	DWORD dwUsedDrives = GetLogicalDrives();
+	if (!bShowDisconnectedNetworkDrives)
+	{
+		/* detect disconnected mapped network shares and removed
+		 * their associated drives from the list
+		 */
+		WCHAR remotePath[512];
+		WCHAR drive[3] = {L'A', L':', 0};
+		DWORD dwLen, status;
+		for (WCHAR i = 0; i <= MAX_MOUNTED_VOLUME_DRIVE_NUMBER; i++)
+		{
+			if ((dwUsedDrives & (1 << i)) == 0)
+			{
+				drive[0] = L'A' + i;
+				dwLen = ARRAYSIZE (remotePath);
+				status =  WNetGetConnection (drive, remotePath, &dwLen);
+				if ((NO_ERROR == status) || (status == ERROR_CONNECTION_UNAVAIL))
+				{
+					/* this is a mapped network share, mark it as used */
+					dwUsedDrives |= (1 << i);
+				}
+			}
+		}
+	}
+
+	return dwUsedDrives;
+}
+
 
 int GetFirstAvailableDrive ()
 {
-	DWORD dwUsedDrives = GetLogicalDrives();
+	DWORD dwUsedDrives = GetUsedLogicalDrives();
 	int i;
 
 	for (i = 0; i < 26; i++)
@@ -6524,7 +6562,7 @@ int GetFirstAvailableDrive ()
 
 int GetLastAvailableDrive ()
 {
-	DWORD dwUsedDrives = GetLogicalDrives();
+	DWORD dwUsedDrives = GetUsedLogicalDrives();
 	int i;
 
 	for (i = 25; i >= 0; i--)
@@ -6539,7 +6577,7 @@ int GetLastAvailableDrive ()
 
 BOOL IsDriveAvailable (int driveNo)
 {
-	return (GetLogicalDrives() & (1 << driveNo)) == 0;
+	return (GetUsedLogicalDrives() & (1 << driveNo)) == 0;
 }
 
 
diff --git a/src/Common/Dlgcode.h b/src/Common/Dlgcode.h
index 7ab74c08..601f7ac7 100644
--- a/src/Common/Dlgcode.h
+++ b/src/Common/Dlgcode.h
@@ -115,6 +115,7 @@ extern HWND MainDlg;
 extern BOOL Silent;
 extern BOOL bHistory;
 extern BOOL bPreserveTimestamp;
+extern BOOL bShowDisconnectedNetworkDrives;
 extern BOOL bStartOnLogon;
 extern BOOL bMountDevicesOnLogon;
 extern BOOL bMountFavoritesOnLogon;
@@ -330,6 +331,7 @@ void CorrectFileName (wchar_t* fileName);
 void IncreaseWrongPwdRetryCount (int count);
 void ResetWrongPwdRetryCount (void);
 BOOL WrongPwdRetryCountOverLimit (void);
+DWORD GetUsedLogicalDrives (void);
 int GetFirstAvailableDrive ();
 int GetLastAvailableDrive ();
 BOOL IsDriveAvailable (int driveNo);
diff --git a/src/Common/Language.xml b/src/Common/Language.xml
index 1fbed144..16821359 100644
--- a/src/Common/Language.xml
+++ b/src/Common/Language.xml
@@ -181,6 +181,7 @@
     <control lang="en" key="IDC_SELECT_DEVICE">Select D&amp;evice...</control>
     <control lang="en" key="IDC_SELECT_FILE">Select &amp;File...</control>
     <control lang="en" key="IDC_SELECT_PKCS11_MODULE">Select &amp;Library...</control>
+    <control lang="en" key="IDC_SHOW_DISCONNECTED_NETWORK_DRIVES">Make disconnected network drives available for mounting</control>
     <control lang="en" key="IDC_SHOW_PASSWORD_CHPWD_NEW">Display password</control>
     <control lang="en" key="IDC_SHOW_PASSWORD_CHPWD_ORI">Display password</control>
     <control lang="en" key="IDC_TRAVEL_OPEN_EXPLORER">Open &amp;Explorer window for mounted volume</control>
-- 
cgit v1.2.3