154d2267b8
When targeting runtimes that support constant literal classes, emit ObjC
literal expressions @(number), @[], and @{} as compile-time constant
data structures rather than runtime msgSend calls. This reduces code
size and runtime overhead at the cost of increased data segment size,
and avoids repeated heap allocation of equivalent literal objects.
The feature is not supported with the fragile ABI or GNU runtimes, where
it is automatically disabled.
The feature can be disabled altogether with -fno-objc-constant-literals,
or individually per literal kind:
-fno-constant-nsnumber-literals
-fno-constant-nsarray-literals
-fno-constant-nsdictionary-literals
Custom backing class names can be specified via:
-fconstant-array-class=<name>
-fconstant-dictionary-class=<name>
-fconstant-integer-number-class=<name>
-fconstant-float-number-class=<name>
-fconstant-double-number-class=<name>
rdar://45380392
rdar://168106035
---------
Co-authored-by: Ben D. Jones <bendjones@apple.com>
360 lines
14 KiB
C++
360 lines
14 KiB
C++
//===- ExprObjC.cpp - (ObjC) Expression AST Node Implementation -----------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements the subclesses of Expr class declared in ExprObjC.h
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "clang/AST/ExprObjC.h"
|
|
#include "clang/AST/ASTContext.h"
|
|
#include "clang/AST/Attr.h"
|
|
#include "clang/AST/ComputeDependence.h"
|
|
#include "clang/AST/SelectorLocationsKind.h"
|
|
#include "clang/AST/Type.h"
|
|
#include "clang/AST/TypeLoc.h"
|
|
#include "llvm/Support/ErrorHandling.h"
|
|
#include <cassert>
|
|
#include <cstdint>
|
|
|
|
using namespace clang;
|
|
|
|
ObjCArrayLiteral::ObjCArrayLiteral(ArrayRef<Expr *> Elements, QualType T,
|
|
ObjCMethodDecl *Method,
|
|
bool ExpressibleAsConstantInitializer,
|
|
SourceRange SR)
|
|
: ObjCObjectLiteral(ObjCArrayLiteralClass, T,
|
|
ExpressibleAsConstantInitializer, VK_PRValue,
|
|
OK_Ordinary),
|
|
NumElements(Elements.size()), Range(SR), ArrayWithObjectsMethod(Method) {
|
|
Expr **SaveElements = getElements();
|
|
for (unsigned I = 0, N = Elements.size(); I != N; ++I)
|
|
SaveElements[I] = Elements[I];
|
|
|
|
setDependence(computeDependence(this));
|
|
}
|
|
|
|
ObjCArrayLiteral *
|
|
ObjCArrayLiteral::Create(const ASTContext &C, ArrayRef<Expr *> Elements,
|
|
QualType T, ObjCMethodDecl *Method,
|
|
bool ExpressibleAsConstantInitializer,
|
|
SourceRange SR) {
|
|
void *Mem = C.Allocate(totalSizeToAlloc<Expr *>(Elements.size()));
|
|
return new (Mem) ObjCArrayLiteral(Elements, T, Method,
|
|
ExpressibleAsConstantInitializer, SR);
|
|
}
|
|
|
|
ObjCArrayLiteral *ObjCArrayLiteral::CreateEmpty(const ASTContext &C,
|
|
unsigned NumElements) {
|
|
void *Mem = C.Allocate(totalSizeToAlloc<Expr *>(NumElements));
|
|
auto *ALE = new (Mem) ObjCArrayLiteral(EmptyShell(), NumElements);
|
|
ALE->setExpressibleAsConstantInitializer(NumElements == 0);
|
|
return ALE;
|
|
}
|
|
|
|
ObjCDictionaryLiteral::ObjCDictionaryLiteral(
|
|
ArrayRef<ObjCDictionaryElement> VK, bool HasPackExpansions, QualType T,
|
|
ObjCMethodDecl *Method, bool ExpressibleAsConstantInitializer,
|
|
SourceRange SR)
|
|
: ObjCObjectLiteral(ObjCDictionaryLiteralClass, T,
|
|
ExpressibleAsConstantInitializer, VK_PRValue,
|
|
OK_Ordinary),
|
|
NumElements(VK.size()), HasPackExpansions(HasPackExpansions), Range(SR),
|
|
DictWithObjectsMethod(Method) {
|
|
KeyValuePair *KeyValues = getTrailingObjects<KeyValuePair>();
|
|
ExpansionData *Expansions =
|
|
HasPackExpansions ? getTrailingObjects<ExpansionData>() : nullptr;
|
|
for (unsigned I = 0; I < NumElements; I++) {
|
|
KeyValues[I].Key = VK[I].Key;
|
|
KeyValues[I].Value = VK[I].Value;
|
|
if (Expansions) {
|
|
Expansions[I].EllipsisLoc = VK[I].EllipsisLoc;
|
|
if (VK[I].NumExpansions)
|
|
Expansions[I].NumExpansionsPlusOne = *VK[I].NumExpansions + 1;
|
|
else
|
|
Expansions[I].NumExpansionsPlusOne = 0;
|
|
}
|
|
}
|
|
setDependence(computeDependence(this));
|
|
}
|
|
|
|
ObjCDictionaryLiteral *ObjCDictionaryLiteral::Create(
|
|
const ASTContext &C, ArrayRef<ObjCDictionaryElement> VK,
|
|
bool HasPackExpansions, QualType T, ObjCMethodDecl *Method,
|
|
bool ExpressibleAsConstantInitializer, SourceRange SR) {
|
|
void *Mem = C.Allocate(totalSizeToAlloc<KeyValuePair, ExpansionData>(
|
|
VK.size(), HasPackExpansions ? VK.size() : 0));
|
|
return new (Mem) ObjCDictionaryLiteral(VK, HasPackExpansions, T, Method,
|
|
ExpressibleAsConstantInitializer, SR);
|
|
}
|
|
|
|
ObjCDictionaryLiteral *
|
|
ObjCDictionaryLiteral::CreateEmpty(const ASTContext &C, unsigned NumElements,
|
|
bool HasPackExpansions) {
|
|
void *Mem = C.Allocate(totalSizeToAlloc<KeyValuePair, ExpansionData>(
|
|
NumElements, HasPackExpansions ? NumElements : 0));
|
|
auto *DLE = new (Mem)
|
|
ObjCDictionaryLiteral(EmptyShell(), NumElements, HasPackExpansions);
|
|
DLE->setExpressibleAsConstantInitializer(NumElements == 0);
|
|
return DLE;
|
|
}
|
|
|
|
QualType ObjCPropertyRefExpr::getReceiverType(const ASTContext &ctx) const {
|
|
if (isClassReceiver())
|
|
return ctx.getObjCInterfaceType(getClassReceiver());
|
|
|
|
if (isSuperReceiver())
|
|
return getSuperReceiverType();
|
|
|
|
return getBase()->getType();
|
|
}
|
|
|
|
ObjCMessageExpr::ObjCMessageExpr(QualType T, ExprValueKind VK,
|
|
SourceLocation LBracLoc,
|
|
SourceLocation SuperLoc, bool IsInstanceSuper,
|
|
QualType SuperType, Selector Sel,
|
|
ArrayRef<SourceLocation> SelLocs,
|
|
SelectorLocationsKind SelLocsK,
|
|
ObjCMethodDecl *Method, ArrayRef<Expr *> Args,
|
|
SourceLocation RBracLoc, bool isImplicit)
|
|
: Expr(ObjCMessageExprClass, T, VK, OK_Ordinary),
|
|
SelectorOrMethod(
|
|
reinterpret_cast<uintptr_t>(Method ? Method : Sel.getAsOpaquePtr())),
|
|
Kind(IsInstanceSuper ? SuperInstance : SuperClass),
|
|
HasMethod(Method != nullptr), IsDelegateInitCall(false),
|
|
IsImplicit(isImplicit), SuperLoc(SuperLoc), LBracLoc(LBracLoc),
|
|
RBracLoc(RBracLoc) {
|
|
initArgsAndSelLocs(Args, SelLocs, SelLocsK);
|
|
setReceiverPointer(SuperType.getAsOpaquePtr());
|
|
setDependence(computeDependence(this));
|
|
}
|
|
|
|
ObjCMessageExpr::ObjCMessageExpr(QualType T, ExprValueKind VK,
|
|
SourceLocation LBracLoc,
|
|
TypeSourceInfo *Receiver, Selector Sel,
|
|
ArrayRef<SourceLocation> SelLocs,
|
|
SelectorLocationsKind SelLocsK,
|
|
ObjCMethodDecl *Method, ArrayRef<Expr *> Args,
|
|
SourceLocation RBracLoc, bool isImplicit)
|
|
: Expr(ObjCMessageExprClass, T, VK, OK_Ordinary),
|
|
SelectorOrMethod(
|
|
reinterpret_cast<uintptr_t>(Method ? Method : Sel.getAsOpaquePtr())),
|
|
Kind(Class), HasMethod(Method != nullptr), IsDelegateInitCall(false),
|
|
IsImplicit(isImplicit), LBracLoc(LBracLoc), RBracLoc(RBracLoc) {
|
|
initArgsAndSelLocs(Args, SelLocs, SelLocsK);
|
|
setReceiverPointer(Receiver);
|
|
setDependence(computeDependence(this));
|
|
}
|
|
|
|
ObjCMessageExpr::ObjCMessageExpr(QualType T, ExprValueKind VK,
|
|
SourceLocation LBracLoc, Expr *Receiver,
|
|
Selector Sel, ArrayRef<SourceLocation> SelLocs,
|
|
SelectorLocationsKind SelLocsK,
|
|
ObjCMethodDecl *Method, ArrayRef<Expr *> Args,
|
|
SourceLocation RBracLoc, bool isImplicit)
|
|
: Expr(ObjCMessageExprClass, T, VK, OK_Ordinary),
|
|
SelectorOrMethod(
|
|
reinterpret_cast<uintptr_t>(Method ? Method : Sel.getAsOpaquePtr())),
|
|
Kind(Instance), HasMethod(Method != nullptr), IsDelegateInitCall(false),
|
|
IsImplicit(isImplicit), LBracLoc(LBracLoc), RBracLoc(RBracLoc) {
|
|
initArgsAndSelLocs(Args, SelLocs, SelLocsK);
|
|
setReceiverPointer(Receiver);
|
|
setDependence(computeDependence(this));
|
|
}
|
|
|
|
void ObjCMessageExpr::initArgsAndSelLocs(ArrayRef<Expr *> Args,
|
|
ArrayRef<SourceLocation> SelLocs,
|
|
SelectorLocationsKind SelLocsK) {
|
|
setNumArgs(Args.size());
|
|
Expr **MyArgs = getArgs();
|
|
for (unsigned I = 0; I != Args.size(); ++I)
|
|
MyArgs[I] = Args[I];
|
|
|
|
SelLocsKind = SelLocsK;
|
|
if (!isImplicit() && SelLocsK == SelLoc_NonStandard)
|
|
llvm::copy(SelLocs, getStoredSelLocs());
|
|
}
|
|
|
|
ObjCMessageExpr *
|
|
ObjCMessageExpr::Create(const ASTContext &Context, QualType T, ExprValueKind VK,
|
|
SourceLocation LBracLoc, SourceLocation SuperLoc,
|
|
bool IsInstanceSuper, QualType SuperType, Selector Sel,
|
|
ArrayRef<SourceLocation> SelLocs,
|
|
ObjCMethodDecl *Method, ArrayRef<Expr *> Args,
|
|
SourceLocation RBracLoc, bool isImplicit) {
|
|
assert((!SelLocs.empty() || isImplicit) &&
|
|
"No selector locs for non-implicit message");
|
|
ObjCMessageExpr *Mem;
|
|
SelectorLocationsKind SelLocsK = SelectorLocationsKind();
|
|
if (isImplicit)
|
|
Mem = alloc(Context, Args.size(), 0);
|
|
else
|
|
Mem = alloc(Context, Args, RBracLoc, SelLocs, Sel, SelLocsK);
|
|
return new (Mem) ObjCMessageExpr(T, VK, LBracLoc, SuperLoc, IsInstanceSuper,
|
|
SuperType, Sel, SelLocs, SelLocsK, Method,
|
|
Args, RBracLoc, isImplicit);
|
|
}
|
|
|
|
ObjCMessageExpr *
|
|
ObjCMessageExpr::Create(const ASTContext &Context, QualType T, ExprValueKind VK,
|
|
SourceLocation LBracLoc, TypeSourceInfo *Receiver,
|
|
Selector Sel, ArrayRef<SourceLocation> SelLocs,
|
|
ObjCMethodDecl *Method, ArrayRef<Expr *> Args,
|
|
SourceLocation RBracLoc, bool isImplicit) {
|
|
assert((!SelLocs.empty() || isImplicit) &&
|
|
"No selector locs for non-implicit message");
|
|
ObjCMessageExpr *Mem;
|
|
SelectorLocationsKind SelLocsK = SelectorLocationsKind();
|
|
if (isImplicit)
|
|
Mem = alloc(Context, Args.size(), 0);
|
|
else
|
|
Mem = alloc(Context, Args, RBracLoc, SelLocs, Sel, SelLocsK);
|
|
return new (Mem)
|
|
ObjCMessageExpr(T, VK, LBracLoc, Receiver, Sel, SelLocs, SelLocsK, Method,
|
|
Args, RBracLoc, isImplicit);
|
|
}
|
|
|
|
ObjCMessageExpr *
|
|
ObjCMessageExpr::Create(const ASTContext &Context, QualType T, ExprValueKind VK,
|
|
SourceLocation LBracLoc, Expr *Receiver, Selector Sel,
|
|
ArrayRef<SourceLocation> SelLocs,
|
|
ObjCMethodDecl *Method, ArrayRef<Expr *> Args,
|
|
SourceLocation RBracLoc, bool isImplicit) {
|
|
assert((!SelLocs.empty() || isImplicit) &&
|
|
"No selector locs for non-implicit message");
|
|
ObjCMessageExpr *Mem;
|
|
SelectorLocationsKind SelLocsK = SelectorLocationsKind();
|
|
if (isImplicit)
|
|
Mem = alloc(Context, Args.size(), 0);
|
|
else
|
|
Mem = alloc(Context, Args, RBracLoc, SelLocs, Sel, SelLocsK);
|
|
return new (Mem)
|
|
ObjCMessageExpr(T, VK, LBracLoc, Receiver, Sel, SelLocs, SelLocsK, Method,
|
|
Args, RBracLoc, isImplicit);
|
|
}
|
|
|
|
ObjCMessageExpr *ObjCMessageExpr::CreateEmpty(const ASTContext &Context,
|
|
unsigned NumArgs,
|
|
unsigned NumStoredSelLocs) {
|
|
ObjCMessageExpr *Mem = alloc(Context, NumArgs, NumStoredSelLocs);
|
|
return new (Mem) ObjCMessageExpr(EmptyShell(), NumArgs);
|
|
}
|
|
|
|
ObjCMessageExpr *ObjCMessageExpr::alloc(const ASTContext &C,
|
|
ArrayRef<Expr *> Args,
|
|
SourceLocation RBraceLoc,
|
|
ArrayRef<SourceLocation> SelLocs,
|
|
Selector Sel,
|
|
SelectorLocationsKind &SelLocsK) {
|
|
SelLocsK = hasStandardSelectorLocs(Sel, SelLocs, Args, RBraceLoc);
|
|
unsigned NumStoredSelLocs =
|
|
(SelLocsK == SelLoc_NonStandard) ? SelLocs.size() : 0;
|
|
return alloc(C, Args.size(), NumStoredSelLocs);
|
|
}
|
|
|
|
ObjCMessageExpr *ObjCMessageExpr::alloc(const ASTContext &C, unsigned NumArgs,
|
|
unsigned NumStoredSelLocs) {
|
|
return (ObjCMessageExpr *)C.Allocate(
|
|
totalSizeToAlloc<void *, SourceLocation>(NumArgs + 1, NumStoredSelLocs),
|
|
alignof(ObjCMessageExpr));
|
|
}
|
|
|
|
void ObjCMessageExpr::getSelectorLocs(
|
|
SmallVectorImpl<SourceLocation> &SelLocs) const {
|
|
for (unsigned i = 0, e = getNumSelectorLocs(); i != e; ++i)
|
|
SelLocs.push_back(getSelectorLoc(i));
|
|
}
|
|
|
|
|
|
QualType ObjCMessageExpr::getCallReturnType(ASTContext &Ctx) const {
|
|
if (const ObjCMethodDecl *MD = getMethodDecl()) {
|
|
QualType QT = MD->getReturnType();
|
|
if (QT == Ctx.getObjCInstanceType()) {
|
|
// instancetype corresponds to expression types.
|
|
return getType();
|
|
}
|
|
return QT;
|
|
}
|
|
return Ctx.getReferenceQualifiedType(this);
|
|
}
|
|
|
|
SourceRange ObjCMessageExpr::getReceiverRange() const {
|
|
switch (getReceiverKind()) {
|
|
case Instance:
|
|
return getInstanceReceiver()->getSourceRange();
|
|
|
|
case Class:
|
|
return getClassReceiverTypeInfo()->getTypeLoc().getSourceRange();
|
|
|
|
case SuperInstance:
|
|
case SuperClass:
|
|
return getSuperLoc();
|
|
}
|
|
|
|
llvm_unreachable("Invalid ReceiverKind!");
|
|
}
|
|
|
|
Selector ObjCMessageExpr::getSelector() const {
|
|
if (HasMethod)
|
|
return reinterpret_cast<const ObjCMethodDecl *>(SelectorOrMethod)
|
|
->getSelector();
|
|
return Selector(SelectorOrMethod);
|
|
}
|
|
|
|
QualType ObjCMessageExpr::getReceiverType() const {
|
|
switch (getReceiverKind()) {
|
|
case Instance:
|
|
return getInstanceReceiver()->getType();
|
|
case Class:
|
|
return getClassReceiver();
|
|
case SuperInstance:
|
|
case SuperClass:
|
|
return getSuperType();
|
|
}
|
|
|
|
llvm_unreachable("unexpected receiver kind");
|
|
}
|
|
|
|
ObjCInterfaceDecl *ObjCMessageExpr::getReceiverInterface() const {
|
|
QualType T = getReceiverType();
|
|
|
|
if (const ObjCObjectPointerType *Ptr = T->getAs<ObjCObjectPointerType>())
|
|
return Ptr->getInterfaceDecl();
|
|
|
|
if (const ObjCObjectType *Ty = T->getAs<ObjCObjectType>())
|
|
return Ty->getInterface();
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
Stmt::child_range ObjCMessageExpr::children() {
|
|
Stmt **begin;
|
|
if (getReceiverKind() == Instance)
|
|
begin = reinterpret_cast<Stmt **>(getTrailingObjects<void *>());
|
|
else
|
|
begin = reinterpret_cast<Stmt **>(getArgs());
|
|
return child_range(begin,
|
|
reinterpret_cast<Stmt **>(getArgs() + getNumArgs()));
|
|
}
|
|
|
|
Stmt::const_child_range ObjCMessageExpr::children() const {
|
|
return const_cast<ObjCMessageExpr *>(this)->children();
|
|
}
|
|
|
|
StringRef ObjCBridgedCastExpr::getBridgeKindName() const {
|
|
switch (getBridgeKind()) {
|
|
case OBC_Bridge:
|
|
return "__bridge";
|
|
case OBC_BridgeTransfer:
|
|
return "__bridge_transfer";
|
|
case OBC_BridgeRetained:
|
|
return "__bridge_retained";
|
|
}
|
|
|
|
llvm_unreachable("Invalid BridgeKind!");
|
|
}
|