ClassNode.java 22.6 KB
Newer Older
S
Skylot 已提交
1 2
package jadx.core.dex.nodes;

3 4 5
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
6
import java.util.LinkedHashSet;
7 8
import java.util.List;
import java.util.Map;
9
import java.util.Set;
10
import java.util.function.BiConsumer;
11
import java.util.function.Consumer;
12
import java.util.stream.Collectors;
13

14
import org.jetbrains.annotations.NotNull;
15 16 17 18
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

S
Skylot 已提交
19
import jadx.api.DecompilationMode;
20
import jadx.api.ICodeCache;
S
Skylot 已提交
21
import jadx.api.ICodeInfo;
S
Skylot 已提交
22
import jadx.api.ICodeWriter;
S
Skylot 已提交
23
import jadx.api.JadxArgs;
24
import jadx.api.JavaClass;
25
import jadx.api.impl.SimpleCodeInfo;
26
import jadx.api.plugins.input.data.IClassData;
27 28
import jadx.api.plugins.input.data.IFieldData;
import jadx.api.plugins.input.data.IMethodData;
29
import jadx.api.plugins.input.data.annotations.EncodedValue;
S
Skylot 已提交
30 31 32 33 34 35
import jadx.api.plugins.input.data.attributes.JadxAttrType;
import jadx.api.plugins.input.data.attributes.types.AnnotationDefaultAttr;
import jadx.api.plugins.input.data.attributes.types.AnnotationDefaultClassAttr;
import jadx.api.plugins.input.data.attributes.types.InnerClassesAttr;
import jadx.api.plugins.input.data.attributes.types.InnerClsInfo;
import jadx.api.plugins.input.data.attributes.types.SourceFileAttr;
36
import jadx.api.plugins.input.data.impl.ListConsumer;
S
Skylot 已提交
37
import jadx.core.Consts;
S
Skylot 已提交
38
import jadx.core.ProcessClass;
39
import jadx.core.dex.attributes.AFlag;
40
import jadx.core.dex.attributes.AType;
41
import jadx.core.dex.attributes.nodes.InlinedAttr;
42
import jadx.core.dex.attributes.nodes.NotificationAttrNode;
S
Skylot 已提交
43 44 45 46 47 48
import jadx.core.dex.info.AccessInfo;
import jadx.core.dex.info.AccessInfo.AFType;
import jadx.core.dex.info.ClassInfo;
import jadx.core.dex.info.FieldInfo;
import jadx.core.dex.info.MethodInfo;
import jadx.core.dex.instructions.args.ArgType;
1
13.beta2 已提交
49
import jadx.core.dex.instructions.args.LiteralArg;
50
import jadx.core.dex.nodes.utils.TypeUtils;
51
import jadx.core.utils.ListUtils;
52
import jadx.core.utils.Utils;
S
Skylot 已提交
53
import jadx.core.utils.exceptions.JadxRuntimeException;
S
Skylot 已提交
54

S
Skylot 已提交
55
import static jadx.core.dex.nodes.ProcessState.LOADED;
56
import static jadx.core.dex.nodes.ProcessState.NOT_LOADED;
57

