From 3f80494ea88c323bbb3e6b778441604e333e2074 Mon Sep 17 00:00:00 2001 From: Mark Paluch Date: Tue, 14 Dec 2021 11:20:25 +0100 Subject: [PATCH 1/2] Prepare issue branch. --- pom.xml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pom.xml b/pom.xml index 014d5831b9..2e41c7f9ee 100644 --- a/pom.xml +++ b/pom.xml @@ -5,7 +5,7 @@ org.springframework.data spring-data-redis - 2.7.0-SNAPSHOT + 2.7.0-GH-2204-SNAPSHOT Spring Data Redis From 6ed3b510ed821f8cea946bea22d0975190fed11a Mon Sep 17 00:00:00 2001 From: Mark Paluch Date: Tue, 14 Dec 2021 11:27:12 +0100 Subject: [PATCH 2/2] Respect ByteBuffer position and limits in ByteUtils.getBytes(ByteBuffer). We now properly extract the byte array from a ByteBuffer by copying its content respecting the read position and limits. Closes #2204 --- .../data/redis/util/ByteUtils.java | 7 +-- .../data/redis/util/ByteUtilsUnitTests.java | 55 +++++++++++++++++++ 2 files changed, 57 insertions(+), 5 deletions(-) create mode 100644 src/test/java/org/springframework/data/redis/util/ByteUtilsUnitTests.java diff --git a/src/main/java/org/springframework/data/redis/util/ByteUtils.java b/src/main/java/org/springframework/data/redis/util/ByteUtils.java index 75737a1862..5eda0c33fb 100644 --- a/src/main/java/org/springframework/data/redis/util/ByteUtils.java +++ b/src/main/java/org/springframework/data/redis/util/ByteUtils.java @@ -129,7 +129,8 @@ public static byte[][] mergeArrays(byte[] firstArray, byte[]... additionalArrays } /** - * Extract a byte array from {@link ByteBuffer} without consuming it. + * Extract a byte array from {@link ByteBuffer} without consuming it. The resulting {@code byte[]} is a copy of the + * buffer's contents and not updated upon changes within the buffer. * * @param byteBuffer must not be {@literal null}. * @return @@ -139,10 +140,6 @@ public static byte[] getBytes(ByteBuffer byteBuffer) { Assert.notNull(byteBuffer, "ByteBuffer must not be null!"); - if (byteBuffer.hasArray()) { - return byteBuffer.array(); - } - ByteBuffer duplicate = byteBuffer.duplicate(); byte[] bytes = new byte[duplicate.remaining()]; duplicate.get(bytes); diff --git a/src/test/java/org/springframework/data/redis/util/ByteUtilsUnitTests.java b/src/test/java/org/springframework/data/redis/util/ByteUtilsUnitTests.java new file mode 100644 index 0000000000..5e7b304dc5 --- /dev/null +++ b/src/test/java/org/springframework/data/redis/util/ByteUtilsUnitTests.java @@ -0,0 +1,55 @@ +/* + * Copyright 2021 the original author or authors. + * + * 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 + * + * https://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 org.springframework.data.redis.util; + +import static org.assertj.core.api.Assertions.*; + +import java.nio.ByteBuffer; +import java.nio.charset.StandardCharsets; + +import org.junit.jupiter.api.Test; + +/** + * Unit tests for {@link ByteUtils}. + * + * @author Mark Paluch + */ +class ByteUtilsUnitTests { + + @Test // GH-2204 + void getBytesShouldUseCorrectHeapBufferSpace() { + + ByteBuffer buffer = ByteBuffer.allocate(16); + buffer.put("hello".getBytes(StandardCharsets.US_ASCII)); + buffer.flip(); + + byte[] bytes = ByteUtils.getBytes(buffer); + + assertThat(bytes).hasSize(5); + } + + @Test // GH-2204 + void getBytesShouldUseCorrectDirectBufferSpace() { + + ByteBuffer buffer = ByteBuffer.allocateDirect(16); + buffer.put("hello".getBytes(StandardCharsets.US_ASCII)); + buffer.flip(); + + byte[] bytes = ByteUtils.getBytes(buffer); + + assertThat(bytes).hasSize(5); + } +}