forked from openrs2/openrs2
parent
4a8704f566
commit
6e9a7f36ef
@ -1,93 +0,0 @@ |
|||||||
package dev.openrs2.bundler.transform; |
|
||||||
|
|
||||||
import java.util.Optional; |
|
||||||
|
|
||||||
import dev.openrs2.asm.InsnMatcher; |
|
||||||
import dev.openrs2.asm.InsnNodeUtilsKt; |
|
||||||
import dev.openrs2.asm.MemberRef; |
|
||||||
import dev.openrs2.asm.MethodNodeUtilsKt; |
|
||||||
import dev.openrs2.asm.classpath.ClassPath; |
|
||||||
import dev.openrs2.asm.classpath.Library; |
|
||||||
import dev.openrs2.asm.transform.Transformer; |
|
||||||
import org.objectweb.asm.tree.ClassNode; |
|
||||||
import org.objectweb.asm.tree.FieldInsnNode; |
|
||||||
import org.objectweb.asm.tree.LdcInsnNode; |
|
||||||
import org.objectweb.asm.tree.MethodNode; |
|
||||||
import org.slf4j.Logger; |
|
||||||
import org.slf4j.LoggerFactory; |
|
||||||
|
|
||||||
public final class BufferSizeTransformer extends Transformer { |
|
||||||
private static final Logger logger = LoggerFactory.getLogger(BufferSizeTransformer.class); |
|
||||||
|
|
||||||
private static final InsnMatcher GPP1_POS_MATCHER = InsnMatcher.compile("LDC (INVOKESPECIAL | INVOKEVIRTUAL) GETSTATIC"); |
|
||||||
private static final InsnMatcher NEW_BUFFER_MATCHER = InsnMatcher.compile("NEW DUP (SIPUSH | LDC) INVOKESPECIAL PUTSTATIC"); |
|
||||||
|
|
||||||
private static Optional<MemberRef> findBuffer(MethodNode method) { |
|
||||||
return GPP1_POS_MATCHER.match(method) |
|
||||||
.filter(match -> { |
|
||||||
var ldc = (LdcInsnNode) match.get(0); |
|
||||||
return ldc.cst.equals("gpp1 pos:"); |
|
||||||
}) |
|
||||||
.map(match -> { |
|
||||||
var getstatic = (FieldInsnNode) match.get(2); |
|
||||||
return new MemberRef(getstatic); |
|
||||||
}) |
|
||||||
.findAny(); |
|
||||||
} |
|
||||||
|
|
||||||
private MemberRef buffer; |
|
||||||
private int buffersResized; |
|
||||||
|
|
||||||
@Override |
|
||||||
protected void preTransform(ClassPath classPath) { |
|
||||||
buffer = null; |
|
||||||
buffersResized = 0; |
|
||||||
|
|
||||||
for (var library : classPath.getLibraries()) { |
|
||||||
for (var clazz : library) { |
|
||||||
for (var method : clazz.methods) { |
|
||||||
if (!MethodNodeUtilsKt.hasCode(method)) { |
|
||||||
continue; |
|
||||||
} |
|
||||||
|
|
||||||
var buffer = findBuffer(method); |
|
||||||
if (buffer.isEmpty()) { |
|
||||||
continue; |
|
||||||
} |
|
||||||
|
|
||||||
this.buffer = buffer.orElseThrow(); |
|
||||||
logger.info("Identified buffer: {}", this.buffer); |
|
||||||
break; |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
} |
|
||||||
|
|
||||||
@Override |
|
||||||
protected boolean transformCode(ClassPath classPath, Library library, ClassNode clazz, MethodNode method) { |
|
||||||
if (buffer == null) { |
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
NEW_BUFFER_MATCHER.match(method).forEach(match -> { |
|
||||||
var putstatic = (FieldInsnNode) match.get(4); |
|
||||||
if (!putstatic.owner.equals(buffer.getOwner())) { |
|
||||||
return; |
|
||||||
} else if (!putstatic.name.equals(buffer.getName())) { |
|
||||||
return; |
|
||||||
} else if (!putstatic.desc.equals(buffer.getDesc())) { |
|
||||||
return; |
|
||||||
} |
|
||||||
|
|
||||||
method.instructions.set(match.get(2), InsnNodeUtilsKt.createIntConstant(65535)); |
|
||||||
buffersResized++; |
|
||||||
}); |
|
||||||
|
|
||||||
return false; |
|
||||||
} |
|
||||||
|
|
||||||
@Override |
|
||||||
protected void postTransform(ClassPath classPath) { |
|
||||||
logger.info("Resized {} buffers to 65535 bytes", buffersResized); |
|
||||||
} |
|
||||||
} |
|
@ -0,0 +1,75 @@ |
|||||||
|
package dev.openrs2.bundler.transform |
||||||
|
|
||||||
|
import com.github.michaelbull.logging.InlineLogger |
||||||
|
import dev.openrs2.asm.InsnMatcher |
||||||
|
import dev.openrs2.asm.MemberRef |
||||||
|
import dev.openrs2.asm.classpath.ClassPath |
||||||
|
import dev.openrs2.asm.classpath.Library |
||||||
|
import dev.openrs2.asm.createIntConstant |
||||||
|
import dev.openrs2.asm.hasCode |
||||||
|
import dev.openrs2.asm.transform.Transformer |
||||||
|
import org.objectweb.asm.tree.ClassNode |
||||||
|
import org.objectweb.asm.tree.FieldInsnNode |
||||||
|
import org.objectweb.asm.tree.LdcInsnNode |
||||||
|
import org.objectweb.asm.tree.MethodNode |
||||||
|
import kotlin.streams.asSequence |
||||||
|
|
||||||
|
class BufferSizeTransformer : Transformer() { |
||||||
|
private var buffer: MemberRef? = null |
||||||
|
private var buffersResized = 0 |
||||||
|
|
||||||
|
override fun preTransform(classPath: ClassPath) { |
||||||
|
buffer = null |
||||||
|
buffersResized = 0 |
||||||
|
|
||||||
|
for (library in classPath.libraries) { |
||||||
|
for (clazz in library) { |
||||||
|
for (method in clazz.methods) { |
||||||
|
if (!method.hasCode()) { |
||||||
|
continue |
||||||
|
} |
||||||
|
|
||||||
|
this.buffer = findBuffer(method) ?: continue |
||||||
|
logger.info { "Identified buffer: ${this.buffer}" } |
||||||
|
break |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
override fun transformCode(classPath: ClassPath, library: Library, clazz: ClassNode, method: MethodNode): Boolean { |
||||||
|
if (buffer == null) { |
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
NEW_BUFFER_MATCHER.match(method).forEach { |
||||||
|
val putstatic = it[4] as FieldInsnNode |
||||||
|
if (MemberRef(putstatic) == buffer!!) { |
||||||
|
method.instructions[it[2]] = createIntConstant(65535) |
||||||
|
buffersResized++ |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return false |
||||||
|
} |
||||||
|
|
||||||
|
override fun postTransform(classPath: ClassPath) { |
||||||
|
logger.info { "Resized $buffersResized buffers to 65535 bytes" } |
||||||
|
} |
||||||
|
|
||||||
|
companion object { |
||||||
|
private val logger = InlineLogger() |
||||||
|
private val GPP1_POS_MATCHER = InsnMatcher.compile("LDC (INVOKESPECIAL | INVOKEVIRTUAL) GETSTATIC") |
||||||
|
private val NEW_BUFFER_MATCHER = InsnMatcher.compile("NEW DUP (SIPUSH | LDC) INVOKESPECIAL PUTSTATIC") |
||||||
|
|
||||||
|
private fun findBuffer(method: MethodNode): MemberRef? { |
||||||
|
return GPP1_POS_MATCHER.match(method).asSequence().filter { |
||||||
|
val ldc = it[0] as LdcInsnNode |
||||||
|
ldc.cst == "gpp1 pos:" |
||||||
|
}.map { |
||||||
|
val getstatic = it[2] as FieldInsnNode |
||||||
|
MemberRef(getstatic) |
||||||
|
}.firstOrNull() |
||||||
|
} |
||||||
|
} |
||||||
|
} |
Loading…
Reference in new issue