Method add_rule

port-forward/src/lib.rs:63–89

Add a forwarding rule. Returns error if the rule already exists.

(&mut self, rule: ForwardRule)

    /// Add a forwarding rule. Returns error if the rule already exists.
    pub fn add_rule(&mut self, rule: ForwardRule) -> Result<()> {
        if self.rules.contains_key(&rule) {
            bail!("rule already exists: {:?}", rule);
        }

        let token = self.cancel.child_token();
        let listen = rule.listen_sock();
        let target = rule.target_sock();

        let task = match rule.protocol {
            Protocol::Tcp => tokio::spawn(tcp::run_tcp_forwarder(listen, target, token.clone())),
            Protocol::Udp => tokio::spawn(udp::run_udp_forwarder(listen, target, token.clone())),
        };

        tracing::info!(
            "added forwarding rule: {listen} -> {target} ({:?})",
            rule.protocol
        );
        self.rules.insert(
            rule,
            RunningRule {
                cancel: token,
                task,
            },
        );
        Ok(())
    }

    /// Remove a forwarding rule and stop its task.
    pub async fn remove_rule(&mut self, rule: &ForwardRule) -> Result<()> {