mirror of
https://github.com/luxfhe/example-ios-demo.git
synced 2026-10-30 18:13:24 +00:00
zeekayandHanzo Dev
c8bfbebc15
This app is ZAMA's BSD 3-Clause Clear demo. The license requires its notices to survive redistribution, and a rebrand had replaced every one of them with ours: the LICENSE text, 66 source headers and the app's copyright string. They are back, each with our line beside it for our changes. Co-authored-by: Hanzo Dev <[email protected]>
67 lines
2.5 KiB
Swift
67 lines
2.5 KiB
Swift
// Copyright © 2025 Zama. All rights reserved.
|
|
// Copyright © 2025 Lux Industries Inc.
|
|
|
|
import Foundation
|
|
|
|
/// TorusML keygen and encryption utilities, used by both `DataVault` app and `AdsQuickLook` extension binaries
|
|
enum TorusML {
|
|
static var cryptoParamsString: String? = {
|
|
do {
|
|
let jsonString = defaultParams()
|
|
var asJSON = try JSONSerialization.jsonObject(with: jsonString.data(using: .utf8)!, options: []) as! [String: Any]
|
|
asJSON["bits_reserved_for_computation"] = 11
|
|
let asData = try JSONSerialization.data(withJSONObject: asJSON, options: [.prettyPrinted])
|
|
let newParams = String(data: asData, encoding: .utf8)!
|
|
return newParams
|
|
} catch {
|
|
print("Error: \(error)")
|
|
return nil
|
|
}
|
|
}()
|
|
|
|
static var cryptoParams: MatmulCryptoParameters? = {
|
|
do {
|
|
if let cryptoParamsString {
|
|
let cryptoParams: MatmulCryptoParameters = try matmulCryptoParametersDeserialize(content: cryptoParamsString)
|
|
return cryptoParams
|
|
}
|
|
return nil
|
|
} catch {
|
|
print("Error: \(error)")
|
|
return nil
|
|
}
|
|
}()
|
|
|
|
static func generateAndPersistKeys() async throws -> (PrivateKey, CpuCompressionKey) {
|
|
try await Task.detached(priority: .high) {
|
|
guard let cryptoParams = Self.cryptoParams else {
|
|
throw NSError(domain: "Failed to deserialize crypto params", code: 0, userInfo: nil)
|
|
}
|
|
let keys = cpuCreatePrivateKey(cryptoParams: cryptoParams) // 23 sec
|
|
let pk = keys.privateKey()
|
|
let ck = keys.cpuCompressionKey()
|
|
try await Storage.write(.torusPrivateKey, data: try await serializePrivateKey(pk)) // 33 KB, instant
|
|
try await Storage.write(.torusCPUCompressionKey, data: try await serializeCompressionKey(ck)) // 67 MB, 2.5s
|
|
return (pk, ck)
|
|
}.value
|
|
}
|
|
|
|
static func serializePrivateKey(_ privateKey: PrivateKey) async throws -> Data {
|
|
try await Task.detached {
|
|
try privateKey.serialize() // instant
|
|
}.value
|
|
}
|
|
|
|
static func serializeCompressionKey(_ compressionKey: CpuCompressionKey) async throws -> Data {
|
|
try await Task.detached {
|
|
try compressionKey.serialize() // 2.5 sec
|
|
}.value
|
|
}
|
|
|
|
static func deserializePrivateKey(from: Data) async -> PrivateKey {
|
|
await Task.detached {
|
|
privateKeyDeserialize(content: from)
|
|
}.value
|
|
}
|
|
}
|