sqlx/mason-mariadb/src/protocol/packets/ssl_request.rs

use super::super::{serialize::Serialize, types::Capabilities};
use crate::connection::Connection;
use bytes::Bytes;
use failure::Error;

#[derive(Default, Debug)]
pub struct SSLRequestPacket {
    pub capabilities: Capabilities,
    pub max_packet_size: u32,
    pub collation: u8,
    pub extended_capabilities: Option<Capabilities>,
}

impl Serialize for SSLRequestPacket {
    fn serialize<'a, 'b>(&self, conn: &mut Connection) -> Result<(), Error> {
        conn.encoder.encode_int_4(self.capabilities.bits() as u32);
        conn.encoder.encode_int_4(self.max_packet_size);
        conn.encoder.encode_int_1(self.collation);

        // Filler
        conn.encoder.encode_byte_fix(&Bytes::from_static(&[0u8; 19]), 19);

        if !(conn.capabilities & Capabilities::CLIENT_MYSQL).is_empty()
            && !(self.capabilities & Capabilities::CLIENT_MYSQL).is_empty()
        {
            if let Some(capabilities) = self.extended_capabilities {
                conn.encoder.encode_int_4(capabilities.bits() as u32);
            }
        } else {
            conn.encoder.encode_byte_fix(&Bytes::from_static(&[0u8; 4]), 4);
        }

        Ok(())
    }
}