1 //===-- ClangExternalASTSourceCommon.cpp ------------------------*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #include "lldb/Symbol/ClangExternalASTSourceCommon.h"
11 #include "lldb/Utility/Stream.h"
12 
13 #include <mutex>
14 
15 using namespace lldb_private;
16 
17 uint64_t g_TotalSizeOfMetadata = 0;
18 
19 typedef llvm::DenseMap<clang::ExternalASTSource *,
20                        ClangExternalASTSourceCommon *>
21     ASTSourceMap;
22 
GetSourceMap(std::unique_lock<std::mutex> & guard)23 static ASTSourceMap &GetSourceMap(std::unique_lock<std::mutex> &guard) {
24   // Intentionally leaked to avoid problems with global destructors.
25   static ASTSourceMap *s_source_map = new ASTSourceMap;
26   static std::mutex s_mutex;
27   std::unique_lock<std::mutex> locked_guard(s_mutex);
28   guard.swap(locked_guard);
29   return *s_source_map;
30 }
31 
32 ClangExternalASTSourceCommon *
Lookup(clang::ExternalASTSource * source)33 ClangExternalASTSourceCommon::Lookup(clang::ExternalASTSource *source) {
34   std::unique_lock<std::mutex> guard;
35   ASTSourceMap &source_map = GetSourceMap(guard);
36 
37   ASTSourceMap::iterator iter = source_map.find(source);
38 
39   if (iter != source_map.end()) {
40     return iter->second;
41   } else {
42     return nullptr;
43   }
44 }
45 
ClangExternalASTSourceCommon()46 ClangExternalASTSourceCommon::ClangExternalASTSourceCommon()
47     : clang::ExternalASTSource() {
48   g_TotalSizeOfMetadata += m_metadata.size();
49   std::unique_lock<std::mutex> guard;
50   GetSourceMap(guard)[this] = this;
51 }
52 
~ClangExternalASTSourceCommon()53 ClangExternalASTSourceCommon::~ClangExternalASTSourceCommon() {
54   std::unique_lock<std::mutex> guard;
55   GetSourceMap(guard).erase(this);
56   g_TotalSizeOfMetadata -= m_metadata.size();
57 }
58 
59 ClangASTMetadata *
GetMetadata(const void * object)60 ClangExternalASTSourceCommon::GetMetadata(const void *object) {
61   if (HasMetadata(object))
62     return &m_metadata[object];
63   else
64     return nullptr;
65 }
66 
SetMetadata(const void * object,ClangASTMetadata & metadata)67 void ClangExternalASTSourceCommon::SetMetadata(const void *object,
68                                                ClangASTMetadata &metadata) {
69   uint64_t orig_size = m_metadata.size();
70   m_metadata[object] = metadata;
71   uint64_t new_size = m_metadata.size();
72   g_TotalSizeOfMetadata += (new_size - orig_size);
73 }
74 
HasMetadata(const void * object)75 bool ClangExternalASTSourceCommon::HasMetadata(const void *object) {
76   return m_metadata.find(object) != m_metadata.end();
77 }
78 
Dump(Stream * s)79 void ClangASTMetadata::Dump(Stream *s) {
80   lldb::user_id_t uid = GetUserID();
81 
82   if (uid != LLDB_INVALID_UID) {
83     s->Printf("uid=0x%" PRIx64, uid);
84   }
85 
86   uint64_t isa_ptr = GetISAPtr();
87   if (isa_ptr != 0) {
88     s->Printf("isa_ptr=0x%" PRIx64, isa_ptr);
89   }
90 
91   const char *obj_ptr_name = GetObjectPtrName();
92   if (obj_ptr_name) {
93     s->Printf("obj_ptr_name=\"%s\" ", obj_ptr_name);
94   }
95 
96   if (m_is_dynamic_cxx) {
97     s->Printf("is_dynamic_cxx=%i ", m_is_dynamic_cxx);
98   }
99   s->EOL();
100 }
101