Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
heliannuuthus committed Jul 15, 2024
1 parent 56cee64 commit 9c53484
Showing 1 changed file with 124 additions and 0 deletions.
124 changes: 124 additions & 0 deletions jose-jwk/src/crypto/k256.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,124 @@
// SPDX-FileCopyrightText: 2022 Profian Inc. <[email protected]>
// SPDX-License-Identifier: Apache-2.0 OR MIT

#![cfg(feature = "k256")]

use k256::elliptic_curve::sec1::{FromEncodedPoint, ToEncodedPoint};
use k256::{EncodedPoint, FieldBytes, PublicKey, SecretKey};

use jose_jwa::{Algorithm, Algorithm::Signing, Signing::*};

use super::Error;
use super::KeyInfo;
use crate::{Ec, EcCurves};

impl KeyInfo for PublicKey {
fn strength(&self) -> usize {
16
}

fn is_supported(&self, algo: &Algorithm) -> bool {
matches!(algo, Signing(Es256K))
}
}

impl KeyInfo for SecretKey {
fn strength(&self) -> usize {
16
}

fn is_supported(&self, algo: &Algorithm) -> bool {
matches!(algo, Signing(Es256K))
}
}

impl From<&PublicKey> for Ec {
fn from(pk: &PublicKey) -> Self {
let ep = pk.to_encoded_point(false);

Self {
crv: EcCurves::P256K,
x: ep.x().expect("unreachable").to_vec().into(),
y: ep.y().expect("unreachable").to_vec().into(),
d: None,
}
}
}

impl From<PublicKey> for Ec {
fn from(sk: PublicKey) -> Self {
(&sk).into()
}
}

impl TryFrom<&Ec> for PublicKey {
type Error = Error;

fn try_from(value: &Ec) -> Result<Self, Self::Error> {
if value.crv != EcCurves::P256K {
return Err(Error::AlgMismatch);
}

let mut x = FieldBytes::default();
if value.x.len() != x.len() {
return Err(Error::Invalid);
}

let mut y = FieldBytes::default();
if value.y.len() != y.len() {
return Err(Error::Invalid);
}

x.copy_from_slice(&value.x);
y.copy_from_slice(&value.y);

let ep = EncodedPoint::from_affine_coordinates(&x, &y, false);
Option::from(Self::from_encoded_point(&ep)).ok_or(Error::Invalid)
}
}

impl TryFrom<Ec> for PublicKey {
type Error = Error;

fn try_from(value: Ec) -> Result<Self, Self::Error> {
(&value).try_into()
}
}

impl From<&SecretKey> for Ec {
fn from(sk: &SecretKey) -> Self {
let mut key: Self = sk.public_key().into();
key.d = Some(sk.to_bytes().to_vec().into());
key
}
}

impl From<SecretKey> for Ec {
fn from(sk: SecretKey) -> Self {
(&sk).into()
}
}

impl TryFrom<&Ec> for SecretKey {
type Error = Error;

fn try_from(value: &Ec) -> Result<Self, Self::Error> {
if value.crv != EcCurves::P256K {
return Err(Error::AlgMismatch);
}

if let Some(d) = value.d.as_ref() {
return Self::from_slice(d).map_err(|_| Error::Invalid);
}

Err(Error::NotPrivate)
}
}

impl TryFrom<Ec> for SecretKey {
type Error = Error;

fn try_from(value: Ec) -> Result<Self, Self::Error> {
(&value).try_into()
}
}

0 comments on commit 9c53484

Please sign in to comment.