Catch up with the latest LWKT msgport updating.
[dragonfly/vkernel-mp.git] / contrib / tnftp / util.c
blob6162fbc5f390230c4588a0bebd24f6610a9dc846
1 /* $NetBSD: util.c,v 1.138 2007/04/17 05:52:04 lukem Exp $ */
3 /*-
4 * Copyright (c) 1997-2007 The NetBSD Foundation, Inc.
5 * All rights reserved.
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Luke Mewburn.
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
16 * are met:
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.
22 * 3. All advertising materials mentioning features or use of this software
23 * must display the following acknowledgement:
24 * This product includes software developed by the NetBSD
25 * Foundation, Inc. and its contributors.
26 * 4. Neither the name of The NetBSD Foundation nor the names of its
27 * contributors may be used to endorse or promote products derived
28 * from this software without specific prior written permission.
30 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
31 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
32 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
33 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
34 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
35 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
36 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
37 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
38 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
39 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
40 * POSSIBILITY OF SUCH DAMAGE.
44 * Copyright (c) 1985, 1989, 1993, 1994
45 * The Regents of the University of California. All rights reserved.
47 * Redistribution and use in source and binary forms, with or without
48 * modification, are permitted provided that the following conditions
49 * are met:
50 * 1. Redistributions of source code must retain the above copyright
51 * notice, this list of conditions and the following disclaimer.
52 * 2. Redistributions in binary form must reproduce the above copyright
53 * notice, this list of conditions and the following disclaimer in the
54 * documentation and/or other materials provided with the distribution.
55 * 3. Neither the name of the University nor the names of its contributors
56 * may be used to endorse or promote products derived from this software
57 * without specific prior written permission.
59 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
60 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
61 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
62 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
63 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
64 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
65 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
66 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
67 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
68 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
69 * SUCH DAMAGE.
72 #include <sys/cdefs.h>
73 #ifndef lint
74 __RCSID("$NetBSD: util.c,v 1.138 2007/04/17 05:52:04 lukem Exp $");
75 #endif /* not lint */
78 * FTP User Program -- Misc support routines
80 #include <sys/param.h>
81 #include <sys/socket.h>
82 #include <sys/ioctl.h>
83 #include <sys/time.h>
84 #include <netinet/in.h>
85 #include <arpa/ftp.h>
87 #include <ctype.h>
88 #include <err.h>
89 #include <errno.h>
90 #include <fcntl.h>
91 #include <glob.h>
92 #include <signal.h>
93 #include <libgen.h>
94 #include <limits.h>
95 #include <netdb.h>
96 #include <stdio.h>
97 #include <stdlib.h>
98 #include <string.h>
99 #include <termios.h>
100 #include <time.h>
101 #include <tzfile.h>
102 #include <unistd.h>
104 #include "ftp_var.h"
107 * Connect to peer server and auto-login, if possible.
109 void
110 setpeer(int argc, char *argv[])
112 char *host;
113 char *port;
115 if (argc == 0)
116 goto usage;
117 if (connected) {
118 fprintf(ttyout, "Already connected to %s, use close first.\n",
119 hostname);
120 code = -1;
121 return;
123 if (argc < 2)
124 (void)another(&argc, &argv, "to");
125 if (argc < 2 || argc > 3) {
126 usage:
127 UPRINTF("usage: %s host-name [port]\n", argv[0]);
128 code = -1;
129 return;
131 if (gatemode)
132 port = gateport;
133 else
134 port = ftpport;
135 if (argc > 2)
136 port = argv[2];
138 if (gatemode) {
139 if (gateserver == NULL || *gateserver == '\0')
140 errx(1, "main: gateserver not defined");
141 host = hookup(gateserver, port);
142 } else
143 host = hookup(argv[1], port);
145 if (host) {
146 if (gatemode && verbose) {
147 fprintf(ttyout,
148 "Connecting via pass-through server %s\n",
149 gateserver);
152 connected = 1;
154 * Set up defaults for FTP.
156 (void)strlcpy(typename, "ascii", sizeof(typename));
157 type = TYPE_A;
158 curtype = TYPE_A;
159 (void)strlcpy(formname, "non-print", sizeof(formname));
160 form = FORM_N;
161 (void)strlcpy(modename, "stream", sizeof(modename));
162 mode = MODE_S;
163 (void)strlcpy(structname, "file", sizeof(structname));
164 stru = STRU_F;
165 (void)strlcpy(bytename, "8", sizeof(bytename));
166 bytesize = 8;
167 if (autologin)
168 (void)ftp_login(argv[1], NULL, NULL);
172 static void
173 parse_feat(const char *line)
177 * work-around broken ProFTPd servers that can't
178 * even obey RFC 2389.
180 while (*line && isspace((int)*line))
181 line++;
183 if (strcasecmp(line, "MDTM") == 0)
184 features[FEAT_MDTM] = 1;
185 else if (strncasecmp(line, "MLST", sizeof("MLST") - 1) == 0) {
186 features[FEAT_MLST] = 1;
187 } else if (strcasecmp(line, "REST STREAM") == 0)
188 features[FEAT_REST_STREAM] = 1;
189 else if (strcasecmp(line, "SIZE") == 0)
190 features[FEAT_SIZE] = 1;
191 else if (strcasecmp(line, "TVFS") == 0)
192 features[FEAT_TVFS] = 1;
196 * Determine the remote system type (SYST) and features (FEAT).
197 * Call after a successful login (i.e, connected = -1)
199 void
200 getremoteinfo(void)
202 int overbose, i;
204 overbose = verbose;
205 if (ftp_debug == 0)
206 verbose = -1;
208 /* determine remote system type */
209 if (command("SYST") == COMPLETE) {
210 if (overbose) {
211 char *cp, c;
213 c = 0;
214 cp = strchr(reply_string + 4, ' ');
215 if (cp == NULL)
216 cp = strchr(reply_string + 4, '\r');
217 if (cp) {
218 if (cp[-1] == '.')
219 cp--;
220 c = *cp;
221 *cp = '\0';
224 fprintf(ttyout, "Remote system type is %s.\n",
225 reply_string + 4);
226 if (cp)
227 *cp = c;
229 if (!strncmp(reply_string, "215 UNIX Type: L8", 17)) {
230 if (proxy)
231 unix_proxy = 1;
232 else
233 unix_server = 1;
235 * Set type to 0 (not specified by user),
236 * meaning binary by default, but don't bother
237 * telling server. We can use binary
238 * for text files unless changed by the user.
240 type = 0;
241 (void)strlcpy(typename, "binary", sizeof(typename));
242 if (overbose)
243 fprintf(ttyout,
244 "Using %s mode to transfer files.\n",
245 typename);
246 } else {
247 if (proxy)
248 unix_proxy = 0;
249 else
250 unix_server = 0;
251 if (overbose &&
252 !strncmp(reply_string, "215 TOPS20", 10))
253 fputs(
254 "Remember to set tenex mode when transferring binary files from this machine.\n",
255 ttyout);
259 /* determine features (if any) */
260 for (i = 0; i < FEAT_max; i++)
261 features[i] = -1;
262 reply_callback = parse_feat;
263 if (command("FEAT") == COMPLETE) {
264 for (i = 0; i < FEAT_max; i++) {
265 if (features[i] == -1)
266 features[i] = 0;
268 features[FEAT_FEAT] = 1;
269 } else
270 features[FEAT_FEAT] = 0;
271 #ifndef NO_DEBUG
272 if (ftp_debug) {
273 #define DEBUG_FEAT(x) fprintf(ttyout, "features[" #x "] = %d\n", features[(x)])
274 DEBUG_FEAT(FEAT_FEAT);
275 DEBUG_FEAT(FEAT_MDTM);
276 DEBUG_FEAT(FEAT_MLST);
277 DEBUG_FEAT(FEAT_REST_STREAM);
278 DEBUG_FEAT(FEAT_SIZE);
279 DEBUG_FEAT(FEAT_TVFS);
280 #undef DEBUG_FEAT
282 #endif
283 reply_callback = NULL;
285 verbose = overbose;
289 * Reset the various variables that indicate connection state back to
290 * disconnected settings.
291 * The caller is responsible for issuing any commands to the remote server
292 * to perform a clean shutdown before this is invoked.
294 void
295 cleanuppeer(void)
298 if (cout)
299 (void)fclose(cout);
300 cout = NULL;
301 connected = 0;
302 unix_server = 0;
303 unix_proxy = 0;
305 * determine if anonftp was specifically set with -a
306 * (1), or implicitly set by auto_fetch() (2). in the
307 * latter case, disable after the current xfer
309 if (anonftp == 2)
310 anonftp = 0;
311 data = -1;
312 epsv4bad = 0;
313 if (username)
314 free(username);
315 username = NULL;
316 if (!proxy)
317 macnum = 0;
321 * Top-level signal handler for interrupted commands.
323 void
324 intr(int signo)
327 sigint_raised = 1;
328 alarmtimer(0);
329 if (fromatty)
330 write(fileno(ttyout), "\n", 1);
331 siglongjmp(toplevel, 1);
335 * Signal handler for lost connections; cleanup various elements of
336 * the connection state, and call cleanuppeer() to finish it off.
338 void
339 lostpeer(int dummy)
341 int oerrno = errno;
343 alarmtimer(0);
344 if (connected) {
345 if (cout != NULL) {
346 (void)shutdown(fileno(cout), 1+1);
347 (void)fclose(cout);
348 cout = NULL;
350 if (data >= 0) {
351 (void)shutdown(data, 1+1);
352 (void)close(data);
353 data = -1;
355 connected = 0;
357 pswitch(1);
358 if (connected) {
359 if (cout != NULL) {
360 (void)shutdown(fileno(cout), 1+1);
361 (void)fclose(cout);
362 cout = NULL;
364 connected = 0;
366 proxflag = 0;
367 pswitch(0);
368 cleanuppeer();
369 errno = oerrno;
374 * Login to remote host, using given username & password if supplied.
375 * Return non-zero if successful.
378 ftp_login(const char *host, const char *luser, const char *lpass)
380 char tmp[80];
381 char *user, *pass, *acct, *p;
382 char emptypass[] = "";
383 const char *errormsg;
384 int n, aflag, rval, nlen;
386 aflag = rval = 0;
387 user = pass = acct = NULL;
388 if (luser)
389 user = ftp_strdup(luser);
390 if (lpass)
391 pass = ftp_strdup(lpass);
393 DPRINTF("ftp_login: user `%s' pass `%s' host `%s'\n",
394 user ? user : "<null>", pass ? pass : "<null>",
395 host ? host : "<null>");
398 * Set up arguments for an anonymous FTP session, if necessary.
400 if (anonftp) {
401 FREEPTR(user);
402 user = ftp_strdup("anonymous"); /* as per RFC 1635 */
403 FREEPTR(pass);
404 pass = ftp_strdup(getoptionvalue("anonpass"));
407 if (ruserpass(host, &user, &pass, &acct) < 0) {
408 code = -1;
409 goto cleanup_ftp_login;
412 while (user == NULL) {
413 if (localname)
414 fprintf(ttyout, "Name (%s:%s): ", host, localname);
415 else
416 fprintf(ttyout, "Name (%s): ", host);
417 errormsg = NULL;
418 nlen = getline(stdin, tmp, sizeof(tmp), &errormsg);
419 if (nlen < 0) {
420 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "login");
421 code = -1;
422 goto cleanup_ftp_login;
423 } else if (nlen == 0) {
424 user = ftp_strdup(localname);
425 } else {
426 user = ftp_strdup(tmp);
430 if (gatemode) {
431 char *nuser;
432 size_t len;
434 len = strlen(user) + 1 + strlen(host) + 1;
435 nuser = ftp_malloc(len);
436 (void)strlcpy(nuser, user, len);
437 (void)strlcat(nuser, "@", len);
438 (void)strlcat(nuser, host, len);
439 FREEPTR(user);
440 user = nuser;
443 n = command("USER %s", user);
444 if (n == CONTINUE) {
445 if (pass == NULL) {
446 p = getpass("Password: ");
447 if (p == NULL)
448 p = emptypass;
449 pass = ftp_strdup(p);
450 memset(p, 0, strlen(p));
452 n = command("PASS %s", pass);
453 memset(pass, 0, strlen(pass));
455 if (n == CONTINUE) {
456 aflag++;
457 if (acct == NULL) {
458 p = getpass("Account: ");
459 if (p == NULL)
460 p = emptypass;
461 acct = ftp_strdup(p);
462 memset(p, 0, strlen(p));
464 if (acct[0] == '\0') {
465 warnx("Login failed");
466 goto cleanup_ftp_login;
468 n = command("ACCT %s", acct);
469 memset(acct, 0, strlen(acct));
471 if ((n != COMPLETE) ||
472 (!aflag && acct != NULL && command("ACCT %s", acct) != COMPLETE)) {
473 warnx("Login failed");
474 goto cleanup_ftp_login;
476 rval = 1;
477 username = ftp_strdup(user);
478 if (proxy)
479 goto cleanup_ftp_login;
481 connected = -1;
482 getremoteinfo();
483 for (n = 0; n < macnum; ++n) {
484 if (!strcmp("init", macros[n].mac_name)) {
485 (void)strlcpy(line, "$init", sizeof(line));
486 makeargv();
487 domacro(margc, margv);
488 break;
491 updatelocalcwd();
492 updateremotecwd();
494 cleanup_ftp_login:
495 FREEPTR(user);
496 if (pass != NULL)
497 memset(pass, 0, strlen(pass));
498 FREEPTR(pass);
499 if (acct != NULL)
500 memset(acct, 0, strlen(acct));
501 FREEPTR(acct);
502 return (rval);
506 * `another' gets another argument, and stores the new argc and argv.
507 * It reverts to the top level (via intr()) on EOF/error.
509 * Returns false if no new arguments have been added.
512 another(int *pargc, char ***pargv, const char *prompt)
514 const char *errormsg;
515 int ret, nlen;
516 size_t len;
518 len = strlen(line);
519 if (len >= sizeof(line) - 3) {
520 fputs("Sorry, arguments too long.\n", ttyout);
521 intr(0);
523 fprintf(ttyout, "(%s) ", prompt);
524 line[len++] = ' ';
525 errormsg = NULL;
526 nlen = getline(stdin, line + len, sizeof(line)-len, &errormsg);
527 if (nlen < 0) {
528 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "operation");
529 intr(0);
531 len += nlen;
532 makeargv();
533 ret = margc > *pargc;
534 *pargc = margc;
535 *pargv = margv;
536 return (ret);
540 * glob files given in argv[] from the remote server.
541 * if errbuf isn't NULL, store error messages there instead
542 * of writing to the screen.
544 char *
545 remglob(char *argv[], int doswitch, const char **errbuf)
547 static char buf[MAXPATHLEN];
548 static FILE *ftemp = NULL;
549 static char **args;
550 char temp[MAXPATHLEN];
551 int oldverbose, oldhash, oldprogress, fd;
552 char *cp;
553 const char *mode;
554 size_t len;
556 if (!mflag || !connected) {
557 if (!doglob)
558 args = NULL;
559 else {
560 if (ftemp) {
561 (void)fclose(ftemp);
562 ftemp = NULL;
565 return (NULL);
567 if (!doglob) {
568 if (args == NULL)
569 args = argv;
570 if ((cp = *++args) == NULL)
571 args = NULL;
572 return (cp);
574 if (ftemp == NULL) {
575 len = strlcpy(temp, tmpdir, sizeof(temp));
576 if (temp[len - 1] != '/')
577 (void)strlcat(temp, "/", sizeof(temp));
578 (void)strlcat(temp, TMPFILE, sizeof(temp));
579 if ((fd = mkstemp(temp)) < 0) {
580 warn("Unable to create temporary file `%s'", temp);
581 return (NULL);
583 close(fd);
584 oldverbose = verbose;
585 verbose = (errbuf != NULL) ? -1 : 0;
586 oldhash = hash;
587 oldprogress = progress;
588 hash = 0;
589 progress = 0;
590 if (doswitch)
591 pswitch(!proxy);
592 for (mode = "w"; *++argv != NULL; mode = "a")
593 recvrequest("NLST", temp, *argv, mode, 0, 0);
594 if ((code / 100) != COMPLETE) {
595 if (errbuf != NULL)
596 *errbuf = reply_string;
598 if (doswitch)
599 pswitch(!proxy);
600 verbose = oldverbose;
601 hash = oldhash;
602 progress = oldprogress;
603 ftemp = fopen(temp, "r");
604 (void)unlink(temp);
605 if (ftemp == NULL) {
606 if (errbuf == NULL)
607 warnx("Can't find list of remote files");
608 else
609 *errbuf =
610 "Can't find list of remote files";
611 return (NULL);
614 if (fgets(buf, sizeof(buf), ftemp) == NULL) {
615 (void)fclose(ftemp);
616 ftemp = NULL;
617 return (NULL);
619 if ((cp = strchr(buf, '\n')) != NULL)
620 *cp = '\0';
621 return (buf);
625 * Glob a local file name specification with the expectation of a single
626 * return value. Can't control multiple values being expanded from the
627 * expression, we return only the first.
628 * Returns NULL on error, or a pointer to a buffer containing the filename
629 * that's the caller's responsiblity to free(3) when finished with.
631 char *
632 globulize(const char *pattern)
634 glob_t gl;
635 int flags;
636 char *p;
638 if (!doglob)
639 return (ftp_strdup(pattern));
641 flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE;
642 memset(&gl, 0, sizeof(gl));
643 if (glob(pattern, flags, NULL, &gl) || gl.gl_pathc == 0) {
644 warnx("Glob pattern `%s' not found", pattern);
645 globfree(&gl);
646 return (NULL);
648 p = ftp_strdup(gl.gl_pathv[0]);
649 globfree(&gl);
650 return (p);
654 * determine size of remote file
656 off_t
657 remotesize(const char *file, int noisy)
659 int overbose, r;
660 off_t size;
662 overbose = verbose;
663 size = -1;
664 if (ftp_debug == 0)
665 verbose = -1;
666 if (! features[FEAT_SIZE]) {
667 if (noisy)
668 fprintf(ttyout,
669 "SIZE is not supported by remote server.\n");
670 goto cleanup_remotesize;
672 r = command("SIZE %s", file);
673 if (r == COMPLETE) {
674 char *cp, *ep;
676 cp = strchr(reply_string, ' ');
677 if (cp != NULL) {
678 cp++;
679 size = STRTOLL(cp, &ep, 10);
680 if (*ep != '\0' && !isspace((unsigned char)*ep))
681 size = -1;
683 } else {
684 if (r == ERROR && code == 500 && features[FEAT_SIZE] == -1)
685 features[FEAT_SIZE] = 0;
686 if (noisy && ftp_debug == 0) {
687 fputs(reply_string, ttyout);
688 putc('\n', ttyout);
691 cleanup_remotesize:
692 verbose = overbose;
693 return (size);
697 * determine last modification time (in GMT) of remote file
699 time_t
700 remotemodtime(const char *file, int noisy)
702 int overbose, ocode, r;
703 time_t rtime;
705 overbose = verbose;
706 ocode = code;
707 rtime = -1;
708 if (ftp_debug == 0)
709 verbose = -1;
710 if (! features[FEAT_MDTM]) {
711 if (noisy)
712 fprintf(ttyout,
713 "MDTM is not supported by remote server.\n");
714 goto cleanup_parse_time;
716 r = command("MDTM %s", file);
717 if (r == COMPLETE) {
718 struct tm timebuf;
719 char *timestr, *frac;
720 int yy, mo, day, hour, min, sec;
723 * time-val = 14DIGIT [ "." 1*DIGIT ]
724 * YYYYMMDDHHMMSS[.sss]
725 * mdtm-response = "213" SP time-val CRLF / error-response
727 timestr = reply_string + 4;
730 * parse fraction.
731 * XXX: ignored for now
733 frac = strchr(timestr, '\r');
734 if (frac != NULL)
735 *frac = '\0';
736 frac = strchr(timestr, '.');
737 if (frac != NULL)
738 *frac++ = '\0';
739 if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
741 * XXX: Workaround for lame ftpd's that return
742 * `19100' instead of `2000'
744 fprintf(ttyout,
745 "Y2K warning! Incorrect time-val `%s' received from server.\n",
746 timestr);
747 timestr++;
748 timestr[0] = '2';
749 timestr[1] = '0';
750 fprintf(ttyout, "Converted to `%s'\n", timestr);
752 if (strlen(timestr) != 14 ||
753 sscanf(timestr, "%04d%02d%02d%02d%02d%02d",
754 &yy, &mo, &day, &hour, &min, &sec) != 6) {
755 bad_parse_time:
756 fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
757 goto cleanup_parse_time;
759 memset(&timebuf, 0, sizeof(timebuf));
760 timebuf.tm_sec = sec;
761 timebuf.tm_min = min;
762 timebuf.tm_hour = hour;
763 timebuf.tm_mday = day;
764 timebuf.tm_mon = mo - 1;
765 timebuf.tm_year = yy - TM_YEAR_BASE;
766 timebuf.tm_isdst = -1;
767 rtime = timegm(&timebuf);
768 if (rtime == -1) {
769 if (noisy || ftp_debug != 0)
770 goto bad_parse_time;
771 else
772 goto cleanup_parse_time;
773 } else
774 DPRINTF("parsed date as: %s", ctime(&rtime));
775 } else {
776 if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
777 features[FEAT_MDTM] = 0;
778 if (noisy && ftp_debug == 0) {
779 fputs(reply_string, ttyout);
780 putc('\n', ttyout);
783 cleanup_parse_time:
784 verbose = overbose;
785 if (rtime == -1)
786 code = ocode;
787 return (rtime);
791 * Update global `localcwd', which contains the state of the local cwd
793 void
794 updatelocalcwd(void)
797 if (getcwd(localcwd, sizeof(localcwd)) == NULL)
798 localcwd[0] = '\0';
799 DPRINTF("got localcwd as `%s'\n", localcwd);
803 * Update global `remotecwd', which contains the state of the remote cwd
805 void
806 updateremotecwd(void)
808 int overbose, ocode, i;
809 char *cp;
811 overbose = verbose;
812 ocode = code;
813 if (ftp_debug == 0)
814 verbose = -1;
815 if (command("PWD") != COMPLETE)
816 goto badremotecwd;
817 cp = strchr(reply_string, ' ');
818 if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
819 goto badremotecwd;
820 cp += 2;
821 for (i = 0; *cp && i < sizeof(remotecwd) - 1; i++, cp++) {
822 if (cp[0] == '"') {
823 if (cp[1] == '"')
824 cp++;
825 else
826 break;
828 remotecwd[i] = *cp;
830 remotecwd[i] = '\0';
831 DPRINTF("got remotecwd as `%s'\n", remotecwd);
832 goto cleanupremotecwd;
833 badremotecwd:
834 remotecwd[0]='\0';
835 cleanupremotecwd:
836 verbose = overbose;
837 code = ocode;
841 * Ensure file is in or under dir.
842 * Returns 1 if so, 0 if not (or an error occurred).
845 fileindir(const char *file, const char *dir)
847 char parentdirbuf[PATH_MAX+1], *parentdir;
848 char realdir[PATH_MAX+1];
849 size_t dirlen;
851 /* determine parent directory of file */
852 (void)strlcpy(parentdirbuf, file, sizeof(parentdirbuf));
853 parentdir = dirname(parentdirbuf);
854 if (strcmp(parentdir, ".") == 0)
855 return 1; /* current directory is ok */
857 /* find the directory */
858 if (realpath(parentdir, realdir) == NULL) {
859 warn("Unable to determine real path of `%s'", parentdir);
860 return 0;
862 if (realdir[0] != '/') /* relative result is ok */
863 return 1;
864 dirlen = strlen(dir);
865 #if 0
866 printf("file %s parent %s realdir %s dir %s [%d]\n",
867 file, parentdir, realdir, dir, dirlen);
868 #endif
869 if (strncmp(realdir, dir, dirlen) == 0 &&
870 (realdir[dirlen] == '/' || realdir[dirlen] == '\0'))
871 return 1;
872 return 0;
876 * List words in stringlist, vertically arranged
878 void
879 list_vertical(StringList *sl)
881 int i, j;
882 int columns, lines;
883 char *p;
884 size_t w, width;
886 width = 0;
888 for (i = 0 ; i < sl->sl_cur ; i++) {
889 w = strlen(sl->sl_str[i]);
890 if (w > width)
891 width = w;
893 width = (width + 8) &~ 7;
895 columns = ttywidth / width;
896 if (columns == 0)
897 columns = 1;
898 lines = (sl->sl_cur + columns - 1) / columns;
899 for (i = 0; i < lines; i++) {
900 for (j = 0; j < columns; j++) {
901 p = sl->sl_str[j * lines + i];
902 if (p)
903 fputs(p, ttyout);
904 if (j * lines + i + lines >= sl->sl_cur) {
905 putc('\n', ttyout);
906 break;
908 if (p) {
909 w = strlen(p);
910 while (w < width) {
911 w = (w + 8) &~ 7;
912 (void)putc('\t', ttyout);
920 * Update the global ttywidth value, using TIOCGWINSZ.
922 void
923 setttywidth(int a)
925 struct winsize winsize;
926 int oerrno = errno;
928 if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
929 winsize.ws_col != 0)
930 ttywidth = winsize.ws_col;
931 else
932 ttywidth = 80;
933 errno = oerrno;
937 * Change the rate limit up (SIGUSR1) or down (SIGUSR2)
939 void
940 crankrate(int sig)
943 switch (sig) {
944 case SIGUSR1:
945 if (rate_get)
946 rate_get += rate_get_incr;
947 if (rate_put)
948 rate_put += rate_put_incr;
949 break;
950 case SIGUSR2:
951 if (rate_get && rate_get > rate_get_incr)
952 rate_get -= rate_get_incr;
953 if (rate_put && rate_put > rate_put_incr)
954 rate_put -= rate_put_incr;
955 break;
956 default:
957 err(1, "crankrate invoked with unknown signal: %d", sig);
963 * Setup or cleanup EditLine structures
965 #ifndef NO_EDITCOMPLETE
966 void
967 controlediting(void)
969 if (editing && el == NULL && hist == NULL) {
970 HistEvent ev;
971 int editmode;
973 el = el_init(getprogname(), stdin, ttyout, stderr);
974 /* init editline */
975 hist = history_init(); /* init the builtin history */
976 history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
977 el_set(el, EL_HIST, history, hist); /* use history */
979 el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
980 el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
981 el_set(el, EL_RPROMPT, rprompt);
983 /* add local file completion, bind to TAB */
984 el_set(el, EL_ADDFN, "ftp-complete",
985 "Context sensitive argument completion",
986 complete);
987 el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
988 el_source(el, NULL); /* read ~/.editrc */
989 if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
990 editing = 0; /* the user doesn't want editing,
991 * so disable, and let statement
992 * below cleanup */
993 else
994 el_set(el, EL_SIGNAL, 1);
996 if (!editing) {
997 if (hist) {
998 history_end(hist);
999 hist = NULL;
1001 if (el) {
1002 el_end(el);
1003 el = NULL;
1007 #endif /* !NO_EDITCOMPLETE */
1010 * Convert the string `arg' to an int, which may have an optional SI suffix
1011 * (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
1014 strsuftoi(const char *arg)
1016 char *cp;
1017 long val;
1019 if (!isdigit((unsigned char)arg[0]))
1020 return (-1);
1022 val = strtol(arg, &cp, 10);
1023 if (cp != NULL) {
1024 if (cp[0] != '\0' && cp[1] != '\0')
1025 return (-1);
1026 switch (tolower((unsigned char)cp[0])) {
1027 case '\0':
1028 case 'b':
1029 break;
1030 case 'k':
1031 val <<= 10;
1032 break;
1033 case 'm':
1034 val <<= 20;
1035 break;
1036 case 'g':
1037 val <<= 30;
1038 break;
1039 default:
1040 return (-1);
1043 if (val < 0 || val > INT_MAX)
1044 return (-1);
1046 return (val);
1050 * Set up socket buffer sizes before a connection is made.
1052 void
1053 setupsockbufsize(int sock)
1056 if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1057 (void *)&sndbuf_size, sizeof(sndbuf_size)) == -1)
1058 warn("Unable to set sndbuf size %d", sndbuf_size);
1060 if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1061 (void *)&rcvbuf_size, sizeof(rcvbuf_size)) == -1)
1062 warn("Unable to set rcvbuf size %d", rcvbuf_size);
1066 * Copy characters from src into dst, \ quoting characters that require it
1068 void
1069 ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
1071 int di, si;
1073 for (di = si = 0;
1074 src[si] != '\0' && di < dstlen && si < srclen;
1075 di++, si++) {
1076 switch (src[si]) {
1077 case '\\':
1078 case ' ':
1079 case '\t':
1080 case '\r':
1081 case '\n':
1082 case '"':
1083 dst[di++] = '\\';
1084 if (di >= dstlen)
1085 break;
1086 /* FALLTHROUGH */
1087 default:
1088 dst[di] = src[si];
1091 dst[di] = '\0';
1095 * Copy src into buf (which is len bytes long), expanding % sequences.
1097 void
1098 formatbuf(char *buf, size_t len, const char *src)
1100 const char *p, *p2, *q;
1101 int i, op, updirs, pdirs;
1103 #define ADDBUF(x) do { \
1104 if (i >= len - 1) \
1105 goto endbuf; \
1106 buf[i++] = (x); \
1107 } while (0)
1109 p = src;
1110 for (i = 0; *p; p++) {
1111 if (*p != '%') {
1112 ADDBUF(*p);
1113 continue;
1115 p++;
1117 switch (op = *p) {
1119 case '/':
1120 case '.':
1121 case 'c':
1122 p2 = connected ? remotecwd : "";
1123 updirs = pdirs = 0;
1125 /* option to determine fixed # of dirs from path */
1126 if (op == '.' || op == 'c') {
1127 int skip;
1129 q = p2;
1130 while (*p2) /* calc # of /'s */
1131 if (*p2++ == '/')
1132 updirs++;
1133 if (p[1] == '0') { /* print <x> or ... */
1134 pdirs = 1;
1135 p++;
1137 if (p[1] >= '1' && p[1] <= '9') {
1138 /* calc # to skip */
1139 skip = p[1] - '0';
1140 p++;
1141 } else
1142 skip = 1;
1144 updirs -= skip;
1145 while (skip-- > 0) {
1146 while ((p2 > q) && (*p2 != '/'))
1147 p2--; /* back up */
1148 if (skip && p2 > q)
1149 p2--;
1151 if (*p2 == '/' && p2 != q)
1152 p2++;
1155 if (updirs > 0 && pdirs) {
1156 if (i >= len - 5)
1157 break;
1158 if (op == '.') {
1159 ADDBUF('.');
1160 ADDBUF('.');
1161 ADDBUF('.');
1162 } else {
1163 ADDBUF('/');
1164 ADDBUF('<');
1165 if (updirs > 9) {
1166 ADDBUF('9');
1167 ADDBUF('+');
1168 } else
1169 ADDBUF('0' + updirs);
1170 ADDBUF('>');
1173 for (; *p2; p2++)
1174 ADDBUF(*p2);
1175 break;
1177 case 'M':
1178 case 'm':
1179 for (p2 = connected && hostname ? hostname : "-";
1180 *p2 ; p2++) {
1181 if (op == 'm' && *p2 == '.')
1182 break;
1183 ADDBUF(*p2);
1185 break;
1187 case 'n':
1188 for (p2 = connected ? username : "-"; *p2 ; p2++)
1189 ADDBUF(*p2);
1190 break;
1192 case '%':
1193 ADDBUF('%');
1194 break;
1196 default: /* display unknown codes literally */
1197 ADDBUF('%');
1198 ADDBUF(op);
1199 break;
1203 endbuf:
1204 buf[i] = '\0';
1208 * Parse `port' into a TCP port number, defaulting to `defport' if `port' is
1209 * an unknown service name. If defport != -1, print a warning upon bad parse.
1212 parseport(const char *port, int defport)
1214 int rv;
1215 long nport;
1216 char *p, *ep;
1218 p = ftp_strdup(port);
1219 nport = strtol(p, &ep, 10);
1220 if (*ep != '\0' && ep == p) {
1221 struct servent *svp;
1223 svp = getservbyname(port, "tcp");
1224 if (svp == NULL) {
1225 badparseport:
1226 if (defport != -1)
1227 warnx("Unknown port `%s', using port %d",
1228 port, defport);
1229 rv = defport;
1230 } else
1231 rv = ntohs(svp->s_port);
1232 } else if (nport < 1 || nport > MAX_IN_PORT_T || *ep != '\0')
1233 goto badparseport;
1234 else
1235 rv = nport;
1236 free(p);
1237 return (rv);
1241 * Determine if given string is an IPv6 address or not.
1242 * Return 1 for yes, 0 for no
1245 isipv6addr(const char *addr)
1247 int rv = 0;
1248 #ifdef INET6
1249 struct addrinfo hints, *res;
1251 memset(&hints, 0, sizeof(hints));
1252 hints.ai_family = PF_INET6;
1253 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
1254 hints.ai_flags = AI_NUMERICHOST;
1255 if (getaddrinfo(addr, "0", &hints, &res) != 0)
1256 rv = 0;
1257 else {
1258 rv = 1;
1259 freeaddrinfo(res);
1261 DPRINTF("isipv6addr: got %d for %s\n", rv, addr);
1262 #endif
1263 return (rv == 1) ? 1 : 0;
1267 * Read a line from the FILE stream into buf/buflen using fgets(), so up
1268 * to buflen-1 chars will be read and the result will be NUL terminated.
1269 * If the line has a trailing newline it will be removed.
1270 * If the line is too long, excess characters will be read until
1271 * newline/EOF/error.
1272 * If EOF/error occurs or a too-long line is encountered and errormsg
1273 * isn't NULL, it will be changed to a description of the problem.
1274 * (The EOF message has a leading \n for cosmetic purposes).
1275 * Returns:
1276 * >=0 length of line (excluding trailing newline) if all ok
1277 * -1 error occurred
1278 * -2 EOF encountered
1279 * -3 line was too long
1282 getline(FILE *stream, char *buf, size_t buflen, const char **errormsg)
1284 int rv, ch;
1285 size_t len;
1287 if (fgets(buf, buflen, stream) == NULL) {
1288 if (feof(stream)) { /* EOF */
1289 rv = -2;
1290 if (errormsg)
1291 *errormsg = "\nEOF received";
1292 } else { /* error */
1293 rv = -1;
1294 if (errormsg)
1295 *errormsg = "Error encountered";
1297 clearerr(stream);
1298 return rv;
1300 len = strlen(buf);
1301 if (buf[len-1] == '\n') { /* clear any trailing newline */
1302 buf[--len] = '\0';
1303 } else if (len == buflen-1) { /* line too long */
1304 while ((ch = getchar()) != '\n' && ch != EOF)
1305 continue;
1306 if (errormsg)
1307 *errormsg = "Input line is too long";
1308 clearerr(stream);
1309 return -3;
1311 if (errormsg)
1312 *errormsg = NULL;
1313 return len;
1317 * Internal version of connect(2); sets socket buffer sizes,
1318 * binds to a specific local address (if set), and
1319 * supports a connection timeout using a non-blocking connect(2) with
1320 * a poll(2).
1321 * Socket fcntl flags are temporarily updated to include O_NONBLOCK;
1322 * these will not be reverted on connection failure.
1323 * Returns 0 on success, or -1 upon failure (with an appropriate
1324 * error message displayed.)
1327 ftp_connect(int sock, const struct sockaddr *name, socklen_t namelen)
1329 int flags, rv, timeout, error;
1330 socklen_t slen;
1331 struct timeval endtime, now, td;
1332 struct pollfd pfd[1];
1333 char hname[NI_MAXHOST];
1335 setupsockbufsize(sock);
1336 if (getnameinfo(name, namelen,
1337 hname, sizeof(hname), NULL, 0, NI_NUMERICHOST) != 0)
1338 strlcpy(hname, "?", sizeof(hname));
1340 if (bindai != NULL) { /* bind to specific addr */
1341 struct addrinfo *ai;
1343 for (ai = bindai; ai != NULL; ai = ai->ai_next) {
1344 if (ai->ai_family == name->sa_family)
1345 break;
1347 if (ai == NULL)
1348 ai = bindai;
1349 if (bind(sock, ai->ai_addr, ai->ai_addrlen) == -1) {
1350 char bname[NI_MAXHOST];
1351 int saveerr;
1353 saveerr = errno;
1354 if (getnameinfo(ai->ai_addr, ai->ai_addrlen,
1355 bname, sizeof(bname), NULL, 0, NI_NUMERICHOST) != 0)
1356 strlcpy(bname, "?", sizeof(bname));
1357 errno = saveerr;
1358 warn("Can't bind to `%s'", bname);
1359 return -1;
1363 /* save current socket flags */
1364 if ((flags = fcntl(sock, F_GETFL, 0)) == -1) {
1365 warn("Can't %s socket flags for connect to `%s'",
1366 "save", hname);
1367 return -1;
1369 /* set non-blocking connect */
1370 if (fcntl(sock, F_SETFL, flags | O_NONBLOCK) == -1) {
1371 warn("Can't set socket non-blocking for connect to `%s'",
1372 hname);
1373 return -1;
1376 /* NOTE: we now must restore socket flags on successful exit */
1378 pfd[0].fd = sock;
1379 pfd[0].events = POLLIN|POLLOUT;
1381 if (quit_time > 0) { /* want a non default timeout */
1382 (void)gettimeofday(&endtime, NULL);
1383 endtime.tv_sec += quit_time; /* determine end time */
1386 rv = connect(sock, name, namelen); /* inititate the connection */
1387 if (rv == -1) { /* connection error */
1388 if (errno != EINPROGRESS) { /* error isn't "please wait" */
1389 connecterror:
1390 warn("Can't connect to `%s'", hname);
1391 return -1;
1394 /* connect EINPROGRESS; wait */
1395 do {
1396 if (quit_time > 0) { /* determine timeout */
1397 (void)gettimeofday(&now, NULL);
1398 timersub(&endtime, &now, &td);
1399 timeout = td.tv_sec * 1000 + td.tv_usec/1000;
1400 if (timeout < 0)
1401 timeout = 0;
1402 } else {
1403 timeout = INFTIM;
1405 pfd[0].revents = 0;
1406 rv = ftp_poll(pfd, 1, timeout);
1407 /* loop until poll ! EINTR */
1408 } while (rv == -1 && errno == EINTR);
1410 if (rv == 0) { /* poll (connect) timed out */
1411 errno = ETIMEDOUT;
1412 goto connecterror;
1415 if (rv == -1) { /* poll error */
1416 goto connecterror;
1417 } else if (pfd[0].revents & (POLLIN|POLLOUT)) {
1418 slen = sizeof(error); /* OK, or pending error */
1419 if (getsockopt(sock, SOL_SOCKET, SO_ERROR,
1420 &error, &slen) == -1) {
1421 /* Solaris pending error */
1422 goto connecterror;
1423 } else if (error != 0) {
1424 errno = error; /* BSD pending error */
1425 goto connecterror;
1427 } else {
1428 errno = EBADF; /* this shouldn't happen ... */
1429 goto connecterror;
1433 if (fcntl(sock, F_SETFL, flags) == -1) {
1434 /* restore socket flags */
1435 warn("Can't %s socket flags for connect to `%s'",
1436 "restore", hname);
1437 return -1;
1439 return 0;
1443 * Internal version of listen(2); sets socket buffer sizes first.
1446 ftp_listen(int sock, int backlog)
1449 setupsockbufsize(sock);
1450 return (listen(sock, backlog));
1454 * Internal version of poll(2), to allow reimplementation by select(2)
1455 * on platforms without the former.
1458 ftp_poll(struct pollfd *fds, int nfds, int timeout)
1460 return poll(fds, nfds, timeout);
1464 * malloc() with inbuilt error checking
1466 void *
1467 ftp_malloc(size_t size)
1469 void *p;
1471 p = malloc(size);
1472 if (p == NULL)
1473 err(1, "Unable to allocate %ld bytes of memory", (long)size);
1474 return (p);
1478 * sl_init() with inbuilt error checking
1480 StringList *
1481 ftp_sl_init(void)
1483 StringList *p;
1485 p = sl_init();
1486 if (p == NULL)
1487 err(1, "Unable to allocate memory for stringlist");
1488 return (p);
1492 * sl_add() with inbuilt error checking
1494 void
1495 ftp_sl_add(StringList *sl, char *i)
1498 if (sl_add(sl, i) == -1)
1499 err(1, "Unable to add `%s' to stringlist", i);
1503 * strdup() with inbuilt error checking
1505 char *
1506 ftp_strdup(const char *str)
1508 char *s;
1510 if (str == NULL)
1511 errx(1, "ftp_strdup: called with NULL argument");
1512 s = strdup(str);
1513 if (s == NULL)
1514 err(1, "Unable to allocate memory for string copy");
1515 return (s);