private MessageKeys GetOrCreateMessageKeys(SessionState sessionState,
IEcPublicKey theirEphemeral,
ChainKey chainKey, uint counter)
{
if (chainKey.GetIndex() > counter)
{
if (sessionState.HasMessageKeys(theirEphemeral, counter))
{
return(sessionState.RemoveMessageKeys(theirEphemeral, counter));
}
else
{
throw new DuplicateMessageException($"Received message with old counter: {chainKey.GetIndex()} , {counter}");
}
}
//Avoiding a uint overflow
uint chainKeyIndex = chainKey.GetIndex();
if ((counter > chainKeyIndex) && (counter - chainKeyIndex > 2000))
{
throw new InvalidMessageException("Over 2000 messages into the future!");
}
while (chainKey.GetIndex() < counter)
{
MessageKeys messageKeys = chainKey.GetMessageKeys();
sessionState.SetMessageKeys(theirEphemeral, messageKeys);
chainKey = chainKey.GetNextChainKey();
}
sessionState.SetReceiverChainKey(theirEphemeral, chainKey.GetNextChainKey());
return(chainKey.GetMessageKeys());
}
public void TestChainKeyDerivationV3()
{
byte[] seed =
{
0x8a, 0xb7, 0x2d, 0x6f, 0x4c,
0xc5, 0xac, 0x0d, 0x38, 0x7e,
0xaf, 0x46, 0x33, 0x78, 0xdd,
0xb2, 0x8e, 0xdd, 0x07, 0x38,
0x5b, 0x1c, 0xb0, 0x12, 0x50,
0xc7, 0x15, 0x98, 0x2e, 0x7a,
0xd4, 0x8f
};
byte[] messageKey =
{
/* 0x02*/
0xbf, 0x51, 0xe9, 0xd7,
0x5e, 0x0e, 0x31, 0x03, 0x10,
0x51, 0xf8, 0x2a, 0x24, 0x91,
0xff, 0xc0, 0x84, 0xfa, 0x29,
0x8b, 0x77, 0x93, 0xbd, 0x9d,
0xb6, 0x20, 0x05, 0x6f, 0xeb,
0xf4, 0x52, 0x17
};
byte[] macKey =
{
0xc6, 0xc7, 0x7d, 0x6a, 0x73,
0xa3, 0x54, 0x33, 0x7a, 0x56,
0x43, 0x5e, 0x34, 0x60, 0x7d,
0xfe, 0x48, 0xe3, 0xac, 0xe1,
0x4e, 0x77, 0x31, 0x4d, 0xc6,
0xab, 0xc1, 0x72, 0xe7, 0xa7,
0x03, 0x0b
};
byte[] nextChainKey =
{
0x28, 0xe8, 0xf8, 0xfe, 0xe5,
0x4b, 0x80, 0x1e, 0xef, 0x7c,
0x5c, 0xfb, 0x2f, 0x17, 0xf3,
0x2c, 0x7b, 0x33, 0x44, 0x85,
0xbb, 0xb7, 0x0f, 0xac, 0x6e,
0xc1, 0x03, 0x42, 0xa2, 0x46,
0xd1, 0x5d
};
ChainKey chainKey = new ChainKey(Hkdf.CreateFor(3), seed, 0);
CollectionAssert.AreEqual(seed, chainKey.GetKey());
CollectionAssert.AreEqual(messageKey, chainKey.GetMessageKeys().GetCipherKey());
CollectionAssert.AreEqual(macKey, chainKey.GetMessageKeys().GetMacKey());
CollectionAssert.AreEqual(nextChainKey, chainKey.GetNextChainKey().GetKey());
Assert.AreEqual <uint>(0, chainKey.GetIndex());
Assert.AreEqual <uint>(0, chainKey.GetMessageKeys().GetCounter());
Assert.AreEqual <uint>(1, chainKey.GetNextChainKey().GetIndex());
Assert.AreEqual <uint>(1, chainKey.GetNextChainKey().GetMessageKeys().GetCounter());
}
public void TestChainKeyDerivationV2()
{
byte[] seed =
{
0x8a, 0xb7, 0x2d, 0x6f, 0x4c,
0xc5, 0xac, 0x0d, 0x38, 0x7e,
0xaf, 0x46, 0x33, 0x78, 0xdd,
0xb2, 0x8e, 0xdd, 0x07, 0x38,
0x5b, 0x1c, 0xb0, 0x12, 0x50,
0xc7, 0x15, 0x98, 0x2e, 0x7a,
0xd4, 0x8f
};
byte[] messageKey =
{
0x02, 0xa9, 0xaa, 0x6c, 0x7d,
0xbd, 0x64, 0xf9, 0xd3, 0xaa,
0x92, 0xf9, 0x2a, 0x27, 0x7b,
0xf5, 0x46, 0x09, 0xda, 0xdf,
0x0b, 0x00, 0x82, 0x8a, 0xcf,
0xc6, 0x1e, 0x3c, 0x72, 0x4b,
0x84, 0xa7
};
byte[] macKey =
{
0xbf, 0xbe, 0x5e, 0xfb, 0x60,
0x30, 0x30, 0x52, 0x67, 0x42,
0xe3, 0xee, 0x89, 0xc7, 0x02,
0x4e, 0x88, 0x4e, 0x44, 0x0f,
0x1f, 0xf3, 0x76, 0xbb, 0x23,
0x17, 0xb2, 0xd6, 0x4d, 0xeb,
0x7c, 0x83
};
byte[] nextChainKey =
{
0x28, 0xe8, 0xf8, 0xfe, 0xe5,
0x4b, 0x80, 0x1e, 0xef, 0x7c,
0x5c, 0xfb, 0x2f, 0x17, 0xf3,
0x2c, 0x7b, 0x33, 0x44, 0x85,
0xbb, 0xb7, 0x0f, 0xac, 0x6e,
0xc1, 0x03, 0x42, 0xa2, 0x46,
0xd1, 0x5d
};
ChainKey chainKey = new ChainKey(Hkdf.CreateFor(2), seed, 0);
Assert.AreEqual(seed, chainKey.GetKey());
CollectionAssert.AreEqual(messageKey, chainKey.GetMessageKeys().GetCipherKey());
CollectionAssert.AreEqual(macKey, chainKey.GetMessageKeys().GetMacKey());
CollectionAssert.AreEqual(nextChainKey, chainKey.GetNextChainKey().GetKey());
Assert.AreEqual <uint>(0, chainKey.GetIndex());
Assert.AreEqual <uint>(0, chainKey.GetMessageKeys().GetCounter());
Assert.AreEqual <uint>(1, chainKey.GetNextChainKey().GetIndex());
Assert.AreEqual <uint>(1, chainKey.GetNextChainKey().GetMessageKeys().GetCounter());
}
public void SetSenderChainKey(ChainKey nextChainKey)
{
SessionStructure.Types.Chain.Types.ChainKey chainKey = new SessionStructure.Types.Chain.Types.ChainKey
{
Key = ByteString.CopyFrom(nextChainKey.GetKey()),
Index = nextChainKey.GetIndex()
};
_sessionStructure.SenderChain.ChainKey = chainKey;
}
public void SetSenderChainKey(ChainKey nextChainKey)
{
Chain.Types.ChainKey chainKey = Chain.Types.ChainKey.CreateBuilder()
.SetKey(ByteString.CopyFrom(nextChainKey.GetKey()))
.SetIndex(nextChainKey.GetIndex())
.Build();
Chain chain = sessionStructure.SenderChain.ToBuilder()
.SetChainKey(chainKey).Build();
this.sessionStructure = this.sessionStructure.ToBuilder().SetSenderChain(chain).Build();
}
public void SetSenderChain(ECKeyPair senderRatchetKeyPair, ChainKey chainKey)
{
Chain.Types.ChainKey chainKeyStructure = Chain.Types.ChainKey.CreateBuilder()
.SetKey(ByteString.CopyFrom(chainKey.GetKey()))
.SetIndex(chainKey.GetIndex())
.Build();
Chain senderChain = Chain.CreateBuilder()
.SetSenderRatchetKey(ByteString.CopyFrom(senderRatchetKeyPair.GetPublicKey().Serialize()))
.SetSenderRatchetKeyPrivate(ByteString.CopyFrom(senderRatchetKeyPair.GetPrivateKey().Serialize()))
.SetChainKey(chainKeyStructure)
.Build();
this.sessionStructure = this.sessionStructure.ToBuilder().SetSenderChain(senderChain).Build();
}
public void SetReceiverChainKey(ECPublicKey senderEphemeral, ChainKey chainKey)
{
Pair <Chain, uint> chainAndIndex = GetReceiverChain(senderEphemeral);
Chain chain = chainAndIndex.First();
Chain.Types.ChainKey chainKeyStructure = Chain.Types.ChainKey.CreateBuilder()
.SetKey(ByteString.CopyFrom(chainKey.GetKey()))
.SetIndex(chainKey.GetIndex())
.Build();
Chain updatedChain = chain.ToBuilder().SetChainKey(chainKeyStructure).Build();
this.sessionStructure = this.sessionStructure.ToBuilder()
.SetReceiverChains((int)chainAndIndex.Second(), updatedChain) // TODO: conv
.Build();
}
public void SetReceiverChainKey(IEcPublicKey senderEphemeral, ChainKey chainKey)
{
Pair <SessionStructure.Types.Chain, uint> chainAndIndex = GetReceiverChain(senderEphemeral);
SessionStructure.Types.Chain chain = chainAndIndex.First();
SessionStructure.Types.Chain.Types.ChainKey chainKeyStructure = new SessionStructure.Types.Chain.Types.ChainKey
{
Key = ByteString.CopyFrom(chainKey.GetKey()),
Index = chainKey.GetIndex()
};
chain.ChainKey = chainKeyStructure;
_sessionStructure.ReceiverChains[(int)chainAndIndex.Second()] = chain;
}
public void SetSenderChain(EcKeyPair senderRatchetKeyPair, ChainKey chainKey)
{
SessionStructure.Types.Chain.Types.ChainKey chainKeyStructure = new SessionStructure.Types.Chain.Types.ChainKey
{
Key = ByteString.CopyFrom(chainKey.GetKey()),
Index = chainKey.GetIndex()
};
SessionStructure.Types.Chain senderChain = new SessionStructure.Types.Chain
{
SenderRatchetKey = ByteString.CopyFrom(senderRatchetKeyPair.GetPublicKey().Serialize()),
SenderRatchetKeyPrivate = ByteString.CopyFrom(senderRatchetKeyPair.GetPrivateKey().Serialize()),
ChainKey = chainKeyStructure
};
_sessionStructure.SenderChain = senderChain;
}
public void AddReceiverChain(ECPublicKey senderRatchetKey, ChainKey chainKey)
{
Chain.Types.ChainKey chainKeyStructure = Chain.Types.ChainKey.CreateBuilder()
.SetKey(ByteString.CopyFrom(chainKey.GetKey()))
.SetIndex(chainKey.GetIndex())
.Build();
Chain chain = Chain.CreateBuilder()
.SetChainKey(chainKeyStructure)
.SetSenderRatchetKey(ByteString.CopyFrom(senderRatchetKey.Serialize()))
.Build();
this.sessionStructure = this.sessionStructure.ToBuilder().AddReceiverChains(chain).Build();
if (this.sessionStructure.ReceiverChainsList.Count > 5)
{
this.sessionStructure = this.sessionStructure.ToBuilder() /*.ClearReceiverChains()*/.Build(); //RemoveReceiverChains(0) TODO: why does it work without
}
}
public void AddReceiverChain(IEcPublicKey senderRatchetKey, ChainKey chainKey)
{
SessionStructure.Types.Chain.Types.ChainKey chainKeyStructure = new SessionStructure.Types.Chain.Types.ChainKey
{
Key = ByteString.CopyFrom(chainKey.GetKey()),
Index = chainKey.GetIndex()
};
SessionStructure.Types.Chain chain = new SessionStructure.Types.Chain
{
ChainKey = chainKeyStructure,
SenderRatchetKey = ByteString.CopyFrom(senderRatchetKey.Serialize())
};
_sessionStructure.ReceiverChains.Add(chain);
while (_sessionStructure.ReceiverChains.Count > 5)
{
_sessionStructure.ReceiverChains.RemoveAt(0); //TODO why was here a TODO?
}
}
public void SetSenderChainKey(ChainKey nextChainKey)
{
StorageProtos.SessionStructure.Types.Chain.Types.ChainKey chainKey = StorageProtos.SessionStructure.Types.Chain.Types.ChainKey.CreateBuilder()
.SetKey(ByteString.CopyFrom(nextChainKey.GetKey()))
.SetIndex(nextChainKey.GetIndex())
.Build();
StorageProtos.SessionStructure.Types.Chain chain = _sessionStructure.SenderChain.ToBuilder()
.SetChainKey(chainKey).Build();
_sessionStructure = _sessionStructure.ToBuilder().SetSenderChain(chain).Build();
}
public void SetSenderChain(EcKeyPair senderRatchetKeyPair, ChainKey chainKey)
{
StorageProtos.SessionStructure.Types.Chain.Types.ChainKey chainKeyStructure = StorageProtos.SessionStructure.Types.Chain.Types.ChainKey.CreateBuilder()
.SetKey(ByteString.CopyFrom(chainKey.GetKey()))
.SetIndex(chainKey.GetIndex())
.Build();
StorageProtos.SessionStructure.Types.Chain senderChain = StorageProtos.SessionStructure.Types.Chain.CreateBuilder()
.SetSenderRatchetKey(ByteString.CopyFrom(senderRatchetKeyPair.GetPublicKey().Serialize()))
.SetSenderRatchetKeyPrivate(ByteString.CopyFrom(senderRatchetKeyPair.GetPrivateKey().Serialize()))
.SetChainKey(chainKeyStructure)
.Build();
_sessionStructure = _sessionStructure.ToBuilder().SetSenderChain(senderChain).Build();
}
/**
* Encrypt a message.
*
* @param paddedMessage The plaintext message bytes, optionally padded to a constant multiple.
* @return A ciphertext message encrypted to the recipient+device tuple.
*/
public CiphertextMessage Encrypt(byte[] paddedMessage)
{
lock (SessionLock)
{
SessionRecord sessionRecord = _sessionStore.LoadSession(_remoteAddress);
SessionState sessionState = sessionRecord.GetSessionState();
ChainKey chainKey = sessionState.GetSenderChainKey();
MessageKeys messageKeys = chainKey.GetMessageKeys();
IEcPublicKey senderEphemeral = sessionState.GetSenderRatchetKey();
uint previousCounter = sessionState.GetPreviousCounter();
uint sessionVersion = sessionState.GetSessionVersion();
byte[] ciphertextBody = GetCiphertext(sessionVersion, messageKeys, paddedMessage);
CiphertextMessage ciphertextMessage = new SignalMessage(sessionVersion, messageKeys.GetMacKey(),
senderEphemeral, chainKey.GetIndex(),
previousCounter, ciphertextBody,
sessionState.GetLocalIdentityKey(),
sessionState.GetRemoteIdentityKey());
if (sessionState.HasUnacknowledgedPreKeyMessage())
{
SessionState.UnacknowledgedPreKeyMessageItems items = sessionState.GetUnacknowledgedPreKeyMessageItems();
uint localRegistrationId = sessionState.GetLocalRegistrationId();
ciphertextMessage = new PreKeySignalMessage(sessionVersion, localRegistrationId, items.GetPreKeyId(),
items.GetSignedPreKeyId(), items.GetBaseKey(),
sessionState.GetLocalIdentityKey(),
(SignalMessage)ciphertextMessage);
}
sessionState.SetSenderChainKey(chainKey.GetNextChainKey());
_sessionStore.StoreSession(_remoteAddress, sessionRecord);
return(ciphertextMessage);
}
}