client side implementation of filesystemOperation

Author: saehejkang

Sources/Containerization/VirtualMachineAgent.swift

@@ -25,6 +25,11 @@ public struct WriteFileFlags {
     public var create = false
 }
 
+public enum FilesystemOperation {
+    case freeze
+    case thaw
+}
+
 /// A protocol for the agent running inside a virtual machine. If an operation isn't
 /// supported the implementation MUST return a ContainerizationError with a code of
 /// `.unsupported`.
@@ -34,6 +39,8 @@ public protocol VirtualMachineAgent: Sendable {
     func standardSetup() async throws
     /// Close any resources held by the agent.
     func close() async throws
+    // Perform a filesystem operation on the given path.
+    func filesystemOperation(operation: FilesystemOperation, path: String) async throws
 
     // POSIX-y
     func getenv(key: String) async throws -> String

Sources/Containerization/Vminitd.swift

@@ -202,6 +202,15 @@ extension Vminitd: VirtualMachineAgent {
             })
     }
 
+    /// Perform a filesystem operation on a path inside the sandbox's environment.
+    public func filesystemOperation(operation: FilesystemOperation, path: String) async throws {
+        _ = try await client.filesystemOperation(
+            .with {
+                $0.operation = operation.toProtoOperation()
+                $0.path = path
+            })
+    }
+
     public func createProcess(
         id: String,
         containerID: String?,
@@ -586,3 +595,15 @@ extension StatCategory {
         return categories
     }
 }
+
+extension FilesystemOperation {
+    /// Convert FilesystemOperation to proto oneof value.
+    func toProtoOperation() -> Com_Apple_Containerization_Sandbox_V3_FilesystemOperationRequest.OneOf_Operation {
+        switch self {
+        case .freeze:
+            return .freeze(.init())
+        case .thaw:
+            return .thaw(.init())
+        }
+    }
+}