1 /* $NetBSD: util.c,v 1.147 2008/05/10 00:05:31 skd Exp $ */
4 * Copyright (c) 1997-2008 The NetBSD Foundation, Inc.
7 * This code is derived from software contributed to The NetBSD Foundation
10 * This code is derived from software contributed to The NetBSD Foundation
11 * by Jason R. Thorpe of the Numerical Aerospace Simulation Facility,
12 * NASA Ames Research Center.
14 * Redistribution and use in source and binary forms, with or without
15 * modification, are permitted provided that the following conditions
17 * 1. Redistributions of source code must retain the above copyright
18 * notice, this list of conditions and the following disclaimer.
19 * 2. Redistributions in binary form must reproduce the above copyright
20 * notice, this list of conditions and the following disclaimer in the
21 * documentation and/or other materials provided with the distribution.
23 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
24 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
25 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
26 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
27 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
28 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
29 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
30 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
31 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
32 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
33 * POSSIBILITY OF SUCH DAMAGE.
37 * Copyright (c) 1985, 1989, 1993, 1994
38 * The Regents of the University of California. All rights reserved.
40 * Redistribution and use in source and binary forms, with or without
41 * modification, are permitted provided that the following conditions
43 * 1. Redistributions of source code must retain the above copyright
44 * notice, this list of conditions and the following disclaimer.
45 * 2. Redistributions in binary form must reproduce the above copyright
46 * notice, this list of conditions and the following disclaimer in the
47 * documentation and/or other materials provided with the distribution.
48 * 3. Neither the name of the University nor the names of its contributors
49 * may be used to endorse or promote products derived from this software
50 * without specific prior written permission.
52 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
53 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
54 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
55 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
56 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
57 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
58 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
59 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
60 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
61 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
65 #include <sys/cdefs.h>
67 __RCSID("$NetBSD: util.c,v 1.147 2008/05/10 00:05:31 skd Exp $");
71 * FTP User Program -- Misc support routines
73 #include <sys/param.h>
74 #include <sys/socket.h>
75 #include <sys/ioctl.h>
77 #include <netinet/in.h>
100 * Connect to peer server and auto-login, if possible.
103 setpeer(int argc, char *argv[])
111 fprintf(ttyout, "Already connected to %s, use close first.\n",
117 (void)another(&argc, &argv, "to");
118 if (argc < 2 || argc > 3) {
120 UPRINTF("usage: %s host-name [port]\n", argv[0]);
132 if (gateserver == NULL || *gateserver == '\0')
133 errx(1, "main: gateserver not defined");
134 host = hookup(gateserver, port);
136 host = hookup(argv[1], port);
139 if (gatemode && verbose) {
141 "Connecting via pass-through server %s\n",
147 * Set up defaults for FTP.
149 (void)strlcpy(typename, "ascii", sizeof(typename));
152 (void)strlcpy(formname, "non-print", sizeof(formname));
154 (void)strlcpy(modename, "stream", sizeof(modename));
156 (void)strlcpy(structname, "file", sizeof(structname));
158 (void)strlcpy(bytename, "8", sizeof(bytename));
161 (void)ftp_login(argv[1], NULL, NULL);
166 parse_feat(const char *line)
170 * work-around broken ProFTPd servers that can't
173 while (*line && isspace((int)*line))
176 if (strcasecmp(line, "MDTM") == 0)
177 features[FEAT_MDTM] = 1;
178 else if (strncasecmp(line, "MLST", sizeof("MLST") - 1) == 0) {
179 features[FEAT_MLST] = 1;
180 } else if (strcasecmp(line, "REST STREAM") == 0)
181 features[FEAT_REST_STREAM] = 1;
182 else if (strcasecmp(line, "SIZE") == 0)
183 features[FEAT_SIZE] = 1;
184 else if (strcasecmp(line, "TVFS") == 0)
185 features[FEAT_TVFS] = 1;
189 * Determine the remote system type (SYST) and features (FEAT).
190 * Call after a successful login (i.e, connected = -1)
201 /* determine remote system type */
202 if (command("SYST") == COMPLETE) {
207 cp = strchr(reply_string + 4, ' ');
209 cp = strchr(reply_string + 4, '\r');
217 fprintf(ttyout, "Remote system type is %s.\n",
222 if (!strncmp(reply_string, "215 UNIX Type: L8", 17)) {
228 * Set type to 0 (not specified by user),
229 * meaning binary by default, but don't bother
230 * telling server. We can use binary
231 * for text files unless changed by the user.
234 (void)strlcpy(typename, "binary", sizeof(typename));
237 "Using %s mode to transfer files.\n",
245 !strncmp(reply_string, "215 TOPS20", 10))
247 "Remember to set tenex mode when transferring binary files from this machine.\n",
252 /* determine features (if any) */
253 for (i = 0; i < FEAT_max; i++)
255 reply_callback = parse_feat;
256 if (command("FEAT") == COMPLETE) {
257 for (i = 0; i < FEAT_max; i++) {
258 if (features[i] == -1)
261 features[FEAT_FEAT] = 1;
263 features[FEAT_FEAT] = 0;
266 #define DEBUG_FEAT(x) fprintf(ttyout, "features[" #x "] = %d\n", features[(x)])
267 DEBUG_FEAT(FEAT_FEAT);
268 DEBUG_FEAT(FEAT_MDTM);
269 DEBUG_FEAT(FEAT_MLST);
270 DEBUG_FEAT(FEAT_REST_STREAM);
271 DEBUG_FEAT(FEAT_SIZE);
272 DEBUG_FEAT(FEAT_TVFS);
276 reply_callback = NULL;
282 * Reset the various variables that indicate connection state back to
283 * disconnected settings.
284 * The caller is responsible for issuing any commands to the remote server
285 * to perform a clean shutdown before this is invoked.
298 * determine if anonftp was specifically set with -a
299 * (1), or implicitly set by auto_fetch() (2). in the
300 * latter case, disable after the current xfer
315 * Top-level signal handler for interrupted commands.
324 write(fileno(ttyout), "\n", 1);
325 siglongjmp(toplevel, 1);
329 * Signal handler for lost connections; cleanup various elements of
330 * the connection state, and call cleanuppeer() to finish it off.
340 (void)shutdown(fileno(cout), 1+1);
345 (void)shutdown(data, 1+1);
354 (void)shutdown(fileno(cout), 1+1);
368 * Login to remote host, using given username & password if supplied.
369 * Return non-zero if successful.
372 ftp_login(const char *host, const char *luser, const char *lpass)
375 char *user, *pass, *acct, *p;
376 char emptypass[] = "";
377 const char *errormsg;
378 int n, aflag, rval, nlen;
381 user = pass = acct = NULL;
383 user = ftp_strdup(luser);
385 pass = ftp_strdup(lpass);
387 DPRINTF("ftp_login: user `%s' pass `%s' host `%s'\n",
388 STRorNULL(user), STRorNULL(pass), STRorNULL(host));
391 * Set up arguments for an anonymous FTP session, if necessary.
395 user = ftp_strdup("anonymous"); /* as per RFC1635 */
397 pass = ftp_strdup(getoptionvalue("anonpass"));
400 if (ruserpass(host, &user, &pass, &acct) < 0) {
402 goto cleanup_ftp_login;
405 while (user == NULL) {
407 fprintf(ttyout, "Name (%s:%s): ", host, localname);
409 fprintf(ttyout, "Name (%s): ", host);
411 nlen = getline(stdin, tmp, sizeof(tmp), &errormsg);
413 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "login");
415 goto cleanup_ftp_login;
416 } else if (nlen == 0) {
417 user = ftp_strdup(localname);
419 user = ftp_strdup(tmp);
427 len = strlen(user) + 1 + strlen(host) + 1;
428 nuser = ftp_malloc(len);
429 (void)strlcpy(nuser, user, len);
430 (void)strlcat(nuser, "@", len);
431 (void)strlcat(nuser, host, len);
436 n = command("USER %s", user);
439 p = getpass("Password: ");
442 pass = ftp_strdup(p);
443 memset(p, 0, strlen(p));
445 n = command("PASS %s", pass);
446 memset(pass, 0, strlen(pass));
451 p = getpass("Account: ");
454 acct = ftp_strdup(p);
455 memset(p, 0, strlen(p));
457 if (acct[0] == '\0') {
458 warnx("Login failed");
459 goto cleanup_ftp_login;
461 n = command("ACCT %s", acct);
462 memset(acct, 0, strlen(acct));
464 if ((n != COMPLETE) ||
465 (!aflag && acct != NULL && command("ACCT %s", acct) != COMPLETE)) {
466 warnx("Login failed");
467 goto cleanup_ftp_login;
470 username = ftp_strdup(user);
472 goto cleanup_ftp_login;
476 for (n = 0; n < macnum; ++n) {
477 if (!strcmp("init", macros[n].mac_name)) {
478 (void)strlcpy(line, "$init", sizeof(line));
480 domacro(margc, margv);
490 memset(pass, 0, strlen(pass));
493 memset(acct, 0, strlen(acct));
499 * `another' gets another argument, and stores the new argc and argv.
500 * It reverts to the top level (via intr()) on EOF/error.
502 * Returns false if no new arguments have been added.
505 another(int *pargc, char ***pargv, const char *prompt)
507 const char *errormsg;
512 if (len >= sizeof(line) - 3) {
513 fputs("Sorry, arguments too long.\n", ttyout);
516 fprintf(ttyout, "(%s) ", prompt);
519 nlen = getline(stdin, line + len, sizeof(line)-len, &errormsg);
521 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "operation");
526 ret = margc > *pargc;
533 * glob files given in argv[] from the remote server.
534 * if errbuf isn't NULL, store error messages there instead
535 * of writing to the screen.
538 remglob(char *argv[], int doswitch, const char **errbuf)
540 static char buf[MAXPATHLEN];
541 static FILE *ftemp = NULL;
543 char temp[MAXPATHLEN];
544 int oldverbose, oldhash, oldprogress, fd;
549 if (!mflag || !connected) {
563 if ((cp = *++args) == NULL)
568 len = strlcpy(temp, tmpdir, sizeof(temp));
569 if (temp[len - 1] != '/')
570 (void)strlcat(temp, "/", sizeof(temp));
571 (void)strlcat(temp, TMPFILE, sizeof(temp));
572 if ((fd = mkstemp(temp)) < 0) {
573 warn("Unable to create temporary file `%s'", temp);
577 oldverbose = verbose;
578 verbose = (errbuf != NULL) ? -1 : 0;
580 oldprogress = progress;
585 for (mode = "w"; *++argv != NULL; mode = "a")
586 recvrequest("NLST", temp, *argv, mode, 0, 0);
587 if ((code / 100) != COMPLETE) {
589 *errbuf = reply_string;
593 verbose = oldverbose;
595 progress = oldprogress;
596 ftemp = fopen(temp, "r");
600 warnx("Can't find list of remote files");
603 "Can't find list of remote files";
607 if (fgets(buf, sizeof(buf), ftemp) == NULL) {
612 if ((cp = strchr(buf, '\n')) != NULL)
618 * Glob a local file name specification with the expectation of a single
619 * return value. Can't control multiple values being expanded from the
620 * expression, we return only the first.
621 * Returns NULL on error, or a pointer to a buffer containing the filename
622 * that's the caller's responsiblity to free(3) when finished with.
625 globulize(const char *pattern)
632 return (ftp_strdup(pattern));
634 flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE;
635 memset(&gl, 0, sizeof(gl));
636 if (glob(pattern, flags, NULL, &gl) || gl.gl_pathc == 0) {
637 warnx("Glob pattern `%s' not found", pattern);
641 p = ftp_strdup(gl.gl_pathv[0]);
647 * determine size of remote file
650 remotesize(const char *file, int noisy)
659 if (! features[FEAT_SIZE]) {
662 "SIZE is not supported by remote server.\n");
663 goto cleanup_remotesize;
665 r = command("SIZE %s", file);
669 cp = strchr(reply_string, ' ');
672 size = STRTOLL(cp, &ep, 10);
673 if (*ep != '\0' && !isspace((unsigned char)*ep))
677 if (r == ERROR && code == 500 && features[FEAT_SIZE] == -1)
678 features[FEAT_SIZE] = 0;
679 if (noisy && ftp_debug == 0) {
680 fputs(reply_string, ttyout);
690 * determine last modification time (in GMT) of remote file
693 remotemodtime(const char *file, int noisy)
695 int overbose, ocode, r;
703 if (! features[FEAT_MDTM]) {
706 "MDTM is not supported by remote server.\n");
707 goto cleanup_parse_time;
709 r = command("MDTM %s", file);
712 char *timestr, *frac;
715 * time-val = 14DIGIT [ "." 1*DIGIT ]
716 * YYYYMMDDHHMMSS[.sss]
717 * mdtm-response = "213" SP time-val CRLF / error-response
719 timestr = reply_string + 4;
723 * XXX: ignored for now
725 frac = strchr(timestr, '\r');
728 frac = strchr(timestr, '.');
731 if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
733 * XXX: Workaround for lame ftpd's that return
734 * `19100' instead of `2000'
737 "Y2K warning! Incorrect time-val `%s' received from server.\n",
742 fprintf(ttyout, "Converted to `%s'\n", timestr);
744 memset(&timebuf, 0, sizeof(timebuf));
745 if (strlen(timestr) != 14 ||
746 (strptime(timestr, "%Y%m%d%H%M%S", &timebuf) == NULL)) {
748 fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
749 goto cleanup_parse_time;
751 timebuf.tm_isdst = -1;
752 rtime = timegm(&timebuf);
754 if (noisy || ftp_debug != 0)
757 goto cleanup_parse_time;
759 DPRINTF("remotemodtime: parsed date `%s' as " LLF
762 rfc2822time(localtime(&rtime)));
764 if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
765 features[FEAT_MDTM] = 0;
766 if (noisy && ftp_debug == 0) {
767 fputs(reply_string, ttyout);
779 * Format tm in an RFC2822 compatible manner, with a trailing \n.
780 * Returns a pointer to a static string containing the result.
783 rfc2822time(const struct tm *tm)
785 static char result[50];
787 if (strftime(result, sizeof(result),
788 "%a, %d %b %Y %H:%M:%S %z\n", tm) == 0)
789 errx(1, "Can't convert RFC2822 time: buffer too small");
794 * Update global `localcwd', which contains the state of the local cwd
800 if (getcwd(localcwd, sizeof(localcwd)) == NULL)
802 DPRINTF("updatelocalcwd: got `%s'\n", localcwd);
806 * Update global `remotecwd', which contains the state of the remote cwd
809 updateremotecwd(void)
811 int overbose, ocode, i;
818 if (command("PWD") != COMPLETE)
820 cp = strchr(reply_string, ' ');
821 if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
824 for (i = 0; *cp && i < sizeof(remotecwd) - 1; i++, cp++) {
834 DPRINTF("updateremotecwd: got `%s'\n", remotecwd);
835 goto cleanupremotecwd;
844 * Ensure file is in or under dir.
845 * Returns 1 if so, 0 if not (or an error occurred).
848 fileindir(const char *file, const char *dir)
850 char parentdirbuf[PATH_MAX+1], *parentdir;
851 char realdir[PATH_MAX+1];
854 /* determine parent directory of file */
855 (void)strlcpy(parentdirbuf, file, sizeof(parentdirbuf));
856 parentdir = dirname(parentdirbuf);
857 if (strcmp(parentdir, ".") == 0)
858 return 1; /* current directory is ok */
860 /* find the directory */
861 if (realpath(parentdir, realdir) == NULL) {
862 warn("Unable to determine real path of `%s'", parentdir);
865 if (realdir[0] != '/') /* relative result is ok */
867 dirlen = strlen(dir);
868 if (strncmp(realdir, dir, dirlen) == 0 &&
869 (realdir[dirlen] == '/' || realdir[dirlen] == '\0'))
875 * List words in stringlist, vertically arranged
878 list_vertical(StringList *sl)
887 for (i = 0 ; i < sl->sl_cur ; i++) {
888 w = strlen(sl->sl_str[i]);
892 width = (width + 8) &~ 7;
894 columns = ttywidth / width;
897 lines = (sl->sl_cur + columns - 1) / columns;
898 for (i = 0; i < lines; i++) {
899 for (j = 0; j < columns; j++) {
900 p = sl->sl_str[j * lines + i];
903 if (j * lines + i + lines >= sl->sl_cur) {
911 (void)putc('\t', ttyout);
919 * Update the global ttywidth value, using TIOCGWINSZ.
924 struct winsize winsize;
927 if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
929 ttywidth = winsize.ws_col;
936 * Change the rate limit up (SIGUSR1) or down (SIGUSR2)
945 rate_get += rate_get_incr;
947 rate_put += rate_put_incr;
950 if (rate_get && rate_get > rate_get_incr)
951 rate_get -= rate_get_incr;
952 if (rate_put && rate_put > rate_put_incr)
953 rate_put -= rate_put_incr;
956 err(1, "crankrate invoked with unknown signal: %d", sig);
962 * Setup or cleanup EditLine structures
964 #ifndef NO_EDITCOMPLETE
968 if (editing && el == NULL && hist == NULL) {
972 el = el_init(getprogname(), stdin, ttyout, stderr);
974 hist = history_init(); /* init the builtin history */
975 history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
976 el_set(el, EL_HIST, history, hist); /* use history */
978 el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
979 el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
980 el_set(el, EL_RPROMPT, rprompt);
982 /* add local file completion, bind to TAB */
983 el_set(el, EL_ADDFN, "ftp-complete",
984 "Context sensitive argument completion",
986 el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
987 el_source(el, NULL); /* read ~/.editrc */
988 if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
989 editing = 0; /* the user doesn't want editing,
990 * so disable, and let statement
993 el_set(el, EL_SIGNAL, 1);
1006 #endif /* !NO_EDITCOMPLETE */
1009 * Convert the string `arg' to an int, which may have an optional SI suffix
1010 * (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
1013 strsuftoi(const char *arg)
1018 if (!isdigit((unsigned char)arg[0]))
1021 val = strtol(arg, &cp, 10);
1023 if (cp[0] != '\0' && cp[1] != '\0')
1025 switch (tolower((unsigned char)cp[0])) {
1042 if (val < 0 || val > INT_MAX)
1049 * Set up socket buffer sizes before a connection is made.
1052 setupsockbufsize(int sock)
1055 if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1056 (void *)&sndbuf_size, sizeof(sndbuf_size)) == -1)
1057 warn("Unable to set sndbuf size %d", sndbuf_size);
1059 if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1060 (void *)&rcvbuf_size, sizeof(rcvbuf_size)) == -1)
1061 warn("Unable to set rcvbuf size %d", rcvbuf_size);
1065 * Copy characters from src into dst, \ quoting characters that require it
1068 ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
1073 src[si] != '\0' && di < dstlen && si < srclen;
1094 * Copy src into buf (which is len bytes long), expanding % sequences.
1097 formatbuf(char *buf, size_t len, const char *src)
1099 const char *p, *p2, *q;
1100 int i, op, updirs, pdirs;
1102 #define ADDBUF(x) do { \
1109 for (i = 0; *p; p++) {
1121 p2 = connected ? remotecwd : "";
1124 /* option to determine fixed # of dirs from path */
1125 if (op == '.' || op == 'c') {
1129 while (*p2) /* calc # of /'s */
1132 if (p[1] == '0') { /* print <x> or ... */
1136 if (p[1] >= '1' && p[1] <= '9') {
1137 /* calc # to skip */
1144 while (skip-- > 0) {
1145 while ((p2 > q) && (*p2 != '/'))
1150 if (*p2 == '/' && p2 != q)
1154 if (updirs > 0 && pdirs) {
1168 ADDBUF('0' + updirs);
1178 for (p2 = connected && hostname ? hostname : "-";
1180 if (op == 'm' && *p2 == '.')
1187 for (p2 = connected ? username : "-"; *p2 ; p2++)
1195 default: /* display unknown codes literally */
1207 * Determine if given string is an IPv6 address or not.
1208 * Return 1 for yes, 0 for no
1211 isipv6addr(const char *addr)
1215 struct addrinfo hints, *res;
1217 memset(&hints, 0, sizeof(hints));
1218 hints.ai_family = PF_INET6;
1219 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
1220 hints.ai_flags = AI_NUMERICHOST;
1221 if (getaddrinfo(addr, "0", &hints, &res) != 0)
1227 DPRINTF("isipv6addr: got %d for %s\n", rv, addr);
1229 return (rv == 1) ? 1 : 0;
1233 * Read a line from the FILE stream into buf/buflen using fgets(), so up
1234 * to buflen-1 chars will be read and the result will be NUL terminated.
1235 * If the line has a trailing newline it will be removed.
1236 * If the line is too long, excess characters will be read until
1237 * newline/EOF/error.
1238 * If EOF/error occurs or a too-long line is encountered and errormsg
1239 * isn't NULL, it will be changed to a description of the problem.
1240 * (The EOF message has a leading \n for cosmetic purposes).
1242 * >=0 length of line (excluding trailing newline) if all ok
1244 * -2 EOF encountered
1245 * -3 line was too long
1248 getline(FILE *stream, char *buf, size_t buflen, const char **errormsg)
1253 if (fgets(buf, buflen, stream) == NULL) {
1254 if (feof(stream)) { /* EOF */
1257 *errormsg = "\nEOF received";
1258 } else { /* error */
1261 *errormsg = "Error encountered";
1267 if (buf[len-1] == '\n') { /* clear any trailing newline */
1269 } else if (len == buflen-1) { /* line too long */
1270 while ((ch = getchar()) != '\n' && ch != EOF)
1273 *errormsg = "Input line is too long";
1283 * Internal version of connect(2); sets socket buffer sizes,
1284 * binds to a specific local address (if set), and
1285 * supports a connection timeout using a non-blocking connect(2) with
1287 * Socket fcntl flags are temporarily updated to include O_NONBLOCK;
1288 * these will not be reverted on connection failure.
1289 * Returns 0 on success, or -1 upon failure (with an appropriate
1290 * error message displayed.)
1293 ftp_connect(int sock, const struct sockaddr *name, socklen_t namelen)
1295 int flags, rv, timeout, error;
1297 struct timeval endtime, now, td;
1298 struct pollfd pfd[1];
1299 char hname[NI_MAXHOST];
1300 char sname[NI_MAXSERV];
1302 setupsockbufsize(sock);
1303 if (getnameinfo(name, namelen,
1304 hname, sizeof(hname), sname, sizeof(sname),
1305 NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
1306 strlcpy(hname, "?", sizeof(hname));
1307 strlcpy(sname, "?", sizeof(sname));
1310 if (bindai != NULL) { /* bind to specific addr */
1311 struct addrinfo *ai;
1313 for (ai = bindai; ai != NULL; ai = ai->ai_next) {
1314 if (ai->ai_family == name->sa_family)
1319 if (bind(sock, ai->ai_addr, ai->ai_addrlen) == -1) {
1320 char bname[NI_MAXHOST];
1324 if (getnameinfo(ai->ai_addr, ai->ai_addrlen,
1325 bname, sizeof(bname), NULL, 0, NI_NUMERICHOST) != 0)
1326 strlcpy(bname, "?", sizeof(bname));
1328 warn("Can't bind to `%s'", bname);
1333 /* save current socket flags */
1334 if ((flags = fcntl(sock, F_GETFL, 0)) == -1) {
1335 warn("Can't %s socket flags for connect to `%s:%s'",
1336 "save", hname, sname);
1339 /* set non-blocking connect */
1340 if (fcntl(sock, F_SETFL, flags | O_NONBLOCK) == -1) {
1341 warn("Can't set socket non-blocking for connect to `%s:%s'",
1346 /* NOTE: we now must restore socket flags on successful exit */
1349 pfd[0].events = POLLIN|POLLOUT;
1351 if (quit_time > 0) { /* want a non default timeout */
1352 (void)gettimeofday(&endtime, NULL);
1353 endtime.tv_sec += quit_time; /* determine end time */
1356 rv = connect(sock, name, namelen); /* inititate the connection */
1357 if (rv == -1) { /* connection error */
1358 if (errno != EINPROGRESS) { /* error isn't "please wait" */
1360 warn("Can't connect to `%s:%s'", hname, sname);
1364 /* connect EINPROGRESS; wait */
1366 if (quit_time > 0) { /* determine timeout */
1367 (void)gettimeofday(&now, NULL);
1368 timersub(&endtime, &now, &td);
1369 timeout = td.tv_sec * 1000 + td.tv_usec/1000;
1376 rv = ftp_poll(pfd, 1, timeout);
1377 /* loop until poll ! EINTR */
1378 } while (rv == -1 && errno == EINTR);
1380 if (rv == 0) { /* poll (connect) timed out */
1385 if (rv == -1) { /* poll error */
1387 } else if (pfd[0].revents & (POLLIN|POLLOUT)) {
1388 slen = sizeof(error); /* OK, or pending error */
1389 if (getsockopt(sock, SOL_SOCKET, SO_ERROR,
1390 &error, &slen) == -1) {
1391 /* Solaris pending error */
1393 } else if (error != 0) {
1394 errno = error; /* BSD pending error */
1398 errno = EBADF; /* this shouldn't happen ... */
1403 if (fcntl(sock, F_SETFL, flags) == -1) {
1404 /* restore socket flags */
1405 warn("Can't %s socket flags for connect to `%s:%s'",
1406 "restore", hname, sname);
1413 * Internal version of listen(2); sets socket buffer sizes first.
1416 ftp_listen(int sock, int backlog)
1419 setupsockbufsize(sock);
1420 return (listen(sock, backlog));
1424 * Internal version of poll(2), to allow reimplementation by select(2)
1425 * on platforms without the former.
1428 ftp_poll(struct pollfd *fds, int nfds, int timeout)
1430 return poll(fds, nfds, timeout);
1434 * malloc() with inbuilt error checking
1437 ftp_malloc(size_t size)
1443 err(1, "Unable to allocate %ld bytes of memory", (long)size);
1448 * sl_init() with inbuilt error checking
1457 err(1, "Unable to allocate memory for stringlist");
1462 * sl_add() with inbuilt error checking
1465 ftp_sl_add(StringList *sl, char *i)
1472 * strdup() with inbuilt error checking
1475 ftp_strdup(const char *str)
1480 errx(1, "ftp_strdup: called with NULL argument");
1483 err(1, "Unable to allocate memory for string copy");