From eea7f13cea9f96e4ebc8176a3357922e25bef013 Mon Sep 17 00:00:00 2001
From: Sidney Cammeresi <sac@readyset.io>
Date: Fri, 25 Oct 2024 13:43:57 -0700
Subject: [PATCH] Add support for cidr 0.3 under separate feature

---
 postgres-types/Cargo.toml     |  2 ++
 postgres-types/src/cidr_03.rs | 44 +++++++++++++++++++++++++++++++++++
 postgres-types/src/lib.rs     |  2 ++
 3 files changed, 48 insertions(+)
 create mode 100644 postgres-types/src/cidr_03.rs

diff --git a/postgres-types/Cargo.toml b/postgres-types/Cargo.toml
index ec8cea4dc..d417664ea 100644
--- a/postgres-types/Cargo.toml
+++ b/postgres-types/Cargo.toml
@@ -16,6 +16,7 @@ array-impls = ["array-init"]
 js = ["postgres-protocol/js"]
 with-bit-vec-0_6 = ["bit-vec-06"]
 with-cidr-0_2 = ["cidr-02"]
+with-cidr-0_3 = ["cidr-03"]
 with-chrono-0_4 = ["chrono-04"]
 with-eui48-0_4 = ["eui48-04"]
 with-eui48-1 = ["eui48-1"]
@@ -41,6 +42,7 @@ chrono-04 = { version = "0.4.16", package = "chrono", default-features = false,
     "clock",
 ], optional = true }
 cidr-02 = { version = "0.2", package = "cidr", optional = true }
+cidr-03 = { version = "0.3", package = "cidr", optional = true }
 # eui48-04 will stop compiling and support will be removed
 # See https://github.com/sfackler/rust-postgres/issues/1073
 eui48-04 = { version = "0.4", package = "eui48", optional = true }
diff --git a/postgres-types/src/cidr_03.rs b/postgres-types/src/cidr_03.rs
new file mode 100644
index 000000000..6a0178711
--- /dev/null
+++ b/postgres-types/src/cidr_03.rs
@@ -0,0 +1,44 @@
+use bytes::BytesMut;
+use cidr_03::{IpCidr, IpInet};
+use postgres_protocol::types;
+use std::error::Error;
+
+use crate::{FromSql, IsNull, ToSql, Type};
+
+impl<'a> FromSql<'a> for IpCidr {
+    fn from_sql(_: &Type, raw: &[u8]) -> Result<Self, Box<dyn Error + Sync + Send>> {
+        let inet = types::inet_from_sql(raw)?;
+        Ok(IpCidr::new(inet.addr(), inet.netmask())?)
+    }
+
+    accepts!(CIDR);
+}
+
+impl ToSql for IpCidr {
+    fn to_sql(&self, _: &Type, w: &mut BytesMut) -> Result<IsNull, Box<dyn Error + Sync + Send>> {
+        types::inet_to_sql(self.first_address(), self.network_length(), w);
+        Ok(IsNull::No)
+    }
+
+    accepts!(CIDR);
+    to_sql_checked!();
+}
+
+impl<'a> FromSql<'a> for IpInet {
+    fn from_sql(_: &Type, raw: &[u8]) -> Result<Self, Box<dyn Error + Sync + Send>> {
+        let inet = types::inet_from_sql(raw)?;
+        Ok(IpInet::new(inet.addr(), inet.netmask())?)
+    }
+
+    accepts!(INET);
+}
+
+impl ToSql for IpInet {
+    fn to_sql(&self, _: &Type, w: &mut BytesMut) -> Result<IsNull, Box<dyn Error + Sync + Send>> {
+        types::inet_to_sql(self.address(), self.network_length(), w);
+        Ok(IsNull::No)
+    }
+
+    accepts!(INET);
+    to_sql_checked!();
+}
diff --git a/postgres-types/src/lib.rs b/postgres-types/src/lib.rs
index 6ad2eff50..c5593cff8 100644
--- a/postgres-types/src/lib.rs
+++ b/postgres-types/src/lib.rs
@@ -268,6 +268,8 @@ mod bit_vec_06;
 mod chrono_04;
 #[cfg(feature = "with-cidr-0_2")]
 mod cidr_02;
+#[cfg(feature = "with-cidr-0_3")]
+mod cidr_03;
 #[cfg(feature = "with-eui48-0_4")]
 mod eui48_04;
 #[cfg(feature = "with-eui48-1")]