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

use base64::{decode, encode};
use hyperx;
use hyperx::header::Scheme;
use std::fmt;
use std::str::FromStr;

#[derive(Clone, PartialEq, Debug)]
pub struct SrdAuthorizationScheme {
    pub msg: Vec<u8>,
}

impl Scheme for SrdAuthorizationScheme {
    fn scheme() -> Option<&'static str> {
        Some("SRD")
    }

    fn fmt_scheme(&self, f: &mut fmt::Formatter) -> fmt::Result {
        f.write_str(&encode(&self.msg))
    }
}

impl FromStr for SrdAuthorizationScheme {
    type Err = hyperx::Error;

    fn from_str(s: &str) -> Result<SrdAuthorizationScheme, Self::Err> {
        match decode(s) {
            Ok(msg) => Ok(SrdAuthorizationScheme { msg }),
            Err(e) => {
                error!("SrdAuthorizationScheme::from_str base64 error: {}", e);
                Err(hyperx::Error::Header)
            }
        }
    }
}