[USETUP] Use correct flags for the NtCreateFile call that creates new file directories.
[reactos.git] / base / setup / usetup / filesup.c
index 030a4ff..e0bd0b1 100644 (file)
@@ -1,27 +1,9 @@
 /*
- *  ReactOS kernel
- *  Copyright (C) 2002 ReactOS Team
- *
- *  This program is free software; you can redistribute it and/or modify
- *  it under the terms of the GNU General Public License as published by
- *  the Free Software Foundation; either version 2 of the License, or
- *  (at your option) any later version.
- *
- *  This program 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 General Public License for more details.
- *
- *  You should have received a copy of the GNU General Public License along
- *  with this program; if not, write to the Free Software Foundation, Inc.,
- *  51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
- */
-/* COPYRIGHT:       See COPYING in the top level directory
+ * COPYRIGHT:       See COPYING in the top level directory
  * PROJECT:         ReactOS text-mode setup
- * FILE:            subsys/system/usetup/filesup.c
+ * FILE:            base/setup/usetup/filesup.c
  * PURPOSE:         File support functions
- * PROGRAMMER:      Eric Kohl
- *                  Casper S. Hornstrup (chorns@users.sourceforge.net)
+ * PROGRAMMER:      Casper S. Hornstrup (chorns@users.sourceforge.net)
  */
 
 /* INCLUDES *****************************************************************/
@@ -37,359 +19,488 @@ static BOOLEAN HasCurrentCabinet = FALSE;
 static WCHAR CurrentCabinetName[MAX_PATH];
 static CAB_SEARCH Search;
 
+static
 NTSTATUS
-SetupCreateDirectory(PWCHAR DirectoryName)
+SetupCreateSingleDirectory(
+    PWCHAR DirectoryName)
 {
-  OBJECT_ATTRIBUTES ObjectAttributes;
-  IO_STATUS_BLOCK IoStatusBlock;
-  UNICODE_STRING PathName;
-  HANDLE DirectoryHandle;
-  NTSTATUS Status;
-
-  RtlCreateUnicodeString(&PathName,
-                        DirectoryName);
-  if (PathName.Length > sizeof(WCHAR) &&
-      PathName.Buffer[PathName.Length / sizeof(WCHAR) - 2] == L'\\' &&
-      PathName.Buffer[PathName.Length / sizeof(WCHAR) - 1] == L'.')
+    OBJECT_ATTRIBUTES ObjectAttributes;
+    IO_STATUS_BLOCK IoStatusBlock;
+    UNICODE_STRING PathName;
+    HANDLE DirectoryHandle;
+    NTSTATUS Status;
+
+    if (!RtlCreateUnicodeString(&PathName, DirectoryName))
+        return STATUS_NO_MEMORY;
+
+    if (PathName.Length > sizeof(WCHAR) &&
+        PathName.Buffer[PathName.Length / sizeof(WCHAR) - 2] == L'\\' &&
+        PathName.Buffer[PathName.Length / sizeof(WCHAR) - 1] == L'.')
     {
-       PathName.Length -= sizeof(WCHAR);
-       PathName.Buffer[PathName.Length / sizeof(WCHAR)] = 0;
+        PathName.Length -= sizeof(WCHAR);
+        PathName.Buffer[PathName.Length / sizeof(WCHAR)] = 0;
     }
 
-  if (PathName.Length > sizeof(WCHAR) &&
-      PathName.Buffer[PathName.Length / sizeof(WCHAR) - 1] == L'\\')
+    if (PathName.Length > sizeof(WCHAR) &&
+        PathName.Buffer[PathName.Length / sizeof(WCHAR) - 1] == L'\\')
     {
-      PathName.Length -= sizeof(WCHAR);
-      PathName.Buffer[PathName.Length / sizeof(WCHAR)] = 0;
-   }
-
-  InitializeObjectAttributes(&ObjectAttributes,
-                            &PathName,
-                            OBJ_CASE_INSENSITIVE | OBJ_INHERIT,
-                            NULL,
-                            NULL);
-
-  Status = NtCreateFile(&DirectoryHandle,
-                       DIRECTORY_ALL_ACCESS,
-                       &ObjectAttributes,
-                       &IoStatusBlock,
-                       NULL,
-                       FILE_ATTRIBUTE_DIRECTORY,
-                       FILE_SHARE_READ | FILE_SHARE_WRITE,
-                       FILE_OPEN_IF,
-                       FILE_DIRECTORY_FILE,
-                       NULL,
-                       0);
-  if (NT_SUCCESS(Status))
+        PathName.Length -= sizeof(WCHAR);
+        PathName.Buffer[PathName.Length / sizeof(WCHAR)] = 0;
+    }
+
+    InitializeObjectAttributes(&ObjectAttributes,
+                               &PathName,
+                               OBJ_CASE_INSENSITIVE | OBJ_INHERIT,
+                               NULL,
+                               NULL);
+
+    Status = NtCreateFile(&DirectoryHandle,
+                          FILE_LIST_DIRECTORY | SYNCHRONIZE,
+                          &ObjectAttributes,
+                          &IoStatusBlock,
+                          NULL,
+                          FILE_ATTRIBUTE_DIRECTORY,
+                          FILE_SHARE_READ | FILE_SHARE_WRITE,
+                          FILE_OPEN_IF,
+                          FILE_OPEN_FOR_BACKUP_INTENT | FILE_DIRECTORY_FILE,
+                          NULL,
+                          0);
+    if (NT_SUCCESS(Status))
     {
-      NtClose(DirectoryHandle);
+        NtClose(DirectoryHandle);
     }
 
