You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
294 lines
10 KiB
294 lines
10 KiB
7 years ago
|
package sig.modules;
|
||
|
|
|
||
|
|
import java.awt.Color;
|
||
|
|
import java.awt.Graphics;
|
||
|
|
import java.awt.Image;
|
||
|
|
import java.awt.geom.Rectangle2D;
|
||
|
|
import java.awt.image.BufferedImage;
|
||
|
|
import java.io.File;
|
||
|
|
import java.io.IOException;
|
||
|
|
import java.util.ArrayList;
|
||
|
|
import java.util.ConcurrentModificationException;
|
||
|
|
import java.util.HashMap;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.concurrent.Executors;
|
||
|
|
import java.util.concurrent.ScheduledExecutorService;
|
||
|
|
import java.util.concurrent.TimeUnit;
|
||
|
|
|
||
|
|
import javax.imageio.ImageIO;
|
||
|
|
|
||
|
|
import com.sun.jna.Memory;
|
||
|
|
import com.sun.jna.Pointer;
|
||
|
|
import com.sun.jna.platform.win32.Kernel32;
|
||
|
|
import com.sun.jna.platform.win32.WinNT;
|
||
|
|
import com.sun.jna.platform.win32.WinNT.HANDLE;
|
||
|
|
|
||
|
|
import sig.Module;
|
||
|
|
import sig.sigIRC;
|
||
|
|
import sig.modules.RabiRace.ColorCycler;
|
||
|
|
import sig.modules.RabiRace.MemoryData;
|
||
|
|
import sig.modules.RabiRace.Profile;
|
||
|
|
import sig.modules.RabiRibi.MemoryOffset;
|
||
|
|
import sig.modules.RabiRibi.MemoryType;
|
||
|
|
import sig.modules.utils.PsapiTools;
|
||
|
|
import sig.utils.DrawUtils;
|
||
|
|
|
||
|
|
public class RabiRaceModule extends Module{
|
||
|
|
final static String ITEMS_DIRECTORY = sigIRC.BASEDIR+"sigIRC/rabi-ribi/items/";
|
||
|
|
final int PROCESS_PERMISSIONS = WinNT.PROCESS_QUERY_INFORMATION | WinNT.PROCESS_VM_READ;
|
||
|
|
boolean foundRabiRibi = false;
|
||
|
|
int rabiRibiPID = -1;
|
||
|
|
long rabiRibiMemOffset = 0;
|
||
|
|
public HANDLE rabiribiProcess = null;
|
||
|
|
public static HashMap<String,Image> image_map = new HashMap<String,Image>();
|
||
|
|
ColorCycler rainbowcycler = new ColorCycler(new Color(255,0,0,96),8);
|
||
|
|
Profile myProfile = new Profile(this);
|
||
|
|
|
||
|
|
public static List<MemoryData> key_items_list = new ArrayList<MemoryData>();
|
||
|
|
public static List<MemoryData> badges_list = new ArrayList<MemoryData>();
|
||
|
|
|
||
|
|
public RabiRaceModule(Rectangle2D bounds, String moduleName) {
|
||
|
|
super(bounds, moduleName);
|
||
|
|
//Initialize();
|
||
|
|
Initialize();
|
||
|
|
|
||
|
|
//System.out.println("Money value is: "+readIntFromMemory(MemoryOffset.MONEY));
|
||
|
|
}
|
||
|
|
|
||
|
|
private void Initialize() {
|
||
|
|
CheckRabiRibiClient();
|
||
|
|
|
||
|
|
ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
|
||
|
|
scheduler.scheduleWithFixedDelay(()->{
|
||
|
|
CheckRabiRibiClient();
|
||
|
|
UpdateMyProfile();
|
||
|
|
}, 5000, 5000, TimeUnit.MILLISECONDS);
|
||
|
|
|
||
|
|
File dir = new File(ITEMS_DIRECTORY);
|
||
|
|
String[] images = dir.list();
|
||
|
|
List<String> filtered_images = new ArrayList<String>();
|
||
|
|
for (String file : images) {
|
||
|
|
File f = new File(ITEMS_DIRECTORY+file);
|
||
|
|
if (!f.isDirectory()) {
|
||
|
|
filtered_images.add(file);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
images = filtered_images.toArray(new String[filtered_images.size()]);
|
||
|
|
for (String image : images) {
|
||
|
|
try {
|
||
|
|
image_map.put(image, ImageIO.read(new File(ITEMS_DIRECTORY+image)));
|
||
|
|
} catch (IOException e) {
|
||
|
|
e.printStackTrace();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for (MemoryData md : MemoryData.values()) {
|
||
|
|
if (md.key_item) {
|
||
|
|
key_items_list.add(md);
|
||
|
|
} else {
|
||
|
|
badges_list.add(md);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void CheckRabiRibiClient() {
|
||
|
|
List<Integer> pids;
|
||
|
|
try {
|
||
|
|
pids = PsapiTools.getInstance().enumProcesses();
|
||
|
|
boolean found=false;
|
||
|
|
for (Integer pid : pids) {
|
||
|
|
HANDLE process = Kernel32.INSTANCE.OpenProcess(PROCESS_PERMISSIONS, true, pid);
|
||
|
|
List<sig.modules.utils.Module> hModules;
|
||
|
|
try {
|
||
|
|
hModules = PsapiTools.getInstance().EnumProcessModules(process);
|
||
|
|
for(sig.modules.utils.Module m: hModules){
|
||
|
|
//System.out.println(m.getFileName()+":"+m.getEntryPoint());
|
||
|
|
if (m.getFileName().contains("rabiribi")) {
|
||
|
|
found=true;
|
||
|
|
if (!foundRabiRibi) {
|
||
|
|
rabiRibiMemOffset = Pointer.nativeValue(m.getLpBaseOfDll().getPointer());
|
||
|
|
System.out.println("Found an instance of Rabi-Ribi at 0x"+Long.toHexString(rabiRibiMemOffset)+" | File:"+m.getFileName()+","+m.getBaseName());
|
||
|
|
rabiRibiPID=pid;
|
||
|
|
foundRabiRibi=true;
|
||
|
|
rabiribiProcess=process;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (found) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
} catch (Exception e) {
|
||
|
|
e.printStackTrace();
|
||
|
|
}
|
||
|
|
if (process!=null) {
|
||
|
|
Kernel32.INSTANCE.CloseHandle(process);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if (!found && foundRabiRibi) {
|
||
|
|
foundRabiRibi=false;
|
||
|
|
System.out.println("Rabi-Ribi process lost.");
|
||
|
|
}
|
||
|
|
} catch (Exception e) {
|
||
|
|
e.printStackTrace();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
public void run() {
|
||
|
|
if (foundRabiRibi) {
|
||
|
|
rainbowcycler.run();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private void UpdateMyProfile() {
|
||
|
|
if (foundRabiRibi) {
|
||
|
|
//System.out.println("Called.");
|
||
|
|
myProfile.rainbowEggCount = readIntFromMemory(MemoryOffset.RAINBOW_EGG_COUNT);
|
||
|
|
myProfile.attackUps = readItemCountFromMemory(MemoryOffset.ATTACKUP_START,MemoryOffset.ATTACKUP_END);
|
||
|
|
myProfile.healthUps = readItemCountFromMemory(MemoryOffset.HEALTHUP_START,MemoryOffset.HEALTHUP_END);
|
||
|
|
myProfile.manaUps = readItemCountFromMemory(MemoryOffset.MANAUP_START,MemoryOffset.MANAUP_END);
|
||
|
|
myProfile.regenUps = readItemCountFromMemory(MemoryOffset.REGENUP_START,MemoryOffset.REGENUP_END);
|
||
|
|
myProfile.packUps = readItemCountFromMemory(MemoryOffset.PACKUP_START,MemoryOffset.PACKUP_END);
|
||
|
|
myProfile.isPaused = readIntFromMemory(MemoryOffset.WARP_TRANSITION_COUNTER)==141;
|
||
|
|
myProfile.itempct = readFloatFromMemory(MemoryOffset.ITEM_PERCENT);
|
||
|
|
myProfile.mappct = readFloatFromMemory(MemoryOffset.MAP_PERCENT);
|
||
|
|
myProfile.playtime = readIntFromMemory(MemoryOffset.PLAYTIME);
|
||
|
|
myProfile.updateClientValues();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/*public int readIntFromErinaData(MemoryOffset val) {
|
||
|
|
return readIntFromPointer(val,MemoryOffset.ENTITY_ARRAY);
|
||
|
|
}
|
||
|
|
|
||
|
|
public float readFloatFromErinaData(MemoryOffset val) {
|
||
|
|
return readFloatFromPointer(val,MemoryOffset.ENTITY_ARRAY);
|
||
|
|
}*/
|
||
|
|
|
||
|
|
public int readIntFromMemory(long offset) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(rabiRibiMemOffset+offset), mem, 4, null);
|
||
|
|
return mem.getInt(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public float readFloatFromMemory(long offset) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(rabiRibiMemOffset+offset), mem, 4, null);
|
||
|
|
return mem.getFloat(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public float readFloatFromMemoryOffset(MemoryOffset val, long pointer) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(pointer+val.getOffset()), mem, 4, null);
|
||
|
|
return mem.getFloat(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public int readIntFromMemoryOffset(MemoryOffset val, long pointer) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(pointer+val.getOffset()), mem, 4, null);
|
||
|
|
return mem.getInt(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public float readDirectFloatFromMemoryLocation(long pointer) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(pointer), mem, 4, null);
|
||
|
|
return mem.getFloat(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public int readDirectIntFromMemoryLocation(long pointer) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(pointer), mem, 4, null);
|
||
|
|
return mem.getInt(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public int readIntFromPointer(MemoryOffset val, MemoryOffset pointer) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(readIntFromMemory(pointer.getOffset())+val.getOffset()), mem, 4, null);
|
||
|
|
return mem.getInt(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public float readFloatFromPointer(MemoryOffset val, MemoryOffset pointer) {
|
||
|
|
Memory mem = new Memory(4);
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(readIntFromMemory(pointer.getOffset())+val.getOffset()), mem, 4, null);
|
||
|
|
return mem.getFloat(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
public int readIntFromMemory(MemoryOffset val) {
|
||
|
|
return (int)readFromMemory(val,MemoryType.INTEGER);
|
||
|
|
}
|
||
|
|
|
||
|
|
public float readFloatFromMemory(MemoryOffset val) {
|
||
|
|
return (float)readFromMemory(val,MemoryType.FLOAT);
|
||
|
|
}
|
||
|
|
|
||
|
|
Object readFromMemory(MemoryOffset val, MemoryType type) {
|
||
|
|
Memory mem = new Memory(type.getSize());
|
||
|
|
Kernel32.INSTANCE.ReadProcessMemory(rabiribiProcess, new Pointer(rabiRibiMemOffset+val.getOffset()), mem, type.getSize(), null);
|
||
|
|
switch (type) {
|
||
|
|
case FLOAT:
|
||
|
|
return mem.getFloat(0);
|
||
|
|
case INTEGER:
|
||
|
|
return mem.getInt(0);
|
||
|
|
default:
|
||
|
|
System.out.println("WARNING! Type "+type+" does not have a defined value.");
|
||
|
|
return -1;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
int readItemCountFromMemory(MemoryOffset start_range,
|
||
|
|
MemoryOffset end_range) {
|
||
|
|
int count=0;
|
||
|
|
for (long i=start_range.getOffset();i<=end_range.getOffset();i++) {
|
||
|
|
if (readIntFromMemory(i)==1) {
|
||
|
|
count++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return count;
|
||
|
|
}
|
||
|
|
|
||
|
|
public void draw(Graphics g) {
|
||
|
|
super.draw(g);
|
||
|
|
if (!foundRabiRibi) {
|
||
|
|
DrawUtils.drawTextFont(g, sigIRC.panel.userFont, position.getX(), position.getY()+26, Color.BLACK, "Rabi-Ribi not found! Please start it.");
|
||
|
|
} else {
|
||
|
|
DrawUtils.drawTextFont(g, sigIRC.panel.userFont, position.getX(), position.getY()+26, Color.BLACK, "Values: "+readIntFromMemory(MemoryOffset.DLC_ITEM1)+","+readIntFromMemory(MemoryOffset.DLC_ITEM2)+","+readIntFromMemory(MemoryOffset.DLC_ITEM3)+","+readIntFromMemory(MemoryOffset.DLC_ITEM4));
|
||
|
|
final int border=20;
|
||
|
|
final int width=(int)(position.getWidth()-border*2);
|
||
|
|
final int spacing=width/5;
|
||
|
|
for (int i=0;i<5;i++) {
|
||
|
|
Image img = image_map.get("easter_egg.png");
|
||
|
|
Color col = (myProfile.rainbowEggCount>i)?rainbowcycler.getCycleColor():new Color(0,0,0,192);
|
||
|
|
DrawUtils.drawImage(g, img, (int)(position.getX()+border+i*spacing-img.getWidth(sigIRC.panel)/4),(int)(position.getY()+36),col,sigIRC.panel);
|
||
|
|
}
|
||
|
|
int size = myProfile.key_items.size();
|
||
|
|
final int icon_size = 24;
|
||
|
|
int count = 0;
|
||
|
|
try {
|
||
|
|
for (String key : myProfile.key_items.keySet()) {
|
||
|
|
MemoryData data = myProfile.key_items.get(key);
|
||
|
|
|
||
|
|
if (size*icon_size<width) {
|
||
|
|
g.drawImage(data.img, (int)(position.getX()+border+((count++)*icon_size)), (int)(position.getY()+96+8), (int)icon_size, (int)icon_size, sigIRC.panel);
|
||
|
|
} else {
|
||
|
|
g.drawImage(data.img, (int)(position.getX()+border+((width/size)*(count++))), (int)(position.getY()+96+8), (int)icon_size, (int)icon_size, sigIRC.panel);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
count=0;
|
||
|
|
size = myProfile.badges.size();
|
||
|
|
for (String key : myProfile.badges.keySet()) {
|
||
|
|
MemoryData data = myProfile.badges.get(key);
|
||
|
|
|
||
|
|
if (size*icon_size<width) {
|
||
|
|
g.drawImage(data.img, (int)(position.getX()+border+((count++)*icon_size)), (int)(position.getY()+96+32), (int)icon_size, (int)icon_size, sigIRC.panel);
|
||
|
|
} else {
|
||
|
|
g.drawImage(data.img, (int)(position.getX()+border+((width/size)*(count++))), (int)(position.getY()+96+32), (int)icon_size, (int)icon_size, sigIRC.panel);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
} catch (ConcurrentModificationException e) {
|
||
|
|
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|