files.c 93.1 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/* $Id$ */
Chris Allegretta's avatar
Chris Allegretta committed
2
3
4
/**************************************************************************
 *   files.c                                                              *
 *                                                                        *
5
 *   Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007,  *
6
 *   2008, 2009, 2010, 2011, 2013, 2014 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>
34
#include <libgen.h>
Chris Allegretta's avatar
Chris Allegretta committed
35

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

54
    /* Start initializing the new buffer. */
55
    openfile->filename = mallocstrcpy(NULL, "");
Chris Allegretta's avatar
Chris Allegretta committed
56

57
    initialize_buffer_text();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
58

59
60
    openfile->current_x = 0;
    openfile->placewewant = 0;
61
    openfile->current_y = 0;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
62

63
    openfile->modified = FALSE;
64
#ifndef NANO_TINY
65
    openfile->mark_set = FALSE;
66
67
68
    openfile->mark_begin = NULL;
    openfile->mark_begin_x = 0;

69
    openfile->fmt = NIX_FILE;
70

71
72
    openfile->undotop = NULL;
    openfile->current_undo = NULL;
73
    openfile->last_action = OTHER;
74
75

    openfile->current_stat = NULL;
76
    openfile->lock_filename = NULL;
77
#endif
78
#ifndef DISABLE_COLOR
79
    openfile->syntax = NULL;
80
81
    openfile->colorstrings = NULL;
#endif
82
}
Chris Allegretta's avatar
Chris Allegretta committed
83

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

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

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

97
#ifndef DISABLE_COLOR
98
99
100
    openfile->fileage->multidata = NULL;
#endif

101
    openfile->totsize = 0;
102
103
}

104
#ifndef NANO_TINY
105
106
107
108
109
110
111
112
113
114
115
/* Actually write the lockfile.  This function will ALWAYS annihilate
 * any previous version of the file.  We'll borrow INSECURE_BACKUP here
 * to decide about lockfile paranoia here as well...
 *
 * Args:
 *     lockfilename: file name for lock
 *     origfilename: name of the file the lock is for
 *     modified: whether to set the modified bit in the file
 *
 * Returns: 1 on success, 0 on failure (but continue loading), -1 on
 * failure and abort. */
116
117
118
119
120
121
122
int write_lockfile(const char *lockfilename, const char *origfilename, bool modified)
{
    int cflags, fd;
    FILE *filestream;
    pid_t mypid;
    uid_t myuid;
    struct passwd *mypwuid;
123
    struct stat fileinfo;
124
125
126
127
128
    char *lockdata = charalloc(1024);
    char myhostname[32];
    ssize_t lockdatalen = 1024;
    ssize_t wroteamt;

129
130
    /* Run things which might fail first before we try and blow away the
     * old state. */
131
132
    myuid = geteuid();
    if ((mypwuid = getpwuid(myuid)) == NULL) {
133
	statusbar(_("Couldn't determine my identity for lock file (getpwuid() failed)"));
Benno Schulenberg's avatar
Benno Schulenberg committed
134
	goto free_and_fail;
135
136
137
138
    }
    mypid = getpid();

    if (gethostname(myhostname, 31) < 0) {
139
140
141
142
143
144
	if (errno == ENAMETOOLONG)
	    myhostname[31] = '\0';
	else {
	    statusbar(_("Couldn't determine hostname for lock file: %s"), strerror(errno));
	    goto free_and_fail;
	}
145
146
    }

147
148
149
    /* Check if the lock exists before we try to delete it...*/
    if (stat(lockfilename, &fileinfo) != -1)
	if (delete_lockfile(lockfilename) < 0)
Benno Schulenberg's avatar
Benno Schulenberg committed
150
	    goto free_and_fail;
151
152

    if (ISSET(INSECURE_BACKUP))
153
	cflags = O_WRONLY | O_CREAT | O_APPEND;
154
    else
155
	cflags = O_WRONLY | O_CREAT | O_EXCL | O_APPEND;
156
157

    fd = open(lockfilename, cflags,
158
		S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
159

160
    /* Maybe we just don't have write access.  Print an error message
161
     * and continue. */
162
    if (fd < 0) {
163
	statusbar(_("Error writing lock file %s: %s"), lockfilename,
164
		    strerror(errno));
Benno Schulenberg's avatar
Benno Schulenberg committed
165
	free(lockdata);
166
	return 0;
167
    }
168

169
170
    /* Now we've got a safe file stream.  If the previous open() call
     * failed, this will return NULL. */
171
172
173
    filestream = fdopen(fd, "wb");

    if (fd < 0 || filestream == NULL) {
174
	statusbar(_("Error writing lock file %s: %s"), lockfilename,
175
		    strerror(errno));
Benno Schulenberg's avatar
Benno Schulenberg committed
176
	goto free_and_fail;
177
178
    }

179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
    /* Okay, so at the moment we're following this state for how to
     * store the lock data:
     *
     * byte 0        - 0x62
     * byte 1        - 0x30
     * bytes 2-12    - program name which created the lock
     * bytes 24,25   - little endian store of creator program's PID
     *                 (b24 = 256^0 column, b25 = 256^1 column)
     * bytes 28-44   - username of who created the lock
     * bytes 68-100  - hostname of where the lock was created
     * bytes 108-876 - filename the lock is for
     * byte 1007     - 0x55 if file is modified
     *
     * Looks like VIM also stores undo state in this file, so we're
     * gonna have to figure out how to slap a 'OMG don't use recover on
     * our lockfile' message in here...
     *
     * This is likely very wrong, so this is a WIP. */
197
    memset(lockdata, 0, lockdatalen);
198
199
200
201
    lockdata[0] = 0x62;
    lockdata[1] = 0x30;
    lockdata[24] = mypid % 256;
    lockdata[25] = mypid / 256;
202
    snprintf(&lockdata[2], 11, "nano %s", VERSION);
203
204
205
    strncpy(&lockdata[28], mypwuid->pw_name, 16);
    strncpy(&lockdata[68], myhostname, 31);
    strncpy(&lockdata[108], origfilename, 768);
206
    if (modified == TRUE)
207
	lockdata[1007] = 0x55;
208
209
210

    wroteamt = fwrite(lockdata, sizeof(char), lockdatalen, filestream);
    if (wroteamt < lockdatalen) {
211
212
	statusbar(_("Error writing lock file %s: %s"),
		lockfilename, ferror(filestream));
Benno Schulenberg's avatar
Benno Schulenberg committed
213
	goto free_and_fail;
214
215
216
    }

#ifdef DEBUG
217
    fprintf(stderr, "In write_lockfile(), write successful (wrote %lu bytes)\n", (unsigned long)wroteamt);
218
#endif
219
220

    if (fclose(filestream) == EOF) {
221
222
	statusbar(_("Error writing lock file %s: %s"),
		lockfilename, strerror(errno));
Benno Schulenberg's avatar
Benno Schulenberg committed
223
	goto free_and_fail;
224
225
    }

226
    openfile->lock_filename = (char *) lockfilename;
227

Benno Schulenberg's avatar
Benno Schulenberg committed
228
    free(lockdata);
229
    return 1;
Benno Schulenberg's avatar
Benno Schulenberg committed
230
231
232
233

  free_and_fail:
    free(lockdata);
    return -1;
234
235
}

236
237
/* Less exciting, delete the lockfile.  Return -1 if unsuccessful and
 * complain on the statusbar, 1 otherwise. */
238
239
240
int delete_lockfile(const char *lockfilename)
{
    if (unlink(lockfilename) < 0 && errno != ENOENT) {
241
242
	statusbar(_("Error deleting lock file %s: %s"), lockfilename,
		  strerror(errno));
243
	return -1;
244
245
246
247
    }
    return 1;
}

248
249
250
251
/* Deal with lockfiles.  Return -1 on refusing to override the lockfile,
 * and 1 on successfully creating it; 0 means we were not successful in
 * creating the lockfile but we should continue to load the file and
 * complain to the user. */
252
253
int do_lockfile(const char *filename)
{
254
255
    char *namecopy1 = (char *) mallocstrcpy(NULL, filename);
    char *namecopy2 = (char *) mallocstrcpy(NULL, filename);
Benno Schulenberg's avatar
Benno Schulenberg committed
256
    size_t locknamesize = strlen(filename) + strlen(locking_prefix)
257
		+ strlen(locking_suffix) + 3;
Benno Schulenberg's avatar
Benno Schulenberg committed
258
    char *lockfilename = charalloc(locknamesize);
259
    char *lockfiledir = NULL;
260
    static char lockprog[11], lockuser[17];
261
262
263
    struct stat fileinfo;
    int lockfd, lockpid;

264
265
266
267
    snprintf(lockfilename, locknamesize, "%s/%s%s%s", dirname(namecopy1),
		locking_prefix, basename(namecopy2), locking_suffix);
    free(namecopy1);
    free(namecopy2);
268
269
#ifdef DEBUG
    fprintf(stderr, "lock file name is %s\n", lockfilename);
270
#endif
271
    if (stat(lockfilename, &fileinfo) != -1) {
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
	ssize_t readtot = 0;
	ssize_t readamt = 0;
	char *lockbuf = charalloc(8192);
	char *promptstr = charalloc(128);
	int ans;
	if ((lockfd = open(lockfilename, O_RDONLY)) < 0) {
	    statusbar(_("Error opening lock file %s: %s"),
			lockfilename, strerror(errno));
	    return -1;
	}
	do {
	    readamt = read(lockfd, &lockbuf[readtot], BUFSIZ);
	    readtot += readamt;
	} while (readtot < 8192 && readamt > 0);

	if (readtot < 48) {
	    statusbar(_("Error reading lock file %s: Not enough data read"),
			lockfilename);
	    return -1;
	}
	strncpy(lockprog, &lockbuf[2], 10);
	lockpid = (unsigned char)lockbuf[25] * 256 + (unsigned char)lockbuf[24];
	strncpy(lockuser, &lockbuf[28], 16);
295
#ifdef DEBUG
296
297
298
	fprintf(stderr, "lockpid = %d\n", lockpid);
	fprintf(stderr, "program name which created this lock file should be %s\n", lockprog);
	fprintf(stderr, "user which created this lock file should be %s\n", lockuser);
299
#endif
300
301
	/* TRANSLATORS: The second %s is the name of the user, the third that of the editor. */
	sprintf(promptstr, _("File %s is being edited (by %s with %s, PID %d); continue?"),
302
303
304
305
306
307
			filename, lockuser, lockprog, lockpid);
	ans = do_yesno_prompt(FALSE, promptstr);
	if (ans < 1) {
	    blank_statusbar();
	    return -1;
	}
308
    } else {
309
310
311
	lockfiledir = mallocstrcpy(NULL, lockfilename);
	lockfiledir = dirname(lockfiledir);
	if (stat(lockfiledir, &fileinfo) == -1) {
312
	    statusbar(_("Error writing lock file: Directory \'%s\' doesn't exist"),
313
		lockfiledir);
314
	    free(lockfiledir);
315
	    return 0;
316
	}
317
	free(lockfiledir);
318
319
320
321
    }

    return write_lockfile(lockfilename, filename, FALSE);
}
322
#endif /* !NANO_TINY */
323

324
325
/* 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. */
326
void open_buffer(const char *filename, bool undoable)
327
{
328
    bool quiet = FALSE;
329
    bool new_buffer = (openfile == NULL
330
#ifndef DISABLE_MULTIBUFFER
331
	 || ISSET(MULTIBUFFER)
332
#endif
333
334
335
336
337
338
	);
	/* 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. */
339

340
341
    assert(filename != NULL);

342
343
344
345
346
347
348
#ifndef DISABLE_OPERATINGDIR
    if (check_operating_dir(filename, FALSE)) {
	statusbar(_("Can't insert file from outside of %s"),
		operating_dir);
	return;
    }
#endif
349

350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
    /* When the specified filename is not empty, and the thing exists,
     * verify that it is a normal file. */
    if (strcmp(filename, "") != 0) {
	struct stat fileinfo;

	if (stat(filename, &fileinfo) == 0 && !S_ISREG(fileinfo.st_mode)) {
	    if (S_ISDIR(fileinfo.st_mode))
		statusbar(_("\"%s\" is a directory"), filename);
	    else
		statusbar(_("\"%s\" is not a normal file"), filename);
	    beep();
	    return;
	}
    }

365
366
    /* If we're going to load into a new buffer, first create the new
     * buffer and lock the corresponding file. */
367
    if (new_buffer) {
368
	make_new_buffer();
369

370
371
372
373
#ifndef NANO_TINY
	if (ISSET(LOCKING) && filename[0] != '\0') {
	    int lockstatus = do_lockfile(filename);
	    if (lockstatus < 0) {
374
#ifndef DISABLE_MULTIBUFFER
375
		if (openfile->next) {
376
		    close_buffer(TRUE);
377
		    return;
378
379
380
381
		}
#endif
	    } else if (lockstatus == 0) {
		quiet = TRUE;
382
	    }
383
	}
384
385
386
#endif
    }

387
388
389
    /* If the filename isn't blank, and we are not in NOREAD_MODE,
     * open the file.  Otherwise, treat it as a new file. */
    rc = (filename[0] != '\0' && !ISSET(NOREAD_MODE)) ?
390
		open_file(filename, new_buffer, quiet, &f) : -2;
391

392
393
    /* If we have a file, and we're loading into a new buffer, update
     * the filename. */
394
395
    if (rc != -1 && new_buffer)
	openfile->filename = mallocstrcpy(openfile->filename, filename);
396

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
397
398
    /* If we have a non-new file, read it in.  Then, if the buffer has
     * no stat, update the stat, if applicable. */
399
    if (rc > 0) {
400
	read_file(f, rc, filename, undoable, new_buffer);
401
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
402
403
404
405
406
	if (openfile->current_stat == NULL) {
	    openfile->current_stat =
		(struct stat *)nmalloc(sizeof(struct stat));
	    stat(filename, openfile->current_stat);
	}
407
#endif
408
409
    }

410
    /* If we have a file, and we're loading into a new buffer, move back
411
412
     * to the beginning of the first line of the buffer. */
    if (rc != -1 && new_buffer) {
413
	openfile->current = openfile->fileage;
414
415
416
	openfile->current_x = 0;
	openfile->placewewant = 0;
    }
417

418
#ifndef DISABLE_COLOR
419
420
    /* If we're loading into a new buffer, update the colors to account
     * for it, if applicable. */
421
422
423
    if (new_buffer)
	color_update();
#endif
424
}
425

426
#ifndef DISABLE_SPELLER
427
428
/* Blow away the text of the current buffer, and then open and read
 * the specified file into its place. */
429
430
431
void replace_buffer(const char *filename)
{
    FILE *f;
432
    int descriptor;
433

434
    assert(filename != NULL && filename[0] != '\0');
435

436
437
    /* Open the file quietly. */
    descriptor = open_file(filename, TRUE, FALSE, &f);
438
439
440
441
442

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

443
444
445
    /* If opening the file succeeded, read it in. */
    if (descriptor > 0)
	read_file(f, descriptor, filename, FALSE, TRUE);
446
447
448

    /* Put current at a place that is certain to exist. */
    openfile->current = openfile->fileage;
449
450
451
}
#endif /* !DISABLE_SPELLER */

452
/* Update the screen to account for the current buffer. */
453
void display_buffer(void)
454
{
455
    /* Update the titlebar, since the filename may have changed. */
456
    titlebar(NULL);
457

458
#ifndef DISABLE_COLOR
459
460
461
    /* Make sure we're using the buffer's associated colors, if
     * applicable. */
    color_init();
462
463
464
#endif

    /* Update the edit window. */
465
    edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
466
467
}

468
#ifndef DISABLE_MULTIBUFFER
469
470
/* Switch to the next file buffer if next_buf is TRUE.  Otherwise,
 * switch to the previous file buffer. */
471
void switch_to_prevnext_buffer(bool next_buf, bool quiet)
Chris Allegretta's avatar
Chris Allegretta committed
472
{
473
    assert(openfile != NULL);
474

475
476
    /* If only one file buffer is open, indicate it on the statusbar and
     * get out. */
477
    if (openfile == openfile->next) {
478
479
	if (quiet == FALSE)
	    statusbar(_("No more open file buffers"));
480
	return;
Chris Allegretta's avatar
Chris Allegretta committed
481
    }
482

483
484
    /* Switch to the next or previous file buffer, depending on the
     * value of next_buf. */
485
    openfile = next_buf ? openfile->next : openfile->prev;
486
487

#ifdef DEBUG
488
    fprintf(stderr, "filename is %s\n", openfile->filename);
489
490
#endif

491
    /* Update the screen to account for the current buffer. */
492
    display_buffer();
493

494
    /* Indicate the switch on the statusbar. */
495
496
    if (quiet == FALSE)
	statusbar(_("Switched to %s"),
497
498
		((openfile->filename[0] == '\0') ?
		_("New Buffer") : openfile->filename));
499
500

#ifdef DEBUG
501
    dump_filestruct(openfile->current);
502
#endif
503
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
504
505
}

506
/* Switch to the previous entry in the openfile filebuffer. */
507
void switch_to_prev_buffer_void(void)
508
{
509
    switch_to_prevnext_buffer(FALSE, FALSE);
510
}
511

512
/* Switch to the next entry in the openfile filebuffer. */
513
void switch_to_next_buffer_void(void)
514
{
515
    switch_to_prevnext_buffer(TRUE, FALSE);
516
}
517

518
/* Delete an entry from the openfile filebuffer, and switch to the one
519
 * after it.  Return TRUE on success, or FALSE if there are no more open
520
521
522
523
 * file buffers.
 * quiet - should we print messages switching bufers
 */
bool close_buffer(bool quiet)
524
{
525
    assert(openfile != NULL);
526

527
    /* If only one file buffer is open, get out. */
528
    if (openfile == openfile->next)
529
	return FALSE;
530

531
#ifndef DISABLE_HISTORIES
532
    update_poshistory(openfile->filename, openfile->current->lineno, xplustabs() + 1);
533
#endif
534

535
    /* Switch to the next file buffer. */
536
    switch_to_prevnext_buffer(TRUE, quiet);
537

538
    /* Close the file buffer we had open before. */
539
    unlink_opennode(openfile->prev);
540

541
542
543
    /* If only one buffer is open now, show Exit in the help lines. */
    if (openfile == openfile->next)
	exitfunc->desc = exit_tag;
544

545
    return TRUE;
546
}
547
#endif /* !DISABLE_MULTIBUFFER */
548

549
550
551
552
553
/* A bit of a copy and paste from open_file(), is_file_writable() just
 * checks whether the file is appendable as a quick permissions check,
 * and we tend to err on the side of permissiveness (reporting TRUE when
 * it might be wrong) to not fluster users editing on odd filesystems by
 * printing incorrect warnings. */
554
555
556
557
558
559
560
561
562
563
564
int is_file_writable(const char *filename)
{
    struct stat fileinfo, fileinfo2;
    int fd;
    FILE *f;
    char *full_filename;
    bool ans = TRUE;

    if (ISSET(VIEW_MODE))
	return TRUE;

565
    assert(filename != NULL);
566
567
568
569

    /* Get the specified file's full path. */
    full_filename = get_full_path(filename);

570
571
    /* Okay, if we can't stat the absolute path due to some component's
     * permissions, just try the relative one. */
572
573
574
    if (full_filename == NULL ||
		(stat(full_filename, &fileinfo) == -1 && stat(filename, &fileinfo2) != -1))
	full_filename = mallocstrcpy(NULL, filename);
575
576

    if ((fd = open(full_filename, O_WRONLY | O_CREAT | O_APPEND, S_IRUSR |
577
578
		S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)) == -1 ||
		(f = fdopen(fd, "a")) == NULL)
579
580
	ans = FALSE;
    else
581
	fclose(f);
582
583
584
585
586
587
    close(fd);

    free(full_filename);
    return ans;
}

588
589
590
591
592
593
/* 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)
594
{
595
    filestruct *fileptr = (filestruct *)nmalloc(sizeof(filestruct));
596

597
    /* Convert nulls to newlines.  buf_len is the string's real length. */
598
    unsunder(buf, buf_len);
599

600
    assert(openfile->fileage != NULL && strlen(buf) == buf_len);
601

602
    fileptr->data = mallocstrcpy(NULL, buf);
603

604
#ifndef NANO_TINY
605
606
607
608
    /* 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';
609
#endif
610

611
#ifndef DISABLE_COLOR
612
    fileptr->multidata = NULL;
613
614
#endif

615
    if (*first_line_ins) {
616
	/* Special case: we're inserting into the first line. */
617
	fileptr->prev = NULL;
618
	fileptr->next = openfile->fileage;
619
	openfile->fileage = fileptr;
620
	fileptr->lineno = 1;
621
	/* Make sure that our edit window stays on the first line. */
622
623
	openfile->edittop = fileptr;
	*first_line_ins = FALSE;
624
625
626
    } else {
	assert(prevnode != NULL);

627
	prevnode->next = fileptr;
628
629
630
	fileptr->prev = prevnode;
	fileptr->next = NULL;
	fileptr->lineno = prevnode->lineno + 1;
631
632
    }

