mirror of
https://github.com/apple/container.git
synced 2026-07-14 21:47:01 +00:00
J Logan
a8dbf294f6
- Closes #150, #394. - Introduces `AttachmentConfiguration` type so that we can add key-value options to `--network` in the future. - Eliminates redundant `ContainersService.Item` type. - Since we now ensure at ContainersService that hostnames will not conflict, the network helper IP allocator now simply provides the existing IP if for an allocation on an existing hostname, which should handle (in an eventually consistent way) the case where a container fails to deallocate an IP on shutdown.
61 lines
2.1 KiB
Swift
61 lines
2.1 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
// Copyright © 2025 Apple Inc. and the container project 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
|
|
//
|
|
// https://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 ContainerizationError
|
|
import ContainerizationExtras
|
|
|
|
actor AttachmentAllocator {
|
|
private let allocator: any AddressAllocator<UInt32>
|
|
private var hostnames: [String: UInt32] = [:]
|
|
|
|
init(lower: UInt32, size: Int) throws {
|
|
allocator = try UInt32.rotatingAllocator(
|
|
lower: lower,
|
|
size: UInt32(size)
|
|
)
|
|
}
|
|
|
|
/// Allocate a network address for a host.
|
|
func allocate(hostname: String) async throws -> UInt32 {
|
|
// Client is responsible for ensuring two containers don't use same hostname, so provide existing IP if hostname exists
|
|
if let index = hostnames[hostname] {
|
|
return index
|
|
}
|
|
|
|
let index = try allocator.allocate()
|
|
hostnames[hostname] = index
|
|
|
|
return index
|
|
}
|
|
|
|
/// Free an allocated network address by hostname.
|
|
func deallocate(hostname: String) async throws {
|
|
if let index = hostnames.removeValue(forKey: hostname) {
|
|
try allocator.release(index)
|
|
}
|
|
}
|
|
|
|
/// If no addresses are allocated, prevent future allocations and return true.
|
|
func disableAllocator() async -> Bool {
|
|
allocator.disableAllocator()
|
|
}
|
|
|
|
/// Retrieve the allocator index for a hostname.
|
|
func lookup(hostname: String) async throws -> UInt32? {
|
|
hostnames[hostname]
|
|
}
|
|
}
|