-  RtlFreeUnicodeString(&PathName);
+    RtlFreeUnicodeString(&PathName);
 
-  return(Status);
+    return Status;
 }
 
+NTSTATUS
+SetupCreateDirectory(
+    PWCHAR PathName)
+{
+    PWCHAR PathBuffer = NULL;
+    PWCHAR Ptr, EndPtr;
+    ULONG BackslashCount;
+    ULONG Size;
+    NTSTATUS Status = STATUS_SUCCESS;
+
+    Size = (wcslen(PathName) + 1) * sizeof(WCHAR);
+    PathBuffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, Size);
+    if (PathBuffer == NULL)
+        return STATUS_INSUFFICIENT_RESOURCES;
+
+    wcscpy(PathBuffer, PathName);
+    EndPtr = PathBuffer + wcslen(PathName);
+
+    Ptr = PathBuffer;
+
+    /* Skip the '\Device\HarddiskX\PartitionY\ part */
+    BackslashCount = 0;
+    while (Ptr < EndPtr && BackslashCount < 4)
+    {
+        if (*Ptr == L'\\')
+            BackslashCount++;
+
+        Ptr++;
+    }
+
+    while (Ptr < EndPtr)
+    {
+        if (*Ptr == L'\\')
+        {
+            *Ptr = 0;
+
+            DPRINT("PathBuffer: %S\n", PathBuffer);
+            if (!DoesPathExist(NULL, PathBuffer))
+            {
+                DPRINT("Create: %S\n", PathBuffer);
+                Status = SetupCreateSingleDirectory(PathBuffer);
+                if (!NT_SUCCESS(Status))
+                    goto done;
+            }
+
+            *Ptr = L'\\';
+        }
+
+        Ptr++;
+    }
+
+    if (!DoesPathExist(NULL, PathBuffer))
+    {
+        DPRINT("Create: %S\n", PathBuffer);
+        Status = SetupCreateSingleDirectory(PathBuffer);
+        if (!NT_SUCCESS(Status))
+            goto done;
+    }
+
+done:
+    DPRINT("Done.\n");
+    if (PathBuffer != NULL)
+        RtlFreeHeap(RtlGetProcessHeap(), 0, PathBuffer);
+
+    return Status;
+}
 
 NTSTATUS
