From f093bf96c44027507cfbc189aec5b90f97d33f4a Mon Sep 17 00:00:00 2001
From: Christian Deacon <christianmdeacon@gmail.com>
Date: Sun, 23 Feb 2025 06:42:37 -0500
Subject: [PATCH] Add more error handling when detaching XDP program.

---
 src/loader/prog.c | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/loader/prog.c b/src/loader/prog.c
index c715a5b..d41701c 100644
--- a/src/loader/prog.c
+++ b/src/loader/prog.c
@@ -1,9 +1,10 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <unistd.h>
-#include <signal.h>
 #include <time.h>
 
+#include <signal.h>
+
 #include <sys/resource.h>
 #include <sys/sysinfo.h>
 #include <sys/stat.h>
@@ -172,12 +173,16 @@ int main(int argc, char *argv[])
         usleep(sleep_time);
     }
 
-    // Detach XDP program.
-    AttachXdp(prog, ifidx, 1, &cmd);
-
-    // Add spacing.
     fprintf(stdout, "\n");
 
+    // Detach XDP program.
+    if (AttachXdp(prog, ifidx, 1, &cmd))
+    {
+        fprintf(stderr, "Failed to detach XDP program from interface '%s'.\n", cfg.interface);
+    }
+
+    fprintf(stdout, "Cleaned up and exiting...\n");
+
     // Exit program successfully.
     return EXIT_SUCCESS;
 }
\ No newline at end of file