mirror of https://github.com/grpc/grpc.git
Merge pull request #12105 from jiangtaoli2016/revert-12101-revert-11977-tsi_grpc
Roll forward "Add TSI zero-copy frame protector"pull/12267/head
Mark D. Roth
8 years ago
committed by
GitHub
commit
e7aa35a51c
22 changed files with 234 additions and 53 deletions
@ -0,0 +1,64 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2017 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 "src/core/tsi/transport_security_grpc.h" |
||||||
|
|
||||||
|
/* This method creates a tsi_zero_copy_grpc_protector object. */ |
||||||
|
tsi_result tsi_handshaker_result_create_zero_copy_grpc_protector( |
||||||
|
const tsi_handshaker_result *self, size_t *max_output_protected_frame_size, |
||||||
|
tsi_zero_copy_grpc_protector **protector) { |
||||||
|
if (self == NULL || self->vtable == NULL || protector == NULL) { |
||||||
|
return TSI_INVALID_ARGUMENT; |
||||||
|
} |
||||||
|
if (self->vtable->create_zero_copy_grpc_protector == NULL) { |
||||||
|
return TSI_UNIMPLEMENTED; |
||||||
|
} |
||||||
|
return self->vtable->create_zero_copy_grpc_protector( |
||||||
|
self, max_output_protected_frame_size, protector); |
||||||
|
} |
||||||
|
|
||||||
|
/* --- tsi_zero_copy_grpc_protector common implementation. ---
|
||||||
|
|
||||||
|
Calls specific implementation after state/input validation. */ |
||||||
|
|
||||||
|
tsi_result tsi_zero_copy_grpc_protector_protect( |
||||||
|
tsi_zero_copy_grpc_protector *self, grpc_slice_buffer *unprotected_slices, |
||||||
|
grpc_slice_buffer *protected_slices) { |
||||||
|
if (self == NULL || self->vtable == NULL || unprotected_slices == NULL || |
||||||
|
protected_slices == NULL) { |
||||||
|
return TSI_INVALID_ARGUMENT; |
||||||
|
} |
||||||
|
if (self->vtable->protect == NULL) return TSI_UNIMPLEMENTED; |
||||||
|
return self->vtable->protect(self, unprotected_slices, protected_slices); |
||||||
|
} |
||||||
|
|
||||||
|
tsi_result tsi_zero_copy_grpc_protector_unprotect( |
||||||
|
tsi_zero_copy_grpc_protector *self, grpc_slice_buffer *protected_slices, |
||||||
|
grpc_slice_buffer *unprotected_slices) { |
||||||
|
if (self == NULL || self->vtable == NULL || protected_slices == NULL || |
||||||
|
unprotected_slices == NULL) { |
||||||
|
return TSI_INVALID_ARGUMENT; |
||||||
|
} |
||||||
|
if (self->vtable->unprotect == NULL) return TSI_UNIMPLEMENTED; |
||||||
|
return self->vtable->unprotect(self, protected_slices, unprotected_slices); |
||||||
|
} |
||||||
|
|
||||||
|
void tsi_zero_copy_grpc_protector_destroy(tsi_zero_copy_grpc_protector *self) { |
||||||
|
if (self == NULL) return; |
||||||
|
self->vtable->destroy(self); |
||||||
|
} |
||||||
@ -0,0 +1,80 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2017 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_TSI_TRANSPORT_SECURITY_GRPC_H |
||||||
|
#define GRPC_CORE_TSI_TRANSPORT_SECURITY_GRPC_H |
||||||
|
|
||||||
|
#include <grpc/slice_buffer.h> |
||||||
|
#include "src/core/tsi/transport_security.h" |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
extern "C" { |
||||||
|
#endif |
||||||
|
|
||||||
|
/* This method creates a tsi_zero_copy_grpc_protector object. It return TSI_OK
|
||||||
|
assuming there is no fatal error. |
||||||
|
The caller is responsible for destroying the protector. */ |
||||||
|
tsi_result tsi_handshaker_result_create_zero_copy_grpc_protector( |
||||||
|
const tsi_handshaker_result *self, size_t *max_output_protected_frame_size, |
||||||
|
tsi_zero_copy_grpc_protector **protector); |
||||||
|
|
||||||
|
/* -- tsi_zero_copy_grpc_protector object -- */ |
||||||
|
|
||||||
|
/* Outputs protected frames.
|
||||||
|
- unprotected_slices is the unprotected data to be protected. |
||||||
|
- protected_slices is the protected output frames. One or more frames |
||||||
|
may be produced in this protect function. |
||||||
|
- This method returns TSI_OK in case of success or a specific error code in |
||||||
|
case of failure. */ |
||||||
|
tsi_result tsi_zero_copy_grpc_protector_protect( |
||||||
|
tsi_zero_copy_grpc_protector *self, grpc_slice_buffer *unprotected_slices, |
||||||
|
grpc_slice_buffer *protected_slices); |
||||||
|
|
||||||
|
/* Outputs unprotected bytes.
|
||||||
|
- protected_slices is the bytes of protected frames. |
||||||
|
- unprotected_slices is the unprotected output data. |
||||||
|
- This method returns TSI_OK in case of success. Success includes cases where |
||||||
|
there is not enough data to output in which case unprotected_slices has 0 |
||||||
|
bytes. */ |
||||||
|
tsi_result tsi_zero_copy_grpc_protector_unprotect( |
||||||
|
tsi_zero_copy_grpc_protector *self, grpc_slice_buffer *protected_slices, |
||||||
|
grpc_slice_buffer *unprotected_slices); |
||||||
|
|
||||||
|
/* Destroys the tsi_zero_copy_grpc_protector object. */ |
||||||
|
void tsi_zero_copy_grpc_protector_destroy(tsi_zero_copy_grpc_protector *self); |
||||||
|
|
||||||
|
/* Base for tsi_zero_copy_grpc_protector implementations. */ |
||||||
|
typedef struct { |
||||||
|
tsi_result (*protect)(tsi_zero_copy_grpc_protector *self, |
||||||
|
grpc_slice_buffer *unprotected_slices, |
||||||
|
grpc_slice_buffer *protected_slices); |
||||||
|
tsi_result (*unprotect)(tsi_zero_copy_grpc_protector *self, |
||||||
|
grpc_slice_buffer *protected_slices, |
||||||
|
grpc_slice_buffer *unprotected_slices); |
||||||
|
void (*destroy)(tsi_zero_copy_grpc_protector *self); |
||||||
|
} tsi_zero_copy_grpc_protector_vtable; |
||||||
|
|
||||||
|
struct tsi_zero_copy_grpc_protector { |
||||||
|
const tsi_zero_copy_grpc_protector_vtable *vtable; |
||||||
|
}; |
||||||
|
|
||||||
|
#ifdef __cplusplus |
||||||
|
} |
||||||
|
#endif |
||||||
|
|
||||||
|
#endif /* GRPC_CORE_TSI_TRANSPORT_SECURITY_GRPC_H */ |
||||||
Loading…
Reference in new issue