OSX/libsecurity_codesigning/lib/antlrplugin.cpp

   1 /*
   2  * Copyright (c) 2007,2011-2012 Apple Inc. All Rights Reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 #include "antlrplugin.h"
  25 #include "cserror.h"
  26 #include "RequirementLexer.hpp"
  27 #include "RequirementParser.hpp"
  28 #include <antlr/TokenStreamException.hpp>
  29
  30
  31 namespace Security {
  32 namespace CodeSigning {
  33
  34 namespace Parser = Security_CodeSigning;
  35
  36
  37 //
  38 // Lexer input adapters
  39 //
  40 class StdioInputStream : public antlr::InputBuffer {
  41 public:
  42         StdioInputStream(FILE *fp) : mFile(fp) { }
  43         int getChar() { return fgetc(mFile); }
  44
  45 private:
  46         FILE *mFile;
  47 };
  48
  49 class StringInputStream : public antlr::InputBuffer {
  50 public:
  51         StringInputStream(const string &s) : mInput(s), mPos(mInput.begin()) { }
  52         int getChar() { return (mPos == mInput.end()) ? EOF : static_cast<unsigned char>(*mPos++); }
  53
  54 private:
  55         string mInput;
  56         string::const_iterator mPos;
  57 };
  58
  59
  60 //
  61 // Generic parser driver
  62 //
  63 template <class Input, class Source, class Result>
  64 const Result *parse(Source source, Result *(Parser::RequirementParser::*rule)(), std::string &errors)
  65 {
  66         Input input(source);
  67         Parser::RequirementLexer lexer(input);
  68         Parser::RequirementParser parser(lexer);
  69         try {
  70                 const Result *result = (parser.*rule)();
  71                 errors = parser.errors;
  72                 if (errors.empty())
  73                         return result;
  74                 else
  75                         ::free((void *)result);
  76         } catch (const antlr::TokenStreamException &ex) {
  77                 errors = ex.toString() + "\n";
  78         }
  79         return NULL;                    // signal failure
  80 }
  81
  82
  83 //
  84 // Hook up each supported parsing action to the plugin interface
  85 //
  86 static
  87 const Requirement *fileRequirement(FILE *source, string &errors)
  88 { return parse<StdioInputStream>(source, &Parser::RequirementParser::requirement, errors); }
  89
  90 static
  91 const Requirement *stringRequirement(string source, string &errors)
  92 { return parse<StringInputStream>(source, &Parser::RequirementParser::requirement, errors); }
  93
  94 static
  95 const Requirements *fileRequirements(FILE *source, string &errors)
  96 { return parse<StdioInputStream>(source, &Parser::RequirementParser::requirementSet, errors); }
  97
  98 static
  99 const Requirements *stringRequirements(string source, string &errors)
 100 { return parse<StringInputStream>(source, &Parser::RequirementParser::requirementSet, errors); }
 101
 102 static
 103 const BlobCore *fileGeneric(FILE *source, string &errors)
 104 { return parse<StdioInputStream>(source, &Parser::RequirementParser::autosense, errors); }
 105
 106 static
 107 const BlobCore *stringGeneric(string source, string &errors)
 108 { return parse<StringInputStream>(source, &Parser::RequirementParser::autosense, errors); }
 109
 110
 111 //
 112 // Basic plugin hookup
 113 //
 114 static AntlrPlugin plugin = {
 115         fileRequirement,
 116         fileRequirements,
 117         fileGeneric,
 118         stringRequirement,
 119         stringRequirements,
 120         stringGeneric
 121 };
 122
 123 AntlrPlugin *findAntlrPlugin()
 124 {
 125         return &plugin;
 126 }
 127
 128
 129 } // end namespace CodeSigning
 130 } // end namespace Security