Sensitive api

src/main/rascal/lang/jimple/toolkit/BasicMetrics.rsc

@@ -2,31 +2,48 @@ module lang::jimple::toolkit::BasicMetrics
 
 import lang::jimple::core::Syntax;
 import lang::jimple::core::Context; 
+import List;
 
 /**
- * Computes the number of classes from 
- * an ExecutionContext. 
- */ 
+ * Computes the number of classes from
+ * an ExecutionContext.
+ */
 public int numberOfClasses(ExecutionContext ctx) {
   int total = 0;
   top-down visit(ctx) {
-    case classDecl(_, _, _, _, _, _): total = total + 1;  
+    case classDecl(_, _, _, _, _, _): total = total + 1;
   }	
-  return total; 
+  return total;
+}
+
+public list[str] classIdentification(ExecutionContext ctx) {
+  list[str] cl = [];
+  top-down visit(ctx) {
+    case classDecl(_, _, _, _, _, list[Method] methods): cl = cl + ["classDecl(_, _, _, _, _, <methods>)"];
+  }
+  return cl;
+}
+
+public list[Method] methodsIdentification(ExecutionContext ctx) {
+  list[Method] lm = [];
+  top-down visit(ctx) {
+    case classDecl(_, _, _, _, _, list[Method] methods): lm = lm + methods;
+  }
+  return lm;
 }
 
 /**
- * Computes the number of public methods from an 
- * execution context. 
- */ 
+ * Computes the number of public methods from an
+ * execution context.
+ */
 public int numberOfPublicMethods(ExecutionContext ctx) {
   int total = 0;
   top-down visit(ctx) {
-    case method(ms, _, _, _, _, _): { 
+    case method(ms, _, _, _, _, _): {
      if(Public() in ms) {
        total = total + 1;
      }
     }
   }	
-  return total; 
+  return total;
 }

src/main/rascal/lang/jimple/toolkit/SensitiveAPI.rsc

@@ -0,0 +1,58 @@
+module lang::jimple::toolkit::SensitiveAPI
+
+import lang::jimple::core::Context;
+import lang::jimple::Syntax;
+import lang::jimple::core::Context;
+import lang::jimple::util::Converters;
+import IO;
+import List;
+import String;
+
+
+public list[str] sensitiveFind(loc fileSensitive){
+	es = [];
+	list[str] listSignature = execute([|project://JimpleFramework/src/test/resources|], es, Analysis(methodSignatureList));
+	list[str] listFileSensitive = readFiles(fileSensitive);
+	list[str] listResult = listFileSensitive & listSignature;
+	return listResult;
+}
+
+
+public list[str] methodSignatureList(ExecutionContext ctx){
+	list[str] mS = [];
+	top-down visit(ctx) {
+    	case invokeStmt(specialInvoke(_,methodS,_)): mS = mS+signature(methodS);
+    	case invokeStmt(virtualInvoke(_,methodS,_)): mS = mS+signature(methodS);
+    	case invokeStmt(interfaceInvoke(_,methodS,_)): mS = mS+signature(methodS);
+    	case invokeStmt(staticMethodInvoke(methodS,_)): mS = mS+signature(methodS);
+    	case invokeStmt(dynamicInvoke(methodS,_,_,_)): mS = mS+signature(methodS);
+    }
+    return mS;
+}
+
+public list[str] readFiles(loc location){
+   res = [];
+   list[str] lines = readFileLines(location);
+   for (str l <- lines){
+      res = res + changeLine(l);
+   };
+   return res;
+}
+
+public str changeLine(str s){
+	str stringFinal = replaceAll(s, "\<", "");
+	stringFinal = replaceAll(stringFinal, "\>", "");
+
+	list[str] partes = split(" ", stringFinal);	
+
+	if (size(partes) >=3) {
+		str caminho = replaceAll(partes[0], ":", "");
+		caminho = replaceAll(caminho, ".", "/");
+		str retorno = partes[1];
+		retorno = replaceAll(retorno, ".", "/");
+		str metodo = partes[2];
+		return caminho + "." + metodo;
+	};
+
+	return "";
+}