Compare commits
4 Commits
download_p
...
341f8f123d
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
341f8f123d | ||
|
|
c59ce1be55 | ||
|
|
f735ab34fc | ||
|
|
b54e044ab1 |
@@ -27,7 +27,7 @@ pub struct P2PClientApp {
|
|||||||
|
|
||||||
// GUI State
|
// GUI State
|
||||||
status_message: String,
|
status_message: String,
|
||||||
known_peers: Vec<String>,
|
known_peers: Vec<(String, bool)>,
|
||||||
connect_address_input: String,
|
connect_address_input: String,
|
||||||
connected_address: String,
|
connected_address: String,
|
||||||
connect_name_input: String,
|
connect_name_input: String,
|
||||||
@@ -62,7 +62,7 @@ impl P2PClientApp {
|
|||||||
network_cmd_tx: cmd_tx,
|
network_cmd_tx: cmd_tx,
|
||||||
network_event_rx: event_rx,
|
network_event_rx: event_rx,
|
||||||
status_message: "Client Initialized. Awaiting network status...".to_string(),
|
status_message: "Client Initialized. Awaiting network status...".to_string(),
|
||||||
known_peers: vec!["bob".to_string()],
|
known_peers: vec![("bob".to_string(), true)],
|
||||||
connect_address_input: "https://jch.irif.fr:8443".to_string(),
|
connect_address_input: "https://jch.irif.fr:8443".to_string(),
|
||||||
connected_address: "".to_string(),
|
connected_address: "".to_string(),
|
||||||
loaded_fs,
|
loaded_fs,
|
||||||
@@ -111,8 +111,8 @@ impl eframe::App for P2PClientApp {
|
|||||||
todo!();
|
todo!();
|
||||||
|
|
||||||
self.status_message = format!("✅ Peer connected: {}", addr);
|
self.status_message = format!("✅ Peer connected: {}", addr);
|
||||||
if !self.known_peers.contains(&addr) {
|
if !self.known_peers.contains(&(addr, true)) {
|
||||||
self.known_peers.push(addr);
|
self.known_peers.push((addr, true));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
NetworkEvent::PeerListUpdated(peers) => {
|
NetworkEvent::PeerListUpdated(peers) => {
|
||||||
@@ -343,28 +343,26 @@ impl eframe::App for P2PClientApp {
|
|||||||
} else {
|
} else {
|
||||||
for peer in &self.known_peers {
|
for peer in &self.known_peers {
|
||||||
let is_active =
|
let is_active =
|
||||||
self.active_peer.as_ref().map_or(false, |id| id == peer); // if peer.id == self.active_peer_id
|
self.active_peer.as_ref().map_or(false, |id| id == &peer.0); // if peer.id == self.active_peer_id
|
||||||
|
|
||||||
let selectable;
|
let selectable;
|
||||||
if &self.active_server == peer {
|
if &self.active_server == &peer.0 {
|
||||||
selectable =
|
selectable =
|
||||||
ui.selectable_label(is_active, format!("{} 📡 🌀", peer))
|
ui.selectable_label(is_active, format!("{} 📡 🌀", peer.0))
|
||||||
} else {
|
} else {
|
||||||
selectable = ui.selectable_label(is_active, format!("{}", peer));
|
selectable = ui.selectable_label(is_active, format!("{}", peer.0));
|
||||||
}
|
}
|
||||||
if selectable.clicked() {
|
if selectable.clicked() {
|
||||||
// switch to displaying this peer's tree
|
// switch to displaying this peer's tree
|
||||||
self.active_peer = Some(peer.clone());
|
self.active_peer = Some(peer.0.clone());
|
||||||
// Request root content if not loaded
|
// Request root content if not loaded
|
||||||
if !self
|
if !self
|
||||||
.loaded_fs
|
.loaded_fs
|
||||||
.contains_key(self.active_peer.as_ref().unwrap())
|
.contains_key(self.active_peer.as_ref().unwrap())
|
||||||
{
|
{
|
||||||
todo!();
|
let _ = self
|
||||||
// let _ = self.network_cmd_tx.send(NetworkCommand::RequestDirectoryContent(
|
.network_cmd_tx
|
||||||
// peer.clone(),
|
.send(NetworkCommand::ConnectPeer(peer.clone()));
|
||||||
// peer.clone(),
|
|
||||||
// ));
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
selectable.context_menu(|ui| {
|
selectable.context_menu(|ui| {
|
||||||
@@ -375,10 +373,10 @@ impl eframe::App for P2PClientApp {
|
|||||||
.button("Utiliser le peer en tant que serveur")
|
.button("Utiliser le peer en tant que serveur")
|
||||||
.clicked()
|
.clicked()
|
||||||
{
|
{
|
||||||
self.active_server = peer.to_string();
|
self.active_server = peer.0.to_string();
|
||||||
let res = self.network_cmd_tx.send(
|
let res = self.network_cmd_tx.send(
|
||||||
NetworkCommand::ServerHandshake(
|
NetworkCommand::ServerHandshake(
|
||||||
peer.to_string(),
|
peer.0.to_string(),
|
||||||
self.connected_address.clone(),
|
self.connected_address.clone(),
|
||||||
),
|
),
|
||||||
);
|
);
|
||||||
@@ -386,10 +384,29 @@ impl eframe::App for P2PClientApp {
|
|||||||
}
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
|
if ui.button("Send Ping").clicked() {
|
||||||
|
let res = self
|
||||||
|
.network_cmd_tx
|
||||||
|
.send(NetworkCommand::Ping(peer.0.to_string()));
|
||||||
|
}
|
||||||
|
if ui.button("Send Nat Traversal Request").clicked() {
|
||||||
|
match self.network_cmd_tx.send(NetworkCommand::NatTraversal(
|
||||||
|
peer.0.to_string(),
|
||||||
|
self.connected_address.clone(),
|
||||||
|
)) {
|
||||||
|
Ok(_) => {
|
||||||
|
print!("[+] successfully sent nat traversal request")
|
||||||
|
}
|
||||||
|
Err(_) => {
|
||||||
|
print!("[-] failed to send nat traversal request")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
if ui.button("Infos").clicked() {
|
if ui.button("Infos").clicked() {
|
||||||
// action 3
|
// action 3
|
||||||
ui.close();
|
ui.close();
|
||||||
}
|
}
|
||||||
|
|
||||||
// ... autres boutons
|
// ... autres boutons
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,19 +12,17 @@ use crate::{
|
|||||||
cryptographic_signature::CryptographicSignature,
|
cryptographic_signature::CryptographicSignature,
|
||||||
message_handling::EventType,
|
message_handling::EventType,
|
||||||
messages_channels::{MultipleSenders, start_receving_thread},
|
messages_channels::{MultipleSenders, start_receving_thread},
|
||||||
registration::{
|
messages_structure::{NATTRAVERSALREQUEST, NATTRAVERSALREQUEST2, construct_message},
|
||||||
get_socket_address, parse_addresses, register_ip_addresses, register_with_the_server,
|
registration::{parse_addresses, register_ip_addresses, register_with_the_server},
|
||||||
},
|
|
||||||
server_communication::{generate_id, get_peer_list},
|
server_communication::{generate_id, get_peer_list},
|
||||||
};
|
};
|
||||||
use std::{
|
use std::{
|
||||||
fmt,
|
io::Error,
|
||||||
sync::{Arc, Mutex},
|
net::{Ipv4Addr, UdpSocket},
|
||||||
};
|
};
|
||||||
use std::{
|
use std::{
|
||||||
io::Error,
|
net::SocketAddr,
|
||||||
net::{SocketAddr, UdpSocket},
|
sync::{Arc, Mutex},
|
||||||
str::FromStr,
|
|
||||||
};
|
};
|
||||||
|
|
||||||
pub struct P2PSharedData {
|
pub struct P2PSharedData {
|
||||||
@@ -35,6 +33,9 @@ pub struct P2PSharedData {
|
|||||||
server_name: Arc<Mutex<String>>,
|
server_name: Arc<Mutex<String>>,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
use bytes::Bytes;
|
||||||
|
use p256::pkcs8::der::pem::Base64Encoder;
|
||||||
|
|
||||||
impl P2PSharedData {
|
impl P2PSharedData {
|
||||||
pub fn new(
|
pub fn new(
|
||||||
username: String,
|
username: String,
|
||||||
@@ -108,9 +109,10 @@ pub enum NetworkCommand {
|
|||||||
ServerHandshake(String, String), // ServerName
|
ServerHandshake(String, String), // ServerName
|
||||||
FetchPeerList(String), // ServerIP
|
FetchPeerList(String), // ServerIP
|
||||||
RegisterAsPeer(String),
|
RegisterAsPeer(String),
|
||||||
Ping(),
|
Ping(String),
|
||||||
ConnectPeer(String), // IP:PORT
|
NatTraversal(String, String),
|
||||||
RequestFileTree(String), // peer_id
|
ConnectPeer((String, bool)), // IP:PORT
|
||||||
|
RequestFileTree(String), // peer_id
|
||||||
RequestDirectoryContent(String, String),
|
RequestDirectoryContent(String, String),
|
||||||
RequestChunk(String, String),
|
RequestChunk(String, String),
|
||||||
Disconnect(),
|
Disconnect(),
|
||||||
@@ -125,7 +127,7 @@ pub enum NetworkEvent {
|
|||||||
Disconnected(),
|
Disconnected(),
|
||||||
Error(String),
|
Error(String),
|
||||||
PeerConnected(String),
|
PeerConnected(String),
|
||||||
PeerListUpdated(Vec<String>),
|
PeerListUpdated(Vec<(String, bool)>),
|
||||||
FileTreeReceived(String, Vec<MerkleNode>), // peer_id, content
|
FileTreeReceived(String, Vec<MerkleNode>), // peer_id, content
|
||||||
DataReceived(String, MerkleNode),
|
DataReceived(String, MerkleNode),
|
||||||
FileTreeRootReceived(String, String),
|
FileTreeRootReceived(String, String),
|
||||||
@@ -177,62 +179,33 @@ pub fn start_p2p_executor(
|
|||||||
let server_addr_query = get_socket_address(username.clone(), ip);
|
let server_addr_query = get_socket_address(username.clone(), ip);
|
||||||
|
|
||||||
match server_addr_query.await {
|
match server_addr_query.await {
|
||||||
Ok(sockaddr_bytes) => {
|
Some(server_address) => {
|
||||||
match String::from_utf8(sockaddr_bytes.to_vec()) {
|
sd.set_servername(username);
|
||||||
Ok(s) => {
|
// first: &SocketAddr
|
||||||
let addresses = parse_addresses(&s);
|
start_receving_thread(
|
||||||
if let Some(first) = addresses.first() {
|
sd,
|
||||||
sd.set_servername(username);
|
server_address, // copie le SocketAddr (implémente Copy pour SocketAddr)
|
||||||
// first: &SocketAddr
|
event_tx.clone(), //
|
||||||
start_receving_thread(
|
);
|
||||||
sd,
|
register_ip_addresses(
|
||||||
*first, // copie le SocketAddr (implémente Copy pour SocketAddr)
|
sd.cryptopair_ref(),
|
||||||
event_tx.clone(), //
|
server_address.to_string(),
|
||||||
);
|
sd.senders_ref(),
|
||||||
register_ip_addresses(
|
sd.messages_list_ref(),
|
||||||
sd.cryptopair_ref(),
|
generate_id(),
|
||||||
first.to_string(),
|
);
|
||||||
sd.senders_ref(),
|
|
||||||
sd.messages_list_ref(),
|
|
||||||
generate_id(),
|
|
||||||
);
|
|
||||||
|
|
||||||
//let res = event_tx
|
|
||||||
// .send(NetworkEvent::());
|
|
||||||
} else {
|
|
||||||
//let res = event_tx.send(NetworkEvent::Error());
|
|
||||||
let err_msg = format!(
|
|
||||||
"no valid socket addresses found in: {}",
|
|
||||||
s
|
|
||||||
)
|
|
||||||
.to_string();
|
|
||||||
let res =
|
|
||||||
event_tx.send(NetworkEvent::Error(err_msg));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Err(e) => {
|
|
||||||
//let res = event_tx.send(NetworkEvent::Error());
|
|
||||||
let err_msg = format!(
|
|
||||||
"invalid UTF-8 in socket address bytes: {}",
|
|
||||||
e
|
|
||||||
)
|
|
||||||
.to_string();
|
|
||||||
let res = event_tx.send(NetworkEvent::Error(err_msg));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
Err(e) => {
|
None => {
|
||||||
let err_msg =
|
let err_msg =
|
||||||
format!("failed to retreive socket address: {}", e)
|
format!("failed to retreive socket address").to_string();
|
||||||
.to_string();
|
|
||||||
let res = event_tx.send(NetworkEvent::Error(err_msg));
|
let res = event_tx.send(NetworkEvent::Error(err_msg));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
NetworkCommand::ConnectPeer(addr) => {
|
NetworkCommand::ConnectPeer((username, connected)) => {
|
||||||
println!("[Network] ConnectPeer() called");
|
println!("[Network] ConnectPeer() called");
|
||||||
println!("[Network] Attempting to connect to: {}", addr);
|
println!("[Network] Attempting to connect to: {}", username);
|
||||||
// Network logic to connect...
|
// Network logic to connect...
|
||||||
// If successful, send an event back:
|
// If successful, send an event back:
|
||||||
// event_tx.send(NetworkEvent::PeerConnected(addr)).unwrap();
|
// event_tx.send(NetworkEvent::PeerConnected(addr)).unwrap();
|
||||||
@@ -295,11 +268,11 @@ pub fn start_p2p_executor(
|
|||||||
match get_peer_list(ip).await {
|
match get_peer_list(ip).await {
|
||||||
Ok(body) => match String::from_utf8(body.to_vec()) {
|
Ok(body) => match String::from_utf8(body.to_vec()) {
|
||||||
Ok(peers_list) => {
|
Ok(peers_list) => {
|
||||||
let mut peers: Vec<String> = Vec::new();
|
let mut peers: Vec<(String, bool)> = Vec::new();
|
||||||
let mut current = String::new();
|
let mut current = String::new();
|
||||||
for i in peers_list.chars() {
|
for i in peers_list.chars() {
|
||||||
if i == '\n' {
|
if i == '\n' {
|
||||||
peers.push(current.clone());
|
peers.push((current.clone(), false));
|
||||||
current.clear();
|
current.clear();
|
||||||
} else {
|
} else {
|
||||||
current.push(i);
|
current.push(i);
|
||||||
@@ -320,7 +293,7 @@ pub fn start_p2p_executor(
|
|||||||
NetworkCommand::RegisterAsPeer(_) => {
|
NetworkCommand::RegisterAsPeer(_) => {
|
||||||
println!("[Network] RegisterAsPeer() called");
|
println!("[Network] RegisterAsPeer() called");
|
||||||
}
|
}
|
||||||
NetworkCommand::Ping() => {
|
NetworkCommand::Ping(String) => {
|
||||||
println!("[Network] Ping() called");
|
println!("[Network] Ping() called");
|
||||||
}
|
}
|
||||||
NetworkCommand::Disconnect() => {
|
NetworkCommand::Disconnect() => {
|
||||||
@@ -339,6 +312,53 @@ pub fn start_p2p_executor(
|
|||||||
println!("no p2p data");
|
println!("no p2p data");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
NetworkCommand::NatTraversal(username, ip) => {
|
||||||
|
if let Some(sd) = shared_data.as_ref() {
|
||||||
|
println!("username:{}, ip:{}", username, ip);
|
||||||
|
// user server to send nattraversal request
|
||||||
|
let server_addr_query =
|
||||||
|
get_socket_address(sd.servername().clone(), ip.clone());
|
||||||
|
let peer_addr_query = get_socket_address(username.clone(), ip.clone());
|
||||||
|
|
||||||
|
match server_addr_query.await {
|
||||||
|
Some(server_addr) => match peer_addr_query.await {
|
||||||
|
Some(peer_addr) => {
|
||||||
|
let payload =
|
||||||
|
parse_pack(peer_addr.clone().to_string().as_str())
|
||||||
|
.expect("couldnt create payload");
|
||||||
|
|
||||||
|
print!("{:?}", payload.clone());
|
||||||
|
|
||||||
|
let natreq = construct_message(
|
||||||
|
NATTRAVERSALREQUEST,
|
||||||
|
payload.clone().to_vec(),
|
||||||
|
generate_id(),
|
||||||
|
&sd.cryptopair(),
|
||||||
|
);
|
||||||
|
|
||||||
|
sd.senders_ref().send_via(
|
||||||
|
0,
|
||||||
|
natreq.expect(
|
||||||
|
"couldnt construct message nattraversalrequest2",
|
||||||
|
),
|
||||||
|
server_addr.to_string(),
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
let err_msg = format!("failed to retreive socket address")
|
||||||
|
.to_string();
|
||||||
|
let res = event_tx.send(NetworkEvent::Error(err_msg));
|
||||||
|
}
|
||||||
|
},
|
||||||
|
None => {
|
||||||
|
let err_msg =
|
||||||
|
format!("failed to retreive socket address").to_string();
|
||||||
|
let res = event_tx.send(NetworkEvent::Error(err_msg));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -352,3 +372,50 @@ pub fn start_p2p_executor(
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn parse_pack(s: &str) -> Option<[u8; 6]> {
|
||||||
|
// split into "ip" and "port"
|
||||||
|
let mut parts = s.rsplitn(2, ':');
|
||||||
|
let port_str = parts.next()?;
|
||||||
|
let ip_str = parts.next()?; // if missing, invalid
|
||||||
|
|
||||||
|
let ip: Ipv4Addr = ip_str.parse().ok()?;
|
||||||
|
let port: u16 = port_str.parse().ok()?;
|
||||||
|
|
||||||
|
let octets = ip.octets();
|
||||||
|
let port_be = port.to_be_bytes();
|
||||||
|
Some([
|
||||||
|
octets[0], octets[1], octets[2], octets[3], port_be[0], port_be[1],
|
||||||
|
])
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
/// sends a get request to the server to get the socket address of the given peer
|
||||||
|
///
|
||||||
|
|
||||||
|
pub async fn get_socket_address(username: String, ip: String) -> Option<SocketAddr> {
|
||||||
|
let client = reqwest::Client::new();
|
||||||
|
let uri = format!("{}/peers/{}/addresses", ip, username);
|
||||||
|
let res = client.get(uri).send().await.expect("couldnt get response");
|
||||||
|
if res.status().is_success() {
|
||||||
|
println!("Successfully retreived the addresses.");
|
||||||
|
} else {
|
||||||
|
eprintln!(
|
||||||
|
"Failed to get the peers addresses from the server. Status: {}",
|
||||||
|
res.status()
|
||||||
|
);
|
||||||
|
}
|
||||||
|
let body: Bytes = res.bytes().await.expect("couldnt get bytes");
|
||||||
|
|
||||||
|
match String::from_utf8(body.to_vec()) {
|
||||||
|
Ok(s) => {
|
||||||
|
let addresses = parse_addresses(&s);
|
||||||
|
if let Some(first) = addresses.first() {
|
||||||
|
Some(first.clone())
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(_) => None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -70,7 +70,14 @@ pub fn handle_recevied_message(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
let resp = parse_message(recevied_message.to_vec(), id, crypto_pair, cmd_tx, ip);
|
let resp = parse_message(
|
||||||
|
recevied_message.to_vec(),
|
||||||
|
id,
|
||||||
|
crypto_pair,
|
||||||
|
cmd_tx,
|
||||||
|
ip,
|
||||||
|
senders,
|
||||||
|
);
|
||||||
|
|
||||||
match resp {
|
match resp {
|
||||||
None => {}
|
None => {}
|
||||||
@@ -152,6 +159,7 @@ pub fn parse_message(
|
|||||||
crypto_pair: &CryptographicSignature,
|
crypto_pair: &CryptographicSignature,
|
||||||
cmd_tx: crossbeam_channel::Sender<NetworkEvent>,
|
cmd_tx: crossbeam_channel::Sender<NetworkEvent>,
|
||||||
ip: SocketAddr,
|
ip: SocketAddr,
|
||||||
|
senders: &MultipleSenders,
|
||||||
) -> Option<Vec<u8>> {
|
) -> Option<Vec<u8>> {
|
||||||
let cmd_tx_clone = cmd_tx.clone();
|
let cmd_tx_clone = cmd_tx.clone();
|
||||||
|
|
||||||
@@ -215,6 +223,48 @@ pub fn parse_message(
|
|||||||
//
|
//
|
||||||
// rien ?
|
// rien ?
|
||||||
// si NATTRAVERSALREQUEST alors
|
// si NATTRAVERSALREQUEST alors
|
||||||
|
NATTRAVERSALREQUEST => {
|
||||||
|
// send ok & send nattraversalrequest2 to peer
|
||||||
|
constructed_message = construct_message(OK, Vec::new(), id, crypto_pair);
|
||||||
|
|
||||||
|
let ilength = u16::from_be_bytes(length_bytes);
|
||||||
|
let received_address =
|
||||||
|
&received_message[LENGTH + EXTENSIONS..LENGTH + ilength as usize];
|
||||||
|
let address = String::from_utf8(received_address.to_vec()).expect("wrong name");
|
||||||
|
|
||||||
|
let natreq2 = construct_message(
|
||||||
|
NATTRAVERSALREQUEST2,
|
||||||
|
ip.to_string().into_bytes(),
|
||||||
|
id,
|
||||||
|
crypto_pair,
|
||||||
|
);
|
||||||
|
|
||||||
|
senders.send_via(
|
||||||
|
0,
|
||||||
|
natreq2.expect("couldnt construct message nattraversalrequest2"),
|
||||||
|
address,
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
NATTRAVERSALREQUEST2 => {
|
||||||
|
// send ok & send ping to peer
|
||||||
|
constructed_message = construct_message(OK, Vec::new(), id, crypto_pair);
|
||||||
|
|
||||||
|
let ilength = u16::from_be_bytes(length_bytes);
|
||||||
|
let received_address =
|
||||||
|
&received_message[LENGTH + EXTENSIONS..LENGTH + ilength as usize];
|
||||||
|
let address = String::from_utf8(received_address.to_vec()).expect("wrong name");
|
||||||
|
|
||||||
|
let pingreq = construct_message(PING, Vec::new(), id, crypto_pair);
|
||||||
|
|
||||||
|
senders.send_via(
|
||||||
|
0,
|
||||||
|
pingreq.expect("couldnt construct message ping request"),
|
||||||
|
address,
|
||||||
|
false,
|
||||||
|
);
|
||||||
|
}
|
||||||
//
|
//
|
||||||
// ERROR
|
// ERROR
|
||||||
//
|
//
|
||||||
|
|||||||
@@ -1,7 +1,4 @@
|
|||||||
use crate::{
|
use crate::cryptographic_signature::{CryptographicSignature, sign_message};
|
||||||
cryptographic_signature::{CryptographicSignature, sign_message},
|
|
||||||
server_communication::generate_id,
|
|
||||||
};
|
|
||||||
|
|
||||||
const ID: usize = 4;
|
const ID: usize = 4;
|
||||||
const TYPE: usize = 5;
|
const TYPE: usize = 5;
|
||||||
@@ -9,18 +6,18 @@ const LENGTH: usize = 7;
|
|||||||
const EXTENSIONS: usize = 4;
|
const EXTENSIONS: usize = 4;
|
||||||
const SIGNATURE: usize = 64;
|
const SIGNATURE: usize = 64;
|
||||||
|
|
||||||
const PING: u8 = 0;
|
pub(crate) const PING: u8 = 0;
|
||||||
const OK: u8 = 128;
|
pub(crate) const OK: u8 = 128;
|
||||||
const ERROR: u8 = 129;
|
pub(crate) const ERROR: u8 = 129;
|
||||||
const HELLO: u8 = 1;
|
pub(crate) const HELLO: u8 = 1;
|
||||||
const HELLOREPLY: u8 = 130;
|
pub(crate) const HELLOREPLY: u8 = 130;
|
||||||
const ROOTREQUEST: u8 = 2;
|
pub(crate) const ROOTREQUEST: u8 = 2;
|
||||||
const ROOTREPLY: u8 = 131;
|
pub(crate) const ROOTREPLY: u8 = 131;
|
||||||
const DATUMREQUEST: u8 = 3;
|
pub(crate) const DATUMREQUEST: u8 = 3;
|
||||||
const NODATUM: u8 = 133;
|
pub(crate) const NODATUM: u8 = 133;
|
||||||
const DATUM: u8 = 132;
|
pub(crate) const DATUM: u8 = 132;
|
||||||
const NATTRAVERSALREQUEST: u8 = 4;
|
pub(crate) const NATTRAVERSALREQUEST: u8 = 4;
|
||||||
const NATTRAVERSALREQUEST2: u8 = 5;
|
pub(crate) const NATTRAVERSALREQUEST2: u8 = 5;
|
||||||
|
|
||||||
pub fn construct_message(
|
pub fn construct_message(
|
||||||
msgtype: u8,
|
msgtype: u8,
|
||||||
@@ -58,8 +55,8 @@ pub fn construct_message(
|
|||||||
ROOTREPLY | NODATUM | DATUM | NATTRAVERSALREQUEST => {
|
ROOTREPLY | NODATUM | DATUM | NATTRAVERSALREQUEST => {
|
||||||
message.extend_from_slice(&payload.len().to_be_bytes());
|
message.extend_from_slice(&payload.len().to_be_bytes());
|
||||||
message.extend_from_slice(&payload);
|
message.extend_from_slice(&payload);
|
||||||
let signature = sign_message(crypto_pair, &message);
|
//let signature = sign_message(crypto_pair, &message);
|
||||||
message.extend_from_slice(&signature);
|
//message.extend_from_slice(&signature);
|
||||||
return Some(message);
|
return Some(message);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -67,3 +64,176 @@ pub fn construct_message(
|
|||||||
}
|
}
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub struct UDPMessage {
|
||||||
|
id: u32,
|
||||||
|
msg_type: u8,
|
||||||
|
length: u16,
|
||||||
|
body: Vec<u8>,
|
||||||
|
signature: Vec<u8>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct HandshakeMessage {
|
||||||
|
pub id: u32,
|
||||||
|
msg_type: u8,
|
||||||
|
length: u16,
|
||||||
|
extensions: u32,
|
||||||
|
pub name: Vec<u8>,
|
||||||
|
pub signature: Vec<u8>,
|
||||||
|
}
|
||||||
|
|
||||||
|
pub struct NatTraversal {}
|
||||||
|
|
||||||
|
impl UDPMessage {
|
||||||
|
pub fn ping(id: u32) -> UDPMessage {
|
||||||
|
UDPMessage {
|
||||||
|
id: id,
|
||||||
|
msg_type: 0,
|
||||||
|
length: 0,
|
||||||
|
body: vec![0; 985],
|
||||||
|
signature: vec![0; 32],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn error(id: u32) -> UDPMessage {
|
||||||
|
UDPMessage {
|
||||||
|
id: id,
|
||||||
|
msg_type: 129,
|
||||||
|
length: 0,
|
||||||
|
body: vec![0; 985],
|
||||||
|
signature: vec![0; 32],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn parse(received_message: Vec<u8>) -> UDPMessage {
|
||||||
|
let id_bytes: [u8; 4] = received_message[0..4]
|
||||||
|
.try_into()
|
||||||
|
.expect("Taille incorrecte");
|
||||||
|
let length_bytes: [u8; 2] = received_message[5..7]
|
||||||
|
.try_into()
|
||||||
|
.expect("Taille incorrecte");
|
||||||
|
let msg_length = u16::from_be_bytes(length_bytes);
|
||||||
|
let name_bytes = &received_message[7..msg_length as usize + 8];
|
||||||
|
let signature_bytes =
|
||||||
|
&received_message[msg_length as usize + 8..msg_length as usize + 9 + 32];
|
||||||
|
UDPMessage {
|
||||||
|
id: u32::from_be_bytes(id_bytes),
|
||||||
|
msg_type: received_message[4],
|
||||||
|
length: u16::from_be_bytes(length_bytes),
|
||||||
|
body: name_bytes.to_vec(),
|
||||||
|
signature: signature_bytes.to_vec(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn display(&self) {
|
||||||
|
println!("ID: {:?}", self.id);
|
||||||
|
println!("Message Type: {}", self.msg_type);
|
||||||
|
println!("Length: {:?}", self.length);
|
||||||
|
let good_length = usize::min(self.length as usize, 985);
|
||||||
|
println!("name: {:?}", &self.body[..good_length]);
|
||||||
|
println!("Signature: {:?}", self.signature);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl HandshakeMessage {
|
||||||
|
pub fn display(&self) {
|
||||||
|
println!("ID: {:?}", self.id);
|
||||||
|
println!("Message Type: {}", self.msg_type);
|
||||||
|
println!("Length: {:?}", self.length);
|
||||||
|
println!("extensions: {:?}", self.extensions);
|
||||||
|
println!("name: {:?}", &self.name[..(self.length - 4) as usize]);
|
||||||
|
println!("Signature: {:?}", self.signature);
|
||||||
|
}
|
||||||
|
pub fn hello(id: u32, length: u16, username: String) -> HandshakeMessage {
|
||||||
|
let name_vec = username.trim_end_matches(char::from(0)).as_bytes().to_vec();
|
||||||
|
HandshakeMessage {
|
||||||
|
id: id,
|
||||||
|
msg_type: 1,
|
||||||
|
length: length,
|
||||||
|
extensions: 0,
|
||||||
|
name: name_vec,
|
||||||
|
signature: vec![0; 64],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn helloReply(id: u32, length: u16, username: String) -> HandshakeMessage {
|
||||||
|
let name_vec = username.trim_end_matches(char::from(0)).as_bytes().to_vec();
|
||||||
|
HandshakeMessage {
|
||||||
|
id: id,
|
||||||
|
msg_type: 130,
|
||||||
|
length: length,
|
||||||
|
extensions: 0,
|
||||||
|
name: name_vec,
|
||||||
|
signature: vec![0; 64],
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn serialize(&self) -> Vec<u8> {
|
||||||
|
let mut out = Vec::with_capacity(4 + 1 + 2 + 4 + self.name.len() + self.signature.len());
|
||||||
|
|
||||||
|
// id: u32 little-endian
|
||||||
|
out.extend_from_slice(&self.id.to_be_bytes());
|
||||||
|
|
||||||
|
// msg_type: u8
|
||||||
|
out.push(self.msg_type);
|
||||||
|
|
||||||
|
out.extend_from_slice(&self.length.to_be_bytes());
|
||||||
|
|
||||||
|
out.extend_from_slice(&self.extensions.to_be_bytes());
|
||||||
|
|
||||||
|
out.extend_from_slice(&self.name);
|
||||||
|
|
||||||
|
out.extend_from_slice(&self.signature);
|
||||||
|
|
||||||
|
out
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn parse(received_message: Vec<u8>) -> HandshakeMessage {
|
||||||
|
let id_bytes: [u8; 4] = received_message[0..4]
|
||||||
|
.try_into()
|
||||||
|
.expect("Taille incorrecte");
|
||||||
|
let length_bytes: [u8; 2] = received_message[5..7]
|
||||||
|
.try_into()
|
||||||
|
.expect("Taille incorrecte");
|
||||||
|
let msg_length = u16::from_be_bytes(length_bytes);
|
||||||
|
let extensions_bytes: [u8; 4] = received_message[7..11]
|
||||||
|
.try_into()
|
||||||
|
.expect("Taille incorrecte");
|
||||||
|
let name_bytes = &received_message[11..(11 + msg_length - 4) as usize];
|
||||||
|
let signature_bytes =
|
||||||
|
&received_message[(11 + msg_length - 4) as usize..(11 + msg_length - 4 + 64) as usize];
|
||||||
|
HandshakeMessage {
|
||||||
|
id: u32::from_be_bytes(id_bytes),
|
||||||
|
msg_type: received_message[4],
|
||||||
|
length: u16::from_be_bytes(length_bytes),
|
||||||
|
extensions: u32::from_be_bytes(extensions_bytes),
|
||||||
|
name: name_bytes.to_vec(),
|
||||||
|
signature: signature_bytes.to_vec(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
// Note this useful idiom: importing names from outer (for mod tests) scope.
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
/// creates an handshake message
|
||||||
|
#[tokio::test]
|
||||||
|
async fn creating_handshake_msg() {
|
||||||
|
let username = String::from("charlie_kirk");
|
||||||
|
let handshake = HandshakeMessage::hello(0, 12, username);
|
||||||
|
handshake.display();
|
||||||
|
}
|
||||||
|
|
||||||
|
/// parses an handshake message
|
||||||
|
#[tokio::test]
|
||||||
|
async fn parse_handshakemessage() {
|
||||||
|
let username = String::from("charlie_kirk");
|
||||||
|
let handshake = HandshakeMessage::hello(0, 12, username);
|
||||||
|
let ser = handshake.serialize();
|
||||||
|
let parsed = HandshakeMessage::parse(ser);
|
||||||
|
handshake.display();
|
||||||
|
parsed.display();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,9 +1,8 @@
|
|||||||
use bytes::Bytes;
|
use bytes::Bytes;
|
||||||
use getrandom::Error;
|
|
||||||
|
|
||||||
use crate::cryptographic_signature::{CryptographicSignature, formatPubKey, sign_message};
|
use crate::cryptographic_signature::{CryptographicSignature, formatPubKey, sign_message};
|
||||||
use crate::message_handling::EventType;
|
use crate::message_handling::EventType;
|
||||||
use crate::messages_channels::{Message, MultipleSenders};
|
use crate::messages_channels::MultipleSenders;
|
||||||
use crate::messages_structure::construct_message;
|
use crate::messages_structure::construct_message;
|
||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
use std::net::SocketAddr;
|
use std::net::SocketAddr;
|
||||||
@@ -29,26 +28,6 @@ pub async fn register_with_the_server(
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
///
|
|
||||||
/// sends a get request to the server to get the socket address of the given peer
|
|
||||||
///
|
|
||||||
|
|
||||||
pub async fn get_socket_address(username: String, ip: String) -> Result<Bytes, reqwest::Error> {
|
|
||||||
let client = reqwest::Client::new();
|
|
||||||
let uri = format!("{}/peers/{}/addresses", ip, username);
|
|
||||||
let res = client.get(uri).send().await?;
|
|
||||||
if res.status().is_success() {
|
|
||||||
println!("Successfully retreived the addresses.");
|
|
||||||
} else {
|
|
||||||
eprintln!(
|
|
||||||
"Failed to get the peers addresses from the server. Status: {}",
|
|
||||||
res.status()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
let body: Bytes = res.bytes().await?;
|
|
||||||
Ok(body)
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn parse_addresses(input: &String) -> Vec<SocketAddr> {
|
pub fn parse_addresses(input: &String) -> Vec<SocketAddr> {
|
||||||
let mut addrs = Vec::new();
|
let mut addrs = Vec::new();
|
||||||
for line in input.lines() {
|
for line in input.lines() {
|
||||||
|
|||||||
41
todo.md
41
todo.md
@@ -1,13 +1,32 @@
|
|||||||
# Todo :
|
# Todo
|
||||||
|
|
||||||
|
|
||||||
## peer discovery
|
## peer discovery
|
||||||
|
|
||||||
## handshake
|
## handshake
|
||||||
|
|
||||||
|
# Todo
|
||||||
|
|
||||||
|
## peer discovery
|
||||||
|
|
||||||
|
- get rsquest to the uri /peers/
|
||||||
|
|
||||||
|
## registration with the server
|
||||||
|
|
||||||
|
- generation of the cryptographic key OK
|
||||||
|
- put request to the uri (check if the peer is already connected) OK
|
||||||
|
- udp handshakes OK
|
||||||
|
- get request to the uri /peers/key to get the public key of a peer OK
|
||||||
|
- get request to the uri /peers/key/addresses OK
|
||||||
|
|
||||||
|
## handshake
|
||||||
|
|
||||||
|
- handshake structure OK
|
||||||
|
|
||||||
- 5min timeout after handshake
|
- 5min timeout after handshake
|
||||||
- matain connection every 4 min
|
- matain connection every 4 min
|
||||||
|
|
||||||
## data transfer
|
## data transfer
|
||||||
|
|
||||||
- request structure
|
- request structure
|
||||||
- root/root reply structure
|
- root/root reply structure
|
||||||
- datum/nodatum and datum structures
|
- datum/nodatum and datum structures
|
||||||
@@ -15,7 +34,16 @@
|
|||||||
- setting in gui to act as a relay
|
- setting in gui to act as a relay
|
||||||
- chunk, directory, big, bigdirectory structures
|
- chunk, directory, big, bigdirectory structures
|
||||||
|
|
||||||
## fonctionnalités application :
|
## fonctionnalités application
|
||||||
|
|
||||||
|
## nat traversal
|
||||||
|
|
||||||
|
- make hello and helloreply messages set the first extension bit to announce that peer is available for nat traversal
|
||||||
|
- implement actual nat traversal requests
|
||||||
|
- implement nat traversal :
|
||||||
|
- if hello/helloreply doesnt work with a peer, find a peer that supports nat traversal (server in priority) then begin protocol
|
||||||
|
|
||||||
|
fonctionnalités :
|
||||||
|
|
||||||
rechercher les fichiers d'un pair
|
rechercher les fichiers d'un pair
|
||||||
telechargement des fichiers
|
telechargement des fichiers
|
||||||
@@ -25,13 +53,11 @@ choisir le nombre de canaux
|
|||||||
handshake server DOING
|
handshake server DOING
|
||||||
se deconnecter du réseau DOING
|
se deconnecter du réseau DOING
|
||||||
|
|
||||||
|
## autre
|
||||||
## autre :
|
|
||||||
|
|
||||||
socket ipv6
|
socket ipv6
|
||||||
|
|
||||||
|
# FAIT
|
||||||
# FAIT :
|
|
||||||
|
|
||||||
- choisir un pseudo OK
|
- choisir un pseudo OK
|
||||||
- get rsquest to the uri /peers/ OK
|
- get rsquest to the uri /peers/ OK
|
||||||
@@ -45,4 +71,3 @@ socket ipv6
|
|||||||
- generer une clé publique OK
|
- generer une clé publique OK
|
||||||
- verifier signature OK
|
- verifier signature OK
|
||||||
- 2 channels -> un pour envoyer et un pour recevoir OK
|
- 2 channels -> un pour envoyer et un pour recevoir OK
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user