mountmgr: Store the contents of the device symlink in the drive object.

This commit is contained in:
Alexandre Julliard 2008-10-23 15:43:46 +02:00
parent 503972980e
commit 5f2d2c7df0
1 changed files with 25 additions and 5 deletions

View File

@ -67,6 +67,7 @@ struct dos_drive
STORAGE_DEVICE_NUMBER devnum; /* device number info */ STORAGE_DEVICE_NUMBER devnum; /* device number info */
struct mount_point *dosdev; /* DosDevices mount point */ struct mount_point *dosdev; /* DosDevices mount point */
struct mount_point *volume; /* Volume{xxx} mount point */ struct mount_point *volume; /* Volume{xxx} mount point */
char *unix_device; /* unix device path */
char *unix_mount; /* unix mount point path */ char *unix_mount; /* unix mount point path */
}; };
@ -88,6 +89,15 @@ static char *get_dosdevices_path( char **drive )
return path; return path;
} }
static char *strdupA( const char *str )
{
char *ret;
if (!str) return NULL;
if ((ret = RtlAllocateHeap( GetProcessHeap(), 0, strlen(str) + 1 ))) strcpy( ret, str );
return ret;
}
/* read a Unix symlink; returned buffer must be freed by caller */ /* read a Unix symlink; returned buffer must be freed by caller */
static char *read_symlink( const char *path ) static char *read_symlink( const char *path )
{ {
@ -186,7 +196,8 @@ static NTSTATUS create_disk_device( const char *udi, DWORD type, struct dos_driv
drive->type = type; drive->type = type;
drive->dosdev = NULL; drive->dosdev = NULL;
drive->volume = NULL; drive->volume = NULL;
drive->unix_mount = NULL; drive->unix_device = NULL;
drive->unix_mount = NULL;
drive->symlink.Buffer = NULL; drive->symlink.Buffer = NULL;
if (udi) if (udi)
{ {
@ -258,6 +269,7 @@ static void delete_disk_device( struct dos_drive *drive )
IoDeleteSymbolicLink( &drive->symlink ); IoDeleteSymbolicLink( &drive->symlink );
RtlFreeUnicodeString( &drive->symlink ); RtlFreeUnicodeString( &drive->symlink );
} }
RtlFreeHeap( GetProcessHeap(), 0, drive->unix_device );
RtlFreeHeap( GetProcessHeap(), 0, drive->unix_mount ); RtlFreeHeap( GetProcessHeap(), 0, drive->unix_mount );
RtlFreeHeap( GetProcessHeap(), 0, drive->udi ); RtlFreeHeap( GetProcessHeap(), 0, drive->udi );
RtlFreeUnicodeString( &drive->name ); RtlFreeUnicodeString( &drive->name );
@ -379,8 +391,7 @@ static BOOL set_unix_mount_point( struct dos_drive *drive, const char *mount_poi
modified = TRUE; modified = TRUE;
} }
RtlFreeHeap( GetProcessHeap(), 0, drive->unix_mount ); RtlFreeHeap( GetProcessHeap(), 0, drive->unix_mount );
if ((drive->unix_mount = RtlAllocateHeap( GetProcessHeap(), 0, strlen(mount_point) + 1 ))) drive->unix_mount = strdupA( mount_point );
strcpy( drive->unix_mount, mount_point );
if (drive->dosdev) set_mount_point_id( drive->dosdev, mount_point, strlen(mount_point) + 1 ); if (drive->dosdev) set_mount_point_id( drive->dosdev, mount_point, strlen(mount_point) + 1 );
if (drive->volume) set_mount_point_id( drive->volume, mount_point, strlen(mount_point) + 1 ); if (drive->volume) set_mount_point_id( drive->volume, mount_point, strlen(mount_point) + 1 );
} }
@ -400,7 +411,7 @@ static BOOL set_unix_mount_point( struct dos_drive *drive, const char *mount_poi
/* create devices for mapped drives */ /* create devices for mapped drives */
static void create_drive_devices(void) static void create_drive_devices(void)
{ {
char *path, *p, *link; char *path, *p, *link, *device;
struct dos_drive *drive; struct dos_drive *drive;
unsigned int i; unsigned int i;
HKEY drives_key; HKEY drives_key;
@ -415,6 +426,8 @@ static void create_drive_devices(void)
p[0] = 'a' + i; p[0] = 'a' + i;
p[2] = 0; p[2] = 0;
if (!(link = read_symlink( path ))) continue; if (!(link = read_symlink( path ))) continue;
p[2] = ':';
device = read_symlink( path );
drive_type = i < 2 ? DRIVE_REMOVABLE : DRIVE_FIXED; drive_type = i < 2 ? DRIVE_REMOVABLE : DRIVE_FIXED;
if (drives_key) if (drives_key)
@ -438,9 +451,14 @@ static void create_drive_devices(void)
if (!create_disk_device( NULL, drive_type, &drive )) if (!create_disk_device( NULL, drive_type, &drive ))
{ {
drive->unix_mount = link; drive->unix_mount = link;
drive->unix_device = device;
set_drive_letter( drive, i ); set_drive_letter( drive, i );
} }
else RtlFreeHeap( GetProcessHeap(), 0, link ); else
{
RtlFreeHeap( GetProcessHeap(), 0, link );
RtlFreeHeap( GetProcessHeap(), 0, device );
}
} }
RegCloseKey( drives_key ); RegCloseKey( drives_key );
RtlFreeHeap( GetProcessHeap(), 0, path ); RtlFreeHeap( GetProcessHeap(), 0, path );
@ -467,6 +485,8 @@ BOOL add_dos_device( const char *udi, const char *device, const char *mount_poin
if (create_disk_device( udi, type, &drive )) return FALSE; if (create_disk_device( udi, type, &drive )) return FALSE;
found: found:
RtlFreeHeap( GetProcessHeap(), 0, drive->unix_device );
drive->unix_device = strdupA( device );
set_drive_letter( drive, letter ); set_drive_letter( drive, letter );
set_unix_mount_point( drive, mount_point ); set_unix_mount_point( drive, mount_point );