├── .github
└── workflows
│ └── build.yml
├── .gitignore
├── .idea
└── .gitignore
├── LICENSE
├── README.md
├── build.gradle.kts
├── gradle.properties
├── gradle
└── wrapper
│ ├── gradle-wrapper.jar
│ └── gradle-wrapper.properties
├── gradlew
├── gradlew.bat
├── screenshots
├── Frida.png
└── Xposed.png
├── settings.gradle.kts
└── src
└── main
├── java
└── com
│ └── lanbaicode
│ ├── FridaHook.java
│ └── XposedHook.java
└── resources
└── META-INF
├── plugin.xml
└── pluginIcon.svg
/.github/workflows/build.yml:
--------------------------------------------------------------------------------
1 | name: Build
2 |
3 | on:
4 | push:
5 | branches: [ "main" ]
6 | pull_request:
7 | branches: [ "main" ]
8 |
9 | jobs:
10 | build:
11 |
12 | runs-on: ubuntu-latest
13 |
14 | steps:
15 | - uses: actions/checkout@v3
16 | - name: Set up JDK 17
17 | uses: actions/setup-java@v3
18 | with:
19 | java-version: '17'
20 | distribution: 'zulu'
21 | cache: gradle
22 |
23 | - name: Grant execute permission for gradlew
24 | run: chmod +x gradlew
25 |
26 | - name: Build with Gradle
27 | run: ./gradlew buildPlugin
28 |
29 | - name: Upload Artifact
30 | uses: actions/upload-artifact@v2
31 | with:
32 | name: plugin-artifact
33 | path: build/libs
34 |
--------------------------------------------------------------------------------
/.gitignore:
--------------------------------------------------------------------------------
1 | .gradle
2 | build/
3 | !gradle/wrapper/gradle-wrapper.jar
4 | !**/src/main/**/build/
5 | !**/src/test/**/build/
6 |
7 | ### IntelliJ IDEA ###
8 | .idea/modules.xml
9 | .idea/jarRepositories.xml
10 | .idea/compiler.xml
11 | .idea/libraries/
12 | .idea/gradle.xml
13 | .idea/kotlinc.xml
14 | .idea/misc.xml
15 | .idea/vcs.xml
16 | .run/
17 |
18 | *.iws
19 | *.iml
20 | *.ipr
21 | out/
22 | !**/src/main/**/out/
23 | !**/src/test/**/out/
24 |
25 |
26 | ### Eclipse ###
27 | .apt_generated
28 | .classpath
29 | .factorypath
30 | .project
31 | .settings
32 | .springBeans
33 | .sts4-cache
34 | bin/
35 | !**/src/main/**/bin/
36 | !**/src/test/**/bin/
37 |
38 | ### NetBeans ###
39 | /nbproject/private/
40 | /nbbuild/
41 | /dist/
42 | /nbdist/
43 | /.nb-gradle/
44 |
45 | ### VS Code ###
46 | .vscode/
47 |
48 | ### Mac OS ###
49 | .DS_Store
--------------------------------------------------------------------------------
/.idea/.gitignore:
--------------------------------------------------------------------------------
1 | # 默认忽略的文件
2 | /shelf/
3 | /workspace.xml
4 | # 基于编辑器的 HTTP 客户端请求
5 | /httpRequests/
6 | # Datasource local storage ignored files
7 | /dataSources/
8 | /dataSources.local.xml
9 |
--------------------------------------------------------------------------------
/LICENSE:
--------------------------------------------------------------------------------
1 | MIT License
2 |
3 | Copyright (c) 2023 LanBaiCode
4 |
5 | Permission is hereby granted, free of charge, to any person obtaining a copy
6 | of this software and associated documentation files (the "Software"), to deal
7 | in the Software without restriction, including without limitation the rights
8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9 | copies of the Software, and to permit persons to whom the Software is
10 | furnished to do so, subject to the following conditions:
11 |
12 | The above copyright notice and this permission notice shall be included in all
13 | copies or substantial portions of the Software.
14 |
15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 | SOFTWARE.
22 |
--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
1 |
2 |
HookCodeGenerator
3 |
HookCodeGenerator Plugin for IntelliJ IDEA.
4 |
5 |

6 |

7 |

