1 /*
2 * Copyright (c) 2013, 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
26 import java.io.ByteArrayOutputStream;
27 import java.io.IOException;
28 import java.io.OutputStream;
29 import java.nio.ByteBuffer;
30
31 import java.util.Base64;
32
33 import static java.nio.charset.StandardCharsets.US_ASCII;
34
35 /**
36 * @test
37 * @bug 8007799
38 * @summary test Encoder with linemax == 0, line separator should not appear in encoded data
39 */
40
41 public class Base64GetEncoderTest {
42
43 public static void main(String args[]) throws Throwable {
44 final Base64.Encoder encoder = Base64.getMimeEncoder(0, "$$$".getBytes(US_ASCII));
45
46 testEncodeToString(encoder);
47
48 testWrapEncode1(encoder);
49
50 testEncodeToStringWithLongInputData(encoder);
51
52 testWrapEncode2(encoder);
53
54 testEncodeWithByteBuffer(encoder);
55
56 }
57
58 private static void testEncodeWithByteBuffer(final Base64.Encoder encoder) {
59 System.err.println("\n\nEncoder.encode with ByteBuffer test ");
60 final byte[] secondTestBuffer =
61 "api/java_util/Base64/index.html#GetEncoderMimeCustom[noLineSeparatorInEncodedString]"
62 .getBytes(US_ASCII);
63 String base64EncodedString;
64 ByteBuffer srcData = ByteBuffer.wrap(secondTestBuffer);
65 ByteBuffer dstData = ByteBuffer.allocate(secondTestBuffer.length * 2);
66
67 encoder.encode(srcData, dstData, 0);
68 dstData.flip();
69 if (dstData.hasArray()) {
70 System.err.println("\nByteBuffer test dstData is Base64 encoding = "
71 + new String(dstData.array(), US_ASCII) + "\n");
72 }
73
74 base64EncodedString = new String(dstData.array(), US_ASCII);
75 if (base64EncodedString.contains("$$$")) {
76 throw new RuntimeException("Base64 encoding contains line separator after Encoder.encode ByteBuffer ... \n");
77 }
78 }
79
80 private static void testWrapEncode2(final Base64.Encoder encoder)
81 throws IOException {
82 System.err.println("\nEncoder.wrap test II ");
83 final byte[] secondTestBuffer =
84 "api/java_util/Base64/index.html#GetEncoderMimeCustom[noLineSeparatorInEncodedString]"
85 .getBytes(US_ASCII);
86 String base64EncodedString;
87 ByteArrayOutputStream secondEncodingStream = new ByteArrayOutputStream();
88 OutputStream base64EncodingStream = encoder.wrap(secondEncodingStream);
89 base64EncodingStream.write(secondTestBuffer);
90 base64EncodingStream.close();
91
92 final byte[] encodedByteArray = secondEncodingStream.toByteArray();
93
94 System.err.print("result = " + new String(encodedByteArray, US_ASCII)
95 + " after wrap Base64 encoding of string");
96
97 base64EncodedString = new String(encodedByteArray, US_ASCII);
98
99 if (base64EncodedString.contains("$$$")) {
100 throw new RuntimeException(
101 "Base64 encoding contains line separator after wrap 2 invoked ... \n");
102 }
103 }
104
105 private static void testEncodeToStringWithLongInputData(
106 final Base64.Encoder encoder) {
107 System.err.println("\n\nEncoder.encodeToStringWithLongInputData test ");
108
109 final byte[] secondTestBuffer =
110 "api/java_util/Base64/index.html#GetEncoderMimeCustom[noLineSeparatorInEncodedString]"
111 .getBytes(US_ASCII);
112 String base64EncodedString;
113 base64EncodedString = encoder.encodeToString(secondTestBuffer);
114
115 System.err.println("Second Base64 encoded string is "
116 + base64EncodedString);
117
118 if (base64EncodedString.contains("$$$")) {
119 throw new RuntimeException(
120 "Base64 encoding contains line separator after encodeToString invoked ... \n");
121 }
122 }
123
124 private static void testWrapEncode1(final Base64.Encoder encoder)
125 throws IOException {
126 System.err.println("\nEncoder.wrap test I ");
127
128 final byte[] bytesIn = "fo".getBytes(US_ASCII);
129 String base64EncodedString;
130 ByteArrayOutputStream encodingStream = new ByteArrayOutputStream();
131 OutputStream encoding = encoder.wrap(encodingStream);
132 encoding.write(bytesIn);
133 encoding.close();
134
135 final byte[] encodedBytes = encodingStream.toByteArray();
136
137 System.err.print("result = " + new String(encodedBytes, US_ASCII)
138 + " after the Base64 encoding \n");
139
140 base64EncodedString = new String(encodedBytes, US_ASCII);
141
142 if (base64EncodedString.contains("$$$")) {
143 throw new RuntimeException(
144 "Base64 encoding contains line separator after wrap I test ... \n");
145 }
146 }
147
148 private static void testEncodeToString(final Base64.Encoder encoder) {
149 final byte[] bytesIn = "fo".getBytes(US_ASCII);
150
151 System.err.println("\nEncoder.encodeToString test ");
152
153 String base64EncodedString = encoder.encodeToString(bytesIn);
154
155 System.err.println("Base64 encoded string is " + base64EncodedString);
156
157 if (base64EncodedString.contains("$$$")) {
158 throw new RuntimeException("Base64 encoding contains line separator after Encoder.encodeToString invoked ... \n");
159 }
160 }
161 }