Jake Probst
6 years ago
6 changed files with 343 additions and 6 deletions
-
1Cargo.toml
-
12psopacket/Cargo.toml
-
255psopacket/src/lib.rs
-
21src/lib.rs
-
2src/patch/mod.rs
-
58src/patch/packet.rs
@ -0,0 +1,12 @@ |
|||
[package] |
|||
name = "psopacket" |
|||
version = "1.0.0" |
|||
authors = ["Jake Probst <jake.probst@gmail.com>"] |
|||
edition = "2018" |
|||
|
|||
[lib] |
|||
proc-macro = true |
|||
|
|||
[dependencies] |
|||
syn = {version = "0.15", features=["full", "extra-traits", "parsing"]} |
|||
quote = "0.6" |
|||
@ -0,0 +1,255 @@ |
|||
#![recursion_limit="128"]
|
|||
|
|||
extern crate proc_macro;
|
|||
|
|||
use proc_macro::TokenStream;
|
|||
use syn::{parse_macro_input, ItemStruct};
|
|||
use quote::quote;
|
|||
|
|||
#[proc_macro_attribute]
|
|||
pub fn pso_packet(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|||
let arg = parse_macro_input!(attr as syn::LitInt);
|
|||
let pkt_cmd = arg.value() as u16;
|
|||
|
|||
let parsed = parse_macro_input!(item as ItemStruct);
|
|||
|
|||
let mut has_flag: bool = false;
|
|||
let mut from_bytes = Vec::new();
|
|||
let mut as_bytes = Vec::new();
|
|||
let mut dbg_write_vars = Vec::new();
|
|||
let mut partialeq = Vec::new();
|
|||
|
|||
for (i, f) in parsed.fields.iter().enumerate() {
|
|||
if let Some(ident) = &f.ident {
|
|||
let ident_str = ident.to_string();
|
|||
|
|||
match &f.ty {
|
|||
syn::Type::Array(arr) => {
|
|||
let array_length = if let syn::Expr::Lit(lit) = &arr.len {
|
|||
if let syn::Lit::Int(int) = &lit.lit {
|
|||
int.value() as usize
|
|||
}
|
|||
else {
|
|||
return syn::Error::new(arr.bracket_token.span, "unknown array size").to_compile_error().into();
|
|||
}
|
|||
} else {
|
|||
return syn::Error::new(arr.bracket_token.span, "unknown array size").to_compile_error().into();
|
|||
};
|
|||
match *arr.elem {
|
|||
syn::Type::Path(ref path) => {
|
|||
dbg_write_vars.push(quote! {
|
|||
write!(f, " {}: {:?}\n", #ident_str, self.#ident.iter()).unwrap();
|
|||
});
|
|||
as_bytes.push(quote! {
|
|||
for f in self.#ident.iter() {
|
|||
buf.extend_from_slice(&f.to_le_bytes())
|
|||
}
|
|||
});
|
|||
let ty = path.path.segments[0].ident.to_string();
|
|||
match ty.as_str() {
|
|||
"u8" => {
|
|||
from_bytes.push(quote! {
|
|||
#ident: {
|
|||
let mut b: [u8; #array_length] = [0; #array_length];
|
|||
if let Ok(len) = cur.read(&mut b) {
|
|||
if len != #array_length {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
}
|
|||
}
|
|||
else {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
};
|
|||
b
|
|||
},
|
|||
});
|
|||
},
|
|||
_ => {
|
|||
return syn::Error::new(path.path.segments[0].ident.span(), "type not supported")
|
|||
.to_compile_error().into();
|
|||
}
|
|||
}
|
|||
partialeq.push(quote! {
|
|||
if self.#ident[..] != other.#ident[..] {
|
|||
return false;
|
|||
}
|
|||
});
|
|||
}
|
|||
_ => {
|
|||
panic!("why");
|
|||
}
|
|||
}
|
|||
},
|
|||
syn::Type::Path(path) => {
|
|||
dbg_write_vars.push(quote! {
|
|||
write!(f, " {}: {:?}\n", #ident_str, self.#ident).unwrap();
|
|||
});
|
|||
as_bytes.push(quote! {
|
|||
buf.extend_from_slice(&self.#ident.to_le_bytes());
|
|||
});
|
|||
let ty = path.path.segments[0].ident.to_string();
|
|||
match ty.as_str() {
|
|||
"u8" => {
|
|||
from_bytes.push(quote! {
|
|||
#ident: {
|
|||
let mut b: [u8; 1] = [0; 1];
|
|||
if let Ok(len) = cur.read(&mut b) {
|
|||
if len != 1 {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
}
|
|||
}
|
|||
else {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
};
|
|||
b[0]
|
|||
},
|
|||
});
|
|||
},
|
|||
"u16" => {
|
|||
from_bytes.push(quote! {
|
|||
#ident: {
|
|||
let mut b: [u8; 2] = [0; 2];
|
|||
if let Ok(len) = cur.read(&mut b) {
|
|||
if len != 2 {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
}
|
|||
}
|
|||
else {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
};
|
|||
u16::from_le_bytes(b)
|
|||
},
|
|||
});
|
|||
},
|
|||
"u32" => {
|
|||
if ident_str == "flag" {
|
|||
if i != 0 {
|
|||
return syn::Error::new(ident.span(), "flag must be first member of struct").to_compile_error().into();
|
|||
}
|
|||
has_flag = true;
|
|||
continue;
|
|||
}
|
|||
from_bytes.push(quote! {
|
|||
#ident: {
|
|||
let mut b: [u8; 4] = [0; 4];
|
|||
if let Ok(len) = cur.read(&mut b) {
|
|||
if len != 4 {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
}
|
|||
}
|
|||
else {
|
|||
return Err(PacketParseError::NotEnoughBytes);
|
|||
};
|
|||
u32::from_le_bytes(b)
|
|||
},
|
|||
});
|
|||
},
|
|||
_ => {
|
|||
return syn::Error::new(path.path.segments[0].ident.span(), "type not supported")
|
|||
.to_compile_error().into();
|
|||
}
|
|||
}
|
|||
partialeq.push(quote! {
|
|||
if self.#ident != other.#ident {
|
|||
return false;
|
|||
}
|
|||
});
|
|||
}
|
|||
_ => {
|
|||
}
|
|||
}
|
|||
}
|
|||
}
|
|||
|
|||
let this_struct = parsed.ident.clone();
|
|||
let this_struct_str = this_struct.to_string();
|
|||
|
|||
let flag_write = if has_flag {
|
|||
quote! {
|
|||
buf.extend_from_slice(&u32::to_le_bytes(self.flag));
|
|||
}
|
|||
}
|
|||
else {
|
|||
quote! {
|
|||
buf.extend_from_slice(&u32::to_le_bytes(0));
|
|||
}
|
|||
};
|
|||
|
|||
let psopacket = quote! {
|
|||
impl PSOPacket for #this_struct {
|
|||
fn from_bytes(data: &Vec<u8>) -> Result<#this_struct, PacketParseError> {
|
|||
// TODO: assert cmd is correct
|
|||
let mut cur = std::io::Cursor::new(data);
|
|||
cur.seek(SeekFrom::Start(2));
|
|||
let mut b: [u8; 2] = [0; 2];
|
|||
cur.read(&mut b);
|
|||
let cmd = u16::from_le_bytes(b);
|
|||
|
|||
if cmd != #pkt_cmd {
|
|||
return Err(PacketParseError::WrongPacketCommand);
|
|||
}
|
|||
|
|||
if #has_flag {
|
|||
cur.seek(SeekFrom::Start(4));
|
|||
}
|
|||
else {
|
|||
cur.seek(SeekFrom::Start(8));
|
|||
}
|
|||
Ok(#this_struct {
|
|||
#(#from_bytes)*
|
|||
})
|
|||
}
|
|||
fn as_bytes(&self) -> Vec<u8> {
|
|||
let mut buf: Vec<u8> = Vec::new();
|
|||
#flag_write
|
|||
#(#as_bytes)*
|
|||
|
|||
let pkt_len = buf.len() as u16;
|
|||
let mut prebuf: Vec<u8> = Vec::new();
|
|||
|
|||
prebuf.extend_from_slice(&u16::to_le_bytes(pkt_len));
|
|||
prebuf.extend_from_slice(&u16::to_le_bytes(#pkt_cmd));
|
|||
prebuf.append(&mut buf);
|
|||
|
|||
prebuf
|
|||
}
|
|||
}
|
|||
};
|
|||
|
|||
|
|||
let psopacket_debug = quote! {
|
|||
impl std::fmt::Debug for #this_struct {
|
|||
fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
|
|||
write!(f, "packet {} {{\n", #this_struct_str).unwrap();
|
|||
#(#dbg_write_vars)*
|
|||
write!(f, "}}")
|
|||
}
|
|||
}
|
|||
};
|
|||
|
|||
|
|||
let psopacket_partialeq = quote! {
|
|||
impl std::cmp::PartialEq for #this_struct {
|
|||
fn eq(&self, other: &Self) -> bool {
|
|||
#(#partialeq)*
|
|||
true
|
|||
}
|
|||
}
|
|||
};
|
|||
|
|||
let q = quote! {
|
|||
#parsed
|
|||
#psopacket
|
|||
#psopacket_debug
|
|||
#psopacket_partialeq
|
|||
};
|
|||
|
|||
//println!("[[[{}]]]", q.to_string());
|
|||
|
|||
q.into()
|
|||
}
|
|||
|
|||
|
|||
#[proc_macro_attribute]
|
|||
pub fn game_command(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|||
item
|
|||
}
|
|||
@ -1,7 +1,16 @@ |
|||
#[cfg(test)]
|
|||
mod tests {
|
|||
#[test]
|
|||
fn it_works() {
|
|||
assert_eq!(2 + 2, 4);
|
|||
}
|
|||
mod patch;
|
|||
|
|||
#[derive(Debug, PartialEq)]
|
|||
pub enum PacketParseError {
|
|||
NotEnoughBytes,
|
|||
WrongPacketCommand,
|
|||
}
|
|||
|
|||
|
|||
pub trait PSOPacket {
|
|||
fn from_bytes(data: &Vec<u8>) -> Result<Self, PacketParseError> where Self: Sized;
|
|||
fn as_bytes(&self) -> Vec<u8>;
|
|||
}
|
|||
|
|||
|
|||
|
|||
@ -0,0 +1,2 @@ |
|||
|
|||
mod packet;
|
|||
@ -0,0 +1,58 @@ |
|||
use psopacket::pso_packet;
|
|||
use crate::{PSOPacket, PacketParseError};
|
|||
|
|||
use std::io::{Read, Seek, SeekFrom};
|
|||
|
|||
|
|||
#[pso_packet(0x02)]
|
|||
struct PatchWelcome {
|
|||
copyright: [u8; 44],
|
|||
padding: [u8; 20],
|
|||
server_key: u32,
|
|||
client_key: u32,
|
|||
}
|
|||
|
|||
impl PatchWelcome {
|
|||
fn new(server_key: u32, client_key: u32) -> PatchWelcome {
|
|||
PatchWelcome {
|
|||
copyright: b"Patch Server. Copyright SonicTeam, LTD. 2001".clone(),
|
|||
padding: [0; 20],
|
|||
server_key: server_key,
|
|||
client_key: client_key,
|
|||
}
|
|||
}
|
|||
}
|
|||
|
|||
enum PatchPackets {
|
|||
PatchWelcome(PatchWelcome)
|
|||
}
|
|||
|
|||
#[cfg(test)]
|
|||
mod tests {
|
|||
#[test]
|
|||
fn patch_welcome() {
|
|||
use super::PSOPacket;
|
|||
|
|||
let pkt = super::PatchWelcome::new(123, 456);
|
|||
|
|||
assert!(pkt.as_bytes() == vec![0x4C, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x50, 0x61, 0x74, 0x63, 0x68, 0x20, 0x53, 0x65,
|
|||
0x72, 0x76, 0x65, 0x72, 0x2E, 0x20, 0x43, 0x6F, 0x70, 0x79, 0x72, 0x69, 0x67, 0x68, 0x74, 0x20,
|
|||
0x53, 0x6F, 0x6E, 0x69, 0x63, 0x54, 0x65, 0x61, 0x6D, 0x2C, 0x20, 0x4C, 0x54, 0x44, 0x2E, 0x20,
|
|||
0x32, 0x30, 0x30, 0x31, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|||
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x7B, 0x00, 0x00, 0x00, 0xC8, 0x01, 0x00, 0x00,
|
|||
]);
|
|||
|
|||
|
|||
let mut bytes = pkt.as_bytes();
|
|||
bytes.splice(32..41, b"Elsewhere".iter().cloned());
|
|||
|
|||
let new_pkt = super::PatchWelcome::from_bytes(&bytes);
|
|||
|
|||
assert!(new_pkt == Ok(super::PatchWelcome {
|
|||
copyright: b"Patch Server. Copyright Elsewhere, LTD. 2001".clone(),
|
|||
padding: [0; 20],
|
|||
server_key: 123,
|
|||
client_key: 456,
|
|||
}))
|
|||
}
|
|||
}
|
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue