Files
swift-mirror/stdlib/public/core/SwiftifyImport.swift
Henrik G. Olsson 48f81c82dd [Swiftify] Add _SwiftifyImportProtocol for safe overloads for protocols
The existing _SwiftifyImport macro is a peer macro, limiting it to only
emitting function wrappers in the same scope as the original function.
Protocols cannot contain function implementations, so these need to be
placed in a separate protocol extension instead. _SwiftifyImportProtocol
is an extension macro rather than a peer macro, to enable this
functionality.

Rather than operating on a single function, like _SwiftifyImport,
_SwiftifyImportProtocol takes information about multiple methods and
creates a single protocol extension with all wrappers in a one-shot
operation.

rdar://144335990
2025-10-21 21:19:54 -07:00

132 lines
6.4 KiB
Swift

public enum _SwiftifyExpr {
case param(_ index: Int)
case `return`
case `self`
}
public enum _DependenceType {
case borrow
case copy
}
/// Different ways to annotate pointer parameters using the `@_SwiftifyImport` macro.
/// All indices into parameter lists start at 1. Indices __must__ be integer literals, and strings
/// __must__ be string literals, because their contents are parsed by the `@_SwiftifyImport` macro.
/// Only 1 instance of `countedBy`, `sizedBy` or `endedBy` can refer to each pointer index, however
/// `nonescaping` is orthogonal to the rest and can (and should) overlap with other annotations.
///
/// This is not marked @available, because _SwiftifyImport is available for any target. Instances
/// of _SwiftifyInfo should ONLY be passed as arguments directly to _SwiftifyImport, so they should
/// not affect linkage since there are never any instances at runtime.
public enum _SwiftifyInfo {
/// Corresponds to the C `__counted_by(count)` attribute.
/// Parameter pointer: index of pointer in function parameter list. Must be of type
/// `Unsafe[Mutable]Pointer<T>[?]`, i.e. not an `UnsafeRawPointer`.
/// Parameter count: string containing valid Swift syntax containing the number of elements in
/// the buffer.
case countedBy(pointer: _SwiftifyExpr, count: String)
/// Corresponds to the C `__sized_by(size)` attribute.
/// Parameter pointer: index of pointer in function parameter list. Must be of type
/// `Unsafe[Mutable]RawPointer[?]`, i.e. not an `UnsafePointer<T>`.
/// Parameter count: string containing valid Swift syntax containing the size of the buffer,
/// in bytes.
case sizedBy(pointer: _SwiftifyExpr, size: String)
/// Corresponds to the C `__ended_by(end)` attribute.
/// Parameter start: index of pointer in function parameter list.
/// Parameter end: index of pointer in function parameter list, pointing one past the end of
/// the same buffer as `start`.
case endedBy(start: _SwiftifyExpr, end: Int)
/// Corresponds to the C `noescape` attribute. Allows generated wrapper to use `Span`-types
/// instead of `UnsafeBuffer`-types, because it is known that the function doesn't capture the
/// object past the lifetime of the function.
/// Parameter pointer: index of pointer in function parameter list.
case nonescaping(pointer: _SwiftifyExpr)
/// Can express lifetime dependencies between inputs and outputs of a function.
/// 'dependsOn' is the input on which the output 'pointer' depends.
case lifetimeDependence(dependsOn: _SwiftifyExpr, pointer: _SwiftifyExpr, type: _DependenceType)
}
/// Generates a safe wrapper for function with Unsafe[Mutable][Raw]Pointer[?] or std::span arguments.
/// Intended to be automatically attached to function declarations imported by ClangImporter.
/// The wrapper function will replace Unsafe[Mutable][Raw]Pointer[?] parameters with
/// [Mutable][Raw]Span[?] or Unsafe[Mutable][Raw]BufferPointer[?] if they have bounds information
/// attached. Where possible "count" parameters will be elided from the wrapper signature, instead
/// fetching the count from the buffer pointer. In these cases the bounds check is also skipped.
/// It will replace some std::span arguments with Swift's Span type when sufficient information is
/// available.
///
/// Currently not supported: nested pointers, pointee "count" parameters, endedBy.
///
/// Parameter paramInfo: information about how the function uses the pointer passed to it. The
/// safety of the generated wrapper function depends on this info being extensive and accurate.
#if hasFeature(Macros)
@attached(peer, names: overloaded)
public macro _SwiftifyImport(_ paramInfo: _SwiftifyInfo...,
spanAvailability: String? = nil,
typeMappings: [String: String] = [:]) =
#externalMacro(module: "SwiftMacros", type: "SwiftifyImportMacro")
#endif
/// Allows annotating pointer parameters in a protocol method using the `@_SwiftifyImportProtocol` macro.
///
/// This is not marked @available, because _SwiftifyImportProtocolMethod is available for any target.
/// Instances of _SwiftifyProtocolMethodInfo should ONLY be passed as arguments directly to
/// _SwiftifyImportProtocolMethod, so they should not affect linkage since there are never any instances
/// at runtime.
public enum _SwiftifyProtocolMethodInfo {
case method(signature: String, paramInfo: [_SwiftifyInfo])
}
/// Like _SwiftifyImport, but since protocols cannot contain function implementations they need to
/// be placed in a separate extension instead. Unlike _SwiftifyImport, which applies to a single
/// function, this macro supports feeding info about multiple methods and generating safe overloads
/// for all of them at once.
#if hasFeature(Macros)
@attached(extension, names: arbitrary)
public macro _SwiftifyImportProtocol(
_ methodInfo: _SwiftifyProtocolMethodInfo...,
spanAvailability: String? = nil,
typeMappings: [String: String] = [:]
) =
#externalMacro(module: "SwiftMacros", type: "SwiftifyImportProtocolMacro")
#endif
/// Unsafely discard any lifetime dependency on the `dependent` argument. Return
/// a value identical to `dependent` with a lifetime dependency on the caller's
/// borrow scope of the `source` argument.
///
/// This mimics the stdlib definition. It is public for use with import macros.
@unsafe
@_unsafeNonescapableResult
@_alwaysEmitIntoClient
@_transparent
@lifetime(borrow source)
public func _swiftifyOverrideLifetime<
T: ~Copyable & ~Escapable, U: ~Copyable & ~Escapable
>(
_ dependent: consuming T, borrowing source: borrowing U
) -> T {
// TODO: Remove @_unsafeNonescapableResult. Instead, the unsafe dependence
// should be expressed by a builtin that is hidden within the function body.
dependent
}
/// Unsafely discard any lifetime dependency on the `dependent` argument. Return
/// a value identical to `dependent` that inherits all lifetime dependencies from
/// the `source` argument.
///
/// This mimics the stdlib definition. It is public for use with import macros.
@unsafe
@_unsafeNonescapableResult
@_alwaysEmitIntoClient
@_transparent
@lifetime(copy source)
public func _swiftifyOverrideLifetime<
T: ~Copyable & ~Escapable, U: ~Copyable & ~Escapable
>(
_ dependent: consuming T, copying source: borrowing U
) -> T {
// TODO: Remove @_unsafeNonescapableResult. Instead, the unsafe dependence
// should be expressed by a builtin that is hidden within the function body.
dependent
}