First draft for LLDBParser::enumerateClassFunctions

Author: ttreyer

oi/type_graph/LLDBParser.cpp

@@ -120,6 +120,7 @@ Class& LLDBParser::enumerateClass(lldb::SBType& type) {
   auto &c = makeType<Class>(type, kind, displayName, name, size, virtuality);
 
   enumerateClassMembers(type, c.members);
+  enumerateClassFunctions(type, c.functions);
 
   return c;
 }
@@ -128,6 +129,7 @@ void LLDBParser::enumerateClassMembers(lldb::SBType& type, std::vector<Member>&
   assert(members.empty());
   members.reserve(type.GetNumberOfFields());
 
+  /* TODO: We are missing the _vptr */
   for (uint32_t i = 0; i < type.GetNumberOfFields(); i++) {
     auto field = type.GetFieldAtIndex(i);
     if (field.GetName() == nullptr)
@@ -146,6 +148,19 @@ void LLDBParser::enumerateClassMembers(lldb::SBType& type, std::vector<Member>&
   });
 }
 
+void LLDBParser::enumerateClassFunctions(lldb::SBType &type, std::vector<Function>& functions) {
+  assert(functions.empty());
+  functions.reserve(type.GetNumberOfMemberFunctions());
+
+  /* TODO: We are missing the default constructors */
+  for (uint32_t i = 0; i < type.GetNumberOfMemberFunctions(); i++) {
+    auto function = type.GetMemberFunctionAtIndex(i);
+
+    /* TODO: We don't know if the function is virtual */
+    functions.emplace_back(function.GetName(), false);
+  }
+}
+
 Enum& LLDBParser::enumerateEnum(lldb::SBType& type) {
   const char *typeName = type.GetName();
   std::string name = typeName ? typeName : "";

oi/type_graph/LLDBParser.h

@@ -68,6 +68,7 @@ class LLDBParser {
   Primitive::Kind primitiveFloatKind(lldb::SBType& type);
 
   void enumerateClassMembers(lldb::SBType& type, std::vector<Member>& members);
+  void enumerateClassFunctions(lldb::SBType &type, std::vector<Function>& functions);
 
   bool chasePointer() const;
 

test/test_lldb_parser.cpp

@@ -542,7 +542,7 @@ TEST_F(LLDBParserTest, VirtualFunctions) {
   testMultiCompiler("oid_test_case_inheritance_polymorphic_a_as_a",
                     R"(
 [1] Pointer
-[0]   Class: A (size: 16)
+[0]   Class: ns_inheritance_polymorphic::A (size: 16)
         Member: _vptr$A (offset: 0)
           Primitive: StubbedPointer
         Member: int_a (offset: 8)