diff options
Diffstat (limited to 'contrib/protoc-bnet/BnetServiceGenerator.cpp')
| -rw-r--r-- | contrib/protoc-bnet/BnetServiceGenerator.cpp | 312 |
1 files changed, 312 insertions, 0 deletions
diff --git a/contrib/protoc-bnet/BnetServiceGenerator.cpp b/contrib/protoc-bnet/BnetServiceGenerator.cpp new file mode 100644 index 00000000000..a47df61f6bf --- /dev/null +++ b/contrib/protoc-bnet/BnetServiceGenerator.cpp @@ -0,0 +1,312 @@ +// +// Created by tea on 10.03.16. +// + +#include "BnetServiceGenerator.h" +#include "method_options.pb.h" +#include "service_options.pb.h" +#include <google/protobuf/descriptor.h> +#include <google/protobuf/io/printer.h> +#include <google/protobuf/stubs/strutil.h> +#include <google/protobuf/compiler/cpp/cpp_helpers.h> +#include "google/protobuf/compiler/cpp/cpp_options.h" + +BnetServiceGenerator::BnetServiceGenerator(pb::ServiceDescriptor const* descriptor, pbcpp::Options const& options) : descriptor_(descriptor) +{ + vars_["classname"] = descriptor_->name(); + vars_["full_name"] = descriptor_->full_name(); + if (options.dllexport_decl.empty()) + vars_["dllexport"] = ""; + else + vars_["dllexport"] = options.dllexport_decl + " "; + + vars_["original_hash"] = " typedef std::integral_constant<uint32, 0x" + pb::ToUpper(pb::ToHex(HashServiceName(descriptor_->options().GetExtension(Battlenet::original_fully_qualified_descriptor_name)))) + "u> OriginalHash;\n"; + vars_["name_hash"] = " typedef std::integral_constant<uint32, 0x" + pb::ToUpper(pb::ToHex(HashServiceName(descriptor_->full_name()))) + "u> NameHash;\n"; +} + +BnetServiceGenerator::~BnetServiceGenerator() { } + +void BnetServiceGenerator::GenerateDeclarations(pb::io::Printer* printer) +{ + GenerateInterface(printer); +} + +void BnetServiceGenerator::GenerateInterface(pb::io::Printer* printer) +{ + printer->Print(vars_, + "class $dllexport$$classname$ : public ServiceBase\n" + "{\n" + " public:\n" + "\n" + " template<typename HashSelector>\n" + " explicit $classname$(HashSelector) : service_hash_(HashSelector::Result::value) { }\n" + "\n" + "$original_hash$" + "$name_hash$"); + + printer->Indent(); + + printer->Print(vars_, + "\n" + "static google::protobuf::ServiceDescriptor const* descriptor();\n" + "\n" + "// client methods --------------------------------------------------\n" + "\n"); + + GenerateClientMethodSignatures(printer); + + printer->Print( + "// server methods --------------------------------------------------\n" + "\n" + "void CallServerMethod(uint32 token, uint32 methodId, MessageBuffer buffer) override final;\n" + "\n"); + + printer->Outdent(); + + printer->Print(" protected:\n "); + + printer->Indent(); + + GenerateServerMethodSignatures(printer); + + printer->Outdent(); + + printer->Print(vars_, + "\n" + " private:\n" + " uint32 service_hash_;\n" + "\n" + " GOOGLE_DISALLOW_EVIL_CONSTRUCTORS($classname$);\n" + "};\n"); +} + +void BnetServiceGenerator::GenerateClientMethodSignatures(pb::io::Printer* printer) +{ + for (int i = 0; i < descriptor_->method_count(); i++) + { + pb::MethodDescriptor const* method = descriptor_->method(i); + if (!method->options().HasExtension(Battlenet::method_id)) + continue; + + std::map<std::string, std::string> sub_vars; + sub_vars["name"] = method->name(); + sub_vars["full_name"] = descriptor_->name() + "." + method->name(); + sub_vars["method_id"] = pb::SimpleItoa(method->options().GetExtension(Battlenet::method_id)); + sub_vars["input_type"] = pbcpp::ClassName(method->input_type(), true); + sub_vars["output_type"] = pbcpp::ClassName(method->output_type(), true); + sub_vars["input_type_name"] = method->input_type()->full_name(); + + if (method->output_type()->name() != "NO_RESPONSE") + printer->Print(sub_vars, "void $name$($input_type$ const* request, std::function<void($output_type$ const*)> responseCallback);\n"); + else + printer->Print(sub_vars, "void $name$($input_type$ const* request);\n"); + } +} + +void BnetServiceGenerator::GenerateServerMethodSignatures(pb::io::Printer* printer) +{ + for (int i = 0; i < descriptor_->method_count(); i++) + { + pb::MethodDescriptor const* method = descriptor_->method(i); + if (!method->options().HasExtension(Battlenet::method_id)) + continue; + + std::map<std::string, std::string> sub_vars; + sub_vars["name"] = method->name(); + sub_vars["input_type"] = pbcpp::ClassName(method->input_type(), true); + sub_vars["output_type"] = pbcpp::ClassName(method->output_type(), true); + + if (method->output_type()->name() != "NO_RESPONSE") + printer->Print(sub_vars, "virtual uint32 Handle$name$($input_type$ const* request, $output_type$* response);\n"); + else + printer->Print(sub_vars, "virtual uint32 Handle$name$($input_type$ const* request);\n"); + } +} + +// =================================================================== + +void BnetServiceGenerator::GenerateDescriptorInitializer(pb::io::Printer* printer, int index) +{ + std::map<std::string, std::string> vars; + vars["classname"] = descriptor_->name(); + vars["index"] = pb::SimpleItoa(index); + + printer->Print(vars, "$classname$_descriptor_ = file->service($index$);\n"); +} + +// =================================================================== + +void BnetServiceGenerator::GenerateImplementation(pb::io::Printer* printer) +{ + printer->Print(vars_, + "google::protobuf::ServiceDescriptor const* $classname$::descriptor() {\n" + " protobuf_AssignDescriptorsOnce();\n" + " return $classname$_descriptor_;\n" + "}\n" + "\n"); + + GenerateClientMethodImplementations(printer); + GenerateServerCallMethod(printer); + GenerateServerImplementations(printer); +} + +void BnetServiceGenerator::GenerateClientMethodImplementations(pb::io::Printer* printer) +{ + for (int i = 0; i < descriptor_->method_count(); i++) + { + pb::MethodDescriptor const* method = descriptor_->method(i); + if (!method->options().HasExtension(Battlenet::method_id)) + continue; + + std::map<std::string, std::string> sub_vars; + sub_vars["classname"] = vars_["classname"]; + sub_vars["name"] = method->name(); + sub_vars["full_name"] = descriptor_->name() + "." + method->name(); + sub_vars["method_id"] = pb::SimpleItoa(method->options().GetExtension(Battlenet::method_id)); + sub_vars["input_type"] = pbcpp::ClassName(method->input_type(), true); + sub_vars["output_type"] = pbcpp::ClassName(method->output_type(), true); + sub_vars["input_type_name"] = method->input_type()->full_name(); + + if (method->output_type()->name() != "NO_RESPONSE") + { + printer->Print(sub_vars, + "void $classname$::$name$($input_type$ const* request, std::function<void($output_type$ const*)> responseCallback) { \n" + " TC_LOG_DEBUG(\"service.protobuf\", \"%s Server called client method $full_name$($input_type_name${ %s })\",\n" + " GetCallerInfo().c_str(), request->ShortDebugString().c_str());\n" + " SendRequest(service_hash_, $method_id$, request, [callback{ std::move(responseCallback) }](MessageBuffer buffer) {\n" + " $output_type$ response;\n" + " if (response.ParseFromArray(buffer.GetReadPointer(), buffer.GetActiveSize()))\n" + " callback(&response);\n" + " });\n" + "}\n" + "\n"); + } + else + { + printer->Print(sub_vars, + "void $classname$::$name$($input_type$ const* request) { \n" + " TC_LOG_DEBUG(\"service.protobuf\", \"%s Server called client method $full_name$($input_type_name${ %s })\",\n" + " GetCallerInfo().c_str(), request->ShortDebugString().c_str());\n" + " SendRequest(service_hash_, $method_id$, request);\n" + "}\n" + "\n"); + } + } +} + +void BnetServiceGenerator::GenerateServerCallMethod(pb::io::Printer* printer) +{ + printer->Print(vars_, + "void $classname$::CallServerMethod(uint32 token, uint32 methodId, MessageBuffer buffer) {\n" + " switch(methodId) {\n"); + + for (int i = 0; i < descriptor_->method_count(); i++) + { + pb::MethodDescriptor const* method = descriptor_->method(i); + if (!method->options().HasExtension(Battlenet::method_id)) + continue; + + std::map<std::string, std::string> sub_vars; + sub_vars["name"] = method->name(); + sub_vars["full_name"] = descriptor_->name() + "." + method->name(); + sub_vars["method_id"] = pb::SimpleItoa(method->options().GetExtension(Battlenet::method_id)); + sub_vars["input_type"] = pbcpp::ClassName(method->input_type(), true); + sub_vars["output_type"] = pbcpp::ClassName(method->output_type(), true); + sub_vars["input_type_name"] = method->input_type()->full_name(); + sub_vars["output_type_name"] = method->output_type()->full_name(); + + printer->Print(sub_vars, + " case $method_id$: {\n" + " $input_type$ request;\n" + " if (!request.ParseFromArray(buffer.GetReadPointer(), buffer.GetActiveSize())) {\n" + " TC_LOG_DEBUG(\"service.protobuf\", \"%s Failed to parse request for $full_name$ server method call.\", GetCallerInfo().c_str());\n" + " SendResponse(service_hash_, $method_id$, token, ERROR_RPC_MALFORMED_REQUEST);\n" + " return;\n" + " }\n" + "\n" + ); + + if (method->output_type()->name() != "NO_RESPONSE") + { + printer->Print(sub_vars, + " $output_type$ response;\n" + " uint32 status = Handle$name$(&request, &response);\n" + " TC_LOG_DEBUG(\"service.protobuf\", \"%s Client called server method $full_name$($input_type_name${ %s }) returned $output_type_name${ %s } status %u.\",\n" + " GetCallerInfo().c_str(), request.ShortDebugString().c_str(), response.ShortDebugString().c_str(), status);\n" + " if (!status)\n" + " SendResponse(service_hash_, $method_id$, token, &response);\n" + " else\n" + " SendResponse(service_hash_, $method_id$, token, status);\n"); + } + else + { + printer->Print(sub_vars, + " uint32 status = Handle$name$(&request);\n" + " TC_LOG_DEBUG(\"service.protobuf\", \"%s Client called server method $full_name$($input_type_name${ %s }) status %u.\",\n" + " GetCallerInfo().c_str(), request.ShortDebugString().c_str(), status);\n" + " if (status)\n" + " SendResponse(service_hash_, $method_id$, token, status);\n"); + } + + printer->Print(sub_vars, + " break;\n" + " }\n"); + } + + printer->Print(vars_, + " default:\n" + " TC_LOG_ERROR(\"service.protobuf\", \"Bad method id %u.\", methodId);\n" + " SendResponse(service_hash_, methodId, token, ERROR_RPC_INVALID_METHOD);\n" + " break;\n" + " }\n" + "}\n" + "\n"); +} + +void BnetServiceGenerator::GenerateServerImplementations(pb::io::Printer* printer) +{ + for (int i = 0; i < descriptor_->method_count(); i++) + { + pb::MethodDescriptor const* method = descriptor_->method(i); + if (!method->options().HasExtension(Battlenet::method_id)) + continue; + + std::map<std::string, std::string> sub_vars; + sub_vars["classname"] = vars_["classname"]; + sub_vars["name"] = method->name(); + sub_vars["full_name"] = descriptor_->name() + "." + method->name(); + sub_vars["input_type"] = pbcpp::ClassName(method->input_type(), true); + sub_vars["output_type"] = pbcpp::ClassName(method->output_type(), true); + + if (method->output_type()->name() != "NO_RESPONSE") + { + printer->Print(sub_vars, "uint32 $classname$::Handle$name$($input_type$ const* request, $output_type$* response) {\n" + " TC_LOG_ERROR(\"service.protobuf\", \"%s Client tried to call not implemented method $full_name$({ %s })\",\n" + " GetCallerInfo().c_str(), request->ShortDebugString().c_str());\n" + " return ERROR_RPC_NOT_IMPLEMENTED;\n" + "}\n" + "\n"); + } + else + { + printer->Print(sub_vars, "uint32 $classname$::Handle$name$($input_type$ const* request) {\n" + " TC_LOG_ERROR(\"service.protobuf\", \"%s Client tried to call not implemented method $full_name$({ %s })\",\n" + " GetCallerInfo().c_str(), request->ShortDebugString().c_str());\n" + " return ERROR_RPC_NOT_IMPLEMENTED;\n" + "}\n" + "\n"); + } + } +} + +std::uint32_t BnetServiceGenerator::HashServiceName(std::string const& name) +{ + std::uint32_t hash = 0x811C9DC5; + for (std::size_t i = 0; i < name.length(); ++i) + { + hash ^= name[i]; + hash *= 0x1000193; + } + + return hash; +} |