177 lines
6.0 KiB
Swift
177 lines
6.0 KiB
Swift
//
|
|
// Archive+MemoryFile.swift
|
|
// ZIPFoundation
|
|
//
|
|
// Copyright © 2017-2025 Thomas Zoechling, https://www.peakstep.com and the ZIP Foundation project authors.
|
|
// Released under the MIT License.
|
|
//
|
|
// See https://github.com/weichsel/ZIPFoundation/blob/master/LICENSE for license information.
|
|
//
|
|
|
|
import Foundation
|
|
|
|
extension Archive {
|
|
var isMemoryArchive: Bool { return self.url.scheme == memoryURLScheme }
|
|
}
|
|
|
|
#if swift(>=5.0)
|
|
|
|
extension Archive {
|
|
|
|
class MemoryFile {
|
|
|
|
private(set) var data: Data
|
|
private var offset = 0
|
|
|
|
init(data: Data = Data()) {
|
|
self.data = data
|
|
}
|
|
|
|
func open(mode: AccessMode) throws -> FILEPointer {
|
|
let cookie = Unmanaged.passRetained(self)
|
|
#if os(macOS) || os(iOS) || os(tvOS) || os(visionOS) || os(watchOS) || os(Android)
|
|
guard let result = mode.isWritable
|
|
? funopen(cookie.toOpaque(), readStub, writeStub, seekStub, closeStub)
|
|
: funopen(cookie.toOpaque(), readStub, nil, seekStub, closeStub)
|
|
else { throw MemoryFileError.invalidMemoryFile }
|
|
#else
|
|
let stubs = cookie_io_functions_t(read: readStub, write: writeStub, seek: seekStub, close: closeStub)
|
|
guard let result = fopencookie(cookie.toOpaque(), mode.posixMode, stubs)
|
|
else { throw MemoryFileError.invalidMemoryFile }
|
|
#endif
|
|
return result
|
|
}
|
|
}
|
|
|
|
/// Returns a `Data` object containing a representation of the receiver.
|
|
public var data: Data? { return self.memoryFile?.data }
|
|
}
|
|
|
|
public enum MemoryFileError: Error {
|
|
case invalidMemoryFile
|
|
}
|
|
|
|
private extension Archive.MemoryFile {
|
|
|
|
func readData(buffer: UnsafeMutableRawBufferPointer) -> Int {
|
|
let size = min(buffer.count, data.count-offset)
|
|
let start = data.startIndex
|
|
self.data.copyBytes(to: buffer.bindMemory(to: UInt8.self), from: start+offset..<start+offset+size)
|
|
self.offset += size
|
|
return size
|
|
}
|
|
|
|
func writeData(buffer: UnsafeRawBufferPointer) -> Int {
|
|
let start = self.data.startIndex
|
|
if self.offset < self.data.count && self.offset+buffer.count > self.data.count {
|
|
self.data.removeSubrange(start+self.offset..<start+self.data.count)
|
|
} else if offset > data.count {
|
|
self.data.append(Data(count: self.offset-self.data.count))
|
|
}
|
|
if self.offset == self.data.count {
|
|
self.data.append(buffer.bindMemory(to: UInt8.self))
|
|
} else {
|
|
let start = self.data.startIndex // May have changed in earlier mutation
|
|
self.data.replaceSubrange(
|
|
start+self.offset..<start+self.offset+buffer.count,
|
|
with: buffer.bindMemory(to: UInt8.self)
|
|
)
|
|
}
|
|
self.offset += buffer.count
|
|
return buffer.count
|
|
}
|
|
|
|
func seek(offset: Int, whence: Int32) -> Int {
|
|
var result = -1
|
|
if whence == SEEK_SET {
|
|
result = offset
|
|
} else if whence == SEEK_CUR {
|
|
result = self.offset + offset
|
|
} else if whence == SEEK_END {
|
|
result = data.count + offset
|
|
}
|
|
self.offset = result
|
|
return self.offset
|
|
}
|
|
}
|
|
|
|
private func fileFromCookie(cookie: UnsafeRawPointer) -> Archive.MemoryFile {
|
|
return Unmanaged<Archive.MemoryFile>.fromOpaque(cookie).takeUnretainedValue()
|
|
}
|
|
|
|
private func closeStub(_ cookie: UnsafeMutableRawPointer?) -> Int32 {
|
|
if let cookie = cookie {
|
|
Unmanaged<Archive.MemoryFile>.fromOpaque(cookie).release()
|
|
}
|
|
return 0
|
|
}
|
|
|
|
#if os(macOS) || os(iOS) || os(tvOS) || os(visionOS) || os(watchOS) || os(Android)
|
|
|
|
private func readStub(_ cookie: UnsafeMutableRawPointer?,
|
|
_ bytePtr: UnsafeMutablePointer<Int8>?,
|
|
_ count: Int32) -> Int32 {
|
|
guard let cookie = cookie, let bytePtr = bytePtr else { return 0 }
|
|
return Int32(fileFromCookie(cookie: cookie).readData(
|
|
buffer: UnsafeMutableRawBufferPointer(start: bytePtr, count: Int(count))))
|
|
}
|
|
|
|
private func writeStub(_ cookie: UnsafeMutableRawPointer?,
|
|
_ bytePtr: UnsafePointer<Int8>?,
|
|
_ count: Int32) -> Int32 {
|
|
guard let cookie = cookie, let bytePtr = bytePtr else { return 0 }
|
|
return Int32(fileFromCookie(cookie: cookie).writeData(
|
|
buffer: UnsafeRawBufferPointer(start: bytePtr, count: Int(count))))
|
|
}
|
|
|
|
private func seekStub(_ cookie: UnsafeMutableRawPointer?,
|
|
_ offset: fpos_t,
|
|
_ whence: Int32) -> fpos_t {
|
|
guard let cookie = cookie else { return 0 }
|
|
return fpos_t(fileFromCookie(cookie: cookie).seek(offset: Int(offset), whence: whence))
|
|
}
|
|
|
|
#else
|
|
|
|
extension Archive.AccessMode {
|
|
|
|
var posixMode: String {
|
|
switch self {
|
|
case .read: return "rb"
|
|
case .create: return "wb+"
|
|
case .update: return "rb+"
|
|
}
|
|
}
|
|
}
|
|
|
|
private func readStub(_ cookie: UnsafeMutableRawPointer?,
|
|
_ bytePtr: UnsafeMutablePointer<Int8>?,
|
|
_ count: Int) -> Int {
|
|
guard let cookie = cookie, let bytePtr = bytePtr else { return 0 }
|
|
return fileFromCookie(cookie: cookie).readData(
|
|
buffer: UnsafeMutableRawBufferPointer(start: bytePtr, count: count))
|
|
}
|
|
|
|
private func writeStub(_ cookie: UnsafeMutableRawPointer?,
|
|
_ bytePtr: UnsafePointer<Int8>?,
|
|
_ count: Int) -> Int {
|
|
guard let cookie = cookie, let bytePtr = bytePtr else { return 0 }
|
|
return fileFromCookie(cookie: cookie).writeData(
|
|
buffer: UnsafeRawBufferPointer(start: bytePtr, count: count))
|
|
}
|
|
|
|
private func seekStub(_ cookie: UnsafeMutableRawPointer?,
|
|
_ offset: UnsafeMutablePointer<Int>?,
|
|
_ whence: Int32) -> Int32 {
|
|
guard let cookie = cookie, let offset = offset else { return 0 }
|
|
let result = fileFromCookie(cookie: cookie).seek(offset: Int(offset.pointee), whence: whence)
|
|
if result >= 0 {
|
|
offset.pointee = result
|
|
return 0
|
|
} else {
|
|
return -1
|
|
}
|
|
}
|
|
#endif
|
|
#endif
|