/*
|
* CDDL HEADER START
|
*
|
* The contents of this file are subject to the terms of the
|
* Common Development and Distribution License, Version 1.0 only
|
* (the "License"). You may not use this file except in compliance
|
* with the License.
|
*
|
* You can obtain a copy of the license at legal-notices/CDDLv1_0.txt
|
* or http://forgerock.org/license/CDDLv1.0.html.
|
* See the License for the specific language governing permissions
|
* and limitations under the License.
|
*
|
* When distributing Covered Code, include this CDDL HEADER in each
|
* file and include the License file at legal-notices/CDDLv1_0.txt.
|
* If applicable, add the following below this CDDL HEADER, with the
|
* fields enclosed by brackets "[]" replaced with your own identifying
|
* information:
|
* Portions Copyright [yyyy] [name of copyright owner]
|
*
|
* CDDL HEADER END
|
*
|
*
|
* Copyright 2006-2008 Sun Microsystems, Inc.
|
* Portions Copyright 2014-2015 ForgeRock AS
|
*/
|
package org.opends.server.util;
|
|
import java.text.ParseException;
|
import java.util.Arrays;
|
|
import org.testng.Assert;
|
import org.testng.annotations.DataProvider;
|
import org.testng.annotations.Test;
|
|
/**
|
* This class defines a set of tests for the
|
* {@link org.opends.server.util.Base64} class.
|
*/
|
public final class TestBase64 extends UtilTestCase {
|
/** Look up table for converting hex chars to byte values. */
|
private static final byte[] hexToByte = { -1, -1, -1, -1, -1, -1, -1, -1,
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
|
-1, -1, -1, -1, -1, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, -1, -1, -1, -1,
|
-1, -1, -1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1,
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1,
|
-1, 10, 11, 12, 13, 14, 15, -1, -1, -1, -1, -1, -1, -1, -1, -1 - 1,
|
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 };
|
|
/**
|
* Base 64 valid test data provider.
|
*
|
* @return Returns an array of decoded and valid encoded base64 data.
|
*/
|
@DataProvider(name = "validData")
|
public Object[][] createValidData() {
|
return new Object[][] {
|
{ "", "" },
|
{ "00", "AA==" },
|
{ "01", "AQ==" },
|
{ "02", "Ag==" },
|
{ "03", "Aw==" },
|
{ "04", "BA==" },
|
{ "05", "BQ==" },
|
{ "06", "Bg==" },
|
{ "07", "Bw==" },
|
{ "0000", "AAA=" },
|
{ "000000", "AAAA" },
|
{ "00000000", "AAAAAA==" },
|
{
|
"000102030405060708090a0b0c0d0e0f"
|
+ "101112131415161718191a1b1c1d1e1f"
|
+ "202122232425262728292a2b2c2d2e2f"
|
+ "303132333435363738393a3b3c3d3e3f"
|
+ "404142434445464748494a4b4c4d4e4f"
|
+ "505152535455565758595a5b5c5d5e5f"
|
+ "606162636465666768696a6b6c6d6e6f"
|
+ "707172737475767778797a7b7c7d7e7f"
|
+ "808182838485868788898a8b8c8d8e8f"
|
+ "909192939495969798999a9b9c9d9e9f"
|
+ "a0a1a2a3a4a5a6a7a8a9aaabacadaeaf"
|
+ "b0b1b2b3b4b5b6b7b8b9babbbcbdbebf"
|
+ "c0c1c2c3c4c5c6c7c8c9cacbcccdcecf"
|
+ "d0d1d2d3d4d5d6d7d8d9dadbdcdddedf"
|
+ "e0e1e2e3e4e5e6e7e8e9eaebecedeeef"
|
+ "f0f1f2f3f4f5f6f7f8f9fafbfcfdfeff",
|
"AAECAwQFBgcICQoLDA0ODxAREhMUFRYXGBkaGxwdHh8gISIjJCUmJygpKissLS4v"
|
+ "MDEyMzQ1Njc4OTo7PD0+P0BBQkNERUZHSElKS0xNTk9QUVJTVFVWV1hZWltcXV5f"
|
+ "YGFiY2RlZmdoaWprbG1ub3BxcnN0dXZ3eHl6e3x9fn+AgYKDhIWGh4iJiouMjY6P"
|
+ "kJGSk5SVlpeYmZqbnJ2en6ChoqOkpaanqKmqq6ytrq+wsbKztLW2t7i5uru8vb6/"
|
+ "wMHCw8TFxsfIycrLzM3Oz9DR0tPU1dbX2Nna29zd3t/g4eLj5OXm5+jp6uvs7e7v"
|
+ "8PHy8/T19vf4+fr7/P3+/w==" }, };
|
}
|
|
/**
|
* Base 64 invalid test data provider.
|
*
|
* @return Returns an array of invalid encoded base64 data.
|
*/
|
@DataProvider(name = "invalidData")
|
public Object[][] createInvalidData() {
|
// FIXME: fix cases ==== and ==x=
|
|
return new Object[][] { { "=" }, { "==" }, { "===" }, { "A" },
|
{ "AA" }, { "AAA" }, { "AA`=" }, { "AA~=" }, { "AA!=" },
|
{ "AA@=" }, { "AA#=" }, { "AA$=" }, { "AA%=" }, { "AA^=" },
|
{ "AA*=" }, { "AA(=" }, { "AA)=" }, { "AA_=" }, { "AA-=" },
|
{ "AA{=" }, { "AA}=" }, { "AA|=" }, { "AA[=" }, { "AA]=" },
|
{ "AA\\=" }, { "AA;=" }, { "AA'=" }, { "AA\"=" }, { "AA:=" },
|
{ "AA,=" }, { "AA.=" }, { "AA<=" }, { "AA>=" }, { "AA?=" },
|
{ "AA;=" } };
|
}
|
|
/**
|
* Tests the encode method.
|
*
|
* @param hexData
|
* The decoded hex data.
|
* @param encodedData
|
* The encoded data.
|
* @throws Exception
|
* If the test failed unexpectedly.
|
*/
|
@Test(dataProvider = "validData")
|
public void testEncode(String hexData, String encodedData)
|
throws Exception {
|
byte[] data = getBytes(hexData);
|
String base64 = Base64.encode(data);
|
Assert.assertEquals(base64, encodedData);
|
}
|
|
/**
|
* Tests the decode method against valid data.
|
*
|
* @param hexData
|
* The decoded hex data.
|
* @param encodedData
|
* The encoded data.
|
* @throws Exception
|
* If the test failed unexpectedly.
|
*/
|
@Test(dataProvider = "validData")
|
public void testDecodeValidData(String hexData, String encodedData)
|
throws Exception {
|
byte[] data = getBytes(hexData);
|
byte[] decodedData = Base64.decode(encodedData);
|
Assert.assertEquals(decodedData, data);
|
}
|
|
/**
|
* Tests the decode method against invalid data.
|
*
|
* @param encodedData
|
* The invalid encoded data.
|
* @throws Exception
|
* If the test failed unexpectedly.
|
*/
|
@Test(dataProvider = "invalidData", expectedExceptions = { ParseException.class })
|
public void testDecodeInvalidData(String encodedData) throws Exception {
|
Assert.fail("Expected exception but got result: "
|
+ Arrays.toString(Base64.decode(encodedData)));
|
}
|
|
/**
|
* Decode a hex string to a byte-array.
|
*
|
* @param hexData
|
* The string of hex.
|
* @return Returns the decoded byte array.
|
*/
|
private byte[] getBytes(String hexData) {
|
int sz = hexData.length();
|
|
if (sz % 2 != 0) {
|
throw new IllegalArgumentException(
|
"Hex string does not contain an even number of hex digits");
|
}
|
|
byte[] bytes = new byte[sz / 2];
|
|
for (int i = 0, j = 0; i < sz; i += 2, j++) {
|
int c = hexData.codePointAt(i);
|
if ((c & 0x7f) != c) {
|
throw new IllegalArgumentException(
|
"Hex string contains non-hex digits");
|
}
|
|
byte b1 = hexToByte[c];
|
if (b1 < 0) {
|
throw new IllegalArgumentException(
|
"Hex string contains non-hex digits");
|
}
|
|
c = hexData.codePointAt(i + 1);
|
if ((c & 0x7f) != c) {
|
throw new IllegalArgumentException(
|
"Hex string contains non-hex digits");
|
}
|
|
byte b2 = hexToByte[c];
|
if (b2 < 0) {
|
throw new IllegalArgumentException(
|
"Hex string contains non-hex digits");
|
}
|
|
bytes[j] = (byte) ((b1 << 4) | b2);
|
}
|
|
return bytes;
|
}
|
}
|
