UseCustomDLL.qs

function UseCustomDLL() {
	var dir = GetDataDirectory(1);
	var finalValue = " 00".repeat(20);
	var dirData = "";
	
	var hasHShield = (exe.getActivePatches().indexOf(15) != -1);	
	if (hasHShield) {
		var aOffset = exe.findString('aossdk.dll', RAW);
		if (aOffset != -1) {
			aOffset = " 00".repeat(8) + (exe.Raw2Rva(aOffset) - exe.getImageBase()).packToHex(4);
		}
	}
	var offset = dir.offset;
	var curValue = exe.fetchHex(offset,20);
	do {
		offset += 20;
		if (!hasHShield || curValue.indexOf(aOffset) !== 3*4) {
			dirData = dirData + curValue;			
		}
		curValue = exe.fetchHex(offset,20);
	} while(curValue != finalValue);
	
	var fp = new TextFile();
	if (!getInputFile(fp, '$customDLL', 'File Input - Use Custom DLL', 'Enter the DLL info file', APP_PATH + "/Input/dlls.txt")) {
		return "Patch Cancelled";
	}
	
	var dllNames = new Array();
	var fnNames = new Array();
	var dptr = -1;
	
	while (!fp.eof()) {
		line = fp.readline().trim();
		if (line === "" || line.indexOf("//") == 0) continue;
		if ((line.indexOf(".dll") - line.length) == -4) {
			dptr++;
			dllNames.push({"offset":0, "value":line});
			fnNames[dptr] = new Array();
		}
		else {
			fnNames[dptr].push({"offset":0, "value":line});
		}
	}
	fp.close();
	
	var dirSize = dirData.hexlength();//Holds the size of Import Directory Table and IAT values
	var strData = "";
	var strSize = 0;//Holds the size of dll names and function names
	
	for (var i = 0; i < dllNames.length; i++) {
		if (fnNames[i].length == 0) continue;		
		var name = dllNames[i].value;
		dllNames[i].offset = strSize;
		strData = strData + name.toHex() + " 00";		
		strSize = strSize + name.length + 1;//Space for name
		dirSize = dirSize + 20 ;//IDIR Entry Size
	
		for (var j = 0; j < fnNames[i].length; j++) {
			var name = fnNames[i][j].value;
			fnNames[i][j].offset = strSize;
			strData = strData + j.packToHex(2) + name.toHex() + " 00";
			strSize = strSize + 2 + name.length + 1;//Space for name
			
			if (name.length % 2 != 0) {//Even the Odds xD
				strData = strData + " 00";
				strSize++;
			}
			
			dirSize = dirSize + 4; //Thunk Value RVAs & Ordinals
		}
		dirSize += 4;//Last Value is 00 00 00 00 after Thunks
	}	
	dirSize += 20;//Accomodate for IAT End Entry

	var free = exe.findZeros(strSize + dirSize);
	var baseAddr = exe.Raw2Rva(free) - exe.getImageBase();
	var prefix = " 00".repeat(12);	
	var dirEntryData = "";
	var dirTableData = "";
	
	var dptr = 0;
	for (var i = 0; i < dllNames.length; i++) {
		if (fnNames[i].length == 0) continue;
		dirTableData = dirTableData + prefix + (baseAddr + dllNames[i].offset).packToHex(4) + (baseAddr + strSize + dptr).packToHex(4);
		
		for (var j = 0; j < fnNames[i].length; j++) {
			dirEntryData = dirEntryData + (baseAddr + fnNames[i][j].offset).packToHex(4);
			dptr += 4;
		}
		dirEntryData = dirEntryData + " 00 00 00 00";
		dptr += 4;	
	}
	dirTableData = dirData + dirTableData + finalValue;	
	exe.insert(free, strSize + dirSize, strData + dirEntryData + dirTableData, PTYPE_HEX);
	
	var PEoffset = exe.find("50 45 00 00", PTYPE_HEX, false);	
	exe.replaceDWord(PEoffset + 0x18 + 0x60 + 0x8, baseAddr + strSize + dirEntryData.hexlength() );
	exe.replaceDWord(PEoffset + 0x18 + 0x60 + 0xC, dirTableData.hexlength() - 20);

	//Hint for HShield Patch to not conflict with each other.
	Imp_DATA = {"offset":free, 
							"valuePre":strData + dirEntryData,
							"valueSuf":dirTableData,
							"tblAddr":baseAddr + strSize + dirEntryData.hexlength(),
							"tblSize":dirTableData.hexlength() - 20
						};

	return true;
}