1 /* $NetBSD: util.c,v 1.21 2009/11/15 10:12:37 lukem Exp $ */
2 /* from NetBSD: util.c,v 1.152 2009/07/13 19:05:41 roy Exp */
3
4 /*-
5 * Copyright (c) 1997-2009 The NetBSD Foundation, Inc.
6 * All rights reserved.
7 *
8 * This code is derived from software contributed to The NetBSD Foundation
9 * by Luke Mewburn.
10 *
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.
14 *
15 * Redistribution and use in source and binary forms, with or without
16 * modification, are permitted provided that the following conditions
17 * are met:
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.
23 *
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.
35 */
36
37 /*
38 * Copyright (c) 1985, 1989, 1993, 1994
39 * The Regents of the University of California. All rights reserved.
40 *
41 * Redistribution and use in source and binary forms, with or without
42 * modification, are permitted provided that the following conditions
43 * are met:
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.
52 *
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
63 * SUCH DAMAGE.
64 */
65
66 #include "tnftp.h"
67
68 #if 0 /* tnftp */
69
70 #include <sys/cdefs.h>
71 #ifndef lint
72 __RCSID(" NetBSD: util.c,v 1.152 2009/07/13 19:05:41 roy Exp ");
73 #endif /* not lint */
74
75 /*
76 * FTP User Program -- Misc support routines
77 */
78 #include <sys/param.h>
79 #include <sys/socket.h>
80 #include <sys/ioctl.h>
81 #include <sys/time.h>
82 #include <netinet/in.h>
83 #include <arpa/ftp.h>
84
85 #include <ctype.h>
86 #include <err.h>
87 #include <errno.h>
88 #include <fcntl.h>
89 #include <glob.h>
90 #include <signal.h>
91 #include <libgen.h>
92 #include <limits.h>
93 #include <netdb.h>
94 #include <stdio.h>
95 #include <stdlib.h>
96 #include <string.h>
97 #include <termios.h>
98 #include <time.h>
99 #include <tzfile.h>
100 #include <unistd.h>
101
102 #endif /* tnftp */
103
104 #include "ftp_var.h"
105
106 /*
107 * Connect to peer server and auto-login, if possible.
108 */
109 void
setpeer(int argc,char * argv[])110 setpeer(int argc, char *argv[])
111 {
112 char *host;
113 const char *port;
114
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;
122 }
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;
130 }
131 if (gatemode)
132 port = gateport;
133 else
134 port = ftpport;
135 if (argc > 2)
136 port = argv[2];
137
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);
144
145 if (host) {
146 if (gatemode && verbose) {
147 fprintf(ttyout,
148 "Connecting via pass-through server %s\n",
149 gateserver);
150 }
151
152 connected = 1;
153 /*
154 * Set up defaults for FTP.
155 */
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);
169 }
170 }
171
172 static void
parse_feat(const char * fline)173 parse_feat(const char *fline)
174 {
175
176 /*
177 * work-around broken ProFTPd servers that can't
178 * even obey RFC2389.
179 */
180 while (*fline && isspace((int)*fline))
181 fline++;
182
183 if (strcasecmp(fline, "MDTM") == 0)
184 features[FEAT_MDTM] = 1;
185 else if (strncasecmp(fline, "MLST", sizeof("MLST") - 1) == 0) {
186 features[FEAT_MLST] = 1;
187 } else if (strcasecmp(fline, "REST STREAM") == 0)
188 features[FEAT_REST_STREAM] = 1;
189 else if (strcasecmp(fline, "SIZE") == 0)
190 features[FEAT_SIZE] = 1;
191 else if (strcasecmp(fline, "TVFS") == 0)
192 features[FEAT_TVFS] = 1;
193 }
194
195 /*
196 * Determine the remote system type (SYST) and features (FEAT).
197 * Call after a successful login (i.e, connected = -1)
198 */
199 void
getremoteinfo(void)200 getremoteinfo(void)
201 {
202 int overbose, i;
203
204 overbose = verbose;
205 if (ftp_debug == 0)
206 verbose = -1;
207
208 /* determine remote system type */
209 if (command("SYST") == COMPLETE) {
210 if (overbose) {
211 char *cp, c;
212
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';
222 }
223
224 fprintf(ttyout, "Remote system type is %s.\n",
225 reply_string + 4);
226 if (cp)
227 *cp = c;
228 }
229 if (!strncmp(reply_string, "215 UNIX Type: L8", 17)) {
230 if (proxy)
231 unix_proxy = 1;
232 else
233 unix_server = 1;
234 /*
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.
239 */
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);
256 }
257 }
258
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;
267 }
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
281 }
282 #endif
283 reply_callback = NULL;
284
285 verbose = overbose;
286 }
287
288 /*
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.
293 */
294 void
cleanuppeer(void)295 cleanuppeer(void)
296 {
297
298 if (cout)
299 (void)fclose(cout);
300 cout = NULL;
301 connected = 0;
302 unix_server = 0;
303 unix_proxy = 0;
304 /*
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
308 */
309 if (anonftp == 2)
310 anonftp = 0;
311 data = -1;
312 epsv4bad = 0;
313 epsv6bad = 0;
314 if (username)
315 free(username);
316 username = NULL;
317 if (!proxy)
318 macnum = 0;
319 }
320
321 /*
322 * Top-level signal handler for interrupted commands.
323 */
324 void
intr(int signo)325 intr(int signo)
326 {
327
328 sigint_raised = 1;
329 alarmtimer(0);
330 if (fromatty)
331 write(fileno(ttyout), "\n", 1);
332 siglongjmp(toplevel, 1);
333 }
334
335 /*
336 * Signal handler for lost connections; cleanup various elements of
337 * the connection state, and call cleanuppeer() to finish it off.
338 */
339 void
lostpeer(int dummy)340 lostpeer(int dummy)
341 {
342 int oerrno = errno;
343
344 alarmtimer(0);
345 if (connected) {
346 if (cout != NULL) {
347 (void)shutdown(fileno(cout), 1+1);
348 (void)fclose(cout);
349 cout = NULL;
350 }
351 if (data >= 0) {
352 (void)shutdown(data, 1+1);
353 (void)close(data);
354 data = -1;
355 }
356 connected = 0;
357 }
358 pswitch(1);
359 if (connected) {
360 if (cout != NULL) {
361 (void)shutdown(fileno(cout), 1+1);
362 (void)fclose(cout);
363 cout = NULL;
364 }
365 connected = 0;
366 }
367 proxflag = 0;
368 pswitch(0);
369 cleanuppeer();
370 errno = oerrno;
371 }
372
373
374 /*
375 * Login to remote host, using given username & password if supplied.
376 * Return non-zero if successful.
377 */
378 int
ftp_login(const char * host,const char * luser,const char * lpass)379 ftp_login(const char *host, const char *luser, const char *lpass)
380 {
381 char tmp[80];
382 char *fuser, *pass, *facct, *p;
383 char emptypass[] = "";
384 const char *errormsg;
385 int n, aflag, rval, nlen;
386
387 aflag = rval = 0;
388 fuser = pass = facct = NULL;
389 if (luser)
390 fuser = ftp_strdup(luser);
391 if (lpass)
392 pass = ftp_strdup(lpass);
393
394 DPRINTF("ftp_login: user `%s' pass `%s' host `%s'\n",
395 STRorNULL(fuser), STRorNULL(pass), STRorNULL(host));
396
397 /*
398 * Set up arguments for an anonymous FTP session, if necessary.
399 */
400 if (anonftp) {
401 FREEPTR(fuser);
402 fuser = ftp_strdup("anonymous"); /* as per RFC1635 */
403 FREEPTR(pass);
404 pass = ftp_strdup(getoptionvalue("anonpass"));
405 }
406
407 if (ruserpass(host, &fuser, &pass, &facct) < 0) {
408 code = -1;
409 goto cleanup_ftp_login;
410 }
411
412 while (fuser == 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 = get_line(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 fuser = ftp_strdup(localname);
425 } else {
426 fuser = ftp_strdup(tmp);
427 }
428 }
429
430 if (gatemode) {
431 char *nuser;
432 size_t len;
433
434 len = strlen(fuser) + 1 + strlen(host) + 1;
435 nuser = ftp_malloc(len);
436 (void)strlcpy(nuser, fuser, len);
437 (void)strlcat(nuser, "@", len);
438 (void)strlcat(nuser, host, len);
439 FREEPTR(fuser);
440 fuser = nuser;
441 }
442
443 n = command("USER %s", fuser);
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));
451 }
452 n = command("PASS %s", pass);
453 memset(pass, 0, strlen(pass));
454 }
455 if (n == CONTINUE) {
456 aflag++;
457 if (facct == NULL) {
458 p = getpass("Account: ");
459 if (p == NULL)
460 p = emptypass;
461 facct = ftp_strdup(p);
462 memset(p, 0, strlen(p));
463 }
464 if (facct[0] == '\0') {
465 warnx("Login failed");
466 goto cleanup_ftp_login;
467 }
468 n = command("ACCT %s", facct);
469 memset(facct, 0, strlen(facct));
470 }
471 if ((n != COMPLETE) ||
472 (!aflag && facct != NULL && command("ACCT %s", facct) != COMPLETE)) {
473 warnx("Login failed");
474 goto cleanup_ftp_login;
475 }
476 rval = 1;
477 username = ftp_strdup(fuser);
478 if (proxy)
479 goto cleanup_ftp_login;
480
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;
489 }
490 }
491 updatelocalcwd();
492 updateremotecwd();
493
494 cleanup_ftp_login:
495 FREEPTR(fuser);
496 if (pass != NULL)
497 memset(pass, 0, strlen(pass));
498 FREEPTR(pass);
499 if (facct != NULL)
500 memset(facct, 0, strlen(facct));
501 FREEPTR(facct);
502 return (rval);
503 }
504
505 /*
506 * `another' gets another argument, and stores the new argc and argv.
507 * It reverts to the top level (via intr()) on EOF/error.
508 *
509 * Returns false if no new arguments have been added.
510 */
511 int
another(int * pargc,char *** pargv,const char * aprompt)512 another(int *pargc, char ***pargv, const char *aprompt)
513 {
514 const char *errormsg;
515 int ret, nlen;
516 size_t len;
517
518 len = strlen(line);
519 if (len >= sizeof(line) - 3) {
520 fputs("Sorry, arguments too long.\n", ttyout);
521 intr(0);
522 }
523 fprintf(ttyout, "(%s) ", aprompt);
524 line[len++] = ' ';
525 errormsg = NULL;
526 nlen = get_line(stdin, line + len, sizeof(line)-len, &errormsg);
527 if (nlen < 0) {
528 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "operation");
529 intr(0);
530 }
531 len += nlen;
532 makeargv();
533 ret = margc > *pargc;
534 *pargc = margc;
535 *pargv = margv;
536 return (ret);
537 }
538
539 /*
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.
543 */
544 char *
remglob(char * argv[],int doswitch,const char ** errbuf)545 remglob(char *argv[], int doswitch, const char **errbuf)
546 {
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 *rmode;
554 size_t len;
555
556 if (!mflag || !connected) {
557 if (!doglob)
558 args = NULL;
559 else {
560 if (ftemp) {
561 (void)fclose(ftemp);
562 ftemp = NULL;
563 }
564 }
565 return (NULL);
566 }
567 if (!doglob) {
568 if (args == NULL)
569 args = argv;
570 if ((cp = *++args) == NULL)
571 args = NULL;
572 return (cp);
573 }
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);
582 }
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 (rmode = "w"; *++argv != NULL; rmode = "a")
593 recvrequest("NLST", temp, *argv, rmode, 0, 0);
594 if ((code / 100) != COMPLETE) {
595 if (errbuf != NULL)
596 *errbuf = reply_string;
597 }
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);
612 }
613 }
614 if (fgets(buf, sizeof(buf), ftemp) == NULL) {
615 (void)fclose(ftemp);
616 ftemp = NULL;
617 return (NULL);
618 }
619 if ((cp = strchr(buf, '\n')) != NULL)
620 *cp = '\0';
621 return (buf);
622 }
623
624 /*
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.
630 */
631 char *
globulize(const char * pattern)632 globulize(const char *pattern)
633 {
634 glob_t gl;
635 int flags;
636 char *p;
637
638 if (!doglob)
639 return (ftp_strdup(pattern));
640
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);
647 }
648 p = ftp_strdup(gl.gl_pathv[0]);
649 globfree(&gl);
650 return (p);
651 }
652
653 /*
654 * determine size of remote file
655 */
656 off_t
remotesize(const char * file,int noisy)657 remotesize(const char *file, int noisy)
658 {
659 int overbose, r;
660 off_t size;
661
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;
671 }
672 r = command("SIZE %s", file);
673 if (r == COMPLETE) {
674 char *cp, *ep;
675
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;
682 }
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);
689 }
690 }
691 cleanup_remotesize:
692 verbose = overbose;
693 return (size);
694 }
695
696 /*
697 * determine last modification time (in GMT) of remote file
698 */
699 time_t
remotemodtime(const char * file,int noisy)700 remotemodtime(const char *file, int noisy)
701 {
702 int overbose, ocode, r;
703 time_t rtime;
704
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;
715 }
716 r = command("MDTM %s", file);
717 if (r == COMPLETE) {
718 struct tm timebuf;
719 char *timestr, *frac;
720
721 /*
722 * time-val = 14DIGIT [ "." 1*DIGIT ]
723 * YYYYMMDDHHMMSS[.sss]
724 * mdtm-response = "213" SP time-val CRLF / error-response
725 */
726 timestr = reply_string + 4;
727
728 /*
729 * parse fraction.
730 * XXX: ignored for now
731 */
732 frac = strchr(timestr, '\r');
733 if (frac != NULL)
734 *frac = '\0';
735 frac = strchr(timestr, '.');
736 if (frac != NULL)
737 *frac++ = '\0';
738 if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
739 /*
740 * XXX: Workaround for lame ftpd's that return
741 * `19100' instead of `2000'
742 */
743 fprintf(ttyout,
744 "Y2K warning! Incorrect time-val `%s' received from server.\n",
745 timestr);
746 timestr++;
747 timestr[0] = '2';
748 timestr[1] = '0';
749 fprintf(ttyout, "Converted to `%s'\n", timestr);
750 }
751 memset(&timebuf, 0, sizeof(timebuf));
752 if (strlen(timestr) != 14 ||
753 (strptime(timestr, "%Y%m%d%H%M%S", &timebuf) == NULL)) {
754 bad_parse_time:
755 fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
756 goto cleanup_parse_time;
757 }
758 timebuf.tm_isdst = -1;
759 rtime = timegm(&timebuf);
760 if (rtime == -1) {
761 if (noisy || ftp_debug != 0)
762 goto bad_parse_time;
763 else
764 goto cleanup_parse_time;
765 } else {
766 DPRINTF("remotemodtime: parsed date `%s' as " LLF
767 ", %s",
768 timestr, (LLT)rtime,
769 rfc2822time(localtime(&rtime)));
770 }
771 } else {
772 if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
773 features[FEAT_MDTM] = 0;
774 if (noisy && ftp_debug == 0) {
775 fputs(reply_string, ttyout);
776 putc('\n', ttyout);
777 }
778 }
779 cleanup_parse_time:
780 verbose = overbose;
781 if (rtime == -1)
782 code = ocode;
783 return (rtime);
784 }
785
786 /*
787 * Format tm in an RFC2822 compatible manner, with a trailing \n.
788 * Returns a pointer to a static string containing the result.
789 */
790 const char *
rfc2822time(const struct tm * tm)791 rfc2822time(const struct tm *tm)
792 {
793 static char result[50];
794
795 if (strftime(result, sizeof(result),
796 "%a, %d %b %Y %H:%M:%S %z\n", tm) == 0)
797 errx(1, "Can't convert RFC2822 time: buffer too small");
798 return result;
799 }
800
801 /*
802 * Update global `localcwd', which contains the state of the local cwd
803 */
804 void
updatelocalcwd(void)805 updatelocalcwd(void)
806 {
807
808 if (getcwd(localcwd, sizeof(localcwd)) == NULL)
809 localcwd[0] = '\0';
810 DPRINTF("updatelocalcwd: got `%s'\n", localcwd);
811 }
812
813 /*
814 * Update global `remotecwd', which contains the state of the remote cwd
815 */
816 void
updateremotecwd(void)817 updateremotecwd(void)
818 {
819 int overbose, ocode;
820 size_t i;
821 char *cp;
822
823 overbose = verbose;
824 ocode = code;
825 if (ftp_debug == 0)
826 verbose = -1;
827 if (command("PWD") != COMPLETE)
828 goto badremotecwd;
829 cp = strchr(reply_string, ' ');
830 if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
831 goto badremotecwd;
832 cp += 2;
833 for (i = 0; *cp && i < sizeof(remotecwd) - 1; i++, cp++) {
834 if (cp[0] == '"') {
835 if (cp[1] == '"')
836 cp++;
837 else
838 break;
839 }
840 remotecwd[i] = *cp;
841 }
842 remotecwd[i] = '\0';
843 DPRINTF("updateremotecwd: got `%s'\n", remotecwd);
844 goto cleanupremotecwd;
845 badremotecwd:
846 remotecwd[0]='\0';
847 cleanupremotecwd:
848 verbose = overbose;
849 code = ocode;
850 }
851
852 /*
853 * Ensure file is in or under dir.
854 * Returns 1 if so, 0 if not (or an error occurred).
855 */
856 int
fileindir(const char * file,const char * dir)857 fileindir(const char *file, const char *dir)
858 {
859 char parentdirbuf[PATH_MAX+1], *parentdir;
860 char realdir[PATH_MAX+1];
861 size_t dirlen;
862
863 /* determine parent directory of file */
864 (void)strlcpy(parentdirbuf, file, sizeof(parentdirbuf));
865 parentdir = dirname(parentdirbuf);
866 if (strcmp(parentdir, ".") == 0)
867 return 1; /* current directory is ok */
868
869 /* find the directory */
870 if (realpath(parentdir, realdir) == NULL) {
871 warn("Unable to determine real path of `%s'", parentdir);
872 return 0;
873 }
874 if (realdir[0] != '/') /* relative result is ok */
875 return 1;
876 dirlen = strlen(dir);
877 if (strncmp(realdir, dir, dirlen) == 0 &&
878 (realdir[dirlen] == '/' || realdir[dirlen] == '\0'))
879 return 1;
880 return 0;
881 }
882
883 /*
884 * List words in stringlist, vertically arranged
885 */
886 void
list_vertical(StringList * sl)887 list_vertical(StringList *sl)
888 {
889 size_t i, j;
890 size_t columns, lines;
891 char *p;
892 size_t w, width;
893
894 width = 0;
895
896 for (i = 0 ; i < sl->sl_cur ; i++) {
897 w = strlen(sl->sl_str[i]);
898 if (w > width)
899 width = w;
900 }
901 width = (width + 8) &~ 7;
902
903 columns = ttywidth / width;
904 if (columns == 0)
905 columns = 1;
906 lines = (sl->sl_cur + columns - 1) / columns;
907 for (i = 0; i < lines; i++) {
908 for (j = 0; j < columns; j++) {
909 p = sl->sl_str[j * lines + i];
910 if (p)
911 fputs(p, ttyout);
912 if (j * lines + i + lines >= sl->sl_cur) {
913 putc('\n', ttyout);
914 break;
915 }
916 if (p) {
917 w = strlen(p);
918 while (w < width) {
919 w = (w + 8) &~ 7;
920 (void)putc('\t', ttyout);
921 }
922 }
923 }
924 }
925 }
926
927 /*
928 * Update the global ttywidth value, using TIOCGWINSZ.
929 */
930 void
setttywidth(int a)931 setttywidth(int a)
932 {
933 struct winsize winsize;
934 int oerrno = errno;
935
936 if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
937 winsize.ws_col != 0)
938 ttywidth = winsize.ws_col;
939 else
940 ttywidth = 80;
941 errno = oerrno;
942 }
943
944 /*
945 * Change the rate limit up (SIGUSR1) or down (SIGUSR2)
946 */
947 void
crankrate(int sig)948 crankrate(int sig)
949 {
950
951 switch (sig) {
952 case SIGUSR1:
953 if (rate_get)
954 rate_get += rate_get_incr;
955 if (rate_put)
956 rate_put += rate_put_incr;
957 break;
958 case SIGUSR2:
959 if (rate_get && rate_get > rate_get_incr)
960 rate_get -= rate_get_incr;
961 if (rate_put && rate_put > rate_put_incr)
962 rate_put -= rate_put_incr;
963 break;
964 default:
965 err(1, "crankrate invoked with unknown signal: %d", sig);
966 }
967 }
968
969
970 /*
971 * Setup or cleanup EditLine structures
972 */
973 #ifndef NO_EDITCOMPLETE
974 void
controlediting(void)975 controlediting(void)
976 {
977 if (editing && el == NULL && hist == NULL) {
978 HistEvent ev;
979 int editmode;
980
981 el = el_init(getprogname(), stdin, ttyout, stderr);
982 /* init editline */
983 hist = history_init(); /* init the builtin history */
984 history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
985 el_set(el, EL_HIST, history, hist); /* use history */
986
987 el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
988 el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
989 el_set(el, EL_RPROMPT, rprompt);
990
991 /* add local file completion, bind to TAB */
992 el_set(el, EL_ADDFN, "ftp-complete",
993 "Context sensitive argument completion",
994 complete);
995 el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
996 el_source(el, NULL); /* read ~/.editrc */
997 if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
998 editing = 0; /* the user doesn't want editing,
999 * so disable, and let statement
1000 * below cleanup */
1001 else
1002 el_set(el, EL_SIGNAL, 1);
1003 }
1004 if (!editing) {
1005 if (hist) {
1006 history_end(hist);
1007 hist = NULL;
1008 }
1009 if (el) {
1010 el_end(el);
1011 el = NULL;
1012 }
1013 }
1014 }
1015 #endif /* !NO_EDITCOMPLETE */
1016
1017 /*
1018 * Convert the string `arg' to an int, which may have an optional SI suffix
1019 * (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
1020 */
1021 int
strsuftoi(const char * arg)1022 strsuftoi(const char *arg)
1023 {
1024 char *cp;
1025 long val;
1026
1027 if (!isdigit((unsigned char)arg[0]))
1028 return (-1);
1029
1030 val = strtol(arg, &cp, 10);
1031 if (cp != NULL) {
1032 if (cp[0] != '\0' && cp[1] != '\0')
1033 return (-1);
1034 switch (tolower((unsigned char)cp[0])) {
1035 case '\0':
1036 case 'b':
1037 break;
1038 case 'k':
1039 val <<= 10;
1040 break;
1041 case 'm':
1042 val <<= 20;
1043 break;
1044 case 'g':
1045 val <<= 30;
1046 break;
1047 default:
1048 return (-1);
1049 }
1050 }
1051 if (val < 0 || val > INT_MAX)
1052 return (-1);
1053
1054 return (val);
1055 }
1056
1057 /*
1058 * Set up socket buffer sizes before a connection is made.
1059 */
1060 void
setupsockbufsize(int sock)1061 setupsockbufsize(int sock)
1062 {
1063 socklen_t slen;
1064
1065 if (0 == rcvbuf_size) {
1066 slen = sizeof(rcvbuf_size);
1067 if (getsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1068 (void *)&rcvbuf_size, &slen) == -1)
1069 err(1, "Unable to determine rcvbuf size");
1070 if (rcvbuf_size <= 0)
1071 rcvbuf_size = 8 * 1024;
1072 if (rcvbuf_size > 8 * 1024 * 1024)
1073 rcvbuf_size = 8 * 1024 * 1024;
1074 DPRINTF("setupsockbufsize: rcvbuf_size determined as %d\n",
1075 rcvbuf_size);
1076 }
1077 if (0 == sndbuf_size) {
1078 slen = sizeof(sndbuf_size);
1079 if (getsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1080 (void *)&sndbuf_size, &slen) == -1)
1081 err(1, "Unable to determine sndbuf size");
1082 if (sndbuf_size <= 0)
1083 sndbuf_size = 8 * 1024;
1084 if (sndbuf_size > 8 * 1024 * 1024)
1085 sndbuf_size = 8 * 1024 * 1024;
1086 DPRINTF("setupsockbufsize: sndbuf_size determined as %d\n",
1087 sndbuf_size);
1088 }
1089
1090 #ifdef __FreeBSD__
1091 DPRINTF("auto_rcvbuf = %d\n", auto_rcvbuf);
1092 if (auto_sndbuf == 0) {
1093 #endif
1094 if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1095 (void *)&sndbuf_size, sizeof(sndbuf_size)) == -1)
1096 warn("Unable to set sndbuf size %d", sndbuf_size);
1097 #ifdef __FreeBSD__
1098 }
1099 #endif
1100
1101 #ifdef __FreeBSD__
1102 DPRINTF("auto_sndbuf = %d\n", auto_sndbuf);
1103 if (auto_rcvbuf == 0) {
1104 #endif
1105 if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1106 (void *)&rcvbuf_size, sizeof(rcvbuf_size)) == -1)
1107 warn("Unable to set rcvbuf size %d", rcvbuf_size);
1108 #ifdef __FreeBSD__
1109 }
1110 #endif
1111 }
1112
1113 /*
1114 * Copy characters from src into dst, \ quoting characters that require it
1115 */
1116 void
ftpvis(char * dst,size_t dstlen,const char * src,size_t srclen)1117 ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
1118 {
1119 size_t di, si;
1120
1121 for (di = si = 0;
1122 src[si] != '\0' && di < dstlen && si < srclen;
1123 di++, si++) {
1124 switch (src[si]) {
1125 case '\\':
1126 case ' ':
1127 case '\t':
1128 case '\r':
1129 case '\n':
1130 case '"':
1131 dst[di++] = '\\';
1132 if (di >= dstlen)
1133 break;
1134 /* FALLTHROUGH */
1135 default:
1136 dst[di] = src[si];
1137 }
1138 }
1139 dst[di] = '\0';
1140 }
1141
1142 /*
1143 * Copy src into buf (which is len bytes long), expanding % sequences.
1144 */
1145 void
formatbuf(char * buf,size_t len,const char * src)1146 formatbuf(char *buf, size_t len, const char *src)
1147 {
1148 const char *p, *p2, *q;
1149 size_t i;
1150 int op, updirs, pdirs;
1151
1152 #define ADDBUF(x) do { \
1153 if (i >= len - 1) \
1154 goto endbuf; \
1155 buf[i++] = (x); \
1156 } while (0)
1157
1158 p = src;
1159 for (i = 0; *p; p++) {
1160 if (*p != '%') {
1161 ADDBUF(*p);
1162 continue;
1163 }
1164 p++;
1165
1166 switch (op = *p) {
1167
1168 case '/':
1169 case '.':
1170 case 'c':
1171 p2 = connected ? remotecwd : "";
1172 updirs = pdirs = 0;
1173
1174 /* option to determine fixed # of dirs from path */
1175 if (op == '.' || op == 'c') {
1176 int skip;
1177
1178 q = p2;
1179 while (*p2) /* calc # of /'s */
1180 if (*p2++ == '/')
1181 updirs++;
1182 if (p[1] == '0') { /* print <x> or ... */
1183 pdirs = 1;
1184 p++;
1185 }
1186 if (p[1] >= '1' && p[1] <= '9') {
1187 /* calc # to skip */
1188 skip = p[1] - '0';
1189 p++;
1190 } else
1191 skip = 1;
1192
1193 updirs -= skip;
1194 while (skip-- > 0) {
1195 while ((p2 > q) && (*p2 != '/'))
1196 p2--; /* back up */
1197 if (skip && p2 > q)
1198 p2--;
1199 }
1200 if (*p2 == '/' && p2 != q)
1201 p2++;
1202 }
1203
1204 if (updirs > 0 && pdirs) {
1205 if (i >= len - 5)
1206 break;
1207 if (op == '.') {
1208 ADDBUF('.');
1209 ADDBUF('.');
1210 ADDBUF('.');
1211 } else {
1212 ADDBUF('/');
1213 ADDBUF('<');
1214 if (updirs > 9) {
1215 ADDBUF('9');
1216 ADDBUF('+');
1217 } else
1218 ADDBUF('0' + updirs);
1219 ADDBUF('>');
1220 }
1221 }
1222 for (; *p2; p2++)
1223 ADDBUF(*p2);
1224 break;
1225
1226 case 'M':
1227 case 'm':
1228 for (p2 = connected && hostname ? hostname : "-";
1229 *p2 ; p2++) {
1230 if (op == 'm' && *p2 == '.')
1231 break;
1232 ADDBUF(*p2);
1233 }
1234 break;
1235
1236 case 'n':
1237 for (p2 = connected ? username : "-"; *p2 ; p2++)
1238 ADDBUF(*p2);
1239 break;
1240
1241 case '%':
1242 ADDBUF('%');
1243 break;
1244
1245 default: /* display unknown codes literally */
1246 ADDBUF('%');
1247 ADDBUF(op);
1248 break;
1249
1250 }
1251 }
1252 endbuf:
1253 buf[i] = '\0';
1254 }
1255
1256 /*
1257 * Determine if given string is an IPv6 address or not.
1258 * Return 1 for yes, 0 for no
1259 */
1260 int
isipv6addr(const char * addr)1261 isipv6addr(const char *addr)
1262 {
1263 int rv = 0;
1264 #ifdef INET6
1265 struct addrinfo hints, *res;
1266
1267 memset(&hints, 0, sizeof(hints));
1268 hints.ai_family = AF_INET6;
1269 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
1270 hints.ai_flags = AI_NUMERICHOST;
1271 if (getaddrinfo(addr, "0", &hints, &res) != 0)
1272 rv = 0;
1273 else {
1274 rv = 1;
1275 freeaddrinfo(res);
1276 }
1277 DPRINTF("isipv6addr: got %d for %s\n", rv, addr);
1278 #endif
1279 return (rv == 1) ? 1 : 0;
1280 }
1281
1282 /*
1283 * Read a line from the FILE stream into buf/buflen using fgets(), so up
1284 * to buflen-1 chars will be read and the result will be NUL terminated.
1285 * If the line has a trailing newline it will be removed.
1286 * If the line is too long, excess characters will be read until
1287 * newline/EOF/error.
1288 * If EOF/error occurs or a too-long line is encountered and errormsg
1289 * isn't NULL, it will be changed to a description of the problem.
1290 * (The EOF message has a leading \n for cosmetic purposes).
1291 * Returns:
1292 * >=0 length of line (excluding trailing newline) if all ok
1293 * -1 error occurred
1294 * -2 EOF encountered
1295 * -3 line was too long
1296 */
1297 int
get_line(FILE * stream,char * buf,size_t buflen,const char ** errormsg)1298 get_line(FILE *stream, char *buf, size_t buflen, const char **errormsg)
1299 {
1300 int rv, ch;
1301 size_t len;
1302
1303 if (fgets(buf, buflen, stream) == NULL) {
1304 if (feof(stream)) { /* EOF */
1305 rv = -2;
1306 if (errormsg)
1307 *errormsg = "\nEOF received";
1308 } else { /* error */
1309 rv = -1;
1310 if (errormsg)
1311 *errormsg = "Error encountered";
1312 }
1313 clearerr(stream);
1314 return rv;
1315 }
1316 len = strlen(buf);
1317 if (buf[len-1] == '\n') { /* clear any trailing newline */
1318 buf[--len] = '\0';
1319 } else if (len == buflen-1) { /* line too long */
1320 while ((ch = getchar()) != '\n' && ch != EOF)
1321 continue;
1322 if (errormsg)
1323 *errormsg = "Input line is too long";
1324 clearerr(stream);
1325 return -3;
1326 }
1327 if (errormsg)
1328 *errormsg = NULL;
1329 return len;
1330 }
1331
1332 /*
1333 * Internal version of connect(2); sets socket buffer sizes,
1334 * binds to a specific local address (if set), and
1335 * supports a connection timeout using a non-blocking connect(2) with
1336 * a poll(2).
1337 * Socket fcntl flags are temporarily updated to include O_NONBLOCK;
1338 * these will not be reverted on connection failure.
1339 * Returns 0 on success, or -1 upon failure (with an appropriate
1340 * error message displayed.)
1341 */
1342 int
ftp_connect(int sock,const struct sockaddr * name,socklen_t namelen)1343 ftp_connect(int sock, const struct sockaddr *name, socklen_t namelen)
1344 {
1345 int flags, rv, timeout, error;
1346 socklen_t slen;
1347 struct timeval endtime, now, td;
1348 struct pollfd pfd[1];
1349 char hname[NI_MAXHOST];
1350 char sname[NI_MAXSERV];
1351
1352 setupsockbufsize(sock);
1353 if (getnameinfo(name, namelen,
1354 hname, sizeof(hname), sname, sizeof(sname),
1355 NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
1356 strlcpy(hname, "?", sizeof(hname));
1357 strlcpy(sname, "?", sizeof(sname));
1358 }
1359
1360 if (bindai != NULL) { /* bind to specific addr */
1361 struct addrinfo *ai;
1362
1363 for (ai = bindai; ai != NULL; ai = ai->ai_next) {
1364 if (ai->ai_family == name->sa_family)
1365 break;
1366 }
1367 if (ai == NULL)
1368 ai = bindai;
1369 if (bind(sock, ai->ai_addr, ai->ai_addrlen) == -1) {
1370 char bname[NI_MAXHOST];
1371 int saveerr;
1372
1373 saveerr = errno;
1374 if (getnameinfo(ai->ai_addr, ai->ai_addrlen,
1375 bname, sizeof(bname), NULL, 0, NI_NUMERICHOST) != 0)
1376 strlcpy(bname, "?", sizeof(bname));
1377 errno = saveerr;
1378 warn("Can't bind to `%s'", bname);
1379 return -1;
1380 }
1381 }
1382
1383 /* save current socket flags */
1384 if ((flags = fcntl(sock, F_GETFL, 0)) == -1) {
1385 warn("Can't %s socket flags for connect to `%s:%s'",
1386 "save", hname, sname);
1387 return -1;
1388 }
1389 /* set non-blocking connect */
1390 if (fcntl(sock, F_SETFL, flags | O_NONBLOCK) == -1) {
1391 warn("Can't set socket non-blocking for connect to `%s:%s'",
1392 hname, sname);
1393 return -1;
1394 }
1395
1396 /* NOTE: we now must restore socket flags on successful exit */
1397
1398 pfd[0].fd = sock;
1399 pfd[0].events = POLLIN|POLLOUT;
1400
1401 if (quit_time > 0) { /* want a non default timeout */
1402 (void)gettimeofday(&endtime, NULL);
1403 endtime.tv_sec += quit_time; /* determine end time */
1404 }
1405
1406 rv = connect(sock, name, namelen); /* inititate the connection */
1407 if (rv == -1) { /* connection error */
1408 if (errno != EINPROGRESS) { /* error isn't "please wait" */
1409 connecterror:
1410 warn("Can't connect to `%s:%s'", hname, sname);
1411 return -1;
1412 }
1413
1414 /* connect EINPROGRESS; wait */
1415 do {
1416 if (quit_time > 0) { /* determine timeout */
1417 (void)gettimeofday(&now, NULL);
1418 timersub(&endtime, &now, &td);
1419 timeout = td.tv_sec * 1000 + td.tv_usec/1000;
1420 if (timeout < 0)
1421 timeout = 0;
1422 } else {
1423 timeout = INFTIM;
1424 }
1425 pfd[0].revents = 0;
1426 rv = ftp_poll(pfd, 1, timeout);
1427 /* loop until poll ! EINTR */
1428 } while (rv == -1 && errno == EINTR);
1429
1430 if (rv == 0) { /* poll (connect) timed out */
1431 errno = ETIMEDOUT;
1432 goto connecterror;
1433 }
1434
1435 if (rv == -1) { /* poll error */
1436 goto connecterror;
1437 } else if (pfd[0].revents & (POLLIN|POLLOUT)) {
1438 slen = sizeof(error); /* OK, or pending error */
1439 if (getsockopt(sock, SOL_SOCKET, SO_ERROR,
1440 &error, &slen) == -1) {
1441 /* Solaris pending error */
1442 goto connecterror;
1443 } else if (error != 0) {
1444 errno = error; /* BSD pending error */
1445 goto connecterror;
1446 }
1447 } else {
1448 errno = EBADF; /* this shouldn't happen ... */
1449 goto connecterror;
1450 }
1451 }
1452
1453 if (fcntl(sock, F_SETFL, flags) == -1) {
1454 /* restore socket flags */
1455 warn("Can't %s socket flags for connect to `%s:%s'",
1456 "restore", hname, sname);
1457 return -1;
1458 }
1459 return 0;
1460 }
1461
1462 /*
1463 * Internal version of listen(2); sets socket buffer sizes first.
1464 */
1465 int
ftp_listen(int sock,int backlog)1466 ftp_listen(int sock, int backlog)
1467 {
1468
1469 setupsockbufsize(sock);
1470 return (listen(sock, backlog));
1471 }
1472
1473 /*
1474 * Internal version of poll(2), to allow reimplementation by select(2)
1475 * on platforms without the former.
1476 */
1477 int
ftp_poll(struct pollfd * fds,int nfds,int timeout)1478 ftp_poll(struct pollfd *fds, int nfds, int timeout)
1479 {
1480 #if defined(HAVE_POLL)
1481 return poll(fds, nfds, timeout);
1482
1483 #elif defined(HAVE_SELECT)
1484 /* implement poll(2) using select(2) */
1485 fd_set rset, wset, xset;
1486 const int rsetflags = POLLIN | POLLRDNORM;
1487 const int wsetflags = POLLOUT | POLLWRNORM;
1488 const int xsetflags = POLLRDBAND;
1489 struct timeval tv, *ptv;
1490 int i, max, rv;
1491
1492 FD_ZERO(&rset); /* build list of read & write events */
1493 FD_ZERO(&wset);
1494 FD_ZERO(&xset);
1495 max = 0;
1496 for (i = 0; i < nfds; i++) {
1497 if (fds[i].fd > FD_SETSIZE) {
1498 warnx("can't select fd %d", fds[i].fd);
1499 errno = EINVAL;
1500 return -1;
1501 } else if (fds[i].fd > max)
1502 max = fds[i].fd;
1503 if (fds[i].events & rsetflags)
1504 FD_SET(fds[i].fd, &rset);
1505 if (fds[i].events & wsetflags)
1506 FD_SET(fds[i].fd, &wset);
1507 if (fds[i].events & xsetflags)
1508 FD_SET(fds[i].fd, &xset);
1509 }
1510
1511 ptv = &tv; /* determine timeout */
1512 if (timeout == -1) { /* wait forever */
1513 ptv = NULL;
1514 } else if (timeout == 0) { /* poll once */
1515 ptv->tv_sec = 0;
1516 ptv->tv_usec = 0;
1517 }
1518 else if (timeout != 0) { /* wait timeout milliseconds */
1519 ptv->tv_sec = timeout / 1000;
1520 ptv->tv_usec = (timeout % 1000) * 1000;
1521 }
1522 rv = select(max + 1, &rset, &wset, &xset, ptv);
1523 if (rv <= 0) /* -1 == error, 0 == timeout */
1524 return rv;
1525
1526 for (i = 0; i < nfds; i++) { /* determine results */
1527 if (FD_ISSET(fds[i].fd, &rset))
1528 fds[i].revents |= (fds[i].events & rsetflags);
1529 if (FD_ISSET(fds[i].fd, &wset))
1530 fds[i].revents |= (fds[i].events & wsetflags);
1531 if (FD_ISSET(fds[i].fd, &xset))
1532 fds[i].revents |= (fds[i].events & xsetflags);
1533 }
1534 return rv;
1535
1536 #else
1537 # error no way to implement xpoll
1538 #endif
1539 }
1540
1541 /*
1542 * malloc() with inbuilt error checking
1543 */
1544 void *
ftp_malloc(size_t size)1545 ftp_malloc(size_t size)
1546 {
1547 void *p;
1548
1549 p = malloc(size);
1550 if (p == NULL)
1551 err(1, "Unable to allocate %ld bytes of memory", (long)size);
1552 return (p);
1553 }
1554
1555 /*
1556 * sl_init() with inbuilt error checking
1557 */
1558 StringList *
ftp_sl_init(void)1559 ftp_sl_init(void)
1560 {
1561 StringList *p;
1562
1563 p = sl_init();
1564 if (p == NULL)
1565 err(1, "Unable to allocate memory for stringlist");
1566 return (p);
1567 }
1568
1569 /*
1570 * sl_add() with inbuilt error checking
1571 */
1572 void
ftp_sl_add(StringList * sl,char * i)1573 ftp_sl_add(StringList *sl, char *i)
1574 {
1575
1576 if (sl_add(sl, i) == -1)
1577 err(1, "Unable to add `%s' to stringlist", i);
1578 }
1579
1580 /*
1581 * strdup() with inbuilt error checking
1582 */
1583 char *
ftp_strdup(const char * str)1584 ftp_strdup(const char *str)
1585 {
1586 char *s;
1587
1588 if (str == NULL)
1589 errx(1, "ftp_strdup: called with NULL argument");
1590 s = strdup(str);
1591 if (s == NULL)
1592 err(1, "Unable to allocate memory for string copy");
1593 return (s);
1594 }
1595