86 lines
3.4 KiB
Kotlin
86 lines
3.4 KiB
Kotlin
package org.thoughtcrime.securesms.loki.protocol
|
|
|
|
import com.google.protobuf.ByteString
|
|
import org.session.libsession.messaging.jobs.Data
|
|
import org.thoughtcrime.securesms.ApplicationContext
|
|
import org.thoughtcrime.securesms.crypto.UnidentifiedAccessUtil
|
|
import org.session.libsession.messaging.threads.Address
|
|
import org.thoughtcrime.securesms.jobmanager.Job
|
|
import org.thoughtcrime.securesms.jobmanager.impl.NetworkConstraint
|
|
import org.thoughtcrime.securesms.jobs.BaseJob
|
|
import org.session.libsignal.utilities.logging.Log
|
|
import org.session.libsession.messaging.threads.recipients.Recipient
|
|
import org.session.libsignal.libsignal.util.guava.Optional
|
|
import org.session.libsignal.service.api.push.SignalServiceAddress
|
|
import org.session.libsignal.service.internal.push.SignalServiceProtos
|
|
import org.session.libsignal.service.loki.protocol.meta.TTLUtilities
|
|
import java.io.IOException
|
|
import java.security.SecureRandom
|
|
import java.util.*
|
|
import java.util.concurrent.TimeUnit
|
|
|
|
class NullMessageSendJob private constructor(parameters: Parameters, private val publicKey: String) : BaseJob(parameters) {
|
|
|
|
companion object {
|
|
const val KEY = "NullMessageSendJob"
|
|
}
|
|
|
|
constructor(publicKey: String) : this(Parameters.Builder()
|
|
.addConstraint(NetworkConstraint.KEY)
|
|
.setQueue(KEY)
|
|
.setLifespan(TimeUnit.DAYS.toMillis(1))
|
|
.setMaxAttempts(1)
|
|
.build(),
|
|
publicKey)
|
|
|
|
override fun serialize(): Data {
|
|
return Data.Builder().putString("publicKey", publicKey).build()
|
|
}
|
|
|
|
override fun getFactoryKey(): String { return KEY }
|
|
|
|
public override fun onRun() {
|
|
val contentMessage = SignalServiceProtos.Content.newBuilder()
|
|
val nullMessage = SignalServiceProtos.NullMessage.newBuilder()
|
|
val sr = SecureRandom()
|
|
val paddingSize = sr.nextInt(512)
|
|
val padding = ByteArray(paddingSize)
|
|
sr.nextBytes(padding)
|
|
nullMessage.padding = ByteString.copyFrom(padding)
|
|
contentMessage.nullMessage = nullMessage.build()
|
|
val serializedContentMessage = contentMessage.build().toByteArray()
|
|
val messageSender = ApplicationContext.getInstance(context).communicationModule.provideSignalMessageSender()
|
|
val address = SignalServiceAddress(publicKey)
|
|
val recipient = Recipient.from(context, Address.fromSerialized(publicKey), false)
|
|
val udAccess = UnidentifiedAccessUtil.getAccessFor(context, recipient)
|
|
val ttl = TTLUtilities.getTTL(TTLUtilities.MessageType.Ephemeral)
|
|
try {
|
|
messageSender.sendMessage(0, address, udAccess.get().targetUnidentifiedAccess,
|
|
Date().time, serializedContentMessage, false, ttl, false,
|
|
false, false, false, Optional.absent())
|
|
} catch (e: Exception) {
|
|
Log.d("Loki", "Failed to send null message to: $publicKey due to error: $e.")
|
|
throw e
|
|
}
|
|
}
|
|
|
|
public override fun onShouldRetry(e: Exception): Boolean {
|
|
// Disable since we have our own retrying
|
|
return false
|
|
}
|
|
|
|
override fun onCanceled() { }
|
|
|
|
class Factory : Job.Factory<NullMessageSendJob> {
|
|
|
|
override fun create(parameters: Parameters, data: Data): NullMessageSendJob {
|
|
try {
|
|
val publicKey = data.getString("publicKey")
|
|
return NullMessageSendJob(parameters, publicKey)
|
|
} catch (e: IOException) {
|
|
throw AssertionError(e)
|
|
}
|
|
}
|
|
}
|
|
}
|