use super::*;
use toxcore::binary_io::*;
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum InnerOnionResponse {
OnionAnnounceResponse(OnionAnnounceResponse),
OnionDataResponse(OnionDataResponse)
}
impl ToBytes for InnerOnionResponse {
fn to_bytes<'a>(&self, buf: (&'a mut [u8], usize)) -> Result<(&'a mut [u8], usize), GenError> {
match *self {
InnerOnionResponse::OnionAnnounceResponse(ref inner) => inner.to_bytes(buf),
InnerOnionResponse::OnionDataResponse(ref inner) => inner.to_bytes(buf),
}
}
}
impl FromBytes for InnerOnionResponse {
named!(from_bytes<InnerOnionResponse>, alt!(
map!(OnionAnnounceResponse::from_bytes, InnerOnionResponse::OnionAnnounceResponse) |
map!(OnionDataResponse::from_bytes, InnerOnionResponse::OnionDataResponse)
));
}
#[cfg(test)]
mod tests {
use super::*;
encode_decode_test!(
inner_onion_announce_response_encode_decode,
InnerOnionResponse::OnionAnnounceResponse(OnionAnnounceResponse {
sendback_data: 12345,
nonce: gen_nonce(),
payload: vec![42; 123]
})
);
encode_decode_test!(
inner_onion_data_response_encode_decode,
InnerOnionResponse::OnionDataResponse(OnionDataResponse {
nonce: gen_nonce(),
temporary_pk: gen_keypair().0,
payload: vec![42; 123]
})
);
}