Error on all .code* directives instead of just .code16 as they
all lead to a silent miscompilation of code.



git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@124603 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/lib/MC/MCParser/AsmParser.cpp b/lib/MC/MCParser/AsmParser.cpp
index b66eeb9..7a65878 100644
--- a/lib/MC/MCParser/AsmParser.cpp
+++ b/lib/MC/MCParser/AsmParser.cpp
@@ -1065,8 +1065,8 @@
     if (IDVal == ".include")
       return ParseDirectiveInclude();
 
-    if (IDVal == ".code16")
-      return TokError(".code16 not supported yet");
+    if (IDVal.startswith(".code"))
+      return TokError(Twine(IDVal) + " not supported yet");
 
     // Look up the handler in the handler table.
     std::pair<MCAsmParserExtension*, DirectiveHandler> Handler =