---

IdentifyRunnableClasses.java

package edu.ksu.cis.bandera.prog;

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
 * Bandera, a Java(TM) analysis and transformation toolkit           *
 * Copyright (C) 1998-2001 SAnToS Laboratories (santos@cis.ksu.edu)  *

 * All rights reserved.                                              *
 *                                                                   *
 * This work was done as a project in the SAnToS Laboratory,         *
 * Department of Computing and Information Sciences, Kansas State    *
 * University, USA (http://www.cis.ksu.edu/santos).                  *
 * It is understood that any modification not identified as such is  *
 * not covered by the preceding statement.                           *
 *                                                                   *
 * This work is free software; you can redistribute it and/or        *
 * modify it under the terms of the GNU Library General Public       *
 * License as published by the Free Software Foundation; either      *
 * version 2 of the License, or (at your option) any later version.  *
 *                                                                   *
 * This work is distributed in the hope that it will be useful,      *
 * but WITHOUT ANY WARRANTY; without even the implied warranty of    *
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU *
 * Library General Public License for more details.                  *
 *                                                                   *
 * You should have received a copy of the GNU Library General Public *
 * License along with this toolkit; if not, write to the             *
 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,      *
 * Boston, MA  02111-1307, USA.                                      *
 *                                                                   *
 * Java is a trademark of Sun Microsystems, Inc.                     *
 *                                                                   *
 * To submit a bug report, send a comment, or get the latest news on *
 * this project and other SAnToS projects, please visit the web-site *
 *                http://www.cis.ksu.edu/santos                      *
 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
import java.util.Vector;
import java.util.Hashtable;
import java.util.HashSet;
import java.util.Enumeration;
import java.io.*;

import ca.mcgill.sable.soot.*;
import ca.mcgill.sable.soot.jimple.*;
import ca.mcgill.sable.util.*;

import edu.ksu.cis.bandera.jjjc.*;

public class IdentifyRunnableClasses {
    private static SootMethod theRun;

private static boolean FindRun(SootClass theClass) {
    // Check to see if this class implements Runnable
        if (theClass.implementsInterface("java.lang.Runnable")) {
      // return run method
          theRun = theClass.getMethod("run", new ArrayList(), VoidType.v());
          return true; 
        }

        // Check to see if this class extends Thread
        //   check to see if it implements a run method
        if (theClass.hasSuperClass() &&
            theClass.getSuperClass().getName().equals("java.lang.Thread")) {
          // if there is no run method generate an empty one
          if (!theClass.declaresMethod("run", new ArrayList(), VoidType.v())) {
        theRun = new SootMethod("run", new ArrayList(), VoidType.v());
        JimpleBody body = (JimpleBody) Jimple.v().newBody(theRun);
        theClass.addMethod(theRun);
          } else {
            theRun = theClass.getMethod("run", new ArrayList(), VoidType.v());
          }
          return true; 
        }

        // search up
        if (theClass.hasSuperClass() &&
            FindRun(theClass.getSuperClass())) {
      // if there is a local "run" method then reset theRun
          if (theClass.declaresMethod("run", new ArrayList(), VoidType.v())) {
            theRun = theClass.getMethod("run", new ArrayList(), VoidType.v());
          }
        } 

    return false;
}
public static java.util.HashSet identify(Enumeration scIter) {
    java.util.HashSet tmpClasses = new java.util.HashSet();

        while (scIter.hasMoreElements()) {
          SootClass theClass = (SootClass) scIter.nextElement();

      // Check to see if this class has a "main" method
      ca.mcgill.sable.util.LinkedList args = new ca.mcgill.sable.util.LinkedList();
      args.add(ca.mcgill.sable.soot.ArrayType.v(RefType.v("java.lang.String"), 1));
          if (theClass.declaresMethod("main", args, VoidType.v())) {
            tmpClasses.add(theClass);
            continue;
          }

      // Check to see if this class implements Runnable
          if (theClass.implementsInterface("java.lang.Runnable")) {
            tmpClasses.add(theClass);
            continue;
          }

          // Check to see if this class extends Thread
          //   check to see if it implements a run method
          if (theClass.hasSuperClass() &&
              theClass.getSuperClass().getName().equals("java.lang.Thread")) {
        // if there is no run method generate an empty one
        if (!theClass.declaresMethod("run", new ArrayList(), VoidType.v())) {
          SootMethod newRun = new SootMethod("run", new ArrayList(), VoidType.v());
          JimpleBody body = (JimpleBody) Jimple.v().newBody(newRun);
          theClass.addMethod(newRun);
        }
            tmpClasses.add(theClass);
            continue;
          }

      // If neither of the above then search up the type hierarchy
          // to see if a supertype extends Thread
          //   recurse up the hierarchy searching
          //   if found then pick up the lowest "run" method coming down
          //
          // Also check to see if a supertype implements Runnable
          // directly or indirectly  
          //   recurse up the hierarchy searching
          //   if found then pick up the lowest "run" method coming down
          {
            theRun = null;
            if (theClass.hasSuperClass() && FindRun(theClass.getSuperClass())) {
          SootMethod newRun = new SootMethod("run", new ArrayList(), VoidType.v());
          JimpleBody body = (JimpleBody) Jimple.v().newBody(newRun);
          body.getStmtList().addAll(((JimpleBody) theRun.getBody(Jimple.v())).getStmtList());
          body.getLocals().addAll(((JimpleBody) theRun.getBody(Jimple.v())).getLocals());
          body.getTraps().addAll(((JimpleBody) theRun.getBody(Jimple.v())).getTraps());
          newRun.storeBody(Jimple.v(), body);
          theClass.addMethod(newRun);
              tmpClasses.add(theClass);
              continue;
            }
          }
    }

    return tmpClasses;
}
public static void main(String args[]) {
    SootClassManager classManager = new SootClassManager();
    CompilationManager.reset();
    CompilationManager.setFilename(args[0]);
    CompilationManager.setClasspath(".");
    CompilationManager.setIncludedPackagesOrTypes(new String[0]);

    //Compile 
    try { CompilationManager.compile(); } 
        catch (Exception e) { 
      System.out.println(e);
      System.out.println("Compilation failed");
    }

    Hashtable storedClasses = CompilationManager.getCompiledClasses();

    Enumeration scIter = storedClasses.elements();
    while (scIter.hasMoreElements()) {
      ((SootClass) scIter.nextElement()).resolveIfNecessary();
    }

    java.util.HashSet rClasses = IdentifyRunnableClasses.identify(storedClasses.elements());

    for (java.util.Iterator cIt = rClasses.iterator(); cIt.hasNext(); ) {
      ((SootClass)cIt.next()).printTo(new StoredBody(Jimple.v()), new PrintWriter(System.out, true));
    }
}
}

---