files.c 24.7 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/* $Id$ */
Chris Allegretta's avatar
Chris Allegretta committed
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
/**************************************************************************
 *   files.c                                                              *
 *                                                                        *
 *   Copyright (C) 1999 Chris Allegretta                                  *
 *   This program is free software; you can redistribute it and/or modify *
 *   it under the terms of the GNU General Public License as published by *
 *   the Free Software Foundation; either version 1, or (at your option)  *
 *   any later version.                                                   *
 *                                                                        *
 *   This program is distributed in the hope that it will be useful,      *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of       *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the        *
 *   GNU General Public License for more details.                         *
 *                                                                        *
 *   You should have received a copy of the GNU General Public License    *
 *   along with this program; if not, write to the Free Software          *
 *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.            *
 *                                                                        *
 **************************************************************************/

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
Chris Allegretta's avatar
Chris Allegretta committed
30
31
#include <ctype.h>
#include <dirent.h>
Chris Allegretta's avatar
Chris Allegretta committed
32
33
34
35

#include "config.h"
#include "proto.h"
#include "nano.h"
Chris Allegretta's avatar
Chris Allegretta committed
36

Chris Allegretta's avatar
Chris Allegretta committed
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
#ifndef NANO_SMALL
#include <libintl.h>
#define _(string) gettext(string)
#else
#define _(string) (string)
#endif

/* Load file into edit buffer - takes data from file struct */
void load_file(void)
{
    current = fileage;
    wmove(edit, current_y, current_x);
}

/* What happens when there is no file to open? aiee! */
void new_file(void)
{
    fileage = nmalloc(sizeof(filestruct));
    fileage->data = nmalloc(1);
    strcpy(fileage->data, "");
    fileage->prev = NULL;
    fileage->next = NULL;
    fileage->lineno = 1;
    filebot = fileage;
    edittop = fileage;
    editbot = fileage;
    current = fileage;
    totlines = 1;
    UNSET(VIEW_MODE);
}


int read_byte(int fd, char *filename, char *input)
{
    static char buf[BUFSIZ];
    static int index = 0;
    static int size = 0;

    if (index == size) {
	index = 0;
	size = read(fd, buf, BUFSIZ);
	if (size == -1) {
	    clear();
	    refresh();
	    resetty();
	    endwin();
	    perror(filename);
	}
	if (!size)
	    return 0;
    }
    *input = buf[index++];
    return 1;
}

filestruct *read_line(char *buf, filestruct * prev, int *line1ins)
{
    filestruct *fileptr;

    fileptr = nmalloc(sizeof(filestruct));
    fileptr->data = nmalloc(strlen(buf) + 2);
    strcpy(fileptr->data, buf);

    if (*line1ins) {
	/* Special case, insert with cursor on 1st line. */
	fileptr->prev = NULL;
	fileptr->next = fileage;
	fileptr->lineno = 1;
	*line1ins = 0;
	/* If we're inserting into the first line of the file, then
	   we want to make sure that our edit buffer stays on the
	   first line (and that fileage stays up to date!) */
	fileage = fileptr;
	edittop = fileptr;
    } else if (fileage == NULL) {
	fileage = fileptr;
	fileage->lineno = 1;
	fileage->next = fileage->prev = NULL;
	fileptr = filebot = fileage;
    } else if (prev) {
	fileptr->prev = prev;
	fileptr->next = NULL;
	fileptr->lineno = prev->lineno + 1;
	prev->next = fileptr;
    } else {
	die(_("read_line: not on first line and prev is NULL"));
    }

    return fileptr;
}


