15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Protocol Buffers - Google's data interchange format 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright 2008 Google Inc. All rights reserved. 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// http://code.google.com/p/protobuf/ 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Redistribution and use in source and binary forms, with or without 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// modification, are permitted provided that the following conditions are 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// met: 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions of source code must retain the above copyright 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// notice, this list of conditions and the following disclaimer. 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Redistributions in binary form must reproduce the above 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// copyright notice, this list of conditions and the following disclaimer 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// in the documentation and/or other materials provided with the 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// distribution. 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// * Neither the name of Google Inc. nor the names of its 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// contributors may be used to endorse or promote products derived from 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// this software without specific prior written permission. 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Author: kenton@google.com (Kenton Varda) 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/compiler/plugin.h> 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <iostream> 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <set> 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _WIN32 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <io.h> 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <fcntl.h> 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef STDIN_FILENO 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define STDIN_FILENO 0 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef STDOUT_FILENO 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define STDOUT_FILENO 1 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#else 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <unistd.h> 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/stubs/common.h> 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/compiler/plugin.pb.h> 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/compiler/code_generator.h> 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/descriptor.h> 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <google/protobuf/io/zero_copy_stream_impl.h> 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace google { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace protobuf { 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace compiler { 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class GeneratorResponseContext : public GeneratorContext { 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GeneratorResponseContext(CodeGeneratorResponse* response, 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const vector<const FileDescriptor*>& parsed_files) 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : response_(response), 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parsed_files_(parsed_files) {} 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~GeneratorResponseContext() {} 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // implements GeneratorContext -------------------------------------- 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual io::ZeroCopyOutputStream* Open(const string& filename) { 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CodeGeneratorResponse::File* file = response_->add_file(); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file->set_name(filename); 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new io::StringOutputStream(file->mutable_content()); 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual io::ZeroCopyOutputStream* OpenForInsert( 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string& filename, const string& insertion_point) { 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CodeGeneratorResponse::File* file = response_->add_file(); 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file->set_name(filename); 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file->set_insertion_point(insertion_point); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new io::StringOutputStream(file->mutable_content()); 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ListParsedFiles(vector<const FileDescriptor*>* output) { 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) *output = parsed_files_; 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CodeGeneratorResponse* response_; 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const vector<const FileDescriptor*>& parsed_files_; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)int PluginMain(int argc, char* argv[], const CodeGenerator* generator) { 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (argc > 1) { 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cerr << argv[0] << ": Unknown option: " << argv[1] << endl; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _WIN32 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _setmode(STDIN_FILENO, _O_BINARY); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) _setmode(STDOUT_FILENO, _O_BINARY); 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CodeGeneratorRequest request; 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!request.ParseFromFileDescriptor(STDIN_FILENO)) { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cerr << argv[0] << ": protoc sent unparseable request to plugin." << endl; 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DescriptorPool pool; 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < request.proto_file_size(); i++) { 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FileDescriptor* file = pool.BuildFile(request.proto_file(i)); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (file == NULL) { 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // BuildFile() already wrote an error message. 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) vector<const FileDescriptor*> parsed_files; 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < request.file_to_generate_size(); i++) { 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) parsed_files.push_back(pool.FindFileByName(request.file_to_generate(i))); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (parsed_files.back() == NULL) { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cerr << argv[0] << ": protoc asked plugin to generate a file but " 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "did not provide a descriptor for the file: " 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << request.file_to_generate(i) << endl; 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CodeGeneratorResponse response; 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GeneratorResponseContext context(&response, parsed_files); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (int i = 0; i < parsed_files.size(); i++) { 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const FileDescriptor* file = parsed_files[i]; 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string error; 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool succeeded = generator->Generate( 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file, request.parameter(), &context, &error); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!succeeded && error.empty()) { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error = "Code generator returned false but provided no error " 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "description."; 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!error.empty()) { 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) response.set_error(file->name() + ": " + error); 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) break; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!response.SerializeToFileDescriptor(STDOUT_FILENO)) { 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) cerr << argv[0] << ": Error writing to stdout." << endl; 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 1; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return 0; 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace compiler 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace protobuf 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace google 164