narayana/another.im-ios
7
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

wip

Browse source
This commit is contained in:
fmodf 2024-09-08 17:06:52 +02:00
parent 0a57e0648f
commit 18083e0b19
1 changed files with 21 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
ConversationsClassic/AppData/Client/Client.swift
View file

@ -110,31 +110,25 @@ extension Client {
func uploadFile(_ localURL: URL) async throws -> String {
// get data from file
guard let data = try? Data(contentsOf: localURL) else {
guard var data = try? Data(contentsOf: localURL) else {
throw AppError.noData
}
// encode data with AES_GSM
guard let iv = try? AESGSMEngine.generateIV(), let key = try? AESGSMEngine.generateKey() else {
throw AppError.securityError
}
var encodedData = Data()
var tag = Data()
guard AESGSMEngine.shared.encrypt(iv: iv, key: key, message: data, output: &encodedData, tag: &tag) else {
throw AppError.securityError
}
// encrypt data if needed
let ((key, iv), encrypted) = try await encryptFile(data)
data = encrypted
// upload
let httpModule = connection.module(HttpFileUploadModule.self)
let components = try await httpModule.findHttpUploadComponents()
guard let component = components.first(where: { $0.maxSize > encodedData.count }) else {
guard let component = components.first(where: { $0.maxSize > data.count }) else {
throw AppError.fileTooBig
}
let slot = try await httpModule.requestUploadSlot(
componentJid: component.jid,
filename: localURL.lastPathComponent,
size: encodedData.count,
size: data.count,
contentType: localURL.mimeType
)
var request = URLRequest(url: slot.putUri)
@ -142,8 +136,8 @@ extension Client {
request.addValue(value, forHTTPHeaderField: key)
}
request.httpMethod = "PUT"
request.httpBody = encodedData
request.addValue(String(encodedData.count), forHTTPHeaderField: "Content-Length")
request.httpBody = data
request.addValue(String(data.count), forHTTPHeaderField: "Content-Length")
request.addValue(localURL.mimeType, forHTTPHeaderField: "Content-Type")
let (_, response) = try await URLSession.shared.data(for: request)
switch response {
@ -198,6 +192,19 @@ private extension Client {
})
}
}
func encryptFile(_ data: Data) async throws -> ((Data, Data), Data) {
guard let iv = try? AESGSMEngine.generateIV(), let key = try? AESGSMEngine.generateKey() else {
throw AppError.securityError
}
var encrypted = Data()
var tag = Data()
guard AESGSMEngine.shared.encrypt(iv: iv, key: key, message: data, output: &encrypted, tag: &tag) else {
throw AppError.securityError
}
return ((key, iv), encrypted)
}
}
extension Client {