extern "C" {
#include <afs/param.h>
#include <afs/stds.h>
+#include <rx/rxkad.h>
+#include <afs/fs_utils.h>
}
-
#include <windows.h>
#include <stdlib.h>
#include <stdio.h>
#include <WINNT/TaLocale.h>
+#undef REALLOC
#include "drivemap.h"
+#include <time.h>
+#include <adssts.h>
+#define DEBUG_VERBOSE
+#include <osilog.h>
+#include <lanahelper.h>
+extern void Config_GetLanAdapter (ULONG *pnLanAdapter);
+extern BOOL Config_ReadNum (LPCTSTR pszLHS, DWORD *pdwRHS);
+extern BOOL Config_ReadString (LPCTSTR pszLHS, LPTSTR pszRHS, size_t cchMax);
/*
* REGISTRY ___________________________________________________________________
*
*/
-static const TCHAR AFSConfigKeyName[] = TEXT("SYSTEM\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters");
+#undef AFSConfigKeyName
+const TCHAR sAFSConfigKeyName[] = TEXT("SYSTEM\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters");
/*
static TCHAR cszINIFILE[] = TEXT("afsdsbmt.ini");
static TCHAR cszSECTION_SUBMOUNTS[] = TEXT("AFS Submounts");
static TCHAR cszSECTION_MAPPINGS[] = TEXT("AFS Mappings");
+static TCHAR cszSECTION_ACTIVE[] = TEXT("AFS Active");
static TCHAR cszAUTOSUBMOUNT[] = TEXT("Auto");
static TCHAR cszLANMANDEVICE[] = TEXT("\\Device\\LanmanRedirector\\");
return fIsWinNT;
}
-
+/* Check if the OS is Windows 2000 or higher.
+*/
BOOL IsWindows2000 (void)
{
static BOOL fChecked = FALSE;
void GetClientNetbiosName (LPTSTR pszName)
{
- *pszName = TEXT('\0');
-
- if (IsWindowsNT())
- {
- HKEY hk;
- if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Control\\ComputerName\\ComputerName"), &hk) == 0)
- {
- DWORD dwSize = MAX_PATH;
- DWORD dwType = REG_SZ;
- RegQueryValueEx (hk, TEXT("ComputerName"), NULL, &dwType, (PBYTE)pszName, &dwSize);
- }
- }
- else // (!IsWindowsNT())
- {
- HKEY hk;
- if (RegOpenKey (HKEY_LOCAL_MACHINE, TEXT("System\\CurrentControlSet\\Services\\TransarcAFSDaemon\\Parameters"), &hk) == 0)
- {
- DWORD dwSize = MAX_PATH;
- DWORD dwType = REG_SZ;
- RegQueryValueEx (hk, TEXT("Gateway"), NULL, &dwType, (PBYTE)pszName, &dwSize);
- }
- }
-
- // Shorten the server name from its FQDN
- //
- for (LPTSTR pch = pszName; *pch; ++pch)
- {
- if (*pch == TEXT('.'))
- {
- *(LPTSTR)pch = TEXT('\0');
- break;
- }
- }
-
- // Form NetBIOS name from client's (possibly truncated) simple host name.
- if (*pszName != TEXT('\0')) {
- pszName[11] = TEXT('\0');
- lstrcat(pszName, TEXT("-AFS"));
- }
+ *pszName = TEXT('\0');
+ lana_GetNetbiosName(pszName, LANA_NETBIOS_NAME_FULL);
}
//
if (!lstrcmpi (pszSubmount, TEXT("all")))
{
- lstrcpy (pszPath, TEXT("/afs"));
+ lstrcpy (pszPath, cm_slash_mount_root);
return TRUE;
}
FreeStringMemory (mszLHS);
}
+BOOL ForceMapActive (TCHAR chDrive)
+{
+ TCHAR szDrive[2];
+ TCHAR szActive[32];
+
+ szDrive[0] = chDrive;
+ szDrive[1] = 0;
+
+ GetPrivateProfileString (cszSECTION_ACTIVE, szDrive, TEXT("0"), szActive, sizeof(szActive), cszINIFILE);
+
+ if ( !lstrcmp(szActive,"1") || !lstrcmpi(szActive,"true") || !lstrcmpi(szActive,"on") || !lstrcmpi(szActive,"yes") )
+ return TRUE;
+ return FALSE;
+}
+
+
+void WriteActiveMap (TCHAR chDrive, BOOL on)
+{
+ TCHAR szDrive[2];
+
+ szDrive[0] = chDrive;
+ szDrive[1] = 0;
+
+ WritePrivateProfileString (cszSECTION_ACTIVE, szDrive, on ? "1" : "0", cszINIFILE);
+}
void QueryDriveMapList_WriteMappings (PDRIVEMAPLIST pList)
{
TCHAR szValue[128];
HKEY hKey;
- _stprintf(szKeyName, TEXT("%s\\GlobalAutoMapper"), AFSConfigKeyName);
+ _stprintf(szKeyName, TEXT("%s\\GlobalAutoMapper"), sAFSConfigKeyName);
if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, szKeyName, 0, KEY_QUERY_VALUE, &hKey) != ERROR_SUCCESS)
return FALSE;
IOInfo.out = (char *)OutData;
IOInfo.out_size = PIOCTL_MAXSIZE;
- ULONG status;
- if ((status = pioctl (0, VIOC_MAKESUBMOUNT, &IOInfo, 1)) != 0)
+ ULONG status = pioctl (0, VIOC_MAKESUBMOUNT, &IOInfo, 1);
+ if (pStatus)
+ *pStatus = status;
+
+ if (status)
return FALSE;
lstrcpy (pszSubmount, (LPCTSTR)OutData);
{
// We can only map drives to places in AFS using this function.
//
- if ( (lstrncmpi (pszMapping, TEXT("/afs"), lstrlen(TEXT("/afs")))) &&
- (lstrncmpi (pszMapping, TEXT("\\afs"), lstrlen(TEXT("\\afs")))) )
+ if ( (lstrncmpi (pszMapping, cm_slash_mount_root, lstrlen(cm_slash_mount_root))) &&
+ (lstrncmpi (pszMapping, cm_back_slash_mount_root, lstrlen(cm_back_slash_mount_root))) )
{
if (pdwStatus)
*pdwStatus = ERROR_BAD_NETPATH;
}
// We now have a submount name and drive letter--map the network drive.
- //
- TCHAR szClient[ MAX_PATH ];
- GetClientNetbiosName (szClient);
-
- TCHAR szLocal[ MAX_PATH ] = TEXT("*:");
- szLocal[0] = chDrive;
-
- TCHAR szRemote[ MAX_PATH ];
- wsprintf (szRemote, TEXT("\\\\%s\\%s"), szClient, szSubmount);
-
- NETRESOURCE Resource;
- memset (&Resource, 0x00, sizeof(NETRESOURCE));
- Resource.dwScope = RESOURCE_GLOBALNET;
- Resource.dwType = RESOURCETYPE_DISK;
- Resource.dwDisplayType = RESOURCEDISPLAYTYPE_SHARE;
- Resource.dwUsage = RESOURCEUSAGE_CONNECTABLE;
- Resource.lpLocalName = szLocal;
- Resource.lpRemoteName = szRemote;
-
- DWORD rc = WNetAddConnection2 (&Resource, NULL, NULL, ((fPersistent) ? CONNECT_UPDATE_PROFILE : 0));
+ DWORD rc=MountDOSDrive(chDrive,szSubmount,fPersistent);
if (rc == NO_ERROR)
return TRUE;
BOOL InactivateDriveMap (TCHAR chDrive, DWORD *pdwStatus)
{
- TCHAR szLocal[ MAX_PATH ] = TEXT("*:");
- szLocal[0] = chDrive;
-
- DWORD rc = WNetCancelConnection (szLocal, FALSE);
+ DWORD rc = DisMountDOSDrive(chDrive, FALSE);
if (rc == NO_ERROR)
return TRUE;
void AdjustAfsPath (LPTSTR pszTarget, LPCTSTR pszSource, BOOL fWantAFS, BOOL fWantForwardSlashes)
{
- if (!*pszSource)
- lstrcpy (pszTarget, (fWantAFS) ? TEXT("/afs") : TEXT(""));
- else if ((*pszSource != TEXT('/')) && (*pszSource != TEXT('\\')))
- wsprintf (pszTarget, TEXT("/afs/%s"), pszSource);
- // We don't want to strip afs off the start if it is part of something for example afscell.company.com
- else if (fWantAFS && (lstrncmpi (&pszSource[1], TEXT("afs"), 3)) || !((pszSource[4] == TEXT('/')) ||
- (pszSource[4] == TEXT('\\')) ||
- (lstrlen(pszSource) == 4)))
- wsprintf (pszTarget, TEXT("/afs%s"), pszSource);
- else if (!fWantAFS && (!lstrncmpi (&pszSource[1], TEXT("afs"), 3) || ((pszSource[4] == TEXT('/')) ||
- (pszSource[4] == TEXT('\\')) ||
- (lstrlen(pszSource) == 4))))
- lstrcpy (pszTarget, &pszSource[4]);
- else
- lstrcpy (pszTarget, pszSource);
+ if (!*pszSource)
+ lstrcpy (pszTarget, (fWantAFS) ? cm_slash_mount_root : TEXT(""));
+ else if ((*pszSource != TEXT('/')) && (*pszSource != TEXT('\\')))
+ wsprintf (pszTarget, TEXT("%s/%s"),cm_slash_mount_root, pszSource);
+ // We don't want to strip afs off the start if it is part of something for example afscell.company.com
+ else if (fWantAFS && (lstrncmpi (&pszSource[1], cm_mount_root, strlen(cm_mount_root))) || !((pszSource[strlen(cm_slash_mount_root)] == TEXT('/')) ||
+ (pszSource[strlen(cm_slash_mount_root)] == TEXT('\\')) ||
+ (lstrlen(pszSource) == strlen(cm_slash_mount_root))))
+ wsprintf (pszTarget, TEXT("%s%s"),cm_slash_mount_root, pszSource);
+ else if (!fWantAFS && (!lstrncmpi (&pszSource[1], cm_mount_root, strlen(cm_mount_root)) && ((pszSource[strlen(cm_slash_mount_root)] == TEXT('/')) ||
+ (pszSource[strlen(cm_slash_mount_root)] == TEXT('\\')) ||
+ (lstrlen(pszSource) == strlen(cm_slash_mount_root)))))
+ lstrcpy (pszTarget, &pszSource[strlen(cm_slash_mount_root)]);
+ else
+ lstrcpy (pszTarget, pszSource);
for (LPTSTR pch = pszTarget; *pch; ++pch)
{
}
}
-
BOOL GetDriveSubmount (TCHAR chDrive, LPTSTR pszSubmountNow)
{
- TCHAR szDrive[] = TEXT("*:");
- szDrive[0] = chDrive;
+ TCHAR szDrive[] = TEXT("*:");
+ szDrive[0] = chDrive;
- TCHAR szMapping[ MAX_PATH ] = TEXT("");
- LPTSTR pszSubmount = szMapping;
+ TCHAR szMapping[ _MAX_PATH ] = TEXT("");
+ LPTSTR pszSubmount = szMapping;
+ TCHAR szNetBiosName[32];
+
+ memset(szNetBiosName, '\0', sizeof(szNetBiosName));
+ lana_GetNetbiosName(szNetBiosName, LANA_NETBIOS_NAME_FULL);
+ _tcscat(szNetBiosName, TEXT("\\"));
if (IsWindowsNT())
- {
- QueryDosDevice (szDrive, szMapping, MAX_PATH);
+ {
+ if (!QueryDosDevice (szDrive, szMapping, MAX_PATH))
+ return FALSE;
// Now if this is an AFS network drive mapping, {szMapping} will be:
//
- // \Device\LanmanRedirector\Q:\machine-afs\submount
+ // \Device\LanmanRedirector\<Drive>:\<netbiosname>\submount
//
// on Windows NT. On Windows 2000, it will be:
//
- // \Device\LanmanRedirector\;Q:0\machine-afs\submount
+ // \Device\LanmanRedirector\;<Drive>:0\<netbiosname>\submount
//
// (This is presumably to support multiple drive mappings with
// Terminal Server).
//
- if (lstrncmpi (szMapping, cszLANMANDEVICE, lstrlen(cszLANMANDEVICE)))
+ // on Windows XP and 2003, it will be :
+ // \Device\LanmanRedirector\;<Drive>:<AuthID>\<netbiosname>\submount
+ //
+ // where : <Drive> : DOS drive letter
+ // <AuthID>: Authentication ID, 16 char hex.
+ // <netbiosname>: Netbios name of server
+ //
+ if (_tcsnicmp(szMapping, cszLANMANDEVICE, _tcslen(cszLANMANDEVICE)))
return FALSE;
- pszSubmount = &szMapping[ lstrlen(cszLANMANDEVICE) ];
+ pszSubmount = &szMapping[ _tcslen(cszLANMANDEVICE) ];
if (IsWindows2000())
+ {
if (*(pszSubmount) != TEXT(';'))
return FALSE;
+ } else
+ --pszSubmount;
if (toupper(*(++pszSubmount)) != chDrive)
return FALSE;
if (*(++pszSubmount) != TEXT(':'))
return FALSE;
+#ifdef COMMENT
+ // No longer a safe assumption on XP
if (IsWindows2000())
if (*(++pszSubmount) != TEXT('0'))
return FALSE;
+#endif
- if (*(++pszSubmount) != TEXT('\\'))
- return FALSE;
+ // scan for next "\"
+ while (*(++pszSubmount) != TEXT('\\'))
+ {
+ if (*pszSubmount==0)
+ return FALSE;
+ }
+
+ // note that szNetBiosName has a '\\' tagged in the end earlier
for (++pszSubmount; *pszSubmount && (*pszSubmount != TEXT('\\')); ++pszSubmount)
- if (!lstrncmpi (pszSubmount, TEXT("-afs\\"), lstrlen(TEXT("-afs\\"))))
+ if (!_tcsncicmp(pszSubmount, szNetBiosName, _tcslen(szNetBiosName)))
break;
if ((!*pszSubmount) || (*pszSubmount == TEXT('\\')))
return FALSE;
- pszSubmount += lstrlen("-afs\\");
+
+ pszSubmount += _tcslen(szNetBiosName);
}
else // (!IsWindowsNT())
{
if (*(pszSubmount++) != TEXT('\\'))
return FALSE;
for ( ; *pszSubmount && (*pszSubmount != TEXT('\\')); ++pszSubmount)
- if (!lstrncmpi (pszSubmount, TEXT("-afs\\"), lstrlen(TEXT("-afs\\"))))
+ if (!lstrncmpi (pszSubmount, szNetBiosName, lstrlen(szNetBiosName)))
break;
if ((!*pszSubmount) || (*pszSubmount == TEXT('\\')))
return FALSE;
- pszSubmount += lstrlen("-afs\\");
+ pszSubmount += lstrlen(szNetBiosName);
}
if (!pszSubmount || !*pszSubmount)
return TRUE;
}
+/* Generate Random User name random acording to time*/
+DWORD dwOldState=0;
+TCHAR pUserName[MAXRANDOMNAMELEN]=TEXT("");
+BOOL fUserName=FALSE;
+#define AFSLogonOptionName TEXT("System\\CurrentControlSet\\Services\\TransarcAFSDaemon\\NetworkProvider")
+
+void SetBitLogonOption(BOOL set,DWORD value)
+{
+
+ RWLogonOption(FALSE,((set)?value | RWLogonOption(TRUE,0):RWLogonOption(TRUE,0) & ~value) );
+}
+
+DWORD RWLogonOption(BOOL read,DWORD value)
+{
+ // if read is true then if value==0 return registry value
+ // if read and value!=0 then use value to test registry, return TRUE if value bits match value read
+ HKEY hk;
+ DWORD dwDisp;
+ DWORD LSPtype, LSPsize;
+ DWORD rval;
+ if (read)
+ {
+ rval=0;
+ if (RegOpenKeyEx(HKEY_LOCAL_MACHINE, AFSLogonOptionName,0, KEY_QUERY_VALUE, &hk)==ERROR_SUCCESS)
+ {
+ LSPsize=sizeof(rval);
+ RegQueryValueEx(hk, "LogonOptions", NULL,
+ &LSPtype, (LPBYTE)&rval, &LSPsize);
+ RegCloseKey (hk);
+ }
+ return (value==0)?rval:((rval & value)==value);
+
+ } else { //write
+ if (RegCreateKeyEx (HKEY_LOCAL_MACHINE, AFSLogonOptionName, 0, NULL, 0, KEY_SET_VALUE, NULL, &hk, &dwDisp) == ERROR_SUCCESS)
+ {
+ RegSetValueEx(hk,TEXT("LogonOptions"),NULL,REG_DWORD,(LPBYTE)&value,sizeof(value));
+ RegCloseKey (hk);
+ }
+ return TRUE;
+ }
+}
+
+void MapShareName(char *pszCmdLineA)
+{
+ fUserName = TRUE;
+ TCHAR *p=pUserName;
+ pszCmdLineA++;
+ while (*pszCmdLineA && (*pszCmdLineA != ' '))
+ {
+ *p++=*pszCmdLineA++;
+ }
+}
+
+void GenRandomName(TCHAR *pname,int len)
+{
+ if (fUserName)
+ { //user name was passed through command line, use once
+ fUserName=FALSE;
+ return;
+ }
+ srand( (unsigned)time( NULL ) );
+ for (int i=0;i<len;i++)
+ pname[i]='a'+(rand() % 26);
+ pname[len]=0;
+ return;
+}
+
+/*
+ Make a connection using users name
+ if fUserName then force a connection
+*/
+
+BOOL TestAndDoMapShare(DWORD dwState)
+{
+ if ((dwState!=SERVICE_RUNNING) || (dwOldState!=SERVICE_START_PENDING))
+ {
+ dwOldState=dwState;
+ return TRUE;
+ }
+ dwOldState=SERVICE_RUNNING;
+ if (RWLogonOption(TRUE,LOGON_OPTION_HIGHSECURITY))
+ return (DoMapShare() && GlobalMountDrive());
+ return GlobalMountDrive();
+}
+
+BOOL IsServiceActive()
+{
+ SC_HANDLE hManager;
+ SERVICE_STATUS Status;
+ if ((hManager = OpenSCManager (NULL, NULL, GENERIC_READ)) != NULL)
+ {
+ SC_HANDLE hService;
+ if ((hService = OpenService (hManager, TEXT("TransarcAFSDaemon"), GENERIC_READ)) != NULL)
+ {
+ QueryServiceStatus (hService, &Status);
+ CloseServiceHandle (hService);
+ }
+
+ CloseServiceHandle (hManager);
+ }
+
+ return (Status.dwCurrentState == SERVICE_RUNNING) ? TRUE : FALSE;
+}
+
+void TestAndDoUnMapShare()
+{
+ if (!RWLogonOption(TRUE,LOGON_OPTION_HIGHSECURITY))
+ return;
+ DoUnMapShare(FALSE);
+}
+
+void DoUnMapShare(BOOL drivemap) //disconnect drivemap
+{
+ TCHAR szMachine[MAX_PATH],szPath[MAX_PATH];
+ DWORD rc=28;
+ HANDLE hEnum;
+ LPNETRESOURCE lpnrLocal,lpnr=NULL;
+ DWORD res;
+ DWORD cbBuffer=16384;
+ DWORD cEntries=-1;
+ CHAR *pSubmount="";
+
+ memset(szMachine, '\0', sizeof(szMachine));
+ lana_GetNetbiosName(szMachine, LANA_NETBIOS_NAME_FULL);
+
+ // Initialize the data structure
+ if ((res=WNetOpenEnum(RESOURCE_CONNECTED,RESOURCETYPE_DISK,RESOURCEUSAGE_CONNECTABLE,lpnr,&hEnum))!=NO_ERROR)
+ return;
+ sprintf(szPath,"\\\\%s\\",szMachine);
+ _strlwr(szPath);
+ lpnrLocal=(LPNETRESOURCE) GlobalAlloc(GPTR,cbBuffer);
+ do {
+ memset(lpnrLocal,0,cbBuffer);
+ if ((res = WNetEnumResource(hEnum,&cEntries,lpnrLocal,&cbBuffer))==NO_ERROR)
+ {
+ for (DWORD i=0;i<cEntries;i++)
+ {
+ if (strstr(_strlwr(lpnrLocal[i].lpRemoteName),szPath))
+ {
+ if ((lpnrLocal[i].lpLocalName) && (strlen(lpnrLocal[i].lpLocalName)>0))
+ {
+ if (drivemap) {
+ DisMountDOSDrive(*lpnrLocal[i].lpLocalName);
+ DEBUG_EVENT1("AFS DriveUnMap","UnMap-Local=%x",res);
+ }
+ } else {
+ DisMountDOSDriveFull(lpnrLocal[i].lpRemoteName);
+ DEBUG_EVENT1("AFS DriveUnMap","UnMap-Remote=%x",res);
+ }
+ }
+ }
+ }
+ } while (res!=ERROR_NO_MORE_ITEMS);
+ GlobalFree((HGLOBAL)lpnrLocal);
+ WNetCloseEnum(hEnum);
+}
+
+BOOL DoMapShareChange()
+{
+ DRIVEMAPLIST List;
+ TCHAR szMachine[ MAX_PATH],szPath[MAX_PATH];
+ DWORD rc=28;
+ HANDLE hEnum;
+ LPNETRESOURCE lpnrLocal,lpnr=NULL;
+ DWORD res;
+ DWORD cEntries=-1;
+ DWORD cbBuffer=16384;
+
+ memset(szMachine, '\0', sizeof(szMachine));
+ lana_GetNetbiosName(szMachine, LANA_NETBIOS_NAME_FULL);
+
+ // Initialize the data structure
+ if (!IsServiceActive())
+ return TRUE;
+ memset (&List, 0x00, sizeof(DRIVEMAPLIST));
+ for (size_t ii = 0; ii < 26; ++ii)
+ List.aDriveMap[ii].chDrive = chDRIVE_A + ii;
+ QueryDriveMapList_ReadSubmounts (&List);
+ if ((res=WNetOpenEnum(RESOURCE_CONNECTED,RESOURCETYPE_DISK,RESOURCEUSAGE_CONNECTABLE,lpnr,&hEnum))!=NO_ERROR)
+ return FALSE;
+ lpnrLocal=(LPNETRESOURCE) GlobalAlloc(GPTR,cbBuffer);
+ sprintf(szPath,"\\\\%s\\",szMachine);
+ _strlwr(szPath);
+ do {
+ memset(lpnrLocal,0,cbBuffer);
+ if ((res = WNetEnumResource(hEnum,&cEntries,lpnrLocal,&cbBuffer))==NO_ERROR)
+ {
+ for (DWORD i=0;i<cEntries;i++)
+ {
+ if (strstr(_strlwr(lpnrLocal[i].lpRemoteName),szPath)==NULL)
+ continue; //only look at real afs mappings
+ CHAR * pSubmount=strrchr(lpnrLocal[i].lpRemoteName,'\\')+1;
+ if (lstrcmpi(pSubmount,"all")==0)
+ continue; // do not remove 'all'
+ for (DWORD j=0;j<List.cSubmounts;j++)
+ {
+ if (
+ (List.aSubmounts[j].szSubmount[0]) &&
+ (lstrcmpi(List.aSubmounts[j].szSubmount,pSubmount)==0)
+ )
+ {
+ List.aSubmounts[j].fInUse=TRUE;
+ goto nextname;
+ }
+ }
+ // wasn't on list so lets remove
+ DisMountDOSDrive(pSubmount);
+ nextname:;
+ }
+ }
+ } while (res!=ERROR_NO_MORE_ITEMS);
+ GlobalFree((HGLOBAL)lpnrLocal);
+ WNetCloseEnum(hEnum);
+ sprintf(szPath,"\\\\%s\\all",szMachine);
+
+ // Lets connect all submounts that weren't connectd
+ DWORD cbUser=MAXRANDOMNAMELEN-1;
+ CHAR szUser[MAXRANDOMNAMELEN];
+ CHAR * pUser = NULL;
+ if (WNetGetUser(szPath,(LPSTR)szUser,&cbUser)!=NO_ERROR) {
+ if (RWLogonOption(TRUE,LOGON_OPTION_HIGHSECURITY)) {
+ if (!pUserName[0]) {
+ GenRandomName(szUser,MAXRANDOMNAMELEN-1);
+ pUser = szUser;
+ } else {
+ pUser = pUserName;
+ }
+ }
+ } else {
+ if ((pUser=strchr(szUser,'\\'))!=NULL)
+ pUser++;
+ }
+
+ for (DWORD j=0;j<List.cSubmounts;j++)
+ {
+ if (List.aSubmounts[j].fInUse)
+ continue;
+ DWORD res=MountDOSDrive(0,List.aSubmounts[j].szSubmount,FALSE,pUser);
+ }
+ return TRUE;
+}
+
+BOOL DoMapShare()
+{
+ DRIVEMAPLIST List;
+ DWORD rc=28;
+ BOOL bMappedAll=FALSE;
+
+ // Initialize the data structure
+ DEBUG_EVENT0("AFS DoMapShare");
+ QueryDriveMapList (&List);
+ DoUnMapShare(TRUE);
+ // All connections have been removed
+ // Lets restore them after making the connection from the random name
+
+ TCHAR szMachine[ MAX_PATH],szPath[MAX_PATH];
+ memset(szMachine, '\0', sizeof(szMachine));
+ lana_GetNetbiosName(szMachine, LANA_NETBIOS_NAME_FULL);
+ sprintf(szPath,"\\\\%s\\all",szMachine);
+
+ // Lets connect all submounts that weren't connectd
+ DWORD cbUser=MAXRANDOMNAMELEN-1;
+ CHAR szUser[MAXRANDOMNAMELEN];
+ CHAR * pUser = NULL;
+ if (WNetGetUser(szPath,(LPSTR)szUser,&cbUser)!=NO_ERROR) {
+ if (RWLogonOption(TRUE,LOGON_OPTION_HIGHSECURITY)) {
+ if (!pUserName[0]) {
+ GenRandomName(szUser,MAXRANDOMNAMELEN-1);
+ pUser = szUser;
+ } else {
+ pUser = pUserName;
+ }
+ }
+ } else {
+ if ((pUser=strchr(szUser,'\\'))!=NULL)
+ pUser++;
+ }
+
+ for (DWORD i=0;i<List.cSubmounts;i++)
+ {
+ if (List.aSubmounts[i].szSubmount[0])
+ {
+ DWORD res=MountDOSDrive(0,List.aSubmounts[i].szSubmount,FALSE,pUser);
+ if (lstrcmpi("all",List.aSubmounts[i].szSubmount)==0)
+ bMappedAll=TRUE;
+ }
+ }
+ if (!bMappedAll) //make sure all is mapped also
+ {
+ DWORD res=MountDOSDrive(0,"all",FALSE,pUser);
+ if (res==ERROR_SESSION_CREDENTIAL_CONFLICT)
+ {
+ DisMountDOSDrive("all");
+ MountDOSDrive(0,"all",FALSE,pUser);
+ }
+ }
+ for (TCHAR chDrive = chDRIVE_A; chDrive <= chDRIVE_Z; ++chDrive)
+ {
+ if (List.aDriveMap[chDrive-chDRIVE_A].fActive ||
+ ForceMapActive(chDrive))
+ {
+ TCHAR szSubmount[ MAX_PATH ];
+ if (List.aDriveMap[chDrive-chDRIVE_A].szSubmount[0])
+ lstrcpy(szSubmount,List.aDriveMap[chDrive-chDRIVE_A].szSubmount);
+ else if (!PathToSubmount (szSubmount, List.aDriveMap[chDrive-chDRIVE_A].szMapping, NULL, NULL))
+ continue;
+
+ BOOL fPersistent = List.aDriveMap[chDrive-chDRIVE_A].fPersistent;
+ if (RWLogonOption(TRUE,LOGON_OPTION_HIGHSECURITY))
+ fPersistent = FALSE;
+ DWORD res=MountDOSDrive(chDrive
+ ,szSubmount
+ ,fPersistent,pUser);
+ }
+ }
+ return TRUE;
+}
+
+BOOL GlobalMountDrive()
+{
+ char szDriveToMapTo[5];
+ DWORD dwResult;
+ char szKeyName[256];
+ HKEY hKey;
+ DWORD dwIndex = 0;
+ DWORD dwDriveSize;
+ DWORD dwSubMountSize;
+ char unsigned szSubMount[256];
+ char cm_HostName[200];
+ DWORD dwType=sizeof(cm_HostName);
+ if (!IsServiceActive())
+ return TRUE;
+ if (!GetComputerName(cm_HostName, &dwType))
+ return TRUE;
+ sprintf(szKeyName, "%s\\GlobalAutoMapper", sAFSConfigKeyName);
+
+ dwResult = RegOpenKeyEx(HKEY_LOCAL_MACHINE, szKeyName, 0, KEY_QUERY_VALUE,
+ &hKey);
+ if (dwResult != ERROR_SUCCESS)
+ return TRUE;
+
+ while (1) {
+ dwDriveSize = sizeof(szDriveToMapTo);
+ dwSubMountSize = sizeof(szSubMount);
+ dwResult = RegEnumValue(hKey, dwIndex++, szDriveToMapTo, &dwDriveSize,
+ 0, &dwType, szSubMount, &dwSubMountSize);
+ if (dwResult != ERROR_MORE_DATA) {
+ if (dwResult != ERROR_SUCCESS) {
+ if (dwResult != ERROR_NO_MORE_ITEMS)
+ {
+ DEBUG_EVENT1("AFS DriveMap","Failed to read GlobalAutoMapper values: %d",dwResult);
+ }
+ break;
+ }
+ }
+ dwResult=MountDOSDrive(*szDriveToMapTo,(const char *)szSubMount,FALSE);
+ }
+ RegCloseKey(hKey);
+ return TRUE;
+}
+
+DWORD MountDOSDrive(char chDrive,const char *szSubmount,BOOL bPersistent,const char * pUsername)
+{
+ TCHAR szPath[MAX_PATH];
+ TCHAR szClient[MAX_PATH];
+ TCHAR szDrive[3] = TEXT("?:");
+ sprintf(szDrive,"%c:",chDrive);
+ GetClientNetbiosName (szClient);
+ sprintf(szPath,"\\\\%s\\%s",szClient,szSubmount);
+ NETRESOURCE nr;
+ memset (&nr, 0x00, sizeof(NETRESOURCE));
+ nr.dwType=RESOURCETYPE_DISK;
+ nr.lpLocalName=szDrive;
+ nr.lpRemoteName=szPath;
+ nr.dwDisplayType = RESOURCEDISPLAYTYPE_SHARE;
+ DWORD res=WNetAddConnection2(&nr,NULL,pUsername,(bPersistent)?CONNECT_UPDATE_PROFILE:0);
+ DEBUG_EVENT5("AFS DriveMap","Mount %s Local[%s] Remote[%s] User[%s]=%x",
+ (bPersistent)?"Persistant" : "NonPresistant",
+ szDrive,szPath,pUsername?pUsername:"NULL",res);
+ return res;
+}
+
+DWORD DisMountDOSDriveFull(const char *szPath,BOOL bForce)
+{
+ DWORD res=WNetCancelConnection(szPath,bForce);
+ DEBUG_EVENT3("AFS DriveMap","%sDismount Remote[%s]=%x",
+ bForce ? "Forced " : "",szPath,res);
+ return (res==ERROR_NOT_CONNECTED)?NO_ERROR:res;
+}
+
+DWORD DisMountDOSDrive(const char *pSubmount,BOOL bForce)
+{
+ TCHAR szPath[MAX_PATH];
+ TCHAR szClient[MAX_PATH];
+ GetClientNetbiosName (szClient);
+ sprintf(szPath,"\\\\%s\\%s",szClient,pSubmount);
+ return DisMountDOSDriveFull(szPath,bForce);
+}
+
+
+DWORD DisMountDOSDrive(const char chDrive,BOOL bForce)
+{
+ TCHAR szPath[MAX_PATH];
+ sprintf(szPath,"%c:",chDrive);
+ return DisMountDOSDriveFull(szPath,bForce);
+}