c-index-test.c revision 936ea3b590117d2cd73b1b92621d06c4a7edbe60
12b8ee6c2994f738e5162ff46b638974870f51662Steve Naroff/* c-index-test.c */ 250398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff 350398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff#include "clang-c/Index.h" 40c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor#include <stdlib.h> 589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff#include <stdio.h> 6af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff#include <string.h> 7f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor#include <assert.h> 8af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff 90d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 100d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Utility functions. */ 110d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 120d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 132e06fc877a633abea3b40a64950c7316dac29ca8John Thompson#ifdef _MSC_VER 142e06fc877a633abea3b40a64950c7316dac29ca8John Thompsonchar *basename(const char* path) 152e06fc877a633abea3b40a64950c7316dac29ca8John Thompson{ 162e06fc877a633abea3b40a64950c7316dac29ca8John Thompson char* base1 = (char*)strrchr(path, '/'); 172e06fc877a633abea3b40a64950c7316dac29ca8John Thompson char* base2 = (char*)strrchr(path, '\\'); 182e06fc877a633abea3b40a64950c7316dac29ca8John Thompson if (base1 && base2) 192e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return((base1 > base2) ? base1 + 1 : base2 + 1); 202e06fc877a633abea3b40a64950c7316dac29ca8John Thompson else if (base1) 212e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return(base1 + 1); 222e06fc877a633abea3b40a64950c7316dac29ca8John Thompson else if (base2) 232e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return(base2 + 1); 242e06fc877a633abea3b40a64950c7316dac29ca8John Thompson 252e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return((char*)path); 262e06fc877a633abea3b40a64950c7316dac29ca8John Thompson} 272e06fc877a633abea3b40a64950c7316dac29ca8John Thompson#else 28ff9e18cb3863defa98257500ef49d93a9d2bc216Steve Naroffextern char *basename(const char *); 292e06fc877a633abea3b40a64950c7316dac29ca8John Thompson#endif 30ff9e18cb3863defa98257500ef49d93a9d2bc216Steve Naroff 315352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregorstatic void PrintDiagnosticCallback(CXDiagnostic Diagnostic, 325352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor CXClientData ClientData); 335352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 341c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenekstatic unsigned CreateTranslationUnit(CXIndex Idx, const char *file, 351c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXTranslationUnit *TU) { 361c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 375352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor *TU = clang_createTranslationUnit(Idx, file, PrintDiagnosticCallback, 0); 381c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!TU) { 391c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Unable to load translation unit from '%s'!\n", file); 401c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 0; 411c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 421c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 431c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 441c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 454db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregorvoid free_remapped_files(struct CXUnsavedFile *unsaved_files, 464db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int num_unsaved_files) { 474db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int i; 484db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (i = 0; i != num_unsaved_files; ++i) { 494db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free((char *)unsaved_files[i].Filename); 504db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free((char *)unsaved_files[i].Contents); 514db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 524db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 534db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 544db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregorint parse_remapped_files(int argc, const char **argv, int start_arg, 554db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile **unsaved_files, 564db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int *num_unsaved_files) { 574db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int i; 584db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int arg; 594db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int prefix_len = strlen("-remap-file="); 604db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 614db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 624db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 634db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Count the number of remapped files. */ 644db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (arg = start_arg; arg < argc; ++arg) { 654db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (strncmp(argv[arg], "-remap-file=", prefix_len)) 664db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor break; 674db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 684db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor ++*num_unsaved_files; 694db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 704db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 714db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (*num_unsaved_files == 0) 724db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return 0; 734db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 744db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files 754db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor = (struct CXUnsavedFile *)malloc(sizeof(struct CXUnsavedFile) * 764db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files); 774db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (arg = start_arg, i = 0; i != *num_unsaved_files; ++i, ++arg) { 784db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile *unsaved = *unsaved_files + i; 794db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor const char *arg_string = argv[arg] + prefix_len; 804db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int filename_len; 814db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor char *filename; 824db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor char *contents; 834db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor FILE *to_file; 844db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor const char *semi = strchr(arg_string, ';'); 854db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (!semi) { 864db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, 874db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor "error: -remap-file=from;to argument is missing semicolon\n"); 884db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 894db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 904db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 914db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 924db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 934db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 944db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Open the file that we're remapping to. */ 954db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor to_file = fopen(semi + 1, "r"); 964db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (!to_file) { 974db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, "error: cannot open file %s that we are remapping to\n", 984db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor semi + 1); 994db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 1004db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 1014db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 1024db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 1034db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 1044db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1054db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Determine the length of the file we're remapping to. */ 1064db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fseek(to_file, 0, SEEK_END); 1074db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Length = ftell(to_file); 1084db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fseek(to_file, 0, SEEK_SET); 1094db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1104db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Read the contents of the file we're remapping to. */ 1114db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor contents = (char *)malloc(unsaved->Length + 1); 1124db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (fread(contents, 1, unsaved->Length, to_file) != unsaved->Length) { 1134db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, "error: unexpected %s reading 'to' file %s\n", 1144db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor (feof(to_file) ? "EOF" : "error"), semi + 1); 1154db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fclose(to_file); 1164db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 1174db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 1184db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 1194db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 1204db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 1214db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor contents[unsaved->Length] = 0; 1224db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Contents = contents; 1234db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1244db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Close the file. */ 1254db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fclose(to_file); 1264db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1274db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Copy the file name that we're remapping from. */ 1284db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename_len = semi - arg_string; 1294db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename = (char *)malloc(filename_len + 1); 1304db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor memcpy(filename, arg_string, filename_len); 1314db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename[filename_len] = 0; 1324db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Filename = filename; 1334db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 1344db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1354db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return 0; 1364db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 1374db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1380d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1390d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Pretty-printing. */ 1400d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1410d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 142af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroffstatic void PrintCursor(CXCursor Cursor) { 14377128ddd3077fc045751a55bb3226802b15d5510Steve Naroff if (clang_isInvalid(Cursor.kind)) 1441c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printf("Invalid Cursor => %s", clang_getCursorKindSpelling(Cursor.kind)); 145699a07d8a0b1579c5178b3baf4bcf9edb6b38108Steve Naroff else { 146ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff CXString string; 147c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor CXCursor Referenced; 1481db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned line, column; 149ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff string = clang_getCursorSpelling(Cursor); 150ff9e18cb3863defa98257500ef49d93a9d2bc216Steve Naroff printf("%s=%s", clang_getCursorKindSpelling(Cursor.kind), 151ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff clang_getCString(string)); 152ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff clang_disposeString(string); 153c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor 154c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor Referenced = clang_getCursorReferenced(Cursor); 155c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor if (!clang_equalCursors(Referenced, clang_getNullCursor())) { 156c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Referenced); 15746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 1581db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor printf(":%d:%d", line, column); 159c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor } 160b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor 161b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor if (clang_isCursorDefinition(Cursor)) 162b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor printf(" (Definition)"); 163699a07d8a0b1579c5178b3baf4bcf9edb6b38108Steve Naroff } 164af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff} 16589922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff 1669298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenekstatic const char* GetCursorSource(CXCursor Cursor) { 1671db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Cursor); 1681db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor const char *source; 1691db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile file; 17046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, &file, 0, 0, 0); 1711db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor source = clang_getFileName(file); 1729298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek if (!source) 1739298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek return "<invalid loc>"; 1749298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek return basename(source); 1759298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek} 1769298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek 1770d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 178ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/* Callbacks. */ 179ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/******************************************************************************/ 180ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 181ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenektypedef void (*PostVisitTU)(CXTranslationUnit); 182ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 1835352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregorstatic void PrintDiagnosticCallback(CXDiagnostic Diagnostic, 1845352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor CXClientData ClientData) { 1855352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor FILE *out = (FILE *)ClientData; 1865352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor CXFile file; 1875352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor unsigned line, column; 1885352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor CXString text; 1895352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor enum CXDiagnosticSeverity severity = clang_getDiagnosticSeverity(Diagnostic); 1905352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 1915352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor /* Ignore diagnostics that should be ignored. */ 1925352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor if (severity == CXDiagnostic_Ignored) 1935352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor return; 1945352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 1955352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor /* Print file:line:column. */ 1965352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor clang_getInstantiationLocation(clang_getDiagnosticLocation(Diagnostic), 1975352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor &file, &line, &column, 0); 1985352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor if (file) 1995352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor fprintf(out, "%s:%d:%d: ", clang_getFileName(file), line, column); 2005352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 2015352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor /* Print warning/error/etc. */ 2025352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor switch (severity) { 2035352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor case CXDiagnostic_Ignored: assert(0 && "impossible"); break; 2045352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor case CXDiagnostic_Note: fprintf(out, "note: "); break; 2055352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor case CXDiagnostic_Warning: fprintf(out, "warning: "); break; 2065352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor case CXDiagnostic_Error: fprintf(out, "error: "); break; 2075352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor case CXDiagnostic_Fatal: fprintf(out, "fatal error: "); break; 2085352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor } 2095352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 2105352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor text = clang_getDiagnosticSpelling(Diagnostic); 2115352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor if (clang_getCString(text)) 2125352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor fprintf(out, "%s\n", clang_getCString(text)); 2135352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor else 2145352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor fprintf(out, "<no diagnostic text>\n"); 2155352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor clang_disposeString(text); 2165352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor} 2175352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 218ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/******************************************************************************/ 219e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor/* Logic for testing traversal. */ 2200d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 2210d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 222fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenekstatic const char *FileCheckPrefix = "CHECK"; 223fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 224a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregorstatic void PrintCursorExtent(CXCursor C) { 225a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor CXSourceRange extent = clang_getCursorExtent(C); 2261db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile begin_file, end_file; 2271db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned begin_line, begin_column, end_line, end_column; 2281db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor 2291db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor clang_getInstantiationLocation(clang_getRangeStart(extent), 23046766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor &begin_file, &begin_line, &begin_column, 0); 2311db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor clang_getInstantiationLocation(clang_getRangeEnd(extent), 23246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor &end_file, &end_line, &end_column, 0); 2331db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor if (!begin_file || !end_file) 23470ee54258035c860ebc71f7e5f803b74f3186889Ted Kremenek return; 2351db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor 2361db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor printf(" [Extent=%d:%d:%d:%d]", begin_line, begin_column, 2371db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor end_line, end_column); 238fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek} 239fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 240e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor/* Data used by all of the visitors. */ 241e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregortypedef struct { 242e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXTranslationUnit TU; 243e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor enum CXCursorKind *Filter; 244e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor} VisitorData; 245fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 246625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 247e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorenum CXChildVisitResult FilteredPrintingVisitor(CXCursor Cursor, 248e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursor Parent, 249e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 250e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 251e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor if (!Data->Filter || (Cursor.kind == *(enum CXCursorKind *)Data->Filter)) { 25298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Cursor); 2531db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned line, column; 25446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 255fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek printf("// %s: %s:%d:%d: ", FileCheckPrefix, 2561db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor GetCursorSource(Cursor), line, column); 257af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff PrintCursor(Cursor); 258a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor PrintCursorExtent(Cursor); 25970ee54258035c860ebc71f7e5f803b74f3186889Ted Kremenek printf("\n"); 260e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Recurse; 261625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar } 262e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor 263e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 264625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar} 265c857ea4d22e1e6dd9ede1f0e84d48b157c6924fdSteve Naroff 266e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorstatic enum CXChildVisitResult FunctionScanVisitor(CXCursor Cursor, 267e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursor Parent, 268e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 269625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar const char *startBuf, *endBuf; 270625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar unsigned startLine, startColumn, endLine, endColumn, curLine, curColumn; 271625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXCursor Ref; 272e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 273625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 274b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor if (Cursor.kind != CXCursor_FunctionDecl || 275b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor !clang_isCursorDefinition(Cursor)) 276e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 277625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 278625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar clang_getDefinitionSpellingAndExtent(Cursor, &startBuf, &endBuf, 279625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar &startLine, &startColumn, 280625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar &endLine, &endColumn); 281625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar /* Probe the entire body, looking for both decls and refs. */ 282625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curLine = startLine; 283625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn = startColumn; 284625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 285625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar while (startBuf < endBuf) { 28698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor CXSourceLocation Loc; 2871db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile file; 28898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor const char *source = 0; 28998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor 290625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar if (*startBuf == '\n') { 291625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar startBuf++; 292625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curLine++; 293625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn = 1; 294625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar } else if (*startBuf != '\t') 295625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn++; 296fbcb2b716bee88c754684bd189913ed9f8c09086Ted Kremenek 29798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor Loc = clang_getCursorLocation(Cursor); 29846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, &file, 0, 0, 0); 2991db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor source = clang_getFileName(file); 30098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor if (source) { 301b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXSourceLocation RefLoc 302b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor = clang_getLocation(Data->TU, file, curLine, curColumn); 303b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor Ref = clang_getCursor(Data->TU, RefLoc); 30498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor if (Ref.kind == CXCursor_NoDeclFound) { 30598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor /* Nothing found here; that's fine. */ 30698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor } else if (Ref.kind != CXCursor_FunctionDecl) { 30798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor printf("// %s: %s:%d:%d: ", FileCheckPrefix, GetCursorSource(Ref), 30898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor curLine, curColumn); 30998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor PrintCursor(Ref); 31098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor printf("\n"); 31198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor } 3124ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff } 313625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar startBuf++; 3142d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff } 315e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor 316e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 31789922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff} 31850398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff 3197d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 3207d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/* USR testing. */ 3217d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 3227d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 323e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorenum CXChildVisitResult USRVisitor(CXCursor C, CXCursor parent, 324e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 325e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 326e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor if (!Data->Filter || (C.kind == *(enum CXCursorKind *)Data->Filter)) { 327cf84aa46def41cccf4dd4c51cd0543b70c11e4ebTed Kremenek CXString USR = clang_getCursorUSR(C); 3287d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (!USR.Spelling) { 3297d40562f83552b7295411e10ee887d8d55470679Ted Kremenek clang_disposeString(USR); 330e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 3317d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 3327d40562f83552b7295411e10ee887d8d55470679Ted Kremenek printf("// %s: %s %s", FileCheckPrefix, GetCursorSource(C), USR.Spelling); 333a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor PrintCursorExtent(C); 3347d40562f83552b7295411e10ee887d8d55470679Ted Kremenek printf("\n"); 3357d40562f83552b7295411e10ee887d8d55470679Ted Kremenek clang_disposeString(USR); 336e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor 337e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Recurse; 338e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor } 339e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor 340e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 3417d40562f83552b7295411e10ee887d8d55470679Ted Kremenek} 3427d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 3437d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 34416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/* Inclusion stack testing. */ 34516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/******************************************************************************/ 34616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 34716b55a71695a33c094383295cc7b7a2080e098daTed Kremenekvoid InclusionVisitor(CXFile includedFile, CXSourceLocation *includeStack, 34816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned includeStackLen, CXClientData data) { 34916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 35016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned i; 35116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek printf("file: %s\nincluded by:\n", clang_getFileName(includedFile)); 35216b55a71695a33c094383295cc7b7a2080e098daTed Kremenek for (i = 0; i < includeStackLen; ++i) { 35316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek CXFile includingFile; 35416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned line, column; 35516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek clang_getInstantiationLocation(includeStack[i], &includingFile, &line, 35616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek &column, 0); 35716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek printf(" %s:%d:%d\n", clang_getFileName(includingFile), line, column); 35816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek } 35916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek printf("\n"); 36016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek} 36116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 36216b55a71695a33c094383295cc7b7a2080e098daTed Kremenekvoid PrintInclusionStack(CXTranslationUnit TU) { 36316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek clang_getInclusions(TU, InclusionVisitor, NULL); 36416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek} 36516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 36616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/******************************************************************************/ 3677d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/* Loading ASTs/source. */ 3687d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 3697d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 370625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbarstatic int perform_test_load(CXIndex Idx, CXTranslationUnit TU, 37198271567b6f10be85bcc75d0bbb67416c8cfb83aTed Kremenek const char *filter, const char *prefix, 372ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek CXCursorVisitor Visitor, 373ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 374e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor 375fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek if (prefix) 376fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek FileCheckPrefix = prefix; 377e3ee02a324559829142f387f75daf3da118442beTed Kremenek 378e3ee02a324559829142f387f75daf3da118442beTed Kremenek if (Visitor) { 379e3ee02a324559829142f387f75daf3da118442beTed Kremenek enum CXCursorKind K = CXCursor_NotImplemented; 380e3ee02a324559829142f387f75daf3da118442beTed Kremenek enum CXCursorKind *ck = &K; 381e3ee02a324559829142f387f75daf3da118442beTed Kremenek VisitorData Data; 382fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 383e3ee02a324559829142f387f75daf3da118442beTed Kremenek /* Perform some simple filtering. */ 384e3ee02a324559829142f387f75daf3da118442beTed Kremenek if (!strcmp(filter, "all") || !strcmp(filter, "local")) ck = NULL; 385e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "category")) K = CXCursor_ObjCCategoryDecl; 386e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "interface")) K = CXCursor_ObjCInterfaceDecl; 387e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "protocol")) K = CXCursor_ObjCProtocolDecl; 388e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "function")) K = CXCursor_FunctionDecl; 389e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "typedef")) K = CXCursor_TypedefDecl; 390e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "scan-function")) Visitor = FunctionScanVisitor; 391e3ee02a324559829142f387f75daf3da118442beTed Kremenek else { 392e3ee02a324559829142f387f75daf3da118442beTed Kremenek fprintf(stderr, "Unknown filter for -test-load-tu: %s\n", filter); 393e3ee02a324559829142f387f75daf3da118442beTed Kremenek return 1; 394e3ee02a324559829142f387f75daf3da118442beTed Kremenek } 3950d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 396e3ee02a324559829142f387f75daf3da118442beTed Kremenek Data.TU = TU; 397e3ee02a324559829142f387f75daf3da118442beTed Kremenek Data.Filter = ck; 398e3ee02a324559829142f387f75daf3da118442beTed Kremenek clang_visitChildren(clang_getTranslationUnitCursor(TU), Visitor, &Data); 399e3ee02a324559829142f387f75daf3da118442beTed Kremenek } 400ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 401ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek if (PV) 402ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PV(TU); 403e3ee02a324559829142f387f75daf3da118442beTed Kremenek 4040d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek clang_disposeTranslationUnit(TU); 4050d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek return 0; 4060d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek} 4070d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 408fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenekint perform_test_load_tu(const char *file, const char *filter, 409ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek const char *prefix, CXCursorVisitor Visitor, 410ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 411625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXIndex Idx; 412625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXTranslationUnit TU; 413625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar Idx = clang_createIndex(/* excludeDeclsFromPCH */ 414936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor !strcmp(filter, "local") ? 1 : 0); 415625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 416625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar if (!CreateTranslationUnit(Idx, file, &TU)) 417625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar return 1; 418625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 419ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load(Idx, TU, filter, prefix, Visitor, PV); 420625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar} 421625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 422ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenekint perform_test_load_source(int argc, const char **argv, 423ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek const char *filter, CXCursorVisitor Visitor, 424ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 4258506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar const char *UseExternalASTs = 4268506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar getenv("CINDEXTEST_USE_EXTERNAL_AST_GENERATION"); 427ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar CXIndex Idx; 428ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar CXTranslationUnit TU; 4294db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 4304db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int num_unsaved_files = 0; 4314db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int result; 4324db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 433ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar Idx = clang_createIndex(/* excludeDeclsFromPCH */ 434936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor !strcmp(filter, "local") ? 1 : 0); 435ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 4368506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar if (UseExternalASTs && strlen(UseExternalASTs)) 4378506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar clang_setUseExternalASTGeneration(Idx, 1); 4388506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar 4394db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (parse_remapped_files(argc, argv, 0, &unsaved_files, &num_unsaved_files)) 4404db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 4414db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 4424db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor TU = clang_createTranslationUnitFromSourceFile(Idx, 0, 4434db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argc - num_unsaved_files, 4444db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argv + num_unsaved_files, 4454db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor num_unsaved_files, 4465352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor unsaved_files, 4475352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor PrintDiagnosticCallback, 4485352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor stderr); 449ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar if (!TU) { 450ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar fprintf(stderr, "Unable to load translation unit!\n"); 451ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar return 1; 452ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar } 453ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 454ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek result = perform_test_load(Idx, TU, filter, NULL, Visitor, PV); 4554db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 4564db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return result; 457ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar} 458ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 4590d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 4601c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/* Logic for testing clang_getCursor(). */ 4611c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/******************************************************************************/ 4621c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 4631c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenekstatic void print_cursor_file_scan(CXCursor cursor, 4641c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek unsigned start_line, unsigned start_col, 4651d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek unsigned end_line, unsigned end_col, 4661d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek const char *prefix) { 4679096a20fc0fd799911b738ce7a9134dbacaf64b7Ted Kremenek printf("// %s: ", FileCheckPrefix); 4681d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek if (prefix) 4691d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek printf("-%s", prefix); 4701d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek printf("{start_line=%d start_col=%d end_line=%d end_col=%d} ", 4711d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek start_line, start_col, end_line, end_col); 4721c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek PrintCursor(cursor); 4731c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printf("\n"); 4741c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 4751c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 4761d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenekstatic int perform_file_scan(const char *ast_file, const char *source_file, 4771d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek const char *prefix) { 4781c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXIndex Idx; 4791c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXTranslationUnit TU; 4801c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek FILE *fp; 4811c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek unsigned line; 4821c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXCursor prevCursor; 483b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXFile file; 4841c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek unsigned printed; 4851c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek unsigned start_line, start_col, last_line, last_col; 4861c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek size_t i; 4871c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 488936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor if (!(Idx = clang_createIndex(/* excludeDeclsFromPCH */ 1))) { 4891c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Could not create Index\n"); 4901c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 4911c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 4921c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 4931c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!CreateTranslationUnit(Idx, ast_file, &TU)) 4941c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 4951c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 4961c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if ((fp = fopen(source_file, "r")) == NULL) { 4971c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Could not open '%s'\n", source_file); 4981c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 4991c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 5001c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5011c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek line = 0; 5021c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek prevCursor = clang_getNullCursor(); 5031c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printed = 0; 5041c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek start_line = last_line = 1; 5051c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek start_col = last_col = 1; 5061c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 507b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor file = clang_getFile(TU, source_file); 5081c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek while (!feof(fp)) { 509a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer size_t len = 0; 510a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer int c; 511a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer 512a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer while ((c = fgetc(fp)) != EOF) { 513a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer len++; 514a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer if (c == '\n') 515a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer break; 516a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer } 517a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer 5181c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek ++line; 5191c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5201c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek for (i = 0; i < len ; ++i) { 5211c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXCursor cursor; 522b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor cursor = clang_getCursor(TU, clang_getLocation(TU, file, line, i+1)); 5231c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5241c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!clang_equalCursors(cursor, prevCursor) && 5251c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek prevCursor.kind != CXCursor_InvalidFile) { 5261c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek print_cursor_file_scan(prevCursor, start_line, start_col, 5271d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek last_line, last_col, prefix); 5281c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printed = 1; 5291c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek start_line = line; 5301c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek start_col = (unsigned) i+1; 5311c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 5321c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek else { 5331c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printed = 0; 5341c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 5351c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5361c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek prevCursor = cursor; 5371c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek last_line = line; 5381c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek last_col = (unsigned) i+1; 5391c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 5401c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 5411c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5421c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!printed && prevCursor.kind != CXCursor_InvalidFile) { 5431c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek print_cursor_file_scan(prevCursor, start_line, start_col, 5441d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek last_line, last_col, prefix); 5451c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 5461c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5471c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fclose(fp); 5481c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 0; 5491c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 5501c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5511c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/******************************************************************************/ 5520d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Logic for testing clang_codeComplete(). */ 5530d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 5540d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 5550c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor/* Parse file:line:column from the input string. Returns 0 on success, non-zero 5560c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor on failure. If successful, the pointer *filename will contain newly-allocated 5570c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor memory (that will be owned by the caller) to store the file name. */ 5580c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorint parse_file_line_column(const char *input, char **filename, unsigned *line, 559fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned *column, unsigned *second_line, 560fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned *second_column) { 56188d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor /* Find the second colon. */ 562fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *last_colon = strrchr(input, ':'); 563fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned values[4], i; 564fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned num_values = (second_line && second_column)? 4 : 2; 565fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 5660c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor char *endptr = 0; 567fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!last_colon || last_colon == input) { 568fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (num_values == 4) 569fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse filename:line:column:line:column in " 570fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor "'%s'\n", input); 571fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor else 572fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse filename:line:column in '%s'\n", input); 5730c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return 1; 5740c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 5750c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 576fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor for (i = 0; i != num_values; ++i) { 577fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *prev_colon; 57888d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor 579fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor /* Parse the next line or column. */ 580fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor values[num_values - i - 1] = strtol(last_colon + 1, &endptr, 10); 581fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (*endptr != 0 && *endptr != ':') { 582fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse %s in '%s'\n", 583fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (i % 2 ? "column" : "line"), input); 584fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return 1; 585fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 586fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 587fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (i + 1 == num_values) 588fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor break; 58988d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor 590fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor /* Find the previous colon. */ 591fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor prev_colon = last_colon - 1; 592fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor while (prev_colon != input && *prev_colon != ':') 593fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor --prev_colon; 594fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (prev_colon == input) { 595fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse %s in '%s'\n", 596fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (i % 2 == 0? "column" : "line"), input); 597fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return 1; 598fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 599fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 600fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor last_colon = prev_colon; 6010c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 602fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 603fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *line = values[0]; 604fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *column = values[1]; 6050c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 606fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (second_line && second_column) { 607fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *second_line = values[2]; 608fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *second_column = values[3]; 609fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 610fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 61188d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor /* Copy the file name. */ 612fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *filename = (char*)malloc(last_colon - input + 1); 613fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor memcpy(*filename, input, last_colon - input); 614fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (*filename)[last_colon - input] = 0; 6150c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return 0; 6160c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 6170c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 6180c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorconst char * 6190c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorclang_getCompletionChunkKindSpelling(enum CXCompletionChunkKind Kind) { 6200c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor switch (Kind) { 6210c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Optional: return "Optional"; 6220c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_TypedText: return "TypedText"; 6230c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Text: return "Text"; 6240c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Placeholder: return "Placeholder"; 6250c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Informative: return "Informative"; 6260c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_CurrentParameter: return "CurrentParameter"; 6270c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftParen: return "LeftParen"; 6280c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightParen: return "RightParen"; 6290c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftBracket: return "LeftBracket"; 6300c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightBracket: return "RightBracket"; 6310c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftBrace: return "LeftBrace"; 6320c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightBrace: return "RightBrace"; 6330c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftAngle: return "LeftAngle"; 6340c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightAngle: return "RightAngle"; 6350c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Comma: return "Comma"; 636ff5ce6eefc7c253ef6edf4d4bfc996fdd82d09aaDouglas Gregor case CXCompletionChunk_ResultType: return "ResultType"; 63701dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_Colon: return "Colon"; 63801dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_SemiColon: return "SemiColon"; 63901dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_Equal: return "Equal"; 64001dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_HorizontalSpace: return "HorizontalSpace"; 64101dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_VerticalSpace: return "VerticalSpace"; 6420c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 6430c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 6440c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return "Unknown"; 6450c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 6460c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 6473ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregorvoid print_completion_string(CXCompletionString completion_string, FILE *file) { 648f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar int I, N; 6493ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor 6503ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor N = clang_getNumCompletionChunks(completion_string); 6510c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor for (I = 0; I != N; ++I) { 652d5a2089663d81faee0de031deabc418fa53ecf3bDouglas Gregor const char *text = 0; 6530c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor enum CXCompletionChunkKind Kind 6543ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor = clang_getCompletionChunkKind(completion_string, I); 6553ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor 6563ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor if (Kind == CXCompletionChunk_Optional) { 6573ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "{Optional "); 6583ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor print_completion_string( 6593ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor clang_getCompletionChunkCompletionString(completion_string, I), 6603ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor file); 6613ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "}"); 6623ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor continue; 6633ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor } 6643ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor 665d5a2089663d81faee0de031deabc418fa53ecf3bDouglas Gregor text = clang_getCompletionChunkText(completion_string, I); 6660c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor fprintf(file, "{%s %s}", 6670c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor clang_getCompletionChunkKindSpelling(Kind), 6680c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor text? text : ""); 6690c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 6703ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor} 6713ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor 6723ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregorvoid print_completion_result(CXCompletionResult *completion_result, 6733ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor CXClientData client_data) { 6743ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor FILE *file = (FILE *)client_data; 6753ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "%s:", 6763ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor clang_getCursorKindSpelling(completion_result->CursorKind)); 6773ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor print_completion_string(completion_result->CompletionString, file); 6780c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor fprintf(file, "\n"); 6790c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 6800c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 681f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekint perform_code_completion(int argc, const char **argv) { 6820c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor const char *input = argv[1]; 6830c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor char *filename = 0; 6840c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor unsigned line; 6850c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor unsigned column; 686f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar CXIndex CIdx; 687f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek int errorCode; 688735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 689735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor int num_unsaved_files = 0; 690ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor CXCodeCompleteResults *results = 0; 691f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar 6920c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor input += strlen("-code-completion-at="); 693fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if ((errorCode = parse_file_line_column(input, &filename, &line, &column, 694fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 0, 0))) 695f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return errorCode; 6960c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 697735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor if (parse_remapped_files(argc, argv, 2, &unsaved_files, &num_unsaved_files)) 698735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor return -1; 699735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor 700936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor CIdx = clang_createIndex(0); 701ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor results = clang_codeComplete(CIdx, 702ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor argv[argc - 1], argc - num_unsaved_files - 3, 703ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor argv + num_unsaved_files + 2, 704ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor num_unsaved_files, unsaved_files, 705936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor filename, line, column, 706936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor PrintDiagnosticCallback, stderr); 707936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor 708ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor if (results) { 709ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor unsigned i, n = results->NumResults; 710ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor for (i = 0; i != n; ++i) 711ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor print_completion_result(results->Results + i, stdout); 712ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor clang_disposeCodeCompleteResults(results); 713ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor } 714ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor 7150c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor clang_disposeIndex(CIdx); 7160c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor free(filename); 717f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 718735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 719735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor 720f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return 0; 7210c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 7220c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 723f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregortypedef struct { 724f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor char *filename; 725f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned line; 726f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned column; 727f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor} CursorSourceLocation; 728f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 729f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregorint inspect_cursor_at(int argc, const char **argv) { 730f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXIndex CIdx; 731f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor int errorCode; 732f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 733f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor int num_unsaved_files = 0; 734f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXTranslationUnit TU; 735f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXCursor Cursor; 736f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CursorSourceLocation *Locations = 0; 737f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned NumLocations = 0, Loc; 7384db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 739f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor /* Count the number of locations. */ 740f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor while (strstr(argv[NumLocations+1], "-cursor-at=") == argv[NumLocations+1]) 741f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor ++NumLocations; 742f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 743f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor /* Parse the locations. */ 744f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor assert(NumLocations > 0 && "Unable to count locations?"); 745f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor Locations = (CursorSourceLocation *)malloc( 746f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor NumLocations * sizeof(CursorSourceLocation)); 747f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor for (Loc = 0; Loc < NumLocations; ++Loc) { 748f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor const char *input = argv[Loc + 1] + strlen("-cursor-at="); 749f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if ((errorCode = parse_file_line_column(input, &Locations[Loc].filename, 750f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor &Locations[Loc].line, 751fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor &Locations[Loc].column, 0, 0))) 752f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return errorCode; 753f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 754f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 755f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (parse_remapped_files(argc, argv, NumLocations + 1, &unsaved_files, 756f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor &num_unsaved_files)) 757f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return -1; 758f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 759936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor CIdx = clang_createIndex(0); 760f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor TU = clang_createTranslationUnitFromSourceFile(CIdx, argv[argc - 1], 761f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor argc - num_unsaved_files - 2 - NumLocations, 7624db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argv + num_unsaved_files + 1 + NumLocations, 7634db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor num_unsaved_files, 7645352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor unsaved_files, 7655352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor PrintDiagnosticCallback, 7665352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor stderr); 767f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (!TU) { 768f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor fprintf(stderr, "unable to parse input\n"); 769f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return -1; 770f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 771f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 772f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor for (Loc = 0; Loc < NumLocations; ++Loc) { 773b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXFile file = clang_getFile(TU, Locations[Loc].filename); 774b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor if (!file) 775b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor continue; 776b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor 777b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor Cursor = clang_getCursor(TU, 778b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor clang_getLocation(TU, file, Locations[Loc].line, 779b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor Locations[Loc].column)); 780f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor PrintCursor(Cursor); 781f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor printf("\n"); 782f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free(Locations[Loc].filename); 783f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 784f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 785f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor clang_disposeTranslationUnit(TU); 786f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor clang_disposeIndex(CIdx); 787f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free(Locations); 788f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 789f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return 0; 790f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor} 791f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 792fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorint perform_token_annotation(int argc, const char **argv) { 793fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *input = argv[1]; 794fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor char *filename = 0; 795fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned line, second_line; 796fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned column, second_column; 797fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXIndex CIdx; 798fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXTranslationUnit TU = 0; 799fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor int errorCode; 800fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 801fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor int num_unsaved_files = 0; 802fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXToken *tokens; 803fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned num_tokens; 804fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceRange range; 805fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceLocation startLoc, endLoc; 806fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXFile file = 0; 807fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXCursor *cursors = 0; 808fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned i; 809fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 810fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor input += strlen("-test-annotate-tokens="); 811fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if ((errorCode = parse_file_line_column(input, &filename, &line, &column, 812fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor &second_line, &second_column))) 813fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return errorCode; 814fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 815fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (parse_remapped_files(argc, argv, 2, &unsaved_files, &num_unsaved_files)) 816fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return -1; 817fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 818936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor CIdx = clang_createIndex(0); 819fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor TU = clang_createTranslationUnitFromSourceFile(CIdx, argv[argc - 1], 820fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor argc - num_unsaved_files - 3, 821fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor argv + num_unsaved_files + 2, 822fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor num_unsaved_files, 8235352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor unsaved_files, 8245352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor PrintDiagnosticCallback, 8255352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor stderr); 826fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!TU) { 827fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "unable to parse input\n"); 828fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeIndex(CIdx); 829fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(filename); 830fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 831fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return -1; 832fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 833fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = 0; 834fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 835fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor file = clang_getFile(TU, filename); 836fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!file) { 837fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "file %s is not in this translation unit\n", filename); 838fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 839fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor goto teardown; 840fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 841fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 842fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor startLoc = clang_getLocation(TU, file, line, column); 843fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (clang_equalLocations(clang_getNullLocation(), startLoc)) { 844fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "invalid source location %s:%d:%d\n", filename, line, 845fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor column); 846fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 847fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor goto teardown; 848fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 849fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 850fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor endLoc = clang_getLocation(TU, file, second_line, second_column); 851fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (clang_equalLocations(clang_getNullLocation(), endLoc)) { 852fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "invalid source location %s:%d:%d\n", filename, 853fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor second_line, second_column); 854fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 855fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor goto teardown; 856fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 857fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 858fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor range = clang_getRange(startLoc, endLoc); 859fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_tokenize(TU, range, &tokens, &num_tokens); 860fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor cursors = (CXCursor *)malloc(num_tokens * sizeof(CXCursor)); 861fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_annotateTokens(TU, tokens, num_tokens, cursors); 862fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor for (i = 0; i != num_tokens; ++i) { 863fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *kind = "<unknown>"; 864fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXString spelling = clang_getTokenSpelling(TU, tokens[i]); 865fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceRange extent = clang_getTokenExtent(TU, tokens[i]); 866fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned start_line, start_column, end_line, end_column; 867fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 868fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor switch (clang_getTokenKind(tokens[i])) { 869fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Punctuation: kind = "Punctuation"; break; 870fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Keyword: kind = "Keyword"; break; 871fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Identifier: kind = "Identifier"; break; 872fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Literal: kind = "Literal"; break; 873fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Comment: kind = "Comment"; break; 874fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 875fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_getInstantiationLocation(clang_getRangeStart(extent), 87646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor 0, &start_line, &start_column, 0); 877fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_getInstantiationLocation(clang_getRangeEnd(extent), 87846766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor 0, &end_line, &end_column, 0); 8790045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf("%s: \"%s\" [%d:%d - %d:%d]", kind, clang_getCString(spelling), 880fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor start_line, start_column, end_line, end_column); 8810045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor if (!clang_isInvalid(cursors[i].kind)) { 8820045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf(" "); 8830045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor PrintCursor(cursors[i]); 8840045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor } 8850045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf("\n"); 886fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 887fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(cursors); 888fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 889fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor teardown: 890fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeTranslationUnit(TU); 891fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeIndex(CIdx); 892fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(filename); 893fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 894fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return errorCode; 895fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} 896fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 8970d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 8980d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Command line processing. */ 8990d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 900f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 901e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorstatic CXCursorVisitor GetVisitor(const char *s) { 9027d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (s[0] == '\0') 903e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return FilteredPrintingVisitor; 9047d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (strcmp(s, "-usrs") == 0) 9057d40562f83552b7295411e10ee887d8d55470679Ted Kremenek return USRVisitor; 9067d40562f83552b7295411e10ee887d8d55470679Ted Kremenek return NULL; 9077d40562f83552b7295411e10ee887d8d55470679Ted Kremenek} 9087d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 909f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekstatic void print_usage(void) { 910f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek fprintf(stderr, 9110d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek "usage: c-index-test -code-completion-at=<site> <compiler arguments>\n" 912f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor " c-index-test -cursor-at=<site> <compiler arguments>\n" 9131d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek " c-index-test -test-file-scan <AST file> <source file> " 9141d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek "[FileCheck prefix]\n" 915fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek " c-index-test -test-load-tu <AST file> <symbol filter> " 916fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek "[FileCheck prefix]\n" 9177d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " c-index-test -test-load-tu-usrs <AST file> <symbol filter> " 9187d40562f83552b7295411e10ee887d8d55470679Ted Kremenek "[FileCheck prefix]\n" 9197d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " c-index-test -test-load-source <symbol filter> {<args>}*\n" 920fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor " c-index-test -test-load-source-usrs <symbol filter> {<args>}*\n"); 921f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor fprintf(stderr, 92216b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-annotate-tokens=<range> {<args>}*\n" 92316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-inclusion-stack-source {<args>}*\n" 92416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-inclusion-stack-tu <AST file>\n\n" 9257d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " <symbol filter> values:\n%s", 9260d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " all - load all symbols, including those from PCH\n" 9270d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " local - load all symbols except those in PCH\n" 9280d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " category - only load ObjC categories (non-PCH)\n" 9290d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " interface - only load ObjC interfaces (non-PCH)\n" 9300d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " protocol - only load ObjC protocols (non-PCH)\n" 9310d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " function - only load functions (non-PCH)\n" 932625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar " typedef - only load typdefs (non-PCH)\n" 933625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar " scan-function - scan function bodies (non-PCH)\n\n"); 934f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek} 935f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 936f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekint main(int argc, const char **argv) { 937f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek if (argc > 2 && strstr(argv[1], "-code-completion-at=") == argv[1]) 938f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return perform_code_completion(argc, argv); 939f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (argc > 2 && strstr(argv[1], "-cursor-at=") == argv[1]) 940f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return inspect_cursor_at(argc, argv); 9417d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strncmp(argv[1], "-test-load-tu", 13) == 0) { 942e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 13); 9437d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (I) 944ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load_tu(argv[2], argv[3], argc >= 5 ? argv[4] : 0, I, 945ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek NULL); 9467d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 9477d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strncmp(argv[1], "-test-load-source", 17) == 0) { 948e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 17); 9497d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (I) 950ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load_source(argc - 3, argv + 3, argv[2], I, NULL); 9517d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 9527d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strcmp(argv[1], "-test-file-scan") == 0) 9531d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek return perform_file_scan(argv[2], argv[3], 9541d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek argc >= 5 ? argv[4] : 0); 955fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor else if (argc > 2 && strstr(argv[1], "-test-annotate-tokens=") == argv[1]) 956fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return perform_token_annotation(argc, argv); 95716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek else if (argc > 2 && strcmp(argv[1], "-test-inclusion-stack-source") == 0) 95816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek return perform_test_load_source(argc - 2, argv + 2, "all", NULL, 95916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek PrintInclusionStack); 96016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek else if (argc > 2 && strcmp(argv[1], "-test-inclusion-stack-tu") == 0) 96116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek return perform_test_load_tu(argv[2], "all", NULL, NULL, 96216b55a71695a33c094383295cc7b7a2080e098daTed Kremenek PrintInclusionStack); 96316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 964f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek print_usage(); 96550398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff return 1; 96650398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff} 967