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
|
// RUN: %target-swift-frontend \
// RUN: -emit-sil -verify \
// RUN: %s \
// RUN: -enable-experimental-feature BuiltinModule \
// RUN: -sil-verify-all
// REQUIRES: asserts
import Builtin
enum MyLittleLayout<T : ~Copyable> {
@_transparent
static var size: Int {
return Int(Builtin.sizeof(T.self))
}
@_transparent
static var stride: Int {
return Int(Builtin.strideof(T.self))
}
}
enum MyLittleResult<Success : ~Copyable, Failure : Error> : ~Copyable {
case success(Success)
case failure(Failure)
}
@usableFromInline
@_alwaysEmitIntoClient
@inline(__always)
func _rethrowsViaClosure<R : ~Copyable>(_ fn: () throws -> R) rethrows -> R {
return try fn()
}
func _withUnprotectedUnsafeTemporaryAllocation<T: ~Copyable, R: ~Copyable>(
of type: T.Type,
capacity: Int,
alignment: Int,
_ body: (Builtin.RawPointer) throws -> R
) rethrows -> R {
let result: MyLittleResult<R, Error>
#if $BuiltinUnprotectedStackAlloc
let stackAddress = Builtin.unprotectedStackAlloc(
capacity._builtinWordValue,
MyLittleLayout<T>.stride._builtinWordValue,
alignment._builtinWordValue
)
#else
let stackAddress = Builtin.stackAlloc(
capacity._builtinWordValue,
MyLittleLayout<T>.stride._builtinWordValue,
alignment._builtinWordValue
)
#endif
do {
result = .success(try body(stackAddress))
Builtin.stackDealloc(stackAddress)
} catch {
result = .failure(error)
Builtin.stackDealloc(stackAddress)
}
switch consume result {
case .success(let success): return success
case .failure(let error): return try _rethrowsViaClosure { throw error }
}
}
|
