mirror of
https://github.com/neondatabase/neon.git
synced 2026-05-17 21:20:37 +00:00
ec9079f483
## Problem The proxy denies using `unwrap()`s in regular code, but we want to use it in test code and so have to allow it for each test block. ## Summary of changes Set `allow-unwrap-in-tests = true` in clippy.toml and remove all exceptions.
69 lines
2.0 KiB
Rust
69 lines
2.0 KiB
Rust
//! Definitions for SASL messages.
|
|
|
|
use pq_proto::{BeAuthenticationSaslMessage, BeMessage};
|
|
|
|
use crate::parse::split_cstr;
|
|
|
|
/// SASL-specific payload of [`PasswordMessage`](pq_proto::FeMessage::PasswordMessage).
|
|
#[derive(Debug)]
|
|
pub(crate) struct FirstMessage<'a> {
|
|
/// Authentication method, e.g. `"SCRAM-SHA-256"`.
|
|
pub(crate) method: &'a str,
|
|
/// Initial client message.
|
|
pub(crate) message: &'a str,
|
|
}
|
|
|
|
impl<'a> FirstMessage<'a> {
|
|
// NB: FromStr doesn't work with lifetimes
|
|
pub(crate) fn parse(bytes: &'a [u8]) -> Option<Self> {
|
|
let (method_cstr, tail) = split_cstr(bytes)?;
|
|
let method = method_cstr.to_str().ok()?;
|
|
|
|
let (len_bytes, bytes) = tail.split_first_chunk()?;
|
|
let len = u32::from_be_bytes(*len_bytes) as usize;
|
|
if len != bytes.len() {
|
|
return None;
|
|
}
|
|
|
|
let message = std::str::from_utf8(bytes).ok()?;
|
|
Some(Self { method, message })
|
|
}
|
|
}
|
|
|
|
/// A single SASL message.
|
|
/// This struct is deliberately decoupled from lower-level
|
|
/// [`BeAuthenticationSaslMessage`].
|
|
#[derive(Debug)]
|
|
pub(super) enum ServerMessage<T> {
|
|
/// We expect to see more steps.
|
|
Continue(T),
|
|
/// This is the final step.
|
|
Final(T),
|
|
}
|
|
|
|
impl<'a> ServerMessage<&'a str> {
|
|
pub(super) fn to_reply(&self) -> BeMessage<'a> {
|
|
BeMessage::AuthenticationSasl(match self {
|
|
ServerMessage::Continue(s) => BeAuthenticationSaslMessage::Continue(s.as_bytes()),
|
|
ServerMessage::Final(s) => BeAuthenticationSaslMessage::Final(s.as_bytes()),
|
|
})
|
|
}
|
|
}
|
|
|
|
#[cfg(test)]
|
|
mod tests {
|
|
use super::*;
|
|
|
|
#[test]
|
|
fn parse_sasl_first_message() {
|
|
let proto = "SCRAM-SHA-256";
|
|
let sasl = "n,,n=,r=KHQ2Gjc7NptyB8aov5/TnUy4";
|
|
let sasl_len = (sasl.len() as u32).to_be_bytes();
|
|
let bytes = [proto.as_bytes(), &[0], sasl_len.as_ref(), sasl.as_bytes()].concat();
|
|
|
|
let password = FirstMessage::parse(&bytes).unwrap();
|
|
assert_eq!(password.method, proto);
|
|
assert_eq!(password.message, sasl);
|
|
}
|
|
}
|