Line data    Source code

       1              : //! Tools for client/server/stored key management.
       2              : 
       3              : /// Faithfully taken from PostgreSQL.
       4              : pub const SCRAM_KEY_LEN: usize = 32;
       5              : 
       6              : /// One of the keys derived from the user's password.
       7              : /// We use the same structure for all keys, i.e.
       8              : /// `ClientKey`, `StoredKey`, and `ServerKey`.
       9           48 : #[derive(Clone, Default, PartialEq, Eq, Debug)]
      10              : #[repr(transparent)]
      11              : pub struct ScramKey {
      12              :     bytes: [u8; SCRAM_KEY_LEN],
      13              : }
      14              : 
      15              : impl ScramKey {
      16           18 :     pub fn sha256(&self) -> Self {
      17           18 :         super::sha256([self.as_ref()]).into()
      18           18 :     }
      19              : 
      20            2 :     pub fn as_bytes(&self) -> [u8; SCRAM_KEY_LEN] {
      21            2 :         self.bytes
      22            2 :     }
      23              : }
      24              : 
      25              : impl From<[u8; SCRAM_KEY_LEN]> for ScramKey {
      26              :     #[inline(always)]
      27           96 :     fn from(bytes: [u8; SCRAM_KEY_LEN]) -> Self {
      28           96 :         Self { bytes }
      29           96 :     }
      30              : }
      31              : 
      32              : impl AsRef<[u8]> for ScramKey {
      33              :     #[inline(always)]
      34           58 :     fn as_ref(&self) -> &[u8] {
      35           58 :         &self.bytes
      36           58 :     }
      37              : }