123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515 |
- import UIKit
- import NextcloudKit
- class NCEndToEndMetadata: NSObject {
- struct E2eeV1: Codable {
- struct Metadata: Codable {
- let metadataKeys: [String: String]
- let version: Double
- }
- struct Encrypted: Codable {
- let key: String
- let filename: String
- let mimetype: String
- }
- struct Files: Codable {
- let initializationVector: String
- let authenticationTag: String?
- let metadataKey: Int
- let encrypted: String
- }
- let metadata: Metadata
- let files: [String: Files]?
- }
- struct E2eeV12: Codable {
- struct Metadata: Codable {
- let metadataKey: String
- let version: Double
- let checksum: String?
- }
- struct Encrypted: Codable {
- let key: String
- let filename: String
- let mimetype: String
- }
- struct Files: Codable {
- let initializationVector: String
- let authenticationTag: String?
- let encrypted: String
- }
- struct Filedrop: Codable {
- let initializationVector: String
- let authenticationTag: String?
- let encrypted: String
- let encryptedKey: String?
- let encryptedTag: String?
- let encryptedInitializationVector: String?
- }
- let metadata: Metadata
- let files: [String: Files]?
- let filedrop: [String: Filedrop]?
- }
-
-
-
- func encoderMetadata(_ items: [tableE2eEncryption], account: String, serverUrl: String) -> String? {
- let encoder = JSONEncoder()
- var metadataKey: String = ""
- let metadataVersion = 1.2
- var files: [String: E2eeV12.Files] = [:]
- var filesCodable: [String: E2eeV12.Files]?
- var filedrop: [String: E2eeV12.Filedrop] = [:]
- var filedropCodable: [String: E2eeV12.Filedrop]?
- let privateKey = CCUtility.getEndToEndPrivateKey(account)
- var fileNameIdentifiers: [String] = []
-
-
-
-
-
-
- if items.isEmpty, let key = NCEndToEndEncryption.sharedManager()?.generateKey() as? NSData {
- if let key = key.base64EncodedString().data(using: .utf8)?.base64EncodedString(),
- let metadataKeyEncrypted = NCEndToEndEncryption.sharedManager().encryptAsymmetricString(key, publicKey: nil, privateKey: privateKey) {
- metadataKey = metadataKeyEncrypted.base64EncodedString()
- }
- } else if let metadatakey = (items.first!.metadataKey.data(using: .utf8)?.base64EncodedString()),
- let metadataKeyEncrypted = NCEndToEndEncryption.sharedManager().encryptAsymmetricString(metadatakey, publicKey: nil, privateKey: privateKey) {
- metadataKey = metadataKeyEncrypted.base64EncodedString()
- }
- for item in items {
-
-
-
- if item.blob == "files" {
- let encrypted = E2eeV12.Encrypted(key: item.key, filename: item.fileName, mimetype: item.mimeType)
- do {
-
- let json = try encoder.encode(encrypted)
- if let encrypted = NCEndToEndEncryption.sharedManager().encryptPayloadFile(String(data: json, encoding: .utf8), key: item.metadataKey) {
- let record = E2eeV12.Files(initializationVector: item.initializationVector, authenticationTag: item.authenticationTag, encrypted: encrypted)
- files.updateValue(record, forKey: item.fileNameIdentifier)
- }
- fileNameIdentifiers.append(item.fileNameIdentifier)
- } catch let error {
- print("Serious internal error in encoding metadata (" + error.localizedDescription + ")")
- return nil
- }
- }
-
-
-
- if item.blob == "filedrop" {
- var encryptedKey: String?
- var encryptedInitializationVector: NSString?
- var encryptedTag: NSString?
- if let metadataKeyFiledrop = (item.metadataKeyFiledrop.data(using: .utf8)?.base64EncodedString()),
- let metadataKeyEncrypted = NCEndToEndEncryption.sharedManager().encryptAsymmetricString(metadataKeyFiledrop, publicKey: nil, privateKey: privateKey) {
- encryptedKey = metadataKeyEncrypted.base64EncodedString()
- }
- let encrypted = E2eeV12.Encrypted(key: item.key, filename: item.fileName, mimetype: item.mimeType)
- do {
-
- let json = try encoder.encode(encrypted)
- if let encrypted = NCEndToEndEncryption.sharedManager().encryptPayloadFile(String(data: json, encoding: .utf8), key: item.metadataKeyFiledrop, initializationVector: &encryptedInitializationVector, authenticationTag: &encryptedTag) {
- let record = E2eeV12.Filedrop(initializationVector: item.initializationVector, authenticationTag: item.authenticationTag, encrypted: encrypted, encryptedKey: encryptedKey, encryptedTag: encryptedTag as? String, encryptedInitializationVector: encryptedInitializationVector as? String)
- filedrop.updateValue(record, forKey: item.fileNameIdentifier)
- }
- } catch let error {
- print("Serious internal error in encoding metadata (" + error.localizedDescription + ")")
- return nil
- }
- }
- }
-
- let passphrase = CCUtility.getEndToEndPassphrase(account).replacingOccurrences(of: " ", with: "")
- let checksum = NCEndToEndEncryption.sharedManager().createSHA256(passphrase + fileNameIdentifiers.sorted().joined() + metadataKey)
-
- let metadata = E2eeV12.Metadata(metadataKey: metadataKey, version: metadataVersion, checksum: checksum)
- if !files.isEmpty { filesCodable = files }
- if !filedrop.isEmpty { filedropCodable = filedrop }
- let e2ee = E2eeV12(metadata: metadata, files: filesCodable, filedrop: filedropCodable)
- do {
- let data = try encoder.encode(e2ee)
- data.printJson()
- let jsonString = String(data: data, encoding: .utf8)
-
- if NCManageDatabase.shared.getE2eMetadata(account: account, serverUrl: serverUrl) == nil {
- NCManageDatabase.shared.setE2eMetadata(account: account, serverUrl: serverUrl, metadataKey: metadataKey, version: metadataVersion)
- }
- return jsonString
- } catch let error {
- print("Serious internal error in encoding e2ee (" + error.localizedDescription + ")")
- return nil
- }
- }
-
-
-
- func decoderMetadata(_ json: String, serverUrl: String, account: String, urlBase: String, userId: String, ownerId: String?) -> (version: Double, metadataKey: String, error: NKError) {
- guard let data = json.data(using: .utf8) else {
- return (0, "", NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decoding JSON"))
- }
- data.printJson()
- let decoder = JSONDecoder()
- if (try? decoder.decode(E2eeV1.self, from: data)) != nil {
- return decoderMetadataV1(json, serverUrl: serverUrl, account: account, urlBase: urlBase, userId: userId)
- } else if (try? decoder.decode(E2eeV12.self, from: data)) != nil {
- return decoderMetadataV12(json, serverUrl: serverUrl, account: account, urlBase: urlBase, userId: userId, ownerId: ownerId)
- } else {
- return (0, "", NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "Server E2EE version " + NCGlobal.shared.capabilityE2EEApiVersion + ", not compatible"))
- }
- }
-
-
-
- func decoderMetadataV1(_ json: String, serverUrl: String, account: String, urlBase: String, userId: String) -> (version: Double, metadataKey: String, error: NKError) {
- guard let data = json.data(using: .utf8) else {
- return (0, "", NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decoding JSON"))
- }
- let decoder = JSONDecoder()
- let privateKey = CCUtility.getEndToEndPrivateKey(account)
- var metadataVersion: Double = 0
- do {
- let json = try decoder.decode(E2eeV1.self, from: data)
- let metadata = json.metadata
- let files = json.files
- var metadataKeys: [String: String] = [:]
- metadataVersion = metadata.version
-
- NCManageDatabase.shared.deleteE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl))
-
-
-
- for metadataKey in metadata.metadataKeys {
- let data = Data(base64Encoded: metadataKey.value)
- if let decrypted = NCEndToEndEncryption.sharedManager().decryptAsymmetricData(data, privateKey: privateKey),
- let keyData = Data(base64Encoded: decrypted) {
- let key = String(data: keyData, encoding: .utf8)
- metadataKeys[metadataKey.key] = key
- }
- }
-
-
-
- if let tableE2eMetadata = NCManageDatabase.shared.getE2eMetadata(account: account, serverUrl: serverUrl) {
- if tableE2eMetadata.version > metadataVersion {
- return (metadataVersion, "", NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error verify version \(tableE2eMetadata.version)"))
- }
- }
-
-
-
- if let files = files {
- for files in files {
- let fileNameIdentifier = files.key
- let files = files.value as E2eeV1.Files
- let encrypted = files.encrypted
- let authenticationTag = files.authenticationTag
- guard let metadataKey = metadataKeys["\(files.metadataKey)"] else { continue }
- let metadataKeyIndex = files.metadataKey
- let initializationVector = files.initializationVector
- if let decrypted = NCEndToEndEncryption.sharedManager().decryptPayloadFile(encrypted, key: metadataKey),
- let decryptedData = Data(base64Encoded: decrypted) {
- do {
- let encrypted = try decoder.decode(E2eeV1.Encrypted.self, from: decryptedData)
- if let metadata = NCManageDatabase.shared.getMetadata(predicate: NSPredicate(format: "account == %@ AND fileName == %@", account, fileNameIdentifier)) {
- let object = tableE2eEncryption()
- object.account = account
- object.authenticationTag = authenticationTag ?? ""
- object.blob = "files"
- object.fileName = encrypted.filename
- object.fileNameIdentifier = fileNameIdentifier
- object.fileNamePath = CCUtility.returnFileNamePath(fromFileName: encrypted.filename, serverUrl: serverUrl, urlBase: urlBase, userId: userId, account: account)
- object.key = encrypted.key
- object.initializationVector = initializationVector
- object.metadataKey = metadataKey
- object.metadataKeyIndex = metadataKeyIndex
- object.metadataVersion = metadataVersion
- object.mimeType = encrypted.mimetype
- object.serverUrl = serverUrl
-
- NCManageDatabase.shared.addE2eEncryption(object)
-
- metadata.fileNameView = encrypted.filename
- let results = NextcloudKit.shared.nkCommonInstance.getInternalType(fileName: encrypted.filename, mimeType: metadata.contentType, directory: metadata.directory)
- metadata.contentType = results.mimeType
- metadata.iconName = results.iconName
- metadata.classFile = results.classFile
- NCManageDatabase.shared.addMetadata(metadata)
- }
- } catch let error {
- return (metadataVersion, metadataKey, NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decrypt file: " + error.localizedDescription))
- }
- }
- }
- }
- } catch let error {
- return (metadataVersion, "", NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: error.localizedDescription))
- }
- return (metadataVersion, "", NKError())
- }
-
-
-
- func decoderMetadataV12(_ json: String, serverUrl: String, account: String, urlBase: String, userId: String, ownerId: String?) -> (version: Double, metadataKey: String, error: NKError) {
- guard let data = json.data(using: .utf8) else {
- return (0, "", NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decoding JSON"))
- }
- let decoder = JSONDecoder()
- let privateKey = CCUtility.getEndToEndPrivateKey(account)
- var metadataVersion: Double = 0
- var metadataKey = ""
- do {
- let json = try decoder.decode(E2eeV12.self, from: data)
- let metadata = json.metadata
- let files = json.files
- let filedrop = json.filedrop
- var fileNameIdentifiers: [String] = []
- metadataVersion = metadata.version
-
-
-
- let data = Data(base64Encoded: metadata.metadataKey)
-
- if let decrypted = NCEndToEndEncryption.sharedManager().decryptAsymmetricData(data, privateKey: privateKey),
- let keyData = Data(base64Encoded: decrypted),
- let key = String(data: keyData, encoding: .utf8) {
- metadataKey = key
- } else {
- return (metadataVersion, "", NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decrypt metadataKey"))
- }
-
- NCManageDatabase.shared.deleteE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl))
-
-
-
- if let files = files {
- for file in files {
- let fileNameIdentifier = file.key
- let files = file.value as E2eeV12.Files
- let encrypted = files.encrypted
- let authenticationTag = files.authenticationTag
- let initializationVector = files.initializationVector
- fileNameIdentifiers.append(fileNameIdentifier)
- if let decrypted = NCEndToEndEncryption.sharedManager().decryptPayloadFile(encrypted, key: metadataKey),
- let decryptedData = Data(base64Encoded: decrypted) {
- do {
- decryptedData.printJson()
- let encrypted = try decoder.decode(E2eeV12.Encrypted.self, from: decryptedData)
- if let metadata = NCManageDatabase.shared.getMetadata(predicate: NSPredicate(format: "account == %@ AND fileName == %@", account, fileNameIdentifier)) {
- let object = tableE2eEncryption()
- object.account = account
- object.authenticationTag = authenticationTag ?? ""
- object.blob = "files"
- object.fileName = encrypted.filename
- object.fileNameIdentifier = fileNameIdentifier
- object.fileNamePath = CCUtility.returnFileNamePath(fromFileName: encrypted.filename, serverUrl: serverUrl, urlBase: urlBase, userId: userId, account: account)
- object.key = encrypted.key
- object.initializationVector = initializationVector
- object.metadataKey = metadataKey
- object.metadataVersion = metadataVersion
- object.mimeType = encrypted.mimetype
- object.serverUrl = serverUrl
-
- NCManageDatabase.shared.addE2eEncryption(object)
-
- metadata.fileNameView = encrypted.filename
- let results = NextcloudKit.shared.nkCommonInstance.getInternalType(fileName: encrypted.filename, mimeType: metadata.contentType, directory: metadata.directory)
- metadata.contentType = results.mimeType
- metadata.iconName = results.iconName
- metadata.classFile = results.classFile
- NCManageDatabase.shared.addMetadata(metadata)
- }
- } catch let error {
- return (metadataVersion, metadataKey, NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decrypt file: " + error.localizedDescription))
- }
- }
- }
- }
-
-
-
- if let filedrop = filedrop {
- for filedrop in filedrop {
- let fileNameIdentifier = filedrop.key
- let filedrop = filedrop.value as E2eeV12.Filedrop
- var metadataKeyFiledrop: String?
- if let encryptedKey = filedrop.encryptedKey,
- let data = Data(base64Encoded: encryptedKey),
- let decrypted = NCEndToEndEncryption.sharedManager().decryptAsymmetricData(data, privateKey: privateKey) {
- let keyData = Data(base64Encoded: decrypted)
- metadataKeyFiledrop = String(data: keyData!, encoding: .utf8)
- }
- if let decrypted = NCEndToEndEncryption.sharedManager().decryptPayloadFile(filedrop.encrypted, key: metadataKeyFiledrop, initializationVector: filedrop.encryptedInitializationVector, authenticationTag: filedrop.encryptedTag),
- let decryptedData = Data(base64Encoded: decrypted) {
- do {
- decryptedData.printJson()
- let encrypted = try decoder.decode(E2eeV1.Encrypted.self, from: decryptedData)
- if let metadata = NCManageDatabase.shared.getMetadata(predicate: NSPredicate(format: "account == %@ AND fileName == %@", account, fileNameIdentifier)) {
- let object = tableE2eEncryption()
- object.account = account
- object.authenticationTag = filedrop.authenticationTag ?? ""
- object.blob = "filedrop"
- object.fileName = encrypted.filename
- object.fileNameIdentifier = fileNameIdentifier
- object.fileNamePath = CCUtility.returnFileNamePath(fromFileName: encrypted.filename, serverUrl: serverUrl, urlBase: urlBase, userId: userId, account: account)
- object.key = encrypted.key
- object.metadataKeyFiledrop = metadataKeyFiledrop ?? ""
- object.initializationVector = filedrop.initializationVector
- object.metadataKey = metadataKey
- object.metadataVersion = metadataVersion
- object.mimeType = encrypted.mimetype
- object.serverUrl = serverUrl
-
- NCManageDatabase.shared.addE2eEncryption(object)
-
- metadata.fileNameView = encrypted.filename
- let results = NextcloudKit.shared.nkCommonInstance.getInternalType(fileName: encrypted.filename, mimeType: metadata.contentType, directory: metadata.directory)
- metadata.contentType = results.mimeType
- metadata.iconName = results.iconName
- metadata.classFile = results.classFile
- NCManageDatabase.shared.addMetadata(metadata)
- }
- } catch let error {
- return (metadataVersion, metadataKey, NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error decrypt filedrop: " + error.localizedDescription))
- }
- }
- }
- }
-
- let passphrase = CCUtility.getEndToEndPassphrase(account).replacingOccurrences(of: " ", with: "")
- let checksum = NCEndToEndEncryption.sharedManager().createSHA256(passphrase + fileNameIdentifiers.sorted().joined() + metadata.metadataKey)
- if metadata.checksum != checksum {
- return (metadataVersion, metadataKey, NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: "Error checksum"))
- }
- } catch let error {
- return (metadataVersion, metadataKey, NKError(errorCode: NCGlobal.shared.errorE2EE, errorDescription: error.localizedDescription))
- }
- return (metadataVersion, metadataKey, NKError())
- }
- }
|