esp-idf-svc 0.45.0

use embedded_svc::ipv4;

use crate::private::mutex::{Mutex, RawMutex};

use esp_idf_sys::*;

use crate::private::common::*;

#[derive(Debug)]
pub struct EspNapt(());

pub enum Protocol {
    UDP,
    TCP,
}

impl Protocol {
    fn get_num_proto(&self) -> u8 {
        match self {
            Self::UDP => 17,
            Self::TCP => 6,
        }
    }
}

static TAKEN: Mutex<bool> = Mutex::wrap(RawMutex::new(), false);

impl EspNapt {
    pub fn new() -> Result<Self, EspError> {
        let mut taken = TAKEN.lock();

        if *taken {
            esp!(ESP_ERR_INVALID_STATE as i32)?;
        }

        *taken = true;
        Ok(Self(()))
    }

    pub fn add_portmap(
        protocol: Protocol,
        external_ip: ipv4::Ipv4Addr,
        external_port: u16,
        internal_ip: ipv4::Ipv4Addr,
        internal_port: u16,
    ) -> bool {
        unsafe {
            ip_portmap_add(
                protocol.get_num_proto(),
                Newtype::<esp_ip4_addr_t>::from(external_ip).0.addr,
                external_port,
                Newtype::<esp_ip4_addr_t>::from(internal_ip).0.addr,
                internal_port,
            ) != 0
        }
    }

    pub fn remove_portmap(protocol: Protocol, external_port: u16) -> bool {
        unsafe { ip_portmap_remove(protocol.get_num_proto(), external_port) != 0 }
    }
}

impl Drop for EspNapt {
    fn drop(&mut self) {
        *TAKEN.lock() = false;
    }
}