files.c 66.3 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/* $Id$ */
Chris Allegretta's avatar
Chris Allegretta committed
2
3
4
/**************************************************************************
 *   files.c                                                              *
 *                                                                        *
5
 *   Copyright (C) 1999-2004 Chris Allegretta                             *
6
 *   Copyright (C) 2005-2006 David Lawrence Ramsey                        *
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 2, 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
137
138
    /* If we have a file and we're loading into a new buffer, update the
     * filename. */
    if (rc != -1 && new_buffer)
	openfile->filename = mallocstrcpy(openfile->filename, filename);
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
299
    /* Convert nulls to newlines.  buf_len is the string's real length
     * here. */
    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 == TRUE) {
313
314
315
	/* Special case: We're inserting with the cursor on the first
	 * line. */
	fileptr->prev = NULL;
316
	fileptr->next = openfile->fileage;
317
318
319
320
321
322
	fileptr->lineno = 1;
	if (*first_line_ins == TRUE) {
	    *first_line_ins = FALSE;
	    /* If we're inserting into the first line of the file, then
	     * we want to make sure that our edit buffer stays on the
	     * first line and that fileage stays up to date. */
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
	    /* If there's a '\r' before the '\n', set format to DOS if
	     * we currently think this is a *nix file, or to both if we
	     * currently think it's a Mac file. */
	    if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r' &&
		(format == 0 || format == 2))
		format++;
391
#endif
392

393
394
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
395

396
397
398
	    /* Reset the line length in preparation for the next
	     * line. */
	    len = 0;
Chris Allegretta's avatar
Chris Allegretta committed
399

400
401
402
	    num_lines++;
	    buf[0] = '\0';
	    i = 0;
403
#ifndef NANO_TINY
404
405
406
407
408
409
410
411
	/* If it's a Mac file ('\r' without '\n'), and file conversion
	 * isn't disabled, handle it! */
	} else if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r') {
	    /* If we currently think the file is a *nix file, set format
	     * to Mac.  If we currently think the file is a DOS file,
	     * set format to both DOS and Mac. */
	    if (format == 0 || format == 1)
		format += 2;
412

413
414
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
415

416
417
418
419
	    /* 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
420

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

431
432
433
434
435
436
437
438
	    /* 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);
439
	    }
440
441
442
443
444
445
446
447
448
449
450

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

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

452
#ifndef NANO_TINY
453
454
455
456
457
458
459
460
    /* 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';
    }
461
#endif
462

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

475
476
477
478
	/* Read in the last line properly. */
	fileptr = read_line(buf, fileptr, &first_line_ins, len);
	num_lines++;
    }
479

480
    free(buf);
481

482
    /* If we didn't get a file and we don't already have one, open a
483
     * blank buffer. */
484
    if (fileptr == NULL)
485
	open_buffer("");
486

487
488
    /* Attach the file we got to the filestruct.  If we got a file of
     * zero bytes, don't do anything. */
489
    if (num_lines > 0) {
490
491
492
493
494
495
496
497
498
499
500
501
	/* 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
502
503
	    /* Tack the text at fileptr onto the beginning of the text
	     * at current. */
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
	    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
519
	    /* Move fileptr back one line and blow away the old fileptr,
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
	     * 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. */
536
	renumber(openfile->current);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
537
    }
538

539
540
    openfile->totsize += get_totsize(openfile->fileage,
	openfile->filebot);
541

542
543
544
545
546
    /* If the NO_NEWLINES flag isn't set, and text has been added to
     * the magicline (i.e, a file that doesn't end in a newline has been
     * 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') {
547
548
549
550
551
552
553
554
555
	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();

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

578
579
580
581
582
583
584
585
586
587
588
589
/* Open the file (and decide if it exists).  If newfie is TRUE, display
 * "New File" if the file is missing.  Otherwise, say "[filename] not
 * found".
 *
 * Return -2 if we say "New File".  Otherwise, -1 if the file isn't
 * opened, 0 otherwise.  The file might still have an error while
 * reading with a 0 return value.  *f is set to the opened file. */
int open_file(const char *filename, bool newfie, FILE **f)
{
    int fd;
    struct stat fileinfo;

590
    assert(filename != NULL && f != NULL);
591

592
    if (stat(filename, &fileinfo) == -1) {
593
594
595
596
597
	if (newfie) {
	    statusbar(_("New File"));
	    return -2;
	}
	statusbar(_("\"%s\" not found"), filename);
598
	beep();
599
600
	return -1;
    } else if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
601
	S_ISBLK(fileinfo.st_mode)) {
602
	/* Don't open character or block files.  Sorry, /dev/sndstat! */
603
604
605
	statusbar(S_ISDIR(fileinfo.st_mode) ?
		_("\"%s\" is a directory") :
		_("File \"%s\" is a device file"), filename);
606
	beep();
607
608
609
	return -1;
    } else if ((fd = open(filename, O_RDONLY)) == -1) {
	statusbar(_("Error reading %s: %s"), filename, strerror(errno));
610
	beep();
611
612
613
614
615
 	return -1;
     } else {
	/* File is A-OK.  Open it in binary mode for our own end-of-line
	 * character munging. */
	*f = fdopen(fd, "rb");
616

617
618
619
	if (*f == NULL) {
	    statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
620
	    beep();
621
622
623
	    close(fd);
	} else
	    statusbar(_("Reading File"));
624
    }
625

626
    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
627
628
}

629
630
631
632
633
/* 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)
634
{
635
    static int ulmax_digits = -1;
636
637
638
    unsigned long i = 0;
    char *buf;
    size_t namelen, suffixlen;
639

640
    assert(name != NULL && suffix != NULL);
641

642
643
644
    if (ulmax_digits == -1)
	ulmax_digits = digits(ULONG_MAX);

645
646
    namelen = strlen(name);
    suffixlen = strlen(suffix);
Chris Allegretta's avatar
Chris Allegretta committed
647

648
    buf = charalloc(namelen + suffixlen + ulmax_digits + 2);
649
    sprintf(buf, "%s%s", name, suffix);
650

651
652
    while (TRUE) {
	struct stat fs;
Chris Allegretta's avatar
Chris Allegretta committed
653

654
655
656
657
	if (stat(buf, &fs) == -1)
	    return buf;
	if (i == ULONG_MAX)
	    break;
658

659
660
661
	i++;
	sprintf(buf + namelen + suffixlen, ".%lu", i);
    }
Chris Allegretta's avatar
Chris Allegretta committed
662

663
664
665
    /* We get here only if there is no possible save file.  Blank out
     * the filename to indicate this. */
    null_at(&buf, 0);
666

667
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
668
669
}

