add LLVMGetDINodeTag to C bindings

Differential Revision: https://reviews.llvm.org/D138415

3 files changed, 32 insertions, 1 deletions

diff --git a/llvm/tools/llvm-c-test/debuginfo.c b/llvm/tools/llvm-c-test/debuginfo.c
index d33e40f..26a063f 100644
--- a/llvm/tools/llvm-c-test/debuginfo.c
+++ b/llvm/tools/llvm-c-test/debuginfo.c
@@ -12,10 +12,10 @@
 \*===----------------------------------------------------------------------===*/
 
 #include "llvm-c-test.h"
-#include "llvm-c/Core.h"
 #include "llvm-c/DebugInfo.h"
 #include <stdio.h>
 #include <string.h>
+#include <assert.h>
 
 static LLVMMetadataRef
 declare_objc_class(LLVMDIBuilderRef DIB, LLVMMetadataRef File) {
@@ -201,3 +201,29 @@ int llvm_test_dibuilder(void) {
 
   return 0;
 }
+
+int llvm_get_di_tag(void) {
+  LLVMModuleRef m = LLVMModuleCreateWithName("Mod");
+  LLVMContextRef context = LLVMGetModuleContext(m);
+
+  LLVMMetadataRef metas[] = {LLVMMDStringInContext2(context, "foo", 3)};
+  LLVMMetadataRef md = LLVMMDNodeInContext2(context, metas, 1);
+  uint16_t tag0 = LLVMGetDINodeTag(md);
+
+  assert(tag0 == 0);
+
+  const char *filename = "metadata.c";
+  LLVMDIBuilderRef builder = LLVMCreateDIBuilder(m);
+  LLVMMetadataRef file =
+      LLVMDIBuilderCreateFile(builder, filename, strlen(filename), ".", 1);
+  LLVMMetadataRef decl = LLVMDIBuilderCreateStructType(
+      builder, file, "TestClass", 9, file, 42, 64, 0,
+      LLVMDIFlagObjcClassComplete, NULL, NULL, 0, 0, NULL, NULL, 0);
+  uint16_t tag1 = LLVMGetDINodeTag(decl);
+
+  assert(tag1 == 0x13);
+
+  LLVMDisposeModule(m);
+
+  return 0;
+}
diff --git a/llvm/tools/llvm-c-test/llvm-c-test.h b/llvm/tools/llvm-c-test/llvm-c-test.h
index 5e5b355..15b8573 100644
--- a/llvm/tools/llvm-c-test/llvm-c-test.h
+++ b/llvm/tools/llvm-c-test/llvm-c-test.h
@@ -37,6 +37,7 @@ int llvm_disassemble(void);
 
 // debuginfo.c
 int llvm_test_dibuilder(void);
+int llvm_get_di_tag(void);
 
 // metadata.c
 int llvm_add_named_metadata_operand(void);
diff --git a/llvm/tools/llvm-c-test/main.c b/llvm/tools/llvm-c-test/main.c
index 79a56f7..78eb346 100644
--- a/llvm/tools/llvm-c-test/main.c
+++ b/llvm/tools/llvm-c-test/main.c
@@ -44,6 +44,8 @@ static void print_usage(void) {
   fprintf(stderr, "    Read lines of triple, hex ascii machine code from stdin "
                   "- print disassembly\n\n");
   fprintf(stderr, "  * --calc\n");
+  fprintf(stderr, "  * --get-di-tag\n");
+  fprintf(stderr, "    Run test for getting MDNode dwarf tag\n");
   fprintf(stderr, "  * --replace-md-operand\n");
   fprintf(stderr, "    Run test for replacing MDNode operands\n");
   fprintf(stderr, "  * --is-a-value-as-metadata\n");
@@ -96,6 +98,8 @@ int main(int argc, char **argv) {
     return llvm_add_named_metadata_operand();
   } else if (argc == 2 && !strcmp(argv[1], "--set-metadata")) {
     return llvm_set_metadata();
+  } else if (argc == 2 && !strcmp(argv[1], "--get-di-tag")) {
+    return llvm_get_di_tag();
   } else if (argc == 2 && !strcmp(argv[1], "--replace-md-operand")) {
     return llvm_replace_md_operand();
   } else if (argc == 2 && !strcmp(argv[1], "--is-a-value-as-metadata")) {