From 7701ea96b5d7f9944d52f55a09ae401237fefb39 Mon Sep 17 00:00:00 2001
From: TsMask <340112800@qq.com>
Date: Fri, 12 Jan 2024 09:56:32 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20UPF=E6=8A=93=E5=8C=85=E9=94=99=E8=AF=AF?=
 =?UTF-8?q?=E6=8D=95=E8=8E=B7?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/modules/trace/controller/tcpdump.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/modules/trace/controller/tcpdump.go b/src/modules/trace/controller/tcpdump.go
index bb7f4449..62dc4f62 100644
--- a/src/modules/trace/controller/tcpdump.go
+++ b/src/modules/trace/controller/tcpdump.go
@@ -243,6 +243,14 @@ func (s *TcpdumpController) NeUPFTask(c *gin.Context) {
 			c.JSON(200, result.ErrMsg(err.Error()))
 			return
 		}
+		if strings.Contains(msg, "command not found") {
+			c.JSON(200, result.ErrMsg("Command [expect] Not Found"))
+			return
+		}
+		if strings.Contains(msg, "Unable to connect to remote host") {
+			c.JSON(200, result.ErrMsg("Connection Refused"))
+			return
+		}
 		s := strings.Index(msg, "pcap dispatch trace:")
 		if s == -1 {
 			s = strings.Index(msg, "Write ")