Read/write from/to PCH the diagnostic mappings that the user set so that e.g. #pragma clang diagnostic can be used in a PCH.

Fixes rdar://8435969.

git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@118303 91177308-0d34-0410-b5e6-96231b3b80d8

1 files changed, 23 insertions, 0 deletions

diff --git a/lib/Serialization/ASTReader.cpp b/lib/Serialization/ASTReader.cpp
index b90203b477..d70546dbbc 100644
--- a/lib/Serialization/ASTReader.cpp
+++ b/lib/Serialization/ASTReader.cpp
@@ -2130,6 +2130,18 @@ ASTReader::ReadASTBlock(PerFileData &F) {
       F.CXXBaseSpecifiersOffsets = (const uint32_t *)BlobStart;
       break;
     }
+
+    case DIAG_USER_MAPPINGS:
+      if (Record.size() % 2 != 0) {
+        Error("invalid DIAG_USER_MAPPINGS block in AST file");
+        return Failure;
+      }
+      if (UserDiagMappings.empty())
+        UserDiagMappings.swap(Record);
+      else
+        UserDiagMappings.insert(UserDiagMappings.end(),
+                                Record.begin(), Record.end());
+      break;
     }
     First = false;
   }
@@ -2457,6 +2469,8 @@ void ASTReader::InitializeContext(ASTContext &Ctx) {
 
   if (SpecialTypes[SPECIAL_TYPE_INT128_INSTALLED])
     Context->setInt128Installed();
+
+  ReadUserDiagnosticMappings(Context->getDiagnostics());
 }
 
 /// \brief Retrieve the name of the original source file name
@@ -2623,6 +2637,15 @@ void ASTReader::ReadPreprocessedEntities() {
   ReadDefinedMacros();
 }
 
+void ASTReader::ReadUserDiagnosticMappings(Diagnostic &Diag) {
+  unsigned Idx = 0;
+  while (Idx < UserDiagMappings.size()) {
+    unsigned DiagID = UserDiagMappings[Idx++];
+    unsigned Map = UserDiagMappings[Idx++];
+    Diag.setDiagnosticMappingInternal(DiagID, Map, /*isUser=*/true);
+  }
+}
+
 /// \brief Get the correct cursor and offset for loading a type.
 ASTReader::RecordLocation ASTReader::TypeCursorForIndex(unsigned Index) {
   PerFileData *F = 0;