add option to templates to generate 14 byte jmp scripts

Author: cheat-engine

Cheat Engine/LuaHandler.pas

@@ -12636,6 +12636,8 @@ function lua_GenerateCodeInjectionScript(L: PLua_state): integer; cdecl;
 var
   script: TStrings;
   address: string;
+
+  farjmp: boolean;
 begin
   result:=0;
   if lua_gettop(L)>=1 then
@@ -12647,8 +12649,13 @@ function lua_GenerateCodeInjectionScript(L: PLua_state): integer; cdecl;
     else
       address:=inttohex(MemoryBrowser.disassemblerview.SelectedAddress,8);
 
+    if lua_gettop(L)>=3 then
+      farjmp:=lua_toboolean(L,3)
+    else
+      farjmp:=false;
+
     try
-      GenerateCodeInjectionScript(script, address);
+      GenerateCodeInjectionScript(script, address,farjmp);
       lua_pushboolean(L,true);
       result:=1;
     except
@@ -12661,6 +12668,7 @@ function lua_GenerateAOBInjectionScript(L: PLua_state): integer; cdecl;
   script: TStrings;
   address, symbolname: string;
   lineCountToCopy: integer;
+  farjmp: boolean;
 begin
   result:=0;
   if lua_gettop(L)>=2 then
@@ -12678,6 +12686,11 @@ function lua_GenerateAOBInjectionScript(L: PLua_state): integer; cdecl;
     else
       linecountToCopy:=20;
 
+    if lua_gettop(L)>=5 then
+      farjmp:=lua_toboolean(L,5)
+    else
+      farjmp:=false;
+
     try
       GenerateAOBInjectionScript(script, address, symbolname, lineCountToCopy);
       lua_pushboolean(L,true);
@@ -12692,6 +12705,7 @@ function lua_GenerateFullInjectionScript(L: PLua_state): integer; cdecl;
   script: TStrings;
   address: string;
   lineCountToCopy: integer;
+  farjmp: boolean;
 begin
   result:=0;
   if lua_gettop(L)>=1 then
@@ -12708,6 +12722,11 @@ function lua_GenerateFullInjectionScript(L: PLua_state): integer; cdecl;
     else
       linecountToCopy:=20;
 
+    if lua_gettop(L)>=4 then
+      farjmp:=lua_toboolean(L,4)
+    else
+      farjmp:=false;
+
     try
       GenerateFullInjectionScript(script, address, lineCountToCopy);
       lua_pushboolean(L,true);

Cheat Engine/bin/celua.txt

