1 /* $NetBSD: util.c,v 1.23 2013/05/05 11:51:43 lukem Exp $ */
2 /* from NetBSD: util.c,v 1.158 2013/02/19 23:29:15 dsl Exp */
5 * Copyright (c) 1997-2009 The NetBSD Foundation, Inc.
8 * This code is derived from software contributed to The NetBSD Foundation
11 * This code is derived from software contributed to The NetBSD Foundation
12 * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
13 * NASA Ames Research Center.
15 * Redistribution and use in source and binary forms, with or without
16 * modification, are permitted provided that the following conditions
18 * 1. Redistributions of source code must retain the above copyright
19 * notice, this list of conditions and the following disclaimer.
20 * 2. Redistributions in binary form must reproduce the above copyright
21 * notice, this list of conditions and the following disclaimer in the
22 * documentation and/or other materials provided with the distribution.
24 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
25 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
26 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
27 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
28 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
29 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
30 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
31 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
32 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
33 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
34 * POSSIBILITY OF SUCH DAMAGE.
38 * Copyright (c) 1985, 1989, 1993, 1994
39 * The Regents of the University of California. All rights reserved.
41 * Redistribution and use in source and binary forms, with or without
42 * modification, are permitted provided that the following conditions
44 * 1. Redistributions of source code must retain the above copyright
45 * notice, this list of conditions and the following disclaimer.
46 * 2. Redistributions in binary form must reproduce the above copyright
47 * notice, this list of conditions and the following disclaimer in the
48 * documentation and/or other materials provided with the distribution.
49 * 3. Neither the name of the University nor the names of its contributors
50 * may be used to endorse or promote products derived from this software
51 * without specific prior written permission.
53 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
54 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
55 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
56 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
57 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
58 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
59 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
60 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
61 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
62 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
70 #include <sys/cdefs.h>
72 __RCSID(" NetBSD: util.c,v 1.158 2013/02/19 23:29:15 dsl Exp ");
76 * FTP User Program -- Misc support routines
78 #include <sys/param.h>
79 #include <sys/socket.h>
80 #include <sys/ioctl.h>
82 #include <netinet/in.h>
108 * Connect to peer server and auto-login, if possible.
111 setpeer(int argc, char *argv[])
119 fprintf(ttyout, "Already connected to %s, use close first.\n",
125 (void)another(&argc, &argv, "to");
126 if (argc < 2 || argc > 3) {
128 UPRINTF("usage: %s host-name [port]\n", argv[0]);
140 if (gateserver == NULL || *gateserver == '\0')
141 errx(1, "main: gateserver not defined");
142 host = hookup(gateserver, port);
144 host = hookup(argv[1], port);
147 if (gatemode && verbose) {
149 "Connecting via pass-through server %s\n",
155 * Set up defaults for FTP.
157 (void)strlcpy(typename, "ascii", sizeof(typename));
160 (void)strlcpy(formname, "non-print", sizeof(formname));
162 (void)strlcpy(modename, "stream", sizeof(modename));
164 (void)strlcpy(structname, "file", sizeof(structname));
166 (void)strlcpy(bytename, "8", sizeof(bytename));
169 (void)ftp_login(argv[1], NULL, NULL);
174 parse_feat(const char *fline)
178 * work-around broken ProFTPd servers that can't
179 * even obey RFC 2389.
181 while (*fline && isspace((int)*fline))
184 if (strcasecmp(fline, "MDTM") == 0)
185 features[FEAT_MDTM] = 1;
186 else if (strncasecmp(fline, "MLST", sizeof("MLST") - 1) == 0) {
187 features[FEAT_MLST] = 1;
188 } else if (strcasecmp(fline, "REST STREAM") == 0)
189 features[FEAT_REST_STREAM] = 1;
190 else if (strcasecmp(fline, "SIZE") == 0)
191 features[FEAT_SIZE] = 1;
192 else if (strcasecmp(fline, "TVFS") == 0)
193 features[FEAT_TVFS] = 1;
197 * Determine the remote system type (SYST) and features (FEAT).
198 * Call after a successful login (i.e, connected = -1)
209 /* determine remote system type */
210 if (command("SYST") == COMPLETE) {
212 int os_len = strcspn(reply_string + 4, " \r\n\t");
213 if (os_len > 1 && reply_string[4 + os_len - 1] == '.')
215 fprintf(ttyout, "Remote system type is %.*s.\n",
216 os_len, reply_string + 4);
219 * Decide whether we should default to bninary.
220 * Traditionally checked for "215 UNIX Type: L8", but
221 * some printers report "Linux" ! so be more forgiving.
222 * In reality we probably almost never want text any more.
224 if (!strncasecmp(reply_string + 4, "unix", 4) ||
225 !strncasecmp(reply_string + 4, "linux", 5)) {
231 * Set type to 0 (not specified by user),
232 * meaning binary by default, but don't bother
233 * telling server. We can use binary
234 * for text files unless changed by the user.
237 (void)strlcpy(typename, "binary", sizeof(typename));
240 "Using %s mode to transfer files.\n",
248 !strncmp(reply_string, "215 TOPS20", 10))
250 "Remember to set tenex mode when transferring binary files from this machine.\n",
255 /* determine features (if any) */
256 for (i = 0; i < FEAT_max; i++)
258 reply_callback = parse_feat;
259 if (command("FEAT") == COMPLETE) {
260 for (i = 0; i < FEAT_max; i++) {
261 if (features[i] == -1)
264 features[FEAT_FEAT] = 1;
266 features[FEAT_FEAT] = 0;
269 #define DEBUG_FEAT(x) fprintf(ttyout, "features[" #x "] = %d\n", features[(x)])
270 DEBUG_FEAT(FEAT_FEAT);
271 DEBUG_FEAT(FEAT_MDTM);
272 DEBUG_FEAT(FEAT_MLST);
273 DEBUG_FEAT(FEAT_REST_STREAM);
274 DEBUG_FEAT(FEAT_SIZE);
275 DEBUG_FEAT(FEAT_TVFS);
279 reply_callback = NULL;
285 * Reset the various variables that indicate connection state back to
286 * disconnected settings.
287 * The caller is responsible for issuing any commands to the remote server
288 * to perform a clean shutdown before this is invoked.
301 * determine if anonftp was specifically set with -a
302 * (1), or implicitly set by auto_fetch() (2). in the
303 * latter case, disable after the current xfer
318 * Top-level signal handler for interrupted commands.
327 write(fileno(ttyout), "\n", 1);
328 siglongjmp(toplevel, 1);
332 * Signal handler for lost connections; cleanup various elements of
333 * the connection state, and call cleanuppeer() to finish it off.
343 (void)shutdown(fileno(cout), 1+1);
348 (void)shutdown(data, 1+1);
357 (void)shutdown(fileno(cout), 1+1);
371 * Login to remote host, using given username & password if supplied.
372 * Return non-zero if successful.
375 ftp_login(const char *host, const char *luser, const char *lpass)
378 char *fuser, *pass, *facct, *p;
379 char emptypass[] = "";
380 const char *errormsg;
381 int n, aflag, rval, nlen;
384 fuser = pass = facct = NULL;
386 fuser = ftp_strdup(luser);
388 pass = ftp_strdup(lpass);
390 DPRINTF("ftp_login: user `%s' pass `%s' host `%s'\n",
391 STRorNULL(fuser), STRorNULL(pass), STRorNULL(host));
394 * Set up arguments for an anonymous FTP session, if necessary.
398 fuser = ftp_strdup("anonymous"); /* as per RFC 1635 */
400 pass = ftp_strdup(getoptionvalue("anonpass"));
403 if (ruserpass(host, &fuser, &pass, &facct) < 0) {
405 goto cleanup_ftp_login;
408 while (fuser == NULL) {
410 fprintf(ttyout, "Name (%s:%s): ", host, localname);
412 fprintf(ttyout, "Name (%s): ", host);
414 nlen = get_line(stdin, tmp, sizeof(tmp), &errormsg);
416 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "login");
418 goto cleanup_ftp_login;
419 } else if (nlen == 0) {
420 fuser = ftp_strdup(localname);
422 fuser = ftp_strdup(tmp);
430 len = strlen(fuser) + 1 + strlen(host) + 1;
431 nuser = ftp_malloc(len);
432 (void)strlcpy(nuser, fuser, len);
433 (void)strlcat(nuser, "@", len);
434 (void)strlcat(nuser, host, len);
439 n = command("USER %s", fuser);
442 p = getpass("Password: ");
445 pass = ftp_strdup(p);
446 memset(p, 0, strlen(p));
448 n = command("PASS %s", pass);
449 memset(pass, 0, strlen(pass));
454 p = getpass("Account: ");
457 facct = ftp_strdup(p);
458 memset(p, 0, strlen(p));
460 if (facct[0] == '\0') {
461 warnx("Login failed");
462 goto cleanup_ftp_login;
464 n = command("ACCT %s", facct);
465 memset(facct, 0, strlen(facct));
467 if ((n != COMPLETE) ||
468 (!aflag && facct != NULL && command("ACCT %s", facct) != COMPLETE)) {
469 warnx("Login failed");
470 goto cleanup_ftp_login;
473 username = ftp_strdup(fuser);
475 goto cleanup_ftp_login;
479 for (n = 0; n < macnum; ++n) {
480 if (!strcmp("init", macros[n].mac_name)) {
481 (void)strlcpy(line, "$init", sizeof(line));
483 domacro(margc, margv);
493 memset(pass, 0, strlen(pass));
496 memset(facct, 0, strlen(facct));
502 * `another' gets another argument, and stores the new argc and argv.
503 * It reverts to the top level (via intr()) on EOF/error.
505 * Returns false if no new arguments have been added.
508 another(int *pargc, char ***pargv, const char *aprompt)
510 const char *errormsg;
515 if (len >= sizeof(line) - 3) {
516 fputs("Sorry, arguments too long.\n", ttyout);
519 fprintf(ttyout, "(%s) ", aprompt);
522 nlen = get_line(stdin, line + len, sizeof(line)-len, &errormsg);
524 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "operation");
529 ret = margc > *pargc;
536 * glob files given in argv[] from the remote server.
537 * if errbuf isn't NULL, store error messages there instead
538 * of writing to the screen.
541 remglob(char *argv[], int doswitch, const char **errbuf)
543 static char buf[MAXPATHLEN];
544 static FILE *ftemp = NULL;
546 char temp[MAXPATHLEN];
547 int oldverbose, oldhash, oldprogress, fd;
552 if (!mflag || !connected) {
566 if ((cp = *++args) == NULL)
571 len = strlcpy(temp, tmpdir, sizeof(temp));
572 if (temp[len - 1] != '/')
573 (void)strlcat(temp, "/", sizeof(temp));
574 (void)strlcat(temp, TMPFILE, sizeof(temp));
575 if ((fd = mkstemp(temp)) < 0) {
576 warn("Unable to create temporary file `%s'", temp);
580 oldverbose = verbose;
581 verbose = (errbuf != NULL) ? -1 : 0;
583 oldprogress = progress;
588 for (rmode = "w"; *++argv != NULL; rmode = "a")
589 recvrequest("NLST", temp, *argv, rmode, 0, 0);
590 if ((code / 100) != COMPLETE) {
592 *errbuf = reply_string;
596 verbose = oldverbose;
598 progress = oldprogress;
599 ftemp = fopen(temp, "r");
603 warnx("Can't find list of remote files");
606 "Can't find list of remote files";
610 if (fgets(buf, sizeof(buf), ftemp) == NULL) {
615 if ((cp = strchr(buf, '\n')) != NULL)
621 * Glob a local file name specification with the expectation of a single
622 * return value. Can't control multiple values being expanded from the
623 * expression, we return only the first.
624 * Returns NULL on error, or a pointer to a buffer containing the filename
625 * that's the caller's responsiblity to free(3) when finished with.
628 globulize(const char *pattern)
635 return (ftp_strdup(pattern));
637 flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE;
638 memset(&gl, 0, sizeof(gl));
639 if (glob(pattern, flags, NULL, &gl) || gl.gl_pathc == 0) {
640 warnx("Glob pattern `%s' not found", pattern);
644 p = ftp_strdup(gl.gl_pathv[0]);
650 * determine size of remote file
653 remotesize(const char *file, int noisy)
662 if (! features[FEAT_SIZE]) {
665 "SIZE is not supported by remote server.\n");
666 goto cleanup_remotesize;
668 r = command("SIZE %s", file);
672 cp = strchr(reply_string, ' ');
675 size = STRTOLL(cp, &ep, 10);
676 if (*ep != '\0' && !isspace((unsigned char)*ep))
680 if (r == ERROR && code == 500 && features[FEAT_SIZE] == -1)
681 features[FEAT_SIZE] = 0;
682 if (noisy && ftp_debug == 0) {
683 fputs(reply_string, ttyout);
693 * determine last modification time (in GMT) of remote file
696 remotemodtime(const char *file, int noisy)
698 int overbose, ocode, r;
706 if (! features[FEAT_MDTM]) {
709 "MDTM is not supported by remote server.\n");
710 goto cleanup_parse_time;
712 r = command("MDTM %s", file);
715 char *timestr, *frac;
718 * time-val = 14DIGIT [ "." 1*DIGIT ]
719 * YYYYMMDDHHMMSS[.sss]
720 * mdtm-response = "213" SP time-val CRLF / error-response
722 timestr = reply_string + 4;
726 * XXX: ignored for now
728 frac = strchr(timestr, '\r');
731 frac = strchr(timestr, '.');
734 if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
736 * XXX: Workaround for lame ftpd's that return
737 * `19100' instead of `2000'
740 "Y2K warning! Incorrect time-val `%s' received from server.\n",
745 fprintf(ttyout, "Converted to `%s'\n", timestr);
747 memset(&timebuf, 0, sizeof(timebuf));
748 if (strlen(timestr) != 14 ||
749 (strptime(timestr, "%Y%m%d%H%M%S", &timebuf) == NULL)) {
751 fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
752 goto cleanup_parse_time;
754 timebuf.tm_isdst = -1;
755 rtime = timegm(&timebuf);
757 if (noisy || ftp_debug != 0)
760 goto cleanup_parse_time;
762 DPRINTF("remotemodtime: parsed time `%s' as " LLF
765 rfc2822time(localtime(&rtime)));
768 if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
769 features[FEAT_MDTM] = 0;
770 if (noisy && ftp_debug == 0) {
771 fputs(reply_string, ttyout);
783 * Format tm in an RFC 2822 compatible manner, with a trailing \n.
784 * Returns a pointer to a static string containing the result.
787 rfc2822time(const struct tm *tm)
789 static char result[50];
791 if (strftime(result, sizeof(result),
792 "%a, %d %b %Y %H:%M:%S %z\n", tm) == 0)
793 errx(1, "Can't convert RFC 2822 time: buffer too small");
798 * Parse HTTP-date as per RFC 2616.
799 * Return a pointer to the next character of the consumed date string,
803 parse_rfc2616time(struct tm *parsed, const char *httpdate)
806 #if defined(HAVE_SETLOCALE)
807 const char *curlocale;
809 /* The representation of %a depends on the current locale. */
810 curlocale = setlocale(LC_TIME, NULL);
811 (void)setlocale(LC_TIME, "C");
814 if ((t = strptime(httpdate, "%a, %d %b %Y %H:%M:%S GMT", parsed)) ||
816 (t = strptime(httpdate, "%a, %d-%b-%y %H:%M:%S GMT", parsed)) ||
818 (t = strptime(httpdate, "%a, %b %d %H:%M:%S %Y", parsed))) {
821 #if defined(HAVE_SETLOCALE)
822 (void)setlocale(LC_TIME, curlocale);
828 * Update global `localcwd', which contains the state of the local cwd
834 if (getcwd(localcwd, sizeof(localcwd)) == NULL)
836 DPRINTF("updatelocalcwd: got `%s'\n", localcwd);
840 * Update global `remotecwd', which contains the state of the remote cwd
843 updateremotecwd(void)
853 if (command("PWD") != COMPLETE)
855 cp = strchr(reply_string, ' ');
856 if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
859 for (i = 0; *cp && i < sizeof(remotecwd) - 1; i++, cp++) {
869 DPRINTF("updateremotecwd: got `%s'\n", remotecwd);
870 goto cleanupremotecwd;
879 * Ensure file is in or under dir.
880 * Returns 1 if so, 0 if not (or an error occurred).
883 fileindir(const char *file, const char *dir)
885 char parentdirbuf[PATH_MAX+1], *parentdir;
886 char realdir[PATH_MAX+1];
889 /* determine parent directory of file */
890 (void)strlcpy(parentdirbuf, file, sizeof(parentdirbuf));
891 parentdir = dirname(parentdirbuf);
892 if (strcmp(parentdir, ".") == 0)
893 return 1; /* current directory is ok */
895 /* find the directory */
896 if (realpath(parentdir, realdir) == NULL) {
897 warn("Unable to determine real path of `%s'", parentdir);
900 if (realdir[0] != '/') /* relative result is ok */
902 dirlen = strlen(dir);
903 if (strncmp(realdir, dir, dirlen) == 0 &&
904 (realdir[dirlen] == '/' || realdir[dirlen] == '\0'))
910 * List words in stringlist, vertically arranged
913 list_vertical(StringList *sl)
916 size_t columns, lines;
922 for (i = 0 ; i < sl->sl_cur ; i++) {
923 w = strlen(sl->sl_str[i]);
927 width = (width + 8) &~ 7;
929 columns = ttywidth / width;
932 lines = (sl->sl_cur + columns - 1) / columns;
933 for (i = 0; i < lines; i++) {
934 for (j = 0; j < columns; j++) {
935 p = sl->sl_str[j * lines + i];
938 if (j * lines + i + lines >= sl->sl_cur) {
946 (void)putc('\t', ttyout);
954 * Update the global ttywidth value, using TIOCGWINSZ.
959 struct winsize winsize;
962 if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
964 ttywidth = winsize.ws_col;
971 * Change the rate limit up (SIGUSR1) or down (SIGUSR2)
980 rate_get += rate_get_incr;
982 rate_put += rate_put_incr;
985 if (rate_get && rate_get > rate_get_incr)
986 rate_get -= rate_get_incr;
987 if (rate_put && rate_put > rate_put_incr)
988 rate_put -= rate_put_incr;
991 err(1, "crankrate invoked with unknown signal: %d", sig);
997 * Setup or cleanup EditLine structures
999 #ifndef NO_EDITCOMPLETE
1001 controlediting(void)
1003 if (editing && el == NULL && hist == NULL) {
1007 el = el_init(getprogname(), stdin, ttyout, stderr);
1009 hist = history_init(); /* init the builtin history */
1010 history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
1011 el_set(el, EL_HIST, history, hist); /* use history */
1013 el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
1014 el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
1015 el_set(el, EL_RPROMPT, rprompt);
1017 /* add local file completion, bind to TAB */
1018 el_set(el, EL_ADDFN, "ftp-complete",
1019 "Context sensitive argument completion",
1021 el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
1022 el_source(el, NULL); /* read ~/.editrc */
1023 if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
1024 editing = 0; /* the user doesn't want editing,
1025 * so disable, and let statement
1028 el_set(el, EL_SIGNAL, 1);
1041 #endif /* !NO_EDITCOMPLETE */
1044 * Convert the string `arg' to an int, which may have an optional SI suffix
1045 * (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
1048 strsuftoi(const char *arg)
1053 if (!isdigit((unsigned char)arg[0]))
1056 val = strtol(arg, &cp, 10);
1058 if (cp[0] != '\0' && cp[1] != '\0')
1060 switch (tolower((unsigned char)cp[0])) {
1077 if (val < 0 || val > INT_MAX)
1084 * Set up socket buffer sizes before a connection is made.
1087 setupsockbufsize(int sock)
1091 if (0 == rcvbuf_size) {
1092 slen = sizeof(rcvbuf_size);
1093 if (getsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1094 (void *)&rcvbuf_size, &slen) == -1)
1095 err(1, "Unable to determine rcvbuf size");
1096 if (rcvbuf_size <= 0)
1097 rcvbuf_size = 8 * 1024;
1098 if (rcvbuf_size > 8 * 1024 * 1024)
1099 rcvbuf_size = 8 * 1024 * 1024;
1100 DPRINTF("setupsockbufsize: rcvbuf_size determined as %d\n",
1103 if (0 == sndbuf_size) {
1104 slen = sizeof(sndbuf_size);
1105 if (getsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1106 (void *)&sndbuf_size, &slen) == -1)
1107 err(1, "Unable to determine sndbuf size");
1108 if (sndbuf_size <= 0)
1109 sndbuf_size = 8 * 1024;
1110 if (sndbuf_size > 8 * 1024 * 1024)
1111 sndbuf_size = 8 * 1024 * 1024;
1112 DPRINTF("setupsockbufsize: sndbuf_size determined as %d\n",
1116 if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1117 (void *)&sndbuf_size, sizeof(sndbuf_size)) == -1)
1118 warn("Unable to set sndbuf size %d", sndbuf_size);
1120 if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1121 (void *)&rcvbuf_size, sizeof(rcvbuf_size)) == -1)
1122 warn("Unable to set rcvbuf size %d", rcvbuf_size);
1126 * Copy characters from src into dst, \ quoting characters that require it
1129 ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
1134 while (src[si] != '\0' && di < dstlen && si < srclen) {
1143 * Need room for two characters and NUL, avoiding
1144 * incomplete escape sequences at end of dst.
1146 if (di >= dstlen - 3)
1151 dst[di] = src[si++];
1160 * Copy src into buf (which is len bytes long), expanding % sequences.
1163 formatbuf(char *buf, size_t len, const char *src)
1165 const char *p, *p2, *q;
1167 int op, updirs, pdirs;
1169 #define ADDBUF(x) do { \
1176 for (i = 0; *p; p++) {
1188 p2 = connected ? remotecwd : "";
1191 /* option to determine fixed # of dirs from path */
1192 if (op == '.' || op == 'c') {
1196 while (*p2) /* calc # of /'s */
1199 if (p[1] == '0') { /* print <x> or ... */
1203 if (p[1] >= '1' && p[1] <= '9') {
1204 /* calc # to skip */
1211 while (skip-- > 0) {
1212 while ((p2 > q) && (*p2 != '/'))
1217 if (*p2 == '/' && p2 != q)
1221 if (updirs > 0 && pdirs) {
1235 ADDBUF('0' + updirs);
1245 for (p2 = connected && hostname ? hostname : "-";
1247 if (op == 'm' && *p2 == '.')
1254 for (p2 = connected ? username : "-"; *p2 ; p2++)
1262 default: /* display unknown codes literally */
1274 * Determine if given string is an IPv6 address or not.
1275 * Return 1 for yes, 0 for no
1278 isipv6addr(const char *addr)
1282 struct addrinfo hints, *res;
1284 memset(&hints, 0, sizeof(hints));
1285 hints.ai_family = AF_INET6;
1286 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
1287 hints.ai_flags = AI_NUMERICHOST;
1288 if (getaddrinfo(addr, "0", &hints, &res) != 0)
1294 DPRINTF("isipv6addr: got %d for %s\n", rv, addr);
1296 return (rv == 1) ? 1 : 0;
1300 * Read a line from the FILE stream into buf/buflen using fgets(), so up
1301 * to buflen-1 chars will be read and the result will be NUL terminated.
1302 * If the line has a trailing newline it will be removed.
1303 * If the line is too long, excess characters will be read until
1304 * newline/EOF/error.
1305 * If EOF/error occurs or a too-long line is encountered and errormsg
1306 * isn't NULL, it will be changed to a description of the problem.
1307 * (The EOF message has a leading \n for cosmetic purposes).
1309 * >=0 length of line (excluding trailing newline) if all ok
1311 * -2 EOF encountered
1312 * -3 line was too long
1315 get_line(FILE *stream, char *buf, size_t buflen, const char **errormsg)
1320 if (fgets(buf, buflen, stream) == NULL) {
1321 if (feof(stream)) { /* EOF */
1324 *errormsg = "\nEOF received";
1325 } else { /* error */
1328 *errormsg = "Error encountered";
1334 if (buf[len-1] == '\n') { /* clear any trailing newline */
1336 } else if (len == buflen-1) { /* line too long */
1337 while ((ch = getchar()) != '\n' && ch != EOF)
1340 *errormsg = "Input line is too long";
1350 * Internal version of connect(2); sets socket buffer sizes,
1351 * binds to a specific local address (if set), and
1352 * supports a connection timeout using a non-blocking connect(2) with
1354 * Socket fcntl flags are temporarily updated to include O_NONBLOCK;
1355 * these will not be reverted on connection failure.
1356 * Returns 0 on success, or -1 upon failure (with an appropriate
1357 * error message displayed.)
1360 ftp_connect(int sock, const struct sockaddr *name, socklen_t namelen, int pe)
1362 int flags, rv, timeout, error;
1364 struct timeval endtime, now, td;
1365 struct pollfd pfd[1];
1366 char hname[NI_MAXHOST];
1367 char sname[NI_MAXSERV];
1369 setupsockbufsize(sock);
1370 if (getnameinfo(name, namelen,
1371 hname, sizeof(hname), sname, sizeof(sname),
1372 NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
1373 strlcpy(hname, "?", sizeof(hname));
1374 strlcpy(sname, "?", sizeof(sname));
1377 if (bindai != NULL) { /* bind to specific addr */
1378 struct addrinfo *ai;
1380 for (ai = bindai; ai != NULL; ai = ai->ai_next) {
1381 if (ai->ai_family == name->sa_family)
1386 if (bind(sock, ai->ai_addr, ai->ai_addrlen) == -1) {
1387 char bname[NI_MAXHOST];
1391 if (getnameinfo(ai->ai_addr, ai->ai_addrlen,
1392 bname, sizeof(bname), NULL, 0, NI_NUMERICHOST) != 0)
1393 strlcpy(bname, "?", sizeof(bname));
1395 warn("Can't bind to `%s'", bname);
1400 /* save current socket flags */
1401 if ((flags = fcntl(sock, F_GETFL, 0)) == -1) {
1402 warn("Can't %s socket flags for connect to `%s:%s'",
1403 "save", hname, sname);
1406 /* set non-blocking connect */
1407 if (fcntl(sock, F_SETFL, flags | O_NONBLOCK) == -1) {
1408 warn("Can't set socket non-blocking for connect to `%s:%s'",
1413 /* NOTE: we now must restore socket flags on successful exit */
1416 pfd[0].events = POLLIN|POLLOUT;
1418 if (quit_time > 0) { /* want a non default timeout */
1419 (void)gettimeofday(&endtime, NULL);
1420 endtime.tv_sec += quit_time; /* determine end time */
1423 rv = connect(sock, name, namelen); /* inititate the connection */
1424 if (rv == -1) { /* connection error */
1425 if (errno != EINPROGRESS) { /* error isn't "please wait" */
1426 if (pe || (errno != EHOSTUNREACH))
1428 warn("Can't connect to `%s:%s'", hname, sname);
1432 /* connect EINPROGRESS; wait */
1434 if (quit_time > 0) { /* determine timeout */
1435 (void)gettimeofday(&now, NULL);
1436 timersub(&endtime, &now, &td);
1437 timeout = td.tv_sec * 1000 + td.tv_usec/1000;
1444 rv = ftp_poll(pfd, 1, timeout);
1445 /* loop until poll ! EINTR */
1446 } while (rv == -1 && errno == EINTR);
1448 if (rv == 0) { /* poll (connect) timed out */
1453 if (rv == -1) { /* poll error */
1455 } else if (pfd[0].revents & (POLLIN|POLLOUT)) {
1456 slen = sizeof(error); /* OK, or pending error */
1457 if (getsockopt(sock, SOL_SOCKET, SO_ERROR,
1458 &error, &slen) == -1) {
1459 /* Solaris pending error */
1461 } else if (error != 0) {
1462 errno = error; /* BSD pending error */
1466 errno = EBADF; /* this shouldn't happen ... */
1471 if (fcntl(sock, F_SETFL, flags) == -1) {
1472 /* restore socket flags */
1473 warn("Can't %s socket flags for connect to `%s:%s'",
1474 "restore", hname, sname);
1481 * Internal version of listen(2); sets socket buffer sizes first.
1484 ftp_listen(int sock, int backlog)
1487 setupsockbufsize(sock);
1488 return (listen(sock, backlog));
1492 * Internal version of poll(2), to allow reimplementation by select(2)
1493 * on platforms without the former.
1496 ftp_poll(struct pollfd *fds, int nfds, int timeout)
1498 #if defined(HAVE_POLL)
1499 return poll(fds, nfds, timeout);
1501 #elif defined(HAVE_SELECT)
1502 /* implement poll(2) using select(2) */
1503 fd_set rset, wset, xset;
1504 const int rsetflags = POLLIN | POLLRDNORM;
1505 const int wsetflags = POLLOUT | POLLWRNORM;
1506 const int xsetflags = POLLRDBAND;
1507 struct timeval tv, *ptv;
1510 FD_ZERO(&rset); /* build list of read & write events */
1514 for (i = 0; i < nfds; i++) {
1515 if (fds[i].fd > FD_SETSIZE) {
1516 warnx("can't select fd %d", fds[i].fd);
1519 } else if (fds[i].fd > max)
1521 if (fds[i].events & rsetflags)
1522 FD_SET(fds[i].fd, &rset);
1523 if (fds[i].events & wsetflags)
1524 FD_SET(fds[i].fd, &wset);
1525 if (fds[i].events & xsetflags)
1526 FD_SET(fds[i].fd, &xset);
1529 ptv = &tv; /* determine timeout */
1530 if (timeout == -1) { /* wait forever */
1532 } else if (timeout == 0) { /* poll once */
1536 else if (timeout != 0) { /* wait timeout milliseconds */
1537 ptv->tv_sec = timeout / 1000;
1538 ptv->tv_usec = (timeout % 1000) * 1000;
1540 rv = select(max + 1, &rset, &wset, &xset, ptv);
1541 if (rv <= 0) /* -1 == error, 0 == timeout */
1544 for (i = 0; i < nfds; i++) { /* determine results */
1545 if (FD_ISSET(fds[i].fd, &rset))
1546 fds[i].revents |= (fds[i].events & rsetflags);
1547 if (FD_ISSET(fds[i].fd, &wset))
1548 fds[i].revents |= (fds[i].events & wsetflags);
1549 if (FD_ISSET(fds[i].fd, &xset))
1550 fds[i].revents |= (fds[i].events & xsetflags);
1555 # error no way to implement xpoll
1560 * malloc() with inbuilt error checking
1563 ftp_malloc(size_t size)
1569 err(1, "Unable to allocate %ld bytes of memory", (long)size);
1574 * sl_init() with inbuilt error checking
1583 err(1, "Unable to allocate memory for stringlist");
1588 * sl_add() with inbuilt error checking
1591 ftp_sl_add(StringList *sl, char *i)
1594 if (sl_add(sl, i) == -1)
1595 err(1, "Unable to add `%s' to stringlist", i);
1599 * strdup() with inbuilt error checking
1602 ftp_strdup(const char *str)
1607 errx(1, "ftp_strdup: called with NULL argument");
1610 err(1, "Unable to allocate memory for string copy");