Get Agent IP every callback (#711)

* added ip to each callback

* move ip refresh to own function

* cleaning

* comment

* oops

.vscode/settings.json

@@ -4,4 +4,5 @@
         "rust-analyzer"
     ],
     "rust-analyzer.check.command": "clippy",
+    "rust-analyzer.showUnlinkedFileNotification": false,
 }

implants/imix/src/agent.rs

@@ -91,6 +91,11 @@ impl Agent {
         loop {
             let start = Instant::now();
 
+            // Sometimes Imix starts too quickly in a boot sequence, a NIC is down during the initial callback,
+            // or the box Imix is on changes its IP. In any case, for each callback we should refresh our claimed
+            // IP.
+            self.cfg.refresh_primary_ip();
+
             match self.callback().await {
                 Ok(_) => {}
                 Err(_err) => {

implants/imix/src/config.rs

@@ -107,6 +107,28 @@ impl Default for Config {
     }
 }
 
+impl Config {
+    pub fn refresh_primary_ip(&mut self) {
+        let fresh_ip = get_primary_ip();
+        if self
+            .info
+            .host
+            .as_ref()
+            .is_some_and(|h| h.primary_ip != fresh_ip)
+        {
+            match self.info.host.as_mut() {
+                Some(h) => {
+                    h.primary_ip = fresh_ip;
+                }
+                None => {
+                    #[cfg(debug_assertions)]
+                    log::error!("host struct was never initialized, failed to set primary ip");
+                }
+            }
+        }
+    }
+}
+
 /*
  * Returns which Platform imix has been compiled for.
  */