From 50a630c4b686642d650360fbaf9a382169668618 Mon Sep 17 00:00:00 2001
From: Charles Han <hanchunchao@inspur.com>
Date: Wed, 18 Oct 2023 20:32:06 +0800
Subject: [PATCH] Fix: Pingtrace:  Limit input parameters

---
 .../net/PingTrace/src/common/options.hpp      | 28 ++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/source/tools/detect/net/PingTrace/src/common/options.hpp b/source/tools/detect/net/PingTrace/src/common/options.hpp
index d59ac148..ac926eb1 100644
--- a/source/tools/detect/net/PingTrace/src/common/options.hpp
+++ b/source/tools/detect/net/PingTrace/src/common/options.hpp
@@ -118,6 +118,32 @@ struct options {
 
 		try {
 			app.parse(argc, argv);
+
+			if (this->max_count == 0) {
+                            std::cout<< "-C, --count Only the integer bigger than 0 is valid." <<std::endl;
+                            throw CLI::CallForHelp();
+                        }
+
+                        if (this->interval_us == 0) {
+                            std::cout<< "-i Only the integer bigger than 0 is valid." <<std::endl;
+                            throw CLI::CallForHelp();
+                        }
+
+                        if (this->runtime == 0) {
+                            std::cout<< "-t Only the integer bigger than 0 is valid." <<std::endl;
+                            throw CLI::CallForHelp();
+                        }
+
+                        if (this->log_max_size == 0) {
+                            std::cout<< "--logsize Only the integer bigger than 0 is valid." <<std::endl;
+                            throw CLI::CallForHelp();
+                        }
+
+                        if (this->log_max_backup == 0) {
+                            std::cout<< "--logbackup Only the integer bigger than 0 is valid." <<std::endl;
+                            throw CLI::CallForHelp();
+                        }
+
 			if (!this->ip.empty())
 			this->run_client = true;
 			if (this->help || (!this->run_client && !this->run_server))
@@ -139,4 +165,4 @@ struct options {
 
 }; // namespace pingtrace
 
-#endif
\ No newline at end of file
+#endif
-- 
Gitee