Skip to content

Commit cb8d9e9

Browse files
committed
fixed #338 - make KeyPair::generate_ed25519 infallible
1 parent 73fa3e5 commit cb8d9e9

File tree

8 files changed

+13
-16
lines changed

8 files changed

+13
-16
lines changed

russh-keys/src/key.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -313,13 +313,13 @@ impl KeyPair {
313313
}
314314

315315
/// Generate a ED25519 key pair.
316-
pub fn generate_ed25519() -> Option<Self> {
316+
pub fn generate_ed25519() -> Self {
317317
let keypair = ed25519_dalek::SigningKey::generate(&mut OsRng {});
318318
assert_eq!(
319319
keypair.verifying_key().as_bytes(),
320320
ed25519_dalek::VerifyingKey::from(&keypair).as_bytes()
321321
);
322-
Some(KeyPair::Ed25519(keypair))
322+
KeyPair::Ed25519(keypair)
323323
}
324324

325325
/// Generate a RSA key pair.

russh/examples/echoserver.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ async fn main() {
1717
inactivity_timeout: Some(std::time::Duration::from_secs(3600)),
1818
auth_rejection_time: std::time::Duration::from_secs(3),
1919
auth_rejection_time_initial: Some(std::time::Duration::from_secs(0)),
20-
keys: vec![russh_keys::key::KeyPair::generate_ed25519().unwrap()],
20+
keys: vec![russh_keys::key::KeyPair::generate_ed25519()],
2121
..Default::default()
2222
};
2323
let config = Arc::new(config);

russh/examples/ratatui_app.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -104,7 +104,7 @@ impl AppServer {
104104
inactivity_timeout: Some(std::time::Duration::from_secs(3600)),
105105
auth_rejection_time: std::time::Duration::from_secs(3),
106106
auth_rejection_time_initial: Some(std::time::Duration::from_secs(0)),
107-
keys: vec![russh_keys::key::KeyPair::generate_ed25519().unwrap()],
107+
keys: vec![russh_keys::key::KeyPair::generate_ed25519()],
108108
..Default::default()
109109
};
110110

russh/examples/ratatui_shared_app.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,7 +106,7 @@ impl AppServer {
106106
inactivity_timeout: Some(std::time::Duration::from_secs(3600)),
107107
auth_rejection_time: std::time::Duration::from_secs(3),
108108
auth_rejection_time_initial: Some(std::time::Duration::from_secs(0)),
109-
keys: vec![russh_keys::key::KeyPair::generate_ed25519().unwrap()],
109+
keys: vec![russh_keys::key::KeyPair::generate_ed25519()],
110110
..Default::default()
111111
};
112112

russh/examples/sftp_server.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ async fn main() {
181181
let config = russh::server::Config {
182182
auth_rejection_time: Duration::from_secs(3),
183183
auth_rejection_time_initial: Some(Duration::from_secs(0)),
184-
keys: vec![KeyPair::generate_ed25519().unwrap()],
184+
keys: vec![KeyPair::generate_ed25519()],
185185
..Default::default()
186186
};
187187

russh/examples/test.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ async fn main() -> anyhow::Result<()> {
1414
config.auth_rejection_time = std::time::Duration::from_secs(3);
1515
config
1616
.keys
17-
.push(russh_keys::key::KeyPair::generate_ed25519().unwrap());
17+
.push(russh_keys::key::KeyPair::generate_ed25519());
1818
let config = Arc::new(config);
1919
let mut sh = Server {
2020
clients: Arc::new(Mutex::new(HashMap::new())),

russh/src/tests.rs

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -19,14 +19,14 @@ mod compress {
1919
async fn compress_local_test() {
2020
let _ = env_logger::try_init();
2121

22-
let client_key = russh_keys::key::KeyPair::generate_ed25519().unwrap();
22+
let client_key = russh_keys::key::KeyPair::generate_ed25519();
2323
let mut config = server::Config::default();
2424
config.preferred = Preferred::COMPRESSED;
2525
config.inactivity_timeout = None; // Some(std::time::Duration::from_secs(3));
2626
config.auth_rejection_time = std::time::Duration::from_secs(3);
2727
config
2828
.keys
29-
.push(russh_keys::key::KeyPair::generate_ed25519().unwrap());
29+
.push(russh_keys::key::KeyPair::generate_ed25519());
3030
let config = Arc::new(config);
3131
let mut sh = Server {
3232
clients: Arc::new(Mutex::new(HashMap::new())),
@@ -162,20 +162,17 @@ mod channels {
162162

163163
let _ = env_logger::try_init();
164164

165-
let client_key = russh_keys::key::KeyPair::generate_ed25519().unwrap();
165+
let client_key = russh_keys::key::KeyPair::generate_ed25519();
166166
let mut config = server::Config::default();
167167
config.inactivity_timeout = None;
168168
config.auth_rejection_time = std::time::Duration::from_secs(3);
169169
config
170170
.keys
171-
.push(russh_keys::key::KeyPair::generate_ed25519().unwrap());
171+
.push(russh_keys::key::KeyPair::generate_ed25519());
172172
let config = Arc::new(config);
173173
let socket = tokio::net::TcpListener::bind("127.0.0.1:0").await.unwrap();
174174
let addr = socket.local_addr().unwrap();
175175

176-
#[derive(Clone)]
177-
struct Server {}
178-
179176
let server_join = tokio::spawn(async move {
180177
let (socket, _) = socket.accept().await.unwrap();
181178

russh/tests/test_data_stream.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ async fn test_reader_and_writer() -> Result<(), anyhow::Error> {
3030

3131
async fn stream(addr: SocketAddr, data: &[u8]) -> Result<(), anyhow::Error> {
3232
let config = Arc::new(client::Config::default());
33-
let key = Arc::new(russh_keys::key::KeyPair::generate_ed25519().unwrap());
33+
let key = Arc::new(russh_keys::key::KeyPair::generate_ed25519());
3434

3535
let mut session = russh::client::connect(config, addr, Client).await?;
3636
let mut channel = match session.authenticate_publickey("user", key).await {
@@ -84,7 +84,7 @@ struct Server;
8484
impl Server {
8585
async fn run(addr: SocketAddr) {
8686
let config = Arc::new(server::Config {
87-
keys: vec![russh_keys::key::KeyPair::generate_ed25519().unwrap()],
87+
keys: vec![russh_keys::key::KeyPair::generate_ed25519()],
8888
window_size: WINDOW_SIZE,
8989
..Default::default()
9090
});

0 commit comments

Comments
 (0)