58
public class ClassNode extends NotificationAttrNode implements ILoadable, ICodeNode, Comparable<ClassNode> {
S
Skylot 已提交
59
	private static final Logger LOG = LoggerFactory.getLogger(ClassNode.class);
S
Skylot 已提交
60

61
	private final RootNode root;
62
	private final IClassData clsData;
63

S
Skylot 已提交
64
	private final ClassInfo clsInfo;
65
	private AccessInfo accessFlags;
S
Skylot 已提交
66 67
	private ArgType superClass;
	private List<ArgType> interfaces;
68
	private List<ArgType> generics = Collections.emptyList();
S
Skylot 已提交
69

70 71
	private List<MethodNode> methods;
	private List<FieldNode> fields;
72
	private List<ClassNode> innerClasses = Collections.emptyList();
S
Skylot 已提交
73

74 75
	private List<ClassNode> inlinedClasses = Collections.emptyList();

76 77
	// store smali
	private String smali;
78 79
	// store parent for inner classes or 'this' otherwise
	private ClassNode parentClass;
S
Skylot 已提交
80

S
Skylot 已提交
81
	private volatile ProcessState state = ProcessState.NOT_LOADED;
82
	private LoadStage loadStage = LoadStage.NONE;
83

84 85 86
	/**
	 * Top level classes used in this class (only for top level classes, empty for inners)
	 */
S
Skylot 已提交
87
	private List<ClassNode> dependencies = Collections.emptyList();
88 89 90 91
	/**
	 * Top level classes needed for code generation stage
	 */
	private List<ClassNode> codegenDeps = Collections.emptyList();
92 93 94
	/**
	 * Classes which uses this class
	 */
95
	private List<ClassNode> useIn = Collections.emptyList();
96 97 98
	/**
	 * Methods which uses this class (by instructions only, definition is excluded)
	 */
99
	private List<MethodNode> useInMth = Collections.emptyList();
100

S
Skylot 已提交
101 102 103
	// cache maps
	private Map<MethodInfo, MethodNode> mthInfoMap = Collections.emptyMap();

104 105
	private JavaClass javaNode;

106 107 108
	public ClassNode(RootNode root, IClassData cls) {
		this.root = root;
		this.clsInfo = ClassInfo.fromType(root, ArgType.object(cls.getType()));
109
		this.clsData = cls.copy();
110
		initialLoad(clsData);
111 112
	}

S
Skylot 已提交
113
	private void initialLoad(IClassData cls) {
S
Skylot 已提交
114
		try {
115 116 117
			addAttrs(cls.getAttributes());
			this.accessFlags = new AccessInfo(getAccessFlags(cls), AFType.CLASS);
			this.superClass = checkSuperType(cls);
118
			this.interfaces = Utils.collectionMap(cls.getInterfacesTypes(), ArgType::object);
S
Skylot 已提交
119

120 121 122
			ListConsumer<IFieldData, FieldNode> fieldsConsumer = new ListConsumer<>(fld -> FieldNode.build(this, fld));
			ListConsumer<IMethodData, MethodNode> methodsConsumer = new ListConsumer<>(mth -> MethodNode.build(this, mth));
			cls.visitFieldsAndMethods(fieldsConsumer, methodsConsumer);
123 124 125 126
			if (this.fields != null && this.methods != null) {
				// TODO: temporary solution for restore usage info in reloaded methods and fields
				restoreUsageData(this.fields, this.methods, fieldsConsumer.getResult(), methodsConsumer.getResult());
			}
127 128
			this.fields = fieldsConsumer.getResult();
			this.methods = methodsConsumer.getResult();
S
Skylot 已提交
129

S
Skylot 已提交
130 131
			initStaticValues(fields);
			processAttributes(this);
S
Skylot 已提交
132
			buildCache();
133 134 135 136 137

			// TODO: implement module attribute parsing
			if (this.accessFlags.isModuleInfo()) {
				this.addWarnComment("Modules not supported yet");
			}
S
Skylot 已提交
138
		} catch (Exception e) {
139
			throw new JadxRuntimeException("Error decode class: " + clsInfo, e);
S
Skylot 已提交
140 141 142
		}
	}

143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160
	private void restoreUsageData(List<FieldNode> oldFields, List<MethodNode> oldMethods,
			List<FieldNode> newFields, List<MethodNode> newMethods) {
		Map<FieldInfo, FieldNode> oldFieldMap = Utils.groupBy(oldFields, FieldNode::getFieldInfo);
		for (FieldNode newField : newFields) {
			FieldNode oldField = oldFieldMap.get(newField.getFieldInfo());
			if (oldField != null) {
				newField.setUseIn(oldField.getUseIn());
			}
		}
		Map<MethodInfo, MethodNode> oldMethodsMap = Utils.groupBy(oldMethods, MethodNode::getMethodInfo);
		for (MethodNode newMethod : newMethods) {
			MethodNode oldMethod = oldMethodsMap.get(newMethod.getMethodInfo());
			if (oldMethod != null) {
				newMethod.setUseIn(oldMethod.getUseIn());
			}
		}
	}

161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176
	private ArgType checkSuperType(IClassData cls) {
		String superType = cls.getSuperType();
		if (superType == null) {
			if (clsInfo.getType().getObject().equals(Consts.CLASS_OBJECT)) {
				// java.lang.Object don't have super class
				return null;
			}
			if (this.accessFlags.isModuleInfo()) {
				// module-info also don't have super class
				return null;
			}
			throw new JadxRuntimeException("No super class in " + clsInfo.getType());
		}
		return ArgType.object(superType);
	}

177 178 179 180 181 182
	public void updateGenericClsData(ArgType superClass, List<ArgType> interfaces, List<ArgType> generics) {
		this.superClass = superClass;
		this.interfaces = interfaces;
		this.generics = generics;
	}

S
Skylot 已提交
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210
	private static void processAttributes(ClassNode cls) {
		// move AnnotationDefault from cls to methods (dex specific)
		AnnotationDefaultClassAttr defAttr = cls.get(JadxAttrType.ANNOTATION_DEFAULT_CLASS);
		if (defAttr != null) {
			cls.remove(JadxAttrType.ANNOTATION_DEFAULT_CLASS);
			for (Map.Entry<String, EncodedValue> entry : defAttr.getValues().entrySet()) {
				MethodNode mth = cls.searchMethodByShortName(entry.getKey());
				if (mth != null) {
					mth.addAttr(new AnnotationDefaultAttr(entry.getValue()));
				} else {
					cls.addWarnComment("Method from annotation default annotation not found: " + entry.getKey());
				}
			}
		}

		// check source file attribute
		if (!cls.checkSourceFilenameAttr()) {
			cls.remove(JadxAttrType.SOURCE_FILE);
		}
	}

