[llvm] Fix D26214: Move error handling out of MC and to the callers.

Summary: Related clang patch; https://reviews.llvm.org/D27360

Reviewers: t.p.northover, grosbach, compnerd, echristo

Subscribers: compnerd, mehdi_amini, llvm-commits

Differential Revision: https://reviews.llvm.org/D27359

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@288763 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Mandeep Singh Grang 2016-12-06 02:49:17 +00:00
parent d37747e061
commit 9196899788
3 changed files with 21 additions and 17 deletions

View File

@ -303,7 +303,7 @@ namespace llvm {
MCSymbol *lookupSymbol(const Twine &Name) const; MCSymbol *lookupSymbol(const Twine &Name) const;
/// Set value for a symbol. /// Set value for a symbol.
int setSymbolValue(MCStreamer &Streamer, std::string &I); void setSymbolValue(MCStreamer &Streamer, StringRef Sym, uint64_t Val);
/// getSymbols - Get a reference for the symbol table for clients that /// getSymbols - Get a reference for the symbol table for clients that
/// want to, for example, iterate over all symbols. 'const' because we /// want to, for example, iterate over all symbols. 'const' because we

View File

@ -260,20 +260,11 @@ MCSymbol *MCContext::lookupSymbol(const Twine &Name) const {
return Symbols.lookup(NameRef); return Symbols.lookup(NameRef);
} }
int MCContext::setSymbolValue(MCStreamer &Streamer, std::string &I) { void MCContext::setSymbolValue(MCStreamer &Streamer,
auto Pair = StringRef(I).split('='); StringRef Sym,
if (Pair.second.empty()) { uint64_t Val) {
errs() << "error: defsym must be of the form: sym=value: " << I << "\n"; auto Symbol = getOrCreateSymbol(Sym);
return 1; Streamer.EmitAssignment(Symbol, MCConstantExpr::create(Val, *this));
}
int64_t Value;
if (Pair.second.getAsInteger(0, Value)) {
errs() << "error: Value is not an integer: " << Pair.second << "\n";
return 1;
}
auto Symbol = getOrCreateSymbol(Pair.first);
Streamer.EmitAssignment(Symbol, MCConstantExpr::create(Value, *this));
return 0;
} }
//===----------------------------------------------------------------------===// //===----------------------------------------------------------------------===//

View File

@ -394,9 +394,22 @@ static int AsLexInput(SourceMgr &SrcMgr, MCAsmInfo &MAI,
} }
static int fillCommandLineSymbols(MCAsmParser &Parser) { static int fillCommandLineSymbols(MCAsmParser &Parser) {
for (auto &I: DefineSymbol) for (auto &I: DefineSymbol) {
if (Parser.getContext().setSymbolValue(Parser.getStreamer(), I)) auto Pair = StringRef(I).split('=');
auto Sym = Pair.first;
auto Val = Pair.second;
if (Sym.empty() || Val.empty()) {
errs() << "error: defsym must be of the form: sym=value: " << I << "\n";
return 1; return 1;
}
int64_t Value;
if (Val.getAsInteger(0, Value)) {
errs() << "error: Value is not an integer: " << Val << "\n";
return 1;
}
Parser.getContext().setSymbolValue(Parser.getStreamer(), Sym, Value);
}
return 0; return 0;
} }