Computer Laboratory

cat.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 "Automaton.h"
33 #include "Manifest.h"
34 
35 #include "tesla.pb.h"
36 
37 #include <google/protobuf/text_format.h>
38 
39 #include <llvm/IR/Function.h>
40 #include <llvm/IR/Instructions.h>
41 #include <llvm/IR/LLVMContext.h>
42 #include <llvm/IR/Module.h>
43 #include <llvm/Support/CommandLine.h>
44 #include <llvm/Support/raw_ostream.h>
45 #include <llvm/Pass.h>
46 
47 
48 using namespace llvm;
49 using namespace tesla;
50 
51 using std::string;
52 
53 cl::list<string> InputFiles(cl::desc("<input files>"),
54  cl::Positional, cl::OneOrMore);
55 
56 cl::opt<string> OutputFile("o", cl::desc("<output file>"), cl::init("-"));
57 
58 
59 int
60 main(int argc, char *argv[]) {
61  cl::ParseCommandLineOptions(argc, argv);
62 
63  auto& err = llvm::errs();
64 
65  ManifestFile Result;
66 
67  for (auto& Filename : InputFiles) {
68  OwningPtr<Manifest> Manifest(Manifest::load(llvm::errs(),
69  Automaton::Unlinked,
70  Filename));
71  if (!Manifest) {
72  err << "Unable to read manifest '" << Filename << "'\n";
73  return 1;
74  }
75 
76  for (auto i : Manifest->AllAutomata())
77  *Result.add_automaton() = *i.second;
78 
79  for (auto i : Manifest->RootAutomata())
80  *Result.add_root() = *i;
81  }
82 
83  string ProtobufText;
84  google::protobuf::TextFormat::PrintToString(Result, &ProtobufText);
85 
86  bool UseFile = (OutputFile != "-");
87  OwningPtr<raw_fd_ostream> outfile;
88 
89  if (UseFile) {
90  string OutErrorInfo;
91  outfile.reset(new raw_fd_ostream(OutputFile.c_str(), OutErrorInfo));
92  }
93  raw_ostream& out = UseFile ? *outfile : llvm::outs();
94  out << ProtobufText;
95 
96  google::protobuf::ShutdownProtobufLibrary();
97 
98  return 0;
99 }
100