	private int getAccessFlags(IClassData cls) {
		InnerClassesAttr innerClassesAttr = get(JadxAttrType.INNER_CLASSES);
		if (innerClassesAttr != null) {
			InnerClsInfo innerClsInfo = innerClassesAttr.getMap().get(cls.getType());
			if (innerClsInfo != null) {
				return innerClsInfo.getAccessFlags();
			}
211
		}
S
Skylot 已提交
212
		return cls.getAccessFlags();
213 214
	}

215
	public static ClassNode addSyntheticClass(RootNode root, String name, int accessFlags) {
216 217 218 219 220 221 222 223 224 225
		ClassInfo clsInfo = ClassInfo.fromName(root, name);
		ClassNode existCls = root.resolveClass(clsInfo);
		if (existCls != null) {
			throw new JadxRuntimeException("Class already exist: " + name);
		}
		return addSyntheticClass(root, clsInfo, accessFlags);
	}

	public static ClassNode addSyntheticClass(RootNode root, ClassInfo clsInfo, int accessFlags) {
		ClassNode cls = new ClassNode(root, clsInfo, accessFlags);
226 227 228 229 230 231 232
		cls.add(AFlag.SYNTHETIC);
		cls.setState(ProcessState.PROCESS_COMPLETE);
		root.addClassNode(cls);
		return cls;
	}

