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

24
#include "proto.h"
25

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

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

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

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

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

62
    initialize_buffer_text();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
63

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

68
    openfile->modified = FALSE;
69
#ifndef NANO_TINY
70
    openfile->mark_set = FALSE;
71

72
73
74
    openfile->mark_begin = NULL;
    openfile->mark_begin_x = 0;

75
    openfile->fmt = NIX_FILE;
76

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

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

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

93
94
95
96
97
    openfile->filebot = openfile->fileage;
    openfile->edittop = openfile->fileage;
    openfile->current = openfile->fileage;

    openfile->totsize = 0;
98
99
}

100
101
/* 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. */
102
void open_buffer(const char *filename)
103
{
104
105
106
    bool new_buffer = (openfile == NULL
#ifdef ENABLE_MULTIBUFFER
	 || ISSET(MULTIBUFFER)
107
#endif
108
109
110
111
112
113
	);
	/* 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. */
114

115
116
    assert(filename != NULL);

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

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

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

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

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

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

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

169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
#ifndef DISABLE_SPELLER
/* If it's not "", filename is a file to open.  We blow away the text of
 * the current buffer, and then open and read the file, if
 * applicable.  Note that we skip the operating directory test when
 * doing this. */
void replace_buffer(const char *filename)
{
    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. */

    assert(filename != NULL);

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

    /* Reinitialize the text of the current buffer. */
    free_filestruct(openfile->fileage);
    initialize_buffer_text();

    /* If we have a non-new file, read it in. */
    if (rc == 0)
	read_file(f, filename);

    /* Move back to the beginning of the first line of the buffer. */
    openfile->current = openfile->fileage;
    openfile->current_x = 0;
    openfile->placewewant = 0;
}
#endif /* !DISABLE_SPELLER */

202
/* Update the screen to account for the current buffer. */
203
void display_buffer(void)
204
{
205
    /* Update the titlebar, since the filename may have changed. */
206
    titlebar(NULL);
207
208

#ifdef ENABLE_COLOR
209
210
211
    /* Make sure we're using the buffer's associated colors, if
     * applicable. */
    color_init();
212
213
214
#endif

    /* Update the edit window. */
215
    edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
216
217
}

218
219
220
221
#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
222
{
223
    assert(openfile != NULL);
224

225
226
    /* If only one file buffer is open, indicate it on the statusbar and
     * get out. */
227
    if (openfile == openfile->next) {
228
229
	statusbar(_("No more open file buffers"));
	return;
Chris Allegretta's avatar
Chris Allegretta committed
230
    }
231

232
233
    /* Switch to the next or previous file buffer, depending on the
     * value of next_buf. */
234
    openfile = next_buf ? openfile->next : openfile->prev;
235
236

#ifdef DEBUG
237
    fprintf(stderr, "filename is %s\n", openfile->filename);
238
239
#endif

240
    /* Update the screen to account for the current buffer. */
241
    display_buffer();
242

243
    /* Indicate the switch on the statusbar. */
244
    statusbar(_("Switched to %s"),
245
246
	((openfile->filename[0] == '\0') ? _("New Buffer") :
	openfile->filename));
247
248

#ifdef DEBUG
249
    dump_filestruct(openfile->current);
250
#endif
Chris Allegretta's avatar
Chris Allegretta committed
251
252
}

253
/* Switch to the previous entry in the openfile filebuffer. */
254
void switch_to_prev_buffer_void(void)
255
{
256
    switch_to_prevnext_buffer(FALSE);
257
}
258

259
/* Switch to the next entry in the openfile filebuffer. */
260
void switch_to_next_buffer_void(void)
261
{
262
    switch_to_prevnext_buffer(TRUE);
263
}
264

265
/* Delete an entry from the openfile filebuffer, and switch to the one
266
267
268
 * after it.  Return TRUE on success, or FALSE if there are no more open
 * file buffers. */
bool close_buffer(void)
269
{
270
    assert(openfile != NULL);
271

272
    /* If only one file buffer is open, get out. */
273
    if (openfile == openfile->next)
274
	return FALSE;
275

276
    /* Switch to the next file buffer. */
277
    switch_to_next_buffer_void();
278

279
    /* Close the file buffer we had open before. */
280
    unlink_opennode(openfile->prev);
281

282
    display_main_list();
283

284
    return TRUE;
285
}
286
#endif /* ENABLE_MULTIBUFFER */
287

288
289
290
291
292
293
/* 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)
294
{
295
    filestruct *fileptr = (filestruct *)nmalloc(sizeof(filestruct));
296

297
298
    /* Convert nulls to newlines.  buf_len is the string's real
     * length. */
299
    unsunder(buf, buf_len);
300

301
    assert(openfile->fileage != NULL && strlen(buf) == buf_len);
302

303
    fileptr->data = mallocstrcpy(NULL, buf);
304

305
#ifndef NANO_TINY
306
307
308
309
    /* 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';
310
#endif
311

312
    if (*first_line_ins) {
313
314
315
	/* Special case: We're inserting with the cursor on the first
	 * line. */
	fileptr->prev = NULL;
316
	fileptr->next = openfile->fileage;
317
	fileptr->lineno = 1;
318
	if (*first_line_ins) {
319
320
321
322
	    *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. */
323
	    openfile->edittop = fileptr;
324
	} else
325
326
	    openfile->filebot = fileptr;
	openfile->fileage = fileptr;
327
328
329
330
331
332
333
    } else {
	assert(prevnode != NULL);

	fileptr->prev = prevnode;
	fileptr->next = NULL;
	fileptr->lineno = prevnode->lineno + 1;
	prevnode->next = fileptr;
334
335
    }

336
337
    return fileptr;
}
338

339
340
/* Read an open file into the current buffer.  f should be set to the
 * open file, and filename should be set to the name of the file. */
341
342
343
344
345
346
347
348
349
350
351
352
353
354
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. */
355
    filestruct *fileptr = openfile->current;
356
357
358
359
360
361
	/* 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. */
362
#ifndef NANO_TINY
363
364
    int format = 0;
	/* 0 = *nix, 1 = DOS, 2 = Mac, 3 = both DOS and Mac. */
365
#endif
366

367
368
    assert(openfile->fileage != NULL && openfile->current != NULL);

369
370
    buf = charalloc(bufx);
    buf[0] = '\0';
371

372
373
374
375
    if (openfile->current == openfile->fileage)
	first_line_ins = TRUE;
    else
	fileptr = openfile->current->prev;
376

377
    /* Read the entire file into the filestruct. */
378
379
380
381
382
383
    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') {
384
#ifndef NANO_TINY
385
386
387
388
389
390
391
392
393
	    /* If it's a DOS file or a DOS/Mac file ('\r' before '\n' on
	     * the first line if we think it's a *nix file, or on any
	     * line otherwise), and file conversion isn't disabled,
	     * handle it! */
	    if (!ISSET(NO_CONVERT) && (num_lines == 0 || format != 0) &&
		i > 0 && buf[i - 1] == '\r') {
		if (format == 0 || format == 2)
		    format++;
	    }
394
#endif
395

396
397
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
398

399
400
401
	    /* Reset the line length in preparation for the next
	     * line. */
	    len = 0;
Chris Allegretta's avatar
Chris Allegretta committed
402

403
404
405
	    num_lines++;
	    buf[0] = '\0';
	    i = 0;
406
#ifndef NANO_TINY
407
408
409
410
411
	/* If it's a Mac file ('\r' without '\n' on the first line if we
	 * think it's a *nix file, or on any line otherwise), and file
	 * conversion isn't disabled, handle it! */
	} else if (!ISSET(NO_CONVERT) && (num_lines == 0 ||
		format != 0) && i > 0 && buf[i - 1] == '\r') {
412
413
414
415
416
	    /* 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;
417

418
419
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
420

421
422
423
424
	    /* 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
425

426
427
428
429
	    num_lines++;
	    buf[0] = input;
	    buf[1] = '\0';
	    i = 1;
430
#endif
431
432
433
434
	} else {
	    /* Calculate the total length of the line.  It might have
	     * nulls in it, so we can't just use strlen() here. */
	    len++;
435

436
437
438
439
440
441
442
443
	    /* 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);
444
	    }
445
446
447
448
449
450
451
452
453
454
455

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

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

457
#ifndef NANO_TINY
458
459
460
461
462
463
464
465
    /* 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';
    }
466
#endif
467

468
469
    /* Did we not get a newline and still have stuff to do? */
    if (len > 0) {
470
#ifndef NANO_TINY
471
472
473
474
475
476
477
	/* 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;
478
479
#endif

480
481
482
483
	/* Read in the last line properly. */
	fileptr = read_line(buf, fileptr, &first_line_ins, len);
	num_lines++;
    }
484

485
    free(buf);
486

487
    /* If we didn't get a file and we don't already have one, open a
488
     * blank buffer. */
489
    if (fileptr == NULL)
490
	open_buffer("");
491

492
493
    /* Attach the file we got to the filestruct.  If we got a file of
     * zero bytes, don't do anything. */
494
    if (num_lines > 0) {
495
496
497
498
499
500
501
502
503
504
505
506
	/* If the file we got doesn't end in a newline, tack its last
	 * line onto the beginning of the line at current. */
	if (len > 0) {
	    size_t current_len = strlen(openfile->current->data);

	    /* Adjust the current x-coordinate to compensate for the
	     * change in the current line. */
	    if (num_lines == 1)
		openfile->current_x += len;
	    else
		openfile->current_x = len;

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
507
508
	    /* Tack the text at fileptr onto the beginning of the text
	     * at current. */
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
	    openfile->current->data =
		charealloc(openfile->current->data, len +
		current_len + 1);
	    charmove(openfile->current->data + len,
		openfile->current->data, current_len + 1);
	    strncpy(openfile->current->data, fileptr->data, len);

	    /* Don't destroy fileage, edittop, or filebot! */
	    if (fileptr == openfile->fileage)
		openfile->fileage = openfile->current;
	    if (fileptr == openfile->edittop)
		openfile->edittop = openfile->current;
	    if (fileptr == openfile->filebot)
		openfile->filebot = openfile->current;

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
524
	    /* Move fileptr back one line and blow away the old fileptr,
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
	     * since its text has been saved. */
	    fileptr = fileptr->prev;
	    if (fileptr != NULL) {
		if (fileptr->next != NULL)
		    free(fileptr->next);
	    }
	}

	/* Attach the line at current after the line at fileptr. */
	if (fileptr != NULL) {
	    fileptr->next = openfile->current;
	    openfile->current->prev = fileptr;
	}

	/* Renumber starting with the last line of the file we
	 * inserted. */
541
	renumber(openfile->current);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
542
    }
543

544
545
    openfile->totsize += get_totsize(openfile->fileage,
	openfile->filebot);
546

547
    /* If the NO_NEWLINES flag isn't set, and text has been added to
548
     * the magicline (i.e. a file that doesn't end in a newline has been
549
550
551
     * inserted at the end of the current buffer), add a new magicline,
     * and move the current line down to it. */
    if (!ISSET(NO_NEWLINES) && openfile->filebot->data[0] != '\0') {
552
553
554
555
556
557
558
559
560
	new_magicline();
	openfile->current = openfile->filebot;
	openfile->current_x = 0;
    }

    /* Set the current place we want to the end of the last line of the
     * file we inserted. */
    openfile->placewewant = xplustabs();

561
#ifndef NANO_TINY
562
563
564
565
566
567
    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) {
568
	openfile->fmt = MAC_FILE;
569
570
571
572
	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) {
573
	openfile->fmt = DOS_FILE;
574
575
576
577
	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
578
#endif
579
580
581
	statusbar(P_("Read %lu line", "Read %lu lines",
		(unsigned long)num_lines), (unsigned long)num_lines);
}
Chris Allegretta's avatar
Chris Allegretta committed
582

583
584
585
586
/* 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".
 *
587
588
589
 * Return -2 if we say "New File", -1 if the file isn't opened, and 0
 * otherwise.  The file might still have an error while reading with a 0
 * return value.  *f is set to the opened file. */
590
591
592
int open_file(const char *filename, bool newfie, FILE **f)
{
    struct stat fileinfo;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
593
    int fd;
594
    char *full_filename;
595

596
    assert(filename != NULL && f != NULL);
597

598
599
600
601
602
603
604
    /* Get the specified file's full path. */
    full_filename = get_full_path(filename);

    if (full_filename == NULL)
	full_filename = mallocstrcpy(NULL, filename);

    if (stat(full_filename, &fileinfo) == -1) {
605
606
607
608
609
	if (newfie) {
	    statusbar(_("New File"));
	    return -2;
	}
	statusbar(_("\"%s\" not found"), filename);
610
	beep();
611
612
	return -1;
    } else if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
613
	S_ISBLK(fileinfo.st_mode)) {
614
615
	/* Don't open directories, character files, or block files.
	 * Sorry, /dev/sndstat! */
616
617
	statusbar(S_ISDIR(fileinfo.st_mode) ?
		_("\"%s\" is a directory") :
618
		_("\"%s\" is a device file"), filename);
619
	beep();
620
	return -1;
621
622
623
    } else if ((fd = open(full_filename, O_RDONLY)) == -1) {
	statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
624
	beep();
625
626
 	return -1;
     } else {
627
	/* The file is A-OK.  Open it. */
628
	*f = fdopen(fd, "rb");
629

630
631
632
	if (*f == NULL) {
	    statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
633
	    beep();
634
635
636
	    close(fd);
	} else
	    statusbar(_("Reading File"));
637
    }
638

639
640
    free(full_filename);

641
    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
642
643
}

644
645
646
647
648
/* 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)
649
{
650
    static int ulmax_digits = -1;
651
652
653
    unsigned long i = 0;
    char *buf;
    size_t namelen, suffixlen;
654

655
    assert(name != NULL && suffix != NULL);
656

657
658
659
    if (ulmax_digits == -1)
	ulmax_digits = digits(ULONG_MAX);

660
661
    namelen = strlen(name);
    suffixlen = strlen(suffix);
Chris Allegretta's avatar
Chris Allegretta committed
662

663
    buf = charalloc(namelen + suffixlen + ulmax_digits + 2);
664
    sprintf(buf, "%s%s", name, suffix);
665

666
667
    while (TRUE) {
	struct stat fs;
Chris Allegretta's avatar
Chris Allegretta committed
668

669
670
671
672
	if (stat(buf, &fs) == -1)
	    return buf;
	if (i == ULONG_MAX)
	    break;
673

674
675
676
	i++;
	sprintf(buf + namelen + suffixlen, ".%lu", i);
    }
Chris Allegretta's avatar
Chris Allegretta committed
677

678
679
680
    /* We get here only if there is no possible save file.  Blank out
     * the filename to indicate this. */
    null_at(&buf, 0);
681

682
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
683
684
}

685
686
687
/* Insert a file into a new buffer if the MULTIBUFFER flag is set, or
 * into the current buffer if it isn't.  If execute is TRUE, insert the
 * output of an executed command instead of a file. */
688
void do_insertfile(
689
#ifndef NANO_TINY
690
691
692
693
694
695
696
697
698
	bool execute
#else
	void
#endif
	)
{
    int i;
    const char *msg;
    char *ans = mallocstrcpy(NULL, "");
699
	/* The last answer the user typed at the statusbar prompt. */
700
    filestruct *edittop_save = openfile->edittop;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
701
    size_t current_x_save = openfile->current_x;
702
    ssize_t current_y_save = openfile->current_y;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
703
    bool edittop_inside = FALSE;
704
#ifndef NANO_TINY
705
    bool right_side_up = FALSE, single_line = FALSE;
706
#endif
707

708
    while (TRUE) {
709
#ifndef NANO_TINY
710
	if (execute) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
711
	    msg =
712
#ifdef ENABLE_MULTIBUFFER
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
713
		ISSET(MULTIBUFFER) ?
714
		_("Command to execute in new buffer [from %s] ") :
715
#endif
716
		_("Command to execute [from %s] ");
717
718
	} else {
#endif
719
	    msg =
720
#ifdef ENABLE_MULTIBUFFER
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
721
		ISSET(MULTIBUFFER) ?
722
		_("File to insert into new buffer [from %s] ") :
723
#endif
724
		_("File to insert [from %s] ");
725
#ifndef NANO_TINY
726
727
	}
#endif
728

729
	i = do_prompt(TRUE,
730
731
732
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
733
#ifndef NANO_TINY
734
		execute ? extcmd_list :
735
#endif
736
		insertfile_list, ans,
737
#ifndef NANO_TINY
738
		NULL,
739
#endif
740
		edit_refresh, msg,
741
#ifndef DISABLE_OPERATINGDIR
742
743
		operating_dir != NULL && strcmp(operating_dir,
		".") != 0 ? operating_dir :
744
745
#endif
		"./");
746

747
	/* If we're in multibuffer mode and the filename or command is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
748
	 * blank, open a new buffer instead of canceling.  If the
749
750
	 * filename or command begins with a newline (i.e. an encoded
	 * null), treat it as though it's blank. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
751
	if (i == -1 || ((i == -2 || *answer == '\n')
752
753
754
755
756
757
758
#ifdef ENABLE_MULTIBUFFER
		&& !ISSET(MULTIBUFFER)
#endif
		)) {
	    statusbar(_("Cancelled"));
	    break;
	} else {
759
	    size_t pww_save = openfile->placewewant;
760

761
	    ans = mallocstrcpy(ans, answer);
762

763
#ifndef NANO_TINY
764
765
766
767
768
769
770
#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
771
#endif
772
773
774
775
776
777
778
	    if (i == NANO_TOOTHERINSERT_KEY) {
		execute = !execute;
		continue;
	    }
#ifndef DISABLE_BROWSER
	    else
#endif
779
#endif /* !NANO_TINY */
780

781
782
783
#ifndef DISABLE_BROWSER
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
784

785
786
		if (tmp == NULL)
		    continue;
787

788
		/* We have a file now.  Indicate this. */
789
790
		free(answer);
		answer = tmp;
791

792
793
794
		i = 0;
	    }
#endif
795

796
797
798
799
800
801
802
803
	    /* 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;
804

805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
#ifndef NANO_TINY
		/* Keep track of whether the mark begins inside the
		 * partition and will need adjustment. */
		if (openfile->mark_set) {
		    filestruct *top, *bot;
		    size_t top_x, bot_x;

		    mark_order((const filestruct **)&top, &top_x,
			(const filestruct **)&bot, &bot_x,
			&right_side_up);

		    single_line = (top == bot);
		}
#endif

820
821
822
823
824
#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
825
826
827
		 * looks like a new buffer, and keep track of whether
		 * the top of the edit window is inside the
		 * partition. */
828
829
830
		filepart = partition_filestruct(openfile->current,
			openfile->current_x, openfile->current,
			openfile->current_x);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
831
832
		edittop_inside =
			(openfile->edittop == openfile->fileage);
833
834
835
#ifdef ENABLE_MULTIBUFFER
	    }
#endif
Chris Allegretta's avatar
Chris Allegretta committed
836

837
838
	    /* Convert newlines to nulls, just before we insert the file
	     * or execute the command. */
839
	    sunder(answer);
840
	    align(&answer);
841

842
#ifndef NANO_TINY
843
844
845
	    if (execute) {
#ifdef ENABLE_MULTIBUFFER
		if (ISSET(MULTIBUFFER))
846
		    /* Open a blank buffer. */
847
848
849
850
		    open_buffer("");
#endif

		/* Save the command's output in the current buffer. */
851
		execute_command(answer);
852
853
854
855
856
857
858
859
860

#ifdef ENABLE_MULTIBUFFER
		if (ISSET(MULTIBUFFER)) {
		    /* Move back to the beginning of the first line of
		     * the buffer. */
		    openfile->current = openfile->fileage;
		    openfile->current_x = 0;
		    openfile->placewewant = 0;
		}
861
#endif
862
863
	    } else {
#endif /* !NANO_TINY */
864
865
		/* Make sure the path to the file specified in answer is
		 * tilde-expanded. */
866
867
		answer = mallocstrassn(answer,
			real_dir_from_tilde(answer));
868
869
870

		/* Save the file specified in answer in the current
		 * buffer. */
871
		open_buffer(answer);
872
#ifndef NANO_TINY
873
	    }
Chris Allegretta's avatar
Chris Allegretta committed
874
#endif
875

876
#ifdef ENABLE_MULTIBUFFER
877
878
879
	    if (ISSET(MULTIBUFFER))
		/* Update the screen to account for the current
		 * buffer. */
880
		display_buffer();
881
	    else
Chris Allegretta's avatar
Chris Allegretta committed
882
#endif
883
	    {
884
		filestruct *top_save = openfile->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
885

886
887
888
		/* If we were at the top of the edit window before, set
		 * the saved value of edittop to the new top of the edit
		 * window. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
889
		if (edittop_inside)
890
		    edittop_save = openfile->fileage;
891
892

		/* Update the current x-coordinate to account for the
893
894
895
896
		 * number of characters inserted on the current line.
		 * If the mark begins inside the partition, adjust the
		 * mark coordinates to compensate for the change in the
		 * current line. */
897
		openfile->current_x = strlen(openfile->filebot->data);
898
899
900
901
		if (openfile->fileage == openfile->filebot) {
#ifndef NANO_TINY
		    if (openfile->mark_set) {
			openfile->mark_begin = openfile->current;
902
			if (!right_side_up)
903
904
905
906
			    openfile->mark_begin_x +=
				openfile->current_x;
		    }
#endif
907
		    openfile->current_x += current_x_save;
908
		}
909
#ifndef NANO_TINY
910
911
912
913
914
915
916
917
918
919
		else if (openfile->mark_set) {
		    if (!right_side_up) {
			if (single_line) {
			    openfile->mark_begin = openfile->current;
			    openfile->mark_begin_x -= current_x_save;
			} else
			    openfile->mark_begin_x -=
				openfile->current_x;
		    }
		}
920
#endif
921
922
923

		/* Update the current y-coordinate to account for the
		 * number of lines inserted. */
924
		openfile->current_y += current_y_save;
925

926
927
928
929
		/* 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. */
930
		unpartition_filestruct(&filepart);
931

932
933
934
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
935

936
		/* Restore the old edittop. */
937
		openfile->edittop = edittop_save;
938

939
940
941
		/* Restore the old place we want. */
		openfile->placewewant = pww_save;

942
943
		/* Mark the file as modified. */
		set_modified();
944

945
946
		/* Update the screen. */
		edit_refresh();
947
	    }
948

949
950
951
952
953
	    break;
	}
    }

    free(ans);
954
955
}

956
957
958
/* Insert a file into a new buffer or the current buffer, depending on
 * whether the MULTIBUFFER flag is set.  If we're in view mode, only
 * allow inserting a file into a new buffer. */
959
void do_insertfile_void(void)
960
{
961
962
#ifdef ENABLE_MULTIBUFFER
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
963
	statusbar(_("Key invalid in non-multibuffer mode"));
964
965
966
    else
#endif
	do_insertfile(
967
#ifndef NANO_TINY
968
969
970
		FALSE
#endif
		);
971
972
973
974

    display_main_list();
}

975
/* When passed "[relative path]" or "[relative path][filename]" in
976
 * origpath, return "[full path]" or "[full path][filename]" on success,
977
978
979
980
 * 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. */
981
char *get_full_path(const char *origpath)
982
{
983
984
985
986
    struct stat fileinfo;
    char *d_here, *d_there, *d_there_file = NULL;
    const char *last_slash;
    bool path_only;
987

988
989
    if (origpath == NULL)
    	return NULL;
990

991
    /* Get the current directory.  If it doesn't exist, back up and try
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
992
993
     * again until we get a directory that does, and use that as the
     * current directory. */
994
995
    d_here = charalloc(PATH_MAX + 1);
    d_here = getcwd(d_here, PATH_MAX + 1);
996

997
    while (d_here == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
998
999
1000
	if (chdir("..") == -1)
	    break;

1001
1002
1003
1004
1005
	d_here = getcwd(d_here, PATH_MAX + 1);
    }

    /* If we succeeded, canonicalize it in d_here. */
    if (d_here != NULL) {
1006
1007
	align(&d_here);

1008
1009
	/* If the current directory isn't "/", tack a slash onto the end
	 * of it. */
1010
	if (strcmp(d_here, "/") != 0) {
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1011
	    d_here = charealloc(d_here, strlen(d_here) + 2);
1012
1013
	    strcat(d_here, "/");
	}
1014
1015
1016
    /* Otherwise, set d_here to "". */
    } else
	d_here = mallocstrcpy(NULL, "");
1017

1018
    d_there = real_dir_from_tilde(origpath);
1019

1020
1021
1022
    /* If stat()ing d_there 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 otherwise. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1023
1024
    path_only = (stat(d_there, &fileinfo) != -1 &&
	S_ISDIR(fileinfo.st_mode));
1025

1026
1027
1028
    /* If path_only is TRUE, make sure d_there ends in a slash. */
    if (path_only) {
	size_t d_there_len = strlen(d_there);
1029

1030
1031
1032
	if (d_there[d_there_len - 1] != '/') {
	    d_there = charealloc(d_there, d_there_len + 2);
	    strcat(d_there, "/");
1033
	}
1034
    }
1035

1036
1037
    /* Search for the last slash in d_there. */
    last_slash = strrchr(d_there, '/');
1038

1039
1040
1041
1042
    /* 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);
1043

1044
1045
1046
1047
1048
1049
1050
1051
	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);

1052
	/* Remove the filename portion of the answer from d_there. */
1053
1054
	null_at(&d_there, last_slash - d_there + 1);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1055
	/* Go to the path specified in d_there. */
1056
1057
1058
	if (chdir(d_there) == -1) {
	    free(d_there);
	    d_there = NULL;
1059
	} else {
1060
1061
1062
1063
1064
1065
1066
1067
1068
1069
1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
1080
1081
1082
1083
	    free(d_there);

	    /* Get the full path. */
	    d_there = charalloc(PATH_MAX + 1);
	    d_there = getcwd(d_there, PATH_MAX + 1);

	    /* If we succeeded, canonicalize it in d_there. */
	    if (d_there != NULL) {
		align(&d_there);

		/* If the current directory isn't "/", tack a slash onto
		 * the end of it. */
		if (strcmp(d_there, "/") != 0) {
		    d_there = charealloc(d_there, strlen(d_there) + 2);
		    strcat(d_there, "/");
		}
	    } else
		/* Otherwise, 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.  We don't check for a chdir()
1084
	     * error, since we can do nothing if we get one. */
1085
	    chdir(d_here);
1086

1087
1088
	    /* Free d_here, since we're done using it. */
	    free(d_here);
1089
	}
1090
    }
1091

1092
1093
1094
1095
1096
1097
1098
    /* At this point, if path_only is FALSE and d_there isn't NULL,
     * 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 the latter onto the end of the former.  d_there will
     * then contain the complete answer. */
    if (!path_only && d_there != NULL) {
	d_there = charealloc(d_there, strlen(d_there) +
1099
		strlen(d_there_file) + 1);
1100
1101
	strcat(d_there, d_there_file);
    }
1102

1103
1104
    /* Free d_there_file, since we're done using it. */
    if (d_there_file != NULL)
1105
1106
	free(d_there_file);

1107
    return d_there;
1108
}
1109

1110
1111
1112
/* 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. */
1113
char *check_writable_directory(const char *path)
1114
{
1115
1116
    char *full_path = get_full_path(path);

1117
    /* If get_full_path() fails, return NULL. */
1118
    if (full_path == NULL)
1119
	return NULL;
1120

1121
1122
1123
    /* 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
1124
	full_path[strlen(full_path) - 1] != '/') {
1125
	free(full_path);
1126
	return NULL;
1127
    }
1128

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1129
    /* Otherwise, return the full path. */
1130
1131
1132
    return full_path;
}

1133
1134
/* This function calls mkstemp(($TMPDIR|P_tmpdir|/tmp/)"nano.XXXXXX").
 * On success, it returns the malloc()ed filename and corresponding FILE
1135
 * stream, opened in "r+b" mode.  On error, it returns NULL for the
1136
1137
 * filename and leaves the FILE stream unchanged. */
char *safe_tempfile(FILE **f)
1138
{
1139
    char *full_tempdir = NULL;
1140
1141
1142
    const char *tmpdir_env;
    int fd;
    mode_t original_umask = 0;
1143

1144
1145
    assert(f != NULL);

1146
1147
1148
    /* If $TMPDIR is set, set tempdir to it, run it through
     * get_full_path(), and save the result in full_tempdir.  Otherwise,
     * leave full_tempdir set to NULL. */
1149
    tmpdir_env = getenv("TMPDIR");
1150
    if (tmpdir_env != NULL)
1151
	full_tempdir = check_writable_directory(tmpdir_env);
1152

1153
1154
    /* If $TMPDIR is unset, empty, or not a writable directory, and
     * full_tempdir is NULL, try P_tmpdir instead. */
1155
    if (full_tempdir == NULL)
1156
	full_tempdir = check_writable_directory(P_tmpdir);
1157

1158
1159
1160
    /* if P_tmpdir is NULL, use /tmp. */
    if (full_tempdir == NULL)
	full_tempdir = mallocstrcpy(NULL, "/tmp/");
1161

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1162
    full_tempdir = charealloc(full_tempdir, strlen(full_tempdir) + 12);
1163
    strcat(full_tempdir, "nano.XXXXXX");
1164

1165
1166
1167
1168
1169
1170
1171
    original_umask = umask(0);
    umask(S_IRWXG | S_IRWXO);

    fd = mkstemp(full_tempdir);

    if (fd != -1)
	*f = fdopen(fd, "r+b");
1172
1173
1174
    else {
	free(full_tempdir);
	full_tempdir = NULL;
1175
    }
1176

1177
1178
    umask(original_umask);

1179
    return full_tempdir;
1180
}
1181
1182

#ifndef DISABLE_OPERATINGDIR
1183
1184
1185
1186
1187
/* Initialize full_operating_dir based on operating_dir. */
void init_operating_dir(void)
{
    assert(full_operating_dir == NULL);

1188
    if (operating_dir == NULL)
1189
	return;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1190

1191
1192
1193
    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
1194
     * inaccessible, unset operating_dir. */
1195
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) {
1196
1197
1198
1199
1200
1201
1202
	free(full_operating_dir);
	full_operating_dir = NULL;
	free(operating_dir);
	operating_dir = NULL;
    }
}

1203
1204
1205
1206
1207
/* 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)
1208
{
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1209
1210
1211
1212
    /* full_operating_dir is global for memory 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. */
1213

1214
    char *fullpath;
1215
    bool retval = FALSE;
1216
    const char *whereami1, *whereami2 = NULL;
1217

1218
    /* If no operating directory is set, don't bother doing anything. */
1219
    if (operating_dir == NULL)
1220
	return FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1221

1222
    assert(full_operating_dir != NULL);
1223
1224

    fullpath = get_full_path(currpath);
1225

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1226
1227
1228
    /* If fullpath is NULL, it means some directory in the path doesn't
     * exist or is unreadable.  If allow_tabcomp is FALSE, then currpath
     * is what the user typed somewhere.  We don't want to report a
1229
     * non-existent directory as being outside the operating directory,
1230
     * so we return FALSE.  If allow_tabcomp is TRUE, then currpath
1231
1232
     * exists, but is not executable.  So we say it isn't in the
     * operating directory. */
1233
    if (fullpath == NULL)
1234
	return allow_tabcomp;
1235
1236
1237
1238
1239

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

1240
    /* If both searches failed, we're outside the operating directory.
1241
     * Otherwise, check the search results.  If the full operating
1242
1243
1244
     * 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. */
1245
    if (whereami1 != fullpath && whereami2 != full_operating_dir)
1246
1247
	retval = TRUE;
    free(fullpath);
1248
1249

    /* Otherwise, we're still inside it. */
1250
    return retval;
1251
}
1252
1253
#endif

1254
#ifndef NANO_TINY
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
1276
1277
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

1278
/* Read from inn, write to out.  We assume inn is opened for reading,
1279
1280
 * and out for writing.  We return 0 on success, -1 on read error, or -2
 * on write error. */
1281
1282
int copy_file(FILE *inn, FILE *out)
{
1283
    int retval = 0;
1284
    char buf[BUFSIZ];
1285
1286
    size_t charsread;

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

1289
1290
1291
1292
1293
1294
1295
1296
1297
1298
1299
    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
1300

1301
1302
1303
1304
    if (fclose(inn) == EOF)
	retval = -1;
    if (fclose(out) == EOF)
	retval = -2;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1305

1306
1307
1308
    return retval;
}

1309
1310
/* Write a file out to disk.  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
1311
 * ourselves.  If tmp is TRUE, we set the umask to disallow anyone else
1312
1313
 * 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
1314
 *
1315
 * tmp means we are writing a temporary file in a secure fashion.  We
1316
1317
1318
1319
 * use it when spell checking or dumping the file on an error.  If
 * append is APPEND, it means we are appending instead of overwriting.
 * If append is PREPEND, it means we are prepending instead of
 * overwriting.  If nonamechange is TRUE, we don't change the current
1320
1321
 * filename.  nonamechange is ignored if tmp is FALSE, we're appending,
 * or we're prepending.
1322
 *
1323
1324
 * Return TRUE on success or FALSE on error. */
bool write_file(const char *name, FILE *f_open, bool tmp, append_type
1325
	append, bool nonamechange)
Chris Allegretta's avatar
Chris Allegretta committed
1326
{
1327
    bool retval = FALSE;
1328
	/* Instead of returning in this function, you should always
1329
	 * set retval and then goto cleanup_and_exit. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1330
    size_t lineswritten = 0;
1331
    const filestruct *fileptr = openfile->fileage;
1332
    int fd;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1333
	/* The file descriptor we use. */
1334
    mode_t original_umask = 0;
1335
	/* Our umask, from when nano started. */
1336
    bool realexists;
1337
	/* The result of stat().  TRUE if the file exists, FALSE
1338
	 * otherwise.  If name is a link that points nowhere, realexists
1339
	 * is FALSE. */
1340
1341
    struct stat st;
	/* The status fields filled in by stat(). */
1342
    bool anyexists;
1343
1344
	/* The result of lstat().  The same as realexists, unless name
	 * is a link. */
1345
1346
1347
    struct stat lst;
	/* The status fields filled in by lstat(). */
    char *realname;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1348
	/* name after tilde expansion. */
1349
    FILE *f = NULL;
1350
1351
1352
	/* 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
1353

1354
    assert(name != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1355

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1356
    if (*name == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1357
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1358

1359
1360
1361
    if (f_open != NULL)
	f = f_open;

1362
1363
    if (!tmp)
	titlebar(NULL);
1364

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1365
    realname = real_dir_from_tilde(name);
1366

1367
#ifndef DISABLE_OPERATINGDIR
1368
    /* If we're writing a temporary file, we're probably going outside
1369
     * the operating directory, so skip the operating directory test. */
1370
    if (!tmp && check_operating_dir(realname, FALSE)) {
1371
1372
	statusbar(_("Can't write outside of %s"), operating_dir);
	goto cleanup_and_exit;
1373
1374
1375
    }
#endif

1376
    anyexists = (lstat(realname, &lst) != -1);
1377

1378
1379
    /* If the temp file exists and isn't already open, give up. */
    if (tmp && anyexists && f_open == NULL)
1380
	goto cleanup_and_exit;
1381

1382
1383
1384
    /* 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)) {
1385
1386
	statusbar(
		_("Cannot prepend or append to a symlink with --nofollow set"));
1387
1388
1389
	goto cleanup_and_exit;
    }

1390
    /* Save the state of the file at the end of the symlink (if there is
1391
     * one). */
1392
    realexists = (stat(realname, &st) != -1);
1393

1394
#ifndef NANO_TINY
1395
    /* We backup only if the backup toggle is set, the file isn't
1396
1397
1398
1399
     * 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. */
1400
1401
1402
    if (ISSET(BACKUP_FILE) && !tmp && realexists && ((append !=
	OVERWRITE || openfile->mark_set) ||
	openfile->current_stat->st_mtime == st.st_mtime)) {
1403
	FILE *backup_file;
1404
	char *backupname;
1405
	struct utimbuf filetime;
1406
	int copy_status;
1407

1408
	/* Save the original file's access and modification times. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1409
1410
	filetime.actime = openfile->current_stat->st_atime;
	filetime.modtime = openfile->current_stat->st_mtime;
1411

1412
1413
1414
	if (f_open == NULL) {
	    /* Open the original file to copy to the backup. */
	    f = fopen(realname, "rb");
1415

1416
1417
1418
	    if (f == NULL) {
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1419
		beep();
1420
1421
1422
1423
		/* If we can't read from the original file, go on, since
		 * only saving the original file is better than saving
		 * nothing. */
		goto skip_backup;
1424
	    }
1425
1426
	}

1427
	/* If backup_dir is set, we set backupname to
1428
1429
1430
1431
	 * 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]. */
1432
	if (backup_dir != NULL) {
1433
	    char *backuptemp = get_full_path(realname);
1434

1435
	    if (backuptemp == NULL)
1436
1437
1438
1439
1440
1441
		/* 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~. */
1442
		backuptemp = mallocstrcpy(NULL, tail(realname));
1443
	    else {
1444
1445
		size_t i = 0;

1446
1447
1448
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1449
1450
1451
1452
		}
	    }

	    backupname = charalloc(strlen(backup_dir) +
1453
1454
1455
1456
		strlen(backuptemp) + 1);
	    sprintf(backupname, "%s%s", backup_dir, backuptemp);
	    free(backuptemp);
	    backuptemp = get_next_filename(backupname, "~");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1457
	    if (*backuptemp == '\0') {
1458
1459
1460
1461
		statusbar(_("Error writing %s: %s"), backupname,
		    _("Too many backup files?"));
		free(backuptemp);
		free(backupname);
1462
1463
1464
1465
		/* If we can't write to the backup, go on, since only
		 * saving the original file is better than saving
		 * nothing. */
		goto skip_backup;
1466
1467
1468
1469
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1470
1471
1472
1473
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1474

1475
1476
1477
	/* Open the destination backup file.  Before we write to it, we
	 * set its permissions, so no unauthorized person can read it as
	 * we write. */
1478
	backup_file = fopen(backupname, "wb");
1479

1480
	if (backup_file == NULL || chmod(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1481
		openfile->current_stat->st_mode) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1482
1483
	    statusbar(_("Error writing %s: %s"), backupname,
		strerror(errno));
1484
	    free(backupname);
1485
1486
	    if (backup_file != NULL)
		fclose(backup_file);
1487
1488
1489
	    /* If we can't write to the backup, go on, since only saving
	     * the original file is better than saving nothing. */
	    goto skip_backup;
1490
1491
1492
	}

#ifdef DEBUG
1493
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1494
1495
#endif

1496
1497
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1498

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1499
	/* And set its metadata. */
1500
	if (copy_status != 0 || chown(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1501
1502
		openfile->current_stat->st_uid,
		openfile->current_stat->st_gid) == -1 ||
1503
		utime(backupname, &filetime) == -1) {
1504
	    if (copy_status == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1505
1506
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1507
1508
		beep();
	    } else
1509
1510
		statusbar(_("Error writing %s: %s"), backupname,
			strerror(errno));
1511
1512
1513
	    /* If we can't read from or write to the backup, go on,
	     * since only saving the original file is better than saving
	     * nothing. */
1514
	}
1515

1516
1517
	free(backupname);
    }
1518
1519

  skip_backup:
1520
#endif /* !NANO_TINY */
1521

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1522
1523
    /* 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
1524
1525
1526
1527
     * overwrite. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode) &&
	unlink(realname) == -1) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1528
	goto cleanup_and_exit;
1529
    }
1530

1531
1532
    if (f_open == NULL) {
	original_umask = umask(0);
1533

1534
	/* If we create a temp file, we don't let anyone else access it.
1535
1536
	 * We create a temp file if tmp is TRUE. */
	if (tmp)
1537
	    umask(S_IRWXG | S_IRWXO);
1538
1539
	else
	    umask(original_umask);
1540
    }
1541

1542
    /* If we're prepending, copy the file to a temp file. */
1543
    if (append == PREPEND) {
1544
1545
1546
	int fd_source;
	FILE *f_source = NULL;

1547
1548
1549
1550
1551
1552
1553
1554
1555
1556
1557
	if (f == NULL) {
	    f = fopen(realname, "rb");

	    if (f == NULL) {
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
		beep();
		goto cleanup_and_exit;
	    }
	}

1558
	tempname = safe_tempfile(&f);
1559

1560
	if (tempname == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1561
	    statusbar(_("Error writing temp file: %s"),
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1562
		strerror(errno));
1563
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1564
	}
1565

1566
1567
	if (f_open == NULL) {
	    fd_source = open(realname, O_RDONLY | O_CREAT);
1568

1569
1570
1571
1572
1573
	    if (fd_source != -1) {
		f_source = fdopen(fd_source, "rb");
		if (f_source == NULL) {
		    statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1574
		    beep();
1575
1576
1577
1578
1579
1580
		    close(fd_source);
		    fclose(f);
		    unlink(tempname);
		    goto cleanup_and_exit;
		}
	    }
1581
1582
1583
	}

	if (copy_file(f_source, f) != 0) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1584
1585
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
1586
	    unlink(tempname);
1587
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1588
1589
1590
	}
    }

1591
1592
1593
    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*. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1594
1595
1596
	fd = open(realname, O_WRONLY | O_CREAT | ((append == APPEND) ?
		O_APPEND : (tmp ? O_EXCL : O_TRUNC)), S_IRUSR |
		S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
1597

1598
1599
	/* Set the umask back to the user's original value. */
	umask(original_umask);
1600

1601
1602
1603
1604
	/* If we couldn't open the file, give up. */
	if (fd == -1) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1605

1606
1607
1608
1609
1610
	    /* tempname has been set only if we're prepending. */
	    if (tempname != NULL)
		unlink(tempname);
	    goto cleanup_and_exit;
	}
1611

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

1614
1615
1616
1617
1618
1619
	if (f == NULL) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
	    close(fd);
	    goto cleanup_and_exit;
	}
1620
1621
    }

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

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

1629
	/* Convert newlines to nulls, just before we write to disk. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1630
1631
	sunder(fileptr->data);

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

1634
1635
	/* Convert nulls to newlines.  data_len is the string's real
	 * length. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1636
1637
	unsunder(fileptr->data, data_len);

1638
	if (size < data_len) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1639
1640
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1641
	    fclose(f);
1642
	    goto cleanup_and_exit;
1643
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1644

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1645
	/* If we're on the last line of the file, don't write a newline
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1646
1647
1648
1649
1650
1651
1652
	 * character after it.  If the last line of the file is blank,
	 * this means that zero bytes are written, in which case we
	 * don't count the last line in the total lines written. */
	if (fileptr == openfile->filebot) {
	    if (fileptr->data[0] == '\0')
		lineswritten--;
	} else {
1653
#ifndef NANO_TINY
1654
1655
1656
1657
	    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
1658
			strerror(errno));
1659
1660
1661
		    fclose(f);
		    goto cleanup_and_exit;
		}
1662
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1663

1664
	    if (openfile->fmt != MAC_FILE) {
1665
#endif
1666
1667
		if (putc('\n', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1668
			strerror(errno));
1669
1670
1671
		    fclose(f);
		    goto cleanup_and_exit;
		}
1672
#ifndef NANO_TINY
1673
	    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1674
#endif
1675
	}
Chris Allegretta's avatar
Chris Allegretta committed
1676
1677
1678
1679
1680

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

1681
    /* If we're prepending, open the temp file, and append it to f. */
1682
    if (append == PREPEND) {
1683
1684
1685
1686
	int fd_source;
	FILE *f_source = NULL;

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

1688
1689
1690
1691
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
1692
	}
1693

1694
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1695
1696
	    statusbar(_("Error reading %s: %s"), tempname,
		strerror(errno));
1697
	    beep();
1698
	    fclose(f);
1699
	    goto cleanup_and_exit;
1700
1701
	}

1702
	if (copy_file(f_source, f) == -1 || unlink(tempname) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1703
1704
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1705
	    goto cleanup_and_exit;
1706
	}
1707
1708
    } else
	fclose(f);
1709

1710
    if (!tmp && append == OVERWRITE) {
Chris Allegretta's avatar
Chris Allegretta committed
1711
	if (!nonamechange) {
1712
1713
	    openfile->filename = mallocstrcpy(openfile->filename,
		realname);
Chris Allegretta's avatar
Chris Allegretta committed
1714
#ifdef ENABLE_COLOR
1715
	    /* We might have changed the filename, so update the colors
1716
1717
	     * to account for it, and then make sure we're using
	     * them. */
1718
	    color_update();
1719
	    color_init();
1720
1721
1722
1723
1724

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

1729
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1730
1731
1732
1733
1734
	/* 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);
1735
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1736

1737
1738
	statusbar(P_("Wrote %lu line", "Wrote %lu lines",
		(unsigned long)lineswritten),
1739
		(unsigned long)lineswritten);
1740
	openfile->modified = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
1741
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
1742
    }
1743

1744
    retval = TRUE;
1745
1746
1747

  cleanup_and_exit:
    free(realname);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1748
1749
    if (tempname != NULL)
	free(tempname);
1750

1751
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1752
1753
}

1754
#ifndef NANO_TINY
1755
1756
/* Write a marked selection from a file out to disk.  Return TRUE on
 * success or FALSE on error. */
1757
bool write_marked_file(const char *name, FILE *f_open, bool tmp,
1758
	append_type append)
1759
{
1760
    bool retval;
1761
1762
    bool old_modified = openfile->modified;
	/* write_file() unsets the modified flag. */
1763
    bool added_magicline = FALSE;
1764
1765
1766
1767
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

1768
1769
    assert(openfile->mark_set);

1770
1771
1772
    /* Partition the filestruct so that it contains only the marked
     * text. */
    mark_order((const filestruct **)&top, &top_x,
1773
	(const filestruct **)&bot, &bot_x, NULL);
1774
    filepart = partition_filestruct(top, top_x, bot, bot_x);
1775

1776
1777
1778
1779
1780
1781
    /* Handle the magicline if the NO_NEWLINES flag isn't set.  If the
     * line at filebot is blank, treat it as the magicline and hence the
     * end of the file.  Otherwise, add a magicline and treat it as the
     * end of the file. */
    if (!ISSET(NO_NEWLINES) &&
	(added_magicline = (openfile->filebot->data[0] != '\0')))
1782
	new_magicline();
1783

1784
    retval = write_file(name, f_open, tmp, append, TRUE);
1785

1786
1787
1788
    /* If the NO_NEWLINES flag isn't set, and we added a magicline,
     * remove it now. */
    if (!ISSET(NO_NEWLINES) && added_magicline)
1789
1790
1791
1792
	remove_magicline();

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

1795
    if (old_modified)
1796
1797
1798
1799
	set_modified();

    return retval;
}
1800
#endif /* !NANO_TINY */
1801

1802
1803
1804
/* Write the current file to disk.  If the mark is on, write the current
 * marked selection to disk.  If exiting is TRUE, write the file to disk
 * regardless of whether the mark is on, and without prompting if the
1805
1806
 * TEMP_FILE flag is set.  Return TRUE on success or FALSE on error. */
bool do_writeout(bool exiting)
Chris Allegretta's avatar
Chris Allegretta committed
1807
{
1808
    int i;
1809
    append_type append = OVERWRITE;
1810
    char *ans;
1811
	/* The last answer the user typed at the statusbar prompt. */
1812
#ifdef NANO_EXTRA
1813
    static bool did_credits = FALSE;
1814
#endif
1815
    bool retval = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
1816

1817
    currshortcut = writefile_list;
1818

1819
    if (exiting && openfile->filename[0] != '\0' && ISSET(TEMP_FILE)) {
1820
	retval = write_file(openfile->filename, NULL, FALSE, OVERWRITE,
1821
		FALSE);
1822
1823

	/* Write succeeded. */
1824
	if (retval)
1825
	    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1826
1827
    }

1828
    ans = mallocstrcpy(NULL,
1829
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1830
	(!exiting && openfile->mark_set) ? "" :
1831
#endif
1832
	openfile->filename);
1833
1834
1835

    while (TRUE) {
	const char *msg;
1836
#ifndef NANO_TINY
1837
1838
	const char *formatstr, *backupstr;

1839
	formatstr = (openfile->fmt == DOS_FILE) ?
1840
1841
		_(" [DOS Format]") : (openfile->fmt == MAC_FILE) ?
		_(" [Mac Format]") : "";
1842

1843
	backupstr = ISSET(BACKUP_FILE) ? _(" [Backup]") : "";
1844

1845
	/* If we're using restricted mode, don't display the "Write
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1846
1847
1848
	 * Selection to File" prompt.  This function is disabled, since
	 * it allows reading from or writing to files not specified on
	 * the command line. */
1849
	if (!ISSET(RESTRICTED) && !exiting && openfile->mark_set)
1850
	    msg = (append == PREPEND) ?
1851
1852
1853
		_("Prepend Selection to File") : (append == APPEND) ?
		_("Append Selection to File") :
		_("Write Selection to File");
1854
	else
1855
#endif /* !NANO_TINY */
1856
1857
1858
	    msg = (append == PREPEND) ? _("File Name to Prepend to") :
		(append == APPEND) ? _("File Name to Append to") :
		_("File Name to Write");
1859

1860
1861
1862
	/* If we're using restricted mode, the filename isn't blank,
	 * and we're at the "Write File" prompt, disable tab
	 * completion. */
1863
	i = do_prompt(!ISSET(RESTRICTED) ||
1864
1865
1866
1867
1868
		openfile->filename[0] == '\0',
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
		writefile_list, ans,
1869
#ifndef NANO_TINY
1870
1871
		NULL,
#endif
1872
		edit_refresh, "%s%s%s", msg,
1873
1874
#ifndef NANO_TINY
		formatstr, backupstr
1875
#else
1876
		"", ""
1877
1878
1879
#endif
		);

1880
1881
	/* If the filename or command begins with a newline (i.e. an
	 * encoded null), treat it as though it's blank. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1882
	if (i < 0 || *answer == '\n') {
1883
	    statusbar(_("Cancelled"));
1884
	    retval = FALSE;
1885
1886
1887
	    break;
	} else {
	    ans = mallocstrcpy(ans, answer);
Chris Allegretta's avatar
Chris Allegretta committed
1888

1889
#ifndef DISABLE_BROWSER
1890
1891
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
1892

1893
1894
		if (tmp == NULL)
		    continue;
1895

1896
		/* We have a file now.  Indicate this. */
1897
1898
1899
		free(answer);
		answer = tmp;
	    } else
1900
#endif /* !DISABLE_BROWSER */
1901
#ifndef NANO_TINY
1902
	    if (i == TOGGLE_DOS_KEY) {
1903
1904
		openfile->fmt = (openfile->fmt == DOS_FILE) ? NIX_FILE :
			DOS_FILE;
1905
1906
		continue;
	    } else if (i == TOGGLE_MAC_KEY) {
1907
1908
		openfile->fmt = (openfile->fmt == MAC_FILE) ? NIX_FILE :
			MAC_FILE;
1909
1910
1911
1912
1913
		continue;
	    } else if (i == TOGGLE_BACKUP_KEY) {
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
1914
#endif /* !NANO_TINY */
1915
	    if (i == NANO_PREPEND_KEY) {
1916
		append = (append == PREPEND) ? OVERWRITE : PREPEND;
1917
1918
		continue;
	    } else if (i == NANO_APPEND_KEY) {
1919
		append = (append == APPEND) ? OVERWRITE : APPEND;
1920
1921
		continue;
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1922

Chris Allegretta's avatar
Chris Allegretta committed
1923
#ifdef DEBUG
1924
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
1925
#endif
1926
1927

#ifdef NANO_EXTRA
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1928
1929
1930
1931
1932
1933
1934
1935
	    /* If the current file has been modified, we've pressed
	     * Ctrl-X at the edit window to exit, we've pressed "y" at
	     * the "Save modified buffer" prompt to save, we've entered
	     * "zzy" as the filename to save under (hence "xyzzy"), and
	     * this is the first time we've done this, show an Easter
	     * egg.  Display the credits. */
	    if (!did_credits && exiting && !ISSET(TEMP_FILE) &&
		strcasecmp(answer, "zzy") == 0) {
1936
		do_credits();
1937
		did_credits = TRUE;
1938
		retval = FALSE;
1939
1940
		break;
	    }
1941
#endif
1942

1943
	    if (append == OVERWRITE) {
1944
		size_t answer_len = strlen(answer);
1945
		bool name_exists, do_warning;
1946
		char *full_answer, *full_filename;
1947
1948
		struct stat st;

1949
1950
1951
		/* Convert newlines to nulls, just before we get the
		 * full path. */
		sunder(answer);
1952

1953
1954
		full_answer = get_full_path(answer);
		full_filename = get_full_path(openfile->filename);
1955
1956
		name_exists = (stat((full_answer == NULL) ? answer :
			full_answer, &st) != -1);
1957
1958
1959
1960
1961
1962
		if (openfile->filename[0] == '\0')
		    do_warning = name_exists;
		else
		    do_warning = (strcmp((full_answer == NULL) ?
			answer : full_answer, (full_filename == NULL) ?
			openfile->filename : full_filename) != 0);
1963

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1964
1965
1966
1967
		/* Convert nulls to newlines.  answer_len is the
		 * string's real length. */
		unsunder(answer, answer_len);

1968
1969
1970
1971
		if (full_filename != NULL)
		    free(full_filename);
		if (full_answer != NULL)
		    free(full_answer);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1972

1973
		if (do_warning) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1974
1975
1976
1977
1978
1979
1980
1981
		    /* If we're using restricted mode, we aren't allowed
		     * to overwrite an existing file with the current
		     * file.  We also aren't allowed to change the name
		     * of the current file if it has one, because that
		     * would allow reading from or writing to files not
		     * specified on the command line. */
		    if (ISSET(RESTRICTED))
			continue;
1982

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1983
		    if (name_exists) {
1984
1985
1986
1987
			i = do_yesno_prompt(FALSE,
				_("File exists, OVERWRITE ? "));
			if (i == 0 || i == -1)
			    continue;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1988
		    } else
1989
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1990
		    if (exiting || !openfile->mark_set)
1991
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1992
		    {
1993
1994
1995
1996
1997
			i = do_yesno_prompt(FALSE,
				_("Save file under DIFFERENT NAME ? "));
			if (i == 0 || i == -1)
			    continue;
		    }
1998
		}
1999
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2000

2001
2002
2003
2004
2005
	    /* Convert newlines to nulls, just before we save the
	     * file. */
	    sunder(answer);
	    align(&answer);

2006
	    /* Here's where we allow the selected text to be written to
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2007
2008
2009
	     * a separate file.  If we're using restricted mode, this
	     * function is disabled, since it allows reading from or
	     * writing to files not specified on the command line. */
2010
2011
2012
2013
	    retval =
#ifndef NANO_TINY
		(!ISSET(RESTRICTED) && !exiting && openfile->mark_set) ?
		write_marked_file(answer, NULL, FALSE, append) :
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2014
#endif
2015
		write_file(answer, NULL, FALSE, append, FALSE);
2016

2017
2018
	    break;
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2019
    }
2020
2021

    free(ans);
2022

2023
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
2024
2025
}

2026
2027
/* Write the current file to disk.  If the mark is on, write the current
 * marked selection to disk. */
2028
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
2029
{
2030
    do_writeout(FALSE);
2031
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
2032
}
Chris Allegretta's avatar
Chris Allegretta committed
2033

2034
2035
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
2036
char *real_dir_from_tilde(const char *buf)
2037
{
2038
    char *retval;
2039

2040
    assert(buf != NULL);
2041

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2042
    if (*buf == '~') {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2043
	size_t i = 1;
2044
	char *tilde_dir;
2045

2046
	/* Figure out how much of the string we need to compare. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2047
	for (; buf[i] != '/' && buf[i] != '\0'; i++)
2048
2049
	    ;

2050
2051
2052
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
2053
	    tilde_dir = mallocstrcpy(NULL, homedir);
2054
2055
2056
	} else {
	    const struct passwd *userdata;

2057
2058
2059
	    tilde_dir = mallocstrncpy(NULL, buf, i + 1);
	    tilde_dir[i] = '\0';

2060
2061
	    do {
		userdata = getpwent();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2062
2063
	    } while (userdata != NULL && strcmp(userdata->pw_name,
		tilde_dir + 1) != 0);
2064
	    endpwent();
2065
	    if (userdata != NULL)
2066
		tilde_dir = mallocstrcpy(tilde_dir, userdata->pw_dir);
Chris Allegretta's avatar
Chris Allegretta committed
2067
	}
2068

2069
2070
	retval = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	sprintf(retval, "%s%s", tilde_dir, buf + i);
2071

2072
2073
2074
	free(tilde_dir);
    } else
	retval = mallocstrcpy(NULL, buf);
2075

2076
    return retval;
2077
2078
}

2079
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
2080
/* Our sort routine for file listings.  Sort alphabetically and
2081
 * case-insensitively, and sort directories before filenames. */
2082
2083
2084
2085
2086
2087
2088
2089
2090
2091
2092
2093
2094
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;

2095
2096
2097
2098
    /* Standard function brain damage: We should be sorting
     * alphabetically and case-insensitively according to the current
     * locale, but there's no standard strcasecoll() function, so we
     * have to use multibyte strcasecmp() instead, */
2099
    return mbstrcasecmp(a, b);
2100
}
2101
2102
2103
2104
2105

/* Free the memory allocated for array, which should contain len
 * elements. */
void free_chararray(char **array, size_t len)
{
2106
2107
    assert(array != NULL);

2108
2109
2110
2111
    for (; len > 0; len--)
	free(array[len - 1]);
    free(array);
}
2112
2113
#endif

Chris Allegretta's avatar
Chris Allegretta committed
2114
#ifndef DISABLE_TABCOMP
2115
/* Is the given path a directory? */
2116
bool is_dir(const char *buf)
2117
{
2118
    char *dirptr;
2119
    struct stat fileinfo;
2120
2121
2122
    bool retval;

    assert(buf != NULL);
2123

2124
2125
2126
2127
    dirptr = real_dir_from_tilde(buf);

    retval = (stat(dirptr, &fileinfo) != -1 &&
	S_ISDIR(fileinfo.st_mode));
2128

2129
    free(dirptr);
2130

2131
    return retval;
2132
}
Chris Allegretta's avatar
Chris Allegretta committed
2133

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2134
2135
2136
/* These functions, username_tab_completion(), cwd_tab_completion()
 * (originally exe_n_cwd_tab_completion()), and input_tab(), were
 * adapted from busybox 0.46 (cmdedit.c).  Here is the notice from that
2137
 * file, with the copyright years updated:
Chris Allegretta's avatar
Chris Allegretta committed
2138
2139
2140
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
2141
 * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007
Chris Allegretta's avatar
Chris Allegretta committed
2142
2143
2144
2145
2146
2147
2148
2149
 *      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.
2150
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
2151

2152
/* We consider the first buf_len characters of buf for ~username tab
2153
2154
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
2155
	size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2156
{
2157
2158
    char **matches = NULL;
    const struct passwd *userdata;
2159

2160
    assert(buf != NULL && num_matches != NULL && buf_len > 0);
2161

2162
    *num_matches = 0;
2163

2164
    while ((userdata = getpwent()) != NULL) {
2165
	if (strncmp(userdata->pw_name, buf + 1, buf_len - 1) == 0) {
2166
2167
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2168

2169
2170
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2171
2172
2173
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
2174
2175
#endif

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2176
2177
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2178
2179
2180
	    matches[*num_matches] =
		charalloc(strlen(userdata->pw_name) + 2);
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
2181
	    ++(*num_matches);
2182
	}
2183
2184
    }
    endpwent();
2185

2186
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2187
2188
}

2189
/* We consider the first buf_len characters of buf for filename tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2190
 * completion. */
2191
char **cwd_tab_completion(const char *buf, bool allow_files, size_t
2192
	*num_matches, size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2193
{
2194
    char *dirname = mallocstrcpy(NULL, buf), *filename;
2195
2196
2197
2198
2199
#ifndef DISABLE_OPERATINGDIR
    size_t dirnamelen;
#endif
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2200
    DIR *dir;
2201
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
2202

2203
    assert(dirname != NULL && num_matches != NULL);
2204

2205
    *num_matches = 0;
2206
    null_at(&dirname, buf_len);
2207
2208
2209
2210
2211
2212
2213
2214

    /* 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
2215
	*tmpdirname = '\0';
2216
2217
2218
	tmpdirname = dirname;
	dirname = real_dir_from_tilde(dirname);
	free(tmpdirname);
Chris Allegretta's avatar
Chris Allegretta committed
2219
    } else {
2220
2221
	filename = dirname;
	dirname = mallocstrcpy(NULL, "./");
Chris Allegretta's avatar
Chris Allegretta committed
2222
2223
    }

2224
    assert(dirname[strlen(dirname) - 1] == '/');
2225

Chris Allegretta's avatar
Chris Allegretta committed
2226
    dir = opendir(dirname);
2227

2228
    if (dir == NULL) {
2229
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
2230
	beep();
2231
2232
2233
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2234
    }
2235
2236
2237
2238
2239
2240

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

2241
    while ((nextdir = readdir(dir)) != NULL) {
2242
2243
	bool skip_match = FALSE;

Chris Allegretta's avatar
Chris Allegretta committed
2244
#ifdef DEBUG
2245
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
2246
#endif
2247
	/* See if this matches. */
2248
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
2249
2250
		(*filename == '.' || (strcmp(nextdir->d_name, ".") !=
		0 && strcmp(nextdir->d_name, "..") != 0))) {
2251
2252
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2253

2254
2255
2256
2257
	    char *tmp = charalloc(strlen(dirname) +
		strlen(nextdir->d_name) + 1);
	    sprintf(tmp, "%s%s", dirname, nextdir->d_name);

2258
2259
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2260
2261
2262
2263
2264
2265
2266
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(tmp, TRUE))
		skip_match = TRUE;
#endif

	    /* ...or unless the match isn't a directory and allow_files
	     * isn't set, in which case just go to the next match. */
2267
	    if (!allow_files && !is_dir(tmp))
2268
2269
2270
		skip_match = TRUE;

	    free(tmp);
2271

2272
	    if (skip_match)
2273
		continue;
2274

2275
2276
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2277
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
2278
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
2279
2280
	}
    }
2281

2282
2283
    closedir(dir);
    free(dirname);
2284
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
2285

2286
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2287
2288
}

2289
/* Do tab completion.  place refers to how much the statusbar cursor
2290
2291
 * position should be advanced.  refresh_func is the function we will
 * call to refresh the edit window. */
2292
char *input_tab(char *buf, bool allow_files, size_t *place, bool
2293
	*lastwastab, void (*refresh_func)(void), bool *list)
Chris Allegretta's avatar
Chris Allegretta committed
2294
{
2295
    size_t num_matches = 0, buf_len;
2296
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2297

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

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2300
    *list = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2301

2302
2303
    /* If the word starts with `~' and there is no slash in the word,
     * then try completing this word as a username. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2304
    if (*place > 0 && *buf == '~') {
2305
	const char *bob = strchr(buf, '/');
Chris Allegretta's avatar
Chris Allegretta committed
2306

2307
2308
2309
2310
	if (bob == NULL || bob >= buf + *place)
	    matches = username_tab_completion(buf, &num_matches,
		*place);
    }
2311

2312
2313
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
2314
2315
	matches = cwd_tab_completion(buf, allow_files, &num_matches,
		*place);
2316

2317
2318
2319
    buf_len = strlen(buf);

    if (num_matches == 0 || *place != buf_len)
2320
2321
2322
2323
	beep();
    else {
	size_t match, common_len = 0;
	char *mzero;
2324
2325
2326
	const char *lastslash = revstrstr(buf, "/", buf + *place);
	size_t lastslash_len = (lastslash == NULL) ? 0 :
		lastslash - buf + 1;
2327
2328
2329
	char *match1_mb = charalloc(mb_cur_max() + 1);
	char *match2_mb = charalloc(mb_cur_max() + 1);
	int match1_mb_len, match2_mb_len;
2330
2331
2332

	while (TRUE) {
	    for (match = 1; match < num_matches; match++) {
2333
2334
		/* Get the number of single-byte characters that all the
		 * matches have in common. */
2335
		match1_mb_len = parse_mbchar(matches[0] + common_len,
2336
			match1_mb, NULL);
2337
		match2_mb_len = parse_mbchar(matches[match] +
2338
			common_len, match2_mb, NULL);
2339
2340
2341
		match1_mb[match1_mb_len] = '\0';
		match2_mb[match2_mb_len] = '\0';
		if (strcmp(match1_mb, match2_mb) != 0)
2342
2343
		    break;
	    }
2344

2345
	    if (match < num_matches || matches[0][common_len] == '\0')
2346
		break;
2347

2348
	    common_len += parse_mbchar(buf + common_len, NULL, NULL);
2349
	}
2350

2351
2352
2353
	free(match1_mb);
	free(match2_mb);

2354
	mzero = charalloc(lastslash_len + common_len + 1);
2355
2356
2357

	strncpy(mzero, buf, lastslash_len);
	strncpy(mzero + lastslash_len, matches[0], common_len);
2358

2359
	common_len += lastslash_len;
2360
	mzero[common_len] = '\0';
2361

2362
	assert(common_len >= *place);
2363

2364
	if (num_matches == 1 && is_dir(mzero)) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2365
	    mzero[common_len++] = '/';
2366

2367
2368
	    assert(common_len > *place);
	}
2369

2370
	if (num_matches > 1 && (common_len != *place || !*lastwastab))
2371
	    beep();
2372

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2373
	/* If there is more of a match to display on the statusbar, show
2374
	 * it.  We reset lastwastab to FALSE: it requires pressing Tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2375
	 * twice in succession with no statusbar changes to see a match
2376
2377
2378
	 * list. */
	if (common_len != *place) {
	    *lastwastab = FALSE;
2379
	    buf = charealloc(buf, common_len + buf_len - *place + 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2380
2381
	    charmove(buf + common_len, buf + *place, buf_len -
		*place + 1);
2382
	    strncpy(buf, mzero, common_len);
2383
	    *place = common_len;
2384
	} else if (!*lastwastab || num_matches < 2)
2385
2386
	    *lastwastab = TRUE;
	else {
2387
	    int longest_name = 0, columns, editline = 0;
2388

2389
2390
2391
2392
2393
2394
2395
2396
	    /* 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
2397

2398
2399
		if (common_len > COLS - 1) {
		    longest_name = COLS - 1;
2400
2401
		    break;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2402

2403
2404
		if (common_len > longest_name)
		    longest_name = common_len;
2405
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2406

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

2409
	    /* Each column will be (longest_name + 2) columns wide, i.e.
2410
2411
	     * two spaces between columns, except that there will be
	     * only one space after the last column. */
2412
	    columns = (COLS + 1) / (longest_name + 2);
2413

2414
2415
2416
2417
	    /* Blank the edit window, and print the matches out
	     * there. */
	    blank_edit();
	    wmove(edit, 0, 0);
2418

2419
2420
	    /* Disable el cursor. */
	    curs_set(0);
2421

2422
2423
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2424

2425
2426
		wmove(edit, editline, (longest_name + 2) *
			(match % columns));
2427

2428
2429
2430
		if (match % columns == 0 &&
			editline == editwinrows - 1 &&
			num_matches - match > columns) {
2431
2432
2433
		    waddstr(edit, _("(more)"));
		    break;
		}
2434

2435
2436
2437
2438
		disp = display_string(matches[match], 0, longest_name,
			FALSE);
		waddstr(edit, disp);
		free(disp);
2439

2440
		if ((match + 1) % columns == 0)
2441
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2442
	    }
2443

2444
	    wnoutrefresh(edit);
2445
	    *list = TRUE;
2446
2447
2448
	}

	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2449
2450
    }

2451
    free_chararray(matches, num_matches);
2452

2453
    /* Only refresh the edit window if we don't have a list of filename
2454
     * matches on it. */
2455
    if (!*list)
2456
	refresh_func();
2457
2458

    /* Enable el cursor. */
2459
    curs_set(1);
2460

2461
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2462
}
2463
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2464

2465
2466
/* Only print the last part of a path.  Isn't there a shell command for
 * this? */
2467
2468
const char *tail(const char *foo)
{
2469
    const char *tmp = strrchr(foo, '/');
2470

2471
2472
    if (tmp == NULL)
	tmp = foo;
2473
    else
2474
2475
2476
2477
2478
	tmp++;

    return tmp;
}

2479
#if !defined(NANO_TINY) && defined(ENABLE_NANORC)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2480
2481
2482
/* Return $HOME/.nano_history, or NULL if we can't find the home
 * directory.  The string is dynamically allocated, and should be
 * freed. */
2483
char *histfilename(void)
2484
{
2485
    char *nanohist = NULL;
2486

2487
2488
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2489

2490
2491
2492
	nanohist = charalloc(homelen + 15);
	strcpy(nanohist, homedir);
	strcpy(nanohist + homelen, "/.nano_history");
Chris Allegretta's avatar
Chris Allegretta committed
2493
    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2494

2495
2496
2497
    return nanohist;
}

2498
/* Load histories from ~/.nano_history. */
2499
2500
2501
void load_history(void)
{
    char *nanohist = histfilename();
Chris Allegretta's avatar
Chris Allegretta committed
2502

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2503
    /* Assume do_rcfile() has reported a missing home directory. */
2504
    if (nanohist != NULL) {
2505
	FILE *hist = fopen(nanohist, "rb");
Chris Allegretta's avatar
Chris Allegretta committed
2506

2507
	if (hist == NULL) {
2508
	    if (errno != ENOENT) {
2509
2510
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
2511
2512
2513
		rcfile_error(N_("Error reading %s: %s"), nanohist,
			strerror(errno));
		fprintf(stderr,
2514
			_("\nPress Enter to continue starting nano.\n"));
2515
2516
		while (getchar() != '\n')
		    ;
2517
	    }
2518
	} else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2519
2520
2521
	    /* Load a history list (first the search history, then the
	     * replace history) from the oldest entry to the newest.
	     * Assume the last history entry is a blank line. */
2522
	    filestruct **history = &search_history;
2523
	    char *line = NULL;
2524
	    size_t buf_len = 0;
2525
2526
	    ssize_t read;

2527
	    while ((read = getline(&line, &buf_len, hist)) >= 0) {
2528
2529
2530
2531
2532
2533
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
		if (read > 0) {
		    unsunder(line, read);
2534
2535
		    update_history(history, line);
		} else
2536
2537
		    history = &replace_history;
	    }
2538

2539
	    fclose(hist);
2540
	    free(line);
2541
	}
2542
2543
2544
2545
	free(nanohist);
    }
}

2546
2547
2548
/* Write the lines of a history list, starting with the line at h, to
 * the open file at hist.  Return TRUE if the write succeeded, and FALSE
 * otherwise. */
2549
bool writehist(FILE *hist, filestruct *h)
2550
{
2551
    filestruct *p;
2552

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2553
2554
    /* Write a history list from the oldest entry to the newest.  Assume
     * the last history entry is a blank line. */
2555
    for (p = h; p != NULL; p = p->next) {
2556
	size_t p_len = strlen(p->data);
2557

2558
	sunder(p->data);
2559

2560
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
2561
2562
		putc('\n', hist) == EOF)
	    return FALSE;
2563
    }
2564

2565
    return TRUE;
2566
2567
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2568
/* Save histories to ~/.nano_history. */
2569
2570
void save_history(void)
{
2571
    char *nanohist;
2572

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2573
    /* Don't save unchanged or empty histories. */
2574
2575
    if (!history_has_changed() || (searchbot->lineno == 1 &&
	replacebot->lineno == 1))
2576
2577
	return;

2578
2579
2580
2581
    nanohist = histfilename();

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

2583
	if (hist == NULL)
2584
2585
	    rcfile_error(N_("Error writing %s: %s"), nanohist,
		strerror(errno));
2586
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2587
2588
	    /* Make sure no one else can read from or write to the
	     * history file. */
2589
	    chmod(nanohist, S_IRUSR | S_IWUSR);
2590

2591
2592
	    if (!writehist(hist, searchage) || !writehist(hist,
		replaceage))
2593
2594
		rcfile_error(N_("Error writing %s: %s"), nanohist,
			strerror(errno));
2595

2596
2597
	    fclose(hist);
	}
2598

2599
2600
2601
	free(nanohist);
    }
}
2602
#endif /* !NANO_TINY && ENABLE_NANORC */