mirror of https://github.com/grpc/grpc.git
Jiangtao Li
7 years ago
committed by
GitHub
parent
9b5da78be9
commit
0211cfb277
22 changed files with 287 additions and 69 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_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_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