Try caching the lookup results

Author: YexuanXiao

clang/include/clang/AST/ASTContext.h

@@ -1218,6 +1218,11 @@ class ASTContext : public RefCountedBase<ASTContext> {
 #define HLSL_INTANGIBLE_TYPE(Name, Id, SingletonId) CanQualType SingletonId;
 #include "clang/Basic/HLSLIntangibleTypes.def"
 
+  // Cache size_t and ptrdiff_t typedefs
+  // (C99 7.17), defined in <stddef.h>.
+  mutable QualType PtrDiffTy;
+  mutable QualType UnsignedSizeTy;
+
   // Types for deductions in C++0x [stmt.ranged]'s desugaring. Built on demand.
   mutable QualType AutoDeductTy;     // Deduction against 'auto'.
   mutable QualType AutoRRefDeductTy; // Deduction against 'auto &&'.
@@ -1941,11 +1946,11 @@ class ASTContext : public RefCountedBase<ASTContext> {
   /// <stddef.h>.
   ///
   /// The sizeof operator requires this (C99 6.5.3.4p4).
-  CanQualType getSizeType() const;
+  QualType getSizeType() const;
 
   /// Return the unique signed counterpart of
   /// the integer type corresponding to size_t.
-  CanQualType getSignedSizeType() const;
+  QualType getSignedSizeType() const;
 
   /// Return the unique type for "intmax_t" (C99 7.18.1.5), defined in
   /// <stdint.h>.
@@ -2442,10 +2447,6 @@ class ASTContext : public RefCountedBase<ASTContext> {
   QualType GetBuiltinType(unsigned ID, GetBuiltinTypeError &Error,
                           unsigned *IntegerConstantArgs = nullptr) const;
 
-  QualType getCGlobalCXXStdNSTypedef(const NamespaceDecl *StdNS,
-                                     StringRef DefName,
-                                     QualType FallBack = {}) const;
-
   /// Types and expressions required to build C++2a three-way comparisons
   /// using operator<=>, including the values return by builtin <=> operators.
   ComparisonCategories CompCategories;

clang/lib/AST/ASTContext.cpp

@@ -6666,16 +6666,55 @@ QualType ASTContext::getTagDeclType(const TagDecl *Decl) const {
   return getTypeDeclType(const_cast<TagDecl*>(Decl));
 }
 
+static QualType LookupCGlobalCXXStdNSTypedef(const ASTContext &Ctx,
+                                             StringRef DefName,
+                                             CanQualType const &CanType) {
+  DeclContextLookupResult Lookup;
+  if (Ctx.getLangOpts().C99) {
+    Lookup = Ctx.getTranslationUnitDecl()->lookup(&Ctx.Idents.get(DefName));
+  } else if (Ctx.getLangOpts().CPlusPlus) {
+    const NamespaceDecl *StdNS = nullptr;
+    auto LookupStdNS =
+        Ctx.getTranslationUnitDecl()->lookup(&Ctx.Idents.get("std"));
+    if (!LookupStdNS.empty()) {
+      StdNS = dyn_cast<NamespaceDecl>(LookupStdNS.front());
+    }
+    if (StdNS) {
+      Lookup = StdNS->lookup(&Ctx.Idents.get(DefName));
+    } else {
+      Lookup = Ctx.getTranslationUnitDecl()->lookup(&Ctx.Idents.get(DefName));
+    }
+  }
+  if (!Lookup.empty()) {
+    if (auto *TD = dyn_cast<TypedefNameDecl>(Lookup.front())) {
+      auto Result = Ctx.getTypeDeclType(TD);
+      if (!Result.isNull() && Ctx.hasSameType(Result, CanType) &&
+          Ctx.getTypeAlign(Result) == Ctx.getTypeAlign(CanType)) {
+        return Result;
+      }
+    }
+  }
+  return QualType();
+}
+
 /// getSizeType - Return the unique type for "size_t" (C99 7.17), the result
 /// of the sizeof operator (C99 6.5.3.4p4). The value is target dependent and
 /// needs to agree with the definition in <stddef.h>.
-CanQualType ASTContext::getSizeType() const {
-  return getFromTargetType(Target->getSizeType());
+QualType ASTContext::getSizeType() const {
+  if (UnsignedSizeTy.isNull()) {
+    auto CanType = getFromTargetType(Target->getSizeType());
+    if (auto TypeDef = LookupCGlobalCXXStdNSTypedef(*this, "size_t", CanType);
+        TypeDef.isNull())
+      return CanType;
+    else
+      UnsignedSizeTy = TypeDef;
+  }
+  return UnsignedSizeTy;
 }
 
 /// Return the unique signed counterpart of the integer type
 /// corresponding to size_t.
-CanQualType ASTContext::getSignedSizeType() const {
+QualType ASTContext::getSignedSizeType() const {
   return getFromTargetType(Target->getSignedSizeType());
 }
 
@@ -6714,7 +6753,16 @@ QualType ASTContext::getUIntPtrType() const {
 /// getPointerDiffType - Return the unique type for "ptrdiff_t" (C99 7.17)
 /// defined in <stddef.h>. Pointer - pointer requires this (C99 6.5.6p9).
 QualType ASTContext::getPointerDiffType() const {
-  return getFromTargetType(Target->getPtrDiffType(LangAS::Default));
+  if (PtrDiffTy.isNull()) {
+    auto CanType = getFromTargetType(Target->getPtrDiffType(LangAS::Default));
+    if (auto TypeDef =
+            LookupCGlobalCXXStdNSTypedef(*this, "ptrdiff_t", CanType);
+        TypeDef.isNull())
+      return CanType;
+    else
+      PtrDiffTy = TypeDef;
+  }
+  return PtrDiffTy;
 }
 
 /// Return the unique unsigned counterpart of "ptrdiff_t"
@@ -12556,35 +12604,6 @@ QualType ASTContext::GetBuiltinType(unsigned Id,
   return getFunctionType(ResType, ArgTypes, EPI);
 }
 
-QualType ASTContext::getCGlobalCXXStdNSTypedef(const NamespaceDecl *StdNS,
-                                               StringRef DefName,
-                                               QualType FallBack) const {
-  DeclContextLookupResult Lookup;
-  if (getLangOpts().C99) {
-    Lookup = getTranslationUnitDecl()->lookup(&Idents.get(DefName));
-  } else if (getLangOpts().CPlusPlus) {
-    if (StdNS == nullptr) {
-      auto LookupStdNS = getTranslationUnitDecl()->lookup(&Idents.get("std"));
-      if (!LookupStdNS.empty()) {
-        StdNS = dyn_cast<NamespaceDecl>(LookupStdNS.front());
-      }
-    }
-    if (StdNS) {
-      Lookup = StdNS->lookup(&Idents.get(DefName));
-    } else {
-      Lookup = getTranslationUnitDecl()->lookup(&Idents.get(DefName));
-    }
-  }
-  if (!Lookup.empty()) {
-    if (auto *TD = dyn_cast<TypedefNameDecl>(Lookup.front())) {
-      if (auto Result = getTypeDeclType(TD); !Result.isNull()) {
-        return Result;
-      }
-    }
-  }
-  return FallBack;
-}
-
 static GVALinkage basicGVALinkageForFunction(const ASTContext &Context,
                                              const FunctionDecl *FD) {
   if (!FD->isExternallyVisible())

clang/lib/AST/ExprCXX.cpp

@@ -1700,10 +1700,8 @@ SizeOfPackExpr *SizeOfPackExpr::Create(ASTContext &Context,
                                        ArrayRef<TemplateArgument> PartialArgs) {
   void *Storage =
       Context.Allocate(totalSizeToAlloc<TemplateArgument>(PartialArgs.size()));
-  return new (Storage) SizeOfPackExpr(
-      Context.getCGlobalCXXStdNSTypedef(nullptr, "size_t",
-                                        Context.getSizeType()),
-      OperatorLoc, Pack, PackLoc, RParenLoc, Length, PartialArgs);
+  return new (Storage) SizeOfPackExpr(Context.getSizeType(), OperatorLoc, Pack,
+                                      PackLoc, RParenLoc, Length, PartialArgs);
 }
 
 SizeOfPackExpr *SizeOfPackExpr::CreateDeserialized(ASTContext &Context,

clang/lib/CodeGen/CGCall.cpp

@@ -191,8 +191,10 @@ static void appendParameterTypes(const CodeGenTypes &CGT,
   assert(ExtInfos.size() == FPT->getNumParams());
   for (unsigned I = 0, E = FPT->getNumParams(); I != E; ++I) {
     prefix.push_back(FPT->getParamType(I));
-    if (ExtInfos[I].hasPassObjectSize())
-      prefix.push_back(CGT.getContext().getSizeType());
+    if (ExtInfos[I].hasPassObjectSize()) {
+      auto &Ctx = CGT.getContext();
+      prefix.push_back(Ctx.getCanonicalType(Ctx.getSizeType()));
+    }
   }
 
   addExtParameterInfosForCall(paramInfos, FPT.getTypePtr(), PrefixSize,

clang/lib/CodeGen/CGCoroutine.cpp

@@ -1002,14 +1002,14 @@ RValue CodeGenFunction::EmitCoroutineIntrinsic(const CallExpr *E,
   }
   case llvm::Intrinsic::coro_size: {
     auto &Context = getContext();
-    CanQualType SizeTy = Context.getSizeType();
+    CanQualType SizeTy = Context.getCanonicalType(Context.getSizeType());
     llvm::IntegerType *T = Builder.getIntNTy(Context.getTypeSize(SizeTy));
     llvm::Function *F = CGM.getIntrinsic(llvm::Intrinsic::coro_size, T);
     return RValue::get(Builder.CreateCall(F));
   }
   case llvm::Intrinsic::coro_align: {
     auto &Context = getContext();
-    CanQualType SizeTy = Context.getSizeType();
+    CanQualType SizeTy = Context.getCanonicalType(Context.getSizeType());
     llvm::IntegerType *T = Builder.getIntNTy(Context.getTypeSize(SizeTy));
     llvm::Function *F = CGM.getIntrinsic(llvm::Intrinsic::coro_align, T);
     return RValue::get(Builder.CreateCall(F));

clang/lib/CodeGen/CGObjCMac.cpp

@@ -285,7 +285,7 @@ class ObjCCommonTypesHelper {
     SmallVector<CanQualType, 5> Params;
     Params.push_back(Ctx.VoidPtrTy);
     Params.push_back(Ctx.VoidPtrTy);
-    Params.push_back(Ctx.getSizeType());
+    Params.push_back(Ctx.getCanonicalType(Ctx.getSizeType()));
     Params.push_back(Ctx.BoolTy);
     Params.push_back(Ctx.BoolTy);
     llvm::FunctionType *FTy = Types.GetFunctionType(

clang/lib/Sema/SemaExpr.cpp

@@ -4025,23 +4025,11 @@ ExprResult Sema::ActOnNumericConstant(const Token &Tok, Scope *UDLScope) {
 
         // Does it fit in size_t?
         if (ResultVal.isIntN(SizeTSize)) {
-          // Does it fit in ssize_t?
-          if (!Literal.isUnsigned && ResultVal[SizeTSize - 1] == 0) {
-            auto SignedSize = Context.getSignedSizeType();
-            if (auto PtrDiff = Context.getCGlobalCXXStdNSTypedef(
-                    getStdNamespace(), "ptrdiff_t");
-                !PtrDiff.isNull() && Context.hasSameType(PtrDiff, SignedSize))
-              Ty = PtrDiff;
-            else if (auto SSize = Context.getCGlobalCXXStdNSTypedef(
-                         getStdNamespace(), "ssize_t");
-                     !SSize.isNull() && Context.hasSameType(SSize, SignedSize))
-              Ty = SSize;
-            else
-              Ty = SignedSize;
-          } else if (AllowUnsigned) {
-            Ty = Context.getCGlobalCXXStdNSTypedef(getStdNamespace(), "size_t",
-                                                   Context.getSizeType());
-          }
+          // Does it fit in ptrdiff_t?
+          if (!Literal.isUnsigned && ResultVal[SizeTSize - 1] == 0)
+            Ty = Context.getPointerDiffType();
+          else if (AllowUnsigned)
+            Ty = Context.getSizeType();
           Width = SizeTSize;
         }
       }
@@ -4714,11 +4702,7 @@ ExprResult Sema::CreateUnaryExprOrTypeTraitExpr(TypeSourceInfo *TInfo,
 
   // C99 6.5.3.4p4: the type (an unsigned integer type) is size_t.
   return new (Context) UnaryExprOrTypeTraitExpr(
-      ExprKind, TInfo,
-      Context.getCGlobalCXXStdNSTypedef(
-          getLangOpts().CPlusPlus ? getStdNamespace() : nullptr, "size_t",
-          Context.getSizeType()),
-      OpLoc, R.getEnd());
+      ExprKind, TInfo, Context.getSizeType(), OpLoc, R.getEnd());
 }
 
 ExprResult
@@ -4761,11 +4745,7 @@ Sema::CreateUnaryExprOrTypeTraitExpr(Expr *E, SourceLocation OpLoc,
 
   // C99 6.5.3.4p4: the type (an unsigned integer type) is size_t.
   return new (Context) UnaryExprOrTypeTraitExpr(
-      ExprKind, E,
-      Context.getCGlobalCXXStdNSTypedef(
-          getLangOpts().CPlusPlus ? getStdNamespace() : nullptr, "size_t",
-          Context.getSizeType()),
-      OpLoc, E->getSourceRange().getEnd());
+      ExprKind, E, Context.getSizeType(), OpLoc, E->getSourceRange().getEnd());
 }
 
 ExprResult
@@ -11375,9 +11355,7 @@ QualType Sema::CheckSubtractionOperands(ExprResult &LHS, ExprResult &RHS,
 
       if (CompLHSTy)
         *CompLHSTy = LHS.get()->getType();
-      return Context.getCGlobalCXXStdNSTypedef(
-          getLangOpts().CPlusPlus ? getStdNamespace() : nullptr, "ptrdiff_t",
-          Context.getPointerDiffType());
+      return Context.getPointerDiffType();
     }
   }
 

clang/lib/StaticAnalyzer/Checkers/VLASizeChecker.cpp

@@ -94,7 +94,7 @@ ProgramStateRef VLASizeChecker::checkVLA(CheckerContext &C,
 
   ASTContext &Ctx = C.getASTContext();
   SValBuilder &SVB = C.getSValBuilder();
-  CanQualType SizeTy = Ctx.getSizeType();
+  CanQualType SizeTy = Ctx.getCanonicalType(Ctx.getSizeType());
   uint64_t SizeMax =
       SVB.getBasicValueFactory().getMaxValue(SizeTy)->getZExtValue();