633
634
    return fileptr;
}
635

636
/* Read an open file into the current buffer.  f should be set to the
637
 * open file, and filename should be set to the name of the file.
638
639
640
 * undoable means do we want to create undo records to try and undo
 * this.  Will also attempt to check file writability if fd > 0 and
 * checkwritable == TRUE. */
641
void read_file(FILE *f, int fd, const char *filename, bool undoable, bool checkwritable)
642
643
644
645
646
647
648
649
650
651
652
653
654
{
    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. */
655
    filestruct *fileptr = openfile->current;
656
657
658
659
660
661
	/* 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. */
662
663
    bool writable = TRUE;
	/* Is the file writable (if we care) */
664
#ifndef NANO_TINY
665
666
    int format = 0;
	/* 0 = *nix, 1 = DOS, 2 = Mac, 3 = both DOS and Mac. */
667
#endif
668

669
670
    assert(openfile->fileage != NULL && openfile->current != NULL);

671
672
    buf = charalloc(bufx);
    buf[0] = '\0';
673

674
675
676
677
678
#ifndef NANO_TINY
    if (undoable)
	add_undo(INSERT);
#endif

679
680
681
682
    if (openfile->current == openfile->fileage)
	first_line_ins = TRUE;
    else
	fileptr = openfile->current->prev;
683

684
    /* Read the entire file into the filestruct. */
685
686
687
688
689
690
    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') {
691
#ifndef NANO_TINY
692
693
694
695
696
	    /* 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) &&
697
			i > 0 && buf[i - 1] == '\r') {
698
699
700
		if (format == 0 || format == 2)
		    format++;
	    }
701
#endif
702

703
704
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
705

706
	    /* Reset the line length in preparation for the next line. */
707
	    len = 0;
Chris Allegretta's avatar
Chris Allegretta committed
708

709
710
711
	    num_lines++;
	    buf[0] = '\0';
	    i = 0;
712
#ifndef NANO_TINY
713
714
715
716
717
	/* 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') {
718
719
720
721
722
	    /* 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;
723

724
725
	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
726

727
728
729
730
	    /* 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
731

732
733
734
735
	    num_lines++;
	    buf[0] = input;
	    buf[1] = '\0';
	    i = 1;
736
#endif
737
738
739
740
	} else {
	    /* Calculate the total length of the line.  It might have
	     * nulls in it, so we can't just use strlen() here. */
	    len++;
741

742
743
744
745
746
747
748
749
	    /* 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);
750
	    }
751
752
753
754
755
756
757
758
759
760
761

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

    /* Perhaps this could use some better handling. */
    if (ferror(f))
	nperror(filename);
    fclose(f);
762
    if (fd > 0 && checkwritable) {
763
	close(fd);
764
765
	writable = is_file_writable(filename);
    }
766

767
#ifndef NANO_TINY
768
769
770
771
772
773
774
775
    /* 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';
    }
776
#endif
777

778
779
    /* Did we not get a newline and still have stuff to do? */
    if (len > 0) {
780
#ifndef NANO_TINY
781
782
783
784
785
786
787
	/* 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;
788
789
#endif

790
791
792
793
	/* Read in the last line properly. */
	fileptr = read_line(buf, fileptr, &first_line_ins, len);
	num_lines++;
    }
794

795
    free(buf);
796

797
    /* If we didn't get a file and we don't already have one, open a
798
     * blank buffer. */
799
    if (fileptr == NULL)
800
	open_buffer("", FALSE);
801

802
803
    /* Attach the file we got to the filestruct.  If we got a file of
     * zero bytes, don't do anything. */
804
    if (num_lines > 0) {
805
806
807
808
809
810
811
812
813
814
815
816
	/* 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
817
818
	    /* Tack the text at fileptr onto the beginning of the text
	     * at current. */
819
820
821
822
	    openfile->current->data = charealloc(openfile->current->data,
						len + current_len + 1);
	    charmove(openfile->current->data + len, openfile->current->data,
			current_len + 1);
823
824
825
826
827
828
829
830
831
832
	    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
833
	    /* Move fileptr back one line and blow away the old fileptr,
834
835
	     * since its text has been saved. */
	    fileptr = fileptr->prev;
836
837
	    if (fileptr != NULL)
		free(fileptr->next);
838
839
840
841
842
843
844
845
	}

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

846
	/* Renumber, starting with the last line of the file we inserted. */
847
	renumber(openfile->current);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
848
    }
849

850
    openfile->totsize += get_totsize(openfile->fileage, openfile->filebot);
851

852
    /* If the NO_NEWLINES flag isn't set, and text has been added to
853
     * the magicline (i.e. a file that doesn't end in a newline has been
854
855
856
     * 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') {
857
858
859
860
861
862
863
864
865
	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();

866
#ifndef NANO_TINY
867
868
869
    if (undoable)
	update_undo(INSERT);

870
871
872
    if (format == 3) {
	if (writable)
	    statusbar(
873
874
875
		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);
876
877
878
879
880
881
	else
	    statusbar(
		P_("Read %lu line (Converted from DOS and Mac format - Warning: No write permission)",
		"Read %lu lines (Converted from DOS and Mac format - Warning: No write permission)",
		(unsigned long)num_lines), (unsigned long)num_lines);
    } else if (format == 2) {
882
	openfile->fmt = MAC_FILE;
883
884
	if (writable)
	    statusbar(P_("Read %lu line (Converted from Mac format)",
885
886
		"Read %lu lines (Converted from Mac format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
887
888
889
890
	else
	    statusbar(P_("Read %lu line (Converted from Mac format - Warning: No write permission)",
		"Read %lu lines (Converted from Mac format - Warning: No write permission)",
		(unsigned long)num_lines), (unsigned long)num_lines);
891
    } else if (format == 1) {
892
	openfile->fmt = DOS_FILE;
893
894
	if (writable)
	    statusbar(P_("Read %lu line (Converted from DOS format)",
895
896
		"Read %lu lines (Converted from DOS format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
897
898
899
900
	else
	    statusbar(P_("Read %lu line (Converted from DOS format - Warning: No write permission)",
		"Read %lu lines (Converted from DOS format - Warning: No write permission)",
		(unsigned long)num_lines), (unsigned long)num_lines);
901
    } else
902
#endif
903
904
905
906
	if (writable)
	    statusbar(P_("Read %lu line", "Read %lu lines",
		(unsigned long)num_lines), (unsigned long)num_lines);
	else
907
	    statusbar(P_("Read %lu line (Warning: No write permission)",
908
		"Read %lu lines (Warning: No write permission)",
909
		(unsigned long)num_lines), (unsigned long)num_lines);
910

911
#ifndef NANO_TINY
912
913
    if (ISSET(MAKE_IT_UNIX))
	openfile->fmt = NIX_FILE;
914
#endif
915
}
Chris Allegretta's avatar
Chris Allegretta committed
916

917
918
919
920
/* 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".
 *
921
 * Return -2 if we say "New File", -1 if the file isn't opened, and the
922
 * fd opened otherwise.  The file might still have an error while reading
923
 * with a 0 return value.  *f is set to the opened file. */
924
int open_file(const char *filename, bool newfie, bool quiet, FILE **f)
925
{
926
    struct stat fileinfo, fileinfo2;
927
    int fd;
928
    char *full_filename;
929

930
    assert(filename != NULL && f != NULL);
931

932
933
934
    /* Get the specified file's full path. */
    full_filename = get_full_path(filename);

935
    /* Okay, if we can't stat the path due to a component's
936
     * permissions, just try the relative one. */
937
938
    if (full_filename == NULL || (stat(full_filename, &fileinfo) == -1 &&
		stat(filename, &fileinfo2) != -1))
939
940
941
	full_filename = mallocstrcpy(NULL, filename);

    if (stat(full_filename, &fileinfo) == -1) {
942
943
	/* Well, maybe we can open the file even if the OS says it's
	 * not there. */
944
	if ((fd = open(filename, O_RDONLY)) != -1) {
945
946
	    if (!quiet)
		statusbar(_("Reading File"));
947
948
949
950
	    free(full_filename);
	    return 0;
	}

951
	if (newfie) {
952
953
	    if (!quiet)
		statusbar(_("New File"));
954
955
956
	    return -2;
	}
	statusbar(_("\"%s\" not found"), filename);
957
	beep();
958
959
	return -1;
    } else if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
960
		S_ISBLK(fileinfo.st_mode)) {
961
962
	/* Don't open directories, character files, or block files.
	 * Sorry, /dev/sndstat! */
963
964
	statusbar(S_ISDIR(fileinfo.st_mode) ?
		_("\"%s\" is a directory") :
965
		_("\"%s\" is a device file"), filename);
966
	beep();
967
	return -1;
968
    } else if ((fd = open(full_filename, O_RDONLY)) == -1) {
969
	statusbar(_("Error reading %s: %s"), filename, strerror(errno));
970
	beep();
971
	return -1;
972
    } else {
973
	/* The file is A-OK.  Open it. */
974
	*f = fdopen(fd, "rb");
975

976
	if (*f == NULL) {
977
	    statusbar(_("Error reading %s: %s"), filename, strerror(errno));
978
	    beep();
979
980
981
	    close(fd);
	} else
	    statusbar(_("Reading File"));
982
    }
983

984
985
    free(full_filename);

986
987
988
    return fd;
}

989
990
991
992
993
/* 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)
994
{
995
996
    unsigned long i = 0;
    char *buf;
997
    size_t wholenamelen;
998

999
    assert(name != NULL && suffix != NULL);
1000

1001
    wholenamelen = strlen(name) + strlen(suffix);
1002

1003
1004
1005
    /* Reserve space for: the name plus the suffix plus a dot plus
     * possibly five digits plus a null byte. */
    buf = charalloc(wholenamelen + 7);
1006
    sprintf(buf, "%s%s", name, suffix);
1007

1008
1009
    while (TRUE) {
	struct stat fs;
Chris Allegretta's avatar
Chris Allegretta committed
1010

1011
1012
	if (stat(buf, &fs) == -1)
	    return buf;
1013
1014
1015

	/* Limit the number of backup files to a hundred thousand. */
	if (++i == 100000)
1016
	    break;
1017

1018
	sprintf(buf + wholenamelen, ".%lu", i);
1019
    }
Chris Allegretta's avatar
Chris Allegretta committed
1020

1021
1022
1023
    /* We get here only if there is no possible save file.  Blank out
     * the filename to indicate this. */
    null_at(&buf, 0);
1024

1025
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
1026
1027
}

1028
1029
1030
/* 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. */
1031
void do_insertfile(
1032
#ifndef NANO_TINY
1033
1034
1035
1036
1037
1038
1039
1040
1041
	bool execute
#else
	void
#endif
	)
{
    int i;
    const char *msg;
    char *ans = mallocstrcpy(NULL, "");
1042
	/* The last answer the user typed at the statusbar prompt. */
1043
    filestruct *edittop_save = openfile->edittop;
1044
    ssize_t was_current_lineno = openfile->current->lineno;
1045
1046
    size_t was_current_x = openfile->current_x;
    ssize_t was_current_y = openfile->current_y;
1047
    bool edittop_inside = FALSE;
1048
#ifndef NANO_TINY
1049
    bool right_side_up = FALSE, single_line = FALSE;
1050
#endif
1051

1052
    while (TRUE) {
1053
#ifndef NANO_TINY
1054
	if (execute) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1055
	    msg =
1056
#ifndef DISABLE_MULTIBUFFER
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1057
		ISSET(MULTIBUFFER) ?
1058
		_("Command to execute in new buffer [from %s] ") :
1059
#endif
1060
		_("Command to execute [from %s] ");
1061
1062
1063
	} else
#endif /* NANO_TINY */
	{
1064
	    msg =
1065
#ifndef DISABLE_MULTIBUFFER
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1066
		ISSET(MULTIBUFFER) ?
1067
		_("File to insert into new buffer [from %s] ") :
1068
#endif
1069
		_("File to insert [from %s] ");
1070
	}
1071

1072
	i = do_prompt(TRUE,
1073
1074
1075
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
1076
#ifndef NANO_TINY
1077
		execute ? MEXTCMD :
1078
#endif
1079
		MINSERTFILE, ans,
1080
#ifndef DISABLE_HISTORIES
1081
		NULL,
1082
#endif
1083
		edit_refresh, msg,
1084
#ifndef DISABLE_OPERATINGDIR
1085
1086
		operating_dir != NULL && strcmp(operating_dir,
		".") != 0 ? operating_dir :
1087
1088
#endif
		"./");
1089

1090
	/* If we're in multibuffer mode and the filename or command is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1091
	 * blank, open a new buffer instead of canceling.  If the
1092
1093
	 * 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
1094
	if (i == -1 || ((i == -2 || *answer == '\n')
1095
#ifndef DISABLE_MULTIBUFFER
1096
1097
1098
1099
1100
1101
		&& !ISSET(MULTIBUFFER)
#endif
		)) {
	    statusbar(_("Cancelled"));
	    break;
	} else {
1102
	    size_t pww_save = openfile->placewewant;
1103
	    functionptrtype func = func_from_key(&i);
1104

1105
	    ans = mallocstrcpy(ans, answer);
1106

1107
#ifndef NANO_TINY
1108
#ifndef DISABLE_MULTIBUFFER
1109
	    if (func == new_buffer_void) {
1110
1111
1112
		/* Don't allow toggling if we're in view mode. */
		if (!ISSET(VIEW_MODE))
		    TOGGLE(MULTIBUFFER);
1113
1114
		else
		    beep();
1115
		continue;
1116
	    }
1117
#endif
1118
	    if (func == flip_execute_void) {
1119
1120
1121
		execute = !execute;
		continue;
	    }
1122
#endif /* !NANO_TINY */
1123

1124
#ifndef DISABLE_BROWSER
1125
	    if (func == to_files_void) {
1126
		char *tmp = do_browse_from(answer);
1127

1128
1129
		if (tmp == NULL)
		    continue;
1130

1131
		/* We have a file now.  Indicate this. */
1132
1133
		free(answer);
		answer = tmp;
1134

1135
1136
1137
		i = 0;
	    }
#endif
1138

1139
	    /* If we don't have a file yet, go back to the statusbar prompt. */
1140
	    if (i != 0
1141
#ifndef DISABLE_MULTIBUFFER
1142
1143
1144
1145
		&& (i != -2 || !ISSET(MULTIBUFFER))
#endif
		)
		continue;
1146

1147
#ifndef NANO_TINY
1148
1149
1150
1151
1152
	    /* 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;
1153

1154
		mark_order((const filestruct **)&top, &top_x,
1155
1156
1157
			(const filestruct **)&bot, &bot_x,
			&right_side_up);

1158
1159
		single_line = (top == bot);
	    }
1160
1161
#endif

1162
#ifndef DISABLE_MULTIBUFFER
1163
1164
1165
1166
	    if (!ISSET(MULTIBUFFER)) {
#endif
		/* If we're not inserting into a new buffer, partition
		 * the filestruct so that it contains no text and hence
1167
		 * looks like a new buffer, and keep track of whether
1168
		 * the top of the edit window is inside the partition. */
1169
1170
1171
		filepart = partition_filestruct(openfile->current,
			openfile->current_x, openfile->current,
			openfile->current_x);
1172
		edittop_inside = (openfile->edittop == openfile->fileage);
1173
#ifndef DISABLE_MULTIBUFFER
1174
1175
	    }
#endif
Chris Allegretta's avatar
Chris Allegretta committed
1176

1177
1178
	    /* Convert newlines to nulls, just before we insert the file
	     * or execute the command. */
1179
	    sunder(answer);
1180
	    align(&answer);
1181

1182
#ifndef NANO_TINY
1183
	    if (execute) {
1184
#ifndef DISABLE_MULTIBUFFER
1185
		if (ISSET(MULTIBUFFER))
1186
		    /* Open a blank buffer. */
1187
		    open_buffer("", FALSE);
1188
1189
1190
#endif

		/* Save the command's output in the current buffer. */
1191
		execute_command(answer);
1192

1193
#ifndef DISABLE_MULTIBUFFER
1194
1195
1196
1197
1198
1199
1200
		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;
		}
1201
#endif
1202
	    } else
1203
#endif /* !NANO_TINY */
1204
	    {
1205
1206
		/* Make sure the path to the file specified in answer is
		 * tilde-expanded. */
1207
		answer = mallocstrassn(answer, real_dir_from_tilde(answer));
1208

1209
		/* Save the file specified in answer in the current buffer. */
1210
		open_buffer(answer, TRUE);
1211
	    }
1212

1213
#if !defined(DISABLE_MULTIBUFFER) && !defined(DISABLE_HISTORIES)
1214
	    if (ISSET(MULTIBUFFER)) {
1215
		/* Update the screen to account for the current buffer. */
1216
		display_buffer();
1217
1218

		ssize_t savedposline, savedposcol;
1219
1220
1221
1222
1223
		if (ISSET(POS_HISTORY) &&
#ifndef NANO_TINY
			!execute &&
#endif
			check_poshistory(answer, &savedposline, &savedposcol))
1224
1225
		    do_gotolinecolumn(savedposline, savedposcol, FALSE, FALSE, FALSE, FALSE);
	    } else
1226
#endif /* !DISABLE_MULTIBUFFER && !DISABLE_HISTORIES */
1227
	    {
1228
		filestruct *top_save = openfile->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
1229

1230
1231
1232
		/* 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
1233
		if (edittop_inside)
1234
		    edittop_save = openfile->fileage;
1235
1236

		/* Update the current x-coordinate to account for the
1237
1238
1239
1240
		 * 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. */
1241
		openfile->current_x = strlen(openfile->filebot->data);
1242
1243
1244
1245
		if (openfile->fileage == openfile->filebot) {
#ifndef NANO_TINY
		    if (openfile->mark_set) {
			openfile->mark_begin = openfile->current;
1246
			if (!right_side_up)
1247
			    openfile->mark_begin_x += openfile->current_x;
1248
1249
		    }
#endif
1250
		    openfile->current_x += was_current_x;
1251
		}
1252
#ifndef NANO_TINY
1253
1254
1255
1256
		else if (openfile->mark_set) {
		    if (!right_side_up) {
			if (single_line) {
			    openfile->mark_begin = openfile->current;
1257
			    openfile->mark_begin_x -= was_current_x;
1258
			} else
1259
			    openfile->mark_begin_x -= openfile->current_x;
1260
1261
		    }
		}
1262
#endif
1263
1264
1265

		/* Update the current y-coordinate to account for the
		 * number of lines inserted. */
1266
		openfile->current_y += was_current_y;
1267

1268
1269
1270
1271
		/* 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. */
1272
		unpartition_filestruct(&filepart);
1273

1274
1275
1276
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
1277

1278
		/* Restore the old edittop. */
1279
		openfile->edittop = edittop_save;
1280

1281
1282
1283
		/* Restore the old place we want. */
		openfile->placewewant = pww_save;

1284
1285
		/* Mark the file as modified if it changed. */
		if (openfile->current->lineno != was_current_lineno ||
1286
			openfile->current_x != was_current_x)
1287
		    set_modified();
1288

1289
1290
		/* Update the screen. */
		edit_refresh();
1291
	    }
1292

1293
1294
1295
1296
	    break;
	}
    }
    free(ans);
1297
1298
}

1299
1300
1301
/* 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. */
1302
void do_insertfile_void(void)
1303
{
1304
    if (ISSET(RESTRICTED)) {
1305
	show_restricted_warning();
1306
1307
1308
	return;
    }

1309
#ifndef DISABLE_MULTIBUFFER
1310
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
1311
	statusbar(_("Key invalid in non-multibuffer mode"));
1312
1313
1314
    else
#endif
	do_insertfile(
1315
#ifndef NANO_TINY
1316
1317
1318
		FALSE
#endif
		);
1319
1320
1321
1322

    display_main_list();
}

1323
/* When passed "[relative path]" or "[relative path][filename]" in
1324
 * origpath, return "[full path]" or "[full path][filename]" on success,
1325
1326
1327
1328
 * 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. */
1329
char *get_full_path(const char *origpath)
1330
{
1331
1332
1333
1334
    struct stat fileinfo;
    char *d_here, *d_there, *d_there_file = NULL;
    const char *last_slash;
    bool path_only;
1335

1336
    if (origpath == NULL)
1337
	return NULL;
1338

1339
    /* Get the current directory.  If it doesn't exist, back up and try
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1340
1341
     * again until we get a directory that does, and use that as the
     * current directory. */
1342
1343
    d_here = charalloc(PATH_MAX + 1);
    d_here = getcwd(d_here, PATH_MAX + 1);
1344

1345
    while (d_here == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1346
1347
1348
	if (chdir("..") == -1)
	    break;

1349
1350
1351
1352
1353
	d_here = getcwd(d_here, PATH_MAX + 1);
    }

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

1356
1357
	/* If the current directory isn't "/", tack a slash onto the end
	 * of it. */
1358
	if (strcmp(d_here, "/") != 0) {
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1359
	    d_here = charealloc(d_here, strlen(d_here) + 2);
1360
1361
	    strcat(d_here, "/");
	}
1362
1363
1364
    /* Otherwise, set d_here to "". */
    } else
	d_here = mallocstrcpy(NULL, "");
1365

1366
    d_there = real_dir_from_tilde(origpath);
1367

1368
1369
1370
    /* 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. */
1371
    path_only = (stat(d_there, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode));
1372

1373
1374
1375
    /* If path_only is TRUE, make sure d_there ends in a slash. */
    if (path_only) {
	size_t d_there_len = strlen(d_there);
1376

1377
1378
1379
	if (d_there[d_there_len - 1] != '/') {
	    d_there = charealloc(d_there, d_there_len + 2);
	    strcat(d_there, "/");
1380
	}
1381
    }
1382

1383
1384
    /* Search for the last slash in d_there. */
    last_slash = strrchr(d_there, '/');
1385

1386
1387
1388
1389
    /* 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);
1390

1391
1392
1393
1394
1395
1396
1397
1398
	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);

1399
	/* Remove the filename portion of the answer from d_there. */
1400
1401
	null_at(&d_there, last_slash - d_there + 1);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1402
	/* Go to the path specified in d_there. */
1403
1404
1405
	if (chdir(d_there) == -1) {
	    free(d_there);
	    d_there = NULL;
1406
	} else {
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416
1417
1418
1419
1420
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
	    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()
1431
	     * error, since we can do nothing if we get one. */
1432
	    IGNORE_CALL_RESULT(chdir(d_here));
1433

1434
1435
	    /* Free d_here, since we're done using it. */
	    free(d_here);
1436
	}
1437
    }
1438

1439
1440
1441
1442
1443
1444
1445
    /* 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) +
1446
		strlen(d_there_file) + 1);
1447
1448
	strcat(d_there, d_there_file);
    }
1449

1450
    /* Free d_there_file, since we're done using it. */
1451
    free(d_there_file);
1452

1453
    return d_there;
1454
}
1455

1456
/* Return the full version of path, as returned by get_full_path().  On
1457
 * error, or if path doesn't reference a directory, or if the directory
1458
 * isn't writable, return NULL. */
1459
char *check_writable_directory(const char *path)
1460
{
1461
1462
    char *full_path = get_full_path(path);

1463
    /* If get_full_path() fails, return NULL. */
1464
    if (full_path == NULL)
1465
	return NULL;
1466

1467
    /* If we can't write to path or path isn't a directory, return NULL. */
1468
    if (access(full_path, W_OK) != 0 ||
1469
		full_path[strlen(full_path) - 1] != '/') {
1470
	free(full_path);
1471
	return NULL;
1472
    }
1473

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1474
    /* Otherwise, return the full path. */
1475
1476
1477
    return full_path;
}

1478
1479
/* This function calls mkstemp(($TMPDIR|P_tmpdir|/tmp/)"nano.XXXXXX").
 * On success, it returns the malloc()ed filename and corresponding FILE
1480
 * stream, opened in "r+b" mode.  On error, it returns NULL for the
1481
1482
 * filename and leaves the FILE stream unchanged. */
char *safe_tempfile(FILE **f)
1483
{
1484
    char *full_tempdir = NULL;
1485
1486
1487
    const char *tmpdir_env;
    int fd;
    mode_t original_umask = 0;
1488

1489
1490
    assert(f != NULL);

1491
1492
1493
    /* 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. */
1494
    tmpdir_env = getenv("TMPDIR");
1495
    if (tmpdir_env != NULL)
1496
	full_tempdir = check_writable_directory(tmpdir_env);
1497

1498
1499
    /* If $TMPDIR is unset, empty, or not a writable directory, and
     * full_tempdir is NULL, try P_tmpdir instead. */
1500
    if (full_tempdir == NULL)
1501
	full_tempdir = check_writable_directory(P_tmpdir);
1502

1503
1504
    /* if P_tmpdir is NULL, use /tmp. */
    if (full_tempdir == NULL)
1505
	full_tempdir = mallocstrcpy(NULL, "/tmp/");
1506

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1507
    full_tempdir = charealloc(full_tempdir, strlen(full_tempdir) + 12);
1508
    strcat(full_tempdir, "nano.XXXXXX");
1509

1510
1511
1512
1513
1514
1515
1516
    original_umask = umask(0);
    umask(S_IRWXG | S_IRWXO);

    fd = mkstemp(full_tempdir);

    if (fd != -1)
	*f = fdopen(fd, "r+b");
1517
1518
1519
    else {
	free(full_tempdir);
	full_tempdir = NULL;
1520
    }
1521

1522
1523
    umask(original_umask);

1524
    return full_tempdir;
1525
}
1526
1527

#ifndef DISABLE_OPERATINGDIR
1528
1529
1530
1531
1532
/* Initialize full_operating_dir based on operating_dir. */
void init_operating_dir(void)
{
    assert(full_operating_dir == NULL);

1533
    if (operating_dir == NULL)
1534
	return;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1535

1536
1537
1538
    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
1539
     * inaccessible, unset operating_dir. */
1540
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) {
1541
1542
1543
1544
1545
1546
1547
	free(full_operating_dir);
	full_operating_dir = NULL;
	free(operating_dir);
	operating_dir = NULL;
    }
}

1548
1549
1550
1551
1552
/* 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)
1553
{
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1554
1555
1556
1557
    /* 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. */
1558

1559
    char *fullpath;
1560
    bool retval = FALSE;
1561
    const char *whereami1, *whereami2 = NULL;
1562

1563
    /* If no operating directory is set, don't bother doing anything. */
1564
    if (operating_dir == NULL)
1565
	return FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1566

1567
    assert(full_operating_dir != NULL);
1568
1569

    fullpath = get_full_path(currpath);
1570

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1571
1572
1573
    /* 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
1574
     * non-existent directory as being outside the operating directory,
1575
     * so we return FALSE.  If allow_tabcomp is TRUE, then currpath
1576
1577
     * exists, but is not executable.  So we say it isn't in the
     * operating directory. */
1578
    if (fullpath == NULL)
1579
	return allow_tabcomp;
1580
1581
1582
1583
1584

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

1585
    /* If both searches failed, we're outside the operating directory.
1586
     * Otherwise, check the search results.  If the full operating
1587
1588
1589
     * 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. */
1590
    if (whereami1 != fullpath && whereami2 != full_operating_dir)
1591
1592
	retval = TRUE;
    free(fullpath);
1593
1594

    /* Otherwise, we're still inside it. */
1595
    return retval;
1596
}
1597
1598
#endif

1599
#ifndef NANO_TINY
1600
1601
/* Although this sucks, it sucks less than having a single 'my system is
 * messed up and I'm blanket allowing insecure file writing operations'. */
1602
1603
1604
int prompt_failed_backupwrite(const char *filename)
{
    static int i;
1605
    static char *prevfile = NULL; /* What was the last file we were
1606
1607
				   * passed so we don't keep asking
				   * this?  Though maybe we should... */
1608
1609
    if (prevfile == NULL || strcmp(filename, prevfile)) {
	i = do_yesno_prompt(FALSE,
1610
		_("Failed to write backup file, continue saving? (Say N if unsure) "));
1611
1612
1613
1614
1615
	prevfile = mallocstrcpy(prevfile, filename);
    }
    return i;
}

1616
1617
1618
1619
1620
1621
1622
1623
1624
1625
1626
1627
1628
1629
1630
1631
1632
1633
1634
1635
1636
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;
    }
}
1637
#endif /* !NANO_TINY */
1638

1639
/* Read from inn, write to out.  We assume inn is opened for reading,
1640
1641
 * and out for writing.  We return 0 on success, -1 on read error, or -2
 * on write error. */
1642
1643
int copy_file(FILE *inn, FILE *out)
{
1644
    int retval = 0;
1645
    char buf[BUFSIZ];
1646
1647
    size_t charsread;

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

1650
1651
1652
1653
1654
1655
1656
1657
1658
1659
1660
    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
1661

1662
1663
1664
1665
    if (fclose(inn) == EOF)
	retval = -1;
    if (fclose(out) == EOF)
	retval = -2;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1666

1667
1668
1669
    return retval;
}

1670
1671
/* 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
1672
 * ourselves.  If tmp is TRUE, we set the umask to disallow anyone else
1673
1674
 * 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
1675
 *
1676
 * tmp means we are writing a temporary file in a secure fashion.  We
1677
1678
1679
1680
 * 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
1681
1682
 * filename.  nonamechange is ignored if tmp is FALSE, we're appending,
 * or we're prepending.
1683
 *
1684
1685
 * Return TRUE on success or FALSE on error. */
bool write_file(const char *name, FILE *f_open, bool tmp, append_type
1686
	append, bool nonamechange)
Chris Allegretta's avatar
Chris Allegretta committed
1687
{
1688
    bool retval = FALSE;
1689
	/* Instead of returning in this function, you should always
1690
	 * set retval and then goto cleanup_and_exit. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1691
    size_t lineswritten = 0;
1692
    const filestruct *fileptr = openfile->fileage;
1693
    int fd;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1694
	/* The file descriptor we use. */
1695
    mode_t original_umask = 0;
1696
	/* Our umask, from when nano started. */
1697
#ifndef NANO_TINY
1698
    bool realexists;
1699
	/* The result of stat().  TRUE if the file exists, FALSE
1700
	 * otherwise.  If name is a link that points nowhere, realexists
1701
	 * is FALSE. */
1702
1703
    struct stat st;
	/* The status fields filled in by stat(). */
1704
#endif
1705
    bool anyexists;
1706
1707
	/* The result of lstat().  The same as realexists, unless name
	 * is a link. */
1708
1709
1710
    struct stat lst;
	/* The status fields filled in by lstat(). */
    char *realname;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1711
	/* name after tilde expansion. */
1712
    FILE *f = NULL;
1713
1714
	/* The actual file, realname, we are writing to. */
    char *tempname = NULL;
1715
	/* The name of the temporary file we write to on prepend. */
Chris Allegretta's avatar
Chris Allegretta committed
1716

1717
    assert(name != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1718

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1719
    if (*name == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1720
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1721

1722
1723
1724
    if (f_open != NULL)
	f = f_open;

1725
1726
    if (!tmp)
	titlebar(NULL);
1727

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1728
    realname = real_dir_from_tilde(name);
1729

1730
#ifndef DISABLE_OPERATINGDIR
1731
    /* If we're writing a temporary file, we're probably going outside
1732
     * the operating directory, so skip the operating directory test. */
1733
    if (!tmp && check_operating_dir(realname, FALSE)) {
1734
1735
	statusbar(_("Can't write outside of %s"), operating_dir);
	goto cleanup_and_exit;
1736
1737
1738
    }
#endif

1739
    anyexists = (lstat(realname, &lst) != -1);
1740

1741
1742
    /* If the temp file exists and isn't already open, give up. */
    if (tmp && anyexists && f_open == NULL)
1743
	goto cleanup_and_exit;
1744

1745
1746
1747
    /* 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)) {
1748
1749
	statusbar(
		_("Cannot prepend or append to a symlink with --nofollow set"));
1750
1751
1752
	goto cleanup_and_exit;
    }

1753
#ifndef NANO_TINY
1754
    /* Check whether the file (at the end of the symlink) exists. */
1755
    realexists = (stat(realname, &st) != -1);
1756

1757
1758
1759
1760
    /* If we haven't stat()d this file before (say, the user just
     * specified it interactively), stat and save the value now,
     * or else we will chase null pointers when we do modtime checks,
     * preserve file times, and so on, during backup. */
1761
1762
    if (openfile->current_stat == NULL && !tmp && realexists) {
	openfile->current_stat = (struct stat *)nmalloc(sizeof(struct stat));
1763
	stat(realname, openfile->current_stat);
1764
    }
1765

1766
    /* We backup only if the backup toggle is set, the file isn't
1767
1768
1769
1770
     * 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. */
1771
1772
1773
    if (ISSET(BACKUP_FILE) && !tmp && realexists && ((append != OVERWRITE ||
		openfile->mark_set) || (openfile->current_stat &&
		openfile->current_stat->st_mtime == st.st_mtime))) {
1774
	int backup_fd;
1775
	FILE *backup_file;
1776
	char *backupname;
1777
	struct utimbuf filetime;
1778
	int copy_status;
1779
	int backup_cflags;
1780

1781
	/* Save the original file's access and modification times. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1782
1783
	filetime.actime = openfile->current_stat->st_atime;
	filetime.modtime = openfile->current_stat->st_mtime;
1784

1785
1786
1787
	if (f_open == NULL) {
	    /* Open the original file to copy to the backup. */
	    f = fopen(realname, "rb");
1788

1789
1790
1791
	    if (f == NULL) {
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1792
		beep();
1793
1794
1795
1796
		/* 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;
1797
	    }
1798
1799
	}

1800
	/* If backup_dir is set, we set backupname to
1801
1802
1803
1804
	 * 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]. */
1805
	if (backup_dir != NULL) {
1806
	    char *backuptemp = get_full_path(realname);
1807

1808
	    if (backuptemp == NULL)
1809
1810
1811
1812
1813
1814
		/* 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~. */
1815
		backuptemp = mallocstrcpy(NULL, tail(realname));
1816
	    else {
1817
1818
		size_t i = 0;

1819
1820
1821
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1822
1823
1824
1825
		}
	    }

	    backupname = charalloc(strlen(backup_dir) +
1826
1827
1828
1829
		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
1830
	    if (*backuptemp == '\0') {
1831
		statusbar(_("Error writing backup file %s: %s"), backupname,
1832
1833
1834
		    _("Too many backup files?"));
		free(backuptemp);
		free(backupname);
1835
1836
1837
1838
1839
		/* If we can't write to the backup, DON'T go on, since
		 * whatever caused the backup file to fail (e.g. disk
		 * full may well cause the real file write to fail,
		 * which means we could lose both the backup and the
		 * original! */
1840
		goto cleanup_and_exit;
1841
1842
1843
1844
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1845
1846
1847
1848
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1849

1850
	/* First, unlink any existing backups.  Next, open the backup
1851
1852
	 * file with O_CREAT and O_EXCL.  If it succeeds, we have a file
	 * descriptor to a new backup file. */
1853
	if (unlink(backupname) < 0 && errno != ENOENT && !ISSET(INSECURE_BACKUP)) {
1854
1855
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1856
1857
	    statusbar(_("Error writing backup file %s: %s"), backupname,
			strerror(errno));
1858
1859
1860
1861
	    free(backupname);
	    goto cleanup_and_exit;
	}

1862
1863
	if (ISSET(INSECURE_BACKUP))
	    backup_cflags = O_WRONLY | O_CREAT | O_APPEND;
1864
	else
1865
1866
1867
	    backup_cflags = O_WRONLY | O_CREAT | O_EXCL | O_APPEND;

	backup_fd = open(backupname, backup_cflags,
1868
1869
		S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
	/* Now we've got a safe file stream.  If the previous open()
1870
	 * call failed, this will return NULL. */
1871
	backup_file = fdopen(backup_fd, "wb");
1872

1873
1874
1875
1876
1877
1878
	if (backup_fd < 0 || backup_file == NULL) {
	    statusbar(_("Error writing backup file %s: %s"), backupname,
			strerror(errno));
	    free(backupname);
	    goto cleanup_and_exit;
	}
1879

1880
1881
	/* We shouldn't worry about chown()ing something if we're not
	 * root, since it's likely to fail! */
1882
	if (geteuid() == NANO_ROOT_UID && fchown(backup_fd,
1883
		openfile->current_stat->st_uid, openfile->current_stat->st_gid) == -1
1884
1885
1886
		&& !ISSET(INSECURE_BACKUP)) {
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1887
1888
1889
1890
1891
1892
1893
	    statusbar(_("Error writing backup file %s: %s"), backupname,
		strerror(errno));
	    free(backupname);
	    fclose(backup_file);
	    goto cleanup_and_exit;
	}

1894
1895
1896
1897
	if (fchmod(backup_fd, openfile->current_stat->st_mode) == -1
		&& !ISSET(INSECURE_BACKUP)) {
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1898
	    statusbar(_("Error writing backup file %s: %s"), backupname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1899
		strerror(errno));
1900
	    free(backupname);
1901
	    fclose(backup_file);
1902
	    /* If we can't write to the backup, DONT go on, since
1903
	     * whatever caused the backup file to fail (e.g. disk
1904
	     * full) may well cause the real file write to fail, which
1905
	     * means we could lose both the backup and the original! */
1906
	    goto cleanup_and_exit;
1907
1908
1909
	}

#ifdef DEBUG
1910
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1911
1912
#endif

1913
1914
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1915

1916
1917
	if (copy_status != 0) {
	    statusbar(_("Error reading %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1918
			strerror(errno));
1919
1920
1921
1922
1923
1924
	    beep();
	    goto cleanup_and_exit;
	}

	/* And set its metadata. */
	if (utime(backupname, &filetime) == -1 && !ISSET(INSECURE_BACKUP)) {
1925
1926
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1927
	    statusbar(_("Error writing backup file %s: %s"), backupname,
1928
			strerror(errno));
1929
1930
1931
1932
	    /* If we can't write to the backup, DON'T go on, since
	     * whatever caused the backup file to fail (e.g. disk full
	     * may well cause the real file write to fail, which means
	     * we could lose both the backup and the original! */
1933
	    goto cleanup_and_exit;
1934
	}
1935

1936
1937
	free(backupname);
    }
1938

1939
    skip_backup:
1940
#endif /* !NANO_TINY */
1941

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1942
1943
    /* 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
1944
1945
     * overwrite. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode) &&
1946
		unlink(realname) == -1) {
1947
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1948
	goto cleanup_and_exit;
1949
    }
1950

1951
1952
    if (f_open == NULL) {
	original_umask = umask(0);
1953

1954
	/* If we create a temp file, we don't let anyone else access it.
1955
1956
	 * We create a temp file if tmp is TRUE. */
	if (tmp)
1957
	    umask(S_IRWXG | S_IRWXO);
1958
1959
	else
	    umask(original_umask);
1960
    }
1961

1962
    /* If we're prepending, copy the file to a temp file. */
1963
    if (append == PREPEND) {
1964
1965
1966
	int fd_source;
	FILE *f_source = NULL;

1967
1968
1969
1970
1971
1972
1973
1974
1975
1976
1977
	if (f == NULL) {
	    f = fopen(realname, "rb");

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

1978
	tempname = safe_tempfile(&f);
1979

1980
	if (tempname == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1981
	    statusbar(_("Error writing temp file: %s"),
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1982
		strerror(errno));
1983
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1984
	}
1985

1986
	if (f_open == NULL) {
1987
	    fd_source = open(realname, O_RDONLY | O_CREAT, S_IRUSR | S_IWUSR);
1988

1989
1990
1991
1992
1993
	    if (fd_source != -1) {
		f_source = fdopen(fd_source, "rb");
		if (f_source == NULL) {
		    statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1994
		    beep();
1995
1996
1997
1998
1999
2000
		    close(fd_source);
		    fclose(f);
		    unlink(tempname);
		    goto cleanup_and_exit;
		}
	    }
2001
2002
	}

2003
	if (f_source == NULL || copy_file(f_source, f) != 0) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2004
2005
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
2006
	    unlink(tempname);
2007
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
2008
2009
2010
	}
    }

2011
2012
2013
    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
2014
2015
2016
	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);
2017

2018
2019
	/* Set the umask back to the user's original value. */
	umask(original_umask);
2020

2021
2022
2023
2024
	/* If we couldn't open the file, give up. */
	if (fd == -1) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
2025

2026
2027
2028
2029
2030
	    /* tempname has been set only if we're prepending. */
	    if (tempname != NULL)
		unlink(tempname);
	    goto cleanup_and_exit;
	}
2031

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

2034
2035
2036
2037
2038
2039
	if (f == NULL) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
	    close(fd);
	    goto cleanup_and_exit;
	}
2040
2041
    }

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

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

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

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

2054
2055
	/* Convert nulls to newlines.  data_len is the string's real
	 * length. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2056
2057
	unsunder(fileptr->data, data_len);

2058
	if (size < data_len) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2059
2060
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
2061
	    fclose(f);
2062
	    goto cleanup_and_exit;
2063
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2064

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2065
	/* If we're on the last line of the file, don't write a newline
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2066
2067
2068
2069
2070
2071
2072
	 * 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 {
2073
#ifndef NANO_TINY
2074
	    if (openfile->fmt == DOS_FILE || openfile->fmt == MAC_FILE) {
2075
2076
		if (putc('\r', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2077
			strerror(errno));
2078
2079
2080
		    fclose(f);
		    goto cleanup_and_exit;
		}
2081
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2082

2083
	    if (openfile->fmt != MAC_FILE)
2084
#endif
2085
2086
		if (putc('\n', f) == EOF) {
		    statusbar(_("Error writing %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2087
			strerror(errno));
2088
2089
2090
2091
		    fclose(f);
		    goto cleanup_and_exit;
		}
	}
Chris Allegretta's avatar
Chris Allegretta committed
2092
2093
2094
2095
2096

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

2097
    /* If we're prepending, open the temp file, and append it to f. */
2098
    if (append == PREPEND) {
2099
2100
2101
	int fd_source;
	FILE *f_source = NULL;

2102
	fd_source = open(tempname, O_RDONLY | O_CREAT, S_IRUSR | S_IWUSR);
2103

2104
2105
2106
2107
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
2108
	}
2109

2110
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2111
2112
	    statusbar(_("Error reading %s: %s"), tempname,
		strerror(errno));
2113
	    beep();
2114
	    fclose(f);
2115
	    goto cleanup_and_exit;
2116
2117
	}

2118
	if (copy_file(f_source, f) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2119
2120
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
2121
	    goto cleanup_and_exit;
2122
	}
2123
2124

	unlink(tempname);
2125
2126
2127
2128
2129
    } else if (fclose(f) != 0) {
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
	    goto cleanup_and_exit;
    }
2130

2131
    if (!tmp && append == OVERWRITE) {
Chris Allegretta's avatar
Chris Allegretta committed
2132
	if (!nonamechange) {
2133
2134
	    openfile->filename = mallocstrcpy(openfile->filename,
		realname);
2135
#ifndef DISABLE_COLOR
2136
	    /* We might have changed the filename, so update the colors
2137
2138
	     * to account for it, and then make sure we're using
	     * them. */
2139
	    color_update();
2140
	    color_init();
2141
2142
2143

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

2149
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2150
2151
2152
2153
	/* 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));
2154
2155
	if (!openfile->mark_set)
	    stat(realname, openfile->current_stat);
2156
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2157

2158
2159
	statusbar(P_("Wrote %lu line", "Wrote %lu lines",
		(unsigned long)lineswritten),
2160
		(unsigned long)lineswritten);
2161
	openfile->modified = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2162
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2163
    }
2164

2165
    retval = TRUE;
2166
2167
2168

  cleanup_and_exit:
    free(realname);
2169
    free(tempname);
2170

2171
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
2172
2173
}

2174
#ifndef NANO_TINY
2175
2176
/* Write a marked selection from a file out to disk.  Return TRUE on
 * success or FALSE on error. */
2177
bool write_marked_file(const char *name, FILE *f_open, bool tmp,
2178
	append_type append)
2179
{
2180
    bool retval;
2181
2182
    bool old_modified = openfile->modified;
	/* write_file() unsets the modified flag. */
2183
    bool added_magicline = FALSE;
2184
2185
2186
2187
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

2188
2189
    assert(openfile->mark_set);

2190
2191
2192
    /* Partition the filestruct so that it contains only the marked
     * text. */
    mark_order((const filestruct **)&top, &top_x,
2193
	(const filestruct **)&bot, &bot_x, NULL);
2194
    filepart = partition_filestruct(top, top_x, bot, bot_x);
2195

2196
2197
2198
2199
2200
    /* 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) &&
2201
		(added_magicline = (openfile->filebot->data[0] != '\0')))
2202
	new_magicline();
2203

2204
    retval = write_file(name, f_open, tmp, append, TRUE);
2205

2206
2207
2208
    /* If the NO_NEWLINES flag isn't set, and we added a magicline,
     * remove it now. */
    if (!ISSET(NO_NEWLINES) && added_magicline)
2209
2210
2211
2212
	remove_magicline();

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

2215
    if (old_modified)
2216
2217
2218
2219
	set_modified();

    return retval;
}
2220

2221
#endif /* !NANO_TINY */
2222

2223
2224
2225
/* 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
2226
2227
 * TEMP_FILE flag is set and the current file has a name.  Return TRUE
 * on success or FALSE on error. */
2228
bool do_writeout(bool exiting)
Chris Allegretta's avatar
Chris Allegretta committed
2229
{
2230
    int i;
2231
    append_type append = OVERWRITE;
2232
    char *ans;
2233
	/* The last answer the user typed at the statusbar prompt. */
2234
#ifndef DISABLE_EXTRA
2235
    static bool did_credits = FALSE;
2236
#endif
2237
    bool retval = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2238

2239
    if (openfile->filename[0] != '\0' && ISSET(TEMP_FILE)) {
2240
	retval = write_file(openfile->filename, NULL, FALSE, OVERWRITE,
2241
		FALSE);
2242
2243

	/* Write succeeded. */
2244
	if (retval)
2245
	    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
2246
2247
    }

2248
    ans = mallocstrcpy(NULL,
2249
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2250
	(!exiting && openfile->mark_set) ? "" :
2251
#endif
2252
	openfile->filename);
2253
2254
2255

    while (TRUE) {
	const char *msg;
2256
#ifndef NANO_TINY
2257
2258
	const char *formatstr, *backupstr;

2259
2260
	formatstr = (openfile->fmt == DOS_FILE) ? _(" [DOS Format]") :
		    (openfile->fmt == MAC_FILE) ? _(" [Mac Format]") : "";
2261

2262
	backupstr = ISSET(BACKUP_FILE) ? _(" [Backup]") : "";
2263

2264
	/* If we're using restricted mode, don't display the "Write
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2265
2266
2267
	 * Selection to File" prompt.  This function is disabled, since
	 * it allows reading from or writing to files not specified on
	 * the command line. */
2268
	if (!ISSET(RESTRICTED) && !exiting && openfile->mark_set)
2269
2270
2271
	    msg = (append == PREPEND) ? _("Prepend Selection to File") :
		  (append == APPEND) ? _("Append Selection to File") :
		  _("Write Selection to File");
2272
	else
2273
#endif /* !NANO_TINY */
2274
	    msg = (append == PREPEND) ? _("File Name to Prepend to") :
2275
2276
		  (append == APPEND) ? _("File Name to Append to") :
		  _("File Name to Write");
2277

2278
2279
2280
2281
2282
2283
2284
2285
	/* If we're not exiting, and the TEMP_FILE flag is set, and
	 * the current file doesn't have a name, warn the user before
	 * prompting for a name.  If we are exiting, we've already
	 * warned the user just before the "Save modified buffer"
	 * prompt, so we don't need to do it again. */
	if (!exiting && openfile->filename[0] == '\0' && ISSET(TEMP_FILE))
	    no_current_file_name_warning();

2286
2287
2288
	/* If we're using restricted mode, the filename isn't blank,
	 * and we're at the "Write File" prompt, disable tab
	 * completion. */
2289
	i = do_prompt(!ISSET(RESTRICTED) ||
2290
2291
2292
2293
		openfile->filename[0] == '\0',
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
2294
		MWRITEFILE, ans,
2295
#ifndef DISABLE_HISTORIES
2296
2297
		NULL,
#endif
2298
		edit_refresh, "%s%s%s", msg,
2299
2300
#ifndef NANO_TINY
		formatstr, backupstr
2301
#else
2302
		"", ""
2303
2304
2305
#endif
		);

2306
2307
	/* 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
2308
	if (i < 0 || *answer == '\n') {
2309
	    statusbar(_("Cancelled"));
2310
	    retval = FALSE;
2311
2312
	    break;
	} else {
2313
2314
	    functionptrtype func = func_from_key(&i);

2315
	    ans = mallocstrcpy(ans, answer);
Chris Allegretta's avatar
Chris Allegretta committed
2316

2317
#ifndef DISABLE_BROWSER
2318
	    if (func == to_files_void) {
2319
		char *tmp = do_browse_from(answer);
2320

2321
2322
		if (tmp == NULL)
		    continue;
2323

2324
		/* We have a file now.  Indicate this. */
2325
2326
2327
		free(answer);
		answer = tmp;
	    } else
2328
#endif /* !DISABLE_BROWSER */
2329
#ifndef NANO_TINY
2330
	    if (func == dos_format_void) {
2331
2332
		openfile->fmt = (openfile->fmt == DOS_FILE) ? NIX_FILE :
			DOS_FILE;
2333
		continue;
2334
	    } else if (func == mac_format_void) {
2335
2336
		openfile->fmt = (openfile->fmt == MAC_FILE) ? NIX_FILE :
			MAC_FILE;
2337
		continue;
2338
	    } else if (func == backup_file_void) {
2339
2340
2341
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
2342
#endif /* !NANO_TINY */
2343
	    if (func == prepend_void) {
2344
		append = (append == PREPEND) ? OVERWRITE : PREPEND;
2345
		continue;
2346
	    } else if (func == append_void) {
2347
		append = (append == APPEND) ? OVERWRITE : APPEND;
2348
		continue;
2349
	    } else if (func == do_help_void) {
2350
		continue;
2351
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2352

Chris Allegretta's avatar
Chris Allegretta committed
2353
#ifdef DEBUG
2354
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
2355
#endif
2356

2357
#ifndef DISABLE_EXTRA
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2358
2359
2360
2361
2362
2363
2364
	    /* 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) &&
2365
			strcasecmp(answer, "zzy") == 0) {
2366
		do_credits();
2367
		did_credits = TRUE;
2368
		retval = FALSE;
2369
2370
		break;
	    }
2371
#endif
2372

2373
	    if (append == OVERWRITE) {
2374
		size_t answer_len = strlen(answer);
2375
		bool name_exists, do_warning;
2376
		char *full_answer, *full_filename;
2377
2378
		struct stat st;

2379
2380
2381
		/* Convert newlines to nulls, just before we get the
		 * full path. */
		sunder(answer);
2382

2383
2384
		full_answer = get_full_path(answer);
		full_filename = get_full_path(openfile->filename);
2385
2386
		name_exists = (stat((full_answer == NULL) ? answer :
			full_answer, &st) != -1);
2387
2388
2389
2390
2391
2392
		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);
2393

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2394
2395
2396
2397
		/* Convert nulls to newlines.  answer_len is the
		 * string's real length. */
		unsunder(answer, answer_len);

2398
2399
		free(full_filename);
		free(full_answer);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2400

2401
		if (do_warning) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2402
2403
2404
2405
2406
2407
2408
2409
		    /* 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;
2410

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2411
		    if (name_exists) {
2412
2413
2414
2415
			i = do_yesno_prompt(FALSE,
				_("File exists, OVERWRITE ? "));
			if (i == 0 || i == -1)
			    continue;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2416
		    } else
2417
#ifndef NANO_TINY
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2418
		    if (exiting || !openfile->mark_set)
2419
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2420
		    {
2421
2422
2423
2424
2425
			i = do_yesno_prompt(FALSE,
				_("Save file under DIFFERENT NAME ? "));
			if (i == 0 || i == -1)
			    continue;
		    }
2426
		}
2427
#ifndef NANO_TINY
2428
2429
2430
		/* Complain if the file exists, the name hasn't changed,
		 * and the stat information we had before does not match
		 * what we have now. */
2431
2432
2433
2434
		else if (name_exists && openfile->current_stat &&
			(openfile->current_stat->st_mtime < st.st_mtime ||
			openfile->current_stat->st_dev != st.st_dev ||
			openfile->current_stat->st_ino != st.st_ino)) {
2435
2436
2437
2438
2439
		    i = do_yesno_prompt(FALSE,
			_("File was modified since you opened it, continue saving ? "));
		    if (i == 0 || i == -1)
			continue;
		}
2440
#endif
2441

2442
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2443

2444
	    /* Convert newlines to nulls, just before we save the file. */
2445
2446
2447
	    sunder(answer);
	    align(&answer);

2448
	    /* Here's where we allow the selected text to be written to
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2449
2450
2451
	     * 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. */
2452
2453
2454
2455
	    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
2456
#endif
2457
		write_file(answer, NULL, FALSE, append, FALSE);
2458

2459
2460
	    break;
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2461
    }
2462
2463

    free(ans);
2464

2465
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
2466
2467
}

2468
2469
/* Write the current file to disk.  If the mark is on, write the current
 * marked selection to disk. */
2470
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
2471
{
2472
    do_writeout(FALSE);
2473
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
2474
}
Chris Allegretta's avatar
Chris Allegretta committed
2475

2476
2477
2478
2479
2480
2481
2482
2483
2484
2485
2486
#ifndef NANO_TINY
/* If it has a name, write the current file to disk without prompting. */
void do_savefile(void)
{
    if (openfile->filename[0] != '\0')
	write_file(openfile->filename, NULL, FALSE, OVERWRITE, FALSE);
    else
	do_writeout_void();
}
#endif

2487
2488
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
2489
char *real_dir_from_tilde(const char *buf)
2490
{
2491
    char *retval;
2492

2493
    assert(buf != NULL);
2494

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2495
    if (*buf == '~') {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2496
	size_t i = 1;
2497
	char *tilde_dir;
2498

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

2503
2504
2505
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
2506
	    tilde_dir = mallocstrcpy(NULL, homedir);
2507
2508
2509
	} else {
	    const struct passwd *userdata;

2510
2511
2512
	    tilde_dir = mallocstrncpy(NULL, buf, i + 1);
	    tilde_dir[i] = '\0';

2513
2514
	    do {
		userdata = getpwent();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2515
2516
	    } while (userdata != NULL && strcmp(userdata->pw_name,
		tilde_dir + 1) != 0);
2517
	    endpwent();
2518
	    if (userdata != NULL)
2519
		tilde_dir = mallocstrcpy(tilde_dir, userdata->pw_dir);
Chris Allegretta's avatar
Chris Allegretta committed
2520
	}
2521

2522
2523
	retval = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	sprintf(retval, "%s%s", tilde_dir, buf + i);
2524

2525
2526
2527
	free(tilde_dir);
    } else
	retval = mallocstrcpy(NULL, buf);
2528

2529
    return retval;
2530
2531
}

2532
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
2533
/* Our sort routine for file listings.  Sort alphabetically and
2534
 * case-insensitively, and sort directories before filenames. */
2535
2536
2537
2538
2539
2540
2541
2542
2543
2544
2545
2546
2547
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;

2548
2549
2550
    /* 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
Benno Schulenberg's avatar
Benno Schulenberg committed
2551
     * have to use multibyte strcasecmp() instead. */
2552
    return mbstrcasecmp(a, b);
2553
}
2554
2555
2556
2557
2558

/* Free the memory allocated for array, which should contain len
 * elements. */
void free_chararray(char **array, size_t len)
{
2559
2560
    if (array == NULL)
	return;
2561

2562
2563
2564
2565
    for (; len > 0; len--)
	free(array[len - 1]);
    free(array);
}
2566
2567
#endif

Chris Allegretta's avatar
Chris Allegretta committed
2568
#ifndef DISABLE_TABCOMP
2569
/* Is the given path a directory? */
2570
bool is_dir(const char *buf)
2571
{
2572
    char *dirptr;
2573
    struct stat fileinfo;
2574
2575
2576
    bool retval;

    assert(buf != NULL);
2577

2578
2579
    dirptr = real_dir_from_tilde(buf);

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

2582
    free(dirptr);
2583

2584
    return retval;
2585
}
Chris Allegretta's avatar
Chris Allegretta committed
2586

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2587
2588
2589
/* 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
2590
 * file, with the copyright years updated:
Chris Allegretta's avatar
Chris Allegretta committed
2591
2592
2593
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
2594
 * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007
Chris Allegretta's avatar
Chris Allegretta committed
2595
2596
2597
2598
2599
2600
2601
2602
 *      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.
2603
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
2604

2605
/* We consider the first buf_len characters of buf for ~username tab
2606
2607
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
2608
	size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2609
{
2610
2611
    char **matches = NULL;
    const struct passwd *userdata;
2612

2613
    assert(buf != NULL && num_matches != NULL && buf_len > 0);
2614

2615
    *num_matches = 0;
2616

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

2622
2623
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2624
2625
2626
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
2627
2628
#endif

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2629
2630
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2631
2632
2633
	    matches[*num_matches] =
		charalloc(strlen(userdata->pw_name) + 2);
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
2634
	    ++(*num_matches);
2635
	}
2636
2637
    }
    endpwent();
2638

2639
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2640
2641
}

2642
/* We consider the first buf_len characters of buf for filename tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2643
 * completion. */
2644
char **cwd_tab_completion(const char *buf, bool allow_files, size_t
2645
	*num_matches, size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2646
{
2647
    char *dirname = mallocstrcpy(NULL, buf), *filename;
2648
2649
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2650
    DIR *dir;
2651
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
2652

2653
    assert(dirname != NULL && num_matches != NULL);
2654

2655
    *num_matches = 0;
2656
    null_at(&dirname, buf_len);
2657
2658
2659
2660
2661
2662
2663
2664

    /* 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
2665
	*tmpdirname = '\0';
2666
2667
2668
	tmpdirname = dirname;
	dirname = real_dir_from_tilde(dirname);
	free(tmpdirname);
Chris Allegretta's avatar
Chris Allegretta committed
2669
    } else {
2670
2671
	filename = dirname;
	dirname = mallocstrcpy(NULL, "./");
Chris Allegretta's avatar
Chris Allegretta committed
2672
2673
    }

2674
    assert(dirname[strlen(dirname) - 1] == '/');
2675

Chris Allegretta's avatar
Chris Allegretta committed
2676
    dir = opendir(dirname);
2677

2678
    if (dir == NULL) {
2679
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
2680
	beep();
2681
2682
2683
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2684
    }
2685
2686
2687

    filenamelen = strlen(filename);

2688
    while ((nextdir = readdir(dir)) != NULL) {
2689
2690
	bool skip_match = FALSE;

Chris Allegretta's avatar
Chris Allegretta committed
2691
#ifdef DEBUG
2692
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
2693
#endif
2694
	/* See if this matches. */
2695
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
2696
2697
		(*filename == '.' || (strcmp(nextdir->d_name, ".") != 0 &&
		strcmp(nextdir->d_name, "..") != 0))) {
2698
2699
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2700

2701
2702
2703
2704
	    char *tmp = charalloc(strlen(dirname) +
		strlen(nextdir->d_name) + 1);
	    sprintf(tmp, "%s%s", dirname, nextdir->d_name);

2705
2706
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2707
2708
2709
2710
2711
2712
2713
	     * 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. */
2714
	    if (!allow_files && !is_dir(tmp))
2715
2716
2717
		skip_match = TRUE;

	    free(tmp);
2718

2719
	    if (skip_match)
2720
		continue;
2721

2722
2723
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2724
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
2725
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
2726
2727
	}
    }
2728

2729
2730
    closedir(dir);
    free(dirname);
2731
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
2732

2733
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2734
2735
}

2736
/* Do tab completion.  place refers to how much the statusbar cursor
2737
2738
 * position should be advanced.  refresh_func is the function we will
 * call to refresh the edit window. */
2739
char *input_tab(char *buf, bool allow_files, size_t *place, bool
2740
	*lastwastab, void (*refresh_func)(void), bool *list)
Chris Allegretta's avatar
Chris Allegretta committed
2741
{
2742
    size_t num_matches = 0, buf_len;
2743
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2744

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

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2748
    *list = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2749

2750
2751
    /* 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
2752
    if (*place > 0 && *buf == '~') {
2753
	const char *bob = strchr(buf, '/');
Chris Allegretta's avatar
Chris Allegretta committed
2754

2755
2756
2757
2758
	if (bob == NULL || bob >= buf + *place)
	    matches = username_tab_completion(buf, &num_matches,
		*place);
    }
2759

2760
2761
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
2762
2763
	matches = cwd_tab_completion(buf, allow_files, &num_matches,
		*place);
2764

2765
2766
2767
    buf_len = strlen(buf);

    if (num_matches == 0 || *place != buf_len)
2768
2769
2770
2771
	beep();
    else {
	size_t match, common_len = 0;
	char *mzero;
2772
2773
2774
	const char *lastslash = revstrstr(buf, "/", buf + *place);
	size_t lastslash_len = (lastslash == NULL) ? 0 :
		lastslash - buf + 1;
2775
2776
2777
	char *match1_mb = charalloc(mb_cur_max() + 1);
	char *match2_mb = charalloc(mb_cur_max() + 1);
	int match1_mb_len, match2_mb_len;
2778
2779
2780

	while (TRUE) {
	    for (match = 1; match < num_matches; match++) {
2781
2782
		/* Get the number of single-byte characters that all the
		 * matches have in common. */
2783
		match1_mb_len = parse_mbchar(matches[0] + common_len,
2784
			match1_mb, NULL);
2785
		match2_mb_len = parse_mbchar(matches[match] +
2786
			common_len, match2_mb, NULL);
2787
2788
2789
		match1_mb[match1_mb_len] = '\0';
		match2_mb[match2_mb_len] = '\0';
		if (strcmp(match1_mb, match2_mb) != 0)
2790
2791
		    break;
	    }
2792

2793
	    if (match < num_matches || matches[0][common_len] == '\0')
2794
		break;
2795

2796
	    common_len += parse_mbchar(buf + common_len, NULL, NULL);
2797
	}
2798

2799
2800
2801
	free(match1_mb);
	free(match2_mb);

2802
	mzero = charalloc(lastslash_len + common_len + 1);
2803
2804
2805

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

2807
	common_len += lastslash_len;
2808
	mzero[common_len] = '\0';
2809

2810
	assert(common_len >= *place);
2811

2812
	if (num_matches == 1 && is_dir(mzero)) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2813
	    mzero[common_len++] = '/';
2814

2815
2816
	    assert(common_len > *place);
	}
2817

2818
	if (num_matches > 1 && (common_len != *place || !*lastwastab))
2819
	    beep();
2820

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2821
	/* If there is more of a match to display on the statusbar, show
2822
	 * it.  We reset lastwastab to FALSE: it requires pressing Tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2823
	 * twice in succession with no statusbar changes to see a match
2824
2825
2826
	 * list. */
	if (common_len != *place) {
	    *lastwastab = FALSE;
2827
	    buf = charealloc(buf, common_len + buf_len - *place + 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2828
2829
	    charmove(buf + common_len, buf + *place, buf_len -
		*place + 1);
2830
	    strncpy(buf, mzero, common_len);
2831
	    *place = common_len;
2832
	} else if (!*lastwastab || num_matches < 2)
2833
2834
	    *lastwastab = TRUE;
	else {
2835
	    int longest_name = 0, ncols, editline = 0;
2836

2837
2838
2839
2840
2841
2842
2843
2844
	    /* 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
2845

2846
2847
		if (common_len > COLS - 1) {
		    longest_name = COLS - 1;
2848
2849
		    break;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2850

2851
2852
		if (common_len > longest_name)
		    longest_name = common_len;
2853
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2854

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

2857
	    /* Each column will be (longest_name + 2) columns wide, i.e.
2858
2859
	     * two spaces between columns, except that there will be
	     * only one space after the last column. */
2860
	    ncols = (COLS + 1) / (longest_name + 2);
2861

2862
2863
2864
2865
	    /* Blank the edit window, and print the matches out
	     * there. */
	    blank_edit();
	    wmove(edit, 0, 0);
2866

2867
2868
	    /* Disable el cursor. */
	    curs_set(0);
2869

2870
2871
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2872

2873
		wmove(edit, editline, (longest_name + 2) *
2874
			(match % ncols));
2875

2876
		if (match % ncols == 0 &&
2877
			editline == editwinrows - 1 &&
2878
			num_matches - match > ncols) {
2879
2880
2881
		    waddstr(edit, _("(more)"));
		    break;
		}
2882

2883
2884
2885
2886
		disp = display_string(matches[match], 0, longest_name,
			FALSE);
		waddstr(edit, disp);
		free(disp);
2887

2888
		if ((match + 1) % ncols == 0)
2889
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2890
	    }
2891

2892
	    wnoutrefresh(edit);
2893
	    *list = TRUE;
2894
2895
2896
	}

	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2897
