-
Notifications
You must be signed in to change notification settings - Fork 40
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Add db table for tracking nat entries * Add endpoint for retrieving changesets * Update instance sagas to update table and trigger RPW * Periodically cleanup soft-deleted entries that no longer need to be sync'd by dendrite. The other half of the RPW lives in Dendrite. It will periodically check for a changeset, or check for a changeset when the trigger endpoint is called by the relevant saga / nexus operation.
- Loading branch information
1 parent
9700d44
commit 5f61f27
Showing
25 changed files
with
1,225 additions
and
79 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,116 @@ | ||
use std::net::{Ipv4Addr, Ipv6Addr}; | ||
|
||
use super::MacAddr; | ||
use crate::{ | ||
schema::{ipv4_nat_entry, nat_gen}, | ||
SqlU16, SqlU32, Vni, | ||
}; | ||
use chrono::{DateTime, Utc}; | ||
use omicron_common::api::external; | ||
use schemars::JsonSchema; | ||
use serde::Serialize; | ||
use uuid::Uuid; | ||
|
||
// TODO correctness | ||
// If we're not going to store ipv4 and ipv6 | ||
// NAT entries in the same table, and we don't | ||
// need any of the special properties of the IpNetwork | ||
// column type, does it make sense to use a different | ||
// column type? | ||
/// Database representation of an Ipv4 NAT Entry. | ||
#[derive(Insertable, Debug, Clone)] | ||
#[diesel(table_name = ipv4_nat_entry)] | ||
pub struct Ipv4NatValues { | ||
pub external_address: ipnetwork::IpNetwork, | ||
pub first_port: SqlU16, | ||
pub last_port: SqlU16, | ||
pub sled_address: ipnetwork::IpNetwork, | ||
pub vni: Vni, | ||
pub mac: MacAddr, | ||
} | ||
|
||
// TODO correctness | ||
// If we're not going to store ipv4 and ipv6 | ||
// NAT entries in the same table, we should probably | ||
// make the types more restrictive to prevent an | ||
// accidental ipv6 entry from being created. | ||
#[derive(Queryable, Debug, Clone, Selectable)] | ||
#[diesel(table_name = ipv4_nat_entry)] | ||
pub struct Ipv4NatEntry { | ||
pub id: Uuid, | ||
pub external_address: ipnetwork::IpNetwork, | ||
pub first_port: SqlU16, | ||
pub last_port: SqlU16, | ||
pub sled_address: ipnetwork::IpNetwork, | ||
pub vni: Vni, | ||
pub mac: MacAddr, | ||
pub gen: SqlU32, | ||
pub time_created: DateTime<Utc>, | ||
pub time_deleted: Option<DateTime<Utc>>, | ||
} | ||
|
||
impl Ipv4NatEntry { | ||
pub fn first_port(&self) -> u16 { | ||
self.first_port.into() | ||
} | ||
|
||
pub fn last_port(&self) -> u16 { | ||
self.last_port.into() | ||
} | ||
|
||
pub fn gen(&self) -> u32 { | ||
self.gen.into() | ||
} | ||
} | ||
|
||
#[derive(Queryable, Debug, Clone, Selectable)] | ||
#[diesel(table_name = nat_gen)] | ||
pub struct Ipv4NatGen { | ||
pub last_value: SqlU32, | ||
pub log_cnt: SqlU32, | ||
pub is_called: bool, | ||
} | ||
|
||
/// NAT Record | ||
#[derive(Clone, Debug, Serialize, JsonSchema)] | ||
pub struct Ipv4NatEntryView { | ||
pub external_address: Ipv4Addr, | ||
pub first_port: u16, | ||
pub last_port: u16, | ||
pub sled_address: Ipv6Addr, | ||
pub vni: external::Vni, | ||
pub mac: external::MacAddr, | ||
pub gen: u32, | ||
pub deleted: bool, | ||
} | ||
|
||
impl From<Ipv4NatEntry> for Ipv4NatEntryView { | ||
fn from(value: Ipv4NatEntry) -> Self { | ||
let external_address = match value.external_address.ip() { | ||
std::net::IpAddr::V4(a) => a, | ||
std::net::IpAddr::V6(_) => unreachable!(), | ||
}; | ||
|
||
let sled_address = match value.sled_address.ip() { | ||
std::net::IpAddr::V4(_) => unreachable!(), | ||
std::net::IpAddr::V6(a) => a, | ||
}; | ||
|
||
Self { | ||
external_address, | ||
first_port: value.first_port(), | ||
last_port: value.last_port(), | ||
sled_address, | ||
vni: value.vni.0, | ||
mac: *value.mac, | ||
gen: value.gen(), | ||
deleted: value.time_deleted.is_some(), | ||
} | ||
} | ||
} | ||
|
||
/// NAT Generation | ||
#[derive(Clone, Debug, Serialize, JsonSchema)] | ||
pub struct Ipv4NatGenView { | ||
pub gen: u32, | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.