1 files changed, 7 insertions, 1 deletions
diff --git a/clang/tools/libclang/CXTranslationUnit.h b/clang/tools/libclang/CXTranslationUnit.h
index 1c713bf2d60..323d4b78aa9 100644
--- a/clang/tools/libclang/CXTranslationUnit.h
+++ b/clang/tools/libclang/CXTranslationUnit.h
@@ -14,6 +14,8 @@
 #ifndef LLVM_CLANG_CXTRANSLATIONUNIT_H
 #define LLVM_CLANG_CXTRANSLATIONUNIT_H
 
+#include "clang-c/Index.h"
+
 extern "C" {
 struct CXTranslationUnitImpl {
   void *CIdx;
@@ -33,7 +35,11 @@ namespace clang {
 namespace cxtu {
 
 CXTranslationUnitImpl *MakeCXTranslationUnit(CIndexer *CIdx, ASTUnit *TU);
-  
+
+static inline ASTUnit *getASTUnit(CXTranslationUnit TU) {
+  return static_cast<ASTUnit *>(TU->TUData);
+}
+
 class CXTUOwner {
   CXTranslationUnitImpl *TU;