670
671
672
/* 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. */
673
void do_insertfile(
674
#ifndef NANO_TINY
675
676
677
678
679
680
681
682
683
684
	bool execute
#else
	void
#endif
	)
{
    int i;
    const char *msg;
    char *ans = mallocstrcpy(NULL, "");
	/* The last answer the user typed on the statusbar. */
685
686
    filestruct *edittop_save = openfile->edittop;
    ssize_t current_y_save = openfile->current_y;
687
688
    bool at_edittop = FALSE;
	/* Whether we're at the top of the edit window. */
689

690
691
692
#ifndef DISABLE_WRAPPING
    wrap_reset();
#endif
693

694
    while (TRUE) {
695
#ifndef NANO_TINY
696
	if (execute) {
697
	    msg = 
698
#ifdef ENABLE_MULTIBUFFER
699
700
		ISSET(MULTIBUFFER) ? 
		N_("Command to execute in new buffer [from %s] ") :
701
#endif
702
		N_("Command to execute [from %s] ");
703
704
	} else {
#endif
705
	    msg =
706
#ifdef ENABLE_MULTIBUFFER
707
708
		ISSET(MULTIBUFFER) ? 
		N_("File to insert into new buffer [from %s] ") :
709
#endif
710
		N_("File to insert [from %s] ");
711
#ifndef NANO_TINY
712
713
	}
#endif
714

715
	i = do_prompt(TRUE,
716
717
718
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
719
#ifndef NANO_TINY
720
		execute ? extcmd_list :
721
#endif
722
		insertfile_list, ans,
723
#ifndef NANO_TINY
724
		NULL,
725
#endif
726
		 edit_refresh, _(msg),
727
728
729
730
731
#ifndef DISABLE_OPERATINGDIR
		operating_dir != NULL && strcmp(operating_dir, ".") != 0 ?
		operating_dir :
#endif
		"./");
732

733
734
735
736
737
738
739
740
741
742
	/* If we're in multibuffer mode and the filename or command is
	 * blank, open a new buffer instead of canceling. */
	if (i == -1 || (i == -2
#ifdef ENABLE_MULTIBUFFER
		&& !ISSET(MULTIBUFFER)
#endif
		)) {
	    statusbar(_("Cancelled"));
	    break;
	} else {
743
	    size_t pww_save = openfile->placewewant;
744

745
	    ans = mallocstrcpy(ans, answer);
746

747
#ifndef NANO_TINY
748
749
750
751
752
753
754
#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
755
#endif
756
757
758
759
760
761
762
	    if (i == NANO_TOOTHERINSERT_KEY) {
		execute = !execute;
		continue;
	    }
#ifndef DISABLE_BROWSER
	    else
#endif
763
#endif /* !NANO_TINY */
764

765
766
767
#ifndef DISABLE_BROWSER
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
768

769
770
		if (tmp == NULL)
		    continue;
771

772
773
		free(answer);
		answer = tmp;
774

775
776
777
		/* We have a file now.  Indicate this and get out of the
		 * statusbar prompt cleanly. */
		i = 0;
778
		do_prompt_abort();
779
780
	    }
#endif
781

782
783
784
785
786
787
788
789
	    /* 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;
790

791
792
793
794
795
796
797
798
#ifdef ENABLE_MULTIBUFFER
	    if (!ISSET(MULTIBUFFER)) {
#endif
		/* If we're not inserting into a new buffer, partition
		 * the filestruct so that it contains no text and hence
		 * looks like a new buffer, and keep track of whether
		 * the top of the partition is the top of the edit
		 * window. */
799
800
801
802
803
		filepart = partition_filestruct(openfile->current,
			openfile->current_x, openfile->current,
			openfile->current_x);
		at_edittop =
			(openfile->fileage == openfile->edittop);
804
805
806
#ifdef ENABLE_MULTIBUFFER
	    }
#endif
Chris Allegretta's avatar
Chris Allegretta committed
807

808
#ifndef NANO_TINY
809
810
811
	    if (execute) {
#ifdef ENABLE_MULTIBUFFER
		if (ISSET(MULTIBUFFER))
812
		    /* Open a blank buffer. */
813
814
815
816
		    open_buffer("");
#endif

		/* Save the command's output in the current buffer. */
817
		execute_command(answer);
818
	    } else {
819
#endif
820
821
		/* Make sure the path to the file specified in answer is
		 * tilde-expanded. */
822
823
		answer = mallocstrassn(answer,
			real_dir_from_tilde(answer));
824
825
826

		/* Save the file specified in answer in the current
		 * buffer. */
827
		open_buffer(answer);
828
#ifndef NANO_TINY
829
	    }
Chris Allegretta's avatar
Chris Allegretta committed
830
#endif
831

832
#ifdef ENABLE_MULTIBUFFER
833
834
835
	    if (ISSET(MULTIBUFFER))
		/* Update the screen to account for the current
		 * buffer. */
836
		display_buffer();
837
	    else
Chris Allegretta's avatar
Chris Allegretta committed
838
#endif
839
	    {
840
		filestruct *top_save = openfile->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
841

842
843
844
845
846
847
		/* If we didn't insert into a new buffer, and we were at
		 * the top of the edit window before, set the saved
		 * value of edittop to the new top of the edit window,
		 * and update the current y-coordinate to account for
		 * the number of lines inserted. */
		if (at_edittop)
848
849
		    edittop_save = openfile->fileage;
		openfile->current_y += current_y_save;
850

851
852
853
854
855
856
		/* If we didn't insert into a new buffer, unpartition
		 * the filestruct so that it contains all the text
		 * again.  Note that we've replaced the non-text
		 * originally in the partition with the text in the
		 * inserted file/executed command output. */
		unpartition_filestruct(&filepart);
857

858
859
860
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
861

862
		/* Restore the old edittop. */
863
		openfile->edittop = edittop_save;
864

865
866
867
		/* Restore the old place we want. */
		openfile->placewewant = pww_save;

868
869
		/* Mark the file as modified. */
		set_modified();
870

871
872
		/* Update the screen. */
		edit_refresh();
873
	    }
874

875
876
877
878
879
	    break;
	}
    }

    free(ans);
880
881
}

882
883
884
/* 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. */
885
void do_insertfile_void(void)
886
{
887
888
889
890
891
892
#ifdef ENABLE_MULTIBUFFER
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
	statusbar(_("Key illegal in non-multibuffer mode"));
    else
#endif
	do_insertfile(
893
#ifndef NANO_TINY
894
895
896
		FALSE
#endif
		);
897
898
899
900

    display_main_list();
}

901
/* When passed "[relative path]" or "[relative path][filename]" in
902
 * origpath, return "[full path]" or "[full path][filename]" on success,
903
904
905
906
 * 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. */
907
char *get_full_path(const char *origpath)
908
{
909
    char *d_here, *d_there = NULL;
910

911
912
    if (origpath == NULL)
    	return NULL;
913

914
915
916
    /* Get the current directory. */
    d_here = charalloc(PATH_MAX + 1);
    d_here = getcwd(d_here, PATH_MAX + 1);
917

918
    if (d_here != NULL) {
919
920
921
922
	const char *last_slash;
	char *d_there_file = NULL;
	bool path_only;
	struct stat fileinfo;
923

924
925
	align(&d_here);

926
927
	/* If the current directory isn't "/", tack a slash onto the end
	 * of it. */
928
	if (strcmp(d_here, "/") != 0) {
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
929
	    d_here = charealloc(d_here, strlen(d_here) + 2);
930
931
	    strcat(d_here, "/");
	}
932

933
934
935
936
937
938
939
940
941
942
943
944
	d_there = real_dir_from_tilde(origpath);

	assert(d_there != NULL);

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

	/* If path_only is TRUE, make sure d_there ends in a slash. */
945
	if (path_only) {
946
947
948
949
	    size_t d_there_len = strlen(d_there);

	    if (d_there[d_there_len - 1] != '/') {
		d_there = charealloc(d_there, d_there_len + 2);
950
951
952
953
		strcat(d_there, "/");
	    }
	}

954
	/* Search for the last slash in d_there. */
955
956
	last_slash = strrchr(d_there, '/');

957
958
959
960
	/* 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);
961

962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
	    d_there_file = d_there;
	    d_there = d_here;
	} else {
	    /* If path_only is FALSE, then save the filename portion of
	     * the answer, everything after the last slash, in
	     * d_there_file. */
	    if (!path_only)
		d_there_file = mallocstrcpy(NULL, last_slash + 1);

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

	    /* Go to the path specified in d_there. */
	    if (chdir(d_there) == -1) {
977
		free(d_there);
978
979
980
981
		d_there = NULL;
	    } else {
		/* Get the full path and save it in d_there. */
		free(d_there);
982

983
984
985
		d_there = charalloc(PATH_MAX + 1);
		d_there = getcwd(d_there, PATH_MAX + 1);

986
987
988
989
990
991
992
993
994
995
		if (d_there != NULL) {
		    align(&d_there);

		    if (strcmp(d_there, "/") != 0) {
			/* Make sure d_there ends in a slash. */
			d_there = charealloc(d_there,
				strlen(d_there) + 2);
			strcat(d_there, "/");
		    }
		} else
996
997
998
999
1000
1001
1002
1003
		    /* If we couldn't get the full path, set path_only
		     * to TRUE so that we clean up correctly, free all
		     * allocated memory, and return NULL. */
		    path_only = TRUE;

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

1006
1007
	    /* Free d_here, since we're done using it. */
	    free(d_here);
1008
	}
1009

1010
1011
1012
1013
1014
	/* At this point, if path_only is FALSE and d_there exists,
	 * d_there contains the path portion of the answer and
	 * d_there_file contains the filename portion of the answer.  If
	 * this is the case, tack d_there_file onto the end of
	 * d_there, so that d_there contains the complete answer. */
1015
	if (!path_only && d_there != NULL) {
1016
1017
1018
1019
1020
1021
	    d_there = charealloc(d_there, strlen(d_there) +
		strlen(d_there_file) + 1);
	    strcat(d_there, d_there_file);
 	}

	/* Free d_there_file, since we're done using it. */
1022
1023
1024
	free(d_there_file);
    }

1025
    return d_there;
1026
}
1027

1028
1029
1030
/* 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. */
1031
char *check_writable_directory(const char *path)
1032
{
1033
1034
    char *full_path = get_full_path(path);

1035
    /* If get_full_path() fails, return NULL. */
1036
    if (full_path == NULL)
1037
	return NULL;
1038

1039
1040
1041
    /* 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
1042
	full_path[strlen(full_path) - 1] != '/') {
1043
	free(full_path);
1044
	return NULL;
1045
    }
1046

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1047
    /* Otherwise, return the full path. */
1048
1049
1050
    return full_path;
}

1051
1052
/* This function calls mkstemp(($TMPDIR|P_tmpdir|/tmp/)"nano.XXXXXX").
 * On success, it returns the malloc()ed filename and corresponding FILE
1053
 * stream, opened in "r+b" mode.  On error, it returns NULL for the
1054
1055
 * filename and leaves the FILE stream unchanged. */
char *safe_tempfile(FILE **f)
1056
{
1057
    char *full_tempdir = NULL;
1058
1059
1060
    const char *tmpdir_env;
    int fd;
    mode_t original_umask = 0;
1061

1062
1063
    assert(f != NULL);

1064
1065
1066
    /* If $TMPDIR is set and non-empty, set tempdir to it, run it
     * through get_full_path(), and save the result in full_tempdir.
     * Otherwise, leave full_tempdir set to NULL. */
1067
1068
1069
    tmpdir_env = getenv("TMPDIR");
    if (tmpdir_env != NULL && tmpdir_env[0] != '\0')
	full_tempdir = check_writable_directory(tmpdir_env);
1070

1071
1072
    /* If $TMPDIR is unset, empty, or not a writable directory, and
     * full_tempdir is NULL, try P_tmpdir instead. */
1073
    if (full_tempdir == NULL)
1074
	full_tempdir = check_writable_directory(P_tmpdir);
1075

1076
1077
1078
    /* if P_tmpdir is NULL, use /tmp. */
    if (full_tempdir == NULL)
	full_tempdir = mallocstrcpy(NULL, "/tmp/");
1079

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1080
    full_tempdir = charealloc(full_tempdir, strlen(full_tempdir) + 12);
1081
    strcat(full_tempdir, "nano.XXXXXX");
1082

1083
1084
1085
1086
1087
1088
1089
    original_umask = umask(0);
    umask(S_IRWXG | S_IRWXO);

    fd = mkstemp(full_tempdir);

    if (fd != -1)
	*f = fdopen(fd, "r+b");
1090
1091
1092
    else {
	free(full_tempdir);
	full_tempdir = NULL;
1093
    }
1094

1095
1096
    umask(original_umask);

1097
    return full_tempdir;
1098
}
1099
1100

#ifndef DISABLE_OPERATINGDIR
1101
1102
1103
1104
1105
/* Initialize full_operating_dir based on operating_dir. */
void init_operating_dir(void)
{
    assert(full_operating_dir == NULL);

1106
    if (operating_dir == NULL)
1107
	return;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1108

1109
1110
1111
    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
1112
     * inaccessible, unset operating_dir. */
1113
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) {
1114
1115
1116
1117
1118
1119
1120
	free(full_operating_dir);
	full_operating_dir = NULL;
	free(operating_dir);
	operating_dir = NULL;
    }
}

1121
1122
1123
1124
1125
/* 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)
1126
{
1127
1128
1129
1130
    /* The char *full_operating_dir is global for mem cleanup.  It
     * should have already been initialized by init_operating_dir().
     * Also, a relative operating directory path will only be handled
     * properly if this is done. */
1131

1132
    char *fullpath;
1133
    bool retval = FALSE;
1134
    const char *whereami1, *whereami2 = NULL;
1135

1136
    /* If no operating directory is set, don't bother doing anything. */
1137
    if (operating_dir == NULL)
1138
	return FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1139

1140
    assert(full_operating_dir != NULL);
1141
1142

    fullpath = get_full_path(currpath);
1143
1144

    /* fullpath == NULL means some directory in the path doesn't exist
1145
     * or is unreadable.  If allow_tabcomp is FALSE, then currpath is
1146
1147
     * what the user typed somewhere.  We don't want to report a
     * non-existent directory as being outside the operating directory,
1148
     * so we return FALSE.  If allow_tabcomp is TRUE, then currpath
1149
1150
     * exists, but is not executable.  So we say it isn't in the
     * operating directory. */
1151
    if (fullpath == NULL)
1152
	return allow_tabcomp;
1153
1154
1155
1156
1157

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

1158
    /* If both searches failed, we're outside the operating directory.
1159
     * Otherwise, check the search results.  If the full operating
1160
1161
1162
     * 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. */
1163
    if (whereami1 != fullpath && whereami2 != full_operating_dir)
1164
1165
	retval = TRUE;
    free(fullpath);
1166
1167

    /* Otherwise, we're still inside it. */
1168
    return retval;
1169
}
1170
1171
#endif

1172
#ifndef NANO_TINY
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
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

1196
/* Read from inn, write to out.  We assume inn is opened for reading,
1197
1198
 * and out for writing.  We return 0 on success, -1 on read error, or -2
 * on write error. */
1199
1200
int copy_file(FILE *inn, FILE *out)
{
1201
    char buf[BUFSIZ];
1202
1203
1204
1205
    size_t charsread;
    int retval = 0;

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

1207
1208
1209
1210
1211
1212
1213
1214
1215
1216
1217
    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
1218

1219
1220
1221
1222
    if (fclose(inn) == EOF)
	retval = -1;
    if (fclose(out) == EOF)
	retval = -2;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1223

1224
1225
1226
    return retval;
}

1227
1228
1229
/* Write a file out.  If f_open isn't NULL, we assume that it is a
 * stream associated with the file, and we don't try to open it
 * ourselves.  If tmp is TRUE, we set the umask to disallow anyone else
1230
1231
 * 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
1232
 *
1233
1234
 * tmp means we are writing a temporary file in a secure fashion.  We
 * use it when spell checking or dumping the file on an error.
1235
 *
1236
1237
 * append == APPEND means we are appending instead of overwriting.
 * append == PREPEND means we are prepending instead of overwriting.
1238
 *
1239
1240
 * nonamechange means don't change the current filename.  It is ignored
 * if tmp is FALSE or if we're appending/prepending.
1241
 *
1242
 * Return 0 on success or -1 on error. */
1243
1244
int write_file(const char *name, FILE *f_open, bool tmp, append_type
	append, bool nonamechange)
Chris Allegretta's avatar
Chris Allegretta committed
1245
{
1246
1247
    int retval = -1;
	/* Instead of returning in this function, you should always
1248
1249
	 * merely set retval and then goto cleanup_and_exit. */
    size_t lineswritten = 0;
1250
    const filestruct *fileptr = openfile->fileage;
1251
    int fd;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1252
	/* The file descriptor we use. */
1253
    mode_t original_umask = 0;
1254
	/* Our umask, from when nano started. */
1255
    bool realexists;
1256
	/* The result of stat().  TRUE if the file exists, FALSE
1257
	 * otherwise.  If name is a link that points nowhere, realexists
1258
	 * is FALSE. */
1259
1260
    struct stat st;
	/* The status fields filled in by stat(). */
1261
    bool anyexists;
1262
1263
	/* The result of lstat().  The same as realexists, unless name
	 * is a link. */
1264
1265
1266
    struct stat lst;
	/* The status fields filled in by lstat(). */
    char *realname;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1267
	/* name after tilde expansion. */
1268
    FILE *f = NULL;
1269
1270
1271
	/* 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
1272

1273
    assert(name != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1274

1275
    if (name[0] == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1276
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1277

1278
1279
1280
    if (f_open != NULL)
	f = f_open;

1281
1282
    if (!tmp)
	titlebar(NULL);
1283

1284
    realname = real_dir_from_tilde(name);
Chris Allegretta's avatar
Chris Allegretta committed
1285

1286
#ifndef DISABLE_OPERATINGDIR
1287
    /* If we're writing a temporary file, we're probably going outside
1288
     * the operating directory, so skip the operating directory test. */
1289
    if (!tmp && check_operating_dir(realname, FALSE)) {
1290
1291
	statusbar(_("Can't write outside of %s"), operating_dir);
	goto cleanup_and_exit;
1292
1293
1294
    }
#endif

1295
    anyexists = (lstat(realname, &lst) != -1);
1296

1297
1298
    /* If the temp file exists and isn't already open, give up. */
    if (tmp && anyexists && f_open == NULL)
1299
	goto cleanup_and_exit;
1300

1301
1302
1303
    /* 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)) {
1304
1305
	statusbar(
		_("Cannot prepend or append to a symlink with --nofollow set"));
1306
1307
1308
	goto cleanup_and_exit;
    }

1309
    /* Save the state of file at the end of the symlink (if there is
1310
     * one). */
1311
    realexists = (stat(realname, &st) != -1);
1312

1313
#ifndef NANO_TINY
1314
    /* We backup only if the backup toggle is set, the file isn't
1315
1316
1317
1318
     * 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. */
1319
1320
1321
    if (ISSET(BACKUP_FILE) && !tmp && realexists && ((append !=
	OVERWRITE || openfile->mark_set) ||
	openfile->current_stat->st_mtime == st.st_mtime)) {
1322
	FILE *backup_file;
1323
	char *backupname;
1324
	struct utimbuf filetime;
1325
	int copy_status;
1326

1327
	/* Save the original file's access and modification times. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1328
1329
	filetime.actime = openfile->current_stat->st_atime;
	filetime.modtime = openfile->current_stat->st_mtime;
1330

1331
1332
1333
	if (f_open == NULL) {
	    /* Open the original file to copy to the backup. */
	    f = fopen(realname, "rb");
1334

1335
1336
1337
	    if (f == NULL) {
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1338
		beep();
1339
1340
		goto cleanup_and_exit;
	    }
1341
1342
	}

1343
	/* If backup_dir is set, we set backupname to
1344
1345
1346
1347
	 * 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]. */
1348
	if (backup_dir != NULL) {
1349
	    char *backuptemp = get_full_path(realname);
1350

1351
	    if (backuptemp == NULL)
1352
1353
1354
1355
1356
1357
		/* 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~. */
1358
		backuptemp = mallocstrcpy(NULL, tail(realname));
1359
	    else {
1360
1361
		size_t i = 0;

1362
1363
1364
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1365
1366
1367
1368
		}
	    }

	    backupname = charalloc(strlen(backup_dir) +
1369
1370
1371
1372
1373
1374
1375
1376
1377
1378
1379
1380
1381
1382
1383
		strlen(backuptemp) + 1);
	    sprintf(backupname, "%s%s", backup_dir, backuptemp);
	    free(backuptemp);
	    backuptemp = get_next_filename(backupname, "~");
	    if (backuptemp[0] == '\0') {
		statusbar(_("Error writing %s: %s"), backupname,
		    _("Too many backup files?"));
		free(backuptemp);
		free(backupname);
		fclose(f);
		goto cleanup_and_exit;
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1384
1385
1386
1387
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1388

1389
1390
1391
	/* Open the destination backup file.  Before we write to it, we
	 * set its permissions, so no unauthorized person can read it as
	 * we write. */
1392
	backup_file = fopen(backupname, "wb");
1393

1394
	if (backup_file == NULL || chmod(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1395
		openfile->current_stat->st_mode) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1396
1397
	    statusbar(_("Error writing %s: %s"), backupname,
		strerror(errno));
1398
	    free(backupname);
1399
1400
1401
1402
	    if (backup_file != NULL)
		fclose(backup_file);
	    fclose(f);
	    goto cleanup_and_exit;
1403
1404
1405
	}

#ifdef DEBUG
1406
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1407
1408
#endif

1409
1410
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1411

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1412
	/* And set its metadata. */
1413
	if (copy_status != 0 || chown(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1414
1415
		openfile->current_stat->st_uid,
		openfile->current_stat->st_gid) == -1 ||
1416
		utime(backupname, &filetime) == -1) {
1417
	    if (copy_status == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1418
1419
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1420
1421
		beep();
	    } else
1422
1423
		statusbar(_("Error writing %s: %s"), backupname,
			strerror(errno));
1424
	    free(backupname);
1425
1426
	    goto cleanup_and_exit;
	}
1427

1428
1429
	free(backupname);
    }
1430
#endif /* !NANO_TINY */
1431

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1432
1433
    /* 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
1434
1435
1436
1437
     * overwrite. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode) &&
	unlink(realname) == -1) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1438
	goto cleanup_and_exit;
1439
    }
1440

1441
1442
    if (f_open == NULL) {
	original_umask = umask(0);
1443

1444
	/* If we create a temp file, we don't let anyone else access it.
1445
1446
	 * We create a temp file if tmp is TRUE. */
	if (tmp)
1447
	    umask(S_IRWXG | S_IRWXO);
1448
1449
	else
	    umask(original_umask);
1450
    }
1451

1452
    /* If we're prepending, copy the file to a temp file. */
1453
    if (append == PREPEND) {
1454
1455
1456
	int fd_source;
	FILE *f_source = NULL;

1457
	tempname = safe_tempfile(&f);
1458

1459
1460
	if (tempname == NULL) {
	    statusbar(_("Prepending to %s failed: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1461
		strerror(errno));
1462
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1463
	}
1464

1465
1466
	if (f_open == NULL) {
	    fd_source = open(realname, O_RDONLY | O_CREAT);
1467

1468
1469
1470
1471
1472
	    if (fd_source != -1) {
		f_source = fdopen(fd_source, "rb");
		if (f_source == NULL) {
		    statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1473
		    beep();
1474
1475
1476
1477
1478
1479
		    close(fd_source);
		    fclose(f);
		    unlink(tempname);
		    goto cleanup_and_exit;
		}
	    }
1480
1481
1482
	}

	if (copy_file(f_source, f) != 0) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1483
1484
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
1485
	    unlink(tempname);
1486
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1487
1488
1489
	}
    }

1490
1491
1492
    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
1493
1494
1495
	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);
1496

1497
1498
	/* Set the umask back to the user's original value. */
	umask(original_umask);
1499

1500
1501
1502
1503
	/* If we couldn't open the file, give up. */
	if (fd == -1) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1504

1505
1506
1507
1508
1509
	    /* tempname has been set only if we're prepending. */
	    if (tempname != NULL)
		unlink(tempname);
	    goto cleanup_and_exit;
	}
1510

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

1513
1514
1515
1516
1517
1518
	if (f == NULL) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
	    close(fd);
	    goto cleanup_and_exit;
	}
1519
1520
    }

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

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

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

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

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

1536
	if (size < data_len) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1537
1538
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1539
	    fclose(f);
1540
	    goto cleanup_and_exit;
1541
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1542

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1543
	/* If we're on the last line of the file, don't write a newline
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1544
1545
1546
1547
1548
1549
1550
	 * 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 {
1551
#ifndef NANO_TINY
1552
1553
1554
1555
	    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
1556
			strerror(errno));
1557
1558
1559
		    fclose(f);
		    goto cleanup_and_exit;
		}
1560
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1561

1562
	    if (openfile->fmt != MAC_FILE) {
1563
#endif
1564
1565
		if (putc('\n', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1566
			strerror(errno));
1567
1568
1569
		    fclose(f);
		    goto cleanup_and_exit;
		}
1570
#ifndef NANO_TINY
1571
	    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1572
#endif
1573
	}
Chris Allegretta's avatar
Chris Allegretta committed
1574
1575
1576
1577
1578

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

1579
    /* If we're prepending, open the temp file, and append it to f. */
1580
    if (append == PREPEND) {
1581
1582
1583
1584
	int fd_source;
	FILE *f_source = NULL;

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

1586
1587
1588
1589
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
1590
	}
1591

1592
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1593
1594
	    statusbar(_("Error reading %s: %s"), tempname,
		strerror(errno));
1595
	    beep();
1596
	    fclose(f);
1597
	    goto cleanup_and_exit;
1598
1599
	}

1600
	if (copy_file(f_source, f) == -1 || unlink(tempname) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1601
1602
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1603
	    goto cleanup_and_exit;
1604
	}
1605
1606
1607
1608
    } else if (fclose(f) == EOF) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
	unlink(tempname);
	goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1609
    }
1610

1611
    if (!tmp && append == OVERWRITE) {
Chris Allegretta's avatar
Chris Allegretta committed
1612
	if (!nonamechange) {
1613
1614
	    openfile->filename = mallocstrcpy(openfile->filename,
		realname);
Chris Allegretta's avatar
Chris Allegretta committed
1615
#ifdef ENABLE_COLOR
1616
	    /* We might have changed the filename, so update the colors
1617
1618
	     * to account for it, and then make sure we're using
	     * them. */
1619
	    color_update();
1620
	    color_init();
1621
1622
1623
1624
1625

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

1630
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1631
1632
1633
1634
1635
	/* 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);
1636
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1637

1638
1639
	statusbar(P_("Wrote %lu line", "Wrote %lu lines",
		(unsigned long)lineswritten),
1640
		(unsigned long)lineswritten);
1641
	openfile->modified = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
1642
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
1643
    }
1644

1645
    retval = 0;
1646
1647
1648

  cleanup_and_exit:
    free(realname);
1649
    free(tempname);
1650

1651
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1652
1653
}

1654
#ifndef NANO_TINY
1655
/* Write a marked selection from a file out. */
1656
1657
int write_marked_file(const char *name, FILE *f_open, bool tmp,
	append_type append)
1658
1659
{
    int retval = -1;
1660
1661
    bool old_modified = openfile->modified;
	/* write_file() unsets the modified flag. */
1662
    bool added_magicline = FALSE;
1663
1664
1665
1666
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

1667
1668
    assert(openfile->mark_set);

1669
1670
1671
    /* Partition the filestruct so that it contains only the marked
     * text. */
    mark_order((const filestruct **)&top, &top_x,
1672
	(const filestruct **)&bot, &bot_x, NULL);
1673
    filepart = partition_filestruct(top, top_x, bot, bot_x);
1674

1675
1676
1677
1678
1679
1680
    /* 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')))
1681
	new_magicline();
1682

1683
    retval = write_file(name, f_open, tmp, append, TRUE);
1684

1685
1686
1687
    /* If the NO_NEWLINES flag isn't set, and we added a magicline,
     * remove it now. */
    if (!ISSET(NO_NEWLINES) && added_magicline)
1688
1689
1690
1691
	remove_magicline();

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

1694
    if (old_modified)
1695
1696
1697
1698
	set_modified();

    return retval;
}
1699
#endif /* !NANO_TINY */
1700

1701
1702
1703
1704
/* 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
 * TEMP_FILE flag is set. */
1705
int do_writeout(bool exiting)
Chris Allegretta's avatar
Chris Allegretta committed
1706
{
1707
1708
    int i, retval = 0;
    append_type append = OVERWRITE;
1709
1710
    char *ans;
	/* The last answer the user typed on the statusbar. */
1711
#ifdef NANO_EXTRA
1712
    static bool did_credits = FALSE;
1713
#endif
Chris Allegretta's avatar
Chris Allegretta committed
1714

1715
    currshortcut = writefile_list;
1716

1717
    if (exiting && openfile->filename[0] != '\0' && ISSET(TEMP_FILE)) {
1718
	retval = write_file(openfile->filename, NULL, FALSE, OVERWRITE,
1719
		FALSE);
1720
1721

	/* Write succeeded. */
1722
	if (retval == 0)
1723
	    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1724
1725
    }

1726
    ans = mallocstrcpy(NULL,
1727
#ifndef NANO_TINY
1728
	(openfile->mark_set && !exiting) ? "" :
1729
#endif
1730
	openfile->filename);
1731
1732
1733

    while (TRUE) {
	const char *msg;
1734
#ifndef NANO_TINY
1735
1736
	const char *formatstr, *backupstr;

1737
1738
1739
1740
1741
	formatstr = (openfile->fmt == DOS_FILE) ?
		N_(" [DOS Format]") : (openfile->fmt == MAC_FILE) ?
		N_(" [Mac Format]") : "";

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

1743
	if (openfile->mark_set && !exiting)
1744
1745
1746
	    msg = (append == PREPEND) ?
		N_("Prepend Selection to File") : (append == APPEND) ?
		N_("Append Selection to File") :
1747
		N_("Write Selection to File");
1748
	else
1749
#endif /* !NANO_TINY */
1750
1751
	    msg = (append == PREPEND) ? N_("File Name to Prepend to") :
		(append == APPEND) ? N_("File Name to Append to") :
1752
		N_("File Name to Write");
1753

1754
1755
1756
	/* If we're using restricted mode, the filename isn't blank,
	 * and we're at the "Write File" prompt, disable tab
	 * completion. */
1757
	i = do_prompt(!ISSET(RESTRICTED) ||
1758
1759
1760
1761
1762
		openfile->filename[0] == '\0',
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
		writefile_list, ans,
1763
#ifndef NANO_TINY
1764
1765
1766
1767
1768
		NULL,
#endif
		 edit_refresh, "%s%s%s", _(msg),
#ifndef NANO_TINY
		formatstr, backupstr
1769
#else
1770
		"", ""
1771
1772
1773
#endif
		);

1774
	if (i < 0) {
1775
	    statusbar(_("Cancelled"));
1776
1777
1778
1779
	    retval = -1;
	    break;
	} else {
	    ans = mallocstrcpy(ans, answer);
Chris Allegretta's avatar
Chris Allegretta committed
1780

1781
#ifndef DISABLE_BROWSER
1782
1783
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
1784

1785
1786
		if (tmp == NULL)
		    continue;
1787

1788
1789
1790
1791
1792
		free(answer);
		answer = tmp;

		/* We have a file now.  Get out of the statusbar prompt
		 * cleanly. */
1793
		do_prompt_abort();
1794
	    } else
1795
#endif /* !DISABLE_BROWSER */
1796
#ifndef NANO_TINY
1797
	    if (i == TOGGLE_DOS_KEY) {
1798
1799
		openfile->fmt = (openfile->fmt == DOS_FILE) ? NIX_FILE :
			DOS_FILE;
1800
1801
		continue;
	    } else if (i == TOGGLE_MAC_KEY) {
1802
1803
		openfile->fmt = (openfile->fmt == MAC_FILE) ? NIX_FILE :
			MAC_FILE;
1804
1805
1806
1807
1808
		continue;
	    } else if (i == TOGGLE_BACKUP_KEY) {
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
1809
#endif /* !NANO_TINY */
1810
	    if (i == NANO_PREPEND_KEY) {
1811
		append = (append == PREPEND) ? OVERWRITE : PREPEND;
1812
1813
		continue;
	    } else if (i == NANO_APPEND_KEY) {
1814
		append = (append == APPEND) ? OVERWRITE : APPEND;
1815
1816
		continue;
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1817

Chris Allegretta's avatar
Chris Allegretta committed
1818
#ifdef DEBUG
1819
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
1820
#endif
1821
1822

#ifdef NANO_EXTRA
1823
	    if (exiting && !ISSET(TEMP_FILE) &&
1824
		strcasecmp(answer, "zzy") == 0 && !did_credits) {
1825
		do_credits();
1826
		did_credits = TRUE;
1827
1828
1829
		retval = -1;
		break;
	    }
1830
#endif
1831
	    if (append == OVERWRITE && strcmp(answer,
1832
		openfile->filename) != 0) {
1833
1834
1835
		struct stat st;

		if (!stat(answer, &st)) {
1836
1837
		    i = do_yesno_prompt(FALSE,
			_("File exists, OVERWRITE ? "));
1838
1839
1840
		    if (i == 0 || i == -1)
			continue;
		/* If we're using restricted mode, we aren't allowed to
1841
		 * change the name of a file once it has one, because
1842
1843
1844
		 * that would allow reading from or writing to files not
		 * specified on the command line.  In this case, don't
		 * bother showing the "Different Name" prompt. */
1845
1846
		} else if (!ISSET(RESTRICTED) &&
			openfile->filename[0] != '\0'
1847
#ifndef NANO_TINY
1848
			&& (exiting || !openfile->mark_set)
1849
#endif
1850
			) {
1851
		    i = do_yesno_prompt(FALSE,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1852
			_("Save file under DIFFERENT NAME ? "));
1853
1854
1855
		    if (i == 0 || i == -1)
			continue;
		}
1856
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1857

1858
#ifndef NANO_TINY
1859
1860
1861
1862
	    /* Here's where we allow the selected text to be written to
	     * a separate file.  If we're using restricted mode, this is
	     * disabled since it allows reading from or writing to files
	     * not specified on the command line. */
1863
	    if (!ISSET(RESTRICTED) && !exiting && openfile->mark_set)
1864
		retval = write_marked_file(answer, NULL, FALSE, append);
1865
	    else
1866
#endif /* !NANO_TINY */
1867
		retval = write_file(answer, NULL, FALSE, append, FALSE);
1868

1869
1870
	    break;
	}
1871
    } /* while (TRUE) */
1872
1873

    free(ans);
1874

1875
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1876
1877
}

1878
1879
/* Write the current file to disk.  If the mark is on, write the current
 * marked selection to disk. */
1880
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
1881
{
1882
    do_writeout(FALSE);
1883
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
1884
}
Chris Allegretta's avatar
Chris Allegretta committed
1885

1886
1887
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
1888
char *real_dir_from_tilde(const char *buf)
1889
{
Chris Allegretta's avatar
Chris Allegretta committed
1890
    char *dirtmp = NULL;
1891

1892
    assert(buf != NULL);
1893

1894
    if (buf[0] == '~') {
1895
	size_t i;
1896
	const char *tilde_dir;
1897

1898
	/* Figure out how much of the str we need to compare. */
1899
1900
1901
	for (i = 1; buf[i] != '/' && buf[i] != '\0'; i++)
	    ;

1902
1903
1904
1905
1906
1907
1908
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
	    tilde_dir = homedir;
	} else {
	    const struct passwd *userdata;

1909
1910
1911
	    do {
		userdata = getpwent();
	    } while (userdata != NULL &&
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1912
		strncmp(userdata->pw_name, buf + 1, i - 1) != 0);
1913
1914
	    endpwent();
	    tilde_dir = userdata->pw_dir;
Chris Allegretta's avatar
Chris Allegretta committed
1915
	}
1916

1917
1918
1919
	if (tilde_dir != NULL) {
	    dirtmp = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	    sprintf(dirtmp, "%s%s", tilde_dir, buf + i);
1920
	}
1921
    }
1922

1923
1924
    /* Set a default value for dirtmp, in case the user's home directory
     * isn't found. */
1925
    if (dirtmp == NULL)
1926
	dirtmp = mallocstrcpy(NULL, buf);
1927

1928
    return dirtmp;
1929
1930
}

1931
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
1932
/* Our sort routine for file listings.  Sort alphabetically and
1933
 * case-insensitively, and sort directories before filenames. */
1934
1935
1936
1937
1938
1939
1940
1941
1942
1943
1944
1945
1946
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;

1947
    return mbstrcasecmp(a, b);
1948
}
1949
1950
1951
1952
1953
1954
1955
1956
1957

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

Chris Allegretta's avatar
Chris Allegretta committed
1960
#ifndef DISABLE_TABCOMP
1961
1962
/* Is the given file a directory? */
int is_dir(const char *buf)
1963
{
1964
    char *dirptr = real_dir_from_tilde(buf);
1965
    struct stat fileinfo;
1966
    int retval = (stat(dirptr, &fileinfo) != -1 &&
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1967
	S_ISDIR(fileinfo.st_mode));
1968

1969
    assert(buf != NULL && dirptr != buf);
1970

1971
    free(dirptr);
1972

1973
    return retval;
1974
}
Chris Allegretta's avatar
Chris Allegretta committed
1975

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1976
1977
1978
1979
/* 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
 * file:
Chris Allegretta's avatar
Chris Allegretta committed
1980
1981
1982
1983
1984
1985
1986
1987
1988
1989
1990
1991
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
 * Copyright (c) 1999
 *      Main code:            Adam Rogoyski <rogoyski@cs.utexas.edu>
 *      Etc:                  Dave Cinege <dcinege@psychosis.com>
 *  Majorly adjusted/re-written for busybox:
 *                            Erik Andersen <andersee@debian.org>
 *
 * You may use this code as you wish, so long as the original author(s)
 * are attributed in any redistributions of the source code.
 * This code is 'as is' with no warranty.
1992
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
1993

1994
1995
1996
1997
/* We consider the first buflen characters of buf for ~username tab
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
	size_t buflen)
Chris Allegretta's avatar
Chris Allegretta committed
1998
{
1999
2000
    char **matches = NULL;
    const struct passwd *userdata;
2001

2002
    assert(buf != NULL && num_matches != NULL && buflen > 0);
2003

2004
    *num_matches = 0;
2005

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

2011
2012
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2013
2014
2015
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
2016
2017
#endif

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2018
2019
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2020
2021
2022
	    matches[*num_matches] =
		charalloc(strlen(userdata->pw_name) + 2);
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
2023
	    ++(*num_matches);
2024
	}
2025
2026
    }
    endpwent();
2027

2028
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2029
2030
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2031
2032
/* We consider the first buflen characters of buf for filename tab
 * completion. */
2033
2034
char **cwd_tab_completion(const char *buf, bool allow_files, size_t
	*num_matches, size_t buflen)
Chris Allegretta's avatar
Chris Allegretta committed
2035
{
2036
    char *dirname = mallocstrcpy(NULL, buf), *filename;
2037
2038
2039
2040
2041
#ifndef DISABLE_OPERATINGDIR
    size_t dirnamelen;
#endif
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2042
    DIR *dir;
2043
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
2044

2045
    assert(dirname != NULL && num_matches != NULL && buflen >= 0);
2046

2047
2048
2049
2050
2051
2052
2053
2054
2055
2056
    *num_matches = 0;
    null_at(&dirname, buflen);

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

	filename = mallocstrcpy(NULL, tmpdirname);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2057
	*tmpdirname = '\0';
2058
2059
2060
	tmpdirname = dirname;
	dirname = real_dir_from_tilde(dirname);
	free(tmpdirname);
Chris Allegretta's avatar
Chris Allegretta committed
2061
    } else {
2062
2063
	filename = dirname;
	dirname = mallocstrcpy(NULL, "./");
Chris Allegretta's avatar
Chris Allegretta committed
2064
2065
    }

2066
    assert(dirname[strlen(dirname) - 1] == '/');
2067

Chris Allegretta's avatar
Chris Allegretta committed
2068
    dir = opendir(dirname);
2069

2070
    if (dir == NULL) {
2071
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
2072
	beep();
2073
2074
2075
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2076
    }
2077
2078
2079
2080
2081
2082

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

2083
    while ((nextdir = readdir(dir)) != NULL) {
2084
2085
	bool skip_match = FALSE;

Chris Allegretta's avatar
Chris Allegretta committed
2086
#ifdef DEBUG
2087
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
2088
#endif
2089
	/* See if this matches. */
2090
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
2091
2092
		(*filename == '.' || (strcmp(nextdir->d_name, ".") !=
		0 && strcmp(nextdir->d_name, "..") != 0))) {
2093
2094
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2095

2096
2097
2098
2099
	    char *tmp = charalloc(strlen(dirname) +
		strlen(nextdir->d_name) + 1);
	    sprintf(tmp, "%s%s", dirname, nextdir->d_name);

2100
2101
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2102
2103
2104
2105
2106
2107
2108
	     * 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. */
2109
	    if (!allow_files && !is_dir(tmp))
2110
2111
2112
		skip_match = TRUE;

	    free(tmp);
2113

2114
	    if (skip_match)
2115
		continue;
2116

2117
2118
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2119
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
2120
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
2121
2122
	}
    }
2123

2124
2125
    closedir(dir);
    free(dirname);
2126
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
2127

2128
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2129
2130
}

2131
/* Do tab completion.  place refers to how much the statusbar cursor
2132
2133
 * position should be advanced.  refresh_func is the function we will
 * call to refresh the edit window. */
2134
char *input_tab(char *buf, bool allow_files, size_t *place, bool
2135
	*lastwastab, void (*refresh_func)(void), bool *list)
Chris Allegretta's avatar
Chris Allegretta committed
2136
{
2137
2138
    size_t num_matches = 0;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2139

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

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2142
    *list = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2143

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

2149
2150
2151
2152
	if (bob == NULL || bob >= buf + *place)
	    matches = username_tab_completion(buf, &num_matches,
		*place);
    }
2153

2154
2155
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
2156
2157
	matches = cwd_tab_completion(buf, allow_files, &num_matches,
		*place);
2158

2159
2160
2161
2162
2163
    if (num_matches <= 0)
	beep();
    else {
	size_t match, common_len = 0;
	char *mzero;
2164
2165
2166
	const char *lastslash = revstrstr(buf, "/", buf + *place);
	size_t lastslash_len = (lastslash == NULL) ? 0 :
		lastslash - buf + 1;
2167
2168
2169
	char *match1_mb = charalloc(mb_cur_max() + 1);
	char *match2_mb = charalloc(mb_cur_max() + 1);
	int match1_mb_len, match2_mb_len;
2170
2171
2172

	while (TRUE) {
	    for (match = 1; match < num_matches; match++) {
2173
2174
		/* Get the number of single-byte characters that all the
		 * matches have in common. */
2175
		match1_mb_len = parse_mbchar(matches[0] + common_len,
2176
			match1_mb, NULL);
2177
		match2_mb_len = parse_mbchar(matches[match] +
2178
			common_len, match2_mb, NULL);
2179
2180
2181
		match1_mb[match1_mb_len] = '\0';
		match2_mb[match2_mb_len] = '\0';
		if (strcmp(match1_mb, match2_mb) != 0)
2182
2183
		    break;
	    }
2184

2185
	    if (match < num_matches || matches[0][common_len] == '\0')
2186
		break;
2187

2188
	    common_len += parse_mbchar(buf + common_len, NULL, NULL);
2189
	}
2190

2191
2192
2193
	free(match1_mb);
	free(match2_mb);

2194
2195
	mzero = charalloc(lastslash_len + common_len + 1);
	sprintf(mzero, "%.*s%.*s", lastslash_len, buf, common_len,
2196
		matches[0]);
2197

2198
	common_len += lastslash_len;
2199

2200
	assert(common_len >= *place);
2201

2202
2203
2204
	if (num_matches == 1 && is_dir(mzero)) {
	    mzero[common_len] = '/';
	    common_len++;
2205

2206
2207
	    assert(common_len > *place);
	}
2208

2209
2210
2211
	if (num_matches > 1 && (common_len != *place ||
		*lastwastab == FALSE))
	    beep();
2212

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2213
	/* If there is more of a match to display on the statusbar, show
2214
	 * it.  We reset lastwastab to FALSE: it requires pressing Tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2215
	 * twice in succession with no statusbar changes to see a match
2216
2217
	 * list. */
	if (common_len != *place) {
2218
	    size_t buf_len = strlen(buf);
2219
2220

	    *lastwastab = FALSE;
2221
	    buf = charealloc(buf, common_len + buf_len - *place + 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2222
2223
	    charmove(buf + common_len, buf + *place, buf_len -
		*place + 1);
2224
	    strncpy(buf, mzero, common_len);
2225
	    *place = common_len;
2226
2227
2228
	} else if (*lastwastab == FALSE || num_matches < 2)
	    *lastwastab = TRUE;
	else {
2229
	    int longest_name = 0, columns, editline = 0;
2230

2231
2232
2233
2234
2235
2236
2237
2238
	    /* 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
2239

2240
2241
		if (common_len > COLS - 1) {
		    longest_name = COLS - 1;
2242
2243
		    break;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2244

2245
2246
		if (common_len > longest_name)
		    longest_name = common_len;
2247
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2248

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

2251
2252
2253
	    /* Each column will be (longest_name + 2) columns wide, i.e,
	     * two spaces between columns, except that there will be
	     * only one space after the last column. */
2254
	    columns = (COLS + 1) / (longest_name + 2);
2255

2256
2257
2258
2259
	    /* Blank the edit window, and print the matches out
	     * there. */
	    blank_edit();
	    wmove(edit, 0, 0);
2260

2261
2262
	    /* Disable el cursor. */
	    curs_set(0);
2263

2264
2265
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2266

2267
2268
		wmove(edit, editline, (longest_name + 2) *
			(match % columns));
2269

2270
2271
2272
		if (match % columns == 0 &&
			editline == editwinrows - 1 &&
			num_matches - match > columns) {
2273
2274
2275
		    waddstr(edit, _("(more)"));
		    break;
		}
2276

2277
2278
2279
2280
		disp = display_string(matches[match], 0, longest_name,
			FALSE);
		waddstr(edit, disp);
		free(disp);
2281

2282
		if ((match + 1) % columns == 0)
2283
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2284
	    }
2285

2286
	    wnoutrefresh(edit);
2287
	    *list = TRUE;
2288
2289
2290
	}

	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2291
2292
    }

2293
    free_chararray(matches, num_matches);
2294

2295
    /* Only refresh the edit window if we don't have a list of filename
2296
     * matches on it. */
2297
    if (*list == FALSE)
2298
	refresh_func();
2299
2300

    /* Enable el cursor. */
2301
    curs_set(1);
2302

2303
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2304
}
2305
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2306

2307
2308
/* Only print the last part of a path.  Isn't there a shell command for
 * this? */
2309
2310
const char *tail(const char *foo)
{
2311
    const char *tmp = strrchr(foo, '/');
2312

2313
2314
2315
    if (tmp == NULL)
	tmp = foo;
    else if (*tmp == '/')
2316
2317
2318
2319
2320
	tmp++;

    return tmp;
}

2321
#if !defined(NANO_TINY) && defined(ENABLE_NANORC)
2322
2323
2324
/* Return $HOME/.nano_history, or NULL if we can't find the homedir.
 * The string is dynamically allocated, and should be freed. */
char *histfilename(void)
2325
{
2326
    char *nanohist = NULL;
2327

2328
2329
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2330

2331
2332
2333
	nanohist = charalloc(homelen + 15);
	strcpy(nanohist, homedir);
	strcpy(nanohist + homelen, "/.nano_history");
Chris Allegretta's avatar
Chris Allegretta committed
2334
    }
2335
2336
2337
    return nanohist;
}

2338
/* Load histories from ~/.nano_history. */
2339
2340
2341
void load_history(void)
{
    char *nanohist = histfilename();
Chris Allegretta's avatar
Chris Allegretta committed
2342

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

2347
	if (hist == NULL) {
2348
	    if (errno != ENOENT) {
2349
2350
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
2351
2352
2353
		rcfile_error(N_("Error reading %s: %s"), nanohist,
			strerror(errno));
		fprintf(stderr,
2354
			_("\nPress Enter to continue starting nano\n"));
2355
2356
		while (getchar() != '\n')
		    ;
2357
	    }
2358
	} else {
2359
2360
2361
2362
	    /* Load a history (first the search history, then the
	     * replace history) from oldest to newest.  Assume the last
	     * history entry is a blank line. */
	    filestruct **history = &search_history;
2363
2364
2365
2366
2367
2368
2369
2370
2371
2372
2373
	    char *line = NULL;
	    size_t buflen = 0;
	    ssize_t read;

	    while ((read = getline(&line, &buflen, hist)) >= 0) {
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
		if (read > 0) {
		    unsunder(line, read);
2374
2375
		    update_history(history, line);
		} else
2376
2377
		    history = &replace_history;
	    }
2378

2379
	    fclose(hist);
2380
	    free(line);
2381
	}
2382
2383
2384
2385
	free(nanohist);
    }
}

2386
2387
2388
/* 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. */
2389
bool writehist(FILE *hist, filestruct *h)
2390
{
2391
    filestruct *p;
2392

2393
2394
2395
    /* Write history from oldest to newest.  Assume the last history
     * entry is a blank line. */
    for (p = h; p != NULL; p = p->next) {
2396
	size_t p_len = strlen(p->data);
2397

2398
	sunder(p->data);
2399

2400
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
2401
2402
		putc('\n', hist) == EOF)
	    return FALSE;
2403
    }
2404

2405
    return TRUE;
2406
2407
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2408
/* Save histories to ~/.nano_history. */
2409
2410
void save_history(void)
{
2411
    char *nanohist;
2412

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2413
    /* Don't save unchanged or empty histories. */
2414
2415
    if (!history_has_changed() || (searchbot->lineno == 1 &&
	replacebot->lineno == 1))
2416
2417
	return;

2418
2419
2420
2421
    nanohist = histfilename();

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

2423
	if (hist == NULL)
2424
2425
	    rcfile_error(N_("Error writing %s: %s"), nanohist,
		strerror(errno));
2426
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2427
2428
	    /* Make sure no one else can read from or write to the
	     * history file. */
2429
	    chmod(nanohist, S_IRUSR | S_IWUSR);
2430

2431
2432
	    if (!writehist(hist, searchage) || !writehist(hist,
		replaceage))
2433
2434
		rcfile_error(N_("Error writing %s: %s"), nanohist,
			strerror(errno));
2435

2436
2437
	    fclose(hist);
	}
2438

2439
2440
2441
	free(nanohist);
    }
}
2442
#endif /* !NANO_TINY && ENABLE_NANORC */