	// Create empty class
233
	private ClassNode(RootNode root, ClassInfo clsInfo, int accessFlags) {
234
		this.root = root;
235
		this.clsData = null;
236
		this.clsInfo = clsInfo;
237 238 239 240
		this.interfaces = new ArrayList<>();
		this.methods = new ArrayList<>();
		this.fields = new ArrayList<>();
		this.accessFlags = new AccessInfo(accessFlags, AFType.CLASS);
241 242 243
		this.parentClass = this;
	}

S
Skylot 已提交
244
	private void initStaticValues(List<FieldNode> fields) {
245 246
		if (fields.isEmpty()) {
			return;
S
Skylot 已提交
247
		}
248
		List<FieldNode> staticFields = fields.stream().filter(FieldNode::isStatic).collect(Collectors.toList());
S
Skylot 已提交
249
		for (FieldNode f : staticFields) {
S
Skylot 已提交
250
			if (f.getAccessFlags().isFinal() && f.get(JadxAttrType.CONSTANT_VALUE) == null) {
251
				// incorrect initialization will be removed if assign found in constructor
S
Skylot 已提交
252
				f.addAttr(EncodedValue.NULL);
S
Skylot 已提交
253 254
			}
		}
255 256 257 258 259
		try {
			// process const fields
			root().getConstValues().processConstFields(this, staticFields);
		} catch (Exception e) {
			this.addWarnComment("Failed to load initial values for static fields", e);
260
		}
S
Skylot 已提交
261 262
	}

S
Skylot 已提交
263 264 265 266
	private boolean checkSourceFilenameAttr() {
		SourceFileAttr sourceFileAttr = get(JadxAttrType.SOURCE_FILE);
		if (sourceFileAttr == null) {
			return true;
267
		}
S
Skylot 已提交
268
		String fileName = sourceFileAttr.getFileName();
269 270 271
		if (fileName.endsWith(".java")) {
			fileName = fileName.substring(0, fileName.length() - 5);
		}
272
		if (fileName.isEmpty() || fileName.equals("SourceFile")) {
S
Skylot 已提交
273
			return false;
274 275 276 277
		}
		if (clsInfo != null) {
			String name = clsInfo.getShortName();
			if (fileName.equals(name)) {
S
Skylot 已提交
278 279 280 281 282 283 284 285 286
				return false;
			}
			ClassInfo parentCls = clsInfo.getParentClass();
			while (parentCls != null) {
				String parentName = parentCls.getShortName();
				if (parentName.equals(fileName) || parentName.startsWith(fileName + '$')) {
					return false;
				}
				parentCls = parentCls.getParentClass();
287
			}
288
			if (fileName.contains("$") && fileName.endsWith('$' + name)) {
S
Skylot 已提交
289 290 291 292
				return false;
			}
			if (name.contains("$") && name.startsWith(fileName)) {
				return false;
S
Skylot 已提交
293
			}
294
		}
S
Skylot 已提交
295
		return true;
296 297
	}

298 299 300 301
	public boolean checkProcessed() {
		return getTopParentClass().getState().isProcessComplete();
	}

302
	public void ensureProcessed() {
303 304
		if (!checkProcessed()) {
			ClassNode topParentClass = getTopParentClass();
305
			throw new JadxRuntimeException("Expected class to be processed at this point,"
306
					+ " class: " + topParentClass + ", state: " + topParentClass.getState());
307
		}
S
Skylot 已提交
308 309
	}

310 311 312 313
	public ICodeInfo decompile() {
		return decompile(true);
	}

S
Skylot 已提交
314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333
	/**
	 * WARNING: Slow operation! Use with caution!
	 */
	public ICodeInfo decompileWithMode(DecompilationMode mode) {
		DecompilationMode baseMode = root.getArgs().getDecompilationMode();
		if (mode == baseMode) {
			return decompile(true);
		}
		JadxArgs args = root.getArgs();
		try {
			unload();
			args.setDecompilationMode(mode);
			ProcessClass process = new ProcessClass(args);
			process.initPasses(root);
			return process.generateCode(this);
		} finally {
			args.setDecompilationMode(baseMode);
		}
	}

334 335 336 337
	public ICodeInfo getCode() {
		return decompile(true);
	}

338 339
	public ICodeInfo reloadCode() {
		add(AFlag.CLASS_DEEP_RELOAD);
340 341 342
		return decompile(false);
	}

343 344 345 346 347 348 349 350 351
	public void unloadCode() {
		if (state == NOT_LOADED) {
			return;
		}
		add(AFlag.CLASS_UNLOADED);
		unloadFromCache();
		deepUnload();
	}

S
Skylot 已提交
352
	public void deepUnload() {
353
		if (clsData == null) {
S
Skylot 已提交
354 355 356
			// manually added class
			return;
		}
357
		unload();
358
		clearAttributes();
359
		root().getConstValues().removeForClass(this);
360
		initialLoad(clsData);
361

362
		innerClasses.forEach(ClassNode::deepUnload);
363 364
	}

365 366 367 368 369 370 371 372
	private void unloadFromCache() {
		if (isInner()) {
			return;
		}
		ICodeCache codeCache = root().getCodeCache();
		codeCache.remove(getRawName());
	}

373
	private synchronized ICodeInfo decompile(boolean searchInCache) {
374 375 376
		if (isInner()) {
			return ICodeInfo.EMPTY;
		}
377
		ICodeCache codeCache = root().getCodeCache();
378
		String clsRawName = getRawName();
379 380
		if (searchInCache) {
			ICodeInfo code = codeCache.get(clsRawName);
381
			if (code != ICodeInfo.EMPTY) {
382 383
				return code;
			}
S
Skylot 已提交
384
		}
385 386 387 388 389 390 391
		ICodeInfo codeInfo;
		try {
			codeInfo = root.getProcessClasses().generateCode(this);
		} catch (Throwable e) {
			addError("Code generation failed", e);
			codeInfo = new SimpleCodeInfo(Utils.getStackTrace(e));
		}
392 393 394
		if (codeInfo != ICodeInfo.EMPTY) {
			codeCache.add(clsRawName, codeInfo);
		}
S
Skylot 已提交
395 396 397
		return codeInfo;
	}

398 399 400 401 402 403 404 405 406 407 408
	@Nullable
	public ICodeInfo getCodeFromCache() {
		ICodeCache codeCache = root().getCodeCache();
		String clsRawName = getRawName();
		ICodeInfo codeInfo = codeCache.get(clsRawName);
		if (codeInfo == ICodeInfo.EMPTY) {
			return null;
		}
		return codeInfo;
	}

S
Skylot 已提交
409
	@Override
S
Skylot 已提交
410
	public void load() {
S
Skylot 已提交
411
		for (MethodNode mth : getMethods()) {
S
Skylot 已提交
412 413
			try {
				mth.load();
414
			} catch (Exception e) {
415
				mth.addError("Method load error", e);
S
Skylot 已提交
416
			}
S
Skylot 已提交
417 418 419 420
		}
		for (ClassNode innerCls : getInnerClasses()) {
			innerCls.load();
		}
S
Skylot 已提交
421
		setState(LOADED);
S
Skylot 已提交
422 423 424 425
	}

