git.oblomov.eu Git - wine/blob - files/drive.c

   1 /*
   2  * DOS drives handling functions
   3  *
   4  * Copyright 1993 Erik Bos
   5  * Copyright 1996 Alexandre Julliard
   6  */
   7
   8 #include "config.h"
   9
  10 #include <assert.h>
  11 #include <ctype.h>
  12 #include <string.h>
  13 #include <stdlib.h>
  14 #include <sys/types.h>
  15 #include <sys/stat.h>
  16 #include <fcntl.h>
  17 #include <errno.h>
  18
  19 #ifdef HAVE_SYS_PARAM_H
  20 # include <sys/param.h>
  21 #endif
  22 #ifdef STATFS_DEFINED_BY_SYS_VFS
  23 # include <sys/vfs.h>
  24 #else
  25 # ifdef STATFS_DEFINED_BY_SYS_MOUNT
  26 #  include <sys/mount.h>
  27 # else
  28 #  ifdef STATFS_DEFINED_BY_SYS_STATFS
  29 #   include <sys/statfs.h>
  30 #  endif
  31 # endif
  32 #endif
  33
  34 #include "windows.h"
  35 #include "winbase.h"
  36 #include "drive.h"
  37 #include "file.h"
  38 #include "heap.h"
  39 #include "msdos.h"
  40 #include "options.h"
  41 #include "task.h"
  42 #include "debug.h"
  43
  44 typedef struct
  45 {
  46     char     *root;      /* root dir in Unix format without trailing / */
  47     char     *dos_cwd;   /* cwd in DOS format without leading or trailing \ */
  48     char     *unix_cwd;  /* cwd in Unix format without leading or trailing / */
  49     char     *device;    /* raw device path */
  50     char      label[12]; /* drive label */
  51     DWORD     serial;    /* drive serial number */
  52     DRIVETYPE type;      /* drive type */
  53     UINT32    flags;     /* drive flags */
  54     dev_t     dev;       /* unix device number */
  55     ino_t     ino;       /* unix inode number */
  56 } DOSDRIVE;
  57
  58
  59 static const char * const DRIVE_Types[] =
  60 {
  61     "floppy",   /* TYPE_FLOPPY */
  62     "hd",       /* TYPE_HD */
  63     "cdrom",    /* TYPE_CDROM */
  64     "network"   /* TYPE_NETWORK */
  65 };
  66
  67
  68 /* Known filesystem types */
  69
  70 typedef struct
  71 {
  72     const char *name;
  73     UINT32      flags;
  74 } FS_DESCR;
  75
  76 static const FS_DESCR DRIVE_Filesystems[] =
  77 {
  78     { "unix",   DRIVE_CASE_SENSITIVE | DRIVE_CASE_PRESERVING },
  79     { "msdos",  DRIVE_SHORT_NAMES },
  80     { "dos",    DRIVE_SHORT_NAMES },
  81     { "fat",    DRIVE_SHORT_NAMES },
  82     { "vfat",   DRIVE_CASE_PRESERVING },
  83     { "win95",  DRIVE_CASE_PRESERVING },
  84     { NULL, 0 }
  85 };
  86
  87
  88 static DOSDRIVE DOSDrives[MAX_DOS_DRIVES];
  89 static int DRIVE_CurDrive = -1;
  90
  91 static HTASK16 DRIVE_LastTask = 0;
  92
  93
  94 /***********************************************************************
  95  *           DRIVE_GetDriveType
  96  */
  97 static DRIVETYPE DRIVE_GetDriveType( const char *name )
  98 {
  99     char buffer[20];
 100     int i;
 101
 102     PROFILE_GetWineIniString( name, "Type", "hd", buffer, sizeof(buffer) );
 103     for (i = 0; i < sizeof(DRIVE_Types)/sizeof(DRIVE_Types[0]); i++)
 104     {
 105         if (!lstrcmpi32A( buffer, DRIVE_Types[i] )) return (DRIVETYPE)i;
 106     }
 107     MSG("%s: unknown type '%s', defaulting to 'hd'.\n", name, buffer );
 108     return TYPE_HD;
 109 }
 110
 111
 112 /***********************************************************************
 113  *           DRIVE_GetFSFlags
 114  */
 115 static UINT32 DRIVE_GetFSFlags( const char *name, const char *value )
 116 {
 117     const FS_DESCR *descr;
 118
 119     for (descr = DRIVE_Filesystems; descr->name; descr++)
 120         if (!lstrcmpi32A( value, descr->name )) return descr->flags;
 121     MSG("%s: unknown filesystem type '%s', defaulting to 'unix'.\n",
 122         name, value );
 123     return DRIVE_CASE_SENSITIVE | DRIVE_CASE_PRESERVING;
 124 }
 125
 126
 127 /***********************************************************************
 128  *           DRIVE_Init
 129  */
 130 int DRIVE_Init(void)
 131 {
 132     int i, len, count = 0;
 133     char name[] = "Drive A";
 134     char path[MAX_PATHNAME_LEN];
 135     char buffer[80];
 136     struct stat drive_stat_buffer;
 137     char *p;
 138     DOSDRIVE *drive;
 139
 140     for (i = 0, drive = DOSDrives; i < MAX_DOS_DRIVES; i++, name[6]++, drive++)
 141     {
 142         PROFILE_GetWineIniString( name, "Path", "", path, sizeof(path)-1 );
 143         if (path[0])
 144         {
 145             p = path + strlen(path) - 1;
 146             while ((p > path) && ((*p == '/') || (*p == '\\'))) *p-- = '\0';
 147             if (!path[0]) strcpy( path, "/" );
 148
 149             if (stat( path, &drive_stat_buffer ))
 150             {
 151                 MSG("Could not stat %s, ignoring drive %c:\n", path, 'A' + i );
 152                 continue;
 153             }
 154             if (!S_ISDIR(drive_stat_buffer.st_mode))
 155             {
 156                 MSG("%s is not a directory, ignoring drive %c:\n",
 157                     path, 'A' + i );
 158                 continue;
 159             }
 160
 161             drive->root = HEAP_strdupA( SystemHeap, 0, path );
 162             drive->dos_cwd  = HEAP_strdupA( SystemHeap, 0, "" );
 163             drive->unix_cwd = HEAP_strdupA( SystemHeap, 0, "" );
 164             drive->type     = DRIVE_GetDriveType( name );
 165             drive->device   = NULL;
 166             drive->flags    = 0;
 167             drive->dev      = drive_stat_buffer.st_dev;
 168             drive->ino      = drive_stat_buffer.st_ino;
 169
 170             /* Get the drive label */
 171             PROFILE_GetWineIniString( name, "Label", name, drive->label, 12 );
 172             if ((len = strlen(drive->label)) < 11)
 173             {
 174                 /* Pad label with spaces */
 175                 memset( drive->label + len, ' ', 11 - len );
 176                 drive->label[12] = '\0';
 177             }
 178
 179             /* Get the serial number */
 180             PROFILE_GetWineIniString( name, "Serial", "12345678",
 181                                       buffer, sizeof(buffer) );
 182             drive->serial = strtoul( buffer, NULL, 16 );
 183
 184             /* Get the filesystem type */
 185             PROFILE_GetWineIniString( name, "Filesystem", "unix",
 186                                       buffer, sizeof(buffer) );
 187             drive->flags = DRIVE_GetFSFlags( name, buffer );
 188
 189             /* Get the device */
 190             PROFILE_GetWineIniString( name, "Device", "",
 191                                       buffer, sizeof(buffer) );
 192             if (buffer[0])
 193                 drive->device = HEAP_strdupA( SystemHeap, 0, buffer );
 194
 195             /* Make the first hard disk the current drive */
 196             if ((DRIVE_CurDrive == -1) && (drive->type == TYPE_HD))
 197                 DRIVE_CurDrive = i;
 198
 199             count++;
 200             TRACE(dosfs, "%s: path=%s type=%s label='%s' serial=%08lx flags=%08x dev=%x ino=%x\n",
 201                            name, path, DRIVE_Types[drive->type],
 202                            drive->label, drive->serial, drive->flags,
 203                            (int)drive->dev, (int)drive->ino );
 204         }
 205         else WARN(dosfs, "%s: not defined\n", name );
 206     }
 207
 208     if (!count)
 209     {
 210         MSG("Warning: no valid DOS drive found, check your configuration file.\n" );
 211         /* Create a C drive pointing to Unix root dir */
 212         DOSDrives[2].root     = HEAP_strdupA( SystemHeap, 0, "/" );
 213         DOSDrives[2].dos_cwd  = HEAP_strdupA( SystemHeap, 0, "" );
 214         DOSDrives[2].unix_cwd = HEAP_strdupA( SystemHeap, 0, "" );
 215         strcpy( DOSDrives[2].label, "Drive C    " );
 216         DOSDrives[2].serial   = 0x12345678;
 217         DOSDrives[2].type     = TYPE_HD;
 218         DOSDrives[2].flags    = 0;
 219         DRIVE_CurDrive = 2;
 220     }
 221
 222     /* Make sure the current drive is valid */
 223     if (DRIVE_CurDrive == -1)
 224     {
 225         for (i = 0, drive = DOSDrives; i < MAX_DOS_DRIVES; i++, drive++)
 226         {
 227             if (drive->root && !(drive->flags & DRIVE_DISABLED))
 228             {
 229                 DRIVE_CurDrive = i;
 230                 break;
 231             }
 232         }
 233     }
 234
 235     return 1;
 236 }
 237
 238
 239 /***********************************************************************
 240  *           DRIVE_IsValid
 241  */
 242 int DRIVE_IsValid( int drive )
 243 {
 244     if ((drive < 0) || (drive >= MAX_DOS_DRIVES)) return 0;
 245     return (DOSDrives[drive].root &&
 246             !(DOSDrives[drive].flags & DRIVE_DISABLED));
 247 }
 248
 249
 250 /***********************************************************************
 251  *           DRIVE_GetCurrentDrive
 252  */
 253 int DRIVE_GetCurrentDrive(void)
 254 {
 255     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
 256     if (pTask && (pTask->curdrive & 0x80)) return pTask->curdrive & ~0x80;
 257     return DRIVE_CurDrive;
 258 }
 259
 260
 261 /***********************************************************************
 262  *           DRIVE_SetCurrentDrive
 263  */
 264 int DRIVE_SetCurrentDrive( int drive )
 265 {
 266     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
 267     if (!DRIVE_IsValid( drive ))
 268     {
 269         DOS_ERROR( ER_InvalidDrive, EC_MediaError, SA_Abort, EL_Disk );
 270         return 0;
 271     }
 272     TRACE(dosfs, "%c:\n", 'A' + drive );
 273     DRIVE_CurDrive = drive;
 274     if (pTask) pTask->curdrive = drive | 0x80;
 275     return 1;
 276 }
 277
 278
 279 /***********************************************************************
 280  *           DRIVE_FindDriveRoot
 281  *
 282  * Find a drive for which the root matches the begginning of the given path.
 283  * This can be used to translate a Unix path into a drive + DOS path.
 284  * Return value is the drive, or -1 on error. On success, path is modified
 285  * to point to the beginning of the DOS path.
 286  */
 287 int DRIVE_FindDriveRoot( const char **path )
 288 {
 289     /* idea: check at all '/' positions.
 290      * If the device and inode of that path is identical with the
 291      * device and inode of the current drive then we found a solution.
 292      * If there is another drive pointing to a deeper position in
 293      * the file tree, we want to find that one, not the earlier solution.
 294      */
 295     int drive, rootdrive = -1;
 296     char buffer[MAX_PATHNAME_LEN];
 297     char *next = buffer;
 298     const char *p = *path;
 299     struct stat st;
 300
 301     strcpy( buffer, "/" );
 302     for (;;)
 303     {
 304         if (stat( buffer, &st ) || !S_ISDIR( st.st_mode )) break;
 305
 306         /* Find the drive */
 307
 308         for (drive = 0; drive < MAX_DOS_DRIVES; drive++)
 309         {
 310            if (!DOSDrives[drive].root ||
 311                (DOSDrives[drive].flags & DRIVE_DISABLED)) continue;
 312
 313            if ((DOSDrives[drive].dev == st.st_dev) &&
 314                (DOSDrives[drive].ino == st.st_ino))
 315            {
 316                rootdrive = drive;
 317                *path = p;
 318            }
 319         }
 320
 321         /* Get the next path component */
 322
 323         *next++ = '/';
 324         while ((*p == '/') || (*p == '\\')) p++;
 325         if (!*p) break;
 326         while (!IS_END_OF_NAME(*p)) *next++ = *p++;
 327         *next = 0;
 328     }
 329     *next = 0;
 330
 331     if (rootdrive != -1)
 332         TRACE(dosfs, "%s -> drive %c:, root='%s', name='%s'\n",
 333                        buffer, 'A' + rootdrive,
 334                        DOSDrives[rootdrive].root, *path );
 335     return rootdrive;
 336 }
 337
 338
 339 /***********************************************************************
 340  *           DRIVE_GetRoot
 341  */
 342 const char * DRIVE_GetRoot( int drive )
 343 {
 344     if (!DRIVE_IsValid( drive )) return NULL;
 345     return DOSDrives[drive].root;
 346 }
 347
 348
 349 /***********************************************************************
 350  *           DRIVE_GetDosCwd
 351  */
 352 const char * DRIVE_GetDosCwd( int drive )
 353 {
 354     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
 355     if (!DRIVE_IsValid( drive )) return NULL;
 356
 357     /* Check if we need to change the directory to the new task. */
 358     if (pTask && (pTask->curdrive & 0x80) &&    /* The task drive is valid */
 359         ((pTask->curdrive & ~0x80) == drive) && /* and it's the one we want */
 360         (DRIVE_LastTask != GetCurrentTask()))   /* and the task changed */
 361     {
 362         /* Perform the task-switch */
 363         if (!DRIVE_Chdir( drive, pTask->curdir )) DRIVE_Chdir( drive, "\\" );
 364         DRIVE_LastTask = GetCurrentTask();
 365     }
 366     return DOSDrives[drive].dos_cwd;
 367 }
 368
 369
 370 /***********************************************************************
 371  *           DRIVE_GetUnixCwd
 372  */
 373 const char * DRIVE_GetUnixCwd( int drive )
 374 {
 375     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
 376     if (!DRIVE_IsValid( drive )) return NULL;
 377
 378     /* Check if we need to change the directory to the new task. */
 379     if (pTask && (pTask->curdrive & 0x80) &&    /* The task drive is valid */
 380         ((pTask->curdrive & ~0x80) == drive) && /* and it's the one we want */
 381         (DRIVE_LastTask != GetCurrentTask()))   /* and the task changed */
 382     {
 383         /* Perform the task-switch */
 384         if (!DRIVE_Chdir( drive, pTask->curdir )) DRIVE_Chdir( drive, "\\" );
 385         DRIVE_LastTask = GetCurrentTask();
 386     }
 387     return DOSDrives[drive].unix_cwd;
 388 }
 389
 390
 391 /***********************************************************************
 392  *           DRIVE_GetLabel
 393  */
 394 const char * DRIVE_GetLabel( int drive )
 395 {
 396     if (!DRIVE_IsValid( drive )) return NULL;
 397     return DOSDrives[drive].label;
 398 }
 399
 400
 401 /***********************************************************************
 402  *           DRIVE_GetSerialNumber
 403  */
 404 DWORD DRIVE_GetSerialNumber( int drive )
 405 {
 406     if (!DRIVE_IsValid( drive )) return 0;
 407     return DOSDrives[drive].serial;
 408 }
 409
 410
 411 /***********************************************************************
 412  *           DRIVE_SetSerialNumber
 413  */
 414 int DRIVE_SetSerialNumber( int drive, DWORD serial )
 415 {
 416     if (!DRIVE_IsValid( drive )) return 0;
 417     DOSDrives[drive].serial = serial;
 418     return 1;
 419 }
 420
 421
 422 /***********************************************************************
 423  *           DRIVE_GetType
 424  */
 425 DRIVETYPE DRIVE_GetType( int drive )
 426 {
 427     if (!DRIVE_IsValid( drive )) return TYPE_INVALID;
 428     return DOSDrives[drive].type;
 429 }
 430
 431
 432 /***********************************************************************
 433  *           DRIVE_GetFlags
 434  */
 435 UINT32 DRIVE_GetFlags( int drive )
 436 {
 437     if ((drive < 0) || (drive >= MAX_DOS_DRIVES)) return 0;
 438     return DOSDrives[drive].flags;
 439 }
 440
 441
 442 /***********************************************************************
 443  *           DRIVE_Chdir
 444  */
 445 int DRIVE_Chdir( int drive, const char *path )
 446 {
 447     DOS_FULL_NAME full_name;
 448     char buffer[MAX_PATHNAME_LEN];
 449     LPSTR unix_cwd;
 450     BY_HANDLE_FILE_INFORMATION info;
 451     TDB *pTask = (TDB *)GlobalLock16( GetCurrentTask() );
 452
 453     TRACE(dosfs, "(%c:,%s)\n", 'A' + drive, path );
 454     strcpy( buffer, "A:" );
 455     buffer[0] += drive;
 456     lstrcpyn32A( buffer + 2, path, sizeof(buffer) - 2 );
 457
 458     if (!DOSFS_GetFullName( buffer, TRUE, &full_name )) return 0;
 459     if (!FILE_Stat( full_name.long_name, &info )) return 0;
 460     if (!(info.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY))
 461     {
 462         DOS_ERROR( ER_FileNotFound, EC_NotFound, SA_Abort, EL_Disk );
 463         return 0;
 464     }
 465     unix_cwd = full_name.long_name + strlen( DOSDrives[drive].root );
 466     while (*unix_cwd == '/') unix_cwd++;
 467
 468     TRACE(dosfs, "(%c:): unix_cwd=%s dos_cwd=%s\n",
 469                    'A' + drive, unix_cwd, full_name.short_name + 3 );
 470
 471     HeapFree( SystemHeap, 0, DOSDrives[drive].dos_cwd );
 472     HeapFree( SystemHeap, 0, DOSDrives[drive].unix_cwd );
 473     DOSDrives[drive].dos_cwd  = HEAP_strdupA( SystemHeap, 0,
 474                                               full_name.short_name + 3 );
 475     DOSDrives[drive].unix_cwd = HEAP_strdupA( SystemHeap, 0, unix_cwd );
 476
 477     if (pTask && (pTask->curdrive & 0x80) &&
 478         ((pTask->curdrive & ~0x80) == drive))
 479     {
 480         lstrcpyn32A( pTask->curdir, full_name.short_name + 2,
 481                      sizeof(pTask->curdir) );
 482         DRIVE_LastTask = GetCurrentTask();
 483     }
 484     return 1;
 485 }
 486
 487
 488 /***********************************************************************
 489  *           DRIVE_Disable
 490  */
 491 int DRIVE_Disable( int drive  )
 492 {
 493     if ((drive < 0) || (drive >= MAX_DOS_DRIVES) || !DOSDrives[drive].root)
 494     {
 495         DOS_ERROR( ER_InvalidDrive, EC_MediaError, SA_Abort, EL_Disk );
 496         return 0;
 497     }
 498     DOSDrives[drive].flags |= DRIVE_DISABLED;
 499     return 1;
 500 }
 501
 502
 503 /***********************************************************************
 504  *           DRIVE_Enable
 505  */
 506 int DRIVE_Enable( int drive  )
 507 {
 508     if ((drive < 0) || (drive >= MAX_DOS_DRIVES) || !DOSDrives[drive].root)
 509     {
 510         DOS_ERROR( ER_InvalidDrive, EC_MediaError, SA_Abort, EL_Disk );
 511         return 0;
 512     }
 513     DOSDrives[drive].flags &= ~DRIVE_DISABLED;
 514     return 1;
 515 }
 516
 517
 518 /***********************************************************************
 519  *           DRIVE_SetLogicalMapping
 520  */
 521 int DRIVE_SetLogicalMapping ( int existing_drive, int new_drive )
 522 {
 523  /* If new_drive is already valid, do nothing and return 0
 524     otherwise, copy DOSDrives[existing_drive] to DOSDrives[new_drive] */
 525
 526     DOSDRIVE *old, *new;
 527
 528     old = DOSDrives + existing_drive;
 529     new = DOSDrives + new_drive;
 530
 531     if ((existing_drive < 0) || (existing_drive >= MAX_DOS_DRIVES) ||
 532         !old->root ||
 533         (new_drive < 0) || (new_drive >= MAX_DOS_DRIVES))
 534     {
 535         DOS_ERROR( ER_InvalidDrive, EC_MediaError, SA_Abort, EL_Disk );
 536         return 0;
 537     }
 538
 539     if ( new->root )
 540     {
 541         TRACE(dosfs, "Can\'t map drive %c to drive %c - "
 542                                 "drive %c already exists\n",
 543                         'A' + existing_drive, 'A' + new_drive,
 544                         'A' + new_drive );
 545         /* it is already mapped there, so return success */
 546         if (!strcmp(old->root,new->root))
 547             return 1;
 548         return 0;
 549     }
 550
 551     new->root = HEAP_strdupA( SystemHeap, 0, old->root );
 552     new->dos_cwd = HEAP_strdupA( SystemHeap, 0, old->dos_cwd );
 553     new->unix_cwd = HEAP_strdupA( SystemHeap, 0, old->unix_cwd );
 554     memcpy ( new->label, old->label, 12 );
 555     new->serial = old->serial;
 556     new->type = old->type;
 557     new->flags = old->flags;
 558     new->dev = old->dev;
 559     new->ino = old->ino;
 560
 561     TRACE(dosfs, "Drive %c is now equal to drive %c\n",
 562                     'A' + new_drive, 'A' + existing_drive );
 563
 564     return 1;
 565 }
 566
 567
 568 /***********************************************************************
 569  *           DRIVE_OpenDevice
 570  *
 571  * Open the drive raw device and return a Unix fd (or -1 on error).
 572  */
 573 int DRIVE_OpenDevice( int drive, int flags )
 574 {
 575     if (!DRIVE_IsValid( drive )) return -1;
 576     return open( DOSDrives[drive].device, flags );
 577 }
 578
 579
 580 /***********************************************************************
 581  *           DRIVE_GetFreeSpace
 582  */
 583 static int DRIVE_GetFreeSpace( int drive, DWORD *size, DWORD *available )
 584 {
 585     struct statfs info;
 586
 587     if (!DRIVE_IsValid(drive))
 588     {
 589         DOS_ERROR( ER_InvalidDrive, EC_MediaError, SA_Abort, EL_Disk );
 590         return 0;
 591     }
 592
 593 /* FIXME: add autoconf check for this */
 594 #if defined(__svr4__) || defined(_SCO_DS)
 595     if (statfs( DOSDrives[drive].root, &info, 0, 0) < 0)
 596 #else
 597     if (statfs( DOSDrives[drive].root, &info) < 0)
 598 #endif
 599     {
 600         FILE_SetDosError();
 601         WARN(dosfs, "cannot do statfs(%s)\n", DOSDrives[drive].root);
 602         return 0;
 603     }
 604
 605     *size = info.f_bsize * info.f_blocks;
 606 #ifdef STATFS_HAS_BAVAIL
 607     *available = info.f_bavail * info.f_bsize;
 608 #else
 609 # ifdef STATFS_HAS_BFREE
 610     *available = info.f_bfree * info.f_bsize;
 611 # else
 612 #  error "statfs has no bfree/bavail member!"
 613 # endif
 614 #endif
 615     return 1;
 616 }
 617
 618
 619 /***********************************************************************
 620  *           GetDiskFreeSpace16   (KERNEL.422)
 621  */
 622 BOOL16 WINAPI GetDiskFreeSpace16( LPCSTR root, LPDWORD cluster_sectors,
 623                                   LPDWORD sector_bytes, LPDWORD free_clusters,
 624                                   LPDWORD total_clusters )
 625 {
 626     return GetDiskFreeSpace32A( root, cluster_sectors, sector_bytes,
 627                                 free_clusters, total_clusters );
 628 }
 629
 630
 631 /***********************************************************************
 632  *           GetDiskFreeSpace32A   (KERNEL32.206)
 633  */
 634 BOOL32 WINAPI GetDiskFreeSpace32A( LPCSTR root, LPDWORD cluster_sectors,
 635                                    LPDWORD sector_bytes, LPDWORD free_clusters,
 636                                    LPDWORD total_clusters )
 637 {
 638     int drive;
 639     DWORD size,available;
 640
 641     if (!root) drive = DRIVE_GetCurrentDrive();
 642     else
 643     {
 644         if ((root[1]) && ((root[1] != ':') || (root[2] != '\\')))
 645         {
 646             WARN(dosfs, "invalid root '%s'\n", root );
 647             return FALSE;
 648         }
 649         drive = toupper(root[0]) - 'A';
 650     }
 651     if (!DRIVE_GetFreeSpace(drive, &size, &available)) return FALSE;
 652
 653     /* Cap the size and available at 2GB as per specs.  */
 654     if (size > 0x7fffffff) size = 0x7fffffff;
 655     if (available > 0x7fffffff) available = 0x7fffffff;
 656
 657     if (DRIVE_GetType(drive)==TYPE_CDROM) {
 658         *sector_bytes    = 2048;
 659         size            /= 2048;
 660         available       /= 2048;
 661     } else {
 662         *sector_bytes    = 512;
 663         size            /= 512;
 664         available       /= 512;
 665     }
 666     /* fixme: probably have to adjust those variables too for CDFS */
 667     *cluster_sectors = 1;
 668     while (*cluster_sectors * 65536 < size) *cluster_sectors *= 2;
 669     *free_clusters   = available/ *cluster_sectors;
 670     *total_clusters  = size/ *cluster_sectors;
 671     return TRUE;
 672 }
 673
 674
 675 /***********************************************************************
 676  *           GetDiskFreeSpace32W   (KERNEL32.207)
 677  */
 678 BOOL32 WINAPI GetDiskFreeSpace32W( LPCWSTR root, LPDWORD cluster_sectors,
 679                                    LPDWORD sector_bytes, LPDWORD free_clusters,
 680                                    LPDWORD total_clusters )
 681 {
 682     LPSTR xroot;
 683     BOOL32 ret;
 684
 685     xroot = HEAP_strdupWtoA( GetProcessHeap(), 0, root);
 686     ret = GetDiskFreeSpace32A( xroot,cluster_sectors, sector_bytes,
 687                                free_clusters, total_clusters );
 688     HeapFree( GetProcessHeap(), 0, xroot );
 689     return ret;
 690 }
 691
 692
 693 /***********************************************************************
 694  *           GetDiskFreeSpaceEx32A   (KERNEL32.871)
 695  */
 696 BOOL32 WINAPI GetDiskFreeSpaceEx32A( LPCSTR root,
 697                                      LPULARGE_INTEGER avail,
 698                                      LPULARGE_INTEGER total,
 699                                      LPULARGE_INTEGER totalfree)
 700 {
 701     int drive;
 702     DWORD size,available;
 703
 704     if (!root) drive = DRIVE_GetCurrentDrive();
 705     else
 706     {
 707         if ((root[1]) && ((root[1] != ':') || (root[2] != '\\')))
 708         {
 709             WARN(dosfs, "invalid root '%s'\n", root );
 710             return FALSE;
 711         }
 712         drive = toupper(root[0]) - 'A';
 713     }
 714     if (!DRIVE_GetFreeSpace(drive, &size, &available)) return FALSE;
 715     /*FIXME: Do we have the number of bytes available to the user? */
 716     avail->HighPart = total->HighPart = 0;
 717     avail->LowPart = available;
 718     total->LowPart = size;
 719     if(totalfree)
 720       {
 721         totalfree->HighPart =0;
 722         totalfree->LowPart=  available;
 723       }
 724     return TRUE;
 725 }
 726
 727 /***********************************************************************
 728  *           GetDiskFreeSpaceEx32W   (KERNEL32.873)
 729  */
 730 BOOL32 WINAPI GetDiskFreeSpaceEx32W( LPCWSTR root, LPULARGE_INTEGER avail,
 731                                      LPULARGE_INTEGER total,
 732                                      LPULARGE_INTEGER  totalfree)
 733 {
 734     LPSTR xroot;
 735     BOOL32 ret;
 736
 737     xroot = HEAP_strdupWtoA( GetProcessHeap(), 0, root);
 738     ret = GetDiskFreeSpaceEx32A( xroot, avail, total, totalfree);
 739     HeapFree( GetProcessHeap(), 0, xroot );
 740     return ret;
 741 }
 742
 743 /***********************************************************************
 744  *           GetDriveType16   (KERNEL.136)
 745  */
 746 UINT16 WINAPI GetDriveType16( UINT16 drive )
 747 {
 748     TRACE(dosfs, "(%c:)\n", 'A' + drive );
 749     switch(DRIVE_GetType(drive))
 750     {
 751     case TYPE_FLOPPY:  return DRIVE_REMOVABLE;
 752     case TYPE_HD:      return DRIVE_FIXED;
 753     case TYPE_CDROM:   return DRIVE_REMOVABLE;
 754     case TYPE_NETWORK: return DRIVE_REMOTE;
 755     case TYPE_INVALID:
 756     default:           return DRIVE_CANNOTDETERMINE;
 757     }
 758 }
 759
 760
 761 /***********************************************************************
 762  *           GetDriveType32A   (KERNEL32.208)
 763  */
 764 UINT32 WINAPI GetDriveType32A( LPCSTR root )
 765 {
 766     TRACE(dosfs, "(%s)\n", root );
 767     if ((root[1]) && (root[1] != ':'))
 768     {
 769         WARN(dosfs, "invalid root '%s'\n", root );
 770         return DRIVE_DOESNOTEXIST;
 771     }
 772     switch(DRIVE_GetType(toupper(root[0]) - 'A'))
 773     {
 774     case TYPE_FLOPPY:  return DRIVE_REMOVABLE;
 775     case TYPE_HD:      return DRIVE_FIXED;
 776     case TYPE_CDROM:   return DRIVE_CDROM;
 777     case TYPE_NETWORK: return DRIVE_REMOTE;
 778     case TYPE_INVALID:
 779     default:           return DRIVE_CANNOTDETERMINE;
 780     }
 781 }
 782
 783
 784 /***********************************************************************
 785  *           GetDriveType32W   (KERNEL32.209)
 786  */
 787 UINT32 WINAPI GetDriveType32W( LPCWSTR root )
 788 {
 789     LPSTR xpath = HEAP_strdupWtoA( GetProcessHeap(), 0, root );
 790     UINT32 ret = GetDriveType32A( xpath );
 791     HeapFree( GetProcessHeap(), 0, xpath );
 792     return ret;
 793 }
 794
 795
 796 /***********************************************************************
 797  *           GetCurrentDirectory16   (KERNEL.411)
 798  */
 799 UINT16 WINAPI GetCurrentDirectory16( UINT16 buflen, LPSTR buf )
 800 {
 801     return (UINT16)GetCurrentDirectory32A( buflen, buf );
 802 }
 803
 804
 805 /***********************************************************************
 806  *           GetCurrentDirectory32A   (KERNEL32.196)
 807  *
 808  * Returns "X:\\path\\etc\\".
 809  *
 810  * Despite the API description, return required length including the
 811  * terminating null when buffer too small. This is the real behaviour.
 812  */
 813 UINT32 WINAPI GetCurrentDirectory32A( UINT32 buflen, LPSTR buf )
 814 {
 815     UINT32 ret;
 816     const char *s = DRIVE_GetDosCwd( DRIVE_GetCurrentDrive() );
 817
 818     assert(s);
 819     ret = strlen(s) + 3; /* length of WHOLE current directory */
 820     if (ret >= buflen) return ret + 1;
 821     lstrcpyn32A( buf, "A:\\", MIN( 4, buflen ) );
 822     if (buflen) buf[0] += DRIVE_GetCurrentDrive();
 823     if (buflen > 3) lstrcpyn32A( buf + 3, s, buflen - 3 );
 824     return ret;
 825 }
 826
 827
 828 /***********************************************************************
 829  *           GetCurrentDirectory32W   (KERNEL32.197)
 830  */
 831 UINT32 WINAPI GetCurrentDirectory32W( UINT32 buflen, LPWSTR buf )
 832 {
 833     LPSTR xpath = HeapAlloc( GetProcessHeap(), 0, buflen+1 );
 834     UINT32 ret = GetCurrentDirectory32A( buflen, xpath );
 835     lstrcpyAtoW( buf, xpath );
 836     HeapFree( GetProcessHeap(), 0, xpath );
 837     return ret;
 838 }
 839
 840
 841 /***********************************************************************
 842  *           SetCurrentDirectory   (KERNEL.412)
 843  */
 844 BOOL16 WINAPI SetCurrentDirectory16( LPCSTR dir )
 845 {
 846     return SetCurrentDirectory32A( dir );
 847 }
 848
 849
 850 /***********************************************************************
 851  *           SetCurrentDirectory32A   (KERNEL32.479)
 852  */
 853 BOOL32 WINAPI SetCurrentDirectory32A( LPCSTR dir )
 854 {
 855     int drive = DRIVE_GetCurrentDrive();
 856
 857     if (!dir) {
 858         ERR(file,"(NULL)!\n");
 859         return FALSE;
 860     }
 861     if (dir[0] && (dir[1]==':'))
 862     {
 863         drive = tolower( *dir ) - 'a';
 864         if (!DRIVE_IsValid( drive ))
 865         {
 866             DOS_ERROR( ER_InvalidDrive, EC_MediaError, SA_Abort, EL_Disk );
 867             return FALSE;
 868         }
 869         dir += 2;
 870     }
 871     /* FIXME: what about empty strings? Add a \\ ? */
 872     if (!DRIVE_Chdir( drive, dir )) return FALSE;
 873     if (drive == DRIVE_GetCurrentDrive()) return TRUE;
 874     return DRIVE_SetCurrentDrive( drive );
 875 }
 876
 877
 878 /***********************************************************************
 879  *           SetCurrentDirectory32W   (KERNEL32.480)
 880  */
 881 BOOL32 WINAPI SetCurrentDirectory32W( LPCWSTR dirW )
 882 {
 883     LPSTR dir = HEAP_strdupWtoA( GetProcessHeap(), 0, dirW );
 884     BOOL32 res = SetCurrentDirectory32A( dir );
 885     HeapFree( GetProcessHeap(), 0, dir );
 886     return res;
 887 }
 888
 889
 890 /***********************************************************************
 891  *           GetLogicalDriveStrings32A   (KERNEL32.231)
 892  */
 893 UINT32 WINAPI GetLogicalDriveStrings32A( UINT32 len, LPSTR buffer )
 894 {
 895     int drive, count;
 896
 897     for (drive = count = 0; drive < MAX_DOS_DRIVES; drive++)
 898         if (DRIVE_IsValid(drive)) count++;
 899     if (count * 4 * sizeof(char) <= len)
 900     {
 901         LPSTR p = buffer;
 902         for (drive = 0; drive < MAX_DOS_DRIVES; drive++)
 903             if (DRIVE_IsValid(drive))
 904             {
 905                 *p++ = 'a' + drive;
 906                 *p++ = ':';
 907                 *p++ = '\\';
 908                 *p++ = '\0';
 909             }
 910         *p = '\0';
 911     }
 912     return count * 4 * sizeof(char);
 913 }
 914
 915
 916 /***********************************************************************
 917  *           GetLogicalDriveStrings32W   (KERNEL32.232)
 918  */
 919 UINT32 WINAPI GetLogicalDriveStrings32W( UINT32 len, LPWSTR buffer )
 920 {
 921     int drive, count;
 922
 923     for (drive = count = 0; drive < MAX_DOS_DRIVES; drive++)
 924         if (DRIVE_IsValid(drive)) count++;
 925     if (count * 4 * sizeof(WCHAR) <= len)
 926     {
 927         LPWSTR p = buffer;
 928         for (drive = 0; drive < MAX_DOS_DRIVES; drive++)
 929             if (DRIVE_IsValid(drive))
 930             {
 931                 *p++ = (WCHAR)('a' + drive);
 932                 *p++ = (WCHAR)':';
 933                 *p++ = (WCHAR)'\\';
 934                 *p++ = (WCHAR)'\0';
 935             }
 936         *p = (WCHAR)'\0';
 937     }
 938     return count * 4 * sizeof(WCHAR);
 939 }
 940
 941
 942 /***********************************************************************
 943  *           GetLogicalDrives   (KERNEL32.233)
 944  */
 945 DWORD WINAPI GetLogicalDrives(void)
 946 {
 947     DWORD ret = 0;
 948     int drive;
 949
 950     for (drive = 0; drive < MAX_DOS_DRIVES; drive++)
 951         if (DRIVE_IsValid(drive)) ret |= (1 << drive);
 952     return ret;
 953 }
 954
 955
 956 /***********************************************************************
 957  *           GetVolumeInformation32A   (KERNEL32.309)
 958  */
 959 BOOL32 WINAPI GetVolumeInformation32A( LPCSTR root, LPSTR label,
 960                                        DWORD label_len, DWORD *serial,
 961                                        DWORD *filename_len, DWORD *flags,
 962                                        LPSTR fsname, DWORD fsname_len )
 963 {
 964     int drive;
 965
 966     /* FIXME, SetLastErrors missing */
 967
 968     if (!root) drive = DRIVE_GetCurrentDrive();
 969     else
 970     {
 971         if ((root[1]) && (root[1] != ':'))
 972         {
 973             WARN(dosfs, "invalid root '%s'\n",root);
 974             return FALSE;
 975         }
 976         drive = toupper(root[0]) - 'A';
 977     }
 978     if (!DRIVE_IsValid( drive )) return FALSE;
 979     if (label) lstrcpyn32A( label, DOSDrives[drive].label, label_len );
 980     if (serial) *serial = DOSDrives[drive].serial;
 981
 982     /* Set the filesystem information */
 983     /* Note: we only emulate a FAT fs at the present */
 984
 985     if (filename_len) *filename_len = 12;
 986     if (flags) *flags = 0;
 987     if (fsname) {
 988         /* Diablo checks that return code ... */
 989         if (DRIVE_GetType(drive)==TYPE_CDROM)
 990             lstrcpyn32A( fsname, "CDFS", fsname_len );
 991         else
 992             lstrcpyn32A( fsname, "FAT", fsname_len );
 993     }
 994     return TRUE;
 995 }
 996
 997
 998 /***********************************************************************
 999  *           GetVolumeInformation32W   (KERNEL32.310)
1000  */
1001 BOOL32 WINAPI GetVolumeInformation32W( LPCWSTR root, LPWSTR label,
1002                                        DWORD label_len, DWORD *serial,
1003                                        DWORD *filename_len, DWORD *flags,
1004                                        LPWSTR fsname, DWORD fsname_len )
1005 {
1006     LPSTR xroot    = HEAP_strdupWtoA( GetProcessHeap(), 0, root );
1007     LPSTR xvolname = label ? HeapAlloc(GetProcessHeap(),0,label_len) : NULL;
1008     LPSTR xfsname  = fsname ? HeapAlloc(GetProcessHeap(),0,fsname_len) : NULL;
1009     BOOL32 ret = GetVolumeInformation32A( xroot, xvolname, label_len, serial,
1010                                           filename_len, flags, xfsname,
1011                                           fsname_len );
1012     if (ret)
1013     {
1014         if (label) lstrcpyAtoW( label, xvolname );
1015         if (fsname) lstrcpyAtoW( fsname, xfsname );
1016     }
1017     HeapFree( GetProcessHeap(), 0, xroot );
1018     HeapFree( GetProcessHeap(), 0, xvolname );
1019     HeapFree( GetProcessHeap(), 0, xfsname );
1020     return ret;
1021 }