| /************************************************************** |
| * |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you under the Apache License, Version 2.0 (the |
| * "License"); you may not use this file except in compliance |
| * with the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| * |
| *************************************************************/ |
| |
| |
| |
| #include "macros.h" |
| |
| #define WININIT_FILENAME "wininit.ini" |
| #define RENAME_SECTION "rename" |
| |
| IMPLEMENT_THUNK( kernel32, WINDOWS, BOOL, WINAPI, MoveFileExA, ( LPCSTR lpExistingFileNameA, LPCSTR lpNewFileNameA, DWORD dwFlags ) ) |
| { |
| BOOL fSuccess = FALSE; // assume failure |
| |
| // Windows 9x has a special mechanism to move files after reboot |
| |
| if ( dwFlags & MOVEFILE_DELAY_UNTIL_REBOOT ) |
| { |
| CHAR szExistingFileNameA[MAX_PATH]; |
| CHAR szNewFileNameA[MAX_PATH] = "NUL"; |
| |
| // Path names in WININIT.INI must be in short path name form |
| |
| if ( |
| GetShortPathNameA( lpExistingFileNameA, szExistingFileNameA, MAX_PATH ) && |
| (!lpNewFileNameA || GetShortPathNameA( lpNewFileNameA, szNewFileNameA, MAX_PATH )) |
| ) |
| { |
| CHAR szBuffer[32767]; // The buffer size must not exceed 32K |
| DWORD dwBufLen = GetPrivateProfileSectionA( RENAME_SECTION, szBuffer, elementsof(szBuffer), WININIT_FILENAME ); |
| |
| CHAR szRename[MAX_PATH]; // This is enough for at most to times 67 chracters |
| strcpy( szRename, szNewFileNameA ); |
| strcat( szRename, "=" ); |
| strcat( szRename, szExistingFileNameA ); |
| size_t lnRename = strlen(szRename); |
| |
| if ( dwBufLen + lnRename + 2 <= elementsof(szBuffer) ) |
| { |
| CopyMemory( &szBuffer[dwBufLen], szRename, lnRename ); |
| szBuffer[dwBufLen + lnRename ] = 0; |
| szBuffer[dwBufLen + lnRename + 1 ] = 0; |
| |
| fSuccess = WritePrivateProfileSectionA( RENAME_SECTION, szBuffer, WININIT_FILENAME ); |
| } |
| else |
| SetLastError( ERROR_BUFFER_OVERFLOW ); |
| } |
| } |
| else |
| { |
| |
| fSuccess = MoveFileA( lpExistingFileNameA, lpNewFileNameA ); |
| |
| if ( !fSuccess && 0 != (dwFlags & (MOVEFILE_COPY_ALLOWED | MOVEFILE_REPLACE_EXISTING)) ) |
| { |
| BOOL bFailIfExist = 0 == (dwFlags & MOVEFILE_REPLACE_EXISTING); |
| |
| fSuccess = CopyFileA( lpExistingFileNameA, lpNewFileNameA, bFailIfExist ); |
| |
| // In case of successfull copy do not return FALSE if delete fails. |
| // Error detection is done by GetLastError() |
| |
| if ( fSuccess ) |
| { |
| SetLastError( NO_ERROR ); |
| DeleteFileA( lpExistingFileNameA ); |
| } |
| } |
| |
| } |
| |
| return fSuccess; |
| } |
| |