	@Override
	public void unload() {
426 427 428
		if (state == NOT_LOADED) {
			return;
		}
429 430 431 432
		methods.forEach(MethodNode::unload);
		innerClasses.forEach(ClassNode::unload);
		fields.forEach(FieldNode::unloadAttributes);
		unloadAttributes();
433
		setState(NOT_LOADED);
434
		this.loadStage = LoadStage.NONE;
435
		this.smali = null;
S
Skylot 已提交
436 437
	}

S
Skylot 已提交
438
	private void buildCache() {
S
Skylot 已提交
439
		mthInfoMap = new HashMap<>(methods.size());
S
Skylot 已提交
440 441 442 443 444
		for (MethodNode mth : methods) {
			mthInfoMap.put(mth.getMethodInfo(), mth);
		}
	}

S
Skylot 已提交
445 446
	@Nullable
	public ArgType getSuperClass() {
S
Skylot 已提交
447 448 449
		return superClass;
	}

S
Skylot 已提交
450
	public List<ArgType> getInterfaces() {
S
Skylot 已提交
451 452 453
		return interfaces;
	}

454
	public List<ArgType> getGenericTypeParameters() {
455
		return generics;
456 457
	}

458 459 460 461 462 463 464 465
	public ArgType getType() {
		ArgType clsType = clsInfo.getType();
		if (Utils.notEmpty(generics)) {
			return ArgType.generic(clsType, generics);
		}
		return clsType;
	}

S
Skylot 已提交
466 467 468 469 470 471 472 473
	public List<MethodNode> getMethods() {
		return methods;
	}

