1 /*

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

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

     4  *

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

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

     7  * published by the Free Software Foundation.

     8  *

     9  * This code is distributed in the hope that it will be useful, but WITHOUT

    10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

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

    13  * accompanied this code).

    14  *

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

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

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

    18  *

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

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

    21  * questions.

    22  */

    23 

    24 /*

    25  * Test reading resource content.

    26  * @test ImageReadTest

    27  * @summary Unit test for JVM_ImageRead() method

    28  * @library /testlibrary /../../test/lib

    29  * @build LocationConstants ImageReadTest

    30  * @run main ClassFileInstaller sun.hotspot.WhiteBox

    31  *                              sun.hotspot.WhiteBox$WhiteBoxPermission

    32  * @run main/othervm -Xbootclasspath/a:. -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -XX:+MemoryMapImage ImageReadTest +

    33  * @run main/othervm -Xbootclasspath/a:. -XX:+UnlockDiagnosticVMOptions -XX:+WhiteBoxAPI -XX:-MemoryMapImage ImageReadTest -

    34  */

    35 

    36 import java.io.File;

    37 import java.nio.ByteBuffer;

    38 import sun.hotspot.WhiteBox;

    39 import static jdk.test.lib.Asserts.*;

    40 

    41 public class ImageReadTest implements LocationConstants {

    42 

    43     public static final WhiteBox wb = WhiteBox.getWhiteBox();

    44 

    45     public static void main(String... args) throws Exception {

    46         String javaHome = System.getProperty("java.home");

    47         String imageFile = javaHome + File.separator + "lib"

    48                 + File.separator + "modules" + File.separator

    49                 + "bootmodules.jimage";

    50 

    51         if (!(new File(imageFile)).exists()) {

    52             System.out.printf("Test skipped.");

    53             return;

    54         }

    55 

    56         boolean isMMap = args[0].equals("+");

    57 

    58         long id = wb.imageOpenImage(imageFile, isMMap);

    59 

    60         final String mm = isMMap ? "-XX:+MemoryMapImage" : "-XX:-MemoryMapImage";

    61         final int magic = 0xCAFEBABE;

    62 

    63         String className = "/java.base/java/lang/String.class";

    64         long[] offsetArr = wb.imageFindAttributes(id, className.getBytes());

    65         long offset = offsetArr[LOCATION_ATTRIBUTE_OFFSET];

    66         long size = offsetArr[LOCATION_ATTRIBUTE_UNCOMPRESSED];

    67 

    68         // positive: read

    69         ByteBuffer buf = ByteBuffer.allocateDirect((int) size);

    70         assertTrue(wb.imageRead(id, offset, buf, size), "Failed. Read operation returned false, should be true");

    71         int m = buf.getInt();

    72         assertTrue(m == magic, "Failed. Read operation returned true but wrong magic = " + magic);

    73 

    74         // positive: mmap

    75         if (isMMap) {

    76             long dataAddr = wb.imageGetDataAddress(id);

    77             assertFalse(dataAddr == 0L, "Failed. Did not obtain data address on mmapped test");

    78             int data = wb.imageGetIntAtAddress(dataAddr, (int) offset, true);

    79             assertTrue(data == magic, "Failed. MMap operation returned true but wrong magic = " + data);

    80         }

    81 

    82         // negative: wrong offset

    83         boolean success = wb.imageRead(id, -100, buf, size);

    84         assertFalse(success, "Failed. Read operation (wrong offset): returned true");

    85 

    86         // negative: too big offset

    87         long filesize = new File(imageFile).length();

    88         success = wb.imageRead(id, filesize + 1, buf, size);

    89         assertFalse(success, "Failed. Read operation (offset > file size) returned true");

    90 

    91         // negative: negative size

    92         success = wb.imageRead(id, offset, buf, -100);

    93         assertFalse(success, "Failed. Read operation (negative size) returned true");

    94 

    95         wb.imageCloseImage(id);

    96     }

    97 }