1 //===-- diagnostic.cpp - tool for testing libLLVM and llvm-c API ----------===//
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 // This file implements the --test-diagnostic-handler command in llvm-c-test.
11 //
12 // This command uses the C API to read a module with a custom diagnostic
13 // handler set to test the diagnostic handler functionality.
14 //
15 //===----------------------------------------------------------------------===//
16
17 #include "llvm-c-test.h"
18 #include "llvm-c/BitReader.h"
19 #include "llvm-c/Core.h"
20
21 #include <stdio.h>
22
diagnosticHandler(LLVMDiagnosticInfoRef DI,void * C)23 static void diagnosticHandler(LLVMDiagnosticInfoRef DI, void *C) {
24 fprintf(stderr, "Executing diagnostic handler\n");
25
26 fprintf(stderr, "Diagnostic severity is of type ");
27 switch (LLVMGetDiagInfoSeverity(DI)) {
28 case LLVMDSError:
29 fprintf(stderr, "error");
30 break;
31 case LLVMDSWarning:
32 fprintf(stderr, "warning");
33 break;
34 case LLVMDSRemark:
35 fprintf(stderr, "remark");
36 break;
37 case LLVMDSNote:
38 fprintf(stderr, "note");
39 break;
40 }
41 fprintf(stderr, "\n");
42
43 (*(int *)C) = 1;
44 }
45
46 static int handlerCalled = 0;
47
llvm_test_diagnostic_handler(void)48 int llvm_test_diagnostic_handler(void) {
49 LLVMContextRef C = LLVMGetGlobalContext();
50 LLVMContextSetDiagnosticHandler(C, diagnosticHandler, &handlerCalled);
51
52 if (LLVMContextGetDiagnosticHandler(C) != diagnosticHandler) {
53 fprintf(stderr, "LLVMContext{Set,Get}DiagnosticHandler failed\n");
54 return 1;
55 }
56
57 int *DC = (int *)LLVMContextGetDiagnosticContext(C);
58 if (DC != &handlerCalled || *DC) {
59 fprintf(stderr, "LLVMContextGetDiagnosticContext failed\n");
60 return 1;
61 }
62
63 LLVMMemoryBufferRef MB;
64 char *msg = NULL;
65 if (LLVMCreateMemoryBufferWithSTDIN(&MB, &msg)) {
66 fprintf(stderr, "Error reading file: %s\n", msg);
67 LLVMDisposeMessage(msg);
68 return 1;
69 }
70
71
72 LLVMModuleRef M;
73 int Ret = LLVMGetBitcodeModule2(MB, &M);
74 if (Ret) {
75 // We do not return if the bitcode was invalid, as we want to test whether
76 // the diagnostic handler was executed.
77 fprintf(stderr, "Error parsing bitcode: %s\n", msg);
78 }
79
80 LLVMDisposeMemoryBuffer(MB);
81
82 if (handlerCalled) {
83 fprintf(stderr, "Diagnostic handler was called while loading module\n");
84 } else {
85 fprintf(stderr, "Diagnostic handler was not called while loading module\n");
86 }
87
88 return 0;
89 }
90