	public List<FieldNode> getFields() {
		return fields;
	}

474
	public void addField(FieldNode fld) {
475 476 477
		if (fields == null || fields.isEmpty()) {
			fields = new ArrayList<>(1);
		}
478 479 480
		fields.add(fld);
	}

481 482 483 484
	public FieldNode getConstField(Object obj) {
		return getConstField(obj, true);
	}

485
	@Nullable
486
	public FieldNode getConstField(Object obj, boolean searchGlobal) {
487
		return root().getConstValues().getConstField(this, obj, searchGlobal);
488 489
	}

490
	@Nullable
1
13.beta2 已提交
491
	public FieldNode getConstFieldByLiteralArg(LiteralArg arg) {
492
		return root().getConstValues().getConstFieldByLiteralArg(this, arg);
1
13.beta2 已提交
493 494
	}

495
	public FieldNode searchField(FieldInfo field) {
S
Skylot 已提交
496
		for (FieldNode f : fields) {
497
			if (f.getFieldInfo().equals(field)) {
S
Skylot 已提交
498
				return f;
S
Skylot 已提交
499
			}
S
Skylot 已提交
500 501 502 503
		}
		return null;
	}

504 505 506 507 508 509 510 511 512
	public FieldNode searchFieldByNameAndType(FieldInfo field) {
		for (FieldNode f : fields) {
			if (f.getFieldInfo().equalsNameAndType(field)) {
				return f;
			}
		}
		return null;
	}

S
Skylot 已提交
513
	public FieldNode searchFieldByName(String name) {
S
Skylot 已提交
514
		for (FieldNode f : fields) {
S
Skylot 已提交
515
			if (f.getName().equals(name)) {
S
Skylot 已提交
516
				return f;
S
Skylot 已提交
517
			}
S
Skylot 已提交
518 519 520 521
		}
		return null;
	}

522 523 524 525 526 527 528 529 530
	public FieldNode searchFieldByShortId(String shortId) {
		for (FieldNode f : fields) {
			if (f.getFieldInfo().getShortId().equals(shortId)) {
				return f;
			}
		}
		return null;
	}

S
Skylot 已提交
531
	public MethodNode searchMethod(MethodInfo mth) {
S
Skylot 已提交
532
		return mthInfoMap.get(mth);
S
Skylot 已提交
533 534
	}

535
	public MethodNode searchMethodByShortId(String shortId) {
S
Skylot 已提交
536
		for (MethodNode m : methods) {
S
Skylot 已提交
537
			if (m.getMethodInfo().getShortId().equals(shortId)) {
S
Skylot 已提交
538
				return m;
S
Skylot 已提交
539
			}
S
Skylot 已提交
540 541 542 543
		}
		return null;
	}

544 545
	/**
	 * Return first method by original short name
546 547
	 * Note: methods are not unique by name (class can have several methods with same name but different
	 * signature)
548 549 550 551 552 553 554 555 556 557 558
	 */
	@Nullable
	public MethodNode searchMethodByShortName(String name) {
		for (MethodNode m : methods) {
			if (m.getMethodInfo().getName().equals(name)) {
				return m;
			}
		}
		return null;
	}

559
	public ClassNode getParentClass() {
560 561 562 563 564 565 566 567 568
		return parentClass;
	}

