1 /**************************************************************
2  *
3  * Licensed to the Apache Software Foundation (ASF) under one
4  * or more contributor license agreements.  See the NOTICE file
5  * distributed with this work for additional information
6  * regarding copyright ownership.  The ASF licenses this file
7  * to you under the Apache License, Version 2.0 (the
8  * "License"); you may not use this file except in compliance
9  * with the License.  You may obtain a copy of the License at
10  *
11  *   http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing,
14  * software distributed under the License is distributed on an
15  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16  * KIND, either express or implied.  See the License for the
17  * specific language governing permissions and limitations
18  * under the License.
19  *
20  *************************************************************/
21 
22 
23 
24 #include "macros.h"
25 
26 #define	WININIT_FILENAME	"wininit.ini"
27 #define RENAME_SECTION		"rename"
28 
29 IMPLEMENT_THUNK( kernel32, WINDOWS, BOOL, WINAPI, MoveFileExA, ( LPCSTR lpExistingFileNameA, LPCSTR lpNewFileNameA, DWORD dwFlags ) )
30 {
31 	BOOL	fSuccess = FALSE;	// assume failure
32 
33 	// Windows 9x has a special mechanism to move files after reboot
34 
35 	if ( dwFlags & MOVEFILE_DELAY_UNTIL_REBOOT )
36 	{
37 		CHAR	szExistingFileNameA[MAX_PATH];
38 		CHAR	szNewFileNameA[MAX_PATH] = "NUL";
39 
40 		// Path names in WININIT.INI must be in short path name form
41 
42 		if (
43 			GetShortPathNameA( lpExistingFileNameA, szExistingFileNameA, MAX_PATH ) &&
44 			(!lpNewFileNameA || GetShortPathNameA( lpNewFileNameA, szNewFileNameA, MAX_PATH ))
45 			)
46 		{
47 			CHAR	szBuffer[32767];	// The buffer size must not exceed 32K
48 			DWORD	dwBufLen = GetPrivateProfileSectionA( RENAME_SECTION, szBuffer, elementsof(szBuffer), WININIT_FILENAME );
49 
50 			CHAR	szRename[MAX_PATH];	// This is enough for at most to times 67 chracters
51 			strcpy( szRename, szNewFileNameA );
52 			strcat( szRename, "=" );
53 			strcat( szRename, szExistingFileNameA );
54 			size_t	lnRename = strlen(szRename);
55 
56 			if ( dwBufLen + lnRename + 2 <= elementsof(szBuffer) )
57 			{
58 				CopyMemory( &szBuffer[dwBufLen], szRename, lnRename );
59 				szBuffer[dwBufLen + lnRename ] = 0;
60 				szBuffer[dwBufLen + lnRename + 1 ] = 0;
61 
62 				fSuccess = WritePrivateProfileSectionA( RENAME_SECTION, szBuffer, WININIT_FILENAME );
63 			}
64 			else
65 				SetLastError( ERROR_BUFFER_OVERFLOW );
66 		}
67 	}
68 	else
69 	{
70 
71 		fSuccess = MoveFileA( lpExistingFileNameA, lpNewFileNameA );
72 
73 		if ( !fSuccess && 0 != (dwFlags & (MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING)) )
74 		{
75 			BOOL	bFailIfExist = 0 == (dwFlags & MOVEFILE_REPLACE_EXISTING);
76 
77 			fSuccess = CopyFileA( lpExistingFileNameA, lpNewFileNameA, bFailIfExist );
78 
79 			// In case of successfull copy do not return FALSE if delete fails.
80 			// Error detection is done by GetLastError()
81 
82 			if ( fSuccess )
83 			{
84 				SetLastError( NO_ERROR );
85 				DeleteFileA( lpExistingFileNameA );
86 			}
87 		}
88 
89 	}
90 
91 	return fSuccess;
92 }
93 
94