-SetupCopyFile(PWCHAR SourceFileName,
-             PWCHAR DestinationFileName)
+SetupCopyFile(
+    PWCHAR SourceFileName,
+    PWCHAR DestinationFileName)
 {
-  OBJECT_ATTRIBUTES ObjectAttributes;
-  HANDLE FileHandleSource;
-  HANDLE FileHandleDest;
-  static IO_STATUS_BLOCK IoStatusBlock;
-  FILE_STANDARD_INFORMATION FileStandard;
-  FILE_BASIC_INFORMATION FileBasic;
-  ULONG RegionSize;
-  UNICODE_STRING FileName;
-  NTSTATUS Status;
-  PVOID SourceFileMap = 0;
-  HANDLE SourceFileSection;
-  SIZE_T SourceSectionSize = 0;
-  LARGE_INTEGER ByteOffset;
-
-#ifdef __REACTOS__
-  RtlInitUnicodeString(&FileName,
-                      SourceFileName);
-
-  InitializeObjectAttributes(&ObjectAttributes,
-                            &FileName,
-                            OBJ_CASE_INSENSITIVE,
-                            NULL,
-                            NULL);
-
-  Status = NtOpenFile(&FileHandleSource,
-                     GENERIC_READ,
-                     &ObjectAttributes,
-                     &IoStatusBlock,
-                     FILE_SHARE_READ,
-                     FILE_SEQUENTIAL_ONLY);
-  if(!NT_SUCCESS(Status))
+    OBJECT_ATTRIBUTES ObjectAttributes;
+    HANDLE FileHandleSource;
+    HANDLE FileHandleDest;
+    static IO_STATUS_BLOCK IoStatusBlock;
+    FILE_STANDARD_INFORMATION FileStandard;
+    FILE_BASIC_INFORMATION FileBasic;
+    ULONG RegionSize;
+    UNICODE_STRING FileName;
+    NTSTATUS Status;
+    PVOID SourceFileMap = 0;
+    HANDLE SourceFileSection;
+    SIZE_T SourceSectionSize = 0;
+    LARGE_INTEGER ByteOffset;
+
+    RtlInitUnicodeString(&FileName,
+                         SourceFileName);
+
+    InitializeObjectAttributes(&ObjectAttributes,
+                               &FileName,
+                               OBJ_CASE_INSENSITIVE,
+                               NULL,
+                               NULL);
+
+    Status = NtOpenFile(&FileHandleSource,
+                        GENERIC_READ,
+                        &ObjectAttributes,
+                        &IoStatusBlock,
+                        FILE_SHARE_READ,
+                        FILE_SEQUENTIAL_ONLY);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtOpenFile failed: %x, %wZ\n", Status, &FileName);
-      goto done;
+        DPRINT1("NtOpenFile failed: %x, %wZ\n", Status, &FileName);
+        goto done;
     }
-#else
-  FileHandleSource = CreateFileW(SourceFileName, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
-  if (FileHandleSource == INVALID_HANDLE_VALUE)
-  {
-    Status = STATUS_UNSUCCESSFUL;
-    goto done;
-  }
-#endif
-
-  Status = NtQueryInformationFile(FileHandleSource,
-                                 &IoStatusBlock,
-                                 &FileStandard,
-                                 sizeof(FILE_STANDARD_INFORMATION),
-                                 FileStandardInformation);
-  if(!NT_SUCCESS(Status))
+
+    Status = NtQueryInformationFile(FileHandleSource,
+                                    &IoStatusBlock,
+                                    &FileStandard,
+                                    sizeof(FILE_STANDARD_INFORMATION),
+                                    FileStandardInformation);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtQueryInformationFile failed: %x\n", Status);
-      goto closesrc;
+        DPRINT1("NtQueryInformationFile failed: %x\n", Status);
+        goto closesrc;
     }
-  Status = NtQueryInformationFile(FileHandleSource,
-                                 &IoStatusBlock,&FileBasic,
-                                 sizeof(FILE_BASIC_INFORMATION),
-                                 FileBasicInformation);
-  if(!NT_SUCCESS(Status))
+
+    Status = NtQueryInformationFile(FileHandleSource,
+                                    &IoStatusBlock,&FileBasic,
+                                    sizeof(FILE_BASIC_INFORMATION),
+                                    FileBasicInformation);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtQueryInformationFile failed: %x\n", Status);
-      goto closesrc;
+        DPRINT1("NtQueryInformationFile failed: %x\n", Status);
+        goto closesrc;
     }
 
-  Status = NtCreateSection( &SourceFileSection,
-                           SECTION_MAP_READ,
-                           NULL,
-                           NULL,
-                           PAGE_READONLY,
-                           SEC_COMMIT,
-                           FileHandleSource);
-  if(!NT_SUCCESS(Status))
+    Status = NtCreateSection(&SourceFileSection,
+                             SECTION_MAP_READ,
+                             NULL,
+                             NULL,
+                             PAGE_READONLY,
+                             SEC_COMMIT,
+                             FileHandleSource);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtCreateSection failed: %x, %wZ\n", Status, SourceFileName);
+      DPRINT1("NtCreateSection failed: %x, %S\n", Status, SourceFileName);
       goto closesrc;
     }
 
