/* * Dynamic devices support * * Copyright 2006 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 "config.h" #include "wine/port.h" #include #include #include #include #include #include #ifdef HAVE_SYS_IOCTL_H # include #endif #define NONAMELESSUNION #include "mountmgr.h" #include "winreg.h" #include "winuser.h" #include "dbt.h" #include "wine/library.h" #include "wine/list.h" #include "wine/unicode.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(mountmgr); #define MAX_DOS_DRIVES 26 #define MAX_PORTS 256 static const WCHAR drive_types[][8] = { { 0 }, /* DEVICE_UNKNOWN */ { 0 }, /* DEVICE_HARDDISK */ {'h','d',0}, /* DEVICE_HARDDISK_VOL */ {'f','l','o','p','p','y',0}, /* DEVICE_FLOPPY */ {'c','d','r','o','m',0}, /* DEVICE_CDROM */ {'c','d','r','o','m',0}, /* DEVICE_DVD */ {'n','e','t','w','o','r','k',0}, /* DEVICE_NETWORK */ {'r','a','m','d','i','s','k',0} /* DEVICE_RAMDISK */ }; static const WCHAR drives_keyW[] = {'S','o','f','t','w','a','r','e','\\', 'W','i','n','e','\\','D','r','i','v','e','s',0}; static const WCHAR ports_keyW[] = {'S','o','f','t','w','a','r','e','\\', 'W','i','n','e','\\','P','o','r','t','s',0}; static const WCHAR scsi_keyW[] = {'H','A','R','D','W','A','R','E','\\','D','E','V','I','C','E','M','A','P','\\','S','c','s','i',0}; static const WCHAR scsi_port_keyW[] = {'S','c','s','i',' ','P','o','r','t',' ','%','d',0}; static const WCHAR scsi_bus_keyW[] = {'S','c','s','i',' ','B','u','s',' ','%','d',0}; static const WCHAR target_id_keyW[] = {'T','a','r','g','e','t',' ','I','d',' ','%','d',0}; static const WCHAR lun_keyW[] = {'L','o','g','i','c','a','l',' ','U','n','i','t',' ','I','d',' ','%','d',0}; static const WCHAR devnameW[] = {'D','e','v','i','c','e','N','a','m','e',0}; enum fs_type { FS_ERROR, /* error accessing the device */ FS_UNKNOWN, /* unknown file system */ FS_FAT1216, FS_FAT32, FS_ISO9660, FS_UDF /* For reference [E] = Ecma-167.pdf, [U] = udf260.pdf */ }; struct disk_device { enum device_type type; /* drive type */ DEVICE_OBJECT *dev_obj; /* disk device allocated for this volume */ UNICODE_STRING name; /* device name */ UNICODE_STRING symlink; /* device symlink if any */ STORAGE_DEVICE_NUMBER devnum; /* device number info */ char *unix_device; /* unix device path */ char *unix_mount; /* unix mount point path */ }; struct volume { struct list entry; /* entry in volumes list */ struct disk_device *device; /* disk device */ char *udi; /* unique identifier for dynamic volumes */ unsigned int ref; /* ref count */ GUID guid; /* volume uuid */ struct mount_point *mount; /* Volume{xxx} mount point */ WCHAR label[256]; /* volume label */ DWORD serial; /* volume serial number */ enum fs_type fs_type; /* file system type */ }; struct dos_drive { struct list entry; /* entry in drives list */ struct volume *volume; /* volume for this drive */ int drive; /* drive letter (0 = A: etc.) */ struct mount_point *mount; /* DosDevices mount point */ }; static struct list drives_list = LIST_INIT(drives_list); static struct list volumes_list = LIST_INIT(volumes_list); static DRIVER_OBJECT *harddisk_driver; static DRIVER_OBJECT *serial_driver; static DRIVER_OBJECT *parallel_driver; static CRITICAL_SECTION device_section; static CRITICAL_SECTION_DEBUG critsect_debug = { 0, 0, &device_section, { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList }, 0, 0, { (DWORD_PTR)(__FILE__ ": device_section") } }; static CRITICAL_SECTION device_section = { &critsect_debug, -1, 0, 0, 0, 0 }; static char *get_dosdevices_path( char **device ) { const char *config_dir = wine_get_config_dir(); size_t len = strlen(config_dir) + sizeof("/dosdevices/com256"); char *path = HeapAlloc( GetProcessHeap(), 0, len ); if (path) { strcpy( path, config_dir ); strcat( path, "/dosdevices/a::" ); *device = path + len - sizeof("com256"); } 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; } static WCHAR *strdupW( const WCHAR *str ) { WCHAR *ret; if (!str) return NULL; if ((ret = RtlAllocateHeap( GetProcessHeap(), 0, (strlenW(str) + 1) * sizeof(WCHAR) ))) strcpyW( ret, str ); return ret; } static const GUID *get_default_uuid( int letter ) { static GUID guid; guid.Data4[7] = 'A' + letter; return &guid; } /* read a Unix symlink; returned buffer must be freed by caller */ static char *read_symlink( const char *path ) { char *buffer; int ret, size = 128; for (;;) { if (!(buffer = RtlAllocateHeap( GetProcessHeap(), 0, size ))) { SetLastError( ERROR_NOT_ENOUGH_MEMORY ); return 0; } ret = readlink( path, buffer, size ); if (ret == -1) { RtlFreeHeap( GetProcessHeap(), 0, buffer ); return 0; } if (ret != size) { buffer[ret] = 0; return buffer; } RtlFreeHeap( GetProcessHeap(), 0, buffer ); size *= 2; } } /* update a symlink if it changed; return TRUE if updated */ static void update_symlink( const char *path, const char *dest, const char *orig_dest ) { if (dest && dest[0]) { if (!orig_dest || strcmp( orig_dest, dest )) { unlink( path ); symlink( dest, path ); } } else unlink( path ); } /* send notification about a change to a given drive */ static void send_notify( int drive, int code ) { DEV_BROADCAST_VOLUME info; info.dbcv_size = sizeof(info); info.dbcv_devicetype = DBT_DEVTYP_VOLUME; info.dbcv_reserved = 0; info.dbcv_unitmask = 1 << drive; info.dbcv_flags = DBTF_MEDIA; BroadcastSystemMessageW( BSF_FORCEIFHUNG|BSF_QUERY, NULL, WM_DEVICECHANGE, code, (LPARAM)&info ); } #define BLOCK_SIZE 2048 #define SUPERBLOCK_SIZE BLOCK_SIZE #define CDFRAMES_PERSEC 75 #define CDFRAMES_PERMIN (CDFRAMES_PERSEC * 60) #define FRAME_OF_ADDR(a) ((a)[1] * CDFRAMES_PERMIN + (a)[2] * CDFRAMES_PERSEC + (a)[3]) #define FRAME_OF_TOC(toc, idx) FRAME_OF_ADDR((toc)->TrackData[(idx) - (toc)->FirstTrack].Address) #define GETWORD(buf,off) MAKEWORD(buf[(off)],buf[(off+1)]) #define GETLONG(buf,off) MAKELONG(GETWORD(buf,off),GETWORD(buf,off+2)) static int open_volume_file( const struct volume *volume, const char *file ) { const char *unix_mount = volume->device->unix_mount; char *path; int fd; if (!unix_mount) return -1; if (unix_mount[0] == '/') { if (!(path = HeapAlloc( GetProcessHeap(), 0, strlen( unix_mount ) + 1 + strlen( file ) + 1 ))) return -1; strcpy( path, unix_mount ); } else { const char *config_dir = wine_get_config_dir(); if (!(path = HeapAlloc( GetProcessHeap(), 0, strlen( config_dir ) + strlen("/dosdevices/") + strlen(unix_mount) + 1 + strlen( file ) + 1 ))) return -1; strcpy( path, config_dir ); strcat( path, "/dosdevices/" ); strcat( path, unix_mount ); } strcat( path, "/" ); strcat( path, file ); fd = open( path, O_RDONLY ); HeapFree( GetProcessHeap(), 0, path ); return fd; } /* get the label by reading it from a file at the root of the filesystem */ static void get_filesystem_label( struct volume *volume ) { int fd; ssize_t size; char buffer[256], *p; volume->label[0] = 0; if ((fd = open_volume_file( volume, ".windows-label" )) == -1) return; size = read( fd, buffer, sizeof(buffer) ); close( fd ); p = buffer + size; while (p > buffer && (p[-1] == ' ' || p[-1] == '\r' || p[-1] == '\n')) p--; *p = 0; if (!MultiByteToWideChar( CP_UNIXCP, 0, buffer, -1, volume->label, ARRAY_SIZE(volume->label) )) volume->label[ARRAY_SIZE(volume->label) - 1] = 0; } /* get the serial number by reading it from a file at the root of the filesystem */ static void get_filesystem_serial( struct volume *volume ) { int fd; ssize_t size; char buffer[32]; volume->serial = 0; if ((fd = open_volume_file( volume, ".windows-serial" )) == -1) return; size = read( fd, buffer, sizeof(buffer) ); close( fd ); if (size < 0) return; buffer[size] = 0; volume->serial = strtoul( buffer, NULL, 16 ); } /****************************************************************** * VOLUME_FindCdRomDataBestVoldesc */ static DWORD VOLUME_FindCdRomDataBestVoldesc( HANDLE handle ) { BYTE cur_vd_type, max_vd_type = 0; BYTE buffer[0x800]; DWORD size, offs, best_offs = 0, extra_offs = 0; for (offs = 0x8000; offs <= 0x9800; offs += 0x800) { /* if 'CDROM' occurs at position 8, this is a pre-iso9660 cd, and * the volume label is displaced forward by 8 */ if (SetFilePointer( handle, offs, NULL, FILE_BEGIN ) != offs) break; if (!ReadFile( handle, buffer, sizeof(buffer), &size, NULL )) break; if (size != sizeof(buffer)) break; /* check for non-ISO9660 signature */ if (!memcmp( buffer + 11, "ROM", 3 )) extra_offs = 8; cur_vd_type = buffer[extra_offs]; if (cur_vd_type == 0xff) /* voldesc set terminator */ break; if (cur_vd_type > max_vd_type) { max_vd_type = cur_vd_type; best_offs = offs + extra_offs; } } return best_offs; } /*********************************************************************** * VOLUME_ReadFATSuperblock */ static enum fs_type VOLUME_ReadFATSuperblock( HANDLE handle, BYTE *buff ) { DWORD size; /* try a fixed disk, with a FAT partition */ if (SetFilePointer( handle, 0, NULL, FILE_BEGIN ) != 0 || !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL )) { if (GetLastError() == ERROR_BAD_DEV_TYPE) return FS_UNKNOWN; /* not a real device */ return FS_ERROR; } if (size < SUPERBLOCK_SIZE) return FS_UNKNOWN; /* FIXME: do really all FAT have their name beginning with * "FAT" ? (At least FAT12, FAT16 and FAT32 have :) */ if (!memcmp(buff+0x36, "FAT", 3) || !memcmp(buff+0x52, "FAT", 3)) { /* guess which type of FAT we have */ int reasonable; unsigned int sectors, sect_per_fat, total_sectors, num_boot_sectors, num_fats, num_root_dir_ents, bytes_per_sector, sectors_per_cluster, nclust; sect_per_fat = GETWORD(buff, 0x16); if (!sect_per_fat) sect_per_fat = GETLONG(buff, 0x24); total_sectors = GETWORD(buff, 0x13); if (!total_sectors) total_sectors = GETLONG(buff, 0x20); num_boot_sectors = GETWORD(buff, 0x0e); num_fats = buff[0x10]; num_root_dir_ents = GETWORD(buff, 0x11); bytes_per_sector = GETWORD(buff, 0x0b); sectors_per_cluster = buff[0x0d]; /* check if the parameters are reasonable and will not cause * arithmetic errors in the calculation */ reasonable = num_boot_sectors < total_sectors && num_fats < 16 && bytes_per_sector >= 512 && bytes_per_sector % 512 == 0 && sectors_per_cluster >= 1; if (!reasonable) return FS_UNKNOWN; sectors = total_sectors - num_boot_sectors - num_fats * sect_per_fat - (num_root_dir_ents * 32 + bytes_per_sector - 1) / bytes_per_sector; nclust = sectors / sectors_per_cluster; if ((buff[0x42] == 0x28 || buff[0x42] == 0x29) && !memcmp(buff+0x52, "FAT", 3)) return FS_FAT32; if (nclust < 65525) { if ((buff[0x26] == 0x28 || buff[0x26] == 0x29) && !memcmp(buff+0x36, "FAT", 3)) return FS_FAT1216; } } return FS_UNKNOWN; } /*********************************************************************** * VOLUME_ReadCDBlock */ static BOOL VOLUME_ReadCDBlock( HANDLE handle, BYTE *buff, INT offs ) { DWORD size, whence = offs >= 0 ? FILE_BEGIN : FILE_END; if (SetFilePointer( handle, offs, NULL, whence ) != offs || !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) || size != SUPERBLOCK_SIZE) return FALSE; return TRUE; } /*********************************************************************** * VOLUME_ReadCDSuperblock */ static enum fs_type VOLUME_ReadCDSuperblock( HANDLE handle, BYTE *buff ) { int i; DWORD offs; /* Check UDF first as UDF and ISO9660 structures can coexist on the same medium * Starting from sector 16, we may find : * - a CD-ROM Volume Descriptor Set (ISO9660) containing one or more Volume Descriptors * - an Extended Area (UDF) -- [E] 2/8.3.1 and [U] 2.1.7 * There is no explicit end so read 16 sectors and then give up */ for( i=16; i<16+16; i++) { if (!VOLUME_ReadCDBlock(handle, buff, i*BLOCK_SIZE)) continue; /* We are supposed to check "BEA01", "NSR0x" and "TEA01" IDs + verify tag checksum * but we assume the volume is well-formatted */ if (!memcmp(&buff[1], "BEA01", 5)) return FS_UDF; } offs = VOLUME_FindCdRomDataBestVoldesc( handle ); if (!offs) return FS_UNKNOWN; if (!VOLUME_ReadCDBlock(handle, buff, offs)) return FS_ERROR; /* check for the iso9660 identifier */ if (!memcmp(&buff[1], "CD001", 5)) return FS_ISO9660; return FS_UNKNOWN; } /************************************************************************** * UDF_Find_PVD * Find the Primary Volume Descriptor */ static BOOL UDF_Find_PVD( HANDLE handle, BYTE pvd[] ) { unsigned int i; DWORD offset; INT locations[] = { 256, -1, -257, 512 }; for(i=0; ifs_type) { case FS_ERROR: label_len = 0; break; case FS_UNKNOWN: get_filesystem_label( volume ); return; case FS_FAT1216: label_ptr = superblock + 0x2b; label_len = 11; break; case FS_FAT32: label_ptr = superblock + 0x47; label_len = 11; break; case FS_ISO9660: { BYTE ver = superblock[0x5a]; if (superblock[0x58] == 0x25 && superblock[0x59] == 0x2f && /* Unicode ID */ ((ver == 0x40) || (ver == 0x43) || (ver == 0x45))) { /* yippee, unicode */ unsigned int i; for (i = 0; i < 16; i++) volume->label[i] = (superblock[40+2*i] << 8) | superblock[41+2*i]; volume->label[i] = 0; while (i && volume->label[i-1] == ' ') volume->label[--i] = 0; return; } label_ptr = superblock + 40; label_len = 32; break; } case FS_UDF: { BYTE pvd[BLOCK_SIZE]; if(!UDF_Find_PVD(handle, pvd)) { label_len = 0; break; } /* [E] 3/10.1.4 and [U] 2.1.1 */ if(pvd[24]==8) { label_ptr = pvd + 24 + 1; label_len = pvd[24+32-1]; break; } else { unsigned int i; label_len = 1 + pvd[24+32-1]; for (i = 0; i < label_len; i += 2) volume->label[i/2] = (pvd[24+1+i] << 8) | pvd[24+1+i+1]; volume->label[label_len] = 0; return; } } } if (label_len) RtlMultiByteToUnicodeN( volume->label, sizeof(volume->label) - sizeof(WCHAR), &label_len, (const char *)label_ptr, label_len ); label_len /= sizeof(WCHAR); volume->label[label_len] = 0; while (label_len && volume->label[label_len-1] == ' ') volume->label[--label_len] = 0; } /************************************************************************** * UDF_Find_FSD_Sector * Find the File Set Descriptor used to compute the serial of a UDF volume */ static int UDF_Find_FSD_Sector( HANDLE handle, BYTE block[] ) { int i, PVD_sector, PD_sector, PD_length; if(!UDF_Find_PVD(handle,block)) goto default_sector; /* Retrieve the tag location of the PVD -- [E] 3/7.2 */ PVD_sector = block[12 + 0]; PVD_sector |= block[12 + 1] << 8; PVD_sector |= block[12 + 2] << 16; PVD_sector |= block[12 + 3] << 24; /* Find the Partition Descriptor */ for(i=PVD_sector+1; ; i++) { if(!VOLUME_ReadCDBlock(handle, block, i*BLOCK_SIZE)) goto default_sector; /* Partition Descriptor Tag Id -- [E] 3/10.5.1 */ if(block[0]==5 && block[1]==0) break; /* Terminating Descriptor Tag Id -- [E] 3/10.9.1 */ if(block[0]==8 && block[1]==0) goto default_sector; } /* Find the partition starting location -- [E] 3/10.5.8 */ PD_sector = block[188 + 0]; PD_sector |= block[188 + 1] << 8; PD_sector |= block[188 + 2] << 16; PD_sector |= block[188 + 3] << 24; /* Find the partition length -- [E] 3/10.5.9 */ PD_length = block[192 + 0]; PD_length |= block[192 + 1] << 8; PD_length |= block[192 + 2] << 16; PD_length |= block[192 + 3] << 24; for(i=PD_sector; ifs_type) { case FS_ERROR: break; case FS_UNKNOWN: get_filesystem_serial( volume ); break; case FS_FAT1216: volume->serial = GETLONG( superblock, 0x27 ); break; case FS_FAT32: volume->serial = GETLONG( superblock, 0x43 ); break; case FS_UDF: FSD_sector = UDF_Find_FSD_Sector(handle, block); if (!VOLUME_ReadCDBlock(handle, block, FSD_sector*BLOCK_SIZE)) break; superblock = block; /* fallthrough */ case FS_ISO9660: { BYTE sum[4]; int i; sum[0] = sum[1] = sum[2] = sum[3] = 0; for (i = 0; i < 2048; i += 4) { /* DON'T optimize this into DWORD !! (breaks overflow) */ sum[0] += superblock[i+0]; sum[1] += superblock[i+1]; sum[2] += superblock[i+2]; sum[3] += superblock[i+3]; } /* * OK, another braindead one... argh. Just believe it. * Me$$ysoft chose to reverse the serial number in NT4/W2K. * It's true and nobody will ever be able to change it. */ if ((GetVersion() & 0x80000000) || volume->fs_type == FS_UDF) volume->serial = (sum[3] << 24) | (sum[2] << 16) | (sum[1] << 8) | sum[0]; else volume->serial = (sum[0] << 24) | (sum[1] << 16) | (sum[2] << 8) | sum[3]; } } } /************************************************************************** * VOLUME_GetAudioCDSerial */ static DWORD VOLUME_GetAudioCDSerial( const CDROM_TOC *toc ) { DWORD serial = 0; int i; for (i = 0; i <= toc->LastTrack - toc->FirstTrack; i++) serial += ((toc->TrackData[i].Address[1] << 16) | (toc->TrackData[i].Address[2] << 8) | toc->TrackData[i].Address[3]); /* * dwStart, dwEnd collect the beginning and end of the disc respectively, in * frames. * There it is collected for correcting the serial when there are less than * 3 tracks. */ if (toc->LastTrack - toc->FirstTrack + 1 < 3) { DWORD dwStart = FRAME_OF_TOC(toc, toc->FirstTrack); DWORD dwEnd = FRAME_OF_TOC(toc, toc->LastTrack + 1); serial += dwEnd - dwStart; } return serial; } /* create the disk device for a given volume */ static NTSTATUS create_disk_device( enum device_type type, struct disk_device **device_ret ) { static const WCHAR harddiskvolW[] = {'\\','D','e','v','i','c','e', '\\','H','a','r','d','d','i','s','k','V','o','l','u','m','e','%','u',0}; static const WCHAR harddiskW[] = {'\\','D','e','v','i','c','e','\\','H','a','r','d','d','i','s','k','%','u',0}; static const WCHAR cdromW[] = {'\\','D','e','v','i','c','e','\\','C','d','R','o','m','%','u',0}; static const WCHAR floppyW[] = {'\\','D','e','v','i','c','e','\\','F','l','o','p','p','y','%','u',0}; static const WCHAR ramdiskW[] = {'\\','D','e','v','i','c','e','\\','R','a','m','d','i','s','k','%','u',0}; static const WCHAR cdromlinkW[] = {'\\','?','?','\\','C','d','R','o','m','%','u',0}; static const WCHAR physdriveW[] = {'\\','?','?','\\','P','h','y','s','i','c','a','l','D','r','i','v','e','%','u',0}; UINT i, first = 0; NTSTATUS status = 0; const WCHAR *format = NULL; const WCHAR *link_format = NULL; UNICODE_STRING name; DEVICE_OBJECT *dev_obj; struct disk_device *device; switch(type) { case DEVICE_UNKNOWN: case DEVICE_HARDDISK: case DEVICE_NETWORK: /* FIXME */ format = harddiskW; link_format = physdriveW; break; case DEVICE_HARDDISK_VOL: format = harddiskvolW; first = 1; /* harddisk volumes start counting from 1 */ break; case DEVICE_FLOPPY: format = floppyW; break; case DEVICE_CDROM: case DEVICE_DVD: format = cdromW; link_format = cdromlinkW; break; case DEVICE_RAMDISK: format = ramdiskW; break; } name.MaximumLength = (strlenW(format) + 10) * sizeof(WCHAR); name.Buffer = RtlAllocateHeap( GetProcessHeap(), 0, name.MaximumLength ); for (i = first; i < 32; i++) { sprintfW( name.Buffer, format, i ); name.Length = strlenW(name.Buffer) * sizeof(WCHAR); status = IoCreateDevice( harddisk_driver, sizeof(*device), &name, 0, 0, FALSE, &dev_obj ); if (status != STATUS_OBJECT_NAME_COLLISION) break; } if (!status) { device = dev_obj->DeviceExtension; device->dev_obj = dev_obj; device->name = name; device->type = type; device->unix_device = NULL; device->unix_mount = NULL; device->symlink.Buffer = NULL; if (link_format) { UNICODE_STRING symlink; symlink.MaximumLength = (strlenW(link_format) + 10) * sizeof(WCHAR); if ((symlink.Buffer = RtlAllocateHeap( GetProcessHeap(), 0, symlink.MaximumLength))) { sprintfW( symlink.Buffer, link_format, i ); symlink.Length = strlenW(symlink.Buffer) * sizeof(WCHAR); if (!IoCreateSymbolicLink( &symlink, &name )) device->symlink = symlink; } } switch (type) { case DEVICE_FLOPPY: case DEVICE_RAMDISK: device->devnum.DeviceType = FILE_DEVICE_DISK; device->devnum.DeviceNumber = i; device->devnum.PartitionNumber = ~0u; break; case DEVICE_CDROM: device->devnum.DeviceType = FILE_DEVICE_CD_ROM; device->devnum.DeviceNumber = i; device->devnum.PartitionNumber = ~0u; break; case DEVICE_DVD: device->devnum.DeviceType = FILE_DEVICE_DVD; device->devnum.DeviceNumber = i; device->devnum.PartitionNumber = ~0u; break; case DEVICE_UNKNOWN: case DEVICE_HARDDISK: case DEVICE_NETWORK: /* FIXME */ device->devnum.DeviceType = FILE_DEVICE_DISK; device->devnum.DeviceNumber = i; device->devnum.PartitionNumber = 0; break; case DEVICE_HARDDISK_VOL: device->devnum.DeviceType = FILE_DEVICE_DISK; device->devnum.DeviceNumber = 0; device->devnum.PartitionNumber = i; break; } *device_ret = device; TRACE( "created device %s\n", debugstr_w(name.Buffer) ); } else { FIXME( "IoCreateDevice %s got %x\n", debugstr_w(name.Buffer), status ); RtlFreeUnicodeString( &name ); } return status; } /* delete the disk device for a given drive */ static void delete_disk_device( struct disk_device *device ) { TRACE( "deleting device %s\n", debugstr_w(device->name.Buffer) ); if (device->symlink.Buffer) { IoDeleteSymbolicLink( &device->symlink ); RtlFreeUnicodeString( &device->symlink ); } RtlFreeHeap( GetProcessHeap(), 0, device->unix_device ); RtlFreeHeap( GetProcessHeap(), 0, device->unix_mount ); RtlFreeUnicodeString( &device->name ); IoDeleteDevice( device->dev_obj ); } /* grab another reference to a volume */ static struct volume *grab_volume( struct volume *volume ) { volume->ref++; return volume; } /* release a volume and delete the corresponding disk device when refcount is 0 */ static unsigned int release_volume( struct volume *volume ) { unsigned int ret = --volume->ref; if (!ret) { TRACE( "%s udi %s\n", debugstr_guid(&volume->guid), debugstr_a(volume->udi) ); assert( !volume->udi ); list_remove( &volume->entry ); if (volume->mount) delete_mount_point( volume->mount ); delete_disk_device( volume->device ); RtlFreeHeap( GetProcessHeap(), 0, volume ); } return ret; } /* set the volume udi */ static void set_volume_udi( struct volume *volume, const char *udi ) { if (udi) { assert( !volume->udi ); /* having a udi means the HAL side holds an extra reference */ if ((volume->udi = strdupA( udi ))) grab_volume( volume ); } else if (volume->udi) { RtlFreeHeap( GetProcessHeap(), 0, volume->udi ); volume->udi = NULL; release_volume( volume ); } } /* create a disk volume */ static NTSTATUS create_volume( const char *udi, enum device_type type, struct volume **volume_ret ) { struct volume *volume; NTSTATUS status; if (!(volume = RtlAllocateHeap( GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*volume) ))) return STATUS_NO_MEMORY; if (!(status = create_disk_device( type, &volume->device ))) { if (udi) set_volume_udi( volume, udi ); list_add_tail( &volumes_list, &volume->entry ); *volume_ret = grab_volume( volume ); } else RtlFreeHeap( GetProcessHeap(), 0, volume ); return status; } /* create the disk device for a given volume */ static NTSTATUS create_dos_device( struct volume *volume, const char *udi, int letter, enum device_type type, struct dos_drive **drive_ret ) { struct dos_drive *drive; NTSTATUS status; if (!(drive = RtlAllocateHeap( GetProcessHeap(), 0, sizeof(*drive) ))) return STATUS_NO_MEMORY; drive->drive = letter; drive->mount = NULL; if (volume) { if (udi) set_volume_udi( volume, udi ); drive->volume = grab_volume( volume ); status = STATUS_SUCCESS; } else status = create_volume( udi, type, &drive->volume ); if (status == STATUS_SUCCESS) { list_add_tail( &drives_list, &drive->entry ); *drive_ret = drive; } else RtlFreeHeap( GetProcessHeap(), 0, drive ); return status; } /* delete the disk device for a given drive */ static void delete_dos_device( struct dos_drive *drive ) { list_remove( &drive->entry ); if (drive->mount) delete_mount_point( drive->mount ); release_volume( drive->volume ); RtlFreeHeap( GetProcessHeap(), 0, drive ); } /* find a volume that matches the parameters */ static struct volume *find_matching_volume( const char *udi, const char *device, const char *mount_point, enum device_type type ) { struct volume *volume; struct disk_device *disk_device; LIST_FOR_EACH_ENTRY( volume, &volumes_list, struct volume, entry ) { int match = 0; /* when we have a udi we only match drives added manually */ if (udi && volume->udi) continue; /* and when we don't have a udi we only match dynamic drives */ if (!udi && !volume->udi) continue; disk_device = volume->device; if (disk_device->type != type) continue; if (device && disk_device->unix_device) { if (strcmp( device, disk_device->unix_device )) continue; match++; } if (mount_point && disk_device->unix_mount) { if (strcmp( mount_point, disk_device->unix_mount )) continue; match++; } if (!match) continue; TRACE( "found matching volume %s for device %s mount %s type %u\n", debugstr_guid(&volume->guid), debugstr_a(device), debugstr_a(mount_point), type ); return grab_volume( volume ); } return NULL; } static BOOL get_volume_device_info( struct volume *volume ) { const char *unix_device = volume->device->unix_device; ANSI_STRING unix_name; UNICODE_STRING nt_name; OBJECT_ATTRIBUTES attr; HANDLE handle; NTSTATUS ret; CDROM_TOC toc; DWORD size; BYTE superblock[SUPERBLOCK_SIZE]; IO_STATUS_BLOCK io; if (!unix_device) return FALSE; RtlInitAnsiString( &unix_name, unix_device ); if ((ret = wine_unix_to_nt_file_name( &unix_name, &nt_name ))) { ERR("Failed to convert %s to NT, status %#x\n", debugstr_a(unix_device), ret); return FALSE; } InitializeObjectAttributes( &attr, &nt_name, OBJ_CASE_INSENSITIVE, 0, NULL ); if ((ret = NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ | FILE_SHARE_WRITE, FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ))) { WARN("Failed to open %s, status %#x\n", debugstr_a(unix_device), ret); RtlFreeUnicodeString( &nt_name ); return FALSE; } if (DeviceIoControl( handle, IOCTL_CDROM_READ_TOC, NULL, 0, &toc, sizeof(toc), &size, 0 )) { if (!(toc.TrackData[0].Control & 0x04)) /* audio track */ { static const WCHAR audiocdW[] = {'A','u','d','i','o',' ','C','D',0}; TRACE( "%s: found audio CD\n", debugstr_a(unix_device) ); lstrcpynW( volume->label, audiocdW, ARRAY_SIZE(volume->label) ); volume->serial = VOLUME_GetAudioCDSerial( &toc ); volume->fs_type = FS_ISO9660; CloseHandle( handle ); return TRUE; } volume->fs_type = VOLUME_ReadCDSuperblock( handle, superblock ); } else { volume->fs_type = VOLUME_ReadFATSuperblock( handle, superblock ); if (volume->fs_type == FS_UNKNOWN) volume->fs_type = VOLUME_ReadCDSuperblock( handle, superblock ); } TRACE( "%s: found fs type %d\n", debugstr_a(unix_device), volume->fs_type ); if (volume->fs_type == FS_ERROR) { CloseHandle( handle ); return FALSE; } VOLUME_GetSuperblockLabel( volume, handle, superblock ); VOLUME_GetSuperblockSerial( volume, handle, superblock ); CloseHandle( handle ); return TRUE; } /* change the information for an existing volume */ static NTSTATUS set_volume_info( struct volume *volume, struct dos_drive *drive, const char *device, const char *mount_point, enum device_type type, const GUID *guid ) { void *id = NULL; unsigned int id_len = 0; struct disk_device *disk_device = volume->device; NTSTATUS status; if (type != disk_device->type) { if ((status = create_disk_device( type, &disk_device ))) return status; if (volume->mount) { delete_mount_point( volume->mount ); volume->mount = NULL; } if (drive && drive->mount) { delete_mount_point( drive->mount ); drive->mount = NULL; } delete_disk_device( volume->device ); volume->device = disk_device; } else { RtlFreeHeap( GetProcessHeap(), 0, disk_device->unix_device ); RtlFreeHeap( GetProcessHeap(), 0, disk_device->unix_mount ); } disk_device->unix_device = strdupA( device ); disk_device->unix_mount = strdupA( mount_point ); if (!get_volume_device_info( volume )) { if (volume->device->type == DEVICE_CDROM) volume->fs_type = FS_ISO9660; else if (volume->device->type == DEVICE_DVD) volume->fs_type = FS_UDF; else volume->fs_type = FS_UNKNOWN; get_filesystem_label( volume ); get_filesystem_serial( volume ); } TRACE("fs_type %#x, label %s, serial %08x\n", volume->fs_type, debugstr_w(volume->label), volume->serial); if (guid && memcmp( &volume->guid, guid, sizeof(volume->guid) )) { volume->guid = *guid; if (volume->mount) { delete_mount_point( volume->mount ); volume->mount = NULL; } } if (!volume->serial) memcpy(&volume->serial, &volume->guid.Data4[4], sizeof(DWORD)); if (!volume->mount) volume->mount = add_volume_mount_point( disk_device->dev_obj, &disk_device->name, &volume->guid ); if (drive && !drive->mount) drive->mount = add_dosdev_mount_point( disk_device->dev_obj, &disk_device->name, drive->drive ); if (disk_device->unix_mount) { id = disk_device->unix_mount; id_len = strlen( disk_device->unix_mount ) + 1; } if (volume->mount) set_mount_point_id( volume->mount, id, id_len ); if (drive && drive->mount) set_mount_point_id( drive->mount, id, id_len ); return STATUS_SUCCESS; } /* change the drive letter or volume for an existing drive */ static void set_drive_info( struct dos_drive *drive, int letter, struct volume *volume ) { if (drive->drive != letter) { if (drive->mount) delete_mount_point( drive->mount ); drive->mount = NULL; drive->drive = letter; } if (drive->volume != volume) { if (drive->mount) delete_mount_point( drive->mount ); drive->mount = NULL; grab_volume( volume ); release_volume( drive->volume ); drive->volume = volume; } } static inline BOOL is_valid_device( struct stat *st ) { #if defined(linux) || defined(__sun__) return S_ISBLK( st->st_mode ); #else /* disks are char devices on *BSD */ return S_ISCHR( st->st_mode ); #endif } /* find or create a DOS drive for the corresponding device */ static int add_drive( const char *device, enum device_type type ) { char *path, *p; char in_use[26]; struct stat dev_st, drive_st; int drive, first, last, avail = 0; if (stat( device, &dev_st ) == -1 || !is_valid_device( &dev_st )) return -1; if (!(path = get_dosdevices_path( &p ))) return -1; memset( in_use, 0, sizeof(in_use) ); switch (type) { case DEVICE_FLOPPY: first = 0; last = 2; break; case DEVICE_CDROM: case DEVICE_DVD: first = 3; last = 26; break; default: first = 2; last = 26; break; } while (avail != -1) { avail = -1; for (drive = first; drive < last; drive++) { if (in_use[drive]) continue; /* already checked */ *p = 'a' + drive; if (stat( path, &drive_st ) == -1) { if (lstat( path, &drive_st ) == -1 && errno == ENOENT) /* this is a candidate */ { if (avail == -1) { p[2] = 0; /* if mount point symlink doesn't exist either, it's available */ if (lstat( path, &drive_st ) == -1 && errno == ENOENT) avail = drive; p[2] = ':'; } } else in_use[drive] = 1; } else { in_use[drive] = 1; if (!is_valid_device( &drive_st )) continue; if (dev_st.st_rdev == drive_st.st_rdev) goto done; } } if (avail != -1) { /* try to use the one we found */ drive = avail; *p = 'a' + drive; if (symlink( device, path ) != -1) goto done; /* failed, retry the search */ } } drive = -1; done: HeapFree( GetProcessHeap(), 0, path ); return drive; } /* create devices for mapped drives */ static void create_drive_devices(void) { char *path, *p, *link, *device; struct dos_drive *drive; struct volume *volume; unsigned int i; HKEY drives_key; enum device_type drive_type; WCHAR driveW[] = {'a',':',0}; if (!(path = get_dosdevices_path( &p ))) return; if (RegOpenKeyW( HKEY_LOCAL_MACHINE, drives_keyW, &drives_key )) drives_key = 0; for (i = 0; i < MAX_DOS_DRIVES; i++) { p[0] = 'a' + i; p[2] = 0; if (!(link = read_symlink( path ))) continue; p[2] = ':'; device = read_symlink( path ); drive_type = i < 2 ? DEVICE_FLOPPY : DEVICE_HARDDISK_VOL; if (drives_key) { WCHAR buffer[32]; DWORD j, type, size = sizeof(buffer); driveW[0] = 'a' + i; if (!RegQueryValueExW( drives_key, driveW, NULL, &type, (BYTE *)buffer, &size ) && type == REG_SZ) { for (j = 0; j < ARRAY_SIZE(drive_types); j++) if (drive_types[j][0] && !strcmpiW( buffer, drive_types[j] )) { drive_type = j; break; } if (drive_type == DEVICE_FLOPPY && i >= 2) drive_type = DEVICE_HARDDISK; } } volume = find_matching_volume( NULL, device, link, drive_type ); if (!create_dos_device( volume, NULL, i, drive_type, &drive )) { /* don't reset uuid if we used an existing volume */ const GUID *guid = volume ? NULL : get_default_uuid(i); set_volume_info( drive->volume, drive, device, link, drive_type, guid ); } else { RtlFreeHeap( GetProcessHeap(), 0, link ); RtlFreeHeap( GetProcessHeap(), 0, device ); } if (volume) release_volume( volume ); } RegCloseKey( drives_key ); RtlFreeHeap( GetProcessHeap(), 0, path ); } /* open the "Logical Unit" key for a given SCSI address */ static HKEY get_scsi_device_lun_key( SCSI_ADDRESS *scsi_addr ) { WCHAR dataW[50]; HKEY scsi_key, port_key, bus_key, target_key, lun_key; if (RegOpenKeyExW( HKEY_LOCAL_MACHINE, scsi_keyW, 0, KEY_READ|KEY_WRITE, &scsi_key )) return NULL; snprintfW( dataW, ARRAY_SIZE( dataW ), scsi_port_keyW, scsi_addr->PortNumber ); if (RegCreateKeyExW( scsi_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &port_key, NULL )) return NULL; RegCloseKey( scsi_key ); snprintfW( dataW, ARRAY_SIZE( dataW ), scsi_bus_keyW, scsi_addr->PathId ); if (RegCreateKeyExW( port_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &bus_key, NULL )) return NULL; RegCloseKey( port_key ); snprintfW( dataW, ARRAY_SIZE( dataW ), target_id_keyW, scsi_addr->TargetId ); if (RegCreateKeyExW( bus_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &target_key, NULL )) return NULL; RegCloseKey( bus_key ); snprintfW( dataW, ARRAY_SIZE( dataW ), lun_keyW, scsi_addr->Lun ); if (RegCreateKeyExW( target_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &lun_key, NULL )) return NULL; RegCloseKey( target_key ); return lun_key; } /* fill in the "Logical Unit" key for a given SCSI address */ void create_scsi_entry( SCSI_ADDRESS *scsi_addr, UINT init_id, const char *driver, UINT type, const char *model, const UNICODE_STRING *dev ) { static UCHAR tape_no = 0; static const WCHAR tapeW[] = {'T','a','p','e','%','d',0}; static const WCHAR init_id_keyW[] = {'I','n','i','t','i','a','t','o','r',' ','I','d',' ','%','d',0}; static const WCHAR driverW[] = {'D','r','i','v','e','r',0}; static const WCHAR bus_time_scanW[] = {'F','i','r','s','t','B','u','s','T','i','m','e','S','c','a','n','I','n','M','s',0}; static const WCHAR typeW[] = {'T','y','p','e',0}; static const WCHAR identW[] = {'I','d','e','n','t','i','f','i','e','r',0}; WCHAR dataW[50]; DWORD sizeW; DWORD value; const char *data; HKEY scsi_key; HKEY port_key; HKEY bus_key; HKEY target_key; HKEY lun_key; if (RegOpenKeyExW( HKEY_LOCAL_MACHINE, scsi_keyW, 0, KEY_READ|KEY_WRITE, &scsi_key )) return; snprintfW( dataW, ARRAY_SIZE( dataW ), scsi_port_keyW, scsi_addr->PortNumber ); if (RegCreateKeyExW( scsi_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &port_key, NULL )) return; RegCloseKey( scsi_key ); RtlMultiByteToUnicodeN( dataW, sizeof(dataW), &sizeW, driver, strlen(driver)+1); RegSetValueExW( port_key, driverW, 0, REG_SZ, (const BYTE *)dataW, sizeW ); value = 10; RegSetValueExW( port_key, bus_time_scanW, 0, REG_DWORD, (const BYTE *)&value, sizeof(value)); value = 0; snprintfW( dataW, ARRAY_SIZE( dataW ), scsi_bus_keyW, scsi_addr->PathId ); if (RegCreateKeyExW( port_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &bus_key, NULL )) return; RegCloseKey( port_key ); snprintfW( dataW, ARRAY_SIZE( dataW ), init_id_keyW, init_id ); if (RegCreateKeyExW( bus_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &target_key, NULL )) return; RegCloseKey( target_key ); snprintfW( dataW, ARRAY_SIZE( dataW ), target_id_keyW, scsi_addr->TargetId ); if (RegCreateKeyExW( bus_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &target_key, NULL )) return; RegCloseKey( bus_key ); snprintfW( dataW, ARRAY_SIZE( dataW ), lun_keyW, scsi_addr->Lun ); if (RegCreateKeyExW( target_key, dataW, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &lun_key, NULL )) return; RegCloseKey( target_key ); switch (type) { case SCSI_DISK_PERIPHERAL: data = "DiskPeripheral"; break; case SCSI_TAPE_PERIPHERAL: data = "TapePeripheral"; break; case SCSI_PRINTER_PERIPHERAL: data = "PrinterPeripheral"; break; case SCSI_WORM_PERIPHERAL: data = "WormPeripheral"; break; case SCSI_CDROM_PERIPHERAL: data = "CdRomPeripheral"; break; case SCSI_SCANNER_PERIPHERAL: data = "ScannerPeripheral"; break; case SCSI_OPTICAL_DISK_PERIPHERAL: data = "OpticalDiskPeripheral"; break; case SCSI_MEDIUM_CHANGER_PERIPHERAL: data = "MediumChangerPeripheral"; break; case SCSI_COMMS_PERIPHERAL: data = "CommunicationsPeripheral"; break; case SCSI_ASC_GRAPHICS_PERIPHERAL: case SCSI_ASC_GRAPHICS2_PERIPHERAL: data = "ASCPrePressGraphicsPeripheral"; break; case SCSI_ARRAY_PERIPHERAL: data = "ArrayPeripheral"; break; case SCSI_ENCLOSURE_PERIPHERAL: data = "EnclosurePeripheral"; break; case SCSI_REDUCED_DISK_PERIPHERAL: data = "RBCPeripheral"; break; case SCSI_CARD_READER_PERIPHERAL: data = "CardReaderPeripheral"; break; case SCSI_BRIDGE_PERIPHERAL: data = "BridgePeripheral"; break; case SCSI_OBJECT_STORAGE_PERIPHERAL: /* Object-based storage devices */ case SCSI_DRIVE_CONTROLLER_PERIPHERAL: /* Automation/drive controllers */ case SCSI_REDUCED_CDROM_PERIPHERAL: /* Reduced-commands MM devices */ case SCSI_PROCESSOR_PERIPHERAL: /* Processor devices (considered to be "Other" by Windows) */ default: data = "OtherPeripheral"; break; } RtlMultiByteToUnicodeN( dataW, sizeof(dataW), &sizeW, data, strlen(data)+1); RegSetValueExW( lun_key, typeW, 0, REG_SZ, (const BYTE *)dataW, sizeW ); RtlMultiByteToUnicodeN( dataW, sizeof(dataW), &sizeW, model, strlen(model)+1); RegSetValueExW( lun_key, identW, 0, REG_SZ, (const BYTE *)dataW, sizeW ); if (dev) { WCHAR *buffer = memchrW( dev->Buffer+1, '\\', dev->Length )+1; ULONG length = dev->Length - (buffer - dev->Buffer)*sizeof(WCHAR); RegSetValueExW( lun_key, devnameW, 0, REG_SZ, (const BYTE *)buffer, length ); } else if (type == SCSI_TAPE_PERIPHERAL) { snprintfW( dataW, ARRAY_SIZE( dataW ), tapeW, tape_no++ ); RegSetValueExW( lun_key, devnameW, 0, REG_SZ, (const BYTE *)dataW, strlenW( dataW ) ); } RegCloseKey( lun_key ); } /* set the "DeviceName" for a given SCSI address */ void set_scsi_device_name( SCSI_ADDRESS *scsi_addr, const UNICODE_STRING *dev ) { HKEY lun_key; WCHAR *buffer; ULONG length; lun_key = get_scsi_device_lun_key( scsi_addr ); buffer = memchrW( dev->Buffer+1, '\\', dev->Length )+1; length = dev->Length - (buffer - dev->Buffer)*sizeof(WCHAR); RegSetValueExW( lun_key, devnameW, 0, REG_SZ, (const BYTE *)buffer, length ); RegCloseKey( lun_key ); } /* create a new disk volume */ NTSTATUS add_volume( const char *udi, const char *device, const char *mount_point, enum device_type type, const GUID *guid ) { struct volume *volume; NTSTATUS status = STATUS_SUCCESS; TRACE( "adding %s device %s mount %s type %u uuid %s\n", debugstr_a(udi), debugstr_a(device), debugstr_a(mount_point), type, debugstr_guid(guid) ); EnterCriticalSection( &device_section ); LIST_FOR_EACH_ENTRY( volume, &volumes_list, struct volume, entry ) if (volume->udi && !strcmp( udi, volume->udi )) { grab_volume( volume ); goto found; } /* udi not found, search for a non-dynamic volume */ if ((volume = find_matching_volume( udi, device, mount_point, type ))) set_volume_udi( volume, udi ); else status = create_volume( udi, type, &volume ); found: if (!status) status = set_volume_info( volume, NULL, device, mount_point, type, guid ); if (volume) release_volume( volume ); LeaveCriticalSection( &device_section ); return status; } /* create a new disk volume */ NTSTATUS remove_volume( const char *udi ) { NTSTATUS status = STATUS_NO_SUCH_DEVICE; struct volume *volume; EnterCriticalSection( &device_section ); LIST_FOR_EACH_ENTRY( volume, &volumes_list, struct volume, entry ) { if (!volume->udi || strcmp( udi, volume->udi )) continue; set_volume_udi( volume, NULL ); status = STATUS_SUCCESS; break; } LeaveCriticalSection( &device_section ); return status; } /* create a new dos drive */ NTSTATUS add_dos_device( int letter, const char *udi, const char *device, const char *mount_point, enum device_type type, const GUID *guid, UNICODE_STRING *devname ) { char *path, *p; HKEY hkey; NTSTATUS status = STATUS_SUCCESS; struct dos_drive *drive, *next; struct volume *volume; int notify = -1; if (!(path = get_dosdevices_path( &p ))) return STATUS_NO_MEMORY; EnterCriticalSection( &device_section ); volume = find_matching_volume( udi, device, mount_point, type ); if (letter == -1) /* auto-assign a letter */ { letter = add_drive( device, type ); if (letter == -1) { status = STATUS_OBJECT_NAME_COLLISION; goto done; } LIST_FOR_EACH_ENTRY_SAFE( drive, next, &drives_list, struct dos_drive, entry ) { if (drive->volume->udi && !strcmp( udi, drive->volume->udi )) goto found; if (drive->drive == letter) delete_dos_device( drive ); } } else /* simply reset the device symlink */ { LIST_FOR_EACH_ENTRY( drive, &drives_list, struct dos_drive, entry ) if (drive->drive == letter) break; *p = 'a' + letter; if (&drive->entry == &drives_list) update_symlink( path, device, NULL ); else { update_symlink( path, device, drive->volume->device->unix_device ); delete_dos_device( drive ); } } if ((status = create_dos_device( volume, udi, letter, type, &drive ))) goto done; found: if (!guid && !volume) guid = get_default_uuid( letter ); if (!volume) volume = grab_volume( drive->volume ); set_drive_info( drive, letter, volume ); p[0] = 'a' + drive->drive; p[2] = 0; update_symlink( path, mount_point, volume->device->unix_mount ); set_volume_info( volume, drive, device, mount_point, type, guid ); TRACE( "added device %c: udi %s for %s on %s type %u\n", 'a' + drive->drive, wine_dbgstr_a(udi), wine_dbgstr_a(device), wine_dbgstr_a(mount_point), type ); /* hack: force the drive type in the registry */ if (!RegCreateKeyW( HKEY_LOCAL_MACHINE, drives_keyW, &hkey )) { const WCHAR *type_name = drive_types[type]; WCHAR name[] = {'a',':',0}; name[0] += drive->drive; if (!type_name[0] && type == DEVICE_HARDDISK) type_name = drive_types[DEVICE_FLOPPY]; if (type_name[0]) RegSetValueExW( hkey, name, 0, REG_SZ, (const BYTE *)type_name, (strlenW(type_name) + 1) * sizeof(WCHAR) ); else RegDeleteValueW( hkey, name ); RegCloseKey( hkey ); } if (udi) notify = drive->drive; if (devname) *devname = volume->device->name; done: if (volume) release_volume( volume ); LeaveCriticalSection( &device_section ); RtlFreeHeap( GetProcessHeap(), 0, path ); if (notify != -1) send_notify( notify, DBT_DEVICEARRIVAL ); return status; } /* remove an existing dos drive, by letter or udi */ NTSTATUS remove_dos_device( int letter, const char *udi ) { NTSTATUS status = STATUS_NO_SUCH_DEVICE; HKEY hkey; struct dos_drive *drive; char *path, *p; int notify = -1; EnterCriticalSection( &device_section ); LIST_FOR_EACH_ENTRY( drive, &drives_list, struct dos_drive, entry ) { if (udi) { if (!drive->volume->udi) continue; if (strcmp( udi, drive->volume->udi )) continue; set_volume_udi( drive->volume, NULL ); } else if (drive->drive != letter) continue; if ((path = get_dosdevices_path( &p ))) { p[0] = 'a' + drive->drive; p[2] = 0; unlink( path ); RtlFreeHeap( GetProcessHeap(), 0, path ); } /* clear the registry key too */ if (!RegOpenKeyW( HKEY_LOCAL_MACHINE, drives_keyW, &hkey )) { WCHAR name[] = {'a',':',0}; name[0] += drive->drive; RegDeleteValueW( hkey, name ); RegCloseKey( hkey ); } if (udi && drive->volume->device->unix_mount) notify = drive->drive; delete_dos_device( drive ); status = STATUS_SUCCESS; break; } LeaveCriticalSection( &device_section ); if (notify != -1) send_notify( notify, DBT_DEVICEREMOVECOMPLETE ); return status; } enum mountmgr_fs_type get_mountmgr_fs_type(enum fs_type fs_type) { switch (fs_type) { case FS_ISO9660: return MOUNTMGR_FS_TYPE_ISO9660; case FS_UDF: return MOUNTMGR_FS_TYPE_UDF; case FS_FAT1216: return MOUNTMGR_FS_TYPE_FAT; case FS_FAT32: return MOUNTMGR_FS_TYPE_FAT32; default: return MOUNTMGR_FS_TYPE_NTFS; } } /* query information about an existing dos drive, by letter or udi */ NTSTATUS query_dos_device( int letter, enum device_type *type, enum mountmgr_fs_type *fs_type, DWORD *serial, char **device, char **mount_point, WCHAR **label ) { NTSTATUS status = STATUS_NO_SUCH_DEVICE; struct dos_drive *drive; struct disk_device *disk_device; EnterCriticalSection( &device_section ); LIST_FOR_EACH_ENTRY( drive, &drives_list, struct dos_drive, entry ) { if (drive->drive != letter) continue; disk_device = drive->volume->device; if (type) *type = disk_device->type; if (fs_type) *fs_type = get_mountmgr_fs_type( drive->volume->fs_type ); if (serial) *serial = drive->volume->serial; if (device) *device = strdupA( disk_device->unix_device ); if (mount_point) *mount_point = strdupA( disk_device->unix_mount ); if (label) *label = strdupW( drive->volume->label ); status = STATUS_SUCCESS; break; } LeaveCriticalSection( &device_section ); return status; } /* query information about an existing unix device, by dev_t */ NTSTATUS query_unix_device( ULONGLONG unix_dev, enum device_type *type, enum mountmgr_fs_type *fs_type, DWORD *serial, char **device, char **mount_point, WCHAR **label ) { NTSTATUS status = STATUS_NO_SUCH_DEVICE; struct volume *volume; struct disk_device *disk_device; struct stat st; EnterCriticalSection( &device_section ); LIST_FOR_EACH_ENTRY( volume, &volumes_list, struct volume, entry ) { disk_device = volume->device; if (!disk_device->unix_device || stat( disk_device->unix_device, &st ) < 0 || st.st_rdev != unix_dev) continue; if (type) *type = disk_device->type; if (fs_type) *fs_type = get_mountmgr_fs_type( volume->fs_type ); if (serial) *serial = volume->serial; if (device) *device = strdupA( disk_device->unix_device ); if (mount_point) *mount_point = strdupA( disk_device->unix_mount ); if (label) *label = strdupW( volume->label ); status = STATUS_SUCCESS; break; } LeaveCriticalSection( &device_section ); return status; } static void query_property(IRP *irp) { IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp ); STORAGE_PROPERTY_QUERY *query = irp->AssociatedIrp.SystemBuffer; if (!irp->AssociatedIrp.SystemBuffer || irpsp->Parameters.DeviceIoControl.InputBufferLength < sizeof(STORAGE_PROPERTY_QUERY)) { irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER; return; } /* Try to persuade application not to check property */ if (query->QueryType == PropertyExistsQuery) { irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED; return; } switch (query->PropertyId) { case StorageDeviceProperty: { STORAGE_DEVICE_DESCRIPTOR *descriptor; if (!irp->UserBuffer || irpsp->Parameters.DeviceIoControl.OutputBufferLength < sizeof(STORAGE_DESCRIPTOR_HEADER)) irp->IoStatus.u.Status = STATUS_INVALID_PARAMETER; else if (irpsp->Parameters.DeviceIoControl.OutputBufferLength < sizeof(STORAGE_DEVICE_DESCRIPTOR)) { descriptor = irp->UserBuffer; descriptor->Version = sizeof(STORAGE_DEVICE_DESCRIPTOR); descriptor->Size = sizeof(STORAGE_DEVICE_DESCRIPTOR); irp->IoStatus.Information = sizeof(STORAGE_DESCRIPTOR_HEADER); irp->IoStatus.u.Status = STATUS_SUCCESS; } else { FIXME( "Faking StorageDeviceProperty data\n" ); memset( irp->UserBuffer, 0, irpsp->Parameters.DeviceIoControl.OutputBufferLength ); descriptor = irp->UserBuffer; descriptor->Version = sizeof(STORAGE_DEVICE_DESCRIPTOR); descriptor->Size = sizeof(STORAGE_DEVICE_DESCRIPTOR); descriptor->DeviceType = FILE_DEVICE_DISK; descriptor->DeviceTypeModifier = 0; descriptor->RemovableMedia = FALSE; descriptor->CommandQueueing = FALSE; descriptor->VendorIdOffset = 0; descriptor->ProductIdOffset = 0; descriptor->ProductRevisionOffset = 0; descriptor->SerialNumberOffset = 0; descriptor->BusType = BusTypeScsi; descriptor->RawPropertiesLength = 0; irp->IoStatus.Information = sizeof(STORAGE_DEVICE_DESCRIPTOR); irp->IoStatus.u.Status = STATUS_SUCCESS; } break; } default: FIXME( "Unsupported property %#x\n", query->PropertyId ); irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED; break; } } /* handler for ioctls on the harddisk device */ static NTSTATUS WINAPI harddisk_ioctl( DEVICE_OBJECT *device, IRP *irp ) { IO_STACK_LOCATION *irpsp = IoGetCurrentIrpStackLocation( irp ); struct disk_device *dev = device->DeviceExtension; TRACE( "ioctl %x insize %u outsize %u\n", irpsp->Parameters.DeviceIoControl.IoControlCode, irpsp->Parameters.DeviceIoControl.InputBufferLength, irpsp->Parameters.DeviceIoControl.OutputBufferLength ); EnterCriticalSection( &device_section ); switch(irpsp->Parameters.DeviceIoControl.IoControlCode) { case IOCTL_DISK_GET_DRIVE_GEOMETRY: { DISK_GEOMETRY info; DWORD len = min( sizeof(info), irpsp->Parameters.DeviceIoControl.OutputBufferLength ); info.Cylinders.QuadPart = 10000; info.MediaType = (dev->devnum.DeviceType == FILE_DEVICE_DISK) ? FixedMedia : RemovableMedia; info.TracksPerCylinder = 255; info.SectorsPerTrack = 63; info.BytesPerSector = 512; memcpy( irp->AssociatedIrp.SystemBuffer, &info, len ); irp->IoStatus.Information = len; irp->IoStatus.u.Status = STATUS_SUCCESS; break; } case IOCTL_DISK_GET_DRIVE_GEOMETRY_EX: { DISK_GEOMETRY_EX info; DWORD len = min( sizeof(info), irpsp->Parameters.DeviceIoControl.OutputBufferLength ); FIXME("The DISK_PARTITION_INFO and DISK_DETECTION_INFO structures will not be filled\n"); info.Geometry.Cylinders.QuadPart = 10000; info.Geometry.MediaType = (dev->devnum.DeviceType == FILE_DEVICE_DISK) ? FixedMedia : RemovableMedia; info.Geometry.TracksPerCylinder = 255; info.Geometry.SectorsPerTrack = 63; info.Geometry.BytesPerSector = 512; info.DiskSize.QuadPart = info.Geometry.Cylinders.QuadPart * info.Geometry.TracksPerCylinder * info.Geometry.SectorsPerTrack * info.Geometry.BytesPerSector; info.Data[0] = 0; memcpy( irp->AssociatedIrp.SystemBuffer, &info, len ); irp->IoStatus.Information = len; irp->IoStatus.u.Status = STATUS_SUCCESS; break; } case IOCTL_STORAGE_GET_DEVICE_NUMBER: { DWORD len = min( sizeof(dev->devnum), irpsp->Parameters.DeviceIoControl.OutputBufferLength ); memcpy( irp->AssociatedIrp.SystemBuffer, &dev->devnum, len ); irp->IoStatus.Information = len; irp->IoStatus.u.Status = STATUS_SUCCESS; break; } case IOCTL_CDROM_READ_TOC: irp->IoStatus.u.Status = STATUS_INVALID_DEVICE_REQUEST; break; case IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS: { DWORD len = min( 32, irpsp->Parameters.DeviceIoControl.OutputBufferLength ); FIXME( "returning zero-filled buffer for IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS\n" ); memset( irp->AssociatedIrp.SystemBuffer, 0, len ); irp->IoStatus.Information = len; irp->IoStatus.u.Status = STATUS_SUCCESS; break; } case IOCTL_STORAGE_QUERY_PROPERTY: query_property( irp ); break; default: { ULONG code = irpsp->Parameters.DeviceIoControl.IoControlCode; FIXME("Unsupported ioctl %x (device=%x access=%x func=%x method=%x)\n", code, code >> 16, (code >> 14) & 3, (code >> 2) & 0xfff, code & 3); irp->IoStatus.u.Status = STATUS_NOT_SUPPORTED; break; } } LeaveCriticalSection( &device_section ); IoCompleteRequest( irp, IO_NO_INCREMENT ); return STATUS_SUCCESS; } /* driver entry point for the harddisk driver */ NTSTATUS WINAPI harddisk_driver_entry( DRIVER_OBJECT *driver, UNICODE_STRING *path ) { struct disk_device *device; harddisk_driver = driver; driver->MajorFunction[IRP_MJ_DEVICE_CONTROL] = harddisk_ioctl; /* create a harddisk0 device that isn't assigned to any drive */ create_disk_device( DEVICE_HARDDISK, &device ); create_drive_devices(); return STATUS_SUCCESS; } /* create a serial or parallel port */ static BOOL create_port_device( DRIVER_OBJECT *driver, int n, const char *unix_path, const char *dosdevices_path, HKEY windows_ports_key ) { static const WCHAR comW[] = {'C','O','M','%','u',0}; static const WCHAR lptW[] = {'L','P','T','%','u',0}; static const WCHAR device_serialW[] = {'\\','D','e','v','i','c','e','\\','S','e','r','i','a','l','%','u',0}; static const WCHAR device_parallelW[] = {'\\','D','e','v','i','c','e','\\','P','a','r','a','l','l','e','l','%','u',0}; static const WCHAR dosdevices_comW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','C','O','M','%','u',0}; static const WCHAR dosdevices_auxW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','A','U','X',0}; static const WCHAR dosdevices_lptW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','L','P','T','%','u',0}; static const WCHAR dosdevices_prnW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','P','R','N',0}; const WCHAR *dos_name_format, *nt_name_format, *reg_value_format, *symlink_format, *default_device; WCHAR dos_name[7], reg_value[256], nt_buffer[32], symlink_buffer[32]; UNICODE_STRING nt_name, symlink_name, default_name; DEVICE_OBJECT *dev_obj; NTSTATUS status; if (driver == serial_driver) { dos_name_format = comW; nt_name_format = device_serialW; reg_value_format = comW; symlink_format = dosdevices_comW; default_device = dosdevices_auxW; } else { dos_name_format = lptW; nt_name_format = device_parallelW; reg_value_format = dosdevices_lptW; symlink_format = dosdevices_lptW; default_device = dosdevices_prnW; } sprintfW( dos_name, dos_name_format, n ); /* create DOS device */ unlink( dosdevices_path ); if (symlink( unix_path, dosdevices_path ) != 0) return FALSE; /* create NT device */ sprintfW( nt_buffer, nt_name_format, n - 1 ); RtlInitUnicodeString( &nt_name, nt_buffer ); status = IoCreateDevice( driver, 0, &nt_name, 0, 0, FALSE, &dev_obj ); if (status != STATUS_SUCCESS) { FIXME( "IoCreateDevice %s got %x\n", debugstr_w(nt_name.Buffer), status ); return FALSE; } sprintfW( symlink_buffer, symlink_format, n ); RtlInitUnicodeString( &symlink_name, symlink_buffer ); IoCreateSymbolicLink( &symlink_name, &nt_name ); if (n == 1) { RtlInitUnicodeString( &default_name, default_device ); IoCreateSymbolicLink( &default_name, &symlink_name ); } /* TODO: store information about the Unix device in the NT device */ /* create registry entry */ sprintfW( reg_value, reg_value_format, n ); RegSetValueExW( windows_ports_key, nt_name.Buffer, 0, REG_SZ, (BYTE *)reg_value, (strlenW( reg_value ) + 1) * sizeof(WCHAR) ); return TRUE; } /* find and create serial or parallel ports */ static void create_port_devices( DRIVER_OBJECT *driver ) { static const char *serial_search_paths[] = { #ifdef linux "/dev/ttyS%u", "/dev/ttyUSB%u", "/dev/ttyACM%u", #elif defined(__FreeBSD__) || defined(__FreeBSD_kernel__) "/dev/cuau%u", #elif defined(__DragonFly__) "/dev/cuaa%u", #endif NULL }; static const char *parallel_search_paths[] = { #ifdef linux "/dev/lp%u", #endif NULL }; static const WCHAR serialcomm_keyW[] = {'H','A','R','D','W','A','R','E','\\', 'D','E','V','I','C','E','M','A','P','\\', 'S','E','R','I','A','L','C','O','M','M',0}; static const WCHAR parallel_ports_keyW[] = {'H','A','R','D','W','A','R','E','\\', 'D','E','V','I','C','E','M','A','P','\\', 'P','A','R','A','L','L','E','L',' ','P','O','R','T','S',0}; static const WCHAR comW[] = {'C','O','M'}; static const WCHAR lptW[] = {'L','P','T'}; const char **search_paths; const WCHAR *windows_ports_key_name; char *dosdevices_path, *p; HKEY wine_ports_key = NULL, windows_ports_key = NULL; char unix_path[256]; const WCHAR *port_prefix; WCHAR reg_value[256]; BOOL used[MAX_PORTS]; WCHAR port[7]; DWORD port_len, type, size; int i, j, n; if (!(dosdevices_path = get_dosdevices_path( &p ))) return; if (driver == serial_driver) { p[0] = 'c'; p[1] = 'o'; p[2] = 'm'; search_paths = serial_search_paths; windows_ports_key_name = serialcomm_keyW; port_prefix = comW; } else { p[0] = 'l'; p[1] = 'p'; p[2] = 't'; search_paths = parallel_search_paths; windows_ports_key_name = parallel_ports_keyW; port_prefix = lptW; } p += 3; RegCreateKeyExW( HKEY_LOCAL_MACHINE, ports_keyW, 0, NULL, 0, KEY_QUERY_VALUE, NULL, &wine_ports_key, NULL ); RegCreateKeyExW( HKEY_LOCAL_MACHINE, windows_ports_key_name, 0, NULL, REG_OPTION_VOLATILE, KEY_ALL_ACCESS, NULL, &windows_ports_key, NULL ); /* add user-defined serial ports */ memset(used, 0, sizeof(used)); for (i = 0; ; i++) { port_len = ARRAY_SIZE(port); size = sizeof(reg_value); if (RegEnumValueW( wine_ports_key, i, port, &port_len, NULL, &type, (BYTE*)reg_value, &size ) != ERROR_SUCCESS) break; if (type != REG_SZ || strncmpiW( port, port_prefix, 3 )) continue; n = atolW( port + 3 ); if (n < 1 || n >= MAX_PORTS) continue; if (!WideCharToMultiByte( CP_UNIXCP, WC_ERR_INVALID_CHARS, reg_value, size/sizeof(WCHAR), unix_path, sizeof(unix_path), NULL, NULL)) continue; used[n - 1] = TRUE; sprintf( p, "%u", n ); create_port_device( driver, n, unix_path, dosdevices_path, windows_ports_key ); } /* look for ports in the usual places */ n = 1; while (n <= MAX_PORTS && used[n - 1]) n++; for (i = 0; search_paths[i]; i++) { for (j = 0; n <= MAX_PORTS; j++) { sprintf( unix_path, search_paths[i], j ); if (access( unix_path, F_OK ) != 0) break; sprintf( p, "%u", n ); create_port_device( driver, n, unix_path, dosdevices_path, windows_ports_key ); n++; while (n <= MAX_PORTS && used[n - 1]) n++; } } RegCloseKey( wine_ports_key ); RegCloseKey( windows_ports_key ); HeapFree( GetProcessHeap(), 0, dosdevices_path ); } /* driver entry point for the serial port driver */ NTSTATUS WINAPI serial_driver_entry( DRIVER_OBJECT *driver, UNICODE_STRING *path ) { serial_driver = driver; /* TODO: fill in driver->MajorFunction */ create_port_devices( driver ); return STATUS_SUCCESS; } /* driver entry point for the parallel port driver */ NTSTATUS WINAPI parallel_driver_entry( DRIVER_OBJECT *driver, UNICODE_STRING *path ) { parallel_driver = driver; /* TODO: fill in driver->MajorFunction */ create_port_devices( driver ); return STATUS_SUCCESS; }