/* * COPYRIGHT: See COPYING in the top level directory * PROJECT: ReactOS system libraries * FILE: dll/win32/kernel32/client/file/move.c * PURPOSE: Directory functions * PROGRAMMER: Ariadne ( ariadne@xs4all.nl) * Gerhard W. Gruber (sparhawk_at_gmx.at) * Dmitry Philippov (shedon@mail.ru) * Pierre Schweitzer (pierre@reactos.org) * UPDATE HISTORY: * Created 01/11/98 * DP (29/07/2006) * Fix some bugs in the add_boot_rename_entry function */ /* INCLUDES *****************************************************************/ #include #include #include #define NDEBUG #include DEBUG_CHANNEL(kernel32file); /* GLOBALS *****************************************************************/ /* DEFINES *****************************************************************/ typedef struct _COPY_PROGRESS_CONTEXT { ULONG Flags; LPPROGRESS_ROUTINE UserRoutine; LPVOID UserData; } COPY_PROGRESS_CONTEXT, *PCOPY_PROGRESS_CONTEXT; /* FUNCTIONS ****************************************************************/ /* * @implemented */ NTSTATUS WINAPI BasepMoveFileDelayed(IN PUNICODE_STRING ExistingPath, IN PUNICODE_STRING NewPath, IN INT KeyId, IN BOOL CreateIfNotFound) { #define STRING_LENGTH 0x400 NTSTATUS Status; HANDLE KeyHandle; PVOID Buffer, BufferBegin; OBJECT_ATTRIBUTES ObjectAttributes; PWSTR PendingOperations, BufferWrite; ULONG DataSize, BufferLength, StringLength = STRING_LENGTH; UNICODE_STRING SessionManagerString, PendingOperationsString; /* +6 because a INT shouldn't take more than 6 chars. Especially given the call path */ WCHAR PendingOperationsBuffer[sizeof(L"PendingFileRenameOperations") / sizeof(WCHAR) + 6]; RtlInitUnicodeString(&SessionManagerString, L"\\Registry\\Machine\\System\\CurrentControlSet\\Control\\Session Manager"); /* Select appropriate key for adding our file */ if (KeyId == 1) { PendingOperations = L"PendingFileRenameOperations"; } else { StringCbPrintfW(PendingOperationsBuffer, sizeof(PendingOperationsBuffer), L"PendingFileRenameOperations%d", KeyId); PendingOperations = PendingOperationsBuffer; } RtlInitUnicodeString(&PendingOperationsString, PendingOperations); InitializeObjectAttributes(&ObjectAttributes, &SessionManagerString, OBJ_OPENIF | OBJ_CASE_INSENSITIVE, NULL, NULL); /* Open parent key */ Status = NtCreateKey(&KeyHandle, GENERIC_READ | GENERIC_WRITE, &ObjectAttributes, 0, NULL, REG_OPTION_NON_VOLATILE, NULL); if (Status == STATUS_ACCESS_DENIED) { Status = NtCreateKey(&KeyHandle, GENERIC_READ | GENERIC_WRITE, &ObjectAttributes, 0, NULL, REG_OPTION_BACKUP_RESTORE, NULL); } if (!NT_SUCCESS(Status)) { return Status; } /* Reserve enough to read previous string + to append our with required null chars */ BufferLength = NewPath->Length + ExistingPath->Length + STRING_LENGTH + 3 * sizeof(UNICODE_NULL); while (TRUE) { /* Allocate output buffer */ Buffer = RtlAllocateHeap(RtlGetProcessHeap(), 0, BufferLength); if (Buffer == NULL) { NtClose(KeyHandle); return STATUS_NO_MEMORY; } Status = NtQueryValueKey(KeyHandle, &PendingOperationsString, KeyValuePartialInformation, Buffer, StringLength, &DataSize); if (Status != STATUS_BUFFER_OVERFLOW) { break; } /* If buffer was too small, then, reallocate one which is big enough */ StringLength = DataSize; RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer); BufferLength = ExistingPath->Length + StringLength + NewPath->Length + 3 * sizeof(UNICODE_NULL); /* Check we didn't overflow */ if (BufferLength < StringLength) { NtClose(KeyHandle); return STATUS_BUFFER_TOO_SMALL; } } /* Check if it existed - if not, create only IF asked to */ if (!NT_SUCCESS(Status) && (Status != STATUS_OBJECT_NAME_NOT_FOUND || !CreateIfNotFound)) { NtClose(KeyHandle); RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer); return Status; } if (!NT_SUCCESS(Status)) { /* We didn't find any - ie, we create, so use complete buffer */ BufferBegin = Buffer; BufferWrite = Buffer; } else { PKEY_VALUE_PARTIAL_INFORMATION PartialInfo = (PKEY_VALUE_PARTIAL_INFORMATION)Buffer; /* Get data, our buffer begin and then where we should append data * (- null char, this is REG_MULTI_SZ, it already includes double termination, we keep only one) */ BufferBegin = PartialInfo->Data; BufferWrite = (PWSTR)((ULONG_PTR)PartialInfo->Data + PartialInfo->DataLength - sizeof(UNICODE_NULL)); } /* First copy existing */ RtlCopyMemory(BufferWrite, ExistingPath->Buffer, ExistingPath->Length); BufferWrite += ExistingPath->Length / sizeof(WCHAR); /* And append null char */ *BufferWrite = UNICODE_NULL; ++BufferWrite; /* Append destination */ RtlCopyMemory(BufferWrite, NewPath->Buffer, NewPath->Length); BufferWrite += NewPath->Length / sizeof(WCHAR); /* And append two null char (end of string) */ *BufferWrite = UNICODE_NULL; ++BufferWrite; *BufferWrite = UNICODE_NULL; /* Set new value */ Status = NtSetValueKey(KeyHandle, &PendingOperationsString, 0, REG_MULTI_SZ, BufferBegin, (ULONG_PTR)BufferWrite - (ULONG_PTR)BufferBegin + sizeof(WCHAR)); NtClose(KeyHandle); RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer); return Status; } /* * @implemented */ DWORD WINAPI BasepGetComputerNameFromNtPath(IN PUNICODE_STRING NewPath, IN HANDLE NewHandle, OUT PWSTR ComputerName, IN OUT PULONG ComputerNameLength) { BOOL Query = FALSE; WCHAR Letter; PWSTR AbsolutePath, EndOfName; USHORT AbsolutePathLength, NameLength; WCHAR TargetDevice[0x105]; WCHAR DeviceName[] = {'A', ':', '\0'}; /* Init to something, will be set later */ UNICODE_STRING UncString = RTL_CONSTANT_STRING(L"\\??\\UNC\\"); UNICODE_STRING GlobalString = RTL_CONSTANT_STRING(L"\\??\\"); DPRINT("BasepGetComputerNameFromNtPath(%wZ, %p, %p, %lu)\n", NewPath, NewHandle, ComputerName, ComputerNameLength); /* If it's an UNC path */ if (RtlPrefixUnicodeString(&UncString, NewPath, TRUE)) { /* Check for broken caller */ if (NewPath->Length <= UncString.Length) { return ERROR_BAD_PATHNAME; } /* Skip UNC prefix */ AbsolutePath = &NewPath->Buffer[UncString.Length / sizeof(WCHAR)]; AbsolutePathLength = NewPath->Length - UncString.Length; /* And query DFS */ Query = TRUE; } /* Otherwise, we have to be in global (NT path!), with drive letter */ else if (RtlPrefixUnicodeString(&GlobalString, NewPath, TRUE) && NewPath->Buffer[5] == ':') { /* Path is like that: \??\C:\Complete Path\To File.ext */ /* Get the letter and upcase it if required */ Letter = NewPath->Buffer[4]; if (Letter >= 'a' && Letter <= 'z') { Letter -= ('a' - 'A'); } DeviceName[0] = Letter; /* Query the associated DOS device */ if (!QueryDosDeviceW(DeviceName, TargetDevice, ARRAYSIZE(TargetDevice))) { return GetLastError(); } /* If that's a network share */ if (TargetDevice == wcsstr(TargetDevice, L"\\Device\\LanmanRedirector\\;")) { /* Path is like that: \Device\LanmanRedirector\;C:0000000000000000\Complete Path\To File.ext */ /* Check we have the correct drive letter */ if (TargetDevice[26] == DeviceName[0] && TargetDevice[27] == ':') { /* Check for the path begin, computer name is before */ PWSTR Path = wcschr(&TargetDevice[28], '\\'); if (Path == NULL) { return ERROR_BAD_PATHNAME; } AbsolutePath = Path + 1; AbsolutePathLength = sizeof(WCHAR) * (ARRAYSIZE(TargetDevice) - (AbsolutePath - TargetDevice)); } else { return ERROR_BAD_PATHNAME; } } /* If it's a local device */ else if (TargetDevice == wcsstr(TargetDevice, L"\\Device\\Harddisk") || TargetDevice == wcsstr(TargetDevice, L"\\Device\\CdRom") || TargetDevice == wcsstr(TargetDevice, L"\\Device\\Floppy")) { /* Just query the computer name */ if (!GetComputerNameW(ComputerName, ComputerNameLength)) { return GetLastError(); } return ERROR_SUCCESS; } /* If it's a DFS share */ else if (TargetDevice == wcsstr(TargetDevice, L"\\Device\\WinDfs\\")) { /* Obviously, query DFS */ Query = TRUE; } else { return ERROR_BAD_PATHNAME; } } else { return ERROR_BAD_PATHNAME; } /* Query DFS, currently not implemented - shouldn't be missing in ReactOS yet ;-) */ if (Query) { UNIMPLEMENTED_DBGBREAK("Querying DFS not implemented!\n"); AbsolutePath = NULL; AbsolutePathLength = 0; } /* Now, properly extract the computer name from the full path */ EndOfName = AbsolutePath; if (AbsolutePathLength) { for (NameLength = 0; NameLength < AbsolutePathLength; NameLength += sizeof(WCHAR)) { /* Look for the next \, it will be the end of computer name */ if (EndOfName[0] == '\\') { break; } /* Computer name cannot contain ., if we get to that point, something went wrong... */ else if (EndOfName[0] == '.') { return ERROR_BAD_PATHNAME; } ++EndOfName; } } NameLength = EndOfName - AbsolutePath; /* Check we didn't overflow and that our computer name isn't ill-formed */ if (NameLength >= AbsolutePathLength || NameLength >= MAX_COMPUTERNAME_LENGTH * sizeof(WCHAR)) { return ERROR_BAD_PATHNAME; } /* Check we can fit */ if (NameLength + sizeof(UNICODE_NULL) > *ComputerNameLength * sizeof(WCHAR)) { return ERROR_BUFFER_OVERFLOW; } /* Write, zero and done! */ RtlCopyMemory(ComputerName, AbsolutePath, NameLength); *ComputerNameLength = NameLength / sizeof(WCHAR); ComputerName[NameLength / sizeof(WCHAR)] = UNICODE_NULL; return ERROR_SUCCESS; } /* * @implemented */ NTSTATUS WINAPI BasepNotifyTrackingService(IN OUT PHANDLE ExistingHandle, IN POBJECT_ATTRIBUTES ObjectAttributes, IN HANDLE NewHandle, IN PUNICODE_STRING NewPath) { NTSTATUS Status; ULONG ComputerNameLength, FileAttributes; WCHAR ComputerName[MAX_COMPUTERNAME_LENGTH + 1]; OEM_STRING ComputerNameStringA; CHAR ComputerNameStringBuffer[0x105]; UNICODE_STRING ComputerNameStringW; IO_STATUS_BLOCK IoStatusBlock; FILE_BASIC_INFORMATION FileBasicInfo; HANDLE hFullWrite; struct { FILE_TRACKING_INFORMATION; CHAR Buffer[(MAX_COMPUTERNAME_LENGTH + 1) * sizeof(WCHAR)]; } FileTrackingInfo; DPRINT("BasepNotifyTrackingService(%p, %p, %p, %wZ)\n", *ExistingHandle, ObjectAttributes, NewHandle, NewPath); Status = STATUS_SUCCESS; ComputerNameLength = ARRAYSIZE(ComputerName); /* Attempt to get computer name of target handle */ if (BasepGetComputerNameFromNtPath(NewPath, NewHandle, ComputerName, &ComputerNameLength)) { /* If we failed to get it, we will just notify with the handle */ FileTrackingInfo.ObjectInformationLength = 0; } else { /* Convert the retrieved computer name to ANSI and attach it to the notification */ ComputerNameStringA.Length = 0; ComputerNameStringA.MaximumLength = ARRAYSIZE(ComputerNameStringBuffer); ComputerNameStringA.Buffer = ComputerNameStringBuffer; RtlInitUnicodeString(&ComputerNameStringW, ComputerName); Status = RtlUnicodeStringToOemString(&ComputerNameStringA, &ComputerNameStringW, 0); if (!NT_SUCCESS(Status)) { return Status; } RtlCopyMemory(FileTrackingInfo.ObjectInformation, ComputerNameStringA.Buffer, ComputerNameStringA.Length); FileTrackingInfo.ObjectInformation[ComputerNameStringA.Length] = 0; FileTrackingInfo.ObjectInformationLength = ComputerNameStringA.Length + 1; } /* Attach the handle we moved */ FileTrackingInfo.DestinationFile = NewHandle; /* Final, notify */ Status = NtSetInformationFile(*ExistingHandle, &IoStatusBlock, &FileTrackingInfo, sizeof(FileTrackingInfo), FileTrackingInformation); if (Status != STATUS_ACCESS_DENIED) { return Status; } /* If we get here, we got access denied error, this comes from a * read-only flag. So, close the file, in order to reopen it with enough * rights to remove said flag and reattempt notification */ CloseHandle(*ExistingHandle); /* Reopen it, to be able to change the destination file attributes */ Status = NtOpenFile(ExistingHandle, SYNCHRONIZE | FILE_READ_ATTRIBUTES | FILE_WRITE_ATTRIBUTES, ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_SYNCHRONOUS_IO_NONALERT); if (!NT_SUCCESS(Status)) { *ExistingHandle = INVALID_HANDLE_VALUE; return Status; } /* Get the file attributes */ Status = NtQueryInformationFile(*ExistingHandle, &IoStatusBlock, &FileBasicInfo, sizeof(FileBasicInfo), FileBasicInformation); if (!NT_SUCCESS(Status)) { return Status; } /* Get rid of the read only flag */ FileAttributes = FileBasicInfo.FileAttributes & ~FILE_ATTRIBUTE_READONLY; RtlZeroMemory(&FileBasicInfo, sizeof(FileBasicInfo)); FileBasicInfo.FileAttributes = FileAttributes; /* Attempt... */ Status = NtSetInformationFile(*ExistingHandle, &IoStatusBlock, &FileBasicInfo, sizeof(FileBasicInfo), FileBasicInformation); if (!NT_SUCCESS(Status)) { return Status; } /* Now, reopen with maximum accesses to notify */ Status = NtOpenFile(&hFullWrite, GENERIC_WRITE | SYNCHRONIZE, ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_SYNCHRONOUS_IO_NONALERT); if (NT_SUCCESS(Status)) { NtClose(*ExistingHandle); *ExistingHandle = hFullWrite; /* Full success, notify! */ Status = NtSetInformationFile(*ExistingHandle, &IoStatusBlock, &FileTrackingInfo, sizeof(FileTrackingInfo), FileTrackingInformation); } /* If opening with full access failed or if notify failed, restore read-only */ if (!NT_SUCCESS(Status)) { FileBasicInfo.FileAttributes |= FILE_ATTRIBUTE_READONLY; Status = NtSetInformationFile(*ExistingHandle, &IoStatusBlock, &FileBasicInfo, sizeof(FileBasicInfo), FileBasicInformation); } /* We're done */ return Status; } /* * @implemented */ NTSTATUS WINAPI BasepOpenFileForMove(IN LPCWSTR File, OUT PUNICODE_STRING RelativeNtName, OUT LPWSTR * NtName, OUT PHANDLE FileHandle, OUT POBJECT_ATTRIBUTES ObjectAttributes, IN ACCESS_MASK DesiredAccess, IN ULONG ShareAccess, IN ULONG OpenOptions) { RTL_RELATIVE_NAME_U RelativeName; NTSTATUS Status; IO_STATUS_BLOCK IoStatusBlock; FILE_ATTRIBUTE_TAG_INFORMATION TagInfo; ULONG IntShareAccess; BOOLEAN HasRelative = FALSE; _SEH2_TRY { /* Zero output */ RelativeNtName->Length = RelativeNtName->MaximumLength = 0; RelativeNtName->Buffer = NULL; *NtName = NULL; if (!RtlDosPathNameToRelativeNtPathName_U(File, RelativeNtName, NULL, &RelativeName)) { Status = STATUS_OBJECT_PATH_NOT_FOUND; _SEH2_LEAVE; } HasRelative = TRUE; *NtName = RelativeNtName->Buffer; if (RelativeName.RelativeName.Length) { RelativeNtName->Length = RelativeName.RelativeName.Length; RelativeNtName->MaximumLength = RelativeName.RelativeName.MaximumLength; RelativeNtName->Buffer = RelativeName.RelativeName.Buffer; } else { RelativeName.ContainingDirectory = 0; } InitializeObjectAttributes(ObjectAttributes, RelativeNtName, OBJ_CASE_INSENSITIVE, RelativeName.ContainingDirectory, NULL); /* Force certain flags here, given ops we'll do */ IntShareAccess = ShareAccess | FILE_SHARE_READ | FILE_SHARE_WRITE; OpenOptions |= FILE_OPEN_FOR_BACKUP_INTENT | FILE_SYNCHRONOUS_IO_NONALERT; /* We'll try to read reparse tag */ Status = NtOpenFile(FileHandle, DesiredAccess | FILE_READ_ATTRIBUTES | SYNCHRONIZE, ObjectAttributes, &IoStatusBlock, IntShareAccess, OpenOptions | FILE_OPEN_REPARSE_POINT); if (NT_SUCCESS(Status)) { /* Attempt the read */ Status = NtQueryInformationFile(*FileHandle, &IoStatusBlock, &TagInfo, sizeof(FILE_ATTRIBUTE_TAG_INFORMATION), FileAttributeTagInformation); /* Return if failure with a status that wouldn't mean the FSD cannot support reparse points */ if (!NT_SUCCESS(Status) && (Status != STATUS_NOT_IMPLEMENTED && Status != STATUS_INVALID_PARAMETER)) { _SEH2_LEAVE; } if (NT_SUCCESS(Status)) { /* This cannot happen on mount points */ if (TagInfo.FileAttributes & FILE_ATTRIBUTE_DEVICE || TagInfo.ReparseTag == IO_REPARSE_TAG_MOUNT_POINT) { _SEH2_LEAVE; } } NtClose(*FileHandle); *FileHandle = INVALID_HANDLE_VALUE; IntShareAccess = ShareAccess | FILE_SHARE_READ | FILE_SHARE_DELETE; } else if (Status == STATUS_INVALID_PARAMETER) { IntShareAccess = ShareAccess | FILE_SHARE_READ | FILE_SHARE_WRITE; } else { _SEH2_LEAVE; } /* Reattempt to open normally, following reparse point if needed */ Status = NtOpenFile(FileHandle, DesiredAccess | SYNCHRONIZE, ObjectAttributes, &IoStatusBlock, IntShareAccess, OpenOptions); } _SEH2_FINALLY { if (HasRelative) { RtlReleaseRelativeName(&RelativeName); } } _SEH2_END; return Status; } /* * @implemented */ DWORD WINAPI BasepMoveFileCopyProgress(IN LARGE_INTEGER TotalFileSize, IN LARGE_INTEGER TotalBytesTransferred, IN LARGE_INTEGER StreamSize, IN LARGE_INTEGER StreamBytesTransferred, IN DWORD dwStreamNumber, IN DWORD dwCallbackReason, IN HANDLE hSourceFile, IN HANDLE hDestinationFile, IN LPVOID lpData OPTIONAL) { DWORD Ret = 0; PCOPY_PROGRESS_CONTEXT Context = (PCOPY_PROGRESS_CONTEXT)lpData; if (Context->Flags & MOVEFILE_WRITE_THROUGH) { if (!dwCallbackReason) { if (StreamBytesTransferred.QuadPart == StreamSize.QuadPart) { FlushFileBuffers(hDestinationFile); } } } if (Context->UserRoutine) { Ret = Context->UserRoutine(TotalFileSize, TotalBytesTransferred, StreamSize, StreamBytesTransferred, dwStreamNumber, dwCallbackReason, hSourceFile, hDestinationFile, Context->UserData); } return Ret; } /* * @implemented */ BOOL WINAPI MoveFileWithProgressW(IN LPCWSTR lpExistingFileName, IN LPCWSTR lpNewFileName, IN LPPROGRESS_ROUTINE lpProgressRoutine, IN LPVOID lpData, IN DWORD dwFlags) { NTSTATUS Status; PWSTR NewBuffer; IO_STATUS_BLOCK IoStatusBlock; COPY_PROGRESS_CONTEXT CopyContext; OBJECT_ATTRIBUTES ObjectAttributes; PFILE_RENAME_INFORMATION RenameInfo; UNICODE_STRING NewPathU, ExistingPathU; FILE_ATTRIBUTE_TAG_INFORMATION FileAttrTagInfo; HANDLE SourceHandle = INVALID_HANDLE_VALUE, NewHandle, ExistingHandle; BOOL Ret = FALSE, ReplaceIfExists, DelayUntilReboot, AttemptReopenWithoutReparse; DPRINT("MoveFileWithProgressW(%S, %S, %p, %p, %x)\n", lpExistingFileName, lpNewFileName, lpProgressRoutine, lpData, dwFlags); NewPathU.Buffer = NULL; ExistingPathU.Buffer = NULL; _SEH2_TRY { /* Don't allow renaming to a disk */ if (lpNewFileName && RtlIsDosDeviceName_U(lpNewFileName)) { BaseSetLastNTError(STATUS_OBJECT_NAME_COLLISION); _SEH2_LEAVE; } ReplaceIfExists = !!(dwFlags & MOVEFILE_REPLACE_EXISTING); /* Get file path */ if (!RtlDosPathNameToNtPathName_U(lpExistingFileName, &ExistingPathU, NULL, NULL)) { BaseSetLastNTError(STATUS_OBJECT_PATH_NOT_FOUND); _SEH2_LEAVE; } /* Sanitize input */ DelayUntilReboot = !!(dwFlags & MOVEFILE_DELAY_UNTIL_REBOOT); if (DelayUntilReboot && (dwFlags & MOVEFILE_CREATE_HARDLINK)) { BaseSetLastNTError(STATUS_INVALID_PARAMETER); _SEH2_LEAVE; } /* Unless we manage a proper opening, we'll attempt to reopen without reparse support */ AttemptReopenWithoutReparse = TRUE; InitializeObjectAttributes(&ObjectAttributes, &ExistingPathU, OBJ_CASE_INSENSITIVE, NULL, NULL); /* Attempt to open source file */ Status = NtOpenFile(&SourceHandle, FILE_READ_ATTRIBUTES | DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_OPEN_FOR_BACKUP_INTENT | ((dwFlags & MOVEFILE_WRITE_THROUGH) ? FILE_WRITE_THROUGH : 0)); if (!NT_SUCCESS(Status)) { /* If we failed and the file doesn't exist, don't attempt to reopen without reparse */ if (DelayUntilReboot && (Status == STATUS_SHARING_VIOLATION || Status == STATUS_OBJECT_NAME_NOT_FOUND || Status == STATUS_OBJECT_PATH_NOT_FOUND)) { /* Here we don't fail completely, as we postpone the operation to reboot * File might exist afterwards, and we don't need a handle here */ SourceHandle = INVALID_HANDLE_VALUE; AttemptReopenWithoutReparse = FALSE; } /* If we failed for any reason than unsupported reparse, fail completely */ else if (Status != STATUS_INVALID_PARAMETER) { BaseSetLastNTError(Status); _SEH2_LEAVE; } } else { /* We managed to open, so query information */ Status = NtQueryInformationFile(SourceHandle, &IoStatusBlock, &FileAttrTagInfo, sizeof(FILE_ATTRIBUTE_TAG_INFORMATION), FileAttributeTagInformation); if (!NT_SUCCESS(Status)) { /* Do not tolerate any other error than something related to not supported operation */ if (Status != STATUS_NOT_IMPLEMENTED && Status != STATUS_INVALID_PARAMETER) { BaseSetLastNTError(Status); _SEH2_LEAVE; } /* Not a reparse point, no need to reopen, it's fine */ AttemptReopenWithoutReparse = FALSE; } /* Validate the reparse point (do we support it?) */ else if (FileAttrTagInfo.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT && FileAttrTagInfo.ReparseTag != IO_REPARSE_TAG_MOUNT_POINT) { NtClose(SourceHandle); SourceHandle = INVALID_HANDLE_VALUE; } else { /* Mount point, let's rename it */ AttemptReopenWithoutReparse = FALSE; } } /* Simply reopen if required */ if (AttemptReopenWithoutReparse) { Status = NtOpenFile(&SourceHandle, DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE, ((dwFlags & MOVEFILE_WRITE_THROUGH) ? FILE_WRITE_THROUGH : 0)); if (!NT_SUCCESS(Status)) { BaseSetLastNTError(Status); _SEH2_LEAVE; } } /* Nullify string if we're to use it */ if (DelayUntilReboot && !lpNewFileName) { RtlInitUnicodeString(&NewPathU, 0); } /* Check whether path exists */ else if (!RtlDosPathNameToNtPathName_U(lpNewFileName, &NewPathU, 0, 0)) { BaseSetLastNTError(STATUS_OBJECT_PATH_NOT_FOUND); _SEH2_LEAVE; } /* Handle postponed renaming */ if (DelayUntilReboot) { /* If new file exists and we're allowed to replace, then mark the path with ! */ if (ReplaceIfExists && NewPathU.Length) { NewBuffer = RtlAllocateHeap(RtlGetProcessHeap(), 0, NewPathU.Length + sizeof(WCHAR)); if (NewBuffer == NULL) { BaseSetLastNTError(STATUS_NO_MEMORY); _SEH2_LEAVE; } NewBuffer[0] = L'!'; RtlCopyMemory(&NewBuffer[1], NewPathU.Buffer, NewPathU.Length); NewPathU.Length += sizeof(WCHAR); NewPathU.MaximumLength += sizeof(WCHAR); RtlFreeHeap(RtlGetProcessHeap(), 0, NewPathU.Buffer); NewPathU.Buffer = NewBuffer; } /* Check whether 'copy' renaming is allowed if required */ if (RtlDetermineDosPathNameType_U(lpExistingFileName) == RtlPathTypeUncAbsolute || dwFlags & MOVEFILE_COPY_ALLOWED) { Status = STATUS_INVALID_PARAMETER; } else { /* First, probe 2nd key to see whether it exists - if so, it will be appended there */ Status = BasepMoveFileDelayed(&ExistingPathU, &NewPathU, 2, FALSE); if (Status == STATUS_OBJECT_NAME_NOT_FOUND) { /* If doesn't exist, append to first key first, creating it if it doesn't exist */ Status = BasepMoveFileDelayed(&ExistingPathU, &NewPathU, 1, TRUE); if (Status == STATUS_INSUFFICIENT_RESOURCES) { /* If it failed because it's too big, then create 2nd key and put it there */ Status = BasepMoveFileDelayed(&ExistingPathU, &NewPathU, 2, TRUE); } } } /* If we failed at some point, return the error */ if (!NT_SUCCESS(Status)) { BaseSetLastNTError(Status); _SEH2_LEAVE; } Ret = TRUE; _SEH2_LEAVE; } /* At that point, we MUST have a source handle */ ASSERT(SourceHandle != INVALID_HANDLE_VALUE); /* Allocate renaming buffer and fill it */ RenameInfo = RtlAllocateHeap(RtlGetProcessHeap(), 0, NewPathU.Length + sizeof(FILE_RENAME_INFORMATION)); if (RenameInfo == NULL) { BaseSetLastNTError(STATUS_NO_MEMORY); _SEH2_LEAVE; } RtlCopyMemory(&RenameInfo->FileName, NewPathU.Buffer, NewPathU.Length); RenameInfo->ReplaceIfExists = ReplaceIfExists; RenameInfo->RootDirectory = 0; RenameInfo->FileNameLength = NewPathU.Length; /* Attempt to rename the file */ Status = NtSetInformationFile(SourceHandle, &IoStatusBlock, RenameInfo, NewPathU.Length + sizeof(FILE_RENAME_INFORMATION), ((dwFlags & MOVEFILE_CREATE_HARDLINK) ? FileLinkInformation : FileRenameInformation)); RtlFreeHeap(RtlGetProcessHeap(), 0, RenameInfo); if (NT_SUCCESS(Status)) { /* If it succeed, all fine, quit */ Ret = TRUE; _SEH2_LEAVE; } /* If we failed for any other reason than not the same device, fail * If we failed because of different devices, only allow renaming if user allowed copy */ if (Status != STATUS_NOT_SAME_DEVICE || !(dwFlags & MOVEFILE_COPY_ALLOWED)) { /* ReactOS hack! To be removed once all FSD have proper renaming support * Just leave status to error and leave */ if (Status == STATUS_NOT_IMPLEMENTED) { DPRINT1("Forcing copy, renaming not supported by FSD\n"); } else { BaseSetLastNTError(Status); _SEH2_LEAVE; } } /* Close source file */ NtClose(SourceHandle); SourceHandle = INVALID_HANDLE_VALUE; /* Issue the copy of the file */ CopyContext.Flags = dwFlags; CopyContext.UserRoutine = lpProgressRoutine; CopyContext.UserData = lpData; NewHandle = INVALID_HANDLE_VALUE; ExistingHandle = INVALID_HANDLE_VALUE; Ret = BasepCopyFileExW(lpExistingFileName, lpNewFileName, BasepMoveFileCopyProgress, &CopyContext, NULL, (ReplaceIfExists == 0) | COPY_FILE_OPEN_SOURCE_FOR_WRITE, 0, &ExistingHandle, &NewHandle); if (!Ret) { /* If it failed, don't leak any handle */ if (ExistingHandle != INVALID_HANDLE_VALUE) { CloseHandle(ExistingHandle); ExistingHandle = INVALID_HANDLE_VALUE; } } else if (ExistingHandle != INVALID_HANDLE_VALUE) { if (NewHandle != INVALID_HANDLE_VALUE) { /* If copying succeed, notify */ Status = BasepNotifyTrackingService(&ExistingHandle, &ObjectAttributes, NewHandle, &NewPathU); if (!NT_SUCCESS(Status)) { /* Fail in case it had to succeed */ if (dwFlags & MOVEFILE_FAIL_IF_NOT_TRACKABLE) { if (NewHandle != INVALID_HANDLE_VALUE) CloseHandle(NewHandle); NewHandle = INVALID_HANDLE_VALUE; DeleteFileW(lpNewFileName); Ret = FALSE; BaseSetLastNTError(Status); } } } CloseHandle(ExistingHandle); ExistingHandle = INVALID_HANDLE_VALUE; } /* In case copy worked, close file */ if (NewHandle != INVALID_HANDLE_VALUE) { CloseHandle(NewHandle); NewHandle = INVALID_HANDLE_VALUE; } /* If it succeed, delete source file */ if (Ret) { if (!DeleteFileW(lpExistingFileName)) { /* Reset file attributes if required */ SetFileAttributesW(lpExistingFileName, FILE_ATTRIBUTE_NORMAL); DeleteFileW(lpExistingFileName); } } } _SEH2_FINALLY { if (SourceHandle != INVALID_HANDLE_VALUE) NtClose(SourceHandle); RtlFreeHeap(RtlGetProcessHeap(), 0, ExistingPathU.Buffer); RtlFreeHeap(RtlGetProcessHeap(), 0, NewPathU.Buffer); } _SEH2_END; return Ret; } /* * @implemented */ BOOL WINAPI MoveFileWithProgressA(IN LPCSTR lpExistingFileName, IN LPCSTR lpNewFileName OPTIONAL, IN LPPROGRESS_ROUTINE lpProgressRoutine OPTIONAL, IN LPVOID lpData OPTIONAL, IN DWORD dwFlags) { BOOL Ret; UNICODE_STRING ExistingFileNameW, NewFileNameW; if (!Basep8BitStringToDynamicUnicodeString(&ExistingFileNameW, lpExistingFileName)) { return FALSE; } if (lpNewFileName) { if (!Basep8BitStringToDynamicUnicodeString(&NewFileNameW, lpNewFileName)) { RtlFreeUnicodeString(&ExistingFileNameW); return FALSE; } } else { NewFileNameW.Buffer = NULL; } Ret = MoveFileWithProgressW(ExistingFileNameW.Buffer, NewFileNameW.Buffer, lpProgressRoutine, lpData, dwFlags); RtlFreeUnicodeString(&ExistingFileNameW); RtlFreeUnicodeString(&NewFileNameW); return Ret; } /* * @implemented */ BOOL WINAPI MoveFileW(IN LPCWSTR lpExistingFileName, IN LPCWSTR lpNewFileName) { return MoveFileWithProgressW(lpExistingFileName, lpNewFileName, NULL, NULL, MOVEFILE_COPY_ALLOWED); } /* * @implemented */ BOOL WINAPI MoveFileExW(IN LPCWSTR lpExistingFileName, IN LPCWSTR lpNewFileName OPTIONAL, IN DWORD dwFlags) { return MoveFileWithProgressW(lpExistingFileName, lpNewFileName, NULL, NULL, dwFlags); } /* * @implemented */ BOOL WINAPI MoveFileA(IN LPCSTR lpExistingFileName, IN LPCSTR lpNewFileName) { return MoveFileWithProgressA(lpExistingFileName, lpNewFileName, NULL, NULL, MOVEFILE_COPY_ALLOWED); } /* * @implemented */ BOOL WINAPI MoveFileExA(IN LPCSTR lpExistingFileName, IN LPCSTR lpNewFileName OPTIONAL, IN DWORD dwFlags) { return MoveFileWithProgressA(lpExistingFileName, lpNewFileName, NULL, NULL, dwFlags); } /* * @implemented */ BOOL WINAPI ReplaceFileA(IN LPCSTR lpReplacedFileName, IN LPCSTR lpReplacementFileName, IN LPCSTR lpBackupFileName OPTIONAL, IN DWORD dwReplaceFlags, IN LPVOID lpExclude, IN LPVOID lpReserved) { BOOL Ret; UNICODE_STRING ReplacedFileNameW, ReplacementFileNameW, BackupFileNameW; if (!lpReplacedFileName || !lpReplacementFileName || lpExclude || lpReserved || dwReplaceFlags & ~(REPLACEFILE_WRITE_THROUGH | REPLACEFILE_IGNORE_MERGE_ERRORS)) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } if (!Basep8BitStringToDynamicUnicodeString(&ReplacedFileNameW, lpReplacedFileName)) { return FALSE; } if (!Basep8BitStringToDynamicUnicodeString(&ReplacementFileNameW, lpReplacementFileName)) { RtlFreeUnicodeString(&ReplacedFileNameW); return FALSE; } if (lpBackupFileName) { if (!Basep8BitStringToDynamicUnicodeString(&BackupFileNameW, lpBackupFileName)) { RtlFreeUnicodeString(&ReplacementFileNameW); RtlFreeUnicodeString(&ReplacedFileNameW); return FALSE; } } else { BackupFileNameW.Buffer = NULL; } Ret = ReplaceFileW(ReplacedFileNameW.Buffer, ReplacementFileNameW.Buffer, BackupFileNameW.Buffer, dwReplaceFlags, 0, 0); if (lpBackupFileName) { RtlFreeUnicodeString(&BackupFileNameW); } RtlFreeUnicodeString(&ReplacementFileNameW); RtlFreeUnicodeString(&ReplacedFileNameW); return Ret; } /* * @unimplemented */ BOOL WINAPI ReplaceFileW( LPCWSTR lpReplacedFileName, LPCWSTR lpReplacementFileName, LPCWSTR lpBackupFileName, DWORD dwReplaceFlags, LPVOID lpExclude, LPVOID lpReserved ) { HANDLE hReplaced = NULL, hReplacement = NULL; UNICODE_STRING NtReplacedName = { 0, 0, NULL }; UNICODE_STRING NtReplacementName = { 0, 0, NULL }; DWORD Error = ERROR_SUCCESS; NTSTATUS Status; BOOL Ret = FALSE; IO_STATUS_BLOCK IoStatusBlock; OBJECT_ATTRIBUTES ObjectAttributes; PVOID Buffer = NULL ; if (dwReplaceFlags) FIXME("Ignoring flags %x\n", dwReplaceFlags); /* First two arguments are mandatory */ if (!lpReplacedFileName || !lpReplacementFileName) { SetLastError(ERROR_INVALID_PARAMETER); return FALSE; } /* Back it up */ if(lpBackupFileName) { if(!CopyFileW(lpReplacedFileName, lpBackupFileName, FALSE)) { Error = GetLastError(); goto Cleanup ; } } /* Open the "replaced" file for reading and writing */ if (!(RtlDosPathNameToNtPathName_U(lpReplacedFileName, &NtReplacedName, NULL, NULL))) { Error = ERROR_PATH_NOT_FOUND; goto Cleanup; } InitializeObjectAttributes(&ObjectAttributes, &NtReplacedName, OBJ_CASE_INSENSITIVE, NULL, NULL); Status = NtOpenFile(&hReplaced, GENERIC_READ | GENERIC_WRITE | DELETE | SYNCHRONIZE | WRITE_DAC, &ObjectAttributes, &IoStatusBlock, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_SYNCHRONOUS_IO_NONALERT | FILE_NON_DIRECTORY_FILE); if (!NT_SUCCESS(Status)) { if (Status == STATUS_OBJECT_NAME_NOT_FOUND) Error = ERROR_FILE_NOT_FOUND; else Error = ERROR_UNABLE_TO_REMOVE_REPLACED; goto Cleanup; } /* Blank it */ SetEndOfFile(hReplaced) ; /* * Open the replacement file for reading, writing, and deleting * (deleting is needed when finished) */ if (!(RtlDosPathNameToNtPathName_U(lpReplacementFileName, &NtReplacementName, NULL, NULL))) { Error = ERROR_PATH_NOT_FOUND; goto Cleanup; } InitializeObjectAttributes(&ObjectAttributes, &NtReplacementName, OBJ_CASE_INSENSITIVE, NULL, NULL); Status = NtOpenFile(&hReplacement, GENERIC_READ | DELETE | SYNCHRONIZE, &ObjectAttributes, &IoStatusBlock, 0, FILE_SYNCHRONOUS_IO_NONALERT | FILE_NON_DIRECTORY_FILE | FILE_DELETE_ON_CLOSE); if (!NT_SUCCESS(Status)) { Error = RtlNtStatusToDosError(Status); goto Cleanup; } Buffer = RtlAllocateHeap(RtlGetProcessHeap(), HEAP_ZERO_MEMORY, 0x10000) ; if (!Buffer) { Error = ERROR_NOT_ENOUGH_MEMORY; goto Cleanup ; } while (Status != STATUS_END_OF_FILE) { Status = NtReadFile(hReplacement, NULL, NULL, NULL, &IoStatusBlock, Buffer, 0x10000, NULL, NULL) ; if (NT_SUCCESS(Status)) { Status = NtWriteFile(hReplaced, NULL, NULL, NULL, &IoStatusBlock, Buffer, IoStatusBlock.Information, NULL, NULL) ; if (!NT_SUCCESS(Status)) { Error = RtlNtStatusToDosError(Status); goto Cleanup; } } else if (Status != STATUS_END_OF_FILE) { Error = RtlNtStatusToDosError(Status); goto Cleanup; } } Ret = TRUE; /* Perform resource cleanup */ Cleanup: if (hReplaced) NtClose(hReplaced); if (hReplacement) NtClose(hReplacement); if (Buffer) RtlFreeHeap(RtlGetProcessHeap(), 0, Buffer); if (NtReplacementName.Buffer) RtlFreeHeap(GetProcessHeap(), 0, NtReplacementName.Buffer); if (NtReplacedName.Buffer) RtlFreeHeap(GetProcessHeap(), 0, NtReplacedName.Buffer); /* If there was an error, set the error code */ if(!Ret) { TRACE("ReplaceFileW failed (error=%lu)\n", Error); SetLastError(Error); } return Ret; } /* * @implemented */ BOOL WINAPI PrivMoveFileIdentityW(IN LPCWSTR lpSource, IN LPCWSTR lpDestination, IN DWORD dwFlags) { ACCESS_MASK SourceAccess; UNICODE_STRING NtSource, NtDestination; LPWSTR RelativeSource, RelativeDestination; HANDLE SourceHandle, DestinationHandle; OBJECT_ATTRIBUTES ObjectAttributesSource, ObjectAttributesDestination; NTSTATUS Status, OldStatus = STATUS_SUCCESS; ACCESS_MASK DestAccess; IO_STATUS_BLOCK IoStatusBlock; FILE_BASIC_INFORMATION SourceInformation, DestinationInformation; FILE_DISPOSITION_INFORMATION FileDispositionInfo; DPRINT("PrivMoveFileIdentityW(%S, %S, %x)\n", lpSource, lpDestination, dwFlags); SourceHandle = INVALID_HANDLE_VALUE; NtSource.Length = NtSource.MaximumLength = 0; NtSource.Buffer = NULL; RelativeSource = NULL; DestinationHandle = INVALID_HANDLE_VALUE; NtDestination.Length = NtDestination.MaximumLength = 0; NtDestination.Buffer = NULL; RelativeDestination = NULL; /* FILE_WRITE_DATA is required for later on notification */ SourceAccess = FILE_READ_ATTRIBUTES | FILE_WRITE_DATA; if (dwFlags & PRIV_DELETE_ON_SUCCESS) { SourceAccess |= DELETE; } _SEH2_TRY { /* We will loop twice: * First we attempt to open with FILE_WRITE_DATA for notification * If it fails and we have flag for non-trackable files, we retry * without FILE_WRITE_DATA. * If that one fails, then, we quit for real */ while (TRUE) { Status = BasepOpenFileForMove(lpSource, &NtSource, &RelativeSource, &SourceHandle, &ObjectAttributesSource, SourceAccess, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_OPEN_NO_RECALL); if (NT_SUCCESS(Status)) { break; } /* If we already attempted the opening without FILE_WRITE_DATA * or if we cannot move on non-trackable files, fail. */ if (!(SourceAccess & FILE_WRITE_DATA) || !(dwFlags & PRIV_ALLOW_NON_TRACKABLE)) { _SEH2_LEAVE; } if (RelativeSource) { RtlFreeHeap(RtlGetProcessHeap(), 0, RelativeSource); RelativeSource = NULL; } if (SourceHandle != INVALID_HANDLE_VALUE) { NtClose(SourceHandle); SourceHandle = INVALID_HANDLE_VALUE; } SourceAccess &= ~FILE_WRITE_DATA; /* Remember fist failure in the path */ if (NT_SUCCESS(OldStatus)) { OldStatus = Status; } } DestAccess = FILE_WRITE_ATTRIBUTES; /* If we could preserve FILE_WRITE_DATA for source, attempt to get it for destination * Still for notification purposes */ if (SourceAccess & FILE_WRITE_DATA) { DestAccess |= FILE_WRITE_DATA; } /* cf comment for first loop */ while (TRUE) { Status = BasepOpenFileForMove(lpDestination, &NtDestination, &RelativeDestination, &DestinationHandle, &ObjectAttributesDestination, DestAccess, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, FILE_OPEN_NO_RECALL); if (NT_SUCCESS(Status)) { break; } /* If we already attempted the opening without FILE_WRITE_DATA * or if we cannot move on non-trackable files, fail. */ if (!(DestAccess & FILE_WRITE_DATA) || !(dwFlags & PRIV_ALLOW_NON_TRACKABLE)) { _SEH2_LEAVE; } if (RelativeDestination) { RtlFreeHeap(RtlGetProcessHeap(), 0, RelativeDestination); RelativeDestination = NULL; } if (DestinationHandle != INVALID_HANDLE_VALUE) { NtClose(DestinationHandle); DestinationHandle = INVALID_HANDLE_VALUE; } DestAccess &= ~FILE_WRITE_DATA; /* Remember fist failure in the path */ if (NT_SUCCESS(OldStatus)) { OldStatus = Status; } } /* Get the creation time from source */ Status = NtQueryInformationFile(SourceHandle, &IoStatusBlock, &SourceInformation, sizeof(SourceInformation), FileBasicInformation); if (NT_SUCCESS(Status)) { /* Then, prepare to set it for destination */ RtlZeroMemory(&DestinationInformation, sizeof(DestinationInformation)); DestinationInformation.CreationTime.QuadPart = SourceInformation.CreationTime.QuadPart; /* And set it, that's all folks! */ Status = NtSetInformationFile(DestinationHandle, &IoStatusBlock, &DestinationInformation, sizeof(DestinationInformation), FileBasicInformation); } if (!NT_SUCCESS(Status)) { if (!(dwFlags & PRIV_ALLOW_NON_TRACKABLE)) { _SEH2_LEAVE; } /* Remember the failure for later notification */ if (NT_SUCCESS(OldStatus)) { OldStatus = Status; } } /* If we could open with FILE_WRITE_DATA both source and destination, * then, notify */ if (DestAccess & FILE_WRITE_DATA && SourceAccess & FILE_WRITE_DATA) { Status = BasepNotifyTrackingService(&SourceHandle, &ObjectAttributesSource, DestinationHandle, &NtDestination); if (!NT_SUCCESS(Status)) { if (dwFlags & PRIV_ALLOW_NON_TRACKABLE) { if (NT_SUCCESS(OldStatus)) OldStatus = Status; /* Reset status, we allow non trackable files */ Status = STATUS_SUCCESS; } } } } _SEH2_FINALLY { if (RelativeSource) RtlFreeHeap(RtlGetProcessHeap(), 0, RelativeSource); if (RelativeDestination) RtlFreeHeap(RtlGetProcessHeap(), 0, RelativeDestination); } _SEH2_END; /* If caller asked for source deletion, if everything succeed, proceed */ if (NT_SUCCESS(Status) && dwFlags & PRIV_DELETE_ON_SUCCESS) { FileDispositionInfo.DeleteFile = TRUE; Status = NtSetInformationFile(SourceHandle, &IoStatusBlock, &FileDispositionInfo, sizeof(FileDispositionInfo), FileDispositionInformation); } /* Cleanup/close portion */ if (DestinationHandle != INVALID_HANDLE_VALUE) { NtClose(DestinationHandle); } if (SourceHandle != INVALID_HANDLE_VALUE) { NtClose(SourceHandle); } /* Set last error if any, and quit */ if (NT_SUCCESS(Status)) { if (!NT_SUCCESS(OldStatus)) { BaseSetLastNTError(OldStatus); } } else { BaseSetLastNTError(Status); } return NT_SUCCESS(Status); } /* EOF */