aboutsummaryrefslogtreecommitdiff
path: root/azalea-protocol/src/packets/login/serverbound_key_packet.rs
blob: 23b3659b35a94a753cc11856930b4177558f07f3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
use azalea_buf::{BufReadError, McBuf};
use azalea_crypto::SaltSignaturePair;
use azalea_protocol_macros::ServerboundLoginPacket;
use std::io::{Cursor, Write};

use azalea_buf::{McBufReadable, McBufWritable};

#[derive(Clone, Debug, McBuf, ServerboundLoginPacket)]
pub struct ServerboundKeyPacket {
    pub key_bytes: Vec<u8>,
    pub nonce_or_salt_signature: NonceOrSaltSignature,
}

#[derive(Clone, Debug)]
pub enum NonceOrSaltSignature {
    Nonce(Vec<u8>),
    SaltSignature(SaltSignaturePair),
}

impl McBufReadable for NonceOrSaltSignature {
    fn read_from(buf: &mut Cursor<&[u8]>) -> Result<Self, BufReadError> {
        let is_nonce = bool::read_from(buf)?;
        if is_nonce {
            Ok(NonceOrSaltSignature::Nonce(Vec::<u8>::read_from(buf)?))
        } else {
            Ok(NonceOrSaltSignature::SaltSignature(
                SaltSignaturePair::read_from(buf)?,
            ))
        }
    }
}

impl McBufWritable for NonceOrSaltSignature {
    fn write_into(&self, buf: &mut impl Write) -> Result<(), std::io::Error> {
        match self {
            NonceOrSaltSignature::Nonce(nonce) => {
                bool::write_into(&true, buf)?;
                nonce.write_into(buf)?;
            }
            NonceOrSaltSignature::SaltSignature(salt_signature) => {
                bool::write_into(&false, buf)?;
                salt_signature.write_into(buf)?;
            }
        }
        Ok(())
    }
}