"src/utils.c" did not exist on "cf2f6563ca7da6fa1b8259f743e35c52fdab26ad"
files.c 72.2 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
79
    openfile->undotop = NULL;
    openfile->current_undo = NULL;
80
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
81
#ifdef ENABLE_COLOR
82
83
    openfile->colorstrings = NULL;
#endif
84
}
Chris Allegretta's avatar
Chris Allegretta committed
85

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

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

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

    openfile->totsize = 0;
100
101
}

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

117
118
    assert(filename != NULL);

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

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

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

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

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

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

#ifdef ENABLE_COLOR
164
165
    /* If we're loading into a new buffer, update the colors to account
     * for it, if applicable. */
166
167
168
    if (new_buffer)
	color_update();
#endif
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
#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)
195
	read_file(f, filename, FALSE);
196
197
198
199
200
201
202
203

    /* 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 */

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

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

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

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

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

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

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

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

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

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

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

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

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

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

278
    /* Switch to the next file buffer. */
279
    switch_to_next_buffer_void();
280

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

284
    display_main_list();
285

286
    return TRUE;
287
}
288
#endif /* ENABLE_MULTIBUFFER */
289

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

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

303
    assert(openfile->fileage != NULL && strlen(buf) == buf_len);
304

305
    fileptr->data = mallocstrcpy(NULL, buf);
306

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

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

	fileptr->prev = prevnode;
	fileptr->next = NULL;
	fileptr->lineno = prevnode->lineno + 1;
	prevnode->next = fileptr;
336
337
    }

338
339
    return fileptr;
}
340

341
/* Read an open file into the current buffer.  f should be set to the
342
343
344
 * open file, and filename should be set to the name of the file.
   undoable  means do we want to create undo records to try and undo this */
void read_file(FILE *f, const char *filename, bool undoable)
345
346
347
348
349
350
351
352
353
354
355
356
357
{
    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. */
358
    filestruct *fileptr = openfile->current;
359
360
361
362
363
364
	/* 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. */
365
#ifndef NANO_TINY
366
367
    int format = 0;
	/* 0 = *nix, 1 = DOS, 2 = Mac, 3 = both DOS and Mac. */
368
#endif
369

370
371
    assert(openfile->fileage != NULL && openfile->current != NULL);

372
373
    buf = charalloc(bufx);
    buf[0] = '\0';
374

375
376
377
378
379
#ifndef NANO_TINY
    if (undoable)
	add_undo(INSERT);
#endif

380
381
382
383
    if (openfile->current == openfile->fileage)
	first_line_ins = TRUE;
    else
	fileptr = openfile->current->prev;
384

385
    /* Read the entire file into the filestruct. */
386
387
388
389
390
391
    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') {
392
#ifndef NANO_TINY
393
394
395
396
397
398
399
400
401
	    /* 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++;
	    }
402
#endif
403

404
405
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
406

407
408
409
	    /* Reset the line length in preparation for the next
	     * line. */
	    len = 0;
Chris Allegretta's avatar
Chris Allegretta committed
410

411
412
413
	    num_lines++;
	    buf[0] = '\0';
	    i = 0;
414
#ifndef NANO_TINY
415
416
417
418
419
	/* 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') {
420
421
422
423
424
	    /* 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;
425

426
427
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
428

429
430
431
432
	    /* 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
433

434
435
436
437
	    num_lines++;
	    buf[0] = input;
	    buf[1] = '\0';
	    i = 1;
438
#endif
439
440
441
442
	} else {
	    /* Calculate the total length of the line.  It might have
	     * nulls in it, so we can't just use strlen() here. */
	    len++;
443

444
445
446
447
448
449
450
451
	    /* 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);
452
	    }
453
454
455
456
457
458
459
460
461
462
463

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

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

465
#ifndef NANO_TINY
466
467
468
469
470
471
472
473
    /* 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';
    }
474
#endif
475

476
477
    /* Did we not get a newline and still have stuff to do? */
    if (len > 0) {
478
#ifndef NANO_TINY
479
480
481
482
483
484
485
	/* 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;
486
487
#endif

488
489
490
491
	/* Read in the last line properly. */
	fileptr = read_line(buf, fileptr, &first_line_ins, len);
	num_lines++;
    }
492

493
    free(buf);
494

495
    /* If we didn't get a file and we don't already have one, open a
496
     * blank buffer. */
497
    if (fileptr == NULL)
498
	open_buffer("", FALSE);
499

500
501
    /* Attach the file we got to the filestruct.  If we got a file of
     * zero bytes, don't do anything. */
502
    if (num_lines > 0) {
503
504
505
506
507
508
509
510
511
512
513
514
	/* 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
515
516
	    /* Tack the text at fileptr onto the beginning of the text
	     * at current. */
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
	    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
532
	    /* Move fileptr back one line and blow away the old fileptr,
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
	     * 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. */
549
	renumber(openfile->current);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
550
    }
551

552
553
    openfile->totsize += get_totsize(openfile->fileage,
	openfile->filebot);
554

555
    /* If the NO_NEWLINES flag isn't set, and text has been added to
556
     * the magicline (i.e. a file that doesn't end in a newline has been
557
558
559
     * 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') {
560
561
562
563
564
565
566
567
568
	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();

569
#ifndef NANO_TINY
570
571
572
    if (undoable)
	update_undo(INSERT);

573
574
575
576
577
578
    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) {
579
	openfile->fmt = MAC_FILE;
580
581
582
583
	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) {
584
	openfile->fmt = DOS_FILE;
585
586
587
588
	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
589
#endif
590
591
592
	statusbar(P_("Read %lu line", "Read %lu lines",
		(unsigned long)num_lines), (unsigned long)num_lines);
}
Chris Allegretta's avatar
Chris Allegretta committed
593

594
595
596
597
/* 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".
 *
598
599
600
 * 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. */
601
602
603
int open_file(const char *filename, bool newfie, FILE **f)
{
    struct stat fileinfo;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
604
    int fd;
605
    char *full_filename;
606

607
    assert(filename != NULL && f != NULL);
608

609
610
611
612
613
614
615
    /* 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) {
616
617
618
619
620
	if (newfie) {
	    statusbar(_("New File"));
	    return -2;
	}
	statusbar(_("\"%s\" not found"), filename);
621
	beep();
622
623
	return -1;
    } else if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
624
	S_ISBLK(fileinfo.st_mode)) {
625
626
	/* Don't open directories, character files, or block files.
	 * Sorry, /dev/sndstat! */
627
628
	statusbar(S_ISDIR(fileinfo.st_mode) ?
		_("\"%s\" is a directory") :
629
		_("\"%s\" is a device file"), filename);
630
	beep();
631
	return -1;
632
633
634
    } else if ((fd = open(full_filename, O_RDONLY)) == -1) {
	statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
635
	beep();
636
637
 	return -1;
     } else {
638
	/* The file is A-OK.  Open it. */
639
	*f = fdopen(fd, "rb");
640

641
642
643
	if (*f == NULL) {
	    statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
644
	    beep();
645
646
647
	    close(fd);
	} else
	    statusbar(_("Reading File"));
648
    }
649

650
651
    free(full_filename);

652
    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
653
654
}

655
656
657
658
659
/* 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)
660
{
661
    static int ulmax_digits = -1;
662
663
664
    unsigned long i = 0;
    char *buf;
    size_t namelen, suffixlen;
665

666
    assert(name != NULL && suffix != NULL);
667

668
669
670
    if (ulmax_digits == -1)
	ulmax_digits = digits(ULONG_MAX);

671
672
    namelen = strlen(name);
    suffixlen = strlen(suffix);
Chris Allegretta's avatar
Chris Allegretta committed
673

674
    buf = charalloc(namelen + suffixlen + ulmax_digits + 2);
675
    sprintf(buf, "%s%s", name, suffix);
676

677
678
    while (TRUE) {
	struct stat fs;
Chris Allegretta's avatar
Chris Allegretta committed
679

680
681
682
683
	if (stat(buf, &fs) == -1)
	    return buf;
	if (i == ULONG_MAX)
	    break;
684

685
686
687
	i++;
	sprintf(buf + namelen + suffixlen, ".%lu", i);
    }
Chris Allegretta's avatar
Chris Allegretta committed
688

689
690
691
    /* We get here only if there is no possible save file.  Blank out
     * the filename to indicate this. */
    null_at(&buf, 0);
692

693
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
694
695
}

696
697
698
/* 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. */
699
void do_insertfile(
700
#ifndef NANO_TINY
701
702
703
704
705
706
707
708
709
	bool execute
#else
	void
#endif
	)
{
    int i;
    const char *msg;
    char *ans = mallocstrcpy(NULL, "");
710
	/* The last answer the user typed at the statusbar prompt. */
711
    filestruct *edittop_save = openfile->edittop;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
712
    size_t current_x_save = openfile->current_x;
713
    ssize_t current_y_save = openfile->current_y;
714
715
    bool edittop_inside = FALSE, meta_key = FALSE, func_key = FALSE;
    const sc *s;
716
#ifndef NANO_TINY
717
    bool right_side_up = FALSE, single_line = FALSE;
718
#endif
719

720
721
    currmenu = MINSERTFILE;

722
    while (TRUE) {
723
#ifndef NANO_TINY
724
	if (execute) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
725
	    msg =
726
#ifdef ENABLE_MULTIBUFFER
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
727
		ISSET(MULTIBUFFER) ?
728
		_("Command to execute in new buffer [from %s] ") :
729
#endif
730
		_("Command to execute [from %s] ");
731
732
	} else {
#endif
733
	    msg =
734
#ifdef ENABLE_MULTIBUFFER
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
735
		ISSET(MULTIBUFFER) ?
736
		_("File to insert into new buffer [from %s] ") :
737
#endif
738
		_("File to insert [from %s] ");
739
#ifndef NANO_TINY
740
741
	}
#endif
742

743
	i = do_prompt(TRUE,
744
745
746
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
747
#ifndef NANO_TINY
748
		execute ? MEXTCMD :
749
#endif
750
		MINSERTFILE, ans,
751
		&meta_key, &func_key,
752
#ifndef NANO_TINY
753
		NULL,
754
#endif
755
		edit_refresh, msg,
756
#ifndef DISABLE_OPERATINGDIR
757
758
		operating_dir != NULL && strcmp(operating_dir,
		".") != 0 ? operating_dir :
759
760
#endif
		"./");
761

762
	/* If we're in multibuffer mode and the filename or command is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
763
	 * blank, open a new buffer instead of canceling.  If the
764
765
	 * 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
766
	if (i == -1 || ((i == -2 || *answer == '\n')
767
768
769
770
771
772
773
#ifdef ENABLE_MULTIBUFFER
		&& !ISSET(MULTIBUFFER)
#endif
		)) {
	    statusbar(_("Cancelled"));
	    break;
	} else {
774
	    size_t pww_save = openfile->placewewant;
775

776
	    ans = mallocstrcpy(ans, answer);
777

778
779
	    s = get_shortcut(currmenu, &i, &meta_key, &func_key);

780
#ifndef NANO_TINY
781
#ifdef ENABLE_MULTIBUFFER
782

783
	    if (s && s->scfunc == (void *) new_buffer_msg) {
784
785
786
787
788
		/* Don't allow toggling if we're in view mode. */
		if (!ISSET(VIEW_MODE))
		    TOGGLE(MULTIBUFFER);
		continue;
	    } else
789
#endif
790
	    if (s && s->scfunc == (void *) ext_cmd_msg) {
791
792
793
794
795
796
		execute = !execute;
		continue;
	    }
#ifndef DISABLE_BROWSER
	    else
#endif
797
#endif /* !NANO_TINY */
798

799
#ifndef DISABLE_BROWSER
800
	    if (s && s->scfunc == (void *) to_files_msg) {
801
		char *tmp = do_browse_from(answer);
802

803
804
		if (tmp == NULL)
		    continue;
805

806
		/* We have a file now.  Indicate this. */
807
808
		free(answer);
		answer = tmp;
809

810
811
812
		i = 0;
	    }
#endif
813

814
815
816
817
818
819
820
821
	    /* 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;
822

823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
#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

838
839
840
841
842
#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
843
844
845
		 * looks like a new buffer, and keep track of whether
		 * the top of the edit window is inside the
		 * partition. */
846
847
848
		filepart = partition_filestruct(openfile->current,
			openfile->current_x, openfile->current,
			openfile->current_x);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
849
850
		edittop_inside =
			(openfile->edittop == openfile->fileage);
851
852
853
#ifdef ENABLE_MULTIBUFFER
	    }
#endif
Chris Allegretta's avatar
Chris Allegretta committed
854

855
856
	    /* Convert newlines to nulls, just before we insert the file
	     * or execute the command. */
857
	    sunder(answer);
858
	    align(&answer);
859

860
#ifndef NANO_TINY
861
862
863
	    if (execute) {
#ifdef ENABLE_MULTIBUFFER
		if (ISSET(MULTIBUFFER))
864
		    /* Open a blank buffer. */
865
		    open_buffer("", FALSE);
866
867
868
#endif

		/* Save the command's output in the current buffer. */
869
		execute_command(answer);
870
871
872
873
874
875
876
877
878

#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;
		}
879
#endif
880
881
	    } else {
#endif /* !NANO_TINY */
882
883
		/* Make sure the path to the file specified in answer is
		 * tilde-expanded. */
884
885
		answer = mallocstrassn(answer,
			real_dir_from_tilde(answer));
886
887
888

		/* Save the file specified in answer in the current
		 * buffer. */
889
		open_buffer(answer, TRUE);
890
#ifndef NANO_TINY
891
	    }
Chris Allegretta's avatar
Chris Allegretta committed
892
#endif
893

894
#ifdef ENABLE_MULTIBUFFER
895
896
897
	    if (ISSET(MULTIBUFFER))
		/* Update the screen to account for the current
		 * buffer. */
898
		display_buffer();
899
	    else
Chris Allegretta's avatar
Chris Allegretta committed
900
#endif
901
	    {
902
		filestruct *top_save = openfile->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
903

904
905
906
		/* 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
907
		if (edittop_inside)
908
		    edittop_save = openfile->fileage;
909
910

		/* Update the current x-coordinate to account for the
911
912
913
914
		 * 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. */
915
		openfile->current_x = strlen(openfile->filebot->data);
916
917
918
919
		if (openfile->fileage == openfile->filebot) {
#ifndef NANO_TINY
		    if (openfile->mark_set) {
			openfile->mark_begin = openfile->current;
920
			if (!right_side_up)
921
922
923
924
			    openfile->mark_begin_x +=
				openfile->current_x;
		    }
#endif
925
		    openfile->current_x += current_x_save;
926
		}
927
#ifndef NANO_TINY
928
929
930
931
932
933
934
935
936
937
		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;
		    }
		}
938
#endif
939
940
941

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

944
945
946
947
		/* 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. */
948
		unpartition_filestruct(&filepart);
949

950
951
952
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
953

954
		/* Restore the old edittop. */
955
		openfile->edittop = edittop_save;
956

