@ -25,7 +25,9 @@ import org.whispersystems.signalservice.api.messages.SignalServiceGroup
import org.whispersystems.signalservice.api.messages.SignalServiceGroup.GroupType
import org.whispersystems.signalservice.api.messages.SignalServiceGroup.GroupType
import org.whispersystems.signalservice.internal.push.SignalServiceProtos
import org.whispersystems.signalservice.internal.push.SignalServiceProtos
import org.whispersystems.signalservice.internal.push.SignalServiceProtos.GroupContext
import org.whispersystems.signalservice.internal.push.SignalServiceProtos.GroupContext
import org.whispersystems.signalservice.loki.api.SnodeAPI
import org.whispersystems.signalservice.loki.protocol.closedgroups.ClosedGroupRatchet
import org.whispersystems.signalservice.loki.protocol.closedgroups.ClosedGroupRatchet
import org.whispersystems.signalservice.loki.protocol.closedgroups.ClosedGroupRatchetCollectionType
import org.whispersystems.signalservice.loki.protocol.closedgroups.ClosedGroupSenderKey
import org.whispersystems.signalservice.loki.protocol.closedgroups.ClosedGroupSenderKey
import org.whispersystems.signalservice.loki.protocol.closedgroups.SharedSenderKeysImplementation
import org.whispersystems.signalservice.loki.protocol.closedgroups.SharedSenderKeysImplementation
import org.whispersystems.signalservice.loki.utilities.hexEncodedPrivateKey
import org.whispersystems.signalservice.loki.utilities.hexEncodedPrivateKey
@ -39,6 +41,12 @@ object ClosedGroupsProtocol {
val isSharedSenderKeysEnabled = true
val isSharedSenderKeysEnabled = true
val groupSizeLimit = 20
val groupSizeLimit = 20
sealed class Error ( val description : String ) : Exception ( ) {
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 InvalidUpdate : Error ( " Invalid group update. " )
}
public fun createClosedGroup ( context : Context , name : String , members : Collection < String > ) : Promise < String , Exception > {
public fun createClosedGroup ( context : Context , name : String , members : Collection < String > ) : Promise < String , Exception > {
val deferred = deferred < String , Exception > ( )
val deferred = deferred < String , Exception > ( )
Thread {
Thread {
@ -98,10 +106,12 @@ object ClosedGroupsProtocol {
val name = group . title
val name = group . title
val oldMembers = group . members . map { it . serialize ( ) } . toSet ( )
val oldMembers = group . members . map { it . serialize ( ) } . toSet ( )
val newMembers = oldMembers . minus ( userPublicKey )
val newMembers = oldMembers . minus ( userPublicKey )
update ( context , groupPublicKey , newMembers , name )
return update ( context , groupPublicKey , newMembers , name ) . get ( )
}
}
public fun update ( context : Context , groupPublicKey : String , members : Collection < String > , name : String ) {
public fun update ( context : Context , groupPublicKey : String , members : Collection < String > , name : String ) : Promise < Unit , Exception > {
val deferred = deferred < Unit , Exception > ( )
Thread {
val userPublicKey = TextSecurePreferences . getLocalNumber ( context )
val userPublicKey = TextSecurePreferences . getLocalNumber ( context )
val sskDatabase = DatabaseFactory . getSSKDatabase ( context )
val sskDatabase = DatabaseFactory . getSSKDatabase ( context )
val groupDB = DatabaseFactory . getGroupDatabase ( context )
val groupDB = DatabaseFactory . getGroupDatabase ( context )
@ -109,7 +119,7 @@ object ClosedGroupsProtocol {
val group = groupDB . getGroup ( groupID ) . orNull ( )
val group = groupDB . getGroup ( groupID ) . orNull ( )
if ( group == null ) {
if ( group == null ) {
Log . d ( " Loki " , " Can't update nonexistent closed group. " )
Log . d ( " Loki " , " Can't update nonexistent closed group. " )
return
return @Thread deferred . reject ( Error . NoThread )
}
}
val oldMembers = group . members . map { it . serialize ( ) } . toSet ( )
val oldMembers = group . members . map { it . serialize ( ) } . toSet ( )
val newMembers = members . minus ( oldMembers )
val newMembers = members . minus ( oldMembers )
@ -119,7 +129,7 @@ object ClosedGroupsProtocol {
val groupPrivateKey = DatabaseFactory . getSSKDatabase ( context ) . getClosedGroupPrivateKey ( groupPublicKey )
val groupPrivateKey = DatabaseFactory . getSSKDatabase ( context ) . getClosedGroupPrivateKey ( groupPublicKey )
if ( groupPrivateKey == null ) {
if ( groupPrivateKey == null ) {
Log . d ( " Loki " , " Couldn't get private key for closed group. " )
Log . d ( " Loki " , " Couldn't get private key for closed group. " )
return
return @Thread deferred . reject ( Error . NoPrivateKey )
}
}
val wasAnyUserRemoved = members . toSet ( ) . intersect ( oldMembers ) != oldMembers . toSet ( )
val wasAnyUserRemoved = members . toSet ( ) . intersect ( oldMembers ) != oldMembers . toSet ( )
val removedMembers = oldMembers . minus ( members )
val removedMembers = oldMembers . minus ( members )
@ -128,16 +138,29 @@ object ClosedGroupsProtocol {
if ( wasAnyUserRemoved ) {
if ( wasAnyUserRemoved ) {
if ( isUserLeaving && removedMembers . count ( ) != 1 ) {
if ( isUserLeaving && removedMembers . count ( ) != 1 ) {
Log . d ( " Loki " , " Can't remove self and others simultaneously. " )
Log . d ( " Loki " , " Can't remove self and others simultaneously. " )
return
return @Thread deferred . reject ( Error . InvalidUpdate )
}
}
// Send the update to the group (don't include new ratchets as everyone should regenerate new ratchets individually)
// Establish sessions if needed
establishSessionsWithMembersIfNeeded ( context , members )
// Send the update to the existing members using established channels (don't include new ratchets as everyone should regenerate new ratchets individually)
for ( member in oldMembers ) {
@Suppress ( " NAME_SHADOWING " )
val closedGroupUpdateKind = ClosedGroupUpdateMessageSendJob . Kind . Info ( Hex . fromStringCondensed ( groupPublicKey ) ,
val closedGroupUpdateKind = ClosedGroupUpdateMessageSendJob . Kind . Info ( Hex . fromStringCondensed ( groupPublicKey ) ,
name , setOf ( ) , membersAsData , adminsAsData )
name , setOf ( ) , membersAsData , adminsAsData )
val job = ClosedGroupUpdateMessageSendJob ( groupPublicKey , closedGroupUpdateKind )
@Suppress ( " NAME_SHADOWING " )
val job = ClosedGroupUpdateMessageSendJob ( member , closedGroupUpdateKind )
job . setContext ( context )
job . setContext ( context )
job . onRun ( ) // Run the job immediately
job . onRun ( ) // Run the job immediately
}
val allOldRatchets = sskDatabase . getAllClosedGroupRatchets ( groupPublicKey , ClosedGroupRatchetCollectionType . Current )
for ( pair in allOldRatchets ) {
val senderPublicKey = pair . first
val ratchet = pair . second
val collection = ClosedGroupRatchetCollectionType . Old
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderPublicKey , ratchet , collection )
}
// Delete all ratchets (it's important that this happens * after * sending out the update)
// Delete all ratchets (it's important that this happens * after * sending out the update)
sskDatabase . removeAllClosedGroupRatchets ( groupPublicKey )
sskDatabase . removeAllClosedGroupRatchets ( groupPublicKey , ClosedGroupRatchetCollectionType . Current )
// Remove the group from the user's set of public keys to poll for if the user is leaving. Otherwise generate a new ratchet and
// Remove the group from the user's set of public keys to poll for if the user is leaving. Otherwise generate a new ratchet and
// send it out to all members (minus the removed ones) using established channels.
// send it out to all members (minus the removed ones) using established channels.
if ( isUserLeaving ) {
if ( isUserLeaving ) {
@ -147,8 +170,6 @@ object ClosedGroupsProtocol {
// Notify the PN server
// Notify the PN server
LokiPushNotificationManager . performOperation ( context , ClosedGroupOperation . Unsubscribe , groupPublicKey , userPublicKey )
LokiPushNotificationManager . performOperation ( context , ClosedGroupOperation . Unsubscribe , groupPublicKey , userPublicKey )
} else {
} else {
// Establish sessions if needed
establishSessionsWithMembersIfNeeded ( context , members )
// Send closed group update messages to any new members using established channels
// Send closed group update messages to any new members using established channels
for ( member in newMembers ) {
for ( member in newMembers ) {
@Suppress ( " NAME_SHADOWING " )
@Suppress ( " NAME_SHADOWING " )
@ -184,7 +205,7 @@ object ClosedGroupsProtocol {
// Establish sessions if needed
// Establish sessions if needed
establishSessionsWithMembersIfNeeded ( context , newMembers )
establishSessionsWithMembersIfNeeded ( context , newMembers )
// Send closed group update messages to the new members using established channels
// Send closed group update messages to the new members using established channels
var allSenderKeys = sskDatabase . getAllClosedGroupSenderKeys ( groupPublicKey );
var allSenderKeys = sskDatabase . getAllClosedGroupSenderKeys ( groupPublicKey , ClosedGroupRatchetCollectionType . Current )
allSenderKeys = allSenderKeys . union ( newSenderKeys )
allSenderKeys = allSenderKeys . union ( newSenderKeys )
for ( member in newMembers ) {
for ( member in newMembers ) {
@Suppress ( " NAME_SHADOWING " )
@Suppress ( " NAME_SHADOWING " )
@ -195,7 +216,7 @@ object ClosedGroupsProtocol {
ApplicationContext . getInstance ( context ) . jobManager . add ( job )
ApplicationContext . getInstance ( context ) . jobManager . add ( job )
}
}
} else {
} else {
val allSenderKeys = sskDatabase . getAllClosedGroupSenderKeys ( groupPublicKey );
val allSenderKeys = sskDatabase . getAllClosedGroupSenderKeys ( groupPublicKey , ClosedGroupRatchetCollectionType . Current )
val closedGroupUpdateKind = ClosedGroupUpdateMessageSendJob . Kind . Info ( Hex . fromStringCondensed ( groupPublicKey ) , name ,
val closedGroupUpdateKind = ClosedGroupUpdateMessageSendJob . Kind . Info ( Hex . fromStringCondensed ( groupPublicKey ) , name ,
allSenderKeys , membersAsData , adminsAsData )
allSenderKeys , membersAsData , adminsAsData )
val job = ClosedGroupUpdateMessageSendJob ( groupPublicKey , closedGroupUpdateKind )
val job = ClosedGroupUpdateMessageSendJob ( groupPublicKey , closedGroupUpdateKind )
@ -208,8 +229,12 @@ object ClosedGroupsProtocol {
groupDB . updateMembers ( groupID , members . map { Address . fromSerialized ( it ) } )
groupDB . updateMembers ( groupID , members . map { Address . fromSerialized ( it ) } )
}
}
// Notify the user
// Notify the user
val infoType = if ( isUserLeaving ) GroupContext . Type . QUIT else GroupContext . Type . UPDATE
val threadID = DatabaseFactory . getThreadDatabase ( context ) . getThreadIdFor ( Recipient . from ( context , Address . fromSerialized ( groupID ) , false ) )
val threadID = DatabaseFactory . getThreadDatabase ( context ) . getThreadIdFor ( Recipient . from ( context , Address . fromSerialized ( groupID ) , false ) )
insertOutgoingInfoMessage ( context , groupID , GroupContext . Type . UPDATE , name , members , admins , threadID )
insertOutgoingInfoMessage ( context , groupID , infoType , name , members , admins , threadID )
deferred . resolve ( Unit )
} . start ( )
return deferred . promise
}
}
@JvmStatic
@JvmStatic
@ -270,7 +295,7 @@ object ClosedGroupsProtocol {
senderKeys . forEach { senderKey ->
senderKeys . forEach { senderKey ->
if ( ! members . contains ( senderKey . publicKey . toHexString ( ) ) ) { return @forEach }
if ( ! members . contains ( senderKey . publicKey . toHexString ( ) ) ) { return @forEach }
val ratchet = ClosedGroupRatchet ( senderKey . chainKey . toHexString ( ) , senderKey . keyIndex , listOf ( ) )
val ratchet = ClosedGroupRatchet ( senderKey . chainKey . toHexString ( ) , senderKey . keyIndex , listOf ( ) )
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderKey . publicKey . toHexString ( ) , ratchet )
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderKey . publicKey . toHexString ( ) , ratchet , ClosedGroupRatchetCollectionType . Current )
}
}
// Sort out any discrepancies between the provided sender keys and what's required
// Sort out any discrepancies between the provided sender keys and what's required
val missingSenderKeys = members . toSet ( ) . subtract ( senderKeys . map { Hex . toStringCondensed ( it . publicKey ) } )
val missingSenderKeys = members . toSet ( ) . subtract ( senderKeys . map { Hex . toStringCondensed ( it . publicKey ) } )
@ -340,7 +365,7 @@ object ClosedGroupsProtocol {
// Store the ratchets for any new members (it's important that this happens before the code below)
// Store the ratchets for any new members (it's important that this happens before the code below)
senderKeys . forEach { senderKey ->
senderKeys . forEach { senderKey ->
val ratchet = ClosedGroupRatchet ( senderKey . chainKey . toHexString ( ) , senderKey . keyIndex , listOf ( ) )
val ratchet = ClosedGroupRatchet ( senderKey . chainKey . toHexString ( ) , senderKey . keyIndex , listOf ( ) )
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderKey . publicKey . toHexString ( ) , ratchet )
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderKey . publicKey . toHexString ( ) , ratchet , ClosedGroupRatchetCollectionType . Current )
}
}
// Delete all ratchets and either:
// Delete all ratchets and either:
// • Send out the user's new ratchet using established channels if other members of the group left or were removed
// • Send out the user's new ratchet using established channels if other members of the group left or were removed
@ -349,7 +374,14 @@ object ClosedGroupsProtocol {
val wasAnyUserRemoved = members . toSet ( ) . intersect ( oldMembers ) != oldMembers . toSet ( )
val wasAnyUserRemoved = members . toSet ( ) . intersect ( oldMembers ) != oldMembers . toSet ( )
val wasSenderRemoved = ! members . contains ( senderPublicKey )
val wasSenderRemoved = ! members . contains ( senderPublicKey )
if ( wasAnyUserRemoved ) {
if ( wasAnyUserRemoved ) {
sskDatabase . removeAllClosedGroupRatchets ( groupPublicKey )
val allOldRatchets = sskDatabase . getAllClosedGroupRatchets ( groupPublicKey , ClosedGroupRatchetCollectionType . Current )
for ( pair in allOldRatchets ) {
@Suppress ( " NAME_SHADOWING " ) val senderPublicKey = pair . first
val ratchet = pair . second
val collection = ClosedGroupRatchetCollectionType . Old
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderPublicKey , ratchet , collection )
}
sskDatabase . removeAllClosedGroupRatchets ( groupPublicKey , ClosedGroupRatchetCollectionType . Current )
if ( wasCurrentUserRemoved ) {
if ( wasCurrentUserRemoved ) {
sskDatabase . removeClosedGroupPrivateKey ( groupPublicKey )
sskDatabase . removeClosedGroupPrivateKey ( groupPublicKey )
groupDB . setActive ( groupID , false )
groupDB . setActive ( groupID , false )
@ -399,7 +431,7 @@ object ClosedGroupsProtocol {
// Respond to the request
// Respond to the request
Log . d ( " Loki " , " Responding to sender key request from: $senderPublicKey . " )
Log . d ( " Loki " , " Responding to sender key request from: $senderPublicKey . " )
ApplicationContext . getInstance ( context ) . sendSessionRequestIfNeeded ( senderPublicKey )
ApplicationContext . getInstance ( context ) . sendSessionRequestIfNeeded ( senderPublicKey )
val userRatchet = DatabaseFactory . getSSKDatabase ( context ) . getClosedGroupRatchet ( groupPublicKey , userPublicKey )
val userRatchet = DatabaseFactory . getSSKDatabase ( context ) . getClosedGroupRatchet ( groupPublicKey , userPublicKey , ClosedGroupRatchetCollectionType . Current )
?: SharedSenderKeysImplementation . shared . generateRatchet ( groupPublicKey , userPublicKey )
?: SharedSenderKeysImplementation . shared . generateRatchet ( groupPublicKey , userPublicKey )
val userSenderKey = ClosedGroupSenderKey ( Hex . fromStringCondensed ( userRatchet . chainKey ) , userRatchet . keyIndex , Hex . fromStringCondensed ( userPublicKey ) )
val userSenderKey = ClosedGroupSenderKey ( Hex . fromStringCondensed ( userRatchet . chainKey ) , userRatchet . keyIndex , Hex . fromStringCondensed ( userPublicKey ) )
val closedGroupUpdateKind = ClosedGroupUpdateMessageSendJob . Kind . SenderKey ( Hex . fromStringCondensed ( groupPublicKey ) , userSenderKey )
val closedGroupUpdateKind = ClosedGroupUpdateMessageSendJob . Kind . SenderKey ( Hex . fromStringCondensed ( groupPublicKey ) , userSenderKey )
@ -424,7 +456,7 @@ object ClosedGroupsProtocol {
// Store the sender key
// Store the sender key
Log . d ( " Loki " , " Received a sender key from: $senderPublicKey . " )
Log . d ( " Loki " , " Received a sender key from: $senderPublicKey . " )
val ratchet = ClosedGroupRatchet ( senderKey . chainKey . toHexString ( ) , senderKey . keyIndex , listOf ( ) )
val ratchet = ClosedGroupRatchet ( senderKey . chainKey . toHexString ( ) , senderKey . keyIndex , listOf ( ) )
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderPublicKey , ratchet )
sskDatabase . setClosedGroupRatchet ( groupPublicKey , senderPublicKey , ratchet , ClosedGroupRatchetCollectionType . Current )
}
}
@JvmStatic
@JvmStatic