c-index-test.c revision 25d9b00ab0b128d651d993c38726a00cd9969124
12b8ee6c2994f738e5162ff46b638974870f51662Steve Naroff/* c-index-test.c */ 250398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff 350398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff#include "clang-c/Index.h" 41e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor#include <ctype.h> 50c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor#include <stdlib.h> 689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff#include <stdio.h> 7af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff#include <string.h> 8f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor#include <assert.h> 9af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff 100d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 110d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Utility functions. */ 120d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 130d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 142e06fc877a633abea3b40a64950c7316dac29ca8John Thompson#ifdef _MSC_VER 152e06fc877a633abea3b40a64950c7316dac29ca8John Thompsonchar *basename(const char* path) 162e06fc877a633abea3b40a64950c7316dac29ca8John Thompson{ 172e06fc877a633abea3b40a64950c7316dac29ca8John Thompson char* base1 = (char*)strrchr(path, '/'); 182e06fc877a633abea3b40a64950c7316dac29ca8John Thompson char* base2 = (char*)strrchr(path, '\\'); 192e06fc877a633abea3b40a64950c7316dac29ca8John Thompson if (base1 && base2) 202e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return((base1 > base2) ? base1 + 1 : base2 + 1); 212e06fc877a633abea3b40a64950c7316dac29ca8John Thompson else if (base1) 222e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return(base1 + 1); 232e06fc877a633abea3b40a64950c7316dac29ca8John Thompson else if (base2) 242e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return(base2 + 1); 252e06fc877a633abea3b40a64950c7316dac29ca8John Thompson 262e06fc877a633abea3b40a64950c7316dac29ca8John Thompson return((char*)path); 272e06fc877a633abea3b40a64950c7316dac29ca8John Thompson} 282e06fc877a633abea3b40a64950c7316dac29ca8John Thompson#else 29ff9e18cb3863defa98257500ef49d93a9d2bc216Steve Naroffextern char *basename(const char *); 302e06fc877a633abea3b40a64950c7316dac29ca8John Thompson#endif 31ff9e18cb3863defa98257500ef49d93a9d2bc216Steve Naroff 3245ba9a1b31110568d0d362c8d31c6133cf9011b7Douglas Gregor/** \brief Return the default parsing options. */ 3344c181aec37789f25f6c15543c164416f72e562aDouglas Gregorstatic unsigned getDefaultParsingOptions() { 3444c181aec37789f25f6c15543c164416f72e562aDouglas Gregor unsigned options = CXTranslationUnit_DetailedPreprocessingRecord; 3544c181aec37789f25f6c15543c164416f72e562aDouglas Gregor 3644c181aec37789f25f6c15543c164416f72e562aDouglas Gregor if (getenv("CINDEXTEST_EDITING")) 37b1c031be513705d924038f497279b9b599868ba1Douglas Gregor options |= clang_defaultEditingTranslationUnitOptions(); 3887c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor if (getenv("CINDEXTEST_COMPLETION_CACHING")) 3987c08a5d6b9e1e44ae6f554df40139d3a6f60b33Douglas Gregor options |= CXTranslationUnit_CacheCompletionResults; 4044c181aec37789f25f6c15543c164416f72e562aDouglas Gregor 4144c181aec37789f25f6c15543c164416f72e562aDouglas Gregor return options; 4244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor} 4344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor 4451b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbarstatic void PrintExtent(FILE *out, unsigned begin_line, unsigned begin_column, 4551b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar unsigned end_line, unsigned end_column) { 4651b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar fprintf(out, "[%d:%d - %d:%d]", begin_line, begin_column, 47d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar end_line, end_column); 4851b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar} 4951b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar 501c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenekstatic unsigned CreateTranslationUnit(CXIndex Idx, const char *file, 511c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXTranslationUnit *TU) { 52e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 53a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor *TU = clang_createTranslationUnit(Idx, file); 546be2a22979e6a4af575af32cccdf6a6190a8edcaDan Gohman if (!*TU) { 551c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Unable to load translation unit from '%s'!\n", file); 561c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 0; 57e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 581c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 591c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 601c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 614db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregorvoid free_remapped_files(struct CXUnsavedFile *unsaved_files, 624db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int num_unsaved_files) { 634db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int i; 644db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (i = 0; i != num_unsaved_files; ++i) { 654db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free((char *)unsaved_files[i].Filename); 664db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free((char *)unsaved_files[i].Contents); 674db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 68653a55f4bc0cb1d0c1f3ff86e9559f744eb3c18fDouglas Gregor free(unsaved_files); 694db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 704db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 714db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregorint parse_remapped_files(int argc, const char **argv, int start_arg, 724db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile **unsaved_files, 734db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int *num_unsaved_files) { 744db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int i; 754db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int arg; 764db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int prefix_len = strlen("-remap-file="); 774db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 784db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 79e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 804db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Count the number of remapped files. */ 814db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (arg = start_arg; arg < argc; ++arg) { 824db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (strncmp(argv[arg], "-remap-file=", prefix_len)) 834db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor break; 84e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 854db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor ++*num_unsaved_files; 864db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 87e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 884db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (*num_unsaved_files == 0) 894db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return 0; 90e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 914db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files 92653a55f4bc0cb1d0c1f3ff86e9559f744eb3c18fDouglas Gregor = (struct CXUnsavedFile *)malloc(sizeof(struct CXUnsavedFile) * 93653a55f4bc0cb1d0c1f3ff86e9559f744eb3c18fDouglas Gregor *num_unsaved_files); 944db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor for (arg = start_arg, i = 0; i != *num_unsaved_files; ++i, ++arg) { 954db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile *unsaved = *unsaved_files + i; 964db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor const char *arg_string = argv[arg] + prefix_len; 974db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int filename_len; 984db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor char *filename; 994db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor char *contents; 1004db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor FILE *to_file; 1014db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor const char *semi = strchr(arg_string, ';'); 1024db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (!semi) { 103e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, 1044db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor "error: -remap-file=from;to argument is missing semicolon\n"); 1054db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 1064db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 1074db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 1084db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 1094db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 110e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1114db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Open the file that we're remapping to. */ 1124db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor to_file = fopen(semi + 1, "r"); 1134db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (!to_file) { 1144db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, "error: cannot open file %s that we are remapping to\n", 1154db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor semi + 1); 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 } 121e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1224db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Determine the length of the file we're remapping to. */ 1234db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fseek(to_file, 0, SEEK_END); 1244db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Length = ftell(to_file); 1254db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fseek(to_file, 0, SEEK_SET); 126e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1274db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Read the contents of the file we're remapping to. */ 1284db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor contents = (char *)malloc(unsaved->Length + 1); 1294db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor if (fread(contents, 1, unsaved->Length, to_file) != unsaved->Length) { 1304db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fprintf(stderr, "error: unexpected %s reading 'to' file %s\n", 1314db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor (feof(to_file) ? "EOF" : "error"), semi + 1); 1324db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fclose(to_file); 1334db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(*unsaved_files, i); 1344db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *unsaved_files = 0; 1354db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor *num_unsaved_files = 0; 1364db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 1374db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 1384db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor contents[unsaved->Length] = 0; 1394db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Contents = contents; 140e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1414db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Close the file. */ 1424db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor fclose(to_file); 143e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1444db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor /* Copy the file name that we're remapping from. */ 1454db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename_len = semi - arg_string; 1464db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename = (char *)malloc(filename_len + 1); 1474db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor memcpy(filename, arg_string, filename_len); 1484db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor filename[filename_len] = 0; 1494db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor unsaved->Filename = filename; 1504db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor } 151e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1524db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return 0; 1534db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor} 1544db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1550d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1560d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Pretty-printing. */ 1570d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1580d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 159af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroffstatic void PrintCursor(CXCursor Cursor) { 160e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (clang_isInvalid(Cursor.kind)) { 161e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek CXString ks = clang_getCursorKindSpelling(Cursor.kind); 162e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("Invalid Cursor => %s", clang_getCString(ks)); 163e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_disposeString(ks); 164e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 165699a07d8a0b1579c5178b3baf4bcf9edb6b38108Steve Naroff else { 166e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek CXString string, ks; 167c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor CXCursor Referenced; 1681db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned line, column; 169e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor CXCursor SpecializationOf; 170e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor 171e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek ks = clang_getCursorKindSpelling(Cursor.kind); 172ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff string = clang_getCursorSpelling(Cursor); 173e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("%s=%s", clang_getCString(ks), 174e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getCString(string)); 175e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_disposeString(ks); 176ef0cef6cec8de5fc60e469a93436eed7212e0dc2Steve Naroff clang_disposeString(string); 177e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 178c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor Referenced = clang_getCursorReferenced(Cursor); 179c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor if (!clang_equalCursors(Referenced, clang_getNullCursor())) { 1801f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor if (clang_getCursorKind(Referenced) == CXCursor_OverloadedDeclRef) { 1811f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor unsigned I, N = clang_getNumOverloadedDecls(Referenced); 1821f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor printf("["); 1831f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor for (I = 0; I != N; ++I) { 1841f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor CXCursor Ovl = clang_getOverloadedDecl(Referenced, I); 1851f6206ed21e026ca7ddaf0bff9599c476301a695Douglas Gregor CXSourceLocation Loc; 1861f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor if (I) 1871f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor printf(", "); 1881f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor 1891f6206ed21e026ca7ddaf0bff9599c476301a695Douglas Gregor Loc = clang_getCursorLocation(Ovl); 1901f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 1911f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor printf("%d:%d", line, column); 1921f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor } 1931f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor printf("]"); 1941f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor } else { 1951f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Referenced); 1961f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 1971f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor printf(":%d:%d", line, column); 1981f60d9ea523fc321d811fe880ba9a1ec74fa8f9bDouglas Gregor } 199c5d1e9375d71e66d22456e7cc52cc7c0a5c65c3fDouglas Gregor } 200b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor 201b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor if (clang_isCursorDefinition(Cursor)) 202b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor printf(" (Definition)"); 20358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor 20458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor switch (clang_getCursorAvailability(Cursor)) { 20558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor case CXAvailability_Available: 20658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor break; 20758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor 20858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor case CXAvailability_Deprecated: 20958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor printf(" (deprecated)"); 21058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor break; 21158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor 21258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor case CXAvailability_NotAvailable: 21358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor printf(" (unavailable)"); 21458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor break; 21558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor } 21695f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek 21795f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek if (Cursor.kind == CXCursor_IBOutletCollectionAttr) { 21895f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek CXType T = 21995f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek clang_getCanonicalType(clang_getIBOutletCollectionType(Cursor)); 22095f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek CXString S = clang_getTypeKindSpelling(T.kind); 22195f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek printf(" [IBOutletCollection=%s]", clang_getCString(S)); 22295f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek clang_disposeString(S); 22395f33555a6d51b6537a9ed3968c3d1c2e4991b51Ted Kremenek } 2243064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek 2253064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek if (Cursor.kind == CXCursor_CXXBaseSpecifier) { 2263064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek enum CX_CXXAccessSpecifier access = clang_getCXXAccessSpecifier(Cursor); 2273064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek unsigned isVirtual = clang_isVirtualBase(Cursor); 2283064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek const char *accessStr = 0; 2293064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek 2303064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek switch (access) { 2313064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek case CX_CXXInvalidAccessSpecifier: 2323064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek accessStr = "invalid"; break; 2333064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek case CX_CXXPublic: 2343064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek accessStr = "public"; break; 2353064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek case CX_CXXProtected: 2363064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek accessStr = "protected"; break; 2373064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek case CX_CXXPrivate: 2383064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek accessStr = "private"; break; 2393064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek } 2403064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek 2413064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek printf(" [access=%s isVirtual=%s]", accessStr, 2423064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek isVirtual ? "true" : "false"); 2433064ef9e604d19a0cfd0d8e3ed3055bfd83f88fdTed Kremenek } 244e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor 245e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor SpecializationOf = clang_getSpecializedCursorTemplate(Cursor); 246e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor if (!clang_equalCursors(SpecializationOf, clang_getNullCursor())) { 247e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor CXSourceLocation Loc = clang_getCursorLocation(SpecializationOf); 248e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor CXString Name = clang_getCursorSpelling(SpecializationOf); 249e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 250e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor printf(" [Specialization of %s:%d:%d]", 251e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor clang_getCString(Name), line, column); 252e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor clang_disposeString(Name); 253e0329acf5c9437e2086a2fb2bf7a95ae2ac96505Douglas Gregor } 254699a07d8a0b1579c5178b3baf4bcf9edb6b38108Steve Naroff } 255af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff} 25689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff 257e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekstatic const char* GetCursorSource(CXCursor Cursor) { 2581db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Cursor); 25974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek CXString source; 2601db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile file; 26146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, &file, 0, 0, 0); 2621db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor source = clang_getFileName(file); 263e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (!clang_getCString(source)) { 26474844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(source); 26574844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek return "<invalid loc>"; 26674844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek } 26774844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek else { 268e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek const char *b = basename(clang_getCString(source)); 26974844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(source); 27074844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek return b; 27174844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek } 2729298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek} 2739298cfc7475c48fa42b168c37f628663d65ddde7Ted Kremenek 2740d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 275ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/* Callbacks. */ 276ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/******************************************************************************/ 277ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 278ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenektypedef void (*PostVisitTU)(CXTranslationUnit); 279ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek 280a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregorvoid PrintDiagnostic(CXDiagnostic Diagnostic) { 281a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor FILE *out = stderr; 2825352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor CXFile file; 283274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor CXString Msg; 2840a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor unsigned display_opts = CXDiagnostic_DisplaySourceLocation 2850a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor | CXDiagnostic_DisplayColumn | CXDiagnostic_DisplaySourceRanges; 2860a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor unsigned i, num_fixits; 287f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 2880a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (clang_getDiagnosticSeverity(Diagnostic) == CXDiagnostic_Ignored) 2895352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor return; 290e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 291274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor Msg = clang_formatDiagnostic(Diagnostic, display_opts); 292274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor fprintf(stderr, "%s\n", clang_getCString(Msg)); 293274f1906f12ebf8fcc179701deeda6d3271120c1Douglas Gregor clang_disposeString(Msg); 294f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 2955352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor clang_getInstantiationLocation(clang_getDiagnosticLocation(Diagnostic), 2960a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor &file, 0, 0, 0); 2970a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (!file) 2980a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor return; 299e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 3000a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor num_fixits = clang_getDiagnosticNumFixIts(Diagnostic); 3010a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor for (i = 0; i != num_fixits; ++i) { 302473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXSourceRange range; 303473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXString insertion_text = clang_getDiagnosticFixIt(Diagnostic, i, &range); 304473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXSourceLocation start = clang_getRangeStart(range); 305473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXSourceLocation end = clang_getRangeEnd(range); 306473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor unsigned start_line, start_column, end_line, end_column; 307473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor CXFile start_file, end_file; 308f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek clang_getInstantiationLocation(start, &start_file, &start_line, 309473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor &start_column, 0); 310473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_getInstantiationLocation(end, &end_file, &end_line, &end_column, 0); 311473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor if (clang_equalLocations(start, end)) { 312473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor /* Insertion. */ 313473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor if (start_file == file) 3140a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "FIX-IT: Insert \"%s\" at %d:%d\n", 315473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_getCString(insertion_text), start_line, start_column); 316473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor } else if (strcmp(clang_getCString(insertion_text), "") == 0) { 317473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor /* Removal. */ 3180a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (start_file == file && end_file == file) { 3190a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "FIX-IT: Remove "); 3200a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor PrintExtent(out, start_line, start_column, end_line, end_column); 3210a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "\n"); 322436f3f0400c633251e4071f81358c47bab964adfDouglas Gregor } 323473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor } else { 324473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor /* Replacement. */ 3250a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (start_file == end_file) { 3260a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor fprintf(out, "FIX-IT: Replace "); 3270a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor PrintExtent(out, start_line, start_column, end_line, end_column); 328473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor fprintf(out, " with \"%s\"\n", clang_getCString(insertion_text)); 32951c6d384551674facc19f745ecf6e289d28dc55fDouglas Gregor } 3300a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor break; 3310a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor } 332473d7019bb54f8a2f0140dca9e9644a935cc6b20Douglas Gregor clang_disposeString(insertion_text); 33351c6d384551674facc19f745ecf6e289d28dc55fDouglas Gregor } 3345352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor} 3355352ac06d8f6194825bb2a99ffa009b61bafb503Douglas Gregor 336a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregorvoid PrintDiagnostics(CXTranslationUnit TU) { 337a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor int i, n = clang_getNumDiagnostics(TU); 338a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor for (i = 0; i != n; ++i) { 339a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor CXDiagnostic Diag = clang_getDiagnostic(TU, i); 340a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostic(Diag); 341a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor clang_disposeDiagnostic(Diag); 342a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor } 343a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor} 344a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor 345ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek/******************************************************************************/ 346e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor/* Logic for testing traversal. */ 3470d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 3480d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 349fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenekstatic const char *FileCheckPrefix = "CHECK"; 350fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 351a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregorstatic void PrintCursorExtent(CXCursor C) { 352a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor CXSourceRange extent = clang_getCursorExtent(C); 3531db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile begin_file, end_file; 3541db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned begin_line, begin_column, end_line, end_column; 355e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 3561db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor clang_getInstantiationLocation(clang_getRangeStart(extent), 35746766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor &begin_file, &begin_line, &begin_column, 0); 3581db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor clang_getInstantiationLocation(clang_getRangeEnd(extent), 35946766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor &end_file, &end_line, &end_column, 0); 3601db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor if (!begin_file || !end_file) 36170ee54258035c860ebc71f7e5f803b74f3186889Ted Kremenek return; 3621db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor 36351b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar printf(" Extent="); 36451b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar PrintExtent(stdout, begin_line, begin_column, end_line, end_column); 365fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek} 366fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 367e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor/* Data used by all of the visitors. */ 368e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregortypedef struct { 369e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXTranslationUnit TU; 370e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor enum CXCursorKind *Filter; 371e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor} VisitorData; 372fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek 373625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 374e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekenum CXChildVisitResult FilteredPrintingVisitor(CXCursor Cursor, 375e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursor Parent, 376e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 377e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 378e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor if (!Data->Filter || (Cursor.kind == *(enum CXCursorKind *)Data->Filter)) { 37998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor CXSourceLocation Loc = clang_getCursorLocation(Cursor); 3801db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor unsigned line, column; 38146766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, 0, &line, &column, 0); 382fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek printf("// %s: %s:%d:%d: ", FileCheckPrefix, 3831db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor GetCursorSource(Cursor), line, column); 384af08ddc8f1c53fed8d8d0ad82aa2a0bb7d654bd1Steve Naroff PrintCursor(Cursor); 385a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor PrintCursorExtent(Cursor); 386e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("\n"); 387e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Recurse; 388625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar } 389e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 390e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 391625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar} 392c857ea4d22e1e6dd9ede1f0e84d48b157c6924fdSteve Naroff 393e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekstatic enum CXChildVisitResult FunctionScanVisitor(CXCursor Cursor, 394e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursor Parent, 395e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 396625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar const char *startBuf, *endBuf; 397625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar unsigned startLine, startColumn, endLine, endColumn, curLine, curColumn; 398625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXCursor Ref; 399e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 400625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 401b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor if (Cursor.kind != CXCursor_FunctionDecl || 402b699866820102a69d83d6ac6941985c5ef4e8c40Douglas Gregor !clang_isCursorDefinition(Cursor)) 403e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 404625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 405625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar clang_getDefinitionSpellingAndExtent(Cursor, &startBuf, &endBuf, 406625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar &startLine, &startColumn, 407625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar &endLine, &endColumn); 408625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar /* Probe the entire body, looking for both decls and refs. */ 409625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curLine = startLine; 410625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn = startColumn; 411625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 412625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar while (startBuf < endBuf) { 41398258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor CXSourceLocation Loc; 4141db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor CXFile file; 41574844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek CXString source; 416e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 417625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar if (*startBuf == '\n') { 418625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar startBuf++; 419625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curLine++; 420625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn = 1; 421625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar } else if (*startBuf != '\t') 422625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar curColumn++; 423e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 42498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor Loc = clang_getCursorLocation(Cursor); 42546766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor clang_getInstantiationLocation(Loc, &file, 0, 0, 0); 426e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 4271db19dea8d221f27be46332d668d1e2decb7f1abDouglas Gregor source = clang_getFileName(file); 428e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (clang_getCString(source)) { 429b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXSourceLocation RefLoc 430b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor = clang_getLocation(Data->TU, file, curLine, curColumn); 431b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor Ref = clang_getCursor(Data->TU, RefLoc); 43298258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor if (Ref.kind == CXCursor_NoDeclFound) { 43398258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor /* Nothing found here; that's fine. */ 43498258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor } else if (Ref.kind != CXCursor_FunctionDecl) { 43598258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor printf("// %s: %s:%d:%d: ", FileCheckPrefix, GetCursorSource(Ref), 43698258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor curLine, curColumn); 43798258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor PrintCursor(Ref); 43898258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor printf("\n"); 43998258afae66bab39b0c57a3efb6b20d4fbb5746cDouglas Gregor } 4404ade6d6eae934f796ca43c81a5aa185e456dde9bSteve Naroff } 44174844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(source); 442625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar startBuf++; 4432d4d629d8a0de5112c7ae9d05c03ddbf6dcd956aSteve Naroff } 444e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 445e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 44689922f86f4e7da383af2a62ef04ad8b93b941220Steve Naroff} 44750398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff 4487d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 4497d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/* USR testing. */ 4507d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 4517d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 452e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorenum CXChildVisitResult USRVisitor(CXCursor C, CXCursor parent, 453e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXClientData ClientData) { 454e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor VisitorData *Data = (VisitorData *)ClientData; 455e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor if (!Data->Filter || (C.kind == *(enum CXCursorKind *)Data->Filter)) { 456cf84aa46def41cccf4dd4c51cd0543b70c11e4ebTed Kremenek CXString USR = clang_getCursorUSR(C); 457e542f7723a929d42bd9e4dfa526b4ede915b91a7Ted Kremenek const char *cstr = clang_getCString(USR); 458e542f7723a929d42bd9e4dfa526b4ede915b91a7Ted Kremenek if (!cstr || cstr[0] == '\0') { 4597d40562f83552b7295411e10ee887d8d55470679Ted Kremenek clang_disposeString(USR); 460e74ef1289d5fff0a6ea573198bf354fa8cd84d51Ted Kremenek return CXChildVisit_Recurse; 4617d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 462e542f7723a929d42bd9e4dfa526b4ede915b91a7Ted Kremenek printf("// %s: %s %s", FileCheckPrefix, GetCursorSource(C), cstr); 463e542f7723a929d42bd9e4dfa526b4ede915b91a7Ted Kremenek 464a7bde20f8c6334ccc3a7ef4dd77243d0921a8497Douglas Gregor PrintCursorExtent(C); 4657d40562f83552b7295411e10ee887d8d55470679Ted Kremenek printf("\n"); 4667d40562f83552b7295411e10ee887d8d55470679Ted Kremenek clang_disposeString(USR); 467e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 468e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Recurse; 469e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 470e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 471e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return CXChildVisit_Continue; 4727d40562f83552b7295411e10ee887d8d55470679Ted Kremenek} 4737d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 4747d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 47516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/* Inclusion stack testing. */ 47616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/******************************************************************************/ 47716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 47816b55a71695a33c094383295cc7b7a2080e098daTed Kremenekvoid InclusionVisitor(CXFile includedFile, CXSourceLocation *includeStack, 47916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned includeStackLen, CXClientData data) { 480e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 48116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned i; 48274844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek CXString fname; 48374844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek 48474844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek fname = clang_getFileName(includedFile); 485e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf("file: %s\nincluded by:\n", clang_getCString(fname)); 48674844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(fname); 487e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 48816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek for (i = 0; i < includeStackLen; ++i) { 48916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek CXFile includingFile; 49016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek unsigned line, column; 49116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek clang_getInstantiationLocation(includeStack[i], &includingFile, &line, 49216b55a71695a33c094383295cc7b7a2080e098daTed Kremenek &column, 0); 49374844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek fname = clang_getFileName(includingFile); 494e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek printf(" %s:%d:%d\n", clang_getCString(fname), line, column); 49574844072411bae91d5dbb89955d200cbe1e0a1c8Ted Kremenek clang_disposeString(fname); 49616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek } 49716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek printf("\n"); 49816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek} 49916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 50016b55a71695a33c094383295cc7b7a2080e098daTed Kremenekvoid PrintInclusionStack(CXTranslationUnit TU) { 501e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getInclusions(TU, InclusionVisitor, NULL); 50216b55a71695a33c094383295cc7b7a2080e098daTed Kremenek} 50316b55a71695a33c094383295cc7b7a2080e098daTed Kremenek 50416b55a71695a33c094383295cc7b7a2080e098daTed Kremenek/******************************************************************************/ 5053bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek/* Linkage testing. */ 5063bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek/******************************************************************************/ 5073bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 5083bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenekstatic enum CXChildVisitResult PrintLinkage(CXCursor cursor, CXCursor p, 5093bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek CXClientData d) { 5103bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek const char *linkage = 0; 5113bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 5123bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek if (clang_isInvalid(clang_getCursorKind(cursor))) 5133bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek return CXChildVisit_Recurse; 5143bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 5153bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek switch (clang_getCursorLinkage(cursor)) { 5163bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek case CXLinkage_Invalid: break; 517c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_NoLinkage: linkage = "NoLinkage"; break; 518c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_Internal: linkage = "Internal"; break; 519c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_UniqueExternal: linkage = "UniqueExternal"; break; 520c2a2b3c2c2da1086d76bf804633ed5c6c48ae722Douglas Gregor case CXLinkage_External: linkage = "External"; break; 5213bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek } 5223bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 5233bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek if (linkage) { 5243bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek PrintCursor(cursor); 5253bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek printf("linkage=%s\n", linkage); 5263bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek } 5273bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 5283bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek return CXChildVisit_Recurse; 5293bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek} 5303bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek 5313bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek/******************************************************************************/ 5328e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek/* Typekind testing. */ 5338e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek/******************************************************************************/ 5348e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek 5358e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenekstatic enum CXChildVisitResult PrintTypeKind(CXCursor cursor, CXCursor p, 5368e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek CXClientData d) { 5378e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek 5388e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek if (!clang_isInvalid(clang_getCursorKind(cursor))) { 5398e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek CXType T = clang_getCursorType(cursor); 5408e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek CXString S = clang_getTypeKindSpelling(T.kind); 5418e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek PrintCursor(cursor); 5428e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek printf(" typekind=%s", clang_getCString(S)); 5438e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek clang_disposeString(S); 544e1403d2dcb06fc306e186c787a7bd0ca30e06edeBenjamin Kramer /* Print the canonical type if it is different. */ 54504c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek { 54604c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek CXType CT = clang_getCanonicalType(T); 54704c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek if (!clang_equalTypes(T, CT)) { 54804c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek CXString CS = clang_getTypeKindSpelling(CT.kind); 54904c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek printf(" [canonical=%s]", clang_getCString(CS)); 55004c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek clang_disposeString(CS); 55104c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek } 55204c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek } 553e1403d2dcb06fc306e186c787a7bd0ca30e06edeBenjamin Kramer /* Print the return type if it exists. */ 55404c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek { 5559a140845438c2fc31e7d48a6dedbc695f4c83c68Ted Kremenek CXType RT = clang_getCursorResultType(cursor); 55604c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek if (RT.kind != CXType_Invalid) { 55704c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek CXString RS = clang_getTypeKindSpelling(RT.kind); 55804c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek printf(" [result=%s]", clang_getCString(RS)); 55904c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek clang_disposeString(RS); 56004c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek } 56104c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek } 5623ce9e7d270e7df86c09c8126b4412d55be7c123bTed Kremenek /* Print if this is a non-POD type. */ 5633ce9e7d270e7df86c09c8126b4412d55be7c123bTed Kremenek printf(" [isPOD=%d]", clang_isPODType(T)); 56404c3cf35a80c09ab78e519f2e71ecccd5c5d8da0Ted Kremenek 5658e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek printf("\n"); 5668e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek } 5678e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek return CXChildVisit_Recurse; 5688e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek} 5698e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek 5708e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek 5718e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek/******************************************************************************/ 5727d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/* Loading ASTs/source. */ 5737d40562f83552b7295411e10ee887d8d55470679Ted Kremenek/******************************************************************************/ 5747d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 575625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbarstatic int perform_test_load(CXIndex Idx, CXTranslationUnit TU, 57698271567b6f10be85bcc75d0bbb67416c8cfb83aTed Kremenek const char *filter, const char *prefix, 577ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek CXCursorVisitor Visitor, 578ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 579e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 580fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek if (prefix) 581e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek FileCheckPrefix = prefix; 582e3ee02a324559829142f387f75daf3da118442beTed Kremenek 583e3ee02a324559829142f387f75daf3da118442beTed Kremenek if (Visitor) { 584e3ee02a324559829142f387f75daf3da118442beTed Kremenek enum CXCursorKind K = CXCursor_NotImplemented; 585e3ee02a324559829142f387f75daf3da118442beTed Kremenek enum CXCursorKind *ck = &K; 586e3ee02a324559829142f387f75daf3da118442beTed Kremenek VisitorData Data; 587e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 588e3ee02a324559829142f387f75daf3da118442beTed Kremenek /* Perform some simple filtering. */ 589e3ee02a324559829142f387f75daf3da118442beTed Kremenek if (!strcmp(filter, "all") || !strcmp(filter, "local")) ck = NULL; 590b1ffee6e574d57b79ce2defd1b57052649221c03Daniel Dunbar else if (!strcmp(filter, "none")) K = (enum CXCursorKind) ~0; 591e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "category")) K = CXCursor_ObjCCategoryDecl; 592e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "interface")) K = CXCursor_ObjCInterfaceDecl; 593e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "protocol")) K = CXCursor_ObjCProtocolDecl; 594e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "function")) K = CXCursor_FunctionDecl; 595e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "typedef")) K = CXCursor_TypedefDecl; 596e3ee02a324559829142f387f75daf3da118442beTed Kremenek else if (!strcmp(filter, "scan-function")) Visitor = FunctionScanVisitor; 597e3ee02a324559829142f387f75daf3da118442beTed Kremenek else { 598e3ee02a324559829142f387f75daf3da118442beTed Kremenek fprintf(stderr, "Unknown filter for -test-load-tu: %s\n", filter); 599e3ee02a324559829142f387f75daf3da118442beTed Kremenek return 1; 600e3ee02a324559829142f387f75daf3da118442beTed Kremenek } 601e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 602e3ee02a324559829142f387f75daf3da118442beTed Kremenek Data.TU = TU; 603e3ee02a324559829142f387f75daf3da118442beTed Kremenek Data.Filter = ck; 604e3ee02a324559829142f387f75daf3da118442beTed Kremenek clang_visitChildren(clang_getTranslationUnitCursor(TU), Visitor, &Data); 605e3ee02a324559829142f387f75daf3da118442beTed Kremenek } 606e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 607ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek if (PV) 608ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PV(TU); 609e3ee02a324559829142f387f75daf3da118442beTed Kremenek 610a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostics(TU); 6110d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek clang_disposeTranslationUnit(TU); 6120d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek return 0; 6130d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek} 6140d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 615fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenekint perform_test_load_tu(const char *file, const char *filter, 616ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek const char *prefix, CXCursorVisitor Visitor, 617ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 618625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXIndex Idx; 619625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar CXTranslationUnit TU; 620020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek int result; 621e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek Idx = clang_createIndex(/* excludeDeclsFromPCH */ 6220a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor !strcmp(filter, "local") ? 1 : 0, 6230a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor /* displayDiagnosics=*/1); 624e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 625020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek if (!CreateTranslationUnit(Idx, file, &TU)) { 626020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 627625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar return 1; 628020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek } 629625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 630020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek result = perform_test_load(Idx, TU, filter, prefix, Visitor, PV); 631020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 632020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek return result; 633625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar} 634625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar 635ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenekint perform_test_load_source(int argc, const char **argv, 636ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek const char *filter, CXCursorVisitor Visitor, 637ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek PostVisitTU PV) { 6388506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar const char *UseExternalASTs = 6398506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar getenv("CINDEXTEST_USE_EXTERNAL_AST_GENERATION"); 640ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar CXIndex Idx; 641ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar CXTranslationUnit TU; 6424db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 6434db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int num_unsaved_files = 0; 6444db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor int result; 645abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 646ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar Idx = clang_createIndex(/* excludeDeclsFromPCH */ 6470a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor !strcmp(filter, "local") ? 1 : 0, 6480a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor /* displayDiagnosics=*/1); 649ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 6508506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar if (UseExternalASTs && strlen(UseExternalASTs)) 6518506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar clang_setUseExternalASTGeneration(Idx, 1); 6528506dde586459887b7e14e23a30af8ac5be5adb6Daniel Dunbar 653020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek if (parse_remapped_files(argc, argv, 0, &unsaved_files, &num_unsaved_files)) { 654020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 6554db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return -1; 656020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek } 6574db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 658e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek TU = clang_createTranslationUnitFromSourceFile(Idx, 0, 659e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek argc - num_unsaved_files, 6604db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argv + num_unsaved_files, 6614db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor num_unsaved_files, 662a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsaved_files); 663ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar if (!TU) { 664ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar fprintf(stderr, "Unable to load translation unit!\n"); 665abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 666020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 667ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar return 1; 668ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar } 669ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 670ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek result = perform_test_load(Idx, TU, filter, NULL, Visitor, PV); 6714db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 672020a095de30ede5b1e3fc65dae501f225a1c89d3Ted Kremenek clang_disposeIndex(Idx); 6734db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor return result; 674ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar} 675ada487d498d82277bbc60312fc6f6479f0afda76Daniel Dunbar 676abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregorint perform_test_reparse_source(int argc, const char **argv, int trials, 677abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor const char *filter, CXCursorVisitor Visitor, 678abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor PostVisitTU PV) { 679abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor const char *UseExternalASTs = 680abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor getenv("CINDEXTEST_USE_EXTERNAL_AST_GENERATION"); 681abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor CXIndex Idx; 682abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor CXTranslationUnit TU; 683abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 684abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor int num_unsaved_files = 0; 685abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor int result; 686abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor int trial; 687abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 688abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor Idx = clang_createIndex(/* excludeDeclsFromPCH */ 689abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor !strcmp(filter, "local") ? 1 : 0, 690abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor /* displayDiagnosics=*/1); 691abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 692abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor if (UseExternalASTs && strlen(UseExternalASTs)) 693abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor clang_setUseExternalASTGeneration(Idx, 1); 694abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 695abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor if (parse_remapped_files(argc, argv, 0, &unsaved_files, &num_unsaved_files)) { 696abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor clang_disposeIndex(Idx); 697abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor return -1; 698abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor } 699abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 700c8a618002ca57495bf70e055bcafe918859cf4caDaniel Dunbar /* Load the initial translation unit -- we do this without honoring remapped 701c8a618002ca57495bf70e055bcafe918859cf4caDaniel Dunbar * files, so that we have a way to test results after changing the source. */ 70244c181aec37789f25f6c15543c164416f72e562aDouglas Gregor TU = clang_parseTranslationUnit(Idx, 0, 70344c181aec37789f25f6c15543c164416f72e562aDouglas Gregor argv + num_unsaved_files, 70444c181aec37789f25f6c15543c164416f72e562aDouglas Gregor argc - num_unsaved_files, 705c8a618002ca57495bf70e055bcafe918859cf4caDaniel Dunbar 0, 0, getDefaultParsingOptions()); 706abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor if (!TU) { 707abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor fprintf(stderr, "Unable to load translation unit!\n"); 708abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 709abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor clang_disposeIndex(Idx); 710abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor return 1; 711abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor } 712abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 713abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor for (trial = 0; trial < trials; ++trial) { 714e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor if (clang_reparseTranslationUnit(TU, num_unsaved_files, unsaved_files, 715e1e13bf568a7e37c95eda6fcfa626659a06e67b1Douglas Gregor clang_defaultReparseOptions(TU))) { 716c8a618002ca57495bf70e055bcafe918859cf4caDaniel Dunbar fprintf(stderr, "Unable to reparse translation unit!\n"); 717abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor clang_disposeTranslationUnit(TU); 718abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 719abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor clang_disposeIndex(Idx); 720abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor return -1; 721abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor } 722abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor } 723abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 724abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor result = perform_test_load(Idx, TU, filter, NULL, Visitor, PV); 725abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 726abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor clang_disposeIndex(Idx); 727abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor return result; 728abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor} 729abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor 7300d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 7311c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/* Logic for testing clang_getCursor(). */ 7321c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/******************************************************************************/ 7331c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 7341c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenekstatic void print_cursor_file_scan(CXCursor cursor, 7351c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek unsigned start_line, unsigned start_col, 7361d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek unsigned end_line, unsigned end_col, 7371d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek const char *prefix) { 7389096a20fc0fd799911b738ce7a9134dbacaf64b7Ted Kremenek printf("// %s: ", FileCheckPrefix); 7391d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek if (prefix) 7401d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek printf("-%s", prefix); 74151b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar PrintExtent(stdout, start_line, start_col, end_line, end_col); 74251b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar printf(" "); 7431c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek PrintCursor(cursor); 7441c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek printf("\n"); 7451c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 7461c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 7471d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenekstatic int perform_file_scan(const char *ast_file, const char *source_file, 7481d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek const char *prefix) { 7491c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXIndex Idx; 7501c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek CXTranslationUnit TU; 7511c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek FILE *fp; 7522389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar CXCursor prevCursor = clang_getNullCursor(); 753b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXFile file; 7542389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar unsigned line = 1, col = 1; 7558f0bf81c0f3ab31881cc7db7914e6978bb2a19afDaniel Dunbar unsigned start_line = 1, start_col = 1; 756e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7570a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor if (!(Idx = clang_createIndex(/* excludeDeclsFromPCH */ 1, 7580a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor /* displayDiagnosics=*/1))) { 7591c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Could not create Index\n"); 7601c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 7611c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 762e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7631c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if (!CreateTranslationUnit(Idx, ast_file, &TU)) 7641c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 765e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7661c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek if ((fp = fopen(source_file, "r")) == NULL) { 7671c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fprintf(stderr, "Could not open '%s'\n", source_file); 7681c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 1; 7691c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 770e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 771b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor file = clang_getFile(TU, source_file); 7722389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar for (;;) { 7732389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar CXCursor cursor; 7742389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar int c = fgetc(fp); 7752389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar 7762389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar if (c == '\n') { 7772389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar ++line; 7782389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar col = 1; 7792389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar } else 7802389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar ++col; 7812389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar 7822389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar /* Check the cursor at this position, and dump the previous one if we have 7832389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar * found something new. 7842389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar */ 7852389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar cursor = clang_getCursor(TU, clang_getLocation(TU, file, line, col)); 7862389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar if ((c == EOF || !clang_equalCursors(cursor, prevCursor)) && 7872389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar prevCursor.kind != CXCursor_InvalidFile) { 7882389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar print_cursor_file_scan(prevCursor, start_line, start_col, 789d52864bd33c66aacc84133460d8c9c0dfcdd5c18Daniel Dunbar line, col, prefix); 7902389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar start_line = line; 7912389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar start_col = col; 792a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer } 7932389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar if (c == EOF) 7942389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar break; 795a9933b98399f656653a0876fc39e5b9093efb732Benjamin Kramer 7962389eff26431b4f77d3383157adb2c8ccc15ff69Daniel Dunbar prevCursor = cursor; 7971c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek } 798e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 7991c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek fclose(fp); 8001c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek return 0; 8011c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek} 8021c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek 8031c6da1735ddc5978a2c24394c5112b4868b347f0Ted Kremenek/******************************************************************************/ 8040d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Logic for testing clang_codeComplete(). */ 8050d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 8060d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek 8070c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor/* Parse file:line:column from the input string. Returns 0 on success, non-zero 8080c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor on failure. If successful, the pointer *filename will contain newly-allocated 8090c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor memory (that will be owned by the caller) to store the file name. */ 810e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenekint parse_file_line_column(const char *input, char **filename, unsigned *line, 811fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned *column, unsigned *second_line, 812fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned *second_column) { 81388d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor /* Find the second colon. */ 814fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *last_colon = strrchr(input, ':'); 815fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned values[4], i; 816fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned num_values = (second_line && second_column)? 4 : 2; 817fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 8180c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor char *endptr = 0; 819fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!last_colon || last_colon == input) { 820fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (num_values == 4) 821fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse filename:line:column:line:column in " 822fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor "'%s'\n", input); 823fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor else 824fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "could not parse filename:line:column in '%s'\n", input); 8250c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return 1; 8260c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 8270c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 828fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor for (i = 0; i != num_values; ++i) { 829fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *prev_colon; 83088d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor 831fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor /* Parse the next line or column. */ 832fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor values[num_values - i - 1] = strtol(last_colon + 1, &endptr, 10); 833fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (*endptr != 0 && *endptr != ':') { 834e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "could not parse %s in '%s'\n", 835fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (i % 2 ? "column" : "line"), input); 836fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return 1; 837fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 838e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 839fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (i + 1 == num_values) 840fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor break; 84188d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor 842fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor /* Find the previous colon. */ 843fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor prev_colon = last_colon - 1; 844fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor while (prev_colon != input && *prev_colon != ':') 845fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor --prev_colon; 846fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (prev_colon == input) { 847e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "could not parse %s in '%s'\n", 848fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (i % 2 == 0? "column" : "line"), input); 849e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek return 1; 850fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 851fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 852fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor last_colon = prev_colon; 8530c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 854fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 855fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *line = values[0]; 856fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *column = values[1]; 857e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 858fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (second_line && second_column) { 859fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *second_line = values[2]; 860fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *second_column = values[3]; 861fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 862fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 86388d23952eadb0737e5bd839654d69a0a578e19bcDouglas Gregor /* Copy the file name. */ 864fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor *filename = (char*)malloc(last_colon - input + 1); 865fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor memcpy(*filename, input, last_colon - input); 866fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor (*filename)[last_colon - input] = 0; 8670c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return 0; 8680c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 8690c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 8700c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorconst char * 8710c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregorclang_getCompletionChunkKindSpelling(enum CXCompletionChunkKind Kind) { 8720c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor switch (Kind) { 8730c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Optional: return "Optional"; 8740c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_TypedText: return "TypedText"; 8750c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Text: return "Text"; 8760c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Placeholder: return "Placeholder"; 8770c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Informative: return "Informative"; 8780c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_CurrentParameter: return "CurrentParameter"; 8790c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftParen: return "LeftParen"; 8800c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightParen: return "RightParen"; 8810c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftBracket: return "LeftBracket"; 8820c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightBracket: return "RightBracket"; 8830c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftBrace: return "LeftBrace"; 8840c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightBrace: return "RightBrace"; 8850c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_LeftAngle: return "LeftAngle"; 8860c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_RightAngle: return "RightAngle"; 8870c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor case CXCompletionChunk_Comma: return "Comma"; 888ff5ce6eefc7c253ef6edf4d4bfc996fdd82d09aaDouglas Gregor case CXCompletionChunk_ResultType: return "ResultType"; 88901dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_Colon: return "Colon"; 89001dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_SemiColon: return "SemiColon"; 89101dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_Equal: return "Equal"; 89201dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_HorizontalSpace: return "HorizontalSpace"; 89301dfea02d1da297e8b53db8eea3d3cc652acda8dDouglas Gregor case CXCompletionChunk_VerticalSpace: return "VerticalSpace"; 8940c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 895e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 8960c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor return "Unknown"; 8970c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 8980c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 8993ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregorvoid print_completion_string(CXCompletionString completion_string, FILE *file) { 900f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar int I, N; 901e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 9023ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor N = clang_getNumCompletionChunks(completion_string); 9030c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor for (I = 0; I != N; ++I) { 9042ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek CXString text; 9052ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek const char *cstr; 9060c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor enum CXCompletionChunkKind Kind 9073ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor = clang_getCompletionChunkKind(completion_string, I); 908e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 9093ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor if (Kind == CXCompletionChunk_Optional) { 9103ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "{Optional "); 9113ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor print_completion_string( 912e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getCompletionChunkCompletionString(completion_string, I), 9133ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor file); 9143ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor fprintf(file, "}"); 9153ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor continue; 9163ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor } 917e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 918d5a2089663d81faee0de031deabc418fa53ecf3bDouglas Gregor text = clang_getCompletionChunkText(completion_string, I); 9192ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek cstr = clang_getCString(text); 920e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(file, "{%s %s}", 9210c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor clang_getCompletionChunkKindSpelling(Kind), 9222ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek cstr ? cstr : ""); 9232ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek clang_disposeString(text); 9240c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor } 9252ef6f8f5a35a60870594c5b04e0aa2bf22c6886fTed Kremenek 9263ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor} 9273ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor 9283ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregorvoid print_completion_result(CXCompletionResult *completion_result, 9293ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor CXClientData client_data) { 9303ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor FILE *file = (FILE *)client_data; 931e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek CXString ks = clang_getCursorKindSpelling(completion_result->CursorKind); 932e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 933e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(file, "%s:", clang_getCString(ks)); 934e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_disposeString(ks); 935e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 9363ac738567fdf4df636b9e76c5a2d3dc3d6f80235Douglas Gregor print_completion_string(completion_result->CompletionString, file); 93758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor fprintf(file, " (%u)", 93812e131385e892e3723483a1081a89bcad29c8a84Douglas Gregor clang_getCompletionPriority(completion_result->CompletionString)); 93958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor switch (clang_getCompletionAvailability(completion_result->CompletionString)){ 94058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor case CXAvailability_Available: 94158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor break; 94258ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor 94358ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor case CXAvailability_Deprecated: 94458ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor fprintf(file, " (deprecated)"); 94558ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor break; 94658ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor 94758ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor case CXAvailability_NotAvailable: 94858ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor fprintf(file, " (unavailable)"); 94958ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor break; 95058ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor } 95158ddb60f409125eda5436c4a1f070f7fa4744295Douglas Gregor fprintf(file, "\n"); 9520c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 9530c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 9541e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregorint my_stricmp(const char *s1, const char *s2) { 9551e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor while (*s1 && *s2) { 9561e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor int c1 = tolower(*s1), c2 = tolower(*s2); 9571e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor if (c1 < c2) 9581e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor return -1; 9591e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor else if (c1 > c2) 9601e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor return 1; 9611e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor 9621e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor ++s1; 9631e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor ++s2; 9641e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor } 9651e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor 9661e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor if (*s1) 9671e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor return 1; 9681e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor else if (*s2) 9691e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor return -1; 9701e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor return 0; 9711e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor} 9721e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor 9731982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregorint perform_code_completion(int argc, const char **argv, int timing_only) { 9740c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor const char *input = argv[1]; 9750c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor char *filename = 0; 9760c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor unsigned line; 9770c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor unsigned column; 978f8297f1512d29c88f20bf2901f04cc04182a3eeeDaniel Dunbar CXIndex CIdx; 979f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek int errorCode; 980735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 981735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor int num_unsaved_files = 0; 982ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor CXCodeCompleteResults *results = 0; 98325d9b00ab0b128d651d993c38726a00cd9969124Dawn Perchik CXTranslationUnit TU = 0; 984df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor 9851982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor if (timing_only) 9861982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor input += strlen("-code-completion-timing="); 9871982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor else 9881982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor input += strlen("-code-completion-at="); 9891982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor 990e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if ((errorCode = parse_file_line_column(input, &filename, &line, &column, 991fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 0, 0))) 992f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return errorCode; 9930c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 994735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor if (parse_remapped_files(argc, argv, 2, &unsaved_files, &num_unsaved_files)) 995735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor return -1; 996735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor 9970a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor CIdx = clang_createIndex(0, 1); 9981abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor if (getenv("CINDEXTEST_EDITING")) { 999d91de2bc0ee43c2da7be2fb879e8919ea74f8afaDaniel Dunbar unsigned I, Repeats = 5; 1000df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor TU = clang_parseTranslationUnit(CIdx, 0, 1001df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor argv + num_unsaved_files + 2, 1002df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor argc - num_unsaved_files - 2, 100343f331d0fdacb343749dd1d6f84617c2b60c3d5bDaniel Dunbar 0, 0, getDefaultParsingOptions()); 10042de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar if (!TU) { 10052de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar fprintf(stderr, "Unable to load translation unit!\n"); 10062de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar return 1; 10072de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar } 1008df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor for (I = 0; I != Repeats; ++I) { 1009df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor results = clang_codeCompleteAt(TU, filename, line, column, 1010df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor unsaved_files, num_unsaved_files, 1011df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor clang_defaultCodeCompleteOptions()); 10122de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar if (!results) { 10132de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar fprintf(stderr, "Unable to perform code completion!\n"); 10142de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar return 1; 10152de41c9c99e0e9ae6488d04c08423a5c1190109eDaniel Dunbar } 1016df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor if (I != Repeats-1) 1017df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor clang_disposeCodeCompleteResults(results); 1018df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor } 10191abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor } else 10201abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor results = clang_codeComplete(CIdx, 10211abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor argv[argc - 1], argc - num_unsaved_files - 3, 10221abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor argv + num_unsaved_files + 2, 10231abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor num_unsaved_files, unsaved_files, 10241abc6bca3d7fb0e7b1e40fbcad6cfb5e10594548Douglas Gregor filename, line, column); 1025936ea3b590117d2cd73b1b92621d06c4a7edbe60Douglas Gregor 1026ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor if (results) { 1027ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor unsigned i, n = results->NumResults; 10281e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor if (!timing_only) { 10291e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor /* Sort the code-completion results based on the typed text. */ 10301e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor clang_sortCodeCompletionResults(results->Results, results->NumResults); 10311e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor 10321982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor for (i = 0; i != n; ++i) 10331982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor print_completion_result(results->Results + i, stdout); 10341e5e6684b0f27701e6f7c65f8c6a32a10cbcc3edDouglas Gregor } 1035a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor n = clang_codeCompleteGetNumDiagnostics(results); 1036a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor for (i = 0; i != n; ++i) { 1037a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor CXDiagnostic diag = clang_codeCompleteGetDiagnostic(results, i); 1038a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostic(diag); 1039a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor clang_disposeDiagnostic(diag); 1040a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor } 1041ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor clang_disposeCodeCompleteResults(results); 1042ec6762c709726bf2ee5f76c21df81e71a56e6f81Douglas Gregor } 1043df95a13ec73d2cdaea79555cb412d767f4963120Douglas Gregor clang_disposeTranslationUnit(TU); 10440c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor clang_disposeIndex(CIdx); 10450c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor free(filename); 1046e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1047735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 1048735df88a38e80c1ca70daa889aa516b8b9f54b50Douglas Gregor 1049f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek return 0; 10500c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor} 10510c8296dfb495f41d6f0de6fe1d03014ffd063674Douglas Gregor 1052f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregortypedef struct { 1053f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor char *filename; 1054f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned line; 1055f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned column; 1056f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor} CursorSourceLocation; 1057f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 1058f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregorint inspect_cursor_at(int argc, const char **argv) { 1059f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXIndex CIdx; 1060f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor int errorCode; 1061f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor struct CXUnsavedFile *unsaved_files = 0; 1062f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor int num_unsaved_files = 0; 1063f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXTranslationUnit TU; 1064f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CXCursor Cursor; 1065f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor CursorSourceLocation *Locations = 0; 1066f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor unsigned NumLocations = 0, Loc; 10674db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor 1068e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek /* Count the number of locations. */ 1069f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor while (strstr(argv[NumLocations+1], "-cursor-at=") == argv[NumLocations+1]) 1070f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor ++NumLocations; 1071e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1072f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor /* Parse the locations. */ 1073f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor assert(NumLocations > 0 && "Unable to count locations?"); 1074f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor Locations = (CursorSourceLocation *)malloc( 1075f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor NumLocations * sizeof(CursorSourceLocation)); 1076f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor for (Loc = 0; Loc < NumLocations; ++Loc) { 1077f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor const char *input = argv[Loc + 1] + strlen("-cursor-at="); 1078e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if ((errorCode = parse_file_line_column(input, &Locations[Loc].filename, 1079e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek &Locations[Loc].line, 1080fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor &Locations[Loc].column, 0, 0))) 1081f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return errorCode; 1082f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 1083e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1084e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek if (parse_remapped_files(argc, argv, NumLocations + 1, &unsaved_files, 1085f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor &num_unsaved_files)) 1086f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return -1; 1087e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 10880a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor CIdx = clang_createIndex(0, 1); 1089f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor TU = clang_createTranslationUnitFromSourceFile(CIdx, argv[argc - 1], 1090f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor argc - num_unsaved_files - 2 - NumLocations, 10914db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor argv + num_unsaved_files + 1 + NumLocations, 10924db64a461cb3442934afe43c83ed3f17f7c11c1dDouglas Gregor num_unsaved_files, 1093a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsaved_files); 1094f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (!TU) { 1095f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor fprintf(stderr, "unable to parse input\n"); 1096f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return -1; 1097f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 1098e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1099f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor for (Loc = 0; Loc < NumLocations; ++Loc) { 1100b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor CXFile file = clang_getFile(TU, Locations[Loc].filename); 1101b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor if (!file) 1102b979034b100be14de2223f2b8f6cc7a3275cbe4fDouglas Gregor continue; 1103e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1104e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek Cursor = clang_getCursor(TU, 1105e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getLocation(TU, file, Locations[Loc].line, 1106e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek Locations[Loc].column)); 1107f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor PrintCursor(Cursor); 1108f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor printf("\n"); 1109f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free(Locations[Loc].filename); 1110f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor } 1111e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek 1112a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostics(TU); 1113f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor clang_disposeTranslationUnit(TU); 1114f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor clang_disposeIndex(CIdx); 1115f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free(Locations); 1116f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 1117f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return 0; 1118f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor} 1119f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor 1120fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregorint perform_token_annotation(int argc, const char **argv) { 1121fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *input = argv[1]; 1122fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor char *filename = 0; 1123fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned line, second_line; 1124fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned column, second_column; 1125fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXIndex CIdx; 1126fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXTranslationUnit TU = 0; 1127fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor int errorCode; 1128fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 1129fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor int num_unsaved_files = 0; 1130fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXToken *tokens; 1131fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned num_tokens; 1132fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceRange range; 1133fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceLocation startLoc, endLoc; 1134fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXFile file = 0; 1135fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXCursor *cursors = 0; 1136fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned i; 1137fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1138fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor input += strlen("-test-annotate-tokens="); 1139fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if ((errorCode = parse_file_line_column(input, &filename, &line, &column, 1140fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor &second_line, &second_column))) 1141fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return errorCode; 1142fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1143fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (parse_remapped_files(argc, argv, 2, &unsaved_files, &num_unsaved_files)) 1144fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return -1; 1145fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 11460a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor CIdx = clang_createIndex(0, 1); 1147fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor TU = clang_createTranslationUnitFromSourceFile(CIdx, argv[argc - 1], 1148fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor argc - num_unsaved_files - 3, 1149fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor argv + num_unsaved_files + 2, 1150fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor num_unsaved_files, 1151a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor unsaved_files); 1152fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!TU) { 1153fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "unable to parse input\n"); 1154fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeIndex(CIdx); 1155fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(filename); 1156fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 1157fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return -1; 1158e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek } 1159fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = 0; 1160fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1161fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor file = clang_getFile(TU, filename); 1162fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (!file) { 1163fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor fprintf(stderr, "file %s is not in this translation unit\n", filename); 1164fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 1165fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor goto teardown; 1166fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 1167fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1168fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor startLoc = clang_getLocation(TU, file, line, column); 1169fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (clang_equalLocations(clang_getNullLocation(), startLoc)) { 1170e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "invalid source location %s:%d:%d\n", filename, line, 1171fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor column); 1172fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 1173e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek goto teardown; 1174fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 1175fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1176fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor endLoc = clang_getLocation(TU, file, second_line, second_column); 1177fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor if (clang_equalLocations(clang_getNullLocation(), endLoc)) { 1178e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek fprintf(stderr, "invalid source location %s:%d:%d\n", filename, 1179fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor second_line, second_column); 1180fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor errorCode = -1; 1181e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek goto teardown; 1182fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 1183fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1184fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor range = clang_getRange(startLoc, endLoc); 1185fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_tokenize(TU, range, &tokens, &num_tokens); 1186fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor cursors = (CXCursor *)malloc(num_tokens * sizeof(CXCursor)); 1187fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_annotateTokens(TU, tokens, num_tokens, cursors); 1188fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor for (i = 0; i != num_tokens; ++i) { 1189fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor const char *kind = "<unknown>"; 1190fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXString spelling = clang_getTokenSpelling(TU, tokens[i]); 1191fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor CXSourceRange extent = clang_getTokenExtent(TU, tokens[i]); 1192fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor unsigned start_line, start_column, end_line, end_column; 1193fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1194fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor switch (clang_getTokenKind(tokens[i])) { 1195fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Punctuation: kind = "Punctuation"; break; 1196fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Keyword: kind = "Keyword"; break; 1197fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Identifier: kind = "Identifier"; break; 1198fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Literal: kind = "Literal"; break; 1199fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor case CXToken_Comment: kind = "Comment"; break; 1200fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 1201e68fff6fc083c6270d835216a3de0b82c6ef0310Ted Kremenek clang_getInstantiationLocation(clang_getRangeStart(extent), 120246766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor 0, &start_line, &start_column, 0); 1203fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_getInstantiationLocation(clang_getRangeEnd(extent), 120446766dc31c09d89024de5aba9e22112a56eadbdfDouglas Gregor 0, &end_line, &end_column, 0); 120551b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar printf("%s: \"%s\" ", kind, clang_getCString(spelling)); 120651b058cb1e726c49fe0fae29404a4ca4308a6a12Daniel Dunbar PrintExtent(stdout, start_line, start_column, end_line, end_column); 12070045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor if (!clang_isInvalid(cursors[i].kind)) { 12080045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf(" "); 12090045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor PrintCursor(cursors[i]); 12100045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor } 12110045e9fe1f7dfc37f1ea7bdb9b70bcdb6700f0c0Douglas Gregor printf("\n"); 1212fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor } 1213fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(cursors); 1214fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 1215fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor teardown: 1216a88084b78fd4ca5d3d858c14b02414f8cc399f02Douglas Gregor PrintDiagnostics(TU); 1217fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeTranslationUnit(TU); 1218fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor clang_disposeIndex(CIdx); 1219fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free(filename); 1220fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 1221fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return errorCode; 1222fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor} 1223fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor 12240d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 1225f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek/* USR printing. */ 1226f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek/******************************************************************************/ 1227f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1228f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekstatic int insufficient_usr(const char *kind, const char *usage) { 1229f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek fprintf(stderr, "USR for '%s' requires: %s\n", kind, usage); 1230f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 1; 1231f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1232f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1233f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekstatic unsigned isUSR(const char *s) { 1234f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return s[0] == 'c' && s[1] == ':'; 1235f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1236f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1237f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekstatic int not_usr(const char *s, const char *arg) { 1238f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek fprintf(stderr, "'%s' argument ('%s') is not a USR\n", s, arg); 1239f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 1; 1240f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1241f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1242f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekstatic void print_usr(CXString usr) { 1243f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek const char *s = clang_getCString(usr); 1244f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek printf("%s\n", s); 1245f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek clang_disposeString(usr); 1246f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1247f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1248f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekstatic void display_usrs() { 1249f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek fprintf(stderr, "-print-usrs options:\n" 1250f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " ObjCCategory <class name> <category name>\n" 1251f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " ObjCClass <class name>\n" 1252f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " ObjCIvar <ivar name> <class USR>\n" 1253f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " ObjCMethod <selector> [0=class method|1=instance method] " 1254f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek "<class USR>\n" 1255f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " ObjCProperty <property name> <class USR>\n" 1256f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " ObjCProtocol <protocol name>\n"); 1257f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1258f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1259f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekint print_usrs(const char **I, const char **E) { 1260f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek while (I != E) { 1261f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek const char *kind = *I; 1262f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek unsigned len = strlen(kind); 1263f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek switch (len) { 1264f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek case 8: 1265f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (memcmp(kind, "ObjCIvar", 8) == 0) { 1266f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I + 2 >= E) 1267f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return insufficient_usr(kind, "<ivar name> <class USR>"); 1268f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (!isUSR(I[2])) 1269f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return not_usr("<class USR>", I[2]); 1270f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else { 1271f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek CXString x; 1272f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek x.Spelling = I[2]; 1273f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek x.MustFreeString = 0; 1274f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek print_usr(clang_constructUSR_ObjCIvar(I[1], x)); 1275f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1276f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1277f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek I += 3; 1278f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1279f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1280f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek break; 1281f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek case 9: 1282f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (memcmp(kind, "ObjCClass", 9) == 0) { 1283f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I + 1 >= E) 1284f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return insufficient_usr(kind, "<class name>"); 1285f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek print_usr(clang_constructUSR_ObjCClass(I[1])); 1286f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek I += 2; 1287f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1288f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1289f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek break; 1290f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek case 10: 1291f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (memcmp(kind, "ObjCMethod", 10) == 0) { 1292f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I + 3 >= E) 1293f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return insufficient_usr(kind, "<method selector> " 1294f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek "[0=class method|1=instance method] <class USR>"); 1295f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (!isUSR(I[3])) 1296f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return not_usr("<class USR>", I[3]); 1297f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else { 1298f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek CXString x; 1299f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek x.Spelling = I[3]; 1300f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek x.MustFreeString = 0; 1301f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek print_usr(clang_constructUSR_ObjCMethod(I[1], atoi(I[2]), x)); 1302f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1303f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek I += 4; 1304f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1305f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1306f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek break; 1307f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek case 12: 1308f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (memcmp(kind, "ObjCCategory", 12) == 0) { 1309f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I + 2 >= E) 1310f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return insufficient_usr(kind, "<class name> <category name>"); 1311f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek print_usr(clang_constructUSR_ObjCCategory(I[1], I[2])); 1312f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek I += 3; 1313f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1314f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1315f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (memcmp(kind, "ObjCProtocol", 12) == 0) { 1316f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I + 1 >= E) 1317f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return insufficient_usr(kind, "<protocol name>"); 1318f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek print_usr(clang_constructUSR_ObjCProtocol(I[1])); 1319f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek I += 2; 1320f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1321f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1322f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (memcmp(kind, "ObjCProperty", 12) == 0) { 1323f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I + 2 >= E) 1324f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return insufficient_usr(kind, "<property name> <class USR>"); 1325f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (!isUSR(I[2])) 1326f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return not_usr("<class USR>", I[2]); 1327f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else { 1328f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek CXString x; 1329f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek x.Spelling = I[2]; 1330f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek x.MustFreeString = 0; 1331f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek print_usr(clang_constructUSR_ObjCProperty(I[1], x)); 1332f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1333f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek I += 3; 1334f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1335f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1336f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek break; 1337f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek default: 1338f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek break; 1339f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1340f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek break; 1341f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1342f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1343f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (I != E) { 1344f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek fprintf(stderr, "Invalid USR kind: %s\n", *I); 1345f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek display_usrs(); 1346f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 1; 1347f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1348f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 0; 1349f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1350f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1351f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenekint print_usrs_file(const char *file_name) { 1352f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek char line[2048]; 1353f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek const char *args[128]; 1354f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek unsigned numChars = 0; 1355f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1356f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek FILE *fp = fopen(file_name, "r"); 1357f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (!fp) { 1358f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek fprintf(stderr, "error: cannot open '%s'\n", file_name); 1359f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 1; 1360f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1361f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1362f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek /* This code is not really all that safe, but it works fine for testing. */ 1363f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek while (!feof(fp)) { 1364f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek char c = fgetc(fp); 1365f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (c == '\n') { 1366f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek unsigned i = 0; 1367f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek const char *s = 0; 1368f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1369f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (numChars == 0) 1370f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1371f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1372f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek line[numChars] = '\0'; 1373f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek numChars = 0; 1374f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1375f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (line[0] == '/' && line[1] == '/') 1376f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek continue; 1377f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1378f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek s = strtok(line, " "); 1379f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek while (s) { 1380f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek args[i] = s; 1381f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek ++i; 1382f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek s = strtok(0, " "); 1383f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1384f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (print_usrs(&args[0], &args[i])) 1385f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 1; 1386f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1387f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else 1388f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek line[numChars++] = c; 1389f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1390f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1391f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek fclose(fp); 1392f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 0; 1393f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek} 1394f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek 1395f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek/******************************************************************************/ 13960d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/* Command line processing. */ 13970d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek/******************************************************************************/ 13987ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregorint write_pch_file(const char *filename, int argc, const char *argv[]) { 13997ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor CXIndex Idx; 14007ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor CXTranslationUnit TU; 14017ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor struct CXUnsavedFile *unsaved_files = 0; 14027ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor int num_unsaved_files = 0; 14037ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 14047ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor Idx = clang_createIndex(/* excludeDeclsFromPCH */1, /* displayDiagnosics=*/1); 14057ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 14067ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor if (parse_remapped_files(argc, argv, 0, &unsaved_files, &num_unsaved_files)) { 14077ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor clang_disposeIndex(Idx); 14087ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor return -1; 14097ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor } 14107ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 14117ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor TU = clang_parseTranslationUnit(Idx, 0, 14127ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor argv + num_unsaved_files, 14137ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor argc - num_unsaved_files, 14147ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor unsaved_files, 14157ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor num_unsaved_files, 14167ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor CXTranslationUnit_Incomplete); 14177ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor if (!TU) { 14187ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor fprintf(stderr, "Unable to load translation unit!\n"); 14197ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 14207ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor clang_disposeIndex(Idx); 14217ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor return 1; 14227ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor } 14237ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 14241999844e7a18786e61e619e1dc6c789827541863Douglas Gregor if (clang_saveTranslationUnit(TU, filename, clang_defaultSaveOptions(TU))) 14257ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor fprintf(stderr, "Unable to write PCH file %s\n", filename); 14267ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor clang_disposeTranslationUnit(TU); 14277ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor free_remapped_files(unsaved_files, num_unsaved_files); 14287ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor clang_disposeIndex(Idx); 14297ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor return 0; 14307ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor} 14317ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 14327ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor/******************************************************************************/ 14337ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor/* Command line processing. */ 14347ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor/******************************************************************************/ 1435f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 1436e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregorstatic CXCursorVisitor GetVisitor(const char *s) { 14377d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (s[0] == '\0') 1438e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor return FilteredPrintingVisitor; 14397d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (strcmp(s, "-usrs") == 0) 14407d40562f83552b7295411e10ee887d8d55470679Ted Kremenek return USRVisitor; 14417d40562f83552b7295411e10ee887d8d55470679Ted Kremenek return NULL; 14427d40562f83552b7295411e10ee887d8d55470679Ted Kremenek} 14437d40562f83552b7295411e10ee887d8d55470679Ted Kremenek 1444f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenekstatic void print_usage(void) { 1445f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek fprintf(stderr, 14460d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek "usage: c-index-test -code-completion-at=<site> <compiler arguments>\n" 14471982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor " c-index-test -code-completion-timing=<site> <compiler arguments>\n" 1448f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor " c-index-test -cursor-at=<site> <compiler arguments>\n" 14491d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek " c-index-test -test-file-scan <AST file> <source file> " 14501d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek "[FileCheck prefix]\n" 1451fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek " c-index-test -test-load-tu <AST file> <symbol filter> " 1452fe6fd3d41a7f48317d6856c9327b6cead32c3498Ted Kremenek "[FileCheck prefix]\n" 14537d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " c-index-test -test-load-tu-usrs <AST file> <symbol filter> " 14547d40562f83552b7295411e10ee887d8d55470679Ted Kremenek "[FileCheck prefix]\n" 14551982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor " c-index-test -test-load-source <symbol filter> {<args>}*\n"); 1456f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor fprintf(stderr, 1457abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor " c-index-test -test-load-source-reparse <trials> <symbol filter> " 1458abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor " {<args>}*\n" 14591982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor " c-index-test -test-load-source-usrs <symbol filter> {<args>}*\n" 146016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-annotate-tokens=<range> {<args>}*\n" 146116b55a71695a33c094383295cc7b7a2080e098daTed Kremenek " c-index-test -test-inclusion-stack-source {<args>}*\n" 14623bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek " c-index-test -test-inclusion-stack-tu <AST file>\n" 1463f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek " c-index-test -test-print-linkage-source {<args>}*\n" 14648e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek " c-index-test -test-print-typekind {<args>}*\n" 146553513d2e24205965ffd5d4780a10f41cc335f17bChandler Carruth " c-index-test -print-usr [<CursorKind> {<args>}]*\n"); 146653513d2e24205965ffd5d4780a10f41cc335f17bChandler Carruth fprintf(stderr, 14677ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor " c-index-test -print-usr-file <file>\n" 14687ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor " c-index-test -write-pch <file> <compiler arguments>\n\n"); 1469caf4bd3d3c66fd95b0159356e2c4da7fbf2897f3Douglas Gregor fprintf(stderr, 14707d40562f83552b7295411e10ee887d8d55470679Ted Kremenek " <symbol filter> values:\n%s", 14710d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " all - load all symbols, including those from PCH\n" 14720d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " local - load all symbols except those in PCH\n" 14730d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " category - only load ObjC categories (non-PCH)\n" 14740d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " interface - only load ObjC interfaces (non-PCH)\n" 14750d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " protocol - only load ObjC protocols (non-PCH)\n" 14760d435191fc8f78338c2ef26d566cfe4d2c52c706Ted Kremenek " function - only load functions (non-PCH)\n" 1477625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar " typedef - only load typdefs (non-PCH)\n" 1478625e4efc2c64b638285c9f836cc00a3ad7d0bd5fDaniel Dunbar " scan-function - scan function bodies (non-PCH)\n\n"); 1479f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek} 1480f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek 14816edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar/***/ 14826edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 14836edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbarint cindextest_main(int argc, const char **argv) { 14840a812cf707da15dadd19fdeb0178b9707f4e01a6Douglas Gregor clang_enableStackTraces(); 1485f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek if (argc > 2 && strstr(argv[1], "-code-completion-at=") == argv[1]) 14861982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor return perform_code_completion(argc, argv, 0); 14871982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor if (argc > 2 && strstr(argv[1], "-code-completion-timing=") == argv[1]) 14881982c18522a4aefd57207bbd2d66d93945c41f92Douglas Gregor return perform_code_completion(argc, argv, 1); 1489f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor if (argc > 2 && strstr(argv[1], "-cursor-at=") == argv[1]) 1490f2c87bd0368775927ead93e0dee3e4f3ca3f9a63Douglas Gregor return inspect_cursor_at(argc, argv); 14917d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strncmp(argv[1], "-test-load-tu", 13) == 0) { 1492e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 13); 14937d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (I) 1494ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load_tu(argv[2], argv[3], argc >= 5 ? argv[4] : 0, I, 1495ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek NULL); 14967d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 1497abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor else if (argc >= 5 && strncmp(argv[1], "-test-load-source-reparse", 25) == 0){ 1498abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 25); 1499abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor if (I) { 1500abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor int trials = atoi(argv[2]); 1501abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor return perform_test_reparse_source(argc - 4, argv + 4, trials, argv[3], I, 1502abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor NULL); 1503abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor } 1504abc563f554951259bbe0315055cad92ee14d87e4Douglas Gregor } 15057d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strncmp(argv[1], "-test-load-source", 17) == 0) { 1506e5b72bac8312f83aeb2e29b3988ebfa30f2b4ca2Douglas Gregor CXCursorVisitor I = GetVisitor(argv[1] + 17); 15077d40562f83552b7295411e10ee887d8d55470679Ted Kremenek if (I) 1508ce2ae88f834c740a28b7e074a4477039918f9bb0Ted Kremenek return perform_test_load_source(argc - 3, argv + 3, argv[2], I, NULL); 15097d40562f83552b7295411e10ee887d8d55470679Ted Kremenek } 15107d40562f83552b7295411e10ee887d8d55470679Ted Kremenek else if (argc >= 4 && strcmp(argv[1], "-test-file-scan") == 0) 15111d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek return perform_file_scan(argv[2], argv[3], 15121d5fdf3d3b5ea2640ebe8673814a0b6ab7cf5eb2Ted Kremenek argc >= 5 ? argv[4] : 0); 1513fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor else if (argc > 2 && strstr(argv[1], "-test-annotate-tokens=") == argv[1]) 1514fc8ea23eb6cbaaa5046f2abb4c033e24c8659efdDouglas Gregor return perform_token_annotation(argc, argv); 151516b55a71695a33c094383295cc7b7a2080e098daTed Kremenek else if (argc > 2 && strcmp(argv[1], "-test-inclusion-stack-source") == 0) 151616b55a71695a33c094383295cc7b7a2080e098daTed Kremenek return perform_test_load_source(argc - 2, argv + 2, "all", NULL, 151716b55a71695a33c094383295cc7b7a2080e098daTed Kremenek PrintInclusionStack); 151816b55a71695a33c094383295cc7b7a2080e098daTed Kremenek else if (argc > 2 && strcmp(argv[1], "-test-inclusion-stack-tu") == 0) 151916b55a71695a33c094383295cc7b7a2080e098daTed Kremenek return perform_test_load_tu(argv[2], "all", NULL, NULL, 152016b55a71695a33c094383295cc7b7a2080e098daTed Kremenek PrintInclusionStack); 15213bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek else if (argc > 2 && strcmp(argv[1], "-test-print-linkage-source") == 0) 15223bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek return perform_test_load_source(argc - 2, argv + 2, "all", PrintLinkage, 15233bed527b10502c6a5916ebb6e6abc49687ce1444Ted Kremenek NULL); 15248e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek else if (argc > 2 && strcmp(argv[1], "-test-print-typekind") == 0) 15258e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek return perform_test_load_source(argc - 2, argv + 2, "all", 15268e0ac174c8c8c980927b3e1447f16f62fbe2a2e4Ted Kremenek PrintTypeKind, 0); 1527f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else if (argc > 1 && strcmp(argv[1], "-print-usr") == 0) { 1528f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek if (argc > 2) 1529f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return print_usrs(argv + 2, argv + argc); 1530f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else { 1531f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek display_usrs(); 1532f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return 1; 1533f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1534f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek } 1535f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek else if (argc > 2 && strcmp(argv[1], "-print-usr-file") == 0) 1536f7b714df46cdbdf9e2ebf26eb2fd7881790d83e6Ted Kremenek return print_usrs_file(argv[2]); 15377ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor else if (argc > 2 && strcmp(argv[1], "-write-pch") == 0) 15387ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor return write_pch_file(argv[2], argc - 3, argv + 3); 15397ae2faafd30524ef5f863bb3b8701977888839bbDouglas Gregor 1540f5d9c9340b1a5ecd6ab62b916600b6e19e69539fTed Kremenek print_usage(); 154150398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff return 1; 154250398199fb10e196a8d92fbf7a062dbe42ed88fdSteve Naroff} 15436edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15446edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar/***/ 15456edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15466edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar/* We intentionally run in a separate thread to ensure we at least minimal 15476edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar * testing of a multithreaded environment (for example, having a reduced stack 15486edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar * size). */ 15496edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15506edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar#include "llvm/Config/config.h" 15516edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar#ifdef HAVE_PTHREAD_H 15526edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15536edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar#include <pthread.h> 15546edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15556edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbartypedef struct thread_info { 15566edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar int argc; 15576edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar const char **argv; 15586edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar int result; 15596edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar} thread_info; 15606edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbarvoid *thread_runner(void *client_data_v) { 15616edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar thread_info *client_data = client_data_v; 15626edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar client_data->result = cindextest_main(client_data->argc, client_data->argv); 15636edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar return 0; 15646edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar} 15656edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15666edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbarint main(int argc, const char **argv) { 15676edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar thread_info client_data; 15686edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar pthread_t thread; 15696edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar int res; 15706edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15716edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar client_data.argc = argc; 15726edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar client_data.argv = argv; 15736edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar res = pthread_create(&thread, 0, thread_runner, &client_data); 15746edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar if (res != 0) { 15756edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar perror("thread creation failed"); 15766edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar return 1; 15776edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar } 15786edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15796edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar res = pthread_join(thread, 0); 15806edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar if (res != 0) { 15816edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar perror("thread join failed"); 15826edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar return 1; 15836edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar } 15846edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15856edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar return client_data.result; 15866edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar} 15876edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15886edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar#else 15896edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15906edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbarint main(int argc, const char **argv) { 15916edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar return cindextest_main(argc, argv); 15926edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar} 15936edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar 15946edc8001f2c2f5a95ed29259d8f2c0b9faa4bc83Daniel Dunbar#endif 1595