Fixes + added def classmetadata reader to module context.

This commit is contained in:
zaxar163 2018-09-21 20:43:59 +03:00
parent 99d5d22eac
commit fb20550724
9 changed files with 147 additions and 134 deletions

View file

@ -1,23 +1,47 @@
package ru.gravit.launchserver.manangers;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collections;
import java.util.jar.JarFile;
import ru.gravit.launcher.Launcher;
import ru.gravit.launcher.LauncherAPI;
import ru.gravit.launcher.LauncherClassLoader;
import ru.gravit.launcher.modules.SimpleModuleManager;
import ru.gravit.launchserver.LaunchServer;
import ru.gravit.launchserver.asm.ClassMetadataReader;
import ru.gravit.launchserver.modules.CoreModule;
import ru.gravit.launchserver.modules.LaunchServerModuleContext;
import ru.gravit.utils.helper.IOHelper;
public class ModulesManager extends SimpleModuleManager {
public final ClassMetadataReader metadataReader;
public ModulesManager(LaunchServer lsrv) {
modules = new ArrayList<>(1);
ClassMetadataReader metadataReader = null;
try {
metadataReader = new ClassMetadataReader(Collections.singletonList(new JarFile(new File(IOHelper.getCodeSource(Launcher.class).toUri()))));
} catch (IOException e) { }
this.metadataReader = metadataReader;
classloader = new LauncherClassLoader(new URL[0], ClassLoader.getSystemClassLoader());
context = new LaunchServerModuleContext(lsrv, classloader);
context = new LaunchServerModuleContext(lsrv, classloader, metadataReader);
registerCoreModule();
}
private void registerCoreModule() {
load(new CoreModule());
}
@Override
@LauncherAPI
public void loadModule(URL jarpath, String classname, boolean preload) throws ClassNotFoundException, IllegalAccessException, InstantiationException {
super.loadModule(jarpath, classname, preload);
try {
if (metadataReader != null) metadataReader.add(new JarFile(new File(IOHelper.getCodeSource(Launcher.class).toUri())));
} catch (IOException e) { }
}
}

View file

