Fix anonymous classes being included in class stubs

Col-E · Jan 11, 2025 · 7da81ca · 7da81ca
1 parent af7d9da
commit 7da81ca
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/recaf-core/src/main/java/software/coley/recaf/services/compile/stub/ClassStubGenerator.java b/recaf-core/src/main/java/software/coley/recaf/services/compile/stub/ClassStubGenerator.java
@@ -279,7 +279,8 @@ else if (method.hasPrivateModifier())
 
 				NameType paramInfo = getInfo("p" + i, paramType.descriptor());
 				code.append(paramInfo.className).append(' ').append(paramInfo.name);
-				if (i < parameterCount - 1) code.append(", ");
+				if (i < parameterCount - 1)
+					code.append(", ");
 			}
 			code.append(") { ");
 			if (isCtor) {
@@ -339,6 +340,8 @@ protected void appendInnerClasses(@Nonnull StringBuilder code) throws Expression
 				continue;
 			if (innerClassName.length() <= className.length())
 				continue;
+			if (!isSafeClassName(innerClassName.replace('/', '.').replace('$', '.')))
+				continue;
 			ClassPathNode innerClassPath = workspace.findClass(innerClassName);
 			if (innerClassPath != null) {
 				ClassInfo innerClassInfo = innerClassPath.getValue();