From b091fe085083a10c8897a10b58f5b4178e6cae72 Mon Sep 17 00:00:00 2001
From: Wang Yu <wangyu6@uniontech.com>
Date: Thu, 17 Sep 2026 19:33:50 +0800
Subject: [PATCH] fix/net-traffic-validate-remote-ip

Signed-off-by: Wang Yu <wangyu6@uniontech.com>
---
 observe/net-traffic.cpp | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/observe/net-traffic.cpp b/observe/net-traffic.cpp
index af25376..1dc3aef 100644
--- a/observe/net-traffic.cpp
+++ b/observe/net-traffic.cpp
@@ -164,7 +164,16 @@ void parse_args(int argc, char **argv)
 			rule.pid = atoi(optarg);
 			break;
 		case 'r': // Remote IP
-			rule.remote_ip = ntohl(inet_addr(optarg));
+			{
+				struct in_addr in;
+				if (inet_pton(AF_INET, optarg, &in) != 1)
+				{
+					fprintf(stderr, "Invalid remote ip: %s\n", optarg);
+					Usage(argv[0]);
+					exit(-1);
+				}
+				rule.remote_ip = ntohl(in.s_addr);
+			}
 			break;
 		case 'P': // Remote port
 			rule.remote_port = atoi(optarg);