wehub-resource-sync
e84fb4a79e
container project - merge build / Invoke build (push) Successful in 1s
73 lines
3.0 KiB
Swift
73 lines
3.0 KiB
Swift
//===----------------------------------------------------------------------===//
|
|
// Copyright © 2026 Apple Inc. and the container project authors.
|
|
//
|
|
// 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 Foundation
|
|
|
|
// TODO: This copies some of the Bindable code from Containerization,
|
|
// but we can't use Bindable as it presumes a fixed length record.
|
|
// We can look at refining this later to see if we can use some common
|
|
// bit fiddling code everywhere.
|
|
|
|
extension [UInt8] {
|
|
/// Copy a value into the buffer at the given offset.
|
|
/// - Returns: The new offset after writing, or nil if the buffer is too small.
|
|
package mutating func copyIn<T>(as type: T.Type, value: T, offset: Int = 0) -> Int? {
|
|
let size = MemoryLayout<T>.size
|
|
guard self.count >= size + offset else {
|
|
return nil
|
|
}
|
|
return self.withUnsafeMutableBytes {
|
|
$0.baseAddress?.advanced(by: offset).assumingMemoryBound(to: T.self).pointee = value
|
|
return offset + size
|
|
}
|
|
}
|
|
|
|
/// Copy a value out of the buffer at the given offset.
|
|
/// - Returns: A tuple of (new offset, value), or nil if the buffer is too small.
|
|
package func copyOut<T>(as type: T.Type, offset: Int = 0) -> (Int, T)? {
|
|
let size = MemoryLayout<T>.size
|
|
guard self.count >= size + offset else {
|
|
return nil
|
|
}
|
|
return self.withUnsafeBytes {
|
|
guard let value = $0.baseAddress?.advanced(by: offset).assumingMemoryBound(to: T.self).pointee else {
|
|
return nil
|
|
}
|
|
return (offset + size, value)
|
|
}
|
|
}
|
|
|
|
/// Copy a byte array into the buffer at the given offset.
|
|
/// - Returns: The new offset after writing, or nil if the buffer is too small.
|
|
package mutating func copyIn(buffer: [UInt8], offset: Int = 0) -> Int? {
|
|
guard offset + buffer.count <= self.count else {
|
|
return nil
|
|
}
|
|
self[offset..<offset + buffer.count] = buffer[0..<buffer.count]
|
|
return offset + buffer.count
|
|
}
|
|
|
|
/// Copy bytes out of the buffer into another buffer.
|
|
/// - Returns: The new offset after reading, or nil if the buffer is too small.
|
|
package func copyOut(buffer: inout [UInt8], offset: Int = 0) -> Int? {
|
|
guard offset + buffer.count <= self.count else {
|
|
return nil
|
|
}
|
|
buffer[0..<buffer.count] = self[offset..<offset + buffer.count]
|
|
return offset + buffer.count
|
|
}
|
|
}
|