2020-12-02 06:39:21 +01:00
|
|
|
package org.session.libsession.messaging.sending_receiving
|
|
|
|
|
|
|
|
import nl.komponents.kovenant.Promise
|
|
|
|
import nl.komponents.kovenant.deferred
|
|
|
|
|
2020-12-10 05:33:57 +01:00
|
|
|
import org.session.libsession.messaging.MessagingConfiguration
|
2020-12-02 06:39:21 +01:00
|
|
|
import org.session.libsession.messaging.jobs.JobQueue
|
|
|
|
import org.session.libsession.messaging.messages.Destination
|
|
|
|
import org.session.libsession.messaging.messages.Message
|
|
|
|
import org.session.libsession.messaging.messages.visible.VisibleMessage
|
|
|
|
import org.session.libsession.messaging.jobs.NotifyPNServerJob
|
2020-12-18 06:44:33 +01:00
|
|
|
import org.session.libsession.messaging.messages.control.ClosedGroupUpdate
|
|
|
|
import org.session.libsession.messaging.messages.visible.Profile
|
2020-12-02 06:39:21 +01:00
|
|
|
import org.session.libsession.messaging.opengroups.OpenGroupAPI
|
|
|
|
import org.session.libsession.messaging.opengroups.OpenGroupMessage
|
|
|
|
import org.session.libsession.messaging.utilities.MessageWrapper
|
|
|
|
import org.session.libsession.snode.RawResponsePromise
|
|
|
|
import org.session.libsession.snode.SnodeAPI
|
|
|
|
import org.session.libsession.snode.SnodeMessage
|
|
|
|
|
|
|
|
import org.session.libsignal.libsignal.logging.Log
|
2020-12-07 05:22:02 +01:00
|
|
|
import org.session.libsignal.service.api.messages.SignalServiceAttachment
|
2020-12-02 06:39:21 +01:00
|
|
|
import org.session.libsignal.service.internal.push.SignalServiceProtos
|
|
|
|
import org.session.libsignal.service.internal.util.Base64
|
|
|
|
import org.session.libsignal.service.loki.api.crypto.ProofOfWork
|
|
|
|
|
2020-11-25 02:06:41 +01:00
|
|
|
|
|
|
|
object MessageSender {
|
2020-12-02 06:39:21 +01:00
|
|
|
|
|
|
|
// Error
|
|
|
|
internal sealed class Error(val description: String) : Exception() {
|
|
|
|
object InvalidMessage : Error("Invalid message.")
|
|
|
|
object ProtoConversionFailed : Error("Couldn't convert message to proto.")
|
|
|
|
object ProofOfWorkCalculationFailed : Error("Proof of work calculation failed.")
|
|
|
|
object NoUserPublicKey : Error("Couldn't find user key pair.")
|
|
|
|
|
|
|
|
// Closed groups
|
|
|
|
object NoThread : Error("Couldn't find a thread associated with the given group public key.")
|
|
|
|
object NoPrivateKey : Error("Couldn't find a private key associated with the given group public key.")
|
|
|
|
object InvalidClosedGroupUpdate : Error("Invalid group update.")
|
|
|
|
|
|
|
|
internal val isRetryable: Boolean = when (this) {
|
|
|
|
is InvalidMessage -> false
|
|
|
|
is ProtoConversionFailed -> false
|
|
|
|
is ProofOfWorkCalculationFailed -> false
|
|
|
|
is InvalidClosedGroupUpdate -> false
|
|
|
|
else -> true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-12-07 05:22:02 +01:00
|
|
|
// Preparation
|
|
|
|
fun prep(signalAttachments: List<SignalServiceAttachment>, message: VisibleMessage) {
|
|
|
|
// TODO: Deal with attachments
|
|
|
|
}
|
|
|
|
|
2020-12-02 06:39:21 +01:00
|
|
|
// Convenience
|
|
|
|
fun send(message: Message, destination: Destination): Promise<Unit, Exception> {
|
|
|
|
if (destination is Destination.OpenGroup) {
|
|
|
|
return sendToOpenGroupDestination(destination, message)
|
|
|
|
}
|
|
|
|
return sendToSnodeDestination(destination, message)
|
|
|
|
}
|
|
|
|
|
|
|
|
// One-on-One Chats & Closed Groups
|
|
|
|
fun sendToSnodeDestination(destination: Destination, message: Message): Promise<Unit, Exception> {
|
|
|
|
val deferred = deferred<Unit, Exception>()
|
|
|
|
val promise = deferred.promise
|
2020-12-10 05:33:57 +01:00
|
|
|
val storage = MessagingConfiguration.shared.storage
|
2020-12-18 06:44:33 +01:00
|
|
|
val userPublicKey = storage.getUserPublicKey()
|
2020-12-02 06:39:21 +01:00
|
|
|
val preconditionFailure = Exception("Destination should not be open groups!")
|
|
|
|
var snodeMessage: SnodeMessage? = null
|
2020-12-18 06:44:33 +01:00
|
|
|
// Set the timestamp, sender and recipient
|
2020-12-02 06:39:21 +01:00
|
|
|
message.sentTimestamp ?: run { message.sentTimestamp = System.currentTimeMillis() } /* Visible messages will already have their sent timestamp set */
|
|
|
|
message.sender = storage.getUserPublicKey()
|
|
|
|
try {
|
|
|
|
when (destination) {
|
|
|
|
is Destination.Contact -> message.recipient = destination.publicKey
|
|
|
|
is Destination.ClosedGroup -> message.recipient = destination.groupPublicKey
|
|
|
|
is Destination.OpenGroup -> throw preconditionFailure
|
|
|
|
}
|
2020-12-18 06:44:33 +01:00
|
|
|
val isSelfSend = (message.recipient == userPublicKey)
|
|
|
|
// Set the failure handler (need it here already for precondition failure handling)
|
|
|
|
fun handleFailure(error: Exception) {
|
|
|
|
handleFailedMessageSend(message, error)
|
|
|
|
if (destination is Destination.Contact && message is VisibleMessage && !isSelfSend) {
|
|
|
|
//TODO Notify user for send failure
|
|
|
|
}
|
|
|
|
deferred.reject(error)
|
|
|
|
}
|
2020-12-02 06:39:21 +01:00
|
|
|
// Validate the message
|
2020-12-02 07:06:28 +01:00
|
|
|
if (!message.isValid()) { throw Error.InvalidMessage }
|
2020-12-18 06:44:33 +01:00
|
|
|
// Stop here if this is a self-send
|
|
|
|
if (isSelfSend) {
|
|
|
|
handleSuccessfulMessageSend(message, destination)
|
|
|
|
deferred.resolve(Unit)
|
|
|
|
return promise
|
|
|
|
}
|
|
|
|
// Attach the user's profile if needed
|
|
|
|
if (message is VisibleMessage) {
|
|
|
|
val displayName = storage.getUserDisplayName()!!
|
|
|
|
val profileKey = storage.getUserProfileKey()
|
|
|
|
val profilePrictureUrl = storage.getUserProfilePictureURL()
|
|
|
|
if (profileKey != null && profilePrictureUrl != null) {
|
|
|
|
message.profile = Profile(displayName, profileKey, profilePrictureUrl)
|
|
|
|
} else {
|
|
|
|
message.profile = Profile(displayName)
|
|
|
|
}
|
|
|
|
}
|
2020-12-02 06:39:21 +01:00
|
|
|
// Convert it to protobuf
|
|
|
|
val proto = message.toProto() ?: throw Error.ProtoConversionFailed
|
|
|
|
// Serialize the protobuf
|
|
|
|
val plaintext = proto.toByteArray()
|
|
|
|
// Encrypt the serialized protobuf
|
2020-12-18 06:44:33 +01:00
|
|
|
if (destination is Destination.Contact && message is VisibleMessage && !isSelfSend) {
|
|
|
|
//TODO Notify user for encrypting message
|
|
|
|
}
|
2020-12-02 06:39:21 +01:00
|
|
|
val ciphertext: ByteArray
|
|
|
|
when (destination) {
|
|
|
|
is Destination.Contact -> ciphertext = MessageSenderEncryption.encryptWithSignalProtocol(plaintext, message, destination.publicKey)
|
|
|
|
is Destination.ClosedGroup -> ciphertext = MessageSenderEncryption.encryptWithSharedSenderKeys(plaintext, destination.groupPublicKey)
|
|
|
|
is Destination.OpenGroup -> throw preconditionFailure
|
|
|
|
}
|
|
|
|
// Wrap the result
|
|
|
|
val kind: SignalServiceProtos.Envelope.Type
|
|
|
|
val senderPublicKey: String
|
|
|
|
when (destination) {
|
|
|
|
is Destination.Contact -> {
|
|
|
|
kind = SignalServiceProtos.Envelope.Type.UNIDENTIFIED_SENDER
|
|
|
|
senderPublicKey = ""
|
|
|
|
}
|
|
|
|
is Destination.ClosedGroup -> {
|
|
|
|
kind = SignalServiceProtos.Envelope.Type.CLOSED_GROUP_CIPHERTEXT
|
|
|
|
senderPublicKey = destination.groupPublicKey
|
|
|
|
}
|
|
|
|
is Destination.OpenGroup -> throw preconditionFailure
|
|
|
|
}
|
|
|
|
val wrappedMessage = MessageWrapper.wrap(kind, message.sentTimestamp!!, senderPublicKey, ciphertext)
|
|
|
|
// Calculate proof of work
|
2020-12-18 06:44:33 +01:00
|
|
|
if (destination is Destination.Contact && message is VisibleMessage && !isSelfSend) {
|
|
|
|
//TODO Notify user for proof of work calculating
|
|
|
|
}
|
2020-12-02 06:39:21 +01:00
|
|
|
val recipient = message.recipient!!
|
|
|
|
val base64EncodedData = Base64.encodeBytes(wrappedMessage)
|
|
|
|
val timestamp = System.currentTimeMillis()
|
|
|
|
val nonce = ProofOfWork.calculate(base64EncodedData, recipient, timestamp, message.ttl.toInt()) ?: throw Error.ProofOfWorkCalculationFailed
|
|
|
|
// Send the result
|
|
|
|
snodeMessage = SnodeMessage(recipient, base64EncodedData, message.ttl, timestamp, nonce)
|
|
|
|
SnodeAPI.sendMessage(snodeMessage).success { promises: Set<RawResponsePromise> ->
|
|
|
|
var isSuccess = false
|
|
|
|
val promiseCount = promises.size
|
|
|
|
var errorCount = 0
|
|
|
|
promises.forEach { promise: RawResponsePromise ->
|
|
|
|
promise.success {
|
|
|
|
if (isSuccess) { return@success } // Succeed as soon as the first promise succeeds
|
|
|
|
isSuccess = true
|
2020-12-18 06:44:33 +01:00
|
|
|
if (destination is Destination.Contact && message is VisibleMessage && !isSelfSend) {
|
|
|
|
//TODO Notify user for message sent
|
|
|
|
}
|
|
|
|
handleSuccessfulMessageSend(message, destination)
|
|
|
|
var shouldNotify = (message is VisibleMessage)
|
|
|
|
if (message is ClosedGroupUpdate && message.kind is ClosedGroupUpdate.Kind.New) {
|
|
|
|
shouldNotify = true
|
|
|
|
}
|
|
|
|
if (shouldNotify) {
|
|
|
|
val notifyPNServerJob = NotifyPNServerJob(snodeMessage)
|
|
|
|
JobQueue.shared.add(notifyPNServerJob)
|
|
|
|
deferred.resolve(Unit)
|
|
|
|
}
|
|
|
|
|
2020-12-02 06:39:21 +01:00
|
|
|
}
|
|
|
|
promise.fail {
|
|
|
|
errorCount += 1
|
|
|
|
if (errorCount != promiseCount) { return@fail } // Only error out if all promises failed
|
2020-12-18 06:44:33 +01:00
|
|
|
handleFailure(it)
|
2020-12-02 06:39:21 +01:00
|
|
|
deferred.reject(it)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}.fail {
|
|
|
|
Log.d("Loki", "Couldn't send message due to error: $it.")
|
|
|
|
deferred.reject(it)
|
|
|
|
}
|
|
|
|
} catch (exception: Exception) {
|
|
|
|
deferred.reject(exception)
|
|
|
|
}
|
|
|
|
return promise
|
|
|
|
}
|
|
|
|
|
|
|
|
// Open Groups
|
|
|
|
fun sendToOpenGroupDestination(destination: Destination, message: Message): Promise<Unit, Exception> {
|
|
|
|
val deferred = deferred<Unit, Exception>()
|
|
|
|
val promise = deferred.promise
|
2020-12-10 05:33:57 +01:00
|
|
|
val storage = MessagingConfiguration.shared.storage
|
2020-12-02 06:39:21 +01:00
|
|
|
val preconditionFailure = Exception("Destination should not be contacts or closed groups!")
|
2020-12-18 06:44:33 +01:00
|
|
|
message.sentTimestamp ?: run { message.sentTimestamp = System.currentTimeMillis() }
|
2020-12-02 06:39:21 +01:00
|
|
|
message.sender = storage.getUserPublicKey()
|
|
|
|
try {
|
|
|
|
val server: String
|
|
|
|
val channel: Long
|
|
|
|
when (destination) {
|
|
|
|
is Destination.Contact -> throw preconditionFailure
|
|
|
|
is Destination.ClosedGroup -> throw preconditionFailure
|
|
|
|
is Destination.OpenGroup -> {
|
|
|
|
message.recipient = "${destination.server}.${destination.channel}"
|
|
|
|
server = destination.server
|
|
|
|
channel = destination.channel
|
|
|
|
}
|
|
|
|
}
|
2020-12-18 06:44:33 +01:00
|
|
|
// Set the failure handler (need it here already for precondition failure handling)
|
|
|
|
fun handleFailure(error: Exception,) {
|
|
|
|
handleFailedMessageSend(message, error)
|
|
|
|
deferred.reject(error)
|
|
|
|
}
|
2020-12-02 06:39:21 +01:00
|
|
|
// Validate the message
|
2020-12-02 07:06:28 +01:00
|
|
|
if (message !is VisibleMessage || !message.isValid()) {
|
2020-12-18 06:44:33 +01:00
|
|
|
handleFailure(Error.InvalidMessage)
|
2020-12-02 06:39:21 +01:00
|
|
|
throw Error.InvalidMessage
|
|
|
|
}
|
|
|
|
// Convert the message to an open group message
|
2020-12-18 06:44:33 +01:00
|
|
|
val openGroupMessage = OpenGroupMessage.from(message, server) ?: kotlin.run {
|
|
|
|
handleFailure(Error.InvalidMessage)
|
|
|
|
throw Error.InvalidMessage
|
|
|
|
}
|
2020-12-02 06:39:21 +01:00
|
|
|
// Send the result
|
|
|
|
OpenGroupAPI.sendMessage(openGroupMessage, channel, server).success {
|
|
|
|
message.openGroupServerMessageID = it.serverID
|
2020-12-18 06:44:33 +01:00
|
|
|
handleSuccessfulMessageSend(message, destination)
|
2020-12-02 06:39:21 +01:00
|
|
|
deferred.resolve(Unit)
|
|
|
|
}.fail {
|
2020-12-18 06:44:33 +01:00
|
|
|
handleFailure(it)
|
2020-12-02 06:39:21 +01:00
|
|
|
}
|
|
|
|
} catch (exception: Exception) {
|
|
|
|
deferred.reject(exception)
|
|
|
|
}
|
|
|
|
return deferred.promise
|
|
|
|
}
|
|
|
|
|
|
|
|
// Result Handling
|
2020-12-18 06:44:33 +01:00
|
|
|
fun handleSuccessfulMessageSend(message: Message, destination: Destination) {
|
|
|
|
MessagingConfiguration.shared.storage.insertMessageOutbox(message)
|
|
|
|
// TODO
|
2020-12-02 06:39:21 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
fun handleFailedMessageSend(message: Message, error: Exception) {
|
2020-12-18 06:44:33 +01:00
|
|
|
MessagingConfiguration.shared.storage.setErrorMessage(message, error)
|
2020-12-02 06:39:21 +01:00
|
|
|
}
|
2020-11-25 02:06:41 +01:00
|
|
|
}
|