1 /*
2 * Copyright (c) 2015, 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 import java.io.ByteArrayInputStream;
25 import java.io.FilterInputStream;
26 import java.io.IOException;
27 import java.io.InputStream;
28 import java.util.Arrays;
29 import java.util.Random;
30 import jdk.test.lib.RandomFactory;
31
32 /*
33 * @test
34 * @bug 8080835 8139206
35 * @library /test/lib
36 * @build jdk.test.lib.RandomFactory
37 * @run main ReadNBytes
38 * @summary Basic test for InputStream.readNBytes
39 * @key randomness
40 */
41
42 public class ReadNBytes {
43
44 private static Random generator = RandomFactory.getRandom();
45
46 public static void main(String[] args) throws IOException {
47 test(new byte[]{1, 2, 3});
48 test(createRandomBytes(1024));
49 for (int shift : new int[] {13, 15, 17}) {
50 for (int offset : new int[] {-1, 0, 1}) {
51 test(createRandomBytes((1 << shift) + offset));
52 }
53 }
54
55 test(-1);
56 test(0);
57 for (int shift : new int[] {13, 15, 17}) {
58 for (int offset : new int[] {-1, 0, 1}) {
59 test((1 << shift) + offset);
60 }
61 }
62 }
63
64 static void test(byte[] inputBytes) throws IOException {
65 int length = inputBytes.length;
66 WrapperInputStream in = new WrapperInputStream(new ByteArrayInputStream(inputBytes));
67 byte[] readBytes = new byte[(length / 2) + 1];
68 int nread = in.readNBytes(readBytes, 0, readBytes.length);
69
70 int x;
71 byte[] tmp;
72 check(nread == readBytes.length,
73 "Expected number of bytes read: " + readBytes.length + ", got: " + nread);
74 check(Arrays.equals((tmp = Arrays.copyOf(inputBytes, nread)), readBytes),
75 "Expected[" + tmp + "], got:[" + readBytes + "]");
76 check(!in.isClosed(), "Stream unexpectedly closed");
77
78 // Read again
79 nread = in.readNBytes(readBytes, 0, readBytes.length);
80
81 check(nread == length - readBytes.length,
82 "Expected number of bytes read: " + (length - readBytes.length) + ", got: " + nread);
83 check(Arrays.equals((tmp = Arrays.copyOfRange(inputBytes, readBytes.length, length)),
84 Arrays.copyOf(readBytes, nread)),
85 "Expected[" + tmp + "], got:[" + readBytes + "]");
86 // Expect end of stream
87 check((x = in.read()) == -1,
88 "Expected end of stream from read(), got " + x);
89 check((x = in.read(tmp)) == -1,
90 "Expected end of stream from read(byte[]), got " + x);
91 check((x = in.read(tmp, 0, tmp.length)) == -1,
92 "Expected end of stream from read(byte[], int, int), got " + x);
93 check((x = in.readNBytes(tmp, 0, tmp.length)) == 0,
94 "Expected end of stream, 0, from readNBytes(byte[], int, int), got " + x);
95 check(!in.isClosed(), "Stream unexpectedly closed");
96 }
97
98 static void test(int max) throws IOException {
99 byte[] inputBytes = max <= 0 ? new byte[0] : createRandomBytes(max);
100 WrapperInputStream in =
101 new WrapperInputStream(new ByteArrayInputStream(inputBytes));
102
103 if (max < 0) {
104 try {
105 in.readNBytes(max);
106 check(false, "Expected IllegalArgumentException not thrown");
107 } catch (IllegalArgumentException iae) {
108 return;
109 }
110 } else if (max == 0) {
111 int x;
112 check((x = in.readNBytes(max).length) == 0,
113 "Expected zero bytes, got " + x);
114 return;
115 }
116
117 int off = Math.toIntExact(in.skip(generator.nextInt(max/2)));
118 int len = generator.nextInt(max - 1 - off);
119 byte[] readBytes = in.readNBytes(len);
120 check(readBytes.length == len,
121 "Expected " + len + " bytes, got " + readBytes.length);
122 check(Arrays.equals(inputBytes, off, off + len, readBytes, 0, len),
123 "Expected[" + Arrays.copyOfRange(inputBytes, off, off + len) +
124 "], got:[" + readBytes + "]");
125
126 int remaining = max - (off + len);
127 readBytes = in.readNBytes(remaining);
128 check(readBytes.length == remaining,
129 "Expected " + remaining + "bytes, got " + readBytes.length);
130 check(Arrays.equals(inputBytes, off + len, max,
131 readBytes, 0, remaining),
132 "Expected[" + Arrays.copyOfRange(inputBytes, off + len, max) +
133 "], got:[" + readBytes + "]");
134
135 check(!in.isClosed(), "Stream unexpectedly closed");
136 }
137
138 static byte[] createRandomBytes(int size) {
139 byte[] bytes = new byte[size];
140 generator.nextBytes(bytes);
141 return bytes;
142 }
143
144 static void check(boolean cond, Object ... failedArgs) {
145 if (cond)
146 return;
147 StringBuilder sb = new StringBuilder();
148 for (Object o : failedArgs)
149 sb.append(o);
150 throw new RuntimeException(sb.toString());
151 }
152
153
154 static class WrapperInputStream extends FilterInputStream {
155 private boolean closed;
156 WrapperInputStream(InputStream in) { super(in); }
157 @Override public void close() throws IOException { closed = true; in.close(); }
158 boolean isClosed() { return closed; }
159 }
160 }