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
|
//
// PrivateKeyWithMetadata.swift
// MullvadVPN
//
// Created by pronebird on 20/06/2019.
// Copyright © 2019 Mullvad VPN AB. All rights reserved.
//
import Foundation
import WireGuardKit
/// A struct holding a private WireGuard key with associated metadata
struct PrivateKeyWithMetadata {
/// When the key was created
let creationDate: Date
/// Private key
let privateKey: PrivateKey
/// Public key metadata
var publicKeyWithMetadata: PublicKeyWithMetadata {
return PublicKeyWithMetadata(publicKey: privateKey.publicKey, createdAt: creationDate)
}
/// Initialize the new private key
init() {
privateKey = PrivateKey()
creationDate = Date()
}
/// Initialize with the existing private key
init(privateKey: PrivateKey, createdAt: Date) {
self.privateKey = privateKey
creationDate = createdAt
}
}
/// A struct holding a public WireGuard key with associated metadata
struct PublicKeyWithMetadata: Equatable {
/// Refers to private key creation date
let creationDate: Date
/// Public key
let publicKey: PublicKey
init(publicKey: PublicKey, createdAt: Date) {
self.publicKey = publicKey
creationDate = createdAt
}
/// Returns a base64 encoded string representation that can be used for displaying the key in
/// the user interface
func stringRepresentation(maxLength: Int? = nil) -> String {
let base64EncodedKey = publicKey.base64Key
if let maxLength = maxLength, maxLength < base64EncodedKey.count {
return base64EncodedKey.prefix(maxLength) + "..."
} else {
return base64EncodedKey
}
}
}
extension PrivateKeyWithMetadata: Codable {
private enum CodingKeys: String, CodingKey {
case privateKeyData, creationDate
}
func encode(to encoder: Encoder) throws {
var container = encoder.container(keyedBy: CodingKeys.self)
try container.encode(privateKey.rawValue, forKey: .privateKeyData)
try container.encode(creationDate, forKey: .creationDate)
}
init(from decoder: Decoder) throws {
let container = try decoder.container(keyedBy: CodingKeys.self)
let privateKeyBytes = try container.decode(Data.self, forKey: .privateKeyData)
guard let privateKey = PrivateKey(rawValue: privateKeyBytes) else {
throw DecodingError.dataCorruptedError(
forKey: CodingKeys.privateKeyData,
in: container,
debugDescription: "Invalid key data"
)
}
self.privateKey = privateKey
self.creationDate = try container.decode(Date.self, forKey: .creationDate)
}
}
|
