xref: /llvm-project/clang/unittests/libclang/LibclangTest.cpp (revision 6189422f08aeef2ac0958cd9533b8d6bc4352d7e)
1 //===- unittests/libclang/LibclangTest.cpp --- libclang tests -------------===//
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 "clang-c/Index.h"
11 #include "gtest/gtest.h"
12 #include "llvm/Support/Debug.h"
13 #include "llvm/Support/FileSystem.h"
14 #include "llvm/Support/Path.h"
15 #include "llvm/Support/raw_ostream.h"
16 #include <fstream>
17 #include <set>
18 #define DEBUG_TYPE "libclang-test"
19 
20 TEST(libclang, clang_parseTranslationUnit2_InvalidArgs) {
21   EXPECT_EQ(CXError_InvalidArguments,
22             clang_parseTranslationUnit2(nullptr, nullptr, nullptr, 0, nullptr,
23                                         0, 0, nullptr));
24 }
25 
26 TEST(libclang, clang_createTranslationUnit_InvalidArgs) {
27   EXPECT_EQ(nullptr, clang_createTranslationUnit(nullptr, nullptr));
28 }
29 
30 TEST(libclang, clang_createTranslationUnit2_InvalidArgs) {
31   EXPECT_EQ(CXError_InvalidArguments,
32             clang_createTranslationUnit2(nullptr, nullptr, nullptr));
33 
34   CXTranslationUnit TU = reinterpret_cast<CXTranslationUnit>(1);
35   EXPECT_EQ(CXError_InvalidArguments,
36             clang_createTranslationUnit2(nullptr, nullptr, &TU));
37   EXPECT_EQ(nullptr, TU);
38 }
39 
40 namespace {
41 struct TestVFO {
42   const char *Contents;
43   CXVirtualFileOverlay VFO;
44 
45   TestVFO(const char *Contents) : Contents(Contents) {
46     VFO = clang_VirtualFileOverlay_create(0);
47   }
48 
49   void map(const char *VPath, const char *RPath) {
50     CXErrorCode Err = clang_VirtualFileOverlay_addFileMapping(VFO, VPath, RPath);
51     EXPECT_EQ(Err, CXError_Success);
52   }
53 
54   void mapError(const char *VPath, const char *RPath, CXErrorCode ExpErr) {
55     CXErrorCode Err = clang_VirtualFileOverlay_addFileMapping(VFO, VPath, RPath);
56     EXPECT_EQ(Err, ExpErr);
57   }
58 
59   ~TestVFO() {
60     if (Contents) {
61       char *BufPtr;
62       unsigned BufSize;
63       clang_VirtualFileOverlay_writeToBuffer(VFO, 0, &BufPtr, &BufSize);
64       std::string BufStr(BufPtr, BufSize);
65       EXPECT_STREQ(Contents, BufStr.c_str());
66       free(BufPtr);
67     }
68     clang_VirtualFileOverlay_dispose(VFO);
69   }
70 };
71 }
72 
73 TEST(libclang, VirtualFileOverlay_Basic) {
74   const char *contents =
75       "{\n"
76       "  'version': 0,\n"
77       "  'roots': [\n"
78       "    {\n"
79       "      'type': 'directory',\n"
80       "      'name': \"/path/virtual\",\n"
81       "      'contents': [\n"
82       "        {\n"
83       "          'type': 'file',\n"
84       "          'name': \"foo.h\",\n"
85       "          'external-contents': \"/real/foo.h\"\n"
86       "        }\n"
87       "      ]\n"
88       "    }\n"
89       "  ]\n"
90       "}\n";
91   TestVFO T(contents);
92   T.map("/path/virtual/foo.h", "/real/foo.h");
93 }
94 
95 TEST(libclang, VirtualFileOverlay_Unicode) {
96   const char *contents =
97       "{\n"
98       "  'version': 0,\n"
99       "  'roots': [\n"
100       "    {\n"
101       "      'type': 'directory',\n"
102       "      'name': \"/path/\\u266B\",\n"
103       "      'contents': [\n"
104       "        {\n"
105       "          'type': 'file',\n"
106       "          'name': \"\\u2602.h\",\n"
107       "          'external-contents': \"/real/\\u2602.h\"\n"
108       "        }\n"
109       "      ]\n"
110       "    }\n"
111       "  ]\n"
112       "}\n";
113   TestVFO T(contents);
114   T.map("/path/♫/☂.h", "/real/☂.h");
115 }
116 
117 TEST(libclang, VirtualFileOverlay_InvalidArgs) {
118   TestVFO T(nullptr);
119   T.mapError("/path/./virtual/../foo.h", "/real/foo.h",
120              CXError_InvalidArguments);
121 }
122 
123 TEST(libclang, VirtualFileOverlay_RemapDirectories) {
124   const char *contents =
125       "{\n"
126       "  'version': 0,\n"
127       "  'roots': [\n"
128       "    {\n"
129       "      'type': 'directory',\n"
130       "      'name': \"/another/dir\",\n"
131       "      'contents': [\n"
132       "        {\n"
133       "          'type': 'file',\n"
134       "          'name': \"foo2.h\",\n"
135       "          'external-contents': \"/real/foo2.h\"\n"
136       "        }\n"
137       "      ]\n"
138       "    },\n"
139       "    {\n"
140       "      'type': 'directory',\n"
141       "      'name': \"/path/virtual/dir\",\n"
142       "      'contents': [\n"
143       "        {\n"
144       "          'type': 'file',\n"
145       "          'name': \"foo1.h\",\n"
146       "          'external-contents': \"/real/foo1.h\"\n"
147       "        },\n"
148       "        {\n"
149       "          'type': 'file',\n"
150       "          'name': \"foo3.h\",\n"
151       "          'external-contents': \"/real/foo3.h\"\n"
152       "        },\n"
153       "        {\n"
154       "          'type': 'directory',\n"
155       "          'name': \"in/subdir\",\n"
156       "          'contents': [\n"
157       "            {\n"
158       "              'type': 'file',\n"
159       "              'name': \"foo4.h\",\n"
160       "              'external-contents': \"/real/foo4.h\"\n"
161       "            }\n"
162       "          ]\n"
163       "        }\n"
164       "      ]\n"
165       "    }\n"
166       "  ]\n"
167       "}\n";
168   TestVFO T(contents);
169   T.map("/path/virtual/dir/foo1.h", "/real/foo1.h");
170   T.map("/another/dir/foo2.h", "/real/foo2.h");
171   T.map("/path/virtual/dir/foo3.h", "/real/foo3.h");
172   T.map("/path/virtual/dir/in/subdir/foo4.h", "/real/foo4.h");
173 }
174 
175 TEST(libclang, VirtualFileOverlay_CaseInsensitive) {
176   const char *contents =
177       "{\n"
178       "  'version': 0,\n"
179       "  'case-sensitive': 'false',\n"
180       "  'roots': [\n"
181       "    {\n"
182       "      'type': 'directory',\n"
183       "      'name': \"/path/virtual\",\n"
184       "      'contents': [\n"
185       "        {\n"
186       "          'type': 'file',\n"
187       "          'name': \"foo.h\",\n"
188       "          'external-contents': \"/real/foo.h\"\n"
189       "        }\n"
190       "      ]\n"
191       "    }\n"
192       "  ]\n"
193       "}\n";
194   TestVFO T(contents);
195   T.map("/path/virtual/foo.h", "/real/foo.h");
196   clang_VirtualFileOverlay_setCaseSensitivity(T.VFO, false);
197 }
198 
199 TEST(libclang, VirtualFileOverlay_SharedPrefix) {
200   const char *contents =
201       "{\n"
202       "  'version': 0,\n"
203       "  'roots': [\n"
204       "    {\n"
205       "      'type': 'directory',\n"
206       "      'name': \"/path/foo\",\n"
207       "      'contents': [\n"
208       "        {\n"
209       "          'type': 'file',\n"
210       "          'name': \"bar\",\n"
211       "          'external-contents': \"/real/bar\"\n"
212       "        },\n"
213       "        {\n"
214       "          'type': 'file',\n"
215       "          'name': \"bar.h\",\n"
216       "          'external-contents': \"/real/bar.h\"\n"
217       "        }\n"
218       "      ]\n"
219       "    },\n"
220       "    {\n"
221       "      'type': 'directory',\n"
222       "      'name': \"/path/foobar\",\n"
223       "      'contents': [\n"
224       "        {\n"
225       "          'type': 'file',\n"
226       "          'name': \"baz.h\",\n"
227       "          'external-contents': \"/real/baz.h\"\n"
228       "        }\n"
229       "      ]\n"
230       "    },\n"
231       "    {\n"
232       "      'type': 'directory',\n"
233       "      'name': \"/path\",\n"
234       "      'contents': [\n"
235       "        {\n"
236       "          'type': 'file',\n"
237       "          'name': \"foobarbaz.h\",\n"
238       "          'external-contents': \"/real/foobarbaz.h\"\n"
239       "        }\n"
240       "      ]\n"
241       "    }\n"
242       "  ]\n"
243       "}\n";
244   TestVFO T(contents);
245   T.map("/path/foo/bar.h", "/real/bar.h");
246   T.map("/path/foo/bar", "/real/bar");
247   T.map("/path/foobar/baz.h", "/real/baz.h");
248   T.map("/path/foobarbaz.h", "/real/foobarbaz.h");
249 }
250 
251 TEST(libclang, VirtualFileOverlay_AdjacentDirectory) {
252   const char *contents =
253       "{\n"
254       "  'version': 0,\n"
255       "  'roots': [\n"
256       "    {\n"
257       "      'type': 'directory',\n"
258       "      'name': \"/path/dir1\",\n"
259       "      'contents': [\n"
260       "        {\n"
261       "          'type': 'file',\n"
262       "          'name': \"foo.h\",\n"
263       "          'external-contents': \"/real/foo.h\"\n"
264       "        },\n"
265       "        {\n"
266       "          'type': 'directory',\n"
267       "          'name': \"subdir\",\n"
268       "          'contents': [\n"
269       "            {\n"
270       "              'type': 'file',\n"
271       "              'name': \"bar.h\",\n"
272       "              'external-contents': \"/real/bar.h\"\n"
273       "            }\n"
274       "          ]\n"
275       "        }\n"
276       "      ]\n"
277       "    },\n"
278       "    {\n"
279       "      'type': 'directory',\n"
280       "      'name': \"/path/dir2\",\n"
281       "      'contents': [\n"
282       "        {\n"
283       "          'type': 'file',\n"
284       "          'name': \"baz.h\",\n"
285       "          'external-contents': \"/real/baz.h\"\n"
286       "        }\n"
287       "      ]\n"
288       "    }\n"
289       "  ]\n"
290       "}\n";
291   TestVFO T(contents);
292   T.map("/path/dir1/foo.h", "/real/foo.h");
293   T.map("/path/dir1/subdir/bar.h", "/real/bar.h");
294   T.map("/path/dir2/baz.h", "/real/baz.h");
295 }
296 
297 TEST(libclang, VirtualFileOverlay_TopLevel) {
298   const char *contents =
299       "{\n"
300       "  'version': 0,\n"
301       "  'roots': [\n"
302       "    {\n"
303       "      'type': 'directory',\n"
304       "      'name': \"/\",\n"
305       "      'contents': [\n"
306       "        {\n"
307       "          'type': 'file',\n"
308       "          'name': \"foo.h\",\n"
309       "          'external-contents': \"/real/foo.h\"\n"
310       "        }\n"
311       "      ]\n"
312       "    }\n"
313       "  ]\n"
314       "}\n";
315   TestVFO T(contents);
316   T.map("/foo.h", "/real/foo.h");
317 }
318 
319 TEST(libclang, ModuleMapDescriptor) {
320   const char *Contents =
321     "framework module TestFrame {\n"
322     "  umbrella header \"TestFrame.h\"\n"
323     "\n"
324     "  export *\n"
325     "  module * { export * }\n"
326     "}\n";
327 
328   CXModuleMapDescriptor MMD = clang_ModuleMapDescriptor_create(0);
329 
330   clang_ModuleMapDescriptor_setFrameworkModuleName(MMD, "TestFrame");
331   clang_ModuleMapDescriptor_setUmbrellaHeader(MMD, "TestFrame.h");
332 
333   char *BufPtr;
334   unsigned BufSize;
335   clang_ModuleMapDescriptor_writeToBuffer(MMD, 0, &BufPtr, &BufSize);
336   std::string BufStr(BufPtr, BufSize);
337   EXPECT_STREQ(Contents, BufStr.c_str());
338   free(BufPtr);
339   clang_ModuleMapDescriptor_dispose(MMD);
340 }
341 
342 class LibclangReparseTest : public ::testing::Test {
343   std::string TestDir;
344   std::set<std::string> Files;
345 public:
346   CXIndex Index;
347   CXTranslationUnit ClangTU;
348   unsigned TUFlags;
349 
350   void SetUp() {
351     llvm::SmallString<256> Dir;
352     ASSERT_FALSE(llvm::sys::fs::createUniqueDirectory("libclang-test", Dir));
353     TestDir = Dir.str();
354     TUFlags = CXTranslationUnit_DetailedPreprocessingRecord |
355               clang_defaultEditingTranslationUnitOptions();
356     Index = clang_createIndex(0, 0);
357   }
358   void TearDown() {
359     clang_disposeTranslationUnit(ClangTU);
360     clang_disposeIndex(Index);
361     for (const std::string &Path : Files)
362       llvm::sys::fs::remove(Path);
363     llvm::sys::fs::remove(TestDir);
364   }
365   void WriteFile(std::string &Filename, const std::string &Contents) {
366     if (!llvm::sys::path::is_absolute(Filename)) {
367       llvm::SmallString<256> Path(TestDir);
368       llvm::sys::path::append(Path, Filename);
369       Filename = Path.str();
370       Files.insert(Filename);
371     }
372     std::ofstream OS(Filename);
373     OS << Contents;
374   }
375   void DisplayDiagnostics() {
376     unsigned NumDiagnostics = clang_getNumDiagnostics(ClangTU);
377     for (unsigned i = 0; i < NumDiagnostics; ++i) {
378       auto Diag = clang_getDiagnostic(ClangTU, i);
379       DEBUG(llvm::dbgs() << clang_getCString(clang_formatDiagnostic(
380           Diag, clang_defaultDiagnosticDisplayOptions())) << "\n");
381       clang_disposeDiagnostic(Diag);
382     }
383   }
384   bool ReparseTU(unsigned num_unsaved_files, CXUnsavedFile* unsaved_files) {
385     if (clang_reparseTranslationUnit(ClangTU, num_unsaved_files, unsaved_files,
386                                      clang_defaultReparseOptions(ClangTU))) {
387       DEBUG(llvm::dbgs() << "Reparse failed\n");
388       return false;
389     }
390     DisplayDiagnostics();
391     return true;
392   }
393 };
394 
395 
396 TEST_F(LibclangReparseTest, Reparse) {
397   const char *HeaderTop = "#ifndef H\n#define H\nstruct Foo { int bar;";
398   const char *HeaderBottom = "\n};\n#endif\n";
399   const char *CppFile = "#include \"HeaderFile.h\"\nint main() {"
400                          " Foo foo; foo.bar = 7; foo.baz = 8; }\n";
401   std::string HeaderName = "HeaderFile.h";
402   std::string CppName = "CppFile.cpp";
403   WriteFile(CppName, CppFile);
404   WriteFile(HeaderName, std::string(HeaderTop) + HeaderBottom);
405 
406   ClangTU = clang_parseTranslationUnit(Index, CppName.c_str(), nullptr, 0,
407                                        nullptr, 0, TUFlags);
408   EXPECT_EQ(1U, clang_getNumDiagnostics(ClangTU));
409   DisplayDiagnostics();
410 
411   // Immedaitely reparse.
412   ASSERT_TRUE(ReparseTU(0, nullptr /* No unsaved files. */));
413   EXPECT_EQ(1U, clang_getNumDiagnostics(ClangTU));
414 
415   std::string NewHeaderContents =
416       std::string(HeaderTop) + "int baz;" + HeaderBottom;
417   WriteFile(HeaderName, NewHeaderContents);
418 
419   // Reparse after fix.
420   ASSERT_TRUE(ReparseTU(0, nullptr /* No unsaved files. */));
421   EXPECT_EQ(0U, clang_getNumDiagnostics(ClangTU));
422 }
423