-  Status = NtMapViewOfSection( SourceFileSection,
-                              NtCurrentProcess(),
-                              &SourceFileMap,
-                              0,
-                              0,
-                              NULL,
-                              &SourceSectionSize,
-                              ViewUnmap,
-                              0,
-                              PAGE_READONLY );
-  if(!NT_SUCCESS(Status))
+    Status = NtMapViewOfSection(SourceFileSection,
+                                NtCurrentProcess(),
+                                &SourceFileMap,
+                                0,
+                                0,
+                                NULL,
+                                &SourceSectionSize,
+                                ViewUnmap,
+                                0,
+                                PAGE_READONLY );
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtMapViewOfSection failed: %x, %wZ\n", Status, SourceFileName);
-      goto closesrcsec;
+        DPRINT1("NtMapViewOfSection failed: %x, %S\n", Status, SourceFileName);
+        goto closesrcsec;
     }
 
-  RtlInitUnicodeString(&FileName,
-                      DestinationFileName);
-
-  InitializeObjectAttributes(&ObjectAttributes,
-                            &FileName,
-                            OBJ_CASE_INSENSITIVE,
-                            NULL,
-                            NULL);
-
-  Status = NtCreateFile(&FileHandleDest,
-                       GENERIC_WRITE | SYNCHRONIZE,
-                       &ObjectAttributes,
-                       &IoStatusBlock,
-                       NULL,
-                       FILE_ATTRIBUTE_NORMAL,
-                       0,
-                       FILE_OVERWRITE_IF,
-                       FILE_NO_INTERMEDIATE_BUFFERING |
-                       FILE_SEQUENTIAL_ONLY |
-                       FILE_SYNCHRONOUS_IO_NONALERT,
-                       NULL,
-                       0);
-  if(!NT_SUCCESS(Status))
+    RtlInitUnicodeString(&FileName,
+                         DestinationFileName);
+
+    InitializeObjectAttributes(&ObjectAttributes,
+                               &FileName,
+                               OBJ_CASE_INSENSITIVE,
+                               NULL,
+                               NULL);
+
+    Status = NtCreateFile(&FileHandleDest,
+                          GENERIC_WRITE | SYNCHRONIZE,
+                          &ObjectAttributes,
+                          &IoStatusBlock,
+                          NULL,
+                          FILE_ATTRIBUTE_NORMAL,
+                          0,
+                          FILE_OVERWRITE_IF,
+                          FILE_NO_INTERMEDIATE_BUFFERING |
+                          FILE_SEQUENTIAL_ONLY |
+                          FILE_SYNCHRONOUS_IO_NONALERT,
+                          NULL,
+                          0);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtCreateFile failed: %x\n", Status);
-      goto unmapsrcsec;
+        /* Open may have failed because the file to overwrite
+         * is in readonly mode
+         */
+        if (Status == STATUS_ACCESS_DENIED)
+        {
+            FILE_BASIC_INFORMATION FileBasicInfo;
+
+            /* Reattempt to open it with limited access */
+            Status = NtCreateFile(&FileHandleDest,
+                                  FILE_WRITE_ATTRIBUTES | SYNCHRONIZE,
+                                  &ObjectAttributes,
+                                  &IoStatusBlock,
+                                  NULL,
+                                  FILE_ATTRIBUTE_NORMAL,
+                                  0,
+                                  FILE_OPEN,
+                                  FILE_NO_INTERMEDIATE_BUFFERING |
+                                  FILE_SEQUENTIAL_ONLY |
+                                  FILE_SYNCHRONOUS_IO_NONALERT,
+                                  NULL,
+                                  0);
+            /* Fail for real if we cannot open it that way */
+            if (!NT_SUCCESS(Status))
+            {
+                DPRINT1("NtCreateFile failed: %x, %wZ\n", Status, &FileName);
+                goto unmapsrcsec;
+            }
+
+            /* Zero our basic info, just to set attributes */
+            RtlZeroMemory(&FileBasicInfo, sizeof(FileBasicInfo));
+            /* Reset attributes to normal, no read-only */
+            FileBasicInfo.FileAttributes = FILE_ATTRIBUTE_NORMAL;
+            /* We basically don't care about whether it succeed:
+             * if it didn't, later open will fail
+             */
+            NtSetInformationFile(FileHandleDest, &IoStatusBlock, &FileBasicInfo,
+                                 sizeof(FileBasicInfo), FileBasicInformation);
+
+            /* Close file */
+            NtClose(FileHandleDest);
+
+            /* And re-attempt overwrite */
+            Status = NtCreateFile(&FileHandleDest,
+                                  GENERIC_WRITE | SYNCHRONIZE,
+                                  &ObjectAttributes,
+                                  &IoStatusBlock,
+                                  NULL,
+                                  FILE_ATTRIBUTE_NORMAL,
+                                  0,
+                                  FILE_OVERWRITE_IF,
+                                  FILE_NO_INTERMEDIATE_BUFFERING |
+                                  FILE_SEQUENTIAL_ONLY |
+                                  FILE_SYNCHRONOUS_IO_NONALERT,
+                                  NULL,
+                                  0);
+        }
+
+        /* We failed */
+        if (!NT_SUCCESS(Status))
+        {
+            DPRINT1("NtCreateFile failed: %x, %wZ\n", Status, &FileName);
+            goto unmapsrcsec;
+        }
     }
 
