NCNetworkingE2EE.swift 32 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548
  1. //
  2. // NCNetworkingE2EE.swift
  3. // Nextcloud
  4. //
  5. // Created by Marino Faggiana on 05/05/2020.
  6. // Copyright © 2020 Marino Faggiana. All rights reserved.
  7. //
  8. // This program is free software: you can redistribute it and/or modify
  9. // it under the terms of the GNU General Public License as published by
  10. // the Free Software Foundation, either version 3 of the License, or
  11. // (at your option) any later version.
  12. //
  13. // This program is distributed in the hope that it will be useful,
  14. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. // GNU General Public License for more details.
  17. //
  18. // You should have received a copy of the GNU General Public License
  19. // along with this program. If not, see <http://www.gnu.org/licenses/>.
  20. //
  21. import UIKit
  22. import OpenSSL
  23. import NextcloudKit
  24. import CFNetwork
  25. import Alamofire
  26. @objc class NCNetworkingE2EE: NSObject {
  27. @objc public static let shared: NCNetworkingE2EE = {
  28. let instance = NCNetworkingE2EE()
  29. return instance
  30. }()
  31. // MARK: - WebDav Create Folder
  32. func createFolder(fileName: String, serverUrl: String, account: String, urlBase: String, userId: String, completion: @escaping (_ error: NKError) -> Void) {
  33. var fileNameFolder = CCUtility.removeForbiddenCharactersServer(fileName)!
  34. var fileNameFolderUrl = ""
  35. var fileNameIdentifier = ""
  36. var key: NSString?
  37. var initializationVector: NSString?
  38. fileNameFolder = NCUtilityFileSystem.shared.createFileName(fileNameFolder, serverUrl: serverUrl, account: account)
  39. if fileNameFolder.count == 0 {
  40. return completion(NKError())
  41. }
  42. fileNameIdentifier = CCUtility.generateRandomIdentifier()
  43. fileNameFolderUrl = serverUrl + "/" + fileNameIdentifier
  44. self.lock(account: account, serverUrl: serverUrl) { directory, e2eToken, error in
  45. if error == .success && e2eToken != nil && directory != nil {
  46. let options = NKRequestOptions(customHeader: ["e2e-token": e2eToken!])
  47. NextcloudKit.shared.createFolder(fileNameFolderUrl, options: options) { account, ocId, _, error in
  48. if error == .success {
  49. guard let fileId = NCUtility.shared.ocIdToFileId(ocId: ocId) else {
  50. // unlock
  51. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  52. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  53. }
  54. return completion(NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "Error convert ocId"))
  55. }
  56. NextcloudKit.shared.markE2EEFolder(fileId: fileId, delete: false) { account, error in
  57. if error == .success {
  58. let object = tableE2eEncryption()
  59. NCEndToEndEncryption.sharedManager()?.encryptkey(&key, initializationVector: &initializationVector)
  60. object.account = account
  61. object.authenticationTag = nil
  62. object.fileName = fileNameFolder
  63. object.fileNameIdentifier = fileNameIdentifier
  64. object.fileNamePath = ""
  65. object.key = key! as String
  66. object.initializationVector = initializationVector! as String
  67. if let result = NCManageDatabase.shared.getE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl)) {
  68. object.metadataKey = result.metadataKey
  69. object.metadataKeyIndex = result.metadataKeyIndex
  70. } else {
  71. object.metadataKey = (NCEndToEndEncryption.sharedManager()?.generateKey(16)?.base64EncodedString(options: []))! as String // AES_KEY_128_LENGTH
  72. object.metadataKeyIndex = 0
  73. }
  74. object.mimeType = "httpd/unix-directory"
  75. object.serverUrl = serverUrl
  76. object.version = 1
  77. NCManageDatabase.shared.addE2eEncryption(object)
  78. self.sendE2EMetadata(account: account, serverUrl: serverUrl, fileNameRename: nil, fileNameNewRename: nil, deleteE2eEncryption: nil, urlBase: urlBase, userId: userId) { e2eToken, error in
  79. // unlock
  80. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  81. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  82. }
  83. if error == .success, let ocId = ocId {
  84. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterCreateFolder, userInfo: ["ocId": ocId, "serverUrl": serverUrl, "account": account, "e2ee": true])
  85. }
  86. completion(error)
  87. }
  88. } else {
  89. // unlock
  90. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  91. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  92. }
  93. completion(error)
  94. }
  95. }
  96. } else {
  97. // unlock
  98. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  99. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  100. }
  101. completion(error)
  102. }
  103. }
  104. } else {
  105. completion(error)
  106. }
  107. }
  108. }
  109. // MARK: - WebDav Delete
  110. func deleteMetadata(_ metadata: tableMetadata, completion: @escaping (_ error: NKError) -> Void) {
  111. self.lock(account: metadata.account, serverUrl: metadata.serverUrl) { directory, e2eToken, error in
  112. if error == .success && e2eToken != nil && directory != nil {
  113. let deleteE2eEncryption = NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileNameIdentifier == %@", metadata.account, metadata.serverUrl, metadata.fileName)
  114. NCNetworking.shared.deleteMetadataPlain(metadata, customHeader: ["e2e-token": e2eToken!]) { error in
  115. let home = NCUtilityFileSystem.shared.getHomeServer(urlBase: metadata.urlBase, userId: metadata.userId)
  116. if metadata.serverUrl != home {
  117. self.sendE2EMetadata(account: metadata.account, serverUrl: metadata.serverUrl, fileNameRename: nil, fileNameNewRename: nil, deleteE2eEncryption: deleteE2eEncryption, urlBase: metadata.urlBase, userId: metadata.userId) { e2eToken, error in
  118. // unlock
  119. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: metadata.account, serverUrl: metadata.serverUrl) {
  120. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  121. }
  122. completion(error)
  123. }
  124. } else {
  125. // unlock
  126. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: metadata.account, serverUrl: metadata.serverUrl) {
  127. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  128. }
  129. completion(error)
  130. }
  131. }
  132. } else {
  133. completion(error)
  134. }
  135. }
  136. }
  137. // MARK: - WebDav Rename
  138. func renameMetadata(_ metadata: tableMetadata, fileNameNew: String, completion: @escaping (_ error: NKError) -> Void) {
  139. // verify if exists the new fileName
  140. if NCManageDatabase.shared.getE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileName == %@", metadata.account, metadata.serverUrl, fileNameNew)) != nil {
  141. return completion(NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "_file_already_exists_"))
  142. } else {
  143. self.sendE2EMetadata(account: metadata.account, serverUrl: metadata.serverUrl, fileNameRename: metadata.fileName, fileNameNewRename: fileNameNew, deleteE2eEncryption: nil, urlBase: metadata.urlBase, userId: metadata.userId) { e2eToken, error in
  144. if error == .success {
  145. NCManageDatabase.shared.setMetadataFileNameView(serverUrl: metadata.serverUrl, fileName: metadata.fileName, newFileNameView: fileNameNew, account: metadata.account)
  146. // Move file system
  147. let atPath = CCUtility.getDirectoryProviderStorageOcId(metadata.ocId) + "/" + metadata.fileNameView
  148. let toPath = CCUtility.getDirectoryProviderStorageOcId(metadata.ocId) + "/" + fileNameNew
  149. do {
  150. try FileManager.default.moveItem(atPath: atPath, toPath: toPath)
  151. } catch { }
  152. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterRenameFile, userInfo: ["ocId": metadata.ocId, "account": metadata.account])
  153. }
  154. // unlock
  155. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: metadata.account, serverUrl: metadata.serverUrl) {
  156. NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE") { _, _, _, _ in }
  157. }
  158. completion(error)
  159. }
  160. }
  161. }
  162. func renameMetadata(_ metadata: tableMetadata, fileNameNew: String) async -> (NKError) {
  163. // verify if exists the new fileName
  164. if NCManageDatabase.shared.getE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@ AND fileName == %@", metadata.account, metadata.serverUrl, fileNameNew)) != nil {
  165. return NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "_file_already_exists_")
  166. } else {
  167. let sendE2EMetadataResults = await sendE2EMetadata(account: metadata.account, serverUrl: metadata.serverUrl, fileNameRename: metadata.fileName, fileNameNewRename: fileNameNew, deleteE2eEncryption: nil, urlBase: metadata.urlBase, userId: metadata.userId)
  168. if sendE2EMetadataResults.error == .success {
  169. NCManageDatabase.shared.setMetadataFileNameView(serverUrl: metadata.serverUrl, fileName: metadata.fileName, newFileNameView: fileNameNew, account: metadata.account)
  170. // Move file system
  171. let atPath = CCUtility.getDirectoryProviderStorageOcId(metadata.ocId) + "/" + metadata.fileNameView
  172. let toPath = CCUtility.getDirectoryProviderStorageOcId(metadata.ocId) + "/" + fileNameNew
  173. do {
  174. try FileManager.default.moveItem(atPath: atPath, toPath: toPath)
  175. } catch { }
  176. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterRenameFile, userInfo: ["ocId": metadata.ocId, "account": metadata.account])
  177. }
  178. // unlock
  179. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: metadata.account, serverUrl: metadata.serverUrl) {
  180. await NextcloudKit.shared.lockE2EEFolder(fileId: tableLock.fileId, e2eToken: tableLock.e2eToken, method: "DELETE")
  181. }
  182. return sendE2EMetadataResults.error
  183. }
  184. }
  185. // MARK: - Upload
  186. func upload(metadata: tableMetadata, start: @escaping () -> Void, completion: @escaping (_ error: NKError) -> Void) {
  187. let objectE2eEncryption = tableE2eEncryption()
  188. var key: NSString?, initializationVector: NSString?, authenticationTag: NSString?
  189. let ocIdTemp = metadata.ocId
  190. let serverUrl = metadata.serverUrl
  191. // Verify max size
  192. if metadata.size > NCGlobal.shared.e2eeMaxFileSize {
  193. NCManageDatabase.shared.deleteMetadata(predicate: NSPredicate(format: "ocId == %@", ocIdTemp))
  194. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadedFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "ocIdTemp": ocIdTemp, "error": NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "E2E Error file too big")])
  195. start()
  196. return completion(NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "E2E Error file too big"))
  197. }
  198. // Update metadata
  199. var metadata = tableMetadata.init(value: metadata)
  200. metadata.fileName = CCUtility.generateRandomIdentifier()!
  201. metadata.e2eEncrypted = true
  202. metadata.session = NKCommon.shared.sessionIdentifierUpload
  203. metadata.sessionError = ""
  204. NCManageDatabase.shared.addMetadata(metadata)
  205. let fileNameLocalPath = CCUtility.getDirectoryProviderStorageOcId(metadata.ocId, fileNameView: metadata.fileName)!
  206. let fileNameLocalPathRequest = CCUtility.getDirectoryProviderStorageOcId(metadata.ocId, fileNameView: metadata.fileNameView)!
  207. let serverUrlFileName = serverUrl + "/" + metadata.fileName
  208. if NCEndToEndEncryption.sharedManager()?.encryptFileName(metadata.fileNameView, fileNameIdentifier: metadata.fileName, directory: CCUtility.getDirectoryProviderStorageOcId(metadata.ocId), key: &key, initializationVector: &initializationVector, authenticationTag: &authenticationTag) == false {
  209. NCManageDatabase.shared.deleteMetadata(predicate: NSPredicate(format: "ocId == %@", ocIdTemp))
  210. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadedFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "ocIdTemp": ocIdTemp, "error": NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "_e2e_error_create_encrypted_")])
  211. start()
  212. return completion(NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "_e2e_error_create_encrypted_"))
  213. }
  214. if let result = NCManageDatabase.shared.getE2eEncryption(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", metadata.account, serverUrl)) {
  215. objectE2eEncryption.metadataKey = result.metadataKey
  216. objectE2eEncryption.metadataKeyIndex = result.metadataKeyIndex
  217. } else {
  218. let key = NCEndToEndEncryption.sharedManager()?.generateKey(16) as NSData?
  219. objectE2eEncryption.metadataKey = key!.base64EncodedString()
  220. objectE2eEncryption.metadataKeyIndex = 0
  221. }
  222. objectE2eEncryption.account = metadata.account
  223. objectE2eEncryption.authenticationTag = authenticationTag as String?
  224. objectE2eEncryption.fileName = metadata.fileNameView
  225. objectE2eEncryption.fileNameIdentifier = metadata.fileName
  226. objectE2eEncryption.fileNamePath = fileNameLocalPath
  227. objectE2eEncryption.key = key! as String
  228. objectE2eEncryption.initializationVector = initializationVector! as String
  229. objectE2eEncryption.mimeType = metadata.contentType
  230. objectE2eEncryption.serverUrl = serverUrl
  231. objectE2eEncryption.version = 1
  232. NCManageDatabase.shared.addE2eEncryption(objectE2eEncryption)
  233. if let getMetadata = NCManageDatabase.shared.getMetadataFromOcId(ocIdTemp) {
  234. metadata = getMetadata
  235. } else {
  236. start()
  237. return completion(NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: "_e2e_error_create_encrypted_"))
  238. }
  239. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterReloadDataSource, userInfo: ["serverUrl": metadata.serverUrl])
  240. NCContentPresenter.shared.noteTop(text: NSLocalizedString("_upload_e2ee_", comment: ""), image: nil, type: NCContentPresenter.messageType.info, delay: NCGlobal.shared.dismissAfterSecond, priority: .max)
  241. NCNetworkingE2EE.shared.sendE2EMetadata(account: metadata.account, serverUrl: serverUrl, fileNameRename: nil, fileNameNewRename: nil, deleteE2eEncryption: nil, urlBase: metadata.urlBase, userId: metadata.userId, upload: true) { e2eToken, error in
  242. start()
  243. if error == .success && e2eToken != nil {
  244. NextcloudKit.shared.upload(serverUrlFileName: serverUrlFileName, fileNameLocalPath: fileNameLocalPath, dateCreationFile: metadata.date as Date, dateModificationFile: metadata.date as Date, addCustomHeaders: ["e2e-token": e2eToken!], requestHandler: { request in
  245. NCNetworking.shared.uploadRequest[fileNameLocalPathRequest] = request
  246. NCManageDatabase.shared.setMetadataSession(ocId: metadata.ocId, session: nil, sessionError: nil, sessionSelector: nil, sessionTaskIdentifier: nil, status: NCGlobal.shared.metadataStatusUploading)
  247. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadStartFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "sessionSelector": metadata.sessionSelector])
  248. }, taskHandler: { _ in
  249. }, progressHandler: { progress in
  250. NotificationCenter.default.postOnMainThread(
  251. name: NCGlobal.shared.notificationCenterProgressTask,
  252. userInfo: [
  253. "account": metadata.account,
  254. "ocId": metadata.ocId,
  255. "fileName": metadata.fileName,
  256. "serverUrl": serverUrl,
  257. "status": NSNumber(value: NCGlobal.shared.metadataStatusInUpload),
  258. "progress": NSNumber(value: progress.fractionCompleted),
  259. "totalBytes": NSNumber(value: progress.totalUnitCount),
  260. "totalBytesExpected": NSNumber(value: progress.completedUnitCount)])
  261. }) { account, ocId, etag, date, _, _, afError, error in
  262. NCNetworkingE2EE.shared.unlock(account: metadata.account, serverUrl: serverUrl) { _, _, errorLock in
  263. NCNetworking.shared.uploadRequest.removeValue(forKey: fileNameLocalPath)
  264. if let metadata = NCManageDatabase.shared.getMetadataFromOcId(metadata.ocId) {
  265. if afError?.isExplicitlyCancelledError ?? false {
  266. CCUtility.removeFile(atPath: CCUtility.getDirectoryProviderStorageOcId(metadata.ocId))
  267. NCManageDatabase.shared.deleteMetadata(predicate: NSPredicate(format: "ocId == %@", metadata.ocId))
  268. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadedFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "ocIdTemp": ocIdTemp, "error": error])
  269. } else if error == .success && ocId != nil {
  270. NCUtilityFileSystem.shared.moveFileInBackground(atPath: CCUtility.getDirectoryProviderStorageOcId(metadata.ocId), toPath: CCUtility.getDirectoryProviderStorageOcId(ocId))
  271. metadata.date = date ?? NSDate()
  272. metadata.etag = etag ?? ""
  273. metadata.ocId = ocId!
  274. metadata.session = ""
  275. metadata.sessionError = ""
  276. metadata.sessionTaskIdentifier = 0
  277. metadata.status = NCGlobal.shared.metadataStatusNormal
  278. NCManageDatabase.shared.addMetadata(metadata)
  279. NCManageDatabase.shared.deleteMetadata(predicate: NSPredicate(format: "ocId == %@", ocIdTemp))
  280. NCManageDatabase.shared.addLocalFile(metadata: metadata)
  281. NCUtility.shared.createImageFrom(fileNameView: metadata.fileNameView, ocId: metadata.ocId, etag: metadata.etag, classFile: metadata.classFile)
  282. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadedFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "ocIdTemp": ocIdTemp, "error": error])
  283. } else {
  284. NCManageDatabase.shared.setMetadataSession(ocId: metadata.ocId, session: nil, sessionError: error.errorDescription, sessionTaskIdentifier: 0, status: NCGlobal.shared.metadataStatusUploadError)
  285. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadedFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "ocIdTemp": ocIdTemp, "error": error])
  286. }
  287. }
  288. completion(error)
  289. }
  290. }
  291. } else {
  292. if let metadata = NCManageDatabase.shared.getMetadataFromOcId(ocIdTemp) {
  293. NCManageDatabase.shared.setMetadataSession(ocId: metadata.ocId, session: nil, sessionError: error.errorDescription, sessionTaskIdentifier: 0, status: NCGlobal.shared.metadataStatusUploadError)
  294. NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterUploadedFile, userInfo: ["ocId": metadata.ocId, "serverUrl": metadata.serverUrl, "account": metadata.account, "fileName": metadata.fileName, "ocIdTemp": ocIdTemp, "error": error])
  295. }
  296. completion(error)
  297. }
  298. }
  299. }
  300. // MARK: - E2EE
  301. @objc func lock(account: String, serverUrl: String, completion: @escaping (_ direcrtory: tableDirectory?, _ e2eToken: String?, _ error: NKError) -> Void) {
  302. var e2eToken: String?
  303. guard let directory = NCManageDatabase.shared.getTableDirectory(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl)) else {
  304. return completion(nil, nil, NKError())
  305. }
  306. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  307. e2eToken = tableLock.e2eToken
  308. }
  309. NextcloudKit.shared.lockE2EEFolder(fileId: directory.fileId, e2eToken: e2eToken, method: "POST") { account, e2eToken, data, error in
  310. if error == .success && e2eToken != nil {
  311. NCManageDatabase.shared.setE2ETokenLock(account: account, serverUrl: serverUrl, fileId: directory.fileId, e2eToken: e2eToken!)
  312. }
  313. completion(directory, e2eToken, error)
  314. }
  315. }
  316. func lock(account: String, serverUrl: String) async -> (directory: tableDirectory?, e2eToken: String?, error: NKError) {
  317. var e2eToken: String?
  318. guard let directory = NCManageDatabase.shared.getTableDirectory(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl)) else {
  319. return (nil, nil, NKError())
  320. }
  321. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  322. e2eToken = tableLock.e2eToken
  323. }
  324. let lockE2EEFolderResults = await NextcloudKit.shared.lockE2EEFolder(fileId: directory.fileId, e2eToken: e2eToken, method: "POST")
  325. if lockE2EEFolderResults.error == .success, let e2eToken = lockE2EEFolderResults.e2eToken {
  326. NCManageDatabase.shared.setE2ETokenLock(account: account, serverUrl: serverUrl, fileId: directory.fileId, e2eToken: e2eToken)
  327. }
  328. return (directory, lockE2EEFolderResults.e2eToken, lockE2EEFolderResults.error)
  329. }
  330. @objc func unlock(account: String, serverUrl: String, completion: @escaping (_ direcrtory: tableDirectory?, _ e2eToken: String?, _ error: NKError) -> Void) {
  331. var e2eToken: String?
  332. guard let directory = NCManageDatabase.shared.getTableDirectory(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl)) else {
  333. return completion(nil, nil, NKError())
  334. }
  335. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  336. e2eToken = tableLock.e2eToken
  337. }
  338. NextcloudKit.shared.lockE2EEFolder(fileId: directory.fileId, e2eToken: e2eToken, method: "DELETE") { account, e2eToken, data, error in
  339. if error == .success {
  340. NCManageDatabase.shared.deteleE2ETokenLock(account: account, serverUrl: serverUrl)
  341. }
  342. completion(directory, e2eToken, error)
  343. }
  344. }
  345. @discardableResult
  346. func unlock(account: String, serverUrl: String) async -> (directory: tableDirectory?, e2eToken: String?, error: NKError) {
  347. var e2eToken: String?
  348. guard let directory = NCManageDatabase.shared.getTableDirectory(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl)) else {
  349. return (nil, nil, NKError())
  350. }
  351. if let tableLock = NCManageDatabase.shared.getE2ETokenLock(account: account, serverUrl: serverUrl) {
  352. e2eToken = tableLock.e2eToken
  353. }
  354. let lockE2EEFolderResults = await NextcloudKit.shared.lockE2EEFolder(fileId: directory.fileId, e2eToken: e2eToken, method: "DELETE")
  355. if lockE2EEFolderResults.error == .success {
  356. NCManageDatabase.shared.deteleE2ETokenLock(account: account, serverUrl: serverUrl)
  357. }
  358. return (directory, lockE2EEFolderResults.e2eToken, lockE2EEFolderResults.error)
  359. }
  360. @objc func sendE2EMetadata(account: String, serverUrl: String, fileNameRename: String?, fileNameNewRename: String?, deleteE2eEncryption: NSPredicate?, urlBase: String, userId: String, upload: Bool = false, completion: @escaping (_ e2eToken: String?, _ error: NKError) -> Void) {
  361. self.lock(account: account, serverUrl: serverUrl) { directory, e2eToken, error in
  362. if error == .success && e2eToken != nil && directory != nil {
  363. NextcloudKit.shared.getE2EEMetadata(fileId: directory!.fileId, e2eToken: e2eToken) { account, e2eMetadata, data, error in
  364. var method = "POST"
  365. var e2eMetadataNew: String?
  366. if error == .success && e2eMetadata != nil {
  367. if !NCEndToEndMetadata.shared.decoderMetadata(e2eMetadata!, privateKey: CCUtility.getEndToEndPrivateKey(account), serverUrl: serverUrl, account: account, urlBase: urlBase, userId: userId) {
  368. return completion(e2eToken, NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: NSLocalizedString("_e2e_error_encode_metadata_", comment: "")))
  369. }
  370. method = "PUT"
  371. }
  372. // Rename
  373. if fileNameRename != nil && fileNameNewRename != nil {
  374. NCManageDatabase.shared.renameFileE2eEncryption(serverUrl: serverUrl, fileNameIdentifier: fileNameRename!, newFileName: fileNameNewRename!, newFileNamePath: CCUtility.returnFileNamePath(fromFileName: fileNameNewRename!, serverUrl: serverUrl, urlBase: urlBase, userId: userId, account: account))
  375. }
  376. // Delete
  377. if deleteE2eEncryption != nil {
  378. NCManageDatabase.shared.deleteE2eEncryption(predicate: deleteE2eEncryption!)
  379. }
  380. // Rebuild metadata for send it
  381. let tableE2eEncryption = NCManageDatabase.shared.getE2eEncryptions(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl))
  382. if tableE2eEncryption != nil {
  383. e2eMetadataNew = NCEndToEndMetadata.shared.encoderMetadata(tableE2eEncryption!, privateKey: CCUtility.getEndToEndPrivateKey(account), serverUrl: serverUrl)
  384. } else {
  385. method = "DELETE"
  386. }
  387. NextcloudKit.shared.putE2EEMetadata(fileId: directory!.fileId, e2eToken: e2eToken!, e2eMetadata: e2eMetadataNew, method: method) { account, _, _, error in
  388. if upload {
  389. completion(e2eToken, error)
  390. } else {
  391. self.unlock(account: account, serverUrl: serverUrl) { _, e2eToken, _ in
  392. completion(e2eToken, error)
  393. }
  394. }
  395. }
  396. }
  397. } else {
  398. completion(e2eToken, error)
  399. }
  400. }
  401. }
  402. func sendE2EMetadata(account: String, serverUrl: String, fileNameRename: String?, fileNameNewRename: String?, deleteE2eEncryption: NSPredicate?, urlBase: String, userId: String, upload: Bool = false) async -> (e2eToken: String?, error: NKError) {
  403. let lockResults = await lock(account: account, serverUrl: serverUrl)
  404. if lockResults.error == .success, let e2eToken = lockResults.e2eToken, let directory = lockResults.directory {
  405. let getE2EEMetadataResults = await NextcloudKit.shared.getE2EEMetadata(fileId: directory.fileId, e2eToken: e2eToken)
  406. var method = "POST"
  407. var e2eMetadataNew: String?
  408. if getE2EEMetadataResults.error == .success, let e2eMetadata = getE2EEMetadataResults.e2eMetadata {
  409. if !NCEndToEndMetadata.shared.decoderMetadata(e2eMetadata, privateKey: CCUtility.getEndToEndPrivateKey(account), serverUrl: serverUrl, account: account, urlBase: urlBase, userId: userId) {
  410. return (e2eToken, NKError(errorCode: NCGlobal.shared.errorInternalError, errorDescription: NSLocalizedString("_e2e_error_encode_metadata_", comment: "")))
  411. }
  412. method = "PUT"
  413. }
  414. // Rename
  415. if let fileNameRename = fileNameRename, let fileNameNewRename = fileNameNewRename {
  416. NCManageDatabase.shared.renameFileE2eEncryption(serverUrl: serverUrl, fileNameIdentifier: fileNameRename, newFileName: fileNameNewRename, newFileNamePath: CCUtility.returnFileNamePath(fromFileName: fileNameNewRename, serverUrl: serverUrl, urlBase: urlBase, userId: userId, account: account))
  417. }
  418. // Delete
  419. if let deleteE2eEncryption = deleteE2eEncryption {
  420. NCManageDatabase.shared.deleteE2eEncryption(predicate: deleteE2eEncryption)
  421. }
  422. // Rebuild metadata for send it
  423. if let tableE2eEncryption = NCManageDatabase.shared.getE2eEncryptions(predicate: NSPredicate(format: "account == %@ AND serverUrl == %@", account, serverUrl)) {
  424. e2eMetadataNew = NCEndToEndMetadata.shared.encoderMetadata(tableE2eEncryption, privateKey: CCUtility.getEndToEndPrivateKey(account), serverUrl: serverUrl)
  425. } else {
  426. method = "DELETE"
  427. }
  428. let putE2EEMetadataResults = await NextcloudKit.shared.putE2EEMetadata(fileId: directory.fileId, e2eToken: e2eToken, e2eMetadata: e2eMetadataNew, method: method)
  429. if upload {
  430. return (e2eToken, putE2EEMetadataResults.error)
  431. } else {
  432. let unlockResults = await unlock(account: account, serverUrl: serverUrl)
  433. return (unlockResults.e2eToken, unlockResults.error)
  434. }
  435. } else {
  436. return (lockResults.e2eToken, lockResults.error)
  437. }
  438. }
  439. }