diff -r 643978a35f6e -r 94691d8e746f test/jdk/sun/security/util/ManifestDigester/LineBreaks.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/test/jdk/sun/security/util/ManifestDigester/LineBreaks.java Thu Jul 18 08:53:06 2019 +0800 @@ -0,0 +1,142 @@ +/* + * Copyright (c) 2019, 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.ByteArrayInputStream; +import java.io.ByteArrayOutputStream; +import java.io.IOException; +import java.util.List; +import java.util.ArrayList; +import java.util.function.Function; +import java.util.jar.Attributes; +import java.util.jar.Manifest; +import sun.security.util.ManifestDigester; + +import org.testng.annotations.DataProvider; +import org.testng.annotations.Test; + +import static java.nio.charset.StandardCharsets.UTF_8; +import static org.testng.Assert.*; + +/** + * @test + * @bug 8217375 + * @modules java.base/sun.security.util + * @compile ../../tools/jarsigner/Utils.java + * @run testng LineBreaks + * @summary Verify {@code ManifestDigester} reads different line breaks well. + * The specifications state: + *
newline: CR LF | LF | CR (not followed by LF)
. + * This test does not verify that the digests are correct. + */ +public class LineBreaks { + + static final String KEY = "Key"; + static final String VALUE = "Value"; + static final String SECTION = "Section"; + static final String FOO = "Foo"; + static final String BAR = "Bar"; + + static final String EXCEED_LINE_WIDTH_LIMIT = "x".repeat(71); + + String breakAndContinue(String str, String lineBreak) { + // assert no multi-byte UTF-8 encoded characters in this test + assertEquals(str.getBytes(UTF_8).length, str.length()); + + int p = 1; + while (p + 71 < str.length()) { + p += 71; + str = str.substring(0, p) + lineBreak + " " + str.substring(p); + p += lineBreak.length() + 1; + } + return str; + } + + byte[] createTestManifest(String lineBreak, boolean onlyMainAttrs, + String excess) throws IOException { + System.out.println("lineBreak = " + + Utils.escapeStringWithNumbers(lineBreak)); + System.out.println("onlyMainAttrs = " + onlyMainAttrs); + String mf = ""; + mf += breakAndContinue( + KEY + ": " + VALUE + excess, lineBreak) + lineBreak; + mf += lineBreak; + if (!onlyMainAttrs) { + mf += breakAndContinue( + "Name: " + SECTION + excess, lineBreak) + lineBreak; + mf += breakAndContinue( + FOO + ": " + BAR + excess, lineBreak) + lineBreak; + mf += lineBreak; + } + byte[] mfBytes = mf.getBytes(UTF_8); + Utils.echoManifest(mfBytes, "binary manifest"); + return mfBytes; + } + + @DataProvider(name = "parameters") + public static Object[][] parameters() { + List tests = new ArrayList<>(); + for (String lineBreak : new String[] { "\n", "\r", "\r\n" }) { + for (boolean onlyMainAttrs : new boolean[] { false, true }) { + for (int numLbs = 0; numLbs < 3; numLbs++) { + tests.add(new Object[]{ lineBreak, onlyMainAttrs, numLbs }); + } + } + } + return tests.toArray(new Object[tests.size()][]); + } + + @Test(dataProvider = "parameters") + public void test(String lineBreak, boolean onlyMainAttrs, int numLbs) + throws IOException { + String excess = EXCEED_LINE_WIDTH_LIMIT.repeat(numLbs); + byte[] mfBytes = createTestManifest(lineBreak, onlyMainAttrs, excess); + + // self-test: make sure the manifest is valid and represents the + // values as expected before attempting to digest it + Manifest mf = new Manifest(new ByteArrayInputStream(mfBytes)); + assertEquals(mf.getMainAttributes().getValue(KEY), VALUE + excess); + Attributes section = mf.getAttributes(SECTION + excess); + if (onlyMainAttrs) { + assertNull(section); + } else { + assertEquals(section.getValue(FOO), BAR + excess); + } + + // verify that ManifestDigester has actually found the individual + // section if and only if it was present thereby also implying based + // on ManifestDigester implementation that the main attributes were + // found before + ManifestDigester md = new ManifestDigester(mfBytes); + assertTrue((md.get(SECTION + excess, false) != null) != onlyMainAttrs); + } + + static List stringToIntList(String string) { + byte[] bytes = string.getBytes(UTF_8); + List list = new ArrayList<>(); + for (int i = 0; i < bytes.length; i++) { + list.add((int) bytes[i]); + } + return list; + } + +}