forked from pool/netcat-openbsd
Peter Simons
221ef298d8
OBS-URL: https://build.opensuse.org/package/show/network:utilities/netcat-openbsd?expand=0&rev=29
54 lines
1.2 KiB
Diff
54 lines
1.2 KiB
Diff
From: Aron Xu <aron@debian.org>
|
|
Date: Mon, 13 Feb 2012 15:29:37 +0800
|
|
Subject: Fix UDP scan timeout
|
|
|
|
---
|
|
netcat.c | 21 ++++++++++++++-------
|
|
1 file changed, 14 insertions(+), 7 deletions(-)
|
|
|
|
diff --git a/netcat.c b/netcat.c
|
|
index 03d339c..dd893ac 100644
|
|
--- a/netcat.c
|
|
+++ b/netcat.c
|
|
@@ -129,6 +129,8 @@
|
|
#define CONNECTION_FAILED 1
|
|
#define CONNECTION_TIMEOUT 2
|
|
|
|
+#define UDP_SCAN_TIMEOUT 3 /* Seconds */
|
|
+
|
|
/* Command Line Options */
|
|
int dflag; /* detached, no stdin */
|
|
int Fflag; /* fdpass sock to stdout */
|
|
@@ -1794,19 +1796,24 @@ build_ports(char *p)
|
|
int
|
|
udptest(int s)
|
|
{
|
|
- int i, ret;
|
|
+ int i, t;
|
|
|
|
/* Only write to the socket in scan mode or interactive mode. */
|
|
if (!zflag && !isatty(STDIN_FILENO))
|
|
return 0;
|
|
|
|
- for (i = 0; i <= 3; i++) {
|
|
- if (write(s, "X", 1) == 1)
|
|
- ret = 1;
|
|
- else
|
|
- ret = -1;
|
|
+ if ((write(s, "X", 1) != 1) ||
|
|
+ ((write(s, "X", 1) != 1) && (errno == ECONNREFUSED)))
|
|
+ return -1;
|
|
+
|
|
+ /* Give the remote host some time to reply. */
|
|
+ for (i = 0, t = (timeout == -1) ? UDP_SCAN_TIMEOUT : (timeout / 1000);
|
|
+ i < t; i++) {
|
|
+ sleep(1);
|
|
+ if ((write(s, "X", 1) != 1) && (errno == ECONNREFUSED))
|
|
+ return -1;
|
|
}
|
|
- return ret;
|
|
+ return 1;
|
|
}
|
|
|
|
void
|