hanchenye-llvm-project/lld/utils/linker-script-test/linker-script-test.cpp

56 lines
1.5 KiB
C++

//===- utils/linker-script-test/linker-script-test.cpp --------------------===//
//
// The LLVM Linker
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
///
/// \file
/// \brief Tool for testing linker script parsing.
///
//===----------------------------------------------------------------------===//
#include "lld/ReaderWriter/LinkerScript.h"
#include "llvm/Support/PrettyStackTrace.h"
#include "llvm/Support/Signals.h"
using namespace llvm;
using namespace lld;
using namespace script;
int main(int argc, const char **argv) {
llvm::sys::PrintStackTraceOnErrorSignal();
llvm::PrettyStackTraceProgram X(argc, argv);
{
llvm::OwningPtr<MemoryBuffer> mb;
if (error_code ec = MemoryBuffer::getFileOrSTDIN(argv[1], mb)) {
llvm::errs() << ec.message() << "\n";
return 1;
}
Lexer l(std::unique_ptr<MemoryBuffer>(mb.take()));
Token tok;
while (true) {
l.lex(tok);
tok.dump(llvm::outs());
if (tok._kind == Token::eof || tok._kind == Token::unknown)
break;
}
}
{
llvm::OwningPtr<MemoryBuffer> mb;
if (error_code ec = MemoryBuffer::getFileOrSTDIN(argv[1], mb)) {
llvm::errs() << ec.message() << "\n";
return 1;
}
Lexer l(std::unique_ptr<MemoryBuffer>(mb.take()));
Parser p(l);
LinkerScript *ls = p.parse();
if (ls)
ls->dump(llvm::outs());
}
}