RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-11-23 11:13:28 +01:00
Code Issues Projects Releases Wiki Activity

llvm-mc: Honor -o option (and add -f).

Browse Source 
llvm-svn: 78640
This commit is contained in:
Daniel Dunbar 2009-08-11 04:34:48 +00:00
parent 9c03cc13de
commit 712f6f40ed
1 changed files with 35 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
tools/llvm-mc/llvm-mc.cpp
View File

@ -36,6 +36,9 @@ static cl::opt<std::string>
OutputFilename("o", cl::desc("Output filename"),
cl::value_desc("filename"));
static cl::opt<bool>
Force("f", cl::desc("Overwrite output files"));
static cl::list<std::string>
IncludeDirs("I", cl::desc("Directory of include files"),
cl::value_desc("directory"), cl::Prefix);
@ -161,6 +164,28 @@ static TargetAsmParser *GetTargetAsmParser(const char *ProgName,
return 0;
}
static raw_ostream *GetOutputStream() {
if (OutputFilename == "" || OutputFilename == "-")
return &outs();
// Make sure that the Out file gets unlinked from the disk if we get a
// SIGINT
sys::RemoveFileOnSignal(sys::Path(OutputFilename));
std::string Err;
raw_fd_ostream *Out = new raw_fd_ostream(OutputFilename.c_str(),
/*Binary=*/false, Force, Err);
if (!Err.empty()) {
errs() << Err << '\n';
if (!Force)
errs() << "Use -f command line argument to force output\n";
delete Out;
return 0;
}
return Out;
}
static int AssembleInput(const char *ProgName) {
std::string Error;
MemoryBuffer *Buffer = MemoryBuffer::getFileOrSTDIN(InputFilename, &Error);
@ -183,7 +208,10 @@ static int AssembleInput(const char *ProgName) {
SrcMgr.setIncludeDirs(IncludeDirs);
MCContext Ctx;
OwningPtr<MCStreamer> Str(createAsmStreamer(Ctx, outs()));
raw_ostream *Out = GetOutputStream();
if (!Out)
return 1;
OwningPtr<MCStreamer> Str(createAsmStreamer(Ctx, *Out));
// FIXME: Target hook & command line option for initial section.
Str.get()->SwitchSection(MCSectionMachO::Create("__TEXT","__text",
@ -196,7 +224,12 @@ static int AssembleInput(const char *ProgName) {
if (!TAP)
return 1;
Parser.setTargetParser(*TAP.get());
return Parser.Run();
int Res = Parser.Run();
if (Out != &outs())
delete Out;
return Res;
}