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

use crate::server::{Validator, Response, Mechanism, MechanismError};
use crate::common::Identity;
use crate::secret;

pub struct Plain<V: Validator<secret::Plain>> {
    validator: V,
}

impl<V: Validator<secret::Plain>> Plain<V> {
    pub fn new(validator: V) -> Plain<V> {
        Plain { validator: validator, }
    }
}

impl<V: Validator<secret::Plain>> Mechanism for Plain<V> {
    fn name(&self) -> &str {
        "PLAIN"
    }

    fn respond(&mut self, payload: &[u8]) -> Result<Response, MechanismError> {
        let mut sp = payload.split(|&b| b == 0);
        sp.next();
        let username = sp.next().ok_or_else(|| MechanismError::NoUsernameSpecified)?;
        let username = String::from_utf8(username.to_vec()).map_err(|_| MechanismError::ErrorDecodingUsername)?;
        let password = sp.next().ok_or_else(|| MechanismError::NoPasswordSpecified)?;
        let password = String::from_utf8(password.to_vec()).map_err(|_| MechanismError::ErrorDecodingPassword)?;
        let ident = Identity::Username(username);
        self.validator.validate(&ident, &secret::Plain(password))?;
        Ok(Response::Success(ident, Vec::new()))
    }
}