Follow macro invocations when checking if a violation happened in a .cc file.

tools/clang/plugins/ChromeClassTester.cpp

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 // A general interface for filtering and only acting on classes in Chromium C++
 // code.
 
 #include "ChromeClassTester.h"
 
 #include <algorithm>
@@ skipping 171 matching lines @@
 
   return false;
 }
 
 std::string ChromeClassTester::GetNamespace(const Decl* record) {
   return GetNamespaceImpl(record->getDeclContext(), "");
 }
 
 bool ChromeClassTester::InImplementationFile(SourceLocation record_location) {
   std::string filename;
-  if (!GetFilename(record_location, &filename))
-    return false;
 
-  if (ends_with(filename, ".cc") || ends_with(filename, ".cpp") ||
-      ends_with(filename, ".mm")) {
-    return true;
+  if (options_.follow_macro_expansion) {
+    // If |record_location| is a macro, check the whole chain of expansions.
+    const SourceManager& source_manager = instance_.getSourceManager();
+    while (true) {
+      if (GetFilename(record_location, &filename)) {
+        if (ends_with(filename, ".cc") || ends_with(filename, ".cpp") ||
+            ends_with(filename, ".mm")) {
+          return true;
+        }
+      }
+      if (!record_location.isMacroID()) {
+        break;
+      }
+      record_location =
+          source_manager.getImmediateExpansionRange(record_location).first;
+    }
+  } else {
+    if (!GetFilename(record_location, &filename))
+      return false;
+
+    if (ends_with(filename, ".cc") || ends_with(filename, ".cpp") ||
+        ends_with(filename, ".mm")) {
+      return true;
+    }
   }
 
   return false;
 }
 
 void ChromeClassTester::BuildBannedLists() {
   banned_namespaces_.emplace("std");
   banned_namespaces_.emplace("__gnu_cxx");
 
   if (options_.enforce_in_thirdparty_webkit) {
@@ skipping 102 matching lines @@
   }
 
   *filename = ploc.getFilename();
   return true;
 }
 
 DiagnosticsEngine::Level ChromeClassTester::getErrorLevel() {
   return diagnostic().getWarningsAsErrors() ? DiagnosticsEngine::Error
                                             : DiagnosticsEngine::Warning;
 }