-  RegionSize = (ULONG)PAGE_ROUND_UP(FileStandard.EndOfFile.u.LowPart);
-  IoStatusBlock.Status = 0;
-  ByteOffset.QuadPart = 0;
-  Status = NtWriteFile(FileHandleDest,
-                      NULL,
-                      NULL,
-                      NULL,
-                      &IoStatusBlock,
-                      SourceFileMap,
-                      RegionSize,
-                      &ByteOffset,
-                      NULL);
-  if(!NT_SUCCESS(Status))
+    RegionSize = (ULONG)PAGE_ROUND_UP(FileStandard.EndOfFile.u.LowPart);
+    IoStatusBlock.Status = 0;
+    ByteOffset.QuadPart = 0ULL;
+    Status = NtWriteFile(FileHandleDest,
+                         NULL,
+                         NULL,
+                         NULL,
+                         &IoStatusBlock,
+                         SourceFileMap,
+                         RegionSize,
+                         &ByteOffset,
+                         NULL);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtWriteFile failed: %x:%x, iosb: %p src: %p, size: %x\n", Status, IoStatusBlock.Status, &IoStatusBlock, SourceFileMap, RegionSize);
-      goto closedest;
+        DPRINT1("NtWriteFile failed: %x:%x, iosb: %p src: %p, size: %x\n", Status, IoStatusBlock.Status, &IoStatusBlock, SourceFileMap, RegionSize);
+        goto closedest;
     }
-  /* Copy file date/time from source file */
-  Status = NtSetInformationFile(FileHandleDest,
-                               &IoStatusBlock,
-                               &FileBasic,
-                               sizeof(FILE_BASIC_INFORMATION),
-                               FileBasicInformation);
-  if(!NT_SUCCESS(Status))
+
+    /* Copy file date/time from source file */
+    Status = NtSetInformationFile(FileHandleDest,
+                                  &IoStatusBlock,
+                                  &FileBasic,
+                                  sizeof(FILE_BASIC_INFORMATION),
+                                  FileBasicInformation);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtSetInformationFile failed: %x\n", Status);
-      goto closedest;
+        DPRINT1("NtSetInformationFile failed: %x\n", Status);
+        goto closedest;
     }
 
-  /* shorten the file back to it's real size after completing the write */
-  Status = NtSetInformationFile(FileHandleDest,
-                      &IoStatusBlock,
-                      &FileStandard.EndOfFile,
-                      sizeof(FILE_END_OF_FILE_INFORMATION),
-                      FileEndOfFileInformation);
-
-  if(!NT_SUCCESS(Status))
+    /* shorten the file back to it's real size after completing the write */
+    Status = NtSetInformationFile(FileHandleDest,
+                                  &IoStatusBlock,
+                                  &FileStandard.EndOfFile,
+                                  sizeof(FILE_END_OF_FILE_INFORMATION),
+                                  FileEndOfFileInformation);
+    if (!NT_SUCCESS(Status))
     {
-      DPRINT1("NtSetInformationFile failed: %x\n", Status);
+        DPRINT1("NtSetInformationFile failed: %x\n", Status);
     }
 
