/*

  * Copyright (c) 2002, 2006, Oracle and/or its affiliates. All rights reserved.

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

  *

  * This code is free software; you can redistribute it and/or modify it

  * under the terms of the GNU General Public License version 2 only, as

  * published by the Free Software Foundation.  Oracle designates this

  * particular file as subject to the "Classpath" exception as provided

  * by Oracle in the LICENSE file that accompanied this code.

  *

  * This code 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 General Public License

  * version 2 for more details (a copy is included in the LICENSE file that

  * accompanied this code).

  *

  * You should have received a copy of the GNU General Public License version

  * 2 along with this work; if not, write to the Free Software Foundation,

  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

  *

  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

  * or visit www.oracle.com if you need additional information or have any

  * questions.

  */

 package sun.tools.javap;

 import java.util.*;

 import java.io.*;

 import java.util.jar.*;

 /**

  * Strores flag values according to command line options

  * and sets path where to find classes.

  *

  * @author  Sucheta Dambalkar

  */

 public class JavapEnvironment {

     //Access flags

     public static final int PRIVATE = 0;

     public static final int PROTECTED  = 1;

     public static final int PACKAGE = 2;

     public static final int PUBLIC  = 3;

     //search path flags.

     private static final int start = 0;

     private static final int  cmdboot= 1;

     private static final int sunboot = 2;

     private static final int  javaclass= 3;

     private static final int  cmdextdir= 4;

     private static final int  javaext= 5;

     private static final int  cmdclasspath= 6;

     private static final int  envclasspath= 7;

     private static final int  javaclasspath= 8;

     private static final int  currentdir = 9;

     // JavapEnvironment flag settings

     boolean showLineAndLocal = false;

     int showAccess = PACKAGE;

     boolean showDisassembled = false;

     boolean showVerbose = false;

     boolean showInternalSigs = false;

     String classPathString = null;

     String bootClassPathString = null;

     String extDirsString = null;

     boolean extDirflag = false;

     boolean nothingToDo = true;

     boolean showallAttr = false;

     String classpath = null;

     int searchpath = start;

     /**

      *  According to which flags are set,

      *  returns file input stream for classfile to disassemble.

      */

     public InputStream getFileInputStream(String Name){

         InputStream fileInStream = null;

         searchpath = cmdboot;

         try{

             if(searchpath == cmdboot){

                 if(bootClassPathString != null){

                     //search in specified bootclasspath.

                     classpath = bootClassPathString;

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path.

                     else searchpath = cmdextdir;

                 }

                 else searchpath = sunboot;

             }

             if(searchpath == sunboot){

                 if(System.getProperty("sun.boot.class.path") != null){

                     //search in sun.boot.class.path

                     classpath = System.getProperty("sun.boot.class.path");

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path

                     else searchpath = cmdextdir;

                 }

                 else searchpath = javaclass;

             }

             if(searchpath == javaclass){

                 if(System.getProperty("java.class.path") != null){

                     //search in java.class.path

                     classpath =System.getProperty("java.class.path");

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path

                     else searchpath = cmdextdir;

                 }

                 else searchpath = cmdextdir;

             }

             if(searchpath == cmdextdir){

                 if(extDirsString != null){

                     //search in specified extdir.

                     classpath = extDirsString;

                     extDirflag = true;

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path

                     else {

                         searchpath = cmdclasspath;

                         extDirflag = false;

                     }

                 }

                 else searchpath = javaext;

             }

             if(searchpath == javaext){

                 if(System.getProperty("java.ext.dirs") != null){

                     //search in java.ext.dirs

                     classpath = System.getProperty("java.ext.dirs");

                     extDirflag = true;

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path

                     else {

                         searchpath = cmdclasspath;

                         extDirflag = false;

                     }

                 }

                 else searchpath = cmdclasspath;

             }

             if(searchpath == cmdclasspath){

                 if(classPathString != null){

                     //search in specified classpath.

                     classpath = classPathString;

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path

                     else searchpath = 8;

                 }

                 else searchpath = envclasspath;

             }

             if(searchpath == envclasspath){

                 if(System.getProperty("env.class.path")!= null){

                     //search in env.class.path

                     classpath = System.getProperty("env.class.path");

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path.

                     else searchpath = javaclasspath;

                 }

                 else searchpath = javaclasspath;

             }

             if(searchpath == javaclasspath){

                 if(("application.home") == null){

                     //search in java.class.path

                     classpath = System.getProperty("java.class.path");

                     if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                     //no classes found in search path.

                     else searchpath = currentdir;

                 }

                 else searchpath = currentdir;

             }

             if(searchpath == currentdir){

                 classpath = ".";

                 //search in current dir.

                 if((fileInStream = resolvefilename(Name)) != null) return fileInStream;

                 else {

                     //no classes found in search path.

                     error("Could not find "+ Name);

                     System.exit(1);

                 }

             }

         }catch(SecurityException excsec){

             excsec.printStackTrace();

             error("fatal exception");

         }catch(NullPointerException excnull){

             excnull.printStackTrace();

             error("fatal exception");

         }catch(IllegalArgumentException excill){

             excill.printStackTrace();

             error("fatal exception");

         }

         return null;

     }

     public void error(String msg) {

         System.err.println("ERROR:" +msg);

     }

     /**

      * Resolves file name for classfile to disassemble.

      */

     public InputStream resolvefilename(String name){

         String classname = name.replace('.', '/') + ".class";

         while (true) {

             InputStream instream = extDirflag

                 ? resolveExdirFilename(classname)

                 : resolveclasspath(classname);

             if (instream != null)

                 return instream;

             int lastindex = classname.lastIndexOf('/');

             if (lastindex == -1) return null;

             classname = classname.substring(0, lastindex) + "$" +

                 classname.substring(lastindex + 1);

         }

     }

     /**

      * Resolves file name for classfile to disassemble if flag exdir is set.

      */

     public InputStream resolveExdirFilename(String classname){

         if(classpath.indexOf(File.pathSeparator) != -1){

             //separates path

             StringTokenizer st = new StringTokenizer(classpath, File.pathSeparator);

             while(st.hasMoreTokens()){

                 String path = st.nextToken();

                 InputStream in = resolveExdirFilenamehelper(path, classname);

                 if (in != null)

                     return in;

             }

         }else return (resolveExdirFilenamehelper(classpath, classname));

         return null;

     }

     /**

      * Resolves file name for classfile to disassemble.

      */

     public InputStream resolveclasspath(String classname){

         if(classpath.indexOf(File.pathSeparator) != -1){

             StringTokenizer st = new StringTokenizer(classpath, File.pathSeparator);

             //separates path.

             while(st.hasMoreTokens()){

                 String path = (st.nextToken()).trim();

                 InputStream in = resolveclasspathhelper(path, classname);

                 if(in != null) return in;

             }

             return null;

         }

         else return (resolveclasspathhelper(classpath, classname));

     }

     /**

      * Returns file input stream for classfile to disassemble if exdir is set.

      */

     public InputStream resolveExdirFilenamehelper(String path, String classname){

         File fileobj = new File(path);

         if(fileobj.isDirectory()){

             // gets list of files in that directory.

             File[] filelist = fileobj.listFiles();

             for(int i = 0; i < filelist.length; i++){

                 try{

                     //file is a jar file.

                     if(filelist[i].toString().endsWith(".jar")){

                         JarFile jfile = new JarFile(filelist[i]);

                         if((jfile.getEntry(classname)) != null){

                             InputStream filein = jfile.getInputStream(jfile.getEntry(classname));

                             int bytearraysize = filein.available();

                             byte []b =  new byte[bytearraysize];

                             int totalread = 0;

                             while(totalread < bytearraysize){

                                 totalread += filein.read(b, totalread, bytearraysize-totalread);

                             }

                             InputStream inbyte = new ByteArrayInputStream(b);

                             filein.close();

                             return inbyte;

                         }

                     } else {

                         //not a jar file.

                         String filename = path+"/"+ classname;

                         File file = new File(filename);

                         if(file.isFile()){

                             return (new FileInputStream(file));

                         }

                     }

                 }catch(FileNotFoundException fnexce){

                     fnexce.printStackTrace();

                     error("cant read file");

                     error("fatal exception");

                 }catch(IOException ioexc){

                     ioexc.printStackTrace();

                     error("fatal exception");

                 }

             }

         }

         return null;

     }

     /**

      * Returns file input stream for classfile to disassemble.

      */

     public InputStream resolveclasspathhelper(String path, String classname){

         File fileobj = new File(path);

         try{

             if(fileobj.isDirectory()){

                 //is a directory.

                 String filename = path+"/"+ classname;

                 File file = new File(filename);

                 if(file.isFile()){

                     return (new FileInputStream(file));

                 }

             }else if(fileobj.isFile()){

                 if(fileobj.toString().endsWith(".jar")){

                     //is a jar file.

                     JarFile jfile = new JarFile(fileobj);

                     if((jfile.getEntry(classname)) != null){

                         InputStream filein = jfile.getInputStream(jfile.getEntry(classname));

                         int bytearraysize = filein.available();

                         byte []b =  new byte[bytearraysize];

                         int totalread = 0;

                         while(totalread < bytearraysize){

                                 totalread += filein.read(b, totalread, bytearraysize-totalread);

                         }

                         InputStream inbyte = new ByteArrayInputStream(b);

                         filein.close();

                          return inbyte;

                     }

                 }

             }

         }catch(FileNotFoundException fnexce){

             fnexce.printStackTrace();

             error("cant read file");

             error("fatal exception");

         }catch(IOException ioexce){

             ioexce.printStackTrace();

             error("fatal exception");

         }

         return null;

     }

 }