PenTest/nmap
1
0
Fork 0
mirror of https://github.com/nmap/nmap.git synced 2026-01-10 08:29:02 +00:00
Code Issues Projects Releases Wiki Activity

recv and send error checking in ncat_listen_dgram.

Browse Source
This commit is contained in:
david
2012-06-19 02:52:58 +00:00
parent a9a621f5bd
commit 04e0503135
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
ncat/ncat_listen.c
View File

@@ -549,7 +549,7 @@ static int ncat_listen_dgram(int proto)
{
int sockfd[NUM_LISTEN_ADDRS];
int i, fdn = -1;
int fdmax, nbytes, fds_ready;
int fdmax, nbytes, n, fds_ready;
char buf[DEFAULT_UDP_BUF_LEN] = { 0 };
char *tempbuf = NULL;
fd_set read_fds;
@@ -674,6 +674,11 @@ static int ncat_listen_dgram(int proto)
/* Dump the current datagram */
nbytes = Recv(socket_n, buf, sizeof(buf), 0);
if (nbytes < 0) {
loguser("%s.\n", socket_strerror(socket_errno()));
close(socket_n);
return 1;
}
ncat_log_recv(buf, nbytes);
}
@@ -730,9 +735,14 @@ static int ncat_listen_dgram(int proto)
fix_line_endings((char *) buf, &nbytes, &tempbuf, &crlf_state);
if (!o.recvonly) {
if (tempbuf != NULL)
send(socket_n, tempbuf, nbytes, 0);
n = send(socket_n, tempbuf, nbytes, 0);
else
send(socket_n, buf, nbytes, 0);
n = send(socket_n, buf, nbytes, 0);
if (n < nbytes) {
loguser("%s.\n", socket_strerror(socket_errno()));
close(socket_n);
return 1;
}
ncat_log_send(buf, nbytes);
}
if (tempbuf != NULL) {