Function hmac_new

crates/stdlib/src/hashlib.rs:759–798

(args: NewHMACHashArgs, vm: &VirtualMachine)

    #[pyfunction]
    fn hmac_new(args: NewHMACHashArgs, vm: &VirtualMachine) -> PyResult<PyHmac> {
        let digestmod = args
            .digestmod
            .into_option()
            .ok_or_else(|| vm.new_type_error("Missing required parameter 'digestmod'."))?;
        let name = resolve_digestmod(&digestmod, vm)?;

        let key_buf = args.key.borrow_buf();
        let msg_data = args.msg.flatten();

        macro_rules! make_hmac {
            ($hash_ty:ty) => {{
                let mut mac = <hmac::Hmac<$hash_ty> as Mac>::new_from_slice(&key_buf)
                    .map_err(|_| vm.new_value_error("invalid key length".to_owned()))?;
                if let Some(ref m) = msg_data {
                    m.with_ref(|bytes| Mac::update(&mut mac, bytes));
                }
                Ok(PyHmac {
                    algo_name: name,
                    digest_size: <$hash_ty as OutputSizeUser>::output_size(),
                    block_size: <$hash_ty as BlockSizeUser>::block_size(),
                    ctx: PyRwLock::new(Box::new(TypedHmac(mac))),
                })
            }};
        }

        match name.as_str() {
            "md5" => make_hmac!(Md5),
            "sha1" => make_hmac!(Sha1),
            "sha224" => make_hmac!(Sha224),
            "sha256" => make_hmac!(Sha256),
            "sha384" => make_hmac!(Sha384),
            "sha512" => make_hmac!(Sha512),
            "sha3_224" => make_hmac!(Sha3_224),
            "sha3_256" => make_hmac!(Sha3_256),
            "sha3_384" => make_hmac!(Sha3_384),
            "sha3_512" => make_hmac!(Sha3_512),
            _ => Err(unsupported_hash(&name, vm)),
        }
    }

    #[pyfunction]
    fn hmac_digest(args: HmacDigestArgs, vm: &VirtualMachine) -> PyResult<PyBytes> {