Skip to content

Commit 5984ce6

Browse files
committed
lint: Fix clippy warnings
Signed-off-by: rrouviere <ruben.rouviere@etu.umontpellier.fr>
1 parent 67fc5ad commit 5984ce6

File tree

4 files changed

+22
-24
lines changed

4 files changed

+22
-24
lines changed

topics/p2p-transfer-protocol/src/client/mod.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,11 +26,11 @@ pub fn run_client(mut file: File, mut stream: TcpStream) -> io::Result<()> {
2626
reader.read_line(&mut response)?;
2727
let response = response.trim();
2828
match response.parse::<Message>() {
29-
Ok(Message::ACK) => {
29+
Ok(Message::Ack) => {
3030
debug!("Received ACK");
3131
sm.transition(ConnectionState::ACKReceived);
3232
}
33-
Ok(Message::NACK) => {
33+
Ok(Message::Nack) => {
3434
debug!("Received NACK");
3535
sm.transition(ConnectionState::NACKReceived);
3636
return Ok(());

topics/p2p-transfer-protocol/src/main.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ fn main() {
6565
fn server_mode(bind_addr: &str, file_path: &str) -> io::Result<()> {
6666
// Start server
6767
// Bind to port: bind_addr
68-
let listener = match TcpListener::bind(&bind_addr) {
68+
let listener = match TcpListener::bind(bind_addr) {
6969
Ok(l) => l,
7070
Err(e) => {
7171
error!("Failed to bind to {}: {}", bind_addr, e);
@@ -76,6 +76,7 @@ fn server_mode(bind_addr: &str, file_path: &str) -> io::Result<()> {
7676

7777
// Open or create output file:
7878
let mut file = OpenOptions::new()
79+
.truncate(true)
7980
.write(true)
8081
.create(true)
8182
.open(file_path)?;

topics/p2p-transfer-protocol/src/protocol/message.rs

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -4,19 +4,19 @@ use std::{fmt::Display, str::FromStr};
44
#[derive(Debug, PartialEq, Eq)]
55
pub enum Message {
66
Hello { file_size: u64 },
7-
ACK,
8-
NACK,
7+
Ack,
8+
Nack,
99
Send,
1010
}
1111

1212
// Wire format
1313
impl Display for Message {
1414
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
1515
match self {
16-
Message::Hello { file_size } => write!(f, "HELLO {file_size}\n"),
17-
Message::ACK => write!(f, "ACK\n"),
18-
Message::NACK => write!(f, "NACK\n"),
19-
Message::Send => write!(f, "SEND\n"),
16+
Message::Hello { file_size } => writeln!(f, "HELLO {file_size}"),
17+
Message::Ack => writeln!(f, "ACK"),
18+
Message::Nack => writeln!(f, "NACK"),
19+
Message::Send => writeln!(f, "SEND"),
2020
}
2121
}
2222
}
@@ -31,22 +31,21 @@ impl FromStr for Message {
3131

3232
if upper.starts_with("HELLO") {
3333
let parts: Vec<&str> = s.split_whitespace().collect();
34-
if parts.len() == 2 {
35-
if let Ok(file_size) = parts[1].parse::<u64>() {
34+
if parts.len() == 2
35+
&& let Ok(file_size) = parts[1].parse::<u64>() {
3636
return Ok(Message::Hello { file_size });
3737
}
38-
}
3938
error!("Invalid HELLO message");
4039
return Err(());
4140
};
4241

4342
// No arguments msgs
44-
return match upper.as_str() {
45-
"ACK" => Ok(Message::ACK),
46-
"NACK" => Ok(Message::NACK),
43+
match upper.as_str() {
44+
"ACK" => Ok(Message::Ack),
45+
"NACK" => Ok(Message::Nack),
4746
"SEND" => Ok(Message::Send),
4847
// Unknown message
4948
_ => Err(()),
50-
};
49+
}
5150
}
5251
}

topics/p2p-transfer-protocol/src/server/mod.rs

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -18,11 +18,11 @@ fn on_hello(file_size: u64, state_machine: &mut StateMachine) -> Message {
1818
// send ack
1919
state_machine.transition(ConnectionState::ACKSent);
2020

21-
Message::ACK
21+
Message::Ack
2222
}
2323

2424
fn on_message(msg: Message, state_machine: &mut StateMachine) -> Option<Message> {
25-
return match msg {
25+
match msg {
2626
Message::Hello { file_size } => Some(on_hello(file_size, state_machine)),
2727
Message::Send => {
2828
info!("Client is starting to send data.");
@@ -34,7 +34,7 @@ fn on_message(msg: Message, state_machine: &mut StateMachine) -> Option<Message>
3434
warn!("Unexpected message from client: {:?}", msg);
3535
None
3636
}
37-
};
37+
}
3838
}
3939

4040
fn send_message(stream: &mut TcpStream, msg: &Message) -> Result<(), io::Error> {
@@ -89,7 +89,7 @@ fn receive_file(
8989

9090
fn message_loop(state_machine: &mut StateMachine, stream: &mut TcpStream) -> Result<(), io::Error> {
9191
let reader_stream = stream.try_clone().expect("Failed to clone stream");
92-
let mut writer_stream = stream;
92+
let writer_stream = stream;
9393
let reader = BufReader::new(reader_stream);
9494

9595
for line in reader.lines() {
@@ -106,10 +106,8 @@ fn message_loop(state_machine: &mut StateMachine, stream: &mut TcpStream) -> Res
106106

107107
// Process message
108108
// FIXME: Do not use unwrap, fix the result type mess.
109-
match on_message(message.unwrap(), state_machine) {
110-
Some(resp) => send_message(&mut writer_stream, &resp)?,
111-
None => {} // No response is needed
112-
}
109+
if let Some(resp) = on_message(message.unwrap(), state_machine) { send_message(writer_stream, &resp)? }
110+
113111
// If we are in Established state, stop the line-based logic and receive the data.
114112
if let ConnectionState::Established = state_machine.current_state() {
115113
break;

0 commit comments

Comments
 (0)