--- trunk/reactos/lib/kernel32/file/move.c 2005-08-19 17:13:37 UTC (rev 17439)
+++ trunk/reactos/lib/kernel32/file/move.c 2005-08-19 19:05:59 UTC (rev 17440)
@@ -5,6 +5,7 @@
* FILE: lib/kernel32/file/file.c
* PURPOSE: Directory functions
* PROGRAMMER: Ariadne ( ariadne@xs4all.nl)
+ * Gerhard W. Gruber (sparhawk_at_gmx.at)
* UPDATE HISTORY:
* Created 01/11/98
*/
@@ -147,7 +148,153 @@
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
*/
@@ -170,6 +317,9 @@
DPRINT("MoveFileWithProgressW()\n");
+ if (dwFlags & MOVEFILE_DELAY_UNTIL_REBOOT)
+ return add_boot_rename_entry( lpExistingFileName, lpNewFileName, dwFlags );
+
hFile = CreateFileW (lpExistingFileName,
GENERIC_ALL,
FILE_SHARE_WRITE|FILE_SHARE_READ,