1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
// RUN: %empty-directory(%t)
// RUN: %target-swift-frontend %S/consuming-parameter-in-cxx.swift -typecheck -module-name Init -clang-header-expose-decls=all-public -emit-clang-header-path %t/consuming.h
// RUN: %target-interop-build-clangxx -c %s -I %t -o %t/swift-consume-execution.o
// RUN: %target-interop-build-swift %S/consuming-parameter-in-cxx.swift -o %t/swift-consume-execution -Xlinker %t/swift-consume-execution.o -module-name Init -Xfrontend -entry-point-function-name -Xfrontend swiftMain
// RUN: %target-codesign %t/swift-consume-execution
// RUN: %target-run %t/swift-consume-execution | %FileCheck %s
// REQUIRES: executable_test
#include <assert.h>
#include <stdint.h>
#include <stdlib.h>
size_t allocCount = 0;
size_t totalAllocs = 0;
void * _Nonnull trackedAlloc(size_t size, size_t align) {
++allocCount;
++totalAllocs;
return malloc(size);
}
void trackedFree(void *_Nonnull p) {
--allocCount;
free(p);
}
#define SWIFT_CXX_INTEROPERABILITY_OVERRIDE_OPAQUE_STORAGE_alloc trackedAlloc
#define SWIFT_CXX_INTEROPERABILITY_OVERRIDE_OPAQUE_STORAGE_free trackedFree
#include "consuming.h"
extern "C" size_t swift_retainCount(void * _Nonnull obj);
size_t getRetainCount(const Init::AKlass & swiftClass) {
void *p = swift::_impl::_impl_RefCountedClass::getOpaquePointer(swiftClass);
return swift_retainCount(p);
}
int main() {
using namespace Init;
{
auto k = AKlass::init();
k.takeKlass();
assert(getRetainCount(k) == 1);
}
// CHECK: destroy AKlass
{
auto k = AKlass::init();
auto x = createSmallStructNonTrivial(k);
auto x2 = InitFromSmall::init(x);
assert(getRetainCount(k) == 2);
}
// CHECK-NEXT: destroy AKlass
{
auto k = AKlass::init();
auto x = createSmallStructNonTrivial(k);
auto c = TheGenericContainer<SmallStructNonTrivial>::init(x);
assert(getRetainCount(k) == 3);
c.takeGenericContainer();
assert(getRetainCount(k) == 3);
}
// CHECK-NEXT: destroy AKlass
// verify that all of the opaque buffers are freed.
assert(allocCount == 0);
assert(totalAllocs != 0);
return 0;
}
|
