use crate::error::Error;
use crate::payload_util::require_at_least;
use alloc::vec::Vec;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[repr(u8)]
#[allow(missing_docs)]
pub enum XwrMode00 {
ReadMode = 0x01,
SetMode = 0x02,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
#[repr(u8)]
#[allow(missing_docs)]
pub enum XwrMode01 {
TransmitApdu = 0x01,
ConnectionDone = 0x02,
SecurePinEntry = 0x03,
SmartCardScan = 0x04,
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct XWrite {
pub mode: u8,
pub payload: Vec<u8>,
}
impl XWrite {
pub fn encode(&self) -> Result<Vec<u8>, Error> {
let mut out = Vec::with_capacity(1 + self.payload.len());
out.push(self.mode);
out.extend_from_slice(&self.payload);
Ok(out)
}
pub fn decode(data: &[u8]) -> Result<Self, Error> {
require_at_least(data, 1, 0xA1)?;
Ok(Self {
mode: data[0],
payload: data[1..].to_vec(),
})
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn roundtrip() {
let body = XWrite {
mode: 0x01,
payload: alloc::vec![0x02, 0xDE, 0xAD],
};
let bytes = body.encode().unwrap();
assert_eq!(bytes, [0x01, 0x02, 0xDE, 0xAD]);
assert_eq!(XWrite::decode(&bytes).unwrap(), body);
}
#[test]
fn mode_only_is_valid() {
let body = XWrite {
mode: 0x00,
payload: Vec::new(),
};
let bytes = body.encode().unwrap();
assert_eq!(bytes, [0x00]);
assert_eq!(XWrite::decode(&bytes).unwrap(), body);
}
#[test]
fn empty_decode_rejected() {
assert!(matches!(
XWrite::decode(&[]),
Err(Error::PayloadTooShort { code: 0xA1, .. })
));
}
}