1 package org.ibex.arenaj;
8 public class Main extends SceneTransformer {
10 public static final int initialSize = 100;
12 private static Main instance = new Main();
14 public static Main v() { return instance; }
16 public static void main(String[] args) throws Exception {
17 if(args.length == 0) {
18 System.out.println("Syntax: java " + v().getClass().getName() + " [soot options]");
21 PackManager.v().getPack("wjtp").add(new Transform("wjtp.tx", Main.v()));
22 String[] args2 = new String[args.length + 14];
23 System.arraycopy(args, 0, args2, 0, args.length-1);
25 System.getProperty("java.class.path") +
27 System.getProperty("sun.boot.class.path");
28 String extDirs = System.getProperty("java.ext.dirs");
29 if (extDirs != null) {
30 StringTokenizer st = new StringTokenizer(extDirs, File.pathSeparatorChar+"");
31 while(st.hasMoreTokens()) {
32 String goo = st.nextToken();
33 System.out.println("goo " + goo);
34 File f = new File(goo);
35 if (!f.exists() || !f.isDirectory()) continue;
36 String[] jars = f.list();
37 for(int i=0; i<jars.length; i++)
38 sootcp += File.pathSeparator + f.getPath() + File.separatorChar + jars[i];
41 args2[args.length - 1] = "-w";
42 args2[args.length + 0] = "-p";
43 args2[args.length + 1] = "cg";
44 args2[args.length + 2] = "enabled:false";
45 args2[args.length + 3] = "-p";
46 args2[args.length + 4] = "wjap";
47 args2[args.length + 5] = "enabled:false";
48 args2[args.length + 6] = "-p";
49 args2[args.length + 7] = "jtp";
50 args2[args.length + 8] = "enabled:false";
51 //args2[args.length + 9] = "-keep-line-number";
52 args2[args.length + 9] = "-allow-phantom-refs";
53 args2[args.length + 10] = "-allow-phantom-refs";
54 args2[args.length + 11] = "-cp";
55 args2[args.length + 12] = sootcp;
56 System.out.println("sootcp => " + sootcp);
57 args2[args.length + 13] = args[args.length-1];
58 for(int i=0; i<args2.length; i++) System.out.println("args2["+i+"] = " + args2[i]);
59 soot.Main.main(args2);
63 public static Local viaLocal(Value v, Body b) {
64 Local l = Jimple.v().newLocal("tmpRef" + (tfr++), v.getType());
66 if (v instanceof IdentityRef) {
67 b.getUnits().addFirst(Jimple.v().newIdentityStmt(l, v));
69 b.getUnits().addFirst(Jimple.v().newAssignStmt(l, v));
73 public static Local viaLocal(Value v, Body b, Unit u) {
74 Local l = Jimple.v().newLocal("tmpRef" + (tfr++), v.getType());
76 if (v instanceof IdentityRef) {
77 b.getUnits().insertBefore(Jimple.v().newIdentityStmt(l, v), u);
79 b.getUnits().insertBefore(Jimple.v().newAssignStmt(l, v), u);
84 HashMap map = new HashMap();
85 HashSet size_added = new HashSet();
87 public boolean isGladiator(SootClass c) { return c.implementsInterface("org.ibex.arenaj.Gladiator"); }
88 public boolean isGladiatorField(SootField f)
89 { return isGladiator(f.getDeclaringClass()) && !f.getName().equals("this") && f.getName().indexOf('$')==-1; }
90 public boolean isGladiatorFieldRef(SootFieldRef f) {
91 return isGladiator(f.declaringClass()) && !f.name().equals("this") && f.name().indexOf('$')==-1;
93 public SootField getGladiatorField(SootField f) { return getGladiatorField(f.makeRef()); }
94 public SootField getGladiatorField(SootFieldRef f) {
95 SootClass c = f.declaringClass();
96 SootClass oc = Scene.v().getSootClass(c.getName().substring(0, c.getName().lastIndexOf('$')));
97 String sig = f.declaringClass().getName()+"."+f.name();
98 if (map.get(sig) != null) return (SootField)map.get(sig);
100 if (t instanceof RefType && isGladiator(((RefType)t).getSootClass())) t = IntType.v();
101 SootField nf = new SootField(c.getShortName().substring(c.getShortName().lastIndexOf('$')+1) + "$" + f.name(),
106 Body body = getInitBody(oc);
107 Expr newArr = Jimple.v().newNewArrayExpr(t, IntConstant.v(initialSize));
108 Local newArrLocal = Jimple.v().newLocal("tmpRef" + (tfr++), f.type().makeArrayType());
109 body.getLocals().add(newArrLocal);
110 InstanceFieldRef ifr = Jimple.v().newInstanceFieldRef(thisLocal(c,body), nf.makeRef());
111 body.getUnits().addFirst(Jimple.v().newAssignStmt(ifr, newArrLocal));
112 body.getUnits().addFirst(Jimple.v().newAssignStmt(newArrLocal, newArr));
118 public Body getInitBody(SootClass c) {
119 c.setApplicationClass();
120 List list = new LinkedList();
121 SootMethod m = c.getMethod("<init>", list);
122 if (!m.hasActiveBody()) {
123 JimpleBody b = (JimpleBody)Jimple.v().newBody(m);
126 return m.getActiveBody();
129 public Local thisLocal(SootClass c, Body b) { return viaLocal(Jimple.v().newThisRef(c.getType()), b); }
130 public SootFieldRef getGladiatorFieldSizeRef(SootClass c) {
131 SootClass mc = getParent(c);
132 String name = c.getShortName().substring(c.getShortName().lastIndexOf('$')+1) + "$$size";
133 if (map.get(name) == null) {
134 SootField f = new SootField(name, IntType.v());
137 Body body = getInitBody(mc);
138 InstanceFieldRef ifr = Jimple.v().newInstanceFieldRef(thisLocal(c,body), f.makeRef());
139 body.getUnits().addFirst(Jimple.v().newAssignStmt(ifr, IntConstant.v(0)));
143 return Scene.v().makeFieldRef(mc, name, IntType.v(), false);
145 public SootFieldRef getGladiatorFieldMaxRef(SootClass c) {
146 SootClass mc = getParent(c);
147 String name = c.getShortName().substring(c.getShortName().lastIndexOf('$')+1) + "$$max";
148 if (map.get(name) == null) {
149 SootField f = new SootField(name, IntType.v());
152 Body body = getInitBody(mc);
153 InstanceFieldRef ifr = Jimple.v().newInstanceFieldRef(thisLocal(c,body), f.makeRef());
154 body.getUnits().addFirst(Jimple.v().newAssignStmt(ifr, IntConstant.v(initialSize)));
158 return Scene.v().makeFieldRef(mc, name, IntType.v(), false);
161 public boolean isGladiatorType(Type t) { return (t instanceof RefType) && isGladiator(((RefType)t).getSootClass()); }
166 boolean done = false;
167 public void internalTransform(String phaseName, Map options) {
168 System.out.println("begun");
169 List ac = new LinkedList();
170 for(Iterator it = Scene.v().getClasses().iterator(); it.hasNext();) {
171 SootClass sc = (SootClass)it.next();
172 sc.setLibraryClass();
174 for(Iterator it = Scene.v().getClasses().iterator(); it.hasNext();) {
175 SootClass sc = (SootClass)it.next();
176 if (isGladiator(sc)) {
177 System.out.println(" application class: " + sc.getName());
178 System.out.println(" application class: " + getParent(sc).getName());
179 sc.setApplicationClass();
180 getParent(sc).setApplicationClass();
182 ac.add(getParent(sc));
183 for(Iterator i2 = sc.getMethods().iterator(); i2.hasNext();) {
184 SootMethod m = (SootMethod)i2.next();
185 if (m.isConcrete()) m.retrieveActiveBody();
187 for(Iterator i2 = getParent(sc).getMethods().iterator(); i2.hasNext();) {
188 SootMethod m = (SootMethod)i2.next();
189 if (m.isConcrete()) m.retrieveActiveBody();
193 for(Iterator it = ac.iterator(); it.hasNext();) {
194 SootClass sc = (SootClass)it.next();
195 if (sc.isInterface()) continue;
196 if (!isGladiator(sc)) continue;
197 System.out.println("fixing Gladiator class: " + sc);
200 for(Iterator it = ac.iterator(); it.hasNext();) {
201 SootClass sc = (SootClass)it.next();
202 if (sc.isInterface()) continue;
203 System.out.println(" updating class: " + sc);
208 public SootClass getParent(SootClass sc) {
209 return Scene.v().getSootClass(sc.getName().substring(0, sc.getName().lastIndexOf('$')));
212 public void fixClass(SootClass sc) {
213 SootClass mc = getParent(sc);
214 String incFuncName = sc.getShortName().substring(sc.getShortName().lastIndexOf('$')+1) + "$$inc";
215 SootMethod method = new SootMethod(incFuncName, new LinkedList(),
216 IntType.v(), Modifier.PRIVATE,
218 mc.addMethod(method);
219 Body body = Jimple.v().newBody(method);
220 method.setActiveBody(body);
221 ((JimpleBody)body).insertIdentityStmts();
222 Local l = Jimple.v().newLocal("tmpRef" + (tfr++), IntType.v());
223 body.getLocals().add(l);
224 Local l2 = Jimple.v().newLocal("tmpRef" + (tfr++), IntType.v());
225 body.getLocals().add(l2);
226 Local l3 = Jimple.v().newLocal("tmpRef" + (tfr++), IntType.v());
227 body.getLocals().add(l3);
228 InstanceFieldRef sfr = Jimple.v().newInstanceFieldRef(body.getThisLocal(), getGladiatorFieldSizeRef(sc));
229 body.getUnits().add(Jimple.v().newAssignStmt(l, sfr));
230 body.getUnits().add(Jimple.v().newAssignStmt(l2, Jimple.v().newAddExpr(l, IntConstant.v(1))));
231 InstanceFieldRef maxField = Jimple.v().newInstanceFieldRef(body.getThisLocal(), getGladiatorFieldMaxRef(sc));
232 body.getUnits().add(Jimple.v().newAssignStmt(l3, maxField));
233 Stmt stmt = Jimple.v().newReturnStmt(l2);
234 body.getUnits().add(Jimple.v().newIfStmt(Jimple.v().newLtExpr(l2, l3), stmt));
236 Local l4 = Jimple.v().newLocal("tmpRef" + (tfr++), IntType.v());
237 body.getLocals().add(l4);
238 body.getUnits().add(Jimple.v().newAssignStmt(l4, Jimple.v().newShlExpr(l3, IntConstant.v(1))));
239 body.getUnits().add(Jimple.v().newAssignStmt(maxField, l4));
241 for(Iterator it = sc.getFields().iterator(); it.hasNext();) {
242 SootField f = getGladiatorField((SootField)it.next());
243 InstanceFieldRef ifr = Jimple.v().newInstanceFieldRef(body.getThisLocal(), f.makeRef());
244 Local ll0 = Jimple.v().newLocal("tmpRef" + (tfr++), f.getType());
245 body.getLocals().add(ll0);
246 body.getUnits().add(Jimple.v().newAssignStmt(ll0, ifr));
247 Local ll = Jimple.v().newLocal("tmpRef" + (tfr++), f.getType());
248 body.getLocals().add(ll);
249 body.getUnits().add(Jimple.v().newAssignStmt(ll,
250 Jimple.v().newNewArrayExpr(((ArrayType)f.getType()).getElementType(),
252 Type ot = Scene.v().getSootClass("java.lang.Object").getType();
253 List types = new LinkedList();
255 types.add(IntType.v());
257 types.add(IntType.v());
258 types.add(IntType.v());
259 SootMethodRef arrayCopy =
260 Scene.v().makeMethodRef(Scene.v().getSootClass("java.lang.System"),
265 List args = new LinkedList();
267 args.add(IntConstant.v(0));
269 args.add(IntConstant.v(0));
271 body.getUnits().add(Jimple.v().newInvokeStmt(Jimple.v().newStaticInvokeExpr(arrayCopy, args)));
272 body.getUnits().add(Jimple.v().newAssignStmt(ifr,ll));
274 for(Iterator it = sc.getMethods().iterator(); it.hasNext();) {
275 SootMethod m = (SootMethod)it.next();
276 if (!m.isConcrete()) continue;
277 if (isGladiator(m.getDeclaringClass()) && m.getName().equals("<init>")) {
278 SootClass c = m.getDeclaringClass();
279 String name = c.getShortName().substring(c.getShortName().lastIndexOf('$')+1) + "$$$init";
280 List li = m.getParameterTypes();
284 SootMethod nm = new SootMethod(name, li, isGladiatorType(m.getReturnType()) ? IntType.v() : m.getReturnType());
285 JimpleBody bod = Jimple.v().newBody(nm);
286 getParent(c).addMethod(nm);
287 //bod.insertIdentityStmts();
288 bod.importBodyContentsFrom(m.getActiveBody());
289 nm.setActiveBody(bod);
292 System.out.println("examining " + sc.getName() + "." + m.getSignature());
293 m.retrieveActiveBody();
294 if (m.isStatic()) continue;
296 String name = sc.getShortName().substring(sc.getShortName().lastIndexOf('$')+1) + "$$" + m.getName();
297 List list = new LinkedList();
298 list.addAll(m.getParameterTypes());
299 list.add(IntType.v());
300 for(Iterator i = list.iterator(); i.hasNext();) System.out.println(i.next());
301 SootMethod m2 = new SootMethod(name, list, m.getReturnType());
302 getParent(sc).addMethod(m2);
304 JimpleBody ab = (JimpleBody)Jimple.v().newBody(m2);
305 ab.importBodyContentsFrom(m.getActiveBody());
306 m2.setActiveBody(ab);
307 Local loc = Jimple.v().newLocal("tmpRef" + (tfr++), getParent(sc).getType());
308 ab.getLocals().add(loc);
309 // FIXME: insert assignment to this
311 for(Iterator z = ab.getLocals().iterator(); z.hasNext();) {
312 Local loc2 = (Local)z.next();
313 if (isGladiatorType(loc2.getType())) {
314 loc2.setType(IntType.v());
318 Chain units = ab.getUnits();
319 for(Iterator stmtIt = units.snapshotIterator(); stmtIt.hasNext();) {
320 Stmt s = (Stmt) stmtIt.next();
321 if (s instanceof IdentityStmt) {
322 IdentityStmt is = (IdentityStmt)s;
323 Local left = (Local)is.getLeftOp();
324 if (is.getRightOp() instanceof ThisRef) {
325 left.setType(IntType.v());
326 is.getRightOpBox().setValue(Jimple.v().newParameterRef(IntType.v(), m.getParameterCount()));
330 for(Iterator i = s.getUseAndDefBoxes().iterator(); i.hasNext();) {
332 if (o instanceof ValueBox) {
333 ValueBox vb = (ValueBox)o;
334 if (vb.getValue() instanceof ThisRef) {
335 System.out.println(s);
346 body.getUnits().add(stmt);
349 public void nuke(SootClass c) {
350 for(Iterator it = c.getFields().iterator(); it.hasNext();) {
351 SootField f = (SootField)it.next();
352 Type t = f.getType();
353 if (t instanceof RefType) {
354 RefType rt = (RefType)t;
355 SootClass sc = rt.getSootClass();
356 if (isGladiator(sc)) f.setType(IntType.v());
357 } else if (t instanceof ArrayType) {
358 ArrayType at = (ArrayType)t;
359 t = at.getElementType();
360 if (!(t instanceof RefType)) continue;
361 RefType rt = (RefType)t;
362 SootClass sc = rt.getSootClass();
363 if (isGladiator(sc)) f.setType(IntType.v().makeArrayType());
367 List list = new LinkedList(); list.addAll(c.getMethods());
368 for(Iterator it = list.iterator(); it.hasNext();) {
369 SootMethod m = (SootMethod)it.next();
371 if (m.hasActiveBody()) b = fixBody(m.getActiveBody(), c, m);
372 List l2 = new LinkedList();
373 List l = m.getParameterTypes();
374 for(Iterator it2 = l.iterator(); it2.hasNext();) {
375 Type t = (Type)it2.next();
376 l2.add(isGladiatorType(t) ? IntType.v() : t);
378 Type t = m.getReturnType();
379 if (isGladiatorType(t)) {
381 if (m.hasActiveBody()) {
382 Body bod = m.getActiveBody();
383 for(Iterator stmtIt = bod.getUnits().snapshotIterator(); stmtIt.hasNext();) {
384 Stmt s = (Stmt) stmtIt.next();
385 if (s instanceof ReturnStmt) {
386 if (((ReturnStmt)s).getOp().getType() instanceof NullType) {
387 ((ReturnStmt)s).getOpBox().setValue(IntConstant.v(-1));
393 String name = m.getName();
394 SootMethod meth = new SootMethod(name, l2, isGladiatorType(t) ? IntType.v() : t, m.getModifiers());
396 JimpleBody b2 = Jimple.v().newBody(meth);
397 b2.importBodyContentsFrom(b);
398 meth.setActiveBody(b2);
406 protected Body fixBody(Body body, SootClass ownerClass, SootMethod smeth) {
407 if (body instanceof JimpleBody) {
408 JimpleBody b2 = Jimple.v().newBody(smeth);
409 //b2.insertIdentityStmts();
410 b2.importBodyContentsFrom(body);
413 Chain units = body.getUnits();
414 System.out.println("fixBody("+body.getMethod()+")");
415 for(Iterator it = body.getLocals().snapshotIterator(); it.hasNext();) {
416 Local l = (Local)it.next();
417 if (isGladiatorType(l.getType())) l.setType(IntType.v());
419 for(int qq=0; qq<2; qq++) for(Iterator stmtIt = units.snapshotIterator(); stmtIt.hasNext();) {
420 Stmt s = (Stmt) stmtIt.next();
421 if (s instanceof DefinitionStmt) {
422 DefinitionStmt ds = (DefinitionStmt)s;
423 if (ds.getLeftOp().getType() instanceof PrimType && ds.getRightOp().getType() instanceof NullType) {
424 ds.getRightOpBox().setValue(IntConstant.v(-1));
427 List l = s.getUseAndDefBoxes();
428 for(Iterator it = l.iterator(); it.hasNext();) {
429 Object o = it.next();
430 if (o instanceof ValueBox) {
431 ValueBox vb = (ValueBox)o;
432 Value v = vb.getValue();
434 if (v instanceof BinopExpr) {
435 BinopExpr boe = (BinopExpr)v;
436 if (boe.getOp1().getType() instanceof PrimType && boe.getOp2().getType() instanceof NullType) {
437 boe.setOp2(IntConstant.v(-1));
439 if (boe.getOp2().getType() instanceof PrimType && boe.getOp1().getType() instanceof NullType) {
440 boe.setOp1(IntConstant.v(-1));
444 if (v instanceof NewExpr) {
445 NewExpr ne = (NewExpr)v;
446 if (isGladiatorType(ne.getBaseType())) {
447 SootClass sc = ((RefType)ne.getBaseType()).getSootClass();
448 SootClass mc = getParent(sc);
449 String incFuncName = sc.getShortName().substring(sc.getShortName().lastIndexOf('$')+1) + "$$inc";
450 SootMethodRef smr = Scene.v().makeMethodRef(mc, incFuncName, new LinkedList(), IntType.v(), false);
451 Expr invokeExpr = Jimple.v().newSpecialInvokeExpr(thisLocal(mc,body), smr);
452 Local ll = viaLocal(invokeExpr, body, s);
459 if (v instanceof InvokeExpr) {
460 InvokeExpr ie = (InvokeExpr)v;
461 SootMethodRef mr = ie.getMethodRef();
462 String name = mr.name();
463 if (v instanceof InstanceInvokeExpr && isGladiator(mr.declaringClass())) {
464 InstanceInvokeExpr iie = (InstanceInvokeExpr)v;
465 List li = new LinkedList();
466 li.addAll(iie.getArgs());
467 LinkedList pl = new LinkedList();
468 pl.addAll(mr.parameterTypes());
469 if (mr.name().equals("<init>") && isGladiator(mr.declaringClass())) {
473 //pl.addFirst(body.getThisLocal());
476 li.add(viaLocal(iie.getBase(),body,s));
477 SootClass sc = mr.declaringClass();
478 name = sc.getShortName().substring(sc.getShortName().lastIndexOf('$')+1) + "$$" + name;
479 mr = Scene.v().makeMethodRef(getParent(sc),
482 isGladiatorType(mr.returnType()) ? IntType.v() : mr.returnType(),
484 ie = Jimple.v().newVirtualInvokeExpr(viaLocal(thisLocal(getParent(sc),body),body,s), mr, li);
487 List l0 = mr.parameterTypes();
488 List l2 = new LinkedList();
489 for(Iterator it2 = l0.iterator(); it2.hasNext();) {
490 Type t = (Type)it2.next();
491 l2.add(isGladiatorType(t) ? IntType.v() : t);
493 mr = Scene.v().makeMethodRef(mr.declaringClass(),
496 isGladiatorType(mr.returnType()) ? IntType.v() : mr.returnType(),
502 for(int i=0; i<ie.getArgCount(); i++) {
503 ValueBox b = ie.getArgBox(i);
504 Value val = b.getValue();
505 if (mr.parameterType(i) instanceof RefType && val.getType() instanceof PrimType) {
506 SootClass intClass = Scene.v().getSootClass("java.lang.Integer");
507 List typelist = new LinkedList();
508 typelist.add(IntType.v());
509 SootMethod intMethod = intClass.getMethod("<init>", typelist);
510 Local loc = viaLocal(Jimple.v().newNewExpr(RefType.v(intClass)), body, s);
511 List list = new LinkedList();
513 units.insertBefore(Jimple.v().newInvokeStmt(Jimple.v().newSpecialInvokeExpr(loc,
519 if (val != null && val.getType() instanceof NullType && mr.parameterType(i) instanceof IntType) {
520 b.setValue(IntConstant.v(-1));
525 } else if (v instanceof CastExpr) {
526 CastExpr ce = (CastExpr)v;
527 if (isGladiatorType(ce.getCastType())) {
528 SootClass mc = getParent(((RefType)ce.getCastType()).getSootClass());
529 SootClass ic = Scene.v().getSootClass("java.lang.Integer");
530 ce.setCastType(ic.getType());
531 // FIXME deal with null
532 SootMethodRef mr = Scene.v().makeMethodRef(ic, "intValue", new LinkedList(), IntType.v(), false);
533 InvokeExpr iie = Jimple.v().newVirtualInvokeExpr(viaLocal(ce, body, s),
536 vb.setValue(viaLocal(iie, body, s));
541 } else if (v instanceof InstanceFieldRef) {
542 InstanceFieldRef ifr = (InstanceFieldRef)v;
543 SootFieldRef fr = ifr.getFieldRef();
545 if (isGladiatorFieldRef(fr)) {
546 SootClass mc = getParent(fr.declaringClass());
547 SootFieldRef sf = getGladiatorField(fr).makeRef();
548 InstanceFieldRef sfr = Jimple.v().newInstanceFieldRef(thisLocal(mc, body), sf);
549 System.out.println("s is " + s);
550 ArrayRef ar = Jimple.v().newArrayRef(viaLocal(sfr, body, s), ifr.getBase());
553 if ((t instanceof RefType) && isGladiator(((RefType)t).getSootClass())) {
554 SootClass tc = ((RefType)t).getSootClass();
555 SootClass mc = getParent(tc);
556 ifr.setFieldRef(Scene.v().makeFieldRef(mc, fr.name(), IntType.v(), false));