From 23793299ff49bf5b43ca7d5ff8aac50f66b5c920 Mon Sep 17 00:00:00 2001
From: Jiang Liu <gerry@linux.alibaba.com>
Date: Tue, 28 Mar 2023 11:19:13 +0800
Subject: [PATCH] nydus: fix a possible panic caused by
 SubCmdArgs::is_present()

Fix a possible panic caused by SubCmdArgs::is_present().

Signed-off-by: Jiang Liu <gerry@linux.alibaba.com>
---
 src/lib.rs | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/lib.rs b/src/lib.rs
index 8b49a92156b..830b529cfab 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -40,7 +40,14 @@ impl<'a> ServiceArgs for SubCmdArgs<'a> {
     }
 
     fn is_present(&self, key: &str) -> bool {
-        self.subargs.get_flag(key) || self.args.get_flag(key)
+        match self.subargs.try_get_one::<bool>(key) {
+            Ok(Some(true)) => true,
+            _ => match self.args.try_get_one::<bool>(key) {
+                Ok(Some(true)) => true,
+                _ => false,
+            },
+        }
+        //self.subargs.try_get_one::<bool>(key).is_ok() || self.args.try_get_one::<bool>(key).is_ok()
     }
 }