[libclang] Add some checks to make sure the given CXTranslationUnit is not null, before using it.
git-svn-id: https://llvm.org/svn/llvm-project/cfe/trunk@178800 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/tools/libclang/CIndex.cpp b/tools/libclang/CIndex.cpp
index 59a7e8d..a81f1e4 100644
--- a/tools/libclang/CIndex.cpp
+++ b/tools/libclang/CIndex.cpp
@@ -2927,6 +2927,9 @@
}
CXCursor clang_getTranslationUnitCursor(CXTranslationUnit TU) {
+ if (!TU)
+ return clang_getNullCursor();
+
ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
return MakeCXCursor(CXXUnit->getASTContext().getTranslationUnitDecl(), TU);
}
@@ -4914,6 +4917,9 @@
if (NumTokens)
*NumTokens = 0;
+ if (!TU)
+ return;
+
ASTUnit *CXXUnit = cxtu::getASTUnit(TU);
if (!CXXUnit || !Tokens || !NumTokens)
return;
@@ -5610,7 +5616,7 @@
void clang_annotateTokens(CXTranslationUnit TU,
CXToken *Tokens, unsigned NumTokens,
CXCursor *Cursors) {
- if (NumTokens == 0 || !Tokens || !Cursors) {
+ if (!TU || NumTokens == 0 || !Tokens || !Cursors) {
LOG_FUNC_SECTION { *Log << "<null input>"; }
return;
}