/*
|
* 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
|
* trunk/opendj3/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
|
* trunk/opendj3/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 2010 Sun Microsystems, Inc.
|
*/
|
|
package com.forgerock.opendj.util;
|
|
|
|
import java.util.Arrays;
|
|
import org.forgerock.i18n.LocalizedIllegalArgumentException;
|
import org.forgerock.opendj.ldap.ByteString;
|
import org.testng.Assert;
|
import org.testng.annotations.DataProvider;
|
import org.testng.annotations.Test;
|
|
import com.forgerock.opendj.util.Base64;
|
|
|
|
/**
|
* This class defines a set of tests for the {@link Base64} class.
|
*/
|
public final class Base64TestCase 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 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;=" } };
|
}
|
|
|
|
/**
|
* 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==" }, };
|
}
|
|
|
|
/**
|
* Tests the decode method against invalid data.
|
*
|
* @param encodedData
|
* The invalid encoded data.
|
* @throws Exception
|
* If the test failed unexpectedly.
|
*/
|
@Test(dataProvider = "invalidData", expectedExceptions = { LocalizedIllegalArgumentException.class })
|
public void testDecodeInvalidData(final String encodedData) throws Exception
|
{
|
Assert.fail("Expected exception but got result: "
|
+ Arrays.toString(new ByteString[] { Base64.decode(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(final String hexData, final String encodedData)
|
throws Exception
|
{
|
final byte[] data = getBytes(hexData);
|
final byte[] decodedData = Base64.decode(encodedData).toByteArray();
|
Assert.assertEquals(decodedData, data);
|
}
|
|
|
|
/**
|
* 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(final String hexData, final String encodedData)
|
throws Exception
|
{
|
final byte[] data = getBytes(hexData);
|
final String base64 = Base64.encode(ByteString.wrap(data));
|
Assert.assertEquals(base64, encodedData);
|
}
|
|
|
|
/**
|
* Decode a hex string to a byte-array.
|
*
|
* @param hexData
|
* The string of hex.
|
* @return Returns the decoded byte array.
|
*/
|
private byte[] getBytes(final String hexData)
|
{
|
final int sz = hexData.length();
|
|
if ((sz % 2) != 0)
|
{
|
throw new IllegalArgumentException(
|
"Hex string does not contain an even number of hex digits");
|
}
|
|
final 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");
|
}
|
|
final 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");
|
}
|
|
final byte b2 = hexToByte[c];
|
if (b2 < 0)
|
{
|
throw new IllegalArgumentException("Hex string contains non-hex digits");
|
}
|
|
bytes[j] = (byte) ((b1 << 4) | b2);
|
}
|
|
return bytes;
|
}
|
}
|
