remove nightly api usage

aschey · Feb 13, 2025 · 4782144 · 4782144
1 parent 4866a0d
commit 4782144
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/.tokeignore b/.tokeignore
@@ -1,3 +1,2 @@
 *_mock.go
 *.pb.go
-old
diff --git a/platuned/server/src/main.rs b/platuned/server/src/main.rs
@@ -1,4 +1,3 @@
-#![feature(ip)]
 mod cert_gen;
 mod ipc_stream;
 mod rpc;

diff --git a/platuned/server/src/services/management.rs b/platuned/server/src/services/management.rs
@@ -118,7 +118,7 @@ async fn get_connection_type<T>(
             })
             .collect();
 
-        if remote_addr.is_global() {
+        if !is_local(remote_addr) {
             if let Ok(mut global_addr) = env::var("PLATUNE_GLOBAL_FILE_URL") {
                 if !global_addr.ends_with('/') {
                     global_addr.push('/');
@@ -146,6 +146,15 @@ async fn get_connection_type<T>(
     }
 }
 
+fn is_local(ip_addr: IpAddr) -> bool {
+    ip_addr.is_loopback()
+        || ip_addr.is_unspecified()
+        || match ip_addr {
+            IpAddr::V4(v4) => v4.is_link_local() || v4.is_private(),
+            IpAddr::V6(v6) => v6.is_unicast_link_local() || v6.is_unique_local(),
+        }
+}
+
 #[tonic::async_trait]
 impl Management for ManagementImpl {
     async fn start_sync(&self, _: Request<()>) -> Result<Response<()>, Status> {