SEC-1752: Fixed Utf8 codec to take account of the limit of the ByteBuffer returned by CharsetEncoder.encode().

This commit is contained in:
Luke Taylor 2011-05-23 18:55:25 +01:00
parent e8a1a6e40b
commit 5a4aed238c
2 changed files with 31 additions and 3 deletions

View File

@ -4,6 +4,7 @@ import java.nio.ByteBuffer;
import java.nio.CharBuffer;
import java.nio.charset.CharacterCodingException;
import java.nio.charset.Charset;
import java.util.*;
/**
* UTF-8 Charset encoder/decoder.
@ -20,7 +21,9 @@ public final class Utf8 {
*/
public static byte[] encode(CharSequence string) {
try {
return CHARSET.newEncoder().encode(CharBuffer.wrap(string)).array();
ByteBuffer bytes = CHARSET.newEncoder().encode(CharBuffer.wrap(string));
return Arrays.copyOfRange(bytes.array(), 0, bytes.limit());
} catch (CharacterCodingException e) {
throw new IllegalArgumentException("Encoding failed", e);
}
@ -31,9 +34,9 @@ public final class Utf8 {
*/
public static String decode(byte[] bytes) {
try {
return new String(CHARSET.newDecoder().decode(ByteBuffer.wrap(bytes)).array());
return CHARSET.newDecoder().decode(ByteBuffer.wrap(bytes)).toString();
} catch (CharacterCodingException e) {
throw new IllegalArgumentException("Encoding failed", e);
throw new IllegalArgumentException("Decoding failed", e);
}
}
}

View File

@ -0,0 +1,25 @@
package org.springframework.security.crypto.codec;
import static org.junit.Assert.*;
import org.junit.*;
import java.util.*;
/**
* @author Luke Taylor
*/
public class Utf8Tests {
// SEC-1752
@Test
public void utf8EncodesAndDecodesCorrectly() throws Exception {
byte[] bytes = Utf8.encode("6048b75ed560785c");
assertEquals(16, bytes.length);
assertTrue(Arrays.equals("6048b75ed560785c".getBytes("UTF-8"), bytes));
String decoded = Utf8.decode(bytes);
assertEquals("6048b75ed560785c", decoded);
}
}