Skip to content

Commit 3b6f638

Browse files
committed
refactor: rename SECRET_KEY_SIZE to PRIVATE_KEY_SIZE across multiple files
1 parent 061c9f3 commit 3b6f638

6 files changed

Lines changed: 16 additions & 18 deletions

File tree

crates/crypt/src/kem.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ pub use libcrux_ml_kem::mlkem768::{
1010
use libcrux_ml_kem::mlkem768::generate_key_pair as ml_kem_generate_key_pair;
1111

1212
// Note, changes to kem size (1024, 768 or 512) will need to update also SECRET_KEY_SIZE and CIPHERTEXT_SIZE
13-
pub const SECRET_KEY_SIZE: usize = 2400;
13+
pub const PRIVATE_KEY_SIZE: usize = 2400;
1414
pub const PUBLIC_KEY_SIZE: usize = 1184;
1515
pub const CIPHERTEXT_SIZE: usize = 1088;
1616

crates/crypt/src/lib.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ mod ticket;
55
pub use ticket::{Ticket, TunnelMaterial};
66

77
mod kem;
8-
pub use kem::{CIPHERTEXT_SIZE, SECRET_KEY_SIZE, PUBLIC_KEY_SIZE, generate_key_pair};
8+
pub use kem::{CIPHERTEXT_SIZE, PRIVATE_KEY_SIZE, PUBLIC_KEY_SIZE, generate_key_pair};
99

1010
#[cfg(test)]
1111
mod tests;

crates/crypt/src/tests.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
use anyhow::Result;
22
use base64::{Engine as _, engine::general_purpose};
33

4-
use crate::{Ticket, SECRET_KEY_SIZE};
4+
use crate::{Ticket, PRIVATE_KEY_SIZE};
55

66
const PRIVATE_KEY_768_TESTING: &str = "TzpPr8sQk1BBjmEFpTqCqdhTNfGdTpK37GBFaQWnigW8AZqMzrlSxRa+grYDdjJ1JiaiuSkpptCtIKsf\
77
6QiD6HRJrAPNCJyxbmihz3KS0IOmjzUx4BYh/Ap/nYbE/0qWZFG0KdGKtSKWnOoQFCph0vOLQKnN8HGq\
@@ -65,11 +65,11 @@ const TEST_TICKET_JSON: &str = r#"{
6565
// }
6666
// }
6767

68-
fn get_private_key_bytes() -> Result<[u8; SECRET_KEY_SIZE]> {
68+
fn get_private_key_bytes() -> Result<[u8; PRIVATE_KEY_SIZE]> {
6969
let kem_private_key_bytes = general_purpose::STANDARD
7070
.decode(PRIVATE_KEY_768_TESTING)
7171
.map_err(|e| anyhow::format_err!("Failed to decode base64 KEM private key: {}", e))?;
72-
let kem_private_key_bytes: [u8; SECRET_KEY_SIZE] = kem_private_key_bytes
72+
let kem_private_key_bytes: [u8; PRIVATE_KEY_SIZE] = kem_private_key_bytes
7373
.try_into()
7474
.map_err(|_| anyhow::format_err!("Invalid KEM private key size"))?;
7575
Ok(kem_private_key_bytes)

crates/crypt/src/ticket.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ use aes_gcm::{
99
};
1010
use base64::{Engine as _, engine::general_purpose};
1111

12-
use super::kem::{CIPHERTEXT_SIZE, CipherText, PrivateKey, SECRET_KEY_SIZE, decapsulate};
12+
use super::kem::{CIPHERTEXT_SIZE, CipherText, PrivateKey, PRIVATE_KEY_SIZE, decapsulate};
1313

1414
#[derive(serde::Serialize, serde::Deserialize, Debug)]
1515
pub struct TunnelMaterial {
@@ -68,7 +68,7 @@ impl Ticket {
6868
pub fn recover_data_from_json(
6969
&self,
7070
ticket_id: &[u8],
71-
kem_private_key: &[u8; SECRET_KEY_SIZE],
71+
kem_private_key: &[u8; PRIVATE_KEY_SIZE],
7272
) -> Result<serde_json::Value> {
7373
let kem_private_key = PrivateKey::from(kem_private_key);
7474

crates/shared/src/broker/api/mod.rs

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ use base64::{Engine as _, engine::general_purpose};
3535
use reqwest::{Client, ClientBuilder};
3636

3737
use crate::{consts, log};
38-
use crypt::{PUBLIC_KEY_SIZE, SECRET_KEY_SIZE, Ticket, generate_key_pair};
38+
use crypt::{PUBLIC_KEY_SIZE, PRIVATE_KEY_SIZE, Ticket, generate_key_pair};
3939

4040
pub mod types;
4141

@@ -55,7 +55,7 @@ pub struct UdsBrokerApi {
5555
broker_url: String,
5656
hostname: String,
5757
public_key: [u8; PUBLIC_KEY_SIZE],
58-
private_key: [u8; SECRET_KEY_SIZE],
58+
private_key: [u8; PRIVATE_KEY_SIZE],
5959
}
6060

6161
impl UdsBrokerApi {
@@ -76,7 +76,6 @@ impl UdsBrokerApi {
7676
builder = builder.no_proxy();
7777
}
7878

79-
8079
// Note: unwraps are intentinonal here, if we cannot build the client, we want to
8180
// abort early.
8281

@@ -96,7 +95,11 @@ impl UdsBrokerApi {
9695

9796
// Only for tests
9897
#[cfg(test)]
99-
pub fn with_keys(self, private_key: [u8; SECRET_KEY_SIZE], public_key: [u8; PUBLIC_KEY_SIZE]) -> Self {
98+
pub fn with_keys(
99+
self,
100+
private_key: [u8; PRIVATE_KEY_SIZE],
101+
public_key: [u8; PUBLIC_KEY_SIZE],
102+
) -> Self {
100103
Self {
101104
public_key,
102105
private_key,
@@ -175,11 +178,6 @@ impl BrokerApi for UdsBrokerApi {
175178
percent: 0,
176179
})
177180
})?
178-
179-
// response
180-
// .json::<types::BrokerResponse<types::Script>>()
181-
// .await?
182-
// .into_result()
183181
}
184182

185183
async fn send_log(&self, log_str: String) -> Result<()> {

crates/shared/src/broker/api/tests.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,11 +106,11 @@ const TICKET_RESPONSE_JSON: &str = r#"{
106106
}"#;
107107
const TICKET_ID: &str = "c6s9FAa5fhb854BVMckqUBJ4hOXg2iE5i1FYPCuktks4eNZD";
108108

109-
fn get_keypair() -> Result<([u8; SECRET_KEY_SIZE], [u8; PUBLIC_KEY_SIZE])> {
109+
fn get_keypair() -> Result<([u8; PRIVATE_KEY_SIZE], [u8; PUBLIC_KEY_SIZE])> {
110110
let kem_private_key_bytes = general_purpose::STANDARD
111111
.decode(PRIVATE_KEY_768_TESTING)
112112
.map_err(|e| anyhow::format_err!("Failed to decode base64 KEM private key: {}", e))?;
113-
let kem_private_key_bytes: [u8; SECRET_KEY_SIZE] = kem_private_key_bytes
113+
let kem_private_key_bytes: [u8; PRIVATE_KEY_SIZE] = kem_private_key_bytes
114114
.try_into()
115115
.map_err(|_| anyhow::format_err!("Invalid KEM private key size"))?;
116116
let kem_public_key_bytes = general_purpose::STANDARD

0 commit comments

Comments
 (0)