Skip to content

Commit 027a178

Browse files
j-huij-hui
authored
Merge pull request #81395 from j-hui/dont-swiftify-invalid-expr
[Swiftify] Do not swiftify non-Swift-like counted_by exprs
2 parents efd70b1 + 203acbb commit 027a178

File tree

4 files changed

+130
-4
lines changed

4 files changed

+130
-4
lines changed

‎lib/ClangImporter/ImportDecl.cpp

Lines changed: 51 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,9 @@
6060
#include "clang/AST/Decl.h"
6161
#include "clang/AST/DeclCXX.h"
6262
#include "clang/AST/DeclObjCCommon.h"
63+
#include "clang/AST/Expr.h"
6364
#include "clang/AST/PrettyPrinter.h"
65+
#include "clang/AST/StmtVisitor.h"
6466
#include "clang/AST/RecordLayout.h"
6567
#include "clang/AST/Type.h"
6668
#include "clang/Basic/Specifiers.h"
@@ -9139,6 +9141,51 @@ class SwiftifyInfoPrinter {
91399141
};
91409142
} // namespace
91419143

9144+
namespace {
9145+
struct CountedByExpressionValidator
9146+
: clang::ConstStmtVisitor<CountedByExpressionValidator, bool> {
9147+
bool VisitDeclRefExpr(const clang::DeclRefExpr *e) { return true; }
9148+
bool VisitIntegerLiteral(const clang::IntegerLiteral *) { return true; }
9149+
bool VisitImplicitCastExpr(const clang::ImplicitCastExpr *c) {
9150+
return this->Visit(c->getSubExpr());
9151+
}
9152+
bool VisitParenExpr(const clang::ParenExpr *p) {
9153+
return this->Visit(p->getSubExpr());
9154+
}
9155+
9156+
#define SUPPORTED_UNOP(UNOP) \
9157+
bool VisitUnary ## UNOP(const clang::UnaryOperator *unop) { \
9158+
return this->Visit(unop->getSubExpr()); \
9159+
}
9160+
SUPPORTED_UNOP(Plus)
9161+
SUPPORTED_UNOP(Minus)
9162+
SUPPORTED_UNOP(Not)
9163+
#undef SUPPORTED_UNOP
9164+
9165+
#define SUPPORTED_BINOP(BINOP) \
9166+
bool VisitBin ## BINOP(const clang::BinaryOperator *binop) { \
9167+
return this->Visit(binop->getLHS()) && this->Visit(binop->getRHS()); \
9168+
}
9169+
SUPPORTED_BINOP(Add)
9170+
SUPPORTED_BINOP(Sub)
9171+
SUPPORTED_BINOP(Div)
9172+
SUPPORTED_BINOP(Mul)
9173+
SUPPORTED_BINOP(Rem)
9174+
SUPPORTED_BINOP(Shl)
9175+
SUPPORTED_BINOP(Shr)
9176+
SUPPORTED_BINOP(And)
9177+
SUPPORTED_BINOP(Xor)
9178+
SUPPORTED_BINOP(Or)
9179+
#undef SUPPORTED_BINOP
9180+
9181+
bool VisitStmt(const clang::Stmt *) { return false; }
9182+
};
9183+
} // namespace
9184+
9185+
static bool SwiftifiableCAT(const clang::CountAttributedType *CAT) {
9186+
return CAT && CountedByExpressionValidator().Visit(CAT->getCountExpr());
9187+
}
9188+
91429189
void ClangImporter::Implementation::swiftify(FuncDecl *MappedDecl) {
91439190
if (!SwiftContext.LangOpts.hasFeature(Feature::SafeInteropWrappers))
91449191
return;
@@ -9174,8 +9221,8 @@ void ClangImporter::Implementation::swiftify(FuncDecl *MappedDecl) {
91749221
SwiftifyInfoPrinter printer(getClangASTContext(), SwiftContext, out);
91759222
bool returnIsStdSpan = registerStdSpanTypeMapping(
91769223
MappedDecl->getResultInterfaceType(), ClangDecl->getReturnType());
9177-
if (auto CAT =
9178-
ClangDecl->getReturnType()->getAs<clang::CountAttributedType>()) {
9224+
auto *CAT = ClangDecl->getReturnType()->getAs<clang::CountAttributedType>();
9225+
if (SwiftifiableCAT(CAT)) {
91799226
printer.printCountedBy(CAT, SwiftifyInfoPrinter::RETURN_VALUE_INDEX);
91809227
attachMacro = true;
91819228
}
@@ -9190,7 +9237,8 @@ void ClangImporter::Implementation::swiftify(FuncDecl *MappedDecl) {
91909237
auto clangParamTy = clangParam->getType();
91919238
auto swiftParam = MappedDecl->getParameters()->get(index);
91929239
bool paramHasBoundsInfo = false;
9193-
if (auto CAT = clangParamTy->getAs<clang::CountAttributedType>()) {
9240+
auto *CAT = clangParamTy->getAs<clang::CountAttributedType>();
9241+
if (SwiftifiableCAT(CAT)) {
91949242
printer.printCountedBy(CAT, index);
91959243
attachMacro = paramHasBoundsInfo = true;
91969244
}

‎test/Interop/C/swiftify-import/Inputs/counted-by.h

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@
44

55
void simple(int len, int * __counted_by(len) p);
66

7+
void simpleFlipped(int * __counted_by(len) p, int len);
8+
79
void swiftAttr(int len, int *p) __attribute__((
810
swift_attr("@_SwiftifyImport(.countedBy(pointer: .param(2), count: \"len\"))")));
911

@@ -18,3 +20,35 @@ void nonnull(int len, int * __counted_by(len) _Nonnull p);
1820
void nullable(int len, int * __counted_by(len) _Nullable p);
1921

2022
int * __counted_by(len) returnPointer(int len);
23+
24+
void offByOne(int len, int * __counted_by(len + 1) p);
25+
26+
void offBySome(int len, int offset, int * __counted_by(len + (1 + offset)) p);
27+
28+
void scalar(int m, int n, int * __counted_by(m * n) p);
29+
30+
void bitwise(int m, int n, int o, int * __counted_by(m & n | ~o) p);
31+
32+
void bitshift(int m, int n, int o, int * __counted_by(m << (n >> o)) p);
33+
34+
void constInt(int * __counted_by(42 * 10) p);
35+
36+
void constFloatCastedToInt(int * __counted_by((int) (4.2 / 12)) p);
37+
38+
void sizeofType(int * __counted_by(sizeof(int *)) p);
39+
40+
void sizeofParam(int * __counted_by(sizeof(p)) p);
41+
42+
void derefLen(int * len, int * __counted_by(*len) p);
43+
44+
void lNot(int len, int * __counted_by(!len) p);
45+
46+
void lAnd(int len, int * __counted_by(len && len) p);
47+
48+
void lOr(int len, int * __counted_by(len || len) p);
49+
50+
void floatCastToInt(float meters, int * __counted_by((int) meters) p);
51+
52+
void pointerCastToInt(int *square, int * __counted_by((int) square) p);
53+
54+
void nanAsInt(int * __counted_by((int) (0 / 0)) p);

‎test/Interop/C/swiftify-import/counted-by-no-swiftify.swift

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
// RUN: %target-swift-ide-test -print-module -module-to-print=CountedByClang -plugin-path %swift-plugin-dir -I %S/Inputs -source-filename=x -enable-experimental-feature SafeInteropWrappers | %FileCheck %s
2+
3+
// REQUIRES: swift_feature_SafeInteropWrappers
4+
5+
// These functions use __counted_by annotations that are not syntactically valid
6+
// in Swift, so they should not be Swiftified
7+
8+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} derefLen
9+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} lNot
10+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} lAnd
11+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} lOr
12+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} floatCastToInt
13+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} pointerCastToInt
14+
// CHECK-NOT: @_alwaysEmitIntoClient {{.*}} nanAsInt

