Skip to content

Commit 8399971

Browse files
author
Gabor Horvath
committed
[cxx-interop] Support lifetime_capture_by in ClangImporter
Import it as lifetime dependencies. rdar://137671377
1 parent 2a8ad3a commit 8399971

File tree

2 files changed

+103
-25
lines changed

2 files changed

+103
-25
lines changed

‎lib/ClangImporter/ImportDecl.cpp

Lines changed: 70 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -25,11 +25,11 @@
2525
#include "swift/AST/ConformanceLookup.h"
2626
#include "swift/AST/Decl.h"
2727
#include "swift/AST/DiagnosticsClangImporter.h"
28-
#include "swift/ClangImporter/ClangImporter.h"
2928
#include "swift/AST/ExistentialLayout.h"
3029
#include "swift/AST/Expr.h"
3130
#include "swift/AST/GenericEnvironment.h"
3231
#include "swift/AST/GenericSignature.h"
32+
#include "swift/AST/LifetimeDependence.h"
3333
#include "swift/AST/Module.h"
3434
#include "swift/AST/NameLookup.h"
3535
#include "swift/AST/NameLookupRequests.h"
@@ -49,6 +49,7 @@
4949
#include "swift/Basic/StringExtras.h"
5050
#include "swift/Basic/Version.h"
5151
#include "swift/ClangImporter/CXXMethodBridging.h"
52+
#include "swift/ClangImporter/ClangImporter.h"
5253
#include "swift/ClangImporter/ClangImporterRequests.h"
5354
#include "swift/ClangImporter/ClangModule.h"
5455
#include "swift/Parse/Lexer.h"
@@ -65,6 +66,7 @@
6566
#include "clang/Sema/Lookup.h"
6667

6768
#include "llvm/ADT/STLExtras.h"
69+
#include "llvm/ADT/SmallBitVector.h"
6870
#include "llvm/ADT/SmallString.h"
6971
#include "llvm/ADT/Statistic.h"
7072
#include "llvm/ADT/StringExtras.h"
@@ -3545,19 +3547,20 @@ namespace {
35453547
return true;
35463548
}
35473549

3548-
static bool
3549-
implicitObjectParamIsLifetimeBound(const clang::FunctionDecl *FD) {
3550+
template <typename T>
3551+
static const T *
3552+
getImplicitObjectParamAnnotation(const clang::FunctionDecl *FD) {
35503553
const clang::TypeSourceInfo *TSI = FD->getTypeSourceInfo();
35513554
if (!TSI)
3552-
return false;
3555+
return nullptr;
35533556
clang::AttributedTypeLoc ATL;
35543557
for (clang::TypeLoc TL = TSI->getTypeLoc();
35553558
(ATL = TL.getAsAdjusted<clang::AttributedTypeLoc>());
35563559
TL = ATL.getModifiedLoc()) {
3557-
if (ATL.getAttrAs<clang::LifetimeBoundAttr>())
3558-
return true;
3560+
if (auto attr = ATL.getAttrAs<T>())
3561+
return attr;
35593562
}
3560-
return false;
3563+
return nullptr;
35613564
}
35623565

35633566
Decl *importFunctionDecl(
@@ -3869,7 +3872,8 @@ namespace {
38693872
if (selfIsInOut)
38703873
func->setSelfAccessKind(SelfAccessKind::Mutating);
38713874
else {
3872-
if (implicitObjectParamIsLifetimeBound(decl))
3875+
if (getImplicitObjectParamAnnotation<clang::LifetimeBoundAttr>(
3876+
decl))
38733877
func->setSelfAccessKind(SelfAccessKind::Borrowing);
38743878
else
38753879
func->setSelfAccessKind(SelfAccessKind::NonMutating);
@@ -3961,26 +3965,68 @@ namespace {
39613965
auto swiftParams = result->getParameters();
39623966
bool hasSelf =
39633967
result->hasImplicitSelfDecl() && !isa<ConstructorDecl>(result);
3964-
SmallBitVector inheritLifetimeParamIndicesForReturn(swiftParams->size() +
3965-
hasSelf);
3966-
SmallBitVector scopedLifetimeParamIndicesForReturn(swiftParams->size() +
3967-
hasSelf);
3968-
for (auto [idx, param] : llvm::enumerate(decl->parameters())) {
3969-
if (param->hasAttr<clang::LifetimeBoundAttr>()) {
3970-
warnForEscapableReturnType();
3971-
if (swiftParams->get(idx)->getInterfaceType()->isEscapable())
3972-
scopedLifetimeParamIndicesForReturn[idx] = true;
3973-
else
3974-
inheritLifetimeParamIndicesForReturn[idx] = true;
3975-
}
3976-
}
3977-
if (implicitObjectParamIsLifetimeBound(decl)) {
3968+
const auto dependencyVecSize = swiftParams->size() + hasSelf;
3969+
SmallBitVector inheritLifetimeParamIndicesForReturn(dependencyVecSize);
3970+
SmallBitVector scopedLifetimeParamIndicesForReturn(dependencyVecSize);
3971+
std::map<unsigned, SmallBitVector> inheritedArgDependences;
3972+
auto processLifetimeBound = [&](unsigned idx, Type ty) {
39783973
warnForEscapableReturnType();
3979-
auto idx = result->getSelfIndex();
3980-
if (result->getImplicitSelfDecl()->getInterfaceType()->isEscapable())
3974+
if (ty->isEscapable())
39813975
scopedLifetimeParamIndicesForReturn[idx] = true;
39823976
else
39833977
inheritLifetimeParamIndicesForReturn[idx] = true;
3978+
};
3979+
auto processLifetimeCaptureBy =
3980+
[&](const clang::LifetimeCaptureByAttr *attr, unsigned idx, Type ty) {
3981+
// FIXME: support scoped lifetimes. This is not straightforward as
3982+
// const T& is imported as taking a value
3983+
// and we assume the address of T would not escape. An
3984+
// annotation in this case contradicts our assumptions. We
3985+
// should diagnose that, and support this for the non-const
3986+
// case.
3987+
if (ty->isEscapable())
3988+
return;
3989+
for (auto param : attr->params()) {
3990+
// FIXME: Swift assumes no escaping to globals. We should diagnose
3991+
// this.
3992+
if (param == clang::LifetimeCaptureByAttr::GLOBAL ||
3993+
param == clang::LifetimeCaptureByAttr::UNKNOWN ||
3994+
param == clang::LifetimeCaptureByAttr::INVALID)
3995+
continue;
3996+
3997+
if (isa<clang::CXXMethodDecl>(decl) &&
3998+
param == clang::LifetimeCaptureByAttr::THIS) {
3999+
auto [it, inserted] = inheritedArgDependences.try_emplace(
4000+
result->getSelfIndex(), SmallBitVector(dependencyVecSize));
4001+
it->second[idx] = true;
4002+
} else {
4003+
auto [it, inserted] = inheritedArgDependences.try_emplace(
4004+
param - isa<clang::CXXMethodDecl>(decl),
4005+
SmallBitVector(dependencyVecSize));
4006+
it->second[idx] = true;
4007+
}
4008+
}
4009+
};
4010+
for (auto [idx, param] : llvm::enumerate(decl->parameters())) {
4011+
if (param->hasAttr<clang::LifetimeBoundAttr>())
4012+
processLifetimeBound(idx, swiftParams->get(idx)->getInterfaceType());
4013+
if (const auto *attr = param->getAttr<clang::LifetimeCaptureByAttr>())
4014+
processLifetimeCaptureBy(attr, idx,
4015+
swiftParams->get(idx)->getInterfaceType());
4016+
}
4017+
if (getImplicitObjectParamAnnotation<clang::LifetimeBoundAttr>(decl))
4018+
processLifetimeBound(result->getSelfIndex(),
4019+
result->getImplicitSelfDecl()->getInterfaceType());
4020+
if (auto attr =
4021+
getImplicitObjectParamAnnotation<clang::LifetimeCaptureByAttr>(
4022+
decl))
4023+
processLifetimeCaptureBy(
4024+
attr, result->getSelfIndex(),
4025+
result->getImplicitSelfDecl()->getInterfaceType());
4026+
4027+
for (auto& [idx, inheritedDepVec]: inheritedArgDependences) {
4028+
lifetimeDependencies.push_back(LifetimeDependenceInfo(inheritedDepVec.any() ? IndexSubset::get(Impl.SwiftContext,
4029+
inheritedDepVec): nullptr, nullptr, idx, /*isImmortal=*/false));
39844030
}
39854031

39864032
if (inheritLifetimeParamIndicesForReturn.any() ||

‎test/Interop/Cxx/class/nonescapable-lifetimebound.swift

Lines changed: 33 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -82,6 +82,29 @@ View returnsImmortal() SWIFT_RETURNS_INDEPENDENT_VALUE {
8282
return View();
8383
}
8484

85+
void copyView(View view1 [[clang::lifetime_capture_by(view2)]], View &view2) {
86+
view2 = view1;
87+
}
88+
89+
struct SWIFT_NONESCAPABLE CaptureView {
90+
CaptureView() : view(nullptr) {}
91+
CaptureView(View p [[clang::lifetimebound]]) : view(p) {}
92+
93+
void captureView(View v [[clang::lifetime_capture_by(this)]]) {
94+
view = v;
95+
}
96+
97+
void handOut(View &v) const [[clang::lifetime_capture_by(v)]] {
98+
v = view;
99+
}
100+
101+
View view;
102+
};
103+
104+
CaptureView getCaptureView(const Owner& owner [[clang::lifetimebound]]) {
105+
return CaptureView(View{&owner.data});
106+
}
107+
85108
// CHECK: sil [clang makeOwner] {{.*}}: $@convention(c) () -> Owner
86109
// CHECK: sil [clang getView] {{.*}} : $@convention(c) (@in_guaranteed Owner) -> @lifetime(borrow 0) @autoreleased View
87110
// CHECK: sil [clang getViewFromFirst] {{.*}} : $@convention(c) (@in_guaranteed Owner, @in_guaranteed Owner) -> @lifetime(borrow 0) @autoreleased View
@@ -92,6 +115,10 @@ View returnsImmortal() SWIFT_RETURNS_INDEPENDENT_VALUE {
92115
// CHECK: sil [clang View.init] {{.*}} : $@convention(c) () -> @lifetime(immortal) @out View
93116
// CHECK: sil [clang OtherView.init] {{.*}} : $@convention(c) (@guaranteed View) -> @lifetime(copy 0) @out OtherView
94117
// CHECK: sil [clang returnsImmortal] {{.*}} : $@convention(c) () -> @lifetime(immortal) @autoreleased View
118+
// CHECK: sil [clang copyView] {{.*}} : $@convention(c) (View, @lifetime(copy 0) @inout View) -> ()
119+
// CHECK: sil [clang getCaptureView] {{.*}} : $@convention(c) (@in_guaranteed Owner) -> @lifetime(borrow 0) @autoreleased CaptureView
120+
// CHECK: sil [clang CaptureView.captureView] {{.*}} : $@convention(cxx_method) (View, @lifetime(copy 0) @inout CaptureView) -> ()
121+
// CHECK: sil [clang CaptureView.handOut] {{.*}} : $@convention(cxx_method) (@lifetime(copy 1) @inout View, @in_guaranteed CaptureView) -> ()
95122

96123
// CHECK-NO-LIFETIMES: nonescapable.h:36:6: error: returning ~Escapable type requires '-enable-experimental-feature LifetimeDependence'
97124
// CHECK-NO-LIFETIMES: nonescapable.h:40:6: error: returning ~Escapable type requires '-enable-experimental-feature LifetimeDependence'
@@ -104,6 +131,7 @@ View returnsImmortal() SWIFT_RETURNS_INDEPENDENT_VALUE {
104131
// CHECK-NO-LIFETIMES: nonescapable.h:13:5: error: returning ~Escapable type requires '-enable-experimental-feature LifetimeDependence'
105132
// CHECK-NO-LIFETIMES: nonescapable.h:14:5: error: returning ~Escapable type requires '-enable-experimental-feature LifetimeDependence'
106133
// CHECK-NO-LIFETIMES: nonescapable.h:68:6: error: returning ~Escapable type requires '-enable-experimental-feature LifetimeDependence'
134+
// CHECK-NO-LIFETIMES: nonescapable.h:91:13: error: returning ~Escapable type requires '-enable-experimental-feature LifetimeDependence'
107135
// CHECK-NO-LIFETIMES-NOT: error
108136

109137
//--- test.swift
@@ -113,7 +141,7 @@ import Test
113141
public func test() {
114142
let o = makeOwner()
115143
let o2 = makeOwner()
116-
let v1 = getView(o)
144+
var v1 = getView(o)
117145
let v2 = getViewFromFirst(o, o2)
118146
let _ = getViewFromEither(o, o2)
119147
let _ = o.handOutView()
@@ -122,4 +150,8 @@ public func test() {
122150
let defaultView = View()
123151
let _ = OtherView(defaultView)
124152
let _ = returnsImmortal()
153+
copyView(v2, &v1)
154+
var cv = getCaptureView(o)
155+
cv.captureView(v1)
156+
cv.handOut(&v1)
125157
}

0 commit comments

Comments
 (0)