c-index-test.c revision c2a2b3c2c2da1086d76bf804633ed5c6c48ae722
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 3151b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbarstatic void PrintExtent(FILE *out, unsigned begin_line, unsigned begin_column, 3251b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar unsigned end_line, unsigned end_column) { 3351b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar fprintf(out, "[%d:%d - %d:%d]", begin_line, begin_column, 34d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar end_line, end_column); 3551b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar} 3651b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar 371c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenekstatic unsigned CreateTranslationUnit(CXIndex Idx, const char *file, 381c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXTranslationUnit *TU) { 39e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 40a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor *TU = clang_createTranslationUnit(Idx, file); 411c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!TU) { 421c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Unable to load translation unit from '%s'!\n", file); 431c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 0; 44e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 451c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 461c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 471c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 484db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregorvoid free_remapped_files(struct CXUnsavedFile *unsaved_files, 494db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int num_unsaved_files) { 504db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int i; 514db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (i = 0; i != num_unsaved_files; ++i) { 524db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free((char *)unsaved_files[i].Filename); 534db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free((char *)unsaved_files[i].Contents); 544db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 554db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 564db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 574db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregorint parse_remapped_files(int argc, const char **argv, int start_arg, 584db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile **unsaved_files, 594db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int *num_unsaved_files) { 604db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int i; 614db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int arg; 624db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int prefix_len = strlen("-remap-file="); 634db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 644db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 65e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 664db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Count the number of remapped files. */ 674db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (arg = start_arg; arg < argc; ++arg) { 684db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (strncmp(argv[arg], "-remap-file=", prefix_len)) 694db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor break; 70e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 714db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor ++*num_unsaved_files; 724db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 73e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 744db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (*num_unsaved_files == 0) 754db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return 0; 76e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 774db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files 78e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek = (struct CXUnsavedFile *)malloc(sizeof(struct CXUnsavedFile) * 794db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files); 804db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (arg = start_arg, i = 0; i != *num_unsaved_files; ++i, ++arg) { 814db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile *unsaved = *unsaved_files + i; 824db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor const char *arg_string = argv[arg] + prefix_len; 834db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int filename_len; 844db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor char *filename; 854db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor char *contents; 864db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor FILE *to_file; 874db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor const char *semi = strchr(arg_string, ';'); 884db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (!semi) { 89e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, 904db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor "error: -remap-file=from;to argument is missing semicolon\n"); 914db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 924db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 934db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 944db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 954db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 96e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 974db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Open the file that we're remapping to. */ 984db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor to_file = fopen(semi + 1, "r"); 994db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (!to_file) { 1004db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, "error: cannot open file %s that we are remapping to\n", 1014db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor semi + 1); 1024db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 1034db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 1044db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 1054db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 1064db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 107e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1084db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Determine the length of the file we're remapping to. */ 1094db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fseek(to_file, 0, SEEK_END); 1104db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Length = ftell(to_file); 1114db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fseek(to_file, 0, SEEK_SET); 112e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1134db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Read the contents of the file we're remapping to. */ 1144db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor contents = (char *)malloc(unsaved->Length + 1); 1154db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (fread(contents, 1, unsaved->Length, to_file) != unsaved->Length) { 1164db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, "error: unexpected %s reading 'to' file %s\n", 1174db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor (feof(to_file) ? "EOF" : "error"), semi + 1); 1184db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fclose(to_file); 1194db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 1204db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 1214db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 1224db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 1234db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 1244db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor contents[unsaved->Length] = 0; 1254db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Contents = contents; 126e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1274db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Close the file. */ 1284db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fclose(to_file); 129e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1304db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Copy the file name that we're remapping from. */ 1314db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename_len = semi - arg_string; 1324db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename = (char *)malloc(filename_len + 1); 1334db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor memcpy(filename, arg_string, filename_len); 1344db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename[filename_len] = 0; 1354db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Filename = filename; 1364db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 137e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1384db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return 0; 1394db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 1404db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1410d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1420d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Pretty-printing. */ 1430d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1440d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 145af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroffstatic void PrintCursor(CXCursor Cursor) { 146e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (clang_isInvalid(Cursor.kind)) { 147e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek CXString ks = clang_getCursorKindSpelling(Cursor.kind); 148e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("Invalid Cursor => %s", clang_getCString(ks)); 149e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_disposeString(ks); 150e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 151699a07d8a0b1579c5178b3baf4bcf9edb6b38108Steve Naroff else { 152e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek CXString string, ks; 153c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor CXCursor Referenced; 1541db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned line, column; 155e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 156e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek ks = clang_getCursorKindSpelling(Cursor.kind); 157ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff string = clang_getCursorSpelling(Cursor); 158e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("%s=%s", clang_getCString(ks), 159e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getCString(string)); 160e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_disposeString(ks); 161ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff clang_disposeString(string); 162e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 163c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor Referenced = clang_getCursorReferenced(Cursor); 164c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor if (!clang_equalCursors(Referenced, clang_getNullCursor())) { 165c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Referenced); 16646766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 1671db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor printf(":%d:%d", line, column); 168c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor } 169b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor 170b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor if (clang_isCursorDefinition(Cursor)) 171b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor printf(" (Definition)"); 172699a07d8a0b1579c5178b3baf4bcf9edb6b38108Steve Naroff } 173af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff} 17489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff 175e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekstatic const char* GetCursorSource(CXCursor Cursor) { 1761db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Cursor); 17774844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek CXString source; 1781db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile file; 17946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, &file, 0, 0, 0); 1801db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor source = clang_getFileName(file); 181e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (!clang_getCString(source)) { 18274844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(source); 18374844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek return "<invalid loc>"; 18474844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek } 18574844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek else { 186e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek const char *b = basename(clang_getCString(source)); 18774844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(source); 18874844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek return b; 18974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek } 1909298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek} 1919298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek 1920d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 193ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/* Callbacks. */ 194ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/******************************************************************************/ 195ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 196ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenektypedef void (*PostVisitTU)(CXTranslationUnit); 197ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 198a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregorvoid PrintDiagnostic(CXDiagnostic Diagnostic) { 199a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor FILE *out = stderr; 2005352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor CXFile file; 201274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor CXString Msg; 2020a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor unsigned display_opts = CXDiagnostic_DisplaySourceLocation 2030a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor | CXDiagnostic_DisplayColumn | CXDiagnostic_DisplaySourceRanges; 2040a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor unsigned i, num_fixits; 205274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor 2060a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (clang_getDiagnosticSeverity(Diagnostic) == CXDiagnostic_Ignored) 2075352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor return; 208e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 209274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor Msg = clang_formatDiagnostic(Diagnostic, display_opts); 210274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor fprintf(stderr, "%s\n", clang_getCString(Msg)); 211274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor clang_disposeString(Msg); 212274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor 2135352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor clang_getInstantiationLocation(clang_getDiagnosticLocation(Diagnostic), 2140a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor &file, 0, 0, 0); 2150a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (!file) 2160a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor return; 217e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 2180a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor num_fixits = clang_getDiagnosticNumFixIts(Diagnostic); 2190a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor for (i = 0; i != num_fixits; ++i) { 220473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXSourceRange range; 221473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXString insertion_text = clang_getDiagnosticFixIt(Diagnostic, i, &range); 222473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXSourceLocation start = clang_getRangeStart(range); 223473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXSourceLocation end = clang_getRangeEnd(range); 224473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor unsigned start_line, start_column, end_line, end_column; 225473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXFile start_file, end_file; 226473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_getInstantiationLocation(start, &start_file, &start_line, 227473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor &start_column, 0); 228473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_getInstantiationLocation(end, &end_file, &end_line, &end_column, 0); 229473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor if (clang_equalLocations(start, end)) { 230473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor /* Insertion. */ 231473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor if (start_file == file) 2320a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "FIX-IT: Insert \"%s\" at %d:%d\n", 233473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_getCString(insertion_text), start_line, start_column); 234473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor } else if (strcmp(clang_getCString(insertion_text), "") == 0) { 235473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor /* Removal. */ 2360a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (start_file == file && end_file == file) { 2370a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "FIX-IT: Remove "); 2380a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor PrintExtent(out, start_line, start_column, end_line, end_column); 2390a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "\n"); 240436f3f0400c633251e4071f81358c47bab964adfDouglas Gregor } 241473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor } else { 242473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor /* Replacement. */ 2430a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (start_file == end_file) { 2440a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "FIX-IT: Replace "); 2450a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor PrintExtent(out, start_line, start_column, end_line, end_column); 246473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor fprintf(out, " with \"%s\"\n", clang_getCString(insertion_text)); 24751c6d384551674facc19f745ecf6e289d28dc55fDouglas Gregor } 2480a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor break; 2490a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor } 250473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_disposeString(insertion_text); 25151c6d384551674facc19f745ecf6e289d28dc55fDouglas Gregor } 2525352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor} 2535352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 254a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregorvoid PrintDiagnostics(CXTranslationUnit TU) { 255a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor int i, n = clang_getNumDiagnostics(TU); 256a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor for (i = 0; i != n; ++i) { 257a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor CXDiagnostic Diag = clang_getDiagnostic(TU, i); 258a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostic(Diag); 259a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor clang_disposeDiagnostic(Diag); 260a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor } 261a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor} 262a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor 263ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/******************************************************************************/ 264e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor/* Logic for testing traversal. */ 2650d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 2660d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 267fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenekstatic const char *FileCheckPrefix = "CHECK"; 268fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 269a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregorstatic void PrintCursorExtent(CXCursor C) { 270a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor CXSourceRange extent = clang_getCursorExtent(C); 2711db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile begin_file, end_file; 2721db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned begin_line, begin_column, end_line, end_column; 273e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 2741db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor clang_getInstantiationLocation(clang_getRangeStart(extent), 27546766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor &begin_file, &begin_line, &begin_column, 0); 2761db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor clang_getInstantiationLocation(clang_getRangeEnd(extent), 27746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor &end_file, &end_line, &end_column, 0); 2781db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor if (!begin_file || !end_file) 27970ee54258035c860ebc71f7e5f803b74f3186889Ted Kremenek return; 2801db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor 28151b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar printf(" Extent="); 28251b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar PrintExtent(stdout, begin_line, begin_column, end_line, end_column); 283fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek} 284fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 285e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor/* Data used by all of the visitors. */ 286e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregortypedef struct { 287e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXTranslationUnit TU; 288e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor enum CXCursorKind *Filter; 289e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor} VisitorData; 290fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 291625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 292e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekenum CXChildVisitResult FilteredPrintingVisitor(CXCursor Cursor, 293e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursor Parent, 294e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 295e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 296e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor if (!Data->Filter || (Cursor.kind == *(enum CXCursorKind *)Data->Filter)) { 29798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Cursor); 2981db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned line, column; 29946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 300fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek printf("// %s: %s:%d:%d: ", FileCheckPrefix, 3011db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor GetCursorSource(Cursor), line, column); 302af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff PrintCursor(Cursor); 303a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor PrintCursorExtent(Cursor); 304e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("\n"); 305e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Recurse; 306625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar } 307e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 308e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 309625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar} 310c857ea4d22e1e6dd9ede1f0e84d48b157c6924fdSteve Naroff 311e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekstatic enum CXChildVisitResult FunctionScanVisitor(CXCursor Cursor, 312e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursor Parent, 313e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 314625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar const char *startBuf, *endBuf; 315625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar unsigned startLine, startColumn, endLine, endColumn, curLine, curColumn; 316625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXCursor Ref; 317e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 318625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 319b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor if (Cursor.kind != CXCursor_FunctionDecl || 320b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor !clang_isCursorDefinition(Cursor)) 321e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 322625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 323625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar clang_getDefinitionSpellingAndExtent(Cursor, &startBuf, &endBuf, 324625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar &startLine, &startColumn, 325625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar &endLine, &endColumn); 326625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar /* Probe the entire body, looking for both decls and refs. */ 327625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curLine = startLine; 328625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn = startColumn; 329625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 330625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar while (startBuf < endBuf) { 33198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor CXSourceLocation Loc; 3321db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile file; 33374844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek CXString source; 334e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 335625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar if (*startBuf == '\n') { 336625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar startBuf++; 337625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curLine++; 338625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn = 1; 339625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar } else if (*startBuf != '\t') 340625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn++; 341e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 34298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor Loc = clang_getCursorLocation(Cursor); 34346766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, &file, 0, 0, 0); 344e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 3451db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor source = clang_getFileName(file); 346e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (clang_getCString(source)) { 347b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXSourceLocation RefLoc 348b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor = clang_getLocation(Data->TU, file, curLine, curColumn); 349b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor Ref = clang_getCursor(Data->TU, RefLoc); 35098258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor if (Ref.kind == CXCursor_NoDeclFound) { 35198258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor /* Nothing found here; that's fine. */ 35298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor } else if (Ref.kind != CXCursor_FunctionDecl) { 35398258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor printf("// %s: %s:%d:%d: ", FileCheckPrefix, GetCursorSource(Ref), 35498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor curLine, curColumn); 35598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor PrintCursor(Ref); 35698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor printf("\n"); 35798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor } 3584ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff } 35974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(source); 360625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar startBuf++; 3612d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff } 362e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 363e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 36489922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff} 36550398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff 3667d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 3677d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/* USR testing. */ 3687d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 3697d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 370e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorenum CXChildVisitResult USRVisitor(CXCursor C, CXCursor parent, 371e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 372e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 373e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor if (!Data->Filter || (C.kind == *(enum CXCursorKind *)Data->Filter)) { 374cf84aa46def41cccf4dd4c51cd0543b70c11e4ebTed Kremenek CXString USR = clang_getCursorUSR(C); 375e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (!clang_getCString(USR)) { 3767d40562f83552b7295411e10ee887d8d55470679Ted Kremenek clang_disposeString(USR); 377e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 3787d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 379e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("// %s: %s %s", FileCheckPrefix, GetCursorSource(C), 380e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getCString(USR)); 381a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor PrintCursorExtent(C); 3827d40562f83552b7295411e10ee887d8d55470679Ted Kremenek printf("\n"); 3837d40562f83552b7295411e10ee887d8d55470679Ted Kremenek clang_disposeString(USR); 384e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 385e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Recurse; 386e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 387e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 388e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 3897d40562f83552b7295411e10ee887d8d55470679Ted Kremenek} 3907d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 3917d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 39216b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/* Inclusion stack testing. */ 39316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/******************************************************************************/ 39416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 39516b55a71695a33c094383295cc7b7a2080e098daTed Kremenekvoid InclusionVisitor(CXFile includedFile, CXSourceLocation *includeStack, 39616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned includeStackLen, CXClientData data) { 397e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 39816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned i; 39974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek CXString fname; 40074844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek 40174844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek fname = clang_getFileName(includedFile); 402e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("file: %s\nincluded by:\n", clang_getCString(fname)); 40374844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(fname); 404e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 40516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek for (i = 0; i < includeStackLen; ++i) { 40616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek CXFile includingFile; 40716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned line, column; 40816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek clang_getInstantiationLocation(includeStack[i], &includingFile, &line, 40916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek &column, 0); 41074844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek fname = clang_getFileName(includingFile); 411e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf(" %s:%d:%d\n", clang_getCString(fname), line, column); 41274844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(fname); 41316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek } 41416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek printf("\n"); 41516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek} 41616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 41716b55a71695a33c094383295cc7b7a2080e098daTed Kremenekvoid PrintInclusionStack(CXTranslationUnit TU) { 418e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getInclusions(TU, InclusionVisitor, NULL); 41916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek} 42016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 42116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/******************************************************************************/ 4223bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek/* Linkage testing. */ 4233bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek/******************************************************************************/ 4243bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 4253bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenekstatic enum CXChildVisitResult PrintLinkage(CXCursor cursor, CXCursor p, 4263bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek CXClientData d) { 4273bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek const char *linkage = 0; 4283bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 4293bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek if (clang_isInvalid(clang_getCursorKind(cursor))) 4303bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek return CXChildVisit_Recurse; 4313bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 4323bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek switch (clang_getCursorLinkage(cursor)) { 4333bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek case CXLinkage_Invalid: break; 434c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_NoLinkage: linkage = "NoLinkage"; break; 435c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_Internal: linkage = "Internal"; break; 436c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_UniqueExternal: linkage = "UniqueExternal"; break; 437c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_External: linkage = "External"; break; 4383bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek } 4393bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 4403bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek if (linkage) { 4413bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek PrintCursor(cursor); 4423bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek printf("linkage=%s\n", linkage); 4433bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek } 4443bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 4453bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek return CXChildVisit_Recurse; 4463bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek} 4473bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 4483bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek/******************************************************************************/ 4497d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/* Loading ASTs/source. */ 4507d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 4517d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 452625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbarstatic int perform_test_load(CXIndex Idx, CXTranslationUnit TU, 45398271567b6f10be85bcc75d0bbb67416c8cfb83aTed Kremenek const char *filter, const char *prefix, 454ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek CXCursorVisitor Visitor, 455ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 456e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 457fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek if (prefix) 458e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek FileCheckPrefix = prefix; 459e3ee02a324559829142f387f75daf3da118442beTed Kremenek 460e3ee02a324559829142f387f75daf3da118442beTed Kremenek if (Visitor) { 461e3ee02a324559829142f387f75daf3da118442beTed Kremenek enum CXCursorKind K = CXCursor_NotImplemented; 462e3ee02a324559829142f387f75daf3da118442beTed Kremenek enum CXCursorKind *ck = &K; 463e3ee02a324559829142f387f75daf3da118442beTed Kremenek VisitorData Data; 464e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 465e3ee02a324559829142f387f75daf3da118442beTed Kremenek /* Perform some simple filtering. */ 466e3ee02a324559829142f387f75daf3da118442beTed Kremenek if (!strcmp(filter, "all") || !strcmp(filter, "local")) ck = NULL; 467b1ffee6e574d57b79ce2defd1b57052649221c03Daniel Dunbar else if (!strcmp(filter, "none")) K = (enum CXCursorKind) ~0; 468e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "category")) K = CXCursor_ObjCCategoryDecl; 469e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "interface")) K = CXCursor_ObjCInterfaceDecl; 470e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "protocol")) K = CXCursor_ObjCProtocolDecl; 471e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "function")) K = CXCursor_FunctionDecl; 472e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "typedef")) K = CXCursor_TypedefDecl; 473e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "scan-function")) Visitor = FunctionScanVisitor; 474e3ee02a324559829142f387f75daf3da118442beTed Kremenek else { 475e3ee02a324559829142f387f75daf3da118442beTed Kremenek fprintf(stderr, "Unknown filter for -test-load-tu: %s\n", filter); 476e3ee02a324559829142f387f75daf3da118442beTed Kremenek return 1; 477e3ee02a324559829142f387f75daf3da118442beTed Kremenek } 478e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 479e3ee02a324559829142f387f75daf3da118442beTed Kremenek Data.TU = TU; 480e3ee02a324559829142f387f75daf3da118442beTed Kremenek Data.Filter = ck; 481e3ee02a324559829142f387f75daf3da118442beTed Kremenek clang_visitChildren(clang_getTranslationUnitCursor(TU), Visitor, &Data); 482e3ee02a324559829142f387f75daf3da118442beTed Kremenek } 483e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 484ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek if (PV) 485ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PV(TU); 486e3ee02a324559829142f387f75daf3da118442beTed Kremenek 487a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostics(TU); 4880d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek clang_disposeTranslationUnit(TU); 4890d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek return 0; 4900d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek} 4910d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 492fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenekint perform_test_load_tu(const char *file, const char *filter, 493ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek const char *prefix, CXCursorVisitor Visitor, 494ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 495625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXIndex Idx; 496625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXTranslationUnit TU; 497020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek int result; 498e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek Idx = clang_createIndex(/* excludeDeclsFromPCH */ 4990a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor !strcmp(filter, "local") ? 1 : 0, 5000a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor /* displayDiagnosics=*/1); 501e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 502020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek if (!CreateTranslationUnit(Idx, file, &TU)) { 503020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 504625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar return 1; 505020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek } 506625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 507020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek result = perform_test_load(Idx, TU, filter, prefix, Visitor, PV); 508020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 509020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek return result; 510625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar} 511625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 512ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenekint perform_test_load_source(int argc, const char **argv, 513ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek const char *filter, CXCursorVisitor Visitor, 514ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 5158506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar const char *UseExternalASTs = 5168506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar getenv("CINDEXTEST_USE_EXTERNAL_AST_GENERATION"); 517ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar CXIndex Idx; 518ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar CXTranslationUnit TU; 5194db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 5204db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int num_unsaved_files = 0; 5214db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int result; 522e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 523ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar Idx = clang_createIndex(/* excludeDeclsFromPCH */ 5240a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor !strcmp(filter, "local") ? 1 : 0, 5250a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor /* displayDiagnosics=*/1); 526ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 5278506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar if (UseExternalASTs && strlen(UseExternalASTs)) 5288506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar clang_setUseExternalASTGeneration(Idx, 1); 5298506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar 530020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek if (parse_remapped_files(argc, argv, 0, &unsaved_files, &num_unsaved_files)) { 531020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 5324db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 533020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek } 5344db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 535e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek TU = clang_createTranslationUnitFromSourceFile(Idx, 0, 536e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek argc - num_unsaved_files, 5374db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argv + num_unsaved_files, 5384db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor num_unsaved_files, 539a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsaved_files); 540ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar if (!TU) { 541ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar fprintf(stderr, "Unable to load translation unit!\n"); 542020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 543ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar return 1; 544ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar } 545ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 546ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek result = perform_test_load(Idx, TU, filter, NULL, Visitor, PV); 5474db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 548020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 5494db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return result; 550ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar} 551ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 5520d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 5531c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/* Logic for testing clang_getCursor(). */ 5541c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/******************************************************************************/ 5551c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5561c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenekstatic void print_cursor_file_scan(CXCursor cursor, 5571c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek unsigned start_line, unsigned start_col, 5581d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek unsigned end_line, unsigned end_col, 5591d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek const char *prefix) { 5609096a20fc0fd799911b738ce7a9134dbacaf64b7Ted Kremenek printf("// %s: ", FileCheckPrefix); 5611d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek if (prefix) 5621d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek printf("-%s", prefix); 56351b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar PrintExtent(stdout, start_line, start_col, end_line, end_col); 56451b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar printf(" "); 5651c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek PrintCursor(cursor); 5661c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printf("\n"); 5671c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 5681c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 5691d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenekstatic int perform_file_scan(const char *ast_file, const char *source_file, 5701d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek const char *prefix) { 5711c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXIndex Idx; 5721c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXTranslationUnit TU; 5731c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek FILE *fp; 5742389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar CXCursor prevCursor = clang_getNullCursor(); 575b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXFile file; 5762389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar unsigned line = 1, col = 1; 5778f0bf81c0f3ab31881cc7db7914e6978bb2a19afDaniel Dunbar unsigned start_line = 1, start_col = 1; 578e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 5790a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (!(Idx = clang_createIndex(/* excludeDeclsFromPCH */ 1, 5800a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor /* displayDiagnosics=*/1))) { 5811c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Could not create Index\n"); 5821c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 5831c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 584e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 5851c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!CreateTranslationUnit(Idx, ast_file, &TU)) 5861c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 587e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 5881c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if ((fp = fopen(source_file, "r")) == NULL) { 5891c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Could not open '%s'\n", source_file); 5901c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 5911c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 592e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 593b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor file = clang_getFile(TU, source_file); 5942389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar for (;;) { 5952389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar CXCursor cursor; 5962389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar int c = fgetc(fp); 5972389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar 5982389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar if (c == '\n') { 5992389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar ++line; 6002389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar col = 1; 6012389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar } else 6022389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar ++col; 6032389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar 6042389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar /* Check the cursor at this position, and dump the previous one if we have 6052389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar * found something new. 6062389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar */ 6072389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar cursor = clang_getCursor(TU, clang_getLocation(TU, file, line, col)); 6082389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar if ((c == EOF || !clang_equalCursors(cursor, prevCursor)) && 6092389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar prevCursor.kind != CXCursor_InvalidFile) { 6102389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar print_cursor_file_scan(prevCursor, start_line, start_col, 611d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar line, col, prefix); 6122389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar start_line = line; 6132389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar start_col = col; 614a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer } 6152389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar if (c == EOF) 6162389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar break; 617a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer 6182389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar prevCursor = cursor; 6191c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 620e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 6211c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fclose(fp); 6221c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 0; 6231c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 6241c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 6251c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/******************************************************************************/ 6260d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Logic for testing clang_codeComplete(). */ 6270d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 6280d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 6290c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor/* Parse file:line:column from the input string. Returns 0 on success, non-zero 6300c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor on failure. If successful, the pointer *filename will contain newly-allocated 6310c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor memory (that will be owned by the caller) to store the file name. */ 632e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekint parse_file_line_column(const char *input, char **filename, unsigned *line, 633fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned *column, unsigned *second_line, 634fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned *second_column) { 63588d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor /* Find the second colon. */ 636fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *last_colon = strrchr(input, ':'); 637fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned values[4], i; 638fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned num_values = (second_line && second_column)? 4 : 2; 639fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 6400c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor char *endptr = 0; 641fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!last_colon || last_colon == input) { 642fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (num_values == 4) 643fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse filename:line:column:line:column in " 644fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor "'%s'\n", input); 645fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor else 646fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse filename:line:column in '%s'\n", input); 6470c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return 1; 6480c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 6490c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 650fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor for (i = 0; i != num_values; ++i) { 651fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *prev_colon; 65288d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor 653fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor /* Parse the next line or column. */ 654fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor values[num_values - i - 1] = strtol(last_colon + 1, &endptr, 10); 655fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (*endptr != 0 && *endptr != ':') { 656e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "could not parse %s in '%s'\n", 657fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (i % 2 ? "column" : "line"), input); 658fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return 1; 659fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 660e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 661fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (i + 1 == num_values) 662fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor break; 66388d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor 664fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor /* Find the previous colon. */ 665fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor prev_colon = last_colon - 1; 666fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor while (prev_colon != input && *prev_colon != ':') 667fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor --prev_colon; 668fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (prev_colon == input) { 669e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "could not parse %s in '%s'\n", 670fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (i % 2 == 0? "column" : "line"), input); 671e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek return 1; 672fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 673fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 674fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor last_colon = prev_colon; 6750c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 676fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 677fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *line = values[0]; 678fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *column = values[1]; 679e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 680fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (second_line && second_column) { 681fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *second_line = values[2]; 682fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *second_column = values[3]; 683fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 684fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 68588d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor /* Copy the file name. */ 686fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *filename = (char*)malloc(last_colon - input + 1); 687fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor memcpy(*filename, input, last_colon - input); 688fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (*filename)[last_colon - input] = 0; 6890c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return 0; 6900c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 6910c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 6920c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorconst char * 6930c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorclang_getCompletionChunkKindSpelling(enum CXCompletionChunkKind Kind) { 6940c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor switch (Kind) { 6950c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Optional: return "Optional"; 6960c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_TypedText: return "TypedText"; 6970c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Text: return "Text"; 6980c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Placeholder: return "Placeholder"; 6990c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Informative: return "Informative"; 7000c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_CurrentParameter: return "CurrentParameter"; 7010c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftParen: return "LeftParen"; 7020c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightParen: return "RightParen"; 7030c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftBracket: return "LeftBracket"; 7040c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightBracket: return "RightBracket"; 7050c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftBrace: return "LeftBrace"; 7060c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightBrace: return "RightBrace"; 7070c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftAngle: return "LeftAngle"; 7080c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightAngle: return "RightAngle"; 7090c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Comma: return "Comma"; 710ff5ce6eefc7c253ef6edf4d4bfc996fdd82d09aaDouglas Gregor case CXCompletionChunk_ResultType: return "ResultType"; 71101dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_Colon: return "Colon"; 71201dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_SemiColon: return "SemiColon"; 71301dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_Equal: return "Equal"; 71401dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_HorizontalSpace: return "HorizontalSpace"; 71501dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_VerticalSpace: return "VerticalSpace"; 7160c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 717e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7180c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return "Unknown"; 7190c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 7200c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 7213ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregorvoid print_completion_string(CXCompletionString completion_string, FILE *file) { 722f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar int I, N; 723e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7243ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor N = clang_getNumCompletionChunks(completion_string); 7250c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor for (I = 0; I != N; ++I) { 7262ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek CXString text; 7272ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek const char *cstr; 7280c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor enum CXCompletionChunkKind Kind 7293ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor = clang_getCompletionChunkKind(completion_string, I); 730e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7313ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor if (Kind == CXCompletionChunk_Optional) { 7323ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "{Optional "); 7333ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor print_completion_string( 734e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getCompletionChunkCompletionString(completion_string, I), 7353ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor file); 7363ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "}"); 7373ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor continue; 7383ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor } 739e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 740d5a2089663d81faee0de031deabc418fa53ecf3bDouglas Gregor text = clang_getCompletionChunkText(completion_string, I); 7412ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek cstr = clang_getCString(text); 742e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(file, "{%s %s}", 7430c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor clang_getCompletionChunkKindSpelling(Kind), 7442ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek cstr ? cstr : ""); 7452ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek clang_disposeString(text); 7460c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 7472ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek 7483ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor} 7493ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor 7503ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregorvoid print_completion_result(CXCompletionResult *completion_result, 7513ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor CXClientData client_data) { 7523ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor FILE *file = (FILE *)client_data; 753e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek CXString ks = clang_getCursorKindSpelling(completion_result->CursorKind); 754e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 755e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(file, "%s:", clang_getCString(ks)); 756e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_disposeString(ks); 757e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7583ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor print_completion_string(completion_result->CompletionString, file); 7590c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor fprintf(file, "\n"); 7600c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 7610c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 762f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekint perform_code_completion(int argc, const char **argv) { 7630c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor const char *input = argv[1]; 7640c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor char *filename = 0; 7650c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor unsigned line; 7660c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor unsigned column; 767f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar CXIndex CIdx; 768f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek int errorCode; 769735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 770735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor int num_unsaved_files = 0; 771ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor CXCodeCompleteResults *results = 0; 772f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar 7730c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor input += strlen("-code-completion-at="); 774e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if ((errorCode = parse_file_line_column(input, &filename, &line, &column, 775fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 0, 0))) 776f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return errorCode; 7770c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 778735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor if (parse_remapped_files(argc, argv, 2, &unsaved_files, &num_unsaved_files)) 779735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor return -1; 780735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor 7810a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor CIdx = clang_createIndex(0, 1); 782e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek results = clang_codeComplete(CIdx, 783e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek argv[argc - 1], argc - num_unsaved_files - 3, 784e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek argv + num_unsaved_files + 2, 785ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor num_unsaved_files, unsaved_files, 786a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor filename, line, column); 787936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor 788ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor if (results) { 789ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor unsigned i, n = results->NumResults; 790ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor for (i = 0; i != n; ++i) 791ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor print_completion_result(results->Results + i, stdout); 792a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor n = clang_codeCompleteGetNumDiagnostics(results); 793a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor for (i = 0; i != n; ++i) { 794a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor CXDiagnostic diag = clang_codeCompleteGetDiagnostic(results, i); 795a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostic(diag); 796a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor clang_disposeDiagnostic(diag); 797a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor } 798ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor clang_disposeCodeCompleteResults(results); 799ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor } 800ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor 8010c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor clang_disposeIndex(CIdx); 8020c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor free(filename); 803e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 804735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 805735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor 806f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return 0; 8070c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 8080c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 809f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregortypedef struct { 810f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor char *filename; 811f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned line; 812f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned column; 813f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor} CursorSourceLocation; 814f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 815f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregorint inspect_cursor_at(int argc, const char **argv) { 816f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXIndex CIdx; 817f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor int errorCode; 818f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 819f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor int num_unsaved_files = 0; 820f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXTranslationUnit TU; 821f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXCursor Cursor; 822f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CursorSourceLocation *Locations = 0; 823f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned NumLocations = 0, Loc; 8244db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 825e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek /* Count the number of locations. */ 826f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor while (strstr(argv[NumLocations+1], "-cursor-at=") == argv[NumLocations+1]) 827f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor ++NumLocations; 828e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 829f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor /* Parse the locations. */ 830f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor assert(NumLocations > 0 && "Unable to count locations?"); 831f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor Locations = (CursorSourceLocation *)malloc( 832f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor NumLocations * sizeof(CursorSourceLocation)); 833f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor for (Loc = 0; Loc < NumLocations; ++Loc) { 834f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor const char *input = argv[Loc + 1] + strlen("-cursor-at="); 835e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if ((errorCode = parse_file_line_column(input, &Locations[Loc].filename, 836e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek &Locations[Loc].line, 837fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor &Locations[Loc].column, 0, 0))) 838f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return errorCode; 839f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 840e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 841e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (parse_remapped_files(argc, argv, NumLocations + 1, &unsaved_files, 842f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor &num_unsaved_files)) 843f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return -1; 844e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 8450a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor CIdx = clang_createIndex(0, 1); 846f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor TU = clang_createTranslationUnitFromSourceFile(CIdx, argv[argc - 1], 847f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor argc - num_unsaved_files - 2 - NumLocations, 8484db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argv + num_unsaved_files + 1 + NumLocations, 8494db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor num_unsaved_files, 850a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsaved_files); 851f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (!TU) { 852f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor fprintf(stderr, "unable to parse input\n"); 853f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return -1; 854f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 855e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 856f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor for (Loc = 0; Loc < NumLocations; ++Loc) { 857b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXFile file = clang_getFile(TU, Locations[Loc].filename); 858b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor if (!file) 859b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor continue; 860e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 861e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek Cursor = clang_getCursor(TU, 862e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getLocation(TU, file, Locations[Loc].line, 863e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek Locations[Loc].column)); 864f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor PrintCursor(Cursor); 865f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor printf("\n"); 866f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free(Locations[Loc].filename); 867f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 868e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 869a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostics(TU); 870f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor clang_disposeTranslationUnit(TU); 871f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor clang_disposeIndex(CIdx); 872f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free(Locations); 873f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 874f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return 0; 875f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor} 876f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 877fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorint perform_token_annotation(int argc, const char **argv) { 878fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *input = argv[1]; 879fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor char *filename = 0; 880fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned line, second_line; 881fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned column, second_column; 882fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXIndex CIdx; 883fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXTranslationUnit TU = 0; 884fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor int errorCode; 885fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 886fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor int num_unsaved_files = 0; 887fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXToken *tokens; 888fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned num_tokens; 889fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceRange range; 890fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceLocation startLoc, endLoc; 891fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXFile file = 0; 892fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXCursor *cursors = 0; 893fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned i; 894fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 895fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor input += strlen("-test-annotate-tokens="); 896fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if ((errorCode = parse_file_line_column(input, &filename, &line, &column, 897fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor &second_line, &second_column))) 898fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return errorCode; 899fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 900fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (parse_remapped_files(argc, argv, 2, &unsaved_files, &num_unsaved_files)) 901fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return -1; 902fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 9030a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor CIdx = clang_createIndex(0, 1); 904fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor TU = clang_createTranslationUnitFromSourceFile(CIdx, argv[argc - 1], 905fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor argc - num_unsaved_files - 3, 906fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor argv + num_unsaved_files + 2, 907fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor num_unsaved_files, 908a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsaved_files); 909fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!TU) { 910fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "unable to parse input\n"); 911fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeIndex(CIdx); 912fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(filename); 913fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 914fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return -1; 915e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 916fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = 0; 917fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 918fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor file = clang_getFile(TU, filename); 919fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!file) { 920fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "file %s is not in this translation unit\n", filename); 921fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 922fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor goto teardown; 923fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 924fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 925fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor startLoc = clang_getLocation(TU, file, line, column); 926fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (clang_equalLocations(clang_getNullLocation(), startLoc)) { 927e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "invalid source location %s:%d:%d\n", filename, line, 928fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor column); 929fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 930e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek goto teardown; 931fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 932fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 933fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor endLoc = clang_getLocation(TU, file, second_line, second_column); 934fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (clang_equalLocations(clang_getNullLocation(), endLoc)) { 935e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "invalid source location %s:%d:%d\n", filename, 936fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor second_line, second_column); 937fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 938e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek goto teardown; 939fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 940fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 941fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor range = clang_getRange(startLoc, endLoc); 942fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_tokenize(TU, range, &tokens, &num_tokens); 943fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor cursors = (CXCursor *)malloc(num_tokens * sizeof(CXCursor)); 944fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_annotateTokens(TU, tokens, num_tokens, cursors); 945fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor for (i = 0; i != num_tokens; ++i) { 946fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *kind = "<unknown>"; 947fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXString spelling = clang_getTokenSpelling(TU, tokens[i]); 948fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceRange extent = clang_getTokenExtent(TU, tokens[i]); 949fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned start_line, start_column, end_line, end_column; 950fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 951fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor switch (clang_getTokenKind(tokens[i])) { 952fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Punctuation: kind = "Punctuation"; break; 953fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Keyword: kind = "Keyword"; break; 954fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Identifier: kind = "Identifier"; break; 955fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Literal: kind = "Literal"; break; 956fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Comment: kind = "Comment"; break; 957fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 958e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getInstantiationLocation(clang_getRangeStart(extent), 95946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor 0, &start_line, &start_column, 0); 960fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_getInstantiationLocation(clang_getRangeEnd(extent), 96146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor 0, &end_line, &end_column, 0); 96251b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar printf("%s: \"%s\" ", kind, clang_getCString(spelling)); 96351b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar PrintExtent(stdout, start_line, start_column, end_line, end_column); 9640045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor if (!clang_isInvalid(cursors[i].kind)) { 9650045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf(" "); 9660045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor PrintCursor(cursors[i]); 9670045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor } 9680045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf("\n"); 969fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 970fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(cursors); 971fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 972fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor teardown: 973a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostics(TU); 974fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeTranslationUnit(TU); 975fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeIndex(CIdx); 976fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(filename); 977fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 978fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return errorCode; 979fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} 980fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 9810d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 9820d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Command line processing. */ 9830d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 984f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 985e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorstatic CXCursorVisitor GetVisitor(const char *s) { 9867d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (s[0] == '\0') 987e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return FilteredPrintingVisitor; 9887d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (strcmp(s, "-usrs") == 0) 9897d40562f83552b7295411e10ee887d8d55470679Ted Kremenek return USRVisitor; 9907d40562f83552b7295411e10ee887d8d55470679Ted Kremenek return NULL; 9917d40562f83552b7295411e10ee887d8d55470679Ted Kremenek} 9927d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 993f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekstatic void print_usage(void) { 994f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek fprintf(stderr, 9950d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek "usage: c-index-test -code-completion-at=<site> <compiler arguments>\n" 996f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor " c-index-test -cursor-at=<site> <compiler arguments>\n" 9971d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek " c-index-test -test-file-scan <AST file> <source file> " 9981d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek "[FileCheck prefix]\n" 999fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek " c-index-test -test-load-tu <AST file> <symbol filter> " 1000fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek "[FileCheck prefix]\n" 10017d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " c-index-test -test-load-tu-usrs <AST file> <symbol filter> " 10027d40562f83552b7295411e10ee887d8d55470679Ted Kremenek "[FileCheck prefix]\n" 10037d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " c-index-test -test-load-source <symbol filter> {<args>}*\n" 1004fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor " c-index-test -test-load-source-usrs <symbol filter> {<args>}*\n"); 1005f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor fprintf(stderr, 100616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-annotate-tokens=<range> {<args>}*\n" 100716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-inclusion-stack-source {<args>}*\n" 10083bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek " c-index-test -test-inclusion-stack-tu <AST file>\n" 10093bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek " c-index-test -test-print-linkage-source {<args>}*\n\n" 10107d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " <symbol filter> values:\n%s", 10110d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " all - load all symbols, including those from PCH\n" 10120d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " local - load all symbols except those in PCH\n" 10130d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " category - only load ObjC categories (non-PCH)\n" 10140d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " interface - only load ObjC interfaces (non-PCH)\n" 10150d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " protocol - only load ObjC protocols (non-PCH)\n" 10160d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " function - only load functions (non-PCH)\n" 1017625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar " typedef - only load typdefs (non-PCH)\n" 1018625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar " scan-function - scan function bodies (non-PCH)\n\n"); 1019f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek} 1020f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 1021f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekint main(int argc, const char **argv) { 10220a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor clang_enableStackTraces(); 1023f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek if (argc > 2 && strstr(argv[1], "-code-completion-at=") == argv[1]) 1024f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return perform_code_completion(argc, argv); 1025f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (argc > 2 && strstr(argv[1], "-cursor-at=") == argv[1]) 1026f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return inspect_cursor_at(argc, argv); 10277d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strncmp(argv[1], "-test-load-tu", 13) == 0) { 1028e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 13); 10297d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (I) 1030ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load_tu(argv[2], argv[3], argc >= 5 ? argv[4] : 0, I, 1031ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek NULL); 10327d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 10337d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strncmp(argv[1], "-test-load-source", 17) == 0) { 1034e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 17); 10357d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (I) 1036ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load_source(argc - 3, argv + 3, argv[2], I, NULL); 10377d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 10387d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strcmp(argv[1], "-test-file-scan") == 0) 10391d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek return perform_file_scan(argv[2], argv[3], 10401d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek argc >= 5 ? argv[4] : 0); 1041fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor else if (argc > 2 && strstr(argv[1], "-test-annotate-tokens=") == argv[1]) 1042fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return perform_token_annotation(argc, argv); 104316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek else if (argc > 2 && strcmp(argv[1], "-test-inclusion-stack-source") == 0) 104416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek return perform_test_load_source(argc - 2, argv + 2, "all", NULL, 104516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek PrintInclusionStack); 104616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek else if (argc > 2 && strcmp(argv[1], "-test-inclusion-stack-tu") == 0) 104716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek return perform_test_load_tu(argv[2], "all", NULL, NULL, 104816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek PrintInclusionStack); 10493bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek else if (argc > 2 && strcmp(argv[1], "-test-print-linkage-source") == 0) 10503bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek return perform_test_load_source(argc - 2, argv + 2, "all", PrintLinkage, 10513bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek NULL); 1052e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1053f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek print_usage(); 105450398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff return 1; 105550398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff} 1056