mirror of https://github.com/grpc/grpc.git
commit
08b5ba1cdd
237 changed files with 9053 additions and 2024 deletions
@ -0,0 +1,7 @@ |
|||||||
|
root = true |
||||||
|
[**] |
||||||
|
end_of_line = LF |
||||||
|
indent_style = space |
||||||
|
indent_size = 2 |
||||||
|
insert_final_newline = true |
||||||
|
tab_width = 8 |
@ -0,0 +1,64 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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 <string.h> |
||||||
|
|
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
|
||||||
|
#include "src/core/json/json.h" |
||||||
|
|
||||||
|
grpc_json *grpc_json_create(grpc_json_type type) { |
||||||
|
grpc_json *json = gpr_malloc(sizeof(grpc_json)); |
||||||
|
memset(json, 0, sizeof(grpc_json)); |
||||||
|
json->type = type; |
||||||
|
|
||||||
|
return json; |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_destroy(grpc_json *json) { |
||||||
|
while (json->child) { |
||||||
|
grpc_json_destroy(json->child); |
||||||
|
} |
||||||
|
|
||||||
|
if (json->next) { |
||||||
|
json->next->prev = json->prev; |
||||||
|
} |
||||||
|
|
||||||
|
if (json->prev) { |
||||||
|
json->prev->next = json->next; |
||||||
|
} else if (json->parent) { |
||||||
|
json->parent->child = json->next; |
||||||
|
} |
||||||
|
|
||||||
|
gpr_free(json); |
||||||
|
} |
@ -0,0 +1,88 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef __GRPC_SRC_CORE_JSON_JSON_H__ |
||||||
|
#define __GRPC_SRC_CORE_JSON_JSON_H__ |
||||||
|
|
||||||
|
#include <stdlib.h> |
||||||
|
|
||||||
|
#include "src/core/json/json_common.h" |
||||||
|
|
||||||
|
/* A tree-like structure to hold json values. The key and value pointers
|
||||||
|
* are not owned by it. |
||||||
|
*/ |
||||||
|
typedef struct grpc_json { |
||||||
|
struct grpc_json* next; |
||||||
|
struct grpc_json* prev; |
||||||
|
struct grpc_json* child; |
||||||
|
struct grpc_json* parent; |
||||||
|
|
||||||
|
grpc_json_type type; |
||||||
|
const char* key; |
||||||
|
const char* value; |
||||||
|
} grpc_json; |
||||||
|
|
||||||
|
/* The next two functions are going to parse the input string, and
|
||||||
|
* destroy it in the process, in order to use its space to store |
||||||
|
* all of the keys and values for the returned object tree. |
||||||
|
* |
||||||
|
* They assume UTF-8 input stream, and will output UTF-8 encoded |
||||||
|
* strings in the tree. The input stream's UTF-8 isn't validated, |
||||||
|
* as in, what you input is what you get as an output. |
||||||
|
* |
||||||
|
* All the keys and values in the grpc_json_t objects will be strings |
||||||
|
* pointing at your input buffer. |
||||||
|
* |
||||||
|
* Delete the allocated tree afterward using grpc_json_destroy(). |
||||||
|
*/ |
||||||
|
grpc_json* grpc_json_parse_string_with_len(char* input, size_t size); |
||||||
|
grpc_json* grpc_json_parse_string(char* input); |
||||||
|
|
||||||
|
/* This function will create a new string using gpr_realloc, and will
|
||||||
|
* deserialize the grpc_json tree into it. It'll be zero-terminated, |
||||||
|
* but will be allocated in chunks of 256 bytes. |
||||||
|
* |
||||||
|
* The indent parameter controls the way the output is formatted. |
||||||
|
* If indent is 0, then newlines will be suppressed as well, and the |
||||||
|
* output will be condensed at its maximum. |
||||||
|
*/ |
||||||
|
char* grpc_json_dump_to_string(grpc_json* json, int indent); |
||||||
|
|
||||||
|
/* Use these to create or delete a grpc_json object.
|
||||||
|
* Deletion is recursive. We will not attempt to free any of the strings |
||||||
|
* in any of the objects of that tree. |
||||||
|
*/ |
||||||
|
grpc_json* grpc_json_create(grpc_json_type type); |
||||||
|
void grpc_json_destroy(grpc_json* json); |
||||||
|
|
||||||
|
#endif /* __GRPC_SRC_CORE_JSON_JSON_H__ */ |
@ -0,0 +1,49 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef __GRPC_SRC_CORE_JSON_JSON_COMMON_H__ |
||||||
|
#define __GRPC_SRC_CORE_JSON_JSON_COMMON_H__ |
||||||
|
|
||||||
|
/* The various json types. */ |
||||||
|
typedef enum { |
||||||
|
GRPC_JSON_OBJECT, |
||||||
|
GRPC_JSON_ARRAY, |
||||||
|
GRPC_JSON_STRING, |
||||||
|
GRPC_JSON_NUMBER, |
||||||
|
GRPC_JSON_TRUE, |
||||||
|
GRPC_JSON_FALSE, |
||||||
|
GRPC_JSON_NULL, |
||||||
|
GRPC_JSON_TOP_LEVEL |
||||||
|
} grpc_json_type; |
||||||
|
|
||||||
|
#endif /* __GRPC_SRC_CORE_JSON_JSON_COMMON_H__ */ |
@ -0,0 +1,653 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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 <string.h> |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "src/core/json/json_reader.h" |
||||||
|
|
||||||
|
static void json_reader_string_clear(grpc_json_reader* reader) { |
||||||
|
reader->vtable->string_clear(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_string_add_char(grpc_json_reader* reader, |
||||||
|
gpr_uint32 c) { |
||||||
|
reader->vtable->string_add_char(reader->userdata, c); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_string_add_utf32(grpc_json_reader* reader, |
||||||
|
gpr_uint32 utf32) { |
||||||
|
reader->vtable->string_add_utf32(reader->userdata, utf32); |
||||||
|
} |
||||||
|
|
||||||
|
static gpr_uint32 |
||||||
|
grpc_json_reader_read_char(grpc_json_reader* reader) { |
||||||
|
return reader->vtable->read_char(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_container_begins(grpc_json_reader* reader, |
||||||
|
grpc_json_type type) { |
||||||
|
reader->vtable->container_begins(reader->userdata, type); |
||||||
|
} |
||||||
|
|
||||||
|
static grpc_json_type |
||||||
|
grpc_json_reader_container_ends(grpc_json_reader* reader) { |
||||||
|
return reader->vtable->container_ends(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_key(grpc_json_reader* reader) { |
||||||
|
reader->vtable->set_key(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_string(grpc_json_reader* reader) { |
||||||
|
reader->vtable->set_string(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static int json_reader_set_number(grpc_json_reader* reader) { |
||||||
|
return reader->vtable->set_number(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_true(grpc_json_reader* reader) { |
||||||
|
reader->vtable->set_true(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_false(grpc_json_reader* reader) { |
||||||
|
reader->vtable->set_false(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_null(grpc_json_reader* reader) { |
||||||
|
reader->vtable->set_null(reader->userdata); |
||||||
|
} |
||||||
|
|
||||||
|
/* Call this function to initialize the reader structure. */ |
||||||
|
void grpc_json_reader_init(grpc_json_reader* reader, |
||||||
|
grpc_json_reader_vtable* vtable, void* userdata) { |
||||||
|
memset(reader, 0, sizeof(grpc_json_reader)); |
||||||
|
reader->vtable = vtable; |
||||||
|
reader->userdata = userdata; |
||||||
|
json_reader_string_clear(reader); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_BEGIN; |
||||||
|
} |
||||||
|
|
||||||
|
int grpc_json_reader_is_complete(grpc_json_reader* reader) { |
||||||
|
return ((reader->depth == 0) && ((reader->state == GRPC_JSON_STATE_END) || |
||||||
|
(reader->state == GRPC_JSON_STATE_VALUE_END))); |
||||||
|
} |
||||||
|
|
||||||
|
grpc_json_reader_status grpc_json_reader_run(grpc_json_reader* reader) { |
||||||
|
gpr_uint32 c, success; |
||||||
|
|
||||||
|
/* This state-machine is a strict implementation of ECMA-404 */ |
||||||
|
for (;;) { |
||||||
|
c = grpc_json_reader_read_char(reader); |
||||||
|
switch (c) { |
||||||
|
/* Let's process the error cases first. */ |
||||||
|
case GRPC_JSON_READ_CHAR_ERROR: |
||||||
|
return GRPC_JSON_READ_ERROR; |
||||||
|
|
||||||
|
case GRPC_JSON_READ_CHAR_EAGAIN: |
||||||
|
return GRPC_JSON_EAGAIN; |
||||||
|
|
||||||
|
case GRPC_JSON_READ_CHAR_EOF: |
||||||
|
if (grpc_json_reader_is_complete(reader)) { |
||||||
|
return GRPC_JSON_DONE; |
||||||
|
} else { |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
/* Processing whitespaces. */ |
||||||
|
case ' ': |
||||||
|
case '\t': |
||||||
|
case '\n': |
||||||
|
case '\r': |
||||||
|
switch (reader->state) { |
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_BEGIN: |
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_END: |
||||||
|
case GRPC_JSON_STATE_VALUE_BEGIN: |
||||||
|
case GRPC_JSON_STATE_VALUE_END: |
||||||
|
case GRPC_JSON_STATE_END: |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_STRING: |
||||||
|
case GRPC_JSON_STATE_VALUE_STRING: |
||||||
|
if (c != ' ') return GRPC_JSON_PARSE_ERROR; |
||||||
|
if (reader->unicode_high_surrogate != 0) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER: |
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_WITH_DECIMAL: |
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_ZERO: |
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_EPM: |
||||||
|
success = json_reader_set_number(reader); |
||||||
|
if (!success) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_clear(reader); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
break; |
||||||
|
|
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
/* Value, object or array terminations. */ |
||||||
|
case ',': |
||||||
|
case '}': |
||||||
|
case ']': |
||||||
|
switch (reader->state) { |
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_STRING: |
||||||
|
case GRPC_JSON_STATE_VALUE_STRING: |
||||||
|
if (reader->unicode_high_surrogate != 0) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER: |
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_WITH_DECIMAL: |
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_ZERO: |
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_EPM: |
||||||
|
success = json_reader_set_number(reader); |
||||||
|
if (!success) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_clear(reader); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
/* The missing break here is intentional. */ |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_END: |
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_BEGIN: |
||||||
|
case GRPC_JSON_STATE_VALUE_BEGIN: |
||||||
|
if (c == ',') { |
||||||
|
if (reader->state != GRPC_JSON_STATE_VALUE_END) { |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
if (reader->in_object) { |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_BEGIN; |
||||||
|
} else { |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_BEGIN; |
||||||
|
} |
||||||
|
} else { |
||||||
|
if (reader->depth-- == 0) return GRPC_JSON_PARSE_ERROR; |
||||||
|
if ((c == '}') && !reader->in_object) { |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
if ((c == '}') && |
||||||
|
(reader->state == GRPC_JSON_STATE_OBJECT_KEY_BEGIN) && |
||||||
|
!reader->container_just_begun) { |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
if ((c == ']') && !reader->in_array) return GRPC_JSON_PARSE_ERROR; |
||||||
|
if ((c == ']') && |
||||||
|
(reader->state == GRPC_JSON_STATE_VALUE_BEGIN) && |
||||||
|
!reader->container_just_begun) { |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
switch (grpc_json_reader_container_ends(reader)) { |
||||||
|
case GRPC_JSON_OBJECT: |
||||||
|
reader->in_object = 1; |
||||||
|
reader->in_array = 0; |
||||||
|
break; |
||||||
|
case GRPC_JSON_ARRAY: |
||||||
|
reader->in_object = 0; |
||||||
|
reader->in_array = 1; |
||||||
|
break; |
||||||
|
case GRPC_JSON_TOP_LEVEL: |
||||||
|
if (reader->depth != 0) return GRPC_JSON_INTERNAL_ERROR; |
||||||
|
reader->in_object = 0; |
||||||
|
reader->in_array = 0; |
||||||
|
reader->state = GRPC_JSON_STATE_END; |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_INTERNAL_ERROR; |
||||||
|
} |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
/* In-string escaping. */ |
||||||
|
case '\\': |
||||||
|
switch (reader->state) { |
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_STRING: |
||||||
|
reader->escaped_string_was_key = 1; |
||||||
|
reader->state = GRPC_JSON_STATE_STRING_ESCAPE; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_STRING: |
||||||
|
reader->escaped_string_was_key = 0; |
||||||
|
reader->state = GRPC_JSON_STATE_STRING_ESCAPE; |
||||||
|
break; |
||||||
|
|
||||||
|
/* This is the \\ case. */ |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE: |
||||||
|
if (reader->unicode_high_surrogate != 0) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_char(reader, '\\'); |
||||||
|
if (reader->escaped_string_was_key) { |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_STRING; |
||||||
|
} else { |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_STRING; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
default: |
||||||
|
reader->container_just_begun = 0; |
||||||
|
switch (reader->state) { |
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_BEGIN: |
||||||
|
if (c != '"') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_STRING; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_STRING: |
||||||
|
if (reader->unicode_high_surrogate != 0) return GRPC_JSON_PARSE_ERROR; |
||||||
|
if (c == '"') { |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_END; |
||||||
|
json_reader_set_key(reader); |
||||||
|
json_reader_string_clear(reader); |
||||||
|
} else { |
||||||
|
if (c <= 0x001f) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_STRING: |
||||||
|
if (reader->unicode_high_surrogate != 0) return GRPC_JSON_PARSE_ERROR; |
||||||
|
if (c == '"') { |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
json_reader_set_string(reader); |
||||||
|
json_reader_string_clear(reader); |
||||||
|
} else { |
||||||
|
if (c < 32) return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_OBJECT_KEY_END: |
||||||
|
if (c != ':') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_BEGIN; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_BEGIN: |
||||||
|
switch (c) { |
||||||
|
case 't': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_TRUE_R; |
||||||
|
break; |
||||||
|
|
||||||
|
case 'f': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_FALSE_A; |
||||||
|
break; |
||||||
|
|
||||||
|
case 'n': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NULL_U; |
||||||
|
break; |
||||||
|
|
||||||
|
case '"': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_STRING; |
||||||
|
break; |
||||||
|
|
||||||
|
case '0': |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_ZERO; |
||||||
|
break; |
||||||
|
|
||||||
|
case '1': |
||||||
|
case '2': |
||||||
|
case '3': |
||||||
|
case '4': |
||||||
|
case '5': |
||||||
|
case '6': |
||||||
|
case '7': |
||||||
|
case '8': |
||||||
|
case '9': |
||||||
|
case '-': |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER; |
||||||
|
break; |
||||||
|
|
||||||
|
case '{': |
||||||
|
reader->container_just_begun = 1; |
||||||
|
json_reader_container_begins(reader, GRPC_JSON_OBJECT); |
||||||
|
reader->depth++; |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_BEGIN; |
||||||
|
reader->in_object = 1; |
||||||
|
reader->in_array = 0; |
||||||
|
break; |
||||||
|
|
||||||
|
case '[': |
||||||
|
reader->container_just_begun = 1; |
||||||
|
json_reader_container_begins(reader, GRPC_JSON_ARRAY); |
||||||
|
reader->depth++; |
||||||
|
reader->in_object = 0; |
||||||
|
reader->in_array = 1; |
||||||
|
break; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE: |
||||||
|
if (reader->escaped_string_was_key) { |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_STRING; |
||||||
|
} else { |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_STRING; |
||||||
|
} |
||||||
|
if (reader->unicode_high_surrogate && c != 'u') |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
switch (c) { |
||||||
|
case '"': |
||||||
|
case '/': |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
break; |
||||||
|
case 'b': |
||||||
|
json_reader_string_add_char(reader, '\b'); |
||||||
|
break; |
||||||
|
case 'f': |
||||||
|
json_reader_string_add_char(reader, '\f'); |
||||||
|
break; |
||||||
|
case 'n': |
||||||
|
json_reader_string_add_char(reader, '\n'); |
||||||
|
break; |
||||||
|
case 'r': |
||||||
|
json_reader_string_add_char(reader, '\r'); |
||||||
|
break; |
||||||
|
case 't': |
||||||
|
json_reader_string_add_char(reader, '\t'); |
||||||
|
break; |
||||||
|
case 'u': |
||||||
|
reader->state = GRPC_JSON_STATE_STRING_ESCAPE_U1; |
||||||
|
reader->unicode_char = 0; |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U1: |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U2: |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U3: |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U4: |
||||||
|
if ((c >= '0') && (c <= '9')) { |
||||||
|
c -= '0'; |
||||||
|
} else if ((c >= 'A') && (c <= 'F')) { |
||||||
|
c -= 'A' - 10; |
||||||
|
} else if ((c >= 'a') && (c <= 'f')) { |
||||||
|
c -= 'a' - 10; |
||||||
|
} else { |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
reader->unicode_char <<= 4; |
||||||
|
reader->unicode_char |= c; |
||||||
|
|
||||||
|
switch (reader->state) { |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U1: |
||||||
|
reader->state = GRPC_JSON_STATE_STRING_ESCAPE_U2; |
||||||
|
break; |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U2: |
||||||
|
reader->state = GRPC_JSON_STATE_STRING_ESCAPE_U3; |
||||||
|
break; |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U3: |
||||||
|
reader->state = GRPC_JSON_STATE_STRING_ESCAPE_U4; |
||||||
|
break; |
||||||
|
case GRPC_JSON_STATE_STRING_ESCAPE_U4: |
||||||
|
/* See grpc_json_writer_escape_string to have a description
|
||||||
|
* of what's going on here. |
||||||
|
*/ |
||||||
|
if ((reader->unicode_char & 0xfc00) == 0xd800) { |
||||||
|
/* high surrogate utf-16 */ |
||||||
|
if (reader->unicode_high_surrogate != 0) |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->unicode_high_surrogate = reader->unicode_char; |
||||||
|
} else if ((reader->unicode_char & 0xfc00) == 0xdc00) { |
||||||
|
/* low surrogate utf-16 */ |
||||||
|
gpr_uint32 utf32; |
||||||
|
if (reader->unicode_high_surrogate == 0) |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
utf32 = 0x10000; |
||||||
|
utf32 += (reader->unicode_high_surrogate - 0xd800) * 0x400; |
||||||
|
utf32 += reader->unicode_char - 0xdc00; |
||||||
|
json_reader_string_add_utf32(reader, utf32); |
||||||
|
reader->unicode_high_surrogate = 0; |
||||||
|
} else { |
||||||
|
/* anything else */ |
||||||
|
if (reader->unicode_high_surrogate != 0) |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_utf32(reader, reader->unicode_char); |
||||||
|
} |
||||||
|
if (reader->escaped_string_was_key) { |
||||||
|
reader->state = GRPC_JSON_STATE_OBJECT_KEY_STRING; |
||||||
|
} else { |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_STRING; |
||||||
|
} |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_INTERNAL_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER: |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
switch (c) { |
||||||
|
case '0': |
||||||
|
case '1': |
||||||
|
case '2': |
||||||
|
case '3': |
||||||
|
case '4': |
||||||
|
case '5': |
||||||
|
case '6': |
||||||
|
case '7': |
||||||
|
case '8': |
||||||
|
case '9': |
||||||
|
break; |
||||||
|
case 'e': |
||||||
|
case 'E': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_E; |
||||||
|
break; |
||||||
|
case '.': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_DOT; |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_WITH_DECIMAL: |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
switch (c) { |
||||||
|
case '0': |
||||||
|
case '1': |
||||||
|
case '2': |
||||||
|
case '3': |
||||||
|
case '4': |
||||||
|
case '5': |
||||||
|
case '6': |
||||||
|
case '7': |
||||||
|
case '8': |
||||||
|
case '9': |
||||||
|
break; |
||||||
|
case 'e': |
||||||
|
case 'E': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_E; |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_ZERO: |
||||||
|
if (c != '.') return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_DOT; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_DOT: |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
switch (c) { |
||||||
|
case '0': |
||||||
|
case '1': |
||||||
|
case '2': |
||||||
|
case '3': |
||||||
|
case '4': |
||||||
|
case '5': |
||||||
|
case '6': |
||||||
|
case '7': |
||||||
|
case '8': |
||||||
|
case '9': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_WITH_DECIMAL; |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_E: |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
switch (c) { |
||||||
|
case '0': |
||||||
|
case '1': |
||||||
|
case '2': |
||||||
|
case '3': |
||||||
|
case '4': |
||||||
|
case '5': |
||||||
|
case '6': |
||||||
|
case '7': |
||||||
|
case '8': |
||||||
|
case '9': |
||||||
|
case '+': |
||||||
|
case '-': |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NUMBER_EPM; |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NUMBER_EPM: |
||||||
|
json_reader_string_add_char(reader, c); |
||||||
|
switch (c) { |
||||||
|
case '0': |
||||||
|
case '1': |
||||||
|
case '2': |
||||||
|
case '3': |
||||||
|
case '4': |
||||||
|
case '5': |
||||||
|
case '6': |
||||||
|
case '7': |
||||||
|
case '8': |
||||||
|
case '9': |
||||||
|
break; |
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_TRUE_R: |
||||||
|
if (c != 'r') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_TRUE_U; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_TRUE_U: |
||||||
|
if (c != 'u') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_TRUE_E; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_TRUE_E: |
||||||
|
if (c != 'e') return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_set_true(reader); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_FALSE_A: |
||||||
|
if (c != 'a') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_FALSE_L; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_FALSE_L: |
||||||
|
if (c != 'l') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_FALSE_S; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_FALSE_S: |
||||||
|
if (c != 's') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_FALSE_E; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_FALSE_E: |
||||||
|
if (c != 'e') return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_set_false(reader); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NULL_U: |
||||||
|
if (c != 'u') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NULL_L1; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NULL_L1: |
||||||
|
if (c != 'l') return GRPC_JSON_PARSE_ERROR; |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_NULL_L2; |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_VALUE_NULL_L2: |
||||||
|
if (c != 'l') return GRPC_JSON_PARSE_ERROR; |
||||||
|
json_reader_set_null(reader); |
||||||
|
reader->state = GRPC_JSON_STATE_VALUE_END; |
||||||
|
break; |
||||||
|
|
||||||
|
/* All of the VALUE_END cases are handled in the specialized case
|
||||||
|
* above. */ |
||||||
|
case GRPC_JSON_STATE_VALUE_END: |
||||||
|
switch (c) { |
||||||
|
case ',': |
||||||
|
case '}': |
||||||
|
case ']': |
||||||
|
return GRPC_JSON_INTERNAL_ERROR; |
||||||
|
break; |
||||||
|
|
||||||
|
default: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
break; |
||||||
|
|
||||||
|
case GRPC_JSON_STATE_END: |
||||||
|
return GRPC_JSON_PARSE_ERROR; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return GRPC_JSON_INTERNAL_ERROR; |
||||||
|
} |
@ -0,0 +1,160 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef __GRPC_SRC_CORE_JSON_JSON_READER_H__ |
||||||
|
#define __GRPC_SRC_CORE_JSON_JSON_READER_H__ |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
#include "src/core/json/json_common.h" |
||||||
|
|
||||||
|
typedef enum { |
||||||
|
GRPC_JSON_STATE_OBJECT_KEY_BEGIN, |
||||||
|
GRPC_JSON_STATE_OBJECT_KEY_STRING, |
||||||
|
GRPC_JSON_STATE_OBJECT_KEY_END, |
||||||
|
GRPC_JSON_STATE_VALUE_BEGIN, |
||||||
|
GRPC_JSON_STATE_VALUE_STRING, |
||||||
|
GRPC_JSON_STATE_STRING_ESCAPE, |
||||||
|
GRPC_JSON_STATE_STRING_ESCAPE_U1, |
||||||
|
GRPC_JSON_STATE_STRING_ESCAPE_U2, |
||||||
|
GRPC_JSON_STATE_STRING_ESCAPE_U3, |
||||||
|
GRPC_JSON_STATE_STRING_ESCAPE_U4, |
||||||
|
GRPC_JSON_STATE_VALUE_NUMBER, |
||||||
|
GRPC_JSON_STATE_VALUE_NUMBER_WITH_DECIMAL, |
||||||
|
GRPC_JSON_STATE_VALUE_NUMBER_ZERO, |
||||||
|
GRPC_JSON_STATE_VALUE_NUMBER_DOT, |
||||||
|
GRPC_JSON_STATE_VALUE_NUMBER_E, |
||||||
|
GRPC_JSON_STATE_VALUE_NUMBER_EPM, |
||||||
|
GRPC_JSON_STATE_VALUE_TRUE_R, |
||||||
|
GRPC_JSON_STATE_VALUE_TRUE_U, |
||||||
|
GRPC_JSON_STATE_VALUE_TRUE_E, |
||||||
|
GRPC_JSON_STATE_VALUE_FALSE_A, |
||||||
|
GRPC_JSON_STATE_VALUE_FALSE_L, |
||||||
|
GRPC_JSON_STATE_VALUE_FALSE_S, |
||||||
|
GRPC_JSON_STATE_VALUE_FALSE_E, |
||||||
|
GRPC_JSON_STATE_VALUE_NULL_U, |
||||||
|
GRPC_JSON_STATE_VALUE_NULL_L1, |
||||||
|
GRPC_JSON_STATE_VALUE_NULL_L2, |
||||||
|
GRPC_JSON_STATE_VALUE_END, |
||||||
|
GRPC_JSON_STATE_END |
||||||
|
} grpc_json_reader_state; |
||||||
|
|
||||||
|
enum { |
||||||
|
/* The first non-unicode value is 0x110000. But let's pick
|
||||||
|
* a value high enough to start our error codes from. These |
||||||
|
* values are safe to return from the read_char function. |
||||||
|
*/ |
||||||
|
GRPC_JSON_READ_CHAR_EOF = 0x7ffffff0, |
||||||
|
GRPC_JSON_READ_CHAR_EAGAIN, |
||||||
|
GRPC_JSON_READ_CHAR_ERROR |
||||||
|
}; |
||||||
|
|
||||||
|
struct grpc_json_reader; |
||||||
|
|
||||||
|
typedef struct grpc_json_reader_vtable { |
||||||
|
/* Clears your internal string scratchpad. */ |
||||||
|
void (*string_clear)(void* userdata); |
||||||
|
/* Adds a char to the string scratchpad. */ |
||||||
|
void (*string_add_char)(void* userdata, gpr_uint32 c); |
||||||
|
/* Adds a utf32 char to the string scratchpad. */ |
||||||
|
void (*string_add_utf32)(void* userdata, gpr_uint32 c); |
||||||
|
/* Reads a character from your input. May be utf-8, 16 or 32. */ |
||||||
|
gpr_uint32 (*read_char)(void* userdata); |
||||||
|
/* Starts a container of type GRPC_JSON_ARRAY or GRPC_JSON_OBJECT. */ |
||||||
|
void (*container_begins)(void* userdata, grpc_json_type type); |
||||||
|
/* Ends the current container. Must return the type of its parent. */ |
||||||
|
grpc_json_type (*container_ends)(void* userdata); |
||||||
|
/* Your internal string scratchpad is an object's key. */ |
||||||
|
void (*set_key)(void* userdata); |
||||||
|
/* Your internal string scratchpad is a string value. */ |
||||||
|
void (*set_string)(void* userdata); |
||||||
|
/* Your internal string scratchpad is a numerical value. Return 1 if valid. */ |
||||||
|
int (*set_number)(void* userdata); |
||||||
|
/* Sets the values true, false or null. */ |
||||||
|
void (*set_true)(void* userdata); |
||||||
|
void (*set_false)(void* userdata); |
||||||
|
void (*set_null)(void* userdata); |
||||||
|
} grpc_json_reader_vtable; |
||||||
|
|
||||||
|
typedef struct grpc_json_reader { |
||||||
|
/* That structure is fully private, and initialized by grpc_json_reader_init.
|
||||||
|
* The definition is public so you can put it on your stack. |
||||||
|
*/ |
||||||
|
|
||||||
|
void* userdata; |
||||||
|
grpc_json_reader_vtable* vtable; |
||||||
|
int depth; |
||||||
|
int in_object; |
||||||
|
int in_array; |
||||||
|
int escaped_string_was_key; |
||||||
|
int container_just_begun; |
||||||
|
gpr_uint16 unicode_char, unicode_high_surrogate; |
||||||
|
grpc_json_reader_state state; |
||||||
|
} grpc_json_reader; |
||||||
|
|
||||||
|
/* The return type of the parser. */ |
||||||
|
typedef enum { |
||||||
|
GRPC_JSON_DONE, /* The parser finished successfully. */ |
||||||
|
GRPC_JSON_EAGAIN, /* The parser yields to get more data. */ |
||||||
|
GRPC_JSON_READ_ERROR, /* The parser passes through a read error. */ |
||||||
|
GRPC_JSON_PARSE_ERROR, /* The parser found an error in the json stream. */ |
||||||
|
GRPC_JSON_INTERNAL_ERROR /* The parser got an internal error. */ |
||||||
|
} grpc_json_reader_status; |
||||||
|
|
||||||
|
/* Call this function to start parsing the input. It will return the following:
|
||||||
|
* . GRPC_JSON_DONE if the input got eof, and the parsing finished |
||||||
|
* successfully. |
||||||
|
* . GRPC_JSON_EAGAIN if the read_char function returned again. Call the |
||||||
|
* parser again as needed. It is okay to call the parser in polling mode, |
||||||
|
* although a bit dull. |
||||||
|
* . GRPC_JSON_READ_ERROR if the read_char function returned an error. The |
||||||
|
* state isn't broken however, and the function can be called again if the |
||||||
|
* error has been corrected. But please use the EAGAIN feature instead for |
||||||
|
* consistency. |
||||||
|
* . GRPC_JSON_PARSE_ERROR if the input was somehow invalid. |
||||||
|
* . GRPC_JSON_INTERNAL_ERROR if the parser somehow ended into an invalid |
||||||
|
* internal state. |
||||||
|
*/ |
||||||
|
grpc_json_reader_status grpc_json_reader_run(grpc_json_reader* reader); |
||||||
|
|
||||||
|
/* Call this function to initialize the reader structure. */ |
||||||
|
void grpc_json_reader_init(grpc_json_reader* reader, |
||||||
|
grpc_json_reader_vtable* vtable, void* userdata); |
||||||
|
|
||||||
|
/* You may call this from the read_char callback if you don't know where is the
|
||||||
|
* end of your input stream, and you'd like the json reader to hint you that it |
||||||
|
* has completed reading its input, so you can return an EOF to it. Note that |
||||||
|
* there might still be trailing whitespaces after that point. |
||||||
|
*/ |
||||||
|
int grpc_json_reader_is_complete(grpc_json_reader* reader); |
||||||
|
|
||||||
|
#endif /* __GRPC_SRC_CORE_JSON_JSON_READER_H__ */ |
@ -0,0 +1,391 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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 <string.h> |
||||||
|
#include <stdlib.h> |
||||||
|
|
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
#include "src/core/json/json.h" |
||||||
|
#include "src/core/json/json_reader.h" |
||||||
|
#include "src/core/json/json_writer.h" |
||||||
|
|
||||||
|
/* The json reader will construct a bunch of grpc_json objects and
|
||||||
|
* link them all up together in a tree-like structure that will represent |
||||||
|
* the json data in memory. |
||||||
|
* |
||||||
|
* It also uses its own input as a scratchpad to store all of the decoded, |
||||||
|
* unescaped strings. So we need to keep track of all these pointers in |
||||||
|
* that opaque structure the reader will carry for us. |
||||||
|
* |
||||||
|
* Note that this works because the act of parsing json always reduces its |
||||||
|
* input size, and never expands it. |
||||||
|
*/ |
||||||
|
typedef struct { |
||||||
|
grpc_json* top; |
||||||
|
grpc_json* current_container; |
||||||
|
grpc_json* current_value; |
||||||
|
gpr_uint8* input; |
||||||
|
gpr_uint8* key; |
||||||
|
gpr_uint8* string; |
||||||
|
gpr_uint8* string_ptr; |
||||||
|
size_t remaining_input; |
||||||
|
} json_reader_userdata; |
||||||
|
|
||||||
|
/* This json writer will put everything in a big string.
|
||||||
|
* The point is that we allocate that string in chunks of 256 bytes. |
||||||
|
*/ |
||||||
|
typedef struct { |
||||||
|
char* output; |
||||||
|
size_t free_space; |
||||||
|
size_t string_len; |
||||||
|
size_t allocated; |
||||||
|
} json_writer_userdata; |
||||||
|
|
||||||
|
|
||||||
|
/* This function checks if there's enough space left in the output buffer,
|
||||||
|
* and will enlarge it if necessary. We're only allocating chunks of 256 |
||||||
|
* bytes at a time (or multiples thereof). |
||||||
|
*/ |
||||||
|
static void json_writer_output_check(void* userdata, size_t needed) { |
||||||
|
json_writer_userdata* state = userdata; |
||||||
|
if (state->free_space >= needed) return; |
||||||
|
needed -= state->free_space; |
||||||
|
/* Round up by 256 bytes. */ |
||||||
|
needed = (needed + 0xff) & ~0xff; |
||||||
|
state->output = gpr_realloc(state->output, state->allocated + needed); |
||||||
|
state->free_space += needed; |
||||||
|
state->allocated += needed; |
||||||
|
} |
||||||
|
|
||||||
|
/* These are needed by the writer's implementation. */ |
||||||
|
static void json_writer_output_char(void* userdata, char c) { |
||||||
|
json_writer_userdata* state = userdata; |
||||||
|
json_writer_output_check(userdata, 1); |
||||||
|
state->output[state->string_len++] = c; |
||||||
|
state->free_space--; |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_output_string_with_len(void* userdata, |
||||||
|
const char* str, size_t len) { |
||||||
|
json_writer_userdata* state = userdata; |
||||||
|
json_writer_output_check(userdata, len); |
||||||
|
memcpy(state->output + state->string_len, str, len); |
||||||
|
state->string_len += len; |
||||||
|
state->free_space -= len; |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_output_string(void* userdata, |
||||||
|
const char* str) { |
||||||
|
size_t len = strlen(str); |
||||||
|
json_writer_output_string_with_len(userdata, str, len); |
||||||
|
} |
||||||
|
|
||||||
|
/* The reader asks us to clear our scratchpad. In our case, we'll simply mark
|
||||||
|
* the end of the current string, and advance our output pointer. |
||||||
|
*/ |
||||||
|
static void json_reader_string_clear(void* userdata) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
if (state->string) { |
||||||
|
GPR_ASSERT(state->string_ptr < state->input); |
||||||
|
*state->string_ptr++ = 0; |
||||||
|
} |
||||||
|
state->string = state->string_ptr; |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_string_add_char(void* userdata, gpr_uint32 c) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
GPR_ASSERT(state->string_ptr < state->input); |
||||||
|
GPR_ASSERT(c <= 0xff); |
||||||
|
*state->string_ptr++ = (char)c; |
||||||
|
} |
||||||
|
|
||||||
|
/* We are converting a UTF-32 character into UTF-8 here,
|
||||||
|
* as described by RFC3629. |
||||||
|
*/ |
||||||
|
static void json_reader_string_add_utf32(void* userdata, gpr_uint32 c) { |
||||||
|
if (c <= 0x7f) { |
||||||
|
json_reader_string_add_char(userdata, c); |
||||||
|
} else if (c <= 0x7ff) { |
||||||
|
int b1 = 0xc0 | ((c >> 6) & 0x1f); |
||||||
|
int b2 = 0x80 | (c & 0x3f); |
||||||
|
json_reader_string_add_char(userdata, b1); |
||||||
|
json_reader_string_add_char(userdata, b2); |
||||||
|
} else if (c <= 0xffff) { |
||||||
|
int b1 = 0xe0 | ((c >> 12) & 0x0f); |
||||||
|
int b2 = 0x80 | ((c >> 6) & 0x3f); |
||||||
|
int b3 = 0x80 | (c & 0x3f); |
||||||
|
json_reader_string_add_char(userdata, b1); |
||||||
|
json_reader_string_add_char(userdata, b2); |
||||||
|
json_reader_string_add_char(userdata, b3); |
||||||
|
} else if (c <= 0x1fffff) { |
||||||
|
int b1 = 0xf0 | ((c >> 18) & 0x07); |
||||||
|
int b2 = 0x80 | ((c >> 12) & 0x3f); |
||||||
|
int b3 = 0x80 | ((c >> 6) & 0x3f); |
||||||
|
int b4 = 0x80 | (c & 0x3f); |
||||||
|
json_reader_string_add_char(userdata, b1); |
||||||
|
json_reader_string_add_char(userdata, b2); |
||||||
|
json_reader_string_add_char(userdata, b3); |
||||||
|
json_reader_string_add_char(userdata, b4); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/* We consider that the input may be a zero-terminated string. So we
|
||||||
|
* can end up hitting eof before the end of the alleged string length. |
||||||
|
*/ |
||||||
|
static gpr_uint32 json_reader_read_char(void* userdata) { |
||||||
|
gpr_uint32 r; |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
|
||||||
|
if (state->remaining_input == 0) return GRPC_JSON_READ_CHAR_EOF; |
||||||
|
|
||||||
|
r = *state->input++; |
||||||
|
state->remaining_input--; |
||||||
|
|
||||||
|
if (r == 0) { |
||||||
|
state->remaining_input = 0; |
||||||
|
return GRPC_JSON_READ_CHAR_EOF; |
||||||
|
} |
||||||
|
|
||||||
|
return r; |
||||||
|
} |
||||||
|
|
||||||
|
/* Helper function to create a new grpc_json object and link it into
|
||||||
|
* our tree-in-progress inside our opaque structure. |
||||||
|
*/ |
||||||
|
static grpc_json* json_create_and_link(void* userdata, |
||||||
|
grpc_json_type type) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
grpc_json* json = grpc_json_create(type); |
||||||
|
|
||||||
|
json->parent = state->current_container; |
||||||
|
json->prev = state->current_value; |
||||||
|
state->current_value = json; |
||||||
|
|
||||||
|
if (json->prev) { |
||||||
|
json->prev->next = json; |
||||||
|
} |
||||||
|
if (json->parent) { |
||||||
|
if (!json->parent->child) { |
||||||
|
json->parent->child = json; |
||||||
|
} |
||||||
|
if (json->parent->type == GRPC_JSON_OBJECT) { |
||||||
|
json->key = (char*) state->key; |
||||||
|
} |
||||||
|
} |
||||||
|
if (!state->top) { |
||||||
|
state->top = json; |
||||||
|
} |
||||||
|
|
||||||
|
return json; |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_container_begins(void* userdata, grpc_json_type type) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
grpc_json* container; |
||||||
|
|
||||||
|
GPR_ASSERT(type == GRPC_JSON_ARRAY || type == GRPC_JSON_OBJECT); |
||||||
|
|
||||||
|
container = json_create_and_link(userdata, type); |
||||||
|
state->current_container = container; |
||||||
|
state->current_value = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
/* It's important to remember that the reader is mostly stateless, so it
|
||||||
|
* isn't trying to remember what the container was prior the one that just |
||||||
|
* ends. Since we're keeping track of these for our own purpose, we are |
||||||
|
* able to return that information back, which is useful for it to validate |
||||||
|
* the input json stream. |
||||||
|
* |
||||||
|
* Also note that if we're at the top of the tree, and the last container |
||||||
|
* ends, we have to return GRPC_JSON_TOP_LEVEL. |
||||||
|
*/ |
||||||
|
static grpc_json_type json_reader_container_ends(void* userdata) { |
||||||
|
grpc_json_type container_type = GRPC_JSON_TOP_LEVEL; |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
|
||||||
|
GPR_ASSERT(state->current_container); |
||||||
|
|
||||||
|
state->current_value = state->current_container; |
||||||
|
state->current_container = state->current_container->parent; |
||||||
|
|
||||||
|
if (state->current_container) { |
||||||
|
container_type = state->current_container->type; |
||||||
|
} |
||||||
|
|
||||||
|
return container_type; |
||||||
|
} |
||||||
|
|
||||||
|
/* The next 3 functions basically are the reader asking us to use our string
|
||||||
|
* scratchpad for one of these 3 purposes. |
||||||
|
* |
||||||
|
* Note that in the set_number case, we're not going to try interpreting it. |
||||||
|
* We'll keep it as a string, and leave it to the caller to evaluate it. |
||||||
|
*/ |
||||||
|
static void json_reader_set_key(void* userdata) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
state->key = state->string; |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_string(void* userdata) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
grpc_json* json = json_create_and_link(userdata, GRPC_JSON_STRING); |
||||||
|
json->value = (char*) state->string; |
||||||
|
} |
||||||
|
|
||||||
|
static int json_reader_set_number(void* userdata) { |
||||||
|
json_reader_userdata* state = userdata; |
||||||
|
grpc_json* json = json_create_and_link(userdata, GRPC_JSON_NUMBER); |
||||||
|
json->value = (char*) state->string; |
||||||
|
return 1; |
||||||
|
} |
||||||
|
|
||||||
|
/* The object types true, false and null are self-sufficient, and don't need
|
||||||
|
* any more information beside their type. |
||||||
|
*/ |
||||||
|
static void json_reader_set_true(void* userdata) { |
||||||
|
json_create_and_link(userdata, GRPC_JSON_TRUE); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_false(void* userdata) { |
||||||
|
json_create_and_link(userdata, GRPC_JSON_FALSE); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_reader_set_null(void* userdata) { |
||||||
|
json_create_and_link(userdata, GRPC_JSON_NULL); |
||||||
|
} |
||||||
|
|
||||||
|
static grpc_json_reader_vtable reader_vtable = { |
||||||
|
json_reader_string_clear, |
||||||
|
json_reader_string_add_char, |
||||||
|
json_reader_string_add_utf32, |
||||||
|
json_reader_read_char, |
||||||
|
json_reader_container_begins, |
||||||
|
json_reader_container_ends, |
||||||
|
json_reader_set_key, |
||||||
|
json_reader_set_string, |
||||||
|
json_reader_set_number, |
||||||
|
json_reader_set_true, |
||||||
|
json_reader_set_false, |
||||||
|
json_reader_set_null |
||||||
|
}; |
||||||
|
|
||||||
|
/* And finally, let's define our public API. */ |
||||||
|
grpc_json* grpc_json_parse_string_with_len(char* input, size_t size) { |
||||||
|
grpc_json_reader reader; |
||||||
|
json_reader_userdata state; |
||||||
|
grpc_json *json = NULL; |
||||||
|
grpc_json_reader_status status; |
||||||
|
|
||||||
|
if (!input) return NULL; |
||||||
|
|
||||||
|
state.top = state.current_container = state.current_value = NULL; |
||||||
|
state.string = state.key = NULL; |
||||||
|
state.string_ptr = state.input = (gpr_uint8*) input; |
||||||
|
state.remaining_input = size; |
||||||
|
grpc_json_reader_init(&reader, &reader_vtable, &state); |
||||||
|
|
||||||
|
status = grpc_json_reader_run(&reader); |
||||||
|
json = state.top; |
||||||
|
|
||||||
|
if ((status != GRPC_JSON_DONE) && json) { |
||||||
|
grpc_json_destroy(json); |
||||||
|
json = NULL; |
||||||
|
} |
||||||
|
|
||||||
|
return json; |
||||||
|
} |
||||||
|
|
||||||
|
#define UNBOUND_JSON_STRING_LENGTH 0x7fffffff |
||||||
|
|
||||||
|
grpc_json* grpc_json_parse_string(char* input) { |
||||||
|
return grpc_json_parse_string_with_len(input, UNBOUND_JSON_STRING_LENGTH); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_dump_recursive(grpc_json_writer* writer, |
||||||
|
grpc_json* json, int in_object) { |
||||||
|
while (json) { |
||||||
|
if (in_object) grpc_json_writer_object_key(writer, json->key); |
||||||
|
|
||||||
|
switch (json->type) { |
||||||
|
case GRPC_JSON_OBJECT: |
||||||
|
case GRPC_JSON_ARRAY: |
||||||
|
grpc_json_writer_container_begins(writer, json->type); |
||||||
|
if (json->child) |
||||||
|
json_dump_recursive(writer, json->child, |
||||||
|
json->type == GRPC_JSON_OBJECT); |
||||||
|
grpc_json_writer_container_ends(writer, json->type); |
||||||
|
break; |
||||||
|
case GRPC_JSON_STRING: |
||||||
|
grpc_json_writer_value_string(writer, json->value); |
||||||
|
break; |
||||||
|
case GRPC_JSON_NUMBER: |
||||||
|
grpc_json_writer_value_raw(writer, json->value); |
||||||
|
break; |
||||||
|
case GRPC_JSON_TRUE: |
||||||
|
grpc_json_writer_value_raw_with_len(writer, "true", 4); |
||||||
|
break; |
||||||
|
case GRPC_JSON_FALSE: |
||||||
|
grpc_json_writer_value_raw_with_len(writer, "false", 5); |
||||||
|
break; |
||||||
|
case GRPC_JSON_NULL: |
||||||
|
grpc_json_writer_value_raw_with_len(writer, "null", 4); |
||||||
|
break; |
||||||
|
default: |
||||||
|
abort(); |
||||||
|
} |
||||||
|
json = json->next; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
static grpc_json_writer_vtable writer_vtable = { |
||||||
|
json_writer_output_char, |
||||||
|
json_writer_output_string, |
||||||
|
json_writer_output_string_with_len |
||||||
|
}; |
||||||
|
|
||||||
|
char* grpc_json_dump_to_string(grpc_json* json, int indent) { |
||||||
|
grpc_json_writer writer; |
||||||
|
json_writer_userdata state; |
||||||
|
|
||||||
|
state.output = NULL; |
||||||
|
state.free_space = state.string_len = state.allocated = 0; |
||||||
|
grpc_json_writer_init(&writer, indent, &writer_vtable, &state); |
||||||
|
|
||||||
|
json_dump_recursive(&writer, json, 0); |
||||||
|
|
||||||
|
json_writer_output_char(&state, 0); |
||||||
|
|
||||||
|
return state.output; |
||||||
|
} |
@ -0,0 +1,252 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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 <string.h> |
||||||
|
|
||||||
|
#include <grpc/support/port_platform.h> |
||||||
|
|
||||||
|
#include "src/core/json/json_writer.h" |
||||||
|
|
||||||
|
static void json_writer_output_char(grpc_json_writer* writer, char c) { |
||||||
|
writer->vtable->output_char(writer->userdata, c); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_output_string(grpc_json_writer* writer, const char* str) { |
||||||
|
writer->vtable->output_string(writer->userdata, str); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_output_string_with_len(grpc_json_writer* writer, const char* str, size_t len) { |
||||||
|
writer->vtable->output_string_with_len(writer->userdata, str, len); |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_init(grpc_json_writer* writer, int indent, |
||||||
|
grpc_json_writer_vtable* vtable, void* userdata) { |
||||||
|
memset(writer, 0, sizeof(grpc_json_writer)); |
||||||
|
writer->container_empty = 1; |
||||||
|
writer->indent = indent; |
||||||
|
writer->vtable = vtable; |
||||||
|
writer->userdata = userdata; |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_output_indent( |
||||||
|
grpc_json_writer* writer) { |
||||||
|
static const char spacesstr[] = |
||||||
|
" " |
||||||
|
" " |
||||||
|
" " |
||||||
|
" "; |
||||||
|
|
||||||
|
unsigned spaces = writer->depth * writer->indent; |
||||||
|
|
||||||
|
if (writer->indent == 0) return; |
||||||
|
|
||||||
|
if (writer->got_key) { |
||||||
|
json_writer_output_char(writer, ' '); |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
while (spaces >= (sizeof(spacesstr) - 1)) { |
||||||
|
json_writer_output_string_with_len(writer, spacesstr, |
||||||
|
sizeof(spacesstr) - 1); |
||||||
|
spaces -= (sizeof(spacesstr) - 1); |
||||||
|
} |
||||||
|
|
||||||
|
if (spaces == 0) return; |
||||||
|
|
||||||
|
json_writer_output_string_with_len( |
||||||
|
writer, spacesstr + sizeof(spacesstr) - 1 - spaces, spaces); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_value_end(grpc_json_writer* writer) { |
||||||
|
if (writer->container_empty) { |
||||||
|
writer->container_empty = 0; |
||||||
|
if ((writer->indent == 0) || (writer->depth == 0)) return; |
||||||
|
json_writer_output_char(writer, '\n'); |
||||||
|
} else { |
||||||
|
json_writer_output_char(writer, ','); |
||||||
|
if (writer->indent == 0) return; |
||||||
|
json_writer_output_char(writer, '\n'); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_escape_utf16(grpc_json_writer* writer, gpr_uint16 utf16) { |
||||||
|
static const char hex[] = "0123456789abcdef"; |
||||||
|
|
||||||
|
json_writer_output_string_with_len(writer, "\\u", 2); |
||||||
|
json_writer_output_char(writer, hex[(utf16 >> 12) & 0x0f]); |
||||||
|
json_writer_output_char(writer, hex[(utf16 >> 8) & 0x0f]); |
||||||
|
json_writer_output_char(writer, hex[(utf16 >> 4) & 0x0f]); |
||||||
|
json_writer_output_char(writer, hex[(utf16) & 0x0f]); |
||||||
|
} |
||||||
|
|
||||||
|
static void json_writer_escape_string(grpc_json_writer* writer, |
||||||
|
const char* string) { |
||||||
|
json_writer_output_char(writer, '"'); |
||||||
|
|
||||||
|
for (;;) { |
||||||
|
gpr_uint8 c = (gpr_uint8)*string++; |
||||||
|
if (c == 0) { |
||||||
|
break; |
||||||
|
} else if ((c >= 32) && (c <= 127)) { |
||||||
|
if ((c == '\\') || (c == '"')) json_writer_output_char(writer, '\\'); |
||||||
|
json_writer_output_char(writer, c); |
||||||
|
} else if (c < 32) { |
||||||
|
switch (c) { |
||||||
|
case '\b': |
||||||
|
json_writer_output_string_with_len(writer, "\\b", 2); |
||||||
|
break; |
||||||
|
case '\f': |
||||||
|
json_writer_output_string_with_len(writer, "\\f", 2); |
||||||
|
break; |
||||||
|
case '\n': |
||||||
|
json_writer_output_string_with_len(writer, "\\n", 2); |
||||||
|
break; |
||||||
|
case '\r': |
||||||
|
json_writer_output_string_with_len(writer, "\\r", 2); |
||||||
|
break; |
||||||
|
case '\t': |
||||||
|
json_writer_output_string_with_len(writer, "\\t", 2); |
||||||
|
break; |
||||||
|
default: |
||||||
|
json_writer_escape_utf16(writer, c); |
||||||
|
break; |
||||||
|
} |
||||||
|
} else { |
||||||
|
gpr_uint32 utf32 = 0; |
||||||
|
int extra = 0; |
||||||
|
int i; |
||||||
|
int valid = 1; |
||||||
|
if ((c & 0xe0) == 0xc0) { |
||||||
|
utf32 = c & 0x1f; |
||||||
|
extra = 1; |
||||||
|
} else if ((c & 0xf0) == 0xe0) { |
||||||
|
utf32 = c & 0x0f; |
||||||
|
extra = 2; |
||||||
|
} else if ((c & 0xf8) == 0xf0) { |
||||||
|
utf32 = c & 0x07; |
||||||
|
extra = 3; |
||||||
|
} else { |
||||||
|
break; |
||||||
|
} |
||||||
|
for (i = 0; i < extra; i++) { |
||||||
|
utf32 <<= 6; |
||||||
|
c = *string++; |
||||||
|
if ((c & 0xc0) != 0x80) { |
||||||
|
valid = 0; |
||||||
|
break; |
||||||
|
} |
||||||
|
utf32 |= c & 0x3f; |
||||||
|
} |
||||||
|
if (!valid) break; |
||||||
|
/* The range 0xd800 - 0xdfff is reserved by the surrogates ad vitam.
|
||||||
|
* Any other range is technically reserved for future usage, so if we |
||||||
|
* don't want the software to break in the future, we have to allow |
||||||
|
* anything else. The first non-unicode character is 0x110000. */ |
||||||
|
if (((utf32 >= 0xd800) && (utf32 <= 0xdfff)) || |
||||||
|
(utf32 >= 0x110000)) break; |
||||||
|
if (utf32 >= 0x10000) { |
||||||
|
/* If utf32 contains a character that is above 0xffff, it needs to be
|
||||||
|
* broken down into a utf-16 surrogate pair. A surrogate pair is first |
||||||
|
* a high surrogate, followed by a low surrogate. Each surrogate holds |
||||||
|
* 10 bits of usable data, thus allowing a total of 20 bits of data. |
||||||
|
* The high surrogate marker is 0xd800, while the low surrogate marker |
||||||
|
* is 0xdc00. The low 10 bits of each will be the usable data. |
||||||
|
* |
||||||
|
* After re-combining the 20 bits of data, one has to add 0x10000 to |
||||||
|
* the resulting value, in order to obtain the original character. |
||||||
|
* This is obviously because the range 0x0000 - 0xffff can be written |
||||||
|
* without any special trick. |
||||||
|
* |
||||||
|
* Since 0x10ffff is the highest allowed character, we're working in |
||||||
|
* the range 0x00000 - 0xfffff after we decrement it by 0x10000. |
||||||
|
* That range is exactly 20 bits. |
||||||
|
*/ |
||||||
|
utf32 -= 0x10000; |
||||||
|
json_writer_escape_utf16(writer, 0xd800 | (utf32 >> 10)); |
||||||
|
json_writer_escape_utf16(writer, 0xdc00 | (utf32 & 0x3ff)); |
||||||
|
} else { |
||||||
|
json_writer_escape_utf16(writer, utf32); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
json_writer_output_char(writer, '"'); |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_container_begins(grpc_json_writer* writer, grpc_json_type type) { |
||||||
|
if (!writer->got_key) json_writer_value_end(writer); |
||||||
|
json_writer_output_indent(writer); |
||||||
|
json_writer_output_char(writer, type == GRPC_JSON_OBJECT ? '{' : '['); |
||||||
|
writer->container_empty = 1; |
||||||
|
writer->got_key = 0; |
||||||
|
writer->depth++; |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_container_ends(grpc_json_writer* writer, grpc_json_type type) { |
||||||
|
if (writer->indent && !writer->container_empty) |
||||||
|
json_writer_output_char(writer, '\n'); |
||||||
|
writer->depth--; |
||||||
|
if (!writer->container_empty) json_writer_output_indent(writer); |
||||||
|
json_writer_output_char(writer, type == GRPC_JSON_OBJECT ? '}' : ']'); |
||||||
|
writer->container_empty = 0; |
||||||
|
writer->got_key = 0; |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_object_key(grpc_json_writer* writer, const char* string) { |
||||||
|
json_writer_value_end(writer); |
||||||
|
json_writer_output_indent(writer); |
||||||
|
json_writer_escape_string(writer, string); |
||||||
|
json_writer_output_char(writer, ':'); |
||||||
|
writer->got_key = 1; |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_value_raw(grpc_json_writer* writer, const char* string) { |
||||||
|
if (!writer->got_key) json_writer_value_end(writer); |
||||||
|
json_writer_output_indent(writer); |
||||||
|
json_writer_output_string(writer, string); |
||||||
|
writer->got_key = 0; |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_value_raw_with_len(grpc_json_writer* writer, const char* string, size_t len) { |
||||||
|
if (!writer->got_key) json_writer_value_end(writer); |
||||||
|
json_writer_output_indent(writer); |
||||||
|
json_writer_output_string_with_len(writer, string, len); |
||||||
|
writer->got_key = 0; |
||||||
|
} |
||||||
|
|
||||||
|
void grpc_json_writer_value_string(grpc_json_writer* writer, const char* string) { |
||||||
|
if (!writer->got_key) json_writer_value_end(writer); |
||||||
|
json_writer_output_indent(writer); |
||||||
|
json_writer_escape_string(writer, string); |
||||||
|
writer->got_key = 0; |
||||||
|
} |
@ -0,0 +1,93 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
/* The idea of the writer is basically symmetrical of the reader. While the
|
||||||
|
* reader emits various calls to your code, the writer takes basically the |
||||||
|
* same calls and emit json out of it. It doesn't try to make any check on |
||||||
|
* the order of the calls you do on it. Meaning you can theorically force |
||||||
|
* it to generate invalid json. |
||||||
|
* |
||||||
|
* Also, unlike the reader, the writer expects UTF-8 encoded input strings. |
||||||
|
* These strings will be UTF-8 validated, and any invalid character will |
||||||
|
* cut the conversion short, before any invalid UTF-8 sequence, thus forming |
||||||
|
* a valid UTF-8 string overall. |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef __GRPC_SRC_CORE_JSON_JSON_WRITER_H__ |
||||||
|
#define __GRPC_SRC_CORE_JSON_JSON_WRITER_H__ |
||||||
|
|
||||||
|
#include <stdlib.h> |
||||||
|
|
||||||
|
#include "src/core/json/json_common.h" |
||||||
|
|
||||||
|
typedef struct grpc_json_writer_vtable { |
||||||
|
/* Adds a character to the output stream. */ |
||||||
|
void (*output_char)(void* userdata, char); |
||||||
|
/* Adds a zero-terminated string to the output stream. */ |
||||||
|
void (*output_string)(void* userdata, const char* str); |
||||||
|
/* Adds a fixed-length string to the output stream. */ |
||||||
|
void (*output_string_with_len)(void* userdata, const char* str, size_t len); |
||||||
|
|
||||||
|
} grpc_json_writer_vtable; |
||||||
|
|
||||||
|
typedef struct grpc_json_writer { |
||||||
|
void* userdata; |
||||||
|
grpc_json_writer_vtable* vtable; |
||||||
|
int indent; |
||||||
|
int depth; |
||||||
|
int container_empty; |
||||||
|
int got_key; |
||||||
|
} grpc_json_writer; |
||||||
|
|
||||||
|
/* Call this to initialize your writer structure. The indent parameter is
|
||||||
|
* specifying the number of spaces to use for indenting the output. If you |
||||||
|
* use indent=0, then the output will not have any newlines either, thus |
||||||
|
* emitting a condensed json output. |
||||||
|
*/ |
||||||
|
void grpc_json_writer_init(grpc_json_writer* writer, int indent, |
||||||
|
grpc_json_writer_vtable* vtable, void* userdata); |
||||||
|
|
||||||
|
/* Signals the beginning of a container. */ |
||||||
|
void grpc_json_writer_container_begins(grpc_json_writer* writer, grpc_json_type type); |
||||||
|
/* Signals the end of a container. */ |
||||||
|
void grpc_json_writer_container_ends(grpc_json_writer* writer, grpc_json_type type); |
||||||
|
/* Writes down an object key for the next value. */ |
||||||
|
void grpc_json_writer_object_key(grpc_json_writer* writer, const char* string); |
||||||
|
/* Sets a raw value. Useful for numbers. */ |
||||||
|
void grpc_json_writer_value_raw(grpc_json_writer* writer, const char* string); |
||||||
|
/* Sets a raw value with its length. Useful for values like true or false. */ |
||||||
|
void grpc_json_writer_value_raw_with_len(grpc_json_writer* writer, const char* string, size_t len); |
||||||
|
/* Sets a string value. It'll be escaped, and utf-8 validated. */ |
||||||
|
void grpc_json_writer_value_string(grpc_json_writer* writer, const char* string); |
||||||
|
|
||||||
|
#endif /* __GRPC_SRC_CORE_JSON_JSON_WRITER_H__ */ |
@ -0,0 +1,62 @@ |
|||||||
|
// 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. |
||||||
|
|
||||||
|
syntax = "proto3"; |
||||||
|
|
||||||
|
package examples; |
||||||
|
|
||||||
|
// Protocol type definitions |
||||||
|
message StockRequest { |
||||||
|
optional string symbol = 1; |
||||||
|
optional int32 num_trades_to_watch = 2 [default=0]; |
||||||
|
}; |
||||||
|
|
||||||
|
message StockReply { |
||||||
|
optional float price = 1; |
||||||
|
optional string symbol = 2; |
||||||
|
}; |
||||||
|
|
||||||
|
|
||||||
|
// Interface exported by the server |
||||||
|
service Stock { |
||||||
|
// Simple blocking RPC |
||||||
|
rpc GetLastTradePrice(StockRequest) returns (StockReply) { |
||||||
|
}; |
||||||
|
// Bidirectional streaming RPC |
||||||
|
rpc GetLastTradePriceMultiple(stream StockRequest) returns |
||||||
|
(stream StockReply) { |
||||||
|
}; |
||||||
|
// Unidirectional server-to-client streaming RPC |
||||||
|
rpc WatchFutureTrades(StockRequest) returns (stream StockReply) { |
||||||
|
}; |
||||||
|
// Unidirectional client-to-server streaming RPC |
||||||
|
rpc GetHighestTradePrice(stream StockRequest) returns (StockReply) { |
||||||
|
}; |
||||||
|
|
||||||
|
}; |
@ -0,0 +1,43 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
var grpc = require('..'); |
||||||
|
var examples = grpc.load(__dirname + '/stock.proto').examples; |
||||||
|
|
||||||
|
/** |
||||||
|
* This exports a client constructor for the Stock service. The usage looks like |
||||||
|
* |
||||||
|
* var StockClient = require('stock_client.js'); |
||||||
|
* var stockClient = new StockClient(server_address); |
||||||
|
*/ |
||||||
|
module.exports = examples.Stock; |
@ -0,0 +1,83 @@ |
|||||||
|
/* |
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
var _ = require('underscore'); |
||||||
|
var grpc = require('..'); |
||||||
|
var examples = grpc.load(__dirname + '/stock.proto').examples; |
||||||
|
|
||||||
|
var StockServer = grpc.makeServerConstructor([examples.Stock.service]); |
||||||
|
|
||||||
|
function getLastTradePrice(call, callback) { |
||||||
|
callback(null, {price: 88}); |
||||||
|
} |
||||||
|
|
||||||
|
function watchFutureTrades(call) { |
||||||
|
for (var i = 0; i < call.request.num_trades_to_watch; i++) { |
||||||
|
call.write({price: 88.00 + i * 10.00}); |
||||||
|
} |
||||||
|
call.end(); |
||||||
|
} |
||||||
|
|
||||||
|
function getHighestTradePrice(call, callback) { |
||||||
|
var trades = []; |
||||||
|
call.on('data', function(data) { |
||||||
|
trades.push({symbol: data.symbol, price: _.random(0, 100)}); |
||||||
|
}); |
||||||
|
call.on('end', function() { |
||||||
|
if(_.isEmpty(trades)) { |
||||||
|
callback(null, {}); |
||||||
|
} else { |
||||||
|
callback(null, _.max(trades, function(trade){return trade.price;})); |
||||||
|
} |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
function getLastTradePriceMultiple(call) { |
||||||
|
call.on('data', function(data) { |
||||||
|
call.write({price: 88}); |
||||||
|
}); |
||||||
|
call.on('end', function() { |
||||||
|
call.end(); |
||||||
|
}); |
||||||
|
} |
||||||
|
|
||||||
|
var stockServer = new StockServer({ |
||||||
|
'examples.Stock' : { |
||||||
|
getLastTradePrice: getLastTradePrice, |
||||||
|
getLastTradePriceMultiple: getLastTradePriceMultiple, |
||||||
|
watchFutureTrades: watchFutureTrades, |
||||||
|
getHighestTradePrice: getHighestTradePrice |
||||||
|
} |
||||||
|
}); |
||||||
|
|
||||||
|
exports.module = stockServer; |
@ -1,52 +1,52 @@ |
|||||||
<?php |
<?php |
||||||
// DO NOT EDIT! Generated by Protobuf-PHP protoc plugin 1.0 |
// DO NOT EDIT! Generated by Protobuf-PHP protoc plugin 1.0 |
||||||
// Source: third_party/stubby/testing/proto/test.proto |
// Source: test/cpp/interop/test.proto |
||||||
// Date: 2014-12-03 22:02:20 |
// Date: 2015-01-30 23:30:46 |
||||||
|
|
||||||
namespace grpc\testing { |
namespace grpc\testing { |
||||||
|
|
||||||
class TestServiceClient extends \Grpc\BaseStub { |
class TestServiceClient{ |
||||||
|
|
||||||
|
private $rpc_impl; |
||||||
|
|
||||||
|
public function __construct($rpc_impl) { |
||||||
|
$this->rpc_impl = $rpc_impl; |
||||||
|
} |
||||||
/** |
/** |
||||||
* @param proto2\EmptyMessage $input |
* @param grpc\testing\EmptyMessage $input |
||||||
* @return proto2\EmptyMessage |
|
||||||
*/ |
*/ |
||||||
public function EmptyCall(\proto2\EmptyMessage $argument, $metadata = array()) { |
public function EmptyCall(\grpc\testing\EmptyMessage $argument, $metadata = array()) { |
||||||
return $this->_simpleRequest('/TestService/EmptyCall', $argument, '\proto2\EmptyMessage::deserialize', $metadata); |
return $this->rpc_impl->_simpleRequest('/grpc.testing.TestService/EmptyCall', $argument, '\grpc\testing\EmptyMessage::deserialize', $metadata); |
||||||
} |
} |
||||||
/** |
/** |
||||||
* @param grpc\testing\SimpleRequest $input |
* @param grpc\testing\SimpleRequest $input |
||||||
* @return grpc\testing\SimpleResponse |
|
||||||
*/ |
*/ |
||||||
public function UnaryCall(\grpc\testing\SimpleRequest $argument, $metadata = array()) { |
public function UnaryCall(\grpc\testing\SimpleRequest $argument, $metadata = array()) { |
||||||
return $this->_simpleRequest('/TestService/UnaryCall', $argument, '\grpc\testing\SimpleResponse::deserialize', $metadata); |
return $this->rpc_impl->_simpleRequest('/grpc.testing.TestService/UnaryCall', $argument, '\grpc\testing\SimpleResponse::deserialize', $metadata); |
||||||
} |
} |
||||||
/** |
/** |
||||||
* @param grpc\testing\StreamingOutputCallRequest $input |
* @param grpc\testing\StreamingOutputCallRequest $input |
||||||
* @return grpc\testing\StreamingOutputCallResponse |
|
||||||
*/ |
*/ |
||||||
public function StreamingOutputCall($argument, $metadata = array()) { |
public function StreamingOutputCall($argument, $metadata = array()) { |
||||||
return $this->_serverStreamRequest('/TestService/StreamingOutputCall', $argument, '\grpc\testing\StreamingOutputCallResponse::deserialize', $metadata); |
return $this->rpc_impl->_serverStreamRequest('/grpc.testing.TestService/StreamingOutputCall', $argument, '\grpc\testing\StreamingOutputCallResponse::deserialize', $metadata); |
||||||
} |
} |
||||||
/** |
/** |
||||||
* @param grpc\testing\StreamingInputCallRequest $input |
* @param grpc\testing\StreamingInputCallRequest $input |
||||||
* @return grpc\testing\StreamingInputCallResponse |
|
||||||
*/ |
*/ |
||||||
public function StreamingInputCall($arguments, $metadata = array()) { |
public function StreamingInputCall($arguments, $metadata = array()) { |
||||||
return $this->_clientStreamRequest('/TestService/StreamingInputCall', $arguments, '\grpc\testing\StreamingInputCallResponse::deserialize', $metadata); |
return $this->rpc_impl->_clientStreamRequest('/grpc.testing.TestService/StreamingInputCall', $arguments, '\grpc\testing\StreamingInputCallResponse::deserialize', $metadata); |
||||||
} |
} |
||||||
/** |
/** |
||||||
* @param grpc\testing\StreamingOutputCallRequest $input |
* @param grpc\testing\StreamingOutputCallRequest $input |
||||||
* @return grpc\testing\StreamingOutputCallResponse |
|
||||||
*/ |
*/ |
||||||
public function FullDuplexCall($metadata = array()) { |
public function FullDuplexCall($metadata = array()) { |
||||||
return $this->_bidiRequest('/TestService/FullDuplexCall', '\grpc\testing\StreamingOutputCallResponse::deserialize', $metadata); |
return $this->rpc_impl->_bidiRequest('/grpc.testing.TestService/FullDuplexCall', '\grpc\testing\StreamingOutputCallResponse::deserialize', $metadata); |
||||||
} |
} |
||||||
/** |
/** |
||||||
* @param grpc\testing\StreamingOutputCallRequest $input |
* @param grpc\testing\StreamingOutputCallRequest $input |
||||||
* @return grpc\testing\StreamingOutputCallResponse |
|
||||||
*/ |
*/ |
||||||
public function HalfDuplexCall($metadata = array()) { |
public function HalfDuplexCall($metadata = array()) { |
||||||
return $this->_bidiRequest('/TestService/HalfDuplexCall', '\grpc\testing\StreamingOutputCallResponse::deserialize', $metadata); |
return $this->rpc_impl->_bidiRequest('/grpc.testing.TestService/HalfDuplexCall', '\grpc\testing\StreamingOutputCallResponse::deserialize', $metadata); |
||||||
} |
} |
||||||
} |
} |
||||||
} |
} |
||||||
|
@ -0,0 +1,86 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""A setup module for the GRPC Python package.""" |
||||||
|
|
||||||
|
from distutils import core as _core |
||||||
|
|
||||||
|
_EXTENSION_SOURCES = ( |
||||||
|
'src/_adapter/_c.c', |
||||||
|
'src/_adapter/_call.c', |
||||||
|
'src/_adapter/_channel.c', |
||||||
|
'src/_adapter/_completion_queue.c', |
||||||
|
'src/_adapter/_error.c', |
||||||
|
'src/_adapter/_server.c', |
||||||
|
) |
||||||
|
|
||||||
|
_EXTENSION_INCLUDE_DIRECTORIES = ( |
||||||
|
'src', |
||||||
|
# TODO(nathaniel): Can this path specification be made to work? |
||||||
|
#'../../include', |
||||||
|
) |
||||||
|
|
||||||
|
_EXTENSION_LIBRARIES = ( |
||||||
|
'gpr', |
||||||
|
'grpc', |
||||||
|
) |
||||||
|
|
||||||
|
_EXTENSION_LIBRARY_DIRECTORIES = ( |
||||||
|
# TODO(nathaniel): Can this path specification be made to work? |
||||||
|
#'../../libs/dbg', |
||||||
|
) |
||||||
|
|
||||||
|
_EXTENSION_MODULE = _core.Extension( |
||||||
|
'_adapter._c', sources=list(_EXTENSION_SOURCES), |
||||||
|
include_dirs=_EXTENSION_INCLUDE_DIRECTORIES, |
||||||
|
libraries=_EXTENSION_LIBRARIES, |
||||||
|
library_dirs=_EXTENSION_LIBRARY_DIRECTORIES) |
||||||
|
|
||||||
|
_PACKAGES=( |
||||||
|
'_adapter', |
||||||
|
'_framework', |
||||||
|
'_framework.base', |
||||||
|
'_framework.base.packets', |
||||||
|
'_framework.common', |
||||||
|
'_framework.face', |
||||||
|
'_framework.face.testing', |
||||||
|
'_framework.foundation', |
||||||
|
'_junkdrawer', |
||||||
|
) |
||||||
|
|
||||||
|
_PACKAGE_DIRECTORIES = { |
||||||
|
'_adapter': 'src/_adapter', |
||||||
|
'_framework': 'src/_framework', |
||||||
|
'_junkdrawer': 'src/_junkdrawer', |
||||||
|
} |
||||||
|
|
||||||
|
_core.setup( |
||||||
|
name='grpc', version='0.0.1', |
||||||
|
ext_modules=[_EXTENSION_MODULE], packages=_PACKAGES, |
||||||
|
package_dir=_PACKAGE_DIRECTORIES) |
@ -0,0 +1,17 @@ |
|||||||
|
"""One of the tests of the Face layer of RPC Framework.""" |
||||||
|
|
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _face_test_case |
||||||
|
from _framework.face.testing import blocking_invocation_inline_service_test_case as test_case |
||||||
|
|
||||||
|
|
||||||
|
class BlockingInvocationInlineServiceTest( |
||||||
|
_face_test_case.FaceTestCase, |
||||||
|
test_case.BlockingInvocationInlineServiceTestCase, |
||||||
|
unittest.TestCase): |
||||||
|
pass |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,77 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
#include "_adapter/_completion_queue.h" |
||||||
|
#include "_adapter/_channel.h" |
||||||
|
#include "_adapter/_call.h" |
||||||
|
#include "_adapter/_server.h" |
||||||
|
|
||||||
|
static PyObject *init(PyObject *self, PyObject *args) { |
||||||
|
grpc_init(); |
||||||
|
Py_RETURN_NONE; |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *shutdown(PyObject *self, PyObject *args) { |
||||||
|
grpc_shutdown(); |
||||||
|
Py_RETURN_NONE; |
||||||
|
} |
||||||
|
|
||||||
|
static PyMethodDef _c_methods[] = { |
||||||
|
{"init", init, METH_VARARGS, "Initialize the module's static state."}, |
||||||
|
{"shut_down", shutdown, METH_VARARGS, |
||||||
|
"Shut down the module's static state."}, |
||||||
|
{NULL}, |
||||||
|
}; |
||||||
|
|
||||||
|
PyMODINIT_FUNC init_c(void) { |
||||||
|
PyObject *module; |
||||||
|
|
||||||
|
module = Py_InitModule3("_c", _c_methods, |
||||||
|
"Wrappings of C structures and functions."); |
||||||
|
|
||||||
|
if (pygrpc_add_completion_queue(module) == -1) { |
||||||
|
return; |
||||||
|
} |
||||||
|
if (pygrpc_add_channel(module) == -1) { |
||||||
|
return; |
||||||
|
} |
||||||
|
if (pygrpc_add_call(module) == -1) { |
||||||
|
return; |
||||||
|
} |
||||||
|
if (pygrpc_add_server(module) == -1) { |
||||||
|
return; |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,141 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Tests for _adapter._c.""" |
||||||
|
|
||||||
|
import threading |
||||||
|
import time |
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _c |
||||||
|
from _adapter import _datatypes |
||||||
|
|
||||||
|
_TIMEOUT = 3 |
||||||
|
_FUTURE = time.time() + 60 * 60 * 24 |
||||||
|
_IDEMPOTENCE_DEMONSTRATION = 7 |
||||||
|
|
||||||
|
|
||||||
|
class _CTest(unittest.TestCase): |
||||||
|
|
||||||
|
def testUpAndDown(self): |
||||||
|
_c.init() |
||||||
|
_c.shut_down() |
||||||
|
|
||||||
|
def testCompletionQueue(self): |
||||||
|
_c.init() |
||||||
|
|
||||||
|
completion_queue = _c.CompletionQueue() |
||||||
|
event = completion_queue.get(0) |
||||||
|
self.assertIsNone(event) |
||||||
|
event = completion_queue.get(time.time()) |
||||||
|
self.assertIsNone(event) |
||||||
|
event = completion_queue.get(time.time() + _TIMEOUT) |
||||||
|
self.assertIsNone(event) |
||||||
|
completion_queue.stop() |
||||||
|
for _ in range(_IDEMPOTENCE_DEMONSTRATION): |
||||||
|
event = completion_queue.get(time.time() + _TIMEOUT) |
||||||
|
self.assertIs(event.kind, _datatypes.Event.Kind.STOP) |
||||||
|
|
||||||
|
del completion_queue |
||||||
|
del event |
||||||
|
|
||||||
|
_c.shut_down() |
||||||
|
|
||||||
|
def testChannel(self): |
||||||
|
_c.init() |
||||||
|
|
||||||
|
channel = _c.Channel('test host:12345') |
||||||
|
del channel |
||||||
|
|
||||||
|
_c.shut_down() |
||||||
|
|
||||||
|
def testCall(self): |
||||||
|
method = 'test method' |
||||||
|
host = 'test host' |
||||||
|
|
||||||
|
_c.init() |
||||||
|
|
||||||
|
channel = _c.Channel('%s:%d' % (host, 12345)) |
||||||
|
call = _c.Call(channel, method, host, time.time() + _TIMEOUT) |
||||||
|
del call |
||||||
|
del channel |
||||||
|
|
||||||
|
_c.shut_down() |
||||||
|
|
||||||
|
def testServer(self): |
||||||
|
_c.init() |
||||||
|
|
||||||
|
completion_queue = _c.CompletionQueue() |
||||||
|
server = _c.Server(completion_queue) |
||||||
|
server.add_http2_addr('[::]:0') |
||||||
|
server.start() |
||||||
|
server.stop() |
||||||
|
completion_queue.stop() |
||||||
|
del server |
||||||
|
del completion_queue |
||||||
|
|
||||||
|
service_tag = object() |
||||||
|
completion_queue = _c.CompletionQueue() |
||||||
|
server = _c.Server(completion_queue) |
||||||
|
server.add_http2_addr('[::]:0') |
||||||
|
server.start() |
||||||
|
server.service(service_tag) |
||||||
|
server.stop() |
||||||
|
completion_queue.stop() |
||||||
|
event = completion_queue.get(time.time() + _TIMEOUT) |
||||||
|
self.assertIs(event.kind, _datatypes.Event.Kind.SERVICE_ACCEPTED) |
||||||
|
self.assertIs(event.tag, service_tag) |
||||||
|
self.assertIsNone(event.service_acceptance) |
||||||
|
for _ in range(_IDEMPOTENCE_DEMONSTRATION): |
||||||
|
event = completion_queue.get(time.time() + _TIMEOUT) |
||||||
|
self.assertIs(event.kind, _datatypes.Event.Kind.STOP) |
||||||
|
del server |
||||||
|
del completion_queue |
||||||
|
|
||||||
|
completion_queue = _c.CompletionQueue() |
||||||
|
server = _c.Server(completion_queue) |
||||||
|
server.add_http2_addr('[::]:0') |
||||||
|
server.start() |
||||||
|
thread = threading.Thread(target=completion_queue.get, args=(_FUTURE,)) |
||||||
|
thread.start() |
||||||
|
time.sleep(1) |
||||||
|
server.stop() |
||||||
|
completion_queue.stop() |
||||||
|
for _ in range(_IDEMPOTENCE_DEMONSTRATION): |
||||||
|
event = completion_queue.get(time.time() + _TIMEOUT) |
||||||
|
self.assertIs(event.kind, _datatypes.Event.Kind.STOP) |
||||||
|
thread.join() |
||||||
|
del server |
||||||
|
del completion_queue |
||||||
|
|
||||||
|
_c.shut_down() |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,292 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2014, 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 "_adapter/_call.h" |
||||||
|
|
||||||
|
#include <math.h> |
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
#include "_adapter/_channel.h" |
||||||
|
#include "_adapter/_completion_queue.h" |
||||||
|
#include "_adapter/_error.h" |
||||||
|
|
||||||
|
static int pygrpc_call_init(Call *self, PyObject *args, PyObject *kwds) { |
||||||
|
const PyObject *channel; |
||||||
|
const char *method; |
||||||
|
const char *host; |
||||||
|
const double deadline; |
||||||
|
|
||||||
|
if (!PyArg_ParseTuple(args, "O!ssd", &pygrpc_ChannelType, &channel, &method, |
||||||
|
&host, &deadline)) { |
||||||
|
self->c_call = NULL; |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
/* TODO(nathaniel): Hoist the gpr_timespec <-> PyFloat arithmetic into its own
|
||||||
|
* function with its own test coverage. |
||||||
|
*/ |
||||||
|
self->c_call = |
||||||
|
grpc_channel_create_call(((Channel *)channel)->c_channel, method, host, |
||||||
|
gpr_time_from_nanos(deadline * GPR_NS_PER_SEC)); |
||||||
|
|
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static void pygrpc_call_dealloc(Call *self) { |
||||||
|
if (self->c_call != NULL) { |
||||||
|
grpc_call_destroy(self->c_call); |
||||||
|
} |
||||||
|
self->ob_type->tp_free((PyObject *)self); |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_invoke(Call *self, PyObject *args) { |
||||||
|
const PyObject *completion_queue; |
||||||
|
const PyObject *metadata_tag; |
||||||
|
const PyObject *finish_tag; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "O!OO", &pygrpc_CompletionQueueType, |
||||||
|
&completion_queue, &metadata_tag, &finish_tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
call_error = grpc_call_invoke( |
||||||
|
self->c_call, ((CompletionQueue *)completion_queue)->c_completion_queue, |
||||||
|
(void *)metadata_tag, (void *)finish_tag, 0); |
||||||
|
|
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(metadata_tag); |
||||||
|
Py_INCREF(finish_tag); |
||||||
|
} |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_write(Call *self, PyObject *args) { |
||||||
|
const char *bytes; |
||||||
|
int length; |
||||||
|
const PyObject *tag; |
||||||
|
gpr_slice slice; |
||||||
|
grpc_byte_buffer *byte_buffer; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "s#O", &bytes, &length, &tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
slice = gpr_slice_from_copied_buffer(bytes, length); |
||||||
|
byte_buffer = grpc_byte_buffer_create(&slice, 1); |
||||||
|
gpr_slice_unref(slice); |
||||||
|
|
||||||
|
call_error = grpc_call_start_write(self->c_call, byte_buffer, (void *)tag, 0); |
||||||
|
|
||||||
|
grpc_byte_buffer_destroy(byte_buffer); |
||||||
|
|
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(tag); |
||||||
|
} |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_complete(Call *self, PyObject *args) { |
||||||
|
const PyObject *tag; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "O", &tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
call_error = grpc_call_writes_done(self->c_call, (void *)tag); |
||||||
|
|
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(tag); |
||||||
|
} |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_accept(Call *self, PyObject *args) { |
||||||
|
const PyObject *completion_queue; |
||||||
|
const PyObject *tag; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "O!O", &pygrpc_CompletionQueueType, |
||||||
|
&completion_queue, &tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
call_error = grpc_call_server_accept( |
||||||
|
self->c_call, ((CompletionQueue *)completion_queue)->c_completion_queue, |
||||||
|
(void *)tag); |
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
|
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(tag); |
||||||
|
} |
||||||
|
|
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_premetadata(Call *self, PyObject *args) { |
||||||
|
/* TODO(b/18702680): Actually support metadata. */ |
||||||
|
return pygrpc_translate_call_error( |
||||||
|
grpc_call_server_end_initial_metadata(self->c_call, 0)); |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_read(Call *self, PyObject *args) { |
||||||
|
const PyObject *tag; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "O", &tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
call_error = grpc_call_start_read(self->c_call, (void *)tag); |
||||||
|
|
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(tag); |
||||||
|
} |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_status(Call *self, PyObject *args) { |
||||||
|
PyObject *status; |
||||||
|
PyObject *code; |
||||||
|
PyObject *details; |
||||||
|
const PyObject *tag; |
||||||
|
grpc_status_code c_code; |
||||||
|
char *c_message; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "OO", &status, &tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
code = PyObject_GetAttrString(status, "code"); |
||||||
|
details = PyObject_GetAttrString(status, "details"); |
||||||
|
c_code = PyInt_AsLong(code); |
||||||
|
c_message = PyBytes_AsString(details); |
||||||
|
Py_DECREF(code); |
||||||
|
Py_DECREF(details); |
||||||
|
|
||||||
|
call_error = grpc_call_start_write_status(self->c_call, c_code, c_message, |
||||||
|
(void *)tag); |
||||||
|
|
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(tag); |
||||||
|
} |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_call_cancel(Call *self) { |
||||||
|
return pygrpc_translate_call_error(grpc_call_cancel(self->c_call)); |
||||||
|
} |
||||||
|
|
||||||
|
static PyMethodDef methods[] = { |
||||||
|
{"invoke", (PyCFunction)pygrpc_call_invoke, METH_VARARGS, |
||||||
|
"Invoke this call."}, |
||||||
|
{"write", (PyCFunction)pygrpc_call_write, METH_VARARGS, |
||||||
|
"Write bytes to this call."}, |
||||||
|
{"complete", (PyCFunction)pygrpc_call_complete, METH_VARARGS, |
||||||
|
"Complete writes to this call."}, |
||||||
|
{"accept", (PyCFunction)pygrpc_call_accept, METH_VARARGS, "Accept an RPC."}, |
||||||
|
{"premetadata", (PyCFunction)pygrpc_call_premetadata, METH_VARARGS, |
||||||
|
"Indicate the end of leading metadata in the response."}, |
||||||
|
{"read", (PyCFunction)pygrpc_call_read, METH_VARARGS, |
||||||
|
"Read bytes from this call."}, |
||||||
|
{"status", (PyCFunction)pygrpc_call_status, METH_VARARGS, |
||||||
|
"Report this call's status."}, |
||||||
|
{"cancel", (PyCFunction)pygrpc_call_cancel, METH_NOARGS, |
||||||
|
"Cancel this call."}, |
||||||
|
{NULL}}; |
||||||
|
|
||||||
|
PyTypeObject pygrpc_CallType = { |
||||||
|
PyObject_HEAD_INIT(NULL)0, /*ob_size*/ |
||||||
|
"_grpc.Call", /*tp_name*/ |
||||||
|
sizeof(Call), /*tp_basicsize*/ |
||||||
|
0, /*tp_itemsize*/ |
||||||
|
(destructor)pygrpc_call_dealloc, /*tp_dealloc*/ |
||||||
|
0, /*tp_print*/ |
||||||
|
0, /*tp_getattr*/ |
||||||
|
0, /*tp_setattr*/ |
||||||
|
0, /*tp_compare*/ |
||||||
|
0, /*tp_repr*/ |
||||||
|
0, /*tp_as_number*/ |
||||||
|
0, /*tp_as_sequence*/ |
||||||
|
0, /*tp_as_mapping*/ |
||||||
|
0, /*tp_hash */ |
||||||
|
0, /*tp_call*/ |
||||||
|
0, /*tp_str*/ |
||||||
|
0, /*tp_getattro*/ |
||||||
|
0, /*tp_setattro*/ |
||||||
|
0, /*tp_as_buffer*/ |
||||||
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/ |
||||||
|
"Wrapping of grpc_call.", /* tp_doc */ |
||||||
|
0, /* tp_traverse */ |
||||||
|
0, /* tp_clear */ |
||||||
|
0, /* tp_richcompare */ |
||||||
|
0, /* tp_weaklistoffset */ |
||||||
|
0, /* tp_iter */ |
||||||
|
0, /* tp_iternext */ |
||||||
|
methods, /* tp_methods */ |
||||||
|
0, /* tp_members */ |
||||||
|
0, /* tp_getset */ |
||||||
|
0, /* tp_base */ |
||||||
|
0, /* tp_dict */ |
||||||
|
0, /* tp_descr_get */ |
||||||
|
0, /* tp_descr_set */ |
||||||
|
0, /* tp_dictoffset */ |
||||||
|
(initproc)pygrpc_call_init, /* tp_init */ |
||||||
|
}; |
||||||
|
|
||||||
|
int pygrpc_add_call(PyObject *module) { |
||||||
|
pygrpc_CallType.tp_new = PyType_GenericNew; |
||||||
|
if (PyType_Ready(&pygrpc_CallType) < 0) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "Error defining pygrpc_CallType!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (PyModule_AddObject(module, "Call", (PyObject *)&pygrpc_CallType) == -1) { |
||||||
|
PyErr_SetString(PyExc_ImportError, "Couldn't add Call type to module!"); |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
@ -0,0 +1,46 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef _ADAPTER__CALL_H_ |
||||||
|
#define _ADAPTER__CALL_H_ |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
typedef struct { PyObject_HEAD grpc_call *c_call; } Call; |
||||||
|
|
||||||
|
PyTypeObject pygrpc_CallType; |
||||||
|
|
||||||
|
int pygrpc_add_call(PyObject *module); |
||||||
|
|
||||||
|
#endif /* _ADAPTER__CALL_H_ */ |
@ -0,0 +1,109 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 "_adapter/_channel.h" |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
static int pygrpc_channel_init(Channel *self, PyObject *args, PyObject *kwds) { |
||||||
|
const char *hostport; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "s", &hostport))) { |
||||||
|
self->c_channel = NULL; |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
self->c_channel = grpc_channel_create(hostport, NULL); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static void pygrpc_channel_dealloc(Channel *self) { |
||||||
|
if (self->c_channel != NULL) { |
||||||
|
grpc_channel_destroy(self->c_channel); |
||||||
|
} |
||||||
|
self->ob_type->tp_free((PyObject *)self); |
||||||
|
} |
||||||
|
|
||||||
|
PyTypeObject pygrpc_ChannelType = { |
||||||
|
PyObject_HEAD_INIT(NULL)0, /*ob_size*/ |
||||||
|
"_grpc.Channel", /*tp_name*/ |
||||||
|
sizeof(Channel), /*tp_basicsize*/ |
||||||
|
0, /*tp_itemsize*/ |
||||||
|
(destructor)pygrpc_channel_dealloc, /*tp_dealloc*/ |
||||||
|
0, /*tp_print*/ |
||||||
|
0, /*tp_getattr*/ |
||||||
|
0, /*tp_setattr*/ |
||||||
|
0, /*tp_compare*/ |
||||||
|
0, /*tp_repr*/ |
||||||
|
0, /*tp_as_number*/ |
||||||
|
0, /*tp_as_sequence*/ |
||||||
|
0, /*tp_as_mapping*/ |
||||||
|
0, /*tp_hash */ |
||||||
|
0, /*tp_call*/ |
||||||
|
0, /*tp_str*/ |
||||||
|
0, /*tp_getattro*/ |
||||||
|
0, /*tp_setattro*/ |
||||||
|
0, /*tp_as_buffer*/ |
||||||
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/ |
||||||
|
"Wrapping of grpc_channel.", /* tp_doc */ |
||||||
|
0, /* tp_traverse */ |
||||||
|
0, /* tp_clear */ |
||||||
|
0, /* tp_richcompare */ |
||||||
|
0, /* tp_weaklistoffset */ |
||||||
|
0, /* tp_iter */ |
||||||
|
0, /* tp_iternext */ |
||||||
|
0, /* tp_methods */ |
||||||
|
0, /* tp_members */ |
||||||
|
0, /* tp_getset */ |
||||||
|
0, /* tp_base */ |
||||||
|
0, /* tp_dict */ |
||||||
|
0, /* tp_descr_get */ |
||||||
|
0, /* tp_descr_set */ |
||||||
|
0, /* tp_dictoffset */ |
||||||
|
(initproc)pygrpc_channel_init, /* tp_init */ |
||||||
|
}; |
||||||
|
|
||||||
|
int pygrpc_add_channel(PyObject *module) { |
||||||
|
pygrpc_ChannelType.tp_new = PyType_GenericNew; |
||||||
|
if (PyType_Ready(&pygrpc_ChannelType) < 0) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "Error defining pygrpc_ChannelType!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (PyModule_AddObject(module, "Channel", (PyObject *)&pygrpc_ChannelType) == |
||||||
|
-1) { |
||||||
|
PyErr_SetString(PyExc_ImportError, "Couldn't add Channel type to module!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
@ -0,0 +1,46 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef _ADAPTER__CHANNEL_H_ |
||||||
|
#define _ADAPTER__CHANNEL_H_ |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
typedef struct { PyObject_HEAD grpc_channel *c_channel; } Channel; |
||||||
|
|
||||||
|
PyTypeObject pygrpc_ChannelType; |
||||||
|
|
||||||
|
int pygrpc_add_channel(PyObject *module); |
||||||
|
|
||||||
|
#endif /* _ADAPTER__CHANNEL_H_ */ |
@ -0,0 +1,76 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""State used by both invocation-side and service-side code.""" |
||||||
|
|
||||||
|
import enum |
||||||
|
|
||||||
|
|
||||||
|
@enum.unique |
||||||
|
class HighWrite(enum.Enum): |
||||||
|
"""The possible categories of high-level write state.""" |
||||||
|
|
||||||
|
OPEN = 'OPEN' |
||||||
|
CLOSED = 'CLOSED' |
||||||
|
|
||||||
|
|
||||||
|
class WriteState(object): |
||||||
|
"""A description of the state of writing to an RPC. |
||||||
|
|
||||||
|
Attributes: |
||||||
|
low: A side-specific value describing the low-level state of writing. |
||||||
|
high: A HighWrite value describing the high-level state of writing. |
||||||
|
pending: A list of bytestrings for the RPC waiting to be written to the |
||||||
|
other side of the RPC. |
||||||
|
""" |
||||||
|
|
||||||
|
def __init__(self, low, high, pending): |
||||||
|
self.low = low |
||||||
|
self.high = high |
||||||
|
self.pending = pending |
||||||
|
|
||||||
|
|
||||||
|
class CommonRPCState(object): |
||||||
|
"""A description of an RPC's state. |
||||||
|
|
||||||
|
Attributes: |
||||||
|
write: A WriteState describing the state of writing to the RPC. |
||||||
|
sequence_number: The lowest-unused sequence number for use in generating |
||||||
|
tickets locally describing the progress of the RPC. |
||||||
|
deserializer: The behavior to be used to deserialize payload bytestreams |
||||||
|
taken off the wire. |
||||||
|
serializer: The behavior to be used to serialize payloads to be sent on the |
||||||
|
wire. |
||||||
|
""" |
||||||
|
|
||||||
|
def __init__(self, write, sequence_number, deserializer, serializer): |
||||||
|
self.write = write |
||||||
|
self.sequence_number = sequence_number |
||||||
|
self.deserializer = deserializer |
||||||
|
self.serializer = serializer |
@ -0,0 +1,541 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 "_adapter/_completion_queue.h" |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
|
||||||
|
#include "_adapter/_call.h" |
||||||
|
|
||||||
|
static PyObject *status_class; |
||||||
|
static PyObject *service_acceptance_class; |
||||||
|
static PyObject *event_class; |
||||||
|
|
||||||
|
static PyObject *ok_status_code; |
||||||
|
static PyObject *cancelled_status_code; |
||||||
|
static PyObject *unknown_status_code; |
||||||
|
static PyObject *invalid_argument_status_code; |
||||||
|
static PyObject *expired_status_code; |
||||||
|
static PyObject *not_found_status_code; |
||||||
|
static PyObject *already_exists_status_code; |
||||||
|
static PyObject *permission_denied_status_code; |
||||||
|
static PyObject *unauthenticated_status_code; |
||||||
|
static PyObject *resource_exhausted_status_code; |
||||||
|
static PyObject *failed_precondition_status_code; |
||||||
|
static PyObject *aborted_status_code; |
||||||
|
static PyObject *out_of_range_status_code; |
||||||
|
static PyObject *unimplemented_status_code; |
||||||
|
static PyObject *internal_error_status_code; |
||||||
|
static PyObject *unavailable_status_code; |
||||||
|
static PyObject *data_loss_status_code; |
||||||
|
|
||||||
|
static PyObject *stop_event_kind; |
||||||
|
static PyObject *write_event_kind; |
||||||
|
static PyObject *complete_event_kind; |
||||||
|
static PyObject *service_event_kind; |
||||||
|
static PyObject *read_event_kind; |
||||||
|
static PyObject *metadata_event_kind; |
||||||
|
static PyObject *finish_event_kind; |
||||||
|
|
||||||
|
static PyObject *pygrpc_as_py_time(gpr_timespec *timespec) { |
||||||
|
return Py_BuildValue("f", |
||||||
|
timespec->tv_sec + ((double)timespec->tv_nsec) / 1.0E9); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_status_code(grpc_status_code c_status_code) { |
||||||
|
switch (c_status_code) { |
||||||
|
case GRPC_STATUS_OK: |
||||||
|
return ok_status_code; |
||||||
|
case GRPC_STATUS_CANCELLED: |
||||||
|
return cancelled_status_code; |
||||||
|
case GRPC_STATUS_UNKNOWN: |
||||||
|
return unknown_status_code; |
||||||
|
case GRPC_STATUS_INVALID_ARGUMENT: |
||||||
|
return invalid_argument_status_code; |
||||||
|
case GRPC_STATUS_DEADLINE_EXCEEDED: |
||||||
|
return expired_status_code; |
||||||
|
case GRPC_STATUS_NOT_FOUND: |
||||||
|
return not_found_status_code; |
||||||
|
case GRPC_STATUS_ALREADY_EXISTS: |
||||||
|
return already_exists_status_code; |
||||||
|
case GRPC_STATUS_PERMISSION_DENIED: |
||||||
|
return permission_denied_status_code; |
||||||
|
case GRPC_STATUS_UNAUTHENTICATED: |
||||||
|
return unauthenticated_status_code; |
||||||
|
case GRPC_STATUS_RESOURCE_EXHAUSTED: |
||||||
|
return resource_exhausted_status_code; |
||||||
|
case GRPC_STATUS_FAILED_PRECONDITION: |
||||||
|
return failed_precondition_status_code; |
||||||
|
case GRPC_STATUS_ABORTED: |
||||||
|
return aborted_status_code; |
||||||
|
case GRPC_STATUS_OUT_OF_RANGE: |
||||||
|
return out_of_range_status_code; |
||||||
|
case GRPC_STATUS_UNIMPLEMENTED: |
||||||
|
return unimplemented_status_code; |
||||||
|
case GRPC_STATUS_INTERNAL: |
||||||
|
return internal_error_status_code; |
||||||
|
case GRPC_STATUS_UNAVAILABLE: |
||||||
|
return unavailable_status_code; |
||||||
|
case GRPC_STATUS_DATA_LOSS: |
||||||
|
return data_loss_status_code; |
||||||
|
default: |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_stop_event_args(grpc_event *c_event) { |
||||||
|
return Py_BuildValue("(OOOOOOO)", stop_event_kind, Py_None, Py_None, Py_None, |
||||||
|
Py_None, Py_None, Py_None); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_write_event_args(grpc_event *c_event) { |
||||||
|
PyObject *write_accepted = |
||||||
|
c_event->data.write_accepted == GRPC_OP_OK ? Py_True : Py_False; |
||||||
|
return Py_BuildValue("(OOOOOOO)", write_event_kind, (PyObject *)c_event->tag, |
||||||
|
write_accepted, Py_None, Py_None, Py_None, Py_None); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_complete_event_args(grpc_event *c_event) { |
||||||
|
PyObject *complete_accepted = |
||||||
|
c_event->data.finish_accepted == GRPC_OP_OK ? Py_True : Py_False; |
||||||
|
return Py_BuildValue("(OOOOOOO)", complete_event_kind, |
||||||
|
(PyObject *)c_event->tag, Py_None, complete_accepted, |
||||||
|
Py_None, Py_None, Py_None); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_service_event_args(grpc_event *c_event) { |
||||||
|
if (c_event->data.server_rpc_new.method == NULL) { |
||||||
|
return Py_BuildValue("(OOOOOOO)", service_event_kind, c_event->tag, |
||||||
|
Py_None, Py_None, Py_None, Py_None, Py_None); |
||||||
|
} else { |
||||||
|
PyObject *method = PyBytes_FromString(c_event->data.server_rpc_new.method); |
||||||
|
PyObject *host = PyBytes_FromString(c_event->data.server_rpc_new.host); |
||||||
|
PyObject *service_deadline = |
||||||
|
pygrpc_as_py_time(&c_event->data.server_rpc_new.deadline); |
||||||
|
|
||||||
|
Call *call; |
||||||
|
PyObject *service_acceptance_args; |
||||||
|
PyObject *service_acceptance; |
||||||
|
PyObject *event_args; |
||||||
|
|
||||||
|
call = PyObject_New(Call, &pygrpc_CallType); |
||||||
|
call->c_call = c_event->call; |
||||||
|
|
||||||
|
service_acceptance_args = |
||||||
|
Py_BuildValue("(OOOO)", call, method, host, service_deadline); |
||||||
|
Py_DECREF(call); |
||||||
|
Py_DECREF(method); |
||||||
|
Py_DECREF(host); |
||||||
|
Py_DECREF(service_deadline); |
||||||
|
|
||||||
|
service_acceptance = |
||||||
|
PyObject_CallObject(service_acceptance_class, service_acceptance_args); |
||||||
|
Py_DECREF(service_acceptance_args); |
||||||
|
|
||||||
|
event_args = Py_BuildValue("(OOOOOOO)", service_event_kind, |
||||||
|
(PyObject *)c_event->tag, Py_None, Py_None, |
||||||
|
service_acceptance, Py_None, Py_None); |
||||||
|
Py_DECREF(service_acceptance); |
||||||
|
return event_args; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_read_event_args(grpc_event *c_event) { |
||||||
|
if (c_event->data.read == NULL) { |
||||||
|
return Py_BuildValue("(OOOOOOO)", read_event_kind, |
||||||
|
(PyObject *)c_event->tag, Py_None, Py_None, Py_None, |
||||||
|
Py_None, Py_None); |
||||||
|
} else { |
||||||
|
size_t length; |
||||||
|
size_t offset; |
||||||
|
grpc_byte_buffer_reader *reader; |
||||||
|
gpr_slice slice; |
||||||
|
char *c_bytes; |
||||||
|
PyObject *bytes; |
||||||
|
PyObject *event_args; |
||||||
|
|
||||||
|
length = grpc_byte_buffer_length(c_event->data.read); |
||||||
|
reader = grpc_byte_buffer_reader_create(c_event->data.read); |
||||||
|
c_bytes = gpr_malloc(length); |
||||||
|
offset = 0; |
||||||
|
while (grpc_byte_buffer_reader_next(reader, &slice)) { |
||||||
|
memcpy(c_bytes + offset, GPR_SLICE_START_PTR(slice), |
||||||
|
GPR_SLICE_LENGTH(slice)); |
||||||
|
offset += GPR_SLICE_LENGTH(slice); |
||||||
|
} |
||||||
|
grpc_byte_buffer_reader_destroy(reader); |
||||||
|
bytes = PyBytes_FromStringAndSize(c_bytes, length); |
||||||
|
gpr_free(c_bytes); |
||||||
|
event_args = |
||||||
|
Py_BuildValue("(OOOOOOO)", read_event_kind, (PyObject *)c_event->tag, |
||||||
|
Py_None, Py_None, Py_None, bytes, Py_None); |
||||||
|
Py_DECREF(bytes); |
||||||
|
return event_args; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_metadata_event_args(grpc_event *c_event) { |
||||||
|
/* TODO(nathaniel): Actual transmission of metadata. */ |
||||||
|
return Py_BuildValue("(OOOOOOO)", metadata_event_kind, |
||||||
|
(PyObject *)c_event->tag, Py_None, Py_None, Py_None, |
||||||
|
Py_None, Py_None); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_finished_event_args(grpc_event *c_event) { |
||||||
|
PyObject *code; |
||||||
|
PyObject *details; |
||||||
|
PyObject *status_args; |
||||||
|
PyObject *status; |
||||||
|
PyObject *event_args; |
||||||
|
|
||||||
|
code = pygrpc_status_code(c_event->data.finished.status); |
||||||
|
if (code == NULL) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "Unrecognized status code!"); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
if (c_event->data.finished.details == NULL) { |
||||||
|
details = PyBytes_FromString(""); |
||||||
|
} else { |
||||||
|
details = PyBytes_FromString(c_event->data.finished.details); |
||||||
|
} |
||||||
|
status_args = Py_BuildValue("(OO)", code, details); |
||||||
|
Py_DECREF(details); |
||||||
|
status = PyObject_CallObject(status_class, status_args); |
||||||
|
Py_DECREF(status_args); |
||||||
|
event_args = |
||||||
|
Py_BuildValue("(OOOOOOO)", finish_event_kind, (PyObject *)c_event->tag, |
||||||
|
Py_None, Py_None, Py_None, Py_None, status); |
||||||
|
Py_DECREF(status); |
||||||
|
return event_args; |
||||||
|
} |
||||||
|
|
||||||
|
static int pygrpc_completion_queue_init(CompletionQueue *self, PyObject *args, |
||||||
|
PyObject *kwds) { |
||||||
|
self->c_completion_queue = grpc_completion_queue_create(); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static void pygrpc_completion_queue_dealloc(CompletionQueue *self) { |
||||||
|
grpc_completion_queue_destroy(self->c_completion_queue); |
||||||
|
self->ob_type->tp_free((PyObject *)self); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_completion_queue_get(CompletionQueue *self, |
||||||
|
PyObject *args) { |
||||||
|
PyObject *deadline; |
||||||
|
double double_deadline; |
||||||
|
gpr_timespec deadline_timespec; |
||||||
|
grpc_event *c_event; |
||||||
|
|
||||||
|
PyObject *event_args; |
||||||
|
PyObject *event; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "O", &deadline))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
if (deadline == Py_None) { |
||||||
|
deadline_timespec = gpr_inf_future; |
||||||
|
} else { |
||||||
|
double_deadline = PyFloat_AsDouble(deadline); |
||||||
|
deadline_timespec = gpr_time_from_nanos((long)(double_deadline * 1.0E9)); |
||||||
|
} |
||||||
|
|
||||||
|
/* TODO(nathaniel): Suppress clang-format in this block and remove the
|
||||||
|
unnecessary and unPythonic semicolons trailing the _ALLOW_THREADS macros. |
||||||
|
(Right now clang-format only understands //-demarcated suppressions.) */
|
||||||
|
Py_BEGIN_ALLOW_THREADS; |
||||||
|
c_event = |
||||||
|
grpc_completion_queue_next(self->c_completion_queue, deadline_timespec); |
||||||
|
Py_END_ALLOW_THREADS; |
||||||
|
|
||||||
|
if (c_event == NULL) { |
||||||
|
Py_RETURN_NONE; |
||||||
|
} |
||||||
|
|
||||||
|
switch (c_event->type) { |
||||||
|
case GRPC_QUEUE_SHUTDOWN: |
||||||
|
event_args = pygrpc_stop_event_args(c_event); |
||||||
|
break; |
||||||
|
case GRPC_WRITE_ACCEPTED: |
||||||
|
event_args = pygrpc_write_event_args(c_event); |
||||||
|
break; |
||||||
|
case GRPC_FINISH_ACCEPTED: |
||||||
|
event_args = pygrpc_complete_event_args(c_event); |
||||||
|
break; |
||||||
|
case GRPC_SERVER_RPC_NEW: |
||||||
|
event_args = pygrpc_service_event_args(c_event); |
||||||
|
break; |
||||||
|
case GRPC_READ: |
||||||
|
event_args = pygrpc_read_event_args(c_event); |
||||||
|
break; |
||||||
|
case GRPC_CLIENT_METADATA_READ: |
||||||
|
event_args = pygrpc_metadata_event_args(c_event); |
||||||
|
break; |
||||||
|
case GRPC_FINISHED: |
||||||
|
event_args = pygrpc_finished_event_args(c_event); |
||||||
|
break; |
||||||
|
default: |
||||||
|
PyErr_SetString(PyExc_Exception, "Unrecognized event type!"); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
if (event_args == NULL) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
event = PyObject_CallObject(event_class, event_args); |
||||||
|
|
||||||
|
Py_DECREF(event_args); |
||||||
|
Py_XDECREF((PyObject *)c_event->tag); |
||||||
|
grpc_event_finish(c_event); |
||||||
|
|
||||||
|
return event; |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_completion_queue_stop(CompletionQueue *self) { |
||||||
|
grpc_completion_queue_shutdown(self->c_completion_queue); |
||||||
|
|
||||||
|
Py_RETURN_NONE; |
||||||
|
} |
||||||
|
|
||||||
|
static PyMethodDef methods[] = { |
||||||
|
{"get", (PyCFunction)pygrpc_completion_queue_get, METH_VARARGS, |
||||||
|
"Get the next event."}, |
||||||
|
{"stop", (PyCFunction)pygrpc_completion_queue_stop, METH_NOARGS, |
||||||
|
"Stop this completion queue."}, |
||||||
|
{NULL}}; |
||||||
|
|
||||||
|
PyTypeObject pygrpc_CompletionQueueType = { |
||||||
|
PyObject_HEAD_INIT(NULL)0, /*ob_size*/ |
||||||
|
"_gprc.CompletionQueue", /*tp_name*/ |
||||||
|
sizeof(CompletionQueue), /*tp_basicsize*/ |
||||||
|
0, /*tp_itemsize*/ |
||||||
|
(destructor)pygrpc_completion_queue_dealloc, /*tp_dealloc*/ |
||||||
|
0, /*tp_print*/ |
||||||
|
0, /*tp_getattr*/ |
||||||
|
0, /*tp_setattr*/ |
||||||
|
0, /*tp_compare*/ |
||||||
|
0, /*tp_repr*/ |
||||||
|
0, /*tp_as_number*/ |
||||||
|
0, /*tp_as_sequence*/ |
||||||
|
0, /*tp_as_mapping*/ |
||||||
|
0, /*tp_hash */ |
||||||
|
0, /*tp_call*/ |
||||||
|
0, /*tp_str*/ |
||||||
|
0, /*tp_getattro*/ |
||||||
|
0, /*tp_setattro*/ |
||||||
|
0, /*tp_as_buffer*/ |
||||||
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/ |
||||||
|
"Wrapping of grpc_completion_queue.", /* tp_doc */ |
||||||
|
0, /* tp_traverse */ |
||||||
|
0, /* tp_clear */ |
||||||
|
0, /* tp_richcompare */ |
||||||
|
0, /* tp_weaklistoffset */ |
||||||
|
0, /* tp_iter */ |
||||||
|
0, /* tp_iternext */ |
||||||
|
methods, /* tp_methods */ |
||||||
|
0, /* tp_members */ |
||||||
|
0, /* tp_getset */ |
||||||
|
0, /* tp_base */ |
||||||
|
0, /* tp_dict */ |
||||||
|
0, /* tp_descr_get */ |
||||||
|
0, /* tp_descr_set */ |
||||||
|
0, /* tp_dictoffset */ |
||||||
|
(initproc)pygrpc_completion_queue_init, /* tp_init */ |
||||||
|
}; |
||||||
|
|
||||||
|
static int pygrpc_get_status_codes(PyObject *datatypes_module) { |
||||||
|
PyObject *code_class = PyObject_GetAttrString(datatypes_module, "Code"); |
||||||
|
if (code_class == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
ok_status_code = PyObject_GetAttrString(code_class, "OK"); |
||||||
|
if (ok_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
cancelled_status_code = PyObject_GetAttrString(code_class, "CANCELLED"); |
||||||
|
if (cancelled_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
unknown_status_code = PyObject_GetAttrString(code_class, "UNKNOWN"); |
||||||
|
if (unknown_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
invalid_argument_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "INVALID_ARGUMENT"); |
||||||
|
if (invalid_argument_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
expired_status_code = PyObject_GetAttrString(code_class, "EXPIRED"); |
||||||
|
if (expired_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
not_found_status_code = PyObject_GetAttrString(code_class, "NOT_FOUND"); |
||||||
|
if (not_found_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
already_exists_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "ALREADY_EXISTS"); |
||||||
|
if (already_exists_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
permission_denied_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "PERMISSION_DENIED"); |
||||||
|
if (permission_denied_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
unauthenticated_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "UNAUTHENTICATED"); |
||||||
|
if (unauthenticated_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
resource_exhausted_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "RESOURCE_EXHAUSTED"); |
||||||
|
if (resource_exhausted_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
failed_precondition_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "FAILED_PRECONDITION"); |
||||||
|
if (failed_precondition_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
aborted_status_code = PyObject_GetAttrString(code_class, "ABORTED"); |
||||||
|
if (aborted_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
out_of_range_status_code = PyObject_GetAttrString(code_class, "OUT_OF_RANGE"); |
||||||
|
if (out_of_range_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
unimplemented_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "UNIMPLEMENTED"); |
||||||
|
if (unimplemented_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
internal_error_status_code = |
||||||
|
PyObject_GetAttrString(code_class, "INTERNAL_ERROR"); |
||||||
|
if (internal_error_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
unavailable_status_code = PyObject_GetAttrString(code_class, "UNAVAILABLE"); |
||||||
|
if (unavailable_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
data_loss_status_code = PyObject_GetAttrString(code_class, "DATA_LOSS"); |
||||||
|
if (data_loss_status_code == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
Py_DECREF(code_class); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static int pygrpc_get_event_kinds(PyObject *event_class) { |
||||||
|
PyObject *kind_class = PyObject_GetAttrString(event_class, "Kind"); |
||||||
|
if (kind_class == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
stop_event_kind = PyObject_GetAttrString(kind_class, "STOP"); |
||||||
|
if (stop_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
write_event_kind = PyObject_GetAttrString(kind_class, "WRITE_ACCEPTED"); |
||||||
|
if (write_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
complete_event_kind = PyObject_GetAttrString(kind_class, "COMPLETE_ACCEPTED"); |
||||||
|
if (complete_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
service_event_kind = PyObject_GetAttrString(kind_class, "SERVICE_ACCEPTED"); |
||||||
|
if (service_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
read_event_kind = PyObject_GetAttrString(kind_class, "READ_ACCEPTED"); |
||||||
|
if (read_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
metadata_event_kind = PyObject_GetAttrString(kind_class, "METADATA_ACCEPTED"); |
||||||
|
if (metadata_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
finish_event_kind = PyObject_GetAttrString(kind_class, "FINISH"); |
||||||
|
if (finish_event_kind == NULL) { |
||||||
|
return -1; |
||||||
|
} |
||||||
|
Py_DECREF(kind_class); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
int pygrpc_add_completion_queue(PyObject *module) { |
||||||
|
char *datatypes_module_path = "_adapter._datatypes"; |
||||||
|
PyObject *datatypes_module = PyImport_ImportModule(datatypes_module_path); |
||||||
|
if (datatypes_module == NULL) { |
||||||
|
PyErr_SetString(PyExc_ImportError, datatypes_module_path); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
status_class = PyObject_GetAttrString(datatypes_module, "Status"); |
||||||
|
service_acceptance_class = |
||||||
|
PyObject_GetAttrString(datatypes_module, "ServiceAcceptance"); |
||||||
|
event_class = PyObject_GetAttrString(datatypes_module, "Event"); |
||||||
|
if (status_class == NULL || service_acceptance_class == NULL || |
||||||
|
event_class == NULL) { |
||||||
|
PyErr_SetString(PyExc_ImportError, "Missing classes in _datatypes module!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (pygrpc_get_status_codes(datatypes_module) == -1) { |
||||||
|
PyErr_SetString(PyExc_ImportError, "Status codes import broken!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (pygrpc_get_event_kinds(event_class) == -1) { |
||||||
|
PyErr_SetString(PyExc_ImportError, "Event kinds import broken!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
Py_DECREF(datatypes_module); |
||||||
|
|
||||||
|
pygrpc_CompletionQueueType.tp_new = PyType_GenericNew; |
||||||
|
if (PyType_Ready(&pygrpc_CompletionQueueType) < 0) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, |
||||||
|
"Error defining pygrpc_CompletionQueueType!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (PyModule_AddObject(module, "CompletionQueue", |
||||||
|
(PyObject *)&pygrpc_CompletionQueueType) == -1) { |
||||||
|
PyErr_SetString(PyExc_ImportError, |
||||||
|
"Couldn't add CompletionQueue type to module!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
@ -0,0 +1,48 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef _ADAPTER__COMPLETION_QUEUE_H_ |
||||||
|
#define _ADAPTER__COMPLETION_QUEUE_H_ |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
typedef struct { |
||||||
|
PyObject_HEAD grpc_completion_queue *c_completion_queue; |
||||||
|
} CompletionQueue; |
||||||
|
|
||||||
|
PyTypeObject pygrpc_CompletionQueueType; |
||||||
|
|
||||||
|
int pygrpc_add_completion_queue(PyObject *module); |
||||||
|
|
||||||
|
#endif /* _ADAPTER__COMPLETION_QUEUE_H_ */ |
@ -0,0 +1,86 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Datatypes passed between Python and C code.""" |
||||||
|
|
||||||
|
import collections |
||||||
|
import enum |
||||||
|
|
||||||
|
|
||||||
|
@enum.unique |
||||||
|
class Code(enum.IntEnum): |
||||||
|
"""One Platform error codes (see status.h and codes.proto).""" |
||||||
|
|
||||||
|
OK = 0 |
||||||
|
CANCELLED = 1 |
||||||
|
UNKNOWN = 2 |
||||||
|
INVALID_ARGUMENT = 3 |
||||||
|
EXPIRED = 4 |
||||||
|
NOT_FOUND = 5 |
||||||
|
ALREADY_EXISTS = 6 |
||||||
|
PERMISSION_DENIED = 7 |
||||||
|
UNAUTHENTICATED = 16 |
||||||
|
RESOURCE_EXHAUSTED = 8 |
||||||
|
FAILED_PRECONDITION = 9 |
||||||
|
ABORTED = 10 |
||||||
|
OUT_OF_RANGE = 11 |
||||||
|
UNIMPLEMENTED = 12 |
||||||
|
INTERNAL_ERROR = 13 |
||||||
|
UNAVAILABLE = 14 |
||||||
|
DATA_LOSS = 15 |
||||||
|
|
||||||
|
|
||||||
|
class Status(collections.namedtuple('Status', ['code', 'details'])): |
||||||
|
"""Describes an RPC's overall status.""" |
||||||
|
|
||||||
|
|
||||||
|
class ServiceAcceptance( |
||||||
|
collections.namedtuple( |
||||||
|
'ServiceAcceptance', ['call', 'method', 'host', 'deadline'])): |
||||||
|
"""Describes an RPC on the service side at the start of service.""" |
||||||
|
|
||||||
|
|
||||||
|
class Event( |
||||||
|
collections.namedtuple( |
||||||
|
'Event', |
||||||
|
['kind', 'tag', 'write_accepted', 'complete_accepted', |
||||||
|
'service_acceptance', 'bytes', 'status'])): |
||||||
|
"""Describes an event emitted from a completion queue.""" |
||||||
|
|
||||||
|
@enum.unique |
||||||
|
class Kind(enum.Enum): |
||||||
|
"""Describes the kind of an event.""" |
||||||
|
|
||||||
|
STOP = object() |
||||||
|
WRITE_ACCEPTED = object() |
||||||
|
COMPLETE_ACCEPTED = object() |
||||||
|
SERVICE_ACCEPTED = object() |
||||||
|
READ_ACCEPTED = object() |
||||||
|
METADATA_ACCEPTED = object() |
||||||
|
FINISH = object() |
@ -0,0 +1,79 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 "_adapter/_error.h" |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
const PyObject *pygrpc_translate_call_error(grpc_call_error call_error) { |
||||||
|
switch (call_error) { |
||||||
|
case GRPC_CALL_OK: |
||||||
|
Py_RETURN_NONE; |
||||||
|
case GRPC_CALL_ERROR: |
||||||
|
PyErr_SetString(PyExc_Exception, "Defect: unknown defect!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_NOT_ON_SERVER: |
||||||
|
PyErr_SetString(PyExc_Exception, |
||||||
|
"Defect: client-only method called on server!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_NOT_ON_CLIENT: |
||||||
|
PyErr_SetString(PyExc_Exception, |
||||||
|
"Defect: server-only method called on client!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_ALREADY_ACCEPTED: |
||||||
|
PyErr_SetString(PyExc_Exception, |
||||||
|
"Defect: attempted to accept already-accepted call!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_ALREADY_INVOKED: |
||||||
|
PyErr_SetString(PyExc_Exception, |
||||||
|
"Defect: attempted to invoke already-invoked call!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_NOT_INVOKED: |
||||||
|
PyErr_SetString(PyExc_Exception, "Defect: Call not yet invoked!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_ALREADY_FINISHED: |
||||||
|
PyErr_SetString(PyExc_Exception, "Defect: Call already finished!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_TOO_MANY_OPERATIONS: |
||||||
|
PyErr_SetString(PyExc_Exception, |
||||||
|
"Defect: Attempted extra read or extra write on call!"); |
||||||
|
return NULL; |
||||||
|
case GRPC_CALL_ERROR_INVALID_FLAGS: |
||||||
|
PyErr_SetString(PyExc_Exception, "Defect: invalid flags!"); |
||||||
|
return NULL; |
||||||
|
default: |
||||||
|
PyErr_SetString(PyExc_Exception, "Defect: Unknown call error!"); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
} |
@ -0,0 +1,42 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef _ADAPTER__ERROR_H_ |
||||||
|
#define _ADAPTER__ERROR_H_ |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
const PyObject *pygrpc_translate_call_error(grpc_call_error call_error); |
||||||
|
|
||||||
|
#endif /* _ADAPTER__ERROR_H_ */ |
@ -0,0 +1,46 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""One of the tests of the Face layer of RPC Framework.""" |
||||||
|
|
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _face_test_case |
||||||
|
from _framework.face.testing import event_invocation_synchronous_event_service_test_case as test_case |
||||||
|
|
||||||
|
|
||||||
|
class EventInvocationSynchronousEventServiceTest( |
||||||
|
_face_test_case.FaceTestCase, |
||||||
|
test_case.EventInvocationSynchronousEventServiceTestCase, |
||||||
|
unittest.TestCase): |
||||||
|
pass |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,124 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Common construction and destruction for GRPC-backed Face-layer tests.""" |
||||||
|
|
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import fore |
||||||
|
from _adapter import rear |
||||||
|
from _framework.base import util |
||||||
|
from _framework.base.packets import implementations as tickets_implementations |
||||||
|
from _framework.face import implementations as face_implementations |
||||||
|
from _framework.face.testing import coverage |
||||||
|
from _framework.face.testing import serial |
||||||
|
from _framework.face.testing import test_case |
||||||
|
from _framework.foundation import logging_pool |
||||||
|
|
||||||
|
_TIMEOUT = 3 |
||||||
|
_MAXIMUM_TIMEOUT = 90 |
||||||
|
_MAXIMUM_POOL_SIZE = 400 |
||||||
|
|
||||||
|
|
||||||
|
class FaceTestCase(test_case.FaceTestCase, coverage.BlockingCoverage): |
||||||
|
"""Provides abstract Face-layer tests a GRPC-backed implementation.""" |
||||||
|
|
||||||
|
def set_up_implementation( |
||||||
|
self, |
||||||
|
name, |
||||||
|
methods, |
||||||
|
inline_value_in_value_out_methods, |
||||||
|
inline_value_in_stream_out_methods, |
||||||
|
inline_stream_in_value_out_methods, |
||||||
|
inline_stream_in_stream_out_methods, |
||||||
|
event_value_in_value_out_methods, |
||||||
|
event_value_in_stream_out_methods, |
||||||
|
event_stream_in_value_out_methods, |
||||||
|
event_stream_in_stream_out_methods, |
||||||
|
multi_method): |
||||||
|
pool = logging_pool.pool(_MAXIMUM_POOL_SIZE) |
||||||
|
|
||||||
|
servicer = face_implementations.servicer( |
||||||
|
pool, |
||||||
|
inline_value_in_value_out_methods=inline_value_in_value_out_methods, |
||||||
|
inline_value_in_stream_out_methods=inline_value_in_stream_out_methods, |
||||||
|
inline_stream_in_value_out_methods=inline_stream_in_value_out_methods, |
||||||
|
inline_stream_in_stream_out_methods=inline_stream_in_stream_out_methods, |
||||||
|
event_value_in_value_out_methods=event_value_in_value_out_methods, |
||||||
|
event_value_in_stream_out_methods=event_value_in_stream_out_methods, |
||||||
|
event_stream_in_value_out_methods=event_stream_in_value_out_methods, |
||||||
|
event_stream_in_stream_out_methods=event_stream_in_stream_out_methods, |
||||||
|
multi_method=multi_method) |
||||||
|
|
||||||
|
serialization = serial.serialization(methods) |
||||||
|
|
||||||
|
fore_link = fore.ForeLink( |
||||||
|
pool, serialization.request_deserializers, |
||||||
|
serialization.response_serializers) |
||||||
|
port = fore_link.start() |
||||||
|
rear_link = rear.RearLink( |
||||||
|
'localhost', port, pool, |
||||||
|
serialization.request_serializers, serialization.response_deserializers) |
||||||
|
rear_link.start() |
||||||
|
front = tickets_implementations.front(pool, pool, pool) |
||||||
|
back = tickets_implementations.back( |
||||||
|
servicer, pool, pool, pool, _TIMEOUT, _MAXIMUM_TIMEOUT) |
||||||
|
fore_link.join_rear_link(back) |
||||||
|
back.join_fore_link(fore_link) |
||||||
|
rear_link.join_fore_link(front) |
||||||
|
front.join_rear_link(rear_link) |
||||||
|
|
||||||
|
server = face_implementations.server() |
||||||
|
stub = face_implementations.stub(front, pool) |
||||||
|
return server, stub, (rear_link, fore_link, front, back) |
||||||
|
|
||||||
|
def tear_down_implementation(self, memo): |
||||||
|
rear_link, fore_link, front, back = memo |
||||||
|
# TODO(nathaniel): Waiting for the front and back to idle possibly should |
||||||
|
# not be necessary - investigate as part of graceful shutdown work. |
||||||
|
util.wait_for_idle(front) |
||||||
|
util.wait_for_idle(back) |
||||||
|
rear_link.stop() |
||||||
|
fore_link.stop() |
||||||
|
|
||||||
|
@unittest.skip('Service-side failure not transmitted by GRPC.') |
||||||
|
def testFailedUnaryRequestUnaryResponse(self): |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@unittest.skip('Service-side failure not transmitted by GRPC.') |
||||||
|
def testFailedUnaryRequestStreamResponse(self): |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@unittest.skip('Service-side failure not transmitted by GRPC.') |
||||||
|
def testFailedStreamRequestUnaryResponse(self): |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@unittest.skip('Service-side failure not transmitted by GRPC.') |
||||||
|
def testFailedStreamRequestStreamResponse(self): |
||||||
|
raise NotImplementedError() |
@ -0,0 +1,46 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""One of the tests of the Face layer of RPC Framework.""" |
||||||
|
|
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _face_test_case |
||||||
|
from _framework.face.testing import future_invocation_asynchronous_event_service_test_case as test_case |
||||||
|
|
||||||
|
|
||||||
|
class FutureInvocationAsynchronousEventServiceTest( |
||||||
|
_face_test_case.FaceTestCase, |
||||||
|
test_case.FutureInvocationAsynchronousEventServiceTestCase, |
||||||
|
unittest.TestCase): |
||||||
|
pass |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,245 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Test of the GRPC-backed ForeLink and RearLink.""" |
||||||
|
|
||||||
|
import threading |
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _proto_scenarios |
||||||
|
from _adapter import _test_links |
||||||
|
from _adapter import fore |
||||||
|
from _adapter import rear |
||||||
|
from _framework.base import interfaces |
||||||
|
from _framework.base.packets import packets as tickets |
||||||
|
from _framework.foundation import logging_pool |
||||||
|
|
||||||
|
_IDENTITY = lambda x: x |
||||||
|
_TIMEOUT = 2 |
||||||
|
|
||||||
|
|
||||||
|
class RoundTripTest(unittest.TestCase): |
||||||
|
|
||||||
|
def setUp(self): |
||||||
|
self.fore_link_pool = logging_pool.pool(80) |
||||||
|
self.rear_link_pool = logging_pool.pool(80) |
||||||
|
|
||||||
|
def tearDown(self): |
||||||
|
self.rear_link_pool.shutdown(wait=True) |
||||||
|
self.fore_link_pool.shutdown(wait=True) |
||||||
|
|
||||||
|
def testZeroMessageRoundTrip(self): |
||||||
|
test_operation_id = object() |
||||||
|
test_method = 'test method' |
||||||
|
test_fore_link = _test_links.ForeLink(None, None) |
||||||
|
def rear_action(front_to_back_ticket, fore_link): |
||||||
|
if front_to_back_ticket.kind in ( |
||||||
|
tickets.Kind.COMPLETION, tickets.Kind.ENTIRE): |
||||||
|
back_to_front_ticket = tickets.BackToFrontPacket( |
||||||
|
front_to_back_ticket.operation_id, 0, tickets.Kind.COMPLETION, None) |
||||||
|
fore_link.accept_back_to_front_ticket(back_to_front_ticket) |
||||||
|
test_rear_link = _test_links.RearLink(rear_action, None) |
||||||
|
|
||||||
|
fore_link = fore.ForeLink( |
||||||
|
self.fore_link_pool, {test_method: None}, {test_method: None}) |
||||||
|
fore_link.join_rear_link(test_rear_link) |
||||||
|
test_rear_link.join_fore_link(fore_link) |
||||||
|
port = fore_link.start() |
||||||
|
|
||||||
|
rear_link = rear.RearLink( |
||||||
|
'localhost', port, self.rear_link_pool, {test_method: None}, |
||||||
|
{test_method: None}) |
||||||
|
rear_link.join_fore_link(test_fore_link) |
||||||
|
test_fore_link.join_rear_link(rear_link) |
||||||
|
rear_link.start() |
||||||
|
|
||||||
|
front_to_back_ticket = tickets.FrontToBackPacket( |
||||||
|
test_operation_id, 0, tickets.Kind.ENTIRE, test_method, interfaces.FULL, |
||||||
|
None, None, _TIMEOUT) |
||||||
|
rear_link.accept_front_to_back_ticket(front_to_back_ticket) |
||||||
|
|
||||||
|
with test_fore_link.condition: |
||||||
|
while (not test_fore_link.tickets or |
||||||
|
test_fore_link.tickets[-1].kind is tickets.Kind.CONTINUATION): |
||||||
|
test_fore_link.condition.wait() |
||||||
|
|
||||||
|
rear_link.stop() |
||||||
|
fore_link.stop() |
||||||
|
|
||||||
|
with test_fore_link.condition: |
||||||
|
self.assertIs(test_fore_link.tickets[-1].kind, tickets.Kind.COMPLETION) |
||||||
|
|
||||||
|
def testEntireRoundTrip(self): |
||||||
|
test_operation_id = object() |
||||||
|
test_method = 'test method' |
||||||
|
test_front_to_back_datum = b'\x07' |
||||||
|
test_back_to_front_datum = b'\x08' |
||||||
|
test_fore_link = _test_links.ForeLink(None, None) |
||||||
|
rear_sequence_number = [0] |
||||||
|
def rear_action(front_to_back_ticket, fore_link): |
||||||
|
if front_to_back_ticket.payload is None: |
||||||
|
payload = None |
||||||
|
else: |
||||||
|
payload = test_back_to_front_datum |
||||||
|
terminal = front_to_back_ticket.kind in ( |
||||||
|
tickets.Kind.COMPLETION, tickets.Kind.ENTIRE) |
||||||
|
if payload is not None or terminal: |
||||||
|
back_to_front_ticket = tickets.BackToFrontPacket( |
||||||
|
front_to_back_ticket.operation_id, rear_sequence_number[0], |
||||||
|
tickets.Kind.COMPLETION if terminal else tickets.Kind.CONTINUATION, |
||||||
|
payload) |
||||||
|
rear_sequence_number[0] += 1 |
||||||
|
fore_link.accept_back_to_front_ticket(back_to_front_ticket) |
||||||
|
test_rear_link = _test_links.RearLink(rear_action, None) |
||||||
|
|
||||||
|
fore_link = fore.ForeLink( |
||||||
|
self.fore_link_pool, {test_method: _IDENTITY}, |
||||||
|
{test_method: _IDENTITY}) |
||||||
|
fore_link.join_rear_link(test_rear_link) |
||||||
|
test_rear_link.join_fore_link(fore_link) |
||||||
|
port = fore_link.start() |
||||||
|
|
||||||
|
rear_link = rear.RearLink( |
||||||
|
'localhost', port, self.rear_link_pool, {test_method: _IDENTITY}, |
||||||
|
{test_method: _IDENTITY}) |
||||||
|
rear_link.join_fore_link(test_fore_link) |
||||||
|
test_fore_link.join_rear_link(rear_link) |
||||||
|
rear_link.start() |
||||||
|
|
||||||
|
front_to_back_ticket = tickets.FrontToBackPacket( |
||||||
|
test_operation_id, 0, tickets.Kind.ENTIRE, test_method, interfaces.FULL, |
||||||
|
None, test_front_to_back_datum, _TIMEOUT) |
||||||
|
rear_link.accept_front_to_back_ticket(front_to_back_ticket) |
||||||
|
|
||||||
|
with test_fore_link.condition: |
||||||
|
while (not test_fore_link.tickets or |
||||||
|
test_fore_link.tickets[-1].kind is not tickets.Kind.COMPLETION): |
||||||
|
test_fore_link.condition.wait() |
||||||
|
|
||||||
|
rear_link.stop() |
||||||
|
fore_link.stop() |
||||||
|
|
||||||
|
with test_rear_link.condition: |
||||||
|
front_to_back_payloads = tuple( |
||||||
|
ticket.payload for ticket in test_rear_link.tickets |
||||||
|
if ticket.payload is not None) |
||||||
|
with test_fore_link.condition: |
||||||
|
back_to_front_payloads = tuple( |
||||||
|
ticket.payload for ticket in test_fore_link.tickets |
||||||
|
if ticket.payload is not None) |
||||||
|
self.assertTupleEqual((test_front_to_back_datum,), front_to_back_payloads) |
||||||
|
self.assertTupleEqual((test_back_to_front_datum,), back_to_front_payloads) |
||||||
|
|
||||||
|
def _perform_scenario_test(self, scenario): |
||||||
|
test_operation_id = object() |
||||||
|
test_method = scenario.method() |
||||||
|
test_fore_link = _test_links.ForeLink(None, None) |
||||||
|
rear_lock = threading.Lock() |
||||||
|
rear_sequence_number = [0] |
||||||
|
def rear_action(front_to_back_ticket, fore_link): |
||||||
|
with rear_lock: |
||||||
|
if front_to_back_ticket.payload is not None: |
||||||
|
response = scenario.response_for_request(front_to_back_ticket.payload) |
||||||
|
else: |
||||||
|
response = None |
||||||
|
terminal = front_to_back_ticket.kind in ( |
||||||
|
tickets.Kind.COMPLETION, tickets.Kind.ENTIRE) |
||||||
|
if response is not None or terminal: |
||||||
|
back_to_front_ticket = tickets.BackToFrontPacket( |
||||||
|
front_to_back_ticket.operation_id, rear_sequence_number[0], |
||||||
|
tickets.Kind.COMPLETION if terminal else tickets.Kind.CONTINUATION, |
||||||
|
response) |
||||||
|
rear_sequence_number[0] += 1 |
||||||
|
fore_link.accept_back_to_front_ticket(back_to_front_ticket) |
||||||
|
test_rear_link = _test_links.RearLink(rear_action, None) |
||||||
|
|
||||||
|
fore_link = fore.ForeLink( |
||||||
|
self.fore_link_pool, {test_method: scenario.deserialize_request}, |
||||||
|
{test_method: scenario.serialize_response}) |
||||||
|
fore_link.join_rear_link(test_rear_link) |
||||||
|
test_rear_link.join_fore_link(fore_link) |
||||||
|
port = fore_link.start() |
||||||
|
|
||||||
|
rear_link = rear.RearLink( |
||||||
|
'localhost', port, self.rear_link_pool, |
||||||
|
{test_method: scenario.serialize_request}, |
||||||
|
{test_method: scenario.deserialize_response}) |
||||||
|
rear_link.join_fore_link(test_fore_link) |
||||||
|
test_fore_link.join_rear_link(rear_link) |
||||||
|
rear_link.start() |
||||||
|
|
||||||
|
commencement_ticket = tickets.FrontToBackPacket( |
||||||
|
test_operation_id, 0, tickets.Kind.COMMENCEMENT, test_method, |
||||||
|
interfaces.FULL, None, None, _TIMEOUT) |
||||||
|
fore_sequence_number = 1 |
||||||
|
rear_link.accept_front_to_back_ticket(commencement_ticket) |
||||||
|
for request in scenario.requests(): |
||||||
|
continuation_ticket = tickets.FrontToBackPacket( |
||||||
|
test_operation_id, fore_sequence_number, tickets.Kind.CONTINUATION, |
||||||
|
None, None, None, request, None) |
||||||
|
fore_sequence_number += 1 |
||||||
|
rear_link.accept_front_to_back_ticket(continuation_ticket) |
||||||
|
completion_ticket = tickets.FrontToBackPacket( |
||||||
|
test_operation_id, fore_sequence_number, tickets.Kind.COMPLETION, None, |
||||||
|
None, None, None, None) |
||||||
|
fore_sequence_number += 1 |
||||||
|
rear_link.accept_front_to_back_ticket(completion_ticket) |
||||||
|
|
||||||
|
with test_fore_link.condition: |
||||||
|
while (not test_fore_link.tickets or |
||||||
|
test_fore_link.tickets[-1].kind is not tickets.Kind.COMPLETION): |
||||||
|
test_fore_link.condition.wait() |
||||||
|
|
||||||
|
rear_link.stop() |
||||||
|
fore_link.stop() |
||||||
|
|
||||||
|
with test_rear_link.condition: |
||||||
|
requests = tuple( |
||||||
|
ticket.payload for ticket in test_rear_link.tickets |
||||||
|
if ticket.payload is not None) |
||||||
|
with test_fore_link.condition: |
||||||
|
responses = tuple( |
||||||
|
ticket.payload for ticket in test_fore_link.tickets |
||||||
|
if ticket.payload is not None) |
||||||
|
self.assertTrue(scenario.verify_requests(requests)) |
||||||
|
self.assertTrue(scenario.verify_responses(responses)) |
||||||
|
|
||||||
|
def testEmptyScenario(self): |
||||||
|
self._perform_scenario_test(_proto_scenarios.EmptyScenario()) |
||||||
|
|
||||||
|
def testBidirectionallyUnaryScenario(self): |
||||||
|
self._perform_scenario_test(_proto_scenarios.BidirectionallyUnaryScenario()) |
||||||
|
|
||||||
|
def testBidirectionallyStreamingScenario(self): |
||||||
|
self._perform_scenario_test( |
||||||
|
_proto_scenarios.BidirectionallyStreamingScenario()) |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,97 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""A test of invocation-side code unconnected to an RPC server.""" |
||||||
|
|
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _test_links |
||||||
|
from _adapter import rear |
||||||
|
from _framework.base import interfaces |
||||||
|
from _framework.base.packets import packets |
||||||
|
from _framework.foundation import logging_pool |
||||||
|
|
||||||
|
_IDENTITY = lambda x: x |
||||||
|
_TIMEOUT = 2 |
||||||
|
|
||||||
|
|
||||||
|
class LonelyRearLinkTest(unittest.TestCase): |
||||||
|
|
||||||
|
def setUp(self): |
||||||
|
self.pool = logging_pool.pool(80) |
||||||
|
|
||||||
|
def tearDown(self): |
||||||
|
self.pool.shutdown(wait=True) |
||||||
|
|
||||||
|
def testUpAndDown(self): |
||||||
|
rear_link = rear.RearLink('nonexistent', 54321, self.pool, {}, {}) |
||||||
|
|
||||||
|
rear_link.start() |
||||||
|
rear_link.stop() |
||||||
|
|
||||||
|
def _perform_lonely_client_test_with_ticket_kind( |
||||||
|
self, front_to_back_ticket_kind): |
||||||
|
test_operation_id = object() |
||||||
|
test_method = 'test method' |
||||||
|
fore_link = _test_links.ForeLink(None, None) |
||||||
|
|
||||||
|
rear_link = rear.RearLink( |
||||||
|
'nonexistent', 54321, self.pool, {test_method: None}, |
||||||
|
{test_method: None}) |
||||||
|
rear_link.join_fore_link(fore_link) |
||||||
|
rear_link.start() |
||||||
|
|
||||||
|
front_to_back_ticket = packets.FrontToBackPacket( |
||||||
|
test_operation_id, 0, front_to_back_ticket_kind, test_method, |
||||||
|
interfaces.FULL, None, None, _TIMEOUT) |
||||||
|
rear_link.accept_front_to_back_ticket(front_to_back_ticket) |
||||||
|
|
||||||
|
with fore_link.condition: |
||||||
|
while True: |
||||||
|
if (fore_link.tickets and |
||||||
|
fore_link.tickets[-1].kind is not packets.Kind.CONTINUATION): |
||||||
|
break |
||||||
|
fore_link.condition.wait() |
||||||
|
|
||||||
|
rear_link.stop() |
||||||
|
|
||||||
|
with fore_link.condition: |
||||||
|
self.assertIsNot(fore_link.tickets[-1].kind, packets.Kind.COMPLETION) |
||||||
|
|
||||||
|
@unittest.skip('TODO(nathaniel): This seems to have broken in the last few weeks; fix it.') |
||||||
|
def testLonelyClientCommencementPacket(self): |
||||||
|
self._perform_lonely_client_test_with_ticket_kind( |
||||||
|
packets.Kind.COMMENCEMENT) |
||||||
|
|
||||||
|
def testLonelyClientEntirePacket(self): |
||||||
|
self._perform_lonely_client_test_with_ticket_kind(packets.Kind.ENTIRE) |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,55 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""A Python interface for GRPC C core structures and behaviors.""" |
||||||
|
|
||||||
|
import atexit |
||||||
|
import gc |
||||||
|
|
||||||
|
from _adapter import _c |
||||||
|
from _adapter import _datatypes |
||||||
|
|
||||||
|
def _shut_down(): |
||||||
|
# force garbage collection before shutting down grpc, to ensure all grpc |
||||||
|
# objects are cleaned up |
||||||
|
gc.collect() |
||||||
|
_c.shut_down() |
||||||
|
|
||||||
|
_c.init() |
||||||
|
atexit.register(_shut_down) |
||||||
|
|
||||||
|
# pylint: disable=invalid-name |
||||||
|
Code = _datatypes.Code |
||||||
|
Status = _datatypes.Status |
||||||
|
Event = _datatypes.Event |
||||||
|
Call = _c.Call |
||||||
|
Channel = _c.Channel |
||||||
|
CompletionQueue = _c.CompletionQueue |
||||||
|
Server = _c.Server |
||||||
|
# pylint: enable=invalid-name |
@ -0,0 +1,371 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Tests for _adapter._low.""" |
||||||
|
|
||||||
|
import time |
||||||
|
import unittest |
||||||
|
|
||||||
|
from _adapter import _low |
||||||
|
|
||||||
|
_STREAM_LENGTH = 300 |
||||||
|
_TIMEOUT = 5 |
||||||
|
_AFTER_DELAY = 2 |
||||||
|
_FUTURE = time.time() + 60 * 60 * 24 |
||||||
|
_BYTE_SEQUENCE = b'\abcdefghijklmnopqrstuvwxyz0123456789' * 200 |
||||||
|
_BYTE_SEQUENCE_SEQUENCE = tuple( |
||||||
|
bytes(bytearray((row + column) % 256 for column in range(row))) |
||||||
|
for row in range(_STREAM_LENGTH)) |
||||||
|
|
||||||
|
|
||||||
|
class LonelyClientTest(unittest.TestCase): |
||||||
|
|
||||||
|
def testLonelyClient(self): |
||||||
|
host = 'nosuchhostexists' |
||||||
|
port = 54321 |
||||||
|
method = 'test method' |
||||||
|
deadline = time.time() + _TIMEOUT |
||||||
|
after_deadline = deadline + _AFTER_DELAY |
||||||
|
metadata_tag = object() |
||||||
|
finish_tag = object() |
||||||
|
|
||||||
|
completion_queue = _low.CompletionQueue() |
||||||
|
channel = _low.Channel('%s:%d' % (host, port)) |
||||||
|
client_call = _low.Call(channel, method, host, deadline) |
||||||
|
|
||||||
|
client_call.invoke(completion_queue, metadata_tag, finish_tag) |
||||||
|
first_event = completion_queue.get(after_deadline) |
||||||
|
self.assertIsNotNone(first_event) |
||||||
|
second_event = completion_queue.get(after_deadline) |
||||||
|
self.assertIsNotNone(second_event) |
||||||
|
kinds = [event.kind for event in (first_event, second_event)] |
||||||
|
self.assertItemsEqual( |
||||||
|
(_low.Event.Kind.METADATA_ACCEPTED, _low.Event.Kind.FINISH), |
||||||
|
kinds) |
||||||
|
|
||||||
|
self.assertIsNone(completion_queue.get(after_deadline)) |
||||||
|
|
||||||
|
completion_queue.stop() |
||||||
|
stop_event = completion_queue.get(_FUTURE) |
||||||
|
self.assertEqual(_low.Event.Kind.STOP, stop_event.kind) |
||||||
|
|
||||||
|
|
||||||
|
class EchoTest(unittest.TestCase): |
||||||
|
|
||||||
|
def setUp(self): |
||||||
|
self.host = 'localhost' |
||||||
|
|
||||||
|
self.server_completion_queue = _low.CompletionQueue() |
||||||
|
self.server = _low.Server(self.server_completion_queue) |
||||||
|
port = self.server.add_http2_addr('[::]:0') |
||||||
|
self.server.start() |
||||||
|
|
||||||
|
self.client_completion_queue = _low.CompletionQueue() |
||||||
|
self.channel = _low.Channel('%s:%d' % (self.host, port)) |
||||||
|
|
||||||
|
def tearDown(self): |
||||||
|
self.server.stop() |
||||||
|
# NOTE(nathaniel): Yep, this is weird; it's a consequence of |
||||||
|
# grpc_server_destroy's being what has the effect of telling the server's |
||||||
|
# completion queue to pump out all pending events/tags immediately rather |
||||||
|
# than gracefully completing all outstanding RPCs while accepting no new |
||||||
|
# ones. |
||||||
|
# TODO(nathaniel): Deallocation of a Python object shouldn't have this kind |
||||||
|
# of observable side effect let alone such an important one. |
||||||
|
del self.server |
||||||
|
self.server_completion_queue.stop() |
||||||
|
self.client_completion_queue.stop() |
||||||
|
while True: |
||||||
|
event = self.server_completion_queue.get(_FUTURE) |
||||||
|
if event is not None and event.kind is _low.Event.Kind.STOP: |
||||||
|
break |
||||||
|
while True: |
||||||
|
event = self.client_completion_queue.get(_FUTURE) |
||||||
|
if event is not None and event.kind is _low.Event.Kind.STOP: |
||||||
|
break |
||||||
|
self.server_completion_queue = None |
||||||
|
self.client_completion_queue = None |
||||||
|
|
||||||
|
def _perform_echo_test(self, test_data): |
||||||
|
method = 'test method' |
||||||
|
details = 'test details' |
||||||
|
deadline = _FUTURE |
||||||
|
metadata_tag = object() |
||||||
|
finish_tag = object() |
||||||
|
write_tag = object() |
||||||
|
complete_tag = object() |
||||||
|
service_tag = object() |
||||||
|
read_tag = object() |
||||||
|
status_tag = object() |
||||||
|
|
||||||
|
server_data = [] |
||||||
|
client_data = [] |
||||||
|
|
||||||
|
client_call = _low.Call(self.channel, method, self.host, deadline) |
||||||
|
|
||||||
|
client_call.invoke(self.client_completion_queue, metadata_tag, finish_tag) |
||||||
|
|
||||||
|
self.server.service(service_tag) |
||||||
|
service_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(service_accepted) |
||||||
|
self.assertIs(service_accepted.kind, _low.Event.Kind.SERVICE_ACCEPTED) |
||||||
|
self.assertIs(service_accepted.tag, service_tag) |
||||||
|
self.assertEqual(method, service_accepted.service_acceptance.method) |
||||||
|
self.assertEqual(self.host, service_accepted.service_acceptance.host) |
||||||
|
self.assertIsNotNone(service_accepted.service_acceptance.call) |
||||||
|
server_call = service_accepted.service_acceptance.call |
||||||
|
server_call.accept(self.server_completion_queue, finish_tag) |
||||||
|
server_call.premetadata() |
||||||
|
|
||||||
|
metadata_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(metadata_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.METADATA_ACCEPTED, metadata_accepted.kind) |
||||||
|
self.assertEqual(metadata_tag, metadata_accepted.tag) |
||||||
|
# TODO(nathaniel): Test transmission and reception of metadata. |
||||||
|
|
||||||
|
for datum in test_data: |
||||||
|
client_call.write(datum, write_tag) |
||||||
|
write_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(write_accepted) |
||||||
|
self.assertIs(write_accepted.kind, _low.Event.Kind.WRITE_ACCEPTED) |
||||||
|
self.assertIs(write_accepted.tag, write_tag) |
||||||
|
self.assertIs(write_accepted.write_accepted, True) |
||||||
|
|
||||||
|
server_call.read(read_tag) |
||||||
|
read_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(read_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.READ_ACCEPTED, read_accepted.kind) |
||||||
|
self.assertEqual(read_tag, read_accepted.tag) |
||||||
|
self.assertIsNotNone(read_accepted.bytes) |
||||||
|
server_data.append(read_accepted.bytes) |
||||||
|
|
||||||
|
server_call.write(read_accepted.bytes, write_tag) |
||||||
|
write_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(write_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.WRITE_ACCEPTED, write_accepted.kind) |
||||||
|
self.assertEqual(write_tag, write_accepted.tag) |
||||||
|
self.assertTrue(write_accepted.write_accepted) |
||||||
|
|
||||||
|
client_call.read(read_tag) |
||||||
|
read_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(read_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.READ_ACCEPTED, read_accepted.kind) |
||||||
|
self.assertEqual(read_tag, read_accepted.tag) |
||||||
|
self.assertIsNotNone(read_accepted.bytes) |
||||||
|
client_data.append(read_accepted.bytes) |
||||||
|
|
||||||
|
client_call.complete(complete_tag) |
||||||
|
complete_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(complete_accepted) |
||||||
|
self.assertIs(complete_accepted.kind, _low.Event.Kind.COMPLETE_ACCEPTED) |
||||||
|
self.assertIs(complete_accepted.tag, complete_tag) |
||||||
|
self.assertIs(complete_accepted.complete_accepted, True) |
||||||
|
|
||||||
|
server_call.read(read_tag) |
||||||
|
read_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(read_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.READ_ACCEPTED, read_accepted.kind) |
||||||
|
self.assertEqual(read_tag, read_accepted.tag) |
||||||
|
self.assertIsNone(read_accepted.bytes) |
||||||
|
|
||||||
|
server_call.status(_low.Status(_low.Code.OK, details), status_tag) |
||||||
|
server_terminal_event_one = self.server_completion_queue.get(_FUTURE) |
||||||
|
server_terminal_event_two = self.server_completion_queue.get(_FUTURE) |
||||||
|
if server_terminal_event_one.kind == _low.Event.Kind.COMPLETE_ACCEPTED: |
||||||
|
status_accepted = server_terminal_event_one |
||||||
|
rpc_accepted = server_terminal_event_two |
||||||
|
else: |
||||||
|
status_accepted = server_terminal_event_two |
||||||
|
rpc_accepted = server_terminal_event_one |
||||||
|
self.assertIsNotNone(status_accepted) |
||||||
|
self.assertIsNotNone(rpc_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.COMPLETE_ACCEPTED, status_accepted.kind) |
||||||
|
self.assertEqual(status_tag, status_accepted.tag) |
||||||
|
self.assertTrue(status_accepted.complete_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.FINISH, rpc_accepted.kind) |
||||||
|
self.assertEqual(finish_tag, rpc_accepted.tag) |
||||||
|
self.assertEqual(_low.Status(_low.Code.OK, ''), rpc_accepted.status) |
||||||
|
|
||||||
|
client_call.read(read_tag) |
||||||
|
client_terminal_event_one = self.client_completion_queue.get(_FUTURE) |
||||||
|
client_terminal_event_two = self.client_completion_queue.get(_FUTURE) |
||||||
|
if client_terminal_event_one.kind == _low.Event.Kind.READ_ACCEPTED: |
||||||
|
read_accepted = client_terminal_event_one |
||||||
|
finish_accepted = client_terminal_event_two |
||||||
|
else: |
||||||
|
read_accepted = client_terminal_event_two |
||||||
|
finish_accepted = client_terminal_event_one |
||||||
|
self.assertIsNotNone(read_accepted) |
||||||
|
self.assertIsNotNone(finish_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.READ_ACCEPTED, read_accepted.kind) |
||||||
|
self.assertEqual(read_tag, read_accepted.tag) |
||||||
|
self.assertIsNone(read_accepted.bytes) |
||||||
|
self.assertEqual(_low.Event.Kind.FINISH, finish_accepted.kind) |
||||||
|
self.assertEqual(finish_tag, finish_accepted.tag) |
||||||
|
self.assertEqual(_low.Status(_low.Code.OK, details), finish_accepted.status) |
||||||
|
|
||||||
|
server_timeout_none_event = self.server_completion_queue.get(0) |
||||||
|
self.assertIsNone(server_timeout_none_event) |
||||||
|
client_timeout_none_event = self.client_completion_queue.get(0) |
||||||
|
self.assertIsNone(client_timeout_none_event) |
||||||
|
|
||||||
|
self.assertSequenceEqual(test_data, server_data) |
||||||
|
self.assertSequenceEqual(test_data, client_data) |
||||||
|
|
||||||
|
def testNoEcho(self): |
||||||
|
self._perform_echo_test(()) |
||||||
|
|
||||||
|
def testOneByteEcho(self): |
||||||
|
self._perform_echo_test([b'\x07']) |
||||||
|
|
||||||
|
def testOneManyByteEcho(self): |
||||||
|
self._perform_echo_test([_BYTE_SEQUENCE]) |
||||||
|
|
||||||
|
def testManyOneByteEchoes(self): |
||||||
|
self._perform_echo_test(_BYTE_SEQUENCE) |
||||||
|
|
||||||
|
def testManyManyByteEchoes(self): |
||||||
|
self._perform_echo_test(_BYTE_SEQUENCE_SEQUENCE) |
||||||
|
|
||||||
|
|
||||||
|
class CancellationTest(unittest.TestCase): |
||||||
|
|
||||||
|
def setUp(self): |
||||||
|
self.host = 'localhost' |
||||||
|
|
||||||
|
self.server_completion_queue = _low.CompletionQueue() |
||||||
|
self.server = _low.Server(self.server_completion_queue) |
||||||
|
port = self.server.add_http2_addr('[::]:0') |
||||||
|
self.server.start() |
||||||
|
|
||||||
|
self.client_completion_queue = _low.CompletionQueue() |
||||||
|
self.channel = _low.Channel('%s:%d' % (self.host, port)) |
||||||
|
|
||||||
|
def tearDown(self): |
||||||
|
self.server.stop() |
||||||
|
del self.server |
||||||
|
self.server_completion_queue.stop() |
||||||
|
self.client_completion_queue.stop() |
||||||
|
while True: |
||||||
|
event = self.server_completion_queue.get(0) |
||||||
|
if event is not None and event.kind is _low.Event.Kind.STOP: |
||||||
|
break |
||||||
|
while True: |
||||||
|
event = self.client_completion_queue.get(0) |
||||||
|
if event is not None and event.kind is _low.Event.Kind.STOP: |
||||||
|
break |
||||||
|
|
||||||
|
def testCancellation(self): |
||||||
|
method = 'test method' |
||||||
|
deadline = _FUTURE |
||||||
|
metadata_tag = object() |
||||||
|
finish_tag = object() |
||||||
|
write_tag = object() |
||||||
|
service_tag = object() |
||||||
|
read_tag = object() |
||||||
|
test_data = _BYTE_SEQUENCE_SEQUENCE |
||||||
|
|
||||||
|
server_data = [] |
||||||
|
client_data = [] |
||||||
|
|
||||||
|
client_call = _low.Call(self.channel, method, self.host, deadline) |
||||||
|
|
||||||
|
client_call.invoke(self.client_completion_queue, metadata_tag, finish_tag) |
||||||
|
|
||||||
|
self.server.service(service_tag) |
||||||
|
service_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
server_call = service_accepted.service_acceptance.call |
||||||
|
|
||||||
|
server_call.accept(self.server_completion_queue, finish_tag) |
||||||
|
server_call.premetadata() |
||||||
|
|
||||||
|
metadata_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(metadata_accepted) |
||||||
|
|
||||||
|
for datum in test_data: |
||||||
|
client_call.write(datum, write_tag) |
||||||
|
write_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
|
||||||
|
server_call.read(read_tag) |
||||||
|
read_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
server_data.append(read_accepted.bytes) |
||||||
|
|
||||||
|
server_call.write(read_accepted.bytes, write_tag) |
||||||
|
write_accepted = self.server_completion_queue.get(_FUTURE) |
||||||
|
self.assertIsNotNone(write_accepted) |
||||||
|
|
||||||
|
client_call.read(read_tag) |
||||||
|
read_accepted = self.client_completion_queue.get(_FUTURE) |
||||||
|
client_data.append(read_accepted.bytes) |
||||||
|
|
||||||
|
client_call.cancel() |
||||||
|
# cancel() is idempotent. |
||||||
|
client_call.cancel() |
||||||
|
client_call.cancel() |
||||||
|
client_call.cancel() |
||||||
|
|
||||||
|
server_call.read(read_tag) |
||||||
|
|
||||||
|
server_terminal_event_one = self.server_completion_queue.get(_FUTURE) |
||||||
|
server_terminal_event_two = self.server_completion_queue.get(_FUTURE) |
||||||
|
if server_terminal_event_one.kind == _low.Event.Kind.READ_ACCEPTED: |
||||||
|
read_accepted = server_terminal_event_one |
||||||
|
rpc_accepted = server_terminal_event_two |
||||||
|
else: |
||||||
|
read_accepted = server_terminal_event_two |
||||||
|
rpc_accepted = server_terminal_event_one |
||||||
|
self.assertIsNotNone(read_accepted) |
||||||
|
self.assertIsNotNone(rpc_accepted) |
||||||
|
self.assertEqual(_low.Event.Kind.READ_ACCEPTED, read_accepted.kind) |
||||||
|
self.assertIsNone(read_accepted.bytes) |
||||||
|
self.assertEqual(_low.Event.Kind.FINISH, rpc_accepted.kind) |
||||||
|
self.assertEqual(_low.Status(_low.Code.CANCELLED, ''), rpc_accepted.status) |
||||||
|
|
||||||
|
finish_event = self.client_completion_queue.get(_FUTURE) |
||||||
|
self.assertEqual(_low.Event.Kind.FINISH, finish_event.kind) |
||||||
|
self.assertEqual(_low.Status(_low.Code.CANCELLED, ''), finish_event.status) |
||||||
|
|
||||||
|
server_timeout_none_event = self.server_completion_queue.get(0) |
||||||
|
self.assertIsNone(server_timeout_none_event) |
||||||
|
client_timeout_none_event = self.client_completion_queue.get(0) |
||||||
|
self.assertIsNone(client_timeout_none_event) |
||||||
|
|
||||||
|
self.assertSequenceEqual(test_data, server_data) |
||||||
|
self.assertSequenceEqual(test_data, client_data) |
||||||
|
|
||||||
|
|
||||||
|
class ExpirationTest(unittest.TestCase): |
||||||
|
|
||||||
|
@unittest.skip('TODO(nathaniel): Expiration test!') |
||||||
|
def testExpiration(self): |
||||||
|
pass |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main() |
@ -0,0 +1,261 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Test scenarios using protocol buffers.""" |
||||||
|
|
||||||
|
import abc |
||||||
|
import threading |
||||||
|
|
||||||
|
from _junkdrawer import math_pb2 |
||||||
|
|
||||||
|
|
||||||
|
class ProtoScenario(object): |
||||||
|
"""An RPC test scenario using protocol buffers.""" |
||||||
|
__metaclass__ = abc.ABCMeta |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def method(self): |
||||||
|
"""Access the test method name. |
||||||
|
|
||||||
|
Returns: |
||||||
|
The test method name. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def serialize_request(self, request): |
||||||
|
"""Serialize a request protocol buffer. |
||||||
|
|
||||||
|
Args: |
||||||
|
request: A request protocol buffer. |
||||||
|
|
||||||
|
Returns: |
||||||
|
The bytestring serialization of the given request protocol buffer. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def deserialize_request(self, request_bytestring): |
||||||
|
"""Deserialize a request protocol buffer. |
||||||
|
|
||||||
|
Args: |
||||||
|
request_bytestring: The bytestring serialization of a request protocol |
||||||
|
buffer. |
||||||
|
|
||||||
|
Returns: |
||||||
|
The request protocol buffer deserialized from the given byte string. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def serialize_response(self, response): |
||||||
|
"""Serialize a response protocol buffer. |
||||||
|
|
||||||
|
Args: |
||||||
|
response: A response protocol buffer. |
||||||
|
|
||||||
|
Returns: |
||||||
|
The bytestring serialization of the given response protocol buffer. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def deserialize_response(self, response_bytestring): |
||||||
|
"""Deserialize a response protocol buffer. |
||||||
|
|
||||||
|
Args: |
||||||
|
response_bytestring: The bytestring serialization of a response protocol |
||||||
|
buffer. |
||||||
|
|
||||||
|
Returns: |
||||||
|
The response protocol buffer deserialized from the given byte string. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def requests(self): |
||||||
|
"""Access the sequence of requests for this scenario. |
||||||
|
|
||||||
|
Returns: |
||||||
|
A sequence of request protocol buffers. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def response_for_request(self, request): |
||||||
|
"""Access the response for a particular request. |
||||||
|
|
||||||
|
Args: |
||||||
|
request: A request protocol buffer. |
||||||
|
|
||||||
|
Returns: |
||||||
|
The response protocol buffer appropriate for the given request. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def verify_requests(self, experimental_requests): |
||||||
|
"""Verify the requests transmitted through the system under test. |
||||||
|
|
||||||
|
Args: |
||||||
|
experimental_requests: The request protocol buffers transmitted through |
||||||
|
the system under test. |
||||||
|
|
||||||
|
Returns: |
||||||
|
True if the requests satisfy this test scenario; False otherwise. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
@abc.abstractmethod |
||||||
|
def verify_responses(self, experimental_responses): |
||||||
|
"""Verify the responses transmitted through the system under test. |
||||||
|
|
||||||
|
Args: |
||||||
|
experimental_responses: The response protocol buffers transmitted through |
||||||
|
the system under test. |
||||||
|
|
||||||
|
Returns: |
||||||
|
True if the responses satisfy this test scenario; False otherwise. |
||||||
|
""" |
||||||
|
raise NotImplementedError() |
||||||
|
|
||||||
|
|
||||||
|
class EmptyScenario(ProtoScenario): |
||||||
|
"""A scenario that transmits no protocol buffers in either direction.""" |
||||||
|
|
||||||
|
def method(self): |
||||||
|
return 'DivMany' |
||||||
|
|
||||||
|
def serialize_request(self, request): |
||||||
|
raise ValueError('This should not be necessary to call!') |
||||||
|
|
||||||
|
def deserialize_request(self, request_bytestring): |
||||||
|
raise ValueError('This should not be necessary to call!') |
||||||
|
|
||||||
|
def serialize_response(self, response): |
||||||
|
raise ValueError('This should not be necessary to call!') |
||||||
|
|
||||||
|
def deserialize_response(self, response_bytestring): |
||||||
|
raise ValueError('This should not be necessary to call!') |
||||||
|
|
||||||
|
def requests(self): |
||||||
|
return () |
||||||
|
|
||||||
|
def response_for_request(self, request): |
||||||
|
raise ValueError('This should not be necessary to call!') |
||||||
|
|
||||||
|
def verify_requests(self, experimental_requests): |
||||||
|
return not experimental_requests |
||||||
|
|
||||||
|
def verify_responses(self, experimental_responses): |
||||||
|
return not experimental_responses |
||||||
|
|
||||||
|
|
||||||
|
class BidirectionallyUnaryScenario(ProtoScenario): |
||||||
|
"""A scenario that transmits no protocol buffers in either direction.""" |
||||||
|
|
||||||
|
_DIVIDEND = 59 |
||||||
|
_DIVISOR = 7 |
||||||
|
_QUOTIENT = 8 |
||||||
|
_REMAINDER = 3 |
||||||
|
|
||||||
|
_REQUEST = math_pb2.DivArgs(dividend=_DIVIDEND, divisor=_DIVISOR) |
||||||
|
_RESPONSE = math_pb2.DivReply(quotient=_QUOTIENT, remainder=_REMAINDER) |
||||||
|
|
||||||
|
def method(self): |
||||||
|
return 'Div' |
||||||
|
|
||||||
|
def serialize_request(self, request): |
||||||
|
return request.SerializeToString() |
||||||
|
|
||||||
|
def deserialize_request(self, request_bytestring): |
||||||
|
return math_pb2.DivArgs.FromString(request_bytestring) |
||||||
|
|
||||||
|
def serialize_response(self, response): |
||||||
|
return response.SerializeToString() |
||||||
|
|
||||||
|
def deserialize_response(self, response_bytestring): |
||||||
|
return math_pb2.DivReply.FromString(response_bytestring) |
||||||
|
|
||||||
|
def requests(self): |
||||||
|
return [self._REQUEST] |
||||||
|
|
||||||
|
def response_for_request(self, request): |
||||||
|
return self._RESPONSE |
||||||
|
|
||||||
|
def verify_requests(self, experimental_requests): |
||||||
|
return tuple(experimental_requests) == (self._REQUEST,) |
||||||
|
|
||||||
|
def verify_responses(self, experimental_responses): |
||||||
|
return tuple(experimental_responses) == (self._RESPONSE,) |
||||||
|
|
||||||
|
|
||||||
|
class BidirectionallyStreamingScenario(ProtoScenario): |
||||||
|
"""A scenario that transmits no protocol buffers in either direction.""" |
||||||
|
|
||||||
|
_STREAM_LENGTH = 200 |
||||||
|
_REQUESTS = tuple( |
||||||
|
math_pb2.DivArgs(dividend=59 + index, divisor=7 + index) |
||||||
|
for index in range(_STREAM_LENGTH)) |
||||||
|
|
||||||
|
def __init__(self): |
||||||
|
self._lock = threading.Lock() |
||||||
|
self._responses = [] |
||||||
|
|
||||||
|
def method(self): |
||||||
|
return 'DivMany' |
||||||
|
|
||||||
|
def serialize_request(self, request): |
||||||
|
return request.SerializeToString() |
||||||
|
|
||||||
|
def deserialize_request(self, request_bytestring): |
||||||
|
return math_pb2.DivArgs.FromString(request_bytestring) |
||||||
|
|
||||||
|
def serialize_response(self, response): |
||||||
|
return response.SerializeToString() |
||||||
|
|
||||||
|
def deserialize_response(self, response_bytestring): |
||||||
|
return math_pb2.DivReply.FromString(response_bytestring) |
||||||
|
|
||||||
|
def requests(self): |
||||||
|
return self._REQUESTS |
||||||
|
|
||||||
|
def response_for_request(self, request): |
||||||
|
quotient, remainder = divmod(request.dividend, request.divisor) |
||||||
|
response = math_pb2.DivReply(quotient=quotient, remainder=remainder) |
||||||
|
with self._lock: |
||||||
|
self._responses.append(response) |
||||||
|
return response |
||||||
|
|
||||||
|
def verify_requests(self, experimental_requests): |
||||||
|
return tuple(experimental_requests) == self._REQUESTS |
||||||
|
|
||||||
|
def verify_responses(self, experimental_responses): |
||||||
|
with self._lock: |
||||||
|
return tuple(experimental_responses) == tuple(self._responses) |
@ -0,0 +1,167 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 "_adapter/_server.h" |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
#include "_adapter/_completion_queue.h" |
||||||
|
#include "_adapter/_error.h" |
||||||
|
|
||||||
|
static int pygrpc_server_init(Server *self, PyObject *args, PyObject *kwds) { |
||||||
|
const PyObject *completion_queue; |
||||||
|
if (!(PyArg_ParseTuple(args, "O!", &pygrpc_CompletionQueueType, |
||||||
|
&completion_queue))) { |
||||||
|
self->c_server = NULL; |
||||||
|
return -1; |
||||||
|
} |
||||||
|
|
||||||
|
self->c_server = grpc_server_create( |
||||||
|
((CompletionQueue *)completion_queue)->c_completion_queue, NULL); |
||||||
|
return 0; |
||||||
|
} |
||||||
|
|
||||||
|
static void pygrpc_server_dealloc(Server *self) { |
||||||
|
if (self->c_server != NULL) { |
||||||
|
grpc_server_destroy(self->c_server); |
||||||
|
} |
||||||
|
self->ob_type->tp_free((PyObject *)self); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_server_add_http2_addr(Server *self, PyObject *args) { |
||||||
|
const char *addr; |
||||||
|
int port; |
||||||
|
PyArg_ParseTuple(args, "s", &addr); |
||||||
|
|
||||||
|
port = grpc_server_add_http2_port(self->c_server, addr); |
||||||
|
if (port == 0) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "Couldn't add port to server!"); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
return PyInt_FromLong(port); |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_server_start(Server *self) { |
||||||
|
grpc_server_start(self->c_server); |
||||||
|
|
||||||
|
Py_RETURN_NONE; |
||||||
|
} |
||||||
|
|
||||||
|
static const PyObject *pygrpc_server_service(Server *self, PyObject *args) { |
||||||
|
const PyObject *tag; |
||||||
|
grpc_call_error call_error; |
||||||
|
const PyObject *result; |
||||||
|
|
||||||
|
if (!(PyArg_ParseTuple(args, "O", &tag))) { |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
|
||||||
|
call_error = grpc_server_request_call(self->c_server, (void *)tag); |
||||||
|
|
||||||
|
result = pygrpc_translate_call_error(call_error); |
||||||
|
if (result != NULL) { |
||||||
|
Py_INCREF(tag); |
||||||
|
} |
||||||
|
return result; |
||||||
|
} |
||||||
|
|
||||||
|
static PyObject *pygrpc_server_stop(Server *self) { |
||||||
|
grpc_server_shutdown(self->c_server); |
||||||
|
|
||||||
|
Py_RETURN_NONE; |
||||||
|
} |
||||||
|
|
||||||
|
static PyMethodDef methods[] = { |
||||||
|
{"add_http2_addr", (PyCFunction)pygrpc_server_add_http2_addr, METH_VARARGS, |
||||||
|
"Add an HTTP2 address."}, |
||||||
|
{"start", (PyCFunction)pygrpc_server_start, METH_NOARGS, |
||||||
|
"Starts the server."}, |
||||||
|
{"service", (PyCFunction)pygrpc_server_service, METH_VARARGS, |
||||||
|
"Services a call."}, |
||||||
|
{"stop", (PyCFunction)pygrpc_server_stop, METH_NOARGS, "Stops the server."}, |
||||||
|
{NULL}}; |
||||||
|
|
||||||
|
static PyTypeObject pygrpc_ServerType = { |
||||||
|
PyObject_HEAD_INIT(NULL)0, /*ob_size*/ |
||||||
|
"_gprc.Server", /*tp_name*/ |
||||||
|
sizeof(Server), /*tp_basicsize*/ |
||||||
|
0, /*tp_itemsize*/ |
||||||
|
(destructor)pygrpc_server_dealloc, /*tp_dealloc*/ |
||||||
|
0, /*tp_print*/ |
||||||
|
0, /*tp_getattr*/ |
||||||
|
0, /*tp_setattr*/ |
||||||
|
0, /*tp_compare*/ |
||||||
|
0, /*tp_repr*/ |
||||||
|
0, /*tp_as_number*/ |
||||||
|
0, /*tp_as_sequence*/ |
||||||
|
0, /*tp_as_mapping*/ |
||||||
|
0, /*tp_hash */ |
||||||
|
0, /*tp_call*/ |
||||||
|
0, /*tp_str*/ |
||||||
|
0, /*tp_getattro*/ |
||||||
|
0, /*tp_setattro*/ |
||||||
|
0, /*tp_as_buffer*/ |
||||||
|
Py_TPFLAGS_DEFAULT, /*tp_flags*/ |
||||||
|
"Wrapping of grpc_server.", /* tp_doc */ |
||||||
|
0, /* tp_traverse */ |
||||||
|
0, /* tp_clear */ |
||||||
|
0, /* tp_richcompare */ |
||||||
|
0, /* tp_weaklistoffset */ |
||||||
|
0, /* tp_iter */ |
||||||
|
0, /* tp_iternext */ |
||||||
|
methods, /* tp_methods */ |
||||||
|
0, /* tp_members */ |
||||||
|
0, /* tp_getset */ |
||||||
|
0, /* tp_base */ |
||||||
|
0, /* tp_dict */ |
||||||
|
0, /* tp_descr_get */ |
||||||
|
0, /* tp_descr_set */ |
||||||
|
0, /* tp_dictoffset */ |
||||||
|
(initproc)pygrpc_server_init, /* tp_init */ |
||||||
|
}; |
||||||
|
|
||||||
|
int pygrpc_add_server(PyObject *module) { |
||||||
|
pygrpc_ServerType.tp_new = PyType_GenericNew; |
||||||
|
if (PyType_Ready(&pygrpc_ServerType) < 0) { |
||||||
|
PyErr_SetString(PyExc_RuntimeError, "Error defining pygrpc_ServerType!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
if (PyModule_AddObject(module, "Server", (PyObject *)&pygrpc_ServerType) == |
||||||
|
-1) { |
||||||
|
PyErr_SetString(PyExc_ImportError, "Couldn't add Server type to module!"); |
||||||
|
return -1; |
||||||
|
} |
||||||
|
return 0; |
||||||
|
} |
@ -0,0 +1,44 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#ifndef _ADAPTER__SERVER_H_ |
||||||
|
#define _ADAPTER__SERVER_H_ |
||||||
|
|
||||||
|
#include <Python.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
|
||||||
|
typedef struct { PyObject_HEAD grpc_server *c_server; } Server; |
||||||
|
|
||||||
|
int pygrpc_add_server(PyObject *module); |
||||||
|
|
||||||
|
#endif /* _ADAPTER__SERVER_H_ */ |
@ -0,0 +1,80 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
"""Links suitable for use in tests.""" |
||||||
|
|
||||||
|
import threading |
||||||
|
|
||||||
|
from _framework.base.packets import interfaces |
||||||
|
|
||||||
|
|
||||||
|
class ForeLink(interfaces.ForeLink): |
||||||
|
"""A ForeLink suitable for use in tests of RearLinks.""" |
||||||
|
|
||||||
|
def __init__(self, action, rear_link): |
||||||
|
self.condition = threading.Condition() |
||||||
|
self.tickets = [] |
||||||
|
self.action = action |
||||||
|
self.rear_link = rear_link |
||||||
|
|
||||||
|
def accept_back_to_front_ticket(self, ticket): |
||||||
|
with self.condition: |
||||||
|
self.tickets.append(ticket) |
||||||
|
self.condition.notify_all() |
||||||
|
action, rear_link = self.action, self.rear_link |
||||||
|
|
||||||
|
if action is not None: |
||||||
|
action(ticket, rear_link) |
||||||
|
|
||||||
|
def join_rear_link(self, rear_link): |
||||||
|
with self.condition: |
||||||
|
self.rear_link = rear_link |
||||||
|
|
||||||
|
|
||||||
|
class RearLink(interfaces.RearLink): |
||||||
|
"""A RearLink suitable for use in tests of ForeLinks.""" |
||||||
|
|
||||||
|
def __init__(self, action, fore_link): |
||||||
|
self.condition = threading.Condition() |
||||||
|
self.tickets = [] |
||||||
|
self.action = action |
||||||
|
self.fore_link = fore_link |
||||||
|
|
||||||
|
def accept_front_to_back_ticket(self, ticket): |
||||||
|
with self.condition: |
||||||
|
self.tickets.append(ticket) |
||||||
|
self.condition.notify_all() |
||||||
|
action, fore_link = self.action, self.fore_link |
||||||
|
|
||||||
|
if action is not None: |
||||||
|
action(ticket, fore_link) |
||||||
|
|
||||||
|
def join_fore_link(self, fore_link): |
||||||
|
with self.condition: |
||||||
|
self.fore_link = fore_link |
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue