use crate::{
impl_command_display, impl_command_ops, impl_default, impl_message_from_buf, impl_message_ops,
len, CommandOps, MessageOps, MessageType, ModulusKey,
};
mod index {
pub const MODULUS: usize = 4;
pub const MODULUS_END: usize = 12;
}
#[repr(C)]
#[derive(Clone, Copy, Debug, PartialEq)]
pub struct SetModulusCommand {
buf: [u8; len::SET_MODULUS_COMMAND],
}
impl SetModulusCommand {
pub fn new() -> Self {
let mut msg = Self {
buf: [0u8; len::SET_MODULUS_COMMAND],
};
msg.init();
msg.set_command(MessageType::SetModulus);
msg
}
pub fn modulus(&self) -> ModulusKey {
let key_bytes: [u8; 8] = self.buf[index::MODULUS..index::MODULUS_END]
.try_into()
.unwrap();
u64::from_le_bytes(key_bytes).into()
}
pub fn set_modulus(&mut self, modulus: &ModulusKey) {
let key_bytes = modulus.as_inner().to_le_bytes();
self.buf[index::MODULUS..index::MODULUS_END].copy_from_slice(key_bytes.as_ref());
}
}
impl_default!(SetModulusCommand);
impl_command_display!(SetModulusCommand);
impl_message_from_buf!(SetModulusCommand);
impl_message_ops!(SetModulusCommand);
impl_command_ops!(SetModulusCommand);