8b14d14209
Module-definition (.def) files are the file containing linker directives, such as export symbols. Because link.exe supports the same features as command line options, just as some Linker Script commands overlaps with command line options, use of module-definition file is not really necessary. It provides an alternative way to specify some linker options. This patch implements EXPORTS directive. Other directives will be implemented in the future. llvm-svn: 198925
120 lines
3.1 KiB
C++
120 lines
3.1 KiB
C++
//===- lib/Driver/WinLinkModuleDef.cpp ------------------------------------===//
|
|
//
|
|
// The LLVM Linker
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
///
|
|
/// \file
|
|
/// \brief Windows module definition file parser.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "lld/Driver/WinLinkModuleDef.h"
|
|
#include "llvm/ADT/StringSwitch.h"
|
|
|
|
namespace lld {
|
|
namespace moduledef {
|
|
|
|
Token Lexer::lex() {
|
|
_buffer = _buffer.trim();
|
|
if (_buffer.empty() || _buffer[0] == '\0')
|
|
return Token(Kind::eof, _buffer);
|
|
|
|
switch (_buffer[0]) {
|
|
case '=':
|
|
_buffer = _buffer.drop_front();
|
|
return Token(Kind::equal, "=");
|
|
case '"': {
|
|
size_t end = _buffer.find('"', 1);
|
|
Token ret;
|
|
if (end == _buffer.npos) {
|
|
ret = Token(Kind::identifier, _buffer.substr(1, end));
|
|
_buffer = "";
|
|
} else {
|
|
ret = Token(Kind::identifier, _buffer.substr(1, end - 1));
|
|
_buffer = _buffer.drop_front(end);
|
|
}
|
|
return ret;
|
|
}
|
|
default: {
|
|
size_t end = _buffer.find_first_not_of(
|
|
"abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
"0123456789_.*~+!@#$%^&*()");
|
|
StringRef word = _buffer.substr(0, end);
|
|
Kind kind = llvm::StringSwitch<Kind>(word)
|
|
.Case("DATA", Kind::kw_data)
|
|
.Case("EXPORTS", Kind::kw_exports)
|
|
.Case("NONAME", Kind::kw_noname)
|
|
.Default(Kind::identifier);
|
|
_buffer = (end == _buffer.npos) ? "" : _buffer.drop_front(end);
|
|
return Token(kind, word);
|
|
}
|
|
}
|
|
}
|
|
|
|
void Parser::consumeToken() {
|
|
if (_tokBuf.empty()) {
|
|
_tok = _lex.lex();
|
|
return;
|
|
}
|
|
_tok = _tokBuf.back();
|
|
_tokBuf.pop_back();
|
|
}
|
|
|
|
void Parser::ungetToken() { _tokBuf.push_back(_tok); }
|
|
|
|
void Parser::error(const Token &tok, Twine msg) {
|
|
_lex.getSourceMgr().PrintMessage(
|
|
llvm::SMLoc::getFromPointer(tok._range.data()), llvm::SourceMgr::DK_Error,
|
|
msg);
|
|
}
|
|
|
|
bool Parser::parse(std::vector<PECOFFLinkingContext::ExportDesc> &result) {
|
|
consumeToken();
|
|
if (_tok._kind != Kind::kw_exports) {
|
|
error(_tok, "Expected EXPORTS");
|
|
return false;
|
|
}
|
|
for (;;) {
|
|
PECOFFLinkingContext::ExportDesc desc;
|
|
if (!parseExport(desc))
|
|
return true;
|
|
result.push_back(desc);
|
|
}
|
|
}
|
|
|
|
bool Parser::parseExport(PECOFFLinkingContext::ExportDesc &result) {
|
|
consumeToken();
|
|
if (_tok._kind != Kind::identifier) {
|
|
ungetToken();
|
|
return false;
|
|
}
|
|
result.name = _tok._range;
|
|
|
|
for (;;) {
|
|
consumeToken();
|
|
if (_tok._kind == Kind::identifier && _tok._range[0] == '@') {
|
|
_tok._range.drop_front().getAsInteger(10, result.ordinal);
|
|
consumeToken();
|
|
if (_tok._kind == Kind::kw_noname) {
|
|
result.noname = true;
|
|
} else {
|
|
ungetToken();
|
|
}
|
|
continue;
|
|
}
|
|
if (_tok._kind == Kind::kw_data) {
|
|
result.isData = true;
|
|
continue;
|
|
}
|
|
ungetToken();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
} // moddef
|
|
} // namespace lld
|