From 3bdae21716d10032f70a0a889070766bbbe4d141 Mon Sep 17 00:00:00 2001 From: Max Audron Date: Sat, 26 Sep 2020 12:01:27 +0200 Subject: split handshake.rs --- src/message/handshake/clientlogin.rs | 49 ++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 src/message/handshake/clientlogin.rs (limited to 'src/message/handshake/clientlogin.rs') diff --git a/src/message/handshake/clientlogin.rs b/src/message/handshake/clientlogin.rs new file mode 100644 index 0000000..769245b --- /dev/null +++ b/src/message/handshake/clientlogin.rs @@ -0,0 +1,49 @@ +use crate::error::ProtocolError; +use crate::primitive::{Variant, VariantMap}; +use crate::{HandshakeDeserialize, HandshakeSerialize}; + +use failure::Error; + +/// Login to the core with user data +/// username and password are transmitted in plain text +#[derive(Debug)] +pub struct ClientLogin { + pub user: String, + pub password: String, +} + +impl HandshakeSerialize for ClientLogin { + fn serialize(&self) -> Result, Error> { + let mut values: VariantMap = VariantMap::new(); + values.insert( + "MsgType".to_string(), + Variant::String("ClientLogin".to_string()), + ); + values.insert("User".to_string(), Variant::String(self.user.clone())); + values.insert( + "Password".to_string(), + Variant::String(self.password.clone()), + ); + return HandshakeSerialize::serialize(&values); + } +} + +impl HandshakeDeserialize for ClientLogin { + fn parse(b: &[u8]) -> Result<(usize, Self), Error> { + let (len, values): (usize, VariantMap) = HandshakeDeserialize::parse(b)?; + + let msgtype = match_variant!(&values["MsgType"], Variant::StringUTF8); + + if msgtype == "ClientLogin" { + return Ok(( + len, + Self { + user: match_variant!(values["User"], Variant::String), + password: match_variant!(values["Password"], Variant::String), + }, + )); + } else { + bail!(ProtocolError::WrongMsgType); + } + } +} -- cgit v1.2.3