diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ae7a295 --- /dev/null +++ b/.gitignore @@ -0,0 +1,41 @@ +.gradle +build/ +!gradle/wrapper/gradle-wrapper.jar +!**/src/main/**/build/ +!**/src/test/**/build/ + +### IntelliJ IDEA ### +.idea/ +*.iws +*.iml +*.ipr +out/ +!**/src/main/**/out/ +!**/src/test/**/out/ + +### Eclipse ### +.apt_generated +.classpath +.factorypath +.project +.settings +.springBeans +.sts4-cache +bin/ +!**/src/main/**/bin/ +!**/src/test/**/bin/ + +### NetBeans ### +/nbproject/private/ +/nbbuild/ +/nbdist/ +/.nb-gradle/ + +### VS Code ### +.vscode/ + +### Mac OS ### +.DS_Store + +obf-sample-test/ +dist/ diff --git a/LICENSE b/LICENSE new file mode 100644 index 0000000..7e77fc5 --- /dev/null +++ b/LICENSE @@ -0,0 +1,21 @@ +MIT License + +Copyright (c) 2024 Matouš Kučera + +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. diff --git a/LICENSE-JASM b/LICENSE-JASM new file mode 100644 index 0000000..b0c7668 --- /dev/null +++ b/LICENSE-JASM @@ -0,0 +1,9 @@ +The MIT License (MIT) + +Copyright (c) 2023-2024 Justus Garbe + +Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. diff --git a/README.md b/README.md new file mode 100644 index 0000000..8a25d42 --- /dev/null +++ b/README.md @@ -0,0 +1,24 @@ +# jasm + +A JavaScript port of the [Jasm dis/assembler](https://github.com/jumanji144/Jasm). + +## Example + +```js +const fs = require("fs"); +const { disassemble } = require("./jasm.js"); // get it from the dist/ directory or jsDelivr + +const data = fs.readFileSync("./your/package/HelloWorld.class"); // read a class file +console.log(disassemble(data, { + indent: " ", // the string that should be used as the indent, defaults to 4 spaces +})); +``` + +Or see the browser-based proof-of-concept in the [docs](./docs) directory. + +## Licensing + +The supporting code for this project and the Jasm dis/assembler are licensed under the MIT License +([supporting code](./LICENSE), [Jasm](./LICENSE-JASM)). + +_This project is not affiliated with, maintained or endorsed by the Jasm project in any way. Do NOT report issues with this project to the Jasm issue tracker._ diff --git a/build.gradle.kts b/build.gradle.kts new file mode 100644 index 0000000..02c7bcb --- /dev/null +++ b/build.gradle.kts @@ -0,0 +1,67 @@ +plugins { + `java-library` + alias(libs.plugins.teavm) // order matters? +} + +val thisVersion = "0.1.0" + +group = "run.slicer" +version = "$thisVersion-${libs.versions.jasm.get()}" +description = "A JavaScript port of the Jasm dis/assembler." + +repositories { + mavenCentral() + maven("https://jitpack.io") +} + +dependencies { + api(libs.jasm.composition.jvm) + implementation("org.ow2.asm:asm:+") // override transitive dep scope + compileOnly(libs.teavm.core) +} + +java.toolchain { + languageVersion = JavaLanguageVersion.of(21) +} + +teavm.js { + mainClass = "run.slicer.jasm.Main" + moduleType = org.teavm.gradle.api.JSModuleType.ES2015 + // obfuscated = false + // optimization = org.teavm.gradle.api.OptimizationLevel.NONE +} + +tasks { + register("copyDist") { + group = "build" + + from("README.md", "LICENSE", "LICENSE-JASM", generateJavaScript, "jasm.d.ts") + into("dist") + + doLast { + file("dist/package.json").writeText( + """ + { + "name": "@run-slicer/jasm", + "version": "${project.version}", + "description": "A JavaScript port of the Jasm dis/assembler (https://github.com/jumanji144/Jasm).", + "main": "jasm.js", + "types": "jasm.d.ts", + "keywords": [ + "assembler", + "disassembler", + "java", + "bytecode" + ], + "author": "run-slicer", + "license": "MIT" + } + """.trimIndent() + ) + } + } + + build { + dependsOn("copyDist") + } +} diff --git a/docs/index.html b/docs/index.html new file mode 100644 index 0000000..7ba2a45 --- /dev/null +++ b/docs/index.html @@ -0,0 +1,81 @@ + + + + + + Jasm in TeaVM + + + + +
+ + + + diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml new file mode 100644 index 0000000..ee3c1db --- /dev/null +++ b/gradle/libs.versions.toml @@ -0,0 +1,10 @@ +[versions] +jasm = "2.5.0" +teavm = "0.10.0" + +[libraries] +jasm-composition-jvm = { group = "com.github.jumanji144.Jasm", name = "jasm-composition-jvm", version.ref = "jasm" } +teavm-core = { group = "org.teavm", name = "teavm-core", version.ref = "teavm" } + +[plugins] +teavm = { id = "org.teavm", version.ref = "teavm" } diff --git a/gradle/wrapper/gradle-wrapper.jar b/gradle/wrapper/gradle-wrapper.jar new file mode 100644 index 0000000..249e583 Binary files /dev/null and b/gradle/wrapper/gradle-wrapper.jar differ diff --git a/gradle/wrapper/gradle-wrapper.properties b/gradle/wrapper/gradle-wrapper.properties new file mode 100644 index 0000000..c255f32 --- /dev/null +++ b/gradle/wrapper/gradle-wrapper.properties @@ -0,0 +1,6 @@ +#Wed Jun 12 22:45:56 CEST 2024 +distributionBase=GRADLE_USER_HOME +distributionPath=wrapper/dists +distributionUrl=https\://services.gradle.org/distributions/gradle-8.7-bin.zip +zipStoreBase=GRADLE_USER_HOME +zipStorePath=wrapper/dists diff --git a/gradlew b/gradlew new file mode 100644 index 0000000..1b6c787 --- /dev/null +++ b/gradlew @@ -0,0 +1,234 @@ +#!/bin/sh + +# +# Copyright © 2015-2021 the original authors. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# https://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +############################################################################## +# +# Gradle start up script for POSIX generated by Gradle. +# +# Important for running: +# +# (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is +# noncompliant, but you have some other compliant shell such as ksh or +# bash, then to run this script, type that shell name before the whole +# command line, like: +# +# ksh Gradle +# +# Busybox and similar reduced shells will NOT work, because this script +# requires all of these POSIX shell features: +# * functions; +# * expansions «$var», «${var}», «${var:-default}», «${var+SET}», +# «${var#prefix}», «${var%suffix}», and «$( cmd )»; +# * compound commands having a testable exit status, especially «case»; +# * various built-in commands including «command», «set», and «ulimit». +# +# Important for patching: +# +# (2) This script targets any POSIX shell, so it avoids extensions provided +# by Bash, Ksh, etc; in particular arrays are avoided. +# +# The "traditional" practice of packing multiple parameters into a +# space-separated string is a well documented source of bugs and security +# problems, so this is (mostly) avoided, by progressively accumulating +# options in "$@", and eventually passing that to Java. +# +# Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS, +# and GRADLE_OPTS) rely on word-splitting, this is performed explicitly; +# see the in-line comments for details. +# +# There are tweaks for specific operating systems such as AIX, CygWin, +# Darwin, MinGW, and NonStop. +# +# (3) This script is generated from the Groovy template +# https://github.com/gradle/gradle/blob/master/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt +# within the Gradle project. +# +# You can find Gradle at https://github.com/gradle/gradle/. +# +############################################################################## + +# Attempt to set APP_HOME + +# Resolve links: $0 may be a link +app_path=$0 + +# Need this for daisy-chained symlinks. +while + APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path + [ -h "$app_path" ] +do + ls=$( ls -ld "$app_path" ) + link=${ls#*' -> '} + case $link in #( + /*) app_path=$link ;; #( + *) app_path=$APP_HOME$link ;; + esac +done + +APP_HOME=$( cd "${APP_HOME:-./}" && pwd -P ) || exit + +APP_NAME="Gradle" +APP_BASE_NAME=${0##*/} + +# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"' + +# Use the maximum available, or set MAX_FD != -1 to use that value. +MAX_FD=maximum + +warn () { + echo "$*" +} >&2 + +die () { + echo + echo "$*" + echo + exit 1 +} >&2 + +# OS specific support (must be 'true' or 'false'). +cygwin=false +msys=false +darwin=false +nonstop=false +case "$( uname )" in #( + CYGWIN* ) cygwin=true ;; #( + Darwin* ) darwin=true ;; #( + MSYS* | MINGW* ) msys=true ;; #( + NONSTOP* ) nonstop=true ;; +esac + +CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar + + +# Determine the Java command to use to start the JVM. +if [ -n "$JAVA_HOME" ] ; then + if [ -x "$JAVA_HOME/jre/sh/java" ] ; then + # IBM's JDK on AIX uses strange locations for the executables + JAVACMD=$JAVA_HOME/jre/sh/java + else + JAVACMD=$JAVA_HOME/bin/java + fi + if [ ! -x "$JAVACMD" ] ; then + die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." + fi +else + JAVACMD=java + which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. + +Please set the JAVA_HOME variable in your environment to match the +location of your Java installation." +fi + +# Increase the maximum file descriptors if we can. +if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then + case $MAX_FD in #( + max*) + MAX_FD=$( ulimit -H -n ) || + warn "Could not query maximum file descriptor limit" + esac + case $MAX_FD in #( + '' | soft) :;; #( + *) + ulimit -n "$MAX_FD" || + warn "Could not set maximum file descriptor limit to $MAX_FD" + esac +fi + +# Collect all arguments for the java command, stacking in reverse order: +# * args from the command line +# * the main class name +# * -classpath +# * -D...appname settings +# * --module-path (only if needed) +# * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables. + +# For Cygwin or MSYS, switch paths to Windows format before running java +if "$cygwin" || "$msys" ; then + APP_HOME=$( cygpath --path --mixed "$APP_HOME" ) + CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" ) + + JAVACMD=$( cygpath --unix "$JAVACMD" ) + + # Now convert the arguments - kludge to limit ourselves to /bin/sh + for arg do + if + case $arg in #( + -*) false ;; # don't mess with options #( + /?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath + [ -e "$t" ] ;; #( + *) false ;; + esac + then + arg=$( cygpath --path --ignore --mixed "$arg" ) + fi + # Roll the args list around exactly as many times as the number of + # args, so each arg winds up back in the position where it started, but + # possibly modified. + # + # NB: a `for` loop captures its iteration list before it begins, so + # changing the positional parameters here affects neither the number of + # iterations, nor the values presented in `arg`. + shift # remove old arg + set -- "$@" "$arg" # push replacement arg + done +fi + +# Collect all arguments for the java command; +# * $DEFAULT_JVM_OPTS, $JAVA_OPTS, and $GRADLE_OPTS can contain fragments of +# shell script including quotes and variable substitutions, so put them in +# double quotes to make sure that they get re-expanded; and +# * put everything else in single quotes, so that it's not re-expanded. + +set -- \ + "-Dorg.gradle.appname=$APP_BASE_NAME" \ + -classpath "$CLASSPATH" \ + org.gradle.wrapper.GradleWrapperMain \ + "$@" + +# Use "xargs" to parse quoted args. +# +# With -n1 it outputs one arg per line, with the quotes and backslashes removed. +# +# In Bash we could simply go: +# +# readarray ARGS < <( xargs -n1 <<<"$var" ) && +# set -- "${ARGS[@]}" "$@" +# +# but POSIX shell has neither arrays nor command substitution, so instead we +# post-process each arg (as a line of input to sed) to backslash-escape any +# character that might be a shell metacharacter, then use eval to reverse +# that process (while maintaining the separation between arguments), and wrap +# the whole thing up as a single "set" statement. +# +# This will of course break if any of these variables contains a newline or +# an unmatched quote. +# + +eval "set -- $( + printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" | + xargs -n1 | + sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' | + tr '\n' ' ' + )" '"$@"' + +exec "$JAVACMD" "$@" diff --git a/gradlew.bat b/gradlew.bat new file mode 100644 index 0000000..107acd3 --- /dev/null +++ b/gradlew.bat @@ -0,0 +1,89 @@ +@rem +@rem Copyright 2015 the original author or authors. +@rem +@rem Licensed under the Apache License, Version 2.0 (the "License"); +@rem you may not use this file except in compliance with the License. +@rem You may obtain a copy of the License at +@rem +@rem https://www.apache.org/licenses/LICENSE-2.0 +@rem +@rem Unless required by applicable law or agreed to in writing, software +@rem distributed under the License is distributed on an "AS IS" BASIS, +@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +@rem See the License for the specific language governing permissions and +@rem limitations under the License. +@rem + +@if "%DEBUG%" == "" @echo off +@rem ########################################################################## +@rem +@rem Gradle startup script for Windows +@rem +@rem ########################################################################## + +@rem Set local scope for the variables with windows NT shell +if "%OS%"=="Windows_NT" setlocal + +set DIRNAME=%~dp0 +if "%DIRNAME%" == "" set DIRNAME=. +set APP_BASE_NAME=%~n0 +set APP_HOME=%DIRNAME% + +@rem Resolve any "." and ".." in APP_HOME to make it shorter. +for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi + +@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script. +set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m" + +@rem Find java.exe +if defined JAVA_HOME goto findJavaFromJavaHome + +set JAVA_EXE=java.exe +%JAVA_EXE% -version >NUL 2>&1 +if "%ERRORLEVEL%" == "0" goto execute + +echo. +echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:findJavaFromJavaHome +set JAVA_HOME=%JAVA_HOME:"=% +set JAVA_EXE=%JAVA_HOME%/bin/java.exe + +if exist "%JAVA_EXE%" goto execute + +echo. +echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% +echo. +echo Please set the JAVA_HOME variable in your environment to match the +echo location of your Java installation. + +goto fail + +:execute +@rem Setup the command line + +set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar + + +@rem Execute Gradle +"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %* + +:end +@rem End local scope for the variables with windows NT shell +if "%ERRORLEVEL%"=="0" goto mainEnd + +:fail +rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of +rem the _cmd.exe /c_ return code! +if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1 +exit /b 1 + +:mainEnd +if "%OS%"=="Windows_NT" endlocal + +:omega diff --git a/jasm.d.ts b/jasm.d.ts new file mode 100644 index 0000000..8efe84c --- /dev/null +++ b/jasm.d.ts @@ -0,0 +1,7 @@ +declare module "@run-slicer/jasm" { + export interface DisassemblyConfig { + indent?: string; + } + + export function disassemble(b: Uint8Array, config?: DisassemblyConfig): string; +} diff --git a/settings.gradle.kts b/settings.gradle.kts new file mode 100644 index 0000000..0ff4cf6 --- /dev/null +++ b/settings.gradle.kts @@ -0,0 +1,5 @@ +plugins { + id("org.gradle.toolchains.foojay-resolver-convention") version "0.8.0" +} + +rootProject.name = "jasm" diff --git a/src/main/java/run/slicer/jasm/DisassemblyOptions.java b/src/main/java/run/slicer/jasm/DisassemblyOptions.java new file mode 100644 index 0000000..aa0581d --- /dev/null +++ b/src/main/java/run/slicer/jasm/DisassemblyOptions.java @@ -0,0 +1,9 @@ +package run.slicer.jasm; + +import org.teavm.jso.JSBody; +import org.teavm.jso.JSObject; + +public interface DisassemblyOptions extends JSObject { + @JSBody(script = "return this.indent || ' ';") // 4 spaces + String indent(); +} diff --git a/src/main/java/run/slicer/jasm/Main.java b/src/main/java/run/slicer/jasm/Main.java new file mode 100644 index 0000000..4d6ddc9 --- /dev/null +++ b/src/main/java/run/slicer/jasm/Main.java @@ -0,0 +1,29 @@ +package run.slicer.jasm; + +import me.darknet.assembler.printer.JvmClassPrinter; +import me.darknet.assembler.printer.PrintContext; +import org.teavm.jso.JSByRef; +import org.teavm.jso.JSExport; +import org.teavm.jso.core.JSObjects; + +import java.io.IOException; +import java.io.UncheckedIOException; + +public class Main { + @JSExport + public static String disassemble(@JSByRef byte[] b, DisassemblyOptions options) { + return disassemble0(b, options == null || JSObjects.isUndefined(options) ? JSObjects.create() : options); + } + + private static String disassemble0(byte[] b, DisassemblyOptions options) { + final var ctx = new PrintContext<>(options.indent()); + + try { + new JvmClassPrinter(b).print(ctx); + } catch (IOException e) { + throw new UncheckedIOException(e); + } + + return ctx.toString(); + } +} diff --git a/src/teavm/java/run/slicer/jasm/teavm/JASMPlugin.java b/src/teavm/java/run/slicer/jasm/teavm/JASMPlugin.java new file mode 100644 index 0000000..9cad497 --- /dev/null +++ b/src/teavm/java/run/slicer/jasm/teavm/JASMPlugin.java @@ -0,0 +1,12 @@ +package run.slicer.jasm.teavm; + +import org.teavm.vm.spi.TeaVMHost; +import org.teavm.vm.spi.TeaVMPlugin; + +public class JASMPlugin implements TeaVMPlugin { + @Override + public void install(TeaVMHost host) { + host.add(new MethodStubTransformer()); + host.add(new MethodDelegationTransformer()); + } +} diff --git a/src/teavm/java/run/slicer/jasm/teavm/MetaAccessors.java b/src/teavm/java/run/slicer/jasm/teavm/MetaAccessors.java new file mode 100644 index 0000000..deb4db9 --- /dev/null +++ b/src/teavm/java/run/slicer/jasm/teavm/MetaAccessors.java @@ -0,0 +1,52 @@ +package run.slicer.jasm.teavm; + +import org.teavm.metaprogramming.CompileTime; +import org.teavm.metaprogramming.Meta; +import org.teavm.metaprogramming.ReflectClass; +import org.teavm.metaprogramming.Value; +import org.teavm.metaprogramming.reflect.ReflectField; + +import static org.teavm.metaprogramming.Metaprogramming.emit; +import static org.teavm.metaprogramming.Metaprogramming.exit; +import static org.teavm.metaprogramming.Metaprogramming.unsupportedCase; + +@CompileTime +public final class MetaAccessors { + private MetaAccessors() { + } + + @Meta + public static native Object getBootstrapMethods(Class cls, Object obj); + + private static void getBootstrapMethods(ReflectClass cls, Value obj) { + if (!cls.getName().equals("org.objectweb.asm.ClassReader")) { + unsupportedCase(); + return; + } + + final ReflectField field = cls.getDeclaredField("bootstrapMethodOffsets"); + if (field != null) { + exit(() -> field.get(obj)); + } else { + throw new RuntimeException("Could not find bootstrapMethodOffsets field"); + } + } + + @Meta + public static native void setOpcodes(Class cls, Object val); + + private static void setOpcodes(ReflectClass cls, Value val) { + if (!cls.getName().equals("me.darknet.assembler.printer.InstructionPrinter")) { + unsupportedCase(); + return; + } + + final ReflectField field = cls.getDeclaredField("OPCODES"); + if (field != null) { + // TeaVM doesn't care about the final modifier thankfully + emit(() -> field.set(null, val)); + } else { + throw new RuntimeException("Could not find OPCODES field"); + } + } +} diff --git a/src/teavm/java/run/slicer/jasm/teavm/MethodDelegates.java b/src/teavm/java/run/slicer/jasm/teavm/MethodDelegates.java new file mode 100644 index 0000000..ccce306 --- /dev/null +++ b/src/teavm/java/run/slicer/jasm/teavm/MethodDelegates.java @@ -0,0 +1,65 @@ +package run.slicer.jasm.teavm; + +import dev.xdark.blw.asm.internal.InternalAsmLibrary; +import dev.xdark.blw.asm.internal.Util; +import dev.xdark.blw.constant.Constant; +import dev.xdark.blw.type.InvokeDynamic; +import dev.xdark.blw.type.MethodHandle; +import dev.xdark.blw.type.Types; +import me.darknet.assembler.printer.InstructionPrinter; +import org.objectweb.asm.ClassReader; +import org.objectweb.asm.Handle; + +import java.util.ArrayList; +import java.util.List; + +import static run.slicer.jasm.teavm.MetaAccessors.getBootstrapMethods; +import static run.slicer.jasm.teavm.MetaAccessors.setOpcodes; + +public final class MethodDelegates { + private static final String[] MNEMONICS = new String[]{ + "nop", "aconst_null", "iconst_m1", "iconst_0", "iconst_1", "iconst_2", "iconst_3", "iconst_4", "iconst_5", "lconst_0", + "lconst_1", "fconst_0", "fconst_1", "fconst_2", "dconst_0", "dconst_1", "bipush", "sipush", "ldc", "ldc_w", + "ldc2_w", "iload", "lload", "fload", "dload", "aload", "iload_0", "iload_1", "iload_2", "iload_3", + "lload_0", "lload_1", "lload_2", "lload_3", "fload_0", "fload_1", "fload_2", "fload_3", "dload_0", "dload_1", + "dload_2", "dload_3", "aload_0", "aload_1", "aload_2", "aload_3", "iaload", "laload", "faload", "daload", + "aaload", "baload", "caload", "saload", "istore", "lstore", "fstore", "dstore", "astore", "istore_0", + "istore_1", "istore_2", "istore_3", "lstore_0", "lstore_1", "lstore_2", "lstore_3", "fstore_0", "fstore_1", "fstore_2", + "fstore_3", "dstore_0", "dstore_1", "dstore_2", "dstore_3", "astore_0", "astore_1", "astore_2", "astore_3", "iastore", + "lastore", "fastore", "dastore", "aastore", "bastore", "castore", "sastore", "pop", "pop2", "dup", "dup_x1", "dup_x2", + "dup2", "dup2_x1", "dup2_x2", "swap", "iadd", "ladd", "fadd", "dadd", "isub", "lsub", "fsub", "dsub", "imul", "lmul", "fmul", + "dmul", "idiv", "ldiv", "fdiv", "ddiv", "irem", "lrem", "frem", "drem", "ineg", "lneg", "fneg", "dneg", "ishl", "lshl", + "ishr", "lshr", "iushr", "lushr", "iand", "land", "ior", "lor", "ixor", "lxor", "iinc", "i2l", "i2f", "i2d", "l2i", "l2f", + "l2d", "f2i", "f2l", "f2d", "d2i", "d2l", "d2f", "i2b", "i2c", "i2s", "lcmp", "fcmpl", "fcmpg", "dcmpl", "dcmpg", + "ifeq", "ifne", "iflt", "ifge", "ifgt", "ifle", "if_icmpeq", "if_icmpne", "if_icmplt", "if_icmpge", "if_icmpgt", + "if_icmple", "if_acmpeq", "if_acmpne", "goto", "jsr", "ret", "tableswitch", "lookupswitch", "ireturn", "lreturn", + "freturn", "dreturn", "areturn", "return", "getstatic", "putstatic", "getfield", "putfield", "invokevirtual", + "invokespecial", "invokestatic", "invokeinterface", "invokedynamic", "new", "newarray", "anewarray", "arraylength", + "athrow", "checkcast", "instanceof", "monitorenter", "monitorexit", "wide", "multianewarray", "ifnull", "ifnonnull", + "goto_w", "jsr_w" + }; + + private MethodDelegates() { + } + + public static InvokeDynamic dev_xdark_blw_asm_internal_InternalAsmLibrary_readInvokeDynamic(InternalAsmLibrary ignored, ClassReader cr, int cpInfoOffset, char[] charBuffer) { + int nameAndTypeCpInfoOffset = cr.getItem(cr.readUnsignedShort(cpInfoOffset + 2)); + String name = cr.readUTF8(nameAndTypeCpInfoOffset, charBuffer); + String descriptor = cr.readUTF8(nameAndTypeCpInfoOffset + 2, charBuffer); + int[] bootstrapMethodOffsets = (int[]) getBootstrapMethods(cr.getClass(), cr); // change: remove VarHandle + int bootstrapMethodOffset = bootstrapMethodOffsets[cr.readUnsignedShort(cpInfoOffset)]; + MethodHandle methodHandle = Util.wrapMethodHandle((Handle) cr.readConst(cr.readUnsignedShort(bootstrapMethodOffset), charBuffer)); + int argCount = cr.readUnsignedShort(bootstrapMethodOffset + 2); + List args = new ArrayList<>(argCount); + bootstrapMethodOffset += 4; + for (int i = 0; i < argCount; i++) { + args.add(Util.wrapConstant(cr.readConst(cr.readUnsignedShort(bootstrapMethodOffset), charBuffer))); + bootstrapMethodOffset += 2; + } + return new InvokeDynamic(name, Types.methodType(descriptor), methodHandle, args); + } + + public static void me_darknet_assembler_printer_InstructionPrinter_LTclinitGT() { + setOpcodes(InstructionPrinter.class, MNEMONICS); + } +} diff --git a/src/teavm/java/run/slicer/jasm/teavm/MethodDelegationTransformer.java b/src/teavm/java/run/slicer/jasm/teavm/MethodDelegationTransformer.java new file mode 100644 index 0000000..f06a9eb --- /dev/null +++ b/src/teavm/java/run/slicer/jasm/teavm/MethodDelegationTransformer.java @@ -0,0 +1,52 @@ +package run.slicer.jasm.teavm; + +import org.teavm.model.*; +import org.teavm.model.instructions.*; + +import java.util.ArrayList; +import java.util.List; +import java.util.Map; +import java.util.Set; + +public final class MethodDelegationTransformer implements ClassHolderTransformer { + private final Map> DELEGATED_METHODS = Map.of( + "dev.xdark.blw.asm.internal.InternalAsmLibrary", Set.of("readInvokeDynamic") + ); + + @Override + public void transformClass(ClassHolder cls, ClassHolderTransformerContext context) { + for (final MethodHolder method : cls.getMethods()) { + if (method.getProgram() == null) continue; + + for (final BasicBlock block : method.getProgram().getBasicBlocks()) { + for (final Instruction insn : block) { + if (insn instanceof InvokeInstruction invokeInsn) { + this.transformInvokeInsn(invokeInsn); + } + } + } + } + } + + private void transformInvokeInsn(InvokeInstruction insn) { + final MethodReference method = insn.getMethod(); + + final Set candidates = DELEGATED_METHODS.get(method.getClassName()); + if (candidates == null || !candidates.contains(method.getName())) return; + + final String name = (method.getClassName() + "_" + method.getName()).replace('.', '_'); + + final List signature = new ArrayList<>(List.of(method.getSignature())); + if (insn.getInstance() != null) { // not a static invocation, remap "this" ref as an argument + signature.addFirst(ValueType.object(method.getClassName())); + + final List args = new ArrayList<>(insn.getArguments()); + args.addFirst(insn.getInstance()); + + insn.setInstance(null); + insn.setArguments(args.toArray(new Variable[0])); + } + + insn.setMethod(new MethodReference("run.slicer.jasm.teavm.MethodDelegates", name, signature.toArray(new ValueType[0]))); + } +} diff --git a/src/teavm/java/run/slicer/jasm/teavm/MethodStubTransformer.java b/src/teavm/java/run/slicer/jasm/teavm/MethodStubTransformer.java new file mode 100644 index 0000000..4c846ae --- /dev/null +++ b/src/teavm/java/run/slicer/jasm/teavm/MethodStubTransformer.java @@ -0,0 +1,62 @@ +package run.slicer.jasm.teavm; + +import org.teavm.model.*; +import org.teavm.model.instructions.ExitInstruction; +import org.teavm.model.instructions.InvocationType; +import org.teavm.model.instructions.InvokeInstruction; + +public class MethodStubTransformer implements ClassHolderTransformer { + @Override + public void transformClass(ClassHolder cls, ClassHolderTransformerContext context) { + switch (cls.getName()) { + case "dev.xdark.blw.asm.internal.InternalAsmLibrary" -> { + this.stubVoid(cls.getMethod(new MethodDescriptor("", void.class))); + } + case "me.darknet.assembler.printer.InstructionPrinter" -> { + this.stubWithCall( + cls.getMethod(new MethodDescriptor("", void.class)), + new MethodReference( + "run.slicer.jasm.teavm.MethodDelegates", + "me_darknet_assembler_printer_InstructionPrinter_LTclinitGT", + ValueType.VOID + ) + ); + } + } + } + + private void stubVoid(MethodHolder method) { + final Program program = this.newProgram(method.parameterCount()); + final BasicBlock block = program.createBasicBlock(); + + block.add(new ExitInstruction()); + + method.setProgram(program); + } + + private void stubWithCall(MethodHolder method, MethodReference target) { + final Program program = this.newProgram(method.parameterCount()); + + final BasicBlock block = program.createBasicBlock(); + + final var invokeInsn = new InvokeInstruction(); + invokeInsn.setType(InvocationType.VIRTUAL); + invokeInsn.setMethod(target); + block.add(invokeInsn); + + block.add(new ExitInstruction()); + + method.setProgram(program); + } + + private Program newProgram(int parameterCount) { + parameterCount++; // type var + + final Program program = new Program(); + for (int i = 0; i < parameterCount; i++) { + program.createVariable(); + } + + return program; + } +} diff --git a/src/teavm/resources/META-INF/services/org.teavm.vm.spi.TeaVMPlugin b/src/teavm/resources/META-INF/services/org.teavm.vm.spi.TeaVMPlugin new file mode 100644 index 0000000..a5dca79 --- /dev/null +++ b/src/teavm/resources/META-INF/services/org.teavm.vm.spi.TeaVMPlugin @@ -0,0 +1 @@ +run.slicer.jasm.teavm.JASMPlugin \ No newline at end of file