blob: 53281628a5826f690c79119d24a7502555d5d5ee (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
|
// Copyright (c) Tailscale Inc & contributors
// SPDX-License-Identifier: BSD-3-Clause
import Foundation
let kDefaultDiskSizeGb: Int64 = 72
let kDefaultMemSizeGb: UInt64 = 72
/// Represents a configuration for a virtual machine
class Config: Codable {
var serverSocket = "/tmp/qemu-dgram.sock"
var memorySize = (kDefaultMemSizeGb * 1024 * 1024 * 1024) as UInt64
var mac = "52:cc:cc:cc:cc:01"
var ethermac = "52:cc:cc:cc:ce:01"
var port: UInt32 = 51009
var sharedDir: String?
// The virtual machines ID. Also double as the directory name under which
// we will store configuration, block device, etc.
let vmID: String
required init(from decoder: Decoder) throws {
let container = try decoder.container(keyedBy: CodingKeys.self)
if let ethermac = try container.decodeIfPresent(String.self, forKey: .ethermac) {
self.ethermac = ethermac
}
if let serverSocket = try container.decodeIfPresent(String.self, forKey: .serverSocket) {
self.serverSocket = serverSocket
}
if let memorySize = try container.decodeIfPresent(UInt64.self, forKey: .memorySize) {
self.memorySize = memorySize
}
if let port = try container.decodeIfPresent(UInt32.self, forKey: .port) {
self.port = port
}
if let mac = try container.decodeIfPresent(String.self, forKey: .mac) {
self.mac = mac
}
if let vmID = try container.decodeIfPresent(String.self, forKey: .vmID) {
self.vmID = vmID
} else {
self.vmID = "default"
}
}
init(_ vmID: String = "default") {
self.vmID = vmID
let configFile = vmDataURL.appendingPathComponent("config.json")
if FileManager.default.fileExists(atPath: configFile.path()) {
print("Using config file at path \(configFile)")
if let jsonData = try? Data(contentsOf: configFile) {
let config = try! JSONDecoder().decode(Config.self, from: jsonData)
self.serverSocket = config.serverSocket
self.memorySize = config.memorySize
self.mac = config.mac
self.port = config.port
self.ethermac = config.ethermac
}
}
}
func persist() {
let configFile = vmDataURL.appendingPathComponent("config.json")
let data = try! JSONEncoder().encode(self)
try! data.write(to: configFile)
}
lazy var restoreImageURL: URL = {
vmBundleURL.appendingPathComponent("RestoreImage.ipsw")
}()
// The VM Data URL holds the specific files composing a unique VM guest instance
// By default, VM's are persisted at ~/VM.bundle/<vmID>
lazy var vmDataURL = {
let dataURL = vmBundleURL.appendingPathComponent(vmID)
return dataURL
}()
lazy var auxiliaryStorageURL = {
vmDataURL.appendingPathComponent("AuxiliaryStorage")
}()
lazy var diskImageURL = {
vmDataURL.appendingPathComponent("Disk.img")
}()
lazy var diskSize: Int64 = {
kDefaultDiskSizeGb * 1024 * 1024 * 1024
}()
lazy var hardwareModelURL = {
vmDataURL.appendingPathComponent("HardwareModel")
}()
lazy var machineIdentifierURL = {
vmDataURL.appendingPathComponent("MachineIdentifier")
}()
lazy var saveFileURL = {
vmDataURL.appendingPathComponent("SaveFile.vzvmsave")
}()
}
// The VM Bundle URL holds the restore image and a set of VM images.
// VMs are stored under ~/.cache/tailscale/vmtest/macos/.
var vmBundleURL: URL = {
let vmBundlePath = NSHomeDirectory() + "/.cache/tailscale/vmtest/macos/"
createDir(vmBundlePath)
let bundleURL = URL(fileURLWithPath: vmBundlePath)
return bundleURL
}()
func createDir(_ path: String) {
do {
try FileManager.default.createDirectory(atPath: path, withIntermediateDirectories: true)
} catch {
fatalError("Unable to create dir at \(path) \(error)")
}
}
|
