[sedwards] 17440: Added support for MOVEFILE_DELAY_UNTIL_REBOOT, based on Wine code by Gerhard W. Gruber and others. Still requires smss to check the key.

Added support for MOVEFILE_DELAY_UNTIL_REBOOT, based on Wine code by
Gerhard W. Gruber and others. Still requires smss to check the key.
Modified: trunk/reactos/lib/kernel32/file/move.c
  _____  

Modified: trunk/reactos/lib/kernel32/file/move.c
--- 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(a)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,