The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
771 lines
26 KiB
771 lines
26 KiB
/* |
|
* |
|
* Copyright 2015, Google Inc. |
|
* All rights reserved. |
|
* |
|
* Redistribution and use in source and binary forms, with or without |
|
* modification, are permitted provided that the following conditions are |
|
* met: |
|
* |
|
* * Redistributions of source code must retain the above copyright |
|
* notice, this list of conditions and the following disclaimer. |
|
* * Redistributions in binary form must reproduce the above |
|
* copyright notice, this list of conditions and the following disclaimer |
|
* in the documentation and/or other materials provided with the |
|
* distribution. |
|
* * Neither the name of Google Inc. nor the names of its |
|
* contributors may be used to endorse or promote products derived from |
|
* this software without specific prior written permission. |
|
* |
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
* |
|
*/ |
|
|
|
#include "src/core/support/string.h" |
|
|
|
#include <grpc/byte_buffer_reader.h> |
|
#include <grpc/support/port_platform.h> |
|
#include <grpc/support/alloc.h> |
|
#include <grpc/support/log.h> |
|
#include <grpc/support/slice.h> |
|
#include <grpc/support/string_util.h> |
|
#include <grpc/support/thd.h> |
|
#include <grpc/grpc.h> |
|
#include <grpc/grpc_security.h> |
|
|
|
#include <string.h> |
|
|
|
#ifdef GPR_WIN32 |
|
#define GPR_EXPORT __declspec(dllexport) |
|
#define GPR_CALLTYPE __stdcall |
|
#endif |
|
|
|
#ifndef GPR_EXPORT |
|
#define GPR_EXPORT |
|
#endif |
|
|
|
#ifndef GPR_CALLTYPE |
|
#define GPR_CALLTYPE |
|
#endif |
|
|
|
grpc_byte_buffer *string_to_byte_buffer(const char *buffer, size_t len) { |
|
gpr_slice slice = gpr_slice_from_copied_buffer(buffer, len); |
|
grpc_byte_buffer *bb = grpc_raw_byte_buffer_create(&slice, 1); |
|
gpr_slice_unref(slice); |
|
return bb; |
|
} |
|
|
|
/* |
|
* Helper to maintain lifetime of batch op inputs and store batch op outputs. |
|
*/ |
|
typedef struct gprcsharp_batch_context { |
|
grpc_metadata_array send_initial_metadata; |
|
grpc_byte_buffer *send_message; |
|
struct { |
|
grpc_metadata_array trailing_metadata; |
|
char *status_details; |
|
} send_status_from_server; |
|
grpc_metadata_array recv_initial_metadata; |
|
grpc_byte_buffer *recv_message; |
|
struct { |
|
grpc_metadata_array trailing_metadata; |
|
grpc_status_code status; |
|
char *status_details; |
|
size_t status_details_capacity; |
|
} recv_status_on_client; |
|
int recv_close_on_server_cancelled; |
|
struct { |
|
grpc_call *call; |
|
grpc_call_details call_details; |
|
grpc_metadata_array request_metadata; |
|
} server_rpc_new; |
|
} grpcsharp_batch_context; |
|
|
|
GPR_EXPORT grpcsharp_batch_context *GPR_CALLTYPE grpcsharp_batch_context_create() { |
|
grpcsharp_batch_context *ctx = gpr_malloc(sizeof(grpcsharp_batch_context)); |
|
memset(ctx, 0, sizeof(grpcsharp_batch_context)); |
|
return ctx; |
|
} |
|
|
|
/* |
|
* Destroys array->metadata. |
|
* The array pointer itself is not freed. |
|
*/ |
|
void grpcsharp_metadata_array_destroy_metadata_only( |
|
grpc_metadata_array *array) { |
|
gpr_free(array->metadata); |
|
} |
|
|
|
/* |
|
* Destroys keys, values and array->metadata. |
|
* The array pointer itself is not freed. |
|
*/ |
|
void grpcsharp_metadata_array_destroy_metadata_including_entries( |
|
grpc_metadata_array *array) { |
|
size_t i; |
|
if (array->metadata) { |
|
for (i = 0; i < array->count; i++) { |
|
gpr_free((void *)array->metadata[i].key); |
|
gpr_free((void *)array->metadata[i].value); |
|
} |
|
} |
|
gpr_free(array->metadata); |
|
} |
|
|
|
/* |
|
* Fully destroys the metadata array. |
|
*/ |
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_metadata_array_destroy_full(grpc_metadata_array *array) { |
|
if (!array) { |
|
return; |
|
} |
|
grpcsharp_metadata_array_destroy_metadata_including_entries(array); |
|
gpr_free(array); |
|
} |
|
|
|
/* |
|
* Creates an empty metadata array with given capacity. |
|
* Array can later be destroyed by grpc_metadata_array_destroy_full. |
|
*/ |
|
GPR_EXPORT grpc_metadata_array *GPR_CALLTYPE |
|
grpcsharp_metadata_array_create(size_t capacity) { |
|
grpc_metadata_array *array = |
|
(grpc_metadata_array *)gpr_malloc(sizeof(grpc_metadata_array)); |
|
grpc_metadata_array_init(array); |
|
array->capacity = capacity; |
|
array->count = 0; |
|
if (capacity > 0) { |
|
array->metadata = |
|
(grpc_metadata *)gpr_malloc(sizeof(grpc_metadata) * capacity); |
|
memset(array->metadata, 0, sizeof(grpc_metadata) * capacity); |
|
} else { |
|
array->metadata = NULL; |
|
} |
|
return array; |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_metadata_array_add(grpc_metadata_array *array, const char *key, |
|
const char *value, size_t value_length) { |
|
size_t i = array->count; |
|
GPR_ASSERT(array->count < array->capacity); |
|
array->metadata[i].key = gpr_strdup(key); |
|
array->metadata[i].value = (char *)gpr_malloc(value_length); |
|
memcpy((void *)array->metadata[i].value, value, value_length); |
|
array->metadata[i].value_length = value_length; |
|
array->count++; |
|
} |
|
|
|
/* Move contents of metadata array */ |
|
void grpcsharp_metadata_array_move(grpc_metadata_array *dest, |
|
grpc_metadata_array *src) { |
|
if (!src) { |
|
dest->capacity = 0; |
|
dest->count = 0; |
|
dest->metadata = NULL; |
|
return; |
|
} |
|
|
|
dest->capacity = src->capacity; |
|
dest->count = src->count; |
|
dest->metadata = src->metadata; |
|
|
|
src->capacity = 0; |
|
src->count = 0; |
|
src->metadata = NULL; |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_batch_context_destroy(grpcsharp_batch_context *ctx) { |
|
if (!ctx) { |
|
return; |
|
} |
|
grpcsharp_metadata_array_destroy_metadata_including_entries( |
|
&(ctx->send_initial_metadata)); |
|
|
|
grpc_byte_buffer_destroy(ctx->send_message); |
|
|
|
grpcsharp_metadata_array_destroy_metadata_including_entries( |
|
&(ctx->send_status_from_server.trailing_metadata)); |
|
gpr_free(ctx->send_status_from_server.status_details); |
|
|
|
grpcsharp_metadata_array_destroy_metadata_only(&(ctx->recv_initial_metadata)); |
|
|
|
grpc_byte_buffer_destroy(ctx->recv_message); |
|
|
|
grpcsharp_metadata_array_destroy_metadata_only( |
|
&(ctx->recv_status_on_client.trailing_metadata)); |
|
gpr_free((void *)ctx->recv_status_on_client.status_details); |
|
|
|
/* NOTE: ctx->server_rpc_new.call is not destroyed because callback handler is |
|
supposed |
|
to take its ownership. */ |
|
|
|
grpc_call_details_destroy(&(ctx->server_rpc_new.call_details)); |
|
grpcsharp_metadata_array_destroy_metadata_only( |
|
&(ctx->server_rpc_new.request_metadata)); |
|
|
|
gpr_free(ctx); |
|
} |
|
|
|
GPR_EXPORT gpr_intptr GPR_CALLTYPE grpcsharp_batch_context_recv_message_length( |
|
const grpcsharp_batch_context *ctx) { |
|
if (!ctx->recv_message) { |
|
return -1; |
|
} |
|
return grpc_byte_buffer_length(ctx->recv_message); |
|
} |
|
|
|
/* |
|
* Copies data from recv_message to a buffer. Fatal error occurs if |
|
* buffer is too small. |
|
*/ |
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_batch_context_recv_message_to_buffer( |
|
const grpcsharp_batch_context *ctx, char *buffer, size_t buffer_len) { |
|
grpc_byte_buffer_reader reader; |
|
gpr_slice slice; |
|
size_t offset = 0; |
|
|
|
grpc_byte_buffer_reader_init(&reader, ctx->recv_message); |
|
|
|
while (grpc_byte_buffer_reader_next(&reader, &slice)) { |
|
size_t len = GPR_SLICE_LENGTH(slice); |
|
GPR_ASSERT(offset + len <= buffer_len); |
|
memcpy(buffer + offset, GPR_SLICE_START_PTR(slice), |
|
GPR_SLICE_LENGTH(slice)); |
|
offset += len; |
|
gpr_slice_unref(slice); |
|
} |
|
} |
|
|
|
GPR_EXPORT grpc_status_code GPR_CALLTYPE |
|
grpcsharp_batch_context_recv_status_on_client_status( |
|
const grpcsharp_batch_context *ctx) { |
|
return ctx->recv_status_on_client.status; |
|
} |
|
|
|
GPR_EXPORT const char *GPR_CALLTYPE |
|
grpcsharp_batch_context_recv_status_on_client_details( |
|
const grpcsharp_batch_context *ctx) { |
|
return ctx->recv_status_on_client.status_details; |
|
} |
|
|
|
GPR_EXPORT grpc_call *GPR_CALLTYPE grpcsharp_batch_context_server_rpc_new_call( |
|
const grpcsharp_batch_context *ctx) { |
|
return ctx->server_rpc_new.call; |
|
} |
|
|
|
GPR_EXPORT const char *GPR_CALLTYPE |
|
grpcsharp_batch_context_server_rpc_new_method( |
|
const grpcsharp_batch_context *ctx) { |
|
return ctx->server_rpc_new.call_details.method; |
|
} |
|
|
|
GPR_EXPORT gpr_int32 GPR_CALLTYPE |
|
grpcsharp_batch_context_recv_close_on_server_cancelled( |
|
const grpcsharp_batch_context *ctx) { |
|
return (gpr_int32) ctx->recv_close_on_server_cancelled; |
|
} |
|
|
|
/* Init & shutdown */ |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_init(void) { grpc_init(); } |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_shutdown(void) { grpc_shutdown(); } |
|
|
|
/* Completion queue */ |
|
|
|
GPR_EXPORT grpc_completion_queue *GPR_CALLTYPE |
|
grpcsharp_completion_queue_create(void) { |
|
return grpc_completion_queue_create(); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_completion_queue_shutdown(grpc_completion_queue *cq) { |
|
grpc_completion_queue_shutdown(cq); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_completion_queue_destroy(grpc_completion_queue *cq) { |
|
grpc_completion_queue_destroy(cq); |
|
} |
|
|
|
GPR_EXPORT grpc_event GPR_CALLTYPE |
|
grpcsharp_completion_queue_next(grpc_completion_queue *cq) { |
|
return grpc_completion_queue_next(cq, gpr_inf_future); |
|
} |
|
|
|
GPR_EXPORT grpc_event GPR_CALLTYPE |
|
grpcsharp_completion_queue_pluck(grpc_completion_queue *cq, void *tag) { |
|
return grpc_completion_queue_pluck(cq, tag, gpr_inf_future); |
|
} |
|
|
|
/* Channel */ |
|
|
|
GPR_EXPORT grpc_channel *GPR_CALLTYPE |
|
grpcsharp_channel_create(const char *target, const grpc_channel_args *args) { |
|
return grpc_channel_create(target, args); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_channel_destroy(grpc_channel *channel) { |
|
grpc_channel_destroy(channel); |
|
} |
|
|
|
GPR_EXPORT grpc_call *GPR_CALLTYPE |
|
grpcsharp_channel_create_call(grpc_channel *channel, grpc_completion_queue *cq, |
|
const char *method, const char *host, |
|
gpr_timespec deadline) { |
|
return grpc_channel_create_call(channel, cq, method, host, deadline); |
|
} |
|
|
|
/* Channel args */ |
|
|
|
GPR_EXPORT grpc_channel_args *GPR_CALLTYPE |
|
grpcsharp_channel_args_create(size_t num_args) { |
|
grpc_channel_args *args = |
|
(grpc_channel_args *)gpr_malloc(sizeof(grpc_channel_args)); |
|
memset(args, 0, sizeof(grpc_channel_args)); |
|
|
|
args->num_args = num_args; |
|
args->args = (grpc_arg *)gpr_malloc(sizeof(grpc_arg) * num_args); |
|
memset(args->args, 0, sizeof(grpc_arg) * num_args); |
|
return args; |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_channel_args_set_string(grpc_channel_args *args, size_t index, |
|
const char *key, const char *value) { |
|
GPR_ASSERT(args); |
|
GPR_ASSERT(index < args->num_args); |
|
args->args[index].type = GRPC_ARG_STRING; |
|
args->args[index].key = gpr_strdup(key); |
|
args->args[index].value.string = gpr_strdup(value); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_channel_args_set_integer(grpc_channel_args *args, size_t index, |
|
const char *key, int value) { |
|
GPR_ASSERT(args); |
|
GPR_ASSERT(index < args->num_args); |
|
args->args[index].type = GRPC_ARG_INTEGER; |
|
args->args[index].key = gpr_strdup(key); |
|
args->args[index].value.integer = value; |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_channel_args_destroy(grpc_channel_args *args) { |
|
size_t i; |
|
if (args) { |
|
for (i = 0; i < args->num_args; i++) { |
|
gpr_free(args->args[i].key); |
|
if (args->args[i].type == GRPC_ARG_STRING) { |
|
gpr_free(args->args[i].value.string); |
|
} |
|
} |
|
gpr_free(args->args); |
|
gpr_free(args); |
|
} |
|
} |
|
|
|
/* Timespec */ |
|
|
|
GPR_EXPORT gpr_timespec GPR_CALLTYPE gprsharp_now(void) { return gpr_now(GPR_CLOCK_REALTIME); } |
|
|
|
GPR_EXPORT gpr_timespec GPR_CALLTYPE gprsharp_inf_future(void) { |
|
return gpr_inf_future; |
|
} |
|
|
|
GPR_EXPORT gpr_int32 GPR_CALLTYPE gprsharp_sizeof_timespec(void) { |
|
return sizeof(gpr_timespec); |
|
} |
|
|
|
/* Call */ |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE grpcsharp_call_cancel(grpc_call *call) { |
|
return grpc_call_cancel(call); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_cancel_with_status(grpc_call *call, grpc_status_code status, |
|
const char *description) { |
|
return grpc_call_cancel_with_status(call, status, description); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_call_destroy(grpc_call *call) { |
|
grpc_call_destroy(call); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_start_unary(grpc_call *call, grpcsharp_batch_context *ctx, |
|
const char *send_buffer, size_t send_buffer_len, |
|
grpc_metadata_array *initial_metadata) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[6]; |
|
ops[0].op = GRPC_OP_SEND_INITIAL_METADATA; |
|
grpcsharp_metadata_array_move(&(ctx->send_initial_metadata), |
|
initial_metadata); |
|
ops[0].data.send_initial_metadata.count = ctx->send_initial_metadata.count; |
|
ops[0].data.send_initial_metadata.metadata = |
|
ctx->send_initial_metadata.metadata; |
|
ops[0].flags = 0; |
|
|
|
ops[1].op = GRPC_OP_SEND_MESSAGE; |
|
ctx->send_message = string_to_byte_buffer(send_buffer, send_buffer_len); |
|
ops[1].data.send_message = ctx->send_message; |
|
ops[1].flags = 0; |
|
|
|
ops[2].op = GRPC_OP_SEND_CLOSE_FROM_CLIENT; |
|
ops[2].flags = 0; |
|
|
|
ops[3].op = GRPC_OP_RECV_INITIAL_METADATA; |
|
ops[3].data.recv_initial_metadata = &(ctx->recv_initial_metadata); |
|
ops[3].flags = 0; |
|
|
|
ops[4].op = GRPC_OP_RECV_MESSAGE; |
|
ops[4].data.recv_message = &(ctx->recv_message); |
|
ops[4].flags = 0; |
|
|
|
ops[5].op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
|
ops[5].data.recv_status_on_client.trailing_metadata = |
|
&(ctx->recv_status_on_client.trailing_metadata); |
|
ops[5].data.recv_status_on_client.status = |
|
&(ctx->recv_status_on_client.status); |
|
/* not using preallocation for status_details */ |
|
ops[5].data.recv_status_on_client.status_details = |
|
&(ctx->recv_status_on_client.status_details); |
|
ops[5].data.recv_status_on_client.status_details_capacity = |
|
&(ctx->recv_status_on_client.status_details_capacity); |
|
ops[5].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_start_client_streaming(grpc_call *call, |
|
grpcsharp_batch_context *ctx, |
|
grpc_metadata_array *initial_metadata) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[4]; |
|
ops[0].op = GRPC_OP_SEND_INITIAL_METADATA; |
|
grpcsharp_metadata_array_move(&(ctx->send_initial_metadata), |
|
initial_metadata); |
|
ops[0].data.send_initial_metadata.count = ctx->send_initial_metadata.count; |
|
ops[0].data.send_initial_metadata.metadata = |
|
ctx->send_initial_metadata.metadata; |
|
ops[0].flags = 0; |
|
|
|
ops[1].op = GRPC_OP_RECV_INITIAL_METADATA; |
|
ops[1].data.recv_initial_metadata = &(ctx->recv_initial_metadata); |
|
ops[1].flags = 0; |
|
|
|
ops[2].op = GRPC_OP_RECV_MESSAGE; |
|
ops[2].data.recv_message = &(ctx->recv_message); |
|
ops[2].flags = 0; |
|
|
|
ops[3].op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
|
ops[3].data.recv_status_on_client.trailing_metadata = |
|
&(ctx->recv_status_on_client.trailing_metadata); |
|
ops[3].data.recv_status_on_client.status = |
|
&(ctx->recv_status_on_client.status); |
|
/* not using preallocation for status_details */ |
|
ops[3].data.recv_status_on_client.status_details = |
|
&(ctx->recv_status_on_client.status_details); |
|
ops[3].data.recv_status_on_client.status_details_capacity = |
|
&(ctx->recv_status_on_client.status_details_capacity); |
|
ops[3].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE grpcsharp_call_start_server_streaming( |
|
grpc_call *call, grpcsharp_batch_context *ctx, const char *send_buffer, |
|
size_t send_buffer_len, grpc_metadata_array *initial_metadata) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[5]; |
|
ops[0].op = GRPC_OP_SEND_INITIAL_METADATA; |
|
grpcsharp_metadata_array_move(&(ctx->send_initial_metadata), |
|
initial_metadata); |
|
ops[0].data.send_initial_metadata.count = ctx->send_initial_metadata.count; |
|
ops[0].data.send_initial_metadata.metadata = |
|
ctx->send_initial_metadata.metadata; |
|
ops[0].flags = 0; |
|
|
|
ops[1].op = GRPC_OP_SEND_MESSAGE; |
|
ctx->send_message = string_to_byte_buffer(send_buffer, send_buffer_len); |
|
ops[1].data.send_message = ctx->send_message; |
|
ops[1].flags = 0; |
|
|
|
ops[2].op = GRPC_OP_SEND_CLOSE_FROM_CLIENT; |
|
ops[2].flags = 0; |
|
|
|
ops[3].op = GRPC_OP_RECV_INITIAL_METADATA; |
|
ops[3].data.recv_initial_metadata = &(ctx->recv_initial_metadata); |
|
ops[3].flags = 0; |
|
|
|
ops[4].op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
|
ops[4].data.recv_status_on_client.trailing_metadata = |
|
&(ctx->recv_status_on_client.trailing_metadata); |
|
ops[4].data.recv_status_on_client.status = |
|
&(ctx->recv_status_on_client.status); |
|
/* not using preallocation for status_details */ |
|
ops[4].data.recv_status_on_client.status_details = |
|
&(ctx->recv_status_on_client.status_details); |
|
ops[4].data.recv_status_on_client.status_details_capacity = |
|
&(ctx->recv_status_on_client.status_details_capacity); |
|
ops[4].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_start_duplex_streaming(grpc_call *call, |
|
grpcsharp_batch_context *ctx, |
|
grpc_metadata_array *initial_metadata) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[3]; |
|
ops[0].op = GRPC_OP_SEND_INITIAL_METADATA; |
|
grpcsharp_metadata_array_move(&(ctx->send_initial_metadata), |
|
initial_metadata); |
|
ops[0].data.send_initial_metadata.count = ctx->send_initial_metadata.count; |
|
ops[0].data.send_initial_metadata.metadata = |
|
ctx->send_initial_metadata.metadata; |
|
ops[0].flags = 0; |
|
|
|
ops[1].op = GRPC_OP_RECV_INITIAL_METADATA; |
|
ops[1].data.recv_initial_metadata = &(ctx->recv_initial_metadata); |
|
ops[1].flags = 0; |
|
|
|
ops[2].op = GRPC_OP_RECV_STATUS_ON_CLIENT; |
|
ops[2].data.recv_status_on_client.trailing_metadata = |
|
&(ctx->recv_status_on_client.trailing_metadata); |
|
ops[2].data.recv_status_on_client.status = |
|
&(ctx->recv_status_on_client.status); |
|
/* not using preallocation for status_details */ |
|
ops[2].data.recv_status_on_client.status_details = |
|
&(ctx->recv_status_on_client.status_details); |
|
ops[2].data.recv_status_on_client.status_details_capacity = |
|
&(ctx->recv_status_on_client.status_details_capacity); |
|
ops[2].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_send_message(grpc_call *call, grpcsharp_batch_context *ctx, |
|
const char *send_buffer, size_t send_buffer_len) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[1]; |
|
ops[0].op = GRPC_OP_SEND_MESSAGE; |
|
ctx->send_message = string_to_byte_buffer(send_buffer, send_buffer_len); |
|
ops[0].data.send_message = ctx->send_message; |
|
ops[0].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_send_close_from_client(grpc_call *call, |
|
grpcsharp_batch_context *ctx) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[1]; |
|
ops[0].op = GRPC_OP_SEND_CLOSE_FROM_CLIENT; |
|
ops[0].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_send_status_from_server(grpc_call *call, |
|
grpcsharp_batch_context *ctx, |
|
grpc_status_code status_code, |
|
const char *status_details) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[1]; |
|
ops[0].op = GRPC_OP_SEND_STATUS_FROM_SERVER; |
|
ops[0].data.send_status_from_server.status = status_code; |
|
ops[0].data.send_status_from_server.status_details = |
|
gpr_strdup(status_details); |
|
ops[0].data.send_status_from_server.trailing_metadata = NULL; |
|
ops[0].data.send_status_from_server.trailing_metadata_count = 0; |
|
ops[0].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_recv_message(grpc_call *call, grpcsharp_batch_context *ctx) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[1]; |
|
ops[0].op = GRPC_OP_RECV_MESSAGE; |
|
ops[0].data.recv_message = &(ctx->recv_message); |
|
ops[0].flags = 0; |
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_call_start_serverside(grpc_call *call, grpcsharp_batch_context *ctx) { |
|
/* TODO: don't use magic number */ |
|
grpc_op ops[2]; |
|
ops[0].op = GRPC_OP_SEND_INITIAL_METADATA; |
|
ops[0].data.send_initial_metadata.count = 0; |
|
ops[0].data.send_initial_metadata.metadata = NULL; |
|
ops[0].flags = 0; |
|
|
|
ops[1].op = GRPC_OP_RECV_CLOSE_ON_SERVER; |
|
ops[1].data.recv_close_on_server.cancelled = |
|
(&ctx->recv_close_on_server_cancelled); |
|
ops[1].flags = 0; |
|
|
|
return grpc_call_start_batch(call, ops, sizeof(ops) / sizeof(ops[0]), ctx); |
|
} |
|
|
|
/* Server */ |
|
|
|
GPR_EXPORT grpc_server *GPR_CALLTYPE |
|
grpcsharp_server_create(grpc_completion_queue *cq, |
|
const grpc_channel_args *args) { |
|
grpc_server *server = grpc_server_create(args); |
|
grpc_server_register_completion_queue(server, cq); |
|
return server; |
|
} |
|
|
|
GPR_EXPORT gpr_int32 GPR_CALLTYPE |
|
grpcsharp_server_add_http2_port(grpc_server *server, const char *addr) { |
|
return grpc_server_add_http2_port(server, addr); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_server_start(grpc_server *server) { |
|
grpc_server_start(server); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_server_shutdown_and_notify_callback(grpc_server *server, |
|
grpc_completion_queue *cq, |
|
grpcsharp_batch_context *ctx) { |
|
grpc_server_shutdown_and_notify(server, cq, ctx); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_server_cancel_all_calls(grpc_server *server) { |
|
grpc_server_cancel_all_calls(server); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_server_destroy(grpc_server *server) { |
|
grpc_server_destroy(server); |
|
} |
|
|
|
GPR_EXPORT grpc_call_error GPR_CALLTYPE |
|
grpcsharp_server_request_call(grpc_server *server, grpc_completion_queue *cq, |
|
grpcsharp_batch_context *ctx) { |
|
return grpc_server_request_call( |
|
server, &(ctx->server_rpc_new.call), &(ctx->server_rpc_new.call_details), |
|
&(ctx->server_rpc_new.request_metadata), cq, cq, ctx); |
|
} |
|
|
|
/* Security */ |
|
|
|
GPR_EXPORT grpc_credentials *GPR_CALLTYPE |
|
grpcsharp_ssl_credentials_create(const char *pem_root_certs, |
|
const char *key_cert_pair_cert_chain, |
|
const char *key_cert_pair_private_key) { |
|
grpc_ssl_pem_key_cert_pair key_cert_pair; |
|
if (key_cert_pair_cert_chain || key_cert_pair_private_key) { |
|
key_cert_pair.cert_chain = key_cert_pair_cert_chain; |
|
key_cert_pair.private_key = key_cert_pair_private_key; |
|
return grpc_ssl_credentials_create(pem_root_certs, &key_cert_pair); |
|
} else { |
|
GPR_ASSERT(!key_cert_pair_cert_chain); |
|
GPR_ASSERT(!key_cert_pair_private_key); |
|
return grpc_ssl_credentials_create(pem_root_certs, NULL); |
|
} |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_credentials_release(grpc_credentials *creds) { |
|
grpc_credentials_release(creds); |
|
} |
|
|
|
GPR_EXPORT grpc_channel *GPR_CALLTYPE |
|
grpcsharp_secure_channel_create(grpc_credentials *creds, const char *target, |
|
const grpc_channel_args *args) { |
|
return grpc_secure_channel_create(creds, target, args); |
|
} |
|
|
|
GPR_EXPORT grpc_server_credentials *GPR_CALLTYPE |
|
grpcsharp_ssl_server_credentials_create( |
|
const char *pem_root_certs, const char **key_cert_pair_cert_chain_array, |
|
const char **key_cert_pair_private_key_array, size_t num_key_cert_pairs) { |
|
size_t i; |
|
grpc_server_credentials *creds; |
|
grpc_ssl_pem_key_cert_pair *key_cert_pairs = |
|
gpr_malloc(sizeof(grpc_ssl_pem_key_cert_pair) * num_key_cert_pairs); |
|
memset(key_cert_pairs, 0, |
|
sizeof(grpc_ssl_pem_key_cert_pair) * num_key_cert_pairs); |
|
|
|
for (i = 0; i < num_key_cert_pairs; i++) { |
|
if (key_cert_pair_cert_chain_array[i] || |
|
key_cert_pair_private_key_array[i]) { |
|
key_cert_pairs[i].cert_chain = key_cert_pair_cert_chain_array[i]; |
|
key_cert_pairs[i].private_key = key_cert_pair_private_key_array[i]; |
|
} |
|
} |
|
creds = grpc_ssl_server_credentials_create(pem_root_certs, key_cert_pairs, |
|
num_key_cert_pairs); |
|
gpr_free(key_cert_pairs); |
|
return creds; |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_server_credentials_release( |
|
grpc_server_credentials *creds) { |
|
grpc_server_credentials_release(creds); |
|
} |
|
|
|
GPR_EXPORT gpr_int32 GPR_CALLTYPE |
|
grpcsharp_server_add_secure_http2_port(grpc_server *server, const char *addr, |
|
grpc_server_credentials *creds) { |
|
return grpc_server_add_secure_http2_port(server, addr, creds); |
|
} |
|
|
|
/* Logging */ |
|
|
|
typedef void(GPR_CALLTYPE *grpcsharp_log_func)(const char *file, gpr_int32 line, |
|
gpr_uint64 thd_id, |
|
const char *severity_string, |
|
const char *msg); |
|
static grpcsharp_log_func log_func = NULL; |
|
|
|
/* Redirects gpr_log to log_func callback */ |
|
static void grpcsharp_log_handler(gpr_log_func_args *args) { |
|
log_func(args->file, args->line, gpr_thd_currentid(), |
|
gpr_log_severity_string(args->severity), args->message); |
|
} |
|
|
|
GPR_EXPORT void GPR_CALLTYPE grpcsharp_redirect_log(grpcsharp_log_func func) { |
|
GPR_ASSERT(func); |
|
log_func = func; |
|
gpr_set_log_function(grpcsharp_log_handler); |
|
} |
|
|
|
typedef void(GPR_CALLTYPE *test_callback_funcptr)(gpr_int32 success); |
|
|
|
/* For testing */ |
|
GPR_EXPORT void GPR_CALLTYPE |
|
grpcsharp_test_callback(test_callback_funcptr callback) { |
|
callback(1); |
|
} |
|
|
|
/* For testing */ |
|
GPR_EXPORT void *GPR_CALLTYPE grpcsharp_test_nop(void *ptr) { return ptr; } |
|
|
|
/* For testing */ |
|
GPR_EXPORT gpr_int32 GPR_CALLTYPE grpcsharp_sizeof_grpc_event(void) { |
|
return sizeof(grpc_event); |
|
}
|
|
|