Initial import.

[mirror_edk2.git] / Tools / Source / CreateMdkPkg / src / org / tianocore / packaging / FrameworkPkg.java

/** @file\r
 \r
 The file is used to install .jar file\r
 \r
 Copyright (c) 2006, Intel Corporation\r
 All rights reserved. This program and the accompanying materials\r
 are licensed and made available under the terms and conditions of the BSD License\r
 which accompanies this distribution.  The full text of the license may be found at\r
 http://opensource.org/licenses/bsd-license.php\r
 \r
 THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
 WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 \r
 **/\r
\r
package org.tianocore.packaging;\r
\r
import java.io.File;\r
import java.io.FileOutputStream;\r
import java.io.IOException;\r
import java.io.InputStream;\r
import java.util.Enumeration;\r
import java.util.jar.JarEntry;\r
import java.util.jar.JarFile;\r
\r
import javax.swing.JFrame;\r
\r
/**\r
 The class is used to install .jar file\r
 \r
 @since CreateMdkPkg 1.0\r
\r
 **/\r
public class FrameworkPkg {\r
    //\r
    // Define class members\r
    //\r
    static JFrame frame;\r
\r
    private String pkg = null;\r
\r
    private JarFile jf = null;\r
\r
    /**\r
     Main clase, used to test\r
     \r
     @param args\r
     \r
     **/\r
    public static void main(String[] args) {\r
        FrameworkPkg fp = new FrameworkPkg("C:\\Documents and Settings\\hchen30\\Desktop\\com.jar");\r
        try {\r
            fp.install("C:\\MyWorkspace" + System.getProperty("file.separator"));\r
        } catch (Exception e) {\r
            e.printStackTrace();\r
        }\r
    }\r
\r
    /**\r
     This is the default constructor\r
     \r
     **/\r
    public FrameworkPkg() {\r
\r
    }\r
\r
    /**\r
     This is the override constructor\r
     \r
     @param package_name\r
     \r
     **/\r
    public FrameworkPkg(String package_name) {\r
        pkg = package_name;\r
    }\r
\r
    /**\r
     Get package name\r
     \r
     @param package_name\r
     \r
     **/\r
    public void setPkg(String package_name) {\r
        pkg = package_name;\r
    }\r
\r
    /**\r
     Set Jarfile\r
     \r
     @throws IOException\r
     \r
     **/\r
    public void setJarFile() throws IOException {\r
        jf = new JarFile(pkg);\r
    }\r
\r
    /**\r
     Install the jar file to specific path\r
     \r
     @param dir The target path\r
     @return 0 - success\r
     @throws IOException\r
     @throws BasePkgNotInstalled\r
     @throws VerNotEqual\r
     @throws GuidNotEqual\r
     @throws SameAll\r
     \r
     **/\r
    public int install(final String dir) throws IOException, BasePkgNotInstalled, VerNotEqual, GuidNotEqual, SameAll {\r
        pre_install();\r
        extract(dir);\r
        post_install();\r
        return 0;\r
    }\r
\r
    /**\r
     \r
     @return\r
     \r
     **/\r
    public int uninstall() {\r
\r
        return 0;\r
    }\r
\r
    /**\r
     Check before install\r
     \r
     @throws IOException\r
     @throws BasePkgNotInstalled\r
     @throws VerNotEqual\r
     @throws GuidNotEqual\r
     @throws SameAll\r
     \r
     **/\r
    protected void pre_install() throws IOException, BasePkgNotInstalled, VerNotEqual, GuidNotEqual, SameAll {\r
        jf = new JarFile(pkg);\r
        if (false) {\r
            throw new BasePkgNotInstalled();\r
        }\r
        if (false) {\r
            throw new VerNotEqual();\r
        }\r
        if (false) {\r
            throw new GuidNotEqual();\r
        }\r
        if (false) {\r
            throw new SameAll();\r
        }\r
    }\r
\r
    /**\r
     End of install\r
     \r
     @throws IOException\r
     \r
     **/\r
    protected void post_install() throws IOException {\r
        jf.close();\r
\r
    }\r
\r
    /**\r
     Extract the jar file to specific dir\r
     \r
     @param dir The target path\r
     @throws IOException\r
     \r
     **/\r
    private synchronized void extract(String dir) throws IOException {\r
\r
        int i = 0;\r
        try {\r
            for (Enumeration e = jf.entries(); e.hasMoreElements(); i++) {\r
                JarEntry je = (JarEntry) e.nextElement();\r
                if (je.getName().contains("META-INF"))\r
                    continue;\r
                if (je.isDirectory()) {\r
                    new File(dir + je.getName()).mkdirs();\r
                    continue;\r
                }\r
\r
                if (je != null) {\r
                    //\r
                    // Get an input stream for the entry.\r
                    //\r
                    InputStream entryStream = jf.getInputStream(je);\r
\r
                    try {\r
                        //\r
                        // Create the output file (clobbering the file if it exists).\r
                        //\r
                        FileOutputStream file = new FileOutputStream(dir + je.getName());\r
\r
                        try {\r
                            //\r
                            // Allocate a buffer for reading the entry data.\r
                            //\r
                            byte[] buffer = new byte[1024];\r
                            int bytesRead;\r
\r
                            //\r
                            // Read the entry data and write it to the output file.\r
                            //    \r
                            while ((bytesRead = entryStream.read(buffer)) != -1) {\r
                                file.write(buffer, 0, bytesRead);\r
                            }\r
\r
                            System.out.println(je.getName() + " extracted.");\r
                        } finally {\r
                            file.close();\r
                        }\r
                    } finally {\r
                        entryStream.close();\r
                    }\r
                }\r
            }\r
        } finally {\r
            jf.close();\r
        }\r
    }\r
}\r
\r
class BasePkgNotInstalled extends Exception {\r
    final static long serialVersionUID = 0;\r
}\r
\r
class VerNotEqual extends Exception {\r
    final static long serialVersionUID = 0;\r
}\r
\r
class GuidNotEqual extends Exception {\r
    final static long serialVersionUID = 0;\r
}\r
\r
class SameAll extends Exception {\r
    final static long serialVersionUID = 0;\r
}\r