#[allow(dead_code)] pub const VOID: u32 = 0x00000000; pub const BOOL: u32 = 0x00000001; pub const QCHAR: u32 = 0x00000007; pub const QVARIANT: u32 = 0x00000090; pub const QVARIANTMAP: u32 = 0x00000008; pub const QVARIANTLIST: u32 = 0x00000009; pub const QSTRING: u32 = 0x0000000a; pub const QSTRINGLIST: u32 = 0x0000000b; pub const QBYTEARRAY: u32 = 0x0000000c; pub const QTIME: u32 = 0x0000000f; pub const QDATETIME: u32 = 0x00000010; pub const USERTYPE: u32 = 0x0000007f; // Basic types pub const LONG: u32 = 0x00000081; // int64_t pub const INT: u32 = 0x00000002; // int32_t pub const SHORT: u32 = 0x00000082; // int16_t pub const CHAR: u32 = 0x00000083; // int8_t pub const ULONG: u32 = 0x00000084; // uint64_t pub const UINT: u32 = 0x00000003; // uint32_t pub const USHORT: u32 = 0x00000085; // uint16_t pub const UCHAR: u32 = 0x00000086; // uint8_t extern crate byteorder; use std::io::Cursor; use byteorder::{ByteOrder, BigEndian, ReadBytesExt}; use std::io::Read; use std::vec::Vec; use std::net::TcpStream; use std::convert::TryInto; use crate::util; use crate::protocol::primitive::{deserialize, serialize, qread}; impl deserialize::Deserialize for bool { fn parse(b: &[u8]) -> (usize, Self) { if b[0] == 0 { return (1, false) } else { return (1, true) }; } } impl qread::QRead for bool { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut [b[0]]).unwrap() } } impl deserialize::Deserialize for u64 { fn parse(b: &[u8]) -> (usize, Self) { return (8, BigEndian::read_u64(b)); } } impl qread::QRead for u64 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut b[0..8]).unwrap() } } impl deserialize::Deserialize for u32 { fn parse(b: &[u8]) -> (usize, Self) { // self = &rdr.read_u32::().unwrap() return (4, BigEndian::read_u32(b)); } } impl qread::QRead for u32 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut b[0..4]).unwrap() } } impl deserialize::Deserialize for u16 { fn parse(b: &[u8]) -> (usize, Self) { return (2, BigEndian::read_u16(b)); } } impl qread::QRead for u16 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut b[0..2]).unwrap() } } impl deserialize::Deserialize for u8 { fn parse(b: &[u8]) -> (usize, Self) { return (1, b[0]); } } impl qread::QRead for u8 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut [b[0]]).unwrap() } } impl deserialize::Deserialize for i64 { fn parse(b: &[u8]) -> (usize, Self) { return (8, BigEndian::read_i64(b)); } } impl qread::QRead for i64 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut b[0..8]).unwrap() } } impl deserialize::Deserialize for i32 { fn parse(b: &[u8]) -> (usize, Self) { let mut rdr = Cursor::new(&b[0..4]); return (4, rdr.read_i32::().unwrap()); } } impl qread::QRead for i32 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut b[0..4]).unwrap() } } impl deserialize::Deserialize for i16 { fn parse(b: &[u8]) -> (usize, Self) { return (2, BigEndian::read_i16(b)); } } impl qread::QRead for i16 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut b[0..2]).unwrap() } } impl deserialize::Deserialize for i8 { fn parse(b: &[u8]) -> (usize, Self) { return (1, b[0].try_into().unwrap()); } } impl qread::QRead for i8 { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { s.read(&mut [b[0]]).unwrap() } } pub type String = std::string::String; impl serialize::Serialize for String { fn serialize(&self) -> Vec { let mut res: Vec = Vec::new(); let utf16: Vec = self.encode_utf16().collect(); for i in utf16 { res.extend(i.to_be_bytes().iter()); } util::prepend_byte_len(&mut res); return res; } } impl serialize::SerializeUTF8 for String { fn serialize_utf8(&self) -> Vec { let mut res: Vec = Vec::new(); res.extend(self.clone().into_bytes()); util::prepend_byte_len(&mut res); return res; } } impl deserialize::Deserialize for String { fn parse(b: &[u8]) -> (usize, Self) { let (_, len) = i32::parse(&b[0..4]); let ulen = len as usize; let mut pos: usize = 4; let mut chars: Vec = Vec::new(); loop { if pos >= (ulen + 4) { break; } let (slen, uchar) = u16::parse(&b[pos..(pos+2)]); chars.push(uchar); pos += slen; } let res: String = String::from_utf16(&chars).unwrap(); return (pos, res); } } impl deserialize::DeserializeUTF8 for String { fn parse_utf8(b: &[u8]) -> (usize, Self) { use crate::protocol::primitive::deserialize::Deserialize; let (_, len) = i32::parse(&b[0..4]); let ulen = len as usize; let res: String = String::from_utf8(b[4..(ulen+4)].to_vec()).unwrap(); return (ulen + 4, res); } } impl qread::QRead for String { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { use crate::protocol::primitive::deserialize::Deserialize; s.read(&mut b[0..4]).unwrap(); let (_, len) = i32::parse(&b[0..4]); let ulen = len as usize; s.read(&mut b[4..(4+ulen)]).unwrap(); return 4 + ulen; } } pub type StringList = Vec; impl serialize::Serialize for StringList { fn serialize(&self) -> Vec { let len: i32 = self.len().try_into().unwrap(); let mut res: Vec = Vec::new(); res.extend(len.to_be_bytes().iter()); for x in self { res.extend(x.serialize()); } return res; } } impl deserialize::Deserialize for StringList { fn parse(b: &[u8]) -> (usize, Self) { let (_, len) = i32::parse(&b[0..4]); let mut res: StringList = StringList::new(); let mut pos = 4; if len > 0 { for _ in 0..len { let (lpos, val) = String::parse(&b[pos..]); pos += lpos; res.push(val); } } return (pos, res); } } impl qread::QRead for StringList { fn read(s: &mut TcpStream, b: &mut [u8]) -> usize { use crate::protocol::primitive::deserialize::Deserialize; s.read(&mut b[0..4]).unwrap(); let (_, len) = i32::parse(&b[0..4]); let mut pos: usize = 0; for _ in 0..len { pos += String::read(s, &mut b[pos..]); } return 4 + pos; } }