pfctl/rule/
route.rs

1// Copyright 2025 Mullvad VPN AB.
2//
3// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
4// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
5// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
6// option. This file may not be copied, modified, or distributed
7// except according to those terms.
8
9use crate::{ffi, pooladdr::PoolAddr};
10
11#[derive(Debug, Default, Clone, PartialEq, Eq, Hash)]
12pub enum Route {
13    #[default]
14    NoRoute,
15    RouteTo(PoolAddr),
16    ReplyTo(PoolAddr),
17    DupTo(PoolAddr),
18}
19
20impl Route {
21    pub fn route_to<T: Into<PoolAddr>>(pool_addr: T) -> Self {
22        Route::RouteTo(pool_addr.into())
23    }
24
25    pub fn reply_to<T: Into<PoolAddr>>(pool_addr: T) -> Self {
26        Route::ReplyTo(pool_addr.into())
27    }
28
29    pub fn dup_to<T: Into<PoolAddr>>(pool_addr: T) -> Self {
30        Route::DupTo(pool_addr.into())
31    }
32
33    pub fn get_pool_addr(&self) -> Option<&PoolAddr> {
34        match *self {
35            Route::NoRoute => None,
36            Route::RouteTo(ref pool_addr) => Some(pool_addr),
37            Route::ReplyTo(ref pool_addr) => Some(pool_addr),
38            Route::DupTo(ref pool_addr) => Some(pool_addr),
39        }
40    }
41}
42
43impl<'a> From<&'a Route> for u8 {
44    fn from(route: &'a Route) -> u8 {
45        match *route {
46            Route::NoRoute => ffi::pfvar::PF_NOPFROUTE as u8,
47            Route::RouteTo(_) => ffi::pfvar::PF_ROUTETO as u8,
48            Route::ReplyTo(_) => ffi::pfvar::PF_REPLYTO as u8,
49            Route::DupTo(_) => ffi::pfvar::PF_DUPTO as u8,
50        }
51    }
52}