Page MenuHomeFreeBSD
Files F103465437

D37423.diff
No OneTemporary
Actions

D37423.diff
View Options

diff --git a/libexec/tftpd/tftp-io.c b/libexec/tftpd/tftp-io.c
--- a/libexec/tftpd/tftp-io.c
+++ b/libexec/tftpd/tftp-io.c
@@ -258,7 +258,7 @@
size += strlen(mode) + 1;
if (options_rfc_enabled) {
- options[OPT_TSIZE].o_request = strdup("0");
+ options_set_request(OPT_TSIZE, "0");
size += make_options(peer, bp, sizeof(buf) - size);
}
diff --git a/libexec/tftpd/tftp-options.h b/libexec/tftpd/tftp-options.h
--- a/libexec/tftpd/tftp-options.h
+++ b/libexec/tftpd/tftp-options.h
@@ -64,3 +64,8 @@
OPT_ROLLOVER,
OPT_WINDOWSIZE,
};
+
+int options_set_request(enum opt_enum, const char *, ...)
+ __printflike(2, 3);
+int options_set_reply(enum opt_enum, const char *, ...)
+ __printflike(2, 3);
diff --git a/libexec/tftpd/tftp-options.c b/libexec/tftpd/tftp-options.c
--- a/libexec/tftpd/tftp-options.c
+++ b/libexec/tftpd/tftp-options.c
@@ -37,6 +37,7 @@
#include <arpa/tftp.h>
#include <ctype.h>
+#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
@@ -64,6 +65,62 @@
int options_rfc_enabled = 1;
int options_extra_enabled = 1;
+int
+options_set_request(enum opt_enum opt, const char *fmt, ...)
+{
+ va_list ap;
+ char *str;
+ int ret;
+
+ if (fmt == NULL) {
+ str = NULL;
+ } else {
+ va_start(ap, fmt);
+ ret = vasprintf(&str, fmt, ap);
+ va_end(ap);
+ if (ret < 0)
+ return (ret);
+ }
+ if (options[opt].o_request != NULL &&
+ options[opt].o_request != options[opt].o_reply)
+ free(options[opt].o_request);
+ options[opt].o_request = str;
+ return (0);
+}
+
+int
+options_set_reply(enum opt_enum opt, const char *fmt, ...)
+{
+ va_list ap;
+ char *str;
+ int ret;
+
+ if (fmt == NULL) {
+ str = NULL;
+ } else {
+ va_start(ap, fmt);
+ ret = vasprintf(&str, fmt, ap);
+ va_end(ap);
+ if (ret < 0)
+ return (ret);
+ }
+ if (options[opt].o_reply != NULL &&
+ options[opt].o_reply != options[opt].o_request)
+ free(options[opt].o_reply);
+ options[opt].o_reply = str;
+ return (0);
+}
+
+static void
+options_set_reply_equal_request(enum opt_enum opt)
+{
+
+ if (options[opt].o_reply != NULL &&
+ options[opt].o_reply != options[opt].o_request)
+ free(options[opt].o_reply);
+ options[opt].o_reply = options[opt].o_request;
+}
+
/*
* Rules for the option handlers:
* - If there is no o_request, there will be no processing.
@@ -90,12 +147,10 @@
return (0);
if (mode == RRQ)
- asprintf(&options[OPT_TSIZE].o_reply,
- "%ju", (uintmax_t)stbuf->st_size);
+ options_set_reply(OPT_TSIZE, "%ju", stbuf->st_size);
else
/* XXX Allows writes of all sizes. */
- options[OPT_TSIZE].o_reply =
- strdup(options[OPT_TSIZE].o_request);
+ options_set_reply_equal_request(OPT_TSIZE);
return (0);
}
@@ -119,8 +174,7 @@
exit(1);
} else {
timeoutpacket = to;
- options[OPT_TIMEOUT].o_reply =
- strdup(options[OPT_TIMEOUT].o_request);
+ options_set_reply_equal_request(OPT_TIMEOUT);
}
settimeouts(timeoutpacket, timeoutnetwork, maxtimeouts);
@@ -151,8 +205,7 @@
}
return (0);
}
- options[OPT_ROLLOVER].o_reply =
- strdup(options[OPT_ROLLOVER].o_request);
+ options_set_reply_equal_request(OPT_ROLLOVER);
if (debug & DEBUG_OPTIONS)
tftp_log(LOG_DEBUG, "Setting rollover to '%s'",
@@ -212,7 +265,7 @@
}
}
- asprintf(&options[OPT_BLKSIZE].o_reply, "%d", size);
+ options_set_reply(OPT_BLKSIZE, "%d", size);
segsize = size;
pktsize = size + 4;
if (debug & DEBUG_OPTIONS)
@@ -266,7 +319,7 @@
/* No need to return */
}
- asprintf(&options[OPT_BLKSIZE2].o_reply, "%d", size);
+ options_set_reply(OPT_BLKSIZE2, "%d", size);
segsize = size;
pktsize = size + 4;
if (debug & DEBUG_OPTIONS)
@@ -301,7 +354,7 @@
}
/* XXX: Should force a windowsize of 1 for non-seekable files. */
- asprintf(&options[OPT_WINDOWSIZE].o_reply, "%d", size);
+ options_set_reply(OPT_WINDOWSIZE, "%d", size);
windowsize = size;
if (debug & DEBUG_OPTIONS)
@@ -391,7 +444,7 @@
for (i = 0; options[i].o_type != NULL; i++) {
if (strcmp(option, options[i].o_type) == 0) {
if (!acting_as_client)
- options[i].o_request = value;
+ options_set_request(i, "%s", value);
if (!options_extra_enabled && !options[i].rfc) {
tftp_log(LOG_INFO,
"Option '%s' with value '%s' found "
@@ -422,5 +475,5 @@
init_options(void)
{
- options[OPT_ROLLOVER].o_request = strdup("0");
+ options_set_request(OPT_ROLLOVER, "0");
}
diff --git a/usr.bin/tftp/main.c b/usr.bin/tftp/main.c
--- a/usr.bin/tftp/main.c
+++ b/usr.bin/tftp/main.c
@@ -496,7 +496,7 @@
close(fd);
return;
}
- asprintf(&options[OPT_TSIZE].o_request, "%ju", sb.st_size);
+ options_set_request(OPT_TSIZE, "%ju", (uintmax_t)sb.st_size);
if (verbose)
printf("putting %s to %s:%s [%s]\n",
@@ -524,7 +524,7 @@
free(path);
continue;
}
- asprintf(&options[OPT_TSIZE].o_request, "%ju", sb.st_size);
+ options_set_request(OPT_TSIZE, "%ju", (uintmax_t)sb.st_size);
if (verbose)
printf("putting %s to %s:%s [%s]\n",
@@ -926,16 +926,13 @@
if (argc == 2) {
if (strcasecmp(argv[1], "never") == 0 ||
strcasecmp(argv[1], "none") == 0) {
- free(options[OPT_ROLLOVER].o_request);
- options[OPT_ROLLOVER].o_request = NULL;
+ options_set_request(OPT_ROLLOVER, NULL);
}
if (strcasecmp(argv[1], "1") == 0) {
- free(options[OPT_ROLLOVER].o_request);
- options[OPT_ROLLOVER].o_request = strdup("1");
+ options_set_request(OPT_ROLLOVER, "1");
}
if (strcasecmp(argv[1], "0") == 0) {
- free(options[OPT_ROLLOVER].o_request);
- options[OPT_ROLLOVER].o_request = strdup("0");
+ options_set_request(OPT_ROLLOVER, "0");
}
}
printf("Support for the rollover options is %s.\n",
@@ -1001,10 +998,9 @@
printf("Blocksize can't be bigger than %ld bytes due "
"to the net.inet.udp.maxdgram sysctl limitation.\n",
maxdgram - 4);
- asprintf(&options[OPT_BLKSIZE].o_request,
- "%ld", maxdgram - 4);
+ options_set_request(OPT_BLKSIZE, "%ld", maxdgram - 4);
} else {
- asprintf(&options[OPT_BLKSIZE].o_request, "%d", size);
+ options_set_request(OPT_BLKSIZE, "%d", size);
}
}
printf("Blocksize is now %s bytes.\n", options[OPT_BLKSIZE].o_request);
@@ -1057,10 +1053,9 @@
for (i = 0; sizes[i+1] != 0; i++) {
if ((int)maxdgram < sizes[i+1]) break;
}
- asprintf(&options[OPT_BLKSIZE2].o_request,
- "%d", sizes[i]);
+ options_set_request(OPT_BLKSIZE2, "%d", sizes[i]);
} else {
- asprintf(&options[OPT_BLKSIZE2].o_request, "%d", size);
+ options_set_request(OPT_BLKSIZE2, "%d", size);
}
}
printf("Blocksize2 is now %s bytes.\n",
@@ -1094,8 +1089,7 @@
"blocks.\n", WINDOWSIZE_MIN, WINDOWSIZE_MAX);
return;
} else {
- asprintf(&options[OPT_WINDOWSIZE].o_request, "%d",
- size);
+ options_set_request(OPT_WINDOWSIZE, "%d", size);
}
}
printf("Windowsize is now %s blocks.\n",

File Metadata

Mime Type
text/plain
Expires
Tue, Nov 26, 9:34 AM (21 h, 57 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
14843350
Default Alt Text
D37423.diff (6 KB)

Event Timeline