mirror of https://github.com/grpc/grpc.git
[slice] Introduce grpc_core::SliceBuffer (#29635)
* A starter SliceBuffer implementation * Add comments and fix sanity checks * Minor fixes * more minor fixes * Addressing review comments and adding a slice_buffer_test * fix sanity checks * regenerate projects * fixing undefined function error * reverting changes from memory_allocator.cc and adding them to src/core/lib/slice/slice_buffer.cc to allow memory allocator lib to build correctly * fix sanity checks * adding an open source slice definition * regnerate projects * fix asan error * Automated change: Fix sanity tests * addressing review comments * fix sanity checks * regenerate projects * update * fix sanity checks * Converting slice buffer to retarin ownership of the underlying ctype * [slice_buffer] Introduce grpc_core::SliceBuffer * add test * Automated change: Fix sanity tests * missing files * Automated change: Fix sanity tests * Automated change: Fix sanity tests * sanity Co-authored-by: Vignesh Babu <vigneshbabu@google.com> Co-authored-by: Vignesh2208 <Vignesh2208@users.noreply.github.com> Co-authored-by: ctiller <ctiller@users.noreply.github.com>pull/29655/head
Craig Tiller
3 years ago
committed by
GitHub
parent
48749f739e
commit
7798e3b741
22 changed files with 618 additions and 263 deletions
@ -0,0 +1,106 @@ |
|||||||
|
// Copyright 2022 gRPC 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
|
||||||
|
//
|
||||||
|
// 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.
|
||||||
|
|
||||||
|
#ifndef GRPC_CORE_LIB_SLICE_SLICE_BUFFER_Habc_SLICE_BUFFER_H |
||||||
|
#define GRPC_CORE_LIB_SLICE_SLICE_BUFFER_Habc_SLICE_BUFFER_H |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include <string.h> |
||||||
|
|
||||||
|
#include <grpc/slice.h> |
||||||
|
#include <grpc/slice_buffer.h> |
||||||
|
|
||||||
|
#include "src/core/lib/slice/slice.h" |
||||||
|
|
||||||
|
namespace grpc_core { |
||||||
|
|
||||||
|
/// A slice buffer holds the memory for a collection of slices.
|
||||||
|
/// The SliceBuffer object itself is meant to only hide the C-style API,
|
||||||
|
/// and won't hold the data itself. In terms of lifespan, the
|
||||||
|
/// grpc_slice_buffer ought to be kept somewhere inside the caller's objects,
|
||||||
|
/// like a transport or an endpoint.
|
||||||
|
///
|
||||||
|
/// This lifespan rule is likely to change in the future, as we may
|
||||||
|
/// collapse the grpc_slice_buffer structure straight into this class.
|
||||||
|
///
|
||||||
|
/// The SliceBuffer API is basically a replica of the grpc_slice_buffer's,
|
||||||
|
/// and its documentation will move here once we remove the C structure,
|
||||||
|
/// which should happen before the Event Engine's API is no longer
|
||||||
|
/// an experimental API.
|
||||||
|
class SliceBuffer { |
||||||
|
public: |
||||||
|
explicit SliceBuffer() { grpc_slice_buffer_init(&slice_buffer_); } |
||||||
|
SliceBuffer(const SliceBuffer& other) = delete; |
||||||
|
SliceBuffer(SliceBuffer&& other) noexcept { |
||||||
|
grpc_slice_buffer_init(&slice_buffer_); |
||||||
|
grpc_slice_buffer_move_into(&slice_buffer_, &other.slice_buffer_); |
||||||
|
} |
||||||
|
/// Upon destruction, the underlying raw slice buffer is cleaned out and all
|
||||||
|
/// slices are unreffed.
|
||||||
|
~SliceBuffer() { grpc_slice_buffer_destroy(&slice_buffer_); } |
||||||
|
|
||||||
|
SliceBuffer& operator=(SliceBuffer&& other) noexcept { |
||||||
|
grpc_slice_buffer_move_into(&slice_buffer_, &other.slice_buffer_); |
||||||
|
return *this; |
||||||
|
} |
||||||
|
|
||||||
|
/// Appends a new slice into the SliceBuffer and makes an attempt to merge
|
||||||
|
/// this slice with the last slice in the SliceBuffer.
|
||||||
|
void Append(Slice slice); |
||||||
|
|
||||||
|
/// Adds a new slice into the SliceBuffer at the next available index.
|
||||||
|
/// Returns the index at which the new slice is added.
|
||||||
|
size_t AppendIndexed(Slice slice); |
||||||
|
|
||||||
|
/// Returns the number of slices held by the SliceBuffer.
|
||||||
|
size_t Count() { return slice_buffer_.count; } |
||||||
|
|
||||||
|
/// Removes/deletes the last n bytes in the SliceBuffer.
|
||||||
|
void RemoveLastNBytes(size_t n) { |
||||||
|
grpc_slice_buffer_trim_end(&slice_buffer_, n, nullptr); |
||||||
|
} |
||||||
|
|
||||||
|
/// Move the first n bytes of the SliceBuffer into a memory pointed to by dst.
|
||||||
|
void MoveFirstNBytesIntoBuffer(size_t n, void* dst) { |
||||||
|
grpc_slice_buffer_move_first_into_buffer(&slice_buffer_, n, dst); |
||||||
|
} |
||||||
|
|
||||||
|
/// Removes and unrefs all slices in the SliceBuffer.
|
||||||
|
void Clear() { grpc_slice_buffer_reset_and_unref(&slice_buffer_); } |
||||||
|
|
||||||
|
/// Removes the first slice in the SliceBuffer and returns it.
|
||||||
|
Slice TakeFirst(); |
||||||
|
|
||||||
|
/// Prepends the slice to the the front of the SliceBuffer.
|
||||||
|
void Prepend(Slice slice); |
||||||
|
|
||||||
|
/// Increased the ref-count of slice at the specified index and returns the
|
||||||
|
/// associated slice.
|
||||||
|
Slice RefSlice(size_t index); |
||||||
|
|
||||||
|
/// The total number of bytes held by the SliceBuffer
|
||||||
|
size_t Length() { return slice_buffer_.length; } |
||||||
|
|
||||||
|
/// Return a pointer to the back raw grpc_slice_buffer
|
||||||
|
grpc_slice_buffer* RawSliceBuffer() { return &slice_buffer_; } |
||||||
|
|
||||||
|
private: |
||||||
|
/// The backing raw slice buffer.
|
||||||
|
grpc_slice_buffer slice_buffer_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc_core
|
||||||
|
|
||||||
|
#endif // GRPC_CORE_LIB_SLICE_SLICE_BUFFER_H
|
||||||
@ -0,0 +1,35 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015 gRPC 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 |
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include <grpc/slice.h> |
||||||
|
#include <grpc/slice_buffer.h> |
||||||
|
|
||||||
|
#include "src/core/lib/iomgr/exec_ctx.h" |
||||||
|
#include "src/core/lib/slice/slice_internal.h" |
||||||
|
|
||||||
|
void grpc_slice_buffer_destroy(grpc_slice_buffer* sb) { |
||||||
|
grpc_core::ExecCtx exec_ctx; |
||||||
|
grpc_slice_buffer_destroy_internal(sb); |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_slice_buffer_reset_and_unref(grpc_slice_buffer* sb) { |
||||||
|
grpc_core::ExecCtx exec_ctx; |
||||||
|
grpc_slice_buffer_reset_and_unref_internal(sb); |
||||||
|
} |
||||||
@ -0,0 +1,193 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2015 gRPC 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 |
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <grpc/slice_buffer.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
#include "src/core/lib/slice/slice_internal.h" |
||||||
|
#include "test/core/util/test_config.h" |
||||||
|
|
||||||
|
static constexpr size_t kTotalDataLength = 4096; |
||||||
|
|
||||||
|
void test_slice_buffer_add() { |
||||||
|
grpc_slice_buffer buf; |
||||||
|
grpc_slice aaa = grpc_slice_from_copied_string("aaa"); |
||||||
|
grpc_slice bb = grpc_slice_from_copied_string("bb"); |
||||||
|
size_t i; |
||||||
|
|
||||||
|
grpc_slice_buffer_init(&buf); |
||||||
|
for (i = 0; i < 10; i++) { |
||||||
|
grpc_slice_ref(aaa); |
||||||
|
grpc_slice_ref(bb); |
||||||
|
grpc_slice_buffer_add(&buf, aaa); |
||||||
|
grpc_slice_buffer_add(&buf, bb); |
||||||
|
} |
||||||
|
GPR_ASSERT(buf.count > 0); |
||||||
|
GPR_ASSERT(buf.length == 50); |
||||||
|
grpc_slice_buffer_reset_and_unref(&buf); |
||||||
|
GPR_ASSERT(buf.count == 0); |
||||||
|
GPR_ASSERT(buf.length == 0); |
||||||
|
for (i = 0; i < 10; i++) { |
||||||
|
grpc_slice_ref(aaa); |
||||||
|
grpc_slice_ref(bb); |
||||||
|
grpc_slice_buffer_add(&buf, aaa); |
||||||
|
grpc_slice_buffer_add(&buf, bb); |
||||||
|
} |
||||||
|
GPR_ASSERT(buf.count > 0); |
||||||
|
GPR_ASSERT(buf.length == 50); |
||||||
|
for (i = 0; i < 10; i++) { |
||||||
|
grpc_slice_buffer_pop(&buf); |
||||||
|
grpc_slice_unref(aaa); |
||||||
|
grpc_slice_unref(bb); |
||||||
|
} |
||||||
|
GPR_ASSERT(buf.count == 0); |
||||||
|
GPR_ASSERT(buf.length == 0); |
||||||
|
grpc_slice_buffer_destroy(&buf); |
||||||
|
} |
||||||
|
|
||||||
|
static void free_data(void* data, size_t len) { |
||||||
|
GPR_ASSERT(len == kTotalDataLength); |
||||||
|
gpr_free(data); |
||||||
|
} |
||||||
|
|
||||||
|
void test_slice_buffer_add_contiguous_slices() { |
||||||
|
grpc_slice_buffer buf; |
||||||
|
grpc_slice_buffer_init(&buf); |
||||||
|
char* data = reinterpret_cast<char*>(gpr_malloc(kTotalDataLength)); |
||||||
|
GPR_ASSERT(data != nullptr); |
||||||
|
grpc_slice a = grpc_slice_new_with_len(data, kTotalDataLength, free_data); |
||||||
|
grpc_slice s1 = grpc_slice_split_head(&a, kTotalDataLength / 4); |
||||||
|
grpc_slice s2 = grpc_slice_split_head(&a, kTotalDataLength / 4); |
||||||
|
grpc_slice s3 = grpc_slice_split_head(&a, kTotalDataLength / 4); |
||||||
|
grpc_slice_buffer_add(&buf, s1); |
||||||
|
GPR_ASSERT(buf.count == 1); |
||||||
|
GPR_ASSERT(buf.length == kTotalDataLength / 4); |
||||||
|
grpc_slice_buffer_add(&buf, s2); |
||||||
|
GPR_ASSERT(buf.count == 1); |
||||||
|
GPR_ASSERT(buf.length == kTotalDataLength / 2); |
||||||
|
grpc_slice_buffer_add(&buf, s3); |
||||||
|
GPR_ASSERT(buf.count == 1); |
||||||
|
GPR_ASSERT(buf.length == 3 * kTotalDataLength / 4); |
||||||
|
grpc_slice_buffer_add(&buf, a); |
||||||
|
GPR_ASSERT(buf.count == 1); |
||||||
|
GPR_ASSERT(buf.length == kTotalDataLength); |
||||||
|
grpc_slice_buffer_destroy(&buf); |
||||||
|
} |
||||||
|
|
||||||
|
void test_slice_buffer_move_first() { |
||||||
|
grpc_slice slices[3]; |
||||||
|
grpc_slice_buffer src; |
||||||
|
grpc_slice_buffer dst; |
||||||
|
int idx = 0; |
||||||
|
size_t src_len = 0; |
||||||
|
size_t dst_len = 0; |
||||||
|
|
||||||
|
slices[0] = grpc_slice_from_copied_string("aaa"); |
||||||
|
slices[1] = grpc_slice_from_copied_string("bbbb"); |
||||||
|
slices[2] = grpc_slice_from_copied_string("ccc"); |
||||||
|
|
||||||
|
grpc_slice_buffer_init(&src); |
||||||
|
grpc_slice_buffer_init(&dst); |
||||||
|
for (idx = 0; idx < 3; idx++) { |
||||||
|
grpc_slice_ref(slices[idx]); |
||||||
|
/* For this test, it is important that we add each slice at a new
|
||||||
|
slice index */ |
||||||
|
grpc_slice_buffer_add_indexed(&src, slices[idx]); |
||||||
|
grpc_slice_buffer_add_indexed(&dst, slices[idx]); |
||||||
|
} |
||||||
|
|
||||||
|
/* Case 1: Move more than the first slice's length from src to dst */ |
||||||
|
src_len = src.length; |
||||||
|
dst_len = dst.length; |
||||||
|
grpc_slice_buffer_move_first(&src, 4, &dst); |
||||||
|
src_len -= 4; |
||||||
|
dst_len += 4; |
||||||
|
GPR_ASSERT(src.length == src_len); |
||||||
|
GPR_ASSERT(dst.length == dst_len); |
||||||
|
|
||||||
|
/* src now has two slices ["bbb"] and ["ccc"] */ |
||||||
|
/* Case 2: Move the first slice from src to dst */ |
||||||
|
grpc_slice_buffer_move_first(&src, 3, &dst); |
||||||
|
src_len -= 3; |
||||||
|
dst_len += 3; |
||||||
|
GPR_ASSERT(src.length == src_len); |
||||||
|
GPR_ASSERT(dst.length == dst_len); |
||||||
|
|
||||||
|
/* src now has one slice ["ccc"] */ |
||||||
|
/* Case 3: Move less than the first slice's length from src to dst*/ |
||||||
|
grpc_slice_buffer_move_first(&src, 2, &dst); |
||||||
|
src_len -= 2; |
||||||
|
dst_len += 2; |
||||||
|
GPR_ASSERT(src.length == src_len); |
||||||
|
GPR_ASSERT(dst.length == dst_len); |
||||||
|
} |
||||||
|
|
||||||
|
void test_slice_buffer_first() { |
||||||
|
grpc_slice slices[3]; |
||||||
|
slices[0] = grpc_slice_from_copied_string("aaa"); |
||||||
|
slices[1] = grpc_slice_from_copied_string("bbbb"); |
||||||
|
slices[2] = grpc_slice_from_copied_string("ccccc"); |
||||||
|
|
||||||
|
grpc_slice_buffer buf; |
||||||
|
grpc_slice_buffer_init(&buf); |
||||||
|
for (int idx = 0; idx < 3; ++idx) { |
||||||
|
grpc_slice_ref(slices[idx]); |
||||||
|
grpc_slice_buffer_add_indexed(&buf, slices[idx]); |
||||||
|
} |
||||||
|
|
||||||
|
grpc_slice* first = grpc_slice_buffer_peek_first(&buf); |
||||||
|
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == GPR_SLICE_LENGTH(slices[0])); |
||||||
|
GPR_ASSERT(buf.count == 3); |
||||||
|
GPR_ASSERT(buf.length == 12); |
||||||
|
|
||||||
|
grpc_slice_buffer_sub_first(&buf, 1, 2); |
||||||
|
first = grpc_slice_buffer_peek_first(&buf); |
||||||
|
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == 1); |
||||||
|
GPR_ASSERT(buf.count == 3); |
||||||
|
GPR_ASSERT(buf.length == 10); |
||||||
|
|
||||||
|
grpc_slice_buffer_remove_first(&buf); |
||||||
|
first = grpc_slice_buffer_peek_first(&buf); |
||||||
|
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == GPR_SLICE_LENGTH(slices[1])); |
||||||
|
GPR_ASSERT(buf.count == 2); |
||||||
|
GPR_ASSERT(buf.length == 9); |
||||||
|
|
||||||
|
grpc_slice_buffer_remove_first(&buf); |
||||||
|
first = grpc_slice_buffer_peek_first(&buf); |
||||||
|
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == GPR_SLICE_LENGTH(slices[2])); |
||||||
|
GPR_ASSERT(buf.count == 1); |
||||||
|
GPR_ASSERT(buf.length == 5); |
||||||
|
|
||||||
|
grpc_slice_buffer_remove_first(&buf); |
||||||
|
GPR_ASSERT(buf.count == 0); |
||||||
|
GPR_ASSERT(buf.length == 0); |
||||||
|
} |
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
grpc::testing::TestEnvironment env(&argc, argv); |
||||||
|
grpc_init(); |
||||||
|
|
||||||
|
test_slice_buffer_add(); |
||||||
|
test_slice_buffer_add_contiguous_slices(); |
||||||
|
test_slice_buffer_move_first(); |
||||||
|
test_slice_buffer_first(); |
||||||
|
|
||||||
|
grpc_shutdown(); |
||||||
|
return 0; |
||||||
|
} |
||||||
@ -1,193 +1,81 @@ |
|||||||
/*
|
// Copyright 2022 The gRPC Authors
|
||||||
* |
//
|
||||||
* Copyright 2015 gRPC authors. |
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
* |
// you may not use this file except in compliance with the License.
|
||||||
* Licensed under the Apache License, Version 2.0 (the "License"); |
// You may obtain a copy of the License at
|
||||||
* 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
|
||||||
* |
//
|
||||||
* 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,
|
||||||
* Unless required by applicable law or agreed to in writing, software |
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
* distributed under the License is distributed on an "AS IS" BASIS, |
// See the License for the specific language governing permissions and
|
||||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
// limitations under the License.
|
||||||
* See the License for the specific language governing permissions and |
|
||||||
* limitations under the License. |
#include <grpc/support/port_platform.h> |
||||||
* |
|
||||||
*/ |
#include "src/core/lib/slice/slice_buffer.h" |
||||||
|
|
||||||
|
#include <gmock/gmock.h> |
||||||
|
#include <gtest/gtest.h> |
||||||
|
|
||||||
|
#include <grpc/event_engine/slice_buffer.h> |
||||||
#include <grpc/grpc.h> |
#include <grpc/grpc.h> |
||||||
#include <grpc/slice_buffer.h> |
|
||||||
#include <grpc/support/log.h> |
|
||||||
|
|
||||||
#include "src/core/lib/slice/slice_internal.h" |
#include "src/core/lib/slice/slice.h" |
||||||
#include "test/core/util/test_config.h" |
#include "test/core/util/test_config.h" |
||||||
|
|
||||||
static constexpr size_t kTotalDataLength = 4096; |
using ::grpc_core::Slice; |
||||||
|
using ::grpc_core::SliceBuffer; |
||||||
void test_slice_buffer_add() { |
|
||||||
grpc_slice_buffer buf; |
|
||||||
grpc_slice aaa = grpc_slice_from_copied_string("aaa"); |
|
||||||
grpc_slice bb = grpc_slice_from_copied_string("bb"); |
|
||||||
size_t i; |
|
||||||
|
|
||||||
grpc_slice_buffer_init(&buf); |
static constexpr int kNewSliceLength = 100; |
||||||
for (i = 0; i < 10; i++) { |
|
||||||
grpc_slice_ref(aaa); |
|
||||||
grpc_slice_ref(bb); |
|
||||||
grpc_slice_buffer_add(&buf, aaa); |
|
||||||
grpc_slice_buffer_add(&buf, bb); |
|
||||||
} |
|
||||||
GPR_ASSERT(buf.count > 0); |
|
||||||
GPR_ASSERT(buf.length == 50); |
|
||||||
grpc_slice_buffer_reset_and_unref(&buf); |
|
||||||
GPR_ASSERT(buf.count == 0); |
|
||||||
GPR_ASSERT(buf.length == 0); |
|
||||||
for (i = 0; i < 10; i++) { |
|
||||||
grpc_slice_ref(aaa); |
|
||||||
grpc_slice_ref(bb); |
|
||||||
grpc_slice_buffer_add(&buf, aaa); |
|
||||||
grpc_slice_buffer_add(&buf, bb); |
|
||||||
} |
|
||||||
GPR_ASSERT(buf.count > 0); |
|
||||||
GPR_ASSERT(buf.length == 50); |
|
||||||
for (i = 0; i < 10; i++) { |
|
||||||
grpc_slice_buffer_pop(&buf); |
|
||||||
grpc_slice_unref(aaa); |
|
||||||
grpc_slice_unref(bb); |
|
||||||
} |
|
||||||
GPR_ASSERT(buf.count == 0); |
|
||||||
GPR_ASSERT(buf.length == 0); |
|
||||||
grpc_slice_buffer_destroy(&buf); |
|
||||||
} |
|
||||||
|
|
||||||
static void free_data(void* data, size_t len) { |
|
||||||
GPR_ASSERT(len == kTotalDataLength); |
|
||||||
gpr_free(data); |
|
||||||
} |
|
||||||
|
|
||||||
void test_slice_buffer_add_contiguous_slices() { |
Slice MakeSlice(size_t len) { |
||||||
grpc_slice_buffer buf; |
GPR_ASSERT(len > 0); |
||||||
grpc_slice_buffer_init(&buf); |
unsigned char* contents = reinterpret_cast<unsigned char*>(new char[len]); |
||||||
char* data = reinterpret_cast<char*>(gpr_malloc(kTotalDataLength)); |
memset(contents, 'a', len); |
||||||
GPR_ASSERT(data != nullptr); |
return Slice(grpc_slice_new(contents, len, gpr_free)); |
||||||
grpc_slice a = grpc_slice_new_with_len(data, kTotalDataLength, free_data); |
|
||||||
grpc_slice s1 = grpc_slice_split_head(&a, kTotalDataLength / 4); |
|
||||||
grpc_slice s2 = grpc_slice_split_head(&a, kTotalDataLength / 4); |
|
||||||
grpc_slice s3 = grpc_slice_split_head(&a, kTotalDataLength / 4); |
|
||||||
grpc_slice_buffer_add(&buf, s1); |
|
||||||
GPR_ASSERT(buf.count == 1); |
|
||||||
GPR_ASSERT(buf.length == kTotalDataLength / 4); |
|
||||||
grpc_slice_buffer_add(&buf, s2); |
|
||||||
GPR_ASSERT(buf.count == 1); |
|
||||||
GPR_ASSERT(buf.length == kTotalDataLength / 2); |
|
||||||
grpc_slice_buffer_add(&buf, s3); |
|
||||||
GPR_ASSERT(buf.count == 1); |
|
||||||
GPR_ASSERT(buf.length == 3 * kTotalDataLength / 4); |
|
||||||
grpc_slice_buffer_add(&buf, a); |
|
||||||
GPR_ASSERT(buf.count == 1); |
|
||||||
GPR_ASSERT(buf.length == kTotalDataLength); |
|
||||||
grpc_slice_buffer_destroy(&buf); |
|
||||||
} |
} |
||||||
|
|
||||||
void test_slice_buffer_move_first() { |
TEST(SliceBufferTest, AddAndRemoveTest) { |
||||||
grpc_slice slices[3]; |
SliceBuffer sb; |
||||||
grpc_slice_buffer src; |
Slice first_slice = MakeSlice(kNewSliceLength); |
||||||
grpc_slice_buffer dst; |
Slice second_slice = MakeSlice(kNewSliceLength); |
||||||
int idx = 0; |
Slice first_slice_copy = first_slice.Copy(); |
||||||
size_t src_len = 0; |
sb.Append(std::move(first_slice)); |
||||||
size_t dst_len = 0; |
sb.Append(std::move(second_slice)); |
||||||
|
ASSERT_EQ(sb.Count(), 2); |
||||||
slices[0] = grpc_slice_from_copied_string("aaa"); |
ASSERT_EQ(sb.Length(), 2 * kNewSliceLength); |
||||||
slices[1] = grpc_slice_from_copied_string("bbbb"); |
Slice popped = sb.TakeFirst(); |
||||||
slices[2] = grpc_slice_from_copied_string("ccc"); |
ASSERT_EQ(popped, first_slice_copy); |
||||||
|
ASSERT_EQ(sb.Count(), 1); |
||||||
grpc_slice_buffer_init(&src); |
ASSERT_EQ(sb.Length(), kNewSliceLength); |
||||||
grpc_slice_buffer_init(&dst); |
sb.Prepend(std::move(popped)); |
||||||
for (idx = 0; idx < 3; idx++) { |
ASSERT_EQ(sb.Count(), 2); |
||||||
grpc_slice_ref(slices[idx]); |
ASSERT_EQ(sb.Length(), 2 * kNewSliceLength); |
||||||
/* For this test, it is important that we add each slice at a new
|
sb.Clear(); |
||||||
slice index */ |
ASSERT_EQ(sb.Count(), 0); |
||||||
grpc_slice_buffer_add_indexed(&src, slices[idx]); |
ASSERT_EQ(sb.Length(), 0); |
||||||
grpc_slice_buffer_add_indexed(&dst, slices[idx]); |
|
||||||
} |
|
||||||
|
|
||||||
/* Case 1: Move more than the first slice's length from src to dst */ |
|
||||||
src_len = src.length; |
|
||||||
dst_len = dst.length; |
|
||||||
grpc_slice_buffer_move_first(&src, 4, &dst); |
|
||||||
src_len -= 4; |
|
||||||
dst_len += 4; |
|
||||||
GPR_ASSERT(src.length == src_len); |
|
||||||
GPR_ASSERT(dst.length == dst_len); |
|
||||||
|
|
||||||
/* src now has two slices ["bbb"] and ["ccc"] */ |
|
||||||
/* Case 2: Move the first slice from src to dst */ |
|
||||||
grpc_slice_buffer_move_first(&src, 3, &dst); |
|
||||||
src_len -= 3; |
|
||||||
dst_len += 3; |
|
||||||
GPR_ASSERT(src.length == src_len); |
|
||||||
GPR_ASSERT(dst.length == dst_len); |
|
||||||
|
|
||||||
/* src now has one slice ["ccc"] */ |
|
||||||
/* Case 3: Move less than the first slice's length from src to dst*/ |
|
||||||
grpc_slice_buffer_move_first(&src, 2, &dst); |
|
||||||
src_len -= 2; |
|
||||||
dst_len += 2; |
|
||||||
GPR_ASSERT(src.length == src_len); |
|
||||||
GPR_ASSERT(dst.length == dst_len); |
|
||||||
} |
} |
||||||
|
|
||||||
void test_slice_buffer_first() { |
TEST(SliceBufferTest, SliceRefTest) { |
||||||
grpc_slice slices[3]; |
SliceBuffer sb; |
||||||
slices[0] = grpc_slice_from_copied_string("aaa"); |
Slice first_slice = MakeSlice(kNewSliceLength); |
||||||
slices[1] = grpc_slice_from_copied_string("bbbb"); |
Slice second_slice = MakeSlice(kNewSliceLength + 1); |
||||||
slices[2] = grpc_slice_from_copied_string("ccccc"); |
Slice first_slice_copy = first_slice.Copy(); |
||||||
|
Slice second_slice_copy = second_slice.Copy(); |
||||||
grpc_slice_buffer buf; |
ASSERT_EQ(sb.AppendIndexed(std::move(first_slice)), 0); |
||||||
grpc_slice_buffer_init(&buf); |
ASSERT_EQ(sb.AppendIndexed(std::move(second_slice)), 1); |
||||||
for (int idx = 0; idx < 3; ++idx) { |
Slice first_reffed = sb.RefSlice(0); |
||||||
grpc_slice_ref(slices[idx]); |
Slice second_reffed = sb.RefSlice(1); |
||||||
grpc_slice_buffer_add_indexed(&buf, slices[idx]); |
ASSERT_EQ(first_reffed, first_slice_copy); |
||||||
} |
ASSERT_EQ(second_reffed, second_slice_copy); |
||||||
|
ASSERT_EQ(sb.Count(), 2); |
||||||
grpc_slice* first = grpc_slice_buffer_peek_first(&buf); |
ASSERT_EQ(sb.Length(), 2 * kNewSliceLength + 1); |
||||||
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == GPR_SLICE_LENGTH(slices[0])); |
sb.Clear(); |
||||||
GPR_ASSERT(buf.count == 3); |
|
||||||
GPR_ASSERT(buf.length == 12); |
|
||||||
|
|
||||||
grpc_slice_buffer_sub_first(&buf, 1, 2); |
|
||||||
first = grpc_slice_buffer_peek_first(&buf); |
|
||||||
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == 1); |
|
||||||
GPR_ASSERT(buf.count == 3); |
|
||||||
GPR_ASSERT(buf.length == 10); |
|
||||||
|
|
||||||
grpc_slice_buffer_remove_first(&buf); |
|
||||||
first = grpc_slice_buffer_peek_first(&buf); |
|
||||||
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == GPR_SLICE_LENGTH(slices[1])); |
|
||||||
GPR_ASSERT(buf.count == 2); |
|
||||||
GPR_ASSERT(buf.length == 9); |
|
||||||
|
|
||||||
grpc_slice_buffer_remove_first(&buf); |
|
||||||
first = grpc_slice_buffer_peek_first(&buf); |
|
||||||
GPR_ASSERT(GPR_SLICE_LENGTH(*first) == GPR_SLICE_LENGTH(slices[2])); |
|
||||||
GPR_ASSERT(buf.count == 1); |
|
||||||
GPR_ASSERT(buf.length == 5); |
|
||||||
|
|
||||||
grpc_slice_buffer_remove_first(&buf); |
|
||||||
GPR_ASSERT(buf.count == 0); |
|
||||||
GPR_ASSERT(buf.length == 0); |
|
||||||
} |
} |
||||||
|
|
||||||
int main(int argc, char** argv) { |
int main(int argc, char** argv) { |
||||||
grpc::testing::TestEnvironment env(&argc, argv); |
testing::InitGoogleTest(&argc, argv); |
||||||
grpc_init(); |
return RUN_ALL_TESTS(); |
||||||
|
|
||||||
test_slice_buffer_add(); |
|
||||||
test_slice_buffer_add_contiguous_slices(); |
|
||||||
test_slice_buffer_move_first(); |
|
||||||
test_slice_buffer_first(); |
|
||||||
|
|
||||||
grpc_shutdown(); |
|
||||||
return 0; |
|
||||||
} |
} |
||||||
|
|||||||
Loading…
Reference in new issue