1 /* 2 * Copyright (c) 1997, 2018, 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 26 * @bug 4017158 8180410 27 * @library /test/lib 28 * @build jdk.test.lib.RandomFactory 29 * @run testng Write 30 * @summary Check for correct implementation of ByteArrayInputStream.write 31 * @key randomness 32 */ 33 34 import java.io.ByteArrayOutputStream; 35 import java.util.Arrays; 36 import java.util.Random; 37 import jdk.test.lib.RandomFactory; 38 import org.testng.annotations.Test; 39 import static org.testng.Assert.*; 40 41 public class Write { 42 private static void doBoundsTest(byte[] b, int off, int len, 43 ByteArrayOutputStream baos) 44 throws Exception { 45 if (b != null) { 46 System.out.println("ByteArrayOutStream.write: b.length = " + 47 b.length + " off = " + off + " len = " + len); 48 } else{ 49 System.out.println("ByteArrayOutStream.write: b is null off = " + 50 off + " len = " + len); 51 } 52 53 try { 54 baos.write(b, off, len); 55 } catch (IndexOutOfBoundsException e) { 56 System.out.println("IndexOutOfBoundsException is thrown: OKAY"); 57 } catch (NullPointerException e) { 58 System.out.println("NullPointerException is thrown: OKAY"); 59 } catch (Throwable e){ 60 throw new RuntimeException("Unexpected Exception is thrown", e); 61 } 62 63 if (b != null) { 64 System.out.println("ByteArrayOutStream.writeBytes: b.length = " + 65 b.length); 66 } else{ 67 System.out.println("ByteArrayOutStream.writeBytes: b is null"); 68 } 69 70 try { 71 baos.writeBytes(b); 72 } catch (NullPointerException e) { 73 System.out.println("NullPointerException is thrown: OKAY"); 74 } catch (Throwable e){ 75 throw new RuntimeException("Unexpected Exception is thrown", e); 76 } 77 } 78 79 @Test 80 public static void boundsTest() throws Exception { 81 byte array1[] = {1 , 2 , 3 , 4 , 5}; // Simple array 82 83 //Create new ByteArrayOutputStream object 84 ByteArrayOutputStream y1 = new ByteArrayOutputStream(5); 85 86 doBoundsTest(array1, 0, Integer.MAX_VALUE , y1); 87 doBoundsTest(array1, 0, array1.length+100, y1); 88 doBoundsTest(array1, -1, 2, y1); 89 doBoundsTest(array1, 0, -1, y1); 90 doBoundsTest(null, 0, 2, y1); 91 } 92 93 @Test 94 public static void writeTest() throws Exception { 95 ByteArrayOutputStream baos = new ByteArrayOutputStream(); 96 Random rnd = RandomFactory.getRandom(); 97 final int size = 17 + rnd.nextInt(128); 98 99 byte[] b = new byte[size]; 100 rnd.nextBytes(b); 101 102 int off1 = rnd.nextInt(size / 4) + 1; 103 int len1 = Math.min(rnd.nextInt(size / 4) + 1, size - off1); 104 int off2 = rnd.nextInt(size / 2) + 1; 105 int len2 = Math.min(rnd.nextInt(size / 2) + 1, size - off2); 106 107 System.out.format("size: %d, off1: %d, len1: %d, off2: %d, len2: %d%n", 108 size, off1, len1, off2, len2); 109 110 baos.write(b, off1, len1); 111 byte[] b1 = baos.toByteArray(); 112 assertEquals(b1.length, len1, "Array length test 1 failed."); 113 assertEquals(b1, Arrays.copyOfRange(b, off1, off1 + len1), 114 "Array equality test 1 failed."); 115 116 baos.write(b, off2, len2); 117 byte[] b2 = baos.toByteArray(); 118 assertEquals(b2.length, len1 + len2, "Array length test 2 failed."); 119 assertEquals(Arrays.copyOfRange(b2, 0, len1), 120 Arrays.copyOfRange(b, off1, off1 + len1), 121 "Array equality test 2A failed."); 122 assertEquals(Arrays.copyOfRange(b2, len1, len1 + len2), 123 Arrays.copyOfRange(b, off2, off2 + len2), 124 "Array equality test 2B failed."); 125 126 baos.writeBytes(b); 127 byte[] b3 = baos.toByteArray(); 128 int len3 = len1 + len2 + b.length; 129 if (b3.length != len1 + len2 + b.length) { 130 throw new RuntimeException("Array length test 3 failed."); 131 } 132 assertEquals(b3.length, len3, "Array length test 3 failed."); 133 assertEquals(Arrays.copyOfRange(b3, 0, len1), 134 Arrays.copyOfRange(b, off1, off1 + len1), 135 "Array equality test 3A failed."); 136 assertEquals(Arrays.copyOfRange(b3, len1, len1 + len2), 137 Arrays.copyOfRange(b, off2, off2 + len2), 138 "Array equality test 3B failed."); 139 assertEquals(Arrays.copyOfRange(b3, len1 + len2, len3), b, 140 "Array equality test 3C failed."); 141 } 142 }