test/jdk/tools/pack200/PackChecksum.java

/*
 * Copyright (c) 2014, 2016, 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.
 *
 * 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.
 */
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.RandomAccessFile;
import java.util.ArrayList;
import java.util.List;
import java.util.jar.JarEntry;
import java.util.jar.JarOutputStream;

/*
 * @test
 * @bug 8000650 8150469
 * @summary unpack200.exe should check gzip crc
 * @compile -XDignore.symbol.file Utils.java PackChecksum.java
 * @run main PackChecksum
 * @author kizune
 */
public class PackChecksum {
    final int TRAILER_LEN = 8;
    final List<String> cmdsList = new ArrayList<>();
    static enum Case {
        CRC32,
        ISIZE,
        BOTH;

    };
    public static void main(String... args) throws Exception {
        new PackChecksum().run();
    }
    void run() throws Exception {
        testBrokenTrailer(Case.CRC32); // negative
        testBrokenTrailer(Case.ISIZE); // negative
        testBrokenTrailer(Case.BOTH);  // negative
        testMultipleSegments();
    }

    void testMultipleSegments() throws Exception {
        File inputJar = new File("input.jar");
        Utils.copyFile(Utils.getGoldenJar(), inputJar);
        cmdsList.clear();

        File testPack = new File("out.jar.pack.gz");

        cmdsList.clear();
        cmdsList.add(Utils.getPack200Cmd());
        // force multiple segments
        cmdsList.add("--segment-limit=100");
        cmdsList.add(testPack.getName());
        cmdsList.add(inputJar.getName());
        Utils.runExec(cmdsList);

        File destFile = new File("dst.jar");
        cmdsList.clear();
        cmdsList.add(Utils.getUnpack200Cmd());
        cmdsList.add(testPack.getName());
        cmdsList.add(destFile.getName());
        try {
            Utils.runExec(cmdsList);
            if (!destFile.exists()) {
                throw new Exception("file not created: " + destFile);
            }
        } finally {
            if (inputJar.exists())
                inputJar.delete();
            if (testPack.exists())
                testPack.delete();
            if (destFile.exists())
                destFile.delete();
        }
    }

    void testBrokenTrailer(Case type) throws Exception {
        System.out.println("Testing: case " + type);
        // Create a fresh .jar file
        File testFile = new File("src_tools.jar");
        File testPack = new File("src_tools.pack.gz");
        generateJar(testFile);

        cmdsList.clear();
        // Create .pack file
        cmdsList.add(Utils.getPack200Cmd());
        cmdsList.add(testPack.getName());
        cmdsList.add(testFile.getName());
        Utils.runExec(cmdsList);

        // mutate the checksum of the packed file
        RandomAccessFile raf = new RandomAccessFile(testPack, "rw");

        switch (type) {
            case CRC32:
                raf.seek(raf.length() - TRAILER_LEN);
                raf.writeInt(0x0dea0a0d);
                break;
            case ISIZE:
                raf.seek(raf.length() - (TRAILER_LEN/2));
                raf.writeInt(0x0b0e0e0f);
                break;
            default:
                raf.seek(raf.length() - (TRAILER_LEN));
                raf.writeLong(0x0dea0a0d0b0e0e0fL);
                break;
        }

        raf.close();

        File dstFile = new File("dst_tools.jar");
        if (dstFile.exists()) {
            dstFile.delete();
        }
        cmdsList.clear();
        cmdsList.add(Utils.getUnpack200Cmd());
        cmdsList.add(testPack.getName());
        cmdsList.add(dstFile.getName());

        boolean processFailed = false;
        try {
            Utils.runExec(cmdsList);
        } catch (RuntimeException re) {
            // unpack200 should exit with non-zero exit code
            processFailed = true;
        } finally {
            // tidy up
            if (testFile.exists())
                testFile.delete();

            if (testPack.exists())
                testPack.delete();

            if (!processFailed) {
                throw new Exception("case " + type +
                        ": file with incorrect CRC, unpacked without the error.");
            }
            if (dstFile.exists()) {
                dstFile.delete();
                throw new Exception("case " + type +
                        ":  file exists: " + dstFile);
            }
        }
    }

    void generateJar(File result) throws IOException {
        if (result.exists()) {
            result.delete();
        }

        try (JarOutputStream output = new JarOutputStream(new FileOutputStream(result)); ) {
            for (int i = 0 ; i < 100 ; i++) {
                JarEntry e = new JarEntry("F-" + i + ".txt");
                output.putNextEntry(e);
            }
            output.flush();
            output.close();
        }
    }

}