|
|
@@ -126,25 +126,30 @@ public class Wala {
|
|
|
|
|
|
|
|
|
public ArrayList<IMethod> getAllMethods() {
|
|
|
- ArrayList<IMethod> methodlist = new ArrayList<>();
|
|
|
- ArrayList<IMethod> mainlist = new ArrayList<>();
|
|
|
- for(IClass klass: cha) {
|
|
|
- if(scope.isApplicationLoader(klass.getClassLoader())) {
|
|
|
- for(IMethod m: klass.getAllMethods()) {
|
|
|
- if(m.getDeclaringClass().equals(klass)) {
|
|
|
- if(m.getSelector().toString().equals("main([Ljava/lang/String;)V")&&m.isStatic()&&m.isPublic())
|
|
|
- mainlist.add(m);
|
|
|
- methodlist.add(m);
|
|
|
+ try {
|
|
|
+ ArrayList<IMethod> methodlist = new ArrayList<>();
|
|
|
+ ArrayList<IMethod> mainlist = new ArrayList<>();
|
|
|
+ for(IClass klass: cha) {
|
|
|
+ if(scope.isApplicationLoader(klass.getClassLoader())) {
|
|
|
+ for(IMethod m: klass.getAllMethods()) {
|
|
|
+ if(m.getDeclaringClass().equals(klass)) {
|
|
|
+ if(m.getSelector().toString().equals("main([Ljava/lang/String;)V")&&m.isStatic()&&m.isPublic())
|
|
|
+ mainlist.add(m);
|
|
|
+ methodlist.add(m);
|
|
|
|
|
|
+ }
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
+ System.out.println(name + " has " + methodlist.size() + " methods\nand " + mainlist.size() + " main functions");
|
|
|
+ if(mainlist.size()!=0){
|
|
|
+ System.out.println(name);
|
|
|
+ }
|
|
|
+ return mainlist.isEmpty() ? methodlist: mainlist;
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error(e);
|
|
|
+ return new ArrayList<>();
|
|
|
}
|
|
|
- System.out.println(name + " has " + methodlist.size() + " methods\nand " + mainlist.size() + " main functions");
|
|
|
- if(mainlist.size()!=0){
|
|
|
- System.out.println(name);
|
|
|
- }
|
|
|
- return mainlist.isEmpty() ? methodlist: mainlist;
|
|
|
}
|
|
|
|
|
|
|