106 lines
2.9 KiB
Rust
106 lines
2.9 KiB
Rust
use std::{
|
|
sync::Arc,
|
|
time::{SystemTime, UNIX_EPOCH},
|
|
};
|
|
|
|
use axum::extract::ws::WebSocket;
|
|
use ed25519_dalek::{Signer, VerifyingKey};
|
|
|
|
use crate::server::Server;
|
|
|
|
use super::*;
|
|
use crate::server::UserConnections;
|
|
|
|
impl UserConnections {
|
|
pub async fn initialize(
|
|
server: &Arc<Server>,
|
|
mut socket: WebSocket,
|
|
) -> anyhow::Result<(WebSocket, VerifyingKey, ClientMeta)> {
|
|
let Some(ServerMethod::Initialize {
|
|
public_key: public_key_string,
|
|
signature,
|
|
|
|
timestamp,
|
|
hostname,
|
|
}) = read_socket(&mut socket).await?
|
|
else {
|
|
send_socket(
|
|
&mut socket,
|
|
&ClientMethod::Error {
|
|
error: Cow::Borrowed("Initialization required"),
|
|
},
|
|
)
|
|
.await?;
|
|
|
|
return Err(anyhow::anyhow!(
|
|
"Failed to initialize: Client sent the wrong method"
|
|
));
|
|
};
|
|
|
|
let Ok(public_key) = crate::signature::from_string(&public_key_string) else {
|
|
send_socket(
|
|
&mut socket,
|
|
&ClientMethod::Error {
|
|
error: Cow::Borrowed("Invalid public key"),
|
|
},
|
|
)
|
|
.await?;
|
|
|
|
return Err(anyhow::anyhow!("Invalid public key"));
|
|
};
|
|
|
|
if public_key
|
|
.verify_strict(
|
|
format!("{timestamp}@{hostname}").as_bytes(),
|
|
&crate::signature::from_string_sig(&signature)?,
|
|
)
|
|
.is_err()
|
|
{
|
|
send_socket(
|
|
&mut socket,
|
|
&ClientMethod::Error {
|
|
error: Cow::Borrowed("Invalid signature"),
|
|
},
|
|
)
|
|
.await?;
|
|
|
|
return Err(anyhow::anyhow!("Invalid signature"));
|
|
}
|
|
|
|
{
|
|
let timestamp = SystemTime::now().duration_since(UNIX_EPOCH)?.as_secs();
|
|
|
|
send_socket(
|
|
&mut socket,
|
|
&ClientMethod::Initialized {
|
|
public_key: crate::signature::to_string(&server.key.verifying_key()),
|
|
signature: server
|
|
.key
|
|
.sign(format!("{timestamp}@{hostname}@{public_key_string}").as_bytes())
|
|
.to_string(),
|
|
|
|
timestamp,
|
|
hostname,
|
|
},
|
|
)
|
|
.await?;
|
|
}
|
|
|
|
let Some(ServerMethod::Meta(meta)) = read_socket(&mut socket).await? else {
|
|
send_socket(
|
|
&mut socket,
|
|
&ClientMethod::Error {
|
|
error: Cow::Borrowed("Expected meta"),
|
|
},
|
|
)
|
|
.await?;
|
|
|
|
return Err(anyhow::anyhow!(
|
|
"Expected meta, client called another method"
|
|
));
|
|
};
|
|
|
|
Ok((socket, public_key, meta))
|
|
}
|
|
}
|