moveonreboot.cpp

#include "common.h"
#include "Support\\LWAnsiString\\LWAnsiString.h"
#include "osver.h"
#include "IAT_ENV.H"
#include "IAT_FILE.H"






void LegacyMarkForDelete(const char* name, LWAnsiString* vebal)
{
	if (! (IAT_DynamicLink_FileApi(IAT_FILE_LINKING_IMPORTS_ANIS) & IAT_FILE_LINKING_IMPORTS_ANIS) == IAT_FILE_LINKING_IMPORTS_ANIS)
	{
		return;
	}
	
	if (!(IAT_DynamicLink_Environment(IAT_ENV_API_ALL_A) & IAT_ENV_API_ALL_A) == IAT_ENV_API_ALL_A)
	{
		return;
	}




	else
	{
		LWAnsiString* WinDir = LWAnsiString_CreateString(MAX_PATH);
		LWAnsiString_MarkLenDirty(WinDir);
		DWORD size = IAT_GetWindowsDirectoryA(WinDir->AnsiData, MAX_PATH);
		if (size > MAX_PATH)
		{
			LWAnsiString_Reserve(WinDir, size + 1);
			SetLastError(0);
			size = IAT_GetWindowsDirectoryA(WinDir->AnsiData, size + 1);
			if (!LWAnsiString_EndsWith(WinDir, "\\", false))
			{
				LWAnsiString_AppendA(WinDir, "\\");
			}
		}
		else
		{
			if (!LWAnsiString_EndsWith(WinDir, "\\", false))
			{
				LWAnsiString_AppendA(WinDir, "\\");
			}
		}

	

		// now the windir
		LWAnsiString_AppendA(WinDir, "wininit.ini");

		{
			DWORD ShortPath = IAT_GetShortPathA(name, NULL, 0);
			if (ShortPath != 0)
			{
				LWAnsiString* EntryLine = LWAnsiString_CreateString(1);
				LWAnsiString_AppendA(EntryLine, "NUL=");
				LWAnsiString_AddReserve(EntryLine, ShortPath + 1);
				char* offset = (char*)LWAnsiString_EndingOffset(EntryLine);
				
				if (IAT_GetShortPathA(name, offset, ShortPath) != 0)
				{
					LWAnsiString_MarkLenDirty(EntryLine);
					LWAnsiString_AppendNewLine(EntryLine);



					HANDLE file = IAT_CreateFileA(LWAnsiString_ToCStr(WinDir), GENERIC_WRITE, 0, 0, OPEN_ALWAYS, FILE_ATTRIBUTE_NORMAL, 0);
					if (file != INVALID_HANDLE_VALUE)
					{
						DWORD wrote = 0;
						if (GetLastError() != ERROR_ALREADY_EXISTS)
						{
							WriteFile(file, "[rename]\r\n", 10, &wrote, 0);
						}
						SetLastError(0);
						if ((IAT_SetFilePointer(file, 0, 0, FILE_END) != INVALID_SET_FILE_POINTER) && (GetLastError() == NO_ERROR))
						{
							if (!WriteFile(file, LWAnsiString_ToCStr(EntryLine), EntryLine->Length * sizeof(char), &wrote, 0))
							{
								LWAnsiString_AppendWithNewLineA(vebal, "Failed to write the new entry to delete on reboot");
							}
							else
							{
								LWAnsiString_AppendA(vebal, "");
								LWAnsiString_AppendA(vebal, LWAnsiString_ToCStr(EntryLine));
								LWAnsiString_AppendA(vebal, "----> registered for deletion in ");
								LWAnsiString_AppendA(vebal, LWAnsiString_ToCStr(WinDir));
								LWAnsiString_AppendWithNewLineA(vebal, " OK ");
							}
						}
						else
						{
							LWAnsiString_AppendWithNewLineA(vebal, "Failed to seek to end of wininit.ini to write the pending delete on reboot");
						}
						CloseHandle(file);
					}
					else
					{
						LWAnsiString_AppendA(vebal, "Failed to Create / Open wininit.ini in the expected location of  ");
						LWAnsiString_AppendA(vebal, LWAnsiString_ToCStr(WinDir));
						LWAnsiString_AppendNewLineA(vebal);
						LWAnsiString_AppendA(vebal, "Error code in attempt is ");
						LWAnsiString_AppendNumberA(GetLastError(), vebal, 0);
						LWAnsiString_AppendNewLineA(vebal);
					}

				}
				else
				{
					LWAnsiString_AppendWithNewLineA(vebal, "Failed to get the short (8.3 DOS naming) size for the pending delete.");
				}
				LWAnsiString_FreeString(EntryLine);
			}
			else
			{
				LWAnsiString_AppendWithNewLineA(vebal, "Failed to get the short (8.3 DOS naming) size for the pending delete.");
			}
		}
		

		// free
		LWAnsiString_FreeString(WinDir);
	}

}
bool DeleteOnReboot(int* result, const char** message_result, const char* argv[], int argc)
{
	MyOSVERSIONINFO osvi;
	bool UnicodeVer = FALSE;
	FetchVersionInfo(&osvi, &UnicodeVer);

	LWAnsiString* vebal = LWAnsiString_CreateString(1);
	BOOL LegacyOS = FALSE;
	if (argc < 2)
	{
		LWAnsiString_FreeString(vebal);
		return false;
	}
	else
	{
		if ((result == 0 || (message_result == 0)))
		{
			LWAnsiString_FreeString(vebal);
			return false;
		}
		else
		{
			if (((IAT_DynamicLink_FileApi(IAT_FILE_LINKING_IMPORTS_ANIS) & (IAT_FILE_LINKING_IMPORTS_ANIS))) != (IAT_FILE_LINKING_IMPORTS_ANIS))
			{
				LWAnsiString_FreeString(vebal);
				return false;
			}


			if ((osvi.A.dwMajorVersion < 5) || (osvi.A.dwPlatformId != VER_PLATFORM_WIN32_WINDOWS))
			{
				LWAnsiString_AppendWithNewLineA(vebal, "Warning: Windows verison reports OS possibly older than Windows 2000. Modern register for delete might not work ");
				LegacyOS = TRUE;
			}

			if (IAT_MoveFileExA == 0)
			{
				if (LegacyOS)
				{
					LWAnsiString_AppendWithNewLineA(vebal, "Failed to get MoveFileExA routine. Falling back to legacy register. Note. This will NOT work on NT based OS.");
					for (int i = 2; i < argc; i++)
					{
						LegacyMarkForDelete(argv[i], vebal);
					}
				}
				else
				{
					LWAnsiString_AppendWithNewLineA(vebal, "Failed to get MoveFileExA routine. No files registered for deletion. ");
				}
			}
			else
			{

				for (int i = 2; i < argc; i++)
				{
					SetLastError(0);
					if (IAT_MoveFileExA(argv[i], NULL, MOVEFILE_DELAY_UNTIL_REBOOT))
					{
						LWAnsiString_AppendA(vebal, argv[i]);
						LWAnsiString_AppendWithNewLineA(vebal, " Registered for deletion. ");
					}
					else
					{
						LWAnsiString_AppendA(vebal, argv[i]);
						LWAnsiString_AppendWithNewLineA(vebal, " Failed to Register for Deletion.");
					}
				}
			}
		}
		WriteStdout(vebal->AnsiData);
		LWAnsiString_FreeString(vebal);
	}
	SetLastError(0);
	return true;

}