8 |
9 |
10 |
11 |
12 |
13 | # 描述
14 | - 这是一个 **IntelliJ IDEA 插件**, 它可以快捷的生成 Xposed 和 Frida 的 Hook 代码
15 | - 支持类名、函数、字段
16 | - 生成的内容自动保存至剪贴板
17 | # 安装
18 | - 你可以通过 [Jetbrains Marketplace](https://plugins.jetbrains.com/plugin/23194-hook-code-generator)安装此插件
19 | - 也可通过这里[下载](https://github.com/LanBaiCode/HookCodeGenerator/releases)并选择磁盘安装
20 |
21 |
22 | # 使用
23 | 安装完成后, 将光标停留在需要 Hook 的位置, 右键选择 `XposedCode` 或者 `FridaCode`, 即可自动生成对应的 Hook 代码
24 |
25 | - https://github.com/LanBaiCode/HookCodeGenerator/assets/72244576/d1c67a66-10ae-4cce-b5eb-eeab08abc0f9
26 |
27 | - https://github.com/LanBaiCode/HookCodeGenerator/assets/72244576/bc508844-a047-4cdf-b07a-70389c9be364
28 |
29 | # 项目编译
30 | - JVM设置: Preferences -> Build, Execution, Deployment -> Gradle JVM (设置为17)
31 | - 运行IDE: Gradle -> Tasks -> intellij -> runIde
32 | - 编译插件: Gradle -> Tasks -> intellij -> buildPlugin
33 |
34 | # Credits
35 | - [jadx](https://github.com/skylot/jadx)
36 |
37 | # License
38 | [MIT License](LICENSE)
39 |
--------------------------------------------------------------------------------
/build.gradle.kts:
--------------------------------------------------------------------------------
1 | plugins {
2 | id("java")
3 | id("org.jetbrains.kotlin.jvm") version "1.9.0"
4 | id("org.jetbrains.intellij") version "1.15.0"
5 | }
6 |
7 | group = "com.LanBaiCode"
8 | version = "1.2"
9 |
10 | repositories {
11 | mavenCentral()
12 | }
13 |
14 | intellij {
15 | version.set("2022.2.5")
16 | type.set("IC") // Target IDE Platform
17 |
18 | plugins.set(listOf("com.intellij.java"))
19 | }
20 |
21 | tasks {
22 | // Set the JVM compatibility versions
23 | withType {
24 | sourceCompatibility = "17"
25 | targetCompatibility = "17"
26 | }
27 | withType {
28 | kotlinOptions.jvmTarget = "17"
29 | }
30 |
31 | patchPluginXml {
32 | sinceBuild.set("222")
33 | untilBuild.set("232.*")
34 | }
35 | publishPlugin {
36 | token.set(System.getenv("token"))
37 | }
38 | }
39 |
--------------------------------------------------------------------------------
/gradle.properties:
--------------------------------------------------------------------------------
1 | # Opt-out flag for bundling Kotlin standard library -> https://jb.gg/intellij-platform-kotlin-stdlib
2 | kotlin.stdlib.default.dependency=false
3 | # Enable Gradle Configuration Cache -> https://docs.gradle.org/current/userguide/configuration_cache.html
4 | org.gradle.configuration-cache=true
5 | # Enable Gradle Build Cache -> https://docs.gradle.org/current/userguide/build_cache.html
6 | org.gradle.caching=true
7 |
8 | platformPlugins = com.intellij.java
--------------------------------------------------------------------------------
/gradle/wrapper/gradle-wrapper.jar:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/LanBaiCode/HookCodeGenerator/d277721ca576703179f4b3462a3e3b4508563f64/gradle/wrapper/gradle-wrapper.jar
--------------------------------------------------------------------------------
/gradle/wrapper/gradle-wrapper.properties:
--------------------------------------------------------------------------------
1 | distributionBase=GRADLE_USER_HOME
2 | distributionPath=wrapper/dists
3 | distributionUrl=https\://services.gradle.org/distributions/gradle-8.2-bin.zip
4 | zipStoreBase=GRADLE_USER_HOME
5 | zipStorePath=wrapper/dists
6 |
--------------------------------------------------------------------------------
/gradlew:
--------------------------------------------------------------------------------
1 | #!/bin/sh
2 |
3 | #
4 | # Copyright © 2015-2021 the original authors.
5 | #
6 | # Licensed under the Apache License, Version 2.0 (the "License");
7 | # you may not use this file except in compliance with the License.
8 | # You may obtain a copy of the License at
9 | #
10 | # https://www.apache.org/licenses/LICENSE-2.0
11 | #
12 | # Unless required by applicable law or agreed to in writing, software
13 | # distributed under the License is distributed on an "AS IS" BASIS,
14 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15 | # See the License for the specific language governing permissions and
16 | # limitations under the License.
17 | #
18 |
19 | ##############################################################################
20 | #
21 | # Gradle start up script for POSIX generated by Gradle.
22 | #
23 | # Important for running:
24 | #
25 | # (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is
26 | # noncompliant, but you have some other compliant shell such as ksh or
27 | # bash, then to run this script, type that shell name before the whole
28 | # command line, like:
29 | #
30 | # ksh Gradle
31 | #
32 | # Busybox and similar reduced shells will NOT work, because this script
33 | # requires all of these POSIX shell features:
34 | # * functions;
35 | # * expansions «$var», «${var}», «${var:-default}», «${var+SET}»,
36 | # «${var#prefix}», «${var%suffix}», and «$( cmd )»;
37 | # * compound commands having a testable exit status, especially «case»;
38 | # * various built-in commands including «command», «set», and «ulimit».
39 | #
40 | # Important for patching:
41 | #
42 | # (2) This script targets any POSIX shell, so it avoids extensions provided
43 | # by Bash, Ksh, etc; in particular arrays are avoided.
44 | #
45 | # The "traditional" practice of packing multiple parameters into a
46 | # space-separated string is a well documented source of bugs and security
47 | # problems, so this is (mostly) avoided, by progressively accumulating
48 | # options in "$@", and eventually passing that to Java.
49 | #
50 | # Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS,
51 | # and GRADLE_OPTS) rely on word-splitting, this is performed explicitly;
52 | # see the in-line comments for details.
53 | #
54 | # There are tweaks for specific operating systems such as AIX, CygWin,
55 | # Darwin, MinGW, and NonStop.
56 | #
57 | # (3) This script is generated from the Groovy template
58 | # https://github.com/gradle/gradle/blob/master/subprojects/plugins/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt
59 | # within the Gradle project.
60 | #
61 | # You can find Gradle at https://github.com/gradle/gradle/.
62 | #
63 | ##############################################################################
64 |
65 | # Attempt to set APP_HOME
66 |
67 | # Resolve links: $0 may be a link
68 | app_path=$0
69 |
70 | # Need this for daisy-chained symlinks.
71 | while
72 | APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path
73 | [ -h "$app_path" ]
74 | do
75 | ls=$( ls -ld "$app_path" )
76 | link=${ls#*' -> '}
77 | case $link in #(
78 | /*) app_path=$link ;; #(
79 | *) app_path=$APP_HOME$link ;;
80 | esac
81 | done
82 |
83 | APP_HOME=$( cd "${APP_HOME:-./}" && pwd -P ) || exit
84 |
85 | APP_NAME="Gradle"
86 | APP_BASE_NAME=${0##*/}
87 |
88 | # Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
89 | DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
90 |
91 | # Use the maximum available, or set MAX_FD != -1 to use that value.
92 | MAX_FD=maximum
93 |
94 | warn () {
95 | echo "$*"
96 | } >&2
97 |
98 | die () {
99 | echo
100 | echo "$*"
101 | echo
102 | exit 1
103 | } >&2
104 |
105 | # OS specific support (must be 'true' or 'false').
106 | cygwin=false
107 | msys=false
108 | darwin=false
109 | nonstop=false
110 | case "$( uname )" in #(
111 | CYGWIN* ) cygwin=true ;; #(
112 | Darwin* ) darwin=true ;; #(
113 | MSYS* | MINGW* ) msys=true ;; #(
114 | NONSTOP* ) nonstop=true ;;
115 | esac
116 |
117 | CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
118 |
119 |
120 | # Determine the Java command to use to start the JVM.
121 | if [ -n "$JAVA_HOME" ] ; then
122 | if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
123 | # IBM's JDK on AIX uses strange locations for the executables
124 | JAVACMD=$JAVA_HOME/jre/sh/java
125 | else
126 | JAVACMD=$JAVA_HOME/bin/java
127 | fi
128 | if [ ! -x "$JAVACMD" ] ; then
129 | die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
130 |
131 | Please set the JAVA_HOME variable in your environment to match the
132 | location of your Java installation."
133 | fi
134 | else
135 | JAVACMD=java
136 | which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
137 |
138 | Please set the JAVA_HOME variable in your environment to match the
139 | location of your Java installation."
140 | fi
141 |
142 | # Increase the maximum file descriptors if we can.
143 | if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then
144 | case $MAX_FD in #(
145 | max*)
146 | MAX_FD=$( ulimit -H -n ) ||
147 | warn "Could not query maximum file descriptor limit"
148 | esac
149 | case $MAX_FD in #(
150 | '' | soft) :;; #(
151 | *)
152 | ulimit -n "$MAX_FD" ||
153 | warn "Could not set maximum file descriptor limit to $MAX_FD"
154 | esac
155 | fi
156 |
157 | # Collect all arguments for the java command, stacking in reverse order:
158 | # * args from the command line
159 | # * the main class name
160 | # * -classpath
161 | # * -D...appname settings
162 | # * --module-path (only if needed)
163 | # * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables.
164 |
165 | # For Cygwin or MSYS, switch paths to Windows format before running java
166 | if "$cygwin" || "$msys" ; then
167 | APP_HOME=$( cygpath --path --mixed "$APP_HOME" )
168 | CLASSPATH=$( cygpath --path --mixed "$CLASSPATH" )
169 |
170 | JAVACMD=$( cygpath --unix "$JAVACMD" )
171 |
172 | # Now convert the arguments - kludge to limit ourselves to /bin/sh
173 | for arg do
174 | if
175 | case $arg in #(
176 | -*) false ;; # don't mess with options #(
177 | /?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath
178 | [ -e "$t" ] ;; #(
179 | *) false ;;
180 | esac
181 | then
182 | arg=$( cygpath --path --ignore --mixed "$arg" )
183 | fi
184 | # Roll the args list around exactly as many times as the number of
185 | # args, so each arg winds up back in the position where it started, but
186 | # possibly modified.
187 | #
188 | # NB: a `for` loop captures its iteration list before it begins, so
189 | # changing the positional parameters here affects neither the number of
190 | # iterations, nor the values presented in `arg`.
191 | shift # remove old arg
192 | set -- "$@" "$arg" # push replacement arg
193 | done
194 | fi
195 |
196 | # Collect all arguments for the java command;
197 | # * $DEFAULT_JVM_OPTS, $JAVA_OPTS, and $GRADLE_OPTS can contain fragments of
198 | # shell script including quotes and variable substitutions, so put them in
199 | # double quotes to make sure that they get re-expanded; and
200 | # * put everything else in single quotes, so that it's not re-expanded.
201 |
202 | set -- \
203 | "-Dorg.gradle.appname=$APP_BASE_NAME" \
204 | -classpath "$CLASSPATH" \
205 | org.gradle.wrapper.GradleWrapperMain \
206 | "$@"
207 |
208 | # Use "xargs" to parse quoted args.
209 | #
210 | # With -n1 it outputs one arg per line, with the quotes and backslashes removed.
211 | #
212 | # In Bash we could simply go:
213 | #
214 | # readarray ARGS < <( xargs -n1 <<<"$var" ) &&
215 | # set -- "${ARGS[@]}" "$@"
216 | #
217 | # but POSIX shell has neither arrays nor command substitution, so instead we
218 | # post-process each arg (as a line of input to sed) to backslash-escape any
219 | # character that might be a shell metacharacter, then use eval to reverse
220 | # that process (while maintaining the separation between arguments), and wrap
221 | # the whole thing up as a single "set" statement.
222 | #
223 | # This will of course break if any of these variables contains a newline or
224 | # an unmatched quote.
225 | #
226 |
227 | eval "set -- $(
228 | printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" |
229 | xargs -n1 |
230 | sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' |
231 | tr '\n' ' '
232 | )" '"$@"'
233 |
234 | exec "$JAVACMD" "$@"
235 |
--------------------------------------------------------------------------------
/gradlew.bat:
--------------------------------------------------------------------------------
1 | @rem
2 | @rem Copyright 2015 the original author or authors.
3 | @rem
4 | @rem Licensed under the Apache License, Version 2.0 (the "License");
5 | @rem you may not use this file except in compliance with the License.
6 | @rem You may obtain a copy of the License at
7 | @rem
8 | @rem https://www.apache.org/licenses/LICENSE-2.0
9 | @rem
10 | @rem Unless required by applicable law or agreed to in writing, software
11 | @rem distributed under the License is distributed on an "AS IS" BASIS,
12 | @rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 | @rem See the License for the specific language governing permissions and
14 | @rem limitations under the License.
15 | @rem
16 |
17 | @if "%DEBUG%" == "" @echo off
18 | @rem ##########################################################################
19 | @rem
20 | @rem Gradle startup script for Windows
21 | @rem
22 | @rem ##########################################################################
23 |
24 | @rem Set local scope for the variables with windows NT shell
25 | if "%OS%"=="Windows_NT" setlocal
26 |
27 | set DIRNAME=%~dp0
28 | if "%DIRNAME%" == "" set DIRNAME=.
29 | set APP_BASE_NAME=%~n0
30 | set APP_HOME=%DIRNAME%
31 |
32 | @rem Resolve any "." and ".." in APP_HOME to make it shorter.
33 | for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
34 |
35 | @rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
36 | set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"
37 |
38 | @rem Find java.exe
39 | if defined JAVA_HOME goto findJavaFromJavaHome
40 |
41 | set JAVA_EXE=java.exe
42 | %JAVA_EXE% -version >NUL 2>&1
43 | if "%ERRORLEVEL%" == "0" goto execute
44 |
45 | echo.
46 | echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
47 | echo.
48 | echo Please set the JAVA_HOME variable in your environment to match the
49 | echo location of your Java installation.
50 |
51 | goto fail
52 |
53 | :findJavaFromJavaHome
54 | set JAVA_HOME=%JAVA_HOME:"=%
55 | set JAVA_EXE=%JAVA_HOME%/bin/java.exe
56 |
57 | if exist "%JAVA_EXE%" goto execute
58 |
59 | echo.
60 | echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
61 | echo.
62 | echo Please set the JAVA_HOME variable in your environment to match the
63 | echo location of your Java installation.
64 |
65 | goto fail
66 |
67 | :execute
68 | @rem Setup the command line
69 |
70 | set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
71 |
72 |
73 | @rem Execute Gradle
74 | "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %*
75 |
76 | :end
77 | @rem End local scope for the variables with windows NT shell
78 | if "%ERRORLEVEL%"=="0" goto mainEnd
79 |
80 | :fail
81 | rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
82 | rem the _cmd.exe /c_ return code!
83 | if not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
84 | exit /b 1
85 |
86 | :mainEnd
87 | if "%OS%"=="Windows_NT" endlocal
88 |
89 | :omega
90 |
--------------------------------------------------------------------------------
/screenshots/Frida.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/LanBaiCode/HookCodeGenerator/d277721ca576703179f4b3462a3e3b4508563f64/screenshots/Frida.png
--------------------------------------------------------------------------------
/screenshots/Xposed.png:
--------------------------------------------------------------------------------
https://raw.githubusercontent.com/LanBaiCode/HookCodeGenerator/d277721ca576703179f4b3462a3e3b4508563f64/screenshots/Xposed.png
--------------------------------------------------------------------------------
/settings.gradle.kts:
--------------------------------------------------------------------------------
1 | pluginManagement {
2 | repositories {
3 | mavenCentral()
4 | gradlePluginPortal()
5 | }
6 | }
7 |
8 | rootProject.name = "HookCodeGenerator"
--------------------------------------------------------------------------------
/src/main/java/com/lanbaicode/FridaHook.java:
--------------------------------------------------------------------------------
1 | package com.lanbaicode;
2 |
3 | import com.intellij.openapi.actionSystem.AnAction;
4 | import com.intellij.openapi.actionSystem.AnActionEvent;
5 | import com.intellij.openapi.ide.CopyPasteManager;
6 | import com.intellij.psi.*;
7 | import com.intellij.psi.util.PsiTreeUtil;
8 |
9 | import javax.swing.*;
10 | import java.awt.datatransfer.StringSelection;
11 | import java.util.Arrays;
12 | import java.util.Map;
13 | import java.util.Objects;
14 |
15 | public class FridaHook extends AnAction {
16 | private static final Map ARRAY_TYPE_MAPPING = Map.of(
17 | "int", "[I",
18 | "byte", "[B",
19 | "short", "[S",
20 | "long", "[J",
21 | "float", "[F",
22 | "double", "[D",
23 | "char", "[C",
24 | "boolean", "[Z"
25 | );
26 |
27 | @Override
28 | public void actionPerformed(AnActionEvent e) {
29 | PsiElement element = (PsiElement) e.getDataContext().getData("psi.Element");
30 | String fridaSnippet = generateFridaSnippet(element);
31 | if (fridaSnippet.isEmpty()) {
32 | showMessage("请选择一个正确的对象");
33 | return;
34 | }
35 | CopyPasteManager.getInstance().setContents(new StringSelection(fridaSnippet));
36 | showMessage("复制成功");
37 | }
38 |
39 | private String generateFridaSnippet(PsiElement element) {
40 | if (element instanceof PsiClass psiClass) {
41 | return generateClassSnippet(psiClass);
42 | } else if (element instanceof PsiMethod psiMethod) {
43 | return generateMethodSnippet(psiMethod);
44 | } else if (element instanceof PsiField psiField) {
45 | return generateFieldSnippet(psiField);
46 | } else {
47 | return "";
48 | }
49 | }
50 |
51 | private String generateClassSnippet(PsiClass psiClass) {
52 | return String.format("""
53 | let %s = Java.use("%s");""", psiClass.getName(), psiClass.getQualifiedName());
54 | }
55 |
56 | private String generateMethodSnippet(PsiMethod psiMethod) {
57 | PsiClass psiClass = PsiTreeUtil.getParentOfType(psiMethod, PsiClass.class);
58 | if (psiClass == null) return "";
59 | String shortClassName = psiClass.getName();
60 |
61 | String args = Arrays.stream(psiMethod.getParameterList().getParameters())
62 | .map(PsiParameter::getName)
63 | .reduce((s1, s2) -> s1 + ", " + s2)
64 | .orElse("");
65 |
66 | String logArgs = Arrays.stream(psiMethod.getParameterList().getParameters())
67 | .map(parameter -> parameter.getName() + "=${" + parameter.getName() + "}")
68 | .reduce((s1, s2) -> s1 + ", " + s2)
69 | .map(result -> ": " + result)
70 | .orElse("");
71 |
72 | //检查是否构造函数
73 | String methodName = psiMethod.isConstructor() ? "$init" : psiMethod.getName();
74 |
75 | //检查此函数是否被重载, 若重载则需要使用overload(x, y, z), 否则只需要使用单独的implementation
76 | String overload = "";
77 | boolean isOverloaded = Arrays.stream(psiMethod.getParent().getChildren())
78 | .filter(element -> element instanceof PsiMethod)
79 | .map(element -> (PsiMethod) element)
80 | .anyMatch(method -> {
81 | //方法名称相同, 但是签名不相同, 则存在重载
82 | return psiMethod.getName().equals(method.getName()) && !method.getSignature(PsiSubstitutor.EMPTY).equals(psiMethod.getSignature(PsiSubstitutor.EMPTY));
83 | });
84 |
85 | if (isOverloaded) {
86 | String argsType = Arrays.stream(psiMethod.getParameterList().getParameters())
87 | .map(parameter -> parseArgType(parameter.getType()))
88 | .reduce((s1, s2) -> s1 + "', '" + s2)
89 | .orElse("");
90 | overload = String.format(".overload('%s')", argsType);
91 | }
92 |
93 | if (psiMethod.isConstructor() || Objects.equals(psiMethod.getReturnType(), PsiType.VOID)) {
94 | // 没有返回值
95 | return String.format("""
96 | %s
97 | %s["%s"]%s.implementation = function (%s) {
98 | console.log(`%s.%s is called%s`);
99 | this["%s"](%s);
100 | };
101 | """,
102 | generateClassSnippet(psiClass),
103 | shortClassName, methodName, overload, args,
104 | shortClassName, methodName, logArgs,
105 | methodName, args
106 | );
107 | } else {
108 | return String.format("""
109 | %s
110 | %s["%s"]%s.implementation = function (%s) {
111 | console.log(`%s.%s is called%s`);
112 | let result = this["%s"](%s);
113 | console.log(`%s.%s result=${result}`);
114 | return result;
115 | };
116 | """,
117 | generateClassSnippet(psiClass),
118 | shortClassName, methodName, overload, args,
119 | shortClassName, methodName, logArgs,
120 | methodName, args,
121 | shortClassName, methodName
122 | );
123 | }
124 | }
125 |
126 | private String parseArgType(PsiType type) {
127 | String CanonicalText = type.getCanonicalText();
128 | if (type instanceof PsiArrayType arrayType) {
129 | CanonicalText = ARRAY_TYPE_MAPPING.getOrDefault(arrayType.getCanonicalText(), "");
130 | //