- closedest:
-  NtClose(FileHandleDest);
- unmapsrcsec:
-  NtUnmapViewOfSection( NtCurrentProcess(), SourceFileMap );
- closesrcsec:
-  NtClose(SourceFileSection);
- closesrc:
-  NtClose(FileHandleSource);
- done:
-  return(Status);
+closedest:
+    NtClose(FileHandleDest);
+
+unmapsrcsec:
+    NtUnmapViewOfSection(NtCurrentProcess(), SourceFileMap);
+
+closesrcsec:
+    NtClose(SourceFileSection);
+
+closesrc:
+    NtClose(FileHandleSource);
+
+done:
+    return Status;
 }
 
-#ifdef __REACTOS__
+
 NTSTATUS
-SetupExtractFile(PWCHAR CabinetFileName,
-        PWCHAR SourceFileName,
-             PWCHAR DestinationPathName)
+SetupExtractFile(
+    PWCHAR CabinetFileName,
+    PWCHAR SourceFileName,
+    PWCHAR DestinationPathName)
 {
-  ULONG CabStatus;
+    ULONG CabStatus;
 
-  DPRINT("SetupExtractFile(CabinetFileName %S, SourceFileName %S, DestinationPathName %S)\n",
-    CabinetFileName, SourceFileName, DestinationPathName);
+    DPRINT("SetupExtractFile(CabinetFileName %S, SourceFileName %S, DestinationPathName %S)\n",
+           CabinetFileName, SourceFileName, DestinationPathName);
 
-  if (HasCurrentCabinet)
+    if (HasCurrentCabinet)
     {
-      DPRINT("CurrentCabinetName: %S\n", CurrentCabinetName);
+        DPRINT("CurrentCabinetName: %S\n", CurrentCabinetName);
     }
 
-  if ((HasCurrentCabinet) && (wcscmp(CabinetFileName, CurrentCabinetName) == 0))
+    if ((HasCurrentCabinet) && (wcscmp(CabinetFileName, CurrentCabinetName) == 0))
     {
-      DPRINT("Using same cabinet as last time\n");
-
-      /* Use our last location because the files should be sequential */
-      CabStatus = CabinetFindNextFileSequential(SourceFileName, &Search);
-      if (CabStatus != CAB_STATUS_SUCCESS)
-      {
-          DPRINT("Sequential miss on file: %S\n", SourceFileName);
-          
-          /* Looks like we got unlucky */
-          CabStatus = CabinetFindFirst(SourceFileName, &Search);
-      }
+        DPRINT("Using same cabinet as last time\n");
+
+        /* Use our last location because the files should be sequential */
+        CabStatus = CabinetFindNextFileSequential(SourceFileName, &Search);
+        if (CabStatus != CAB_STATUS_SUCCESS)
+        {
+            DPRINT("Sequential miss on file: %S\n", SourceFileName);
+
+            /* Looks like we got unlucky */
+            CabStatus = CabinetFindFirst(SourceFileName, &Search);
+        }
     }
-  else
+    else
     {
-      DPRINT("Using new cabinet\n");
+        DPRINT("Using new cabinet\n");
 
-      if (HasCurrentCabinet)
+        if (HasCurrentCabinet)
         {
-          CabinetCleanup();
+            CabinetCleanup();
         }
 
-      wcscpy(CurrentCabinetName, CabinetFileName);
+        wcscpy(CurrentCabinetName, CabinetFileName);
 
-      CabinetInitialize();
-      CabinetSetEventHandlers(NULL, NULL, NULL);
-      CabinetSetCabinetName(CabinetFileName);
+        CabinetInitialize();
+        CabinetSetEventHandlers(NULL, NULL, NULL);
+        CabinetSetCabinetName(CabinetFileName);
 
-      CabStatus = CabinetOpen();
-      if (CabStatus == CAB_STATUS_SUCCESS)
+        CabStatus = CabinetOpen();
+        if (CabStatus == CAB_STATUS_SUCCESS)
         {
-          DPRINT("Opened cabinet %S\n", CabinetGetCabinetName());
-          HasCurrentCabinet = TRUE;
+            DPRINT("Opened cabinet %S\n", CabinetGetCabinetName());
+            HasCurrentCabinet = TRUE;
         }
-      else
+        else
         {
-          DPRINT("Cannot open cabinet (%d)\n", CabStatus);
-          return STATUS_UNSUCCESSFUL;
+            DPRINT("Cannot open cabinet (%d)\n", CabStatus);
+            return STATUS_UNSUCCESSFUL;
         }
-        
-      /* We have to start at the beginning here */
-      CabStatus = CabinetFindFirst(SourceFileName, &Search);
+
+        /* We have to start at the beginning here */
+        CabStatus = CabinetFindFirst(SourceFileName, &Search);
     }
-    
-  if (CabStatus != CAB_STATUS_SUCCESS)
-  {
-      DPRINT1("Unable to find '%S' in cabinet '%S'\n", SourceFileName, CabinetGetCabinetName());
-      return STATUS_UNSUCCESSFUL;
-  }
-
-  CabinetSetDestinationPath(DestinationPathName);
-  CabStatus = CabinetExtractFile(&Search);
-  if (CabStatus != CAB_STATUS_SUCCESS)
+
+    if (CabStatus != CAB_STATUS_SUCCESS)
     {
-      DPRINT("Cannot extract file %S (%d)\n", SourceFileName, CabStatus);
-      return STATUS_UNSUCCESSFUL;
+        DPRINT1("Unable to find '%S' in cabinet '%S'\n", SourceFileName, CabinetGetCabinetName());
+        return STATUS_UNSUCCESSFUL;
     }
 
-  return STATUS_SUCCESS;
+    CabinetSetDestinationPath(DestinationPathName);
+    CabStatus = CabinetExtractFile(&Search);
+    if (CabStatus != CAB_STATUS_SUCCESS)
+    {
+        DPRINT("Cannot extract file %S (%d)\n", SourceFileName, CabStatus);
+        return STATUS_UNSUCCESSFUL;
+    }
+
+    return STATUS_SUCCESS;
 }
