undefect. CWE-407 — 63 sites patched across 27 ecosystems
Authors: russell@unturf.com · brackishbert@gmail.com · foxhop.net · TimeHexOn.com Patches, unit tests, benchmarks, whitepaper, and outreach briefs. Public domain — no copyright claimed. Use freely.
This commit is contained in:
commit
0a580b313d
70422 changed files with 17213626 additions and 0 deletions
181
test/jdk/java/util/Base64/TestBase64Golden.java
Normal file
181
test/jdk/java/util/Base64/TestBase64Golden.java
Normal file
|
|
@ -0,0 +1,181 @@
|
|||
/*
|
||||
* Copyright (c) 2012, 2013, 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.
|
||||
*/
|
||||
|
||||
/*
|
||||
* @test
|
||||
* @bug 4235519
|
||||
* @author Eric Wang <yiming.wang@oracle.com>
|
||||
* @summary tests java.util.Base64
|
||||
*/
|
||||
|
||||
import java.io.BufferedReader;
|
||||
import java.io.FileReader;
|
||||
import java.nio.ByteBuffer;
|
||||
import java.nio.charset.Charset;
|
||||
import java.nio.charset.StandardCharsets;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
import java.util.Base64;
|
||||
import java.util.Base64.Decoder;
|
||||
import java.util.Base64.Encoder;
|
||||
import java.util.Objects;
|
||||
import java.util.Random;
|
||||
|
||||
public class TestBase64Golden {
|
||||
|
||||
public static void main(String[] args) throws Exception {
|
||||
test0(Base64Type.BASIC, Base64.getEncoder(), Base64.getDecoder(),
|
||||
"plain.txt", "baseEncode.txt");
|
||||
|
||||
test0(Base64Type.URLSAFE, Base64.getUrlEncoder(), Base64.getUrlDecoder(),
|
||||
"plain.txt", "urlEncode.txt");
|
||||
|
||||
test0(Base64Type.MIME, Base64.getMimeEncoder(), Base64.getMimeDecoder(),
|
||||
"plain.txt", "mimeEncode.txt");
|
||||
test1();
|
||||
}
|
||||
|
||||
public static void test0(Base64Type type, Encoder encoder, Decoder decoder,
|
||||
String srcFile, String encodedFile) throws Exception {
|
||||
|
||||
String[] srcLns = Files.readAllLines(Paths.get(SRCDIR, srcFile), DEF_CHARSET)
|
||||
.toArray(new String[0]);
|
||||
String[] encodedLns = Files.readAllLines(Paths.get(SRCDIR, encodedFile),
|
||||
DEF_CHARSET)
|
||||
.toArray(new String[0]);
|
||||
int lns = 0;
|
||||
for (String srcStr : srcLns) {
|
||||
String encodedStr = null;
|
||||
if (type != Base64Type.MIME) {
|
||||
encodedStr = encodedLns[lns++];
|
||||
} else {
|
||||
while (lns < encodedLns.length) {
|
||||
String s = encodedLns[lns++];
|
||||
if (s.length() == 0)
|
||||
break;
|
||||
if (encodedStr != null) {
|
||||
encodedStr += DEFAULT_CRLF + s;
|
||||
} else {
|
||||
encodedStr = s;
|
||||
}
|
||||
}
|
||||
if (encodedStr == null && srcStr.length() == 0) {
|
||||
encodedStr = "";
|
||||
}
|
||||
}
|
||||
System.out.printf("%n src[%d]: %s%n", srcStr.length(), srcStr);
|
||||
System.out.printf("encoded[%d]: %s%n", encodedStr.length(), encodedStr);
|
||||
|
||||
byte[] srcArr = srcStr.getBytes(DEF_CHARSET);
|
||||
byte[] encodedArr = encodedStr.getBytes(DEF_CHARSET);
|
||||
|
||||
ByteBuffer srcBuf = ByteBuffer.wrap(srcArr);
|
||||
ByteBuffer encodedBuf = ByteBuffer.wrap(encodedArr);
|
||||
byte[] resArr = new byte[encodedArr.length];
|
||||
|
||||
// test int encode(byte[], byte[])
|
||||
int len = encoder.encode(srcArr, resArr);
|
||||
assertEqual(len, encodedArr.length);
|
||||
assertEqual(resArr, encodedArr);
|
||||
|
||||
// test byte[] encode(byte[])
|
||||
resArr = encoder.encode(srcArr);
|
||||
assertEqual(resArr, encodedArr);
|
||||
|
||||
// test ByteBuffer encode(ByteBuffer)
|
||||
int limit = srcBuf.limit();
|
||||
ByteBuffer resBuf = encoder.encode(srcBuf);
|
||||
assertEqual(srcBuf.position(), limit);
|
||||
assertEqual(srcBuf.limit(), limit);
|
||||
assertEqual(resBuf, encodedBuf);
|
||||
srcBuf.rewind(); // reset for next test
|
||||
|
||||
// test String encodeToString(byte[])
|
||||
String resEncodeStr = encoder.encodeToString(srcArr);
|
||||
assertEqual(resEncodeStr, encodedStr);
|
||||
|
||||
// test int decode(byte[], byte[])
|
||||
resArr = new byte[srcArr.length];
|
||||
len = decoder.decode(encodedArr, resArr);
|
||||
assertEqual(len, srcArr.length);
|
||||
assertEqual(resArr, srcArr);
|
||||
|
||||
// test byte[] decode(byte[])
|
||||
resArr = decoder.decode(encodedArr);
|
||||
assertEqual(resArr, srcArr);
|
||||
|
||||
// test ByteBuffer decode(ByteBuffer)
|
||||
limit = encodedBuf.limit();
|
||||
resBuf = decoder.decode(encodedBuf);
|
||||
assertEqual(encodedBuf.position(), limit);
|
||||
assertEqual(encodedBuf.limit(), limit);
|
||||
assertEqual(resBuf, srcBuf);
|
||||
encodedBuf.rewind(); // reset for next test
|
||||
|
||||
// test byte[] decode(String)
|
||||
resArr = decoder.decode(encodedStr);
|
||||
assertEqual(resArr, srcArr);
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
private static void test1() throws Exception {
|
||||
byte[] src = new byte[] {
|
||||
46, -97, -35, -44, 127, -60, -39, -4, -112, 34, -57, 47, -14, 67,
|
||||
40, 18, 90, -59, 68, 112, 23, 121, -91, 94, 35, 49, 104, 17, 30,
|
||||
-80, -104, -3, -53, 27, 38, -72, -47, 113, -52, 18, 5, -126 };
|
||||
Encoder encoder = Base64.getMimeEncoder(49, new byte[] { 0x7e });
|
||||
byte[] encoded = encoder.encode(src);
|
||||
Decoder decoder = Base64.getMimeDecoder();
|
||||
byte[] decoded = decoder.decode(encoded);
|
||||
if (!Objects.deepEquals(src, decoded)) {
|
||||
throw new RuntimeException();
|
||||
}
|
||||
}
|
||||
|
||||
// helper
|
||||
enum Base64Type {
|
||||
BASIC, URLSAFE, MIME
|
||||
}
|
||||
|
||||
private static final String SRCDIR = System.getProperty("test.src", ".");
|
||||
private static final Charset DEF_CHARSET = StandardCharsets.US_ASCII;
|
||||
private static final String DEF_EXCEPTION_MSG =
|
||||
"Assertion failed! The result is not same as expected";
|
||||
private static final String DEFAULT_CRLF = "\r\n";
|
||||
|
||||
private static void assertEqual(Object result, Object expect) {
|
||||
if (!Objects.deepEquals(result, expect)) {
|
||||
String resultStr = result.toString();
|
||||
String expectStr = expect.toString();
|
||||
if (result instanceof byte[]) {
|
||||
resultStr = new String((byte[]) result, DEF_CHARSET);
|
||||
}
|
||||
if (expect instanceof byte[]) {
|
||||
expectStr = new String((byte[]) expect, DEF_CHARSET);
|
||||
}
|
||||
throw new RuntimeException(DEF_EXCEPTION_MSG +
|
||||
" result: " + resultStr + " expected: " + expectStr);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue