1 // 2 // Copyright 2016 gRPC authors. 3 // 4 // Licensed under the Apache License, Version 2.0 (the "License"); 5 // you may not use this file except in compliance with the License. 6 // You may obtain a copy of the License at 7 // 8 // http://www.apache.org/licenses/LICENSE-2.0 9 // 10 // Unless required by applicable law or agreed to in writing, software 11 // distributed under the License is distributed on an "AS IS" BASIS, 12 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 // See the License for the specific language governing permissions and 14 // limitations under the License. 15 // 16 17 #ifndef GRPC_CORE_EXT_FILTERS_MESSAGE_SIZE_MESSAGE_SIZE_FILTER_H 18 #define GRPC_CORE_EXT_FILTERS_MESSAGE_SIZE_MESSAGE_SIZE_FILTER_H 19 20 #include <grpc/support/port_platform.h> 21 22 #include "src/core/ext/service_config/service_config_parser.h" 23 #include "src/core/lib/channel/channel_stack.h" 24 25 extern const grpc_channel_filter grpc_message_size_filter; 26 27 namespace grpc_core { 28 29 class MessageSizeParsedConfig : public ServiceConfigParser::ParsedConfig { 30 public: 31 struct message_size_limits { 32 int max_send_size; 33 int max_recv_size; 34 }; 35 MessageSizeParsedConfig(int max_send_size,int max_recv_size)36 MessageSizeParsedConfig(int max_send_size, int max_recv_size) { 37 limits_.max_send_size = max_send_size; 38 limits_.max_recv_size = max_recv_size; 39 } 40 limits()41 const message_size_limits& limits() const { return limits_; } 42 43 static const MessageSizeParsedConfig* GetFromCallContext( 44 const grpc_call_context_element* context); 45 46 private: 47 message_size_limits limits_; 48 }; 49 50 class MessageSizeParser : public ServiceConfigParser::Parser { 51 public: 52 std::unique_ptr<ServiceConfigParser::ParsedConfig> ParsePerMethodParams( 53 const grpc_channel_args* /*args*/, const Json& json, 54 grpc_error_handle* error) override; 55 56 static void Register(); 57 58 static size_t ParserIndex(); 59 }; 60 61 int GetMaxRecvSizeFromChannelArgs(const grpc_channel_args* args); 62 int GetMaxSendSizeFromChannelArgs(const grpc_channel_args* args); 63 64 } // namespace grpc_core 65 66 #endif /* GRPC_CORE_EXT_FILTERS_MESSAGE_SIZE_MESSAGE_SIZE_FILTER_H */ 67