resource_control.cpp

#include <windows.h>
#include <stdint.h>
#include "utils.h"

char aPatchC[] = "data\\patch\\";
char aDataC[] = "data\\";

char aLocale_res[] = "locale.res";

bool _stdcall TestMap(const char* file) // if this returns true, the file's path is unchanged for loading
{
	std::string filename = TruncateSlashes(FixSlashes(ToLower(file)));
	if ((filename.find("graphics/") == 0) ||
		(filename.find("locale/") == 0) ||
		(filename.find("main/") == 0) ||
		(filename.find("movies/") == 0) ||
		(filename.find("music/") == 0) ||
		(filename.find("patch/") == 0) ||
		(filename.find("scenario/") == 0) ||
		(filename.find("sfx/") == 0) ||
		(filename.find("speech/") == 0) ||
		(filename.find("video/") == 0) ||
		(filename.find("world/") == 0))
			return false;

	return true;
}

int __declspec(naked) RES_loadFile(int x)
/* èçìåíåííàÿ ïðîöåäóðà çàãðóçêè ðåñóðñîâ;
 * ñíà÷àëà ïûòàåìñÿ çàãðóçèòü ôàéë èç patch11.res
 * ïîòîì èç a25graph.res, åñëè îí ïîäãðóæåí,
 * ïîñëå èç îðèãèíàëüíîãî ðåñóðñà */
// ZZYZX EDIT: a25graph.res íå èñïîëüçóåòñÿ.
{
	__asm
	{ // 004EBEDA
		call	sub_4EB99D

		mov		[ebp-0x814], eax
		cmp		eax, 0
		jz		loc_4EBF9B
		mov		edx, 0x004EBEEE
		jmp		edx
loc_4EBF9B:
		mov		edx, 0x004EBF9B
		jmp		edx

sub_4EB99D:
		push	ebp
		mov		ebp, esp

		sub		esp, 8
		mov		[ebp-4], ecx

		push	[ebp+8]
		call	TestMap
		and		eax, 0xFF
		test	eax, eax
		jnz		regular_load

		push	0x200
		mov		edx, 0x0058CA80
		call	edx	/// malloc
		add		esp, 4
		mov		[ebp-8], eax

		push	offset aPatchC
		push	[ebp-8]
		mov		edx, 0x0058CDF0
		call	edx	// strcpy
		add		esp, 8

		push	dword ptr [ebp+8]
		push	dword ptr [ebp-8]
		mov		edx, 0x0058CE00 // strcat
		call	edx
		add		esp, 8

		push	dword ptr [ebp-8]
		mov		ecx, [ebp-4]
		mov		edx, 0x004EB99D
		call	edx
		test	eax, eax
		jnz		en

		// concat with data

		push	offset aDataC
		push	[ebp-8]
		mov		edx, 0x0058CDF0
		call	edx	// strcpy
		add		esp, 8

		push	dword ptr [ebp+8]
		push	dword ptr [ebp-8]
		mov		edx, 0x0058CE00 // strcat
		call	edx
		add		esp, 8

		push	dword ptr [ebp-8]
		mov		ecx, [ebp-4]
		mov		edx, 0x004EB99D
		call	edx
		jmp		en

regular_load:
		push	dword ptr [ebp+8]
		mov		ecx, [ebp-4]
		mov		edx, 0x004EB99D
		call	edx
		jmp		en_nofree

en:

		mov		[ebp-4], eax
		push	dword ptr [ebp-8]
		mov		edx, 0x0058CB50
		call	edx /// free
		add		esp, 4
		mov		eax, [ebp-4]

en_nofree:

		mov		esp, ebp
		pop		ebp
		retn	4

loc_retzero:
		mov		eax, 0
		
		mov		esp, ebp
		pop		ebp
		retn	4
	}
}

void __declspec(naked) RES_loadException()
{ // 004EB723
	__asm
	{
		mov	edx, 0x004EB72E
		jmp	edx
	}
}

void _stdcall CheckMusic()
{
	// check if data/music directory exists
	// if not, disable music
	uint32_t dwAttrib = GetFileAttributes("data\\music");

	bool exists = (dwAttrib != INVALID_FILE_ATTRIBUTES && 
				  (dwAttrib & FILE_ATTRIBUTE_DIRECTORY));

	if (!exists)
		*(uint32_t*)(0x0062C870) = 0; // disable music
}

void __declspec(naked) RES_loadDirectories()
{ // 004867E6
	__asm
	{
		push	offset aLocale_res
		mov		edx, 0x004EB741
		call	edx
		add		esp, 4

		mov		byte ptr [ebp-0x04], 4

		push	0x005FD168 // offset aMusic_res; "music.res"
		mov		edx, 0x004EB741
		call	edx
		add		esp, 4

		call	CheckMusic

		mov		edx, 0x0048681A
		jmp		edx
	}
}

void __declspec(naked) RES_noLoad()
{
	__asm
	{ // 4EB741
		push	ebp
		mov		ebp, esp
		// do nothing
		mov		esp, ebp
		pop		ebp
		retn
	}
}