Computer Laboratory

AST.cpp
Go to the documentation of this file.
1 
2 /*
3  * Copyright (c) 2013 Jonathan Anderson
4  * All rights reserved.
5  *
6  * This software was developed by SRI International and the University of
7  * Cambridge Computer Laboratory under DARPA/AFRL contract (FA8750-10-C-0237)
8  * ("CTSRD"), as part of the DARPA CRASH research programme.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  * notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  * notice, this list of conditions and the following disclaimer in the
17  * documentation and/or other materials provided with the distribution.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #include "AST.h"
33 #include "Debug.h"
34 #include "Parser.h"
35 #include "Visitor.h"
36 
37 #include <clang/AST/ASTContext.h>
38 #include <clang/Tooling/Tooling.h>
39 
40 #include <llvm/Support/raw_ostream.h>
41 
42 #include <google/protobuf/text_format.h>
43 
44 #include <string>
45 
46 using namespace clang;
47 using namespace tesla;
48 using std::string;
49 
50 
51 namespace tesla {
52 
53 TeslaConsumer::TeslaConsumer(llvm::StringRef In, llvm::StringRef Out)
54  : InFile(In), OutFile(Out)
55 {
56 }
57 
58 void TeslaConsumer::HandleTranslationUnit(ASTContext &Context) {
59  TeslaVisitor Visitor(InFile, &Context);
60 
61  if (!Visitor.TraverseDecl(Context.getTranslationUnitDecl()))
62  panic("error analysing '" + InFile + "'");
63 
64  string ErrorInfo;
65  llvm::raw_fd_ostream Out(OutFile.str().c_str(), ErrorInfo);
66  if (Out.has_error())
67  panic("unable to open '" + OutFile + "': " + ErrorInfo);
68 
69  ManifestFile Result;
70  for (const AutomatonDescription *A : Visitor.GetAutomata())
71  *Result.add_automaton() = *A;
72 
73  for (const Usage *U : Visitor.RootAutomata())
74  *Result.add_root() = *U;
75 
76  string ProtobufText;
77  google::protobuf::TextFormat::PrintToString(Result, &ProtobufText);
78  Out << ProtobufText;
79 }
80 
81 
82 ASTConsumer* TeslaAction::CreateASTConsumer(CompilerInstance &C,
83  llvm::StringRef InFile)
84 {
85  return new TeslaConsumer(InFile, OutFile);
86 }
87 
88 
89 FrontendAction* TeslaActionFactory::create() {
90  return new TeslaAction(OutFile);
91 }
92 
93 } // namespace tesla
94