coydog/zmap-freebsd
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:zmap/zmap

Browse Source
This commit is contained in:
Zakir Durumeric 2013-08-18 13:34:13 -04:00
commit 8ed7a2fe36
2 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/probe_modules/module_udp.c
View File

@ -29,10 +29,10 @@
#define MAX_UDP_PAYLOAD_LEN 1472 #define MAX_UDP_PAYLOAD_LEN 1472
char *udp_send_msg = NULL; // Must be null-terminated char *udp_send_msg = NULL;
int udp_send_msg_len = 0; int udp_send_msg_len = 0;
const char *udp_send_msg_default = "GET / HTTP/1.1\r\n\r\n"; const char *udp_send_msg_default = "GET / HTTP/1.1\r\nHost: www\r\n\r\n";
static int num_ports = 1; static int num_ports = 1;
@ -66,6 +66,7 @@ int udp_global_initialize(struct state_conf * zconf) {
*c++ = 0; *c++ = 0;
if (strcmp(args, "text") == 0) { if (strcmp(args, "text") == 0) {
free(udp_send_msg);
udp_send_msg = strdup(c); udp_send_msg = strdup(c);
udp_send_msg_len = strlen(udp_send_msg); udp_send_msg_len = strlen(udp_send_msg);
@ -77,10 +78,10 @@ int udp_global_initialize(struct state_conf * zconf) {
log_fatal("udp", "could not open UDP data file '%s'\n", c); log_fatal("udp", "could not open UDP data file '%s'\n", c);
exit(1); exit(1);
} }
free(udp_send_msg);
udp_send_msg = malloc(MAX_UDP_PAYLOAD_LEN); udp_send_msg = malloc(MAX_UDP_PAYLOAD_LEN);
if (! udp_send_msg) { if (! udp_send_msg) {
free(args); free(args);
free(udp_send_msg);
log_fatal("udp", "failed to malloc payload buffer"); log_fatal("udp", "failed to malloc payload buffer");
exit(1); exit(1);
} }
@ -89,10 +90,10 @@ int udp_global_initialize(struct state_conf * zconf) {
} else if (strcmp(args, "hex") == 0) { } else if (strcmp(args, "hex") == 0) {
udp_send_msg_len = strlen(c) / 2; udp_send_msg_len = strlen(c) / 2;
free(udp_send_msg);
udp_send_msg = malloc(udp_send_msg_len); udp_send_msg = malloc(udp_send_msg_len);
if (! udp_send_msg) { if (! udp_send_msg) {
free(args); free(args);
free(udp_send_msg);
log_fatal("udp", "failed to malloc payload buffer"); log_fatal("udp", "failed to malloc payload buffer");
exit(1); exit(1);
} }
@ -108,12 +109,13 @@ int udp_global_initialize(struct state_conf * zconf) {
} }
} else { } else {
log_fatal("udp", "unknown UDP probe specification (expected file:/path, text:STRING, or hex:01020304)"); log_fatal("udp", "unknown UDP probe specification (expected file:/path, text:STRING, or hex:01020304)");
free(udp_send_msg);
free(args); free(args);
exit(1); exit(1);
} }
if (udp_send_msg_len > MAX_UDP_PAYLOAD_LEN) { if (udp_send_msg_len > MAX_UDP_PAYLOAD_LEN) {
fprintf(stderr, "warning: reducing UDP payload to %d bytes (from %d) to fit on the wire\n", log_warn("udp", "warning: reducing UDP payload to %d bytes (from %d) to fit on the wire\n",
MAX_UDP_PAYLOAD_LEN, udp_send_msg_len); MAX_UDP_PAYLOAD_LEN, udp_send_msg_len);
udp_send_msg_len = MAX_UDP_PAYLOAD_LEN; udp_send_msg_len = MAX_UDP_PAYLOAD_LEN;
} }

2
src/zmap.c
View File

@ -359,7 +359,7 @@ int main(int argc, char *argv[])
zconf.output_module = get_output_module_by_name(args.output_module_arg); zconf.output_module = get_output_module_by_name(args.output_module_arg);
if (!zconf.output_module) { if (!zconf.output_module) {
fprintf(stderr, "%s: specified output module (%s) does not exist\n", fprintf(stderr, "%s: specified output module (%s) does not exist\n",
args.output_module_arg, CMDLINE_PARSER_PACKAGE); CMDLINE_PARSER_PACKAGE, args.output_module_arg);
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
zconf.probe_module = get_probe_module_by_name(args.probe_module_arg); zconf.probe_module = get_probe_module_by_name(args.probe_module_arg);