‎test/Interop/C/swiftify-import/counted-by.swift

Lines changed: 31 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,20 +10,35 @@
1010

1111
import CountedByClang
1212

13-
// CHECK: @_alwaysEmitIntoClient public func complexExpr(_ len: Int{{.*}}, _ offset: Int{{.*}}, _ p: UnsafeMutableBufferPointer<Int{{.*}}>)
13+
// CHECK: @_alwaysEmitIntoClient public func bitshift(_ m: Int32, _ n: Int32, _ o: Int32, _ p: UnsafeMutableBufferPointer<Int32>)
14+
// CHECK-NEXT: @_alwaysEmitIntoClient public func bitwise(_ m: Int32, _ n: Int32, _ o: Int32, _ p: UnsafeMutableBufferPointer<Int32>)
15+
// CHECK-NEXT: @_alwaysEmitIntoClient public func complexExpr(_ len: Int{{.*}}, _ offset: Int{{.*}}, _ p: UnsafeMutableBufferPointer<Int{{.*}}>)
16+
// CHECK-NEXT: @_alwaysEmitIntoClient public func constFloatCastedToInt(_ p: UnsafeMutableBufferPointer<Int32>)
17+
// CHECK-NEXT: @_alwaysEmitIntoClient public func constInt(_ p: UnsafeMutableBufferPointer<Int32>)
1418
// CHECK-NEXT: @_alwaysEmitIntoClient public func nonnull(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
1519
// CHECK-NEXT: @_alwaysEmitIntoClient public func nullUnspecified(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
1620
// CHECK-NEXT: @_alwaysEmitIntoClient public func nullable(_ p: UnsafeMutableBufferPointer<Int{{.*}}>?)
21+
// CHECK-NEXT: @_alwaysEmitIntoClient public func offByOne(_ len: Int32, _ p: UnsafeMutableBufferPointer<Int32>)
22+
// CHECK-NEXT: @_alwaysEmitIntoClient public func offBySome(_ len: Int32, _ offset: Int32, _ p: UnsafeMutableBufferPointer<Int32>)
1723
// CHECK-NEXT: @_alwaysEmitIntoClient @_disfavoredOverload public func returnPointer(_ len: Int{{.*}}) -> UnsafeMutableBufferPointer<Int{{.*}}>
24+
// CHECK-NEXT: @_alwaysEmitIntoClient public func scalar(_ m: Int32, _ n: Int32, _ p: UnsafeMutableBufferPointer<Int32>)
1825
// CHECK-NEXT: @_alwaysEmitIntoClient public func shared(_ len: Int{{.*}}, _ p1: UnsafeMutableBufferPointer<Int{{.*}}>, _ p2: UnsafeMutableBufferPointer<Int{{.*}}>)
1926
// CHECK-NEXT: @_alwaysEmitIntoClient public func simple(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
27+
// CHECK-NEXT: @_alwaysEmitIntoClient public func simpleFlipped(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
28+
// CHECK-NEXT: @_alwaysEmitIntoClient public func sizeofParam(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
29+
// CHECK-NEXT: @_alwaysEmitIntoClient public func sizeofType(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
2030
// CHECK-NEXT: @_alwaysEmitIntoClient public func swiftAttr(_ p: UnsafeMutableBufferPointer<Int{{.*}}>)
2131

