You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
160 lines
3.6 KiB
160 lines
3.6 KiB
use std::io;
|
|
|
|
use crypto::md5::Md5;
|
|
use crypto::digest::Digest;
|
|
|
|
use super::constants::*;
|
|
use super::super::packet::Packet;
|
|
|
|
trait WriteToPacket {
|
|
fn write_to_packet(&self, &mut Packet) -> io::Result<()>;
|
|
}
|
|
|
|
/*================*
|
|
* SERVER REQUEST *
|
|
*================*/
|
|
|
|
#[derive(Debug)]
|
|
pub enum ServerRequest {
|
|
LoginRequest(LoginRequest),
|
|
PeerAddressRequest(PeerAddressRequest),
|
|
RoomListRequest(RoomListRequest),
|
|
SetListenPortRequest(SetListenPortRequest),
|
|
}
|
|
|
|
impl ServerRequest {
|
|
pub fn to_packet(&self) -> io::Result<Packet> {
|
|
let (mut packet, request): (Packet, &WriteToPacket) = match *self {
|
|
ServerRequest::LoginRequest(ref request) =>
|
|
(Packet::new(CODE_LOGIN), request),
|
|
|
|
ServerRequest::PeerAddressRequest(ref request) =>
|
|
(Packet::new(CODE_PEER_ADDRESS), request),
|
|
|
|
ServerRequest::RoomListRequest(ref request) =>
|
|
(Packet::new(CODE_ROOM_LIST), request),
|
|
|
|
ServerRequest::SetListenPortRequest(ref request) =>
|
|
(Packet::new(CODE_SET_LISTEN_PORT), request),
|
|
};
|
|
try!(request.write_to_packet(&mut packet));
|
|
Ok(packet)
|
|
}
|
|
}
|
|
|
|
fn md5_str(string: &str) -> String {
|
|
let mut hasher = Md5::new();
|
|
hasher.input_str(string);
|
|
hasher.result_str()
|
|
}
|
|
|
|
/*=======*
|
|
* LOGIN *
|
|
*=======*/
|
|
|
|
#[derive(Debug)]
|
|
pub struct LoginRequest {
|
|
username: String,
|
|
password: String,
|
|
major: u32,
|
|
minor: u32,
|
|
}
|
|
|
|
impl LoginRequest {
|
|
pub fn new(username: &str, password: &str, major: u32, minor: u32)
|
|
-> Result<Self, &'static str> {
|
|
if password.len() > 0 {
|
|
Ok(LoginRequest {
|
|
username: username.to_string(),
|
|
password: password.to_string(),
|
|
major: major,
|
|
minor: minor,
|
|
})
|
|
} else {
|
|
Err("Empty password")
|
|
}
|
|
}
|
|
}
|
|
|
|
impl WriteToPacket for LoginRequest {
|
|
fn write_to_packet(&self, packet: &mut Packet) -> io::Result<()> {
|
|
let userpass = String::new() + &self.username + &self.password;
|
|
let userpass_md5 = md5_str(&userpass);
|
|
|
|
try!(packet.write_str(&self.username));
|
|
try!(packet.write_str(&self.password));
|
|
try!(packet.write_uint(self.major));
|
|
try!(packet.write_str(&userpass_md5));
|
|
try!(packet.write_uint(self.minor));
|
|
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/*==============*
|
|
* PEER ADDRESS *
|
|
*==============*/
|
|
|
|
#[derive(Debug)]
|
|
pub struct PeerAddressRequest {
|
|
username: String,
|
|
}
|
|
|
|
impl PeerAddressRequest {
|
|
fn new(username: &str) -> Self {
|
|
PeerAddressRequest {
|
|
username: username.to_string(),
|
|
}
|
|
}
|
|
}
|
|
|
|
impl WriteToPacket for PeerAddressRequest {
|
|
fn write_to_packet(&self, packet: &mut Packet) -> io::Result<()> {
|
|
try!(packet.write_str(&self.username));
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/*===========*
|
|
* ROOM LIST *
|
|
*===========*/
|
|
|
|
#[derive(Debug)]
|
|
pub struct RoomListRequest;
|
|
|
|
impl RoomListRequest {
|
|
pub fn new() -> Self {
|
|
RoomListRequest
|
|
}
|
|
}
|
|
|
|
impl WriteToPacket for RoomListRequest {
|
|
fn write_to_packet(&self, _: &mut Packet) -> io::Result<()> {
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/*=================*
|
|
* SET LISTEN PORT *
|
|
*=================*/
|
|
|
|
#[derive(Debug)]
|
|
pub struct SetListenPortRequest {
|
|
port: u16,
|
|
}
|
|
|
|
impl SetListenPortRequest {
|
|
fn new(port: u16) -> Self {
|
|
SetListenPortRequest {
|
|
port: port,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl WriteToPacket for SetListenPortRequest {
|
|
fn write_to_packet(&self, packet: &mut Packet) -> io::Result<()> {
|
|
try!(packet.write_uint(self.port as u32));
|
|
Ok(())
|
|
}
|
|
}
|
|
|