Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion ssh-key/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -177,7 +177,7 @@ pub use sha2;
pub use crate::{
certificate::Certificate,
known_hosts::KnownHosts,
mpint::MPInt,
mpint::Mpint,
signature::{Signature, SigningKey},
sshsig::SshSig,
};
Expand Down
72 changes: 36 additions & 36 deletions ssh-key/src/mpint.rs
Original file line number Diff line number Diff line change
Expand Up @@ -39,12 +39,12 @@ use zeroize::Zeroizing;
/// |-1234 | `00 00 00 02 ed cc`
/// | -deadbeef | `00 00 00 05 ff 21 52 41 11`
#[derive(Clone, PartialOrd, Ord)]
pub struct MPInt {
pub struct Mpint {
/// Inner big endian-serialized integer value
inner: Vec<u8>,
}

impl MPInt {
impl Mpint {
/// Create a new multiple precision integer from the given
/// big endian-encoded byte slice.
///
Expand Down Expand Up @@ -74,7 +74,7 @@ impl MPInt {
/// Get the big integer data encoded as big endian bytes.
///
/// This slice will contain a leading zero if the value is positive but the
/// MSB is also set. Use [`MPInt::as_positive_bytes`] to ensure the number
/// MSB is also set. Use [`Mpint::as_positive_bytes`] to ensure the number
/// is positive and strip the leading zero byte if it exists.
pub fn as_bytes(&self) -> &[u8] {
&self.inner
Expand All @@ -94,35 +94,35 @@ impl MPInt {
}
}

impl AsRef<[u8]> for MPInt {
impl AsRef<[u8]> for Mpint {
fn as_ref(&self) -> &[u8] {
self.as_bytes()
}
}

impl ConstantTimeEq for MPInt {
impl ConstantTimeEq for Mpint {
fn ct_eq(&self, other: &Self) -> Choice {
self.as_ref().ct_eq(other.as_ref())
}
}

impl Eq for MPInt {}
impl Eq for Mpint {}

impl PartialEq for MPInt {
impl PartialEq for Mpint {
fn eq(&self, other: &Self) -> bool {
self.ct_eq(other).into()
}
}

impl Decode for MPInt {
impl Decode for Mpint {
type Error = Error;

fn decode(reader: &mut impl Reader) -> Result<Self> {
Vec::decode(reader)?.try_into()
}
}

impl Encode for MPInt {
impl Encode for Mpint {
type Error = Error;

fn encoded_len(&self) -> Result<usize> {
Expand All @@ -135,15 +135,15 @@ impl Encode for MPInt {
}
}

impl TryFrom<&[u8]> for MPInt {
impl TryFrom<&[u8]> for Mpint {
type Error = Error;

fn try_from(bytes: &[u8]) -> Result<Self> {
Vec::from(bytes).try_into()
}
}

impl TryFrom<Vec<u8>> for MPInt {
impl TryFrom<Vec<u8>> for Mpint {
type Error = Error;

fn try_from(bytes: Vec<u8>) -> Result<Self> {
Expand All @@ -157,25 +157,25 @@ impl TryFrom<Vec<u8>> for MPInt {
}
}

impl Zeroize for MPInt {
impl Zeroize for Mpint {
fn zeroize(&mut self) {
self.inner.zeroize();
}
}

impl fmt::Debug for MPInt {
impl fmt::Debug for Mpint {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "MPInt({self:X})")
write!(f, "Mpint({self:X})")
}
}

impl fmt::Display for MPInt {
impl fmt::Display for Mpint {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{self:X}")
}
}

impl fmt::LowerHex for MPInt {
impl fmt::LowerHex for Mpint {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
for byte in self.as_bytes() {
write!(f, "{byte:02x}")?;
Expand All @@ -184,7 +184,7 @@ impl fmt::LowerHex for MPInt {
}
}

impl fmt::UpperHex for MPInt {
impl fmt::UpperHex for Mpint {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
for byte in self.as_bytes() {
write!(f, "{byte:02X}")?;
Expand All @@ -194,38 +194,38 @@ impl fmt::UpperHex for MPInt {
}

#[cfg(any(feature = "dsa", feature = "rsa"))]
impl TryFrom<bigint::BigUint> for MPInt {
impl TryFrom<bigint::BigUint> for Mpint {
type Error = Error;

fn try_from(uint: bigint::BigUint) -> Result<MPInt> {
MPInt::try_from(&uint)
fn try_from(uint: bigint::BigUint) -> Result<Mpint> {
Mpint::try_from(&uint)
}
}

#[cfg(any(feature = "dsa", feature = "rsa"))]
impl TryFrom<&bigint::BigUint> for MPInt {
impl TryFrom<&bigint::BigUint> for Mpint {
type Error = Error;

fn try_from(uint: &bigint::BigUint) -> Result<MPInt> {
fn try_from(uint: &bigint::BigUint) -> Result<Mpint> {
let bytes = Zeroizing::new(uint.to_bytes_be());
MPInt::from_positive_bytes(bytes.as_slice())
Mpint::from_positive_bytes(bytes.as_slice())
}
}

#[cfg(any(feature = "dsa", feature = "rsa"))]
impl TryFrom<MPInt> for bigint::BigUint {
impl TryFrom<Mpint> for bigint::BigUint {
type Error = Error;

fn try_from(mpint: MPInt) -> Result<bigint::BigUint> {
fn try_from(mpint: Mpint) -> Result<bigint::BigUint> {
bigint::BigUint::try_from(&mpint)
}
}

#[cfg(any(feature = "dsa", feature = "rsa"))]
impl TryFrom<&MPInt> for bigint::BigUint {
impl TryFrom<&Mpint> for bigint::BigUint {
type Error = Error;

fn try_from(mpint: &MPInt) -> Result<bigint::BigUint> {
fn try_from(mpint: &Mpint) -> Result<bigint::BigUint> {
mpint
.as_positive_bytes()
.map(bigint::BigUint::from_bytes_be)
Expand All @@ -235,30 +235,30 @@ impl TryFrom<&MPInt> for bigint::BigUint {

#[cfg(test)]
mod tests {
use super::MPInt;
use super::Mpint;
use hex_literal::hex;

#[test]
fn decode_0() {
let n = MPInt::from_bytes(b"").unwrap();
let n = Mpint::from_bytes(b"").unwrap();
assert_eq!(b"", n.as_bytes())
}

#[test]
fn reject_extra_leading_zeroes() {
assert!(MPInt::from_bytes(&hex!("00")).is_err());
assert!(MPInt::from_bytes(&hex!("00 00")).is_err());
assert!(MPInt::from_bytes(&hex!("00 01")).is_err());
assert!(Mpint::from_bytes(&hex!("00")).is_err());
assert!(Mpint::from_bytes(&hex!("00 00")).is_err());
assert!(Mpint::from_bytes(&hex!("00 01")).is_err());
}

#[test]
fn decode_9a378f9b2e332a7() {
assert!(MPInt::from_bytes(&hex!("09 a3 78 f9 b2 e3 32 a7")).is_ok());
assert!(Mpint::from_bytes(&hex!("09 a3 78 f9 b2 e3 32 a7")).is_ok());
}

#[test]
fn decode_80() {
let n = MPInt::from_bytes(&hex!("00 80")).unwrap();
let n = Mpint::from_bytes(&hex!("00 80")).unwrap();

// Leading zero stripped
assert_eq!(&hex!("80"), n.as_positive_bytes().unwrap())
Expand All @@ -267,14 +267,14 @@ mod tests {
// TODO(tarcieri): drop support for negative numbers?
#[test]
fn decode_neg_1234() {
let n = MPInt::from_bytes(&hex!("ed cc")).unwrap();
let n = Mpint::from_bytes(&hex!("ed cc")).unwrap();
assert!(n.as_positive_bytes().is_none());
}

// TODO(tarcieri): drop support for negative numbers?
#[test]
fn decode_neg_deadbeef() {
let n = MPInt::from_bytes(&hex!("ff 21 52 41 11")).unwrap();
let n = Mpint::from_bytes(&hex!("ff 21 52 41 11")).unwrap();
assert!(n.as_positive_bytes().is_none());
}
}
10 changes: 5 additions & 5 deletions ssh-key/src/private/dsa.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
//! Digital Signature Algorithm (DSA) private keys.

use crate::{public::DsaPublicKey, Error, MPInt, Result};
use crate::{public::DsaPublicKey, Error, Mpint, Result};
use core::fmt;
use encoding::{CheckedSum, Decode, Encode, Reader, Writer};
use subtle::{Choice, ConstantTimeEq};
Expand All @@ -18,7 +18,7 @@ use rand_core::CryptoRngCore;
#[derive(Clone)]
pub struct DsaPrivateKey {
/// Integer representing a DSA private key.
inner: MPInt,
inner: Mpint,
}

impl DsaPrivateKey {
Expand All @@ -27,8 +27,8 @@ impl DsaPrivateKey {
self.inner.as_bytes()
}

/// Get the inner [`MPInt`].
pub fn as_mpint(&self) -> &MPInt {
/// Get the inner [`Mpint`].
pub fn as_mpint(&self) -> &Mpint {
&self.inner
}
}
Expand Down Expand Up @@ -58,7 +58,7 @@ impl Decode for DsaPrivateKey {

fn decode(reader: &mut impl Reader) -> Result<Self> {
Ok(Self {
inner: MPInt::decode(reader)?,
inner: Mpint::decode(reader)?,
})
}
}
Expand Down
22 changes: 11 additions & 11 deletions ssh-key/src/private/rsa.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
//! Rivest–Shamir–Adleman (RSA) private keys.

use crate::{public::RsaPublicKey, Error, MPInt, Result};
use crate::{public::RsaPublicKey, Error, Mpint, Result};
use core::fmt;
use encoding::{CheckedSum, Decode, Encode, Reader, Writer};
use subtle::{Choice, ConstantTimeEq};
Expand All @@ -17,16 +17,16 @@ use {
#[derive(Clone)]
pub struct RsaPrivateKey {
/// RSA private exponent.
pub d: MPInt,
pub d: Mpint,

/// CRT coefficient: `(inverse of q) mod p`.
pub iqmp: MPInt,
pub iqmp: Mpint,

/// First prime factor of `n`.
pub p: MPInt,
pub p: Mpint,

/// Second prime factor of `n`.
pub q: MPInt,
pub q: Mpint,
}

impl ConstantTimeEq for RsaPrivateKey {
Expand All @@ -50,10 +50,10 @@ impl Decode for RsaPrivateKey {
type Error = Error;

fn decode(reader: &mut impl Reader) -> Result<Self> {
let d = MPInt::decode(reader)?;
let iqmp = MPInt::decode(reader)?;
let p = MPInt::decode(reader)?;
let q = MPInt::decode(reader)?;
let d = Mpint::decode(reader)?;
let iqmp = Mpint::decode(reader)?;
let p = Mpint::decode(reader)?;
let q = Mpint::decode(reader)?;
Ok(Self { d, iqmp, p, q })
}
}
Expand Down Expand Up @@ -133,8 +133,8 @@ impl Decode for RsaKeypair {
type Error = Error;

fn decode(reader: &mut impl Reader) -> Result<Self> {
let n = MPInt::decode(reader)?;
let e = MPInt::decode(reader)?;
let n = Mpint::decode(reader)?;
let e = Mpint::decode(reader)?;
let public = RsaPublicKey { n, e };
let private = RsaPrivateKey::decode(reader)?;
Ok(RsaKeypair { public, private })
Expand Down
18 changes: 9 additions & 9 deletions ssh-key/src/public/dsa.rs
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
//! Digital Signature Algorithm (DSA) public keys.

use crate::{Error, MPInt, Result};
use crate::{Error, Mpint, Result};
use encoding::{CheckedSum, Decode, Encode, Reader, Writer};

/// Digital Signature Algorithm (DSA) public key.
Expand All @@ -9,27 +9,27 @@ use encoding::{CheckedSum, Decode, Encode, Reader, Writer};
#[derive(Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
pub struct DsaPublicKey {
/// Prime modulus.
pub p: MPInt,
pub p: Mpint,

/// Prime divisor of `p - 1`.
pub q: MPInt,
pub q: Mpint,

/// Generator of a subgroup of order `q` in the multiplicative group
/// `GF(p)`, such that `1 < g < p`.
pub g: MPInt,
pub g: Mpint,

/// The public key, where `y = gˣ mod p`.
pub y: MPInt,
pub y: Mpint,
}

impl Decode for DsaPublicKey {
type Error = Error;

fn decode(reader: &mut impl Reader) -> Result<Self> {
let p = MPInt::decode(reader)?;
let q = MPInt::decode(reader)?;
let g = MPInt::decode(reader)?;
let y = MPInt::decode(reader)?;
let p = Mpint::decode(reader)?;
let q = Mpint::decode(reader)?;
let g = Mpint::decode(reader)?;
let y = Mpint::decode(reader)?;
Ok(Self { p, q, g, y })
}
}
Expand Down
Loading