/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/. */

use std::{ffi::CString, process};

use windows_sys::Win32::Foundation::{ERROR_ACCESS_DENIED, ERROR_ADDRESS_ALREADY_ASSOCIATED};

use crate::{
    ipc_channel::IPCChannelError,
    ipc_listener::IPCListenerError,
    platform::windows::{server_addr, PlatformError},
    IPCConnector, IPCListener, Pid,
};

pub struct IPCChannel {
    listener: IPCListener,
    client_endpoint: IPCConnector,
    server_endpoint: IPCConnector,
}

impl IPCChannel {
    /// Create a new IPC channel for use between the browser main process and
    /// the crash helper. This includes a listener that will use the current
    /// process PID as part of its address and two connected endpoints.
    pub fn new() -> Result<IPCChannel, IPCChannelError> {
        let pid = process::id() as Pid;
        let mut listener = IPCListener::new(server_addr(pid))?;
        listener.listen()?;
        let client_endpoint = IPCConnector::connect(listener.address())?;
        let server_endpoint = listener.accept()?;

        Ok(IPCChannel {
            listener,
            client_endpoint,
            server_endpoint,
        })
    }

    /// Deconstruct the IPC channel, returning the listener, the connected
    /// server-side endpoint and the connected client-side endpoint.
    pub fn deconstruct(self) -> (IPCListener, IPCConnector, IPCConnector) {
        (self.listener, self.server_endpoint, self.client_endpoint)
    }
}

pub struct IPCClientChannel {
    client_endpoint: IPCConnector,
    server_endpoint: IPCConnector,
}

impl IPCClientChannel {
    /// Create a new IPC channel for use between one of the browser's child
    /// processes and the crash helper.
    pub fn new() -> Result<IPCClientChannel, IPCChannelError> {
        let mut listener = Self::create_listener()?;
        listener.listen()?;
        let client_endpoint = IPCConnector::connect(listener.address())?;
        let server_endpoint = listener.accept()?;

        Ok(IPCClientChannel {
            client_endpoint,
            server_endpoint,
        })
    }

    fn create_listener() -> Result<IPCListener, IPCListenerError> {
        const ATTEMPTS: u32 = 5;

        // We pick the listener name at random, as unlikely as it may be there
        // could be clashes so try a few times before giving up.
        for _i in 0..ATTEMPTS {
            let Ok(random_id) = getrandom::u64() else {
                continue;
            };

            let pipe_name = CString::new(format!(
                "\\\\.\\pipe\\gecko-crash-helper-child-pipe.{random_id:}"
            ))
            .unwrap();
            match IPCListener::new(pipe_name) {
                Ok(listener) => return Ok(listener),
                Err(
                    _error @ IPCListenerError::CreationError(PlatformError::CreatePipeFailure(
                        ERROR_ACCESS_DENIED,
                    )),
                ) => {} // Try again
                Err(error) => return Err(error),
            }
        }

        // If we got to this point give up.
        Err(IPCListenerError::CreationError(
            PlatformError::CreatePipeFailure(ERROR_ADDRESS_ALREADY_ASSOCIATED),
        ))
    }

    /// Deconstruct the IPC channel, returning the connected server-side
    /// endpoint and the connected client-side endpoint.
    pub fn deconstruct(self) -> (IPCConnector, IPCConnector) {
        (self.server_endpoint, self.client_endpoint)
    }
}