proto(utils): add TransientDigest trait (#2071)

easytier/src/proto/mod.rs

@@ -12,6 +12,7 @@ pub mod web;
 
 #[cfg(test)]
 pub mod tests;
+pub mod utils;
 
 const DESCRIPTOR_POOL_BYTES: &[u8] =
     include_bytes!(concat!(env!("OUT_DIR"), "/file_descriptor_set.bin"));

easytier/src/proto/utils.rs

@@ -0,0 +1,20 @@
+use prost::Message;
+use sha2::{Digest, Sha256};
+
+/// Generates a stable digest strictly within the lifecycle of the current process.
+///
+/// ⚠️ WARNING:
+/// - This digest is ONLY guaranteed to be deterministic within a **single process and the exact same binary build**.
+pub trait TransientDigest: Message {
+    fn digest(&self) -> [u8; 32]
+    where
+        Self: Sized,
+    {
+        let buf = self.encode_to_vec();
+        let mut hasher = Sha256::new();
+        hasher.update(buf);
+        hasher.finalize().into()
+    }
+}
+
+impl<S: Message> TransientDigest for S {}

easytier/src/utils.rs

@@ -1,5 +1,6 @@
 use crate::common::log;
 use indoc::formatdoc;
+use std::sync::Arc;
 use std::{fs::OpenOptions, str::FromStr};
 
 pub type PeerRoutePair = crate::proto::api::instance::PeerRoutePair;
@@ -136,7 +137,7 @@ pub fn find_free_tcp_port(mut range: std::ops::Range<u16>) -> Option<u16> {
     range.find(|&port| check_tcp_available(port))
 }
 
-pub fn weak_upgrade<T>(weak: &std::sync::Weak<T>) -> anyhow::Result<std::sync::Arc<T>> {
+pub fn weak_upgrade<T>(weak: &std::sync::Weak<T>) -> anyhow::Result<Arc<T>> {
     weak.upgrade()
         .ok_or_else(|| anyhow::anyhow!("{} not available", std::any::type_name::<T>()))
 }