util.c revision 1.114 1 /* $NetBSD: util.c,v 1.114 2003/08/07 11:13:57 agc Exp $ */
2
3 /*-
4 * Copyright (c) 1997-2003 The NetBSD Foundation, Inc.
5 * All rights reserved.
6 *
7 * This code is derived from software contributed to The NetBSD Foundation
8 * by Luke Mewburn.
9 *
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.
13 *
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.
29 *
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.
41 */
42
43 /*
44 * Copyright (c) 1985, 1989, 1993, 1994
45 * The Regents of the University of California. All rights reserved.
46 *
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.
58 *
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.
70 */
71
72 #include <sys/cdefs.h>
73 #ifndef lint
74 __RCSID("$NetBSD: util.c,v 1.114 2003/08/07 11:13:57 agc Exp $");
75 #endif /* not lint */
76
77 /*
78 * FTP User Program -- Misc support routines
79 */
80 #include <sys/types.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>
86
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 <limits.h>
94 #include <netdb.h>
95 #include <stdio.h>
96 #include <stdlib.h>
97 #include <string.h>
98 #include <termios.h>
99 #include <time.h>
100 #include <tzfile.h>
101 #include <unistd.h>
102
103 #include "ftp_var.h"
104
105 /*
106 * Connect to peer server and auto-login, if possible.
107 */
108 void
109 setpeer(int argc, char *argv[])
110 {
111 char *host;
112 char *port;
113
114 if (argc == 0)
115 goto usage;
116 if (connected) {
117 fprintf(ttyout, "Already connected to %s, use close first.\n",
118 hostname);
119 code = -1;
120 return;
121 }
122 if (argc < 2)
123 (void)another(&argc, &argv, "to");
124 if (argc < 2 || argc > 3) {
125 usage:
126 fprintf(ttyout, "usage: %s host-name [port]\n", argv[0]);
127 code = -1;
128 return;
129 }
130 if (gatemode)
131 port = gateport;
132 else
133 port = ftpport;
134 if (argc > 2)
135 port = argv[2];
136
137 if (gatemode) {
138 if (gateserver == NULL || *gateserver == '\0')
139 errx(1, "gateserver not defined (shouldn't happen)");
140 host = hookup(gateserver, port);
141 } else
142 host = hookup(argv[1], port);
143
144 if (host) {
145 if (gatemode && verbose) {
146 fprintf(ttyout,
147 "Connecting via pass-through server %s\n",
148 gateserver);
149 }
150
151 connected = 1;
152 /*
153 * Set up defaults for FTP.
154 */
155 (void)strlcpy(typename, "ascii", sizeof(typename));
156 type = TYPE_A;
157 curtype = TYPE_A;
158 (void)strlcpy(formname, "non-print", sizeof(formname));
159 form = FORM_N;
160 (void)strlcpy(modename, "stream", sizeof(modename));
161 mode = MODE_S;
162 (void)strlcpy(structname, "file", sizeof(structname));
163 stru = STRU_F;
164 (void)strlcpy(bytename, "8", sizeof(bytename));
165 bytesize = 8;
166 if (autologin)
167 (void)ftp_login(argv[1], NULL, NULL);
168 }
169 }
170
171 static void
172 parse_feat(const char *line)
173 {
174
175 /*
176 * work-around broken ProFTPd servers that can't
177 * even obey RFC 2389.
178 */
179 while (*line && isspace((int)*line))
180 line++;
181
182 if (strcasecmp(line, "MDTM") == 0)
183 features[FEAT_MDTM] = 1;
184 else if (strncasecmp(line, "MLST", sizeof("MLST") - 1) == 0) {
185 features[FEAT_MLST] = 1;
186 } else if (strcasecmp(line, "REST STREAM") == 0)
187 features[FEAT_REST_STREAM] = 1;
188 else if (strcasecmp(line, "SIZE") == 0)
189 features[FEAT_SIZE] = 1;
190 else if (strcasecmp(line, "TVFS") == 0)
191 features[FEAT_TVFS] = 1;
192 }
193
194 /*
195 * Determine the remote system type (SYST) and features (FEAT).
196 * Call after a successful login (i.e, connected = -1)
197 */
198 void
199 getremoteinfo(void)
200 {
201 int overbose, i;
202
203 overbose = verbose;
204 if (debug == 0)
205 verbose = -1;
206
207 /* determine remote system type */
208 if (command("SYST") == COMPLETE) {
209 if (overbose) {
210 char *cp, c;
211
212 c = 0;
213 cp = strchr(reply_string + 4, ' ');
214 if (cp == NULL)
215 cp = strchr(reply_string + 4, '\r');
216 if (cp) {
217 if (cp[-1] == '.')
218 cp--;
219 c = *cp;
220 *cp = '\0';
221 }
222
223 fprintf(ttyout, "Remote system type is %s.\n",
224 reply_string + 4);
225 if (cp)
226 *cp = c;
227 }
228 if (!strncmp(reply_string, "215 UNIX Type: L8", 17)) {
229 if (proxy)
230 unix_proxy = 1;
231 else
232 unix_server = 1;
233 /*
234 * Set type to 0 (not specified by user),
235 * meaning binary by default, but don't bother
236 * telling server. We can use binary
237 * for text files unless changed by the user.
238 */
239 type = 0;
240 (void)strlcpy(typename, "binary", sizeof(typename));
241 if (overbose)
242 fprintf(ttyout,
243 "Using %s mode to transfer files.\n",
244 typename);
245 } else {
246 if (proxy)
247 unix_proxy = 0;
248 else
249 unix_server = 0;
250 if (overbose &&
251 !strncmp(reply_string, "215 TOPS20", 10))
252 fputs(
253 "Remember to set tenex mode when transferring binary files from this machine.\n",
254 ttyout);
255 }
256 }
257
258 /* determine features (if any) */
259 for (i = 0; i < FEAT_max; i++)
260 features[i] = -1;
261 reply_callback = parse_feat;
262 if (command("FEAT") == COMPLETE) {
263 for (i = 0; i < FEAT_max; i++) {
264 if (features[i] == -1)
265 features[i] = 0;
266 }
267 features[FEAT_FEAT] = 1;
268 } else
269 features[FEAT_FEAT] = 0;
270 if (debug) {
271 #define DEBUG_FEAT(x) fprintf(ttyout, "features[" #x "] = %d\n", features[(x)])
272 DEBUG_FEAT(FEAT_FEAT);
273 DEBUG_FEAT(FEAT_MDTM);
274 DEBUG_FEAT(FEAT_MLST);
275 DEBUG_FEAT(FEAT_REST_STREAM);
276 DEBUG_FEAT(FEAT_SIZE);
277 DEBUG_FEAT(FEAT_TVFS);
278 #undef DEBUG_FEAT
279 }
280 reply_callback = NULL;
281
282 verbose = overbose;
283 }
284
285 /*
286 * Reset the various variables that indicate connection state back to
287 * disconnected settings.
288 * The caller is responsible for issuing any commands to the remote server
289 * to perform a clean shutdown before this is invoked.
290 */
291 void
292 cleanuppeer(void)
293 {
294
295 if (cout)
296 (void)fclose(cout);
297 cout = NULL;
298 connected = 0;
299 unix_server = 0;
300 unix_proxy = 0;
301 /*
302 * determine if anonftp was specifically set with -a
303 * (1), or implicitly set by auto_fetch() (2). in the
304 * latter case, disable after the current xfer
305 */
306 if (anonftp == 2)
307 anonftp = 0;
308 data = -1;
309 epsv4bad = 0;
310 if (username)
311 free(username);
312 username = NULL;
313 if (!proxy)
314 macnum = 0;
315 }
316
317 /*
318 * Top-level signal handler for interrupted commands.
319 */
320 void
321 intr(int dummy)
322 {
323
324 alarmtimer(0);
325 if (fromatty)
326 write(fileno(ttyout), "\n", 1);
327 siglongjmp(toplevel, 1);
328 }
329
330 /*
331 * Signal handler for lost connections; cleanup various elements of
332 * the connection state, and call cleanuppeer() to finish it off.
333 */
334 void
335 lostpeer(int dummy)
336 {
337 int oerrno = errno;
338
339 alarmtimer(0);
340 if (connected) {
341 if (cout != NULL) {
342 (void)shutdown(fileno(cout), 1+1);
343 (void)fclose(cout);
344 cout = NULL;
345 }
346 if (data >= 0) {
347 (void)shutdown(data, 1+1);
348 (void)close(data);
349 data = -1;
350 }
351 connected = 0;
352 }
353 pswitch(1);
354 if (connected) {
355 if (cout != NULL) {
356 (void)shutdown(fileno(cout), 1+1);
357 (void)fclose(cout);
358 cout = NULL;
359 }
360 connected = 0;
361 }
362 proxflag = 0;
363 pswitch(0);
364 cleanuppeer();
365 errno = oerrno;
366 }
367
368
369 /*
370 * Login to remote host, using given username & password if supplied.
371 * Return non-zero if successful.
372 */
373 int
374 ftp_login(const char *host, const char *user, const char *pass)
375 {
376 char tmp[80];
377 const char *acct;
378 int n, aflag, rval, freeuser, freepass, freeacct;
379
380 acct = NULL;
381 aflag = rval = freeuser = freepass = freeacct = 0;
382
383 if (debug)
384 fprintf(ttyout, "ftp_login: user `%s' pass `%s' host `%s'\n",
385 user ? user : "<null>", pass ? pass : "<null>",
386 host ? host : "<null>");
387
388
389 /*
390 * Set up arguments for an anonymous FTP session, if necessary.
391 */
392 if (anonftp) {
393 user = "anonymous"; /* as per RFC 1635 */
394 pass = getoptionvalue("anonpass");
395 }
396
397 if (user == NULL)
398 freeuser = 1;
399 if (pass == NULL)
400 freepass = 1;
401 freeacct = 1;
402 if (ruserpass(host, &user, &pass, &acct) < 0) {
403 code = -1;
404 goto cleanup_ftp_login;
405 }
406
407 while (user == NULL) {
408 if (localname)
409 fprintf(ttyout, "Name (%s:%s): ", host, localname);
410 else
411 fprintf(ttyout, "Name (%s): ", host);
412 *tmp = '\0';
413 if (fgets(tmp, sizeof(tmp) - 1, stdin) == NULL) {
414 fprintf(ttyout, "\nEOF received; login aborted.\n");
415 clearerr(stdin);
416 code = -1;
417 goto cleanup_ftp_login;
418 }
419 tmp[strlen(tmp) - 1] = '\0';
420 freeuser = 0;
421 if (*tmp == '\0')
422 user = localname;
423 else
424 user = tmp;
425 }
426
427 if (gatemode) {
428 char *nuser;
429 int len;
430
431 len = strlen(user) + 1 + strlen(host) + 1;
432 nuser = xmalloc(len);
433 (void)strlcpy(nuser, user, len);
434 (void)strlcat(nuser, "@", len);
435 (void)strlcat(nuser, host, len);
436 freeuser = 1;
437 user = nuser;
438 }
439
440 n = command("USER %s", user);
441 if (n == CONTINUE) {
442 if (pass == NULL) {
443 freepass = 0;
444 pass = getpass("Password:");
445 }
446 n = command("PASS %s", pass);
447 }
448 if (n == CONTINUE) {
449 aflag++;
450 if (acct == NULL) {
451 freeacct = 0;
452 acct = getpass("Account:");
453 }
454 if (acct[0] == '\0') {
455 warnx("Login failed.");
456 goto cleanup_ftp_login;
457 }
458 n = command("ACCT %s", acct);
459 }
460 if ((n != COMPLETE) ||
461 (!aflag && acct != NULL && command("ACCT %s", acct) != COMPLETE)) {
462 warnx("Login failed.");
463 goto cleanup_ftp_login;
464 }
465 rval = 1;
466 username = xstrdup(user);
467 if (proxy)
468 goto cleanup_ftp_login;
469
470 connected = -1;
471 getremoteinfo();
472 for (n = 0; n < macnum; ++n) {
473 if (!strcmp("init", macros[n].mac_name)) {
474 (void)strlcpy(line, "$init", sizeof(line));
475 makeargv();
476 domacro(margc, margv);
477 break;
478 }
479 }
480 updateremotepwd();
481
482 cleanup_ftp_login:
483 if (user != NULL && freeuser)
484 free((char *)user);
485 if (pass != NULL && freepass)
486 free((char *)pass);
487 if (acct != NULL && freeacct)
488 free((char *)acct);
489 return (rval);
490 }
491
492 /*
493 * `another' gets another argument, and stores the new argc and argv.
494 * It reverts to the top level (via intr()) on EOF/error.
495 *
496 * Returns false if no new arguments have been added.
497 */
498 int
499 another(int *pargc, char ***pargv, const char *prompt)
500 {
501 int len = strlen(line), ret;
502
503 if (len >= sizeof(line) - 3) {
504 fputs("sorry, arguments too long.\n", ttyout);
505 intr(0);
506 }
507 fprintf(ttyout, "(%s) ", prompt);
508 line[len++] = ' ';
509 if (fgets(&line[len], sizeof(line) - len, stdin) == NULL) {
510 clearerr(stdin);
511 intr(0);
512 }
513 len += strlen(&line[len]);
514 if (len > 0 && line[len - 1] == '\n')
515 line[len - 1] = '\0';
516 makeargv();
517 ret = margc > *pargc;
518 *pargc = margc;
519 *pargv = margv;
520 return (ret);
521 }
522
523 /*
524 * glob files given in argv[] from the remote server.
525 * if errbuf isn't NULL, store error messages there instead
526 * of writing to the screen.
527 */
528 char *
529 remglob(char *argv[], int doswitch, char **errbuf)
530 {
531 char temp[MAXPATHLEN];
532 static char buf[MAXPATHLEN];
533 static FILE *ftemp = NULL;
534 static char **args;
535 int oldverbose, oldhash, oldprogress, fd, len;
536 char *cp, *mode;
537
538 if (!mflag || !connected) {
539 if (!doglob)
540 args = NULL;
541 else {
542 if (ftemp) {
543 (void)fclose(ftemp);
544 ftemp = NULL;
545 }
546 }
547 return (NULL);
548 }
549 if (!doglob) {
550 if (args == NULL)
551 args = argv;
552 if ((cp = *++args) == NULL)
553 args = NULL;
554 return (cp);
555 }
556 if (ftemp == NULL) {
557 len = strlcpy(temp, tmpdir, sizeof(temp));
558 if (temp[len - 1] != '/')
559 (void)strlcat(temp, "/", sizeof(temp));
560 (void)strlcat(temp, TMPFILE, sizeof(temp));
561 if ((fd = mkstemp(temp)) < 0) {
562 warn("unable to create temporary file %s", temp);
563 return (NULL);
564 }
565 close(fd);
566 oldverbose = verbose;
567 verbose = (errbuf != NULL) ? -1 : 0;
568 oldhash = hash;
569 oldprogress = progress;
570 hash = 0;
571 progress = 0;
572 if (doswitch)
573 pswitch(!proxy);
574 for (mode = "w"; *++argv != NULL; mode = "a")
575 recvrequest("NLST", temp, *argv, mode, 0, 0);
576 if ((code / 100) != COMPLETE) {
577 if (errbuf != NULL)
578 *errbuf = reply_string;
579 }
580 if (doswitch)
581 pswitch(!proxy);
582 verbose = oldverbose;
583 hash = oldhash;
584 progress = oldprogress;
585 ftemp = fopen(temp, "r");
586 (void)unlink(temp);
587 if (ftemp == NULL) {
588 if (errbuf == NULL)
589 fputs(
590 "can't find list of remote files, oops.\n",
591 ttyout);
592 else
593 *errbuf =
594 "can't find list of remote files, oops.";
595 return (NULL);
596 }
597 }
598 if (fgets(buf, sizeof(buf), ftemp) == NULL) {
599 (void)fclose(ftemp);
600 ftemp = NULL;
601 return (NULL);
602 }
603 if ((cp = strchr(buf, '\n')) != NULL)
604 *cp = '\0';
605 return (buf);
606 }
607
608 /*
609 * Glob a local file name specification with the expectation of a single
610 * return value. Can't control multiple values being expanded from the
611 * expression, we return only the first.
612 * Returns NULL on error, or a pointer to a buffer containing the filename
613 * that's the caller's responsiblity to free(3) when finished with.
614 */
615 char *
616 globulize(const char *pattern)
617 {
618 glob_t gl;
619 int flags;
620 char *p;
621
622 if (!doglob)
623 return (xstrdup(pattern));
624
625 flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE;
626 memset(&gl, 0, sizeof(gl));
627 if (glob(pattern, flags, NULL, &gl) || gl.gl_pathc == 0) {
628 warnx("%s: not found", pattern);
629 globfree(&gl);
630 return (NULL);
631 }
632 p = xstrdup(gl.gl_pathv[0]);
633 globfree(&gl);
634 return (p);
635 }
636
637 /*
638 * determine size of remote file
639 */
640 off_t
641 remotesize(const char *file, int noisy)
642 {
643 int overbose, r;
644 off_t size;
645
646 overbose = verbose;
647 size = -1;
648 if (debug == 0)
649 verbose = -1;
650 if (! features[FEAT_SIZE]) {
651 if (noisy)
652 fprintf(ttyout,
653 "SIZE is not supported by remote server.\n");
654 goto cleanup_remotesize;
655 }
656 r = command("SIZE %s", file);
657 if (r == COMPLETE) {
658 char *cp, *ep;
659
660 cp = strchr(reply_string, ' ');
661 if (cp != NULL) {
662 cp++;
663 size = STRTOLL(cp, &ep, 10);
664 if (*ep != '\0' && !isspace((unsigned char)*ep))
665 size = -1;
666 }
667 } else {
668 if (r == ERROR && code == 500 && features[FEAT_SIZE] == -1)
669 features[FEAT_SIZE] = 0;
670 if (noisy && debug == 0) {
671 fputs(reply_string, ttyout);
672 putc('\n', ttyout);
673 }
674 }
675 cleanup_remotesize:
676 verbose = overbose;
677 return (size);
678 }
679
680 /*
681 * determine last modification time (in GMT) of remote file
682 */
683 time_t
684 remotemodtime(const char *file, int noisy)
685 {
686 int overbose, ocode, r;
687 time_t rtime;
688
689 overbose = verbose;
690 ocode = code;
691 rtime = -1;
692 if (debug == 0)
693 verbose = -1;
694 if (! features[FEAT_MDTM]) {
695 if (noisy)
696 fprintf(ttyout,
697 "MDTM is not supported by remote server.\n");
698 goto cleanup_parse_time;
699 }
700 r = command("MDTM %s", file);
701 if (r == COMPLETE) {
702 struct tm timebuf;
703 char *timestr, *frac;
704 int yy, mo, day, hour, min, sec;
705
706 /*
707 * time-val = 14DIGIT [ "." 1*DIGIT ]
708 * YYYYMMDDHHMMSS[.sss]
709 * mdtm-response = "213" SP time-val CRLF / error-response
710 */
711 timestr = reply_string + 4;
712
713 /*
714 * parse fraction.
715 * XXX: ignored for now
716 */
717 frac = strchr(timestr, '\r');
718 if (frac != NULL)
719 *frac = '\0';
720 frac = strchr(timestr, '.');
721 if (frac != NULL)
722 *frac++ = '\0';
723 if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
724 /*
725 * XXX: Workaround for lame ftpd's that return
726 * `19100' instead of `2000'
727 */
728 fprintf(ttyout,
729 "Y2K warning! Incorrect time-val `%s' received from server.\n",
730 timestr);
731 timestr++;
732 timestr[0] = '2';
733 timestr[1] = '0';
734 fprintf(ttyout, "Converted to `%s'\n", timestr);
735 }
736 if (strlen(timestr) != 14 ||
737 sscanf(timestr, "%04d%02d%02d%02d%02d%02d",
738 &yy, &mo, &day, &hour, &min, &sec) != 6) {
739 bad_parse_time:
740 fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
741 goto cleanup_parse_time;
742 }
743 memset(&timebuf, 0, sizeof(timebuf));
744 timebuf.tm_sec = sec;
745 timebuf.tm_min = min;
746 timebuf.tm_hour = hour;
747 timebuf.tm_mday = day;
748 timebuf.tm_mon = mo - 1;
749 timebuf.tm_year = yy - TM_YEAR_BASE;
750 timebuf.tm_isdst = -1;
751 rtime = timegm(&timebuf);
752 if (rtime == -1) {
753 if (noisy || debug != 0)
754 goto bad_parse_time;
755 else
756 goto cleanup_parse_time;
757 } else if (debug)
758 fprintf(ttyout, "parsed date as: %s", ctime(&rtime));
759 } else {
760 if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
761 features[FEAT_MDTM] = 0;
762 if (noisy && debug == 0) {
763 fputs(reply_string, ttyout);
764 putc('\n', ttyout);
765 }
766 }
767 cleanup_parse_time:
768 verbose = overbose;
769 if (rtime == -1)
770 code = ocode;
771 return (rtime);
772 }
773
774 /*
775 * update global `remotepwd', which contains the state of the remote cwd
776 */
777 void
778 updateremotepwd(void)
779 {
780 int overbose, ocode, i;
781 char *cp;
782
783 overbose = verbose;
784 ocode = code;
785 if (debug == 0)
786 verbose = -1;
787 if (command("PWD") != COMPLETE)
788 goto badremotepwd;
789 cp = strchr(reply_string, ' ');
790 if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
791 goto badremotepwd;
792 cp += 2;
793 for (i = 0; *cp && i < sizeof(remotepwd) - 1; i++, cp++) {
794 if (cp[0] == '"') {
795 if (cp[1] == '"')
796 cp++;
797 else
798 break;
799 }
800 remotepwd[i] = *cp;
801 }
802 remotepwd[i] = '\0';
803 if (debug)
804 fprintf(ttyout, "got remotepwd as `%s'\n", remotepwd);
805 goto cleanupremotepwd;
806 badremotepwd:
807 remotepwd[0]='\0';
808 cleanupremotepwd:
809 verbose = overbose;
810 code = ocode;
811 }
812
813
814 /*
815 * List words in stringlist, vertically arranged
816 */
817 void
818 list_vertical(StringList *sl)
819 {
820 int i, j, w;
821 int columns, width, lines;
822 char *p;
823
824 width = 0;
825
826 for (i = 0 ; i < sl->sl_cur ; i++) {
827 w = strlen(sl->sl_str[i]);
828 if (w > width)
829 width = w;
830 }
831 width = (width + 8) &~ 7;
832
833 columns = ttywidth / width;
834 if (columns == 0)
835 columns = 1;
836 lines = (sl->sl_cur + columns - 1) / columns;
837 for (i = 0; i < lines; i++) {
838 for (j = 0; j < columns; j++) {
839 p = sl->sl_str[j * lines + i];
840 if (p)
841 fputs(p, ttyout);
842 if (j * lines + i + lines >= sl->sl_cur) {
843 putc('\n', ttyout);
844 break;
845 }
846 w = strlen(p);
847 while (w < width) {
848 w = (w + 8) &~ 7;
849 (void)putc('\t', ttyout);
850 }
851 }
852 }
853 }
854
855 /*
856 * Update the global ttywidth value, using TIOCGWINSZ.
857 */
858 void
859 setttywidth(int a)
860 {
861 struct winsize winsize;
862 int oerrno = errno;
863
864 if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
865 winsize.ws_col != 0)
866 ttywidth = winsize.ws_col;
867 else
868 ttywidth = 80;
869 errno = oerrno;
870 }
871
872 /*
873 * Change the rate limit up (SIGUSR1) or down (SIGUSR2)
874 */
875 void
876 crankrate(int sig)
877 {
878
879 switch (sig) {
880 case SIGUSR1:
881 if (rate_get)
882 rate_get += rate_get_incr;
883 if (rate_put)
884 rate_put += rate_put_incr;
885 break;
886 case SIGUSR2:
887 if (rate_get && rate_get > rate_get_incr)
888 rate_get -= rate_get_incr;
889 if (rate_put && rate_put > rate_put_incr)
890 rate_put -= rate_put_incr;
891 break;
892 default:
893 err(1, "crankrate invoked with unknown signal: %d", sig);
894 }
895 }
896
897
898 /*
899 * Setup or cleanup EditLine structures
900 */
901 #ifndef NO_EDITCOMPLETE
902 void
903 controlediting(void)
904 {
905 if (editing && el == NULL && hist == NULL) {
906 HistEvent ev;
907 int editmode;
908
909 el = el_init(getprogname(), stdin, ttyout, stderr);
910 /* init editline */
911 hist = history_init(); /* init the builtin history */
912 history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
913 el_set(el, EL_HIST, history, hist); /* use history */
914
915 el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
916 el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
917 el_set(el, EL_RPROMPT, rprompt);
918
919 /* add local file completion, bind to TAB */
920 el_set(el, EL_ADDFN, "ftp-complete",
921 "Context sensitive argument completion",
922 complete);
923 el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
924 el_source(el, NULL); /* read ~/.editrc */
925 if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
926 editing = 0; /* the user doesn't want editing,
927 * so disable, and let statement
928 * below cleanup */
929 else
930 el_set(el, EL_SIGNAL, 1);
931 }
932 if (!editing) {
933 if (hist) {
934 history_end(hist);
935 hist = NULL;
936 }
937 if (el) {
938 el_end(el);
939 el = NULL;
940 }
941 }
942 }
943 #endif /* !NO_EDITCOMPLETE */
944
945 /*
946 * Convert the string `arg' to an int, which may have an optional SI suffix
947 * (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
948 */
949 int
950 strsuftoi(const char *arg)
951 {
952 char *cp;
953 long val;
954
955 if (!isdigit((unsigned char)arg[0]))
956 return (-1);
957
958 val = strtol(arg, &cp, 10);
959 if (cp != NULL) {
960 if (cp[0] != '\0' && cp[1] != '\0')
961 return (-1);
962 switch (tolower((unsigned char)cp[0])) {
963 case '\0':
964 case 'b':
965 break;
966 case 'k':
967 val <<= 10;
968 break;
969 case 'm':
970 val <<= 20;
971 break;
972 case 'g':
973 val <<= 30;
974 break;
975 default:
976 return (-1);
977 }
978 }
979 if (val < 0 || val > INT_MAX)
980 return (-1);
981
982 return (val);
983 }
984
985 /*
986 * Set up socket buffer sizes before a connection is made.
987 */
988 void
989 setupsockbufsize(int sock)
990 {
991
992 if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, (void *) &sndbuf_size,
993 sizeof(rcvbuf_size)) < 0)
994 warn("unable to set sndbuf size %d", sndbuf_size);
995
996 if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF, (void *) &rcvbuf_size,
997 sizeof(rcvbuf_size)) < 0)
998 warn("unable to set rcvbuf size %d", rcvbuf_size);
999 }
1000
1001 /*
1002 * Copy characters from src into dst, \ quoting characters that require it
1003 */
1004 void
1005 ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
1006 {
1007 int di, si;
1008
1009 for (di = si = 0;
1010 src[si] != '\0' && di < dstlen && si < srclen;
1011 di++, si++) {
1012 switch (src[si]) {
1013 case '\\':
1014 case ' ':
1015 case '\t':
1016 case '\r':
1017 case '\n':
1018 case '"':
1019 dst[di++] = '\\';
1020 if (di >= dstlen)
1021 break;
1022 /* FALLTHROUGH */
1023 default:
1024 dst[di] = src[si];
1025 }
1026 }
1027 dst[di] = '\0';
1028 }
1029
1030 /*
1031 * Copy src into buf (which is len bytes long), expanding % sequences.
1032 */
1033 void
1034 formatbuf(char *buf, size_t len, const char *src)
1035 {
1036 const char *p;
1037 char *p2, *q;
1038 int i, op, updirs, pdirs;
1039
1040 #define ADDBUF(x) do { \
1041 if (i >= len - 1) \
1042 goto endbuf; \
1043 buf[i++] = (x); \
1044 } while (0)
1045
1046 p = src;
1047 for (i = 0; *p; p++) {
1048 if (*p != '%') {
1049 ADDBUF(*p);
1050 continue;
1051 }
1052 p++;
1053
1054 switch (op = *p) {
1055
1056 case '/':
1057 case '.':
1058 case 'c':
1059 p2 = connected ? remotepwd : "";
1060 updirs = pdirs = 0;
1061
1062 /* option to determine fixed # of dirs from path */
1063 if (op == '.' || op == 'c') {
1064 int skip;
1065
1066 q = p2;
1067 while (*p2) /* calc # of /'s */
1068 if (*p2++ == '/')
1069 updirs++;
1070 if (p[1] == '0') { /* print <x> or ... */
1071 pdirs = 1;
1072 p++;
1073 }
1074 if (p[1] >= '1' && p[1] <= '9') {
1075 /* calc # to skip */
1076 skip = p[1] - '0';
1077 p++;
1078 } else
1079 skip = 1;
1080
1081 updirs -= skip;
1082 while (skip-- > 0) {
1083 while ((p2 > q) && (*p2 != '/'))
1084 p2--; /* back up */
1085 if (skip && p2 > q)
1086 p2--;
1087 }
1088 if (*p2 == '/' && p2 != q)
1089 p2++;
1090 }
1091
1092 if (updirs > 0 && pdirs) {
1093 if (i >= len - 5)
1094 break;
1095 if (op == '.') {
1096 ADDBUF('.');
1097 ADDBUF('.');
1098 ADDBUF('.');
1099 } else {
1100 ADDBUF('/');
1101 ADDBUF('<');
1102 if (updirs > 9) {
1103 ADDBUF('9');
1104 ADDBUF('+');
1105 } else
1106 ADDBUF('0' + updirs);
1107 ADDBUF('>');
1108 }
1109 }
1110 for (; *p2; p2++)
1111 ADDBUF(*p2);
1112 break;
1113
1114 case 'M':
1115 case 'm':
1116 for (p2 = connected && username ? username : "-";
1117 *p2 ; p2++) {
1118 if (op == 'm' && *p2 == '.')
1119 break;
1120 ADDBUF(*p2);
1121 }
1122 break;
1123
1124 case 'n':
1125 for (p2 = connected ? username : "-"; *p2 ; p2++)
1126 ADDBUF(*p2);
1127 break;
1128
1129 case '%':
1130 ADDBUF('%');
1131 break;
1132
1133 default: /* display unknown codes literally */
1134 ADDBUF('%');
1135 ADDBUF(op);
1136 break;
1137
1138 }
1139 }
1140 endbuf:
1141 buf[i] = '\0';
1142 }
1143
1144 /*
1145 * Parse `port' into a TCP port number, defaulting to `defport' if `port' is
1146 * an unknown service name. If defport != -1, print a warning upon bad parse.
1147 */
1148 int
1149 parseport(const char *port, int defport)
1150 {
1151 int rv;
1152 long nport;
1153 char *p, *ep;
1154
1155 p = xstrdup(port);
1156 nport = strtol(p, &ep, 10);
1157 if (*ep != '\0' && ep == p) {
1158 struct servent *svp;
1159
1160 svp = getservbyname(port, "tcp");
1161 if (svp == NULL) {
1162 badparseport:
1163 if (defport != -1)
1164 warnx("Unknown port `%s', using port %d",
1165 port, defport);
1166 rv = defport;
1167 } else
1168 rv = ntohs(svp->s_port);
1169 } else if (nport < 1 || nport > MAX_IN_PORT_T || *ep != '\0')
1170 goto badparseport;
1171 else
1172 rv = nport;
1173 free(p);
1174 return (rv);
1175 }
1176
1177 /*
1178 * Determine if given string is an IPv6 address or not.
1179 * Return 1 for yes, 0 for no
1180 */
1181 int
1182 isipv6addr(const char *addr)
1183 {
1184 int rv = 0;
1185 #ifdef INET6
1186 struct addrinfo hints, *res;
1187
1188 memset(&hints, 0, sizeof(hints));
1189 hints.ai_family = PF_INET6;
1190 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
1191 hints.ai_flags = AI_NUMERICHOST;
1192 if (getaddrinfo(addr, "0", &hints, &res) != 0)
1193 rv = 0;
1194 else {
1195 rv = 1;
1196 freeaddrinfo(res);
1197 }
1198 if (debug)
1199 fprintf(ttyout, "isipv6addr: got %d for %s\n", rv, addr);
1200 #endif
1201 return (rv == 1) ? 1 : 0;
1202 }
1203
1204
1205 /*
1206 * Internal version of connect(2); sets socket buffer sizes first.
1207 */
1208 int
1209 xconnect(int sock, const struct sockaddr *name, int namelen)
1210 {
1211
1212 setupsockbufsize(sock);
1213 return (connect(sock, name, namelen));
1214 }
1215
1216 /*
1217 * Internal version of listen(2); sets socket buffer sizes first.
1218 */
1219 int
1220 xlisten(int sock, int backlog)
1221 {
1222
1223 setupsockbufsize(sock);
1224 return (listen(sock, backlog));
1225 }
1226
1227 /*
1228 * malloc() with inbuilt error checking
1229 */
1230 void *
1231 xmalloc(size_t size)
1232 {
1233 void *p;
1234
1235 p = malloc(size);
1236 if (p == NULL)
1237 err(1, "Unable to allocate %ld bytes of memory", (long)size);
1238 return (p);
1239 }
1240
1241 /*
1242 * sl_init() with inbuilt error checking
1243 */
1244 StringList *
1245 xsl_init(void)
1246 {
1247 StringList *p;
1248
1249 p = sl_init();
1250 if (p == NULL)
1251 err(1, "Unable to allocate memory for stringlist");
1252 return (p);
1253 }
1254
1255 /*
1256 * sl_add() with inbuilt error checking
1257 */
1258 void
1259 xsl_add(StringList *sl, char *i)
1260 {
1261
1262 if (sl_add(sl, i) == -1)
1263 err(1, "Unable to add `%s' to stringlist", i);
1264 }
1265
1266 /*
1267 * strdup() with inbuilt error checking
1268 */
1269 char *
1270 xstrdup(const char *str)
1271 {
1272 char *s;
1273
1274 if (str == NULL)
1275 errx(1, "xstrdup() called with NULL argument");
1276 s = strdup(str);
1277 if (s == NULL)
1278 err(1, "Unable to allocate memory for string copy");
1279 return (s);
1280 }
1281