element-ios/Riot/Utils/DataProtectionHelper.swift

48 lines
1.7 KiB
Swift

/*
Copyright 2020-2024 New Vector Ltd.
SPDX-License-Identifier: AGPL-3.0-only
Please see LICENSE in the repository root for full details.
*/
import Foundation
final class DataProtectionHelper {
/// Detects after reboot, before unlocked state. Does this by trying to write a file to the filesystem (to the Caches directory) and read it back.
/// - Parameter appGroupIdentifier: App-group identifier to be used when deciding where it'll try to write the file.
/// - Returns: true if the state detected
static func isDeviceInRebootedAndLockedState(appGroupIdentifier: String? = nil) -> Bool {
let dummyString = String.vc_unique
guard let dummyData = dummyString.data(using: .utf8) else {
return true
}
do {
var url: URL
if let identifier = appGroupIdentifier,
let containerURL = FileManager.default.containerURL(forSecurityApplicationGroupIdentifier: identifier) {
url = containerURL
} else {
url = FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask)[0]
}
// add a unique filename
url = url.appendingPathComponent(String.vc_unique)
try dummyData.write(to: url, options: .completeFileProtectionUntilFirstUserAuthentication)
let readData = try Data(contentsOf: url)
let readString = String(data: readData, encoding: .utf8)
try FileManager.default.removeItem(at: url)
if readString != dummyString {
return true
}
} catch {
return true
}
return false
}
}