| OLD | NEW |
| 1 /* Copyright 2013 The Native Client Authors. All rights reserved. | 1 /* Copyright 2013 The Native Client Authors. All rights reserved. |
| 2 * Use of this source code is governed by a BSD-style license that can | 2 * Use of this source code is governed by a BSD-style license that can |
| 3 * be found in the LICENSE file. | 3 * be found in the LICENSE file. |
| 4 */ | 4 */ |
| 5 | 5 |
| 6 //===-- pnacl-freeze.cpp - The low-level NaCl bitcode freezer --------===// | 6 //===-- pnacl-thaw.cpp - The low-level NaCl bitcode thawer ----------------===// |
| 7 // | 7 // |
| 8 //===----------------------------------------------------------------------===// | 8 //===----------------------------------------------------------------------===// |
| 9 // | 9 // |
| 10 // Generates NaCl pexe wire format. | 10 // Converts NaCl wire format back to LLVM bitcode. |
| 11 // | 11 // |
| 12 //===----------------------------------------------------------------------===// | 12 //===----------------------------------------------------------------------===// |
| 13 | 13 |
| 14 #include "llvm/IR/LLVMContext.h" | 14 #include "llvm/IR/LLVMContext.h" |
| 15 // Note: We need the following to provide the API for calling the NaCl | 15 // Note: We need the following to provide the API for calling the NaCl |
| 16 // Bitcode Writer to generate the frozen file. | 16 // Bitcode Reader to read the frozen file. |
| 17 #include "llvm/Bitcode/NaCl/NaClReaderWriter.h" | 17 #include "llvm/Bitcode/NaCl/NaClReaderWriter.h" |
| 18 // Note: We need the following to provide the API for calling the (LLVM) | 18 // Note: We need the following to provide the API for calling the (LLVM) |
| 19 // Bitcode Reader to read in the corresonding pexe file to freeze. | 19 // Bitcode Writer to generate the corresponding LLVM bitcode file. |
| 20 #include "llvm/Bitcode/ReaderWriter.h" | 20 #include "llvm/Bitcode/ReaderWriter.h" |
| 21 #include "llvm/IR/Module.h" | 21 #include "llvm/IR/Module.h" |
| 22 #include "llvm/Support/CommandLine.h" | 22 #include "llvm/Support/CommandLine.h" |
| 23 #include "llvm/Support/DataStream.h" | 23 #include "llvm/Support/DataStream.h" |
| 24 #include "llvm/Support/ManagedStatic.h" | 24 #include "llvm/Support/ManagedStatic.h" |
| 25 #include "llvm/Support/PrettyStackTrace.h" | 25 #include "llvm/Support/PrettyStackTrace.h" |
| 26 #include "llvm/Support/Signals.h" | 26 #include "llvm/Support/Signals.h" |
| 27 #include "llvm/Support/ToolOutputFile.h" | 27 #include "llvm/Support/ToolOutputFile.h" |
| 28 | 28 |
| 29 using namespace llvm; | 29 using namespace llvm; |
| 30 | 30 |
| 31 | |
| 32 static cl::opt<std::string> | 31 static cl::opt<std::string> |
| 33 OutputFilename("o", cl::desc("Specify output filename"), | 32 OutputFilename("o", cl::desc("Specify thawed pexe filename"), |
| 34 cl::value_desc("filename")); | 33 cl::value_desc("filename")); |
| 35 | 34 |
| 36 static cl::opt<std::string> | 35 static cl::opt<std::string> |
| 37 InputFilename(cl::Positional, cl::desc("<pexe file>"), cl::Required); | 36 InputFilename(cl::Positional, cl::desc("<frozen file>"), cl::Required); |
| 38 | 37 |
| 39 static void WriteOutputFile(const Module *M) { | 38 static void WriteOutputFile(const Module *M) { |
| 40 | 39 |
| 41 std::string FrozenFilename = | 40 std::string ThawedFilename = |
| 42 (OutputFilename.size() == 0 ? (InputFilename + ".frozen") : OutputFilename); | 41 (OutputFilename.size() == 0 ? (InputFilename + ".thawed") : OutputFilename); |
| 43 | 42 |
| 44 std::string ErrorInfo; | 43 std::string ErrorInfo; |
| 45 OwningPtr<tool_output_file> Out | 44 OwningPtr<tool_output_file> Out |
| 46 (new tool_output_file(FrozenFilename.c_str(), ErrorInfo, | 45 (new tool_output_file(ThawedFilename.c_str(), ErrorInfo, |
| 47 raw_fd_ostream::F_Binary)); | 46 raw_fd_ostream::F_Binary)); |
| 48 if (!ErrorInfo.empty()) { | 47 if (!ErrorInfo.empty()) { |
| 49 errs() << ErrorInfo << '\n'; | 48 errs() << ErrorInfo << '\n'; |
| 50 exit(1); | 49 exit(1); |
| 51 } | 50 } |
| 52 | 51 |
| 53 NaClWriteBitcodeToFile(M, Out->os()); | 52 WriteBitcodeToFile(M, Out->os()); |
| 54 | 53 |
| 55 // Declare success. | 54 // Declare success. |
| 56 Out->keep(); | 55 Out->keep(); |
| 57 } | 56 } |
| 58 | 57 |
| 59 int main(int argc, char **argv) { | 58 int main(int argc, char **argv) { |
| 60 // Print a stack trace if we signal out. | 59 // Print a stack trace if we signal out. |
| 61 sys::PrintStackTraceOnErrorSignal(); | 60 sys::PrintStackTraceOnErrorSignal(); |
| 62 PrettyStackTraceProgram X(argc, argv); | 61 PrettyStackTraceProgram X(argc, argv); |
| 63 | 62 |
| 64 LLVMContext &Context = getGlobalContext(); | 63 LLVMContext &Context = getGlobalContext(); |
| 65 llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. | 64 llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. |
| 66 | 65 |
| 67 cl::ParseCommandLineOptions(argc, argv, "Generates NaCl pexe wire format\n"); | 66 cl::ParseCommandLineOptions( |
| 67 argc, argv, "Converts NaCl pexe wire format into LLVM bitcode format\n"); |
| 68 | 68 |
| 69 std::string ErrorMessage; | 69 std::string ErrorMessage; |
| 70 std::auto_ptr<Module> M; | 70 std::auto_ptr<Module> M; |
| 71 | 71 |
| 72 // Use the bitcode streaming interface | 72 // Use the bitcode streaming interface |
| 73 DataStreamer *streamer = getDataFileStreamer(InputFilename, &ErrorMessage); | 73 DataStreamer *streamer = getDataFileStreamer(InputFilename, &ErrorMessage); |
| 74 if (streamer) { | 74 if (streamer) { |
| 75 std::string DisplayFilename = InputFilename; | 75 std::string DisplayFilename = InputFilename; |
| 76 M.reset(getStreamedBitcodeModule(DisplayFilename, streamer, Context, | 76 M.reset(getNaClStreamedBitcodeModule(DisplayFilename, streamer, Context, |
| 77 &ErrorMessage)); | 77 &ErrorMessage)); |
| 78 if(M.get() != 0 && M->MaterializeAllPermanently(&ErrorMessage)) { | 78 if(M.get() != 0 && M->MaterializeAllPermanently(&ErrorMessage)) { |
| 79 M.reset(); | 79 M.reset(); |
| 80 } | 80 } |
| 81 } | 81 } |
| 82 | 82 |
| 83 if (M.get() == 0) { | 83 if (M.get() == 0) { |
| 84 errs() << argv[0] << ": "; | 84 errs() << argv[0] << ": "; |
| 85 if (ErrorMessage.size()) | 85 if (ErrorMessage.size()) |
| 86 errs() << ErrorMessage << "\n"; | 86 errs() << ErrorMessage << "\n"; |
| 87 else | 87 else |
| 88 errs() << "bitcode didn't read correctly.\n"; | 88 errs() << "bitcode didn't read correctly.\n"; |
| 89 return 1; | 89 return 1; |
| 90 } | 90 } |
| 91 | 91 |
| 92 WriteOutputFile(M.get()); | 92 WriteOutputFile(M.get()); |
| 93 return 0; | 93 return 0; |
| 94 } | 94 } |
| OLD | NEW |