Feat support addr label

examples/add_address.rs

@@ -42,7 +42,12 @@ async fn add_address(
     if let Some(link) = links.try_next().await? {
         handle
             .address()
-            .add(link.header.index, ip.ip(), ip.prefix())
+            .add(
+                link.header.index,
+                ip.ip(),
+                ip.prefix(),
+                Some(format!("{}:vip", link_name)),
+            )
             .execute()
             .await?
     }

examples/get_address.rs

@@ -2,16 +2,30 @@
 
 use futures::stream::TryStreamExt;
 use rtnetlink::{new_connection, Error, Handle};
+use std::env;
 
 #[tokio::main]
 async fn main() -> Result<(), ()> {
+    let args: Vec<String> = env::args().collect();
+    if args.len() != 2 {
+        return Ok(());
+    }
+
+    let link_name = &args[1];
+
     let (connection, handle, _) = new_connection().unwrap();
     tokio::spawn(connection);
 
     let link = "lo".to_string();
     println!("dumping address for link \"{link}\"");
 
-    if let Err(e) = dump_addresses(handle, link).await {
+    if let Err(e) = dump_addresses(handle.clone(), link).await {
+        eprintln!("{e}");
+    }
+
+    let label = format!("{link_name}:vip");
+    println!("\ndumping address with label \"{label}\"");
+    if let Err(e) = dump_addresses_with_label(handle, &label).await {
         eprintln!("{e}");
     }
 
@@ -35,3 +49,16 @@ async fn dump_addresses(handle: Handle, link: String) -> Result<(), Error> {
         Ok(())
     }
 }
+
+async fn dump_addresses_with_label(
+    handle: Handle,
+    label: &str,
+) -> Result<(), Error> {
+    let mut addrs = handle.address().get().set_label_filter(label).execute();
+
+    while let Some(msg) = addrs.try_next().await? {
+        println!("{msg:?}");
+    }
+
+    Ok(())
+}

src/addr/add.rs

@@ -28,6 +28,7 @@ impl AddressAddRequest {
         index: u32,
         address: IpAddr,
         prefix_len: u8,
+        label: Option<String>,
     ) -> Self {
         let mut message = AddressMessage::default();
 
@@ -54,6 +55,10 @@ impl AddressAddRequest {
         } else {
             message.nlas.push(Nla::Address(address_vec.clone()));
 
+            if let Some(l) = label {
+                message.nlas.push(Nla::Label(l));
+            }
+
             // for IPv4 the IFA_LOCAL address can be set to the same value as
             // IFA_ADDRESS
             message.nlas.push(Nla::Local(address_vec.clone()));

src/addr/get.rs

@@ -71,6 +71,11 @@ impl AddressGetRequest {
         self.filter_builder.address = Some(address);
         self
     }
+
+    pub fn set_label_filter(mut self, label: &str) -> Self {
+        self.filter_builder.label = Some(label.to_string());
+        self
+    }
 }
 
 // The reason for having filters, is that we cannot retrieve addresses
@@ -84,6 +89,7 @@ struct AddressFilterBuilder {
     index: Option<u32>,
     address: Option<IpAddr>,
     prefix_len: Option<u8>,
+    label: Option<String>,
 }
 
 impl AddressFilterBuilder {
@@ -108,11 +114,13 @@ impl AddressFilterBuilder {
             }
 
             if let Some(address) = self.address {
+                let mut is_match: bool = false;
+
                 for nla in msg.nlas.iter() {
                     if let Unspec(x) | Address(x) | Local(x) | Multicast(x)
                     | Anycast(x) = nla
                     {
-                        let is_match = match address {
+                        is_match = match address {
                             IpAddr::V4(address) => {
                                 x[..] == address.octets()[..]
                             }
@@ -121,12 +129,33 @@ impl AddressFilterBuilder {
                             }
                         };
                         if is_match {
-                            return true;
+                            break;
+                        }
+                    }
+                }
+
+                if !is_match {
+                    return false;
+                }
+            }
+
+            if let Some(ref label) = self.label {
+                let mut is_match: bool = false;
+
+                for nla in msg.nlas.iter() {
+                    if let Label(l) = nla {
+                        if label == l {
+                            is_match = true;
+                            break;
                         }
                     }
                 }
-                return false;
+
+                if !is_match {
+                    return false;
+                }
             }
+
             true
         }
     }

src/addr/handle.rs

@@ -25,8 +25,15 @@ impl AddressHandle {
         index: u32,
         address: IpAddr,
         prefix_len: u8,
+        label: Option<String>,
     ) -> AddressAddRequest {
-        AddressAddRequest::new(self.0.clone(), index, address, prefix_len)
+        AddressAddRequest::new(
+            self.0.clone(),
+            index,
+            address,
+            prefix_len,
+            label,
+        )
     }
 
     /// Delete the given address