diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/MC/MCContext.cpp | 21 | ||||
-rw-r--r-- | lib/MC/MCParser/AsmParser.cpp | 8 |
2 files changed, 12 insertions, 17 deletions
diff --git a/lib/MC/MCContext.cpp b/lib/MC/MCContext.cpp index 45d2c02536..63264f63c4 100644 --- a/lib/MC/MCContext.cpp +++ b/lib/MC/MCContext.cpp @@ -23,16 +23,8 @@ MCContext::~MCContext() { // we don't need to free them here. } -MCSymbol *MCContext::CreateSymbol(StringRef Name) { - assert(Name[0] != '\0' && "Normal symbols cannot be unnamed!"); - - // Create and bind the symbol, and ensure that names are unique. - MCSymbol *&Entry = Symbols[Name]; - assert(!Entry && "Duplicate symbol definition!"); - return Entry = new (*this) MCSymbol(Name, false); -} - MCSymbol *MCContext::GetOrCreateSymbol(StringRef Name) { + assert(!Name.empty() && "Normal symbols cannot be unnamed!"); MCSymbol *&Entry = Symbols[Name]; if (Entry) return Entry; @@ -46,17 +38,24 @@ MCSymbol *MCContext::GetOrCreateSymbol(const Twine &Name) { } -MCSymbol *MCContext::CreateTemporarySymbol(StringRef Name) { +MCSymbol *MCContext::GetOrCreateTemporarySymbol(StringRef Name) { // If unnamed, just create a symbol. if (Name.empty()) new (*this) MCSymbol("", true); // Otherwise create as usual. MCSymbol *&Entry = Symbols[Name]; - assert(!Entry && "Duplicate symbol definition!"); + if (Entry) return Entry; return Entry = new (*this) MCSymbol(Name, true); } +MCSymbol *MCContext::GetOrCreateTemporarySymbol(const Twine &Name) { + SmallString<128> NameSV; + Name.toVector(NameSV); + return GetOrCreateTemporarySymbol(NameSV.str()); +} + + MCSymbol *MCContext::LookupSymbol(StringRef Name) const { return Symbols.lookup(Name); } diff --git a/lib/MC/MCParser/AsmParser.cpp b/lib/MC/MCParser/AsmParser.cpp index 6185c30f52..fc8d549b8a 100644 --- a/lib/MC/MCParser/AsmParser.cpp +++ b/lib/MC/MCParser/AsmParser.cpp @@ -240,14 +240,10 @@ bool AsmParser::ParseParenExpr(const MCExpr *&Res, SMLoc &EndLoc) { } MCSymbol *AsmParser::CreateSymbol(StringRef Name) { - if (MCSymbol *S = Ctx.LookupSymbol(Name)) - return S; - // If the label starts with L it is an assembler temporary label. if (Name.startswith("L")) - return Ctx.CreateTemporarySymbol(Name); - - return Ctx.CreateSymbol(Name); + return Ctx.GetOrCreateTemporarySymbol(Name); + return Ctx.GetOrCreateSymbol(Name); } /// ParsePrimaryExpr - Parse a primary expression and return it. |