Jan Tattermusch
9 years ago
commit
f8bd01f934
122 changed files with 11294 additions and 204 deletions
@ -0,0 +1,85 @@ |
|||||||
|
GRPC C++ STYLE GUIDE |
||||||
|
===================== |
||||||
|
|
||||||
|
Background |
||||||
|
---------- |
||||||
|
|
||||||
|
Here we document style rules for C++ usage in the gRPC C++ bindings |
||||||
|
and tests. |
||||||
|
|
||||||
|
General |
||||||
|
------- |
||||||
|
|
||||||
|
- The majority of gRPC's C++ requirements are drawn from the [Google C++ style |
||||||
|
guide] (https://google.github.io/styleguide/cppguide.html) |
||||||
|
- However, gRPC has some additional requirements to maintain |
||||||
|
[portability] (#portability) |
||||||
|
- As in C, layout rules are defined by clang-format, and all code |
||||||
|
should be passed through clang-format. A (docker-based) script to do |
||||||
|
so is included in [tools/distrib/clang\_format\_code.sh] |
||||||
|
(../tools/distrib/clang_format_code.sh). |
||||||
|
|
||||||
|
<a name="portability"></a> |
||||||
|
Portability Restrictions |
||||||
|
------------------- |
||||||
|
|
||||||
|
gRPC supports a large number of compilers, ranging from those that are |
||||||
|
missing many key C++11 features to those that have quite detailed |
||||||
|
analysis. As a result, gRPC compiles with a high level of warnings and |
||||||
|
treat all warnings as errors. gRPC also forbids the use of some common |
||||||
|
C++11 constructs. Here are some guidelines, to be extended as needed: |
||||||
|
- Do not use range-based for. Expressions of the form |
||||||
|
```c |
||||||
|
for (auto& i: vec) { |
||||||
|
// code |
||||||
|
} |
||||||
|
``` |
||||||
|
|
||||||
|
are not allowed and should be replaced with code such as |
||||||
|
```c |
||||||
|
for (auto it = vec.begin; it != vec.end(); it++) { |
||||||
|
auto& i = *it; |
||||||
|
// code |
||||||
|
} |
||||||
|
``` |
||||||
|
|
||||||
|
- Do not use lambda of any kind (no capture, explicit capture, or |
||||||
|
default capture). Other C++ functional features such as |
||||||
|
`std::function` or `std::bind` are allowed |
||||||
|
- Do not use brace-list initializers. |
||||||
|
- Do not compare a pointer to `nullptr` . This is because gcc 4.4 |
||||||
|
does not support `nullptr` directly and gRPC implements a subset of |
||||||
|
its features in [include/grpc++/impl/codegen/config.h] |
||||||
|
(../include/grpc++/impl/codegen/config.h). Instead, pointers should |
||||||
|
be checked for validity using their implicit conversion to `bool`. |
||||||
|
In other words, use `if (p)` rather than `if (p != nullptr)` |
||||||
|
- Do not use `final` or `override` as these are not supported by some |
||||||
|
compilers. Instead use `GRPC_FINAL` and `GRPC_OVERRIDE` . These |
||||||
|
compile down to the traditional C++ forms for compilers that support |
||||||
|
them but are just elided if the compiler does not support those features. |
||||||
|
- In the [include] (../../../tree/master/include/grpc++) and [src] |
||||||
|
(../../../tree/master/src/cpp) directory trees, you should also not |
||||||
|
use certain STL objects like `std::mutex`, `std::lock_guard`, |
||||||
|
`std::unique_lock`, `std::nullptr`, `std::thread` . Instead, use |
||||||
|
`grpc::mutex`, `grpc::lock_guard`, etc., which are gRPC |
||||||
|
implementations of the prominent features of these objects that are |
||||||
|
not always available. You can use the `std` versions of those in [test] |
||||||
|
(../../../tree/master/test/cpp) |
||||||
|
- Similarly, in the same directories, do not use `std::chrono` unless |
||||||
|
it is guarded by `#ifndef GRPC_CXX0X_NO_CHRONO` . For platforms that |
||||||
|
lack`std::chrono,` there is a C-language timer called gpr_timespec that can |
||||||
|
be used instead. |
||||||
|
- `std::unique_ptr` must be used with extreme care in any kind of |
||||||
|
collection. For example `vector<std::unique_ptr>` does not work in |
||||||
|
gcc 4.4 if the vector is constructed to its full size at |
||||||
|
initialization but does work if elements are added to the vector |
||||||
|
using functions like `push_back`. `map` and other pair-based |
||||||
|
collections do not work with `unique_ptr` under gcc 4.4. The issue |
||||||
|
is that many of these collection implementations assume a copy |
||||||
|
constructor |
||||||
|
to be available. |
||||||
|
- Don't use `std::this_thread` . Use `gpr_sleep_until` for sleeping a thread. |
||||||
|
- [Some adjacent character combinations cause problems] |
||||||
|
(https://en.wikipedia.org/wiki/Digraphs_and_trigraphs#C). If declaring a |
||||||
|
template against some class relative to the global namespace, |
||||||
|
`<::name` will be non-portable. Separate the `<` from the `:` and use `< ::name`. |
||||||
@ -0,0 +1,69 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 GRPCXX_EXT_PROTO_SERVER_REFLECTION_PLUGIN_H |
||||||
|
#define GRPCXX_EXT_PROTO_SERVER_REFLECTION_PLUGIN_H |
||||||
|
|
||||||
|
#include <grpc++/impl/server_builder_plugin.h> |
||||||
|
#include <grpc++/support/config.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
class ServerInitializer; |
||||||
|
class ProtoServerReflection; |
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace reflection { |
||||||
|
|
||||||
|
class ProtoServerReflectionPlugin : public ::grpc::ServerBuilderPlugin { |
||||||
|
public: |
||||||
|
ProtoServerReflectionPlugin(); |
||||||
|
::grpc::string name() GRPC_OVERRIDE; |
||||||
|
void InitServer(::grpc::ServerInitializer* si) GRPC_OVERRIDE; |
||||||
|
void Finish(::grpc::ServerInitializer* si) GRPC_OVERRIDE; |
||||||
|
void ChangeArguments(const ::grpc::string& name, void* value) GRPC_OVERRIDE; |
||||||
|
bool has_async_methods() const GRPC_OVERRIDE; |
||||||
|
bool has_sync_methods() const GRPC_OVERRIDE; |
||||||
|
|
||||||
|
private: |
||||||
|
std::shared_ptr<::grpc::ProtoServerReflection> reflection_service_; |
||||||
|
}; |
||||||
|
|
||||||
|
// Add proto reflection plugin to ServerBuilder. This function should be called
|
||||||
|
// at the static initialization time.
|
||||||
|
void InitProtoReflectionServerBuilderPlugin(); |
||||||
|
|
||||||
|
} // namespace reflection
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPCXX_EXT_PROTO_SERVER_REFLECTION_PLUGIN_H
|
||||||
@ -0,0 +1,184 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
|
||||||
|
// Generated by the gRPC protobuf plugin.
|
||||||
|
// If you make any local change, they will be lost.
|
||||||
|
// source: reflection.proto
|
||||||
|
// Original file comments:
|
||||||
|
// Copyright 2016, 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.
|
||||||
|
//
|
||||||
|
// Service exported by server reflection
|
||||||
|
//
|
||||||
|
#ifndef GRPC_reflection_2eproto__INCLUDED |
||||||
|
#define GRPC_reflection_2eproto__INCLUDED |
||||||
|
|
||||||
|
#include <grpc++/ext/reflection.pb.h> |
||||||
|
|
||||||
|
#include <grpc++/impl/codegen/async_stream.h> |
||||||
|
#include <grpc++/impl/codegen/async_unary_call.h> |
||||||
|
#include <grpc++/impl/codegen/proto_utils.h> |
||||||
|
#include <grpc++/impl/codegen/rpc_method.h> |
||||||
|
#include <grpc++/impl/codegen/service_type.h> |
||||||
|
#include <grpc++/impl/codegen/status.h> |
||||||
|
#include <grpc++/impl/codegen/stub_options.h> |
||||||
|
#include <grpc++/impl/codegen/sync_stream.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
class CompletionQueue; |
||||||
|
class Channel; |
||||||
|
class RpcService; |
||||||
|
class ServerCompletionQueue; |
||||||
|
class ServerContext; |
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace reflection { |
||||||
|
namespace v1alpha { |
||||||
|
|
||||||
|
class ServerReflection GRPC_FINAL { |
||||||
|
public: |
||||||
|
class StubInterface { |
||||||
|
public: |
||||||
|
virtual ~StubInterface() {} |
||||||
|
// The reflection service is structured as a bidirectional stream, ensuring
|
||||||
|
// all related requests go to a single server.
|
||||||
|
std::unique_ptr< ::grpc::ClientReaderWriterInterface< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>> ServerReflectionInfo(::grpc::ClientContext* context) { |
||||||
|
return std::unique_ptr< ::grpc::ClientReaderWriterInterface< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>>(ServerReflectionInfoRaw(context)); |
||||||
|
} |
||||||
|
std::unique_ptr< ::grpc::ClientAsyncReaderWriterInterface< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>> AsyncServerReflectionInfo(::grpc::ClientContext* context, ::grpc::CompletionQueue* cq, void* tag) { |
||||||
|
return std::unique_ptr< ::grpc::ClientAsyncReaderWriterInterface< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>>(AsyncServerReflectionInfoRaw(context, cq, tag)); |
||||||
|
} |
||||||
|
private: |
||||||
|
virtual ::grpc::ClientReaderWriterInterface< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>* ServerReflectionInfoRaw(::grpc::ClientContext* context) = 0; |
||||||
|
virtual ::grpc::ClientAsyncReaderWriterInterface< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>* AsyncServerReflectionInfoRaw(::grpc::ClientContext* context, ::grpc::CompletionQueue* cq, void* tag) = 0; |
||||||
|
}; |
||||||
|
class Stub GRPC_FINAL : public StubInterface { |
||||||
|
public: |
||||||
|
Stub(const std::shared_ptr< ::grpc::ChannelInterface>& channel); |
||||||
|
std::unique_ptr< ::grpc::ClientReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>> ServerReflectionInfo(::grpc::ClientContext* context) { |
||||||
|
return std::unique_ptr< ::grpc::ClientReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>>(ServerReflectionInfoRaw(context)); |
||||||
|
} |
||||||
|
std::unique_ptr< ::grpc::ClientAsyncReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>> AsyncServerReflectionInfo(::grpc::ClientContext* context, ::grpc::CompletionQueue* cq, void* tag) { |
||||||
|
return std::unique_ptr< ::grpc::ClientAsyncReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>>(AsyncServerReflectionInfoRaw(context, cq, tag)); |
||||||
|
} |
||||||
|
|
||||||
|
private: |
||||||
|
std::shared_ptr< ::grpc::ChannelInterface> channel_; |
||||||
|
::grpc::ClientReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>* ServerReflectionInfoRaw(::grpc::ClientContext* context) GRPC_OVERRIDE; |
||||||
|
::grpc::ClientAsyncReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>* AsyncServerReflectionInfoRaw(::grpc::ClientContext* context, ::grpc::CompletionQueue* cq, void* tag) GRPC_OVERRIDE; |
||||||
|
const ::grpc::RpcMethod rpcmethod_ServerReflectionInfo_; |
||||||
|
}; |
||||||
|
static std::unique_ptr<Stub> NewStub(const std::shared_ptr< ::grpc::ChannelInterface>& channel, const ::grpc::StubOptions& options = ::grpc::StubOptions()); |
||||||
|
|
||||||
|
class Service : public ::grpc::Service { |
||||||
|
public: |
||||||
|
Service(); |
||||||
|
virtual ~Service(); |
||||||
|
// The reflection service is structured as a bidirectional stream, ensuring
|
||||||
|
// all related requests go to a single server.
|
||||||
|
virtual ::grpc::Status ServerReflectionInfo(::grpc::ServerContext* context, ::grpc::ServerReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionResponse, ::grpc::reflection::v1alpha::ServerReflectionRequest>* stream); |
||||||
|
}; |
||||||
|
template <class BaseClass> |
||||||
|
class WithAsyncMethod_ServerReflectionInfo : public BaseClass { |
||||||
|
private: |
||||||
|
void BaseClassMustBeDerivedFromService(const Service *service) {} |
||||||
|
public: |
||||||
|
WithAsyncMethod_ServerReflectionInfo() { |
||||||
|
::grpc::Service::MarkMethodAsync(0); |
||||||
|
} |
||||||
|
~WithAsyncMethod_ServerReflectionInfo() GRPC_OVERRIDE { |
||||||
|
BaseClassMustBeDerivedFromService(this); |
||||||
|
} |
||||||
|
// disable synchronous version of this method
|
||||||
|
::grpc::Status ServerReflectionInfo(::grpc::ServerContext* context, ::grpc::ServerReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionResponse, ::grpc::reflection::v1alpha::ServerReflectionRequest>* stream) GRPC_FINAL GRPC_OVERRIDE { |
||||||
|
abort(); |
||||||
|
return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, ""); |
||||||
|
} |
||||||
|
void RequestServerReflectionInfo(::grpc::ServerContext* context, ::grpc::ServerAsyncReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionResponse, ::grpc::reflection::v1alpha::ServerReflectionRequest>* stream, ::grpc::CompletionQueue* new_call_cq, ::grpc::ServerCompletionQueue* notification_cq, void *tag) { |
||||||
|
::grpc::Service::RequestAsyncBidiStreaming(0, context, stream, new_call_cq, notification_cq, tag); |
||||||
|
} |
||||||
|
}; |
||||||
|
typedef WithAsyncMethod_ServerReflectionInfo<Service > AsyncService; |
||||||
|
template <class BaseClass> |
||||||
|
class WithGenericMethod_ServerReflectionInfo : public BaseClass { |
||||||
|
private: |
||||||
|
void BaseClassMustBeDerivedFromService(const Service *service) {} |
||||||
|
public: |
||||||
|
WithGenericMethod_ServerReflectionInfo() { |
||||||
|
::grpc::Service::MarkMethodGeneric(0); |
||||||
|
} |
||||||
|
~WithGenericMethod_ServerReflectionInfo() GRPC_OVERRIDE { |
||||||
|
BaseClassMustBeDerivedFromService(this); |
||||||
|
} |
||||||
|
// disable synchronous version of this method
|
||||||
|
::grpc::Status ServerReflectionInfo(::grpc::ServerContext* context, ::grpc::ServerReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionResponse, ::grpc::reflection::v1alpha::ServerReflectionRequest>* stream) GRPC_FINAL GRPC_OVERRIDE { |
||||||
|
abort(); |
||||||
|
return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, ""); |
||||||
|
} |
||||||
|
}; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace v1alpha
|
||||||
|
} // namespace reflection
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
|
||||||
|
#endif // GRPC_reflection_2eproto__INCLUDED
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,232 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "src/core/ext/transport/chttp2/transport/bin_decoder.h" |
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include "src/core/lib/support/string.h" |
||||||
|
|
||||||
|
static uint8_t decode_table[] = { |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 62, 0x40, 0x40, 0x40, 63, |
||||||
|
52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0, 1, 2, 3, 4, 5, 6, |
||||||
|
7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, |
||||||
|
19, 20, 21, 22, 23, 24, 25, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, |
||||||
|
37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, |
||||||
|
49, 50, 51, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, 0x40, |
||||||
|
0x40, 0x40, 0x40, 0x40}; |
||||||
|
|
||||||
|
static const uint8_t tail_xtra[4] = {0, 0, 1, 2}; |
||||||
|
|
||||||
|
static bool input_is_valid(uint8_t *input_ptr, size_t length) { |
||||||
|
size_t i; |
||||||
|
|
||||||
|
for (i = 0; i < length; ++i) { |
||||||
|
if ((decode_table[input_ptr[i]] & 0xC0) != 0) { |
||||||
|
gpr_log(GPR_ERROR, |
||||||
|
"Base64 decoding failed, invalid character '%c' in base64 " |
||||||
|
"input.\n", |
||||||
|
(char)(*input_ptr)); |
||||||
|
return false; |
||||||
|
} |
||||||
|
} |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
#define COMPOSE_OUTPUT_BYTE_0(input_ptr) \ |
||||||
|
(uint8_t)((decode_table[input_ptr[0]] << 2) | \
|
||||||
|
(decode_table[input_ptr[1]] >> 4)) |
||||||
|
|
||||||
|
#define COMPOSE_OUTPUT_BYTE_1(input_ptr) \ |
||||||
|
(uint8_t)((decode_table[input_ptr[1]] << 4) | \
|
||||||
|
(decode_table[input_ptr[2]] >> 2)) |
||||||
|
|
||||||
|
#define COMPOSE_OUTPUT_BYTE_2(input_ptr) \ |
||||||
|
(uint8_t)((decode_table[input_ptr[2]] << 6) | decode_table[input_ptr[3]]) |
||||||
|
|
||||||
|
bool grpc_base64_decode_partial(struct grpc_base64_decode_context *ctx) { |
||||||
|
size_t input_tail; |
||||||
|
|
||||||
|
if (ctx->input_cur > ctx->input_end || ctx->output_cur > ctx->output_end) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
// Process a block of 4 input characters and 3 output bytes
|
||||||
|
while (ctx->input_end >= ctx->input_cur + 4 && |
||||||
|
ctx->output_end >= ctx->output_cur + 3) { |
||||||
|
if (!input_is_valid(ctx->input_cur, 4)) return false; |
||||||
|
ctx->output_cur[0] = COMPOSE_OUTPUT_BYTE_0(ctx->input_cur); |
||||||
|
ctx->output_cur[1] = COMPOSE_OUTPUT_BYTE_1(ctx->input_cur); |
||||||
|
ctx->output_cur[2] = COMPOSE_OUTPUT_BYTE_2(ctx->input_cur); |
||||||
|
ctx->output_cur += 3; |
||||||
|
ctx->input_cur += 4; |
||||||
|
} |
||||||
|
|
||||||
|
// Process the tail of input data
|
||||||
|
input_tail = (size_t)(ctx->input_end - ctx->input_cur); |
||||||
|
if (input_tail == 4) { |
||||||
|
// Process the input data with pad chars
|
||||||
|
if (ctx->input_cur[3] == '=') { |
||||||
|
if (ctx->input_cur[2] == '=' && ctx->output_end >= ctx->output_cur + 1) { |
||||||
|
if (!input_is_valid(ctx->input_cur, 2)) return false; |
||||||
|
*(ctx->output_cur++) = COMPOSE_OUTPUT_BYTE_0(ctx->input_cur); |
||||||
|
ctx->input_cur += 4; |
||||||
|
} else if (ctx->output_end >= ctx->output_cur + 2) { |
||||||
|
if (!input_is_valid(ctx->input_cur, 3)) return false; |
||||||
|
*(ctx->output_cur++) = COMPOSE_OUTPUT_BYTE_0(ctx->input_cur); |
||||||
|
*(ctx->output_cur++) = COMPOSE_OUTPUT_BYTE_1(ctx->input_cur); |
||||||
|
; |
||||||
|
ctx->input_cur += 4; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} else if (ctx->contains_tail && input_tail > 1) { |
||||||
|
// Process the input data without pad chars, but constains_tail is set
|
||||||
|
if (ctx->output_end >= ctx->output_cur + tail_xtra[input_tail]) { |
||||||
|
if (!input_is_valid(ctx->input_cur, input_tail)) return false; |
||||||
|
switch (input_tail) { |
||||||
|
case 3: |
||||||
|
ctx->output_cur[1] = COMPOSE_OUTPUT_BYTE_1(ctx->input_cur); |
||||||
|
case 2: |
||||||
|
ctx->output_cur[0] = COMPOSE_OUTPUT_BYTE_0(ctx->input_cur); |
||||||
|
} |
||||||
|
ctx->output_cur += tail_xtra[input_tail]; |
||||||
|
ctx->input_cur += input_tail; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
gpr_slice grpc_chttp2_base64_decode(gpr_slice input) { |
||||||
|
size_t input_length = GPR_SLICE_LENGTH(input); |
||||||
|
size_t output_length = input_length / 4 * 3; |
||||||
|
struct grpc_base64_decode_context ctx; |
||||||
|
gpr_slice output; |
||||||
|
|
||||||
|
if (input_length % 4 != 0) { |
||||||
|
gpr_log(GPR_ERROR, |
||||||
|
"Base64 decoding failed, input of " |
||||||
|
"grpc_chttp2_base64_decode has a length of %d, which is not a " |
||||||
|
"multiple of 4.\n", |
||||||
|
(int)input_length); |
||||||
|
return gpr_empty_slice(); |
||||||
|
} |
||||||
|
|
||||||
|
if (input_length > 0) { |
||||||
|
uint8_t *input_end = GPR_SLICE_END_PTR(input); |
||||||
|
if (*(--input_end) == '=') { |
||||||
|
output_length--; |
||||||
|
if (*(--input_end) == '=') { |
||||||
|
output_length--; |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
output = gpr_slice_malloc(output_length); |
||||||
|
|
||||||
|
ctx.input_cur = GPR_SLICE_START_PTR(input); |
||||||
|
ctx.input_end = GPR_SLICE_END_PTR(input); |
||||||
|
ctx.output_cur = GPR_SLICE_START_PTR(output); |
||||||
|
ctx.output_end = GPR_SLICE_END_PTR(output); |
||||||
|
ctx.contains_tail = false; |
||||||
|
|
||||||
|
if (!grpc_base64_decode_partial(&ctx)) { |
||||||
|
char *s = gpr_dump_slice(input, GPR_DUMP_ASCII); |
||||||
|
gpr_log(GPR_ERROR, "Base64 decoding failed, input string:\n%s\n", s); |
||||||
|
gpr_free(s); |
||||||
|
gpr_slice_unref(output); |
||||||
|
return gpr_empty_slice(); |
||||||
|
} |
||||||
|
GPR_ASSERT(ctx.output_cur == GPR_SLICE_END_PTR(output)); |
||||||
|
GPR_ASSERT(ctx.input_cur == GPR_SLICE_END_PTR(input)); |
||||||
|
return output; |
||||||
|
} |
||||||
|
|
||||||
|
gpr_slice grpc_chttp2_base64_decode_with_length(gpr_slice input, |
||||||
|
size_t output_length) { |
||||||
|
size_t input_length = GPR_SLICE_LENGTH(input); |
||||||
|
gpr_slice output = gpr_slice_malloc(output_length); |
||||||
|
struct grpc_base64_decode_context ctx; |
||||||
|
|
||||||
|
// The length of a base64 string cannot be 4 * n + 1
|
||||||
|
if (input_length % 4 == 1) { |
||||||
|
gpr_log(GPR_ERROR, |
||||||
|
"Base64 decoding failed, input of " |
||||||
|
"grpc_chttp2_base64_decode_with_length has a length of %d, which " |
||||||
|
"has a tail of 1 byte.\n", |
||||||
|
(int)input_length); |
||||||
|
gpr_slice_unref(output); |
||||||
|
return gpr_empty_slice(); |
||||||
|
} |
||||||
|
|
||||||
|
if (output_length > input_length / 4 * 3 + tail_xtra[input_length % 4]) { |
||||||
|
gpr_log(GPR_ERROR, |
||||||
|
"Base64 decoding failed, output_length %d is longer " |
||||||
|
"than the max possible output length %d.\n", |
||||||
|
(int)output_length, |
||||||
|
(int)(input_length / 4 * 3 + tail_xtra[input_length % 4])); |
||||||
|
gpr_slice_unref(output); |
||||||
|
return gpr_empty_slice(); |
||||||
|
} |
||||||
|
|
||||||
|
ctx.input_cur = GPR_SLICE_START_PTR(input); |
||||||
|
ctx.input_end = GPR_SLICE_END_PTR(input); |
||||||
|
ctx.output_cur = GPR_SLICE_START_PTR(output); |
||||||
|
ctx.output_end = GPR_SLICE_END_PTR(output); |
||||||
|
ctx.contains_tail = true; |
||||||
|
|
||||||
|
if (!grpc_base64_decode_partial(&ctx)) { |
||||||
|
char *s = gpr_dump_slice(input, GPR_DUMP_ASCII); |
||||||
|
gpr_log(GPR_ERROR, "Base64 decoding failed, input string:\n%s\n", s); |
||||||
|
gpr_free(s); |
||||||
|
gpr_slice_unref(output); |
||||||
|
return gpr_empty_slice(); |
||||||
|
} |
||||||
|
GPR_ASSERT(ctx.output_cur == GPR_SLICE_END_PTR(output)); |
||||||
|
GPR_ASSERT(ctx.input_cur <= GPR_SLICE_END_PTR(input)); |
||||||
|
return output; |
||||||
|
} |
||||||
@ -0,0 +1,66 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_BIN_DECODER_H |
||||||
|
#define GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_BIN_DECODER_H |
||||||
|
|
||||||
|
#include <grpc/support/slice.h> |
||||||
|
#include <stdbool.h> |
||||||
|
|
||||||
|
struct grpc_base64_decode_context { |
||||||
|
/* input/output: */ |
||||||
|
uint8_t *input_cur; |
||||||
|
uint8_t *input_end; |
||||||
|
uint8_t *output_cur; |
||||||
|
uint8_t *output_end; |
||||||
|
/* Indicate if the decoder should handle the tail of input data*/ |
||||||
|
bool contains_tail; |
||||||
|
}; |
||||||
|
|
||||||
|
/* base64 decode a grpc_base64_decode_context util either input_end is reached
|
||||||
|
or output_end is reached. When input_end is reached, (input_end - input_cur) |
||||||
|
is less than 4. When output_end is reached, (output_end - output_cur) is less |
||||||
|
than 3. Returns false if decoding is failed. */ |
||||||
|
bool grpc_base64_decode_partial(struct grpc_base64_decode_context *ctx); |
||||||
|
|
||||||
|
/* base64 decode a slice with pad chars. Returns a new slice, does not take
|
||||||
|
ownership of the input. Returns an empty slice if decoding is failed. */ |
||||||
|
gpr_slice grpc_chttp2_base64_decode(gpr_slice input); |
||||||
|
|
||||||
|
/* base64 decode a slice without pad chars, data length is needed. Returns a new
|
||||||
|
slice, does not take ownership of the input. Returns an empty slice if |
||||||
|
decoding is failed. */ |
||||||
|
gpr_slice grpc_chttp2_base64_decode_with_length(gpr_slice input, |
||||||
|
size_t output_length); |
||||||
|
|
||||||
|
#endif /* GRPC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_BIN_DECODER_H */ |
||||||
@ -0,0 +1,227 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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 <unordered_set> |
||||||
|
#include <vector> |
||||||
|
|
||||||
|
#include <grpc++/grpc++.h> |
||||||
|
|
||||||
|
#include "src/cpp/ext/proto_server_reflection.h" |
||||||
|
|
||||||
|
using grpc::Status; |
||||||
|
using grpc::StatusCode; |
||||||
|
using grpc::reflection::v1alpha::ServerReflectionRequest; |
||||||
|
using grpc::reflection::v1alpha::ExtensionRequest; |
||||||
|
using grpc::reflection::v1alpha::ServerReflectionResponse; |
||||||
|
using grpc::reflection::v1alpha::ListServiceResponse; |
||||||
|
using grpc::reflection::v1alpha::ServiceResponse; |
||||||
|
using grpc::reflection::v1alpha::ExtensionNumberResponse; |
||||||
|
using grpc::reflection::v1alpha::ErrorResponse; |
||||||
|
using grpc::reflection::v1alpha::FileDescriptorResponse; |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
ProtoServerReflection::ProtoServerReflection() |
||||||
|
: descriptor_pool_(protobuf::DescriptorPool::generated_pool()) {} |
||||||
|
|
||||||
|
void ProtoServerReflection::SetServiceList( |
||||||
|
const std::vector<grpc::string>* services) { |
||||||
|
services_ = services; |
||||||
|
} |
||||||
|
|
||||||
|
Status ProtoServerReflection::ServerReflectionInfo( |
||||||
|
ServerContext* context, |
||||||
|
ServerReaderWriter<ServerReflectionResponse, ServerReflectionRequest>* |
||||||
|
stream) { |
||||||
|
ServerReflectionRequest request; |
||||||
|
ServerReflectionResponse response; |
||||||
|
Status status; |
||||||
|
while (stream->Read(&request)) { |
||||||
|
switch (request.message_request_case()) { |
||||||
|
case ServerReflectionRequest::MessageRequestCase::kFileByFilename: |
||||||
|
status = GetFileByName(context, request.file_by_filename(), &response); |
||||||
|
break; |
||||||
|
case ServerReflectionRequest::MessageRequestCase::kFileContainingSymbol: |
||||||
|
status = GetFileContainingSymbol( |
||||||
|
context, request.file_containing_symbol(), &response); |
||||||
|
break; |
||||||
|
case ServerReflectionRequest::MessageRequestCase:: |
||||||
|
kFileContainingExtension: |
||||||
|
status = GetFileContainingExtension( |
||||||
|
context, &request.file_containing_extension(), &response); |
||||||
|
break; |
||||||
|
case ServerReflectionRequest::MessageRequestCase:: |
||||||
|
kAllExtensionNumbersOfType: |
||||||
|
status = GetAllExtensionNumbers( |
||||||
|
context, request.all_extension_numbers_of_type(), |
||||||
|
response.mutable_all_extension_numbers_response()); |
||||||
|
break; |
||||||
|
case ServerReflectionRequest::MessageRequestCase::kListServices: |
||||||
|
status = |
||||||
|
ListService(context, response.mutable_list_services_response()); |
||||||
|
break; |
||||||
|
default: |
||||||
|
status = Status(StatusCode::UNIMPLEMENTED, ""); |
||||||
|
} |
||||||
|
|
||||||
|
if (!status.ok()) { |
||||||
|
FillErrorResponse(status, response.mutable_error_response()); |
||||||
|
} |
||||||
|
response.set_valid_host(request.host()); |
||||||
|
response.set_allocated_original_request( |
||||||
|
new ServerReflectionRequest(request)); |
||||||
|
stream->Write(response); |
||||||
|
} |
||||||
|
|
||||||
|
return Status::OK; |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoServerReflection::FillErrorResponse(const Status& status, |
||||||
|
ErrorResponse* error_response) { |
||||||
|
error_response->set_error_code(status.error_code()); |
||||||
|
error_response->set_error_message(status.error_message()); |
||||||
|
} |
||||||
|
|
||||||
|
Status ProtoServerReflection::ListService(ServerContext* context, |
||||||
|
ListServiceResponse* response) { |
||||||
|
if (services_ == nullptr) { |
||||||
|
return Status(StatusCode::NOT_FOUND, "Services not found."); |
||||||
|
} |
||||||
|
for (auto it = services_->begin(); it != services_->end(); ++it) { |
||||||
|
ServiceResponse* service_response = response->add_service(); |
||||||
|
service_response->set_name(*it); |
||||||
|
} |
||||||
|
return Status::OK; |
||||||
|
} |
||||||
|
|
||||||
|
Status ProtoServerReflection::GetFileByName( |
||||||
|
ServerContext* context, const grpc::string& filename, |
||||||
|
ServerReflectionResponse* response) { |
||||||
|
if (descriptor_pool_ == nullptr) { |
||||||
|
return Status::CANCELLED; |
||||||
|
} |
||||||
|
|
||||||
|
const protobuf::FileDescriptor* file_desc = |
||||||
|
descriptor_pool_->FindFileByName(filename); |
||||||
|
if (file_desc == nullptr) { |
||||||
|
return Status(StatusCode::NOT_FOUND, "File not found."); |
||||||
|
} |
||||||
|
std::unordered_set<grpc::string> seen_files; |
||||||
|
FillFileDescriptorResponse(file_desc, response, &seen_files); |
||||||
|
return Status::OK; |
||||||
|
} |
||||||
|
|
||||||
|
Status ProtoServerReflection::GetFileContainingSymbol( |
||||||
|
ServerContext* context, const grpc::string& symbol, |
||||||
|
ServerReflectionResponse* response) { |
||||||
|
if (descriptor_pool_ == nullptr) { |
||||||
|
return Status::CANCELLED; |
||||||
|
} |
||||||
|
|
||||||
|
const protobuf::FileDescriptor* file_desc = |
||||||
|
descriptor_pool_->FindFileContainingSymbol(symbol); |
||||||
|
if (file_desc == nullptr) { |
||||||
|
return Status(StatusCode::NOT_FOUND, "Symbol not found."); |
||||||
|
} |
||||||
|
std::unordered_set<grpc::string> seen_files; |
||||||
|
FillFileDescriptorResponse(file_desc, response, &seen_files); |
||||||
|
return Status::OK; |
||||||
|
} |
||||||
|
|
||||||
|
Status ProtoServerReflection::GetFileContainingExtension( |
||||||
|
ServerContext* context, const ExtensionRequest* request, |
||||||
|
ServerReflectionResponse* response) { |
||||||
|
if (descriptor_pool_ == nullptr) { |
||||||
|
return Status::CANCELLED; |
||||||
|
} |
||||||
|
|
||||||
|
const protobuf::Descriptor* desc = |
||||||
|
descriptor_pool_->FindMessageTypeByName(request->containing_type()); |
||||||
|
if (desc == nullptr) { |
||||||
|
return Status(StatusCode::NOT_FOUND, "Type not found."); |
||||||
|
} |
||||||
|
|
||||||
|
const protobuf::FieldDescriptor* field_desc = |
||||||
|
descriptor_pool_->FindExtensionByNumber(desc, |
||||||
|
request->extension_number()); |
||||||
|
if (field_desc == nullptr) { |
||||||
|
return Status(StatusCode::NOT_FOUND, "Extension not found."); |
||||||
|
} |
||||||
|
std::unordered_set<grpc::string> seen_files; |
||||||
|
FillFileDescriptorResponse(field_desc->file(), response, &seen_files); |
||||||
|
return Status::OK; |
||||||
|
} |
||||||
|
|
||||||
|
Status ProtoServerReflection::GetAllExtensionNumbers( |
||||||
|
ServerContext* context, const grpc::string& type, |
||||||
|
ExtensionNumberResponse* response) { |
||||||
|
if (descriptor_pool_ == nullptr) { |
||||||
|
return Status::CANCELLED; |
||||||
|
} |
||||||
|
|
||||||
|
const protobuf::Descriptor* desc = |
||||||
|
descriptor_pool_->FindMessageTypeByName(type); |
||||||
|
if (desc == nullptr) { |
||||||
|
return Status(StatusCode::NOT_FOUND, "Type not found."); |
||||||
|
} |
||||||
|
|
||||||
|
std::vector<const protobuf::FieldDescriptor*> extensions; |
||||||
|
descriptor_pool_->FindAllExtensions(desc, &extensions); |
||||||
|
for (auto extension : extensions) { |
||||||
|
response->add_extension_number(extension->number()); |
||||||
|
} |
||||||
|
response->set_base_type_name(type); |
||||||
|
return Status::OK; |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoServerReflection::FillFileDescriptorResponse( |
||||||
|
const protobuf::FileDescriptor* file_desc, |
||||||
|
ServerReflectionResponse* response, |
||||||
|
std::unordered_set<grpc::string>* seen_files) { |
||||||
|
if (seen_files->find(file_desc->name()) != seen_files->end()) { |
||||||
|
return; |
||||||
|
} |
||||||
|
seen_files->insert(file_desc->name()); |
||||||
|
|
||||||
|
protobuf::FileDescriptorProto file_desc_proto; |
||||||
|
grpc::string data; |
||||||
|
file_desc->CopyTo(&file_desc_proto); |
||||||
|
file_desc_proto.SerializeToString(&data); |
||||||
|
response->mutable_file_descriptor_response()->add_file_descriptor_proto(data); |
||||||
|
|
||||||
|
for (int i = 0; i < file_desc->dependency_count(); ++i) { |
||||||
|
FillFileDescriptorResponse(file_desc->dependency(i), response, seen_files); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
@ -0,0 +1,94 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* 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 GRPC_INTERNAL_CPP_EXT_PROTO_SERVER_REFLECTION_H |
||||||
|
#define GRPC_INTERNAL_CPP_EXT_PROTO_SERVER_REFLECTION_H |
||||||
|
|
||||||
|
#include <unordered_set> |
||||||
|
#include <vector> |
||||||
|
|
||||||
|
#include <grpc++/ext/reflection.grpc.pb.h> |
||||||
|
#include <grpc++/grpc++.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
class ProtoServerReflection GRPC_FINAL |
||||||
|
: public reflection::v1alpha::ServerReflection::Service { |
||||||
|
public: |
||||||
|
ProtoServerReflection(); |
||||||
|
|
||||||
|
// Add the full names of registered services
|
||||||
|
void SetServiceList(const std::vector<grpc::string>* services); |
||||||
|
|
||||||
|
// implementation of ServerReflectionInfo(stream ServerReflectionRequest) rpc
|
||||||
|
// in ServerReflection service
|
||||||
|
Status ServerReflectionInfo( |
||||||
|
ServerContext* context, |
||||||
|
ServerReaderWriter<reflection::v1alpha::ServerReflectionResponse, |
||||||
|
reflection::v1alpha::ServerReflectionRequest>* stream) |
||||||
|
GRPC_OVERRIDE; |
||||||
|
|
||||||
|
private: |
||||||
|
Status ListService(ServerContext* context, |
||||||
|
reflection::v1alpha::ListServiceResponse* response); |
||||||
|
|
||||||
|
Status GetFileByName(ServerContext* context, const grpc::string& file_name, |
||||||
|
reflection::v1alpha::ServerReflectionResponse* response); |
||||||
|
|
||||||
|
Status GetFileContainingSymbol( |
||||||
|
ServerContext* context, const grpc::string& symbol, |
||||||
|
reflection::v1alpha::ServerReflectionResponse* response); |
||||||
|
|
||||||
|
Status GetFileContainingExtension( |
||||||
|
ServerContext* context, |
||||||
|
const reflection::v1alpha::ExtensionRequest* request, |
||||||
|
reflection::v1alpha::ServerReflectionResponse* response); |
||||||
|
|
||||||
|
Status GetAllExtensionNumbers( |
||||||
|
ServerContext* context, const grpc::string& type, |
||||||
|
reflection::v1alpha::ExtensionNumberResponse* response); |
||||||
|
|
||||||
|
void FillFileDescriptorResponse( |
||||||
|
const protobuf::FileDescriptor* file_desc, |
||||||
|
reflection::v1alpha::ServerReflectionResponse* response, |
||||||
|
std::unordered_set<grpc::string>* seen_files); |
||||||
|
|
||||||
|
void FillErrorResponse(const Status& status, |
||||||
|
reflection::v1alpha::ErrorResponse* error_response); |
||||||
|
|
||||||
|
const protobuf::DescriptorPool* descriptor_pool_; |
||||||
|
const std::vector<string>* services_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_INTERNAL_CPP_EXT_PROTO_SERVER_REFLECTION_H
|
||||||
@ -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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <grpc++/ext/proto_server_reflection_plugin.h> |
||||||
|
#include <grpc++/impl/server_builder_plugin.h> |
||||||
|
#include <grpc++/impl/server_initializer.h> |
||||||
|
#include <grpc++/server.h> |
||||||
|
|
||||||
|
#include "src/cpp/ext/proto_server_reflection.h" |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace reflection { |
||||||
|
|
||||||
|
ProtoServerReflectionPlugin::ProtoServerReflectionPlugin() |
||||||
|
: reflection_service_(new grpc::ProtoServerReflection()) {} |
||||||
|
|
||||||
|
grpc::string ProtoServerReflectionPlugin::name() { |
||||||
|
return "proto_server_reflection"; |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoServerReflectionPlugin::InitServer(grpc::ServerInitializer* si) { |
||||||
|
si->RegisterService(reflection_service_); |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoServerReflectionPlugin::Finish(grpc::ServerInitializer* si) { |
||||||
|
reflection_service_->SetServiceList(si->GetServiceList()); |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoServerReflectionPlugin::ChangeArguments(const grpc::string& name, |
||||||
|
void* value) {} |
||||||
|
|
||||||
|
bool ProtoServerReflectionPlugin::has_sync_methods() const { |
||||||
|
if (reflection_service_ != nullptr) { |
||||||
|
return reflection_service_->has_synchronous_methods(); |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
bool ProtoServerReflectionPlugin::has_async_methods() const { |
||||||
|
if (reflection_service_ != nullptr) { |
||||||
|
return reflection_service_->has_async_methods(); |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
static std::unique_ptr<::grpc::ServerBuilderPlugin> CreateProtoReflection() { |
||||||
|
return std::unique_ptr<::grpc::ServerBuilderPlugin>( |
||||||
|
new ProtoServerReflectionPlugin()); |
||||||
|
} |
||||||
|
|
||||||
|
void InitProtoReflectionServerBuilderPlugin() { |
||||||
|
static bool already_here = false; |
||||||
|
if (already_here) return; |
||||||
|
already_here = true; |
||||||
|
::grpc::ServerBuilder::InternalAddPluginFactory(&CreateProtoReflection); |
||||||
|
} |
||||||
|
|
||||||
|
// Force InitProtoReflectionServerBuilderPlugin() to be called at static
|
||||||
|
// initialization time.
|
||||||
|
struct StaticProtoReflectionPluginInitializer { |
||||||
|
StaticProtoReflectionPluginInitializer() { |
||||||
|
InitProtoReflectionServerBuilderPlugin(); |
||||||
|
} |
||||||
|
} static_proto_reflection_plugin_initializer; |
||||||
|
|
||||||
|
} // namespace reflection
|
||||||
|
} // namespace grpc
|
||||||
@ -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. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
|
||||||
|
// Generated by the gRPC protobuf plugin.
|
||||||
|
// If you make any local change, they will be lost.
|
||||||
|
// source: reflection.proto
|
||||||
|
|
||||||
|
#include <grpc++/ext/reflection.pb.h> |
||||||
|
#include <grpc++/ext/reflection.grpc.pb.h> |
||||||
|
|
||||||
|
#include <grpc++/impl/codegen/async_stream.h> |
||||||
|
#include <grpc++/impl/codegen/async_unary_call.h> |
||||||
|
#include <grpc++/impl/codegen/channel_interface.h> |
||||||
|
#include <grpc++/impl/codegen/client_unary_call.h> |
||||||
|
#include <grpc++/impl/codegen/method_handler_impl.h> |
||||||
|
#include <grpc++/impl/codegen/rpc_service_method.h> |
||||||
|
#include <grpc++/impl/codegen/service_type.h> |
||||||
|
#include <grpc++/impl/codegen/sync_stream.h> |
||||||
|
namespace grpc { |
||||||
|
namespace reflection { |
||||||
|
namespace v1alpha { |
||||||
|
|
||||||
|
static const char* ServerReflection_method_names[] = { |
||||||
|
"/grpc.reflection.v1alpha.ServerReflection/ServerReflectionInfo", |
||||||
|
}; |
||||||
|
|
||||||
|
std::unique_ptr< ServerReflection::Stub> ServerReflection::NewStub(const std::shared_ptr< ::grpc::ChannelInterface>& channel, const ::grpc::StubOptions& options) { |
||||||
|
std::unique_ptr< ServerReflection::Stub> stub(new ServerReflection::Stub(channel)); |
||||||
|
return stub; |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflection::Stub::Stub(const std::shared_ptr< ::grpc::ChannelInterface>& channel) |
||||||
|
: channel_(channel), rpcmethod_ServerReflectionInfo_(ServerReflection_method_names[0], ::grpc::RpcMethod::BIDI_STREAMING, channel) |
||||||
|
{} |
||||||
|
|
||||||
|
::grpc::ClientReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>* ServerReflection::Stub::ServerReflectionInfoRaw(::grpc::ClientContext* context) { |
||||||
|
return new ::grpc::ClientReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>(channel_.get(), rpcmethod_ServerReflectionInfo_, context); |
||||||
|
} |
||||||
|
|
||||||
|
::grpc::ClientAsyncReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>* ServerReflection::Stub::AsyncServerReflectionInfoRaw(::grpc::ClientContext* context, ::grpc::CompletionQueue* cq, void* tag) { |
||||||
|
return new ::grpc::ClientAsyncReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>(channel_.get(), cq, rpcmethod_ServerReflectionInfo_, context, tag); |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflection::Service::Service() { |
||||||
|
(void)ServerReflection_method_names; |
||||||
|
AddMethod(new ::grpc::RpcServiceMethod( |
||||||
|
ServerReflection_method_names[0], |
||||||
|
::grpc::RpcMethod::BIDI_STREAMING, |
||||||
|
new ::grpc::BidiStreamingHandler< ServerReflection::Service, ::grpc::reflection::v1alpha::ServerReflectionRequest, ::grpc::reflection::v1alpha::ServerReflectionResponse>( |
||||||
|
std::mem_fn(&ServerReflection::Service::ServerReflectionInfo), this))); |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflection::Service::~Service() { |
||||||
|
} |
||||||
|
|
||||||
|
::grpc::Status ServerReflection::Service::ServerReflectionInfo(::grpc::ServerContext* context, ::grpc::ServerReaderWriter< ::grpc::reflection::v1alpha::ServerReflectionResponse, ::grpc::reflection::v1alpha::ServerReflectionRequest>* stream) { |
||||||
|
(void) context; |
||||||
|
(void) stream; |
||||||
|
return ::grpc::Status(::grpc::StatusCode::UNIMPLEMENTED, ""); |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
|
} // namespace reflection
|
||||||
|
} // namespace v1alpha
|
||||||
|
|
||||||
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,59 @@ |
|||||||
|
#region Copyright notice and license |
||||||
|
|
||||||
|
// 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. |
||||||
|
|
||||||
|
#endregion |
||||||
|
|
||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
|
||||||
|
namespace Grpc.Core.Logging |
||||||
|
{ |
||||||
|
/// <summary>Standard logging levels.</summary> |
||||||
|
public enum LogLevel |
||||||
|
{ |
||||||
|
/// <summary> |
||||||
|
/// Debug severity. |
||||||
|
/// </summary> |
||||||
|
Debug = 0, |
||||||
|
/// <summary> |
||||||
|
/// Info severity. |
||||||
|
/// </summary> |
||||||
|
Info, |
||||||
|
/// <summary> |
||||||
|
/// Warning severity. |
||||||
|
/// </summary> |
||||||
|
Warning, |
||||||
|
/// <summary> |
||||||
|
/// Error severity. |
||||||
|
/// </summary> |
||||||
|
Error |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,160 @@ |
|||||||
|
#region Copyright notice and license |
||||||
|
|
||||||
|
// 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. |
||||||
|
|
||||||
|
#endregion |
||||||
|
|
||||||
|
using System; |
||||||
|
using System.Collections.Generic; |
||||||
|
using System.Globalization; |
||||||
|
using System.IO; |
||||||
|
using Grpc.Core.Utils; |
||||||
|
|
||||||
|
namespace Grpc.Core.Logging |
||||||
|
{ |
||||||
|
/// <summary>Logger that filters out messages below certain log level.</summary> |
||||||
|
public class LogLevelFilterLogger : ILogger |
||||||
|
{ |
||||||
|
readonly ILogger innerLogger; |
||||||
|
readonly LogLevel logLevel; |
||||||
|
|
||||||
|
/// <summary> |
||||||
|
/// Creates and instance of <c>LogLevelFilter.</c> |
||||||
|
/// </summary> |
||||||
|
public LogLevelFilterLogger(ILogger logger, LogLevel logLevel) |
||||||
|
{ |
||||||
|
this.innerLogger = GrpcPreconditions.CheckNotNull(logger); |
||||||
|
this.logLevel = logLevel; |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary> |
||||||
|
/// Returns a logger associated with the specified type. |
||||||
|
/// </summary> |
||||||
|
public virtual ILogger ForType<T>() |
||||||
|
{ |
||||||
|
var newInnerLogger = innerLogger.ForType<T>(); |
||||||
|
if (object.ReferenceEquals(this.innerLogger, newInnerLogger)) |
||||||
|
{ |
||||||
|
return this; |
||||||
|
} |
||||||
|
return new LogLevelFilterLogger(newInnerLogger, logLevel); |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a message with severity Debug.</summary> |
||||||
|
public void Debug(string message) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Debug) |
||||||
|
{ |
||||||
|
innerLogger.Debug(message); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a formatted message with severity Debug.</summary> |
||||||
|
public void Debug(string format, params object[] formatArgs) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Debug) |
||||||
|
{ |
||||||
|
innerLogger.Debug(format, formatArgs); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a message with severity Info.</summary> |
||||||
|
public void Info(string message) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Info) |
||||||
|
{ |
||||||
|
innerLogger.Info(message); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a formatted message with severity Info.</summary> |
||||||
|
public void Info(string format, params object[] formatArgs) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Info) |
||||||
|
{ |
||||||
|
innerLogger.Info(format, formatArgs); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a message with severity Warning.</summary> |
||||||
|
public void Warning(string message) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Warning) |
||||||
|
{ |
||||||
|
innerLogger.Warning(message); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a formatted message with severity Warning.</summary> |
||||||
|
public void Warning(string format, params object[] formatArgs) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Warning) |
||||||
|
{ |
||||||
|
innerLogger.Warning(format, formatArgs); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a message and an associated exception with severity Warning.</summary> |
||||||
|
public void Warning(Exception exception, string message) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Warning) |
||||||
|
{ |
||||||
|
innerLogger.Warning(exception, message); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a message with severity Error.</summary> |
||||||
|
public void Error(string message) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Error) |
||||||
|
{ |
||||||
|
innerLogger.Error(message); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a formatted message with severity Error.</summary> |
||||||
|
public void Error(string format, params object[] formatArgs) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Error) |
||||||
|
{ |
||||||
|
innerLogger.Error(format, formatArgs); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
/// <summary>Logs a message and an associated exception with severity Error.</summary> |
||||||
|
public void Error(Exception exception, string message) |
||||||
|
{ |
||||||
|
if (logLevel <= LogLevel.Error) |
||||||
|
{ |
||||||
|
innerLogger.Error(exception, message); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
@ -0,0 +1,583 @@ |
|||||||
|
# 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. |
||||||
|
|
||||||
|
import collections |
||||||
|
from concurrent import futures |
||||||
|
import contextlib |
||||||
|
import distutils.spawn |
||||||
|
import errno |
||||||
|
import os |
||||||
|
import shutil |
||||||
|
import subprocess |
||||||
|
import sys |
||||||
|
import tempfile |
||||||
|
import threading |
||||||
|
import unittest |
||||||
|
|
||||||
|
from six import moves |
||||||
|
|
||||||
|
import grpc |
||||||
|
from tests.unit.framework.common import test_constants |
||||||
|
|
||||||
|
# Identifiers of entities we expect to find in the generated module. |
||||||
|
STUB_IDENTIFIER = 'TestServiceStub' |
||||||
|
SERVICER_IDENTIFIER = 'TestServiceServicer' |
||||||
|
ADD_SERVICER_TO_SERVER_IDENTIFIER = 'add_TestServiceServicer_to_server' |
||||||
|
|
||||||
|
|
||||||
|
class _ServicerMethods(object): |
||||||
|
|
||||||
|
def __init__(self, response_pb2, payload_pb2): |
||||||
|
self._condition = threading.Condition() |
||||||
|
self._paused = False |
||||||
|
self._fail = False |
||||||
|
self._response_pb2 = response_pb2 |
||||||
|
self._payload_pb2 = payload_pb2 |
||||||
|
|
||||||
|
@contextlib.contextmanager |
||||||
|
def pause(self): # pylint: disable=invalid-name |
||||||
|
with self._condition: |
||||||
|
self._paused = True |
||||||
|
yield |
||||||
|
with self._condition: |
||||||
|
self._paused = False |
||||||
|
self._condition.notify_all() |
||||||
|
|
||||||
|
@contextlib.contextmanager |
||||||
|
def fail(self): # pylint: disable=invalid-name |
||||||
|
with self._condition: |
||||||
|
self._fail = True |
||||||
|
yield |
||||||
|
with self._condition: |
||||||
|
self._fail = False |
||||||
|
|
||||||
|
def _control(self): # pylint: disable=invalid-name |
||||||
|
with self._condition: |
||||||
|
if self._fail: |
||||||
|
raise ValueError() |
||||||
|
while self._paused: |
||||||
|
self._condition.wait() |
||||||
|
|
||||||
|
def UnaryCall(self, request, unused_rpc_context): |
||||||
|
response = self._response_pb2.SimpleResponse() |
||||||
|
response.payload.payload_type = self._payload_pb2.COMPRESSABLE |
||||||
|
response.payload.payload_compressable = 'a' * request.response_size |
||||||
|
self._control() |
||||||
|
return response |
||||||
|
|
||||||
|
def StreamingOutputCall(self, request, unused_rpc_context): |
||||||
|
for parameter in request.response_parameters: |
||||||
|
response = self._response_pb2.StreamingOutputCallResponse() |
||||||
|
response.payload.payload_type = self._payload_pb2.COMPRESSABLE |
||||||
|
response.payload.payload_compressable = 'a' * parameter.size |
||||||
|
self._control() |
||||||
|
yield response |
||||||
|
|
||||||
|
def StreamingInputCall(self, request_iter, unused_rpc_context): |
||||||
|
response = self._response_pb2.StreamingInputCallResponse() |
||||||
|
aggregated_payload_size = 0 |
||||||
|
for request in request_iter: |
||||||
|
aggregated_payload_size += len(request.payload.payload_compressable) |
||||||
|
response.aggregated_payload_size = aggregated_payload_size |
||||||
|
self._control() |
||||||
|
return response |
||||||
|
|
||||||
|
def FullDuplexCall(self, request_iter, unused_rpc_context): |
||||||
|
for request in request_iter: |
||||||
|
for parameter in request.response_parameters: |
||||||
|
response = self._response_pb2.StreamingOutputCallResponse() |
||||||
|
response.payload.payload_type = self._payload_pb2.COMPRESSABLE |
||||||
|
response.payload.payload_compressable = 'a' * parameter.size |
||||||
|
self._control() |
||||||
|
yield response |
||||||
|
|
||||||
|
def HalfDuplexCall(self, request_iter, unused_rpc_context): |
||||||
|
responses = [] |
||||||
|
for request in request_iter: |
||||||
|
for parameter in request.response_parameters: |
||||||
|
response = self._response_pb2.StreamingOutputCallResponse() |
||||||
|
response.payload.payload_type = self._payload_pb2.COMPRESSABLE |
||||||
|
response.payload.payload_compressable = 'a' * parameter.size |
||||||
|
self._control() |
||||||
|
responses.append(response) |
||||||
|
for response in responses: |
||||||
|
yield response |
||||||
|
|
||||||
|
|
||||||
|
class _Service( |
||||||
|
collections.namedtuple( |
||||||
|
'_Service', ('servicer_methods', 'server', 'stub',))): |
||||||
|
"""A live and running service. |
||||||
|
|
||||||
|
Attributes: |
||||||
|
servicer_methods: The _ServicerMethods servicing RPCs. |
||||||
|
server: The grpc.Server servicing RPCs. |
||||||
|
stub: A stub on which to invoke RPCs. |
||||||
|
""" |
||||||
|
|
||||||
|
|
||||||
|
def _CreateService(service_pb2, response_pb2, payload_pb2): |
||||||
|
"""Provides a servicer backend and a stub. |
||||||
|
|
||||||
|
Args: |
||||||
|
service_pb2: The service_pb2 module generated by this test. |
||||||
|
response_pb2: The response_pb2 module generated by this test. |
||||||
|
payload_pb2: The payload_pb2 module generated by this test. |
||||||
|
|
||||||
|
Returns: |
||||||
|
A _Service with which to test RPCs. |
||||||
|
""" |
||||||
|
servicer_methods = _ServicerMethods(response_pb2, payload_pb2) |
||||||
|
|
||||||
|
class Servicer(getattr(service_pb2, SERVICER_IDENTIFIER)): |
||||||
|
|
||||||
|
def UnaryCall(self, request, context): |
||||||
|
return servicer_methods.UnaryCall(request, context) |
||||||
|
|
||||||
|
def StreamingOutputCall(self, request, context): |
||||||
|
return servicer_methods.StreamingOutputCall(request, context) |
||||||
|
|
||||||
|
def StreamingInputCall(self, request_iter, context): |
||||||
|
return servicer_methods.StreamingInputCall(request_iter, context) |
||||||
|
|
||||||
|
def FullDuplexCall(self, request_iter, context): |
||||||
|
return servicer_methods.FullDuplexCall(request_iter, context) |
||||||
|
|
||||||
|
def HalfDuplexCall(self, request_iter, context): |
||||||
|
return servicer_methods.HalfDuplexCall(request_iter, context) |
||||||
|
|
||||||
|
server = grpc.server( |
||||||
|
(), futures.ThreadPoolExecutor(max_workers=test_constants.POOL_SIZE)) |
||||||
|
getattr(service_pb2, ADD_SERVICER_TO_SERVER_IDENTIFIER)(Servicer(), server) |
||||||
|
port = server.add_insecure_port('[::]:0') |
||||||
|
server.start() |
||||||
|
channel = grpc.insecure_channel('localhost:{}'.format(port)) |
||||||
|
stub = getattr(service_pb2, STUB_IDENTIFIER)(channel) |
||||||
|
return _Service(servicer_methods, server, stub) |
||||||
|
|
||||||
|
|
||||||
|
def _CreateIncompleteService(service_pb2): |
||||||
|
"""Provides a servicer backend that fails to implement methods and its stub. |
||||||
|
|
||||||
|
Args: |
||||||
|
service_pb2: The service_pb2 module generated by this test. |
||||||
|
|
||||||
|
Returns: |
||||||
|
A _Service with which to test RPCs. The returned _Service's |
||||||
|
servicer_methods implements none of the methods required of it. |
||||||
|
""" |
||||||
|
|
||||||
|
class Servicer(getattr(service_pb2, SERVICER_IDENTIFIER)): |
||||||
|
pass |
||||||
|
|
||||||
|
server = grpc.server( |
||||||
|
(), futures.ThreadPoolExecutor(max_workers=test_constants.POOL_SIZE)) |
||||||
|
getattr(service_pb2, ADD_SERVICER_TO_SERVER_IDENTIFIER)(Servicer(), server) |
||||||
|
port = server.add_insecure_port('[::]:0') |
||||||
|
server.start() |
||||||
|
channel = grpc.insecure_channel('localhost:{}'.format(port)) |
||||||
|
stub = getattr(service_pb2, STUB_IDENTIFIER)(channel) |
||||||
|
return _Service(None, server, stub) |
||||||
|
|
||||||
|
|
||||||
|
def _streaming_input_request_iterator(request_pb2, payload_pb2): |
||||||
|
for _ in range(3): |
||||||
|
request = request_pb2.StreamingInputCallRequest() |
||||||
|
request.payload.payload_type = payload_pb2.COMPRESSABLE |
||||||
|
request.payload.payload_compressable = 'a' |
||||||
|
yield request |
||||||
|
|
||||||
|
|
||||||
|
def _streaming_output_request(request_pb2): |
||||||
|
request = request_pb2.StreamingOutputCallRequest() |
||||||
|
sizes = [1, 2, 3] |
||||||
|
request.response_parameters.add(size=sizes[0], interval_us=0) |
||||||
|
request.response_parameters.add(size=sizes[1], interval_us=0) |
||||||
|
request.response_parameters.add(size=sizes[2], interval_us=0) |
||||||
|
return request |
||||||
|
|
||||||
|
|
||||||
|
def _full_duplex_request_iterator(request_pb2): |
||||||
|
request = request_pb2.StreamingOutputCallRequest() |
||||||
|
request.response_parameters.add(size=1, interval_us=0) |
||||||
|
yield request |
||||||
|
request = request_pb2.StreamingOutputCallRequest() |
||||||
|
request.response_parameters.add(size=2, interval_us=0) |
||||||
|
request.response_parameters.add(size=3, interval_us=0) |
||||||
|
yield request |
||||||
|
|
||||||
|
|
||||||
|
class PythonPluginTest(unittest.TestCase): |
||||||
|
"""Test case for the gRPC Python protoc-plugin. |
||||||
|
|
||||||
|
While reading these tests, remember that the futures API |
||||||
|
(`stub.method.future()`) only gives futures for the *response-unary* |
||||||
|
methods and does not exist for response-streaming methods. |
||||||
|
""" |
||||||
|
|
||||||
|
def setUp(self): |
||||||
|
# Assume that the appropriate protoc and grpc_python_plugins are on the |
||||||
|
# path. |
||||||
|
protoc_command = 'protoc' |
||||||
|
protoc_plugin_filename = distutils.spawn.find_executable( |
||||||
|
'grpc_python_plugin') |
||||||
|
if not os.path.isfile(protoc_command): |
||||||
|
# Assume that if we haven't built protoc that it's on the system. |
||||||
|
protoc_command = 'protoc' |
||||||
|
|
||||||
|
# Ensure that the output directory exists. |
||||||
|
self.outdir = tempfile.mkdtemp() |
||||||
|
|
||||||
|
# Find all proto files |
||||||
|
paths = [] |
||||||
|
root_dir = os.path.dirname(os.path.realpath(__file__)) |
||||||
|
proto_dir = os.path.join(root_dir, 'protos') |
||||||
|
for walk_root, _, filenames in os.walk(proto_dir): |
||||||
|
for filename in filenames: |
||||||
|
if filename.endswith('.proto'): |
||||||
|
path = os.path.join(walk_root, filename) |
||||||
|
paths.append(path) |
||||||
|
|
||||||
|
# Invoke protoc with the plugin. |
||||||
|
cmd = [ |
||||||
|
protoc_command, |
||||||
|
'--plugin=protoc-gen-python-grpc=%s' % protoc_plugin_filename, |
||||||
|
'-I %s' % root_dir, |
||||||
|
'--python_out=%s' % self.outdir, |
||||||
|
'--python-grpc_out=%s' % self.outdir |
||||||
|
] + paths |
||||||
|
subprocess.check_call(' '.join(cmd), shell=True, env=os.environ, |
||||||
|
cwd=os.path.dirname(os.path.realpath(__file__))) |
||||||
|
|
||||||
|
# Generated proto directories dont include __init__.py, but |
||||||
|
# these are needed for python package resolution |
||||||
|
for walk_root, _, _ in os.walk(os.path.join(self.outdir, 'protos')): |
||||||
|
path = os.path.join(walk_root, '__init__.py') |
||||||
|
open(path, 'a').close() |
||||||
|
|
||||||
|
sys.path.insert(0, self.outdir) |
||||||
|
|
||||||
|
import protos.payload.test_payload_pb2 as payload_pb2 |
||||||
|
import protos.requests.r.test_requests_pb2 as request_pb2 |
||||||
|
import protos.responses.test_responses_pb2 as response_pb2 |
||||||
|
import protos.service.test_service_pb2 as service_pb2 |
||||||
|
self._payload_pb2 = payload_pb2 |
||||||
|
self._request_pb2 = request_pb2 |
||||||
|
self._response_pb2 = response_pb2 |
||||||
|
self._service_pb2 = service_pb2 |
||||||
|
|
||||||
|
def tearDown(self): |
||||||
|
try: |
||||||
|
shutil.rmtree(self.outdir) |
||||||
|
except OSError as exc: |
||||||
|
if exc.errno != errno.ENOENT: |
||||||
|
raise |
||||||
|
sys.path.remove(self.outdir) |
||||||
|
|
||||||
|
def testImportAttributes(self): |
||||||
|
# check that we can access the generated module and its members. |
||||||
|
self.assertIsNotNone( |
||||||
|
getattr(self._service_pb2, STUB_IDENTIFIER, None)) |
||||||
|
self.assertIsNotNone( |
||||||
|
getattr(self._service_pb2, SERVICER_IDENTIFIER, None)) |
||||||
|
self.assertIsNotNone( |
||||||
|
getattr(self._service_pb2, ADD_SERVICER_TO_SERVER_IDENTIFIER, None)) |
||||||
|
|
||||||
|
def testUpDown(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
self.assertIsNotNone(service.servicer_methods) |
||||||
|
self.assertIsNotNone(service.server) |
||||||
|
self.assertIsNotNone(service.stub) |
||||||
|
|
||||||
|
def testIncompleteServicer(self): |
||||||
|
service = _CreateIncompleteService(self._service_pb2) |
||||||
|
request = self._request_pb2.SimpleRequest(response_size=13) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
service.stub.UnaryCall(request) |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.UNIMPLEMENTED) |
||||||
|
|
||||||
|
def testUnaryCall(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = self._request_pb2.SimpleRequest(response_size=13) |
||||||
|
response = service.stub.UnaryCall(request) |
||||||
|
expected_response = service.servicer_methods.UnaryCall( |
||||||
|
request, 'not a real context!') |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testUnaryCallFuture(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = self._request_pb2.SimpleRequest(response_size=13) |
||||||
|
# Check that the call does not block waiting for the server to respond. |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
response_future = service.stub.UnaryCall.future(request) |
||||||
|
response = response_future.result() |
||||||
|
expected_response = service.servicer_methods.UnaryCall( |
||||||
|
request, 'not a real RpcContext!') |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testUnaryCallFutureExpired(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = self._request_pb2.SimpleRequest(response_size=13) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
response_future = service.stub.UnaryCall.future( |
||||||
|
request, timeout=test_constants.SHORT_TIMEOUT) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
response_future.result() |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
self.assertIs(response_future.code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
|
||||||
|
def testUnaryCallFutureCancelled(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = self._request_pb2.SimpleRequest(response_size=13) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
response_future = service.stub.UnaryCall.future(request) |
||||||
|
response_future.cancel() |
||||||
|
self.assertTrue(response_future.cancelled()) |
||||||
|
self.assertIs(response_future.code(), grpc.StatusCode.CANCELLED) |
||||||
|
|
||||||
|
def testUnaryCallFutureFailed(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = self._request_pb2.SimpleRequest(response_size=13) |
||||||
|
with service.servicer_methods.fail(): |
||||||
|
response_future = service.stub.UnaryCall.future(request) |
||||||
|
self.assertIsNotNone(response_future.exception()) |
||||||
|
self.assertIs(response_future.code(), grpc.StatusCode.UNKNOWN) |
||||||
|
|
||||||
|
def testStreamingOutputCall(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = _streaming_output_request(self._request_pb2) |
||||||
|
responses = service.stub.StreamingOutputCall(request) |
||||||
|
expected_responses = service.servicer_methods.StreamingOutputCall( |
||||||
|
request, 'not a real RpcContext!') |
||||||
|
for expected_response, response in moves.zip_longest( |
||||||
|
expected_responses, responses): |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testStreamingOutputCallExpired(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = _streaming_output_request(self._request_pb2) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
responses = service.stub.StreamingOutputCall( |
||||||
|
request, timeout=test_constants.SHORT_TIMEOUT) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
list(responses) |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
|
||||||
|
def testStreamingOutputCallCancelled(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = _streaming_output_request(self._request_pb2) |
||||||
|
responses = service.stub.StreamingOutputCall(request) |
||||||
|
next(responses) |
||||||
|
responses.cancel() |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
next(responses) |
||||||
|
self.assertIs(responses.code(), grpc.StatusCode.CANCELLED) |
||||||
|
|
||||||
|
def testStreamingOutputCallFailed(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request = _streaming_output_request(self._request_pb2) |
||||||
|
with service.servicer_methods.fail(): |
||||||
|
responses = service.stub.StreamingOutputCall(request) |
||||||
|
self.assertIsNotNone(responses) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
next(responses) |
||||||
|
self.assertIs(exception_context.exception.code(), grpc.StatusCode.UNKNOWN) |
||||||
|
|
||||||
|
def testStreamingInputCall(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
response = service.stub.StreamingInputCall( |
||||||
|
_streaming_input_request_iterator( |
||||||
|
self._request_pb2, self._payload_pb2)) |
||||||
|
expected_response = service.servicer_methods.StreamingInputCall( |
||||||
|
_streaming_input_request_iterator(self._request_pb2, self._payload_pb2), |
||||||
|
'not a real RpcContext!') |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testStreamingInputCallFuture(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
response_future = service.stub.StreamingInputCall.future( |
||||||
|
_streaming_input_request_iterator( |
||||||
|
self._request_pb2, self._payload_pb2)) |
||||||
|
response = response_future.result() |
||||||
|
expected_response = service.servicer_methods.StreamingInputCall( |
||||||
|
_streaming_input_request_iterator(self._request_pb2, self._payload_pb2), |
||||||
|
'not a real RpcContext!') |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testStreamingInputCallFutureExpired(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
response_future = service.stub.StreamingInputCall.future( |
||||||
|
_streaming_input_request_iterator( |
||||||
|
self._request_pb2, self._payload_pb2), |
||||||
|
timeout=test_constants.SHORT_TIMEOUT) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
response_future.result() |
||||||
|
self.assertIsInstance(response_future.exception(), grpc.RpcError) |
||||||
|
self.assertIs( |
||||||
|
response_future.exception().code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
|
||||||
|
def testStreamingInputCallFutureCancelled(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
response_future = service.stub.StreamingInputCall.future( |
||||||
|
_streaming_input_request_iterator( |
||||||
|
self._request_pb2, self._payload_pb2)) |
||||||
|
response_future.cancel() |
||||||
|
self.assertTrue(response_future.cancelled()) |
||||||
|
with self.assertRaises(grpc.FutureCancelledError): |
||||||
|
response_future.result() |
||||||
|
|
||||||
|
def testStreamingInputCallFutureFailed(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with service.servicer_methods.fail(): |
||||||
|
response_future = service.stub.StreamingInputCall.future( |
||||||
|
_streaming_input_request_iterator( |
||||||
|
self._request_pb2, self._payload_pb2)) |
||||||
|
self.assertIsNotNone(response_future.exception()) |
||||||
|
self.assertIs(response_future.code(), grpc.StatusCode.UNKNOWN) |
||||||
|
|
||||||
|
def testFullDuplexCall(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
responses = service.stub.FullDuplexCall( |
||||||
|
_full_duplex_request_iterator(self._request_pb2)) |
||||||
|
expected_responses = service.servicer_methods.FullDuplexCall( |
||||||
|
_full_duplex_request_iterator(self._request_pb2), |
||||||
|
'not a real RpcContext!') |
||||||
|
for expected_response, response in moves.zip_longest( |
||||||
|
expected_responses, responses): |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testFullDuplexCallExpired(self): |
||||||
|
request_iterator = _full_duplex_request_iterator(self._request_pb2) |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with service.servicer_methods.pause(): |
||||||
|
responses = service.stub.FullDuplexCall( |
||||||
|
request_iterator, timeout=test_constants.SHORT_TIMEOUT) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
list(responses) |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
|
||||||
|
def testFullDuplexCallCancelled(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
request_iterator = _full_duplex_request_iterator(self._request_pb2) |
||||||
|
responses = service.stub.FullDuplexCall(request_iterator) |
||||||
|
next(responses) |
||||||
|
responses.cancel() |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
next(responses) |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.CANCELLED) |
||||||
|
|
||||||
|
def testFullDuplexCallFailed(self): |
||||||
|
request_iterator = _full_duplex_request_iterator(self._request_pb2) |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with service.servicer_methods.fail(): |
||||||
|
responses = service.stub.FullDuplexCall(request_iterator) |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
next(responses) |
||||||
|
self.assertIs(exception_context.exception.code(), grpc.StatusCode.UNKNOWN) |
||||||
|
|
||||||
|
def testHalfDuplexCall(self): |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
def half_duplex_request_iterator(): |
||||||
|
request = self._request_pb2.StreamingOutputCallRequest() |
||||||
|
request.response_parameters.add(size=1, interval_us=0) |
||||||
|
yield request |
||||||
|
request = self._request_pb2.StreamingOutputCallRequest() |
||||||
|
request.response_parameters.add(size=2, interval_us=0) |
||||||
|
request.response_parameters.add(size=3, interval_us=0) |
||||||
|
yield request |
||||||
|
responses = service.stub.HalfDuplexCall(half_duplex_request_iterator()) |
||||||
|
expected_responses = service.servicer_methods.HalfDuplexCall( |
||||||
|
half_duplex_request_iterator(), 'not a real RpcContext!') |
||||||
|
for expected_response, response in moves.zip_longest( |
||||||
|
expected_responses, responses): |
||||||
|
self.assertEqual(expected_response, response) |
||||||
|
|
||||||
|
def testHalfDuplexCallWedged(self): |
||||||
|
condition = threading.Condition() |
||||||
|
wait_cell = [False] |
||||||
|
@contextlib.contextmanager |
||||||
|
def wait(): # pylint: disable=invalid-name |
||||||
|
# Where's Python 3's 'nonlocal' statement when you need it? |
||||||
|
with condition: |
||||||
|
wait_cell[0] = True |
||||||
|
yield |
||||||
|
with condition: |
||||||
|
wait_cell[0] = False |
||||||
|
condition.notify_all() |
||||||
|
def half_duplex_request_iterator(): |
||||||
|
request = self._request_pb2.StreamingOutputCallRequest() |
||||||
|
request.response_parameters.add(size=1, interval_us=0) |
||||||
|
yield request |
||||||
|
with condition: |
||||||
|
while wait_cell[0]: |
||||||
|
condition.wait() |
||||||
|
service = _CreateService( |
||||||
|
self._service_pb2, self._response_pb2, self._payload_pb2) |
||||||
|
with wait(): |
||||||
|
responses = service.stub.HalfDuplexCall( |
||||||
|
half_duplex_request_iterator(), timeout=test_constants.SHORT_TIMEOUT) |
||||||
|
# half-duplex waits for the client to send all info |
||||||
|
with self.assertRaises(grpc.RpcError) as exception_context: |
||||||
|
next(responses) |
||||||
|
self.assertIs( |
||||||
|
exception_context.exception.code(), grpc.StatusCode.DEADLINE_EXCEEDED) |
||||||
|
|
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main(verbosity=2) |
||||||
@ -0,0 +1,117 @@ |
|||||||
|
# Copyright 2016, 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 CleanupThread.""" |
||||||
|
|
||||||
|
import threading |
||||||
|
import time |
||||||
|
import unittest |
||||||
|
|
||||||
|
from grpc import _common |
||||||
|
|
||||||
|
_SHORT_TIME = 0.5 |
||||||
|
_LONG_TIME = 2.0 |
||||||
|
_EPSILON = 0.1 |
||||||
|
|
||||||
|
|
||||||
|
def cleanup(timeout): |
||||||
|
if timeout is not None: |
||||||
|
time.sleep(timeout) |
||||||
|
else: |
||||||
|
time.sleep(_LONG_TIME) |
||||||
|
|
||||||
|
|
||||||
|
def slow_cleanup(timeout): |
||||||
|
# Don't respect timeout |
||||||
|
time.sleep(_LONG_TIME) |
||||||
|
|
||||||
|
|
||||||
|
class CleanupThreadTest(unittest.TestCase): |
||||||
|
|
||||||
|
def testTargetInvocation(self): |
||||||
|
event = threading.Event() |
||||||
|
def target(arg1, arg2, arg3=None): |
||||||
|
self.assertEqual('arg1', arg1) |
||||||
|
self.assertEqual('arg2', arg2) |
||||||
|
self.assertEqual('arg3', arg3) |
||||||
|
event.set() |
||||||
|
|
||||||
|
cleanup_thread = _common.CleanupThread(behavior=lambda x: None, |
||||||
|
target=target, name='test-name', |
||||||
|
args=('arg1', 'arg2'), kwargs={'arg3': 'arg3'}) |
||||||
|
cleanup_thread.start() |
||||||
|
cleanup_thread.join() |
||||||
|
self.assertEqual(cleanup_thread.name, 'test-name') |
||||||
|
self.assertTrue(event.is_set()) |
||||||
|
|
||||||
|
def testJoinNoTimeout(self): |
||||||
|
cleanup_thread = _common.CleanupThread(behavior=cleanup) |
||||||
|
cleanup_thread.start() |
||||||
|
start_time = time.time() |
||||||
|
cleanup_thread.join() |
||||||
|
end_time = time.time() |
||||||
|
self.assertAlmostEqual(_LONG_TIME, end_time - start_time, delta=_EPSILON) |
||||||
|
|
||||||
|
def testJoinTimeout(self): |
||||||
|
cleanup_thread = _common.CleanupThread(behavior=cleanup) |
||||||
|
cleanup_thread.start() |
||||||
|
start_time = time.time() |
||||||
|
cleanup_thread.join(_SHORT_TIME) |
||||||
|
end_time = time.time() |
||||||
|
self.assertAlmostEqual(_SHORT_TIME, end_time - start_time, delta=_EPSILON) |
||||||
|
|
||||||
|
def testJoinTimeoutSlowBehavior(self): |
||||||
|
cleanup_thread = _common.CleanupThread(behavior=slow_cleanup) |
||||||
|
cleanup_thread.start() |
||||||
|
start_time = time.time() |
||||||
|
cleanup_thread.join(_SHORT_TIME) |
||||||
|
end_time = time.time() |
||||||
|
self.assertAlmostEqual(_LONG_TIME, end_time - start_time, delta=_EPSILON) |
||||||
|
|
||||||
|
def testJoinTimeoutSlowTarget(self): |
||||||
|
event = threading.Event() |
||||||
|
def target(): |
||||||
|
event.wait(_LONG_TIME) |
||||||
|
cleanup_thread = _common.CleanupThread(behavior=cleanup, target=target) |
||||||
|
cleanup_thread.start() |
||||||
|
start_time = time.time() |
||||||
|
cleanup_thread.join(_SHORT_TIME) |
||||||
|
end_time = time.time() |
||||||
|
self.assertAlmostEqual(_SHORT_TIME, end_time - start_time, delta=_EPSILON) |
||||||
|
event.set() |
||||||
|
|
||||||
|
def testJoinZeroTimeout(self): |
||||||
|
cleanup_thread = _common.CleanupThread(behavior=cleanup) |
||||||
|
cleanup_thread.start() |
||||||
|
start_time = time.time() |
||||||
|
cleanup_thread.join(0) |
||||||
|
end_time = time.time() |
||||||
|
self.assertAlmostEqual(0, end_time - start_time, delta=_EPSILON) |
||||||
|
|
||||||
|
if __name__ == '__main__': |
||||||
|
unittest.main(verbosity=2) |
||||||
@ -0,0 +1,144 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, Google Inc. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions are |
||||||
|
* met: |
||||||
|
* |
||||||
|
* * Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* * Redistributions in binary form must reproduce the above |
||||||
|
* copyright notice, this list of conditions and the following disclaimer |
||||||
|
* in the documentation and/or other materials provided with the |
||||||
|
* distribution. |
||||||
|
* * Neither the name of Google Inc. nor the names of its |
||||||
|
* contributors may be used to endorse or promote products derived from |
||||||
|
* this software without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||||
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||||
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||||
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||||
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||||
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||||
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||||
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||||
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||||
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||||
|
* |
||||||
|
*/ |
||||||
|
|
||||||
|
#include "src/core/ext/transport/chttp2/transport/bin_decoder.h" |
||||||
|
|
||||||
|
#include <string.h> |
||||||
|
|
||||||
|
#include <grpc/support/alloc.h> |
||||||
|
#include <grpc/support/log.h> |
||||||
|
#include "src/core/ext/transport/chttp2/transport/bin_encoder.h" |
||||||
|
#include "src/core/lib/support/string.h" |
||||||
|
|
||||||
|
static int all_ok = 1; |
||||||
|
|
||||||
|
static void expect_slice_eq(gpr_slice expected, gpr_slice slice, char *debug, |
||||||
|
int line) { |
||||||
|
if (0 != gpr_slice_cmp(slice, expected)) { |
||||||
|
char *hs = gpr_dump_slice(slice, GPR_DUMP_HEX | GPR_DUMP_ASCII); |
||||||
|
char *he = gpr_dump_slice(expected, GPR_DUMP_HEX | GPR_DUMP_ASCII); |
||||||
|
gpr_log(GPR_ERROR, "FAILED:%d: %s\ngot: %s\nwant: %s", line, debug, hs, |
||||||
|
he); |
||||||
|
gpr_free(hs); |
||||||
|
gpr_free(he); |
||||||
|
all_ok = 0; |
||||||
|
} |
||||||
|
gpr_slice_unref(expected); |
||||||
|
gpr_slice_unref(slice); |
||||||
|
} |
||||||
|
|
||||||
|
static gpr_slice base64_encode(const char *s) { |
||||||
|
gpr_slice ss = gpr_slice_from_copied_string(s); |
||||||
|
gpr_slice out = grpc_chttp2_base64_encode(ss); |
||||||
|
gpr_slice_unref(ss); |
||||||
|
return out; |
||||||
|
} |
||||||
|
|
||||||
|
static gpr_slice base64_decode(const char *s) { |
||||||
|
gpr_slice ss = gpr_slice_from_copied_string(s); |
||||||
|
gpr_slice out = grpc_chttp2_base64_decode(ss); |
||||||
|
gpr_slice_unref(ss); |
||||||
|
return out; |
||||||
|
} |
||||||
|
|
||||||
|
static gpr_slice base64_decode_with_length(const char *s, |
||||||
|
size_t output_length) { |
||||||
|
gpr_slice ss = gpr_slice_from_copied_string(s); |
||||||
|
gpr_slice out = grpc_chttp2_base64_decode_with_length(ss, output_length); |
||||||
|
gpr_slice_unref(ss); |
||||||
|
return out; |
||||||
|
} |
||||||
|
|
||||||
|
#define EXPECT_SLICE_EQ(expected, slice) \ |
||||||
|
expect_slice_eq( \
|
||||||
|
gpr_slice_from_copied_buffer(expected, sizeof(expected) - 1), slice, \
|
||||||
|
#slice, __LINE__); |
||||||
|
|
||||||
|
#define ENCODE_AND_DECODE(s) \ |
||||||
|
EXPECT_SLICE_EQ( \
|
||||||
|
s, grpc_chttp2_base64_decode_with_length(base64_encode(s), strlen(s))); |
||||||
|
|
||||||
|
int main(int argc, char **argv) { |
||||||
|
/* ENCODE_AND_DECODE tests grpc_chttp2_base64_decode_with_length(), which
|
||||||
|
takes encoded base64 strings without pad chars, but output length is |
||||||
|
required. */ |
||||||
|
/* Base64 test vectors from RFC 4648 */ |
||||||
|
ENCODE_AND_DECODE(""); |
||||||
|
ENCODE_AND_DECODE("f"); |
||||||
|
ENCODE_AND_DECODE("foo"); |
||||||
|
ENCODE_AND_DECODE("fo"); |
||||||
|
ENCODE_AND_DECODE("foob"); |
||||||
|
ENCODE_AND_DECODE("fooba"); |
||||||
|
ENCODE_AND_DECODE("foobar"); |
||||||
|
|
||||||
|
ENCODE_AND_DECODE("\xc0\xc1\xc2\xc3\xc4\xc5"); |
||||||
|
|
||||||
|
/* Base64 test vectors from RFC 4648, with pad chars */ |
||||||
|
/* BASE64("") = "" */ |
||||||
|
EXPECT_SLICE_EQ("", base64_decode("")); |
||||||
|
/* BASE64("f") = "Zg==" */ |
||||||
|
EXPECT_SLICE_EQ("f", base64_decode("Zg==")); |
||||||
|
/* BASE64("fo") = "Zm8=" */ |
||||||
|
EXPECT_SLICE_EQ("fo", base64_decode("Zm8=")); |
||||||
|
/* BASE64("foo") = "Zm9v" */ |
||||||
|
EXPECT_SLICE_EQ("foo", base64_decode("Zm9v")); |
||||||
|
/* BASE64("foob") = "Zm9vYg==" */ |
||||||
|
EXPECT_SLICE_EQ("foob", base64_decode("Zm9vYg==")); |
||||||
|
/* BASE64("fooba") = "Zm9vYmE=" */ |
||||||
|
EXPECT_SLICE_EQ("fooba", base64_decode("Zm9vYmE=")); |
||||||
|
/* BASE64("foobar") = "Zm9vYmFy" */ |
||||||
|
EXPECT_SLICE_EQ("foobar", base64_decode("Zm9vYmFy")); |
||||||
|
|
||||||
|
EXPECT_SLICE_EQ("\xc0\xc1\xc2\xc3\xc4\xc5", base64_decode("wMHCw8TF")); |
||||||
|
|
||||||
|
// Test illegal input length in grpc_chttp2_base64_decode
|
||||||
|
EXPECT_SLICE_EQ("", base64_decode("a")); |
||||||
|
EXPECT_SLICE_EQ("", base64_decode("ab")); |
||||||
|
EXPECT_SLICE_EQ("", base64_decode("abc")); |
||||||
|
|
||||||
|
// Test illegal charactors in grpc_chttp2_base64_decode
|
||||||
|
EXPECT_SLICE_EQ("", base64_decode("Zm:v")); |
||||||
|
EXPECT_SLICE_EQ("", base64_decode("Zm=v")); |
||||||
|
|
||||||
|
// Test output_length longer than max possible output length in
|
||||||
|
// grpc_chttp2_base64_decode_with_length
|
||||||
|
EXPECT_SLICE_EQ("", base64_decode_with_length("Zg", 2)); |
||||||
|
EXPECT_SLICE_EQ("", base64_decode_with_length("Zm8", 3)); |
||||||
|
EXPECT_SLICE_EQ("", base64_decode_with_length("Zm9v", 4)); |
||||||
|
|
||||||
|
// Test illegal charactors in grpc_chttp2_base64_decode_with_length
|
||||||
|
EXPECT_SLICE_EQ("", base64_decode_with_length("Zm:v", 3)); |
||||||
|
EXPECT_SLICE_EQ("", base64_decode_with_length("Zm=v", 3)); |
||||||
|
|
||||||
|
return all_ok ? 0 : 1; |
||||||
|
} |
||||||
@ -0,0 +1,166 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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 <google/protobuf/descriptor.h> |
||||||
|
#include <grpc++/channel.h> |
||||||
|
#include <grpc++/client_context.h> |
||||||
|
#include <grpc++/create_channel.h> |
||||||
|
#include <grpc++/ext/proto_server_reflection_plugin.h> |
||||||
|
#include <grpc++/security/credentials.h> |
||||||
|
#include <grpc++/security/server_credentials.h> |
||||||
|
#include <grpc++/server.h> |
||||||
|
#include <grpc++/server_builder.h> |
||||||
|
#include <grpc++/server_context.h> |
||||||
|
#include <grpc/grpc.h> |
||||||
|
#include <gtest/gtest.h> |
||||||
|
|
||||||
|
#include "src/proto/grpc/testing/echo.grpc.pb.h" |
||||||
|
#include "test/core/util/port.h" |
||||||
|
#include "test/core/util/test_config.h" |
||||||
|
#include "test/cpp/end2end/test_service_impl.h" |
||||||
|
#include "test/cpp/util/proto_reflection_descriptor_database.h" |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
namespace testing { |
||||||
|
|
||||||
|
class ProtoServerReflectionTest : public ::testing::Test { |
||||||
|
public: |
||||||
|
ProtoServerReflectionTest() {} |
||||||
|
|
||||||
|
void SetUp() GRPC_OVERRIDE { |
||||||
|
port_ = grpc_pick_unused_port_or_die(); |
||||||
|
ref_desc_pool_ = google::protobuf::DescriptorPool::generated_pool(); |
||||||
|
|
||||||
|
ServerBuilder builder; |
||||||
|
grpc::string server_address = "localhost:" + to_string(port_); |
||||||
|
builder.AddListeningPort(server_address, InsecureServerCredentials()); |
||||||
|
server_ = builder.BuildAndStart(); |
||||||
|
} |
||||||
|
|
||||||
|
void ResetStub() { |
||||||
|
string target = "dns:localhost:" + to_string(port_); |
||||||
|
std::shared_ptr<Channel> channel = |
||||||
|
CreateChannel(target, InsecureChannelCredentials()); |
||||||
|
stub_ = grpc::testing::EchoTestService::NewStub(channel); |
||||||
|
desc_db_.reset(new ProtoReflectionDescriptorDatabase(channel)); |
||||||
|
desc_pool_.reset(new google::protobuf::DescriptorPool(desc_db_.get())); |
||||||
|
} |
||||||
|
|
||||||
|
string to_string(const int number) { |
||||||
|
std::stringstream strs; |
||||||
|
strs << number; |
||||||
|
return strs.str(); |
||||||
|
} |
||||||
|
|
||||||
|
void CompareService(const grpc::string& service) { |
||||||
|
const google::protobuf::ServiceDescriptor* service_desc = |
||||||
|
desc_pool_->FindServiceByName(service); |
||||||
|
const google::protobuf::ServiceDescriptor* ref_service_desc = |
||||||
|
ref_desc_pool_->FindServiceByName(service); |
||||||
|
EXPECT_TRUE(service_desc != nullptr); |
||||||
|
EXPECT_TRUE(ref_service_desc != nullptr); |
||||||
|
EXPECT_EQ(service_desc->DebugString(), ref_service_desc->DebugString()); |
||||||
|
|
||||||
|
const google::protobuf::FileDescriptor* file_desc = service_desc->file(); |
||||||
|
if (known_files_.find(file_desc->package() + "/" + file_desc->name()) != |
||||||
|
known_files_.end()) { |
||||||
|
EXPECT_EQ(file_desc->DebugString(), |
||||||
|
ref_service_desc->file()->DebugString()); |
||||||
|
known_files_.insert(file_desc->package() + "/" + file_desc->name()); |
||||||
|
} |
||||||
|
|
||||||
|
for (int i = 0; i < service_desc->method_count(); ++i) { |
||||||
|
CompareMethod(service_desc->method(i)->full_name()); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
void CompareMethod(const grpc::string& method) { |
||||||
|
const google::protobuf::MethodDescriptor* method_desc = |
||||||
|
desc_pool_->FindMethodByName(method); |
||||||
|
const google::protobuf::MethodDescriptor* ref_method_desc = |
||||||
|
ref_desc_pool_->FindMethodByName(method); |
||||||
|
EXPECT_TRUE(method_desc != nullptr); |
||||||
|
EXPECT_TRUE(ref_method_desc != nullptr); |
||||||
|
EXPECT_EQ(method_desc->DebugString(), ref_method_desc->DebugString()); |
||||||
|
|
||||||
|
CompareType(method_desc->input_type()->full_name()); |
||||||
|
CompareType(method_desc->output_type()->full_name()); |
||||||
|
} |
||||||
|
|
||||||
|
void CompareType(const grpc::string& type) { |
||||||
|
if (known_types_.find(type) != known_types_.end()) { |
||||||
|
return; |
||||||
|
} |
||||||
|
|
||||||
|
const google::protobuf::Descriptor* desc = |
||||||
|
desc_pool_->FindMessageTypeByName(type); |
||||||
|
const google::protobuf::Descriptor* ref_desc = |
||||||
|
ref_desc_pool_->FindMessageTypeByName(type); |
||||||
|
EXPECT_TRUE(desc != nullptr); |
||||||
|
EXPECT_TRUE(ref_desc != nullptr); |
||||||
|
EXPECT_EQ(desc->DebugString(), ref_desc->DebugString()); |
||||||
|
} |
||||||
|
|
||||||
|
protected: |
||||||
|
std::unique_ptr<Server> server_; |
||||||
|
std::unique_ptr<grpc::testing::EchoTestService::Stub> stub_; |
||||||
|
std::unique_ptr<ProtoReflectionDescriptorDatabase> desc_db_; |
||||||
|
std::unique_ptr<google::protobuf::DescriptorPool> desc_pool_; |
||||||
|
std::unordered_set<string> known_files_; |
||||||
|
std::unordered_set<string> known_types_; |
||||||
|
const google::protobuf::DescriptorPool* ref_desc_pool_; |
||||||
|
int port_; |
||||||
|
reflection::ProtoServerReflectionPlugin plugin_; |
||||||
|
}; |
||||||
|
|
||||||
|
TEST_F(ProtoServerReflectionTest, CheckResponseWithLocalDescriptorPool) { |
||||||
|
ResetStub(); |
||||||
|
|
||||||
|
std::vector<std::string> services; |
||||||
|
desc_db_->GetServices(&services); |
||||||
|
// The service list has at least one service (reflection servcie).
|
||||||
|
EXPECT_TRUE(services.size() > 0); |
||||||
|
|
||||||
|
for (auto it = services.begin(); it != services.end(); ++it) { |
||||||
|
CompareService(*it); |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace testing
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
int main(int argc, char** argv) { |
||||||
|
grpc_test_init(argc, argv); |
||||||
|
::testing::InitGoogleTest(&argc, argv); |
||||||
|
return RUN_ALL_TESTS(); |
||||||
|
} |
||||||
@ -0,0 +1,316 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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 "test/cpp/util/proto_reflection_descriptor_database.h" |
||||||
|
|
||||||
|
#include <vector> |
||||||
|
|
||||||
|
#include <grpc/support/log.h> |
||||||
|
|
||||||
|
using grpc::reflection::v1alpha::ServerReflection; |
||||||
|
using grpc::reflection::v1alpha::ServerReflectionRequest; |
||||||
|
using grpc::reflection::v1alpha::ServerReflectionResponse; |
||||||
|
using grpc::reflection::v1alpha::ListServiceResponse; |
||||||
|
using grpc::reflection::v1alpha::ErrorResponse; |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
ProtoReflectionDescriptorDatabase::ProtoReflectionDescriptorDatabase( |
||||||
|
std::unique_ptr<ServerReflection::Stub> stub) |
||||||
|
: stub_(std::move(stub)) {} |
||||||
|
|
||||||
|
ProtoReflectionDescriptorDatabase::ProtoReflectionDescriptorDatabase( |
||||||
|
std::shared_ptr<grpc::Channel> channel) |
||||||
|
: stub_(ServerReflection::NewStub(channel)) {} |
||||||
|
|
||||||
|
ProtoReflectionDescriptorDatabase::~ProtoReflectionDescriptorDatabase() {} |
||||||
|
|
||||||
|
bool ProtoReflectionDescriptorDatabase::FindFileByName( |
||||||
|
const string& filename, google::protobuf::FileDescriptorProto* output) { |
||||||
|
if (cached_db_.FindFileByName(filename, output)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
if (known_files_.find(filename) != known_files_.end()) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflectionRequest request; |
||||||
|
request.set_file_by_filename(filename); |
||||||
|
ServerReflectionResponse response; |
||||||
|
|
||||||
|
DoOneRequest(request, response); |
||||||
|
|
||||||
|
if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kFileDescriptorResponse) { |
||||||
|
AddFileFromResponse(response.file_descriptor_response()); |
||||||
|
} else if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kErrorResponse) { |
||||||
|
const ErrorResponse error = response.error_response(); |
||||||
|
if (error.error_code() == StatusCode::NOT_FOUND) { |
||||||
|
gpr_log(GPR_INFO, "NOT_FOUND from server for FindFileByName(%s)", |
||||||
|
filename.c_str()); |
||||||
|
} else { |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Error on FindFileByName(%s)\n\tError code: %d\n" |
||||||
|
"\tError Message: %s", |
||||||
|
filename.c_str(), error.error_code(), |
||||||
|
error.error_message().c_str()); |
||||||
|
} |
||||||
|
} else { |
||||||
|
gpr_log( |
||||||
|
GPR_INFO, |
||||||
|
"Error on FindFileByName(%s) response type\n" |
||||||
|
"\tExpecting: %d\n\tReceived: %d", |
||||||
|
filename.c_str(), |
||||||
|
ServerReflectionResponse::MessageResponseCase::kFileDescriptorResponse, |
||||||
|
response.message_response_case()); |
||||||
|
} |
||||||
|
|
||||||
|
return cached_db_.FindFileByName(filename, output); |
||||||
|
} |
||||||
|
|
||||||
|
bool ProtoReflectionDescriptorDatabase::FindFileContainingSymbol( |
||||||
|
const string& symbol_name, google::protobuf::FileDescriptorProto* output) { |
||||||
|
if (cached_db_.FindFileContainingSymbol(symbol_name, output)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
if (missing_symbols_.find(symbol_name) != missing_symbols_.end()) { |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflectionRequest request; |
||||||
|
request.set_file_containing_symbol(symbol_name); |
||||||
|
ServerReflectionResponse response; |
||||||
|
|
||||||
|
DoOneRequest(request, response); |
||||||
|
|
||||||
|
if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kFileDescriptorResponse) { |
||||||
|
AddFileFromResponse(response.file_descriptor_response()); |
||||||
|
} else if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kErrorResponse) { |
||||||
|
const ErrorResponse error = response.error_response(); |
||||||
|
if (error.error_code() == StatusCode::NOT_FOUND) { |
||||||
|
missing_symbols_.insert(symbol_name); |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"NOT_FOUND from server for FindFileContainingSymbol(%s)", |
||||||
|
symbol_name.c_str()); |
||||||
|
} else { |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Error on FindFileContainingSymbol(%s)\n" |
||||||
|
"\tError code: %d\n\tError Message: %s", |
||||||
|
symbol_name.c_str(), error.error_code(), |
||||||
|
error.error_message().c_str()); |
||||||
|
} |
||||||
|
} else { |
||||||
|
gpr_log( |
||||||
|
GPR_INFO, |
||||||
|
"Error on FindFileContainingSymbol(%s) response type\n" |
||||||
|
"\tExpecting: %d\n\tReceived: %d", |
||||||
|
symbol_name.c_str(), |
||||||
|
ServerReflectionResponse::MessageResponseCase::kFileDescriptorResponse, |
||||||
|
response.message_response_case()); |
||||||
|
} |
||||||
|
return cached_db_.FindFileContainingSymbol(symbol_name, output); |
||||||
|
} |
||||||
|
|
||||||
|
bool ProtoReflectionDescriptorDatabase::FindFileContainingExtension( |
||||||
|
const string& containing_type, int field_number, |
||||||
|
google::protobuf::FileDescriptorProto* output) { |
||||||
|
if (cached_db_.FindFileContainingExtension(containing_type, field_number, |
||||||
|
output)) { |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
if (missing_extensions_.find(containing_type) != missing_extensions_.end() && |
||||||
|
missing_extensions_[containing_type].find(field_number) != |
||||||
|
missing_extensions_[containing_type].end()) { |
||||||
|
gpr_log(GPR_INFO, "nested map."); |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflectionRequest request; |
||||||
|
request.mutable_file_containing_extension()->set_containing_type( |
||||||
|
containing_type); |
||||||
|
request.mutable_file_containing_extension()->set_extension_number( |
||||||
|
field_number); |
||||||
|
ServerReflectionResponse response; |
||||||
|
|
||||||
|
DoOneRequest(request, response); |
||||||
|
|
||||||
|
if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kFileDescriptorResponse) { |
||||||
|
AddFileFromResponse(response.file_descriptor_response()); |
||||||
|
} else if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kErrorResponse) { |
||||||
|
const ErrorResponse error = response.error_response(); |
||||||
|
if (error.error_code() == StatusCode::NOT_FOUND) { |
||||||
|
if (missing_extensions_.find(containing_type) == |
||||||
|
missing_extensions_.end()) { |
||||||
|
missing_extensions_[containing_type] = {}; |
||||||
|
} |
||||||
|
missing_extensions_[containing_type].insert(field_number); |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"NOT_FOUND from server for FindFileContainingExtension(%s, %d)", |
||||||
|
containing_type.c_str(), field_number); |
||||||
|
} else { |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Error on FindFileContainingExtension(%s, %d)\n" |
||||||
|
"\tError code: %d\n\tError Message: %s", |
||||||
|
containing_type.c_str(), field_number, error.error_code(), |
||||||
|
error.error_message().c_str()); |
||||||
|
} |
||||||
|
} else { |
||||||
|
gpr_log( |
||||||
|
GPR_INFO, |
||||||
|
"Error on FindFileContainingExtension(%s, %d) response type\n" |
||||||
|
"\tExpecting: %d\n\tReceived: %d", |
||||||
|
containing_type.c_str(), field_number, |
||||||
|
ServerReflectionResponse::MessageResponseCase::kFileDescriptorResponse, |
||||||
|
response.message_response_case()); |
||||||
|
} |
||||||
|
|
||||||
|
return cached_db_.FindFileContainingExtension(containing_type, field_number, |
||||||
|
output); |
||||||
|
} |
||||||
|
|
||||||
|
bool ProtoReflectionDescriptorDatabase::FindAllExtensionNumbers( |
||||||
|
const string& extendee_type, std::vector<int>* output) { |
||||||
|
if (cached_extension_numbers_.find(extendee_type) != |
||||||
|
cached_extension_numbers_.end()) { |
||||||
|
*output = cached_extension_numbers_[extendee_type]; |
||||||
|
return true; |
||||||
|
} |
||||||
|
|
||||||
|
ServerReflectionRequest request; |
||||||
|
request.set_all_extension_numbers_of_type(extendee_type); |
||||||
|
ServerReflectionResponse response; |
||||||
|
|
||||||
|
DoOneRequest(request, response); |
||||||
|
|
||||||
|
if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase:: |
||||||
|
kAllExtensionNumbersResponse) { |
||||||
|
auto number = response.all_extension_numbers_response().extension_number(); |
||||||
|
*output = std::vector<int>(number.begin(), number.end()); |
||||||
|
cached_extension_numbers_[extendee_type] = *output; |
||||||
|
return true; |
||||||
|
} else if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kErrorResponse) { |
||||||
|
const ErrorResponse error = response.error_response(); |
||||||
|
if (error.error_code() == StatusCode::NOT_FOUND) { |
||||||
|
gpr_log(GPR_INFO, "NOT_FOUND from server for FindAllExtensionNumbers(%s)", |
||||||
|
extendee_type.c_str()); |
||||||
|
} else { |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Error on FindAllExtensionNumbersExtension(%s)\n" |
||||||
|
"\tError code: %d\n\tError Message: %s", |
||||||
|
extendee_type.c_str(), error.error_code(), |
||||||
|
error.error_message().c_str()); |
||||||
|
} |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
bool ProtoReflectionDescriptorDatabase::GetServices( |
||||||
|
std::vector<std::string>* output) { |
||||||
|
ServerReflectionRequest request; |
||||||
|
request.set_list_services(""); |
||||||
|
ServerReflectionResponse response; |
||||||
|
|
||||||
|
DoOneRequest(request, response); |
||||||
|
|
||||||
|
if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kListServicesResponse) { |
||||||
|
const ListServiceResponse ls_response = response.list_services_response(); |
||||||
|
for (int i = 0; i < ls_response.service_size(); ++i) { |
||||||
|
(*output).push_back(ls_response.service(i).name()); |
||||||
|
} |
||||||
|
return true; |
||||||
|
} else if (response.message_response_case() == |
||||||
|
ServerReflectionResponse::MessageResponseCase::kErrorResponse) { |
||||||
|
const ErrorResponse error = response.error_response(); |
||||||
|
gpr_log(GPR_INFO, |
||||||
|
"Error on GetServices()\n\tError code: %d\n" |
||||||
|
"\tError Message: %s", |
||||||
|
error.error_code(), error.error_message().c_str()); |
||||||
|
} else { |
||||||
|
gpr_log( |
||||||
|
GPR_INFO, |
||||||
|
"Error on GetServices() response type\n\tExpecting: %d\n\tReceived: %d", |
||||||
|
ServerReflectionResponse::MessageResponseCase::kListServicesResponse, |
||||||
|
response.message_response_case()); |
||||||
|
} |
||||||
|
return false; |
||||||
|
} |
||||||
|
|
||||||
|
const google::protobuf::FileDescriptorProto |
||||||
|
ProtoReflectionDescriptorDatabase::ParseFileDescriptorProtoResponse( |
||||||
|
const std::string& byte_fd_proto) { |
||||||
|
google::protobuf::FileDescriptorProto file_desc_proto; |
||||||
|
file_desc_proto.ParseFromString(byte_fd_proto); |
||||||
|
return file_desc_proto; |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoReflectionDescriptorDatabase::AddFileFromResponse( |
||||||
|
const grpc::reflection::v1alpha::FileDescriptorResponse& response) { |
||||||
|
for (int i = 0; i < response.file_descriptor_proto_size(); ++i) { |
||||||
|
const google::protobuf::FileDescriptorProto file_proto = |
||||||
|
ParseFileDescriptorProtoResponse(response.file_descriptor_proto(i)); |
||||||
|
if (known_files_.find(file_proto.name()) == known_files_.end()) { |
||||||
|
known_files_.insert(file_proto.name()); |
||||||
|
cached_db_.Add(file_proto); |
||||||
|
} |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
const std::shared_ptr<ProtoReflectionDescriptorDatabase::ClientStream> |
||||||
|
ProtoReflectionDescriptorDatabase::GetStream() { |
||||||
|
if (stream_ == nullptr) { |
||||||
|
stream_ = stub_->ServerReflectionInfo(&ctx_); |
||||||
|
} |
||||||
|
return stream_; |
||||||
|
} |
||||||
|
|
||||||
|
void ProtoReflectionDescriptorDatabase::DoOneRequest( |
||||||
|
const ServerReflectionRequest& request, |
||||||
|
ServerReflectionResponse& response) { |
||||||
|
stream_mutex_.lock(); |
||||||
|
GetStream()->Write(request); |
||||||
|
GetStream()->Read(&response); |
||||||
|
stream_mutex_.unlock(); |
||||||
|
} |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
@ -0,0 +1,131 @@ |
|||||||
|
/*
|
||||||
|
* |
||||||
|
* Copyright 2016, 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_TEST_CPP_PROTO_SERVER_REFLECTION_DATABSE_H |
||||||
|
#define GRPC_TEST_CPP_PROTO_SERVER_REFLECTION_DATABSE_H |
||||||
|
|
||||||
|
#include <mutex> |
||||||
|
#include <unordered_map> |
||||||
|
#include <unordered_set> |
||||||
|
#include <vector> |
||||||
|
|
||||||
|
#include <google/protobuf/descriptor.h> |
||||||
|
#include <google/protobuf/descriptor.pb.h> |
||||||
|
#include <google/protobuf/descriptor_database.h> |
||||||
|
#include <grpc++/ext/reflection.grpc.pb.h> |
||||||
|
#include <grpc++/grpc++.h> |
||||||
|
|
||||||
|
namespace grpc { |
||||||
|
|
||||||
|
// ProtoReflectionDescriptorDatabase takes a stub of ServerReflection and
|
||||||
|
// provides the methods defined by DescriptorDatabase interfaces. It can be used
|
||||||
|
// to feed a DescriptorPool instance.
|
||||||
|
class ProtoReflectionDescriptorDatabase |
||||||
|
: public google::protobuf::DescriptorDatabase { |
||||||
|
public: |
||||||
|
explicit ProtoReflectionDescriptorDatabase( |
||||||
|
std::unique_ptr<reflection::v1alpha::ServerReflection::Stub> stub); |
||||||
|
|
||||||
|
explicit ProtoReflectionDescriptorDatabase( |
||||||
|
std::shared_ptr<grpc::Channel> channel); |
||||||
|
|
||||||
|
virtual ~ProtoReflectionDescriptorDatabase(); |
||||||
|
|
||||||
|
// The following four methods implement DescriptorDatabase interfaces.
|
||||||
|
//
|
||||||
|
// Find a file by file name. Fills in in *output and returns true if found.
|
||||||
|
// Otherwise, returns false, leaving the contents of *output undefined.
|
||||||
|
bool FindFileByName(const string& filename, |
||||||
|
google::protobuf::FileDescriptorProto* output) |
||||||
|
GRPC_OVERRIDE; |
||||||
|
|
||||||
|
// Find the file that declares the given fully-qualified symbol name.
|
||||||
|
// If found, fills in *output and returns true, otherwise returns false
|
||||||
|
// and leaves *output undefined.
|
||||||
|
bool FindFileContainingSymbol(const string& symbol_name, |
||||||
|
google::protobuf::FileDescriptorProto* output) |
||||||
|
GRPC_OVERRIDE; |
||||||
|
|
||||||
|
// Find the file which defines an extension extending the given message type
|
||||||
|
// with the given field number. If found, fills in *output and returns true,
|
||||||
|
// otherwise returns false and leaves *output undefined. containing_type
|
||||||
|
// must be a fully-qualified type name.
|
||||||
|
bool FindFileContainingExtension( |
||||||
|
const string& containing_type, int field_number, |
||||||
|
google::protobuf::FileDescriptorProto* output) GRPC_OVERRIDE; |
||||||
|
|
||||||
|
// Finds the tag numbers used by all known extensions of
|
||||||
|
// extendee_type, and appends them to output in an undefined
|
||||||
|
// order. This method is best-effort: it's not guaranteed that the
|
||||||
|
// database will find all extensions, and it's not guaranteed that
|
||||||
|
// FindFileContainingExtension will return true on all of the found
|
||||||
|
// numbers. Returns true if the search was successful, otherwise
|
||||||
|
// returns false and leaves output unchanged.
|
||||||
|
bool FindAllExtensionNumbers(const string& extendee_type, |
||||||
|
std::vector<int>* output) GRPC_OVERRIDE; |
||||||
|
|
||||||
|
// Provide a list of full names of registered services
|
||||||
|
bool GetServices(std::vector<std::string>* output); |
||||||
|
|
||||||
|
private: |
||||||
|
typedef ClientReaderWriter< |
||||||
|
grpc::reflection::v1alpha::ServerReflectionRequest, |
||||||
|
grpc::reflection::v1alpha::ServerReflectionResponse> |
||||||
|
ClientStream; |
||||||
|
|
||||||
|
const google::protobuf::FileDescriptorProto ParseFileDescriptorProtoResponse( |
||||||
|
const std::string& byte_fd_proto); |
||||||
|
|
||||||
|
void AddFileFromResponse( |
||||||
|
const grpc::reflection::v1alpha::FileDescriptorResponse& response); |
||||||
|
|
||||||
|
const std::shared_ptr<ClientStream> GetStream(); |
||||||
|
|
||||||
|
void DoOneRequest( |
||||||
|
const grpc::reflection::v1alpha::ServerReflectionRequest& request, |
||||||
|
grpc::reflection::v1alpha::ServerReflectionResponse& response); |
||||||
|
|
||||||
|
std::shared_ptr<ClientStream> stream_; |
||||||
|
grpc::ClientContext ctx_; |
||||||
|
std::unique_ptr<grpc::reflection::v1alpha::ServerReflection::Stub> stub_; |
||||||
|
std::unordered_set<string> known_files_; |
||||||
|
std::unordered_set<string> missing_symbols_; |
||||||
|
std::unordered_map<string, std::unordered_set<int>> missing_extensions_; |
||||||
|
std::unordered_map<string, std::vector<int>> cached_extension_numbers_; |
||||||
|
std::mutex stream_mutex_; |
||||||
|
|
||||||
|
google::protobuf::SimpleDescriptorDatabase cached_db_; |
||||||
|
}; |
||||||
|
|
||||||
|
} // namespace grpc
|
||||||
|
|
||||||
|
#endif // GRPC_TEST_CPP_METRICS_SERVER_H
|
||||||
@ -0,0 +1,52 @@ |
|||||||
|
#!/bin/bash |
||||||
|
# Copyright 2016, 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. |
||||||
|
|
||||||
|
set -ex |
||||||
|
|
||||||
|
git clone --recursive $EXTERNAL_GIT_ROOT |
||||||
|
cd grpc |
||||||
|
|
||||||
|
cd third_party/protobuf && ./autogen.sh && \ |
||||||
|
./configure && make -j4 && make check && make install && ldconfig |
||||||
|
|
||||||
|
cd ../.. && make -j4 && make install |
||||||
|
|
||||||
|
cd examples/cpp/helloworld |
||||||
|
|
||||||
|
make |
||||||
|
|
||||||
|
make clean |
||||||
|
|
||||||
|
cd ../../../examples/cpp/route_guide |
||||||
|
|
||||||
|
make |
||||||
|
|
||||||
|
make clean |
||||||
|
|
||||||
@ -0,0 +1,73 @@ |
|||||||
|
#!/bin/bash |
||||||
|
|
||||||
|
# Copyright 2016, 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. |
||||||
|
|
||||||
|
PROTO_DIR="src/proto/grpc/reflection/v1alpha" |
||||||
|
PROTO_FILE="reflection" |
||||||
|
HEADER_DIR="include/grpc++/ext" |
||||||
|
SRC_DIR="src/cpp/ext" |
||||||
|
INCLUDE_DIR="grpc++/ext" |
||||||
|
TMP_DIR="tmp" |
||||||
|
GRPC_PLUGIN="bins/opt/grpc_cpp_plugin" |
||||||
|
PROTOC=third_party/protobuf/src/protoc |
||||||
|
|
||||||
|
set -e |
||||||
|
|
||||||
|
TMP_DIR=${TMP_DIR}_${PROTO_FILE} |
||||||
|
|
||||||
|
cd $(dirname $0)/../../.. |
||||||
|
|
||||||
|
[ ! -d $HEADER_DIR ] && mkdir -p $HEADER_DIR || : |
||||||
|
[ ! -d $SRC_DIR ] && mkdir -p $SRC_DIR || : |
||||||
|
[ ! -d $TMP_DIR ] && mkdir -p $TMP_DIR || : |
||||||
|
|
||||||
|
$PROTOC -I$PROTO_DIR --cpp_out=$TMP_DIR ${PROTO_DIR}/${PROTO_FILE}.proto |
||||||
|
$PROTOC -I$PROTO_DIR --grpc_out=$TMP_DIR --plugin=protoc-gen-grpc=${GRPC_PLUGIN} ${PROTO_DIR}/${PROTO_FILE}.proto |
||||||
|
|
||||||
|
sed -i "s/\"${PROTO_FILE}.pb.h\"/<${INCLUDE_DIR/\//\\\/}\/${PROTO_FILE}.pb.h>/g" ${TMP_DIR}/${PROTO_FILE}.pb.cc |
||||||
|
sed -i "s/\"${PROTO_FILE}.pb.h\"/<${INCLUDE_DIR/\//\\\/}\/${PROTO_FILE}.pb.h>/g" ${TMP_DIR}/${PROTO_FILE}.grpc.pb.h |
||||||
|
sed -i "s/\"${PROTO_FILE}.pb.h\"/<${INCLUDE_DIR/\//\\\/}\/${PROTO_FILE}.pb.h>/g" ${TMP_DIR}/${PROTO_FILE}.grpc.pb.cc |
||||||
|
sed -i "s/\"${PROTO_FILE}.grpc.pb.h\"/<${INCLUDE_DIR/\//\\\/}\/${PROTO_FILE}.grpc.pb.h>/g" ${TMP_DIR}/${PROTO_FILE}.grpc.pb.cc |
||||||
|
|
||||||
|
/bin/cp LICENSE ${TMP_DIR}/TMP_LICENSE |
||||||
|
sed -i -e "s/./ &/" -e "s/.*/ \*&/" ${TMP_DIR}/TMP_LICENSE |
||||||
|
sed -i -r "\$a\ *\n *\/\n\n" ${TMP_DIR}/TMP_LICENSE |
||||||
|
|
||||||
|
sed -i -e "1s/^/ *\n/" -e "1s/^/\/*\n/" ${TMP_DIR}/*.pb.h |
||||||
|
sed -i -e "1s/^/ *\n/" -e "1s/^/\/*\n/" ${TMP_DIR}/*.pb.cc |
||||||
|
|
||||||
|
sed -i "2r ${TMP_DIR}/TMP_LICENSE" ${TMP_DIR}/*.pb.h |
||||||
|
sed -i "2r ${TMP_DIR}/TMP_LICENSE" ${TMP_DIR}/*.pb.cc |
||||||
|
|
||||||
|
/bin/mv ${TMP_DIR}/${PROTO_FILE}.pb.h ${HEADER_DIR} |
||||||
|
/bin/mv ${TMP_DIR}/${PROTO_FILE}.grpc.pb.h ${HEADER_DIR} |
||||||
|
/bin/mv ${TMP_DIR}/${PROTO_FILE}.pb.cc ${SRC_DIR} |
||||||
|
/bin/mv ${TMP_DIR}/${PROTO_FILE}.grpc.pb.cc ${SRC_DIR} |
||||||
|
/bin/rm -r $TMP_DIR |
||||||
@ -0,0 +1,40 @@ |
|||||||
|
#!/bin/bash |
||||||
|
# Copyright 2016, 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. |
||||||
|
|
||||||
|
set -ex |
||||||
|
|
||||||
|
# change to root directory |
||||||
|
cd $(dirname $0)/../.. |
||||||
|
|
||||||
|
# build grpc_check_generated_pb_files docker image |
||||||
|
docker build -t grpc_check_generated_pb_files tools/dockerfile/grpc_check_generated_pb_files |
||||||
|
|
||||||
|
# run check_pb_files against the checked out codebase |
||||||
|
docker run -e TEST=$TEST --rm=true -v ${HOST_GIT_ROOT:-`pwd`}:/var/local/jenkins/grpc -t grpc_check_generated_pb_files /var/local/jenkins/grpc/tools/dockerfile/grpc_check_generated_pb_files/check_pb_files.sh |
||||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue