feat(mgmt): add grpc server addr option

Signed-off-by: Sergey Matov <[email protected]>

Author: sergeymatov

dataplane/src/args.rs

@@ -4,6 +4,7 @@
 #![allow(unused)]
 
 pub(crate) use clap::Parser;
+use std::net::SocketAddr;
 use tracing::debug;
 
 #[derive(Parser)]
@@ -30,7 +31,16 @@ pub(crate) struct CmdArgs {
     driver: Option<String>,
     #[arg(long, value_name = "name of kernel interface")]
     interface: Vec<String>,
+
+    // gRPC server address
+    #[arg(
+        long,
+        value_name = "gRPC server address",
+        default_value = "[::1]:50051"
+    )]
+    grpc_address: String,
 }
+
 impl CmdArgs {
     pub fn get_driver_name(&self) -> &str {
         match &self.driver {
@@ -92,4 +102,15 @@ impl CmdArgs {
 
         out
     }
+
+    /// Get the gRPC server address
+    pub fn get_grpc_address(&self) -> SocketAddr {
+        match self.grpc_address.parse() {
+            Ok(addr) => addr,
+            Err(e) => {
+                eprintln!("Error: Invalid gRPC address '{}': {}", self.grpc_address, e);
+                panic!("Process receives unexpected gRPC address. Aborting...");
+            }
+        }
+    }
 }

dataplane/src/main.rs

@@ -61,11 +61,11 @@ fn main() {
     /* parse cmd line args */
     let args = CmdArgs::parse();
 
-    let grpc_address = "[::1]:50051".parse().expect("Bad grpc address");
-
-    if let Err(e) = start_mgmt(grpc_address) {
+    // Get the gRPC address from command line args
+    let grpc_addr = args.get_grpc_address();
+    if let Err(e) = start_mgmt(grpc_addr) {
         error!("Failed to start management service: {e}");
-        std::process::exit(0);
+        panic!("Management service failed to start. Aborting...");
     }
 
     debug!("Starting pipeline....");
@@ -82,7 +82,7 @@ fn main() {
         }
         other => {
             error!("Unknown driver '{other}'. Aborting...");
-            std::process::exit(0);
+            panic!("Packet processing pipeline failed to start. Aborting...");
         }
     }