oxen-io
/
session-ios
mirror of https://github.com/oxen-io/session-ios.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Don't hardcode scheme and port

This commit is contained in:
Niels Andriesse 2021-04-20 11:13:55 +10:00
parent 5a242ce4e8
commit 8375b598ad
2 changed files with 11 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
SessionSnodeKit/OnionRequestAPI+Encryption.swift
View File

@ -27,7 +27,7 @@ internal extension OnionRequestAPI {
let plaintext = try encode(ciphertext: payloadAsData, json: [ "headers" : "" ])
let result = try AESGCM.encrypt(plaintext, for: snodeX25519PublicKey)
seal.fulfill(result)
case .server(_, _, let serverX25519PublicKey):
case .server(_, _, let serverX25519PublicKey, _, _):
let plaintext = try JSONSerialization.data(withJSONObject: payload, options: [ .fragmentsAllowed ])
let result = try AESGCM.encrypt(plaintext, for: serverX25519PublicKey)
seal.fulfill(result)
@ -48,9 +48,9 @@ internal extension OnionRequestAPI {
case .snode(let snode):
let snodeED25519PublicKey = snode.publicKeySet.ed25519Key
parameters = [ "destination" : snodeED25519PublicKey ]
case .server(let host, let target, _):
// FIXME: protocol and port shouldn't be hardcoded
parameters = [ "host" : host, "target" : target, "method" : "POST", "protocol" : "http", "port" : 80 ]
case .server(let host, let target, _, let scheme, let port):
parameters = [ "host" : host, "target" : target, "method" : "POST", "protocol" : scheme ?? "https" ]
if let port = port { parameters["port"] = port }
}
parameters["ephemeral_key"] = previousEncryptionResult.ephemeralPublicKey.toHexString()
let x25519PublicKey: String
@ -58,7 +58,7 @@ internal extension OnionRequestAPI {
case .snode(let snode):
let snodeX25519PublicKey = snode.publicKeySet.x25519Key
x25519PublicKey = snodeX25519PublicKey
case .server(_, _, let serverX25519PublicKey):
case .server(_, _, let serverX25519PublicKey, _, _):
x25519PublicKey = serverX25519PublicKey
}
do {

15
SessionSnodeKit/OnionRequestAPI.swift
View File

@ -29,7 +29,7 @@ public enum OnionRequestAPI {
// MARK: Destination
public enum Destination {
case snode(Snode)
case server(host: String, target: String, x25519PublicKey: String)
case server(host: String, target: String, x25519PublicKey: String, scheme: String?, port: UInt16?)
}
// MARK: Error
@ -306,13 +306,10 @@ public enum OnionRequestAPI {
default: return value
}
}
guard let url = request.url?.absoluteString, let host = request.url?.host else { return Promise(error: Error.invalidURL) }
var endpoint = ""
if server.count < url.count {
guard let serverEndIndex = url.range(of: server)?.upperBound else { return Promise(error: Error.invalidURL) }
let endpointStartIndex = url.index(after: serverEndIndex)
endpoint = String(url[endpointStartIndex..<url.endIndex])
}
guard let url = request.url, let host = request.url?.host else { return Promise(error: Error.invalidURL) }
let endpoint = url.path.removingPrefix("/")
let scheme = url.scheme
let port = given(url.port) { UInt16($0) }
let parametersAsString: String
if let tsRequest = request as? TSRequest {
headers["Content-Type"] = "application/json"
@ -339,7 +336,7 @@ public enum OnionRequestAPI {
"method" : request.httpMethod!,
"headers" : headers
]
let destination = Destination.server(host: host, target: target, x25519PublicKey: x25519PublicKey)
let destination = Destination.server(host: host, target: target, x25519PublicKey: x25519PublicKey, scheme: scheme, port: port)
let promise = sendOnionRequest(with: payload, to: destination, isJSONRequired: isJSONRequired)
promise.catch2 { error in
SNLog("Couldn't reach server: \(url) due to error: \(error).")