2232
@inlinable
2333
public func callComplexExpr(_ p: UnsafeMutableBufferPointer<CInt>) {
2434
complexExpr(CInt(p.count), 1, p)
2535
}
2636

37+
@inlinable
38+
public func callConstInt(_ p: UnsafeMutableBufferPointer<CInt>) {
39+
constInt(p)
40+
}
41+
2742
@inlinable
2843
public func callNonnull(_ p: UnsafeMutableBufferPointer<CInt>) {
2944
nonnull(p)
@@ -39,12 +54,22 @@ public func callNullable(_ p: UnsafeMutableBufferPointer<CInt>?) {
3954
nullable(p)
4055
}
4156

57+
@inlinable
58+
public func callOffByOne(_ p: UnsafeMutableBufferPointer<CInt>) {
59+
offByOne(0, p)
60+
}
61+
4262
@inlinable
4363
public func callReturnPointer() {
4464
let a: UnsafeMutableBufferPointer<CInt>? = returnPointer(4) // call wrapper
4565
let b: UnsafeMutablePointer<CInt>? = returnPointer(4) // call unsafe interop
4666
}
4767

68+
@inlinable
69+
public func callScalar(_ p: UnsafeMutableBufferPointer<CInt>) {
70+
scalar(4, 2, p)
71+
}
72+
4873
@inlinable
4974
public func callShared(_ p: UnsafeMutableBufferPointer<CInt>, _ p2: UnsafeMutableBufferPointer<CInt>) {
5075
shared(CInt(p.count), p, p2)
@@ -55,6 +80,11 @@ public func callSimple(_ p: UnsafeMutableBufferPointer<CInt>) {
5580
simple(p)
5681
}
5782

83+
@inlinable
84+
public func callSimpleFlipped(_ p: UnsafeMutableBufferPointer<CInt>) {
85+
simpleFlipped(p)
86+
}
87+
5888
@inlinable
5989
public func callSwiftAttr(_ p: UnsafeMutableBufferPointer<CInt>) {
6090
swiftAttr(p)

0 commit comments

Comments
 (0)