@ -4,14 +4,18 @@
import ru.gravit.launcher.modules.ModuleContext;
import ru.gravit.launcher.modules.ModulesManagerInterface;
import ru.gravit.launchserver.LaunchServer;
import ru.gravit.launchserver.asm.ClassMetadataReader;
public class LaunchServerModuleContext implements ModuleContext {
public final LaunchServer launchServer;
public final LauncherClassLoader classloader;
public LaunchServerModuleContext(LaunchServer server, LauncherClassLoader classloader)
public final ClassMetadataReader metadataReader;
public LaunchServerModuleContext(LaunchServer server, LauncherClassLoader classloader, ClassMetadataReader metadataReader)
{
launchServer = server;
this.classloader = classloader;
this.metadataReader = metadataReader;
}
@Override
public Type getType() {

View file

@ -40,6 +40,7 @@ public void run() {
}
}
}
public static final String NAME = Launcher.getConfig().projectname;
public static String avn32 = null, avn64 = null;
public static Path wrap32 = null, wrap64 = null;
@ -64,14 +65,13 @@ private static Path handle(Path mustdiedll, String resource) {
public static void loadVared() {
if (JVMHelper.JVM_BITS == 32)
GuardBind.startAbs(System.getProperty("avn32"));
loadLib(System.getProperty("avn32"));
else if (JVMHelper.JVM_BITS == 64)
GuardBind.startAbs(System.getProperty("avn64"));
loadLib(System.getProperty("avn64"));
}
public static void main(boolean init) {
if (init)
GuardBind.init();
public static void start() {
loadVared();
GuardBind.avnRegisterThreatNotifier((int threatType) -> {
System.err.println("Threat " + GuardBind.ThreatType.getThreat(threatType).name());
LogHelper.error("Cheating == crash!");
@ -102,7 +102,7 @@ private static void processArched(Path arch32, Path arch64, Path wrapper32, Path
wrap64 = IOHelper.toAbsPath(wrapper64);
}
public static void start(Path path1) throws IOException {
public static void init(Path path1) throws IOException {
Path path = path1.resolve("guard");
processArched(handle(path.resolve("Avanguard32.dll"), "Avanguard32.dll"),
handle(path.resolve("Avanguard64.dll"), "Avanguard64.dll"),
@ -120,4 +120,10 @@ private static void transfer(byte[] orig, Path mustdiedll) throws IOException {
mustdiedll.toFile().createNewFile();
IOHelper.transfer(orig, mustdiedll, false);
}
public static void loadLib(String path) {
LogHelper.debug("Anti-Cheat loading");
System.load(path);
LogHelper.debug("Anti-Cheat loaded");
}
}

View file

@ -67,7 +67,8 @@
import ru.gravit.launcher.serialize.stream.StreamObject;
public class LauncherEngine {
@LauncherAPI
private static final boolean isUsingGuard = true;
public static void addLauncherClassBindings(Map<String, Object> bindings) {
bindings.put("LauncherClass", Launcher.class);
bindings.put("LauncherConfigClass", LauncherConfig.class);
@ -188,7 +189,6 @@ private void setScriptBindings() {
addLauncherClassBindings(bindings);
}
@LauncherAPI
public void start(String... args) throws Throwable {
Launcher.modulesManager = new ClientModuleManager(this);
LauncherConfig.getAutogenConfig().initModules(); //INIT
@ -205,9 +205,9 @@ public void start(String... args) throws Throwable {
LogHelper.info("Invoking start() function");
Invocable invoker = (Invocable) engine;
if (JVMHelper.OS_TYPE == JVMHelper.OS.MUSTDIE) {
AvanguardStarter.start((Path) invoker.invokeFunction("getPathDirHelper"));
AvanguardStarter.init((Path) invoker.invokeFunction("getPathDirHelper"));
AvanguardStarter.loadVared();
AvanguardStarter.main(false);
if (isUsingGuard) AvanguardStarter.start();
}
Launcher.modulesManager.postInitModules();
invoker.invokeFunction("start", (Object) args);

View file

@ -149,6 +149,7 @@ public void write(HOutput output) throws IOException {
private static final String[] EMPTY_ARRAY = new String[0];
private static final String MAGICAL_INTEL_OPTION = "-XX:HeapDumpPath=ThisTricksIntelDriversForPerformance_javaw.exe_minecraft.exe.heapdump";
private static final boolean isUsingWrapper = true;
private static final boolean isUsingGuard = true;
@LauncherAPI
public static final String TITLE_PROPERTY = "launcher.title";
@ -378,10 +379,7 @@ public static void main(String... args) throws Throwable {
Launcher.modulesManager = new ClientModuleManager(null);
LauncherConfig.getAutogenConfig().initModules(); //INIT
Launcher.modulesManager.preInitModules();
if (JVMHelper.OS_TYPE == OS.MUSTDIE) {
AvanguardStarter.loadVared();
AvanguardStarter.main(false);
}
if (JVMHelper.OS_TYPE == OS.MUSTDIE) runAvn();
checkJVMBitsAndVersion();
JVMHelper.verifySystemProperties(ClientLauncher.class, true);
LogHelper.printVersion("Client Launcher");
@ -436,6 +434,10 @@ public static void main(String... args) throws Throwable {
}
}
private static void runAvn() {
if (isUsingGuard) AvanguardStarter.start(); // так компилятор с гарантией вырежет
}
private static URL[] resolveClassPath(Path clientDir, String... classPath) throws IOException {
Collection<Path> result = new LinkedList<>();
for (String classPathEntry : classPath) {

View file

@ -1,10 +1,6 @@
package ru.zaxar163;
import java.nio.file.Path;
import ru.gravit.launcher.LauncherAPI;
import ru.gravit.utils.helper.JVMHelper;
import ru.gravit.utils.helper.LogHelper;
@LauncherAPI
public final class GuardBind {
@ -74,28 +70,6 @@ public int getValue() {
@LauncherAPI
public static native int getCheckTime();
public static void init() {
LogHelper.debug("Anti-Cheat loading");
if (JVMHelper.OS_TYPE == JVMHelper.OS.MUSTDIE)
if (JVMHelper.JVM_BITS == 32)
System.loadLibrary("Avanguard32");
else if (JVMHelper.JVM_BITS == 64)
System.loadLibrary("Avanguard64");
LogHelper.debug("Anti-Cheat loaded");
}
@LauncherAPI
public static native void setCheckTime(int time);
public static void start(Path path) {
LogHelper.debug("Anti-Cheat loading");
System.load(path.normalize().toAbsolutePath().toFile().getAbsolutePath());
LogHelper.debug("Anti-Cheat loaded");
}
public static void startAbs(String path) {
LogHelper.debug("Anti-Cheat loading");
System.load(path);
LogHelper.debug("Anti-Cheat loaded");
}
}

View file

@ -81,6 +81,7 @@ public void initModules() {
@LauncherAPI
public void load(Module module) {
modules.add(module);
module.preInit(context);
}
@Override
@ -107,9 +108,7 @@ public void loadModule(URL jarpath, String classname, boolean preload) throws Cl
classloader.addURL(jarpath);
Class<?> moduleclass = Class.forName(classname, true, classloader);
Module module = (Module) moduleclass.newInstance();
modules.add(module);
module.preInit(context);
if(!preload) module.init(context);
load(module, preload);
LogHelper.info("Module %s version: %s loaded",module.getName(),module.getVersion());
}

View file

@ -34,7 +34,7 @@ public void init(ModuleContext context1) {
// Config may has boolean variable "hardAntiDecomp", which enables hard mode (needs -noverify to JVM)
LaunchServerModuleContext context = (LaunchServerModuleContext) context1;
boolean hobf = context.launchServer.config.block.hasEntry("hardAntiDecomp") ? context.launchServer.config.block.getEntryValue("hardAntiDecomp", BooleanConfigEntry.class) : false;
context.launchServer.buildHookManager.registerClassTransformer(new TransformerClass(hobf));
context.launchServer.buildHookManager.registerClassTransformer(new TransformerClass(hobf, context.metadataReader));
}
}

View file

@ -3,20 +3,24 @@
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassWriter;
import ru.gravit.launchserver.asm.ClassMetadataReader;
import ru.gravit.launchserver.asm.SafeClassWriter;
import ru.gravit.launchserver.manangers.BuildHookManager.Transformer;
public class TransformerClass implements Transformer {
private final boolean context;
private final ClassMetadataReader reader;
public TransformerClass(boolean hobf) {
public TransformerClass(boolean hobf, ClassMetadataReader reader) {
this.context = hobf;
this.reader = reader;
}
@Override
public byte[] transform(byte[] input, CharSequence classname) {
ClassReader classReader = new ClassReader(input);
ClassWriter writer = new ClassWriter(ClassWriter.COMPUTE_MAXS | ClassWriter.COMPUTE_FRAMES);
ClassWriter writer = new SafeClassWriter(reader, ClassWriter.COMPUTE_MAXS | ClassWriter.COMPUTE_FRAMES);
classReader.accept(new AntiDecompileClassVisitor(writer, context), 0);
return writer.toByteArray();
}