/*
- * 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 *****************************************************************/
static WCHAR CurrentCabinetName[MAX_PATH];
static CAB_SEARCH Search;
+static
NTSTATUS
-SetupCreateDirectory(
+SetupCreateSingleDirectory(
PWCHAR DirectoryName)
{
OBJECT_ATTRIBUTES ObjectAttributes;
HANDLE DirectoryHandle;
NTSTATUS Status;
- RtlCreateUnicodeString(&PathName,
- DirectoryName);
+ 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'.')
NULL);
Status = NtCreateFile(&DirectoryHandle,
- DIRECTORY_ALL_ACCESS,
+ FILE_LIST_DIRECTORY | SYNCHRONIZE,
&ObjectAttributes,
&IoStatusBlock,
NULL,
FILE_ATTRIBUTE_DIRECTORY,
FILE_SHARE_READ | FILE_SHARE_WRITE,
FILE_OPEN_IF,
- FILE_DIRECTORY_FILE,
+ FILE_OPEN_FOR_BACKUP_INTENT | FILE_DIRECTORY_FILE,
NULL,
0);
if (NT_SUCCESS(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(
SIZE_T SourceSectionSize = 0;
LARGE_INTEGER ByteOffset;
-#ifdef __REACTOS__
RtlInitUnicodeString(&FileName,
SourceFileName);
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,
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;
+ ByteOffset.QuadPart = 0ULL;
Status = NtWriteFile(FileHandleDest,
NULL,
NULL,
return Status;
}
-#ifdef __REACTOS__
+
NTSTATUS
SetupExtractFile(
PWCHAR CabinetFileName,
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;
+ UINT i, Length;
- wcscpy(FullName, PathName);
- if (FileName != NULL)
- {
- if (FileName[0] != L'\\')
- wcscat(FullName, L"\\");
- wcscat(FullName, FileName);
- }
+ Length = wcslen(InstallDir);
- RtlInitUnicodeString(&Name,
- FullName);
+ // TODO: Add check for 8.3 too.
- InitializeObjectAttributes(&ObjectAttributes,
- &Name,
- OBJ_CASE_INSENSITIVE,
- NULL,
- NULL);
+ /* Path must be at least 2 characters long */
+// if (Length < 2)
+// return FALSE;
- Status = NtOpenFile(&FileHandle,
- GENERIC_READ | SYNCHRONIZE,
- &ObjectAttributes,
- &IoStatusBlock,
- 0,
- FILE_SYNCHRONOUS_IO_NONALERT);
- if (!NT_SUCCESS(Status))
+ /* 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++)
{
- return FALSE;
+ if (iswspace(InstallDir[i]))
+ return FALSE;
+ }
+
+ /* Path component must not end with a dot */
+ for (i = 0; i < Length; i++)
+ {
+ if (InstallDir[i] == L'\\' && i > 0)
+ {
+ if (InstallDir[i - 1] == L'.')
+ return FALSE;
+ }
}
- NtClose(FileHandle);
+ if (InstallDir[Length - 1] == L'.')
+ return FALSE;
return TRUE;
}
projects / reactos.git / blobdiff