X-Git-Url: http://git.megacz.com/?p=org.ibex.core.git;a=blobdiff_plain;f=src%2Forg%2Fibex%2Futil%2FNanoGoat.java;h=2b5b772d3eca2957d17ec42db8691166e7d06ada;hp=7f1a50e108bf3e0511260ee4cdf121a825438f27;hb=01732f3955ae92c0520283b44f52d1ec69f9f9f2;hpb=a41252ba1e0f5dd90100aa2064c011268809c038

diff --git a/src/org/ibex/util/NanoGoat.java b/src/org/ibex/util/NanoGoat.java
index 7f1a50e..2b5b772 100644
--- a/src/org/ibex/util/NanoGoat.java
+++ b/src/org/ibex/util/NanoGoat.java
@@ -46,27 +46,31 @@ public class NanoGoat {
             if (meths[i].getName().equals(methodname))
                 visitJavaMethod(repo.loadClass(classname), meths[i]);
     }
-    public static void main(String[] s) throws Exception {
+    public static void main(String[] args) throws Exception {
         int start = 1;
-        repo = SyntheticRepository.getInstance(new ClassPath(s[0]));
+        repo = SyntheticRepository.getInstance(new ClassPath(args[0]));
 
         NanoGoat bcp = new NanoGoat();
-        for(int i=start; i<s.length; i++) {
+        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
+        for(String s = br.readLine(); s != null; s = br.readLine()) {
+            s = s.trim();
+            if (s.length() == 0) continue;
             try {
-                if (s[i].endsWith(".class")) {
-                    bcp.visitJavaClass(repo.loadClass(s[i].substring(0, s[i].length() - 6)));
+                if (s.endsWith("$")) s = s.substring(0, s.length() - 1);
+                if (s.endsWith(".class")) {
+                    bcp.visitJavaClass(repo.loadClass(s.substring(0, s.length() - 6)));
                 } else {
-                    JavaClass cl = repo.loadClass(s[i].substring(0, s[i].lastIndexOf('.')));;
+                    JavaClass cl = repo.loadClass(s.substring(0, s.lastIndexOf('.')));;
                     bcp.visitJavaClass(cl);
-                    bcp.loadMethod(s[i]);
+                    bcp.loadMethod(s);
                     Field[] fields = cl.getFields();
                     for(int j=0; j<fields.length; j++) {
-                        if (fields[j].getName().equals(s[i].substring(s[i].lastIndexOf('.') + 1)))
+                        if (fields[j].getName().equals(s.substring(s.lastIndexOf('.') + 1)))
                             bcp.visitJavaField(fields[j], cl);
                     }
                 }
             } catch (Exception e) {
-                System.out.println("WARNING: couldn't load class for " + s[i]);
+                System.out.println("WARNING: couldn't load class for " + s);
                 e.printStackTrace();
             }
         }
@@ -124,8 +128,8 @@ public class NanoGoat {
         System.out.println();
 
         System.out.println("Dumping...");
-        ZipFile zf = new ZipFile(s[0]);
-        ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(s[0] + ".tmp"));
+        ZipFile zf = new ZipFile(args[0]);
+        ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(args[0] + ".tmp"));
         Enumeration e = zf.entries();
         while(e.hasMoreElements()) {
             ZipEntry ze = ((ZipEntry)e.nextElement());
@@ -137,7 +141,7 @@ public class NanoGoat {
         }
         zos.close();
         zf.close();
-        new File(s[0] + ".tmp").renameTo(new File(s[0] + ".pruned"));
+        new File(args[0] + ".tmp").renameTo(new File(args[0] + ".pruned"));
     }
 
     public static void dump(JavaClass clazz, ZipOutputStream zos) throws Exception {