| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651 |
- /*
- * Copyright 2019, gRPC Authors All rights reserved.
- *
- * Licensed under the Apache License, Version 2.0 (the "License");
- * you may not use this file except in compliance with the License.
- * You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
- import Foundation
- import Logging
- import NIOCore
- import NIOExtras
- import NIOHTTP1
- import NIOHTTP2
- import NIOPosix
- import NIOTransportServices
- #if canImport(NIOSSL)
- import NIOSSL
- #endif
- #if canImport(Network)
- import Network
- #endif
- /// Wrapper object to manage the lifecycle of a gRPC server.
- ///
- /// The pipeline is configured in three stages detailed below. Note: handlers marked with
- /// a '*' are responsible for handling errors.
- ///
- /// 1. Initial stage, prior to pipeline configuration.
- ///
- /// ┌─────────────────────────────────┐
- /// │ GRPCServerPipelineConfigurator* │
- /// └────▲───────────────────────┬────┘
- /// ByteBuffer│ │ByteBuffer
- /// ┌─┴───────────────────────▼─┐
- /// │ NIOSSLHandler │
- /// └─▲───────────────────────┬─┘
- /// ByteBuffer│ │ByteBuffer
- /// │ ▼
- ///
- /// The `NIOSSLHandler` is optional and depends on how the framework user has configured
- /// their server. The `GRPCServerPipelineConfigurator` detects which HTTP version is being used
- /// (via ALPN if TLS is used or by parsing the first bytes on the connection otherwise) and
- /// configures the pipeline accordingly.
- ///
- /// 2. HTTP version detected. "HTTP Handlers" depends on the HTTP version determined by
- /// `GRPCServerPipelineConfigurator`. In the case of HTTP/2:
- ///
- /// ┌─────────────────────────────────┐
- /// │ HTTP2StreamMultiplexer │
- /// └─▲─────────────────────────────┬─┘
- /// HTTP2Frame│ │HTTP2Frame
- /// ┌─┴─────────────────────────────▼─┐
- /// │ HTTP2Handler │
- /// └─▲─────────────────────────────┬─┘
- /// ByteBuffer│ │ByteBuffer
- /// ┌─┴─────────────────────────────▼─┐
- /// │ NIOSSLHandler │
- /// └─▲─────────────────────────────┬─┘
- /// ByteBuffer│ │ByteBuffer
- /// │ ▼
- ///
- /// The `HTTP2StreamMultiplexer` provides one `Channel` for each HTTP/2 stream (and thus each
- /// RPC).
- ///
- /// 3. The frames for each stream channel are routed by the `HTTP2ToRawGRPCServerCodec` handler to
- /// a handler containing the user-implemented logic provided by a `CallHandlerProvider`:
- ///
- /// ┌─────────────────────────────────┐
- /// │ BaseCallHandler* │
- /// └─▲─────────────────────────────┬─┘
- /// GRPCServerRequestPart│ │GRPCServerResponsePart
- /// ┌─┴─────────────────────────────▼─┐
- /// │ HTTP2ToRawGRPCServerCodec │
- /// └─▲─────────────────────────────┬─┘
- /// HTTP2Frame.FramePayload│ │HTTP2Frame.FramePayload
- /// │ ▼
- ///
- /// - Note: This class is thread safe. It's marked as `@unchecked Sendable` because the non-Sendable
- /// `errorDelegate` property is mutated, but it's done thread-safely, as it only happens inside the `EventLoop`.
- public final class Server: @unchecked Sendable {
- /// Makes and configures a `ServerBootstrap` using the provided configuration.
- public class func makeBootstrap(configuration: Configuration) -> ServerBootstrapProtocol {
- let bootstrap = PlatformSupport.makeServerBootstrap(group: configuration.eventLoopGroup)
- // Backlog is only available on `ServerBootstrap`.
- if bootstrap is ServerBootstrap {
- // Specify a backlog to avoid overloading the server.
- _ = bootstrap.serverChannelOption(ChannelOptions.backlog, value: 256)
- }
- #if canImport(NIOSSL)
- // Making a `NIOSSLContext` is expensive, we should only do it once per TLS configuration so
- // we'll do it now, before accepting connections. Unfortunately our API isn't throwing so we'll
- // only surface any error when initializing a child channel.
- //
- // 'nil' means we're not using TLS, or we're using the Network.framework TLS backend. If we're
- // using the Network.framework TLS backend we'll apply the settings just below.
- let sslContext: Result<NIOSSLContext, Error>?
- if let tlsConfiguration = configuration.tlsConfiguration {
- do {
- sslContext = try tlsConfiguration.makeNIOSSLContext().map { .success($0) }
- } catch {
- sslContext = .failure(error)
- }
- } else {
- // No TLS configuration, no SSL context.
- sslContext = nil
- }
- #endif // canImport(NIOSSL)
- #if canImport(Network)
- if let tlsConfiguration = configuration.tlsConfiguration {
- if #available(macOS 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, *),
- let transportServicesBootstrap = bootstrap as? NIOTSListenerBootstrap
- {
- _ = transportServicesBootstrap.tlsOptions(from: tlsConfiguration)
- }
- }
- if #available(macOS 10.14, iOS 12.0, watchOS 6.0, tvOS 12.0, *),
- let configurator = configuration.listenerNWParametersConfigurator,
- let transportServicesBootstrap = bootstrap as? NIOTSListenerBootstrap
- {
- _ = transportServicesBootstrap.configureNWParameters(configurator)
- }
- if #available(macOS 10.14, iOS 12.0, watchOS 6.0, tvOS 12.0, *),
- let configurator = configuration.childChannelNWParametersConfigurator,
- let transportServicesBootstrap = bootstrap as? NIOTSListenerBootstrap
- {
- _ = transportServicesBootstrap.configureChildNWParameters(configurator)
- }
- #endif // canImport(Network)
- return
- bootstrap
- // Enable `SO_REUSEADDR` to avoid "address already in use" error.
- .serverChannelOption(
- ChannelOptions.socket(SocketOptionLevel(SOL_SOCKET), SO_REUSEADDR),
- value: 1
- )
- // Set the handlers that are applied to the accepted Channels
- .childChannelInitializer { channel in
- var configuration = configuration
- configuration.logger[metadataKey: MetadataKey.connectionID] = "\(UUID().uuidString)"
- configuration.logger.addIPAddressMetadata(
- local: channel.localAddress,
- remote: channel.remoteAddress
- )
- do {
- let sync = channel.pipeline.syncOperations
- #if canImport(NIOSSL)
- if let sslContext = try sslContext?.get() {
- let sslHandler: NIOSSLServerHandler
- if let verify = configuration.tlsConfiguration?.nioSSLCustomVerificationCallback {
- sslHandler = NIOSSLServerHandler(
- context: sslContext,
- customVerificationCallback: verify
- )
- } else {
- sslHandler = NIOSSLServerHandler(context: sslContext)
- }
- try sync.addHandler(sslHandler)
- }
- #endif // canImport(NIOSSL)
- // Configures the pipeline based on whether the connection uses TLS or not.
- try sync.addHandler(GRPCServerPipelineConfigurator(configuration: configuration))
- // Work around the zero length write issue, if needed.
- let requiresZeroLengthWorkaround = PlatformSupport.requiresZeroLengthWriteWorkaround(
- group: configuration.eventLoopGroup,
- hasTLS: configuration.tlsConfiguration != nil
- )
- if requiresZeroLengthWorkaround,
- #available(macOS 10.14, iOS 12.0, tvOS 12.0, watchOS 6.0, *)
- {
- try sync.addHandler(NIOFilterEmptyWritesHandler())
- }
- } catch {
- return channel.eventLoop.makeFailedFuture(error)
- }
- // Run the debug initializer, if there is one.
- if let debugAcceptedChannelInitializer = configuration.debugChannelInitializer {
- return debugAcceptedChannelInitializer(channel)
- } else {
- return channel.eventLoop.makeSucceededVoidFuture()
- }
- }
- // Enable TCP_NODELAY and SO_REUSEADDR for the accepted Channels
- .childChannelOption(ChannelOptions.socket(IPPROTO_TCP, TCP_NODELAY), value: 1)
- .childChannelOption(
- ChannelOptions.socket(SocketOptionLevel(SOL_SOCKET), SO_REUSEADDR),
- value: 1
- )
- }
- /// Starts a server with the given configuration. See `Server.Configuration` for the options
- /// available to configure the server.
- public static func start(configuration: Configuration) -> EventLoopFuture<Server> {
- let quiescingHelper = ServerQuiescingHelper(group: configuration.eventLoopGroup)
- return self.makeBootstrap(configuration: configuration)
- .serverChannelInitializer { channel in
- channel.pipeline.addHandler(quiescingHelper.makeServerChannelHandler(channel: channel))
- }
- .bind(to: configuration.target)
- .map { channel in
- Server(
- channel: channel,
- quiescingHelper: quiescingHelper,
- errorDelegate: configuration.errorDelegate
- )
- }
- }
- public let channel: Channel
- private let quiescingHelper: ServerQuiescingHelper
- private var errorDelegate: ServerErrorDelegate?
- private init(
- channel: Channel,
- quiescingHelper: ServerQuiescingHelper,
- errorDelegate: ServerErrorDelegate?
- ) {
- self.channel = channel
- self.quiescingHelper = quiescingHelper
- // Maintain a strong reference to ensure it lives as long as the server.
- self.errorDelegate = errorDelegate
- // If we have an error delegate, add a server channel error handler as well. We don't need to wait for the handler to
- // be added.
- if let errorDelegate = errorDelegate {
- _ = channel.pipeline.addHandler(ServerChannelErrorHandler(errorDelegate: errorDelegate))
- }
- // nil out errorDelegate to avoid retain cycles.
- self.onClose.whenComplete { _ in
- self.errorDelegate = nil
- }
- }
- /// Fired when the server shuts down.
- public var onClose: EventLoopFuture<Void> {
- return self.channel.closeFuture
- }
- /// Initiates a graceful shutdown. Existing RPCs may run to completion, any new RPCs or
- /// connections will be rejected.
- public func initiateGracefulShutdown(promise: EventLoopPromise<Void>?) {
- self.quiescingHelper.initiateShutdown(promise: promise)
- }
- /// Initiates a graceful shutdown. Existing RPCs may run to completion, any new RPCs or
- /// connections will be rejected.
- public func initiateGracefulShutdown() -> EventLoopFuture<Void> {
- let promise = self.channel.eventLoop.makePromise(of: Void.self)
- self.initiateGracefulShutdown(promise: promise)
- return promise.futureResult
- }
- /// Shutdown the server immediately. Active RPCs and connections will be terminated.
- public func close(promise: EventLoopPromise<Void>?) {
- self.channel.close(mode: .all, promise: promise)
- }
- /// Shutdown the server immediately. Active RPCs and connections will be terminated.
- public func close() -> EventLoopFuture<Void> {
- return self.channel.close(mode: .all)
- }
- }
- public typealias BindTarget = ConnectionTarget
- extension Server {
- /// The configuration for a server.
- public struct Configuration {
- /// The target to bind to.
- public var target: BindTarget
- /// The event loop group to run the connection on.
- public var eventLoopGroup: EventLoopGroup
- /// Providers the server should use to handle gRPC requests.
- public var serviceProviders: [CallHandlerProvider] {
- get {
- return Array(self.serviceProvidersByName.values)
- }
- set {
- self
- .serviceProvidersByName = Dictionary(
- uniqueKeysWithValues:
- newValue
- .map { ($0.serviceName, $0) }
- )
- }
- }
- /// An error delegate which is called when errors are caught. Provided delegates **must not
- /// maintain a strong reference to this `Server`**. Doing so will cause a retain cycle.
- public var errorDelegate: ServerErrorDelegate?
- #if canImport(NIOSSL)
- /// TLS configuration for this connection. `nil` if TLS is not desired.
- @available(*, deprecated, renamed: "tlsConfiguration")
- public var tls: TLS? {
- get {
- return self.tlsConfiguration?.asDeprecatedServerConfiguration
- }
- set {
- self.tlsConfiguration = newValue.map { GRPCTLSConfiguration(transforming: $0) }
- }
- }
- #endif // canImport(NIOSSL)
- public var tlsConfiguration: GRPCTLSConfiguration?
- /// The connection keepalive configuration.
- public var connectionKeepalive = ServerConnectionKeepalive()
- /// The amount of time to wait before closing connections. The idle timeout will start only
- /// if there are no RPCs in progress and will be cancelled as soon as any RPCs start.
- public var connectionIdleTimeout: TimeAmount = .nanoseconds(.max)
- /// The compression configuration for requests and responses.
- ///
- /// If compression is enabled for the server it may be disabled for responses on any RPC by
- /// setting `compressionEnabled` to `false` on the context of the call.
- ///
- /// Compression may also be disabled at the message-level for streaming responses (i.e. server
- /// streaming and bidirectional streaming RPCs) by passing setting `compression` to `.disabled`
- /// in `sendResponse(_:compression)`.
- ///
- /// Defaults to ``ServerMessageEncoding/disabled``.
- public var messageEncoding: ServerMessageEncoding = .disabled
- /// The maximum size in bytes of a message which may be received from a client. Defaults to 4MB.
- public var maximumReceiveMessageLength: Int = 4 * 1024 * 1024 {
- willSet {
- precondition(newValue >= 0, "maximumReceiveMessageLength must be positive")
- }
- }
- /// The HTTP/2 flow control target window size. Defaults to 8MB. Values are clamped between
- /// 1 and 2^31-1 inclusive.
- public var httpTargetWindowSize = 8 * 1024 * 1024 {
- didSet {
- self.httpTargetWindowSize = self.httpTargetWindowSize.clamped(to: 1 ... Int(Int32.max))
- }
- }
- /// The HTTP/2 max number of concurrent streams. Defaults to 100. Must be non-negative.
- public var httpMaxConcurrentStreams: Int = 100 {
- willSet {
- precondition(newValue >= 0, "httpMaxConcurrentStreams must be non-negative")
- }
- }
- /// The HTTP/2 max frame size. Defaults to 16384. Value is clamped between 2^14 and 2^24-1
- /// octets inclusive (the minimum and maximum allowable values - HTTP/2 RFC 7540 4.2).
- public var httpMaxFrameSize: Int = 16384 {
- didSet {
- self.httpMaxFrameSize = self.httpMaxFrameSize.clamped(to: 16384 ... 16_777_215)
- }
- }
- /// The HTTP/2 max number of reset streams. Defaults to 32. Must be non-negative.
- public var httpMaxResetStreams: Int = 32 {
- willSet {
- precondition(newValue >= 0, "httpMaxResetStreams must be non-negative")
- }
- }
- /// The root server logger. Accepted connections will branch from this logger and RPCs on
- /// each connection will use a logger branched from the connections logger. This logger is made
- /// available to service providers via `context`. Defaults to a no-op logger.
- public var logger = Logger(label: "io.grpc", factory: { _ in SwiftLogNoOpLogHandler() })
- /// A channel initializer which will be run after gRPC has initialized each accepted channel.
- /// This may be used to add additional handlers to the pipeline and is intended for debugging.
- /// This is analogous to `NIO.ServerBootstrap.childChannelInitializer`.
- ///
- /// - Warning: The initializer closure may be invoked *multiple times*. More precisely: it will
- /// be invoked at most once per accepted connection.
- public var debugChannelInitializer: ((Channel) -> EventLoopFuture<Void>)?
- /// A calculated private cache of the service providers by name.
- ///
- /// This is how gRPC consumes the service providers internally. Caching this as stored data avoids
- /// the need to recalculate this dictionary each time we receive an rpc.
- internal var serviceProvidersByName: [Substring: CallHandlerProvider]
- #if canImport(Network)
- /// A closure allowing to customise the listener's `NWParameters` used when establishing a connection using `NIOTransportServices`.
- @available(macOS 10.14, iOS 12.0, watchOS 6.0, tvOS 12.0, *)
- public var listenerNWParametersConfigurator: (@Sendable (NWParameters) -> Void)? {
- get {
- self._listenerNWParametersConfigurator as! (@Sendable (NWParameters) -> Void)?
- }
- set {
- self._listenerNWParametersConfigurator = newValue
- }
- }
- private var _listenerNWParametersConfigurator: (any Sendable)?
- /// A closure allowing to customise the child channels' `NWParameters` used when establishing connections using `NIOTransportServices`.
- @available(macOS 10.14, iOS 12.0, watchOS 6.0, tvOS 12.0, *)
- public var childChannelNWParametersConfigurator: (@Sendable (NWParameters) -> Void)? {
- get {
- self._childChannelNWParametersConfigurator as! (@Sendable (NWParameters) -> Void)?
- }
- set {
- self._childChannelNWParametersConfigurator = newValue
- }
- }
- private var _childChannelNWParametersConfigurator: (any Sendable)?
- #endif
- /// CORS configuration for gRPC-Web support.
- public var webCORS = Configuration.CORS()
- #if canImport(NIOSSL)
- /// Create a `Configuration` with some pre-defined defaults.
- ///
- /// - Parameters:
- /// - target: The target to bind to.
- /// - eventLoopGroup: The event loop group to run the server on.
- /// - serviceProviders: An array of `CallHandlerProvider`s which the server should use
- /// to handle requests.
- /// - errorDelegate: The error delegate, defaulting to a logging delegate.
- /// - tls: TLS configuration, defaulting to `nil`.
- /// - connectionKeepalive: The keepalive configuration to use.
- /// - connectionIdleTimeout: The amount of time to wait before closing the connection, this is
- /// indefinite by default.
- /// - messageEncoding: Message compression configuration, defaulting to no compression.
- /// - httpTargetWindowSize: The HTTP/2 flow control target window size.
- /// - logger: A logger. Defaults to a no-op logger.
- /// - debugChannelInitializer: A channel initializer which will be called for each connection
- /// the server accepts after gRPC has initialized the channel. Defaults to `nil`.
- @available(*, deprecated, renamed: "default(target:eventLoopGroup:serviceProviders:)")
- public init(
- target: BindTarget,
- eventLoopGroup: EventLoopGroup,
- serviceProviders: [CallHandlerProvider],
- errorDelegate: ServerErrorDelegate? = nil,
- tls: TLS? = nil,
- connectionKeepalive: ServerConnectionKeepalive = ServerConnectionKeepalive(),
- connectionIdleTimeout: TimeAmount = .nanoseconds(.max),
- messageEncoding: ServerMessageEncoding = .disabled,
- httpTargetWindowSize: Int = 8 * 1024 * 1024,
- logger: Logger = Logger(label: "io.grpc", factory: { _ in SwiftLogNoOpLogHandler() }),
- debugChannelInitializer: ((Channel) -> EventLoopFuture<Void>)? = nil
- ) {
- self.target = target
- self.eventLoopGroup = eventLoopGroup
- self.serviceProvidersByName = Dictionary(
- uniqueKeysWithValues: serviceProviders.map { ($0.serviceName, $0) }
- )
- self.errorDelegate = errorDelegate
- self.tlsConfiguration = tls.map { GRPCTLSConfiguration(transforming: $0) }
- self.connectionKeepalive = connectionKeepalive
- self.connectionIdleTimeout = connectionIdleTimeout
- self.messageEncoding = messageEncoding
- self.httpTargetWindowSize = httpTargetWindowSize
- self.logger = logger
- self.debugChannelInitializer = debugChannelInitializer
- }
- #endif // canImport(NIOSSL)
- private init(
- eventLoopGroup: EventLoopGroup,
- target: BindTarget,
- serviceProviders: [CallHandlerProvider]
- ) {
- self.eventLoopGroup = eventLoopGroup
- self.target = target
- self.serviceProvidersByName = Dictionary(
- uniqueKeysWithValues: serviceProviders.map {
- ($0.serviceName, $0)
- }
- )
- }
- /// Make a new configuration using default values.
- ///
- /// - Parameters:
- /// - target: The target to bind to.
- /// - eventLoopGroup: The `EventLoopGroup` the server should run on.
- /// - serviceProviders: An array of `CallHandlerProvider`s which the server should use
- /// to handle requests.
- /// - Returns: A configuration with default values set.
- public static func `default`(
- target: BindTarget,
- eventLoopGroup: EventLoopGroup,
- serviceProviders: [CallHandlerProvider]
- ) -> Configuration {
- return .init(
- eventLoopGroup: eventLoopGroup,
- target: target,
- serviceProviders: serviceProviders
- )
- }
- }
- }
- extension Server.Configuration {
- public struct CORS: Hashable, Sendable {
- /// Determines which 'origin' header field values are permitted in a CORS request.
- public var allowedOrigins: AllowedOrigins
- /// Sets the headers which are permitted in a response to a CORS request.
- public var allowedHeaders: [String]
- /// Enabling this value allows sets the "access-control-allow-credentials" header field
- /// to "true" in respones to CORS requests. This must be enabled if the client intends to send
- /// credentials.
- public var allowCredentialedRequests: Bool
- /// The maximum age in seconds which pre-flight CORS requests may be cached for.
- public var preflightCacheExpiration: Int
- public init(
- allowedOrigins: AllowedOrigins = .all,
- allowedHeaders: [String] = ["content-type", "x-grpc-web", "x-user-agent"],
- allowCredentialedRequests: Bool = false,
- preflightCacheExpiration: Int = 86400
- ) {
- self.allowedOrigins = allowedOrigins
- self.allowedHeaders = allowedHeaders
- self.allowCredentialedRequests = allowCredentialedRequests
- self.preflightCacheExpiration = preflightCacheExpiration
- }
- }
- }
- extension Server.Configuration.CORS {
- public struct AllowedOrigins: Hashable, Sendable {
- enum Wrapped: Hashable, Sendable {
- case all
- case originBased
- case only([String])
- case custom(AnyCustomCORSAllowedOrigin)
- }
- private(set) var wrapped: Wrapped
- private init(_ wrapped: Wrapped) {
- self.wrapped = wrapped
- }
- /// Allow all origin values.
- public static let all = Self(.all)
- /// Allow all origin values; similar to `all` but returns the value of the origin header field
- /// in the 'access-control-allow-origin' response header (rather than "*").
- public static let originBased = Self(.originBased)
- /// Allow only the given origin values.
- public static func only(_ allowed: [String]) -> Self {
- return Self(.only(allowed))
- }
- /// Provide a custom CORS origin check.
- ///
- /// - Parameter checkOrigin: A closure which is called with the value of the 'origin' header
- /// and returns the value to use in the 'access-control-allow-origin' response header,
- /// or `nil` if the origin is not allowed.
- public static func custom<C: GRPCCustomCORSAllowedOrigin>(_ custom: C) -> Self {
- return Self(.custom(AnyCustomCORSAllowedOrigin(custom)))
- }
- }
- }
- extension ServerBootstrapProtocol {
- fileprivate func bind(to target: BindTarget) -> EventLoopFuture<Channel> {
- switch target.wrapped {
- case let .hostAndPort(host, port):
- return self.bind(host: host, port: port)
- case let .unixDomainSocket(path):
- return self.bind(unixDomainSocketPath: path)
- case let .socketAddress(address):
- return self.bind(to: address)
- case let .connectedSocket(socket):
- return self.withBoundSocket(socket)
- case let .vsockAddress(address):
- return self.bind(to: address)
- }
- }
- }
- extension Comparable {
- internal func clamped(to range: ClosedRange<Self>) -> Self {
- return min(max(self, range.lowerBound), range.upperBound)
- }
- }
- public protocol GRPCCustomCORSAllowedOrigin: Sendable, Hashable {
- /// Returns the value to use for the 'access-control-allow-origin' response header for the given
- /// value of the 'origin' request header.
- ///
- /// - Parameter origin: The value of the 'origin' request header field.
- /// - Returns: The value to use for the 'access-control-allow-origin' header field or `nil` if no
- /// CORS related headers should be returned.
- func check(origin: String) -> String?
- }
- extension Server.Configuration.CORS.AllowedOrigins {
- struct AnyCustomCORSAllowedOrigin: GRPCCustomCORSAllowedOrigin {
- private var checkOrigin: @Sendable (String) -> String?
- private let hashInto: @Sendable (inout Hasher) -> Void
- private let isEqualTo: @Sendable (any GRPCCustomCORSAllowedOrigin) -> Bool
- init<W: GRPCCustomCORSAllowedOrigin>(_ wrap: W) {
- self.checkOrigin = { wrap.check(origin: $0) }
- self.hashInto = { wrap.hash(into: &$0) }
- self.isEqualTo = { wrap == ($0 as? W) }
- }
- func check(origin: String) -> String? {
- return self.checkOrigin(origin)
- }
- func hash(into hasher: inout Hasher) {
- self.hashInto(&hasher)
- }
- static func == (
- lhs: Server.Configuration.CORS.AllowedOrigins.AnyCustomCORSAllowedOrigin,
- rhs: Server.Configuration.CORS.AllowedOrigins.AnyCustomCORSAllowedOrigin
- ) -> Bool {
- return lhs.isEqualTo(rhs)
- }
- }
- }
|