aboutsummaryrefslogtreecommitdiffhomepage
path: root/test/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'test/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector_test.cc')
-rw-r--r--test/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector_test.cc289
1 files changed, 0 insertions, 289 deletions
diff --git a/test/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector_test.cc b/test/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector_test.cc
deleted file mode 100644
index 2388be95cd..0000000000
--- a/test/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector_test.cc
+++ /dev/null
@@ -1,289 +0,0 @@
-/*
- *
- * Copyright 2018 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/slice_buffer.h>
-#include <grpc/support/alloc.h>
-#include <grpc/support/log.h>
-
-#include "src/core/lib/slice/slice_internal.h"
-#include "src/core/tsi/alts/crypt/gsec.h"
-#include "src/core/tsi/alts/zero_copy_frame_protector/alts_zero_copy_grpc_protector.h"
-#include "src/core/tsi/transport_security_grpc.h"
-#include "test/core/tsi/alts/crypt/gsec_test_util.h"
-
-/* TODO: tests zero_copy_grpc_protector under TSI test library, which
- * has more comprehensive tests. */
-
-constexpr size_t kSealRepeatTimes = 50;
-constexpr size_t kSmallBufferSize = 16;
-constexpr size_t kLargeBufferSize = 16384;
-constexpr size_t kChannelMaxSize = 2048;
-constexpr size_t kChannelMinSize = 128;
-
-/* Test fixtures for each test cases. */
-struct alts_zero_copy_grpc_protector_test_fixture {
- tsi_zero_copy_grpc_protector* client;
- tsi_zero_copy_grpc_protector* server;
-};
-
-/* Test input variables for protect/unprotect operations. */
-struct alts_zero_copy_grpc_protector_test_var {
- grpc_slice_buffer original_sb;
- grpc_slice_buffer duplicate_sb;
- grpc_slice_buffer staging_sb;
- grpc_slice_buffer protected_sb;
- grpc_slice_buffer unprotected_sb;
-};
-
-/* --- Test utility functions. --- */
-
-static void create_random_slice_buffer(grpc_slice_buffer* sb,
- grpc_slice_buffer* dup_sb,
- size_t length) {
- GPR_ASSERT(sb != nullptr);
- GPR_ASSERT(dup_sb != nullptr);
- GPR_ASSERT(length > 0);
- grpc_slice slice = GRPC_SLICE_MALLOC(length);
- gsec_test_random_bytes(GRPC_SLICE_START_PTR(slice), length);
- grpc_slice_buffer_add(sb, grpc_slice_ref(slice));
- grpc_slice_buffer_add(dup_sb, slice);
-}
-
-static uint8_t* pointer_to_nth_byte(grpc_slice_buffer* sb, size_t index) {
- GPR_ASSERT(sb != nullptr);
- GPR_ASSERT(index < sb->length);
- for (size_t i = 0; i < sb->count; i++) {
- if (index < GRPC_SLICE_LENGTH(sb->slices[i])) {
- return GRPC_SLICE_START_PTR(sb->slices[i]) + index;
- } else {
- index -= GRPC_SLICE_LENGTH(sb->slices[i]);
- }
- }
- return nullptr;
-}
-
-/* Checks if two slice buffer contents are the same. It is not super efficient,
- * but OK for testing. */
-static bool are_slice_buffers_equal(grpc_slice_buffer* first,
- grpc_slice_buffer* second) {
- GPR_ASSERT(first != nullptr);
- GPR_ASSERT(second != nullptr);
- if (first->length != second->length) {
- return false;
- }
- for (size_t i = 0; i < first->length; i++) {
- uint8_t* first_ptr = pointer_to_nth_byte(first, i);
- uint8_t* second_ptr = pointer_to_nth_byte(second, i);
- GPR_ASSERT(first_ptr != nullptr && second_ptr != nullptr);
- if ((*first_ptr) != (*second_ptr)) {
- return false;
- }
- }
- return true;
-}
-
-static alts_zero_copy_grpc_protector_test_fixture*
-alts_zero_copy_grpc_protector_test_fixture_create(bool rekey,
- bool integrity_only) {
- alts_zero_copy_grpc_protector_test_fixture* fixture =
- static_cast<alts_zero_copy_grpc_protector_test_fixture*>(
- gpr_zalloc(sizeof(alts_zero_copy_grpc_protector_test_fixture)));
- grpc_core::ExecCtx exec_ctx;
- size_t key_length = rekey ? kAes128GcmRekeyKeyLength : kAes128GcmKeyLength;
- uint8_t* key;
- size_t max_protected_frame_size = 1024;
- gsec_test_random_array(&key, key_length);
- GPR_ASSERT(alts_zero_copy_grpc_protector_create(
- key, key_length, rekey, /*is_client=*/true, integrity_only,
- &max_protected_frame_size, &fixture->client) == TSI_OK);
- GPR_ASSERT(alts_zero_copy_grpc_protector_create(
- key, key_length, rekey, /*is_client=*/false, integrity_only,
- &max_protected_frame_size, &fixture->server) == TSI_OK);
- gpr_free(key);
- grpc_core::ExecCtx::Get()->Flush();
- return fixture;
-}
-
-static void alts_zero_copy_grpc_protector_test_fixture_destroy(
- alts_zero_copy_grpc_protector_test_fixture* fixture) {
- if (fixture == nullptr) {
- return;
- }
- grpc_core::ExecCtx exec_ctx;
- tsi_zero_copy_grpc_protector_destroy(fixture->client);
- tsi_zero_copy_grpc_protector_destroy(fixture->server);
- grpc_core::ExecCtx::Get()->Flush();
- gpr_free(fixture);
-}
-
-static alts_zero_copy_grpc_protector_test_var*
-alts_zero_copy_grpc_protector_test_var_create() {
- alts_zero_copy_grpc_protector_test_var* var =
- static_cast<alts_zero_copy_grpc_protector_test_var*>(
- gpr_zalloc(sizeof(alts_zero_copy_grpc_protector_test_var)));
- grpc_slice_buffer_init(&var->original_sb);
- grpc_slice_buffer_init(&var->duplicate_sb);
- grpc_slice_buffer_init(&var->staging_sb);
- grpc_slice_buffer_init(&var->protected_sb);
- grpc_slice_buffer_init(&var->unprotected_sb);
- return var;
-}
-
-static void alts_zero_copy_grpc_protector_test_var_destroy(
- alts_zero_copy_grpc_protector_test_var* var) {
- if (var == nullptr) {
- return;
- }
- grpc_slice_buffer_destroy_internal(&var->original_sb);
- grpc_slice_buffer_destroy_internal(&var->duplicate_sb);
- grpc_slice_buffer_destroy_internal(&var->staging_sb);
- grpc_slice_buffer_destroy_internal(&var->protected_sb);
- grpc_slice_buffer_destroy_internal(&var->unprotected_sb);
- gpr_free(var);
-}
-
-/* --- ALTS zero-copy protector tests. --- */
-
-static void seal_unseal_small_buffer(tsi_zero_copy_grpc_protector* sender,
- tsi_zero_copy_grpc_protector* receiver) {
- grpc_core::ExecCtx exec_ctx;
- for (size_t i = 0; i < kSealRepeatTimes; i++) {
- alts_zero_copy_grpc_protector_test_var* var =
- alts_zero_copy_grpc_protector_test_var_create();
- /* Creates a random small slice buffer and calls protect(). */
- create_random_slice_buffer(&var->original_sb, &var->duplicate_sb,
- kSmallBufferSize);
- GPR_ASSERT(tsi_zero_copy_grpc_protector_protect(
- sender, &var->original_sb, &var->protected_sb) == TSI_OK);
- /* Splits protected slice buffer into two: first one is staging_sb, and
- * second one is is protected_sb. */
- uint32_t staging_sb_size =
- gsec_test_bias_random_uint32(
- static_cast<uint32_t>(var->protected_sb.length - 1)) +
- 1;
- grpc_slice_buffer_move_first(&var->protected_sb, staging_sb_size,
- &var->staging_sb);
- /* Unprotects one by one. */
- GPR_ASSERT(tsi_zero_copy_grpc_protector_unprotect(
- receiver, &var->staging_sb, &var->unprotected_sb) == TSI_OK);
- GPR_ASSERT(var->unprotected_sb.length == 0);
- GPR_ASSERT(tsi_zero_copy_grpc_protector_unprotect(
- receiver, &var->protected_sb, &var->unprotected_sb) ==
- TSI_OK);
- GPR_ASSERT(
- are_slice_buffers_equal(&var->unprotected_sb, &var->duplicate_sb));
- alts_zero_copy_grpc_protector_test_var_destroy(var);
- }
- grpc_core::ExecCtx::Get()->Flush();
-}
-
-static void seal_unseal_large_buffer(tsi_zero_copy_grpc_protector* sender,
- tsi_zero_copy_grpc_protector* receiver) {
- grpc_core::ExecCtx exec_ctx;
- for (size_t i = 0; i < kSealRepeatTimes; i++) {
- alts_zero_copy_grpc_protector_test_var* var =
- alts_zero_copy_grpc_protector_test_var_create();
- /* Creates a random large slice buffer and calls protect(). */
- create_random_slice_buffer(&var->original_sb, &var->duplicate_sb,
- kLargeBufferSize);
- GPR_ASSERT(tsi_zero_copy_grpc_protector_protect(
- sender, &var->original_sb, &var->protected_sb) == TSI_OK);
- /* Splits protected slice buffer into multiple pieces. Receiver unprotects
- * each slice buffer one by one. */
- uint32_t channel_size = gsec_test_bias_random_uint32(static_cast<uint32_t>(
- kChannelMaxSize + 1 - kChannelMinSize)) +
- static_cast<uint32_t>(kChannelMinSize);
- while (var->protected_sb.length > channel_size) {
- grpc_slice_buffer_reset_and_unref_internal(&var->staging_sb);
- grpc_slice_buffer_move_first(&var->protected_sb, channel_size,
- &var->staging_sb);
- GPR_ASSERT(tsi_zero_copy_grpc_protector_unprotect(
- receiver, &var->staging_sb, &var->unprotected_sb) ==
- TSI_OK);
- }
- GPR_ASSERT(tsi_zero_copy_grpc_protector_unprotect(
- receiver, &var->protected_sb, &var->unprotected_sb) ==
- TSI_OK);
- GPR_ASSERT(
- are_slice_buffers_equal(&var->unprotected_sb, &var->duplicate_sb));
- alts_zero_copy_grpc_protector_test_var_destroy(var);
- }
- grpc_core::ExecCtx::Get()->Flush();
-}
-
-/* --- Test cases. --- */
-
-static void alts_zero_copy_protector_seal_unseal_small_buffer_tests() {
- alts_zero_copy_grpc_protector_test_fixture* fixture =
- alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/false, /*integrity_only=*/true);
- seal_unseal_small_buffer(fixture->client, fixture->server);
- seal_unseal_small_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-
- fixture = alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/false, /*integrity_only=*/false);
- seal_unseal_small_buffer(fixture->client, fixture->server);
- seal_unseal_small_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-
- fixture = alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/true, /*integrity_only=*/true);
- seal_unseal_small_buffer(fixture->client, fixture->server);
- seal_unseal_small_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-
- fixture = alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/true, /*integrity_only=*/false);
- seal_unseal_small_buffer(fixture->client, fixture->server);
- seal_unseal_small_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-}
-
-static void alts_zero_copy_protector_seal_unseal_large_buffer_tests() {
- alts_zero_copy_grpc_protector_test_fixture* fixture =
- alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/false, /*integrity_only=*/true);
- seal_unseal_large_buffer(fixture->client, fixture->server);
- seal_unseal_large_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-
- fixture = alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/false, /*integrity_only=*/false);
- seal_unseal_large_buffer(fixture->client, fixture->server);
- seal_unseal_large_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-
- fixture = alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/true, /*integrity_only=*/true);
- seal_unseal_large_buffer(fixture->client, fixture->server);
- seal_unseal_large_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-
- fixture = alts_zero_copy_grpc_protector_test_fixture_create(
- /*rekey=*/true, /*integrity_only=*/false);
- seal_unseal_large_buffer(fixture->client, fixture->server);
- seal_unseal_large_buffer(fixture->server, fixture->client);
- alts_zero_copy_grpc_protector_test_fixture_destroy(fixture);
-}
-
-int main(int argc, char** argv) {
- alts_zero_copy_protector_seal_unseal_small_buffer_tests();
- alts_zero_copy_protector_seal_unseal_large_buffer_tests();
- return 0;
-}
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-25 01:13:53 +0000