Quelle  Reader.java   Sprache: JAVA

/*
 * Copyright (c) 1997, 2021, 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.
 */


/*
 * The Original Code is HAT. The Initial Developer of the
 * Original Code is Bill Foote, with contributions from others
 * at JavaSoft/Sun.
 */

package jdk.test.lib.hprof.parser;

import java.io.*;
import java.util.zip.GZIPInputStream;
import jdk.test.lib.hprof.model.*;

/**
 * Abstract base class for reading object dump files.  A reader need not be
 * thread-safe.
 *
 * @author      Bill Foote
 */


public abstract class Reader {
    protected PositionDataInputStream in;
    // Magic number of gzip dump file header.
    private static final int GZIP_HEADER_MAGIC = 0x1f8b08;

    protected Reader(PositionDataInputStream in) {
        this.in = in;
    }

    /**
     * Read a snapshot from a data input stream.  It is assumed that the magic
     * number has already been read.
     */
    abstract public Snapshot read() throws IOException;

    /**
     * Read a snapshot from a file.
     *
     * @param heapFile The name of a file containing a heap dump
     * @param callStack If true, read the call stack of allocaation sites
     */
    public static Snapshot readFile(String heapFile, boolean callStack,
                                    int debugLevel)
            throws IOException {
        int dumpNumber = 1;
        int pos = heapFile.lastIndexOf('#');
        if (pos > -1) {
            String num = heapFile.substring(pos+1, heapFile.length());
            try {
                dumpNumber = Integer.parseInt(num, 10);
            } catch (java.lang.NumberFormatException ex) {
                String msg = "In file name \"" + heapFile
                             + "\", a dump number was "
                             + "expected after the :, but \""
                             + num + "\" was found instead.";
                System.err.println(msg);
                throw new IOException(msg);
            }
            heapFile = heapFile.substring(0, pos);
        }
        try (FileInputStream fis = new FileInputStream(heapFile);
             BufferedInputStream bis = new BufferedInputStream(fis);
             PositionDataInputStream in = new PositionDataInputStream(bis)) {
            int i = in.readInt();
            if (i == HprofReader.MAGIC_NUMBER) {
                Reader r
                    = new HprofReader(heapFile, in, dumpNumber,
                                      callStack, debugLevel);
                return r.read();
            } else if ((i >>> 8) == GZIP_HEADER_MAGIC) {
                // Possible gziped file, try decompress it and get the stack trace.
                in.close();
                String deCompressedFile = "heapdump" + System.currentTimeMillis() + ".hprof";
                File out = new File(deCompressedFile);
                // Decompress to get dump file.
                try (FileInputStream heapFis = new FileInputStream(heapFile);
                     GZIPInputStream gis = new GZIPInputStream(heapFis);
                     FileOutputStream fos = new FileOutputStream(out)) {
                    byte[] buffer = new byte[1024 * 1024];
                    int len = 0;
                    while ((len = gis.read(buffer)) > 0) {
                        fos.write(buffer, 0, len);
                    }
                } catch (Exception e) {
                    out.delete();
                    throw new IOException("Cannot decompress the compressed hprof file", e);
                }
                // Check dump data header and print stack trace.
                try (FileInputStream outFis = new FileInputStream(out);
                     BufferedInputStream outBis = new BufferedInputStream(outFis);
                     PositionDataInputStream pdin = new PositionDataInputStream(outBis)) {
                    i = pdin.readInt();
                    if (i == HprofReader.MAGIC_NUMBER) {
                        HprofReader r
                            = new HprofReader(deCompressedFile, pdin, dumpNumber,
                                              true, debugLevel);
                        return r.read();
                    } else {
                        throw new IOException("Unrecognized magic number found in decompressed data: " + i);
                    }
                } finally {
                    out.delete();
                }
            } else {
                throw new IOException("Unrecognized magic number: " + i);
            }
        }
    }

    /**
     * Get Stack Traces from a Hprof file.
     *
     * @param heapFile The name of a file containing a heap dump
     */
    public static String getStack(String heapFile, int debugLevel)
            throws IOException {
        int dumpNumber = 1;
        int pos = heapFile.lastIndexOf('#');
        if (pos > -1) {
            String num = heapFile.substring(pos+1, heapFile.length());
            try {
                dumpNumber = Integer.parseInt(num, 10);
            } catch (java.lang.NumberFormatException ex) {
                String msg = "In file name \"" + heapFile
                             + "\", a dump number was "
                             + "expected after the :, but \""
                             + num + "\" was found instead.";
                System.err.println(msg);
                throw new IOException(msg);
            }
            heapFile = heapFile.substring(0, pos);
        }
        try (FileInputStream fis= new FileInputStream(heapFile);
             BufferedInputStream bis = new BufferedInputStream(fis);
             PositionDataInputStream in = new PositionDataInputStream(bis)) {
            int i = in.readInt();
            if (i == HprofReader.MAGIC_NUMBER) {
                HprofReader r
                    = new HprofReader(heapFile, in, dumpNumber,
                                      true, debugLevel);
                r.read();
                return r.printStackTraces();
            } else if ((i >>> 8) == GZIP_HEADER_MAGIC) {
                // Possible gziped file, try decompress it and get the stack trace.
                in.close();
                String deCompressedFile = "heapdump" + System.currentTimeMillis() + ".hprof";
                File out = new File(deCompressedFile);
                // Decompress to get dump file.
                try (FileInputStream heapFis = new FileInputStream(heapFile);
                     GZIPInputStream gis = new GZIPInputStream(heapFis);
                     FileOutputStream fos = new FileOutputStream(out)) {
                    byte[] buffer = new byte[1024 * 1024];
                    int len = 0;
                    while ((len = gis.read(buffer)) > 0) {
                        fos.write(buffer, 0, len);
                    }
                } catch (Exception e) {
                    out.delete();
                    throw new IOException("Cannot decompress the compressed hprof file", e);
                }
                // Check dump data header and print stack trace.
                try (FileInputStream outFis = new FileInputStream(out);
                     BufferedInputStream outBis = new BufferedInputStream(outFis);
                     PositionDataInputStream pdin = new PositionDataInputStream(outBis)) {
                    i = pdin.readInt();
                    if (i == HprofReader.MAGIC_NUMBER) {
                        HprofReader r
                            = new HprofReader(deCompressedFile, pdin, dumpNumber,
                                              true, debugLevel);
                        r.read();
                        return r.printStackTraces();
                    } else {
                        throw new IOException("Unrecognized magic number found in decompressed data: " + i);
                    }
                } finally {
                    out.delete();
                }
            } else {
                throw new IOException("Unrecognized magic number: " + i);
            }
        }
    }
}