mirror of
https://github.com/pj1234678/RustyDHCP.git
synced 2025-10-26 08:26:07 +08:00
Run cargo fmt
This commit is contained in:
@@ -25,7 +25,9 @@ impl server::Handler for MyServer {
|
|||||||
};
|
};
|
||||||
println!(
|
println!(
|
||||||
"{}\t{}\t{}\tOnline",
|
"{}\t{}\t{}\tOnline",
|
||||||
time::OffsetDateTime::try_now_local().unwrap().format("%Y-%m-%dT%H:%M:%S"),
|
time::OffsetDateTime::try_now_local()
|
||||||
|
.unwrap()
|
||||||
|
.format("%Y-%m-%dT%H:%M:%S"),
|
||||||
chaddr(&in_packet.chaddr),
|
chaddr(&in_packet.chaddr),
|
||||||
Ipv4Addr::from(req_ip)
|
Ipv4Addr::from(req_ip)
|
||||||
);
|
);
|
||||||
|
|||||||
@@ -1,9 +1,9 @@
|
|||||||
use std::collections::HashMap;
|
use std::collections::HashMap;
|
||||||
|
use std::fs::File;
|
||||||
|
use std::io::{BufRead, BufReader};
|
||||||
use std::net::{Ipv4Addr, UdpSocket};
|
use std::net::{Ipv4Addr, UdpSocket};
|
||||||
use std::ops::Add;
|
use std::ops::Add;
|
||||||
use std::time::{Duration, Instant};
|
use std::time::{Duration, Instant};
|
||||||
use std::fs::File;
|
|
||||||
use std::io::{BufRead, BufReader};
|
|
||||||
|
|
||||||
use dhcp4r::{options, packet, server};
|
use dhcp4r::{options, packet, server};
|
||||||
|
|
||||||
@@ -24,7 +24,6 @@ const LEASE_NUM: u32 = 252;
|
|||||||
const IP_START_NUM: u32 = u32::from_be_bytes(IP_START);
|
const IP_START_NUM: u32 = u32::from_be_bytes(IP_START);
|
||||||
const INFINITE_LEASE: Option<Instant> = None; // Special value for infinite lease
|
const INFINITE_LEASE: Option<Instant> = None; // Special value for infinite lease
|
||||||
|
|
||||||
|
|
||||||
fn main() {
|
fn main() {
|
||||||
let socket = UdpSocket::bind("0.0.0.0:67").unwrap();
|
let socket = UdpSocket::bind("0.0.0.0:67").unwrap();
|
||||||
socket.set_broadcast(true).unwrap();
|
socket.set_broadcast(true).unwrap();
|
||||||
@@ -76,7 +75,6 @@ impl server::Handler for MyServer {
|
|||||||
fn handle_request(&mut self, server: &server::Server, in_packet: packet::Packet) {
|
fn handle_request(&mut self, server: &server::Server, in_packet: packet::Packet) {
|
||||||
match in_packet.message_type() {
|
match in_packet.message_type() {
|
||||||
Ok(options::MessageType::Discover) => {
|
Ok(options::MessageType::Discover) => {
|
||||||
|
|
||||||
// Otherwise prefer existing (including expired if available)
|
// Otherwise prefer existing (including expired if available)
|
||||||
if let Some(ip) = self.current_lease(&in_packet.chaddr) {
|
if let Some(ip) = self.current_lease(&in_packet.chaddr) {
|
||||||
println!("Sending Reply to discover");
|
println!("Sending Reply to discover");
|
||||||
@@ -126,7 +124,13 @@ impl server::Handler for MyServer {
|
|||||||
nak(server, in_packet, "Requested IP not available");
|
nak(server, in_packet, "Requested IP not available");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
self.leases.insert(req_ip, (in_packet.chaddr, Some(Instant::now().add(self.lease_duration))));
|
self.leases.insert(
|
||||||
|
req_ip,
|
||||||
|
(
|
||||||
|
in_packet.chaddr,
|
||||||
|
Some(Instant::now().add(self.lease_duration)),
|
||||||
|
),
|
||||||
|
);
|
||||||
println!("Sending Reply to Request");
|
println!("Sending Reply to Request");
|
||||||
reply(server, options::MessageType::Ack, in_packet, &req_ip);
|
reply(server, options::MessageType::Ack, in_packet, &req_ip);
|
||||||
}
|
}
|
||||||
@@ -160,7 +164,6 @@ impl MyServer {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn current_lease(&self, chaddr: &[u8; 6]) -> Option<Ipv4Addr> {
|
fn current_lease(&self, chaddr: &[u8; 6]) -> Option<Ipv4Addr> {
|
||||||
|
|
||||||
for (i, v) in &self.leases {
|
for (i, v) in &self.leases {
|
||||||
if v.0 == *chaddr {
|
if v.0 == *chaddr {
|
||||||
return Some(*i);
|
return Some(*i);
|
||||||
|
|||||||
@@ -19,8 +19,6 @@ pub enum ErrorKind {
|
|||||||
|
|
||||||
type IResult<I, O> = Result<(I, O), CustomErr<I>>;
|
type IResult<I, O> = Result<(I, O), CustomErr<I>>;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/// DHCP Packet Structure
|
/// DHCP Packet Structure
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
pub struct Packet {
|
pub struct Packet {
|
||||||
@@ -87,10 +85,12 @@ pub fn decode_option(input: &[u8]) -> IResult<&[u8], DhcpOption> {
|
|||||||
let (input, len) = custom_be_u8(input)?;
|
let (input, len) = custom_be_u8(input)?;
|
||||||
let (input, data) = custom_take(len.into())(input)?;
|
let (input, data) = custom_take(len.into())(input)?;
|
||||||
let option = match code {
|
let option = match code {
|
||||||
DHCP_MESSAGE_TYPE => DhcpOption::DhcpMessageType(match MessageType::from(custom_be_u8(data)?.1) {
|
DHCP_MESSAGE_TYPE => {
|
||||||
|
DhcpOption::DhcpMessageType(match MessageType::from(custom_be_u8(data)?.1) {
|
||||||
Ok(x) => x,
|
Ok(x) => x,
|
||||||
Err(_) => return Err(CustomErr::UnrecognizedMessageType),
|
Err(_) => return Err(CustomErr::UnrecognizedMessageType),
|
||||||
}),
|
})
|
||||||
|
}
|
||||||
SERVER_IDENTIFIER => DhcpOption::ServerIdentifier(decode_ipv4(data)?.1),
|
SERVER_IDENTIFIER => DhcpOption::ServerIdentifier(decode_ipv4(data)?.1),
|
||||||
PARAMETER_REQUEST_LIST => DhcpOption::ParameterRequestList(data.to_vec()),
|
PARAMETER_REQUEST_LIST => DhcpOption::ParameterRequestList(data.to_vec()),
|
||||||
REQUESTED_IP_ADDRESS => DhcpOption::RequestedIpAddress(decode_ipv4(data)?.1),
|
REQUESTED_IP_ADDRESS => DhcpOption::RequestedIpAddress(decode_ipv4(data)?.1),
|
||||||
@@ -174,7 +174,6 @@ fn decode(input: &[u8]) -> IResult<&[u8], Packet> {
|
|||||||
let (input, giaddr) = decode_ipv4(input)?;
|
let (input, giaddr) = decode_ipv4(input)?;
|
||||||
|
|
||||||
if hlen != 6 {
|
if hlen != 6 {
|
||||||
|
|
||||||
return Err(CustomErr::InvalidHlen);
|
return Err(CustomErr::InvalidHlen);
|
||||||
}
|
}
|
||||||
let (_, chaddr) = custom_take(6usize)(input)?;
|
let (_, chaddr) = custom_take(6usize)(input)?;
|
||||||
@@ -182,7 +181,6 @@ fn decode(input: &[u8]) -> IResult<&[u8], Packet> {
|
|||||||
let input = options_input;
|
let input = options_input;
|
||||||
let (input, _) = custom_tag(&COOKIE)(input)?;
|
let (input, _) = custom_tag(&COOKIE)(input)?;
|
||||||
|
|
||||||
|
|
||||||
let mut options = Vec::new();
|
let mut options = Vec::new();
|
||||||
let mut rest = input;
|
let mut rest = input;
|
||||||
|
|
||||||
@@ -223,7 +221,6 @@ fn decode(input: &[u8]) -> IResult<&[u8], Packet> {
|
|||||||
|
|
||||||
impl Packet {
|
impl Packet {
|
||||||
pub fn from(input: &[u8]) -> Result<Packet, CustomErr<&[u8]>> {
|
pub fn from(input: &[u8]) -> Result<Packet, CustomErr<&[u8]>> {
|
||||||
|
|
||||||
Ok(decode(input)?.1)
|
Ok(decode(input)?.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -84,7 +84,6 @@ impl Server {
|
|||||||
server_ip,
|
server_ip,
|
||||||
broadcast_ip,
|
broadcast_ip,
|
||||||
src: SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 0),
|
src: SocketAddr::new(IpAddr::V4(Ipv4Addr::new(0, 0, 0, 0)), 0),
|
||||||
|
|
||||||
};
|
};
|
||||||
loop {
|
loop {
|
||||||
match s.socket.recv_from(&mut in_buf) {
|
match s.socket.recv_from(&mut in_buf) {
|
||||||
@@ -154,9 +153,7 @@ impl Server {
|
|||||||
/// Checks the packet see if it was intended for this DHCP server (as opposed to some other also on the network).
|
/// Checks the packet see if it was intended for this DHCP server (as opposed to some other also on the network).
|
||||||
pub fn for_this_server(&self, packet: &Packet) -> bool {
|
pub fn for_this_server(&self, packet: &Packet) -> bool {
|
||||||
match packet.option(options::SERVER_IDENTIFIER) {
|
match packet.option(options::SERVER_IDENTIFIER) {
|
||||||
Some(DhcpOption::ServerIdentifier(x)) => {
|
Some(DhcpOption::ServerIdentifier(x)) => x == &self.server_ip,
|
||||||
x == &self.server_ip
|
|
||||||
},
|
|
||||||
_ => false,
|
_ => false,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user