	public void updateParentClass() {
		if (clsInfo.isInner()) {
			ClassNode parent = root.resolveClass(clsInfo.getParentClass());
			if (parent != null) {
				parentClass = parent;
				return;
569 570
			}
		}
571
		parentClass = this;
572 573
	}

574 575
	public ClassNode getTopParentClass() {
		ClassNode parent = getParentClass();
576
		return parent == this ? this : parent.getTopParentClass();
577 578
	}

579 580 581 582 583 584 585 586 587 588
	public void visitParentClasses(Consumer<ClassNode> consumer) {
		ClassNode currentCls = this;
		ClassNode parentCls = currentCls.getParentClass();
		while (parentCls != currentCls) {
			consumer.accept(parentCls);
			currentCls = parentCls;
			parentCls = currentCls.getParentClass();
		}
	}

589 590 591 592 593 594 595 596 597 598 599 600 601
	public void visitSuperTypes(BiConsumer<ArgType, ArgType> consumer) {
		TypeUtils typeUtils = root.getTypeUtils();
		ArgType thisType = this.getType();
		if (!superClass.equals(ArgType.OBJECT)) {
			consumer.accept(thisType, superClass);
			typeUtils.visitSuperTypes(superClass, consumer);
		}
		for (ArgType iface : interfaces) {
			consumer.accept(thisType, iface);
			typeUtils.visitSuperTypes(iface, consumer);
		}
	}

602 603 604 605 606 607 608 609 610 611 612
	public boolean hasNotGeneratedParent() {
		if (contains(AFlag.DONT_GENERATE)) {
			return true;
		}
		ClassNode parent = getParentClass();
		if (parent == this) {
			return false;
		}
		return parent.hasNotGeneratedParent();
	}

S
Skylot 已提交
613 614 615 616
	public List<ClassNode> getInnerClasses() {
		return innerClasses;
	}

617 618 619 620
	public List<ClassNode> getInlinedClasses() {
		return inlinedClasses;
	}

621
	/**
622
	 * Get all inner and inlined classes recursively
623
	 *
624 625
	 * @param resultClassesSet
	 *                         all identified inner and inlined classes are added to this set
626
	 */
627 628 629 630 631 632 633 634 635 636
	public void getInnerAndInlinedClassesRecursive(Set<ClassNode> resultClassesSet) {
		for (ClassNode innerCls : innerClasses) {
			if (resultClassesSet.add(innerCls)) {
				innerCls.getInnerAndInlinedClassesRecursive(resultClassesSet);
			}
		}
		for (ClassNode inlinedCls : inlinedClasses) {
			if (resultClassesSet.add(inlinedCls)) {
				inlinedCls.getInnerAndInlinedClassesRecursive(resultClassesSet);
			}
637 638 639
		}
	}

S
Skylot 已提交
640
	public void addInnerClass(ClassNode cls) {
641 642 643
		if (innerClasses.isEmpty()) {
			innerClasses = new ArrayList<>(5);
		}
S
Skylot 已提交
644
		innerClasses.add(cls);
645
		cls.parentClass = this;
S
Skylot 已提交
646 647
	}

648 649 650 651
	public void addInlinedClass(ClassNode cls) {
		if (inlinedClasses.isEmpty()) {
			inlinedClasses = new ArrayList<>(5);
		}
652
		cls.addAttr(new InlinedAttr(this));
653 654 655
		inlinedClasses.add(cls);
	}

656
	public boolean isEnum() {
S
Skylot 已提交
657 658 659
		return getAccessFlags().isEnum()
				&& getSuperClass() != null
				&& getSuperClass().getObject().equals(ArgType.ENUM.getObject());
660 661
	}

S
Skylot 已提交
662
	public boolean isAnonymous() {
663
		return contains(AType.ANONYMOUS_CLASS);
664 665
	}

666 667 668 669
	public boolean isSynthetic() {
		return contains(AFlag.SYNTHETIC);
	}

670
	public boolean isInner() {
671
		return parentClass != this;
672 673
	}

674 675 676 677
	public boolean isTopClass() {
		return parentClass == this;
	}

678 679
	@Nullable
	public MethodNode getClassInitMth() {
680
		return searchMethodByShortId("<clinit>()V");
681 682 683
	}

	@Nullable
S
Skylot 已提交
684 685
	public MethodNode getDefaultConstructor() {
		for (MethodNode mth : methods) {
686
			if (mth.isDefaultConstructor()) {
S
Skylot 已提交
687
				return mth;
S
Skylot 已提交
688 689
			}
		}
S
Skylot 已提交
690
		return null;
S
Skylot 已提交
691 692
	}

693
	@Override
S
Skylot 已提交
694 695 696 697
	public AccessInfo getAccessFlags() {
		return accessFlags;
	}

698 699 700 701 702
	@Override
	public void setAccessFlags(AccessInfo accessFlags) {
		this.accessFlags = accessFlags;
	}

703 704
	@Override
	public RootNode root() {
705
		return root;
706 707
	}

708 709 710 711 712
	@Override
	public String typeName() {
		return "class";
	}

S
Skylot 已提交
713 714 715 716 717 718 719
	public String getRawName() {
		return clsInfo.getRawName();
	}

	/**
	 * Internal class info (don't use in code generation and external api).
	 */
S
Skylot 已提交
720 721 722 723 724
	public ClassInfo getClassInfo() {
		return clsInfo;
	}

	public String getShortName() {
725
		return clsInfo.getAliasShortName();
S
Skylot 已提交
726 727 728
	}

	public String getFullName() {
729
		return clsInfo.getAliasFullName();
S
Skylot 已提交
730 731 732
	}

