/* * Mount manager service implementation * * Copyright 2008 Alexandre Julliard * * This library is free software; you can redistribute it and/or * modify it under the terms of the GNU Lesser General Public * License as published by the Free Software Foundation; either * version 2.1 of the License, or (at your option) any later version. * * This library is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU * Lesser General Public License for more details. * * You should have received a copy of the GNU Lesser General Public * License along with this library; if not, write to the Free Software * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA */ #include <stdarg.h> #include <unistd.h> #define NONAMELESSUNION #define NONAMELESSSTRUCT #include "mountmgr.h" #include "winreg.h" #include "wine/library.h" #include "wine/list.h" #include "wine/unicode.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(mountmgr); #define MIN_ID_LEN 4 struct mount_point { struct list entry; /* entry in mount points list */ DEVICE_OBJECT *device; /* disk device */ UNICODE_STRING name; /* device name */ UNICODE_STRING link; /* DOS device symlink */ void *id; /* device unique id */ unsigned int id_len; }; static struct list mount_points_list = LIST_INIT(mount_points_list); static HKEY mount_key; void set_mount_point_id( struct mount_point *mount, const void *id, unsigned int id_len ) { RtlFreeHeap( GetProcessHeap(), 0, mount->id ); mount->id_len = max( MIN_ID_LEN, id_len ); if ((mount->id = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY, mount->id_len ))) { memcpy( mount->id, id, id_len ); RegSetValueExW( mount_key, mount->link.Buffer, 0, REG_BINARY, mount->id, mount->id_len ); } else mount->id_len = 0; } static struct mount_point *add_mount_point( DEVICE_OBJECT *device, UNICODE_STRING *device_name, const WCHAR *link, const void *id, unsigned int id_len ) { struct mount_point *mount; WCHAR *str; UINT len = (strlenW(link) + 1) * sizeof(WCHAR) + device_name->Length + sizeof(WCHAR); if (!(mount = RtlAllocateHeap( GetProcessHeap(), 0, sizeof(*mount) + len ))) return NULL; str = (WCHAR *)(mount + 1); strcpyW( str, link ); RtlInitUnicodeString( &mount->link, str ); str += strlenW(str) + 1; memcpy( str, device_name->Buffer, device_name->Length ); str[device_name->Length / sizeof(WCHAR)] = 0; mount->name.Buffer = str; mount->name.Length = device_name->Length; mount->name.MaximumLength = device_name->Length + sizeof(WCHAR); mount->device = device; mount->id = NULL; list_add_tail( &mount_points_list, &mount->entry ); IoCreateSymbolicLink( &mount->link, device_name ); set_mount_point_id( mount, id, id_len ); TRACE( "created %s id %s for %s\n", debugstr_w(mount->link.Buffer), debugstr_a(mount->id), debugstr_w(mount->name.Buffer) ); return mount; } /* create the DosDevices mount point symlink for a new device */ struct mount_point *add_dosdev_mount_point( DEVICE_OBJECT *device, UNICODE_STRING *device_name, int drive, const void *id, unsigned int id_len ) { static const WCHAR driveW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','%','c',':',0}; WCHAR link[sizeof(driveW)]; sprintfW( link, driveW, 'A' + drive ); return add_mount_point( device, device_name, link, id, id_len ); } /* create the Volume mount point symlink for a new device */ struct mount_point *add_volume_mount_point( DEVICE_OBJECT *device, UNICODE_STRING *device_name, int drive, const void *id, unsigned int id_len ) { static const WCHAR volumeW[] = {'\\','?','?','\\','V','o','l','u','m','e','{', '%','0','8','x','-','%','0','4','x','-','%','0','4','x','-', '%','0','2','x','%','0','2','x','-','%','0','2','x','%','0','2','x', '%','0','2','x','%','0','2','x','%','0','2','x','%','0','2','x','}',0}; WCHAR link[sizeof(volumeW)]; GUID guid; memset( &guid, 0, sizeof(guid) ); /* FIXME */ guid.Data4[7] = 'A' + drive; sprintfW( link, volumeW, guid.Data1, guid.Data2, guid.Data3, guid.Data4[0], guid.Data4[1], guid.Data4[2], guid.Data4[3], guid.Data4[4], guid.Data4[5], guid.Data4[6], guid.Data4[7]); return add_mount_point( device, device_name, link, id, id_len ); } /* delete the mount point symlinks when a device goes away */ void delete_mount_point( struct mount_point *mount ) { TRACE( "deleting %s\n", debugstr_w(mount->link.Buffer) ); list_remove( &mount->entry ); RegDeleteValueW( mount_key, mount->link.Buffer ); IoDeleteSymbolicLink( &mount->link ); RtlFreeHeap( GetProcessHeap(), 0, mount->id ); RtlFreeHeap( GetProcessHeap(), 0, mount ); } /* check if a given mount point matches the requested specs */ static BOOL matching_mount_point( const struct mount_point *mount, const MOUNTMGR_MOUNT_POINT *spec ) { if (spec->SymbolicLinkNameOffset) { const WCHAR *name = (const WCHAR *)((const char *)spec + spec->SymbolicLinkNameOffset); if (spec->SymbolicLinkNameLength != mount->link.Length) return FALSE; if (memicmpW( name, mount->link.Buffer, mount->link.Length/sizeof(WCHAR))) return FALSE; } if (spec->DeviceNameOffset) { const WCHAR *name = (const WCHAR *)((const char *)spec + spec->DeviceNameOffset); if (spec->DeviceNameLength != mount->name.Length) return FALSE; if (memicmpW( name, mount->name.Buffer, mount->name.Length/sizeof(WCHAR))) return FALSE; } if (spec->UniqueIdOffset) { const void *id = ((const char *)spec + spec->UniqueIdOffset); if (spec->UniqueIdLength != mount->id_len) return FALSE; if (memcmp( id, mount->id, mount->id_len )) return FALSE; } return TRUE; } /* implementation of IOCTL_MOUNTMGR_QUERY_POINTS */ static NTSTATUS query_mount_points( const void *in_buff, SIZE_T insize, void *out_buff, SIZE_T outsize, IO_STATUS_BLOCK *iosb ) { UINT count, pos, size; const MOUNTMGR_MOUNT_POINT *input = in_buff; MOUNTMGR_MOUNT_POINTS *info = out_buff; struct mount_point *mount; /* sanity checks */ if (input->SymbolicLinkNameOffset + input->SymbolicLinkNameLength > insize || input->UniqueIdOffset + input->UniqueIdLength > insize || input->DeviceNameOffset + input->DeviceNameLength > insize || input->SymbolicLinkNameOffset + input->SymbolicLinkNameLength < input->SymbolicLinkNameOffset || input->UniqueIdOffset + input->UniqueIdLength < input->UniqueIdOffset || input->DeviceNameOffset + input->DeviceNameLength < input->DeviceNameOffset) return STATUS_INVALID_PARAMETER; count = size = 0; LIST_FOR_EACH_ENTRY( mount, &mount_points_list, struct mount_point, entry ) { if (!matching_mount_point( mount, input )) continue; size += mount->name.Length; size += mount->link.Length; size += mount->id_len; size = (size + sizeof(WCHAR) - 1) & ~(sizeof(WCHAR) - 1); count++; } pos = FIELD_OFFSET( MOUNTMGR_MOUNT_POINTS, MountPoints[count] ); size += pos; if (size > outsize) { if (size >= sizeof(info->Size)) info->Size = size; iosb->Information = sizeof(info->Size); return STATUS_MORE_ENTRIES; } info->NumberOfMountPoints = count; count = 0; LIST_FOR_EACH_ENTRY( mount, &mount_points_list, struct mount_point, entry ) { if (!matching_mount_point( mount, input )) continue; info->MountPoints[count].DeviceNameOffset = pos; info->MountPoints[count].DeviceNameLength = mount->name.Length; memcpy( (char *)out_buff + pos, mount->name.Buffer, mount->name.Length ); pos += mount->name.Length; info->MountPoints[count].SymbolicLinkNameOffset = pos; info->MountPoints[count].SymbolicLinkNameLength = mount->link.Length; memcpy( (char *)out_buff + pos, mount->link.Buffer, mount->link.Length ); pos += mount->link.Length; info->MountPoints[count].UniqueIdOffset = pos; info->MountPoints[count].UniqueIdLength = mount->id_len; memcpy( (char *)out_buff + pos, mount->id, mount->id_len ); pos += mount->id_len; pos = (pos + sizeof(WCHAR) - 1) & ~(sizeof(WCHAR) - 1); count++; } info->Size = pos; iosb->Information = pos; return STATUS_SUCCESS; } /* implementation of IOCTL_MOUNTMGR_DEFINE_UNIX_DRIVE */ static NTSTATUS define_unix_drive( const void *in_buff, SIZE_T insize ) { const struct mountmgr_unix_drive *input = in_buff; const char *mount_point = NULL, *device = NULL; unsigned int i; WCHAR letter = tolowerW( input->letter ); if (letter < 'a' || letter > 'z') return STATUS_INVALID_PARAMETER; if (input->type > DRIVE_RAMDISK) return STATUS_INVALID_PARAMETER; if (input->mount_point_offset > insize || input->device_offset > insize) return STATUS_INVALID_PARAMETER; /* make sure string are null-terminated */ if (input->mount_point_offset) { mount_point = (const char *)in_buff + input->mount_point_offset; for (i = input->mount_point_offset; i < insize; i++) if (!*((const char *)in_buff + i)) break; if (i >= insize) return STATUS_INVALID_PARAMETER; } if (input->device_offset) { device = (const char *)in_buff + input->device_offset; for (i = input->device_offset; i < insize; i++) if (!*((const char *)in_buff + i)) break; if (i >= insize) return STATUS_INVALID_PARAMETER; } if (input->type != DRIVE_NO_ROOT_DIR) { enum device_type type = DEVICE_UNKNOWN; TRACE( "defining %c: dev %s mount %s type %u\n", letter, debugstr_a(device), debugstr_a(mount_point), input->type ); switch (input->type) { case DRIVE_REMOVABLE: type = (letter >= 'c') ? DEVICE_HARDDISK : DEVICE_FLOPPY; break; case DRIVE_REMOTE: type = DEVICE_NETWORK; break; case DRIVE_CDROM: type = DEVICE_CDROM; break; case DRIVE_RAMDISK: type = DEVICE_RAMDISK; break; case DRIVE_FIXED: type = DEVICE_HARDDISK_VOL; break; } return add_dos_device( letter - 'a', NULL, device, mount_point, type ); } else { TRACE( "removing %c:\n", letter ); return remove_dos_device( letter - 'a', NULL ); } } /* implementation of IOCTL_MOUNTMGR_QUERY_UNIX_DRIVE */ static NTSTATUS query_unix_drive( const void *in_buff, SIZE_T insize, void *out_buff, SIZE_T outsize, IO_STATUS_BLOCK *iosb ) { const struct mountmgr_unix_drive *input = in_buff; struct mountmgr_unix_drive *output = out_buff; const char *device, *mount_point; int letter = tolowerW( input->letter ); NTSTATUS status; DWORD size, type = DEVICE_UNKNOWN; enum device_type device_type; char *ptr; if (letter < 'a' || letter > 'z') return STATUS_INVALID_PARAMETER; if ((status = query_dos_device( letter - 'a', &device_type, &device, &mount_point ))) return status; switch (device_type) { case DEVICE_UNKNOWN: type = DRIVE_UNKNOWN; break; case DEVICE_HARDDISK: type = DRIVE_REMOVABLE; break; case DEVICE_HARDDISK_VOL: type = DRIVE_FIXED; break; case DEVICE_FLOPPY: type = DRIVE_REMOVABLE; break; case DEVICE_CDROM: type = DRIVE_CDROM; break; case DEVICE_NETWORK: type = DRIVE_REMOTE; break; case DEVICE_RAMDISK: type = DRIVE_RAMDISK; break; } size = sizeof(*output); if (device) size += strlen(device) + 1; if (mount_point) size += strlen(mount_point) + 1; if (size > outsize) { iosb->Information = 0; if (size >= FIELD_OFFSET( struct mountmgr_unix_drive, size ) + sizeof(output->size)) { output->size = size; iosb->Information = FIELD_OFFSET( struct mountmgr_unix_drive, size ) + sizeof(output->size); } if (size >= FIELD_OFFSET( struct mountmgr_unix_drive, type ) + sizeof(output->type)) { output->type = type; iosb->Information = FIELD_OFFSET( struct mountmgr_unix_drive, type ) + sizeof(output->type); } return STATUS_MORE_ENTRIES; } output->size = size; output->letter = letter; output->type = type; ptr = (char *)(output + 1); if (mount_point) { output->mount_point_offset = ptr - (char *)output; strcpy( ptr, mount_point ); ptr += strlen(ptr) + 1; } else output->mount_point_offset = 0; if (device) { output->device_offset = ptr - (char *)output; strcpy( ptr, device ); ptr += strlen(ptr) + 1; } else output->device_offset = 0; TRACE( "returning %c: dev %s mount %s type %u\n", letter, debugstr_a(device), debugstr_a(mount_point), type ); iosb->Information = ptr - (char *)output; return STATUS_SUCCESS; } /* handler for ioctls on the mount manager device */ static NTSTATUS WINAPI mountmgr_ioctl( DEVICE_OBJECT *device, IRP *irp ) { IO_STACK_LOCATION *irpsp = irp->Tail.Overlay.s.u.CurrentStackLocation; TRACE( "ioctl %x insize %u outsize %u\n", irpsp->Parameters.DeviceIoControl.IoControlCode, irpsp->Parameters.DeviceIoControl.InputBufferLength, irpsp->Parameters.DeviceIoControl.OutputBufferLength ); switch(irpsp->Parameters.DeviceIoControl.IoControlCode) { case IOCTL_MOUNTMGR_QUERY_POINTS: if (irpsp->Parameters.DeviceIoControl.InputBufferLength < sizeof(MOUNTMGR_MOUNT_POINT)) return STATUS_INVALID_PARAMETER; irp->IoStatus.u.Status = query_mount_points( irpsp->Parameters.DeviceIoControl.Type3InputBuffer, irpsp->Parameters.DeviceIoControl.InputBufferLength, irp->MdlAddress->StartVa, irpsp->Parameters.DeviceIoControl.OutputBufferLength, &irp->IoStatus ); break; case IOCTL_MOUNTMGR_DEFINE_UNIX_DRIVE: if (irpsp->Parameters.DeviceIoControl.InputBufferLength < sizeof(struct mountmgr_unix_drive)) return STATUS_INVALID_PARAMETER; irp->IoStatus.Information = 0; irp->IoStatus.u.Status = define_unix_drive( irpsp->Parameters.DeviceIoControl.Type3InputBuffer, irpsp->Parameters.DeviceIoControl.InputBufferLength ); break; case IOCTL_MOUNTMGR_QUERY_UNIX_DRIVE: if (irpsp->Parameters.DeviceIoControl.InputBufferLength < sizeof(struct mountmgr_unix_drive)) return STATUS_INVALID_PARAMETER; irp->IoStatus.u.Status = query_unix_drive( irpsp->Parameters.DeviceIoControl.Type3InputBuffer, irpsp->Parameters.DeviceIoControl.InputBufferLength, irp->MdlAddress->StartVa, irpsp->Parameters.DeviceIoControl.OutputBufferLength, &irp->IoStatus ); break; default: FIXME( "ioctl %x not supported\n", irpsp->Parameters.DeviceIoControl.IoControlCode ); irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED; break; } return irp->IoStatus.u.Status; } /* main entry point for the mount point manager driver */ NTSTATUS WINAPI DriverEntry( DRIVER_OBJECT *driver, UNICODE_STRING *path ) { static const WCHAR mounted_devicesW[] = {'S','y','s','t','e','m','\\','M','o','u','n','t','e','d','D','e','v','i','c','e','s',0}; static const WCHAR device_mountmgrW[] = {'\\','D','e','v','i','c','e','\\','M','o','u','n','t','P','o','i','n','t','M','a','n','a','g','e','r',0}; static const WCHAR link_mountmgrW[] = {'\\','?','?','\\','M','o','u','n','t','P','o','i','n','t','M','a','n','a','g','e','r',0}; static const WCHAR harddiskW[] = {'\\','D','r','i','v','e','r','\\','H','a','r','d','d','i','s','k',0}; UNICODE_STRING nameW, linkW; DEVICE_OBJECT *device; NTSTATUS status; TRACE( "%s\n", debugstr_w(path->Buffer) ); driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = mountmgr_ioctl; RtlInitUnicodeString( &nameW, device_mountmgrW ); RtlInitUnicodeString( &linkW, link_mountmgrW ); if (!(status = IoCreateDevice( driver, 0, &nameW, 0, 0, FALSE, &device ))) status = IoCreateSymbolicLink( &linkW, &nameW ); if (status) { FIXME( "failed to create device error %x\n", status ); return status; } RegCreateKeyExW( HKEY_LOCAL_MACHINE, mounted_devicesW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &mount_key, NULL ); RtlInitUnicodeString( &nameW, harddiskW ); status = IoCreateDriver( &nameW, harddisk_driver_entry ); initialize_hal(); initialize_diskarbitration(); return status; }