mirror of
https://github.com/reactos/reactos.git
synced 2024-12-28 10:04:49 +00:00
Added support for MOVEFILE_DELAY_UNTIL_REBOOT, based on Wine code by Gerhard W. Gruber and others. Still requires smss to check the key.
svn path=/trunk/; revision=17440
This commit is contained in:
Steven Edwards
parent
c84b875eac
commit
ceb380247b
1 changed files with 150 additions and 0 deletions
|
|
@ -5,6 +5,7 @@
|
||||||
* FILE: lib/kernel32/file/file.c
|
* FILE: lib/kernel32/file/file.c
|
||||||
* PURPOSE: Directory functions
|
* PURPOSE: Directory functions
|
||||||
* PROGRAMMER: Ariadne ( ariadne@xs4all.nl)
|
* PROGRAMMER: Ariadne ( ariadne@xs4all.nl)
|
||||||
|
* Gerhard W. Gruber (sparhawk_at_gmx.at)
|
||||||
* UPDATE HISTORY:
|
* UPDATE HISTORY:
|
||||||
* Created 01/11/98
|
* Created 01/11/98
|
||||||
*/
|
*/
|
||||||
|
|
@ -147,6 +148,152 @@ AdjustFileAttributes (
|
||||||
return Result;
|
return Result;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/***********************************************************************
|
||||||
|
* add_boot_rename_entry
|
||||||
|
*
|
||||||
|
* Adds an entry to the registry that is loaded when windows boots and
|
||||||
|
* checks if there are some files to be removed or renamed/moved.
|
||||||
|
* <fn1> has to be valid and <fn2> may be NULL. If both pointers are
|
||||||
|
* non-NULL then the file is moved, otherwise it is deleted. The
|
||||||
|
* entry of the registrykey is always appended with two zero
|
||||||
|
* terminated strings. If <fn2> is NULL then the second entry is
|
||||||
|
* simply a single 0-byte. Otherwise the second filename goes
|
||||||
|
* there. The entries are prepended with \??\ before the path and the
|
||||||
|
* second filename gets also a '!' as the first character if
|
||||||
|
* MOVEFILE_REPLACE_EXISTING is set. After the final string another
|
||||||
|
* 0-byte follows to indicate the end of the strings.
|
||||||
|
* i.e.:
|
||||||
|
* \??\D:\test\file1[0]
|
||||||
|
* !\??\D:\test\file1_renamed[0]
|
||||||
|
* \??\D:\Test|delete[0]
|
||||||
|
* [0] <- file is to be deleted, second string empty
|
||||||
|
* \??\D:\test\file2[0]
|
||||||
|
* !\??\D:\test\file2_renamed[0]
|
||||||
|
* [0] <- indicates end of strings
|
||||||
|
*
|
||||||
|
* or:
|
||||||
|
* \??\D:\test\file1[0]
|
||||||
|
* !\??\D:\test\file1_renamed[0]
|
||||||
|
* \??\D:\Test|delete[0]
|
||||||
|
* [0] <- file is to be deleted, second string empty
|
||||||
|
* [0] <- indicates end of strings
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
static BOOL add_boot_rename_entry( LPCWSTR source, LPCWSTR dest, DWORD flags )
|
||||||
|
{
|
||||||
|
static const WCHAR ValueName[] = {'P','e','n','d','i','n','g',
|
||||||
|
'F','i','l','e','R','e','n','a','m','e',
|
||||||
|
'O','p','e','r','a','t','i','o','n','s',0};
|
||||||
|
static const WCHAR SessionW[] = {'M','a','c','h','i','n','e','\\',
|
||||||
|
'S','y','s','t','e','m','\\',
|
||||||
|
'C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t','\\',
|
||||||
|
'C','o','n','t','r','o','l','\\',
|
||||||
|
'S','e','s','s','i','o','n',' ','M','a','n','a','g','e','r',0};
|
||||||
|
static const int info_size = FIELD_OFFSET( KEY_VALUE_PARTIAL_INFORMATION, Data );
|
||||||
|
|
||||||
|
OBJECT_ATTRIBUTES attr;
|
||||||
|
UNICODE_STRING nameW, source_name, dest_name;
|
||||||
|
KEY_VALUE_PARTIAL_INFORMATION *info;
|
||||||
|
BOOL rc = FALSE;
|
||||||
|
HANDLE Reboot = 0;
|
||||||
|
DWORD len1, len2;
|
||||||
|
DWORD DataSize = 0;
|
||||||
|
BYTE *Buffer = NULL;
|
||||||
|
WCHAR *p;
|
||||||
|
|
||||||
|
DPRINT("Add support to smss for keys created by MOVEFILE_DELAY_UNTIL_REBOOT\n");
|
||||||
|
|
||||||
|
if (!RtlDosPathNameToNtPathName_U( (LPWSTR)source, &source_name, NULL, NULL ))
|
||||||
|
{
|
||||||
|
SetLastError( ERROR_PATH_NOT_FOUND );
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
dest_name.Buffer = NULL;
|
||||||
|
if (dest && !RtlDosPathNameToNtPathName_U( (LPWSTR)dest, &dest_name, NULL, NULL ))
|
||||||
|
{
|
||||||
|
RtlFreeUnicodeString( &source_name );
|
||||||
|
SetLastError( ERROR_PATH_NOT_FOUND );
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
attr.Length = sizeof(attr);
|
||||||
|
attr.RootDirectory = 0;
|
||||||
|
attr.ObjectName = &nameW;
|
||||||
|
attr.Attributes = 0;
|
||||||
|
attr.SecurityDescriptor = NULL;
|
||||||
|
attr.SecurityQualityOfService = NULL;
|
||||||
|
RtlInitUnicodeString( &nameW, SessionW );
|
||||||
|
|
||||||
|
if (NtCreateKey( &Reboot, KEY_ALL_ACCESS, &attr, 0, NULL, 0, NULL ) != STATUS_SUCCESS)
|
||||||
|
{
|
||||||
|
DPRINT1("Error creating key for reboot managment [%s]\n",
|
||||||
|
"SYSTEM\\CurrentControlSet\\Control\\Session Manager");
|
||||||
|
RtlFreeUnicodeString( &source_name );
|
||||||
|
RtlFreeUnicodeString( &dest_name );
|
||||||
|
return FALSE;
|
||||||
|
}
|
||||||
|
|
||||||
|
len1 = source_name.Length + sizeof(WCHAR);
|
||||||
|
if (dest)
|
||||||
|
{
|
||||||
|
len2 = dest_name.Length + sizeof(WCHAR);
|
||||||
|
if (flags & MOVEFILE_REPLACE_EXISTING)
|
||||||
|
len2 += sizeof(WCHAR); /* Plus 1 because of the leading '!' */
|
||||||
|
}
|
||||||
|
else len2 = sizeof(WCHAR); /* minimum is the 0 characters for the empty second string */
|
||||||
|
|
||||||
|
RtlInitUnicodeString( &nameW, ValueName );
|
||||||
|
|
||||||
|
/* First we check if the key exists and if so how many bytes it already contains. */
|
||||||
|
if (NtQueryValueKey( Reboot, &nameW, KeyValuePartialInformation,
|
||||||
|
NULL, 0, &DataSize ) == STATUS_BUFFER_OVERFLOW)
|
||||||
|
{
|
||||||
|
if (!(Buffer = HeapAlloc( GetProcessHeap(), 0, DataSize + len1 + len2 + sizeof(WCHAR) )))
|
||||||
|
goto Quit;
|
||||||
|
if (NtQueryValueKey( Reboot, &nameW, KeyValuePartialInformation,
|
||||||
|
Buffer, DataSize, &DataSize )) goto Quit;
|
||||||
|
info = (KEY_VALUE_PARTIAL_INFORMATION *)Buffer;
|
||||||
|
if (info->Type != REG_MULTI_SZ) goto Quit;
|
||||||
|
if (DataSize > sizeof(info)) DataSize -= sizeof(WCHAR); /* remove terminating null (will be added back later) */
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
DataSize = info_size;
|
||||||
|
if (!(Buffer = HeapAlloc( GetProcessHeap(), 0, DataSize + len1 + len2 + sizeof(WCHAR) )))
|
||||||
|
goto Quit;
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy( Buffer + DataSize, source_name.Buffer, len1 );
|
||||||
|
DataSize += len1;
|
||||||
|
p = (WCHAR *)(Buffer + DataSize);
|
||||||
|
if (dest)
|
||||||
|
{
|
||||||
|
if (flags & MOVEFILE_REPLACE_EXISTING)
|
||||||
|
*p++ = '!';
|
||||||
|
memcpy( p, dest_name.Buffer, len2 );
|
||||||
|
DataSize += len2;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
*p = 0;
|
||||||
|
DataSize += sizeof(WCHAR);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* add final null */
|
||||||
|
p = (WCHAR *)(Buffer + DataSize);
|
||||||
|
*p = 0;
|
||||||
|
DataSize += sizeof(WCHAR);
|
||||||
|
|
||||||
|
rc = !NtSetValueKey(Reboot, &nameW, 0, REG_MULTI_SZ, Buffer + info_size, DataSize - info_size);
|
||||||
|
|
||||||
|
Quit:
|
||||||
|
RtlFreeUnicodeString( &source_name );
|
||||||
|
RtlFreeUnicodeString( &dest_name );
|
||||||
|
if (Reboot) NtClose(Reboot);
|
||||||
|
HeapFree( GetProcessHeap(), 0, Buffer );
|
||||||
|
return(rc);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* @implemented
|
* @implemented
|
||||||
|
|
@ -170,6 +317,9 @@ MoveFileWithProgressW (
|
||||||
|
|
||||||
DPRINT("MoveFileWithProgressW()\n");
|
DPRINT("MoveFileWithProgressW()\n");
|
||||||
|
|
||||||
|
if (dwFlags & MOVEFILE_DELAY_UNTIL_REBOOT)
|
||||||
|
return add_boot_rename_entry( lpExistingFileName, lpNewFileName, dwFlags );
|
||||||
|
|
||||||
hFile = CreateFileW (lpExistingFileName,
|
hFile = CreateFileW (lpExistingFileName,
|
||||||
GENERIC_ALL,
|
GENERIC_ALL,
|
||||||
FILE_SHARE_WRITE|FILE_SHARE_READ,
|
FILE_SHARE_WRITE|FILE_SHARE_READ,
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue