files.c 73.6 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/* $Id$ */
Chris Allegretta's avatar
Chris Allegretta committed
2
3
4
/**************************************************************************
 *   files.c                                                              *
 *                                                                        *
5
 *   Copyright (C) 1999-2005 Chris Allegretta                             *
Chris Allegretta's avatar
Chris Allegretta committed
6
7
 *   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 *
8
 *   the Free Software Foundation; either version 2, or (at your option)  *
Chris Allegretta's avatar
Chris Allegretta committed
9
10
 *   any later version.                                                   *
 *                                                                        *
11
12
13
14
 *   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.                             *
Chris Allegretta's avatar
Chris Allegretta committed
15
16
17
 *                                                                        *
 *   You should have received a copy of the GNU General Public License    *
 *   along with this program; if not, write to the Free Software          *
18
19
 *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA            *
 *   02110-1301, USA.                                                     *
Chris Allegretta's avatar
Chris Allegretta committed
20
21
22
 *                                                                        *
 **************************************************************************/

23
24
25
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
26

Chris Allegretta's avatar
Chris Allegretta committed
27
28
29
#include <string.h>
#include <stdio.h>
#include <unistd.h>
30
#include <utime.h>
Chris Allegretta's avatar
Chris Allegretta committed
31
32
#include <fcntl.h>
#include <errno.h>
Chris Allegretta's avatar
Chris Allegretta committed
33
#include <ctype.h>
34
#include <pwd.h>
Chris Allegretta's avatar
Chris Allegretta committed
35
#include "proto.h"
Chris Allegretta's avatar
Chris Allegretta committed
36

37
38
39
/* Add an entry to the openfile openfilestruct.  This should only be
 * called from open_buffer(). */
void make_new_buffer(void)
40
{
41
42
    /* If there are no entries in openfile, make the first one and
     * move to it. */
43
44
45
    if (openfile == NULL) {
	openfile = make_new_opennode();
	splice_opennode(openfile, openfile, openfile);
46
47
48
    /* Otherwise, make a new entry for openfile, splice it in after
     * the current entry, and move to it. */
    } else {
49
50
	splice_opennode(openfile, make_new_opennode(), openfile->next);
	openfile = openfile->next;
Chris Allegretta's avatar
Chris Allegretta committed
51
    }
52

53
54
55
56
57
58
59
60
61
    /* Initialize the new buffer. */
    initialize_buffer();
}

/* Initialize the current entry of the openfile openfilestruct. */
void initialize_buffer(void)
{
    assert(openfile != NULL);

62
    openfile->filename = mallocstrcpy(NULL, "");
Chris Allegretta's avatar
Chris Allegretta committed
63

64
    initialize_buffer_text();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
65

66
67
    openfile->current_x = 0;
    openfile->placewewant = 0;
68
    openfile->current_y = 0;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
69

70
    openfile->modified = FALSE;
71
#ifndef NANO_SMALL
72
    openfile->mark_set = FALSE;
73

74
75
76
    openfile->mark_begin = NULL;
    openfile->mark_begin_x = 0;

77
    openfile->fmt = NIX_FILE;
78

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
79
    openfile->current_stat = NULL;
80
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
81
#ifdef ENABLE_COLOR
82
83
    openfile->colorstrings = NULL;
#endif
84
}
Chris Allegretta's avatar
Chris Allegretta committed
85

86
87
88
/* Initialize the text of the current entry of the openfile
 * openfilestruct. */
void initialize_buffer_text(void)
89
{
90
    assert(openfile != NULL);
91

92
93
    openfile->fileage = make_new_node(NULL);
    openfile->fileage->data = mallocstrcpy(NULL, "");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
94

95
96
97
98
99
    openfile->filebot = openfile->fileage;
    openfile->edittop = openfile->fileage;
    openfile->current = openfile->fileage;

    openfile->totsize = 0;
100
101
}

102
103
/* If it's not "", filename is a file to open.  We make a new buffer, if
 * necessary, and then open and read the file, if applicable. */
104
void open_buffer(const char *filename)
105
{
106
107
108
    bool new_buffer = (openfile == NULL
#ifdef ENABLE_MULTIBUFFER
	 || ISSET(MULTIBUFFER)
109
#endif
110
111
112
113
114
115
	);
	/* Whether we load into this buffer or a new one. */
    FILE *f;
    int rc;
	/* rc == -2 means that we have a new file.  -1 means that the
	 * open() failed.  0 means that the open() succeeded. */
116

117
118
    assert(filename != NULL);

119
120
121
122
123
124
125
#ifndef DISABLE_OPERATINGDIR
    if (check_operating_dir(filename, FALSE)) {
	statusbar(_("Can't insert file from outside of %s"),
		operating_dir);
	return;
    }
#endif
126

127
128
    /* If the filename isn't blank, open the file.  Otherwise, treat it
     * as a new file. */
129
130
    rc = (filename[0] != '\0') ? open_file(filename, new_buffer, &f) :
	-2;
131

132
133
    /* If we're loading into a new buffer, add a new entry to
     * openfile. */
134
135
    if (new_buffer)
	make_new_buffer();
136

137
138
139
140
    /* If we have a file and we're loading into a new buffer, update the
     * filename. */
    if (rc != -1 && new_buffer)
	openfile->filename = mallocstrcpy(openfile->filename, filename);
141

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
142
143
    /* If we have a non-new file, read it in.  Then, if the buffer has
     * no stat, update the stat, if applicable. */
144
145
    if (rc == 0) {
	read_file(f, filename);
146
#ifndef NANO_SMALL
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
147
148
149
150
151
	if (openfile->current_stat == NULL) {
	    openfile->current_stat =
		(struct stat *)nmalloc(sizeof(struct stat));
	    stat(filename, openfile->current_stat);
	}
152
#endif
153
154
    }

155
156
157
158
    /* If we have a file and we're loading into a new buffer, move back
     * to the first line of the buffer. */
    if (rc != -1 && new_buffer)
	openfile->current = openfile->fileage;
159
160

#ifdef ENABLE_COLOR
161
162
    /* If we're loading into a new buffer, update the colors to account
     * for it, if applicable. */
163
164
165
    if (new_buffer)
	color_update();
#endif
166
}
167

168
/* Update the screen to account for the current buffer. */
169
void display_buffer(void)
170
{
171
    /* Update the titlebar, since the filename may have changed. */
172
    titlebar(NULL);
173
174

#ifdef ENABLE_COLOR
175
176
177
    /* Make sure we're using the buffer's associated colors, if
     * applicable. */
    color_init();
178
179
180
#endif

    /* Update the edit window. */
181
    edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
182
183
}

184
185
186
187
#ifdef ENABLE_MULTIBUFFER
/* Switch to the next file buffer if next_buf is TRUE.  Otherwise,
 * switch to the previous file buffer. */
void switch_to_prevnext_buffer(bool next_buf)
Chris Allegretta's avatar
Chris Allegretta committed
188
{
189
    assert(openfile != NULL);
190

191
192
    /* If only one file buffer is open, indicate it on the statusbar and
     * get out. */
193
    if (openfile == openfile->next) {
194
195
	statusbar(_("No more open file buffers"));
	return;
Chris Allegretta's avatar
Chris Allegretta committed
196
    }
197

198
199
    /* Switch to the next or previous file buffer, depending on the
     * value of next_buf. */
200
    openfile = next_buf ? openfile->next : openfile->prev;
201
202

#ifdef DEBUG
203
    fprintf(stderr, "filename is %s\n", openfile->filename);
204
205
#endif

206
    /* Update the screen to account for the current buffer. */
207
    display_buffer();
208

209
    /* Indicate the switch on the statusbar. */
210
    statusbar(_("Switched to %s"),
211
212
	((openfile->filename[0] == '\0') ? _("New Buffer") :
	openfile->filename));
213
214

#ifdef DEBUG
215
    dump_filestruct(openfile->current);
216
#endif
Chris Allegretta's avatar
Chris Allegretta committed
217
218
}

219
/* Switch to the previous entry in the openfile filebuffer. */
220
void switch_to_prev_buffer_void(void)
221
{
222
    switch_to_prevnext_buffer(FALSE);
223
}
224

225
/* Switch to the next entry in the openfile filebuffer. */
226
void switch_to_next_buffer_void(void)
227
{
228
    switch_to_prevnext_buffer(TRUE);
229
}
230

231
/* Delete an entry from the openfile filebuffer, and switch to the one
232
233
234
 * after it.  Return TRUE on success, or FALSE if there are no more open
 * file buffers. */
bool close_buffer(void)
235
{
236
    assert(openfile != NULL);
237

238
    /* If only one file buffer is open, get out. */
239
    if (openfile == openfile->next)
240
	return FALSE;
241

242
    /* Switch to the next file buffer. */
243
    switch_to_next_buffer_void();
244

245
    /* Close the file buffer we had open before. */
246
    unlink_opennode(openfile->prev);
247

248
    display_main_list();
249

250
    return TRUE;
251
}
252
#endif /* ENABLE_MULTIBUFFER */
253

254
255
256
257
258
259
/* We make a new line of text from buf.  buf is length buf_len.  If
 * first_line_ins is TRUE, then we put the new line at the top of the
 * file.  Otherwise, we assume prevnode is the last line of the file,
 * and put our line after prevnode. */
filestruct *read_line(char *buf, filestruct *prevnode, bool
	*first_line_ins, size_t buf_len)
260
{
261
    filestruct *fileptr = (filestruct *)nmalloc(sizeof(filestruct));
262

263
264
265
    /* Convert nulls to newlines.  buf_len is the string's real length
     * here. */
    unsunder(buf, buf_len);
266

267
    assert(openfile->fileage != NULL && strlen(buf) == buf_len);
268

269
    fileptr->data = mallocstrcpy(NULL, buf);
270

Chris Allegretta's avatar
Chris Allegretta committed
271
#ifndef NANO_SMALL
272
273
274
275
    /* If it's a DOS file ("\r\n"), and file conversion isn't disabled,
     * strip the '\r' part from fileptr->data. */
    if (!ISSET(NO_CONVERT) && buf_len > 0 && buf[buf_len - 1] == '\r')
	fileptr->data[buf_len - 1] = '\0';
276
#endif
277

278
    if (*first_line_ins == TRUE) {
279
280
281
	/* Special case: We're inserting with the cursor on the first
	 * line. */
	fileptr->prev = NULL;
282
	fileptr->next = openfile->fileage;
283
284
285
286
287
288
	fileptr->lineno = 1;
	if (*first_line_ins == TRUE) {
	    *first_line_ins = FALSE;
	    /* 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. */
289
	    openfile->edittop = fileptr;
290
	} else
291
292
	    openfile->filebot = fileptr;
	openfile->fileage = fileptr;
293
294
295
296
297
298
299
    } else {
	assert(prevnode != NULL);

	fileptr->prev = prevnode;
	fileptr->next = NULL;
	fileptr->lineno = prevnode->lineno + 1;
	prevnode->next = fileptr;
300
301
    }

302
303
    return fileptr;
}
304

305
306
307
308
309
310
311
312
313
314
315
316
317
318
void read_file(FILE *f, const char *filename)
{
    size_t num_lines = 0;
	/* The number of lines in the file. */
    size_t len = 0;
	/* The length of the current line of the file. */
    size_t i = 0;
	/* The position in the current line of the file. */
    size_t bufx = MAX_BUF_SIZE;
	/* The size of each chunk of the file that we read. */
    char input = '\0';
	/* The current input character. */
    char *buf;
	/* The buffer where we store chunks of the file. */
319
    filestruct *fileptr = openfile->current;
320
321
322
323
324
325
	/* The current line of the file. */
    bool first_line_ins = FALSE;
	/* Whether we're inserting with the cursor on the first line. */
    int input_int;
	/* The current value we read from the file, whether an input
	 * character or EOF. */
326
#ifndef NANO_SMALL
327
328
    int format = 0;
	/* 0 = *nix, 1 = DOS, 2 = Mac, 3 = both DOS and Mac. */
329
#endif
330

331
332
    assert(openfile->fileage != NULL && openfile->current != NULL);

333
334
    buf = charalloc(bufx);
    buf[0] = '\0';
335

336
337
338
339
    if (openfile->current == openfile->fileage)
	first_line_ins = TRUE;
    else
	fileptr = openfile->current->prev;
340

341
    /* Read the entire file into the filestruct. */
342
343
344
345
346
347
    while ((input_int = getc(f)) != EOF) {
	input = (char)input_int;

	/* If it's a *nix file ("\n") or a DOS file ("\r\n"), and file
	 * conversion isn't disabled, handle it! */
	if (input == '\n') {
Chris Allegretta's avatar
Chris Allegretta committed
348
#ifndef NANO_SMALL
349
350
351
352
353
354
	    /* If there's a '\r' before the '\n', set format to DOS if
	     * we currently think this is a *nix file, or to both if we
	     * currently think it's a Mac file. */
	    if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r' &&
		(format == 0 || format == 2))
		format++;
355
#endif
356

357
358
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
359

360
361
362
	    /* Reset the line length in preparation for the next
	     * line. */
	    len = 0;
Chris Allegretta's avatar
Chris Allegretta committed
363

364
365
366
	    num_lines++;
	    buf[0] = '\0';
	    i = 0;
367
#ifndef NANO_SMALL
368
369
370
371
372
373
374
375
	/* If it's a Mac file ('\r' without '\n'), and file conversion
	 * isn't disabled, handle it! */
	} else if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r') {
	    /* If we currently think the file is a *nix file, set format
	     * to Mac.  If we currently think the file is a DOS file,
	     * set format to both DOS and Mac. */
	    if (format == 0 || format == 1)
		format += 2;
376

377
378
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
379

380
381
382
383
	    /* Reset the line length in preparation for the next line.
	     * Since we've already read in the next character, reset it
	     * to 1 instead of 0. */
	    len = 1;
Chris Allegretta's avatar
Chris Allegretta committed
384

385
386
387
388
	    num_lines++;
	    buf[0] = input;
	    buf[1] = '\0';
	    i = 1;
389
#endif
390
391
392
393
	} else {
	    /* Calculate the total length of the line.  It might have
	     * nulls in it, so we can't just use strlen() here. */
	    len++;
394

395
396
397
398
399
400
401
402
	    /* Now we allocate a bigger buffer MAX_BUF_SIZE 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) {
		bufx += MAX_BUF_SIZE;
		buf = charealloc(buf, bufx);
403
	    }
404
405
406
407
408
409
410
411
412
413
414

	    buf[i] = input;
	    buf[i + 1] = '\0';
	    i++;
	}
    }

    /* Perhaps this could use some better handling. */
    if (ferror(f))
	nperror(filename);
    fclose(f);
415

416
#ifndef NANO_SMALL
417
418
419
420
421
422
423
424
    /* If file conversion isn't disabled and the last character in this
     * file is '\r', read it in properly as a Mac format line. */
    if (len == 0 && !ISSET(NO_CONVERT) && input == '\r') {
	len = 1;

	buf[0] = input;
	buf[1] = '\0';
    }
425
#endif
426

427
428
429
430
431
432
433
434
435
436
    /* Did we not get a newline and still have stuff to do? */
    if (len > 0) {
#ifndef NANO_SMALL
	/* If file conversion isn't disabled and the last character in
	 * this file is '\r', set format to Mac if we currently think
	 * the file is a *nix file, or to both DOS and Mac if we
	 * currently think the file is a DOS file. */
	if (!ISSET(NO_CONVERT) && buf[len - 1] == '\r' &&
		(format == 0 || format == 1))
	    format += 2;
437
438
#endif

439
440
441
442
	/* Read in the last line properly. */
	fileptr = read_line(buf, fileptr, &first_line_ins, len);
	num_lines++;
    }
443

444
    free(buf);
445

446
    /* If we didn't get a file and we don't already have one, open a
447
     * blank buffer. */
448
    if (fileptr == NULL)
449
	open_buffer("");
450

451
    /* Did we try to insert a file of zero bytes? */
452
453
454
455
456
457
    if (num_lines > 0) {
	fileptr->next = openfile->current;
	openfile->current->prev = fileptr;
	renumber(openfile->current);
	openfile->current_x = 0;
	openfile->placewewant = 0;
458
    }
459

460
461
    openfile->totsize += get_totsize(openfile->fileage,
	openfile->filebot);
462

463
464
465
466
467
468
469
#ifndef NANO_SMALL
    if (format == 3)
	statusbar(
		P_("Read %lu line (Converted from DOS and Mac format)",
		"Read %lu lines (Converted from DOS and Mac format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
    else if (format == 2) {
470
	openfile->fmt = MAC_FILE;
471
472
473
474
	statusbar(P_("Read %lu line (Converted from Mac format)",
		"Read %lu lines (Converted from Mac format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
    } else if (format == 1) {
475
	openfile->fmt = DOS_FILE;
476
477
478
479
	statusbar(P_("Read %lu line (Converted from DOS format)",
		"Read %lu lines (Converted from DOS format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
    } else
480
#endif
481
482
483
	statusbar(P_("Read %lu line", "Read %lu lines",
		(unsigned long)num_lines), (unsigned long)num_lines);
}
Chris Allegretta's avatar
Chris Allegretta committed
484

485
486
487
488
489
490
491
492
493
494
495
496
/* Open the file (and decide if it exists).  If newfie is TRUE, display
 * "New File" if the file is missing.  Otherwise, say "[filename] not
 * found".
 *
 * Return -2 if we say "New File".  Otherwise, -1 if the file isn't
 * opened, 0 otherwise.  The file might still have an error while
 * reading with a 0 return value.  *f is set to the opened file. */
int open_file(const char *filename, bool newfie, FILE **f)
{
    int fd;
    struct stat fileinfo;

497
    assert(filename != NULL && f != NULL);
498

499
    if (stat(filename, &fileinfo) == -1) {
500
501
502
503
504
505
506
	if (newfie) {
	    statusbar(_("New File"));
	    return -2;
	}
	statusbar(_("\"%s\" not found"), filename);
	return -1;
    } else if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
507
	S_ISBLK(fileinfo.st_mode)) {
508
	/* Don't open character or block files.  Sorry, /dev/sndstat! */
509
510
511
	statusbar(S_ISDIR(fileinfo.st_mode) ?
		_("\"%s\" is a directory") :
		_("File \"%s\" is a device file"), filename);
512
513
514
515
516
517
518
519
	return -1;
    } else if ((fd = open(filename, O_RDONLY)) == -1) {
	statusbar(_("Error reading %s: %s"), filename, strerror(errno));
 	return -1;
     } else {
	/* File is A-OK.  Open it in binary mode for our own end-of-line
	 * character munging. */
	*f = fdopen(fd, "rb");
520

521
522
523
524
525
526
	if (*f == NULL) {
	    statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
	    close(fd);
	} else
	    statusbar(_("Reading File"));
527
    }
528

529
    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
530
531
}

532
533
534
535
536
/* This function will return the name of the first available extension
 * of a filename (starting with [name][suffix], then [name][suffix].1,
 * etc.).  Memory is allocated for the return value.  If no writable
 * extension exists, we return "". */
char *get_next_filename(const char *name, const char *suffix)
537
{
538
539
540
    unsigned long i = 0;
    char *buf;
    size_t namelen, suffixlen;
541

542
    assert(name != NULL && suffix != NULL);
543

544
545
    namelen = strlen(name);
    suffixlen = strlen(suffix);
Chris Allegretta's avatar
Chris Allegretta committed
546

547
548
    buf = charalloc(namelen + suffixlen + digits(ULONG_MAX) + 2);
    sprintf(buf, "%s%s", name, suffix);
549

550
551
    while (TRUE) {
	struct stat fs;
Chris Allegretta's avatar
Chris Allegretta committed
552

553
554
555
556
	if (stat(buf, &fs) == -1)
	    return buf;
	if (i == ULONG_MAX)
	    break;
557

558
559
560
	i++;
	sprintf(buf + namelen + suffixlen, ".%lu", i);
    }
Chris Allegretta's avatar
Chris Allegretta committed
561

562
563
564
    /* We get here only if there is no possible save file.  Blank out
     * the filename to indicate this. */
    null_at(&buf, 0);
565

566
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
567
568
}

569
570
571
572
573
574
575
576
577
578
579
580
void do_insertfile(
#ifndef NANO_SMALL
	bool execute
#else
	void
#endif
	)
{
    int i;
    const char *msg;
    char *ans = mallocstrcpy(NULL, "");
	/* The last answer the user typed on the statusbar. */
581
582
    filestruct *edittop_save = openfile->edittop;
    ssize_t current_y_save = openfile->current_y;
583
584
    bool at_edittop = FALSE;
	/* Whether we're at the top of the edit window. */
585

586
587
588
#ifndef DISABLE_WRAPPING
    wrap_reset();
#endif
589

590
591
592
    while (TRUE) {
#ifndef NANO_SMALL
	if (execute) {
593
	    msg = 
594
#ifdef ENABLE_MULTIBUFFER
595
596
		ISSET(MULTIBUFFER) ? 
		N_("Command to execute in new buffer [from %s] ") :
597
#endif
598
		N_("Command to execute [from %s] ");
599
600
	} else {
#endif
601
	    msg =
602
#ifdef ENABLE_MULTIBUFFER
603
604
		ISSET(MULTIBUFFER) ? 
		N_("File to insert into new buffer [from %s] ") :
605
#endif
606
		N_("File to insert [from %s] ");
607
608
609
#ifndef NANO_SMALL
	}
#endif
610

611
	i = statusq(TRUE,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
612
#ifndef NANO_SMALL
613
		execute ? extcmd_list :
614
#endif
615
616
617
		insertfile_list, ans,
#ifndef NANO_SMALL
		NULL,
618
#endif
619
620
621
622
623
624
		_(msg),
#ifndef DISABLE_OPERATINGDIR
		operating_dir != NULL && strcmp(operating_dir, ".") != 0 ?
		operating_dir :
#endif
		"./");
625

626
627
628
629
630
631
632
633
634
635
	/* If we're in multibuffer mode and the filename or command is
	 * blank, open a new buffer instead of canceling. */
	if (i == -1 || (i == -2
#ifdef ENABLE_MULTIBUFFER
		&& !ISSET(MULTIBUFFER)
#endif
		)) {
	    statusbar(_("Cancelled"));
	    break;
	} else {
636
	    size_t pww_save = openfile->placewewant;
637

638
	    ans = mallocstrcpy(ans, answer);
639

640
#ifndef NANO_SMALL
641
642
643
644
645
646
647
#ifdef ENABLE_MULTIBUFFER
	    if (i == TOGGLE_MULTIBUFFER_KEY) {
		/* Don't allow toggling if we're in view mode. */
		if (!ISSET(VIEW_MODE))
		    TOGGLE(MULTIBUFFER);
		continue;
	    } else
648
#endif
649
650
651
652
653
654
655
656
	    if (i == NANO_TOOTHERINSERT_KEY) {
		execute = !execute;
		continue;
	    }
#ifndef DISABLE_BROWSER
	    else
#endif
#endif /* !NANO_SMALL */
657

658
659
660
#ifndef DISABLE_BROWSER
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
661

662
663
		if (tmp == NULL)
		    continue;
664

665
666
		free(answer);
		answer = tmp;
667

668
669
670
671
672
673
		/* We have a file now.  Indicate this and get out of the
		 * statusbar prompt cleanly. */
		i = 0;
		statusq_abort();
	    }
#endif
674

675
676
677
678
679
680
681
682
	    /* If we don't have a file yet, go back to the statusbar
	     * prompt. */
	    if (i != 0
#ifdef ENABLE_MULTIBUFFER
		&& (i != -2 || !ISSET(MULTIBUFFER))
#endif
		)
		continue;
683

684
685
686
687
688
689
690
691
#ifdef ENABLE_MULTIBUFFER
	    if (!ISSET(MULTIBUFFER)) {
#endif
		/* If we're not inserting into a new buffer, partition
		 * the filestruct so that it contains no text and hence
		 * looks like a new buffer, and keep track of whether
		 * the top of the partition is the top of the edit
		 * window. */
692
693
694
695
696
		filepart = partition_filestruct(openfile->current,
			openfile->current_x, openfile->current,
			openfile->current_x);
		at_edittop =
			(openfile->fileage == openfile->edittop);
697
698
699
#ifdef ENABLE_MULTIBUFFER
	    }
#endif
Chris Allegretta's avatar
Chris Allegretta committed
700
701

#ifndef NANO_SMALL
702
703
704
	    if (execute) {
#ifdef ENABLE_MULTIBUFFER
		if (ISSET(MULTIBUFFER))
705
		    /* Open a blank buffer. */
706
707
708
709
		    open_buffer("");
#endif

		/* Save the command's output in the current buffer. */
710
		execute_command(answer);
711
	    } else {
712
#endif
713
714
		/* Make sure the path to the file specified in answer is
		 * tilde-expanded. */
715
716
		answer = mallocstrassn(answer,
			real_dir_from_tilde(answer));
717
718
719

		/* Save the file specified in answer in the current
		 * buffer. */
720
		open_buffer(answer);
721
722
#ifndef NANO_SMALL
	    }
Chris Allegretta's avatar
Chris Allegretta committed
723
#endif
724

725
#ifdef ENABLE_MULTIBUFFER
726
727
728
	    if (ISSET(MULTIBUFFER))
		/* Update the screen to account for the current
		 * buffer. */
729
		display_buffer();
730
	    else
Chris Allegretta's avatar
Chris Allegretta committed
731
#endif
732
	    {
733
		filestruct *top_save = openfile->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
734

735
736
737
738
739
740
		/* If we didn't insert into a new buffer, and we were at
		 * the top of the edit window before, set the saved
		 * value of edittop to the new top of the edit window,
		 * and update the current y-coordinate to account for
		 * the number of lines inserted. */
		if (at_edittop)
741
742
		    edittop_save = openfile->fileage;
		openfile->current_y += current_y_save;
743

744
745
746
747
748
749
		/* If we didn't insert into a new buffer, unpartition
		 * the filestruct so that it contains all the text
		 * again.  Note that we've replaced the non-text
		 * originally in the partition with the text in the
		 * inserted file/executed command output. */
		unpartition_filestruct(&filepart);
750

751
752
753
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
754

755
		/* Restore the old edittop. */
756
		openfile->edittop = edittop_save;
757

758
759
760
		/* Restore the old place we want. */
		openfile->placewewant = pww_save;

761
762
		/* Mark the file as modified. */
		set_modified();
763

764
765
		/* Update the screen. */
		edit_refresh();
766
	    }
767

768
769
770
771
772
	    break;
	}
    }

    free(ans);
773
774
}

775
void do_insertfile_void(void)
776
{
777
778
779
780
781
782
783
784
785
786
#ifdef ENABLE_MULTIBUFFER
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
	statusbar(_("Key illegal in non-multibuffer mode"));
    else
#endif
	do_insertfile(
#ifndef NANO_SMALL
		FALSE
#endif
		);
787
788
789
790

    display_main_list();
}

791
/* When passed "[relative path]" or "[relative path][filename]" in
792
 * origpath, return "[full path]" or "[full path][filename]" on success,
793
794
795
796
 * or NULL on error.  Do this if the file doesn't exist but the relative
 * path does, since the file could exist in memory but not yet on disk).
 * Don't do this if the relative path doesn't exist, since we won't be
 * able to go there. */
797
char *get_full_path(const char *origpath)
798
{
799
    char *d_here, *d_there = NULL;
800

801
802
    if (origpath == NULL)
    	return NULL;
803

804
805
806
    /* Get the current directory. */
    d_here = charalloc(PATH_MAX + 1);
    d_here = getcwd(d_here, PATH_MAX + 1);
807

808
    if (d_here != NULL) {
809
810
811
812
	const char *last_slash;
	char *d_there_file = NULL;
	bool path_only;
	struct stat fileinfo;
813

814
815
	align(&d_here);

816
817
	/* If the current directory isn't "/", tack a slash onto the end
	 * of it. */
818
	if (strcmp(d_here, "/") != 0) {
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
819
	    d_here = charealloc(d_here, strlen(d_here) + 2);
820
821
	    strcat(d_here, "/");
	}
822

823
824
825
826
827
828
829
830
831
832
833
834
	d_there = real_dir_from_tilde(origpath);

	assert(d_there != NULL);

	/* Stat d_there.  If stat() fails, assume that d_there refers to
	 * a new file that hasn't been saved to disk yet.  Set path_only
	 * to TRUE if d_there refers to a directory, and FALSE if
	 * d_there refers to a file. */
	path_only = !stat(d_there, &fileinfo) &&
		S_ISDIR(fileinfo.st_mode);

	/* If path_only is TRUE, make sure d_there ends in a slash. */
835
	if (path_only) {
836
837
838
839
	    size_t d_there_len = strlen(d_there);

	    if (d_there[d_there_len - 1] != '/') {
		d_there = charealloc(d_there, d_there_len + 2);
840
841
842
843
		strcat(d_there, "/");
	    }
	}

844
	/* Search for the last slash in d_there. */
845
846
	last_slash = strrchr(d_there, '/');

847
848
849
850
	/* If we didn't find one, then make sure the answer is in the
	 * format "d_here/d_there". */
	if (last_slash == NULL) {
	    assert(!path_only);
851

852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
	    d_there_file = d_there;
	    d_there = d_here;
	} else {
	    /* If path_only is FALSE, then save the filename portion of
	     * the answer, everything after the last slash, in
	     * d_there_file. */
	    if (!path_only)
		d_there_file = mallocstrcpy(NULL, last_slash + 1);

	    /* And remove the filename portion of the answer from
	     * d_there. */
	    null_at(&d_there, last_slash - d_there + 1);

	    /* Go to the path specified in d_there. */
	    if (chdir(d_there) == -1) {
867
		free(d_there);
868
869
870
871
		d_there = NULL;
	    } else {
		/* Get the full path and save it in d_there. */
		free(d_there);
872

873
874
875
		d_there = charalloc(PATH_MAX + 1);
		d_there = getcwd(d_there, PATH_MAX + 1);

876
877
878
879
880
881
882
883
884
885
		if (d_there != NULL) {
		    align(&d_there);

		    if (strcmp(d_there, "/") != 0) {
			/* Make sure d_there ends in a slash. */
			d_there = charealloc(d_there,
				strlen(d_there) + 2);
			strcat(d_there, "/");
		    }
		} else
886
887
888
889
890
891
892
893
		    /* If we couldn't get the full path, set path_only
		     * to TRUE so that we clean up correctly, free all
		     * allocated memory, and return NULL. */
		    path_only = TRUE;

		/* Finally, go back to the path specified in d_here,
		 * where we were before. */
		chdir(d_here);
894
895
	    }

896
897
	    /* Free d_here, since we're done using it. */
	    free(d_here);
898
	}
899

900
901
902
903
904
	/* At this point, if path_only is FALSE and d_there exists,
	 * d_there contains the path portion of the answer and
	 * d_there_file contains the filename portion of the answer.  If
	 * this is the case, tack d_there_file onto the end of
	 * d_there, so that d_there contains the complete answer. */
905
	if (!path_only && d_there != NULL) {
906
907
908
909
910
911
	    d_there = charealloc(d_there, strlen(d_there) +
		strlen(d_there_file) + 1);
	    strcat(d_there, d_there_file);
 	}

	/* Free d_there_file, since we're done using it. */
912
913
914
	free(d_there_file);
    }

915
    return d_there;
916
}
917

918
919
920
/* Return the full version of path, as returned by get_full_path().  On
 * error, if path doesn't reference a directory, or if the directory
 * isn't writable, return NULL. */
921
char *check_writable_directory(const char *path)
922
{
923
924
    char *full_path = get_full_path(path);

925
    /* If get_full_path() fails, return NULL. */
926
    if (full_path == NULL)
927
	return NULL;
928

929
930
931
    /* If we can't write to path or path isn't a directory, return
     * NULL. */
    if (access(full_path, W_OK) != 0 ||
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
932
	full_path[strlen(full_path) - 1] != '/') {
933
	free(full_path);
934
	return NULL;
935
    }
936

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
937
    /* Otherwise, return the full path. */
938
939
940
    return full_path;
}

941
942
/* This function calls mkstemp(($TMPDIR|P_tmpdir|/tmp/)"nano.XXXXXX").
 * On success, it returns the malloc()ed filename and corresponding FILE
943
 * stream, opened in "r+b" mode.  On error, it returns NULL for the
944
945
 * filename and leaves the FILE stream unchanged. */
char *safe_tempfile(FILE **f)
946
{
947
    char *full_tempdir = NULL;
948
949
950
    const char *tmpdir_env;
    int fd;
    mode_t original_umask = 0;
951

952
953
    assert(f != NULL);

954
955
956
    /* If $TMPDIR is set and non-empty, set tempdir to it, run it
     * through get_full_path(), and save the result in full_tempdir.
     * Otherwise, leave full_tempdir set to NULL. */
957
958
959
    tmpdir_env = getenv("TMPDIR");
    if (tmpdir_env != NULL && tmpdir_env[0] != '\0')
	full_tempdir = check_writable_directory(tmpdir_env);
960

961
962
    /* If $TMPDIR is unset, empty, or not a writable directory, and
     * full_tempdir is NULL, try P_tmpdir instead. */
963
    if (full_tempdir == NULL)
964
	full_tempdir = check_writable_directory(P_tmpdir);
965

966
967
968
    /* if P_tmpdir is NULL, use /tmp. */
    if (full_tempdir == NULL)
	full_tempdir = mallocstrcpy(NULL, "/tmp/");
969

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
970
    full_tempdir = charealloc(full_tempdir, strlen(full_tempdir) + 12);
971
    strcat(full_tempdir, "nano.XXXXXX");
972

973
974
975
976
977
978
979
    original_umask = umask(0);
    umask(S_IRWXG | S_IRWXO);

    fd = mkstemp(full_tempdir);

    if (fd != -1)
	*f = fdopen(fd, "r+b");
980
981
982
    else {
	free(full_tempdir);
	full_tempdir = NULL;
983
    }
984

985
986
    umask(original_umask);

987
    return full_tempdir;
988
}
989
990

#ifndef DISABLE_OPERATINGDIR
991
992
993
994
995
/* Initialize full_operating_dir based on operating_dir. */
void init_operating_dir(void)
{
    assert(full_operating_dir == NULL);

996
    if (operating_dir == NULL)
997
	return;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
998

999
1000
1001
    full_operating_dir = get_full_path(operating_dir);

    /* If get_full_path() failed or the operating directory is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1002
     * inaccessible, unset operating_dir. */
1003
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) {
1004
1005
1006
1007
1008
1009
1010
	free(full_operating_dir);
	full_operating_dir = NULL;
	free(operating_dir);
	operating_dir = NULL;
    }
}

1011
1012
1013
1014
1015
/* Check to see if we're inside the operating directory.  Return FALSE
 * if we are, or TRUE otherwise.  If allow_tabcomp is TRUE, allow
 * incomplete names that would be matches for the operating directory,
 * so that tab completion will work. */
bool check_operating_dir(const char *currpath, bool allow_tabcomp)
1016
{
1017
1018
1019
1020
    /* The char *full_operating_dir is global for mem cleanup.  It
     * should have already been initialized by init_operating_dir().
     * Also, a relative operating directory path will only be handled
     * properly if this is done. */
1021

1022
    char *fullpath;
1023
    bool retval = FALSE;
1024
    const char *whereami1, *whereami2 = NULL;
1025

1026
    /* If no operating directory is set, don't bother doing anything. */
1027
    if (operating_dir == NULL)
1028
	return FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1029

1030
    assert(full_operating_dir != NULL);
1031
1032

    fullpath = get_full_path(currpath);
1033
1034

    /* fullpath == NULL means some directory in the path doesn't exist
1035
     * or is unreadable.  If allow_tabcomp is FALSE, then currpath is
1036
1037
     * what the user typed somewhere.  We don't want to report a
     * non-existent directory as being outside the operating directory,
1038
     * so we return FALSE.  If allow_tabcomp is TRUE, then currpath
1039
1040
     * exists, but is not executable.  So we say it isn't in the
     * operating directory. */
1041
    if (fullpath == NULL)
1042
	return allow_tabcomp;
1043
1044
1045
1046
1047

    whereami1 = strstr(fullpath, full_operating_dir);
    if (allow_tabcomp)
	whereami2 = strstr(full_operating_dir, fullpath);

1048
    /* If both searches failed, we're outside the operating directory.
1049
     * Otherwise, check the search results.  If the full operating
1050
1051
1052
     * directory path is not at the beginning of the full current path
     * (for normal usage) and vice versa (for tab completion, if we're
     * allowing it), we're outside the operating directory. */
1053
    if (whereami1 != fullpath && whereami2 != full_operating_dir)
1054
1055
	retval = TRUE;
    free(fullpath);
1056
1057

    /* Otherwise, we're still inside it. */
1058
    return retval;
1059
}
1060
1061
#endif

1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
1084
1085
#ifndef NANO_SMALL
void init_backup_dir(void)
{
    char *full_backup_dir;

    if (backup_dir == NULL)
	return;

    full_backup_dir = get_full_path(backup_dir);

    /* If get_full_path() failed or the backup directory is
     * inaccessible, unset backup_dir. */
    if (full_backup_dir == NULL ||
	full_backup_dir[strlen(full_backup_dir) - 1] != '/') {
	free(full_backup_dir);
	free(backup_dir);
	backup_dir = NULL;
    } else {
	free(backup_dir);
	backup_dir = full_backup_dir;
    }
}
#endif

1086
/* Read from inn, write to out.  We assume inn is opened for reading,
1087
1088
 * and out for writing.  We return 0 on success, -1 on read error, or -2
 * on write error. */
1089
1090
int copy_file(FILE *inn, FILE *out)
{
1091
    char buf[BUFSIZ];
1092
1093
1094
1095
    size_t charsread;
    int retval = 0;

    assert(inn != NULL && out != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1096

1097
1098
1099
1100
1101
1102
1103
1104
1105
1106
1107
    do {
	charsread = fread(buf, sizeof(char), BUFSIZ, inn);
	if (charsread == 0 && ferror(inn)) {
	    retval = -1;
	    break;
	}
	if (fwrite(buf, sizeof(char), charsread, out) < charsread) {
	    retval = -2;
	    break;
	}
    } while (charsread > 0);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1108

1109
1110
1111
1112
    if (fclose(inn) == EOF)
	retval = -1;
    if (fclose(out) == EOF)
	retval = -2;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1113

1114
1115
1116
    return retval;
}

1117
1118
1119
/* Write a file out.  If f_open isn't NULL, we assume that it is a
 * stream associated with the file, and we don't try to open it
 * ourselves.  If tmp is TRUE, we set the umask to disallow anyone else
1120
1121
 * from accessing the file, we don't set the filename to its name, and
 * we don't print out how many lines we wrote on the statusbar.
Chris Allegretta's avatar
Chris Allegretta committed
1122
 *
1123
1124
 * tmp means we are writing a temporary file in a secure fashion.  We
 * use it when spell checking or dumping the file on an error.
1125
 *
1126
1127
 * append == APPEND means we are appending instead of overwriting.
 * append == PREPEND means we are prepending instead of overwriting.
1128
 *
1129
1130
 * nonamechange means don't change the current filename.  It is ignored
 * if tmp is FALSE or if we're appending/prepending.
1131
 *
1132
 * Return 0 on success or -1 on error. */
1133
1134
int write_file(const char *name, FILE *f_open, bool tmp, append_type
	append, bool nonamechange)
Chris Allegretta's avatar
Chris Allegretta committed
1135
{
1136
1137
    int retval = -1;
	/* Instead of returning in this function, you should always
1138
1139
	 * merely set retval and then goto cleanup_and_exit. */
    size_t lineswritten = 0;
1140
    const filestruct *fileptr = openfile->fileage;
1141
    int fd;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1142
	/* The file descriptor we use. */
1143
    mode_t original_umask = 0;
1144
	/* Our umask, from when nano started. */
1145
    bool realexists;
1146
	/* The result of stat().  TRUE if the file exists, FALSE
1147
	 * otherwise.  If name is a link that points nowhere, realexists
1148
	 * is FALSE. */
1149
1150
    struct stat st;
	/* The status fields filled in by stat(). */
1151
    bool anyexists;
1152
1153
	/* The result of lstat().  The same as realexists, unless name
	 * is a link. */
1154
1155
1156
    struct stat lst;
	/* The status fields filled in by lstat(). */
    char *realname;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1157
	/* name after tilde expansion. */
1158
    FILE *f = NULL;
1159
1160
1161
	/* The actual file, realname, we are writing to. */
    char *tempname = NULL;
	/* The temp file name we write to on prepend. */
Chris Allegretta's avatar
Chris Allegretta committed
1162

1163
    assert(name != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1164

1165
    if (name[0] == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1166
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1167

1168
1169
1170
    if (f_open != NULL)
	f = f_open;

1171
1172
    if (!tmp)
	titlebar(NULL);
1173

1174
    realname = real_dir_from_tilde(name);
Chris Allegretta's avatar
Chris Allegretta committed
1175

1176
#ifndef DISABLE_OPERATINGDIR
1177
    /* If we're writing a temporary file, we're probably going outside
1178
     * the operating directory, so skip the operating directory test. */
1179
    if (!tmp && check_operating_dir(realname, FALSE)) {
1180
1181
	statusbar(_("Can't write outside of %s"), operating_dir);
	goto cleanup_and_exit;
1182
1183
1184
    }
#endif

1185
    anyexists = (lstat(realname, &lst) != -1);
1186

1187
1188
    /* If the temp file exists and isn't already open, give up. */
    if (tmp && anyexists && f_open == NULL)
1189
	goto cleanup_and_exit;
1190

1191
1192
1193
    /* If NOFOLLOW_SYMLINKS is set, it doesn't make sense to prepend or
     * append to a symlink.  Here we warn about the contradiction. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode)) {
1194
1195
	statusbar(
		_("Cannot prepend or append to a symlink with --nofollow set"));
1196
1197
1198
	goto cleanup_and_exit;
    }

1199
    /* Save the state of file at the end of the symlink (if there is
1200
     * one). */
1201
    realexists = (stat(realname, &st) != -1);
1202

1203
1204
#ifndef NANO_SMALL
    /* We backup only if the backup toggle is set, the file isn't
1205
1206
1207
1208
     * temporary, and the file already exists.  Furthermore, if we
     * aren't appending, prepending, or writing a selection, we backup
     * only if the file has not been modified by someone else since nano
     * opened it. */
1209
1210
1211
    if (ISSET(BACKUP_FILE) && !tmp && realexists && ((append !=
	OVERWRITE || openfile->mark_set) ||
	openfile->current_stat->st_mtime == st.st_mtime)) {
1212
	FILE *backup_file;
1213
	char *backupname;
1214
	struct utimbuf filetime;
1215
	int copy_status;
1216

1217
	/* Save the original file's access and modification times. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1218
1219
	filetime.actime = openfile->current_stat->st_atime;
	filetime.modtime = openfile->current_stat->st_mtime;
1220

1221
1222
1223
	if (f_open == NULL) {
	    /* Open the original file to copy to the backup. */
	    f = fopen(realname, "rb");
1224

1225
1226
1227
1228
1229
	    if (f == NULL) {
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
		goto cleanup_and_exit;
	    }
1230
1231
	}

1232
	/* If backup_dir is set, we set backupname to
1233
1234
1235
1236
	 * backup_dir/backupname~[.number], where backupname is the
	 * canonicalized absolute pathname of realname with every '/'
	 * replaced with a '!'.  This means that /home/foo/file is
	 * backed up in backup_dir/!home!foo!file~[.number]. */
1237
	if (backup_dir != NULL) {
1238
	    char *backuptemp = get_full_path(realname);
1239

1240
	    if (backuptemp == NULL)
1241
1242
1243
1244
1245
1246
		/* If get_full_path() failed, we don't have a
		 * canonicalized absolute pathname, so just use the
		 * filename portion of the pathname.  We use tail() so
		 * that e.g. ../backupname will be backed up in
		 * backupdir/backupname~ instead of
		 * backupdir/../backupname~. */
1247
		backuptemp = mallocstrcpy(NULL, tail(realname));
1248
	    else {
1249
1250
		size_t i = 0;

1251
1252
1253
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1254
1255
1256
1257
		}
	    }

	    backupname = charalloc(strlen(backup_dir) +
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
		strlen(backuptemp) + 1);
	    sprintf(backupname, "%s%s", backup_dir, backuptemp);
	    free(backuptemp);
	    backuptemp = get_next_filename(backupname, "~");
	    if (backuptemp[0] == '\0') {
		statusbar(_("Error writing %s: %s"), backupname,
		    _("Too many backup files?"));
		free(backuptemp);
		free(backupname);
		fclose(f);
		goto cleanup_and_exit;
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1273
1274
1275
1276
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1277

1278
1279
1280
	/* Open the destination backup file.  Before we write to it, we
	 * set its permissions, so no unauthorized person can read it as
	 * we write. */
1281
	backup_file = fopen(backupname, "wb");
1282

1283
	if (backup_file == NULL || chmod(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1284
		openfile->current_stat->st_mode) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1285
1286
	    statusbar(_("Error writing %s: %s"), backupname,
		strerror(errno));
1287
	    free(backupname);
1288
1289
1290
1291
	    if (backup_file != NULL)
		fclose(backup_file);
	    fclose(f);
	    goto cleanup_and_exit;
1292
1293
1294
	}

#ifdef DEBUG
1295
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1296
1297
#endif

1298
1299
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1300

1301
	/* And set metadata. */
1302
	if (copy_status != 0 || chown(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1303
1304
		openfile->current_stat->st_uid,
		openfile->current_stat->st_gid) == -1 ||
1305
		utime(backupname, &filetime) == -1) {
1306
1307
	    free(backupname);
	    if (copy_status == -1)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1308
1309
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1310
1311
1312
1313
1314
	    else
		statusbar(_("Error writing %s: %s"), backupname,
			strerror(errno));
	    goto cleanup_and_exit;
	}
1315

1316
1317
	free(backupname);
    }
1318
#endif /* !NANO_SMALL */
1319

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1320
1321
    /* If NOFOLLOW_SYMLINKS is set and the file is a link, we aren't
     * doing prepend or append.  So we delete the link first, and just
1322
1323
1324
1325
     * overwrite. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode) &&
	unlink(realname) == -1) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1326
	goto cleanup_and_exit;
1327
    }
1328

1329
1330
    if (f_open == NULL) {
	original_umask = umask(0);
1331

1332
	/* If we create a temp file, we don't let anyone else access it.
1333
1334
	 * We create a temp file if tmp is TRUE. */
	if (tmp)
1335
	    umask(S_IRWXG | S_IRWXO);
1336
1337
	else
	    umask(original_umask);
1338
    }
1339

1340
    /* If we're prepending, copy the file to a temp file. */
1341
    if (append == PREPEND) {
1342
1343
1344
	int fd_source;
	FILE *f_source = NULL;

1345
	tempname = safe_tempfile(&f);
1346

1347
1348
	if (tempname == NULL) {
	    statusbar(_("Prepending to %s failed: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1349
		strerror(errno));
1350
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1351
	}
1352

1353
1354
	if (f_open == NULL) {
	    fd_source = open(realname, O_RDONLY | O_CREAT);
1355

1356
1357
1358
1359
1360
1361
1362
1363
1364
1365
1366
	    if (fd_source != -1) {
		f_source = fdopen(fd_source, "rb");
		if (f_source == NULL) {
		    statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
		    close(fd_source);
		    fclose(f);
		    unlink(tempname);
		    goto cleanup_and_exit;
		}
	    }
1367
1368
1369
	}

	if (copy_file(f_source, f) != 0) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1370
1371
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
1372
	    unlink(tempname);
1373
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1374
1375
1376
	}
    }

1377
1378
1379
1380
    if (f_open == NULL) {
	/* Now open the file in place.  Use O_EXCL if tmp is TRUE.  This
	 * is copied from joe, because wiggy says so *shrug*. */
	fd = open(realname, O_WRONLY | O_CREAT |
1381
1382
1383
		((append == APPEND) ? O_APPEND : (tmp ? O_EXCL :
		O_TRUNC)), S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP |
		S_IROTH | S_IWOTH);
1384

1385
1386
	/* Set the umask back to the user's original value. */
	umask(original_umask);
1387

1388
1389
1390
1391
	/* If we couldn't open the file, give up. */
	if (fd == -1) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1392

1393
1394
1395
1396
1397
	    /* tempname has been set only if we're prepending. */
	    if (tempname != NULL)
		unlink(tempname);
	    goto cleanup_and_exit;
	}
1398

1399
	f = fdopen(fd, (append == APPEND) ? "ab" : "wb");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1400

1401
1402
1403
1404
1405
1406
	if (f == NULL) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
	    close(fd);
	    goto cleanup_and_exit;
	}
1407
1408
    }

1409
    /* There might not be a magicline.  There won't be when writing out
1410
     * a selection. */
1411
    assert(openfile->fileage != NULL && openfile->filebot != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1412

1413
    while (fileptr != NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1414
	size_t data_len = strlen(fileptr->data), size;
1415

1416
	/* Newlines to nulls, just before we write to disk. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1417
1418
	sunder(fileptr->data);

1419
	size = fwrite(fileptr->data, sizeof(char), data_len, f);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1420

1421
	/* Nulls to newlines; data_len is the string's real length. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1422
1423
	unsunder(fileptr->data, data_len);

1424
	if (size < data_len) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1425
1426
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1427
	    fclose(f);
1428
	    goto cleanup_and_exit;
1429
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1430

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1431
1432
1433
	/* If we're on the last line of the file, don't write a newline
	 * character after it. */
	if (fileptr != openfile->filebot) {
1434
#ifndef NANO_SMALL
1435
1436
1437
1438
	    if (openfile->fmt == DOS_FILE || openfile->fmt ==
		MAC_FILE) {
		if (putc('\r', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1439
			strerror(errno));
1440
1441
1442
		    fclose(f);
		    goto cleanup_and_exit;
		}
1443
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1444

1445
	    if (openfile->fmt != MAC_FILE) {
1446
#endif
1447
1448
		if (putc('\n', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1449
			strerror(errno));
1450
1451
1452
		    fclose(f);
		    goto cleanup_and_exit;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1453
#ifndef NANO_SMALL
1454
	    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1455
#endif
1456
	}
Chris Allegretta's avatar
Chris Allegretta committed
1457
1458
1459
1460
1461

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

1462
    /* If we're prepending, open the temp file, and append it to f. */
1463
    if (append == PREPEND) {
1464
1465
1466
1467
	int fd_source;
	FILE *f_source = NULL;

	fd_source = open(tempname, O_RDONLY | O_CREAT);
1468

1469
1470
1471
1472
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
1473
	}
1474

1475
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1476
1477
	    statusbar(_("Error reading %s: %s"), tempname,
		strerror(errno));
1478
	    fclose(f);
1479
	    goto cleanup_and_exit;
1480
1481
	}

1482
	if (copy_file(f_source, f) == -1 || unlink(tempname) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1483
1484
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1485
	    goto cleanup_and_exit;
1486
	}
1487
1488
1489
1490
    } else if (fclose(f) == EOF) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
	unlink(tempname);
	goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1491
    }
1492

1493
    if (!tmp && append == OVERWRITE) {
Chris Allegretta's avatar
Chris Allegretta committed
1494
	if (!nonamechange) {
1495
1496
	    openfile->filename = mallocstrcpy(openfile->filename,
		realname);
Chris Allegretta's avatar
Chris Allegretta committed
1497
#ifdef ENABLE_COLOR
1498
	    /* We might have changed the filename, so update the colors
1499
1500
	     * to account for it, and then make sure we're using
	     * them. */
1501
	    color_update();
1502
	    color_init();
1503
1504
1505
1506
1507

	    /* If color syntaxes are available and turned on, we need to
	     * call edit_refresh(). */
	    if (openfile->colorstrings != NULL &&
		!ISSET(NO_COLOR_SYNTAX))
1508
		edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
1509
1510
#endif
	}
1511

1512
#ifndef NANO_SMALL
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1513
1514
1515
1516
1517
	/* Update current_stat to reference the file as it is now. */
	if (openfile->current_stat == NULL)
	    openfile->current_stat =
		(struct stat *)nmalloc(sizeof(struct stat));
	stat(realname, openfile->current_stat);
1518
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1519

1520
1521
	statusbar(P_("Wrote %lu line", "Wrote %lu lines",
		(unsigned long)lineswritten),
1522
		(unsigned long)lineswritten);
1523
	openfile->modified = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
1524
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
1525
    }
1526

1527
    retval = 0;
1528
1529
1530

  cleanup_and_exit:
    free(realname);
1531
    free(tempname);
1532

1533
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1534
1535
}

1536
#ifndef NANO_SMALL
1537
/* Write a marked selection from a file out. */
1538
1539
int write_marked_file(const char *name, FILE *f_open, bool tmp,
	append_type append)
1540
1541
{
    int retval = -1;
1542
1543
    bool old_modified = openfile->modified;
	/* write_file() unsets the modified flag. */
1544
1545
1546
1547
1548
    bool added_magicline;
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

1549
1550
    assert(openfile->mark_set);

1551
1552
1553
    /* Partition the filestruct so that it contains only the marked
     * text. */
    mark_order((const filestruct **)&top, &top_x,
1554
	(const filestruct **)&bot, &bot_x, NULL);
1555
    filepart = partition_filestruct(top, top_x, bot, bot_x);
1556
1557

    /* If the line at filebot is blank, treat it as the magicline and
1558
1559
     * hence the end of the file.  Otherwise, add a magicline and treat
     * it as the end of the file. */
1560
    added_magicline = (openfile->filebot->data[0] != '\0');
1561
1562
    if (added_magicline)
	new_magicline();
1563

1564
    retval = write_file(name, f_open, tmp, append, TRUE);
1565

1566
1567
1568
1569
1570
1571
    /* If we added a magicline, remove it now. */
    if (added_magicline)
	remove_magicline();

    /* Unpartition the filestruct so that it contains all the text
     * again. */
1572
    unpartition_filestruct(&filepart);
1573

1574
    if (old_modified)
1575
1576
1577
1578
1579
1580
	set_modified();

    return retval;
}
#endif /* !NANO_SMALL */

1581
int do_writeout(bool exiting)
Chris Allegretta's avatar
Chris Allegretta committed
1582
{
1583
1584
    int i, retval = 0;
    append_type append = OVERWRITE;
1585
1586
    char *ans;
	/* The last answer the user typed on the statusbar. */
1587
#ifdef NANO_EXTRA
1588
    static bool did_credits = FALSE;
1589
#endif
Chris Allegretta's avatar
Chris Allegretta committed
1590

1591
    currshortcut = writefile_list;
1592

1593
    if (exiting && openfile->filename[0] != '\0' && ISSET(TEMP_FILE)) {
1594
	retval = write_file(openfile->filename, NULL, FALSE, OVERWRITE,
1595
		FALSE);
1596
1597

	/* Write succeeded. */
1598
	if (retval == 0)
1599
	    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1600
1601
    }

1602
    ans = mallocstrcpy(NULL,
1603
#ifndef NANO_SMALL
1604
	(openfile->mark_set && !exiting) ? "" :
1605
#endif
1606
	openfile->filename);
1607
1608
1609
1610

    while (TRUE) {
	const char *msg;
#ifndef NANO_SMALL
1611
1612
	const char *formatstr, *backupstr;

1613
1614
1615
1616
1617
	formatstr = (openfile->fmt == DOS_FILE) ?
		N_(" [DOS Format]") : (openfile->fmt == MAC_FILE) ?
		N_(" [Mac Format]") : "";

	backupstr = ISSET(BACKUP_FILE) ? N_(" [Backup]") : "";
1618

1619
	if (openfile->mark_set && !exiting)
1620
1621
1622
	    msg = (append == PREPEND) ?
		N_("Prepend Selection to File") : (append == APPEND) ?
		N_("Append Selection to File") :
1623
		N_("Write Selection to File");
1624
	else
1625
#endif /* !NANO_SMALL */
1626
1627
	    msg = (append == PREPEND) ? N_("File Name to Prepend to") :
		(append == APPEND) ? N_("File Name to Append to") :
1628
		N_("File Name to Write");
1629

1630
1631
1632
	/* If we're using restricted mode, the filename isn't blank,
	 * and we're at the "Write File" prompt, disable tab
	 * completion. */
1633
1634
	i = statusq(!ISSET(RESTRICTED) ||
		openfile->filename[0] == '\0', writefile_list, ans,
1635
#ifndef NANO_SMALL
1636
		NULL, "%s%s%s", _(msg), formatstr, backupstr
1637
#else
1638
		"%s", _(msg)
1639
1640
1641
#endif
		);

1642
	if (i < 0) {
1643
	    statusbar(_("Cancelled"));
1644
1645
1646
1647
	    retval = -1;
	    break;
	} else {
	    ans = mallocstrcpy(ans, answer);
Chris Allegretta's avatar
Chris Allegretta committed
1648

1649
#ifndef DISABLE_BROWSER
1650
1651
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
1652

1653
1654
1655
1656
		currshortcut = writefile_list;

		if (tmp == NULL)
		    continue;
1657

1658
1659
1660
1661
1662
1663
1664
		free(answer);
		answer = tmp;

		/* We have a file now.  Get out of the statusbar prompt
		 * cleanly. */
		statusq_abort();
	    } else
1665
#endif /* !DISABLE_BROWSER */
Chris Allegretta's avatar
Chris Allegretta committed
1666
#ifndef NANO_SMALL
1667
	    if (i == TOGGLE_DOS_KEY) {
1668
1669
		openfile->fmt = (openfile->fmt == DOS_FILE) ? NIX_FILE :
			DOS_FILE;
1670
1671
		continue;
	    } else if (i == TOGGLE_MAC_KEY) {
1672
1673
		openfile->fmt = (openfile->fmt == MAC_FILE) ? NIX_FILE :
			MAC_FILE;
1674
1675
1676
1677
1678
		continue;
	    } else if (i == TOGGLE_BACKUP_KEY) {
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
1679
#endif /* !NANO_SMALL */
1680
	    if (i == NANO_PREPEND_KEY) {
1681
		append = (append == PREPEND) ? OVERWRITE : PREPEND;
1682
1683
		continue;
	    } else if (i == NANO_APPEND_KEY) {
1684
		append = (append == APPEND) ? OVERWRITE : APPEND;
1685
1686
		continue;
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1687

Chris Allegretta's avatar
Chris Allegretta committed
1688
#ifdef DEBUG
1689
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
1690
#endif
1691
1692

#ifdef NANO_EXTRA
1693
	    if (exiting && !ISSET(TEMP_FILE) &&
1694
		strcasecmp(answer, "zzy") == 0 && !did_credits) {
1695
		do_credits();
1696
		did_credits = TRUE;
1697
1698
1699
		retval = -1;
		break;
	    }
1700
#endif
1701
	    if (append == OVERWRITE && strcmp(answer,
1702
		openfile->filename) != 0) {
1703
1704
1705
1706
1707
1708
1709
		struct stat st;

		if (!stat(answer, &st)) {
		    i = do_yesno(FALSE, _("File exists, OVERWRITE ? "));
		    if (i == 0 || i == -1)
			continue;
		/* If we're using restricted mode, we aren't allowed to
1710
		 * change the name of a file once it has one, because
1711
1712
1713
		 * that would allow reading from or writing to files not
		 * specified on the command line.  In this case, don't
		 * bother showing the "Different Name" prompt. */
1714
1715
		} else if (!ISSET(RESTRICTED) &&
			openfile->filename[0] != '\0'
1716
#ifndef NANO_SMALL
1717
			&& (exiting || !openfile->mark_set)
1718
#endif
1719
			) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1720
1721
		    i = do_yesno(FALSE,
			_("Save file under DIFFERENT NAME ? "));
1722
1723
1724
		    if (i == 0 || i == -1)
			continue;
		}
1725
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1726

1727
#ifndef NANO_SMALL
1728
1729
1730
1731
	    /* Here's where we allow the selected text to be written to
	     * a separate file.  If we're using restricted mode, this is
	     * disabled since it allows reading from or writing to files
	     * not specified on the command line. */
1732
	    if (!ISSET(RESTRICTED) && !exiting && openfile->mark_set)
1733
		retval = write_marked_file(answer, NULL, FALSE, append);
1734
	    else
1735
#endif /* !NANO_SMALL */
1736
		retval = write_file(answer, NULL, FALSE, append, FALSE);
1737

1738
1739
	    break;
	}
1740
    } /* while (TRUE) */
1741
1742

    free(ans);
1743

1744
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1745
1746
}

1747
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
1748
{
1749
    do_writeout(FALSE);
1750
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
1751
}
Chris Allegretta's avatar
Chris Allegretta committed
1752

1753
1754
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
1755
char *real_dir_from_tilde(const char *buf)
1756
{
Chris Allegretta's avatar
Chris Allegretta committed
1757
    char *dirtmp = NULL;
1758

1759
    assert(buf != NULL);
1760

1761
    if (buf[0] == '~') {
1762
	size_t i;
1763
	const char *tilde_dir;
1764

1765
	/* Figure out how much of the str we need to compare. */
1766
1767
1768
	for (i = 1; buf[i] != '/' && buf[i] != '\0'; i++)
	    ;

1769
1770
1771
1772
1773
1774
1775
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
	    tilde_dir = homedir;
	} else {
	    const struct passwd *userdata;

1776
1777
1778
	    do {
		userdata = getpwent();
	    } while (userdata != NULL &&
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1779
		strncmp(userdata->pw_name, buf + 1, i - 1) != 0);
1780
1781
	    endpwent();
	    tilde_dir = userdata->pw_dir;
Chris Allegretta's avatar
Chris Allegretta committed
1782
	}
1783

1784
1785
1786
	if (tilde_dir != NULL) {
	    dirtmp = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	    sprintf(dirtmp, "%s%s", tilde_dir, buf + i);
1787
	}
1788
    }
1789

1790
1791
    /* Set a default value for dirtmp, in case the user's home directory
     * isn't found. */
1792
    if (dirtmp == NULL)
1793
	dirtmp = mallocstrcpy(NULL, buf);
1794

1795
    return dirtmp;
1796
1797
}

1798
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
1799
/* Our sort routine for file listings.  Sort alphabetically and
1800
 * case-insensitively, and sort directories before filenames. */
1801
1802
1803
1804
1805
1806
1807
1808
1809
1810
1811
1812
1813
int diralphasort(const void *va, const void *vb)
{
    struct stat fileinfo;
    const char *a = *(const char *const *)va;
    const char *b = *(const char *const *)vb;
    bool aisdir = stat(a, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode);
    bool bisdir = stat(b, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode);

    if (aisdir && !bisdir)
	return -1;
    if (!aisdir && bisdir)
	return 1;

1814
    return mbstrcasecmp(a, b);
1815
}
1816
1817
1818
1819
1820
1821
1822
1823
1824

/* Free the memory allocated for array, which should contain len
 * elements. */
void free_chararray(char **array, size_t len)
{
    for (; len > 0; len--)
	free(array[len - 1]);
    free(array);
}
1825
1826
#endif

Chris Allegretta's avatar
Chris Allegretta committed
1827
#ifndef DISABLE_TABCOMP
1828
1829
/* Is the given file a directory? */
int is_dir(const char *buf)
1830
{
1831
    char *dirptr = real_dir_from_tilde(buf);
1832
1833
    struct stat fileinfo;

1834
    int ret = (stat(dirptr, &fileinfo) != -1 &&
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1835
	S_ISDIR(fileinfo.st_mode));
1836

1837
    assert(buf != NULL && dirptr != buf);
1838

1839
    free(dirptr);
1840

1841
    return ret;
1842
}
Chris Allegretta's avatar
Chris Allegretta committed
1843

1844
/* These functions (username_tab_completion(), cwd_tab_completion(), and
Chris Allegretta's avatar
Chris Allegretta committed
1845
 * input_tab()) were taken from busybox 0.46 (cmdedit.c).  Here is the
1846
 * notice from that file:
Chris Allegretta's avatar
Chris Allegretta committed
1847
1848
1849
1850
1851
1852
1853
1854
1855
1856
1857
1858
 *
 * 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.
1859
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
1860

1861
1862
1863
1864
/* We consider the first buflen characters of buf for ~username tab
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
	size_t buflen)
Chris Allegretta's avatar
Chris Allegretta committed
1865
{
1866
1867
    char **matches = NULL;
    const struct passwd *userdata;
1868

1869
    assert(buf != NULL && num_matches != NULL && buflen > 0);
1870

1871
    *num_matches = 0;
1872

1873
    while ((userdata = getpwent()) != NULL) {
1874
1875
1876
	if (strncmp(userdata->pw_name, buf + 1, buflen - 1) == 0) {
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
1877

1878
1879
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
1880
1881
1882
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
1883
1884
#endif

1885
1886
	    matches = (char **)nrealloc(matches,
		(*num_matches + 1) * sizeof(char *));
1887
1888
1889
	    matches[*num_matches] =
		charalloc(strlen(userdata->pw_name) + 2);
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
1890
	    ++(*num_matches);
1891
	}
1892
1893
    }
    endpwent();
1894

1895
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
1896
1897
}

1898
/* This was originally called exe_n_cwd_tab_completion(), but we're not
1899
1900
1901
 * worried about executables, only filenames :> */
char **cwd_tab_completion(const char *buf, size_t *num_matches, size_t
	buflen)
Chris Allegretta's avatar
Chris Allegretta committed
1902
{
1903
    char *dirname = mallocstrcpy(NULL, buf), *filename;
1904
1905
1906
1907
1908
#ifndef DISABLE_OPERATINGDIR
    size_t dirnamelen;
#endif
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
1909
    DIR *dir;
1910
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
1911

1912
    assert(dirname != NULL && num_matches != NULL && buflen >= 0);
1913

1914
1915
1916
1917
1918
1919
1920
1921
1922
1923
    *num_matches = 0;
    null_at(&dirname, buflen);

    /* Okie, if there's a / in the buffer, strip out the directory
     * part. */
    filename = strrchr(dirname, '/');
    if (filename != NULL) {
	char *tmpdirname = filename + 1;

	filename = mallocstrcpy(NULL, tmpdirname);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1924
	*tmpdirname = '\0';
1925
1926
1927
	tmpdirname = dirname;
	dirname = real_dir_from_tilde(dirname);
	free(tmpdirname);
Chris Allegretta's avatar
Chris Allegretta committed
1928
    } else {
1929
1930
	filename = dirname;
	dirname = mallocstrcpy(NULL, "./");
Chris Allegretta's avatar
Chris Allegretta committed
1931
1932
    }

1933
    assert(dirname[strlen(dirname) - 1] == '/');
1934

Chris Allegretta's avatar
Chris Allegretta committed
1935
    dir = opendir(dirname);
1936

1937
    if (dir == NULL) {
1938
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
1939
	beep();
1940
1941
1942
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
1943
    }
1944
1945
1946
1947
1948
1949

#ifndef DISABLE_OPERATINGDIR
    dirnamelen = strlen(dirname);
#endif
    filenamelen = strlen(filename);

1950
    while ((nextdir = readdir(dir)) != NULL) {
Chris Allegretta's avatar
Chris Allegretta committed
1951
#ifdef DEBUG
1952
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
1953
#endif
1954
	/* See if this matches. */
1955
1956
1957
1958
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
		(*filename == '.' ||
		(strcmp(nextdir->d_name, ".") != 0 &&
		strcmp(nextdir->d_name, "..") != 0))) {
1959
1960
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
1961
1962
1963

#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
1964
1965
1966
1967
1968
	     * directory, in which case just go to the next match.  To
	     * properly do operating directory checking, we have to add
	     * the directory name to the beginning of the proposed match
	     * before we check it. */
	    char *tmp2 = charalloc(strlen(dirname) +
1969
		strlen(nextdir->d_name) + 1);
1970

1971
	    sprintf(tmp2, "%s%s", dirname, nextdir->d_name);
1972
1973
1974
	    if (check_operating_dir(tmp2, TRUE)) {
		free(tmp2);
		continue;
1975
	    }
1976
	    free(tmp2);
1977
1978
#endif

1979
1980
	    matches = (char **)nrealloc(matches,
		(*num_matches + 1) * sizeof(char *));
1981
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
1982
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
1983
1984
	}
    }
1985

1986
1987
    closedir(dir);
    free(dirname);
1988
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
1989

1990
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
1991
1992
}

1993
1994
/* Do tab completion.  place refers to how much the statusbar cursor
 * position should be advanced. */
1995
char *input_tab(char *buf, size_t *place, bool *lastwastab, bool *list)
Chris Allegretta's avatar
Chris Allegretta committed
1996
{
1997
1998
    size_t num_matches = 0;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
1999

2000
    assert(buf != NULL && place != NULL && *place <= strlen(buf) && lastwastab != NULL && list != NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2001

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2002
    *list = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2003

2004
2005
2006
2007
    /* If the word starts with `~' and there is no slash in the word,
     * then try completing this word as a username. */
    if (*place > 0 && *buf == '~') {
	const char *bob = strchr(buf, '/');
Chris Allegretta's avatar
Chris Allegretta committed
2008

2009
2010
2011
2012
	if (bob == NULL || bob >= buf + *place)
	    matches = username_tab_completion(buf, &num_matches,
		*place);
    }
2013

2014
2015
2016
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
	matches = cwd_tab_completion(buf, &num_matches, *place);
2017

2018
2019
2020
2021
2022
    if (num_matches <= 0)
	beep();
    else {
	size_t match, common_len = 0;
	char *mzero;
2023
2024
2025
	const char *lastslash = revstrstr(buf, "/", buf + *place);
	size_t lastslash_len = (lastslash == NULL) ? 0 :
		lastslash - buf + 1;
2026
2027
2028
	char *match1_mb = charalloc(mb_cur_max() + 1);
	char *match2_mb = charalloc(mb_cur_max() + 1);
	int match1_mb_len, match2_mb_len;
2029
2030
2031

	while (TRUE) {
	    for (match = 1; match < num_matches; match++) {
2032
2033
		/* Get the number of single-byte characters that all the
		 * matches have in common. */
2034
		match1_mb_len = parse_mbchar(matches[0] + common_len,
2035
			match1_mb, NULL);
2036
		match2_mb_len = parse_mbchar(matches[match] +
2037
			common_len, match2_mb, NULL);
2038
2039
2040
		match1_mb[match1_mb_len] = '\0';
		match2_mb[match2_mb_len] = '\0';
		if (strcmp(match1_mb, match2_mb) != 0)
2041
2042
		    break;
	    }
2043

2044
	    if (match < num_matches || matches[0][common_len] == '\0')
2045
		break;
2046

2047
	    common_len += parse_mbchar(buf + common_len, NULL, NULL);
2048
	}
2049

2050
2051
2052
	free(match1_mb);
	free(match2_mb);

2053
2054
	mzero = charalloc(lastslash_len + common_len + 1);
	sprintf(mzero, "%.*s%.*s", lastslash_len, buf, common_len,
2055
		matches[0]);
2056

2057
	common_len += lastslash_len;
2058

2059
	assert(common_len >= *place);
2060

2061
2062
2063
	if (num_matches == 1 && is_dir(mzero)) {
	    mzero[common_len] = '/';
	    common_len++;
2064

2065
2066
	    assert(common_len > *place);
	}
2067

2068
2069
2070
	if (num_matches > 1 && (common_len != *place ||
		*lastwastab == FALSE))
	    beep();
2071

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2072
	/* If there is more of a match to display on the statusbar, show
2073
	 * it.  We reset lastwastab to FALSE: it requires pressing Tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2074
	 * twice in succession with no statusbar changes to see a match
2075
2076
	 * list. */
	if (common_len != *place) {
2077
	    size_t buf_len = strlen(buf);
2078
2079

	    *lastwastab = FALSE;
2080
	    buf = charealloc(buf, common_len + buf_len - *place + 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2081
	    charmove(buf + common_len, buf + *place,
2082
		buf_len - *place + 1);
2083
	    strncpy(buf, mzero, common_len);
2084
	    *place = common_len;
2085
2086
2087
	} else if (*lastwastab == FALSE || num_matches < 2)
	    *lastwastab = TRUE;
	else {
2088
	    int longest_name = 0, columns, editline = 0;
2089

2090
2091
2092
2093
2094
2095
2096
2097
	    /* Now we show a list of the available choices. */
	    assert(num_matches > 1);

	    /* Sort the list. */
	    qsort(matches, num_matches, sizeof(char *), diralphasort);

	    for (match = 0; match < num_matches; match++) {
		common_len = strnlenpt(matches[match], COLS - 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2098

2099
2100
		if (common_len > COLS - 1) {
		    longest_name = COLS - 1;
2101
2102
		    break;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2103

2104
2105
		if (common_len > longest_name)
		    longest_name = common_len;
2106
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2107

2108
	    assert(longest_name <= COLS - 1);
Chris Allegretta's avatar
Chris Allegretta committed
2109

2110
2111
2112
	    /* Each column will be (longest_name + 2) columns wide, i.e,
	     * two spaces between columns, except that there will be
	     * only one space after the last column. */
2113
	    columns = (COLS + 1) / (longest_name + 2);
2114

2115
2116
2117
2118
	    /* Blank the edit window, and print the matches out
	     * there. */
	    blank_edit();
	    wmove(edit, 0, 0);
2119

2120
2121
	    /* Disable el cursor. */
	    curs_set(0);
2122

2123
2124
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2125

2126
2127
		wmove(edit, editline, (longest_name + 2) *
			(match % columns));
2128

2129
2130
2131
		if (match % columns == 0 &&
			editline == editwinrows - 1 &&
			num_matches - match > columns) {
2132
2133
2134
		    waddstr(edit, _("(more)"));
		    break;
		}
2135

2136
2137
2138
2139
		disp = display_string(matches[match], 0, longest_name,
			FALSE);
		waddstr(edit, disp);
		free(disp);
2140

2141
		if ((match + 1) % columns == 0)
2142
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2143
	    }
2144

2145
	    wnoutrefresh(edit);
2146
	    *list = TRUE;
2147
2148
2149
	}

	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2150
2151
    }

2152
    free_chararray(matches, num_matches);
2153

2154
    /* Only refresh the edit window if we don't have a list of filename
2155
     * matches on it. */
2156
    if (*list == FALSE)
2157
	edit_refresh();
2158
2159

    /* Enable el cursor. */
2160
    curs_set(1);
2161

2162
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2163
}
2164
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2165

2166
2167
/* Only print the last part of a path.  Isn't there a shell command for
 * this? */
2168
2169
const char *tail(const char *foo)
{
2170
    const char *tmp = strrchr(foo, '/');
2171

2172
2173
2174
    if (tmp == NULL)
	tmp = foo;
    else if (*tmp == '/')
2175
2176
2177
2178
2179
	tmp++;

    return tmp;
}

2180
#ifndef DISABLE_BROWSER
2181
2182
/* Strip one directory from the end of path. */
void striponedir(char *path)
Chris Allegretta's avatar
Chris Allegretta committed
2183
{
2184
    char *tmp;
Chris Allegretta's avatar
Chris Allegretta committed
2185

2186
    assert(path != NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2187

2188
    tmp = strrchr(path, '/');
2189

2190
2191
    if (tmp != NULL)
 	*tmp = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
2192
2193
}

2194
2195
2196
2197
2198
2199
/* Return a list of files contained in the directory path.  *longest is
 * the maximum display length of a file, up to COLS - 1 (but at least
 * 7).  *numents is the number of files.  We assume path exists and is a
 * directory.  If neither is true, we return NULL. */
char **browser_init(const char *path, int *longest, size_t *numents, DIR
	*dir)
2200
{
2201
    const struct dirent *nextdir;
2202
    char **filelist;
2203
    size_t i = 0, path_len;
2204

2205
    assert(dir != NULL);
2206

2207
    *longest = 0;
2208

2209
    while ((nextdir = readdir(dir)) != NULL) {
2210
2211
	size_t dlen;

2212
	/* Don't show the "." entry. */
2213
	if (strcmp(nextdir->d_name, ".") == 0)
2214
	   continue;
2215
2216
	i++;

2217
	dlen = strlenpt(nextdir->d_name);
2218
	if (dlen > *longest)
2219
	    *longest = (dlen > COLS - 1) ? COLS - 1 : dlen;
2220
    }
2221
2222

    *numents = i;
2223
2224
2225
    rewinddir(dir);
    *longest += 10;

2226
    filelist = (char **)nmalloc(*numents * sizeof(char *));
2227

2228
2229
    path_len = strlen(path);

2230
2231
    i = 0;

2232
    while ((nextdir = readdir(dir)) != NULL && i < *numents) {
2233
	/* Don't show the "." entry. */
2234
	if (strcmp(nextdir->d_name, ".") == 0)
2235
2236
	   continue;

2237
2238
	filelist[i] = charalloc(path_len + strlen(nextdir->d_name) + 1);
	sprintf(filelist[i], "%s%s", path, nextdir->d_name);
2239
2240
	i++;
    }
2241
2242
2243
2244
2245

    /* Maybe the number of files in the directory changed between the
     * first time we scanned and the second.  i is the actual length of
     * filelist, so record it. */
    *numents = i;
2246
    closedir(dir);
2247
2248
2249

    if (*longest > COLS - 1)
	*longest = COLS - 1;
2250
2251
    if (*longest < 7)
	*longest = 7;
2252
2253
2254
2255

    return filelist;
}

2256
2257
2258
/* Our browser function.  path is the path to start browsing from.
 * Assume path has already been tilde-expanded. */
char *do_browser(char *path, DIR *dir)
Chris Allegretta's avatar
Chris Allegretta committed
2259
{
2260
    int kbinput, longest, selected, width;
2261
    bool meta_key, func_key, old_const_update = ISSET(CONST_UPDATE);
2262
2263
2264
2265
2266
2267
    size_t numents;
    char **filelist, *retval = NULL;

    curs_set(0);
    blank_statusbar();
    bottombars(browser_list);
2268
    wnoutrefresh(bottomwin);
2269
2270
2271
2272
2273

#if !defined(DISABLE_HELP) || !defined(DISABLE_MOUSE)
    /* Set currshortcut so the user can click in the shortcut area, and
     * so the browser help screen will come up. */
    currshortcut = browser_list;
2274
#endif
Chris Allegretta's avatar
Chris Allegretta committed
2275

2276
    UNSET(CONST_UPDATE);
2277

2278
2279
2280
2281
2282
2283
2284
2285
  change_browser_directory:
	/* We go here after the user selects a new directory. */

    kbinput = ERR;
    selected = 0;
    width = 0;

    path = mallocstrassn(path, get_full_path(path));
Chris Allegretta's avatar
Chris Allegretta committed
2286

2287
2288
    /* Assume that path exists and ends with a slash. */
    assert(path != NULL && path[strlen(path) - 1] == '/');
Chris Allegretta's avatar
Chris Allegretta committed
2289

2290
2291
2292
2293
    /* Get the list of files. */
    filelist = browser_init(path, &longest, &numents, dir);

    assert(filelist != NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2294

2295
    /* Sort the list. */
Chris Allegretta's avatar
Chris Allegretta committed
2296
2297
2298
    qsort(filelist, numents, sizeof(char *), diralphasort);

    titlebar(path);
Chris Allegretta's avatar
Chris Allegretta committed
2299
2300

    /* Loop invariant: Microsoft sucks. */
Chris Allegretta's avatar
Chris Allegretta committed
2301
    do {
2302
	bool abort = FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2303
	int j, col = 0, editline = 0, fileline;
2304
2305
2306
2307
	int filecols = 0;
	    /* Used only if width == 0, to calculate the number of files
	     * per row below. */
	struct stat st;
2308
	char *new_path;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2309
	    /* Used by the "Go To Directory" prompt. */
2310
2311
2312
#ifndef DISABLE_MOUSE
	MEVENT mevent;
#endif
Rocco Corsi's avatar
   
Rocco Corsi committed
2313

2314
	check_statusblank();
Rocco Corsi's avatar
   
Rocco Corsi committed
2315

2316
2317
	/* Compute the line number we're on now, so that we don't divide
	 * by zero later. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2318
	fileline = selected;
2319
	if (width != 0)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2320
	    fileline /= width;
Chris Allegretta's avatar
Chris Allegretta committed
2321
2322

	switch (kbinput) {
2323
#ifndef DISABLE_MOUSE
2324
2325
2326
	    case KEY_MOUSE:
		if (getmouse(&mevent) == ERR)
		    break;
2327

2328
2329
2330
		/* If we clicked in the edit window, we probably clicked
		 * on a file. */
		if (wenclose(edit, mevent.y, mevent.x)) {
2331
		    int old_selected = selected;
2332

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2333
		    /* Subtract out the size of topwin. */
2334
		    mevent.y -= 2 - no_more_space();
2335
2336
2337

		    /* longest is the width of each column.  There are
		     * two spaces between each column. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2338
		    selected = (fileline / editwinrows) * editwinrows *
2339
2340
2341
2342
2343
2344
2345
2346
2347
2348
2349
2350
2351
			width + mevent.y * width + mevent.x /
			(longest + 2);

		    /* If they clicked beyond the end of a row, select
		     * the end of that row. */
		    if (mevent.x > width * (longest + 2))
			selected--;

		    /* If we're off the screen, reset to the last item.
		     * If we clicked the same place as last time, select
		     * this name! */
		    if (selected > numents - 1)
			selected = numents - 1;
2352
		    else if (old_selected == selected)
2353
2354
2355
2356
2357
2358
2359
2360
			/* Put back the 'select' key. */
			unget_kbinput('s', FALSE, FALSE);
		} else {
		    /* We must have clicked a shortcut.  Put back the
		     * equivalent shortcut key. */
		    int mouse_x, mouse_y;
		    get_mouseinput(&mouse_x, &mouse_y, TRUE);
		}
2361

2362
		break;
2363
#endif
2364
2365
2366
2367
2368
2369
2370
2371
2372
2373
2374
2375
2376
2377
2378
2379
2380
2381
2382
	    case NANO_PREVLINE_KEY:
		if (selected >= width)
		    selected -= width;
		break;
	    case NANO_BACK_KEY:
		if (selected > 0)
		    selected--;
		break;
	    case NANO_NEXTLINE_KEY:
		if (selected + width <= numents - 1)
		    selected += width;
		break;
	    case NANO_FORWARD_KEY:
		if (selected < numents - 1)
		    selected++;
		break;
	    case NANO_PREVPAGE_KEY:
	    case NANO_PREVPAGE_FKEY:
	    case '-': /* Pico compatibility. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2383
		if (selected >= (editwinrows + fileline % editwinrows) *
2384
			width)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2385
		    selected -= (editwinrows + fileline % editwinrows) *
2386
2387
2388
2389
2390
2391
2392
			width;
		else
		    selected = 0;
		break;
	    case NANO_NEXTPAGE_KEY:
	    case NANO_NEXTPAGE_FKEY:
	    case ' ': /* Pico compatibility. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2393
		selected += (editwinrows - fileline % editwinrows) *
2394
2395
2396
2397
2398
2399
2400
			width;
		if (selected >= numents)
		    selected = numents - 1;
		break;
	    case NANO_HELP_KEY:
	    case NANO_HELP_FKEY:
	    case '?': /* Pico compatibility. */
2401
#ifndef DISABLE_HELP
2402
2403
		do_help();
		curs_set(0);
2404
#else
2405
		nano_disabled_msg();
2406
#endif
Rocco Corsi's avatar
   
Rocco Corsi committed
2407
		break;
2408
2409
2410
2411
2412
2413
2414
2415
2416
	    case NANO_ENTER_KEY:
	    case 'S': /* Pico compatibility. */
	    case 's':
		/* You can't move up from "/". */
		if (strcmp(filelist[selected], "/..") == 0) {
		    statusbar(_("Can't move up a directory"));
		    beep();
		    break;
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2417

2418
#ifndef DISABLE_OPERATINGDIR
2419
2420
2421
2422
2423
2424
		/* Note: the selected file can be outside the operating
		 * directory if it's ".." or if it's a symlink to a
		 * directory outside the operating directory. */
		if (check_operating_dir(filelist[selected], FALSE)) {
		    statusbar(
			_("Can't go outside of %s in restricted mode"),
2425
			operating_dir);
2426
2427
2428
		    beep();
		    break;
		}
2429
2430
#endif

2431
2432
2433
2434
2435
2436
		if (stat(filelist[selected], &st) == -1) {
		    statusbar(_("Error reading %s: %s"),
			filelist[selected], strerror(errno));
		    beep();
		    break;
		}
2437

2438
2439
2440
2441
2442
		if (!S_ISDIR(st.st_mode)) {
		    retval = mallocstrcpy(retval, filelist[selected]);
		    abort = TRUE;
		    break;
		}
2443

2444
2445
2446
2447
2448
2449
2450
		dir = opendir(filelist[selected]);
		if (dir == NULL) {
		    /* We can't open this dir for some reason.
		     * Complain. */
		    statusbar(_("Error reading %s: %s"),
			filelist[selected], strerror(errno));
		    break;
Chris Allegretta's avatar
Chris Allegretta committed
2451
		}
Chris Allegretta's avatar
Chris Allegretta committed
2452

2453
		path = mallocstrcpy(path, filelist[selected]);
2454

2455
		/* Start over again with the new path value. */
2456
		free_chararray(filelist, numents);
2457
2458
		goto change_browser_directory;

2459
2460
	    /* Refresh the screen. */
	    case NANO_REFRESH_KEY:
2461
		total_redraw();
2462
2463
		break;

2464
2465
2466
2467
2468
2469
2470
2471
	    /* Go to a specific directory. */
	    case NANO_GOTOLINE_KEY:
	    case NANO_GOTOLINE_FKEY:
	    case 'G': /* Pico compatibility. */
	    case 'g':
		curs_set(1);

		j = statusq(FALSE, gotodir_list, "",
Chris Allegretta's avatar
Chris Allegretta committed
2472
#ifndef NANO_SMALL
2473
			NULL,
Chris Allegretta's avatar
Chris Allegretta committed
2474
#endif
2475
			_("Go To Directory"));
Rocco Corsi's avatar
   
Rocco Corsi committed
2476

2477
2478
		curs_set(0);
		bottombars(browser_list);
Rocco Corsi's avatar
   
Rocco Corsi committed
2479

2480
2481
2482
2483
		if (j < 0) {
		    statusbar(_("Cancelled"));
		    break;
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2484

2485
2486
2487
2488
2489
2490
2491
		new_path = real_dir_from_tilde(answer);

		if (new_path[0] != '/') {
		    new_path = charealloc(new_path, strlen(path) +
			strlen(answer) + 1);
		    sprintf(new_path, "%s%s", path, answer);
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2492

2493
#ifndef DISABLE_OPERATINGDIR
2494
2495
2496
2497
2498
2499
2500
		if (check_operating_dir(new_path, FALSE)) {
		    statusbar(
			_("Can't go outside of %s in restricted mode"),
			operating_dir);
		    free(new_path);
		    break;
		}
2501
2502
#endif

2503
2504
2505
2506
2507
2508
2509
2510
2511
		dir = opendir(new_path);
		if (dir == NULL) {
		    /* We can't open this dir for some reason.
		     * Complain. */
		    statusbar(_("Error reading %s: %s"), answer,
			strerror(errno));
		    free(new_path);
		    break;
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2512

2513
2514
2515
		/* Start over again with the new path value. */
		free(path);
		path = new_path;
2516
		free_chararray(filelist, numents);
2517
2518
2519
2520
2521
2522
2523
2524
2525
		goto change_browser_directory;

	    /* Abort the browser. */
	    case NANO_EXIT_KEY:
	    case NANO_EXIT_FKEY:
	    case 'E': /* Pico compatibility. */
	    case 'e':
		abort = TRUE;
		break;
Chris Allegretta's avatar
Chris Allegretta committed
2526
	}
2527

Chris Allegretta's avatar
Chris Allegretta committed
2528
2529
2530
	if (abort)
	    break;

Rocco Corsi's avatar
   
Rocco Corsi committed
2531
2532
	blank_edit();

2533
	if (width != 0)
2534
2535
	    j = width * editwinrows *
		((selected / width) / editwinrows);
Chris Allegretta's avatar
Chris Allegretta committed
2536
	else
2537
	    j = 0;
Chris Allegretta's avatar
Chris Allegretta committed
2538
2539
2540

	wmove(edit, 0, 0);

2541
	{
2542
	    size_t foo_len = mb_cur_max() * 7;
2543
2544
2545
2546
2547
2548
2549
2550
2551
2552
	    char *foo = charalloc(foo_len + 1);

	    for (; j < numents && editline <= editwinrows - 1; j++) {
		char *disp = display_string(tail(filelist[j]), 0,
			longest, FALSE);

		/* Highlight the currently selected file/dir. */
		if (j == selected)
		    wattron(edit, A_REVERSE);

2553
		blank_line(edit, editline, col, longest);
2554
2555
2556
2557
2558
2559
2560
2561
2562
2563
2564
2565
2566
2567
2568
2569
		mvwaddstr(edit, editline, col, disp);
		free(disp);

		col += longest;
		filecols++;

		/* Show file info also.  We don't want to report file
		 * sizes for links, so we use lstat().  Also, stat() and
		 * lstat() return an error if, for example, the file is
		 * deleted while the file browser is open.  In that
		 * case, we report "--" as the file info. */
		if (lstat(filelist[j], &st) == -1 ||
			S_ISLNK(st.st_mode)) {
		    /* Aha!  It's a symlink!  Now, is it a dir?  If so,
		     * mark it as such. */
		    if (stat(filelist[j], &st) == 0 &&
2570
			S_ISDIR(st.st_mode)) {
2571
			strncpy(foo, _("(dir)"), foo_len);
2572
2573
			foo[foo_len] = '\0';
		    } else
2574
			strcpy(foo, "--");
2575
		} else if (S_ISDIR(st.st_mode)) {
2576
		    strncpy(foo, _("(dir)"), foo_len);
2577
2578
		    foo[foo_len] = '\0';
		} else if (st.st_size < (1 << 10)) /* less than 1 k. */
2579
2580
2581
2582
2583
2584
2585
2586
2587
2588
2589
2590
2591
2592
2593
2594
2595
2596
2597
2598
2599
2600
2601
2602
2603
2604
2605
2606
		    sprintf(foo, "%4u  B", (unsigned int)st.st_size);
		else if (st.st_size < (1 << 20)) /* less than 1 meg. */
		    sprintf(foo, "%4u KB",
			(unsigned int)(st.st_size >> 10));
		else if (st.st_size < (1 << 30)) /* less than 1 gig. */
		    sprintf(foo, "%4u MB",
			(unsigned int)(st.st_size >> 20));
		else
		    sprintf(foo, "%4u GB",
			(unsigned int)(st.st_size >> 30));

		mvwaddnstr(edit, editline, col - strlen(foo), foo,
			foo_len);

		if (j == selected)
		    wattroff(edit, A_REVERSE);

		/* Add some space between the columns. */
		col += 2;

		/* If the next entry isn't going to fit on the line,
		 * move to the next line. */
		if (col > COLS - longest) {
		    editline++;
		    col = 0;
		    if (width == 0)
			width = filecols;
		}
2607

2608
		wmove(edit, editline, col);
Chris Allegretta's avatar
Chris Allegretta committed
2609
	    }
2610
2611

	    free(foo);
Chris Allegretta's avatar
Chris Allegretta committed
2612
	}
2613

2614
	wnoutrefresh(edit);
2615
2616
    } while ((kbinput = get_kbinput(edit, &meta_key, &func_key)) !=
	NANO_EXIT_KEY && kbinput != NANO_EXIT_FKEY);
2617

Chris Allegretta's avatar
Chris Allegretta committed
2618
    blank_edit();
Chris Allegretta's avatar
Chris Allegretta committed
2619
    titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2620
    edit_refresh();
2621
    curs_set(1);
2622
2623
    if (old_const_update)
	SET(CONST_UPDATE);
Chris Allegretta's avatar
Chris Allegretta committed
2624

2625
    /* Clean up. */
2626
    free_chararray(filelist, numents);
2627
2628
    free(path);

Chris Allegretta's avatar
Chris Allegretta committed
2629
2630
    return retval;
}
2631

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2632
2633
2634
/* The file browser front end.  We check to see if inpath has a dir in
 * it.  If it does, we start do_browser() from there.  Otherwise, we
 * start do_browser() from the current directory. */
2635
char *do_browse_from(const char *inpath)
2636
2637
{
    struct stat st;
2638
    char *path;
2639
	/* This holds the tilde-expanded version of inpath. */
2640
    DIR *dir = NULL;
2641

2642
2643
2644
    assert(inpath != NULL);

    path = real_dir_from_tilde(inpath);
2645

2646
2647
2648
2649
2650
2651
    /* Perhaps path is a directory.  If so, we'll pass it to
     * do_browser().  Or perhaps path is a directory / a file.  If so,
     * we'll try stripping off the last path element and passing it to
     * do_browser().  Or perhaps path doesn't have a directory portion
     * at all.  If so, we'll just pass the current directory to
     * do_browser(). */
2652
2653
    if (stat(path, &st) == -1 || !S_ISDIR(st.st_mode)) {
	striponedir(path);
2654
2655
	if (stat(path, &st) == -1 || !S_ISDIR(st.st_mode)) {
	    free(path);
2656

2657
2658
	    path = charalloc(PATH_MAX + 1);
	    path = getcwd(path, PATH_MAX + 1);
2659
2660
2661

	    if (path != NULL)
		align(&path);
2662
	}
2663
    }
2664

2665
#ifndef DISABLE_OPERATINGDIR
2666
2667
2668
2669
2670
2671
2672
    /* If the resulting path isn't in the operating directory, use
     * the operating directory instead. */
    if (check_operating_dir(path, FALSE)) {
	if (path != NULL)
	    free(path);
	path = mallocstrcpy(NULL, operating_dir);
    }
2673
2674
#endif

2675
2676
2677
    if (path != NULL)
	dir = opendir(path);

2678
    if (dir == NULL) {
2679
	beep();
2680
2681
2682
2683
2684
	free(path);
	return NULL;
    }

    return do_browser(path, dir);
2685
}
Chris Allegretta's avatar
Chris Allegretta committed
2686
#endif /* !DISABLE_BROWSER */
2687

2688
#if !defined(NANO_SMALL) && defined(ENABLE_NANORC)
2689
2690
2691
/* Return $HOME/.nano_history, or NULL if we can't find the homedir.
 * The string is dynamically allocated, and should be freed. */
char *histfilename(void)
2692
{
2693
    char *nanohist = NULL;
2694

2695
2696
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2697

2698
2699
2700
	nanohist = charalloc(homelen + 15);
	strcpy(nanohist, homedir);
	strcpy(nanohist + homelen, "/.nano_history");
Chris Allegretta's avatar
Chris Allegretta committed
2701
    }
2702
2703
2704
    return nanohist;
}

2705
/* Load histories from ~/.nano_history. */
2706
2707
2708
void load_history(void)
{
    char *nanohist = histfilename();
Chris Allegretta's avatar
Chris Allegretta committed
2709

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2710
    /* Assume do_rcfile() has reported a missing home directory. */
2711
2712
    if (nanohist != NULL) {
	FILE *hist = fopen(nanohist, "r");
Chris Allegretta's avatar
Chris Allegretta committed
2713

2714
	if (hist == NULL) {
2715
	    if (errno != ENOENT) {
2716
2717
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
2718
2719
2720
2721
		rcfile_error(N_("Error reading %s: %s"), nanohist,
			strerror(errno));
		fprintf(stderr,
			_("\nPress Return to continue starting nano\n"));
2722
2723
		while (getchar() != '\n')
		    ;
2724
	    }
2725
	} else {
2726
2727
2728
2729
	    /* Load a history (first the search history, then the
	     * replace history) from oldest to newest.  Assume the last
	     * history entry is a blank line. */
	    filestruct **history = &search_history;
2730
2731
2732
2733
2734
2735
2736
2737
2738
2739
2740
	    char *line = NULL;
	    size_t buflen = 0;
	    ssize_t read;

	    while ((read = getline(&line, &buflen, hist)) >= 0) {
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
		if (read > 0) {
		    unsunder(line, read);
2741
2742
		    update_history(history, line);
		} else
2743
2744
		    history = &replace_history;
	    }
2745

2746
	    fclose(hist);
2747
	    free(line);
2748
	}
2749
2750
2751
2752
	free(nanohist);
    }
}

2753
bool writehist(FILE *hist, filestruct *h)
2754
{
2755
    filestruct *p;
2756

2757
2758
2759
    /* Write history from oldest to newest.  Assume the last history
     * entry is a blank line. */
    for (p = h; p != NULL; p = p->next) {
2760
	size_t p_len = strlen(p->data);
2761

2762
	sunder(p->data);
2763

2764
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
2765
2766
		putc('\n', hist) == EOF)
	    return FALSE;
2767
    }
2768

2769
    return TRUE;
2770
2771
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2772
/* Save histories to ~/.nano_history. */
2773
2774
void save_history(void)
{
2775
    char *nanohist;
2776

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2777
    /* Don't save unchanged or empty histories. */
2778
2779
    if (!history_has_changed() || (searchbot->lineno == 1 &&
	replacebot->lineno == 1))
2780
2781
	return;

2782
2783
2784
2785
    nanohist = histfilename();

    if (nanohist != NULL) {
	FILE *hist = fopen(nanohist, "wb");
Chris Allegretta's avatar
Chris Allegretta committed
2786

2787
	if (hist == NULL)
2788
2789
	    rcfile_error(N_("Error writing %s: %s"), nanohist,
		strerror(errno));
2790
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2791
2792
	    /* Make sure no one else can read from or write to the
	     * history file. */
2793
	    chmod(nanohist, S_IRUSR | S_IWUSR);
2794

2795
2796
	    if (!writehist(hist, searchage) || !writehist(hist,
		replaceage))
2797
2798
		rcfile_error(N_("Error writing %s: %s"), nanohist,
			strerror(errno));
2799

2800
2801
	    fclose(hist);
	}
2802

2803
2804
2805
	free(nanohist);
    }
}
2806
#endif /* !NANO_SMALL && ENABLE_NANORC */