blink_gc_plugin: detect singletons with embedded ScriptWrappables.

tools/clang/blink_gc_plugin/CollectVisitor.cpp

 // Copyright 2015 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.
 
 #include "CollectVisitor.h"
 
 #include "Config.h"
 
 using namespace clang;
 
-CollectVisitor::CollectVisitor() {
-}
+CollectVisitor::CollectVisitor(bool check_var_decls)
+    : check_var_decls_(check_var_decls), var_visitor_(this) {}
 
 CollectVisitor::RecordVector& CollectVisitor::record_decls() {
   return record_decls_;
 }
 
 CollectVisitor::MethodVector& CollectVisitor::trace_decls() {
   return trace_decls_;
 }
 
+const CollectVisitor::SingletonVector& CollectVisitor::singleton_decls() const {
+  return singleton_decls_;
+}
+
 bool CollectVisitor::VisitCXXRecordDecl(CXXRecordDecl* record) {
   if (record->hasDefinition() && record->isCompleteDefinition())
     record_decls_.push_back(record);
   return true;
 }
 
 bool CollectVisitor::VisitCXXMethodDecl(CXXMethodDecl* method) {
   if (method->isThisDeclarationADefinition() && Config::IsTraceMethod(method))
     trace_decls_.push_back(method);
   return true;
 }
+
+bool CollectVisitor::LocalVarsVisitor::VisitVarDecl(VarDecl* decl) {
+  if (Config::IsStaticSingleton(decl) && !Config::IsIgnoreAnnotated(decl) &&
+      decl->isStaticLocal())
+    parent_->singleton_decls_.push_back(decl);
+  return true;
+}
+
+bool CollectVisitor::VisitFunctionDecl(FunctionDecl* decl) {
+  if (!check_var_decls_)
+    return true;
+
+  if (!decl->isThisDeclarationADefinition())
+    return true;
+
+  if (Stmt* body = decl->getBody())
+    var_visitor_.TraverseStmt(body);
+  return true;
+}