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 |
||||
// DO NOT EDIT! Generated by Protobuf-PHP protoc plugin 1.0 |
||||
// Source: third_party/stubby/testing/proto/test.proto |
||||
// Date: 2014-12-03 22:02:20 |
||||
// Source: test/cpp/interop/test.proto |
||||
// Date: 2015-01-30 23:30:46 |
||||
|
||||
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 |
||||
* @return proto2\EmptyMessage |
||||
* @param grpc\testing\EmptyMessage $input |
||||
*/ |
||||
public function EmptyCall(\proto2\EmptyMessage $argument, $metadata = array()) { |
||||
return $this->_simpleRequest('/TestService/EmptyCall', $argument, '\proto2\EmptyMessage::deserialize', $metadata); |
||||
public function EmptyCall(\grpc\testing\EmptyMessage $argument, $metadata = array()) { |
||||
return $this->rpc_impl->_simpleRequest('/grpc.testing.TestService/EmptyCall', $argument, '\grpc\testing\EmptyMessage::deserialize', $metadata); |
||||
} |
||||
/** |
||||
* @param grpc\testing\SimpleRequest $input |
||||
* @return grpc\testing\SimpleResponse |
||||
*/ |
||||
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 |
||||
* @return grpc\testing\StreamingOutputCallResponse |
||||
*/ |
||||
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 |
||||
* @return grpc\testing\StreamingInputCallResponse |
||||
*/ |
||||
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 |
||||
* @return grpc\testing\StreamingOutputCallResponse |
||||
*/ |
||||
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 |
||||
* @return grpc\testing\StreamingOutputCallResponse |
||||
*/ |
||||
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