	public String getPackage() {
733
		return clsInfo.getAliasPkg();
734 735
	}

S
Skylot 已提交
736
	public String getDisassembledCode() {
737
		if (smali == null) {
738
			StringBuilder sb = new StringBuilder();
S
Skylot 已提交
739 740
			getDisassembledCode(sb);
			sb.append(ICodeWriter.NL);
741 742 743
			Set<ClassNode> allInlinedClasses = new LinkedHashSet<>();
			getInnerAndInlinedClassesRecursive(allInlinedClasses);
			for (ClassNode innerClass : allInlinedClasses) {
S
Skylot 已提交
744 745
				innerClass.getDisassembledCode(sb);
				sb.append(ICodeWriter.NL);
746
			}
747
			smali = sb.toString();
748
		}
749 750 751
		return smali;
	}

S
Skylot 已提交
752 753
	protected void getDisassembledCode(StringBuilder sb) {
		if (clsData == null) {
754 755
			sb.append(String.format("###### Class %s is created by jadx", getFullName()));
			return;
756
		}
757
		sb.append(String.format("###### Class %s (%s)", getFullName(), getRawName()));
S
Skylot 已提交
758 759
		sb.append(ICodeWriter.NL);
		sb.append(clsData.getDisassembledCode());
760 761
	}

762 763
	public IClassData getClsData() {
		return clsData;
764 765
	}

766 767 768 769 770 771 772 773
	public ProcessState getState() {
		return state;
	}

	public void setState(ProcessState state) {
		this.state = state;
	}

774 775 776 777
	public LoadStage getLoadStage() {
		return loadStage;
	}

778 779
	public void setLoadStage(LoadStage loadStage) {
		this.loadStage = loadStage;
780 781 782 783 784
	}

	public void reloadAtCodegenStage() {
		ClassNode topCls = this.getTopParentClass();
		if (topCls.getLoadStage() == LoadStage.CODEGEN_STAGE) {
785
			throw new JadxRuntimeException("Class not yet loaded at codegen stage: " + topCls);
786 787 788 789
		}
		topCls.add(AFlag.RELOAD_AT_CODEGEN_STAGE);
	}

S
Skylot 已提交
790
	public List<ClassNode> getDependencies() {
791 792 793
		return dependencies;
	}

S
Skylot 已提交
794 795 796 797
	public void setDependencies(List<ClassNode> dependencies) {
		this.dependencies = dependencies;
	}

798 799 800 801
	public void removeDependency(ClassNode dep) {
		this.dependencies = ListUtils.safeRemoveAndTrim(this.dependencies, dep);
	}

802 803 804 805 806 807 808 809
	public List<ClassNode> getCodegenDeps() {
		return codegenDeps;
	}

	public void setCodegenDeps(List<ClassNode> codegenDeps) {
		this.codegenDeps = codegenDeps;
	}

810
	public void addCodegenDep(ClassNode dep) {
811 812 813
		if (!codegenDeps.contains(dep)) {
			this.codegenDeps = ListUtils.safeAdd(this.codegenDeps, dep);
		}
814 815
	}

816 817 818 819
	public int getTotalDepsCount() {
		return dependencies.size() + codegenDeps.size();
	}

820 821 822 823 824 825 826 827 828 829
	public List<ClassNode> getUseIn() {
		return useIn;
	}

	public void setUseIn(List<ClassNode> useIn) {
		this.useIn = useIn;
	}

	public List<MethodNode> getUseInMth() {
		return useInMth;
830 831
	}

832 833
	public void setUseInMth(List<MethodNode> useInMth) {
		this.useInMth = useInMth;
834 835
	}

836
	@Override
837 838
	public String getInputFileName() {
		return clsData == null ? "synthetic" : clsData.getInputFileName();
839 840
	}

841 842 843 844 845 846 847 848
	public JavaClass getJavaNode() {
		return javaNode;
	}

	public void setJavaNode(JavaClass javaNode) {
		this.javaNode = javaNode;
	}

849 850 851 852 853
	@Override
	public AnnType getAnnType() {
		return AnnType.CLASS;
	}

854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870
	@Override
	public int hashCode() {
		return clsInfo.hashCode();
	}

	@Override
	public boolean equals(Object o) {
		if (this == o) {
			return true;
		}
		if (o instanceof ClassNode) {
			ClassNode other = (ClassNode) o;
			return clsInfo.equals(other.clsInfo);
		}
		return false;
	}

871 872 873 874 875
	@Override
	public int compareTo(@NotNull ClassNode o) {
		return this.getFullName().compareTo(o.getFullName());
	}

S
Skylot 已提交
876 877
	@Override
	public String toString() {
S
Skylot 已提交
878
		return clsInfo.getFullName();
S
Skylot 已提交
879 880
	}
}