Skip to content

Commit

Permalink
[1+IDENTITY_KEY] add docs to identity_key.rs
Browse files Browse the repository at this point in the history
  • Loading branch information
cosmicexplorer committed Jun 27, 2022
1 parent 2a46a5b commit 7674fc4
Showing 1 changed file with 33 additions and 2 deletions.
35 changes: 33 additions & 2 deletions rust/protocol/src/identity_key.rs
Original file line number Diff line number Diff line change
@@ -1,11 +1,20 @@
//
// Copyright 2020 Signal Messenger, LLC.
// Copyright 2020-2022 Signal Messenger, LLC.
// SPDX-License-Identifier: AGPL-3.0-only
//

//! Wrappers over cryptographic primitives from [crate::curve] to represent [identity].
//!
//! [identity]: https://en.wikipedia.org/wiki/Online_identity
#![warn(missing_docs)]

use crate::proto;
use crate::{KeyPair, PrivateKey, PublicKey, Result, SignalProtocolError};

#[cfg(doc)]
use crate::{protocol::PreKeySignalMessage, state::SessionRecord, storage::IdentityKeyStore};

use rand::{CryptoRng, Rng};
use std::convert::TryFrom;

Expand All @@ -15,31 +24,39 @@ use prost::Message;
const ALTERNATE_IDENTITY_SIGNATURE_PREFIX_1: &[u8] = &[0xFF; 32];
const ALTERNATE_IDENTITY_SIGNATURE_PREFIX_2: &[u8] = b"Signal_PNI_Signature";

/// The public identity of a user, used in [IdentityKeyStore].
///
/// Wrapper for [PublicKey].
#[derive(Debug, PartialOrd, Ord, PartialEq, Eq, Clone, Copy)]
pub struct IdentityKey {
public_key: PublicKey,
}

impl IdentityKey {
/// Initialize a public-facing identity from a public key.
pub fn new(public_key: PublicKey) -> Self {
Self { public_key }
}

/// Return a public key representing the public identity.
#[inline]
pub fn public_key(&self) -> &PublicKey {
&self.public_key
}

/// Return an owned byte slice which can be deserialized with [Self::decode].
#[inline]
pub fn serialize(&self) -> Box<[u8]> {
self.public_key.serialize()
}

/// Deserialize a public identity from a byte slice.
pub fn decode(value: &[u8]) -> Result<Self> {
let pk = PublicKey::try_from(value)?;
Ok(Self { public_key: pk })
}

/// Verify `signature` matches a provided identity key `other` (*not* `self`)!
pub fn verify_alternate_identity(&self, other: &IdentityKey, signature: &[u8]) -> Result<bool> {
self.public_key.verify_signature_for_multipart_message(
&[
Expand Down Expand Up @@ -72,20 +89,29 @@ impl From<IdentityKey> for PublicKey {
}
}

#[derive(Copy, Clone)]
/// The private identity of a user.
///
/// This cryptographic identity is used to sign pre-keys in [PreKeySignalMessage], to conceal
/// sealed-sender [crate::sealed_sender::UnidentifiedSenderMessage]s, or to initiate the key
/// derivation function for a [SessionRecord].
///
/// Also see [KeyPair].
#[derive(Copy, Clone, Debug)]
pub struct IdentityKeyPair {
identity_key: IdentityKey,
private_key: PrivateKey,
}

impl IdentityKeyPair {
/// Create a key pair from a public `identity_key` and a private `private_key`.
pub fn new(identity_key: IdentityKey, private_key: PrivateKey) -> Self {
Self {
identity_key,
private_key,
}
}

/// Generate a random new identity from randomness in `csprng`.
pub fn generate<R: CryptoRng + Rng>(csprng: &mut R) -> Self {
let keypair = KeyPair::generate(csprng);

Expand All @@ -95,21 +121,25 @@ impl IdentityKeyPair {
}
}

/// Return the public identity of this user.
#[inline]
pub fn identity_key(&self) -> &IdentityKey {
&self.identity_key
}

/// Return the public key that defines this identity.
#[inline]
pub fn public_key(&self) -> &PublicKey {
self.identity_key.public_key()
}

/// Return the private key that defines this identity.
#[inline]
pub fn private_key(&self) -> &PrivateKey {
&self.private_key
}

/// Return a byte slice which can later be deserialized with [Self::try_from].
pub fn serialize(&self) -> Box<[u8]> {
let structure = proto::storage::IdentityKeyPairStructure {
public_key: self.identity_key.serialize().to_vec(),
Expand All @@ -120,6 +150,7 @@ impl IdentityKeyPair {
result.into_boxed_slice()
}

/// Generate a signature for a provided identity key `other` (*not* `self`)!
pub fn sign_alternate_identity<R: Rng + CryptoRng>(
&self,
other: &IdentityKey,
Expand Down

0 comments on commit 7674fc4

Please sign in to comment.