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 import java.io.ByteArrayInputStream;
  25 import java.io.ByteArrayOutputStream;
  26 import java.io.DataOutputStream;
  27 import java.io.IOException;
  28 import java.io.InputStream;
  29 import java.nio.ByteOrder;
  30 import java.nio.file.Path;
  31 import java.util.ArrayList;
  32 import java.util.Collections;
  33 import java.util.HashSet;
  34 import java.util.List;
  35 import java.util.Set;
  36 import java.util.stream.Stream;
  37 import jdk.tools.jlink.internal.Archive;
  38 import jdk.tools.jlink.internal.ImageFileCreator;
  39 import jdk.tools.jlink.internal.ImagePluginStack;
  40 import jdk.tools.jlink.plugin.ExecutableImage;
  41 import jdk.tools.jlink.builder.ImageBuilder;
  42 import jdk.tools.jlink.plugin.Pool;
  43 
  44 
  45 /*
  46  * @test
  47  * @summary ImageFileCreator class test
  48  * @author Jean-Francois Denise
  49  * @modules jdk.jlink/jdk.tools.jlink.internal
  50  *          java.base/jdk.internal.jimage
  51  * @run main/othervm -verbose:gc -Xmx1g ImageFileCreatorTest
  52  */
  53 public class ImageFileCreatorTest {
  54 
  55     private static class TestArchive implements Archive {
  56 
  57         private final String name;
  58         private final List<Entry> entries = new ArrayList<>();
  59 
  60         private TestArchive(String name, List<String> entries) {
  61             this.name = name;
  62             for (String p : entries) {
  63                 this.entries.add(new TestEntry(p, p));
  64             }
  65         }
  66 
  67         @Override
  68         public String moduleName() {
  69             return name;
  70         }
  71 
  72         @Override
  73         public Stream<Entry> entries() {
  74             return entries.stream();
  75         }
  76 
  77         @Override
  78         public Path getPath() {
  79             return null;
  80         }
  81 
  82         @Override
  83         public void open() throws IOException {
  84         }
  85 
  86         @Override
  87         public void close() throws IOException {
  88         }
  89 
  90         private class TestEntry extends Entry {
  91 
  92             TestEntry(String path, String name) {
  93                 super(TestArchive.this, path, name, Entry.EntryType.CLASS_OR_RESOURCE);
  94             }
  95 
  96             @Override
  97             public long size() {
  98                 return 0;
  99             }
 100 
 101             @Override
 102             public InputStream stream() throws IOException {
 103                 return new ByteArrayInputStream(new byte[0]);
 104             }
 105         }
 106     }
 107 
 108     public static void main(String[] args) throws Exception {
 109 
 110         {
 111             List<String> entries = new ArrayList<>();
 112             entries.add("classes/class");
 113             test(entries);
 114         }
 115 
 116         {
 117             // Add an entry that is a directory, that is wrong
 118             List<String> entries = new ArrayList<>();
 119             entries.add("classes");
 120             entries.add("classes/class");
 121             test(entries);
 122         }
 123 
 124         {
 125             // Add an entry that is wrongly prefixed by /
 126             // /bad//classes/class is the resource added
 127             // /bad/classes/class is the metadata node built.
 128             List<String> entries = new ArrayList<>();
 129             entries.add("/classes/class");
 130             test(entries);
 131         }
 132 
 133         {
 134             // Trailing '/' is wrong
 135             List<String> entries = new ArrayList<>();
 136             entries.add("classes/class/");
 137             test(entries);
 138         }
 139 
 140         {
 141             // Too much '/' characters
 142             List<String> entries = new ArrayList<>();
 143             entries.add("classes//class/");
 144             test(entries);
 145         }
 146 
 147         {
 148             // Too much '/' characters
 149             List<String> entries = new ArrayList<>();
 150             entries.add("classes/////class/");
 151             test(entries);
 152         }
 153 
 154         {
 155             // Single '/' character
 156             List<String> entries = new ArrayList<>();
 157             entries.add("/");
 158             test(entries);
 159         }
 160 
 161         {
 162             // 2 '/' characters
 163             List<String> entries = new ArrayList<>();
 164             entries.add("//");
 165             test(entries);
 166         }
 167 
 168         {
 169             // 3 '/' characters
 170             List<String> entries = new ArrayList<>();
 171             entries.add("///");
 172             test(entries);
 173         }
 174 
 175         {
 176             // no character
 177             List<String> entries = new ArrayList<>();
 178             entries.add("");
 179             test(entries);
 180         }
 181 
 182         {
 183             // all together
 184             List<String> entries = new ArrayList<>();
 185             entries.add("");
 186             entries.add("///");
 187             entries.add("//");
 188             entries.add("/");
 189             entries.add("classes/////class/");
 190             entries.add("classes//class/");
 191             entries.add("classes/class/");
 192             entries.add("/classes/class");
 193             entries.add("classes");
 194             entries.add("classes/class");
 195             test(entries);
 196         }
 197 
 198     }
 199 
 200     private static void test(List<String> entries) throws Exception {
 201         TestArchive arch = new TestArchive("bad", entries);
 202         Set<Archive> archives = new HashSet<>();
 203         archives.add(arch);
 204         ImageBuilder noopBuilder = new ImageBuilder() {
 205 
 206             @Override
 207             public DataOutputStream getJImageOutputStream() {
 208                 return new DataOutputStream(new ByteArrayOutputStream());
 209             }
 210 
 211             @Override
 212             public ExecutableImage getExecutableImage() {
 213                 return null;
 214             }
 215 
 216             @Override
 217             public void storeFiles(Pool content, String bom) {
 218 
 219             }
 220 
 221         };
 222 
 223         ImagePluginStack stack = new ImagePluginStack(noopBuilder, Collections.emptyList(),
 224                 null, Collections.emptyList(), "");
 225 
 226         ImageFileCreator.create(archives, ByteOrder.nativeOrder(), stack);
 227     }
 228 }