2898
    }

2899
    free_chararray(matches, num_matches);
2900

2901
    /* Only refresh the edit window if we don't have a list of filename
2902
     * matches on it. */
2903
    if (!*list)
2904
	refresh_func();
2905
2906

    /* Enable el cursor. */
2907
    curs_set(1);
2908

2909
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2910
}
2911
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2912

2913
2914
/* Only print the last part of a path.  Isn't there a shell command for
 * this? */
2915
2916
const char *tail(const char *foo)
{
2917
    const char *tmp = strrchr(foo, '/');
2918

2919
2920
    if (tmp == NULL)
	tmp = foo;
2921
    else
2922
2923
2924
2925
2926
	tmp++;

    return tmp;
}

2927
2928
#ifndef DISABLE_HISTORIES
/* Return the constructed dirfile path, or NULL if we can't find the home
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2929
2930
 * directory.  The string is dynamically allocated, and should be
 * freed. */
2931
char *construct_filename(const char *str)
2932
{
2933
    char *newstr = NULL;
2934

2935
2936
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2937

2938
2939
2940
	newstr = charalloc(homelen + strlen(str) + 1);
	strcpy(newstr, homedir);
	strcpy(newstr + homelen, str);
Chris Allegretta's avatar
Chris Allegretta committed
2941
    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2942

2943
2944
2945
2946
2947
    return newstr;
}

char *histfilename(void)
{
2948
    return construct_filename("/.nano/search_history");
2949
2950
}

2951
2952
/* Construct the legacy history filename.
 * (Deprecate in 2.5, delete later.) */
2953
2954
2955
2956
2957
2958
2959
char *legacyhistfilename(void)
{
    return construct_filename("/.nano_history");
}

char *poshistfilename(void)
{
2960
    return construct_filename("/.nano/filepos_history");
2961
2962
2963
2964
}

void history_error(const char *msg, ...)
{
2965
    va_list ap;
2966
2967
2968
2969
2970
2971
2972
2973
2974
2975

    va_start(ap, msg);
    vfprintf(stderr, _(msg), ap);
    va_end(ap);

    fprintf(stderr, _("\nPress Enter to continue\n"));
    while (getchar() != '\n')
	;
}

2976
2977
2978
/* Now that we have more than one history file, let's just rely on a
 * .nano dir for this stuff.  Return 1 if the dir exists or was
 * successfully created, and return 0 otherwise. */
2979
2980
int check_dotnano(void)
{
2981
    int ret = 1;
2982
2983
2984
2985
2986
    struct stat dirstat;
    char *nanodir = construct_filename("/.nano");

    if (stat(nanodir, &dirstat) == -1) {
	if (mkdir(nanodir, S_IRWXU | S_IRWXG | S_IRWXO) == -1) {
2987
2988
2989
	    history_error(N_("Unable to create directory %s: %s\n"
			     "It is required for saving/loading search history or cursor positions.\n"),
				nanodir, strerror(errno));
2990
	    ret = 0;
2991
2992
	}
    } else if (!S_ISDIR(dirstat.st_mode)) {
2993
2994
2995
	history_error(N_("Path %s is not a directory and needs to be.\n"
			 "Nano will be unable to load or save search history or cursor positions.\n"),
				nanodir);
2996
	ret = 0;
2997
    }
2998
2999
3000

    free(nanodir);
    return ret;
3001
3002
}

3003
/* Load the search and replace histories from ~/.nano/search_history. */
3004
3005
3006
void load_history(void)
{
    char *nanohist = histfilename();
3007
3008
3009
    char *legacyhist = legacyhistfilename();
    struct stat hstat;

3010
    if (stat(legacyhist, &hstat) != -1 && stat(nanohist, &hstat) == -1) {
3011
	if (rename(legacyhist, nanohist) == -1)
3012
3013
3014
	    history_error(N_("Detected a legacy nano history file (%s) which I tried to move\n"
			     "to the preferred location (%s) but encountered an error: %s"),
				legacyhist, nanohist, strerror(errno));
3015
	else
3016
3017
3018
	    history_error(N_("Detected a legacy nano history file (%s) which I moved\n"
			     "to the preferred location (%s)\n(see the nano FAQ about this change)"),
				legacyhist, nanohist);
3019
3020
    }

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

3025
	if (hist == NULL) {
3026
	    if (errno != ENOENT) {
3027
3028
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
3029
		history_error(N_("Error reading %s: %s"), nanohist,
3030
			strerror(errno));
3031
	    }
3032
	} else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
3033
3034
3035
	    /* 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. */
3036
	    filestruct **history = &search_history;
3037
	    char *line = NULL;
3038
	    size_t buf_len = 0;
3039
3040
	    ssize_t read;

3041
	    while ((read = getline(&line, &buf_len, hist)) >= 0) {
3042
3043
3044
3045
3046
3047
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
		if (read > 0) {
		    unsunder(line, read);
3048
3049
		    update_history(history, line);
		} else
3050
3051
		    history = &replace_history;
	    }
3052

3053
	    fclose(hist);
3054
	    free(line);
3055
	}
3056
	free(nanohist);
3057
	free(legacyhist);
3058
3059
3060
    }
}

3061
3062
3063
/* 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. */
3064
bool writehist(FILE *hist, filestruct *h)
3065
{
3066
    filestruct *p;
3067

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

3073
	sunder(p->data);
3074

3075
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
3076
3077
		putc('\n', hist) == EOF)
	    return FALSE;
3078
    }
3079

3080
    return TRUE;
3081
3082
}

3083
/* Save the search and replace histories to ~/.nano/search_history. */
3084
3085
void save_history(void)
{
3086
    char *nanohist;
3087

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
3088
    /* Don't save unchanged or empty histories. */
3089
    if (!history_has_changed() || (searchbot->lineno == 1 &&
3090
		replacebot->lineno == 1))
3091
3092
	return;

3093
3094
3095
3096
    nanohist = histfilename();

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

3098
	if (hist == NULL)
3099
	    history_error(N_("Error writing %s: %s"), nanohist,
3100
		strerror(errno));
3101
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
3102
3103
	    /* Make sure no one else can read from or write to the
	     * history file. */
3104
	    chmod(nanohist, S_IRUSR | S_IWUSR);
3105

3106
	    if (!writehist(hist, searchage) || !writehist(hist, replaceage))
3107
		history_error(N_("Error writing %s: %s"), nanohist,
3108
			strerror(errno));
3109

3110
3111
	    fclose(hist);
	}
3112

3113
3114
3115
	free(nanohist);
    }
}
3116

3117
/* Save the recorded last file positions to ~/.nano/filepos_history. */
3118
3119
3120
void save_poshistory(void)
{
    char *poshist;
3121
3122
    char *statusstr = NULL;
    poshiststruct *posptr;
3123
3124
3125
3126
3127
3128
3129
3130
3131
3132
3133
3134
3135
3136

    poshist = poshistfilename();

    if (poshist != NULL) {
	FILE *hist = fopen(poshist, "wb");

	if (hist == NULL)
	    history_error(N_("Error writing %s: %s"), poshist,
		strerror(errno));
	else {
	    /* Make sure no one else can read from or write to the
	     * history file. */
	    chmod(poshist, S_IRUSR | S_IWUSR);

3137
	    for (posptr = poshistory; posptr != NULL; posptr = posptr->next) {
3138
		statusstr = charalloc(strlen(posptr->filename) + 2 * sizeof(ssize_t) + 4);
3139
3140
		sprintf(statusstr, "%s %ld %ld\n", posptr->filename, (long)posptr->lineno,
			(long)posptr->xno);
3141
3142
3143
		if (fwrite(statusstr, sizeof(char), strlen(statusstr), hist) < strlen(statusstr))
		    history_error(N_("Error writing %s: %s"), poshist,
			strerror(errno));
3144
		free(statusstr);
3145
3146
3147
3148
3149
3150
3151
	    }
	    fclose(hist);
	}
	free(poshist);
    }
}

3152
3153
/* Update the recorded last file positions, given a filename, a line
 * and a column.  If no entry is found, add a new one at the end. */
3154
3155
void update_poshistory(char *filename, ssize_t lineno, ssize_t xpos)
{
3156
3157
    poshiststruct *posptr, *posprev = NULL;
    char *fullpath = get_full_path(filename);
3158
3159

    if (fullpath == NULL)
3160
	return;
3161
3162

    for (posptr = poshistory; posptr != NULL; posptr = posptr->next) {
3163
	if (!strcmp(posptr->filename, fullpath)) {
3164
3165
	    posptr->lineno = lineno;
	    posptr->xno = xpos;
3166
	    free(fullpath);
3167
3168
	    return;
	}
3169
3170
3171
3172
	posprev = posptr;
    }

    /* Didn't find it, make a new node yo! */
3173
    posptr = (poshiststruct *)nmalloc(sizeof(poshiststruct));
3174
3175
3176
3177
3178
3179
3180
3181
3182
3183
3184
3185
3186
    posptr->filename = mallocstrcpy(NULL, fullpath);
    posptr->lineno = lineno;
    posptr->xno = xpos;
    posptr->next = NULL;

    if (!poshistory)
	poshistory = posptr;
    else
	posprev->next = posptr;

    free(fullpath);
}

3187
3188
3189
/* Check the recorded last file positions to see if the given file
 * matches an existing entry.  If so, return 1 and set line and column
 * to the retrieved values.  Otherwise, return 0. */
3190
3191
3192
3193
3194
3195
int check_poshistory(const char *file, ssize_t *line, ssize_t *column)
{
    poshiststruct *posptr;
    char *fullpath = get_full_path(file);

    if (fullpath == NULL)
3196
	return 0;
3197
3198
3199
3200
3201

    for (posptr = poshistory; posptr != NULL; posptr = posptr->next) {
	if (!strcmp(posptr->filename, fullpath)) {
	    *line = posptr->lineno;
	    *column = posptr->xno;
3202
	    free(fullpath);
3203
3204
3205
	    return 1;
	}
    }
3206
    free(fullpath);
3207
3208
3209
    return 0;
}

3210
/* Load the recorded file positions from ~/.nano/filepos_history. */
3211
3212
3213
3214
3215
3216
3217
3218
3219
3220
3221
void load_poshistory(void)
{
    char *nanohist = poshistfilename();

    /* Assume do_rcfile() has reported a missing home directory. */
    if (nanohist != NULL) {
	FILE *hist = fopen(nanohist, "rb");

	if (hist == NULL) {
	    if (errno != ENOENT) {
		/* Don't save history when we quit. */
3222
		UNSET(POS_HISTORY);
3223
3224
3225
3226
3227
3228
3229
3230
3231
		history_error(N_("Error reading %s: %s"), nanohist,
			strerror(errno));
	    }
	} else {
	    char *line = NULL, *lineptr, *xptr;
	    size_t buf_len = 0;
	    ssize_t read, lineno, xno;
	    poshiststruct *posptr;

3232
3233
	    /* Read and parse each line, and put the data into the
	     * positions history structure. */
3234
3235
3236
3237
3238
	    while ((read = getline(&line, &buf_len, hist)) >= 0) {
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
3239
		if (read > 0)
3240
3241
3242
3243
3244
3245
		    unsunder(line, read);
		lineptr = parse_next_word(line);
		xptr = parse_next_word(lineptr);
		lineno = atoi(lineptr);
		xno = atoi(xptr);
		if (poshistory == NULL) {
3246
		    poshistory = (poshiststruct *)nmalloc(sizeof(poshiststruct));
3247
3248
3249
3250
3251
		    poshistory->filename = mallocstrcpy(NULL, line);
		    poshistory->lineno = lineno;
		    poshistory->xno = xno;
		    poshistory->next = NULL;
		} else {
3252
		    for (posptr = poshistory; posptr->next != NULL; posptr = posptr->next)
3253
			;
3254
		    posptr->next = (poshiststruct *)nmalloc(sizeof(poshiststruct));
3255
3256
3257
3258
3259
3260
3261
3262
3263
3264
3265
3266
3267
		    posptr->next->filename = mallocstrcpy(NULL, line);
		    posptr->next->lineno = lineno;
		    posptr->next->xno = xno;
		    posptr->next->next = NULL;
		}
	    }

	    fclose(hist);
	    free(line);
	}
	free(nanohist);
    }
}
3268
#endif /* !DISABLE_HISTORIES */