957
958
959
		/* Restore the old place we want. */
		openfile->placewewant = pww_save;

960
961
		/* Mark the file as modified. */
		set_modified();
962

963
964
		/* Update the screen. */
		edit_refresh();
965
	    }
966

967
968
969
	    break;
	}
    }
970
    shortcut_init(FALSE);
971
972

    free(ans);
973
974
}

975
976
977
/* 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. */
978
void do_insertfile_void(void)
979
{
980
981
#ifdef ENABLE_MULTIBUFFER
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
982
	statusbar(_("Key invalid in non-multibuffer mode"));
983
984
985
    else
#endif
	do_insertfile(
986
#ifndef NANO_TINY
987
988
989
		FALSE
#endif
		);
990
991
992
993

    display_main_list();
}

994
/* When passed "[relative path]" or "[relative path][filename]" in
995
 * origpath, return "[full path]" or "[full path][filename]" on success,
996
997
998
999
 * 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. */
1000
char *get_full_path(const char *origpath)
1001
{
1002
1003
1004
1005
    struct stat fileinfo;
    char *d_here, *d_there, *d_there_file = NULL;
    const char *last_slash;
    bool path_only;
1006

1007
1008
    if (origpath == NULL)
    	return NULL;
1009

1010
    /* Get the current directory.  If it doesn't exist, back up and try
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1011
1012
     * again until we get a directory that does, and use that as the
     * current directory. */
1013
1014
    d_here = charalloc(PATH_MAX + 1);
    d_here = getcwd(d_here, PATH_MAX + 1);
1015

1016
    while (d_here == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1017
1018
1019
	if (chdir("..") == -1)
	    break;

1020
1021
1022
1023
1024
	d_here = getcwd(d_here, PATH_MAX + 1);
    }

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

1027
1028
	/* If the current directory isn't "/", tack a slash onto the end
	 * of it. */
1029
	if (strcmp(d_here, "/") != 0) {
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1030
	    d_here = charealloc(d_here, strlen(d_here) + 2);
1031
1032
	    strcat(d_here, "/");
	}
1033
1034
1035
    /* Otherwise, set d_here to "". */
    } else
	d_here = mallocstrcpy(NULL, "");
1036

1037
    d_there = real_dir_from_tilde(origpath);
1038

1039
1040
1041
    /* 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
1042
1043
    path_only = (stat(d_there, &fileinfo) != -1 &&
	S_ISDIR(fileinfo.st_mode));
1044

1045
1046
1047
    /* If path_only is TRUE, make sure d_there ends in a slash. */
    if (path_only) {
	size_t d_there_len = strlen(d_there);
1048

1049
1050
1051
	if (d_there[d_there_len - 1] != '/') {
	    d_there = charealloc(d_there, d_there_len + 2);
	    strcat(d_there, "/");
1052
	}
1053
    }
1054

1055
1056
    /* Search for the last slash in d_there. */
    last_slash = strrchr(d_there, '/');
1057

1058
1059
1060
1061
    /* 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);
1062

1063
1064
1065
1066
1067
1068
1069
1070
	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);

1071
	/* Remove the filename portion of the answer from d_there. */
1072
1073
	null_at(&d_there, last_slash - d_there + 1);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1074
	/* Go to the path specified in d_there. */
1075
1076
1077
	if (chdir(d_there) == -1) {
	    free(d_there);
	    d_there = NULL;
1078
	} else {
1079
1080
1081
1082
1083
1084
1085
1086
1087
1088
1089
1090
1091
1092
1093
1094
1095
1096
1097
1098
1099
1100
1101
1102
	    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()
1103
	     * error, since we can do nothing if we get one. */
1104
	    chdir(d_here);
1105

1106
1107
	    /* Free d_here, since we're done using it. */
	    free(d_here);
1108
	}
1109
    }
1110

1111
1112
1113
1114
1115
1116
1117
    /* 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) +
1118
		strlen(d_there_file) + 1);
1119
1120
	strcat(d_there, d_there_file);
    }
1121

1122
1123
    /* Free d_there_file, since we're done using it. */
    if (d_there_file != NULL)
1124
1125
	free(d_there_file);

1126
    return d_there;
1127
}
1128

1129
1130
1131
/* 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. */
1132
char *check_writable_directory(const char *path)
1133
{
1134
1135
    char *full_path = get_full_path(path);

1136
    /* If get_full_path() fails, return NULL. */
1137
    if (full_path == NULL)
1138
	return NULL;
1139

1140
1141
1142
    /* 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
1143
	full_path[strlen(full_path) - 1] != '/') {
1144
	free(full_path);
1145
	return NULL;
1146
    }
1147

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1148
    /* Otherwise, return the full path. */
1149
1150
1151
    return full_path;
}

1152
1153
/* This function calls mkstemp(($TMPDIR|P_tmpdir|/tmp/)"nano.XXXXXX").
 * On success, it returns the malloc()ed filename and corresponding FILE
1154
 * stream, opened in "r+b" mode.  On error, it returns NULL for the
1155
1156
 * filename and leaves the FILE stream unchanged. */
char *safe_tempfile(FILE **f)
1157
{
1158
    char *full_tempdir = NULL;
1159
1160
1161
    const char *tmpdir_env;
    int fd;
    mode_t original_umask = 0;
1162

1163
1164
    assert(f != NULL);

1165
1166
1167
    /* 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. */
1168
    tmpdir_env = getenv("TMPDIR");
1169
    if (tmpdir_env != NULL)
1170
	full_tempdir = check_writable_directory(tmpdir_env);
1171

1172
1173
    /* If $TMPDIR is unset, empty, or not a writable directory, and
     * full_tempdir is NULL, try P_tmpdir instead. */
1174
    if (full_tempdir == NULL)
1175
	full_tempdir = check_writable_directory(P_tmpdir);
1176

1177
1178
1179
    /* if P_tmpdir is NULL, use /tmp. */
    if (full_tempdir == NULL)
	full_tempdir = mallocstrcpy(NULL, "/tmp/");
1180

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1181
    full_tempdir = charealloc(full_tempdir, strlen(full_tempdir) + 12);
1182
    strcat(full_tempdir, "nano.XXXXXX");
1183

1184
1185
1186
1187
1188
1189
1190
    original_umask = umask(0);
    umask(S_IRWXG | S_IRWXO);

    fd = mkstemp(full_tempdir);

    if (fd != -1)
	*f = fdopen(fd, "r+b");
1191
1192
1193
    else {
	free(full_tempdir);
	full_tempdir = NULL;
1194
    }
1195

1196
1197
    umask(original_umask);

1198
    return full_tempdir;
1199
}
1200
1201

#ifndef DISABLE_OPERATINGDIR
1202
1203
1204
1205
1206
/* Initialize full_operating_dir based on operating_dir. */
void init_operating_dir(void)
{
    assert(full_operating_dir == NULL);

1207
    if (operating_dir == NULL)
1208
	return;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1209

1210
1211
1212
    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
1213
     * inaccessible, unset operating_dir. */
1214
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) {
1215
1216
1217
1218
1219
1220
1221
	free(full_operating_dir);
	full_operating_dir = NULL;
	free(operating_dir);
	operating_dir = NULL;
    }
}

1222
1223
1224
1225
1226
/* 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)
1227
{
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1228
1229
1230
1231
    /* 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. */
1232

1233
    char *fullpath;
1234
    bool retval = FALSE;
1235
    const char *whereami1, *whereami2 = NULL;
1236

1237
    /* If no operating directory is set, don't bother doing anything. */
1238
    if (operating_dir == NULL)
1239
	return FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1240

1241
    assert(full_operating_dir != NULL);
1242
1243

    fullpath = get_full_path(currpath);
1244

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1245
1246
1247
    /* 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
1248
     * non-existent directory as being outside the operating directory,
1249
     * so we return FALSE.  If allow_tabcomp is TRUE, then currpath
1250
1251
     * exists, but is not executable.  So we say it isn't in the
     * operating directory. */
1252
    if (fullpath == NULL)
1253
	return allow_tabcomp;
1254
1255
1256
1257
1258

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

1259
    /* If both searches failed, we're outside the operating directory.
1260
     * Otherwise, check the search results.  If the full operating
1261
1262
1263
     * 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. */
1264
    if (whereami1 != fullpath && whereami2 != full_operating_dir)
1265
1266
	retval = TRUE;
    free(fullpath);
1267
1268

    /* Otherwise, we're still inside it. */
1269
    return retval;
1270
}
1271
1272
#endif

1273
#ifndef NANO_TINY
1274
1275
1276
1277
1278
1279
1280
1281
1282
1283
1284
1285
1286
1287
1288
1289
1290
1291
1292
1293
1294
1295
1296
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

1297
/* Read from inn, write to out.  We assume inn is opened for reading,
1298
1299
 * and out for writing.  We return 0 on success, -1 on read error, or -2
 * on write error. */
1300
1301
int copy_file(FILE *inn, FILE *out)
{
1302
    int retval = 0;
1303
    char buf[BUFSIZ];
1304
1305
    size_t charsread;

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

1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318
    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
1319

1320
1321
1322
1323
    if (fclose(inn) == EOF)
	retval = -1;
    if (fclose(out) == EOF)
	retval = -2;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1324

1325
1326
1327
    return retval;
}

1328
1329
/* 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
1330
 * ourselves.  If tmp is TRUE, we set the umask to disallow anyone else
1331
1332
 * 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
1333
 *
1334
 * tmp means we are writing a temporary file in a secure fashion.  We
1335
1336
1337
1338
 * 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
1339
1340
 * filename.  nonamechange is ignored if tmp is FALSE, we're appending,
 * or we're prepending.
1341
 *
1342
1343
 * Return TRUE on success or FALSE on error. */
bool write_file(const char *name, FILE *f_open, bool tmp, append_type
1344
	append, bool nonamechange)
Chris Allegretta's avatar
Chris Allegretta committed
1345
{
1346
    bool retval = FALSE;
1347
	/* Instead of returning in this function, you should always
1348
	 * set retval and then goto cleanup_and_exit. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1349
    size_t lineswritten = 0;
1350
    const filestruct *fileptr = openfile->fileage;
1351
    int fd;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1352
	/* The file descriptor we use. */
1353
    mode_t original_umask = 0;
1354
	/* Our umask, from when nano started. */
1355
    bool realexists;
1356
	/* The result of stat().  TRUE if the file exists, FALSE
1357
	 * otherwise.  If name is a link that points nowhere, realexists
1358
	 * is FALSE. */
1359
1360
    struct stat st;
	/* The status fields filled in by stat(). */
1361
    bool anyexists;
1362
1363
	/* The result of lstat().  The same as realexists, unless name
	 * is a link. */
1364
1365
1366
    struct stat lst;
	/* The status fields filled in by lstat(). */
    char *realname;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1367
	/* name after tilde expansion. */
1368
    FILE *f = NULL;
1369
1370
1371
	/* 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
1372

1373
    assert(name != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1374

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1375
    if (*name == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1376
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1377

1378
1379
1380
    if (f_open != NULL)
	f = f_open;

1381
1382
    if (!tmp)
	titlebar(NULL);
1383

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1384
    realname = real_dir_from_tilde(name);
1385

1386
#ifndef DISABLE_OPERATINGDIR
1387
    /* If we're writing a temporary file, we're probably going outside
1388
     * the operating directory, so skip the operating directory test. */
1389
    if (!tmp && check_operating_dir(realname, FALSE)) {
1390
1391
	statusbar(_("Can't write outside of %s"), operating_dir);
	goto cleanup_and_exit;
1392
1393
1394
    }
#endif

1395
    anyexists = (lstat(realname, &lst) != -1);
1396

1397
1398
    /* If the temp file exists and isn't already open, give up. */
    if (tmp && anyexists && f_open == NULL)
1399
	goto cleanup_and_exit;
1400

1401
1402
1403
    /* 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)) {
1404
1405
	statusbar(
		_("Cannot prepend or append to a symlink with --nofollow set"));
1406
1407
1408
	goto cleanup_and_exit;
    }

1409
    /* Save the state of the file at the end of the symlink (if there is
1410
     * one). */
1411
    realexists = (stat(realname, &st) != -1);
1412

1413
#ifndef NANO_TINY
1414
    /* We backup only if the backup toggle is set, the file isn't
1415
1416
1417
1418
     * 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. */
1419
1420
1421
    if (ISSET(BACKUP_FILE) && !tmp && realexists && ((append !=
	OVERWRITE || openfile->mark_set) ||
	openfile->current_stat->st_mtime == st.st_mtime)) {
1422
	FILE *backup_file;
1423
	char *backupname;
1424
	struct utimbuf filetime;
1425
	int copy_status;
1426

1427
	/* Save the original file's access and modification times. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1428
1429
	filetime.actime = openfile->current_stat->st_atime;
	filetime.modtime = openfile->current_stat->st_mtime;
1430

1431
1432
1433
	if (f_open == NULL) {
	    /* Open the original file to copy to the backup. */
	    f = fopen(realname, "rb");
1434

1435
1436
1437
	    if (f == NULL) {
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1438
		beep();
1439
1440
1441
1442
		/* 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;
1443
	    }
1444
1445
	}

1446
	/* If backup_dir is set, we set backupname to
1447
1448
1449
1450
	 * 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]. */
1451
	if (backup_dir != NULL) {
1452
	    char *backuptemp = get_full_path(realname);
1453

1454
	    if (backuptemp == NULL)
1455
1456
1457
1458
1459
1460
		/* 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~. */
1461
		backuptemp = mallocstrcpy(NULL, tail(realname));
1462
	    else {
1463
1464
		size_t i = 0;

1465
1466
1467
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1468
1469
1470
1471
		}
	    }

	    backupname = charalloc(strlen(backup_dir) +
1472
1473
1474
1475
		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
1476
	    if (*backuptemp == '\0') {
1477
1478
1479
1480
		statusbar(_("Error writing %s: %s"), backupname,
		    _("Too many backup files?"));
		free(backuptemp);
		free(backupname);
1481
1482
1483
1484
		/* If we can't write to the backup, go on, since only
		 * saving the original file is better than saving
		 * nothing. */
		goto skip_backup;
1485
1486
1487
1488
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1489
1490
1491
1492
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1493

1494
1495
1496
	/* Open the destination backup file.  Before we write to it, we
	 * set its permissions, so no unauthorized person can read it as
	 * we write. */
1497
	backup_file = fopen(backupname, "wb");
1498

1499
	if (backup_file == NULL || chmod(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1500
		openfile->current_stat->st_mode) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1501
1502
	    statusbar(_("Error writing %s: %s"), backupname,
		strerror(errno));
1503
	    free(backupname);
1504
1505
	    if (backup_file != NULL)
		fclose(backup_file);
1506
1507
1508
	    /* If we can't write to the backup, go on, since only saving
	     * the original file is better than saving nothing. */
	    goto skip_backup;
1509
1510
1511
	}

#ifdef DEBUG
1512
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1513
1514
#endif

1515
1516
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1517

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1518
	/* And set its metadata. */
1519
	if (copy_status != 0 || chown(backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1520
1521
		openfile->current_stat->st_uid,
		openfile->current_stat->st_gid) == -1 ||
1522
		utime(backupname, &filetime) == -1) {
1523
	    if (copy_status == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1524
1525
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1526
1527
		beep();
	    } else
1528
1529
		statusbar(_("Error writing %s: %s"), backupname,
			strerror(errno));
1530
1531
1532
	    /* If we can't read from or write to the backup, go on,
	     * since only saving the original file is better than saving
	     * nothing. */
1533
	}
1534

1535
1536
	free(backupname);
    }
1537
1538

  skip_backup:
1539
#endif /* !NANO_TINY */
1540

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1541
1542
    /* 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
1543
1544
1545
1546
     * overwrite. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode) &&
	unlink(realname) == -1) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1547
	goto cleanup_and_exit;
1548
    }
1549

1550
1551
    if (f_open == NULL) {
	original_umask = umask(0);
1552

1553
	/* If we create a temp file, we don't let anyone else access it.
1554
1555
	 * We create a temp file if tmp is TRUE. */
	if (tmp)
1556
	    umask(S_IRWXG | S_IRWXO);
1557
1558
	else
	    umask(original_umask);
1559
    }
1560

1561
    /* If we're prepending, copy the file to a temp file. */
1562
    if (append == PREPEND) {
1563
1564
1565
	int fd_source;
	FILE *f_source = NULL;

1566
1567
1568
1569
1570
1571
1572
1573
1574
1575
1576
	if (f == NULL) {
	    f = fopen(realname, "rb");

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

1577
	tempname = safe_tempfile(&f);
1578

1579
	if (tempname == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1580
	    statusbar(_("Error writing temp file: %s"),
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1581
		strerror(errno));
1582
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1583
	}
1584

1585
1586
	if (f_open == NULL) {
	    fd_source = open(realname, O_RDONLY | O_CREAT);
1587

1588
1589
1590
1591
1592
	    if (fd_source != -1) {
		f_source = fdopen(fd_source, "rb");
		if (f_source == NULL) {
		    statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1593
		    beep();
1594
1595
1596
1597
1598
1599
		    close(fd_source);
		    fclose(f);
		    unlink(tempname);
		    goto cleanup_and_exit;
		}
	    }
1600
1601
1602
	}

	if (copy_file(f_source, f) != 0) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1603
1604
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
1605
	    unlink(tempname);
1606
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1607
1608
1609
	}
    }

1610
1611
1612
    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
1613
1614
1615
	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);
1616

1617
1618
	/* Set the umask back to the user's original value. */
	umask(original_umask);
1619

1620
1621
1622
1623
	/* If we couldn't open the file, give up. */
	if (fd == -1) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1624

1625
1626
1627
1628
1629
	    /* tempname has been set only if we're prepending. */
	    if (tempname != NULL)
		unlink(tempname);
	    goto cleanup_and_exit;
	}
1630

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

1633
1634
1635
1636
1637
1638
	if (f == NULL) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
	    close(fd);
	    goto cleanup_and_exit;
	}
1639
1640
    }

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

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

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

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

1653
1654
	/* Convert nulls to newlines.  data_len is the string's real
	 * length. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1655
1656
	unsunder(fileptr->data, data_len);

1657
	if (size < data_len) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1658
1659
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1660
	    fclose(f);
1661
	    goto cleanup_and_exit;
1662
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1663

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1664
	/* If we're on the last line of the file, don't write a newline
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1665
1666
1667
1668
1669
1670
1671
	 * 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 {
1672
#ifndef NANO_TINY
1673
1674
1675
1676
	    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
1677
			strerror(errno));
1678
1679
1680
		    fclose(f);
		    goto cleanup_and_exit;
		}
1681
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1682

1683
	    if (openfile->fmt != MAC_FILE) {
1684
#endif
1685
1686
		if (putc('\n', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1687
			strerror(errno));
1688
1689
1690
		    fclose(f);
		    goto cleanup_and_exit;
		}
1691
#ifndef NANO_TINY
1692
	    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1693
#endif
1694
	}
Chris Allegretta's avatar
Chris Allegretta committed
1695
1696
1697
1698
1699

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

1700
    /* If we're prepending, open the temp file, and append it to f. */
1701
    if (append == PREPEND) {
1702
1703
1704
1705
	int fd_source;
	FILE *f_source = NULL;

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

1707
1708
1709
1710
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
1711
	}
1712

1713
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1714
1715
	    statusbar(_("Error reading %s: %s"), tempname,
		strerror(errno));
1716
	    beep();
1717
	    fclose(f);
1718
	    goto cleanup_and_exit;
1719
1720
	}

1721
	if (copy_file(f_source, f) == -1 || unlink(tempname) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1722
1723
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1724
	    goto cleanup_and_exit;
1725
	}
1726
1727
    } else
	fclose(f);
1728

1729
    if (!tmp && append == OVERWRITE) {
Chris Allegretta's avatar
Chris Allegretta committed
1730
	if (!nonamechange) {
1731
1732
	    openfile->filename = mallocstrcpy(openfile->filename,
		realname);
Chris Allegretta's avatar
Chris Allegretta committed
1733
#ifdef ENABLE_COLOR
1734
	    /* We might have changed the filename, so update the colors
1735
1736
	     * to account for it, and then make sure we're using
	     * them. */
1737
	    color_update();
1738
	    color_init();
1739
1740
1741
1742
1743

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

1748
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1749
1750
1751
1752
1753
	/* 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);
1754
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1755

1756
1757
	statusbar(P_("Wrote %lu line", "Wrote %lu lines",
		(unsigned long)lineswritten),
1758
		(unsigned long)lineswritten);
1759
	openfile->modified = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
1760
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
1761
    }
1762

1763
    retval = TRUE;
1764
1765
1766

  cleanup_and_exit:
    free(realname);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1767
1768
    if (tempname != NULL)
	free(tempname);
1769

1770
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1771
1772
}

1773
#ifndef NANO_TINY
1774
1775
/* Write a marked selection from a file out to disk.  Return TRUE on
 * success or FALSE on error. */
1776
bool write_marked_file(const char *name, FILE *f_open, bool tmp,
1777
	append_type append)
1778
{
1779
    bool retval;
1780
1781
    bool old_modified = openfile->modified;
	/* write_file() unsets the modified flag. */
1782
    bool added_magicline = FALSE;
1783
1784
1785
1786
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

1787
1788
    assert(openfile->mark_set);

1789
1790
1791
    /* Partition the filestruct so that it contains only the marked
     * text. */
    mark_order((const filestruct **)&top, &top_x,
1792
	(const filestruct **)&bot, &bot_x, NULL);
1793
    filepart = partition_filestruct(top, top_x, bot, bot_x);
1794

1795
1796
1797
1798
1799
1800
    /* 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')))
1801
	new_magicline();
1802

1803
    retval = write_file(name, f_open, tmp, append, TRUE);
1804

1805
1806
1807
    /* If the NO_NEWLINES flag isn't set, and we added a magicline,
     * remove it now. */
    if (!ISSET(NO_NEWLINES) && added_magicline)
1808
1809
1810
1811
	remove_magicline();

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

1814
    if (old_modified)
1815
1816
1817
1818
	set_modified();

    return retval;
}
1819
#endif /* !NANO_TINY */
1820

1821
1822
1823
/* 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
1824
1825
 * 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
1826
{
1827
    int i;
1828
    append_type append = OVERWRITE;
1829
    char *ans;
1830
	/* The last answer the user typed at the statusbar prompt. */
1831
#ifdef NANO_EXTRA
1832
    static bool did_credits = FALSE;
1833
#endif
1834
1835
    bool retval = FALSE, meta_key = FALSE, func_key = FALSE;
    const sc *s;
Chris Allegretta's avatar
Chris Allegretta committed
1836

1837
    currmenu = MWRITEFILE;
1838

1839
    if (exiting && openfile->filename[0] != '\0' && ISSET(TEMP_FILE)) {
1840
	retval = write_file(openfile->filename, NULL, FALSE, OVERWRITE,
1841
		FALSE);
1842
1843

	/* Write succeeded. */
1844
	if (retval)
1845
	    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1846
1847
    }

1848
    ans = mallocstrcpy(NULL,
1849
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1850
	(!exiting && openfile->mark_set) ? "" :
1851
#endif
1852
	openfile->filename);
1853
1854
1855

    while (TRUE) {
	const char *msg;
1856
#ifndef NANO_TINY
1857
1858
	const char *formatstr, *backupstr;

1859
	formatstr = (openfile->fmt == DOS_FILE) ?
1860
1861
		_(" [DOS Format]") : (openfile->fmt == MAC_FILE) ?
		_(" [Mac Format]") : "";
1862

1863
	backupstr = ISSET(BACKUP_FILE) ? _(" [Backup]") : "";
1864

1865
	/* If we're using restricted mode, don't display the "Write
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1866
1867
1868
	 * Selection to File" prompt.  This function is disabled, since
	 * it allows reading from or writing to files not specified on
	 * the command line. */
1869
	if (!ISSET(RESTRICTED) && !exiting && openfile->mark_set)
1870
	    msg = (append == PREPEND) ?
1871
1872
1873
		_("Prepend Selection to File") : (append == APPEND) ?
		_("Append Selection to File") :
		_("Write Selection to File");
1874
	else
1875
#endif /* !NANO_TINY */
1876
1877
1878
	    msg = (append == PREPEND) ? _("File Name to Prepend to") :
		(append == APPEND) ? _("File Name to Append to") :
		_("File Name to Write");
1879

1880
1881
1882
	/* If we're using restricted mode, the filename isn't blank,
	 * and we're at the "Write File" prompt, disable tab
	 * completion. */
1883
	i = do_prompt(!ISSET(RESTRICTED) ||
1884
1885
1886
1887
		openfile->filename[0] == '\0',
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
1888
		MWRITEFILE, ans,
1889
		&meta_key, &func_key,
1890
#ifndef NANO_TINY
1891
1892
		NULL,
#endif
1893
		edit_refresh, "%s%s%s", msg,
1894
1895
#ifndef NANO_TINY
		formatstr, backupstr
1896
#else
1897
		"", ""
1898
1899
1900
#endif
		);

1901
1902
	/* 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
1903
	if (i < 0 || *answer == '\n') {
1904
	    statusbar(_("Cancelled"));
1905
	    retval = FALSE;
1906
1907
1908
	    break;
	} else {
	    ans = mallocstrcpy(ans, answer);
1909
            s = get_shortcut(currmenu, &i, &meta_key, &func_key);
Chris Allegretta's avatar
Chris Allegretta committed
1910

1911
#ifndef DISABLE_BROWSER
1912
	    if (s && s->scfunc == (void *) to_files_msg) {
1913
		char *tmp = do_browse_from(answer);
1914

1915
1916
		if (tmp == NULL)
		    continue;
1917

1918
		/* We have a file now.  Indicate this. */
1919
1920
1921
		free(answer);
		answer = tmp;
	    } else
1922
#endif /* !DISABLE_BROWSER */
1923
#ifndef NANO_TINY
1924
	    if (s && s->scfunc == (void *) dos_format_msg) {
1925
1926
		openfile->fmt = (openfile->fmt == DOS_FILE) ? NIX_FILE :
			DOS_FILE;
1927
		continue;
1928
	    } else if (s && s->scfunc == (void *) mac_format_msg) {
1929
1930
		openfile->fmt = (openfile->fmt == MAC_FILE) ? NIX_FILE :
			MAC_FILE;
1931
		continue;
1932
	    } else if (s && s->scfunc == (void *) backup_file_msg) {
1933
1934
1935
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
1936
#endif /* !NANO_TINY */
1937
	    if (s && s->scfunc == (void *) prepend_msg) {
1938
		append = (append == PREPEND) ? OVERWRITE : PREPEND;
1939
		continue;
1940
	    } else if (s && s->scfunc == (void *) append_msg) {
1941
		append = (append == APPEND) ? OVERWRITE : APPEND;
1942
1943
		continue;
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1944

Chris Allegretta's avatar
Chris Allegretta committed
1945
#ifdef DEBUG
1946
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
1947
#endif
1948
1949

#ifdef NANO_EXTRA
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1950
1951
1952
1953
1954
1955
1956
1957
	    /* 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) {
1958
		do_credits();
1959
		did_credits = TRUE;
1960
		retval = FALSE;
1961
1962
		break;
	    }
1963
#endif
1964

1965
	    if (append == OVERWRITE) {
1966
		size_t answer_len = strlen(answer);
1967
		bool name_exists, do_warning;
1968
		char *full_answer, *full_filename;
1969
1970
		struct stat st;

1971
1972
1973
		/* Convert newlines to nulls, just before we get the
		 * full path. */
		sunder(answer);
1974

1975
1976
		full_answer = get_full_path(answer);
		full_filename = get_full_path(openfile->filename);
1977
1978
		name_exists = (stat((full_answer == NULL) ? answer :
			full_answer, &st) != -1);
1979
1980
1981
1982
1983
1984
		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);
1985

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1986
1987
1988
1989
		/* Convert nulls to newlines.  answer_len is the
		 * string's real length. */
		unsunder(answer, answer_len);

1990
1991
1992
1993
		if (full_filename != NULL)
		    free(full_filename);
		if (full_answer != NULL)
		    free(full_answer);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1994

1995
		if (do_warning) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1996
1997
1998
1999
2000
2001
2002
2003
		    /* 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;
2004

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2005
		    if (name_exists) {
2006
2007
2008
2009
			i = do_yesno_prompt(FALSE,
				_("File exists, OVERWRITE ? "));
			if (i == 0 || i == -1)
			    continue;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2010
		    } else
2011
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2012
		    if (exiting || !openfile->mark_set)
2013
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2014
		    {
2015
2016
2017
2018
2019
			i = do_yesno_prompt(FALSE,
				_("Save file under DIFFERENT NAME ? "));
			if (i == 0 || i == -1)
			    continue;
		    }
2020
		}
2021
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2022

2023
2024
2025
2026
2027
	    /* Convert newlines to nulls, just before we save the
	     * file. */
	    sunder(answer);
	    align(&answer);

2028
	    /* Here's where we allow the selected text to be written to
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2029
2030
2031
	     * 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. */
2032
2033
2034
2035
	    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
2036
#endif
2037
		write_file(answer, NULL, FALSE, append, FALSE);
2038

2039
2040
	    break;
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2041
    }
2042
2043

    free(ans);
2044

2045
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
2046
2047
}

2048
2049
/* Write the current file to disk.  If the mark is on, write the current
 * marked selection to disk. */
2050
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
2051
{
2052
    do_writeout(FALSE);
2053
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
2054
}
Chris Allegretta's avatar
Chris Allegretta committed
2055

2056
2057
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
2058
char *real_dir_from_tilde(const char *buf)
2059
{
2060
    char *retval;
2061

2062
    assert(buf != NULL);
2063

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2064
    if (*buf == '~') {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2065
	size_t i = 1;
2066
	char *tilde_dir;
2067

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

2072
2073
2074
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
2075
	    tilde_dir = mallocstrcpy(NULL, homedir);
2076
2077
2078
	} else {
	    const struct passwd *userdata;

2079
2080
2081
	    tilde_dir = mallocstrncpy(NULL, buf, i + 1);
	    tilde_dir[i] = '\0';

2082
2083
	    do {
		userdata = getpwent();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2084
2085
	    } while (userdata != NULL && strcmp(userdata->pw_name,
		tilde_dir + 1) != 0);
2086
	    endpwent();
2087
	    if (userdata != NULL)
2088
		tilde_dir = mallocstrcpy(tilde_dir, userdata->pw_dir);
Chris Allegretta's avatar
Chris Allegretta committed
2089
	}
2090

2091
2092
	retval = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	sprintf(retval, "%s%s", tilde_dir, buf + i);
2093

2094
2095
2096
	free(tilde_dir);
    } else
	retval = mallocstrcpy(NULL, buf);
2097

2098
    return retval;
2099
2100
}

2101
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
2102
/* Our sort routine for file listings.  Sort alphabetically and
2103
 * case-insensitively, and sort directories before filenames. */
2104
2105
2106
2107
2108
2109
2110
2111
2112
2113
2114
2115
2116
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;

2117
2118
2119
2120
    /* 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, */
2121
    return mbstrcasecmp(a, b);
2122
}
2123
2124
2125
2126
2127

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

2130
2131
2132
2133
    for (; len > 0; len--)
	free(array[len - 1]);
    free(array);
}
2134
2135
#endif

Chris Allegretta's avatar
Chris Allegretta committed
2136
#ifndef DISABLE_TABCOMP
2137
/* Is the given path a directory? */
2138
bool is_dir(const char *buf)
2139
{
2140
    char *dirptr;
2141
    struct stat fileinfo;
2142
2143
2144
    bool retval;

    assert(buf != NULL);
2145

2146
2147
2148
2149
    dirptr = real_dir_from_tilde(buf);

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

2151
    free(dirptr);
2152

2153
    return retval;
2154
}
Chris Allegretta's avatar
Chris Allegretta committed
2155

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2156
2157
2158
/* 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
2159
 * file, with the copyright years updated:
Chris Allegretta's avatar
Chris Allegretta committed
2160
2161
2162
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
2163
 * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007
Chris Allegretta's avatar
Chris Allegretta committed
2164
2165
2166
2167
2168
2169
2170
2171
 *      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.
2172
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
2173

2174
/* We consider the first buf_len characters of buf for ~username tab
2175
2176
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
2177
	size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2178
{
2179
2180
    char **matches = NULL;
    const struct passwd *userdata;
2181

2182
    assert(buf != NULL && num_matches != NULL && buf_len > 0);
2183

2184
    *num_matches = 0;
2185

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

2191
2192
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2193
2194
2195
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
2196
2197
#endif

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2198
2199
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2200
2201
2202
	    matches[*num_matches] =
		charalloc(strlen(userdata->pw_name) + 2);
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
2203
	    ++(*num_matches);
2204
	}
2205
2206
    }
    endpwent();
2207

2208
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2209
2210
}

2211
/* We consider the first buf_len characters of buf for filename tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2212
 * completion. */
2213
char **cwd_tab_completion(const char *buf, bool allow_files, size_t
2214
	*num_matches, size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2215
{
2216
    char *dirname = mallocstrcpy(NULL, buf), *filename;
2217
2218
2219
2220
2221
#ifndef DISABLE_OPERATINGDIR
    size_t dirnamelen;
#endif
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2222
    DIR *dir;
2223
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
2224

2225
    assert(dirname != NULL && num_matches != NULL);
2226

2227
    *num_matches = 0;
2228
    null_at(&dirname, buf_len);
2229
2230
2231
2232
2233
2234
2235
2236

    /* 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
2237
	*tmpdirname = '\0';
2238
2239
2240
	tmpdirname = dirname;
	dirname = real_dir_from_tilde(dirname);
	free(tmpdirname);
Chris Allegretta's avatar
Chris Allegretta committed
2241
    } else {
2242
2243
	filename = dirname;
	dirname = mallocstrcpy(NULL, "./");
Chris Allegretta's avatar
Chris Allegretta committed
2244
2245
    }

2246
    assert(dirname[strlen(dirname) - 1] == '/');
2247

Chris Allegretta's avatar
Chris Allegretta committed
2248
    dir = opendir(dirname);
2249

2250
    if (dir == NULL) {
2251
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
2252
	beep();
2253
2254
2255
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2256
    }
2257
2258
2259
2260
2261
2262

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

2263
    while ((nextdir = readdir(dir)) != NULL) {
2264
2265
	bool skip_match = FALSE;

Chris Allegretta's avatar
Chris Allegretta committed
2266
#ifdef DEBUG
2267
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
2268
#endif
2269
	/* See if this matches. */
2270
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
2271
2272
		(*filename == '.' || (strcmp(nextdir->d_name, ".") !=
		0 && strcmp(nextdir->d_name, "..") != 0))) {
2273
2274
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2275

2276
2277
2278
2279
	    char *tmp = charalloc(strlen(dirname) +
		strlen(nextdir->d_name) + 1);
	    sprintf(tmp, "%s%s", dirname, nextdir->d_name);

2280
2281
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2282
2283
2284
2285
2286
2287
2288
	     * 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. */
2289
	    if (!allow_files && !is_dir(tmp))
2290
2291
2292
		skip_match = TRUE;

	    free(tmp);
2293

2294
	    if (skip_match)
2295
		continue;
2296

2297
2298
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2299
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
2300
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
2301
2302
	}
    }
2303

2304
2305
    closedir(dir);
    free(dirname);
2306
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
2307

2308
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2309
2310
}

2311
/* Do tab completion.  place refers to how much the statusbar cursor
2312
2313
 * position should be advanced.  refresh_func is the function we will
 * call to refresh the edit window. */
2314
char *input_tab(char *buf, bool allow_files, size_t *place, bool
2315
	*lastwastab, void (*refresh_func)(void), bool *list)
Chris Allegretta's avatar
Chris Allegretta committed
2316
{
2317
    size_t num_matches = 0, buf_len;
2318
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2319

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

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2322
    *list = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2323

2324
2325
    /* 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
2326
    if (*place > 0 && *buf == '~') {
2327
	const char *bob = strchr(buf, '/');
Chris Allegretta's avatar
Chris Allegretta committed
2328

2329
2330
2331
2332
	if (bob == NULL || bob >= buf + *place)
	    matches = username_tab_completion(buf, &num_matches,
		*place);
    }
2333

2334
2335
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
2336
2337
	matches = cwd_tab_completion(buf, allow_files, &num_matches,
		*place);
2338

2339
2340
2341
    buf_len = strlen(buf);

    if (num_matches == 0 || *place != buf_len)
2342
2343
2344
2345
	beep();
    else {
	size_t match, common_len = 0;
	char *mzero;
2346
2347
2348
	const char *lastslash = revstrstr(buf, "/", buf + *place);
	size_t lastslash_len = (lastslash == NULL) ? 0 :
		lastslash - buf + 1;
2349
2350
2351
	char *match1_mb = charalloc(mb_cur_max() + 1);
	char *match2_mb = charalloc(mb_cur_max() + 1);
	int match1_mb_len, match2_mb_len;
2352
2353
2354

	while (TRUE) {
	    for (match = 1; match < num_matches; match++) {
2355
2356
		/* Get the number of single-byte characters that all the
		 * matches have in common. */
2357
		match1_mb_len = parse_mbchar(matches[0] + common_len,
2358
			match1_mb, NULL);
2359
		match2_mb_len = parse_mbchar(matches[match] +
2360
			common_len, match2_mb, NULL);
2361
2362
2363
		match1_mb[match1_mb_len] = '\0';
		match2_mb[match2_mb_len] = '\0';
		if (strcmp(match1_mb, match2_mb) != 0)
2364
2365
		    break;
	    }
2366

2367
	    if (match < num_matches || matches[0][common_len] == '\0')
2368
		break;
2369

2370
	    common_len += parse_mbchar(buf + common_len, NULL, NULL);
2371
	}
2372

2373
2374
2375
	free(match1_mb);
	free(match2_mb);

2376
	mzero = charalloc(lastslash_len + common_len + 1);
2377
2378
2379

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

2381
	common_len += lastslash_len;
2382
	mzero[common_len] = '\0';
2383

2384
	assert(common_len >= *place);
2385

2386
	if (num_matches == 1 && is_dir(mzero)) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2387
	    mzero[common_len++] = '/';
2388

2389
2390
	    assert(common_len > *place);
	}
2391

2392
	if (num_matches > 1 && (common_len != *place || !*lastwastab))
2393
	    beep();
2394

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2395
	/* If there is more of a match to display on the statusbar, show
2396
	 * it.  We reset lastwastab to FALSE: it requires pressing Tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2397
	 * twice in succession with no statusbar changes to see a match
2398
2399
2400
	 * list. */
	if (common_len != *place) {
	    *lastwastab = FALSE;
2401
	    buf = charealloc(buf, common_len + buf_len - *place + 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2402
2403
	    charmove(buf + common_len, buf + *place, buf_len -
		*place + 1);
2404
	    strncpy(buf, mzero, common_len);
2405
	    *place = common_len;
2406
	} else if (!*lastwastab || num_matches < 2)
2407
2408
	    *lastwastab = TRUE;
	else {
2409
	    int longest_name = 0, ncols, editline = 0;
2410

2411
2412
2413
2414
2415
2416
2417
2418
	    /* 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
2419

2420
2421
		if (common_len > COLS - 1) {
		    longest_name = COLS - 1;
2422
2423
		    break;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2424

2425
2426
		if (common_len > longest_name)
		    longest_name = common_len;
2427
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2428

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

2431
	    /* Each column will be (longest_name + 2) columns wide, i.e.
2432
2433
	     * two spaces between columns, except that there will be
	     * only one space after the last column. */
2434
	    ncols = (COLS + 1) / (longest_name + 2);
2435

2436
2437
2438
2439
	    /* Blank the edit window, and print the matches out
	     * there. */
	    blank_edit();
	    wmove(edit, 0, 0);
2440

2441
2442
	    /* Disable el cursor. */
	    curs_set(0);
2443

2444
2445
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2446

2447
		wmove(edit, editline, (longest_name + 2) *
2448
			(match % ncols));
2449

2450
		if (match % ncols == 0 &&
2451
			editline == editwinrows - 1 &&
2452
			num_matches - match > ncols) {
2453
2454
2455
		    waddstr(edit, _("(more)"));
		    break;
		}
2456

2457
2458
2459
2460
		disp = display_string(matches[match], 0, longest_name,
			FALSE);
		waddstr(edit, disp);
		free(disp);
2461

2462
		if ((match + 1) % ncols == 0)
2463
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2464
	    }
2465

2466
	    wnoutrefresh(edit);
2467
	    *list = TRUE;
2468
2469
2470
	}

	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2471
2472
    }

2473
    free_chararray(matches, num_matches);
2474

2475
    /* Only refresh the edit window if we don't have a list of filename
2476
     * matches on it. */
2477
    if (!*list)
2478
	refresh_func();
2479
2480

    /* Enable el cursor. */
2481
    curs_set(1);
2482

2483
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2484
}
2485
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2486

2487
2488
/* Only print the last part of a path.  Isn't there a shell command for
 * this? */
2489
2490
const char *tail(const char *foo)
{
2491
    const char *tmp = strrchr(foo, '/');
2492

2493
2494
    if (tmp == NULL)
	tmp = foo;
2495
    else
2496
2497
2498
2499
2500
	tmp++;

    return tmp;
}

2501
#if !defined(NANO_TINY) && defined(ENABLE_NANORC)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2502
2503
2504
/* Return $HOME/.nano_history, or NULL if we can't find the home
 * directory.  The string is dynamically allocated, and should be
 * freed. */
2505
char *histfilename(void)
2506
{
2507
    char *nanohist = NULL;
2508

2509
2510
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2511

2512
2513
2514
	nanohist = charalloc(homelen + 15);
	strcpy(nanohist, homedir);
	strcpy(nanohist + homelen, "/.nano_history");
Chris Allegretta's avatar
Chris Allegretta committed
2515
    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2516

2517
2518
2519
    return nanohist;
}

2520
/* Load histories from ~/.nano_history. */
2521
2522
2523
void load_history(void)
{
    char *nanohist = histfilename();
Chris Allegretta's avatar
Chris Allegretta committed
2524

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

2529
	if (hist == NULL) {
2530
	    if (errno != ENOENT) {
2531
2532
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
2533
2534
2535
		rcfile_error(N_("Error reading %s: %s"), nanohist,
			strerror(errno));
		fprintf(stderr,
2536
			_("\nPress Enter to continue starting nano.\n"));
2537
2538
		while (getchar() != '\n')
		    ;
2539
	    }
2540
	} else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2541
2542
2543
	    /* 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. */
2544
	    filestruct **history = &search_history;
2545
	    char *line = NULL;
2546
	    size_t buf_len = 0;
2547
2548
	    ssize_t read;

2549
	    while ((read = getline(&line, &buf_len, hist)) >= 0) {
2550
2551
2552
2553
2554
2555
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
		if (read > 0) {
		    unsunder(line, read);
2556
2557
		    update_history(history, line);
		} else
2558
2559
		    history = &replace_history;
	    }
2560

2561
	    fclose(hist);
2562
	    free(line);
2563
	}
2564
2565
2566
2567
	free(nanohist);
    }
}

2568
2569
2570
/* 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. */
2571
bool writehist(FILE *hist, filestruct *h)
2572
{
2573
    filestruct *p;
2574

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

2580
	sunder(p->data);
2581

2582
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
2583
2584
		putc('\n', hist) == EOF)
	    return FALSE;
2585
    }
2586

2587
    return TRUE;
2588
2589
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2590
/* Save histories to ~/.nano_history. */
2591
2592
void save_history(void)
{
2593
    char *nanohist;
2594

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2595
    /* Don't save unchanged or empty histories. */
2596
2597
    if (!history_has_changed() || (searchbot->lineno == 1 &&
	replacebot->lineno == 1))
2598
2599
	return;

2600
2601
2602
2603
    nanohist = histfilename();

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

2605
	if (hist == NULL)
2606
2607
	    rcfile_error(N_("Error writing %s: %s"), nanohist,
		strerror(errno));
2608
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2609
2610
	    /* Make sure no one else can read from or write to the
	     * history file. */
2611
	    chmod(nanohist, S_IRUSR | S_IWUSR);
2612

2613
2614
	    if (!writehist(hist, searchage) || !writehist(hist,
		replaceage))
2615
2616
		rcfile_error(N_("Error writing %s: %s"), nanohist,
			strerror(errno));
2617

2618
2619
	    fclose(hist);
	}
2620

2621
2622
2623
	free(nanohist);
    }
}
2624
#endif /* !NANO_TINY && ENABLE_NANORC */