1 /* $NetBSD: util.c,v 1.158 2013/02/19 23:29:15 dsl Exp $ */
2
3 /*-
4 * Copyright (c) 1997-2009 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 *
23 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
24 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
25 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
26 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
27 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
28 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
29 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
30 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
31 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
32 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
33 * POSSIBILITY OF SUCH DAMAGE.
34 */
35
36 /*
37 * Copyright (c) 1985, 1989, 1993, 1994
38 * The Regents of the University of California. All rights reserved.
39 *
40 * Redistribution and use in source and binary forms, with or without
41 * modification, are permitted provided that the following conditions
42 * are met:
43 * 1. Redistributions of source code must retain the above copyright
44 * notice, this list of conditions and the following disclaimer.
45 * 2. Redistributions in binary form must reproduce the above copyright
46 * notice, this list of conditions and the following disclaimer in the
47 * documentation and/or other materials provided with the distribution.
48 * 3. Neither the name of the University nor the names of its contributors
49 * may be used to endorse or promote products derived from this software
50 * without specific prior written permission.
51 *
52 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
53 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
54 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
55 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
56 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
57 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
58 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
59 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
60 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
61 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
62 * SUCH DAMAGE.
63 */
64
65 #include <sys/cdefs.h>
66 #ifndef lint
67 __RCSID("$NetBSD: util.c,v 1.158 2013/02/19 23:29:15 dsl Exp $");
68 #endif /* not lint */
69
70 /*
71 * FTP User Program -- Misc support routines
72 */
73 #include <sys/param.h>
74 #include <sys/socket.h>
75 #include <sys/ioctl.h>
76 #include <sys/time.h>
77 #include <netinet/in.h>
78 #include <arpa/ftp.h>
79
80 #include <ctype.h>
81 #include <err.h>
82 #include <errno.h>
83 #include <fcntl.h>
84 #include <glob.h>
85 #include <signal.h>
86 #include <libgen.h>
87 #include <limits.h>
88 #include <locale.h>
89 #include <netdb.h>
90 #include <stdio.h>
91 #include <stdlib.h>
92 #include <string.h>
93 #include <termios.h>
94 #include <time.h>
95 #include <tzfile.h>
96 #include <unistd.h>
97
98 #include "ftp_var.h"
99
100 /*
101 * Connect to peer server and auto-login, if possible.
102 */
103 void
setpeer(int argc,char * argv[])104 setpeer(int argc, char *argv[])
105 {
106 char *host;
107 const char *port;
108
109 if (argc == 0)
110 goto usage;
111 if (connected) {
112 fprintf(ttyout, "Already connected to %s, use close first.\n",
113 hostname);
114 code = -1;
115 return;
116 }
117 if (argc < 2)
118 (void)another(&argc, &argv, "to");
119 if (argc < 2 || argc > 3) {
120 usage:
121 UPRINTF("usage: %s host-name [port]\n", argv[0]);
122 code = -1;
123 return;
124 }
125 if (gatemode)
126 port = gateport;
127 else
128 port = ftpport;
129 if (argc > 2)
130 port = argv[2];
131
132 if (gatemode) {
133 if (gateserver == NULL || *gateserver == '\0')
134 errx(1, "main: gateserver not defined");
135 host = hookup(gateserver, port);
136 } else
137 host = hookup(argv[1], port);
138
139 if (host) {
140 if (gatemode && verbose) {
141 fprintf(ttyout,
142 "Connecting via pass-through server %s\n",
143 gateserver);
144 }
145
146 connected = 1;
147 /*
148 * Set up defaults for FTP.
149 */
150 (void)strlcpy(typename, "ascii", sizeof(typename));
151 type = TYPE_A;
152 curtype = TYPE_A;
153 (void)strlcpy(formname, "non-print", sizeof(formname));
154 form = FORM_N;
155 (void)strlcpy(modename, "stream", sizeof(modename));
156 mode = MODE_S;
157 (void)strlcpy(structname, "file", sizeof(structname));
158 stru = STRU_F;
159 (void)strlcpy(bytename, "8", sizeof(bytename));
160 bytesize = 8;
161 if (autologin)
162 (void)ftp_login(argv[1], NULL, NULL);
163 }
164 }
165
166 static void
parse_feat(const char * fline)167 parse_feat(const char *fline)
168 {
169
170 /*
171 * work-around broken ProFTPd servers that can't
172 * even obey RFC 2389.
173 */
174 while (*fline && isspace((int)*fline))
175 fline++;
176
177 if (strcasecmp(fline, "MDTM") == 0)
178 features[FEAT_MDTM] = 1;
179 else if (strncasecmp(fline, "MLST", sizeof("MLST") - 1) == 0) {
180 features[FEAT_MLST] = 1;
181 } else if (strcasecmp(fline, "REST STREAM") == 0)
182 features[FEAT_REST_STREAM] = 1;
183 else if (strcasecmp(fline, "SIZE") == 0)
184 features[FEAT_SIZE] = 1;
185 else if (strcasecmp(fline, "TVFS") == 0)
186 features[FEAT_TVFS] = 1;
187 }
188
189 /*
190 * Determine the remote system type (SYST) and features (FEAT).
191 * Call after a successful login (i.e, connected = -1)
192 */
193 void
getremoteinfo(void)194 getremoteinfo(void)
195 {
196 int overbose, i;
197
198 overbose = verbose;
199 if (ftp_debug == 0)
200 verbose = -1;
201
202 /* determine remote system type */
203 if (command("SYST") == COMPLETE) {
204 if (overbose) {
205 int os_len = strcspn(reply_string + 4, " \r\n\t");
206 if (os_len > 1 && reply_string[4 + os_len - 1] == '.')
207 os_len--;
208 fprintf(ttyout, "Remote system type is %.*s.\n",
209 os_len, reply_string + 4);
210 }
211 /*
212 * Decide whether we should default to bninary.
213 * Traditionally checked for "215 UNIX Type: L8", but
214 * some printers report "Linux" ! so be more forgiving.
215 * In reality we probably almost never want text any more.
216 */
217 if (!strncasecmp(reply_string + 4, "unix", 4) ||
218 !strncasecmp(reply_string + 4, "linux", 5)) {
219 if (proxy)
220 unix_proxy = 1;
221 else
222 unix_server = 1;
223 /*
224 * Set type to 0 (not specified by user),
225 * meaning binary by default, but don't bother
226 * telling server. We can use binary
227 * for text files unless changed by the user.
228 */
229 type = 0;
230 (void)strlcpy(typename, "binary", sizeof(typename));
231 if (overbose)
232 fprintf(ttyout,
233 "Using %s mode to transfer files.\n",
234 typename);
235 } else {
236 if (proxy)
237 unix_proxy = 0;
238 else
239 unix_server = 0;
240 if (overbose &&
241 !strncmp(reply_string, "215 TOPS20", 10))
242 fputs(
243 "Remember to set tenex mode when transferring binary files from this machine.\n",
244 ttyout);
245 }
246 }
247
248 /* determine features (if any) */
249 for (i = 0; i < FEAT_max; i++)
250 features[i] = -1;
251 reply_callback = parse_feat;
252 if (command("FEAT") == COMPLETE) {
253 for (i = 0; i < FEAT_max; i++) {
254 if (features[i] == -1)
255 features[i] = 0;
256 }
257 features[FEAT_FEAT] = 1;
258 } else
259 features[FEAT_FEAT] = 0;
260 #ifndef NO_DEBUG
261 if (ftp_debug) {
262 #define DEBUG_FEAT(x) fprintf(ttyout, "features[" #x "] = %d\n", features[(x)])
263 DEBUG_FEAT(FEAT_FEAT);
264 DEBUG_FEAT(FEAT_MDTM);
265 DEBUG_FEAT(FEAT_MLST);
266 DEBUG_FEAT(FEAT_REST_STREAM);
267 DEBUG_FEAT(FEAT_SIZE);
268 DEBUG_FEAT(FEAT_TVFS);
269 #undef DEBUG_FEAT
270 }
271 #endif
272 reply_callback = NULL;
273
274 verbose = overbose;
275 }
276
277 /*
278 * Reset the various variables that indicate connection state back to
279 * disconnected settings.
280 * The caller is responsible for issuing any commands to the remote server
281 * to perform a clean shutdown before this is invoked.
282 */
283 void
cleanuppeer(void)284 cleanuppeer(void)
285 {
286
287 if (cout)
288 (void)fclose(cout);
289 cout = NULL;
290 connected = 0;
291 unix_server = 0;
292 unix_proxy = 0;
293 /*
294 * determine if anonftp was specifically set with -a
295 * (1), or implicitly set by auto_fetch() (2). in the
296 * latter case, disable after the current xfer
297 */
298 if (anonftp == 2)
299 anonftp = 0;
300 data = -1;
301 epsv4bad = 0;
302 epsv6bad = 0;
303 if (username)
304 free(username);
305 username = NULL;
306 if (!proxy)
307 macnum = 0;
308 }
309
310 /*
311 * Top-level signal handler for interrupted commands.
312 */
313 void
intr(int signo)314 intr(int signo)
315 {
316
317 sigint_raised = 1;
318 alarmtimer(0);
319 if (fromatty)
320 write(fileno(ttyout), "\n", 1);
321 siglongjmp(toplevel, 1);
322 }
323
324 /*
325 * Signal handler for lost connections; cleanup various elements of
326 * the connection state, and call cleanuppeer() to finish it off.
327 */
328 void
lostpeer(int dummy)329 lostpeer(int dummy)
330 {
331 int oerrno = errno;
332
333 alarmtimer(0);
334 if (connected) {
335 if (cout != NULL) {
336 (void)shutdown(fileno(cout), 1+1);
337 (void)fclose(cout);
338 cout = NULL;
339 }
340 if (data >= 0) {
341 (void)shutdown(data, 1+1);
342 (void)close(data);
343 data = -1;
344 }
345 connected = 0;
346 }
347 pswitch(1);
348 if (connected) {
349 if (cout != NULL) {
350 (void)shutdown(fileno(cout), 1+1);
351 (void)fclose(cout);
352 cout = NULL;
353 }
354 connected = 0;
355 }
356 proxflag = 0;
357 pswitch(0);
358 cleanuppeer();
359 errno = oerrno;
360 }
361
362
363 /*
364 * Login to remote host, using given username & password if supplied.
365 * Return non-zero if successful.
366 */
367 int
ftp_login(const char * host,const char * luser,const char * lpass)368 ftp_login(const char *host, const char *luser, const char *lpass)
369 {
370 char tmp[80];
371 char *fuser, *pass, *facct, *p;
372 char emptypass[] = "";
373 const char *errormsg;
374 int n, aflag, rval, nlen;
375
376 aflag = rval = 0;
377 fuser = pass = facct = NULL;
378 if (luser)
379 fuser = ftp_strdup(luser);
380 if (lpass)
381 pass = ftp_strdup(lpass);
382
383 DPRINTF("ftp_login: user `%s' pass `%s' host `%s'\n",
384 STRorNULL(fuser), STRorNULL(pass), STRorNULL(host));
385
386 /*
387 * Set up arguments for an anonymous FTP session, if necessary.
388 */
389 if (anonftp) {
390 FREEPTR(fuser);
391 fuser = ftp_strdup("anonymous"); /* as per RFC 1635 */
392 FREEPTR(pass);
393 pass = ftp_strdup(getoptionvalue("anonpass"));
394 }
395
396 if (ruserpass(host, &fuser, &pass, &facct) < 0) {
397 code = -1;
398 goto cleanup_ftp_login;
399 }
400
401 while (fuser == NULL) {
402 if (localname)
403 fprintf(ttyout, "Name (%s:%s): ", host, localname);
404 else
405 fprintf(ttyout, "Name (%s): ", host);
406 errormsg = NULL;
407 nlen = get_line(stdin, tmp, sizeof(tmp), &errormsg);
408 if (nlen < 0) {
409 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "login");
410 code = -1;
411 goto cleanup_ftp_login;
412 } else if (nlen == 0) {
413 fuser = ftp_strdup(localname);
414 } else {
415 fuser = ftp_strdup(tmp);
416 }
417 }
418
419 if (gatemode) {
420 char *nuser;
421 size_t len;
422
423 len = strlen(fuser) + 1 + strlen(host) + 1;
424 nuser = ftp_malloc(len);
425 (void)strlcpy(nuser, fuser, len);
426 (void)strlcat(nuser, "@", len);
427 (void)strlcat(nuser, host, len);
428 FREEPTR(fuser);
429 fuser = nuser;
430 }
431
432 n = command("USER %s", fuser);
433 if (n == CONTINUE) {
434 if (pass == NULL) {
435 p = getpass("Password: ");
436 if (p == NULL)
437 p = emptypass;
438 pass = ftp_strdup(p);
439 memset(p, 0, strlen(p));
440 }
441 n = command("PASS %s", pass);
442 memset(pass, 0, strlen(pass));
443 }
444 if (n == CONTINUE) {
445 aflag++;
446 if (facct == NULL) {
447 p = getpass("Account: ");
448 if (p == NULL)
449 p = emptypass;
450 facct = ftp_strdup(p);
451 memset(p, 0, strlen(p));
452 }
453 if (facct[0] == '\0') {
454 warnx("Login failed");
455 goto cleanup_ftp_login;
456 }
457 n = command("ACCT %s", facct);
458 memset(facct, 0, strlen(facct));
459 }
460 if ((n != COMPLETE) ||
461 (!aflag && facct != NULL && command("ACCT %s", facct) != COMPLETE)) {
462 warnx("Login failed");
463 goto cleanup_ftp_login;
464 }
465 rval = 1;
466 username = ftp_strdup(fuser);
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 updatelocalcwd();
481 updateremotecwd();
482
483 cleanup_ftp_login:
484 FREEPTR(fuser);
485 if (pass != NULL)
486 memset(pass, 0, strlen(pass));
487 FREEPTR(pass);
488 if (facct != NULL)
489 memset(facct, 0, strlen(facct));
490 FREEPTR(facct);
491 return (rval);
492 }
493
494 /*
495 * `another' gets another argument, and stores the new argc and argv.
496 * It reverts to the top level (via intr()) on EOF/error.
497 *
498 * Returns false if no new arguments have been added.
499 */
500 int
another(int * pargc,char *** pargv,const char * aprompt)501 another(int *pargc, char ***pargv, const char *aprompt)
502 {
503 const char *errormsg;
504 int ret, nlen;
505 size_t len;
506
507 len = strlen(line);
508 if (len >= sizeof(line) - 3) {
509 fputs("Sorry, arguments too long.\n", ttyout);
510 intr(0);
511 }
512 fprintf(ttyout, "(%s) ", aprompt);
513 line[len++] = ' ';
514 errormsg = NULL;
515 nlen = get_line(stdin, line + len, sizeof(line)-len, &errormsg);
516 if (nlen < 0) {
517 fprintf(ttyout, "%s; %s aborted.\n", errormsg, "operation");
518 intr(0);
519 }
520 len += nlen;
521 makeargv();
522 ret = margc > *pargc;
523 *pargc = margc;
524 *pargv = margv;
525 return (ret);
526 }
527
528 /*
529 * glob files given in argv[] from the remote server.
530 * if errbuf isn't NULL, store error messages there instead
531 * of writing to the screen.
532 */
533 char *
remglob(char * argv[],int doswitch,const char ** errbuf)534 remglob(char *argv[], int doswitch, const char **errbuf)
535 {
536 static char buf[MAXPATHLEN];
537 static FILE *ftemp = NULL;
538 static char **args;
539 char temp[MAXPATHLEN];
540 int oldverbose, oldhash, oldprogress, fd;
541 char *cp;
542 const char *rmode;
543 size_t len;
544
545 if (!mflag || !connected) {
546 if (!doglob)
547 args = NULL;
548 else {
549 if (ftemp) {
550 (void)fclose(ftemp);
551 ftemp = NULL;
552 }
553 }
554 return (NULL);
555 }
556 if (!doglob) {
557 if (args == NULL)
558 args = argv;
559 if ((cp = *++args) == NULL)
560 args = NULL;
561 return (cp);
562 }
563 if (ftemp == NULL) {
564 len = strlcpy(temp, tmpdir, sizeof(temp));
565 if (temp[len - 1] != '/')
566 (void)strlcat(temp, "/", sizeof(temp));
567 (void)strlcat(temp, TMPFILE, sizeof(temp));
568 if ((fd = mkstemp(temp)) < 0) {
569 warn("Unable to create temporary file `%s'", temp);
570 return (NULL);
571 }
572 close(fd);
573 oldverbose = verbose;
574 verbose = (errbuf != NULL) ? -1 : 0;
575 oldhash = hash;
576 oldprogress = progress;
577 hash = 0;
578 progress = 0;
579 if (doswitch)
580 pswitch(!proxy);
581 for (rmode = "w"; *++argv != NULL; rmode = "a")
582 recvrequest("NLST", temp, *argv, rmode, 0, 0);
583 if ((code / 100) != COMPLETE) {
584 if (errbuf != NULL)
585 *errbuf = reply_string;
586 }
587 if (doswitch)
588 pswitch(!proxy);
589 verbose = oldverbose;
590 hash = oldhash;
591 progress = oldprogress;
592 ftemp = fopen(temp, "r");
593 (void)unlink(temp);
594 if (ftemp == NULL) {
595 if (errbuf == NULL)
596 warnx("Can't find list of remote files");
597 else
598 *errbuf =
599 "Can't find list of remote files";
600 return (NULL);
601 }
602 }
603 if (fgets(buf, sizeof(buf), ftemp) == NULL) {
604 (void)fclose(ftemp);
605 ftemp = NULL;
606 return (NULL);
607 }
608 if ((cp = strchr(buf, '\n')) != NULL)
609 *cp = '\0';
610 return (buf);
611 }
612
613 /*
614 * Glob a local file name specification with the expectation of a single
615 * return value. Can't control multiple values being expanded from the
616 * expression, we return only the first.
617 * Returns NULL on error, or a pointer to a buffer containing the filename
618 * that's the caller's responsiblity to free(3) when finished with.
619 */
620 char *
globulize(const char * pattern)621 globulize(const char *pattern)
622 {
623 glob_t gl;
624 int flags;
625 char *p;
626
627 if (!doglob)
628 return (ftp_strdup(pattern));
629
630 flags = GLOB_BRACE|GLOB_NOCHECK|GLOB_TILDE;
631 memset(&gl, 0, sizeof(gl));
632 if (glob(pattern, flags, NULL, &gl) || gl.gl_pathc == 0) {
633 warnx("Glob pattern `%s' not found", pattern);
634 globfree(&gl);
635 return (NULL);
636 }
637 p = ftp_strdup(gl.gl_pathv[0]);
638 globfree(&gl);
639 return (p);
640 }
641
642 /*
643 * determine size of remote file
644 */
645 off_t
remotesize(const char * file,int noisy)646 remotesize(const char *file, int noisy)
647 {
648 int overbose, r;
649 off_t size;
650
651 overbose = verbose;
652 size = -1;
653 if (ftp_debug == 0)
654 verbose = -1;
655 if (! features[FEAT_SIZE]) {
656 if (noisy)
657 fprintf(ttyout,
658 "SIZE is not supported by remote server.\n");
659 goto cleanup_remotesize;
660 }
661 r = command("SIZE %s", file);
662 if (r == COMPLETE) {
663 char *cp, *ep;
664
665 cp = strchr(reply_string, ' ');
666 if (cp != NULL) {
667 cp++;
668 size = STRTOLL(cp, &ep, 10);
669 if (*ep != '\0' && !isspace((unsigned char)*ep))
670 size = -1;
671 }
672 } else {
673 if (r == ERROR && code == 500 && features[FEAT_SIZE] == -1)
674 features[FEAT_SIZE] = 0;
675 if (noisy && ftp_debug == 0) {
676 fputs(reply_string, ttyout);
677 putc('\n', ttyout);
678 }
679 }
680 cleanup_remotesize:
681 verbose = overbose;
682 return (size);
683 }
684
685 /*
686 * determine last modification time (in GMT) of remote file
687 */
688 time_t
remotemodtime(const char * file,int noisy)689 remotemodtime(const char *file, int noisy)
690 {
691 int overbose, ocode, r;
692 time_t rtime;
693
694 overbose = verbose;
695 ocode = code;
696 rtime = -1;
697 if (ftp_debug == 0)
698 verbose = -1;
699 if (! features[FEAT_MDTM]) {
700 if (noisy)
701 fprintf(ttyout,
702 "MDTM is not supported by remote server.\n");
703 goto cleanup_parse_time;
704 }
705 r = command("MDTM %s", file);
706 if (r == COMPLETE) {
707 struct tm timebuf;
708 char *timestr, *frac;
709
710 /*
711 * time-val = 14DIGIT [ "." 1*DIGIT ]
712 * YYYYMMDDHHMMSS[.sss]
713 * mdtm-response = "213" SP time-val CRLF / error-response
714 */
715 timestr = reply_string + 4;
716
717 /*
718 * parse fraction.
719 * XXX: ignored for now
720 */
721 frac = strchr(timestr, '\r');
722 if (frac != NULL)
723 *frac = '\0';
724 frac = strchr(timestr, '.');
725 if (frac != NULL)
726 *frac++ = '\0';
727 if (strlen(timestr) == 15 && strncmp(timestr, "191", 3) == 0) {
728 /*
729 * XXX: Workaround for lame ftpd's that return
730 * `19100' instead of `2000'
731 */
732 fprintf(ttyout,
733 "Y2K warning! Incorrect time-val `%s' received from server.\n",
734 timestr);
735 timestr++;
736 timestr[0] = '2';
737 timestr[1] = '0';
738 fprintf(ttyout, "Converted to `%s'\n", timestr);
739 }
740 memset(&timebuf, 0, sizeof(timebuf));
741 if (strlen(timestr) != 14 ||
742 (strptime(timestr, "%Y%m%d%H%M%S", &timebuf) == NULL)) {
743 bad_parse_time:
744 fprintf(ttyout, "Can't parse time `%s'.\n", timestr);
745 goto cleanup_parse_time;
746 }
747 timebuf.tm_isdst = -1;
748 rtime = timegm(&timebuf);
749 if (rtime == -1) {
750 if (noisy || ftp_debug != 0)
751 goto bad_parse_time;
752 else
753 goto cleanup_parse_time;
754 } else {
755 DPRINTF("remotemodtime: parsed time `%s' as " LLF
756 ", %s",
757 timestr, (LLT)rtime,
758 rfc2822time(localtime(&rtime)));
759 }
760 } else {
761 if (r == ERROR && code == 500 && features[FEAT_MDTM] == -1)
762 features[FEAT_MDTM] = 0;
763 if (noisy && ftp_debug == 0) {
764 fputs(reply_string, ttyout);
765 putc('\n', ttyout);
766 }
767 }
768 cleanup_parse_time:
769 verbose = overbose;
770 if (rtime == -1)
771 code = ocode;
772 return (rtime);
773 }
774
775 /*
776 * Format tm in an RFC 2822 compatible manner, with a trailing \n.
777 * Returns a pointer to a static string containing the result.
778 */
779 const char *
rfc2822time(const struct tm * tm)780 rfc2822time(const struct tm *tm)
781 {
782 static char result[50];
783
784 if (strftime(result, sizeof(result),
785 "%a, %d %b %Y %H:%M:%S %z\n", tm) == 0)
786 errx(1, "Can't convert RFC 2822 time: buffer too small");
787 return result;
788 }
789
790 /*
791 * Parse HTTP-date as per RFC 2616.
792 * Return a pointer to the next character of the consumed date string,
793 * or NULL if failed.
794 */
795 const char *
parse_rfc2616time(struct tm * parsed,const char * httpdate)796 parse_rfc2616time(struct tm *parsed, const char *httpdate)
797 {
798 const char *t;
799 const char *curlocale;
800
801 /* The representation of %a depends on the current locale. */
802 curlocale = setlocale(LC_TIME, NULL);
803 (void)setlocale(LC_TIME, "C");
804 /* RFC 1123 */
805 if ((t = strptime(httpdate, "%a, %d %b %Y %H:%M:%S GMT", parsed)) ||
806 /* RFC 850 */
807 (t = strptime(httpdate, "%a, %d-%b-%y %H:%M:%S GMT", parsed)) ||
808 /* asctime */
809 (t = strptime(httpdate, "%a, %b %d %H:%M:%S %Y", parsed))) {
810 ; /* do nothing */
811 }
812 (void)setlocale(LC_TIME, curlocale);
813 return t;
814 }
815
816 /*
817 * Update global `localcwd', which contains the state of the local cwd
818 */
819 void
updatelocalcwd(void)820 updatelocalcwd(void)
821 {
822
823 if (getcwd(localcwd, sizeof(localcwd)) == NULL)
824 localcwd[0] = '\0';
825 DPRINTF("updatelocalcwd: got `%s'\n", localcwd);
826 }
827
828 /*
829 * Update global `remotecwd', which contains the state of the remote cwd
830 */
831 void
updateremotecwd(void)832 updateremotecwd(void)
833 {
834 int overbose, ocode;
835 size_t i;
836 char *cp;
837
838 overbose = verbose;
839 ocode = code;
840 if (ftp_debug == 0)
841 verbose = -1;
842 if (command("PWD") != COMPLETE)
843 goto badremotecwd;
844 cp = strchr(reply_string, ' ');
845 if (cp == NULL || cp[0] == '\0' || cp[1] != '"')
846 goto badremotecwd;
847 cp += 2;
848 for (i = 0; *cp && i < sizeof(remotecwd) - 1; i++, cp++) {
849 if (cp[0] == '"') {
850 if (cp[1] == '"')
851 cp++;
852 else
853 break;
854 }
855 remotecwd[i] = *cp;
856 }
857 remotecwd[i] = '\0';
858 DPRINTF("updateremotecwd: got `%s'\n", remotecwd);
859 goto cleanupremotecwd;
860 badremotecwd:
861 remotecwd[0]='\0';
862 cleanupremotecwd:
863 verbose = overbose;
864 code = ocode;
865 }
866
867 /*
868 * Ensure file is in or under dir.
869 * Returns 1 if so, 0 if not (or an error occurred).
870 */
871 int
fileindir(const char * file,const char * dir)872 fileindir(const char *file, const char *dir)
873 {
874 char parentdirbuf[PATH_MAX+1], *parentdir;
875 char realdir[PATH_MAX+1];
876 size_t dirlen;
877
878 /* determine parent directory of file */
879 (void)strlcpy(parentdirbuf, file, sizeof(parentdirbuf));
880 parentdir = dirname(parentdirbuf);
881 if (strcmp(parentdir, ".") == 0)
882 return 1; /* current directory is ok */
883
884 /* find the directory */
885 if (realpath(parentdir, realdir) == NULL) {
886 warn("Unable to determine real path of `%s'", parentdir);
887 return 0;
888 }
889 if (realdir[0] != '/') /* relative result is ok */
890 return 1;
891 dirlen = strlen(dir);
892 if (strncmp(realdir, dir, dirlen) == 0 &&
893 (realdir[dirlen] == '/' || realdir[dirlen] == '\0'))
894 return 1;
895 return 0;
896 }
897
898 /*
899 * List words in stringlist, vertically arranged
900 */
901 void
list_vertical(StringList * sl)902 list_vertical(StringList *sl)
903 {
904 size_t i, j;
905 size_t columns, lines;
906 char *p;
907 size_t w, width;
908
909 width = 0;
910
911 for (i = 0 ; i < sl->sl_cur ; i++) {
912 w = strlen(sl->sl_str[i]);
913 if (w > width)
914 width = w;
915 }
916 width = (width + 8) &~ 7;
917
918 columns = ttywidth / width;
919 if (columns == 0)
920 columns = 1;
921 lines = (sl->sl_cur + columns - 1) / columns;
922 for (i = 0; i < lines; i++) {
923 for (j = 0; j < columns; j++) {
924 p = sl->sl_str[j * lines + i];
925 if (p)
926 fputs(p, ttyout);
927 if (j * lines + i + lines >= sl->sl_cur) {
928 putc('\n', ttyout);
929 break;
930 }
931 if (p) {
932 w = strlen(p);
933 while (w < width) {
934 w = (w + 8) &~ 7;
935 (void)putc('\t', ttyout);
936 }
937 }
938 }
939 }
940 }
941
942 /*
943 * Update the global ttywidth value, using TIOCGWINSZ.
944 */
945 void
setttywidth(int a)946 setttywidth(int a)
947 {
948 struct winsize winsize;
949 int oerrno = errno;
950
951 if (ioctl(fileno(ttyout), TIOCGWINSZ, &winsize) != -1 &&
952 winsize.ws_col != 0)
953 ttywidth = winsize.ws_col;
954 else
955 ttywidth = 80;
956 errno = oerrno;
957 }
958
959 /*
960 * Change the rate limit up (SIGUSR1) or down (SIGUSR2)
961 */
962 void
crankrate(int sig)963 crankrate(int sig)
964 {
965
966 switch (sig) {
967 case SIGUSR1:
968 if (rate_get)
969 rate_get += rate_get_incr;
970 if (rate_put)
971 rate_put += rate_put_incr;
972 break;
973 case SIGUSR2:
974 if (rate_get && rate_get > rate_get_incr)
975 rate_get -= rate_get_incr;
976 if (rate_put && rate_put > rate_put_incr)
977 rate_put -= rate_put_incr;
978 break;
979 default:
980 err(1, "crankrate invoked with unknown signal: %d", sig);
981 }
982 }
983
984
985 /*
986 * Setup or cleanup EditLine structures
987 */
988 #ifndef NO_EDITCOMPLETE
989 void
controlediting(void)990 controlediting(void)
991 {
992 if (editing && el == NULL && hist == NULL) {
993 HistEvent ev;
994 int editmode;
995
996 el = el_init(getprogname(), stdin, ttyout, stderr);
997 /* init editline */
998 hist = history_init(); /* init the builtin history */
999 history(hist, &ev, H_SETSIZE, 100);/* remember 100 events */
1000 el_set(el, EL_HIST, history, hist); /* use history */
1001
1002 el_set(el, EL_EDITOR, "emacs"); /* default editor is emacs */
1003 el_set(el, EL_PROMPT, prompt); /* set the prompt functions */
1004 el_set(el, EL_RPROMPT, rprompt);
1005
1006 /* add local file completion, bind to TAB */
1007 el_set(el, EL_ADDFN, "ftp-complete",
1008 "Context sensitive argument completion",
1009 complete);
1010 el_set(el, EL_BIND, "^I", "ftp-complete", NULL);
1011 el_source(el, NULL); /* read ~/.editrc */
1012 if ((el_get(el, EL_EDITMODE, &editmode) != -1) && editmode == 0)
1013 editing = 0; /* the user doesn't want editing,
1014 * so disable, and let statement
1015 * below cleanup */
1016 else
1017 el_set(el, EL_SIGNAL, 1);
1018 }
1019 if (!editing) {
1020 if (hist) {
1021 history_end(hist);
1022 hist = NULL;
1023 }
1024 if (el) {
1025 el_end(el);
1026 el = NULL;
1027 }
1028 }
1029 }
1030 #endif /* !NO_EDITCOMPLETE */
1031
1032 /*
1033 * Convert the string `arg' to an int, which may have an optional SI suffix
1034 * (`b', `k', `m', `g'). Returns the number for success, -1 otherwise.
1035 */
1036 int
strsuftoi(const char * arg)1037 strsuftoi(const char *arg)
1038 {
1039 char *cp;
1040 long val;
1041
1042 if (!isdigit((unsigned char)arg[0]))
1043 return (-1);
1044
1045 val = strtol(arg, &cp, 10);
1046 if (cp != NULL) {
1047 if (cp[0] != '\0' && cp[1] != '\0')
1048 return (-1);
1049 switch (tolower((unsigned char)cp[0])) {
1050 case '\0':
1051 case 'b':
1052 break;
1053 case 'k':
1054 val <<= 10;
1055 break;
1056 case 'm':
1057 val <<= 20;
1058 break;
1059 case 'g':
1060 val <<= 30;
1061 break;
1062 default:
1063 return (-1);
1064 }
1065 }
1066 if (val < 0 || val > INT_MAX)
1067 return (-1);
1068
1069 return (val);
1070 }
1071
1072 /*
1073 * Set up socket buffer sizes before a connection is made.
1074 */
1075 void
setupsockbufsize(int sock)1076 setupsockbufsize(int sock)
1077 {
1078 socklen_t slen;
1079
1080 if (0 == rcvbuf_size) {
1081 slen = sizeof(rcvbuf_size);
1082 if (getsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1083 (void *)&rcvbuf_size, &slen) == -1)
1084 err(1, "Unable to determine rcvbuf size");
1085 if (rcvbuf_size <= 0)
1086 rcvbuf_size = 8 * 1024;
1087 if (rcvbuf_size > 8 * 1024 * 1024)
1088 rcvbuf_size = 8 * 1024 * 1024;
1089 DPRINTF("setupsockbufsize: rcvbuf_size determined as %d\n",
1090 rcvbuf_size);
1091 }
1092 if (0 == sndbuf_size) {
1093 slen = sizeof(sndbuf_size);
1094 if (getsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1095 (void *)&sndbuf_size, &slen) == -1)
1096 err(1, "Unable to determine sndbuf size");
1097 if (sndbuf_size <= 0)
1098 sndbuf_size = 8 * 1024;
1099 if (sndbuf_size > 8 * 1024 * 1024)
1100 sndbuf_size = 8 * 1024 * 1024;
1101 DPRINTF("setupsockbufsize: sndbuf_size determined as %d\n",
1102 sndbuf_size);
1103 }
1104
1105 if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF,
1106 (void *)&sndbuf_size, sizeof(sndbuf_size)) == -1)
1107 warn("Unable to set sndbuf size %d", sndbuf_size);
1108
1109 if (setsockopt(sock, SOL_SOCKET, SO_RCVBUF,
1110 (void *)&rcvbuf_size, sizeof(rcvbuf_size)) == -1)
1111 warn("Unable to set rcvbuf size %d", rcvbuf_size);
1112 }
1113
1114 /*
1115 * Copy characters from src into dst, \ quoting characters that require it
1116 */
1117 void
ftpvis(char * dst,size_t dstlen,const char * src,size_t srclen)1118 ftpvis(char *dst, size_t dstlen, const char *src, size_t srclen)
1119 {
1120 size_t di, si;
1121
1122 di = si = 0;
1123 while (src[si] != '\0' && di < dstlen && si < srclen) {
1124 switch (src[si]) {
1125 case '\\':
1126 case ' ':
1127 case '\t':
1128 case '\r':
1129 case '\n':
1130 case '"':
1131 /*
1132 * Need room for two characters and NUL, avoiding
1133 * incomplete escape sequences at end of dst.
1134 */
1135 if (di >= dstlen - 3)
1136 break;
1137 dst[di++] = '\\';
1138 /* FALLTHROUGH */
1139 default:
1140 dst[di] = src[si++];
1141 if (di < dstlen)
1142 di++;
1143 }
1144 }
1145 dst[di] = '\0';
1146 }
1147
1148 /*
1149 * Copy src into buf (which is len bytes long), expanding % sequences.
1150 */
1151 void
formatbuf(char * buf,size_t len,const char * src)1152 formatbuf(char *buf, size_t len, const char *src)
1153 {
1154 const char *p, *p2, *q;
1155 size_t i;
1156 int op, updirs, pdirs;
1157
1158 #define ADDBUF(x) do { \
1159 if (i >= len - 1) \
1160 goto endbuf; \
1161 buf[i++] = (x); \
1162 } while (0)
1163
1164 p = src;
1165 for (i = 0; *p; p++) {
1166 if (*p != '%') {
1167 ADDBUF(*p);
1168 continue;
1169 }
1170 p++;
1171
1172 switch (op = *p) {
1173
1174 case '/':
1175 case '.':
1176 case 'c':
1177 p2 = connected ? remotecwd : "";
1178 updirs = pdirs = 0;
1179
1180 /* option to determine fixed # of dirs from path */
1181 if (op == '.' || op == 'c') {
1182 int skip;
1183
1184 q = p2;
1185 while (*p2) /* calc # of /'s */
1186 if (*p2++ == '/')
1187 updirs++;
1188 if (p[1] == '0') { /* print <x> or ... */
1189 pdirs = 1;
1190 p++;
1191 }
1192 if (p[1] >= '1' && p[1] <= '9') {
1193 /* calc # to skip */
1194 skip = p[1] - '0';
1195 p++;
1196 } else
1197 skip = 1;
1198
1199 updirs -= skip;
1200 while (skip-- > 0) {
1201 while ((p2 > q) && (*p2 != '/'))
1202 p2--; /* back up */
1203 if (skip && p2 > q)
1204 p2--;
1205 }
1206 if (*p2 == '/' && p2 != q)
1207 p2++;
1208 }
1209
1210 if (updirs > 0 && pdirs) {
1211 if (i >= len - 5)
1212 break;
1213 if (op == '.') {
1214 ADDBUF('.');
1215 ADDBUF('.');
1216 ADDBUF('.');
1217 } else {
1218 ADDBUF('/');
1219 ADDBUF('<');
1220 if (updirs > 9) {
1221 ADDBUF('9');
1222 ADDBUF('+');
1223 } else
1224 ADDBUF('0' + updirs);
1225 ADDBUF('>');
1226 }
1227 }
1228 for (; *p2; p2++)
1229 ADDBUF(*p2);
1230 break;
1231
1232 case 'M':
1233 case 'm':
1234 for (p2 = connected && hostname ? hostname : "-";
1235 *p2 ; p2++) {
1236 if (op == 'm' && *p2 == '.')
1237 break;
1238 ADDBUF(*p2);
1239 }
1240 break;
1241
1242 case 'n':
1243 for (p2 = connected ? username : "-"; *p2 ; p2++)
1244 ADDBUF(*p2);
1245 break;
1246
1247 case '%':
1248 ADDBUF('%');
1249 break;
1250
1251 default: /* display unknown codes literally */
1252 ADDBUF('%');
1253 ADDBUF(op);
1254 break;
1255
1256 }
1257 }
1258 endbuf:
1259 buf[i] = '\0';
1260 }
1261
1262 /*
1263 * Determine if given string is an IPv6 address or not.
1264 * Return 1 for yes, 0 for no
1265 */
1266 int
isipv6addr(const char * addr)1267 isipv6addr(const char *addr)
1268 {
1269 int rv = 0;
1270 #ifdef INET6
1271 struct addrinfo hints, *res;
1272
1273 memset(&hints, 0, sizeof(hints));
1274 hints.ai_family = AF_INET6;
1275 hints.ai_socktype = SOCK_DGRAM; /*dummy*/
1276 hints.ai_flags = AI_NUMERICHOST;
1277 if (getaddrinfo(addr, "0", &hints, &res) != 0)
1278 rv = 0;
1279 else {
1280 rv = 1;
1281 freeaddrinfo(res);
1282 }
1283 DPRINTF("isipv6addr: got %d for %s\n", rv, addr);
1284 #endif
1285 return (rv == 1) ? 1 : 0;
1286 }
1287
1288 /*
1289 * Read a line from the FILE stream into buf/buflen using fgets(), so up
1290 * to buflen-1 chars will be read and the result will be NUL terminated.
1291 * If the line has a trailing newline it will be removed.
1292 * If the line is too long, excess characters will be read until
1293 * newline/EOF/error.
1294 * If EOF/error occurs or a too-long line is encountered and errormsg
1295 * isn't NULL, it will be changed to a description of the problem.
1296 * (The EOF message has a leading \n for cosmetic purposes).
1297 * Returns:
1298 * >=0 length of line (excluding trailing newline) if all ok
1299 * -1 error occurred
1300 * -2 EOF encountered
1301 * -3 line was too long
1302 */
1303 int
get_line(FILE * stream,char * buf,size_t buflen,const char ** errormsg)1304 get_line(FILE *stream, char *buf, size_t buflen, const char **errormsg)
1305 {
1306 int rv, ch;
1307 size_t len;
1308
1309 if (fgets(buf, buflen, stream) == NULL) {
1310 if (feof(stream)) { /* EOF */
1311 rv = -2;
1312 if (errormsg)
1313 *errormsg = "\nEOF received";
1314 } else { /* error */
1315 rv = -1;
1316 if (errormsg)
1317 *errormsg = "Error encountered";
1318 }
1319 clearerr(stream);
1320 return rv;
1321 }
1322 len = strlen(buf);
1323 if (buf[len-1] == '\n') { /* clear any trailing newline */
1324 buf[--len] = '\0';
1325 } else if (len == buflen-1) { /* line too long */
1326 while ((ch = getchar()) != '\n' && ch != EOF)
1327 continue;
1328 if (errormsg)
1329 *errormsg = "Input line is too long";
1330 clearerr(stream);
1331 return -3;
1332 }
1333 if (errormsg)
1334 *errormsg = NULL;
1335 return len;
1336 }
1337
1338 /*
1339 * Internal version of connect(2); sets socket buffer sizes,
1340 * binds to a specific local address (if set), and
1341 * supports a connection timeout using a non-blocking connect(2) with
1342 * a poll(2).
1343 * Socket fcntl flags are temporarily updated to include O_NONBLOCK;
1344 * these will not be reverted on connection failure.
1345 * Returns 0 on success, or -1 upon failure (with an appropriate
1346 * error message displayed.)
1347 */
1348 int
ftp_connect(int sock,const struct sockaddr * name,socklen_t namelen,int pe)1349 ftp_connect(int sock, const struct sockaddr *name, socklen_t namelen, int pe)
1350 {
1351 int flags, rv, timeout, error;
1352 socklen_t slen;
1353 struct timeval endtime, now, td;
1354 struct pollfd pfd[1];
1355 char hname[NI_MAXHOST];
1356 char sname[NI_MAXSERV];
1357
1358 setupsockbufsize(sock);
1359 if (getnameinfo(name, namelen,
1360 hname, sizeof(hname), sname, sizeof(sname),
1361 NI_NUMERICHOST | NI_NUMERICSERV) != 0) {
1362 strlcpy(hname, "?", sizeof(hname));
1363 strlcpy(sname, "?", sizeof(sname));
1364 }
1365
1366 if (bindai != NULL) { /* bind to specific addr */
1367 struct addrinfo *ai;
1368
1369 for (ai = bindai; ai != NULL; ai = ai->ai_next) {
1370 if (ai->ai_family == name->sa_family)
1371 break;
1372 }
1373 if (ai == NULL)
1374 ai = bindai;
1375 if (bind(sock, ai->ai_addr, ai->ai_addrlen) == -1) {
1376 char bname[NI_MAXHOST];
1377 int saveerr;
1378
1379 saveerr = errno;
1380 if (getnameinfo(ai->ai_addr, ai->ai_addrlen,
1381 bname, sizeof(bname), NULL, 0, NI_NUMERICHOST) != 0)
1382 strlcpy(bname, "?", sizeof(bname));
1383 errno = saveerr;
1384 warn("Can't bind to `%s'", bname);
1385 return -1;
1386 }
1387 }
1388
1389 /* save current socket flags */
1390 if ((flags = fcntl(sock, F_GETFL, 0)) == -1) {
1391 warn("Can't %s socket flags for connect to `%s:%s'",
1392 "save", hname, sname);
1393 return -1;
1394 }
1395 /* set non-blocking connect */
1396 if (fcntl(sock, F_SETFL, flags | O_NONBLOCK) == -1) {
1397 warn("Can't set socket non-blocking for connect to `%s:%s'",
1398 hname, sname);
1399 return -1;
1400 }
1401
1402 /* NOTE: we now must restore socket flags on successful exit */
1403
1404 pfd[0].fd = sock;
1405 pfd[0].events = POLLIN|POLLOUT;
1406
1407 if (quit_time > 0) { /* want a non default timeout */
1408 (void)gettimeofday(&endtime, NULL);
1409 endtime.tv_sec += quit_time; /* determine end time */
1410 }
1411
1412 rv = connect(sock, name, namelen); /* inititate the connection */
1413 if (rv == -1) { /* connection error */
1414 if (errno != EINPROGRESS) { /* error isn't "please wait" */
1415 if (pe || (errno != EHOSTUNREACH))
1416 connecterror:
1417 warn("Can't connect to `%s:%s'", hname, sname);
1418 return -1;
1419 }
1420
1421 /* connect EINPROGRESS; wait */
1422 do {
1423 if (quit_time > 0) { /* determine timeout */
1424 (void)gettimeofday(&now, NULL);
1425 timersub(&endtime, &now, &td);
1426 timeout = td.tv_sec * 1000 + td.tv_usec/1000;
1427 if (timeout < 0)
1428 timeout = 0;
1429 } else {
1430 timeout = INFTIM;
1431 }
1432 pfd[0].revents = 0;
1433 rv = ftp_poll(pfd, 1, timeout);
1434 /* loop until poll ! EINTR */
1435 } while (rv == -1 && errno == EINTR);
1436
1437 if (rv == 0) { /* poll (connect) timed out */
1438 errno = ETIMEDOUT;
1439 goto connecterror;
1440 }
1441
1442 if (rv == -1) { /* poll error */
1443 goto connecterror;
1444 } else if (pfd[0].revents & (POLLIN|POLLOUT)) {
1445 slen = sizeof(error); /* OK, or pending error */
1446 if (getsockopt(sock, SOL_SOCKET, SO_ERROR,
1447 &error, &slen) == -1) {
1448 /* Solaris pending error */
1449 goto connecterror;
1450 } else if (error != 0) {
1451 errno = error; /* BSD pending error */
1452 goto connecterror;
1453 }
1454 } else {
1455 errno = EBADF; /* this shouldn't happen ... */
1456 goto connecterror;
1457 }
1458 }
1459
1460 if (fcntl(sock, F_SETFL, flags) == -1) {
1461 /* restore socket flags */
1462 warn("Can't %s socket flags for connect to `%s:%s'",
1463 "restore", hname, sname);
1464 return -1;
1465 }
1466 return 0;
1467 }
1468
1469 /*
1470 * Internal version of listen(2); sets socket buffer sizes first.
1471 */
1472 int
ftp_listen(int sock,int backlog)1473 ftp_listen(int sock, int backlog)
1474 {
1475
1476 setupsockbufsize(sock);
1477 return (listen(sock, backlog));
1478 }
1479
1480 /*
1481 * Internal version of poll(2), to allow reimplementation by select(2)
1482 * on platforms without the former.
1483 */
1484 int
ftp_poll(struct pollfd * fds,int nfds,int timeout)1485 ftp_poll(struct pollfd *fds, int nfds, int timeout)
1486 {
1487 return poll(fds, nfds, timeout);
1488 }
1489
1490 /*
1491 * malloc() with inbuilt error checking
1492 */
1493 void *
ftp_malloc(size_t size)1494 ftp_malloc(size_t size)
1495 {
1496 void *p;
1497
1498 p = malloc(size);
1499 if (p == NULL)
1500 err(1, "Unable to allocate %ld bytes of memory", (long)size);
1501 return (p);
1502 }
1503
1504 /*
1505 * sl_init() with inbuilt error checking
1506 */
1507 StringList *
ftp_sl_init(void)1508 ftp_sl_init(void)
1509 {
1510 StringList *p;
1511
1512 p = sl_init();
1513 if (p == NULL)
1514 err(1, "Unable to allocate memory for stringlist");
1515 return (p);
1516 }
1517
1518 /*
1519 * sl_add() with inbuilt error checking
1520 */
1521 void
ftp_sl_add(StringList * sl,char * i)1522 ftp_sl_add(StringList *sl, char *i)
1523 {
1524
1525 if (sl_add(sl, i) == -1)
1526 err(1, "Unable to add `%s' to stringlist", i);
1527 }
1528
1529 /*
1530 * strdup() with inbuilt error checking
1531 */
1532 char *
ftp_strdup(const char * str)1533 ftp_strdup(const char *str)
1534 {
1535 char *s;
1536
1537 if (str == NULL)
1538 errx(1, "ftp_strdup: called with NULL argument");
1539 s = strdup(str);
1540 if (s == NULL)
1541 err(1, "Unable to allocate memory for string copy");
1542 return (s);
1543 }
1544