-#endif
+
 
 BOOLEAN
-DoesFileExist(PWSTR PathName,
-             PWSTR FileName)
+IsValidPath(
+    IN PCWSTR InstallDir)
 {
-  OBJECT_ATTRIBUTES ObjectAttributes;
-  IO_STATUS_BLOCK IoStatusBlock;
-  UNICODE_STRING Name;
-  WCHAR FullName[MAX_PATH];
-  HANDLE FileHandle;
-  NTSTATUS Status;
-
-  wcscpy(FullName, PathName);
-  if (FileName != NULL)
+    UINT i, Length;
+
+    Length = wcslen(InstallDir);
+
+    // TODO: Add check for 8.3 too.
+
+    /* Path must be at least 2 characters long */
+//    if (Length < 2)
+//        return FALSE;
+
+    /* Path must start with a backslash */
+//    if (InstallDir[0] != L'\\')
+//        return FALSE;
+
+    /* Path must not end with a backslash */
+    if (InstallDir[Length - 1] == L'\\')
+        return FALSE;
+
+    /* Path must not contain whitespace characters */
+    for (i = 0; i < Length; i++)
     {
-      if (FileName[0] != L'\\')
-       wcscat(FullName, L"\\");
-      wcscat(FullName, FileName);
+        if (iswspace(InstallDir[i]))
+            return FALSE;
     }
 
-  RtlInitUnicodeString(&Name,
-                      FullName);
-
-  InitializeObjectAttributes(&ObjectAttributes,
-                            &Name,
-                            OBJ_CASE_INSENSITIVE,
-                            NULL,
-                            NULL);
-
-  Status = NtOpenFile(&FileHandle,
-                     GENERIC_READ | SYNCHRONIZE,
-                     &ObjectAttributes,
-                     &IoStatusBlock,
-                     0,
-                     FILE_SYNCHRONOUS_IO_NONALERT);
-  if (!NT_SUCCESS(Status))
+    /* Path component must not end with a dot */
+    for (i = 0; i < Length; i++)
     {
-      return(FALSE);
+        if (InstallDir[i] == L'\\' && i > 0)
+        {
+            if (InstallDir[i - 1] == L'.')
+                return FALSE;
+        }
     }
 
-  NtClose(FileHandle);
+    if (InstallDir[Length - 1] == L'.')
+        return FALSE;
 
-  return(TRUE);
+    return TRUE;
 }
 
 /* EOF */