Merge "a new java implementation of base64 for android-common"
This commit is contained in:
Doug Zongker
497
common/java/com/android/common/Base64.java
Normal file
497
common/java/com/android/common/Base64.java
Normal file
@@ -0,0 +1,497 @@
|
||||
/*
|
||||
* Copyright (C) 2010 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.common;
|
||||
|
||||
/**
|
||||
* Utilities for encoding and decoding the Base64 encoding. See RFCs
|
||||
* 2045 and 3548.
|
||||
*/
|
||||
public class Base64 {
|
||||
/**
|
||||
* Encoder flag bit to indicate you want the padding '='
|
||||
* characters at the end (if any) to be omitted.
|
||||
*/
|
||||
public static final int NO_PADDING = 1;
|
||||
|
||||
/**
|
||||
* Encoder flag bit to indicate you want all line terminators to
|
||||
* be omitted (ie, the output will be on one long line).
|
||||
*/
|
||||
public static final int NO_WRAP = 2;
|
||||
|
||||
/**
|
||||
* Encoder flag bit to indicate you want lines to be ended with
|
||||
* CRLF instead of just LF.
|
||||
*/
|
||||
public static final int CRLF = 4;
|
||||
|
||||
/**
|
||||
* Encoder/decoder flag bit to indicate using the "web safe"
|
||||
* variant of Base64 (see RFC 3548 section 4) where '-' and '_'
|
||||
* are used in place of '+' and '/'.
|
||||
*/
|
||||
public static final int WEB_SAFE = 8;
|
||||
|
||||
/**
|
||||
* Lookup table for turning bytes into their position in the
|
||||
* Base64 alphabet.
|
||||
*/
|
||||
private static final int DECODE[] = {
|
||||
-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, 62, -1, -1, -1, 63,
|
||||
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -2, -1, -1,
|
||||
-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
|
||||
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, -1,
|
||||
-1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
|
||||
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -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, -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, -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,
|
||||
};
|
||||
|
||||
/**
|
||||
* Decode lookup table for the "web safe" variant (RFC 3548
|
||||
* sec. 4) where - and _ replace + and /.
|
||||
*/
|
||||
private static final int DECODE_WEBSAFE[] = {
|
||||
-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, 62, -1, -1,
|
||||
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, -1, -1, -1, -2, -1, -1,
|
||||
-1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14,
|
||||
15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, -1, -1, -1, -1, 63,
|
||||
-1, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40,
|
||||
41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, -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, -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, -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,
|
||||
};
|
||||
|
||||
/** Non-data values in the DECODE arrays. */
|
||||
private static final int SKIP = -1;
|
||||
private static final int EQUALS = -2;
|
||||
|
||||
/**
|
||||
* Decode the Base64-encoded data in input and return the data in
|
||||
* a new byte array.
|
||||
*
|
||||
* The padding '=' characters at the end are considered optional, but
|
||||
* if any are present, there must be the correct number of them.
|
||||
*
|
||||
* @param input the input String to decode, which is converted to
|
||||
* bytes using the default charset
|
||||
* @param flags controls certain features of the decoded output.
|
||||
* Passing 0 to decode standard Base64.
|
||||
*
|
||||
* @throws IllegalArgumentException if the input contains
|
||||
* incorrect padding
|
||||
*/
|
||||
public static byte[] decode(String str, int flags) {
|
||||
return decode(str.getBytes(), flags);
|
||||
}
|
||||
|
||||
/**
|
||||
* Decode the Base64-encoded data in input and return the data in
|
||||
* a new byte array.
|
||||
*
|
||||
* The padding '=' characters at the end are considered optional, but
|
||||
* if any are present, there must be the correct number of them.
|
||||
*
|
||||
* @param input the input array to decode
|
||||
* @param flags controls certain features of the decoded output.
|
||||
* Passing 0 to decode standard Base64.
|
||||
*
|
||||
* @throws IllegalArgumentException if the input contains
|
||||
* incorrect padding
|
||||
*/
|
||||
public static byte[] decode(byte[] input, int flags) {
|
||||
return decode(input, 0, input.length, flags);
|
||||
}
|
||||
|
||||
/**
|
||||
* Decode the Base64-encoded data in input and return the data in
|
||||
* a new byte array.
|
||||
*
|
||||
* The padding '=' characters at the end are considered optional, but
|
||||
* if any are present, there must be the correct number of them.
|
||||
*
|
||||
* @param input the data to decode
|
||||
* @param offset the position within the input array at which to start
|
||||
* @param len the number of bytes of input to decode
|
||||
* @param flags controls certain features of the decoded output.
|
||||
* Passing 0 to decode standard Base64.
|
||||
*
|
||||
* @throws IllegalArgumentException if the input contains
|
||||
* incorrect padding
|
||||
*/
|
||||
public static byte[] decode(byte[] input, int offset, int len, int flags) {
|
||||
int p = offset;
|
||||
// Allocate space for the most data the input could represent.
|
||||
// (It could contain less if it contains whitespace, etc.)
|
||||
byte[] output = new byte[len*3/4];
|
||||
len += offset;
|
||||
int op = 0;
|
||||
|
||||
final int[] decode = ((flags & WEB_SAFE) == 0) ?
|
||||
DECODE : DECODE_WEBSAFE;
|
||||
|
||||
int state = 0;
|
||||
int value = 0;
|
||||
|
||||
while (p < len) {
|
||||
|
||||
// Try the fast path: we're starting a new tuple and the
|
||||
// next four bytes of the input stream are all data
|
||||
// bytes. This corresponds to going through states
|
||||
// 0-1-2-3-0. We expect to use this method for most of
|
||||
// the data.
|
||||
//
|
||||
// If any of the next four bytes of input are non-data
|
||||
// (whitespace, etc.), value will end up negative. (All
|
||||
// the non-data values in decode are small negative
|
||||
// numbers, so shifting any of them up and or'ing them
|
||||
// together will result in a value with its top bit set.)
|
||||
//
|
||||
// You can remove this whole block and the output should
|
||||
// be the same, just slower.
|
||||
if (state == 0 && p+4 <= len &&
|
||||
(value = ((decode[input[p] & 0xff] << 18) |
|
||||
(decode[input[p+1] & 0xff] << 12) |
|
||||
(decode[input[p+2] & 0xff] << 6) |
|
||||
(decode[input[p+3] & 0xff]))) >= 0) {
|
||||
output[op+2] = (byte) value;
|
||||
output[op+1] = (byte) (value >> 8);
|
||||
output[op] = (byte) (value >> 16);
|
||||
op += 3;
|
||||
p += 4;
|
||||
continue;
|
||||
}
|
||||
|
||||
// The fast path isn't available -- either we've read a
|
||||
// partial tuple, or the next four input bytes aren't all
|
||||
// data, or whatever. Fall back to the slower state
|
||||
// machine implementation.
|
||||
//
|
||||
// States 0-3 are reading through the next input tuple.
|
||||
// State 4 is having read one '=' and expecting exactly
|
||||
// one more.
|
||||
// State 5 is expecting no more data or padding characters
|
||||
// in the input.
|
||||
|
||||
int d = decode[input[p++] & 0xff];
|
||||
|
||||
switch (state) {
|
||||
case 0:
|
||||
if (d >= 0) {
|
||||
value = d;
|
||||
++state;
|
||||
} else if (d != SKIP) {
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
}
|
||||
break;
|
||||
|
||||
case 1:
|
||||
if (d >= 0) {
|
||||
value = (value << 6) | d;
|
||||
++state;
|
||||
} else if (d != SKIP) {
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
}
|
||||
break;
|
||||
|
||||
case 2:
|
||||
if (d >= 0) {
|
||||
value = (value << 6) | d;
|
||||
++state;
|
||||
} else if (d == EQUALS) {
|
||||
// Emit the last (partial) output tuple;
|
||||
// expect exactly one more padding character.
|
||||
output[op++] = (byte) (value >> 4);
|
||||
state = 4;
|
||||
} else if (d != SKIP) {
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
}
|
||||
break;
|
||||
|
||||
case 3:
|
||||
if (d >= 0) {
|
||||
// Emit the output triple and return to state 0.
|
||||
value = (value << 6) | d;
|
||||
output[op+2] = (byte) value;
|
||||
output[op+1] = (byte) (value >> 8);
|
||||
output[op] = (byte) (value >> 16);
|
||||
op += 3;
|
||||
state = 0;
|
||||
} else if (d == EQUALS) {
|
||||
// Emit the last (partial) output tuple;
|
||||
// expect no further data or padding characters.
|
||||
output[op+1] = (byte) (value >> 2);
|
||||
output[op] = (byte) (value >> 10);
|
||||
op += 2;
|
||||
state = 5;
|
||||
} else if (d != SKIP) {
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
}
|
||||
break;
|
||||
|
||||
case 4:
|
||||
if (d == EQUALS) {
|
||||
++state;
|
||||
} else if (d != SKIP) {
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
}
|
||||
break;
|
||||
|
||||
case 5:
|
||||
if (d != SKIP) {
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
}
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
// Done reading input. Now figure out where we are left in
|
||||
// the state machine and finish up.
|
||||
|
||||
switch (state) {
|
||||
case 0:
|
||||
// Output length is a multiple of three. Fine.
|
||||
break;
|
||||
case 1:
|
||||
// Read one extra input byte, which isn't enough to
|
||||
// make another output byte. Illegal.
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
case 2:
|
||||
// Read two extra input bytes, enough to emit 1 more
|
||||
// output byte. Fine.
|
||||
output[op++] = (byte) (value >> 4);
|
||||
break;
|
||||
case 3:
|
||||
// Read three extra input bytes, enough to emit 2 more
|
||||
// output bytes. Fine.
|
||||
output[op+1] = (byte) (value >> 2);
|
||||
output[op] = (byte) (value >> 10);
|
||||
op += 2;
|
||||
break;
|
||||
case 4:
|
||||
// Read one padding '=' when we expected 2. Illegal.
|
||||
throw new IllegalArgumentException("bad base-64");
|
||||
case 5:
|
||||
// Read all the padding '='s we expected and no more.
|
||||
// Fine.
|
||||
break;
|
||||
}
|
||||
|
||||
// Maybe we got lucky and allocated exactly enough output space.
|
||||
if (op == output.length) {
|
||||
return output;
|
||||
}
|
||||
|
||||
// Need to shorten the array, so allocate a new one of the
|
||||
// right size and copy.
|
||||
byte[] temp = new byte[op];
|
||||
System.arraycopy(output, 0, temp, 0, op);
|
||||
return temp;
|
||||
}
|
||||
|
||||
/**
|
||||
* Emit a new line every this many output tuples. Corresponds to
|
||||
* a 76-character line length (the maximum allowable according to
|
||||
* RFC 2045).
|
||||
*/
|
||||
private static final int LINE_GROUPS = 19;
|
||||
|
||||
/**
|
||||
* Lookup table for turning Base64 alphabet positions (6 bits)
|
||||
* into output bytes.
|
||||
*/
|
||||
private static final byte ENCODE[] = {
|
||||
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
|
||||
'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
|
||||
'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X',
|
||||
'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f',
|
||||
'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
|
||||
'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
|
||||
'w', 'x', 'y', 'z', '0', '1', '2', '3',
|
||||
'4', '5', '6', '7', '8', '9', '+', '/',
|
||||
};
|
||||
|
||||
/**
|
||||
* Lookup table for turning Base64 alphabet positions (6 bits)
|
||||
* into output bytes.
|
||||
*/
|
||||
private static final byte ENCODE_WEBSAFE[] = {
|
||||
'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H',
|
||||
'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P',
|
||||
'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X',
|
||||
'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f',
|
||||
'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n',
|
||||
'o', 'p', 'q', 'r', 's', 't', 'u', 'v',
|
||||
'w', 'x', 'y', 'z', '0', '1', '2', '3',
|
||||
'4', '5', '6', '7', '8', '9', '-', '_',
|
||||
};
|
||||
|
||||
/**
|
||||
* Base64-encode the given data and return a newly allocated
|
||||
* String with the result.
|
||||
*
|
||||
* @param input the data to encode
|
||||
* @param flags controls certain features of the encoded output.
|
||||
* Passing 0 results in output that adheres to RFC
|
||||
* 2045.
|
||||
*/
|
||||
public static String encodeString(byte[] input, int flags) {
|
||||
return new String(encode(input, flags));
|
||||
}
|
||||
|
||||
/**
|
||||
* Base64-encode the given data and return a newly allocated
|
||||
* String with the result.
|
||||
*
|
||||
* @param input the data to encode
|
||||
* @param offset the position within the input array at which to
|
||||
* start
|
||||
* @param len the number of bytes of input to encode
|
||||
* @param flags controls certain features of the encoded output.
|
||||
* Passing 0 results in output that adheres to RFC
|
||||
* 2045.
|
||||
*/
|
||||
public static String encodeString(byte[] input, int offset, int len, int flags) {
|
||||
return new String(encode(input, offset, len, flags));
|
||||
}
|
||||
|
||||
/**
|
||||
* Base64-encode the given data and return a newly allocated
|
||||
* byte[] with the result.
|
||||
*
|
||||
* @param input the data to encode
|
||||
* @param flags controls certain features of the encoded output.
|
||||
* Passing 0 results in output that adheres to RFC
|
||||
* 2045.
|
||||
*/
|
||||
public static byte[] encode(byte[] input, int flags) {
|
||||
return encode(input, 0, input.length, flags);
|
||||
}
|
||||
|
||||
/**
|
||||
* Base64-encode the given data and return a newly allocated
|
||||
* byte[] with the result.
|
||||
*
|
||||
* @param input the data to encode
|
||||
* @param offset the position within the input array at which to
|
||||
* start
|
||||
* @param len the number of bytes of input to encode
|
||||
* @param flags controls certain features of the encoded output.
|
||||
* Passing 0 results in output that adheres to RFC
|
||||
* 2045.
|
||||
*/
|
||||
public static byte[] encode(byte[] input, int offset, int len, int flags) {
|
||||
final boolean do_padding = (flags & NO_PADDING) == 0;
|
||||
final boolean do_newline = (flags & NO_WRAP) == 0;
|
||||
final boolean do_cr = (flags & CRLF) != 0;
|
||||
|
||||
final byte[] encode = ((flags & WEB_SAFE) == 0) ? ENCODE : ENCODE_WEBSAFE;
|
||||
|
||||
// Compute the exact length of the array we will produce.
|
||||
int output_len = len / 3 * 4;
|
||||
|
||||
// Account for the tail of the data and the padding bytes, if any.
|
||||
if (do_padding) {
|
||||
if (len % 3 > 0) {
|
||||
output_len += 4;
|
||||
}
|
||||
} else {
|
||||
switch (len % 3) {
|
||||
case 0: break;
|
||||
case 1: output_len += 2; break;
|
||||
case 2: output_len += 3; break;
|
||||
}
|
||||
}
|
||||
|
||||
// Account for the newlines, if any.
|
||||
if (do_newline && len > 0) {
|
||||
output_len += (((len-1) / (3 * LINE_GROUPS)) + 1) * (do_cr ? 2 : 1);
|
||||
}
|
||||
|
||||
int op = 0;
|
||||
byte[] output = new byte[output_len];
|
||||
|
||||
// The main loop, turning 3 input bytes into 4 output bytes on
|
||||
// each iteration.
|
||||
int count = do_newline ? LINE_GROUPS : -1;
|
||||
int p = offset;
|
||||
len += offset;
|
||||
while (p+3 <= len) {
|
||||
int v = ((input[p++] & 0xff) << 16) |
|
||||
((input[p++] & 0xff) << 8) |
|
||||
(input[p++] & 0xff);
|
||||
output[op++] = encode[(v >> 18) & 0x3f];
|
||||
output[op++] = encode[(v >> 12) & 0x3f];
|
||||
output[op++] = encode[(v >> 6) & 0x3f];
|
||||
output[op++] = encode[v & 0x3f];
|
||||
if (--count == 0) {
|
||||
if (do_cr) output[op++] = '\r';
|
||||
output[op++] = '\n';
|
||||
count = LINE_GROUPS;
|
||||
}
|
||||
}
|
||||
|
||||
// Finish up the tail of the input.
|
||||
if (p == len-1) {
|
||||
int v = (input[p] & 0xff) << 4;
|
||||
output[op++] = encode[(v >> 6) & 0x3f];
|
||||
output[op++] = encode[v & 0x3f];
|
||||
if (do_padding) {
|
||||
output[op++] = '=';
|
||||
output[op++] = '=';
|
||||
}
|
||||
if (do_newline) {
|
||||
if (do_cr) output[op++] = '\r';
|
||||
output[op++] = '\n';
|
||||
}
|
||||
} else if (p == len-2) {
|
||||
int v = ((input[p] & 0xff) << 10) | ((input[p+1] & 0xff) << 2);
|
||||
output[op++] = encode[(v >> 12) & 0x3f];
|
||||
output[op++] = encode[(v >> 6) & 0x3f];
|
||||
output[op++] = encode[v & 0x3f];
|
||||
if (do_padding) {
|
||||
output[op++] = '=';
|
||||
}
|
||||
if (do_newline) {
|
||||
if (do_cr) output[op++] = '\r';
|
||||
output[op++] = '\n';
|
||||
}
|
||||
} else if (do_newline && op > 0 && count != LINE_GROUPS) {
|
||||
if (do_cr) output[op++] = '\r';
|
||||
output[op++] = '\n';
|
||||
}
|
||||
|
||||
assert op == output.length;
|
||||
return output;
|
||||
}
|
||||
}
|
||||
208
common/tests/src/com/android/common/Base64Test.java
Normal file
208
common/tests/src/com/android/common/Base64Test.java
Normal file
@@ -0,0 +1,208 @@
|
||||
/*
|
||||
* Copyright (C) 2010 The Android Open Source Project
|
||||
*
|
||||
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||
* you may not use this file except in compliance with the License.
|
||||
* You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
|
||||
package com.android.common;
|
||||
|
||||
import junit.framework.TestCase;
|
||||
|
||||
public class Base64Test extends TestCase {
|
||||
private static final String TAG = "B64Test";
|
||||
|
||||
/** Decodes a string, returning a string. */
|
||||
private String decodeString(String in) throws Exception {
|
||||
byte[] out = Base64.decode(in, 0);
|
||||
return new String(out);
|
||||
}
|
||||
|
||||
/**
|
||||
* Encodes the string 'in' using 'flags'. Asserts that decoding
|
||||
* gives the same string. Returns the encoded string.
|
||||
*/
|
||||
private String encodeString(String in, int flags) throws Exception {
|
||||
String b64 = Base64.encodeString(in.getBytes(), flags);
|
||||
String dec = decodeString(b64);
|
||||
assertEquals(in, dec);
|
||||
return b64;
|
||||
}
|
||||
|
||||
/** Assert that decoding 'in' throws IllegalArgumentException. */
|
||||
private void assertBad(String in) throws Exception {
|
||||
try {
|
||||
byte[] out = Base64.decode(in, 0);
|
||||
fail("should have failed to decode");
|
||||
} catch (IllegalArgumentException e) {
|
||||
}
|
||||
}
|
||||
|
||||
/** Assert that actual equals the first len bytes of expected. */
|
||||
private void assertEquals(byte[] expected, int len, byte[] actual) {
|
||||
assertEquals(len, actual.length);
|
||||
for (int i = 0; i < len; ++i) {
|
||||
assertEquals(expected[i], actual[i]);
|
||||
}
|
||||
}
|
||||
|
||||
public void testDecodeExtraChars() throws Exception {
|
||||
// padding 0
|
||||
assertEquals("hello, world", decodeString("aGVsbG8sIHdvcmxk"));
|
||||
assertBad("aGVsbG8sIHdvcmxk=");
|
||||
assertBad("aGVsbG8sIHdvcmxk==");
|
||||
assertBad("aGVsbG8sIHdvcmxk =");
|
||||
assertBad("aGVsbG8sIHdvcmxk = = ");
|
||||
assertEquals("hello, world", decodeString(" aGVs bG8s IHdv cmxk "));
|
||||
assertEquals("hello, world", decodeString(" aGV sbG8 sIHd vcmx k "));
|
||||
assertEquals("hello, world", decodeString(" aG VsbG 8sIH dvcm xk "));
|
||||
assertEquals("hello, world", decodeString(" a GVsb G8sI Hdvc mxk "));
|
||||
assertEquals("hello, world", decodeString(" a G V s b G 8 s I H d v c m x k "));
|
||||
assertEquals("hello, world", decodeString("_a*G_V*s_b*G_8*s_I*H_d*v_c*m_x*k_"));
|
||||
assertEquals("hello, world", decodeString("aGVsbG8sIHdvcmxk"));
|
||||
|
||||
// padding 1
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPyE="));
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPyE"));
|
||||
assertBad("aGVsbG8sIHdvcmxkPyE==");
|
||||
assertBad("aGVsbG8sIHdvcmxkPyE ==");
|
||||
assertBad("aGVsbG8sIHdvcmxkPyE = = ");
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPy E="));
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPy E"));
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPy E ="));
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPy E "));
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPy E = "));
|
||||
assertEquals("hello, world?!", decodeString("aGVsbG8sIHdvcmxkPy E "));
|
||||
|
||||
// padding 2
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkLg=="));
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkLg"));
|
||||
assertBad("aGVsbG8sIHdvcmxkLg=");
|
||||
assertBad("aGVsbG8sIHdvcmxkLg =");
|
||||
assertBad("aGVsbG8sIHdvcmxkLg = ");
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkL g=="));
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkL g"));
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkL g =="));
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkL g "));
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkL g = = "));
|
||||
assertEquals("hello, world.", decodeString("aGVsbG8sIHdvcmxkL g "));
|
||||
}
|
||||
|
||||
private static final byte[] BYTES = { (byte) 0xff, (byte) 0xee, (byte) 0xdd,
|
||||
(byte) 0xcc, (byte) 0xbb, (byte) 0xaa,
|
||||
(byte) 0x99, (byte) 0x88, (byte) 0x77 };
|
||||
|
||||
public void testBinaryDecode() throws Exception {
|
||||
assertEquals(BYTES, 0, Base64.decode("", 0));
|
||||
assertEquals(BYTES, 1, Base64.decode("/w==", 0));
|
||||
assertEquals(BYTES, 2, Base64.decode("/+4=", 0));
|
||||
assertEquals(BYTES, 3, Base64.decode("/+7d", 0));
|
||||
assertEquals(BYTES, 4, Base64.decode("/+7dzA==", 0));
|
||||
assertEquals(BYTES, 5, Base64.decode("/+7dzLs=", 0));
|
||||
assertEquals(BYTES, 6, Base64.decode("/+7dzLuq", 0));
|
||||
assertEquals(BYTES, 7, Base64.decode("/+7dzLuqmQ==", 0));
|
||||
assertEquals(BYTES, 8, Base64.decode("/+7dzLuqmYg=", 0));
|
||||
}
|
||||
|
||||
public void testWebSafe() throws Exception {
|
||||
assertEquals(BYTES, 0, Base64.decode("", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 1, Base64.decode("_w==", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 2, Base64.decode("_-4=", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 3, Base64.decode("_-7d", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 4, Base64.decode("_-7dzA==", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 5, Base64.decode("_-7dzLs=", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 6, Base64.decode("_-7dzLuq", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 7, Base64.decode("_-7dzLuqmQ==", Base64.WEB_SAFE));
|
||||
assertEquals(BYTES, 8, Base64.decode("_-7dzLuqmYg=", Base64.WEB_SAFE));
|
||||
|
||||
assertEquals("", Base64.encodeString(BYTES, 0, 0, Base64.WEB_SAFE));
|
||||
assertEquals("_w==\n", Base64.encodeString(BYTES, 0, 1, Base64.WEB_SAFE));
|
||||
assertEquals("_-4=\n", Base64.encodeString(BYTES, 0, 2, Base64.WEB_SAFE));
|
||||
assertEquals("_-7d\n", Base64.encodeString(BYTES, 0, 3, Base64.WEB_SAFE));
|
||||
assertEquals("_-7dzA==\n", Base64.encodeString(BYTES, 0, 4, Base64.WEB_SAFE));
|
||||
assertEquals("_-7dzLs=\n", Base64.encodeString(BYTES, 0, 5, Base64.WEB_SAFE));
|
||||
assertEquals("_-7dzLuq\n", Base64.encodeString(BYTES, 0, 6, Base64.WEB_SAFE));
|
||||
assertEquals("_-7dzLuqmQ==\n", Base64.encodeString(BYTES, 0, 7, Base64.WEB_SAFE));
|
||||
assertEquals("_-7dzLuqmYg=\n", Base64.encodeString(BYTES, 0, 8, Base64.WEB_SAFE));
|
||||
}
|
||||
|
||||
public void testFlags() throws Exception {
|
||||
assertEquals("YQ==\n", encodeString("a", 0));
|
||||
assertEquals("YQ==", encodeString("a", Base64.NO_WRAP));
|
||||
assertEquals("YQ\n", encodeString("a", Base64.NO_PADDING));
|
||||
assertEquals("YQ", encodeString("a", Base64.NO_PADDING | Base64.NO_WRAP));
|
||||
assertEquals("YQ==\r\n", encodeString("a", Base64.CRLF));
|
||||
assertEquals("YQ\r\n", encodeString("a", Base64.CRLF | Base64.NO_PADDING));
|
||||
|
||||
assertEquals("YWI=\n", encodeString("ab", 0));
|
||||
assertEquals("YWI=", encodeString("ab", Base64.NO_WRAP));
|
||||
assertEquals("YWI\n", encodeString("ab", Base64.NO_PADDING));
|
||||
assertEquals("YWI", encodeString("ab", Base64.NO_PADDING | Base64.NO_WRAP));
|
||||
assertEquals("YWI=\r\n", encodeString("ab", Base64.CRLF));
|
||||
assertEquals("YWI\r\n", encodeString("ab", Base64.CRLF | Base64.NO_PADDING));
|
||||
|
||||
assertEquals("YWJj\n", encodeString("abc", 0));
|
||||
assertEquals("YWJj", encodeString("abc", Base64.NO_WRAP));
|
||||
assertEquals("YWJj\n", encodeString("abc", Base64.NO_PADDING));
|
||||
assertEquals("YWJj", encodeString("abc", Base64.NO_PADDING | Base64.NO_WRAP));
|
||||
assertEquals("YWJj\r\n", encodeString("abc", Base64.CRLF));
|
||||
assertEquals("YWJj\r\n", encodeString("abc", Base64.CRLF | Base64.NO_PADDING));
|
||||
|
||||
assertEquals("YWJjZA==\n", encodeString("abcd", 0));
|
||||
assertEquals("YWJjZA==", encodeString("abcd", Base64.NO_WRAP));
|
||||
assertEquals("YWJjZA\n", encodeString("abcd", Base64.NO_PADDING));
|
||||
assertEquals("YWJjZA", encodeString("abcd", Base64.NO_PADDING | Base64.NO_WRAP));
|
||||
assertEquals("YWJjZA==\r\n", encodeString("abcd", Base64.CRLF));
|
||||
assertEquals("YWJjZA\r\n", encodeString("abcd", Base64.CRLF | Base64.NO_PADDING));
|
||||
}
|
||||
|
||||
public void testLineLength() throws Exception {
|
||||
String in_56 = "abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcd";
|
||||
String in_57 = in_56 + "e";
|
||||
String in_58 = in_56 + "ef";
|
||||
String in_59 = in_56 + "efg";
|
||||
String in_60 = in_56 + "efgh";
|
||||
String in_61 = in_56 + "efghi";
|
||||
|
||||
String prefix = "YWJjZGVmZ2hpamtsbW5vcHFyc3R1dnd4eXphYmNkZWZnaGlqa2xtbm9wcXJzdHV2d3h5emFi";
|
||||
String out_56 = prefix + "Y2Q=\n";
|
||||
String out_57 = prefix + "Y2Rl\n";
|
||||
String out_58 = prefix + "Y2Rl\nZg==\n";
|
||||
String out_59 = prefix + "Y2Rl\nZmc=\n";
|
||||
String out_60 = prefix + "Y2Rl\nZmdo\n";
|
||||
String out_61 = prefix + "Y2Rl\nZmdoaQ==\n";
|
||||
|
||||
// no newline for an empty input array.
|
||||
assertEquals("", encodeString("", 0));
|
||||
|
||||
assertEquals(out_56, encodeString(in_56, 0));
|
||||
assertEquals(out_57, encodeString(in_57, 0));
|
||||
assertEquals(out_58, encodeString(in_58, 0));
|
||||
assertEquals(out_59, encodeString(in_59, 0));
|
||||
assertEquals(out_60, encodeString(in_60, 0));
|
||||
assertEquals(out_61, encodeString(in_61, 0));
|
||||
|
||||
assertEquals(out_56.replaceAll("=", ""), encodeString(in_56, Base64.NO_PADDING));
|
||||
assertEquals(out_57.replaceAll("=", ""), encodeString(in_57, Base64.NO_PADDING));
|
||||
assertEquals(out_58.replaceAll("=", ""), encodeString(in_58, Base64.NO_PADDING));
|
||||
assertEquals(out_59.replaceAll("=", ""), encodeString(in_59, Base64.NO_PADDING));
|
||||
assertEquals(out_60.replaceAll("=", ""), encodeString(in_60, Base64.NO_PADDING));
|
||||
assertEquals(out_61.replaceAll("=", ""), encodeString(in_61, Base64.NO_PADDING));
|
||||
|
||||
assertEquals(out_56.replaceAll("\n", ""), encodeString(in_56, Base64.NO_WRAP));
|
||||
assertEquals(out_57.replaceAll("\n", ""), encodeString(in_57, Base64.NO_WRAP));
|
||||
assertEquals(out_58.replaceAll("\n", ""), encodeString(in_58, Base64.NO_WRAP));
|
||||
assertEquals(out_59.replaceAll("\n", ""), encodeString(in_59, Base64.NO_WRAP));
|
||||
assertEquals(out_60.replaceAll("\n", ""), encodeString(in_60, Base64.NO_WRAP));
|
||||
assertEquals(out_61.replaceAll("\n", ""), encodeString(in_61, Base64.NO_WRAP));
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user