Struct pg_wire::HandShakeProcess

pub struct HandShakeProcess { /* fields omitted */ }

Encapsulate protocol hand shake process

Examples

use pg_wire::{HandShakeProcess, HandShakeStatus, HandShakeRequest};

let mut stream = accept_tcp_connection();
let mut process = HandShakeProcess::start();
let mut buffer: Option<Vec<u8>> = None;
loop {
    match process.next_stage(buffer.as_deref()) {
        Ok(HandShakeStatus::Requesting(HandShakeRequest::Buffer(len))) => {
            let mut buf = vec![b'0'; len];
            buffer = Some(stream.read(&mut buf));
        }
        Ok(HandShakeStatus::Requesting(HandShakeRequest::UpgradeToSsl)) => {
            stream.write_all(&[b'S']); // accepting tls connection from client
            stream = tls_stream(stream);
        }
        Ok(HandShakeStatus::Cancel(conn_id, secret_key)) => {
            handle_request_cancellation(conn_id, secret_key);
            break;
        }
        Ok(HandShakeStatus::Done(props)) => {
            handle_authentication_and_other_stuff();
            break;
        }
        Err(protocol_error) => {
            handle_protocol_error(protocol_error);
            break;
        }
    }
}

Implementations

impl Process

pub fn start() -> Process

Creates new process to make client <-> server hand shake

pub fn next_stage(&mut self, payload: Option<&[u8]>) -> Result<Status>

Proceed to the next stage of client <-> server hand shake