diff --git a/llvm/include/llvm/MC/MCContext.h b/llvm/include/llvm/MC/MCContext.h index 2ce2780d4f22e..aa302dbe0a218 100644 --- a/llvm/include/llvm/MC/MCContext.h +++ b/llvm/include/llvm/MC/MCContext.h @@ -303,7 +303,7 @@ namespace llvm { MCSymbol *lookupSymbol(const Twine &Name) const; /// 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 /// want to, for example, iterate over all symbols. 'const' because we diff --git a/llvm/lib/MC/MCContext.cpp b/llvm/lib/MC/MCContext.cpp index afe8ee84918e3..8298aff928ccf 100644 --- a/llvm/lib/MC/MCContext.cpp +++ b/llvm/lib/MC/MCContext.cpp @@ -260,20 +260,11 @@ MCSymbol *MCContext::lookupSymbol(const Twine &Name) const { return Symbols.lookup(NameRef); } -int MCContext::setSymbolValue(MCStreamer &Streamer, std::string &I) { - auto Pair = StringRef(I).split('='); - if (Pair.second.empty()) { - errs() << "error: defsym must be of the form: sym=value: " << I << "\n"; - return 1; - } - 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; +void MCContext::setSymbolValue(MCStreamer &Streamer, + StringRef Sym, + uint64_t Val) { + auto Symbol = getOrCreateSymbol(Sym); + Streamer.EmitAssignment(Symbol, MCConstantExpr::create(Val, *this)); } //===----------------------------------------------------------------------===// diff --git a/llvm/tools/llvm-mc/llvm-mc.cpp b/llvm/tools/llvm-mc/llvm-mc.cpp index 8a904a8ad7cc9..497fb1987764e 100644 --- a/llvm/tools/llvm-mc/llvm-mc.cpp +++ b/llvm/tools/llvm-mc/llvm-mc.cpp @@ -394,9 +394,22 @@ static int AsLexInput(SourceMgr &SrcMgr, MCAsmInfo &MAI, } static int fillCommandLineSymbols(MCAsmParser &Parser) { - for (auto &I: DefineSymbol) - if (Parser.getContext().setSymbolValue(Parser.getStreamer(), I)) + for (auto &I: DefineSymbol) { + 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; + } + 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; }