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