@@ -320,9 +320,9 @@ registerAutoAssemblerTemplate(name, function(script: TStrings; sender: TFrmAutoI
 unregisterAutoAssemblerTemplate(ID)
 
 
-generateCodeInjectionScript(script: Tstrings, address: string) - Adds a default codeinjection script to the given script
-generateAOBInjectionScript(script: Tstrings, symbolname: string, address: string, commentradius(default 10)) - Adds an AOB injection script to the given script
-generateFullInjectionScript(script: Tstrings, address: string, commentradius(default 10)) - Adds a Full Injection script to the given script
+generateCodeInjectionScript(script: Tstrings, address: string, farjmp: boolean) - Adds a default codeinjection script to the given script
+generateAOBInjectionScript(script: Tstrings, symbolname: string, address: string, commentradius(default 10), farjmp: boolean) - Adds an AOB injection script to the given script
+generateFullInjectionScript(script: Tstrings, address: string, commentradius(default 10), farjmp: boolean) - Adds a Full Injection script to the given script
 
 getNextAllocNumber(script: TStrings): integer - scans the given script for alloc(newmem## and returns the next unused newmem number)
 addSnapshotAsComment(script: TStrings, address: integer, radius(Default 10)) - creates a comment section for AA scripts that contains a snapshot of the original code

Cheat Engine/frmautoinjectunit.pas

@@ -384,9 +384,9 @@   TfrmAutoInject = class(TForm)
 
 procedure Getjumpandoverwrittenbytes(address,addressto: ptrUINT; jumppart,originalcodepart: tstrings);
 procedure generateAPIHookScript(script: tstrings; address: string; addresstogoto: string; addresstostoreneworiginalfunction: string=''; nameextension:string='0'; targetself: boolean=false);
-procedure GenerateCodeInjectionScript(script: tstrings; addressstring: string);
-procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolname: string; commentradius: integer=10);
-procedure GenerateFullInjectionScript(Script: tstrings; address: string; commentradius: integer=10);
+procedure GenerateCodeInjectionScript(script: tstrings; addressstring: string; farjmp: boolean=false);
+procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolname: string; commentradius: integer=10; farjmp: boolean=false);
+procedure GenerateFullInjectionScript(Script: tstrings; address: string; commentradius: integer=10; farjmp: boolean=false);
 
 function registerAutoAssemblerTemplate(name: string; m: TAutoAssemblerTemplateCallback; shortcut: TShortCut=0): integer;
 procedure unregisterAutoAssemblerTemplate(id: integer);
@@ -984,7 +984,7 @@ procedure TfrmAutoInject.FormClose(Sender: TObject;
 {$endif}
 end;
 
-procedure GenerateCodeInjectionScript(script: tstrings; addressstring: string);
+procedure GenerateCodeInjectionScript(script: tstrings; addressstring: string; farjmp: boolean=false);
 function inttostr(i:int64):string;
 begin
   if i=0 then result:='' else result:=sysutils.IntToStr(i);
@@ -1005,7 +1005,13 @@ function inttostr(i:int64):string;
     disablepos: integer;
     enablecode: tstringlist;
     disablecode: tstringlist;
+    jmpsize: integer;
 begin
+  if not processhandler.is64Bit then
+    farjmp:=false;
+
+  jmpsize:=ifthen(farjmp, 14, 5);
+
   try
     a:=StrToQWordEx('$'+addressstring);
   except
@@ -1020,7 +1026,7 @@ function inttostr(i:int64):string;
   setlength(originalcode,0);
   codesize:=0;
 
-  while codesize<5 do
+  while codesize<jmpsize do
   begin
     setlength(originalcode,length(originalcode)+1);
     originalcode[length(originalcode)-1]:=disassemble(c,x);
@@ -1038,7 +1044,7 @@ function inttostr(i:int64):string;
   try
     with enablecode do
     begin
-      if processhandler.is64bit then
+      if processhandler.is64bit and (not farjmp) then
         add('alloc(newmem'+inttostr(injectnr)+',2048,'+addressstring+') ')
       else
         add('alloc(newmem'+inttostr(injectnr)+',2048)');
@@ -1059,11 +1065,14 @@ function inttostr(i:int64):string;
 
       add('');
       add(addressstring+':');
-      add('jmp newmem'+inttostr(injectnr)+'');
-      if codesize>5 then
+      if farjmp then
+        add('jmp far newmem'+inttostr(injectnr)+'')
+      else
+        add('jmp newmem'+inttostr(injectnr)+'');
+      if codesize>jmpsize then
       begin
-        if codesize-5>1 then
-          add('nop '+inttohex(codesize-5,1))
+        if codesize-jmpsize>1 then
+          add('nop '+inttohex(codesize-jmpsize,1))
         else
           add('nop');
       end;
@@ -1134,7 +1143,7 @@ procedure TfrmAutoInject.Codeinjection1Click(Sender: TObject);
     address:=symhandler.getNameFromAddress(a);
 
   if inputquery(rsCodeInjectTemplate, rsOnWhatAddressDoYouWantTheJump, address) then
-    GenerateCodeInjectionScript(assemblescreen.lines, address);
+    GenerateCodeInjectionScript(assemblescreen.lines, address, ssCtrl in GetKeyShiftState);
 end;
 
 procedure TfrmAutoInject.Panel1Resize(Sender: TObject);
@@ -2839,7 +2848,7 @@ function GetNextAllocNumber(Script: TStrings): integer;
 end;
 
 // \/   http://forum.cheatengine.org/viewtopic.php?t=566415 (jgoemat and some mods by db)
-procedure GenerateFullInjectionScript(Script: tstrings; address: string; commentRadius: integer=10);
+procedure GenerateFullInjectionScript(Script: tstrings; address: string; commentRadius: integer=10; farjmp: boolean=false);
 var
   originalcode: array of string;
   originalbytes: array of byte;
@@ -2874,7 +2883,12 @@ procedure GenerateFullInjectionScript(Script: tstrings; address: string; comment
   ddBytes: string;
 
   mi: TModuleInfo;
+  jmpsize: integer;
 begin
+  if not processhandler.is64Bit then
+    farjmp:=false;
+
+  jmpsize:=ifthen(farjmp, 14, 5);
 
   try
     a:=StrToQWordEx('$'+address);
@@ -2896,7 +2910,7 @@ procedure GenerateFullInjectionScript(Script: tstrings; address: string; comment
   setlength(originalcode,0);
   codesize:=0;
 
-  while codesize<5 do
+  while codesize<jmpsize do
   begin
     setlength(originalcode,length(originalcode)+1);
     originalcode[length(originalcode)-1]:=disassemble(c,x);
@@ -2954,7 +2968,7 @@ procedure GenerateFullInjectionScript(Script: tstrings; address: string; comment
     with enablecode do
     begin
       add('assert(address'+nr+',bytes'+nr+')');
-      if processhandler.is64bit then
+      if processhandler.is64bit and (not farjmp) then
         add('alloc(newmem' + nr + ',$1000,' + address + ')')
       else
         add('alloc(newmem' + nr + ',$1000)');
@@ -2972,11 +2986,14 @@ procedure GenerateFullInjectionScript(Script: tstrings; address: string; comment
 
       add('');
       add('address'+nr+':');
-      add('  jmp newmem'+nr+'');
-      if codesize>5 then
+      if farjmp then
+        add('  jmp far newmem'+nr+'')
+      else
+        add('  jmp newmem'+nr+'');
+      if codesize>jmpsize then
       begin
-        if codesize-5>1 then
-          add('  nop '+inttohex(codesize-5,1))
+        if codesize-jmpsize>1 then
+          add('  nop '+inttohex(codesize-jmpsize,1))
         else
           add('  nop');
       end;
@@ -3058,7 +3075,7 @@ procedure TfrmAutoInject.menuFullInjectionClick(Sender: TObject);
     address:=inttohex(a,8);
 
   if inputquery(rsCodeInjectTemplate, rsOnWhatAddressDoYouWantTheJump, address) then
-    generateFullInjectionScript(assemblescreen.Lines, address);
+    generateFullInjectionScript(assemblescreen.Lines, address, 10, ssCtrl in GetKeyShiftState);
 end;
 
 
@@ -3127,7 +3144,7 @@ procedure TfrmAutoInject.MenuItem1Click(Sender: TObject);
   frmHighlighterEditor.free;
 end;
 
-procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolname: string; commentradius: integer=10);
+procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolname: string; commentradius: integer=10; farjmp: boolean=false);
 var
   a,a2: ptrUint;                  // pointer to injection point
   originalcode: array of string;  // disassembled code we're replacing
@@ -3168,8 +3185,12 @@ procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolna
   resultAOB: String;
   resultOffset: Integer;
   symbolNameWithOffset: String;
+  jmpsize: integer;
 begin
-  // now heavily modified code from "Code injection" menu
+  if not processhandler.is64Bit then
+    farjmp:=false;
+
+  jmpsize:=ifthen(farjmp, 14, 5);
 
   try
     a:=StrToQWordEx('$'+address);
@@ -3190,7 +3211,7 @@ procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolna
   setlength(originalcode,0);
   codesize:=0;
 
-  while codesize<5 do
+  while codesize<jmpsize do
   begin
     setlength(originalcode,length(originalcode)+1);
     originalcode[length(originalcode)-1]:=disassemble(c,x);
@@ -3253,7 +3274,7 @@ procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolna
       else
         add('aobscan(' + symbolName + ',' + resultAOB + ') // should be unique');
 
-      if processhandler.is64bit then
+      if processhandler.is64bit and (not farjmp) then
         add('alloc(newmem' + nr + ',$1000,' + symbolname + ')')
       else
         add('alloc(newmem' + nr + ',$1000)');
@@ -3271,11 +3292,14 @@ procedure GenerateAOBInjectionScript(script: TStrings; address: string; symbolna
 
       add('');
       add(symbolNameWithOffset + ':');
-      add('  jmp newmem' + nr + '');
-      if codesize>5 then
+      if farjmp then
+        add('  jmp far newmem' + nr + '')
+      else
+        add('  jmp newmem' + nr + '');
+      if codesize>jmpsize then
       begin
-        if codesize-5>1 then
-          add('  nop '+inttohex(codesize-5,1))
+        if codesize-jmpsize>1 then
+          add('  nop '+inttohex(codesize-jmpsize,1))
         else
           add('  nop');
       end;
@@ -3370,7 +3394,7 @@ procedure TfrmAutoInject.menuAOBInjectionClick(Sender: TObject);
     symbolname:='INJECT'+nr;
 
     if inputquery(rsCodeInjectTemplate, rsWhatIdentifierDoYouWantToUse, symbolName) then
-      GenerateAOBInjectionScript(assemblescreen.Lines, address, symbolname);
+      GenerateAOBInjectionScript(assemblescreen.Lines, address, symbolname, 10, ssCtrl in GetKeyShiftState);
   end;
 end;