tftp: Style cleanup
MFC after: 1 week Reviewed by: markj Differential Revision: https://reviews.freebsd.org/D57073
This commit is contained in:
+8
-24
@@ -76,7 +76,7 @@ static int margc;
|
|||||||
static char *margv[MAX_MARGV];
|
static char *margv[MAX_MARGV];
|
||||||
|
|
||||||
int verbose;
|
int verbose;
|
||||||
static char *port = NULL;
|
static char *port;
|
||||||
|
|
||||||
static void get(int, char **);
|
static void get(int, char **);
|
||||||
static void help(int, char **);
|
static void help(int, char **);
|
||||||
@@ -224,11 +224,11 @@ main(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
urihandling(char *URI)
|
urihandling(char *URI)
|
||||||
{
|
{
|
||||||
|
const char *tmode = "octet";
|
||||||
char meth[] = "get";
|
char meth[] = "get";
|
||||||
char *host = NULL;
|
char *host = NULL;
|
||||||
char *path = NULL;
|
char *path = NULL;
|
||||||
char *opts = NULL;
|
char *opts = NULL;
|
||||||
const char *tmode = "octet";
|
|
||||||
char *s;
|
char *s;
|
||||||
int i;
|
int i;
|
||||||
|
|
||||||
@@ -271,8 +271,8 @@ static void
|
|||||||
setpeer0(char *host, const char *lport)
|
setpeer0(char *host, const char *lport)
|
||||||
{
|
{
|
||||||
struct addrinfo hints, *res0, *res;
|
struct addrinfo hints, *res0, *res;
|
||||||
int error;
|
|
||||||
const char *cause = "unknown";
|
const char *cause = "unknown";
|
||||||
|
int error;
|
||||||
|
|
||||||
if (connected) {
|
if (connected) {
|
||||||
close(peer);
|
close(peer);
|
||||||
@@ -285,7 +285,7 @@ setpeer0(char *host, const char *lport)
|
|||||||
hints.ai_socktype = SOCK_DGRAM;
|
hints.ai_socktype = SOCK_DGRAM;
|
||||||
hints.ai_protocol = IPPROTO_UDP;
|
hints.ai_protocol = IPPROTO_UDP;
|
||||||
hints.ai_flags = AI_CANONNAME;
|
hints.ai_flags = AI_CANONNAME;
|
||||||
if (!lport)
|
if (lport == NULL)
|
||||||
lport = "tftp";
|
lport = "tftp";
|
||||||
error = getaddrinfo(host, lport, &hints, &res0);
|
error = getaddrinfo(host, lport, &hints, &res0);
|
||||||
if (error) {
|
if (error) {
|
||||||
@@ -347,7 +347,7 @@ setpeer(int argc, char *argv[])
|
|||||||
argc = margc;
|
argc = margc;
|
||||||
argv = margv;
|
argv = margv;
|
||||||
}
|
}
|
||||||
if ((argc < 2) || (argc > 3)) {
|
if (argc < 2 || argc > 3) {
|
||||||
printf("usage: %s [host [port]]\n", argv[0]);
|
printf("usage: %s [host [port]]\n", argv[0]);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@@ -393,21 +393,18 @@ modecmd(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setbinary(int argc __unused, char *argv[] __unused)
|
setbinary(int argc __unused, char *argv[] __unused)
|
||||||
{
|
{
|
||||||
|
|
||||||
settftpmode("octet");
|
settftpmode("octet");
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
setascii(int argc __unused, char *argv[] __unused)
|
setascii(int argc __unused, char *argv[] __unused)
|
||||||
{
|
{
|
||||||
|
|
||||||
settftpmode("netascii");
|
settftpmode("netascii");
|
||||||
}
|
}
|
||||||
|
|
||||||
static void
|
static void
|
||||||
settftpmode(const char *newmode)
|
settftpmode(const char *newmode)
|
||||||
{
|
{
|
||||||
|
|
||||||
strlcpy(mode, newmode, sizeof(mode));
|
strlcpy(mode, newmode, sizeof(mode));
|
||||||
if (verbose)
|
if (verbose)
|
||||||
printf("mode set to %s\n", mode);
|
printf("mode set to %s\n", mode);
|
||||||
@@ -422,9 +419,9 @@ put(int argc, char *argv[])
|
|||||||
{
|
{
|
||||||
static char *line;
|
static char *line;
|
||||||
static size_t sz;
|
static size_t sz;
|
||||||
int fd, n;
|
|
||||||
char *cp, *targ, *path;
|
|
||||||
struct stat sb;
|
struct stat sb;
|
||||||
|
char *cp, *targ, *path;
|
||||||
|
int fd, n;
|
||||||
|
|
||||||
if (argc < 2) {
|
if (argc < 2) {
|
||||||
printf("(file) ");
|
printf("(file) ");
|
||||||
@@ -517,7 +514,6 @@ put(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
putusage(char *s)
|
putusage(char *s)
|
||||||
{
|
{
|
||||||
|
|
||||||
printf("usage: %s file [remotename]\n", s);
|
printf("usage: %s file [remotename]\n", s);
|
||||||
printf(" %s file host:remotename\n", s);
|
printf(" %s file host:remotename\n", s);
|
||||||
printf(" %s file1 file2 ... fileN [[host:]remote-directory]\n", s);
|
printf(" %s file1 file2 ... fileN [[host:]remote-directory]\n", s);
|
||||||
@@ -531,8 +527,8 @@ get(int argc, char *argv[])
|
|||||||
{
|
{
|
||||||
static char *line;
|
static char *line;
|
||||||
static size_t sz;
|
static size_t sz;
|
||||||
int fd, n;
|
|
||||||
char *cp, *src;
|
char *cp, *src;
|
||||||
|
int fd, n;
|
||||||
|
|
||||||
if (argc < 2) {
|
if (argc < 2) {
|
||||||
printf("(files) ");
|
printf("(files) ");
|
||||||
@@ -608,7 +604,6 @@ get(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
getusage(char *s)
|
getusage(char *s)
|
||||||
{
|
{
|
||||||
|
|
||||||
printf("usage: %s file [localname]\n", s);
|
printf("usage: %s file [localname]\n", s);
|
||||||
printf(" %s [host:]file [localname]\n", s);
|
printf(" %s [host:]file [localname]\n", s);
|
||||||
printf(" %s [host1:]file1 [host2:]file2 ... [hostN:]fileN\n", s);
|
printf(" %s [host1:]file1 [host2:]file2 ... [hostN:]fileN\n", s);
|
||||||
@@ -671,7 +666,6 @@ settimeoutnetwork(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
showstatus(int argc __unused, char *argv[] __unused)
|
showstatus(int argc __unused, char *argv[] __unused)
|
||||||
{
|
{
|
||||||
|
|
||||||
printf("Remote host: %s\n",
|
printf("Remote host: %s\n",
|
||||||
connected ? hostname : "none specified yet");
|
connected ? hostname : "none specified yet");
|
||||||
printf("RFC2347 Options support: %s\n",
|
printf("RFC2347 Options support: %s\n",
|
||||||
@@ -692,7 +686,6 @@ showstatus(int argc __unused, char *argv[] __unused)
|
|||||||
static void
|
static void
|
||||||
intr(int dummy __unused)
|
intr(int dummy __unused)
|
||||||
{
|
{
|
||||||
|
|
||||||
signal(SIGALRM, SIG_IGN);
|
signal(SIGALRM, SIG_IGN);
|
||||||
alarm(0);
|
alarm(0);
|
||||||
longjmp(toplevel, -1);
|
longjmp(toplevel, -1);
|
||||||
@@ -717,7 +710,6 @@ tail(char *filename)
|
|||||||
static const char *
|
static const char *
|
||||||
command_prompt(void)
|
command_prompt(void)
|
||||||
{
|
{
|
||||||
|
|
||||||
return ("tftp> ");
|
return ("tftp> ");
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -835,7 +827,6 @@ makeargv(char *argv0, char *line)
|
|||||||
static void
|
static void
|
||||||
quit(int argc __unused, char *argv[] __unused)
|
quit(int argc __unused, char *argv[] __unused)
|
||||||
{
|
{
|
||||||
|
|
||||||
exit(txrx_error);
|
exit(txrx_error);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -873,7 +864,6 @@ help(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setverbose(int argc __unused, char *argv[] __unused)
|
setverbose(int argc __unused, char *argv[] __unused)
|
||||||
{
|
{
|
||||||
|
|
||||||
verbose = !verbose;
|
verbose = !verbose;
|
||||||
printf("Verbose mode %s.\n", verbose ? "on" : "off");
|
printf("Verbose mode %s.\n", verbose ? "on" : "off");
|
||||||
}
|
}
|
||||||
@@ -881,7 +871,6 @@ setverbose(int argc __unused, char *argv[] __unused)
|
|||||||
static void
|
static void
|
||||||
setoptions(int argc, char *argv[])
|
setoptions(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
if (argc == 2) {
|
if (argc == 2) {
|
||||||
if (strcasecmp(argv[1], "enable") == 0 ||
|
if (strcasecmp(argv[1], "enable") == 0 ||
|
||||||
strcasecmp(argv[1], "on") == 0) {
|
strcasecmp(argv[1], "on") == 0) {
|
||||||
@@ -911,7 +900,6 @@ setoptions(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setrollover(int argc, char *argv[])
|
setrollover(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
if (argc == 2) {
|
if (argc == 2) {
|
||||||
if (strcasecmp(argv[1], "never") == 0 ||
|
if (strcasecmp(argv[1], "never") == 0 ||
|
||||||
strcasecmp(argv[1], "none") == 0) {
|
strcasecmp(argv[1], "none") == 0) {
|
||||||
@@ -962,7 +950,6 @@ setdebug(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setblocksize(int argc, char *argv[])
|
setblocksize(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
if (!options_rfc_enabled)
|
if (!options_rfc_enabled)
|
||||||
printf("RFC2347 style options are not enabled "
|
printf("RFC2347 style options are not enabled "
|
||||||
"(but proceeding anyway)\n");
|
"(but proceeding anyway)\n");
|
||||||
@@ -998,7 +985,6 @@ setblocksize(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setblocksize2(int argc, char *argv[])
|
setblocksize2(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
if (!options_rfc_enabled || !options_extra_enabled)
|
if (!options_rfc_enabled || !options_extra_enabled)
|
||||||
printf(
|
printf(
|
||||||
"RFC2347 style or non-RFC defined options are not enabled "
|
"RFC2347 style or non-RFC defined options are not enabled "
|
||||||
@@ -1054,7 +1040,6 @@ setblocksize2(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setpacketdrop(int argc, char *argv[])
|
setpacketdrop(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
if (argc != 1)
|
if (argc != 1)
|
||||||
packetdroppercentage = atoi(argv[1]);
|
packetdroppercentage = atoi(argv[1]);
|
||||||
|
|
||||||
@@ -1065,7 +1050,6 @@ setpacketdrop(int argc, char *argv[])
|
|||||||
static void
|
static void
|
||||||
setwindowsize(int argc, char *argv[])
|
setwindowsize(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
|
|
||||||
if (!options_rfc_enabled)
|
if (!options_rfc_enabled)
|
||||||
printf("RFC2347 style options are not enabled "
|
printf("RFC2347 style options are not enabled "
|
||||||
"(but proceeding anyway)\n");
|
"(but proceeding anyway)\n");
|
||||||
|
|||||||
+20
-20
@@ -63,13 +63,13 @@
|
|||||||
int
|
int
|
||||||
xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
||||||
{
|
{
|
||||||
|
char recvbuffer[MAXPKTSIZE];
|
||||||
|
struct sockaddr_storage serv; /* valid server port number */
|
||||||
|
struct tftp_stats tftp_stats;
|
||||||
struct tftphdr *rp;
|
struct tftphdr *rp;
|
||||||
struct servent *se;
|
struct servent *se;
|
||||||
int n, i, ret = 0;
|
int n, i, ret = 0;
|
||||||
uint16_t block, portn;
|
uint16_t block, portn;
|
||||||
struct sockaddr_storage serv; /* valid server port number */
|
|
||||||
char recvbuffer[MAXPKTSIZE];
|
|
||||||
struct tftp_stats tftp_stats;
|
|
||||||
|
|
||||||
stats_init(&tftp_stats);
|
stats_init(&tftp_stats);
|
||||||
|
|
||||||
@@ -82,7 +82,7 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
portn = htons(atoi(port));
|
portn = htons(atoi(port));
|
||||||
if (portn == 0) {
|
if (portn == 0) {
|
||||||
printf("Invalid port '%s'.\n", port ? port : "tftp");
|
printf("Invalid port '%s'.\n", port ? port : "tftp");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
((struct sockaddr_in *)&peer_sock)->sin_port = portn;
|
((struct sockaddr_in *)&peer_sock)->sin_port = portn;
|
||||||
|
|
||||||
@@ -96,7 +96,7 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
n = send_wrq(peer, name, mode);
|
n = send_wrq(peer, name, mode);
|
||||||
if (n > 0) {
|
if (n > 0) {
|
||||||
printf("Cannot send WRQ packet\n");
|
printf("Cannot send WRQ packet\n");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@@ -125,11 +125,11 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
}
|
}
|
||||||
if (i == 12) {
|
if (i == 12) {
|
||||||
printf("Transfer timed out.\n");
|
printf("Transfer timed out.\n");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
if (rp->th_opcode == ERROR) {
|
if (rp->th_opcode == ERROR) {
|
||||||
printf("Got ERROR, aborted\n");
|
printf("Got ERROR, aborted\n");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@@ -140,7 +140,7 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
if (!options_rfc_enabled) {
|
if (!options_rfc_enabled) {
|
||||||
printf("Got OACK while options are not enabled!\n");
|
printf("Got OACK while options are not enabled!\n");
|
||||||
send_error(peer, EBADOP);
|
send_error(peer, EBADOP);
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
parse_options(peer, rp->th_stuff, n + 2);
|
parse_options(peer, rp->th_stuff, n + 2);
|
||||||
@@ -148,7 +148,7 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
|
|
||||||
if (read_init(fd, NULL, mode) < 0) {
|
if (read_init(fd, NULL, mode) < 0) {
|
||||||
warn("read_init()");
|
warn("read_init()");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
block = 1;
|
block = 1;
|
||||||
@@ -158,7 +158,7 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
read_close();
|
read_close();
|
||||||
if (tftp_stats.amount > 0)
|
if (tftp_stats.amount > 0)
|
||||||
printstats("Sent", verbose, &tftp_stats);
|
printstats("Sent", verbose, &tftp_stats);
|
||||||
return ret;
|
return (ret);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@@ -167,12 +167,12 @@ xmitfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
int
|
int
|
||||||
recvfile(int peer, char *port, int fd, char *name, char *mode)
|
recvfile(int peer, char *port, int fd, char *name, char *mode)
|
||||||
{
|
{
|
||||||
|
char recvbuffer[MAXPKTSIZE];
|
||||||
|
struct tftp_stats tftp_stats;
|
||||||
struct tftphdr *rp;
|
struct tftphdr *rp;
|
||||||
struct servent *se;
|
struct servent *se;
|
||||||
uint16_t block, portn;
|
|
||||||
char recvbuffer[MAXPKTSIZE];
|
|
||||||
int n, i, ret = 0;
|
int n, i, ret = 0;
|
||||||
struct tftp_stats tftp_stats;
|
uint16_t block, portn;
|
||||||
|
|
||||||
stats_init(&tftp_stats);
|
stats_init(&tftp_stats);
|
||||||
|
|
||||||
@@ -198,7 +198,7 @@ recvfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
n = send_rrq(peer, name, mode);
|
n = send_rrq(peer, name, mode);
|
||||||
if (n > 0) {
|
if (n > 0) {
|
||||||
printf("Cannot send RRQ packet\n");
|
printf("Cannot send RRQ packet\n");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@@ -227,16 +227,16 @@ recvfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
}
|
}
|
||||||
if (i == 12) {
|
if (i == 12) {
|
||||||
printf("Transfer timed out.\n");
|
printf("Transfer timed out.\n");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
if (rp->th_opcode == ERROR) {
|
if (rp->th_opcode == ERROR) {
|
||||||
tftp_log(LOG_ERR, "Error code %d: %s", rp->th_code, rp->th_msg);
|
tftp_log(LOG_ERR, "Error code %d: %s", rp->th_code, rp->th_msg);
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (write_init(fd, NULL, mode) < 0) {
|
if (write_init(fd, NULL, mode) < 0) {
|
||||||
warn("write_init");
|
warn("write_init");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
@@ -247,7 +247,7 @@ recvfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
if (!options_rfc_enabled) {
|
if (!options_rfc_enabled) {
|
||||||
printf("Got OACK while options are not enabled!\n");
|
printf("Got OACK while options are not enabled!\n");
|
||||||
send_error(peer, EBADOP);
|
send_error(peer, EBADOP);
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
parse_options(peer, rp->th_stuff, n + 2);
|
parse_options(peer, rp->th_stuff, n + 2);
|
||||||
@@ -255,7 +255,7 @@ recvfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
n = send_ack(peer, 0);
|
n = send_ack(peer, 0);
|
||||||
if (n > 0) {
|
if (n > 0) {
|
||||||
printf("Cannot send ACK on OACK.\n");
|
printf("Cannot send ACK on OACK.\n");
|
||||||
return -1;
|
return (-1);
|
||||||
}
|
}
|
||||||
block = 0;
|
block = 0;
|
||||||
if (tftp_receive(peer, &block, &tftp_stats, NULL, 0) != 0)
|
if (tftp_receive(peer, &block, &tftp_stats, NULL, 0) != 0)
|
||||||
@@ -268,5 +268,5 @@ recvfile(int peer, char *port, int fd, char *name, char *mode)
|
|||||||
|
|
||||||
if (tftp_stats.amount > 0)
|
if (tftp_stats.amount > 0)
|
||||||
printstats("Received", verbose, &tftp_stats);
|
printstats("Received", verbose, &tftp_stats);
|
||||||
return ret;
|
return (ret);
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user