int read_file(int fd, char *filename)
{
131
    long size, num_lines = 0, linetemp = 0;
Chris Allegretta's avatar
Chris Allegretta committed
132
133
134
135
136
137
138
    char input[2];		/* buffer */
    char *buf;
    long i = 0, bufx = 128;
    filestruct *fileptr = current, *tmp = NULL;
    int line1ins = 0;

    buf = nmalloc(bufx);
139
    buf[0] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154

    if (fileptr != NULL && fileptr->prev != NULL) {
	fileptr = fileptr->prev;
	tmp = fileptr;
    } else if (fileptr != NULL && fileptr->prev == NULL) {
	tmp = fileage;
	current = fileage;
	line1ins = 1;
    }
    input[1] = 0;
    /* Read the entire file into file struct */
    while ((size = read_byte(fd, filename, input)) > 0) {
	linetemp = 0;
	if (input[0] == '\n') {
	    fileptr = read_line(buf, fileptr, &line1ins);
155
	    num_lines++;
Chris Allegretta's avatar
Chris Allegretta committed
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
	    buf[0] = 0;
	    i = 0;
	} else {
	    /* Now we allocate a bigger buffer 128 characters at a time.
	       If we allocate a lot of space for one line, we may indeed 
	       have to use a buffer this big later on, so we don't
	       decrease it at all.  We do free it at the end though. */

	    if (i >= bufx - 1) {
		buf = nrealloc(buf, bufx + 128);
		bufx += 128;
	    }
	    buf[i] = input[0];
	    buf[i + 1] = 0;
	    i++;
	}
	totsize += size;
    }

    /* Did we not get a newline but still have stuff to do? */
    if (buf[0]) {
	fileptr = read_line(buf, fileptr, &line1ins);
178
	num_lines++;
Chris Allegretta's avatar
Chris Allegretta committed
179
180
181
	buf[0] = 0;
    }
    /* Did we even GET a file? */
182
    if (totsize == 0 || fileptr == NULL) {
Chris Allegretta's avatar
Chris Allegretta committed
183
	new_file();
184
	statusbar(_("Read %d lines"), num_lines);
Chris Allegretta's avatar
Chris Allegretta committed
185
186
	return 1;
    }
Robert Siemborski's avatar
Robert Siemborski committed
187

Chris Allegretta's avatar
Chris Allegretta committed
188
189
190
191
192
193
194
195
    if (current != NULL) {
	fileptr->next = current;
	current->prev = fileptr;
	renumber(current);
	current_x = 0;
	placewewant = 0;
    } else if (fileptr->next == NULL) {
	filebot = fileptr;
Robert Siemborski's avatar
Robert Siemborski committed
196
197
198
	new_magicline();

	/* Update the edit buffer */
Chris Allegretta's avatar
Chris Allegretta committed
199
200
	load_file();
    }
201
202
    statusbar(_("Read %d lines"), num_lines);
    totlines += num_lines;
Chris Allegretta's avatar
Chris Allegretta committed
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232

    free(buf);
    close(fd);

    return 1;
}

/* Open the file (and decide if it exists) */
int open_file(char *filename, int insert, int quiet)
{
    int fd;
    struct stat fileinfo;

    if (!strcmp(filename, "") || stat(filename, &fileinfo) == -1) {
	if (insert) {
	    if (!quiet)
		statusbar(_("\"%s\" not found"), filename);
	    return -1;
	} else {
	    /* We have a new file */
	    statusbar(_("New File"));
	    new_file();
	}
    } else if ((fd = open(filename, O_RDONLY)) == -1) {
	if (!quiet)
	    statusbar("%s: %s", strerror(errno), filename);
	return -1;
    } else {			/* File is A-OK */
	if (S_ISDIR(fileinfo.st_mode)) {
	    statusbar(_("File \"%s\" is a directory"), filename);
Chris Allegretta's avatar
Chris Allegretta committed
233
234
	    if (!insert)
		new_file();
Chris Allegretta's avatar
Chris Allegretta committed
235
236
237
238
239
240
241
242
243
244
245
246
247
	    return -1;
	}
	if (!quiet)
	    statusbar(_("Reading File"));
	read_file(fd, filename);
    }

    return 1;
}

int do_insertfile(void)
{
    int i;
248
    char *realname = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
249
250

    wrap_reset();
251
    i = statusq(1, writefile_list, WRITEFILE_LIST_LEN, "",
Chris Allegretta's avatar
Chris Allegretta committed
252
253
254
255
256
257
258
		_("File to insert [from ./] "));
    if (i != -1) {

#ifdef DEBUG
	fprintf(stderr, "filename is %s", answer);
#endif

259
#ifndef DISABLE_TABCOMP
260
        realname = real_dir_from_tilde(answer);
261
262
263
#else
        realname = mallocstrcpy(realname, answer);
#endif
264
265
266

	i = open_file(realname, 1, 0);
	free(realname);
Chris Allegretta's avatar
Chris Allegretta committed
267
268
269
270
271

	dump_buffer(fileage);
	set_modified();

	/* Here we want to rebuild the edit window */
Robert Siemborski's avatar
Robert Siemborski committed
272
	fix_editbot();
Chris Allegretta's avatar
Chris Allegretta committed
273
274

	/* If we've gone off the bottom, recenter, otherwise just redraw */
Chris Allegretta's avatar
Chris Allegretta committed
275
	if (current->lineno > editbot->lineno)
276
	    edit_update(current, CENTER);
Chris Allegretta's avatar
Chris Allegretta committed
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
	else
	    edit_refresh();

	UNSET(KEEP_CUTBUFFER);
	display_main_list();
	return i;
    } else {
	statusbar(_("Cancelled"));
	UNSET(KEEP_CUTBUFFER);
	display_main_list();
	return 0;
    }
}

/*
 * Write a file out.  If tmp is nonzero, we set the umask to 0600,
 * we don't set the global variable filename to it's name, and don't
 * print out how many lines we wrote on the statusbar.
 * 
296
297
 * tmp means we are writing a tmp file in a secute fashion.  We use
 * it when spell checking or dumping the file on an error.
Chris Allegretta's avatar
Chris Allegretta committed
298
299
300
301
302
303
 */
int write_file(char *name, int tmp)
{
    long size, lineswritten = 0;
    char buf[PATH_MAX + 1];
    filestruct *fileptr;
304
305
    int fd, mask = 0, realexists, anyexists;
    struct stat st, st2;
306
    static char *realname = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
307
308
309
310
311
312
313

    if (!strcmp(name, "")) {
	statusbar(_("Cancelled"));
	return -1;
    }
    titlebar();
    fileptr = fileage;
314
315
316
317

    if (realname != NULL)
	free(realname);

318
319
320
321
322
#ifndef DISABLE_TABCOMP
    realname = real_dir_from_tilde(name);
#else
    realname = mallocstrcpy(realname, name);
#endif
Chris Allegretta's avatar
Chris Allegretta committed
323

324
325
326
    /* Save the state of file at the end of the symlink */
    realexists = stat(realname, &st);

327
328
329
    /* Check to see if the file is a regular file and FOLLOW_SYMLINKS is
       set.  If so then don't do the delete and recreate code which would
       cause unexpected behavior */
330
    anyexists = lstat(realname, &st);
331

332
333
    /* New case: if the file exists, just give up */
    if (tmp && anyexists != -1)
334
	 return -1;
335
    else if (ISSET(FOLLOW_SYMLINKS) || !S_ISLNK(st.st_mode)) {
336

Chris Allegretta's avatar
Chris Allegretta committed
337
338
339
340
341
342
343
	/* Use O_EXCL if tmp == 1, I suppose */
	if (tmp)
	    fd = open(realname, O_WRONLY | O_CREAT | O_EXCL | O_TRUNC,
			S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH |
			S_IWOTH);
	else 
	    fd = open(realname, O_WRONLY | O_CREAT | O_TRUNC,
344
345
346
347
			S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH |
			S_IWOTH);

	/* First, just give up if we couldn't even open the file */
348
	if (fd == -1) {
349
	    if (!tmp && ISSET(TEMP_OPT)) {
Chris Allegretta's avatar
Chris Allegretta committed
350
351
352
		UNSET(TEMP_OPT);
		return do_writeout(1);
	    }
Chris Allegretta's avatar
Chris Allegretta committed
353
354
	    statusbar(_("Could not open file for writing: %s"),
		      strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
355
	    fprintf(stderr, "1\n");
356
	    free(realname);
Chris Allegretta's avatar
Chris Allegretta committed
357
358
	    return -1;
	}
359
360
361
362
363
364

	/* Now we fstat() the file, to make sure it's the same file still!
	   Thanks to Oliver Friedrichs(?) for this code from securityfocus */

 	if (fstat(fd, &st2) != 0) {
	    close(fd);
Chris Allegretta's avatar
Chris Allegretta committed
365
	    fprintf(stderr, "2\n");
366
367
368
	    return -1; 
	}
      
Chris Allegretta's avatar
Chris Allegretta committed
369
370
371
    }
    /* Don't follow symlink.  Create new file. */
    else {
372
	if (strlen(realname) > (PATH_MAX - 7)) {
Chris Allegretta's avatar
Chris Allegretta committed
373
374
375
376
377
	    statusbar(_("Could not open file: Path length exceeded."));
	    return -1;
	}

	memset(buf, 0x00, PATH_MAX + 1);
378
	strcat(buf, realname);
Chris Allegretta's avatar
Chris Allegretta committed
379
380
	strcat(buf, ".XXXXXX");
	if ((fd = mkstemp(buf)) == -1) {
Chris Allegretta's avatar
Chris Allegretta committed
381
382
383
	    if (ISSET(TEMP_OPT)) {
		UNSET(TEMP_OPT);
		return do_writeout(1);
384
	    }
Chris Allegretta's avatar
Chris Allegretta committed
385
386
	    statusbar(_("Could not open file for writing: %s"),
		      strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
387
	    fprintf(stderr, "3\n");
Chris Allegretta's avatar
Chris Allegretta committed
388
389
390
391
392
393
	    return -1;
	}
    }

    dump_buffer(fileage);
    while (fileptr != NULL && fileptr->next != NULL) {
Robert Siemborski's avatar
Robert Siemborski committed
394
	/* Next line is so we discount the "magic line" */
395
396
	if (filebot == fileptr && fileptr->data[0] == '\0')
	    break;
Robert Siemborski's avatar
Robert Siemborski committed
397

Chris Allegretta's avatar
Chris Allegretta committed
398
399
400
401
	size = write(fd, fileptr->data, strlen(fileptr->data));
	if (size == -1) {
	    statusbar(_("Could not open file for writing: %s"),
		      strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
402
	    fprintf(stderr, "4\n");
Chris Allegretta's avatar
Chris Allegretta committed
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
	    return -1;
	} else {
#ifdef DEBUG
	    fprintf(stderr, _("Wrote >%s\n"), fileptr->data);
#endif
	}
	write(fd, "\n", 1);

	fileptr = fileptr->next;
	lineswritten++;
    }

    if (fileptr != NULL) {
	size = write(fd, fileptr->data, strlen(fileptr->data));
	if (size == -1) {
	    statusbar(_("Could not open file for writing: %s"),
		      strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
420
	    fprintf(stderr, "5\n");
Chris Allegretta's avatar
Chris Allegretta committed
421
422
423
424
425
426
	    return -1;
	} else if (size > 0) {
	    size = write(fd, "\n", 1);
	    if (size == -1) {
		statusbar(_("Could not open file for writing: %s"),
			  strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
427
	    fprintf(stderr, "6\n");
Chris Allegretta's avatar
Chris Allegretta committed
428
429
430
431
432
433
434
		return -1;
	    }
	}
    }


    if (close(fd) == -1) {
435
	statusbar(_("Could not close %s: %s"), realname, strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
436
	unlink(buf);
Chris Allegretta's avatar
Chris Allegretta committed
437
	    fprintf(stderr, "7\n");
Chris Allegretta's avatar
Chris Allegretta committed
438
439
440
441
	return -1;
    }

    if (!ISSET(FOLLOW_SYMLINKS) || tmp) {
442
	if (realexists == -1) {
Chris Allegretta's avatar
Chris Allegretta committed
443
444
445
446
447
448
449
450
451
452
453
454
	    /* Use default umask as file permisions if file is a new file. */
	    mask = umask(0);
	    umask(mask);

	    if (tmp)		/* We don't want anyone reading our temporary file! */
		mask = 0600;
	    else
		mask = 0666 & ~mask;

	} else {
	    /* Use permissions from file we are overwriting. */
	    mask = st.st_mode;
455
	    if (unlink(realname) == -1) {
Chris Allegretta's avatar
Chris Allegretta committed
456
457
		if (errno != ENOENT) {
		    statusbar(_("Could not open %s for writing: %s"),
458
			      realname, strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
459
	    fprintf(stderr, "8\n");
Chris Allegretta's avatar
Chris Allegretta committed
460
461
462
463
464
465
		    unlink(buf);
		    return -1;
		}
	    }
	}

466
467
468
469
470
	if (!tmp) {
	    if (link(buf, realname) != -1)
		unlink(buf);
	    else if (errno != EPERM) {
		statusbar(_("Could not open %s for writing: %s"),
Chris Allegretta's avatar
Chris Allegretta committed
471
		      name, strerror(errno));
472
		unlink(buf);
Chris Allegretta's avatar
Chris Allegretta committed
473
	    fprintf(stderr, "9\n");
474
475
476
		return -1;
	    } else if (rename(buf, realname) == -1) {	/* Try a rename?? */
 		statusbar(_("Could not open %s for writing: %s"),
477
		      realname, strerror(errno));
478
		unlink(buf);
Chris Allegretta's avatar
Chris Allegretta committed
479
	    fprintf(stderr, "10\n");
480
481
		return -1;
	    }
Chris Allegretta's avatar
Chris Allegretta committed
482
	}
483
	if (chmod(realname, mask) == -1)
Chris Allegretta's avatar
Chris Allegretta committed
484
	    statusbar(_("Could not set permissions %o on %s: %s"),
485
		      mask, realname, strerror(errno));
Chris Allegretta's avatar
Chris Allegretta committed
486
487
488

    }
    if (!tmp) {
489
	strncpy(filename, realname, 132);
Chris Allegretta's avatar
Chris Allegretta committed
490
	statusbar(_("Wrote %d lines"), lineswritten);
491
492
        UNSET(MODIFIED);
        titlebar();
Chris Allegretta's avatar
Chris Allegretta committed
493
494
495
496
497
498
499
    }
    return 1;
}

int do_writeout(int exiting)
{
    int i = 0;
500
501
502
#ifdef NANO_EXTRA
    static int did_cred = 0;
#endif
Chris Allegretta's avatar
Chris Allegretta committed
503

504
    answer = mallocstrcpy(answer, filename);
Chris Allegretta's avatar
Chris Allegretta committed
505

506
    if ((exiting) && (ISSET(TEMP_OPT))) {
507
	if (filename[0]) {
508
509
510
	    i = write_file(answer, 0);
	    display_main_list();
	    return i;
511
	} else {
Chris Allegretta's avatar
Chris Allegretta committed
512
513
514
515
516
	    UNSET(TEMP_OPT);
	    do_exit();

	    /* They cancelled, abort quit */
	    return -1;
517
	}
Chris Allegretta's avatar
Chris Allegretta committed
518
519
520
    }

    while (1) {
521
	i = statusq(1, writefile_list, WRITEFILE_LIST_LEN, answer,
Chris Allegretta's avatar
Chris Allegretta committed
522
		    _("File Name to write"));
Chris Allegretta's avatar
Chris Allegretta committed
523

Chris Allegretta's avatar
Chris Allegretta committed
524
	if (i != -1) {
Chris Allegretta's avatar
Chris Allegretta committed
525
526
527
528

#ifdef DEBUG
	    fprintf(stderr, _("filename is %s"), answer);
#endif
529
530
531
532
533
534
535
536

#ifdef NANO_EXTRA
	    if (exiting && !ISSET(TEMP_OPT) && !strcasecmp(answer, "zzy") && !did_cred) {
		do_credits();
		did_cred = 1;
		return - 1;
	    }
#endif
537
	    if (strcmp(answer, filename)) {
Chris Allegretta's avatar
Chris Allegretta committed
538
539
540
541
542
543
		struct stat st;
		if (!stat(answer, &st)) {
		    i = do_yesno(0, 0, _("File exists, OVERWRITE ?"));

		    if (!i || (i == -1))
			continue;
Chris Allegretta's avatar
Chris Allegretta committed
544
545
		}
	    }
Chris Allegretta's avatar
Chris Allegretta committed
546
547
548
549
	    i = write_file(answer, 0);

	    display_main_list();
	    return i;
Chris Allegretta's avatar
Chris Allegretta committed
550
	} else {
Chris Allegretta's avatar
Chris Allegretta committed
551
552
553
	    statusbar(_("Cancelled"));
	    display_main_list();
	    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
554
	}
Chris Allegretta's avatar
Chris Allegretta committed
555
556
557
558
559
560
561
    }
}

int do_writeout_void(void)
{
    return do_writeout(0);
}
Chris Allegretta's avatar
Chris Allegretta committed
562

563
564
565
566
567
568
569
570
#ifndef DISABLE_TABCOMP
static char **homedirs;

/* Return a malloc()ed string containing the actual directory, used
 * to convert ~user and ~/ notation...
 */
char *real_dir_from_tilde (char *buf)
{
571
572
    char *dirtmp = NULL, *line = NULL, byte[1], *lineptr;
    int fd, i, status, searchctr = 1;
573
574

    if (buf[0] == '~') {
575
576
577
	if (buf[1] == '~')
	    goto abort;		/* Handle ~~ without segfaulting =) */
	else if (buf[1] == '/') {
578
579
580
581
582
583
584
585
	    if (getenv("HOME") != NULL) {
		dirtmp = nmalloc(strlen(buf) + 2 + strlen(getenv("HOME")));

		sprintf(dirtmp, "%s/%s", getenv("HOME"), &buf[2]);
	    }
	}
	else if (buf[1] != 0) {

586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
	    if((fd = open("/etc/passwd", O_RDONLY)) == -1)
		goto abort;

	    /* Figure how how much of of the str we need to compare */
	    for (searchctr = 1; buf[searchctr] != '/' &&
			buf[searchctr] != 0; searchctr++)
		;

	    do {
		i = 0;
		line = nmalloc(1);
		while ((status = read(fd, byte, 1)) != 0 && byte[0] != '\n') {

		    line[i] = byte[0];
		    i++;
		    line = nrealloc(line, i+1);
		}
		line[i] = 0;

		if (i == 0)
		     goto abort;

		line[i] = 0;
		lineptr = strtok(line, ":");

		if (!strncmp(lineptr, &buf[1], searchctr - 1)) {

		    /* Okay, skip to the password portion now */
		    for (i = 0; i <= 4 && lineptr != NULL; i++)
			lineptr = strtok(NULL, ":");

		    if (lineptr == NULL)
			goto abort;

		     /* Else copy the new string into the new buf */
		    dirtmp = nmalloc(strlen(buf) + 2 + strlen(lineptr));

		    sprintf(dirtmp, "%s%s", lineptr, &buf[searchctr]);
		    free(line);
		    break;
		}
	
		free(line);

	    } while (status != 0);
631
632
633
634
635
636
	}
    }
    else
	dirtmp = mallocstrcpy(dirtmp, buf);

    return dirtmp;	
637
638
639
640

abort:		
    dirtmp = mallocstrcpy(dirtmp, buf);
    return dirtmp;
641
642
643
}

/* Tack a slash onto the string we're completing if it's a directory */
644
int append_slash_if_dir(char *buf, int *lastWasTab, int *place)
645
646
647
{
    char *dirptr;
    struct stat fileinfo;
648
    int ret = 0;
649
650
651
652

    dirptr = real_dir_from_tilde(buf);

    if (stat(dirptr, &fileinfo) == -1)
653
    	ret = 0;
654
655
656
657
658
    else if (S_ISDIR(fileinfo.st_mode)) {
	strncat(buf, "/", 1);
	*place += 1;
	/* now we start over again with # of tabs so far */
	*lastWasTab = 0;
659
	ret = 1;
660
661
662
663
    }

    if (dirptr != buf)
    	free(dirptr);
664
665

    return ret;
666
}
Chris Allegretta's avatar
Chris Allegretta committed
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688

/*
 * These functions (username_tab_completion, cwd_tab_completion, and
 * input_tab were taken from busybox 0.46 (cmdedit.c).  Here is the notice
 * from that file:
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
 * Copyright (c) 1999
 *      Main code:            Adam Rogoyski <rogoyski@cs.utexas.edu>
 *      Etc:                  Dave Cinege <dcinege@psychosis.com>
 *  Majorly adjusted/re-written for busybox:
 *                            Erik Andersen <andersee@debian.org>
 *
 * You may use this code as you wish, so long as the original author(s)
 * are attributed in any redistributions of the source code.
 * This code is 'as is' with no warranty.
 * This code may safely be consumed by a BSD or GPL license.
 */

char **username_tab_completion(char *buf, int *num_matches)
{
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
    char **matches = (char **) NULL, *line = NULL, *lineptr;
    char *matchline = NULL, *matchdir = NULL;
	
    int fd, i = 0, status = 1;
    char byte[1];

    if((fd = open("/etc/passwd", O_RDONLY)) == -1) {
	return NULL;
    }

    if (homedirs != NULL) {
	for (i = 0; i < *num_matches; i++)
	    free(homedirs[i]);
	free(homedirs);
	homedirs = (char **) NULL;
	*num_matches = 0;
    }
    matches = nmalloc(BUFSIZ);
    homedirs = nmalloc(BUFSIZ);
    strcat(buf, "*");
    do {
	i = 0;
	line = nmalloc(1);
	while ((status = read(fd, byte, 1)) != 0 && byte[0] != '\n') {

	    line[i] = byte[0];
	    i++;
	    line = nrealloc(line, i+1);
	}

	if (i == 0)
	    break;

	line[i] = 0;
	lineptr = strtok(line, ":");

	if (check_wildcard_match(line, &buf[1]) == TRUE) {

	    if (*num_matches == BUFSIZ)
		break;

	    /* Cool, found a match.  Add it to the list
	     * This makes a lot more sense to me (Chris) this way...
	     */
	    matchline = nmalloc(strlen(line) + 2);
	    sprintf(matchline, "~%s", line);

	    for (i = 0; i <= 4 && lineptr != NULL; i++)
		lineptr = strtok(NULL, ":");

	    if (lineptr == NULL)
		break;

	    matchdir = mallocstrcpy(matchdir, lineptr);
	    homedirs[*num_matches] = matchdir;
	    matches[*num_matches] = matchline;

	    ++*num_matches;

	    /* If there's no more room, bail out */
	    if (*num_matches == BUFSIZ)
		break;
	}

	free(line);

    } while (status != 0);

    close(fd);
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
759
760
761
762
763
764
765
766
767
768
769
#ifdef DEBUG
    fprintf(stderr, "\nin username_tab_completion\n");
#endif
    return (matches);
}

/* This was originally called exe_n_cwd_tab_completion, but we're not
   worried about executables, only filenames :> */

char **cwd_tab_completion(char *buf, int *num_matches)
{
770
    char *dirName, *dirtmp = NULL, *tmp = NULL, *tmp2 = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
771
772
773
774
    char **matches = (char **) NULL;
    DIR *dir;
    struct dirent *next;

775
    matches = nmalloc(BUFSIZ);
Chris Allegretta's avatar
Chris Allegretta committed
776
777
778
779

    /* Stick a wildcard onto the buf, for later use */
    strcat(buf, "*");

780
    /* Okie, if there's a / in the buffer, strip out the directory part */
Chris Allegretta's avatar
Chris Allegretta committed
781
782
783
784
    if (strcmp(buf, "") && strstr(buf, "/")) {
	dirName = malloc(strlen(buf) + 1);
	tmp = buf + strlen(buf);
	while (*tmp != '/' && tmp != buf)
785
	    tmp--;
786

Chris Allegretta's avatar
Chris Allegretta committed
787
788
	tmp++;

789
790
791
	strncpy(dirName, buf, tmp - buf + 1);
	dirName[tmp - buf] = 0;

Chris Allegretta's avatar
Chris Allegretta committed
792
793
794
795
796
797
798
799
800
801
802
803
    } else {
	if ((dirName = getcwd(NULL, 0)) == NULL)
	    return matches;
	else
	    tmp = buf;
    }

#ifdef DEBUG
    fprintf(stderr, "\nDir = %s\n", dirName);
    fprintf(stderr, "\nbuf = %s\n", buf);
    fprintf(stderr, "\ntmp = %s\n", tmp);
#endif
804

805
806
807
808
809
810
811
812
813
814
815
    dirtmp = real_dir_from_tilde(dirName);
    free(dirName);
    dirName = dirtmp;

#ifdef DEBUG
    fprintf(stderr, "\nDir = %s\n", dirName);
    fprintf(stderr, "\nbuf = %s\n", buf);
    fprintf(stderr, "\ntmp = %s\n", tmp);
#endif


Chris Allegretta's avatar
Chris Allegretta committed
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
    dir = opendir(dirName);
    if (!dir) {
	/* Don't print an error, just shut up and return */
	*num_matches = 0;
	beep();
	return (matches);
    }
    while ((next = readdir(dir)) != NULL) {

	/* Some quick sanity checks */
	if ((strcmp(next->d_name, "..") == 0)
	    || (strcmp(next->d_name, ".") == 0)) {
	    continue;
	}
#ifdef DEBUG
831
	fprintf(stderr, "Comparing \'%s\'\n", next->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
832
833
834
#endif
	/* See if this matches */
	if (check_wildcard_match(next->d_name, tmp) == TRUE) {
835
836
837
838
839
840
841
842

	    /* Cool, found a match.  Add it to the list
	     * This makes a lot more sense to me (Chris) this way...
	     */
	    tmp2 = NULL;
	    tmp2 = nmalloc(strlen(next->d_name) + 1);
	    strcpy(tmp2, next->d_name);
	    matches[*num_matches] = tmp2;
Chris Allegretta's avatar
Chris Allegretta committed
843
	    ++*num_matches;
844
845
846
847

	    /* If there's no more room, bail out */
	    if (*num_matches == BUFSIZ)
		break;
Chris Allegretta's avatar
Chris Allegretta committed
848
849
850
851
852
853
	}
    }

    return (matches);
}

854
/* This function now has an arg which refers to how much the 
Chris Allegretta's avatar
Chris Allegretta committed
855
856
 * statusbar (place) should be advanced, i.e. the new cursor pos.
 */
857
char *input_tab(char *buf, int place, int *lastWasTab, int *newplace)
Chris Allegretta's avatar
Chris Allegretta committed
858
859
{
    /* Do TAB completion */
860
    static int num_matches = 0, match_matches = 0;
Chris Allegretta's avatar
Chris Allegretta committed
861
    static char **matches = (char **) NULL;
862
    int pos = place, i = 0, col = 0, editline = 0;
863
    int longestname = 0, is_dir = 0;
864
    char *foo;
Chris Allegretta's avatar
Chris Allegretta committed
865

866
    if (*lastWasTab == FALSE) {
867
	char *tmp, *copyto, *matchBuf;
Chris Allegretta's avatar
Chris Allegretta committed
868

869
870
	*lastWasTab = 1;

Chris Allegretta's avatar
Chris Allegretta committed
871
872
	/* Make a local copy of the string -- up to the position of the
	   cursor */
873
874
	matchBuf = (char *) calloc(strlen(buf) + 2, sizeof(char));

Chris Allegretta's avatar
Chris Allegretta committed
875
876
877
878
	strncpy(matchBuf, buf, place);
	tmp = matchBuf;

	/* skip any leading white space */
879
	while (*tmp && isspace((int) *tmp))
Chris Allegretta's avatar
Chris Allegretta committed
880
881
882
883
	    ++tmp;

	/* Free up any memory already allocated */
	if (matches != NULL) {
884
885
	    for (i = i; i < num_matches; i++)
		free(matches[i]);
Chris Allegretta's avatar
Chris Allegretta committed
886
887
	    free(matches);
	    matches = (char **) NULL;
888
	    num_matches = 0;
Chris Allegretta's avatar
Chris Allegretta committed
889
890
891
892
893
	}

	/* If the word starts with `~' and there is no slash in the word, 
	 * then try completing this word as a username. */

894
	/* FIXME -- this check is broken! */
895
	   if (*tmp == '~' && !strchr(tmp, '/'))
896
	   matches = username_tab_completion(tmp, &num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
897
898
899
900
901
902
903
904
905

	/* Try to match everything in the current working directory that
	 * matches.  */
	if (!matches)
	    matches = cwd_tab_completion(tmp, &num_matches);

	/* Don't leak memory */
	free(matchBuf);

906
907
908
#ifdef DEBUG
	fprintf(stderr, "%d matches found...\n", num_matches);
#endif
Chris Allegretta's avatar
Chris Allegretta committed
909
	/* Did we find exactly one match? */
910
	switch (num_matches) {
911
	case 0:
912
	    blank_edit();
913
	    wrefresh(edit);
914
915
	    break;
	case 1:
916
917
918
919

	    buf = nrealloc(buf, strlen(buf) + strlen(matches[0]) + 1);

	    if (strcmp(buf, "") && strstr(buf, "/")) {
920
921
		for (tmp = buf + strlen(buf); *tmp != '/' && tmp != buf;
		     tmp--);
922
		tmp++;
923
	    } else
924
925
		tmp = buf;

926
927
928
929
930
	    if (!strcmp(tmp, matches[0]))
		is_dir = append_slash_if_dir(buf, lastWasTab, newplace);

	    if (is_dir)
		break;
931
932

	    copyto = tmp;
933
934
	    for (pos = 0; *tmp == matches[0][pos] &&
		 pos <= strlen(matches[0]); pos++)
935
936
		tmp++;

937
	    /* write out the matched name */
938
	    strncpy(copyto, matches[0], strlen(matches[0]) + 1);
939
940
	    *newplace += strlen(matches[0]) - pos;

941
942
943
	    /* Is it a directory? */
	    append_slash_if_dir(buf, lastWasTab, newplace);

944
945
	    break;
	default:
946
	    /* Check to see if all matches share a beginning, and if so
947
	       tack it onto buf and then beep */
948

949
	    if (strcmp(buf, "") && strstr(buf, "/")) {
950
951
		for (tmp = buf + strlen(buf); *tmp != '/' && tmp != buf;
		     tmp--);
952
		tmp++;
953
	    } else
954
955
956
		tmp = buf;

	    for (pos = 0; *tmp == matches[0][pos] && *tmp != 0 &&
957
		 pos <= strlen(matches[0]); pos++)
958
959
		tmp++;

960
961
962
963
964
965
966
967
968
	    while (1) {
		match_matches = 0;

		for (i = 0; i < num_matches; i++) {
		    if (matches[i][pos] == 0)
			break;
		    else if (matches[i][pos] == matches[0][pos])
			match_matches++;
		}
969
970
		if (match_matches == num_matches &&
		    (i == num_matches || matches[i] != 0)) {
971
		    /* All the matches have the same character at pos+1,
972
		       so paste it into buf... */
973
		    buf = nrealloc(buf, strlen(buf) + 2);
974
		    strncat(buf, matches[0] + pos, 1);
975
		    *newplace += 1;
976
		    pos++;
977
		} else {
978
979
980
981
		    beep();
		    break;
		}
	    }
982
	    break;
983
	}
Chris Allegretta's avatar
Chris Allegretta committed
984
985
986
987
988
989
990
991
992
993
    } else {
	/* Ok -- the last char was a TAB.  Since they
	 * just hit TAB again, print a list of all the
	 * available choices... */
	if (matches && num_matches > 0) {

	    /* Blank the edit window, and print the matches out there */
	    blank_edit();
	    wmove(edit, 0, 0);

994
	    editline = 0;
995

996
997
998
999
1000
1001
1002
1003
	    /* Figure out the length of the longest filename */
	    for (i = 0; i < num_matches; i++)
		if (strlen(matches[i]) > longestname)
		    longestname = strlen(matches[i]);

	    if (longestname > COLS - 1)
		longestname = COLS - 1;

1004
1005
	    foo = nmalloc(longestname + 5);

Chris Allegretta's avatar
Chris Allegretta committed
1006
1007
	    /* Print the list of matches */
	    for (i = 0, col = 0; i < num_matches; i++) {
1008

1009
		/* make each filename shown be the same length as the longest
1010
		   filename, with two spaces at the end */
1011
1012
1013
1014
1015
1016
		snprintf(foo, longestname + 1, matches[i]);
		while (strlen(foo) < longestname)
		    strcat(foo, " ");

		strcat(foo, "  ");

1017
1018
		/* Disable el cursor */
		curs_set(0);
1019
1020
1021
1022
1023
1024
		/* now, put the match on the screen */
		waddnstr(edit, foo, strlen(foo));
		col += strlen(foo);

		/* And if the next match isn't going to fit on the
		   line, move to the next one */
1025
		if (col > (COLS - longestname) && matches[i + 1] != NULL) {
1026
1027
		    editline++;
		    wmove(edit, editline, 0);
1028
1029
1030
1031
		    if (editline == editwinrows - 1) {
			waddstr(edit, _("(more)"));
			break;
		    }
Chris Allegretta's avatar
Chris Allegretta committed
1032
1033
1034
		    col = 0;
		}
	    }
1035
	    free(foo);
Chris Allegretta's avatar
Chris Allegretta committed
1036
	    wrefresh(edit);
1037
1038
	} else
	    beep();
Chris Allegretta's avatar
Chris Allegretta committed
1039
1040
1041
1042

    }

    edit_refresh();
1043
1044
    curs_set(1);
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
1045
}
1046
#endif