From 41cd8500a62ef3a486107645299671322b649a80 Mon Sep 17 00:00:00 2001
From: Shuduo Sang <sangshuduo@gmail.com>
Date: Fri, 27 Aug 2021 16:52:33 +0800
Subject: [PATCH] [TD-6280]<fix>: taosdump -P determine input. (#7607)

---
 src/kit/taosdump/taosdump.c | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/src/kit/taosdump/taosdump.c b/src/kit/taosdump/taosdump.c
index cca04f926b..ae2193a82e 100644
--- a/src/kit/taosdump/taosdump.c
+++ b/src/kit/taosdump/taosdump.c
@@ -367,6 +367,15 @@ struct arguments g_args = {
     false       // performance_print
 };
 
+static void errorPrintReqArg2(char *program, char *wrong_arg)
+{
+    fprintf(stderr,
+            "%s: option requires a number argument '-%s'\n",
+            program, wrong_arg);
+    fprintf(stderr,
+            "Try `taosdump --help' or `taosdump --usage' for more information.\n");
+}
+
 static void errorPrintReqArg3(char *program, char *wrong_arg)
 {
     fprintf(stderr,
@@ -396,6 +405,10 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state) {
         case 'p':
             break;
         case 'P':
+            if (!isStringNumber(arg)) {
+                errorPrintReqArg2("taosdump", "P");
+                exit(EXIT_FAILURE);
+            }
             g_args.port = atoi(arg);
             break;
         case 'q':
-- 
GitLab