[llvm-commits] [llvm] r103137 - in /llvm/trunk: lib/MC/MCParser/AsmLexer.cpp test/MC/AsmParser/X86/x86_operands.s test/MC/AsmParser/labels.s

[Chris Lattner]

Author: lattner
Date: Wed May  5 18:51:28 2010
New Revision: 103137

URL: http://llvm.org/viewvc/llvm-project?rev=103137&view=rev
Log:
fix rdar://7946934 - in some limited cases, the assembler should
allow $ at the start of a symbol name.


Modified:
    llvm/trunk/lib/MC/MCParser/AsmLexer.cpp
    llvm/trunk/test/MC/AsmParser/X86/x86_operands.s
    llvm/trunk/test/MC/AsmParser/labels.s

Modified: llvm/trunk/lib/MC/MCParser/AsmLexer.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/MC/MCParser/AsmLexer.cpp?rev=103137&r1=103136&r2=103137&view=diff
==============================================================================
--- llvm/trunk/lib/MC/MCParser/AsmLexer.cpp (original)
+++ llvm/trunk/lib/MC/MCParser/AsmLexer.cpp Wed May  5 18:51:28 2010
@@ -69,16 +69,26 @@
   }
 }
 
-/// LexIdentifier: [a-zA-Z_.][a-zA-Z0-9_$.@]*
+/// LexIdentifier: [a-zA-Z_.$][a-zA-Z0-9_$.@]*
+/// LexIdentifier: .
+/// LexIdentifier: $
 AsmToken AsmLexer::LexIdentifier() {
   while (isalnum(*CurPtr) || *CurPtr == '_' || *CurPtr == '$' ||
          *CurPtr == '.' || *CurPtr == '@')
     ++CurPtr;
   
   // Handle . as a special case.
-  if (CurPtr == TokStart+1 && TokStart[0] == '.')
-    return AsmToken(AsmToken::Dot, StringRef(TokStart, 1));
-  
+  if (CurPtr == TokStart+1)
+    if (TokStart[0] == '.')
+      return AsmToken(AsmToken::Dot, StringRef(TokStart, 1));
+
+  // Handle $ as a special case. $foo is an identifier, $42 is not.
+  if (TokStart[0] == '$' &&
+      (CurPtr-TokStart == 1 || isdigit(TokStart[1]) || TokStart[1] == '"')) {
+    CurPtr = TokStart+1;
+    return AsmToken(AsmToken::Dollar, StringRef(TokStart, 1));
+  }
+
   return AsmToken(AsmToken::Identifier, StringRef(TokStart, CurPtr - TokStart));
 }
 
@@ -252,8 +262,8 @@
 
   switch (CurChar) {
   default:
-    // Handle identifier: [a-zA-Z_.][a-zA-Z0-9_$.@]*
-    if (isalpha(CurChar) || CurChar == '_' || CurChar == '.')
+    // Handle identifier: [a-zA-Z_.$][a-zA-Z0-9_$.@]*
+    if (isalpha(CurChar) || CurChar == '_' || CurChar == '.' || CurChar == '$')
       return LexIdentifier();
     
     // Unknown character, emit an error.
@@ -279,7 +289,6 @@
   case '}': return AsmToken(AsmToken::RCurly, StringRef(TokStart, 1));
   case '*': return AsmToken(AsmToken::Star, StringRef(TokStart, 1));
   case ',': return AsmToken(AsmToken::Comma, StringRef(TokStart, 1));
-  case '$': return AsmToken(AsmToken::Dollar, StringRef(TokStart, 1));
   case '=': 
     if (*CurPtr == '=')
       return ++CurPtr, AsmToken(AsmToken::EqualEqual, StringRef(TokStart, 2));

Modified: llvm/trunk/test/MC/AsmParser/X86/x86_operands.s
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/MC/AsmParser/X86/x86_operands.s?rev=103137&r1=103136&r2=103137&view=diff
==============================================================================
--- llvm/trunk/test/MC/AsmParser/X86/x86_operands.s (original)
+++ llvm/trunk/test/MC/AsmParser/X86/x86_operands.s Wed May  5 18:51:28 2010
@@ -1,5 +1,3 @@
-// FIXME: Actually test that we get the expected results.
-        
 // RUN: llvm-mc -triple i386-unknown-unknown %s | FileCheck %s
 
 # Immediates
@@ -7,7 +5,7 @@
         addl $1, %eax
 # CHECK: addl $3, %eax
         addl $(1+2), %eax
-# CHECK: addl $a, %eax
+# CHECK: addl ($a), %eax
         addl $a, %eax
 # CHECK: addl $3, %eax
         addl $1 + 2, %eax

Modified: llvm/trunk/test/MC/AsmParser/labels.s
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/MC/AsmParser/labels.s?rev=103137&r1=103136&r2=103137&view=diff
==============================================================================
--- llvm/trunk/test/MC/AsmParser/labels.s (original)
+++ llvm/trunk/test/MC/AsmParser/labels.s Wed May  5 18:51:28 2010
@@ -57,3 +57,10 @@
 
 // CHECK: .long "a 9"
         .long "a 9"
+
+
+// rdar://7946934
+// CHECK: .globl $abc
+        .globl $abc
+        
+        
\ No newline at end of file