summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--tools/llvm-mc/AsmLexer.cpp6
-rw-r--r--tools/llvm-mc/AsmLexer.h2
2 files changed, 3 insertions, 5 deletions
diff --git a/tools/llvm-mc/AsmLexer.cpp b/tools/llvm-mc/AsmLexer.cpp
index be8f60d2dd..99055c6855 100644
--- a/tools/llvm-mc/AsmLexer.cpp
+++ b/tools/llvm-mc/AsmLexer.cpp
@@ -243,10 +243,8 @@ StringRef AsmLexer::LexUntilEndOfStatement() {
}
bool AsmLexer::isAtStartOfComment(char Char) {
- for (const char *p = MAI.getCommentString(); *p != 0; ++p)
- if (Char == *p)
- return true;
- return false;
+ // FIXME: This won't work for multi-character comment indicators like "//".
+ return Char == *MAI.getCommentString();
}
AsmToken AsmLexer::LexToken() {
diff --git a/tools/llvm-mc/AsmLexer.h b/tools/llvm-mc/AsmLexer.h
index 0c2ec13892..0696abc887 100644
--- a/tools/llvm-mc/AsmLexer.h
+++ b/tools/llvm-mc/AsmLexer.h
@@ -30,7 +30,7 @@ class MCAsmInfo;
/// AsmLexer - Lexer class for assembly files.
class AsmLexer : public MCAsmLexer {
SourceMgr &SrcMgr;
- const MCAsmInfo MAI;
+ const MCAsmInfo &MAI;
const char *CurPtr;
const MemoryBuffer *CurBuf;