files.c 93.6 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/**************************************************************************
2
 *   files.c  --  This file is part of GNU nano.                          *
Chris Allegretta's avatar
Chris Allegretta committed
3
 *                                                                        *
4
 *   Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007,  *
5
 *   2008, 2009, 2010, 2011, 2013, 2014 Free Software Foundation, Inc.    *
6
7
 *   Copyright (C) 2015, 2016 Benno Schulenberg                           *
 *                                                                        *
8
9
10
11
 *   GNU nano is free software: you can redistribute it and/or modify     *
 *   it under the terms of the GNU General Public License as published    *
 *   by the Free Software Foundation, either version 3 of the License,    *
 *   or (at your option) any later version.                               *
Chris Allegretta's avatar
Chris Allegretta committed
12
 *                                                                        *
13
14
15
16
 *   GNU nano 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
17
18
 *                                                                        *
 *   You should have received a copy of the GNU General Public License    *
19
 *   along with this program.  If not, see http://www.gnu.org/licenses/.  *
Chris Allegretta's avatar
Chris Allegretta committed
20
21
22
 *                                                                        *
 **************************************************************************/

23
#include "proto.h"
24

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

35
36
#define LOCKBUFSIZE 8192

37
38
/* Verify that the containing directory of the given filename exists. */
bool has_valid_path(const char *filename)
39
{
40
41
    char *namecopy = mallocstrcpy(NULL, filename);
    char *parentdir = dirname(namecopy);
42
    struct stat parentinfo;
43
    bool validity = FALSE;
44

45
46
    if (stat(parentdir, &parentinfo) == -1) {
	if (errno == ENOENT)
47
	    statusline(ALERT, _("Directory '%s' does not exist"), parentdir);
48
	else
49
	    statusline(ALERT, _("Path '%s': %s"), parentdir, strerror(errno));
50
    } else if (!S_ISDIR(parentinfo.st_mode))
51
	statusline(ALERT, _("Path '%s' is not a directory"), parentdir);
52
    else if (access(parentdir, X_OK) == -1)
53
	statusline(ALERT, _("Path '%s' is not accessible"), parentdir);
54
55
    else if (ISSET(LOCKING) && access(parentdir, W_OK) == -1)
	statusline(MILD, _("Directory '%s' is not writable"), parentdir);
56
57
    else
	validity = TRUE;
58

59
    free(namecopy);
60

61
    return validity;
62
63
}

64
/* Add an entry to the circular list of openfile structs. */
65
void make_new_buffer(void)
66
{
67
68
    if (openfile == NULL) {
	openfile = make_new_opennode();
69

70
	/* Make the first open file the only element in the list. */
71
72
	openfile->prev = openfile;
	openfile->next = openfile;
73
    } else {
74
75
76
77
78
79
80
81
82
83
84
85
	openfilestruct *newnode = make_new_opennode();

	/* Add the new open file after the current one in the list. */
	newnode->prev = openfile;
	newnode->next = openfile->next;
	openfile->next->prev = newnode;
	openfile->next = newnode;

	/* Make the new file the current one. */
	openfile = newnode;

	/* There is more than one file open: show Close in help lines. */
86
	exitfunc->desc = close_tag;
Chris Allegretta's avatar
Chris Allegretta committed
87
    }
88

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

92
    initialize_buffer_text();
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
93

94
95
    openfile->current_x = 0;
    openfile->placewewant = 0;
96
    openfile->current_y = 0;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
97

98
    openfile->modified = FALSE;
99
#ifndef NANO_TINY
100
    openfile->mark_set = FALSE;
101
102
    openfile->mark_begin = NULL;
    openfile->mark_begin_x = 0;
103
    openfile->kind_of_mark = SOFTMARK;
104

105
    openfile->fmt = NIX_FILE;
106

107
108
    openfile->undotop = NULL;
    openfile->current_undo = NULL;
109
    openfile->last_action = OTHER;
110
111

    openfile->current_stat = NULL;
112
    openfile->lock_filename = NULL;
113
#endif
114
#ifndef DISABLE_COLOR
115
    openfile->syntax = NULL;
116
117
    openfile->colorstrings = NULL;
#endif
118
}
Chris Allegretta's avatar
Chris Allegretta committed
119

120
/* Initialize the text of the current openfile struct. */
121
void initialize_buffer_text(void)
122
{
123
    assert(openfile != NULL);
124

125
126
    openfile->fileage = make_new_node(NULL);
    openfile->fileage->data = mallocstrcpy(NULL, "");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
127

128
129
130
131
132
    openfile->filebot = openfile->fileage;
    openfile->edittop = openfile->fileage;
    openfile->current = openfile->fileage;

    openfile->totsize = 0;
133
134
}

135
136
137
138
139
140
141
142
143
144
145
/* Mark the current file as modified if it isn't already, and then
 * update the titlebar to display the file's new status. */
void set_modified(void)
{
    if (openfile->modified)
	return;

    openfile->modified = TRUE;
    titlebar(NULL);

#ifndef NANO_TINY
146
    if (!ISSET(LOCKING) || openfile->filename[0] == '\0')
147
148
149
	return;

    if (openfile->lock_filename == NULL) {
150
	/* TRANSLATORS: Keep the next ten messages at most 76 characters. */
151
	statusline(ALERT, _("Warning: Modifying a file which is not locked,"
152
153
154
155
156
157
158
159
160
			" check directory permission?"));
    } else {
	char *fullname = get_full_path(openfile->filename);
	write_lockfile(openfile->lock_filename, fullname, TRUE);
	free(fullname);
    }
#endif
}

161
#ifndef NANO_TINY
162
163
164
165
166
167
168
169
170
/* 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
 *
171
 * Returns 1 on success, and 0 on failure (but continue anyway). */
172
173
174
175
176
177
178
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;
179
    struct stat fileinfo;
180
181
    char *lockdata = charalloc(1024);
    char myhostname[32];
182
183
    size_t lockdatalen = 1024;
    size_t wroteamt;
184

185
    mypid = getpid();
186
    myuid = geteuid();
187
188

    /* First run things that might fail before blowing away the old state. */
189
    if ((mypwuid = getpwuid(myuid)) == NULL) {
190
	statusline(MILD, _("Couldn't determine my identity for lock file "
191
				"(getpwuid() failed)"));
192
	goto free_the_data;
193
194
195
    }

    if (gethostname(myhostname, 31) < 0) {
196
197
198
	if (errno == ENAMETOOLONG)
	    myhostname[31] = '\0';
	else {
199
	    statusline(MILD, _("Couldn't determine hostname for lock file: %s"),
200
			strerror(errno));
201
	    goto free_the_data;
202
	}
203
204
    }

205
    /* If the lockfile exists, try to delete it. */
206
207
    if (stat(lockfilename, &fileinfo) != -1)
	if (delete_lockfile(lockfilename) < 0)
208
	    goto free_the_data;
209
210

    if (ISSET(INSECURE_BACKUP))
211
	cflags = O_WRONLY | O_CREAT | O_APPEND;
212
    else
213
	cflags = O_WRONLY | O_CREAT | O_EXCL | O_APPEND;
214

215
    /* Try to create the lockfile. */
216
    fd = open(lockfilename, cflags,
217
		S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
218
    if (fd < 0) {
219
	statusline(MILD, _("Error writing lock file %s: %s"),
220
			lockfilename, strerror(errno));
221
	goto free_the_data;
222
    }
223

224
    /* Try to associate a stream with the now open lockfile. */
225
226
    filestream = fdopen(fd, "wb");

227
    if (filestream == NULL) {
228
	statusline(MILD, _("Error writing lock file %s: %s"), lockfilename,
229
			strerror(errno));
230
	goto free_the_data;
231
232
    }

233
    /* This is the lock data we will store:
234
235
236
237
     *
     * byte 0        - 0x62
     * byte 1        - 0x30
     * bytes 2-12    - program name which created the lock
238
     * bytes 24-27   - PID (little endian) of creator process
239
240
241
242
243
244
245
246
247
248
     * 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. */
249
    memset(lockdata, 0, lockdatalen);
250
251
252
    lockdata[0] = 0x62;
    lockdata[1] = 0x30;
    lockdata[24] = mypid % 256;
253
254
255
    lockdata[25] = (mypid / 256) % 256;
    lockdata[26] = (mypid / (256 * 256)) % 256;
    lockdata[27] = mypid / (256 * 256 * 256);
256
    snprintf(&lockdata[2], 11, "nano %s", VERSION);
257
258
    strncpy(&lockdata[28], mypwuid->pw_name, 16);
    strncpy(&lockdata[68], myhostname, 31);
259
260
    if (origfilename != NULL)
	strncpy(&lockdata[108], origfilename, 768);
261
    if (modified == TRUE)
262
	lockdata[1007] = 0x55;
263
264
265

    wroteamt = fwrite(lockdata, sizeof(char), lockdatalen, filestream);
    if (wroteamt < lockdatalen) {
266
	statusline(MILD, _("Error writing lock file %s: %s"),
267
			lockfilename, ferror(filestream));
268
	goto free_the_data;
269
270
271
    }

#ifdef DEBUG
272
    fprintf(stderr, "In write_lockfile(), write successful (wrote %lu bytes)\n", (unsigned long)wroteamt);
273
#endif
274
275

    if (fclose(filestream) == EOF) {
276
	statusline(MILD, _("Error writing lock file %s: %s"),
277
			lockfilename, strerror(errno));
278
	goto free_the_data;
279
280
    }

281
    openfile->lock_filename = (char *) lockfilename;
282

Benno Schulenberg's avatar
Benno Schulenberg committed
283
    free(lockdata);
284
    return 1;
Benno Schulenberg's avatar
Benno Schulenberg committed
285

286
  free_the_data:
Benno Schulenberg's avatar
Benno Schulenberg committed
287
    free(lockdata);
288
    return 0;
289
290
}

291
/* Delete the lockfile.  Return -1 if unsuccessful, and 1 otherwise. */
292
293
294
int delete_lockfile(const char *lockfilename)
{
    if (unlink(lockfilename) < 0 && errno != ENOENT) {
295
	statusline(MILD, _("Error deleting lock file %s: %s"), lockfilename,
296
			strerror(errno));
297
	return -1;
298
299
300
301
    }
    return 1;
}

302
303
/* Deal with lockfiles.  Return -1 on refusing to override the lockfile,
 * and 1 on successfully creating it; 0 means we were not successful in
304
 * creating the lockfile but we should continue to load the file. */
305
306
int do_lockfile(const char *filename)
{
307
308
    char *namecopy = (char *) mallocstrcpy(NULL, filename);
    char *secondcopy = (char *) mallocstrcpy(NULL, filename);
Benno Schulenberg's avatar
Benno Schulenberg committed
309
    size_t locknamesize = strlen(filename) + strlen(locking_prefix)
310
		+ strlen(locking_suffix) + 3;
Benno Schulenberg's avatar
Benno Schulenberg committed
311
    char *lockfilename = charalloc(locknamesize);
312
    static char lockprog[11], lockuser[17];
313
    struct stat fileinfo;
314
    int lockfd, lockpid, retval = -1;
315

316
317
318
    snprintf(lockfilename, locknamesize, "%s/%s%s%s", dirname(namecopy),
		locking_prefix, basename(secondcopy), locking_suffix);
    free(secondcopy);
319
320
#ifdef DEBUG
    fprintf(stderr, "lock file name is %s\n", lockfilename);
321
#endif
322
    if (stat(lockfilename, &fileinfo) != -1) {
323
324
	size_t readtot = 0;
	size_t readamt = 0;
325
	char *lockbuf, *question, *pidstring, *postedname, *promptstr;
326
	int room, response;
327

328
	if ((lockfd = open(lockfilename, O_RDONLY)) < 0) {
329
	    statusline(MILD, _("Error opening lock file %s: %s"),
330
			lockfilename, strerror(errno));
331
	    goto free_the_name;
332
	}
333

334
	lockbuf = charalloc(LOCKBUFSIZE);
335
	do {
336
	    readamt = read(lockfd, &lockbuf[readtot], LOCKBUFSIZE - readtot);
337
	    readtot += readamt;
338
	} while (readamt > 0 && readtot < LOCKBUFSIZE);
339

340
341
	close(lockfd);

342
	if (readtot < 48) {
343
	    statusline(MILD, _("Error reading lock file %s: "
344
			"Not enough data read"), lockfilename);
345
346
	    free(lockbuf);
	    goto free_the_name;
347
	}
348

349
	strncpy(lockprog, &lockbuf[2], 10);
350
351
	lockpid = (((unsigned char)lockbuf[27] * 256 + (unsigned char)lockbuf[26]) * 256 +
			(unsigned char)lockbuf[25]) * 256 + (unsigned char)lockbuf[24];
352
	strncpy(lockuser, &lockbuf[28], 16);
353
354
	free(lockbuf);

355
356
357
	pidstring = charalloc(11);
	sprintf (pidstring, "%u", (unsigned int)lockpid);

358
#ifdef DEBUG
359
360
361
	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);
362
#endif
363
	/* TRANSLATORS: The second %s is the name of the user, the third that of the editor. */
364
365
366
	question = _("File %s is being edited (by %s with %s, PID %s); continue?");
	room = COLS - strlenpt(question) + 7 - strlenpt(lockuser) -
				strlenpt(lockprog) - strlenpt(pidstring);
367
	if (room < 4)
368
369
370
371
372
373
374
375
376
377
	    postedname = mallocstrcpy(NULL, "_");
	else if (room < strlenpt(filename)) {
	    char *fragment = display_string(filename,
				strlenpt(filename) - room + 3, room, FALSE);
	    postedname = charalloc(strlen(fragment) + 4);
	    strcpy(postedname, "...");
	    strcat(postedname, fragment);
	    free(fragment);
	} else
	    postedname = mallocstrcpy(NULL, filename);
378

379
	/* Allow extra space for username (14), program name (8), PID (8),
380
	 * and terminating \0 (1), minus the %s (2) for the file name. */
381
382
	promptstr = charalloc(strlen(question) + 29 + strlen(postedname));
	sprintf(promptstr, question, postedname, lockuser, lockprog, pidstring);
383
	free(postedname);
384
	free(pidstring);
385

386
	response = do_yesno_prompt(FALSE, promptstr);
387
388
	free(promptstr);

389
	if (response < 1) {
390
	    blank_statusbar();
391
	    goto free_the_name;
392
	}
393
    }
394
395
396
397

    retval = write_lockfile(lockfilename, filename, FALSE);

  free_the_name:
398
    free(namecopy);
399
400
401
402
    if (retval < 1)
	free(lockfilename);

    return retval;
403
}
404
405
406
407
408
409
410
411
412
413
414
415
416
417

/* Perform a stat call on the given filename, allocating a stat struct
 * if necessary.  On success, *pstat points to the stat's result.  On
 * failure, *pstat is freed and made NULL. */
void stat_with_alloc(const char *filename, struct stat **pstat)
{
    if (*pstat == NULL)
	*pstat = (struct stat *)nmalloc(sizeof(struct stat));

    if (stat(filename, *pstat) != 0) {
	free(*pstat);
	*pstat = NULL;
    }
}
418
#endif /* !NANO_TINY */
419

420
421
422
/* This does one of three things.  If the filename is "", just create a new
 * empty buffer.  Otherwise, read the given file into the existing buffer,
 * or into a new buffer when MULTIBUFFER is set or there is no buffer yet. */
423
bool open_buffer(const char *filename, bool undoable)
424
{
425
426
    bool new_buffer = (openfile == NULL || ISSET(MULTIBUFFER));
	/* Whether we load into the current buffer or a new one. */
427
428
    char *realname;
	/* The filename after tilde expansion. */
429
430
431
432
    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. */
433

434
435
    assert(filename != NULL);

436
437
#ifndef DISABLE_OPERATINGDIR
    if (check_operating_dir(filename, FALSE)) {
438
	statusline(ALERT, _("Can't insert file from outside of %s"),
439
				full_operating_dir);
440
	return FALSE;
441
442
    }
#endif
443

444
445
    realname = real_dir_from_tilde(filename);

446
447
    /* When the specified filename is not empty, and the corresponding
     * file exists, verify that it is a normal file. */
448
449
450
    if (strcmp(filename, "") != 0) {
	struct stat fileinfo;

451
	if (stat(realname, &fileinfo) == 0 && !S_ISREG(fileinfo.st_mode)) {
452
	    if (S_ISDIR(fileinfo.st_mode))
453
		statusline(ALERT, _("\"%s\" is a directory"), realname);
454
	    else
455
		statusline(ALERT, _("\"%s\" is not a normal file"), realname);
456
	    free(realname);
457
	    return FALSE;
458
459
460
	}
    }

461
    /* If we're going to load into a new buffer, first create the new
462
     * buffer and (if possible) lock the corresponding file. */
463
    if (new_buffer) {
464
	make_new_buffer();
465

466
	if (has_valid_path(realname)) {
467
#ifndef NANO_TINY
468
	    if (ISSET(LOCKING) && filename[0] != '\0') {
469
		/* When not overriding an existing lock, discard the buffer. */
470
		if (do_lockfile(realname) < 0) {
471
#ifndef DISABLE_MULTIBUFFER
472
		    close_buffer();
473
#endif
474
475
		    free(realname);
		    return FALSE;
476
		}
477
	    }
478
#endif /* !NANO_TINY */
479
	}
480
481
    }

482
483
484
    /* 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)) ?
485
		open_file(realname, new_buffer, FALSE, &f) : -2;
486

487
488
    /* If we have a file, and we're loading into a new buffer, update
     * the filename. */
489
    if (rc != -1 && new_buffer)
490
	openfile->filename = mallocstrcpy(openfile->filename, realname);
491

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
492
493
    /* If we have a non-new file, read it in.  Then, if the buffer has
     * no stat, update the stat, if applicable. */
494
    if (rc > 0) {
495
	read_file(f, rc, realname, undoable, new_buffer);
496
#ifndef NANO_TINY
497
	if (openfile->current_stat == NULL)
498
	    stat_with_alloc(realname, &openfile->current_stat);
499
#endif
500
501
    }

502
    /* If we have a file, and we're loading into a new buffer, move back
503
504
     * to the beginning of the first line of the buffer. */
    if (rc != -1 && new_buffer) {
505
	openfile->current = openfile->fileage;
506
507
508
	openfile->current_x = 0;
	openfile->placewewant = 0;
    }
509

510
#ifndef DISABLE_COLOR
511
512
    /* If we're loading into a new buffer, update the colors to account
     * for it, if applicable. */
513
514
515
    if (new_buffer)
	color_update();
#endif
516
    free(realname);
517
    return TRUE;
518
}
519

520
#ifndef DISABLE_SPELLER
521
522
/* Open the specified file, and if that succeeds, blow away the text of
 * the current buffer and read the file contents into its place. */
523
524
525
void replace_buffer(const char *filename)
{
    FILE *f;
526
    int descriptor;
527

528
    assert(filename != NULL && filename[0] != '\0');
529

530
    /* Open the file quietly. */
531
532
533
534
535
    descriptor = open_file(filename, FALSE, TRUE, &f);

    /* If opening failed, forget it. */
    if (descriptor < 0)
	return;
536
537
538
539
540

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

541
542
    /* Insert the processed file into its place. */
    read_file(f, descriptor, filename, FALSE, TRUE);
543
544
545

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

549
/* Update the screen to account for the current buffer. */
550
void display_buffer(void)
551
{
552
    /* Update the titlebar, since the filename may have changed. */
553
    titlebar(NULL);
554

555
#ifndef DISABLE_COLOR
556
    /* Make sure we're using the buffer's associated colors. */
557
    color_init();
558
559
560
561
562
563

    /* If there are multiline coloring regexes, and there is no
     * multiline cache data yet, precalculate it now. */
    if (openfile->syntax && openfile->syntax->nmultis > 0 &&
		openfile->fileage->multidata == NULL)
	precalc_multicolorinfo();
564
565
#endif

566
    refresh_needed = TRUE;
Chris Allegretta's avatar
Chris Allegretta committed
567
568
}

569
#ifndef DISABLE_MULTIBUFFER
570
571
/* Switch to a neighbouring file buffer; to the next if to_next is TRUE;
 * otherwise, to the previous one. */
572
void switch_to_prevnext_buffer(bool to_next)
Chris Allegretta's avatar
Chris Allegretta committed
573
{
574
    assert(openfile != NULL);
575

576
    /* If only one file buffer is open, say so and get out. */
577
    if (openfile == openfile->next) {
578
	statusbar(_("No more open file buffers"));
579
	return;
Chris Allegretta's avatar
Chris Allegretta committed
580
    }
581

582
583
    /* Switch to the next or previous file buffer. */
    openfile = to_next ? openfile->next : openfile->prev;
584
585

#ifdef DEBUG
586
    fprintf(stderr, "filename is %s\n", openfile->filename);
587
588
#endif

589
    /* Update the screen to account for the current buffer. */
590
    display_buffer();
591

592
    /* Indicate the switch on the statusbar. */
593
    statusline(HUSH, _("Switched to %s"),
594
595
		((openfile->filename[0] == '\0') ?
		_("New Buffer") : openfile->filename));
596
597

#ifdef DEBUG
598
    dump_filestruct(openfile->current);
599
#endif
Chris Allegretta's avatar
Chris Allegretta committed
600
601
}

602
/* Switch to the previous entry in the openfile filebuffer. */
603
void switch_to_prev_buffer_void(void)
604
{
605
    switch_to_prevnext_buffer(FALSE);
606
}
607

608
/* Switch to the next entry in the openfile filebuffer. */
609
void switch_to_next_buffer_void(void)
610
{
611
    switch_to_prevnext_buffer(TRUE);
612
}
613

614
615
616
617
/* Delete an entry from the circular list of open files, and switch to the
 * one after it.  Return TRUE on success, and FALSE if there are no other
 * open buffers. */
bool close_buffer(void)
618
{
619
    assert(openfile != NULL);
620

621
    /* If only one file buffer is open, get out. */
622
    if (openfile == openfile->next)
623
	return FALSE;
624

625
#ifndef DISABLE_HISTORIES
626
627
628
    if (ISSET(POS_HISTORY))
	update_poshistory(openfile->filename,
			openfile->current->lineno, xplustabs() + 1);
629
#endif
630

631
    /* Switch to the next file buffer. */
632
    switch_to_prevnext_buffer(TRUE);
633

634
    /* Close the file buffer we had open before. */
635
    unlink_opennode(openfile->prev);
636

637
638
639
    /* If only one buffer is open now, show Exit in the help lines. */
    if (openfile == openfile->next)
	exitfunc->desc = exit_tag;
640

641
    return TRUE;
642
}
643
#endif /* !DISABLE_MULTIBUFFER */
644

645
646
647
648
/* Do a quick permissions check by verifying whether the file is appendable.
 * 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. */
649
650
651
652
653
654
int is_file_writable(const char *filename)
{
    struct stat fileinfo, fileinfo2;
    int fd;
    FILE *f;
    char *full_filename;
655
    bool result = TRUE;
656
657
658
659

    if (ISSET(VIEW_MODE))
	return TRUE;

660
    assert(filename != NULL);
661
662
663
664

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

665
666
    /* Okay, if we can't stat the absolute path due to some component's
     * permissions, just try the relative one. */
667
668
669
    if (full_filename == NULL ||
		(stat(full_filename, &fileinfo) == -1 && stat(filename, &fileinfo2) != -1))
	full_filename = mallocstrcpy(NULL, filename);
670
671

    if ((fd = open(full_filename, O_WRONLY | O_CREAT | O_APPEND, S_IRUSR |
672
673
		S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH)) == -1 ||
		(f = fdopen(fd, "a")) == NULL)
674
	result = FALSE;
675
    else
676
	fclose(f);
677

678
    close(fd);
679
    free(full_filename);
680
681

    return result;
682
683
}

684
685
686
/* Make a new line of text from the given buf, which is of length buf_len.
 * Then attach this line after prevnode. */
filestruct *read_line(char *buf, size_t buf_len, filestruct *prevnode)
687
{
688
    filestruct *freshline = (filestruct *)nmalloc(sizeof(filestruct));
689

690
    /* Convert nulls to newlines.  buf_len is the string's real length. */
691
    unsunder(buf, buf_len);
692
    buf[buf_len] = '\0';
693

694
    assert(openfile->fileage != NULL && strlen(buf) == buf_len);
695

696
#ifndef NANO_TINY
697
698
699
    /* If file conversion isn't disabled, strip a '\r' from the line. */
    if (buf_len > 0 && buf[buf_len - 1] == '\r' && !ISSET(NO_CONVERT))
	buf[buf_len - 1] = '\0';
700
#endif
701

702
703
    freshline->data = mallocstrcpy(NULL, buf);

704
#ifndef DISABLE_COLOR
705
    freshline->multidata = NULL;
706
707
#endif

708
    freshline->prev = prevnode;
709
710

    if (prevnode == NULL) {
711
	/* Special case: we're inserting into the first line. */
712
713
714
	freshline->next = openfile->fileage;
	openfile->fileage = freshline;
	freshline->lineno = 1;
715
    } else {
716
717
718
	prevnode->next = freshline;
	freshline->next = NULL;
	freshline->lineno = prevnode->lineno + 1;
719
720
    }

721
    return freshline;
722
}
723

724
/* Read an open file into the current buffer.  f should be set to the
725
 * open file, and filename should be set to the name of the file.
726
727
728
 * 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. */
729
void read_file(FILE *f, int fd, const char *filename, bool undoable, bool checkwritable)
730
731
732
733
734
735
736
737
738
739
740
{
    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 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. */
741
742
    filestruct *fileptr = openfile->current->prev;
	/* The line after which to start inserting. */
743
744
745
    int input_int;
	/* The current value we read from the file, whether an input
	 * character or EOF. */
746
747
    bool writable = TRUE;
	/* Is the file writable (if we care) */
748
#ifndef NANO_TINY
749
750
    int format = 0;
	/* 0 = *nix, 1 = DOS, 2 = Mac, 3 = both DOS and Mac. */
751
#endif
752

753
    buf = charalloc(bufx);
754

755
756
757
758
759
#ifndef NANO_TINY
    if (undoable)
	add_undo(INSERT);
#endif

760
    /* Read the entire file into the filestruct. */
761
762
763
764
765
766
    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') {
767
#ifndef NANO_TINY
768
769
770
771
	    /* 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! */
772
	    if ((num_lines == 0 || format != 0) && !ISSET(NO_CONVERT) &&
773
			len > 0 && buf[len - 1] == '\r') {
774
775
776
		if (format == 0 || format == 2)
		    format++;
	    }
777
#endif
778
	    /* Read in the line properly. */
779
	    fileptr = read_line(buf, len, fileptr);
780
	    num_lines++;
781

782
	    /* Reset the length in preparation for the next line. */
783
	    len = 0;
784
#ifndef NANO_TINY
785
786
787
	/* 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! */
788
789
	} else if ((num_lines == 0 || format != 0) && !ISSET(NO_CONVERT) &&
			len > 0 && buf[len - 1] == '\r') {
790
791
792
793
794
	    /* 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;
795

796
	    /* Read in the line properly. */
797
	    fileptr = read_line(buf, len, fileptr);
798
	    num_lines++;
799
800
801

	    /* Store the character after the \r as the first character
	     * of the next line. */
802
	    buf[0] = input;
803
	    len = 1;
804
#endif
805
	} else {
806
807
808
	    /* Store the character. */
	    buf[len] = input;

809
810
	    /* Keep track of the total length of the line.  It might have
	     * nulls in it, so we can't just use strlen() later. */
811
	    len++;
812

813
814
	    /* If needed, increase the buffer size, MAX_BUF_SIZE characters at
	     * a time.  Don't bother decreasing it; it is freed at the end. */
815
	    if (len == bufx) {
816
817
		bufx += MAX_BUF_SIZE;
		buf = charealloc(buf, bufx);
818
	    }
819
820
821
822
823
824
825
	}
    }

    /* Perhaps this could use some better handling. */
    if (ferror(f))
	nperror(filename);
    fclose(f);
826
    if (fd > 0 && checkwritable) {
827
	close(fd);
828
829
	writable = is_file_writable(filename);
    }
830

831
832
    /* Did we not get a newline and still have stuff to do? */
    if (len > 0) {
833
#ifndef NANO_TINY
834
	/* If file conversion isn't disabled and the last character in
835
	 * This file is '\r', set format to Mac if we currently think
836
837
	 * the file is a *nix file, or to both DOS and Mac if we
	 * currently think the file is a DOS file. */
838
	if (buf[len - 1] == '\r' && !ISSET(NO_CONVERT) && format < 2)
839
	    format += 2;
840
#endif
841
	/* Read in the last line properly. */
842
	fileptr = read_line(buf, len, fileptr);
843
844
	num_lines++;
    }
845

846
    free(buf);
847

848
849
    /* Attach the file we got to the filestruct.  If we got a file of
     * zero bytes, don't do anything. */
850
    if (num_lines > 0) {
851
852
853
	/* If the file we got doesn't end in a newline (nor in a Mac return),
	 * tack its last line onto the beginning of the line at current. */
	if (len > 0 && (input != '\r' || ISSET(NO_CONVERT))) {
854
	    filestruct *dropline = fileptr;
855
856
857
858
859
860
861
862
863
	    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
864
865
	    /* Tack the text at fileptr onto the beginning of the text
	     * at current. */
866
867
868
869
	    openfile->current->data = charealloc(openfile->current->data,
						len + current_len + 1);
	    charmove(openfile->current->data + len, openfile->current->data,
			current_len + 1);
870
871
	    strncpy(openfile->current->data, fileptr->data, len);

872
873
	    /* Step back one line, and blow away the unterminated line,
	     * since its text has been copied into current. */
874
	    fileptr = fileptr->prev;
875
	    delete_node(dropline);
876
877
	}

878
879
880
881
882
883
	if (fileptr == NULL)
	    /* After inserting a single unterminated line at the top,
	     * readjust the top-of-file pointer. */
	    openfile->fileage = openfile->current;
	else {
	    /* Attach the line at current after the line at fileptr. */
884
885
886
887
	    fileptr->next = openfile->current;
	    openfile->current->prev = fileptr;
	}

888
	/* Renumber, starting with the last line of the file we inserted. */
889
	renumber(openfile->current);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
890
    }
891

892
    openfile->totsize += get_totsize(openfile->fileage, openfile->filebot);
893

894
    /* If the NO_NEWLINES flag isn't set, and text has been added to
895
     * the magicline (i.e. a file that doesn't end in a newline has been
896
897
898
     * 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') {
899
900
901
902
903
904
905
906
907
	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();

908
    if (!writable)
909
	statusline(ALERT, _("File '%s' is unwritable"), filename);
910
#ifndef NANO_TINY
911
    else if (format == 3) {
912
913
914
915
	/* TRANSLATORS: Keep the next four messages at most 78 characters. */
	statusline(HUSH, 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);
916
    } else if (format == 2) {
917
	openfile->fmt = MAC_FILE;
918
919
920
	statusline(HUSH, P_("Read %lu line (Converted from Mac format)",
			"Read %lu lines (Converted from Mac format)",
			(unsigned long)num_lines), (unsigned long)num_lines);
921
    } else if (format == 1) {
922
	openfile->fmt = DOS_FILE;
923
924
925
	statusline(HUSH, P_("Read %lu line (Converted from DOS format)",
			"Read %lu lines (Converted from DOS format)",
			(unsigned long)num_lines), (unsigned long)num_lines);
926
    }
927
#endif
928
    else
929
930
	statusline(HUSH, P_("Read %lu line", "Read %lu lines",
			(unsigned long)num_lines), (unsigned long)num_lines);
931

932
933
934
    if (num_lines < editwinrows)
	focusing = FALSE;

935
#ifndef NANO_TINY
936
937
938
    if (undoable)
	update_undo(INSERT);

939
940
    if (ISSET(MAKE_IT_UNIX))
	openfile->fmt = NIX_FILE;
941
#endif
942
}
Chris Allegretta's avatar
Chris Allegretta committed
943

944
945
/* Open the file with the given name.  If the file does not exist, display
 * "New File" if newfie is TRUE, and say "File not found" otherwise.
946
 * Return -2 if we say "New File", -1 if the file isn't opened, and the
947
 * obtained fd otherwise.  *f is set to the opened file. */
948
int open_file(const char *filename, bool newfie, bool quiet, FILE **f)
949
{
950
    struct stat fileinfo, fileinfo2;
951
    int fd;
952
    char *full_filename;
953

954
    assert(filename != NULL && f != NULL);
955

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

959
    /* Okay, if we can't stat the path due to a component's
960
     * permissions, just try the relative one. */
961
    if (full_filename == NULL || (stat(full_filename, &fileinfo) == -1 &&
962
					stat(filename, &fileinfo2) != -1))
963
	full_filename = mallocstrcpy(full_filename, filename);
964
965

    if (stat(full_filename, &fileinfo) == -1) {
966
	if (newfie) {
967
	    if (!quiet)
968
		statusbar(_("New File"));
969
	    free(full_filename);
970
971
	    return -2;
	}
972

973
	statusline(ALERT, _("File \"%s\" not found"), filename);
974
	free(full_filename);
975
976
	return -1;
    }
977

978
979
980
    /* Don't open directories, character files, or block files. */
    if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
				S_ISBLK(fileinfo.st_mode)) {
981
	statusline(ALERT, S_ISDIR(fileinfo.st_mode) ?
982
983
			_("\"%s\" is a directory") :
			_("\"%s\" is a device file"), filename);
984
	free(full_filename);
985
	return -1;
986
987
988
989
990
991
992
993
994
    }

    /* Try opening the file. */
    fd = open(full_filename, O_RDONLY);

    if (fd == -1)
	statusline(ALERT, _("Error reading %s: %s"), filename, strerror(errno));
    else {
	/* The file is A-OK.  Associate a stream with it. */
995
	*f = fdopen(fd, "rb");
996

997
	if (*f == NULL) {
998
	    statusline(ALERT, _("Error reading %s: %s"), filename, strerror(errno));
999
1000
1001
	    close(fd);
	} else
	    statusbar(_("Reading File"));
1002
    }
1003

1004
1005
    free(full_filename);

1006
1007
1008
    return fd;
}

1009
1010
1011
1012
1013
/* 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)
1014
{
1015
1016
    unsigned long i = 0;
    char *buf;
1017
    size_t wholenamelen;
1018

1019
    assert(name != NULL && suffix != NULL);
1020

1021
    wholenamelen = strlen(name) + strlen(suffix);
1022

1023
1024
1025
    /* Reserve space for: the name plus the suffix plus a dot plus
     * possibly five digits plus a null byte. */
    buf = charalloc(wholenamelen + 7);
1026
    sprintf(buf, "%s%s", name, suffix);
1027

1028
1029
    while (TRUE) {
	struct stat fs;
Chris Allegretta's avatar
Chris Allegretta committed
1030

1031
1032
	if (stat(buf, &fs) == -1)
	    return buf;
1033
1034
1035

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

1038
	sprintf(buf + wholenamelen, ".%lu", i);
1039
    }
Chris Allegretta's avatar
Chris Allegretta committed
1040

1041
1042
1043
    /* We get here only if there is no possible save file.  Blank out
     * the filename to indicate this. */
    null_at(&buf, 0);
1044

1045
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
1046
1047
}

1048
1049
1050
/* Insert a file into the current buffer, or into a new buffer when
 * the MULTIBUFFER flag is set. */
void do_insertfile(void)
1051
1052
1053
{
    int i;
    const char *msg;
1054
    char *given = mallocstrcpy(NULL, "");
1055
	/* The last answer the user typed at the statusbar prompt. */
1056
#ifndef NANO_TINY
1057
    bool execute = FALSE, right_side_up = FALSE, single_line = FALSE;
1058
#endif
1059

1060
    while (TRUE) {
1061
#ifndef NANO_TINY
1062
	if (execute) {
1063
#ifndef DISABLE_MULTIBUFFER
1064
1065
1066
	    if (ISSET(MULTIBUFFER))
		msg = _("Command to execute in new buffer [from %s] ");
	    else
1067
#endif
1068
		msg = _("Command to execute [from %s] ");
1069
1070
1071
	} else
#endif /* NANO_TINY */
	{
1072
#ifndef DISABLE_MULTIBUFFER
1073
1074
1075
	    if (ISSET(MULTIBUFFER))
		msg = _("File to insert into new buffer [from %s] ");
	    else
1076
#endif
1077
		msg = _("File to insert [from %s] ");
1078
	}
1079

1080
1081
	present_path = mallocstrcpy(present_path, "./");

1082
	i = do_prompt(TRUE,
1083
1084
1085
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
1086
#ifndef NANO_TINY
1087
		execute ? MEXTCMD :
1088
#endif
1089
		MINSERTFILE, given,
1090
#ifndef DISABLE_HISTORIES
1091
		NULL,
1092
#endif
1093
		edit_refresh, msg,
1094
#ifndef DISABLE_OPERATINGDIR
1095
1096
		operating_dir != NULL && strcmp(operating_dir, ".") != 0 ?
		operating_dir :
1097
1098
#endif
		"./");
1099

1100
	/* If we're in multibuffer mode and the filename or command is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1101
	 * blank, open a new buffer instead of canceling.  If the
1102
1103
	 * filename or command begins with a newline (i.e. an encoded
	 * null), treat it as though it's blank. */
1104
	if (i == -1 || (!ISSET(MULTIBUFFER) && (i == -2 || *answer == '\n'))) {
1105
1106
1107
	    statusbar(_("Cancelled"));
	    break;
	} else {
1108
1109
1110
1111
1112
	    filestruct *edittop_save = openfile->edittop;
	    ssize_t was_current_lineno = openfile->current->lineno;
	    size_t was_current_x = openfile->current_x;
	    ssize_t was_current_y = openfile->current_y;
	    bool current_was_at_top = FALSE;
1113
	    size_t pww_save = openfile->placewewant;
1114
#if !defined(NANO_TINY) || !defined(DISABLE_BROWSER)
1115
	    functionptrtype func = func_from_key(&i);
1116
#endif
1117
	    given = mallocstrcpy(given, answer);
1118

1119
#ifndef NANO_TINY
1120
#ifndef DISABLE_MULTIBUFFER
1121
	    if (func == new_buffer_void) {
1122
		/* Don't allow toggling when in view mode. */
1123
1124
		if (!ISSET(VIEW_MODE))
		    TOGGLE(MULTIBUFFER);
1125
1126
		else
		    beep();
1127
		continue;
1128
	    }
1129
#endif
1130
	    if (func == flip_execute_void) {
1131
1132
1133
		execute = !execute;
		continue;
	    }
1134
#endif /* !NANO_TINY */
1135

1136
#ifndef DISABLE_BROWSER
1137
	    if (func == to_files_void) {
1138
		char *chosen = do_browse_from(answer);
1139

1140
		/* If no file was chosen, go back to the prompt. */
1141
		if (chosen == NULL)
1142
		    continue;
1143

1144
		free(answer);
1145
		answer = chosen;
1146
1147
1148
		i = 0;
	    }
#endif
1149
1150
	    /* If we don't have a file yet, go back to the prompt. */
	    if (i != 0 && (!ISSET(MULTIBUFFER) || i != -2))
1151
		continue;
1152

1153
#ifndef NANO_TINY
1154
1155
1156
1157
1158
	    /* 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;
1159

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

1163
1164
		single_line = (top == bot);
	    }
1165
#endif
1166
1167
1168
1169
	    /* When not inserting into a new buffer, partition the filestruct
	     * so that it contains no text and hence looks like a new buffer,
	     * and remember whether the current line is the first on screen. */
	    if (!ISSET(MULTIBUFFER)) {
1170
1171
1172
		filepart = partition_filestruct(openfile->current,
			openfile->current_x, openfile->current,
			openfile->current_x);
1173
		current_was_at_top = (openfile->edittop == openfile->fileage);
1174
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1175

1176
	    /* Convert newlines to nulls in the given filename. */
1177
1178
	    sunder(answer);

1179
#ifndef NANO_TINY
1180
	    if (execute) {
1181
#ifndef DISABLE_MULTIBUFFER
1182
		/* When in multibuffer mode, first open a blank buffer. */
1183
		if (ISSET(MULTIBUFFER))
1184
		    open_buffer("", FALSE);
1185
1186
#endif
		/* Save the command's output in the current buffer. */
1187
		execute_command(answer);
1188

1189
#ifndef DISABLE_MULTIBUFFER
1190
		/* If this is a new buffer, put the cursor at the top. */
1191
1192
1193
1194
		if (ISSET(MULTIBUFFER)) {
		    openfile->current = openfile->fileage;
		    openfile->current_x = 0;
		    openfile->placewewant = 0;
1195
1196

		    set_modified();
1197
		}
1198
#endif
1199
	    } else
1200
#endif /* !NANO_TINY */
1201
	    {
1202
1203
		/* Make sure the path to the file specified in answer is
		 * tilde-expanded. */
1204
		answer = free_and_assign(answer, real_dir_from_tilde(answer));
1205

1206
		/* Save the file specified in answer in the current buffer. */
1207
		open_buffer(answer, TRUE);
1208
	    }
1209

1210
#ifndef DISABLE_MULTIBUFFER
1211
	    if (ISSET(MULTIBUFFER)) {
1212
1213
1214
#ifndef DISABLE_HISTORIES
		if (ISSET(POS_HISTORY)) {
		    ssize_t priorline, priorcol;
1215
#ifndef NANO_TINY
1216
		    if (!execute)
1217
#endif
1218
1219
1220
1221
		    if (check_poshistory(answer, &priorline, &priorcol))
			do_gotolinecolumn(priorline, priorcol, FALSE, FALSE);
		}
#endif /* !DISABLE_HISTORIES */
1222
		/* Update stuff to account for the current buffer. */
1223
		display_buffer();
1224
	    } else
1225
#endif /* !DISABLE_MULTIBUFFER */
1226
	    {
1227
		filestruct *top_save = openfile->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
1228

1229
1230
1231
		/* If we were at the top of the edit window before, change the
		 * saved value of edittop to the start of inserted stuff. */
		if (current_was_at_top)
1232
		    edittop_save = openfile->fileage;
1233
1234

		/* Update the current x-coordinate to account for the
1235
		 * number of characters inserted on the current line.
1236
1237
1238
		 * If the mark was positioned after the cursor and on the
		 * same line, adjust the mark's coordinates to compensate
		 * for the change in this line. */
1239
		openfile->current_x = strlen(openfile->filebot->data);
1240
1241
1242
1243
		if (openfile->fileage == openfile->filebot) {
#ifndef NANO_TINY
		    if (openfile->mark_set) {
			openfile->mark_begin = openfile->current;
1244
			if (!right_side_up)
1245
			    openfile->mark_begin_x += openfile->current_x;
1246
1247
		    }
#endif
1248
		    openfile->current_x += was_current_x;
1249
		}
1250
#ifndef NANO_TINY
1251
1252
1253
1254
		else if (openfile->mark_set) {
		    if (!right_side_up) {
			if (single_line) {
			    openfile->mark_begin = openfile->current;
1255
			    openfile->mark_begin_x -= was_current_x;
1256
			} else
1257
			    openfile->mark_begin_x -= openfile->current_x;
1258
1259
		    }
		}
1260
#endif
1261
1262
		/* Update the current y-coordinate to account for the
		 * number of lines inserted. */
1263
		openfile->current_y += was_current_y;
1264

1265
1266
1267
1268
		/* 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. */
1269
		unpartition_filestruct(&filepart);
1270

1271
1272
1273
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
1274

1275
		/* Restore the old edittop. */
1276
		openfile->edittop = edittop_save;
1277

1278
1279
1280
		/* Restore the old place we want. */
		openfile->placewewant = pww_save;

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

1286
		refresh_needed = TRUE;
1287
	    }
1288

1289
1290
1291
	    break;
	}
    }
1292
1293

    free(given);
1294
1295
}

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

1306
#ifndef DISABLE_MULTIBUFFER
1307
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
1308
	statusbar(_("Key invalid in non-multibuffer mode"));
1309
1310
    else
#endif
1311
	do_insertfile();
1312
1313
}

1314
/* When passed "[relative path]" or "[relative path][filename]" in
1315
 * origpath, return "[full path]" or "[full path][filename]" on success,
1316
1317
1318
1319
 * 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. */
1320
char *get_full_path(const char *origpath)
1321
{
1322
    int attempts = 0;
1323
	/* How often we've tried climbing back up the tree. */
1324
    struct stat fileinfo;
1325
    char *currentdir, *d_here, *d_there, *d_there_file = NULL;
1326
1327
    const char *last_slash;
    bool path_only;
1328

1329
    if (origpath == NULL)
1330
	return NULL;
1331

1332
    /* Get the current directory.  If it doesn't exist, back up and try
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1333
1334
     * again until we get a directory that does, and use that as the
     * current directory. */
1335
1336
    currentdir = charalloc(PATH_MAX + 1);
    d_here = getcwd(currentdir, PATH_MAX + 1);
1337

1338
1339
    while (d_here == NULL && attempts < 20) {
	IGNORE_CALL_RESULT(chdir(".."));
Benno Schulenberg's avatar
Benno Schulenberg committed
1340
	d_here = getcwd(currentdir, PATH_MAX + 1);
1341
	attempts++;
1342
1343
1344
1345
    }

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

1348
1349
	/* If the current directory isn't "/", tack a slash onto the end
	 * of it. */
1350
	if (strcmp(d_here, "/") != 0) {
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1351
	    d_here = charealloc(d_here, strlen(d_here) + 2);
1352
1353
	    strcat(d_here, "/");
	}
1354
    /* Otherwise, set d_here to "". */
1355
    } else {
1356
	d_here = mallocstrcpy(NULL, "");
1357
1358
	free(currentdir);
    }
1359

1360
    d_there = real_dir_from_tilde(origpath);
1361

1362
1363
1364
    /* 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. */
1365
    path_only = (stat(d_there, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode));
1366

1367
1368
1369
    /* If path_only is TRUE, make sure d_there ends in a slash. */
    if (path_only) {
	size_t d_there_len = strlen(d_there);
1370

1371
1372
1373
	if (d_there[d_there_len - 1] != '/') {
	    d_there = charealloc(d_there, d_there_len + 2);
	    strcat(d_there, "/");
1374
	}
1375
    }
1376

1377
1378
    /* Search for the last slash in d_there. */
    last_slash = strrchr(d_there, '/');
1379

1380
1381
1382
1383
    /* 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);
1384

1385
1386
1387
1388
1389
1390
1391
1392
	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);

1393
	/* Remove the filename portion of the answer from d_there. */
1394
1395
	null_at(&d_there, last_slash - d_there + 1);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1396
	/* Go to the path specified in d_there. */
1397
1398
1399
	if (chdir(d_there) == -1) {
	    free(d_there);
	    d_there = NULL;
1400
	} else {
1401
1402
1403
	    free(d_there);

	    /* Get the full path. */
1404
1405
	    currentdir = charalloc(PATH_MAX + 1);
	    d_there = getcwd(currentdir, PATH_MAX + 1);
1406
1407
1408
1409
1410
1411
1412
1413
1414
1415
1416

	    /* 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, "/");
		}
1417
1418
	    /* Otherwise, make sure that we return NULL. */
	    } else {
1419
		path_only = TRUE;
1420
1421
		free(currentdir);
	    }
1422
1423
1424

	    /* Finally, go back to the path specified in d_here,
	     * where we were before.  We don't check for a chdir()
1425
	     * error, since we can do nothing if we get one. */
1426
	    IGNORE_CALL_RESULT(chdir(d_here));
1427
	}
1428
1429
1430

	/* Free d_here, since we're done using it. */
	free(d_here);
1431
    }
1432

1433
1434
1435
1436
1437
1438
1439
    /* 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) +
1440
		strlen(d_there_file) + 1);
1441
1442
	strcat(d_there, d_there_file);
    }
1443

1444
    /* Free d_there_file, since we're done using it. */
1445
    free(d_there_file);
1446

1447
    return d_there;
1448
}
1449

1450
/* Return the full version of path, as returned by get_full_path().  On
1451
 * error, or if path doesn't reference a directory, or if the directory
1452
 * isn't writable, return NULL. */
1453
char *check_writable_directory(const char *path)
1454
{
1455
1456
    char *full_path = get_full_path(path);

1457
    if (full_path == NULL)
1458
	return NULL;
1459

1460
    /* If we can't write to path or path isn't a directory, return NULL. */
1461
    if (access(full_path, W_OK) != 0 ||
1462
		full_path[strlen(full_path) - 1] != '/') {
1463
	free(full_path);
1464
	return NULL;
1465
    }
1466
1467
1468
1469

    return full_path;
}

1470
1471
/* This function calls mkstemp(($TMPDIR|P_tmpdir|/tmp/)"nano.XXXXXX").
 * On success, it returns the malloc()ed filename and corresponding FILE
1472
 * stream, opened in "r+b" mode.  On error, it returns NULL for the
1473
1474
 * filename and leaves the FILE stream unchanged. */
char *safe_tempfile(FILE **f)
1475
{
1476
    char *full_tempdir = NULL;
1477
1478
1479
    const char *tmpdir_env;
    int fd;
    mode_t original_umask = 0;
1480

1481
1482
    assert(f != NULL);

1483
1484
1485
    /* 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. */
1486
    tmpdir_env = getenv("TMPDIR");
1487
    if (tmpdir_env != NULL)
1488
	full_tempdir = check_writable_directory(tmpdir_env);
1489

1490
1491
    /* If $TMPDIR is unset, empty, or not a writable directory, and
     * full_tempdir is NULL, try P_tmpdir instead. */
1492
    if (full_tempdir == NULL)
1493
	full_tempdir = check_writable_directory(P_tmpdir);
1494

1495
1496
    /* if P_tmpdir is NULL, use /tmp. */
    if (full_tempdir == NULL)
1497
	full_tempdir = mallocstrcpy(NULL, "/tmp/");
1498

David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
1499
    full_tempdir = charealloc(full_tempdir, strlen(full_tempdir) + 12);
1500
    strcat(full_tempdir, "nano.XXXXXX");
1501

1502
1503
1504
1505
1506
1507
1508
    original_umask = umask(0);
    umask(S_IRWXG | S_IRWXO);

    fd = mkstemp(full_tempdir);

    if (fd != -1)
	*f = fdopen(fd, "r+b");
1509
1510
1511
    else {
	free(full_tempdir);
	full_tempdir = NULL;
1512
    }
1513

1514
1515
    umask(original_umask);

1516
    return full_tempdir;
1517
}
1518
1519

#ifndef DISABLE_OPERATINGDIR
1520
1521
1522
1523
1524
/* Initialize full_operating_dir based on operating_dir. */
void init_operating_dir(void)
{
    assert(full_operating_dir == NULL);

1525
    if (operating_dir == NULL)
1526
	return;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1527

1528
1529
    full_operating_dir = get_full_path(operating_dir);

1530
1531
1532
    /* If the operating directory is inaccessible, fail. */
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1)
	die("Invalid operating directory\n");
1533
1534
}

1535
1536
1537
1538
1539
/* 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)
1540
{
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1541
1542
1543
1544
    /* 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. */
1545

1546
    char *fullpath;
1547
    bool retval = FALSE;
1548
    const char *whereami1, *whereami2 = NULL;
1549

1550
    /* If no operating directory is set, don't bother doing anything. */
1551
    if (operating_dir == NULL)
1552
	return FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1553

1554
    assert(full_operating_dir != NULL);
1555
1556

    fullpath = get_full_path(currpath);
1557

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1558
1559
1560
    /* 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
1561
     * non-existent directory as being outside the operating directory,
1562
     * so we return FALSE.  If allow_tabcomp is TRUE, then currpath
1563
1564
     * exists, but is not executable.  So we say it isn't in the
     * operating directory. */
1565
    if (fullpath == NULL)
1566
	return allow_tabcomp;
1567
1568
1569
1570
1571

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

1572
    /* If both searches failed, we're outside the operating directory.
1573
     * Otherwise, check the search results.  If the full operating
1574
1575
1576
     * 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. */
1577
    if (whereami1 != fullpath && whereami2 != full_operating_dir)
1578
1579
	retval = TRUE;
    free(fullpath);
1580
1581

    /* Otherwise, we're still inside it. */
1582
    return retval;
1583
}
1584
1585
#endif

1586
#ifndef NANO_TINY
1587
1588
/* Although this sucks, it sucks less than having a single 'my system is
 * messed up and I'm blanket allowing insecure file writing operations'. */
1589
1590
int prompt_failed_backupwrite(const char *filename)
{
1591
    static int response;
1592
    static char *prevfile = NULL; /* What was the last file we were
1593
1594
				   * passed so we don't keep asking
				   * this?  Though maybe we should... */
1595
    if (prevfile == NULL || strcmp(filename, prevfile)) {
1596
1597
	response = do_yesno_prompt(FALSE, _("Failed to write backup file; "
			"continue saving? (Say N if unsure.) "));
1598
1599
	prevfile = mallocstrcpy(prevfile, filename);
    }
1600
1601

    return response;
1602
1603
}

1604
1605
1606
1607
1608
1609
1610
1611
1612
1613
1614
1615
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 ||
1616
		full_backup_dir[strlen(full_backup_dir) - 1] != '/') {
1617
1618
1619
1620
1621
1622
1623
1624
	free(full_backup_dir);
	free(backup_dir);
	backup_dir = NULL;
    } else {
	free(backup_dir);
	backup_dir = full_backup_dir;
    }
}
1625
#endif /* !NANO_TINY */
1626

1627
/* Read from inn, write to out.  We assume inn is opened for reading,
1628
1629
 * and out for writing.  We return 0 on success, -1 on read error, or -2
 * on write error. */
1630
1631
int copy_file(FILE *inn, FILE *out)
{
1632
    int retval = 0;
1633
    char buf[BUFSIZ];
1634
1635
    size_t charsread;

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

1638
1639
1640
1641
1642
1643
1644
1645
1646
1647
1648
    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
1649

1650
1651
1652
1653
    if (fclose(inn) == EOF)
	retval = -1;
    if (fclose(out) == EOF)
	retval = -2;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1654

1655
1656
1657
    return retval;
}

1658
1659
/* 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
1660
 * ourselves.  If tmp is TRUE, we set the umask to disallow anyone else
1661
1662
 * 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
1663
 *
1664
 * tmp means we are writing a temporary file in a secure fashion.  We
1665
 * use it when spell checking or dumping the file on an error.  If
1666
1667
 * method is APPEND, it means we are appending instead of overwriting.
 * If method is PREPEND, it means we are prepending instead of
1668
 * overwriting.  If nonamechange is TRUE, we don't change the current
1669
1670
 * filename.  nonamechange is irrelevant when appending or prepending,
 * or when writing a temporary file.
1671
 *
1672
 * Return TRUE on success or FALSE on error. */
1673
1674
bool write_file(const char *name, FILE *f_open, bool tmp,
	kind_of_writing_type method, bool nonamechange)
Chris Allegretta's avatar
Chris Allegretta committed
1675
{
1676
    bool retval = FALSE;
1677
	/* Instead of returning in this function, you should always
1678
	 * set retval and then goto cleanup_and_exit. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1679
    size_t lineswritten = 0;
1680
    const filestruct *fileptr = openfile->fileage;
1681
    int fd;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1682
	/* The file descriptor we use. */
1683
    mode_t original_umask = 0;
1684
	/* Our umask, from when nano started. */
1685
#ifndef NANO_TINY
1686
    bool realexists;
1687
	/* The result of stat().  TRUE if the file exists, FALSE
1688
	 * otherwise.  If name is a link that points nowhere, realexists
1689
	 * is FALSE. */
1690
#endif
1691
1692
1693
    struct stat st;
	/* The status fields filled in by stat(). */
    char *realname;
1694
	/* The filename after tilde expansion. */
1695
    FILE *f = f_open;
1696
1697
	/* The actual file, realname, we are writing to. */
    char *tempname = NULL;
1698
	/* The name of the temporary file we write to on prepend. */
Chris Allegretta's avatar
Chris Allegretta committed
1699

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1700
    if (*name == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1701
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1702

1703
1704
    if (!tmp)
	titlebar(NULL);
1705

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1706
    realname = real_dir_from_tilde(name);
1707

1708
#ifndef DISABLE_OPERATINGDIR
1709
    /* If we're writing a temporary file, we're probably going outside
1710
     * the operating directory, so skip the operating directory test. */
1711
    if (!tmp && check_operating_dir(realname, FALSE)) {
1712
	statusline(ALERT, _("Can't write outside of %s"), full_operating_dir);
1713
	goto cleanup_and_exit;
1714
1715
1716
    }
#endif

1717
    /* If the temp file exists and isn't already open, give up. */
1718
    if (tmp && (lstat(realname, &st) != -1) && f_open == NULL)
1719
	goto cleanup_and_exit;
1720

1721
#ifndef NANO_TINY
1722
    /* Check whether the file (at the end of the symlink) exists. */
1723
    realexists = (stat(realname, &st) != -1);
1724

1725
1726
1727
1728
    /* 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. */
1729
1730
    if (openfile->current_stat == NULL && !tmp && realexists)
	stat_with_alloc(realname, &openfile->current_stat);
1731

1732
    /* We backup only if the backup toggle is set, the file isn't
1733
1734
1735
1736
     * 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. */
1737
    if (ISSET(BACKUP_FILE) && !tmp && realexists && openfile->current_stat &&
1738
		(method != OVERWRITE || openfile->mark_set ||
1739
		openfile->current_stat->st_mtime == st.st_mtime)) {
1740
	int backup_fd;
1741
	FILE *backup_file;
1742
	char *backupname;
1743
	struct utimbuf filetime;
1744
	int copy_status;
1745
	int backup_cflags;
1746

1747
	/* Save the original file's access and modification times. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1748
1749
	filetime.actime = openfile->current_stat->st_atime;
	filetime.modtime = openfile->current_stat->st_mtime;
1750

1751
1752
1753
	if (f_open == NULL) {
	    /* Open the original file to copy to the backup. */
	    f = fopen(realname, "rb");
1754

1755
	    if (f == NULL) {
1756
		statusline(ALERT, _("Error reading %s: %s"), realname,
1757
			strerror(errno));
1758
1759
1760
1761
		/* 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;
1762
	    }
1763
1764
	}

1765
	/* If backup_dir is set, we set backupname to
1766
1767
1768
1769
	 * 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]. */
1770
	if (backup_dir != NULL) {
1771
	    char *backuptemp = get_full_path(realname);
1772

1773
	    if (backuptemp == NULL)
1774
1775
1776
1777
1778
1779
		/* 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~. */
1780
		backuptemp = mallocstrcpy(NULL, tail(realname));
1781
	    else {
1782
1783
		size_t i = 0;

1784
1785
1786
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1787
1788
1789
		}
	    }

1790
	    backupname = charalloc(strlen(backup_dir) + strlen(backuptemp) + 1);
1791
1792
1793
	    sprintf(backupname, "%s%s", backup_dir, backuptemp);
	    free(backuptemp);
	    backuptemp = get_next_filename(backupname, "~");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1794
	    if (*backuptemp == '\0') {
1795
1796
		statusline(HUSH, _("Error writing backup file %s: %s"),
			backupname, _("Too many backup files?"));
1797
1798
		free(backuptemp);
		free(backupname);
1799
1800
1801
1802
1803
		/* 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! */
1804
		goto cleanup_and_exit;
1805
1806
1807
1808
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1809
1810
1811
1812
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1813

1814
	/* First, unlink any existing backups.  Next, open the backup
1815
1816
	 * file with O_CREAT and O_EXCL.  If it succeeds, we have a file
	 * descriptor to a new backup file. */
1817
	if (unlink(backupname) < 0 && errno != ENOENT && !ISSET(INSECURE_BACKUP)) {
1818
1819
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1820
1821
	    statusline(HUSH, _("Error writing backup file %s: %s"),
			backupname, strerror(errno));
1822
1823
1824
1825
	    free(backupname);
	    goto cleanup_and_exit;
	}

1826
1827
	if (ISSET(INSECURE_BACKUP))
	    backup_cflags = O_WRONLY | O_CREAT | O_APPEND;
1828
	else
1829
1830
1831
	    backup_cflags = O_WRONLY | O_CREAT | O_EXCL | O_APPEND;

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

1837
	if (backup_fd < 0 || backup_file == NULL) {
1838
1839
	    statusline(HUSH, _("Error writing backup file %s: %s"),
			backupname, strerror(errno));
1840
1841
1842
	    free(backupname);
	    goto cleanup_and_exit;
	}
1843

1844
1845
	/* We shouldn't worry about chown()ing something if we're not
	 * root, since it's likely to fail! */
1846
	if (geteuid() == NANO_ROOT_UID && fchown(backup_fd,
1847
		openfile->current_stat->st_uid, openfile->current_stat->st_gid) == -1
1848
1849
1850
		&& !ISSET(INSECURE_BACKUP)) {
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1851
1852
	    statusline(HUSH, _("Error writing backup file %s: %s"),
			backupname, strerror(errno));
1853
1854
1855
1856
1857
	    free(backupname);
	    fclose(backup_file);
	    goto cleanup_and_exit;
	}

1858
1859
1860
1861
	if (fchmod(backup_fd, openfile->current_stat->st_mode) == -1
		&& !ISSET(INSECURE_BACKUP)) {
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1862
1863
	    statusline(HUSH, _("Error writing backup file %s: %s"),
			backupname, strerror(errno));
1864
	    free(backupname);
1865
	    fclose(backup_file);
1866
	    /* If we can't write to the backup, DONT go on, since
1867
	     * whatever caused the backup file to fail (e.g. disk
1868
	     * full) may well cause the real file write to fail, which
1869
	     * means we could lose both the backup and the original! */
1870
	    goto cleanup_and_exit;
1871
1872
1873
	}

#ifdef DEBUG
1874
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1875
1876
#endif

1877
1878
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1879

1880
	if (copy_status != 0) {
1881
	    statusline(ALERT, _("Error reading %s: %s"), realname,
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1882
			strerror(errno));
1883
1884
1885
1886
1887
	    goto cleanup_and_exit;
	}

	/* And set its metadata. */
	if (utime(backupname, &filetime) == -1 && !ISSET(INSECURE_BACKUP)) {
1888
1889
	    if (prompt_failed_backupwrite(backupname))
		goto skip_backup;
1890
1891
	    statusline(HUSH, _("Error writing backup file %s: %s"),
			backupname, strerror(errno));
1892
1893
1894
1895
	    /* 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! */
1896
	    goto cleanup_and_exit;
1897
	}
1898

1899
1900
	free(backupname);
    }
1901

1902
    skip_backup:
1903
#endif /* !NANO_TINY */
1904

1905
1906
    if (f_open == NULL) {
	original_umask = umask(0);
1907

1908
	/* If we create a temp file, we don't let anyone else access it.
1909
1910
	 * We create a temp file if tmp is TRUE. */
	if (tmp)
1911
	    umask(S_IRWXG | S_IRWXO);
1912
1913
	else
	    umask(original_umask);
1914
    }
1915

1916
    /* If we're prepending, copy the file to a temp file. */
1917
    if (method == PREPEND) {
1918
1919
1920
	int fd_source;
	FILE *f_source = NULL;

1921
1922
1923
1924
	if (f == NULL) {
	    f = fopen(realname, "rb");

	    if (f == NULL) {
1925
		statusline(ALERT, _("Error reading %s: %s"), realname,
1926
1927
1928
1929
1930
			strerror(errno));
		goto cleanup_and_exit;
	    }
	}

1931
	tempname = safe_tempfile(&f);
1932

1933
	if (tempname == NULL) {
1934
	    statusline(ALERT, _("Error writing temp file: %s"),
1935
			strerror(errno));
1936
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1937
	}
1938

1939
	if (f_open == NULL) {
1940
	    fd_source = open(realname, O_RDONLY | O_CREAT, S_IRUSR | S_IWUSR);
1941

1942
1943
1944
	    if (fd_source != -1) {
		f_source = fdopen(fd_source, "rb");
		if (f_source == NULL) {
1945
		    statusline(ALERT, _("Error reading %s: %s"), realname,
1946
				strerror(errno));
1947
1948
1949
1950
1951
1952
		    close(fd_source);
		    fclose(f);
		    unlink(tempname);
		    goto cleanup_and_exit;
		}
	    }
1953
1954
	}

1955
	if (f_source == NULL || copy_file(f_source, f) != 0) {
1956
	    statusline(ALERT, _("Error writing temp file: %s"),
1957
			strerror(errno));
1958
	    unlink(tempname);
1959
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1960
1961
1962
	}
    }

1963
1964
1965
    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*. */
1966
	fd = open(realname, O_WRONLY | O_CREAT | ((method == APPEND) ?
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1967
1968
		O_APPEND : (tmp ? O_EXCL : O_TRUNC)), S_IRUSR |
		S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
1969

1970
1971
	/* Set the umask back to the user's original value. */
	umask(original_umask);
1972

1973
1974
	/* If we couldn't open the file, give up. */
	if (fd == -1) {
1975
	    statusline(ALERT, _("Error writing %s: %s"), realname,
1976
			strerror(errno));
1977
1978
1979
1980
	    if (tempname != NULL)
		unlink(tempname);
	    goto cleanup_and_exit;
	}
1981

1982
	f = fdopen(fd, (method == APPEND) ? "ab" : "wb");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1983

1984
	if (f == NULL) {
1985
	    statusline(ALERT, _("Error writing %s: %s"), realname,
1986
			strerror(errno));
1987
1988
1989
	    close(fd);
	    goto cleanup_and_exit;
	}
1990
1991
    }

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

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

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

2000
2001
	/* Convert nulls to newlines.  data_len is the string's real
	 * length. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2002
2003
	unsunder(fileptr->data, data_len);

2004
	if (size < data_len) {
2005
	    statusline(ALERT, _("Error writing %s: %s"), realname,
2006
			strerror(errno));
2007
	    fclose(f);
2008
	    goto cleanup_and_exit;
2009
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2010

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2011
	/* If we're on the last line of the file, don't write a newline
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2012
2013
2014
2015
2016
2017
2018
	 * 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 {
2019
#ifndef NANO_TINY
2020
	    if (openfile->fmt == DOS_FILE || openfile->fmt == MAC_FILE) {
2021
		if (putc('\r', f) == EOF) {
2022
		    statusline(ALERT, _("Error writing %s: %s"), realname,
2023
				strerror(errno));
2024
2025
2026
		    fclose(f);
		    goto cleanup_and_exit;
		}
2027
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2028

2029
	    if (openfile->fmt != MAC_FILE)
2030
#endif
2031
		if (putc('\n', f) == EOF) {
2032
		    statusline(ALERT, _("Error writing %s: %s"), realname,
2033
				strerror(errno));
2034
2035
2036
2037
		    fclose(f);
		    goto cleanup_and_exit;
		}
	}
Chris Allegretta's avatar
Chris Allegretta committed
2038
2039
2040
2041
2042

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

2043
    /* If we're prepending, open the temp file, and append it to f. */
2044
    if (method == PREPEND) {
2045
2046
2047
	int fd_source;
	FILE *f_source = NULL;

2048
	fd_source = open(tempname, O_RDONLY | O_CREAT, S_IRUSR | S_IWUSR);
2049

2050
2051
2052
2053
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
2054
	}
2055

2056
	if (f_source == NULL) {
2057
2058
	    statusline(ALERT, _("Error reading %s: %s"), tempname,
			strerror(errno));
2059
	    fclose(f);
2060
	    goto cleanup_and_exit;
2061
2062
	}

2063
	if (copy_file(f_source, f) == -1) {
2064
	    statusline(ALERT, _("Error writing %s: %s"), realname,
2065
			strerror(errno));
2066
	    goto cleanup_and_exit;
2067
	}
2068
2069

	unlink(tempname);
2070
    } else if (fclose(f) != 0) {
2071
	statusline(ALERT, _("Error writing %s: %s"), realname,
2072
			strerror(errno));
2073
	goto cleanup_and_exit;
2074
    }
2075

2076
    if (method == OVERWRITE && !tmp) {
2077
2078
	/* If we must set the filename, and it changed, adjust things. */
	if (!nonamechange && strcmp(openfile->filename, realname) != 0) {
2079
#ifndef DISABLE_COLOR
2080
	    char *oldname = openfile->syntax ? openfile->syntax->name : "";
2081
2082
2083
	    filestruct *line = openfile->fileage;
#endif
	    openfile->filename = mallocstrcpy(openfile->filename, realname);
2084

2085
2086
#ifndef DISABLE_COLOR
	    /* See if the applicable syntax has changed. */
2087
	    color_update();
2088
	    color_init();
2089

2090
2091
	    char *newname = openfile->syntax ? openfile->syntax->name : "";

2092
	    /* If the syntax changed, discard and recompute the multidata. */
2093
	    if (strcmp(oldname, newname) != 0) {
2094
2095
2096
2097
2098
		for (; line != NULL; line = line->next) {
		    free(line->multidata);
		    line->multidata = NULL;
		}
		precalc_multicolorinfo();
2099
		refresh_needed = TRUE;
2100
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2101
2102
#endif
	}
2103

2104
#ifndef NANO_TINY
2105
	if (!openfile->mark_set)
2106
2107
	    /* Get or update the stat info to reflect the current state. */
	    stat_with_alloc(realname, &openfile->current_stat);
2108
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2109

2110
2111
	statusline(HUSH, P_("Wrote %lu line", "Wrote %lu lines",
		(unsigned long)lineswritten), (unsigned long)lineswritten);
2112
	openfile->modified = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2113
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2114
    }
2115

2116
    retval = TRUE;
2117
2118
2119

  cleanup_and_exit:
    free(realname);
2120
    free(tempname);
2121

2122
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
2123
2124
}

2125
#ifndef NANO_TINY
2126
2127
/* Write a marked selection from a file out to disk.  Return TRUE on
 * success or FALSE on error. */
2128
bool write_marked_file(const char *name, FILE *f_open, bool tmp,
2129
	kind_of_writing_type method)
2130
{
2131
    bool retval;
2132
    bool old_modified = openfile->modified;
2133
	/* Save the status, because write_file() unsets the modified flag. */
2134
    bool added_magicline = FALSE;
2135
2136
2137
2138
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

2139
    /* Partition the filestruct so that it contains only the marked text. */
2140
    mark_order((const filestruct **)&top, &top_x,
2141
		(const filestruct **)&bot, &bot_x, NULL);
2142
    filepart = partition_filestruct(top, top_x, bot, bot_x);
2143

2144
2145
2146
    /* If we are doing magicline, and the last line of the partition
     * isn't blank, then add a newline at the end of the buffer. */
    if (!ISSET(NO_NEWLINES) && openfile->filebot->data[0] != '\0') {
2147
	new_magicline();
2148
2149
	added_magicline = TRUE;
    }
2150

2151
    retval = write_file(name, f_open, tmp, method, TRUE);
2152

2153
2154
    /* If we added a magicline, remove it now. */
    if (added_magicline)
2155
2156
	remove_magicline();

2157
    /* Unpartition the filestruct so that it contains all the text again. */
2158
    unpartition_filestruct(&filepart);
2159

2160
    if (old_modified)
2161
2162
2163
2164
	set_modified();

    return retval;
}
2165

2166
#endif /* !NANO_TINY */
2167

2168
/* Write the current file to disk.  If the mark is on, write the current
2169
2170
2171
2172
2173
 * marked selection to disk.  If exiting is TRUE, write the entire file
 * to disk regardless of whether the mark is on, and without prompting if
 * the TEMP_FILE flag is set and the current file has a name.  Return 0
 * on error, 1 on success, and 2 when the buffer is to be discarded. */
int do_writeout(bool exiting)
Chris Allegretta's avatar
Chris Allegretta committed
2174
{
2175
    int i;
2176
    bool result = FALSE;
2177
    kind_of_writing_type method = OVERWRITE;
2178
2179
    char *given;
	/* The filename we offer, or what the user typed so far. */
2180
    bool maychange = (openfile->filename[0] == '\0');
2181
	/* Whether it's okay to save the file under a different name. */
2182
#ifndef DISABLE_EXTRA
2183
    static bool did_credits = FALSE;
2184
#endif
Chris Allegretta's avatar
Chris Allegretta committed
2185

2186
2187
2188
2189
    if (exiting && ISSET(TEMP_FILE) && openfile->filename[0] != '\0') {
	if (write_file(openfile->filename, NULL, FALSE, OVERWRITE, FALSE))
	    return 1;
	/* If writing the file failed, go on to prompt for a new name. */
Chris Allegretta's avatar
Chris Allegretta committed
2190
2191
    }

2192
    given = mallocstrcpy(NULL,
2193
#ifndef NANO_TINY
2194
	(openfile->mark_set && !exiting) ? "" :
2195
#endif
2196
	openfile->filename);
2197
2198
2199

    while (TRUE) {
	const char *msg;
2200
#ifndef NANO_TINY
2201
2202
	const char *formatstr, *backupstr;

2203
	formatstr = (openfile->fmt == DOS_FILE) ? _(" [DOS Format]") :
2204
			(openfile->fmt == MAC_FILE) ? _(" [Mac Format]") : "";
2205

2206
	backupstr = ISSET(BACKUP_FILE) ? _(" [Backup]") : "";
2207

2208
	/* If we're using restricted mode, don't display the "Write
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2209
2210
2211
	 * Selection to File" prompt.  This function is disabled, since
	 * it allows reading from or writing to files not specified on
	 * the command line. */
2212
	if (openfile->mark_set && !exiting && !ISSET(RESTRICTED))
2213
	    /* TRANSLATORS: The next six strings are prompts. */
2214
2215
	    msg = (method == PREPEND) ? _("Prepend Selection to File") :
			(method == APPEND) ? _("Append Selection to File") :
2216
			_("Write Selection to File");
2217
	else
2218
#endif /* !NANO_TINY */
2219
2220
	    msg = (method == PREPEND) ? _("File Name to Prepend to") :
			(method == APPEND) ? _("File Name to Append to") :
2221
			_("File Name to Write");
2222

2223
2224
	present_path = mallocstrcpy(present_path, "./");

2225
2226
	/* If we're using restricted mode, and the filename isn't blank,
	 * disable tab completion. */
2227
	i = do_prompt(!ISSET(RESTRICTED) || openfile->filename[0] == '\0',
2228
2229
2230
#ifndef DISABLE_TABCOMP
		TRUE,
#endif
2231
		MWRITEFILE, given,
2232
#ifndef DISABLE_HISTORIES
2233
2234
		NULL,
#endif
2235
		edit_refresh, "%s%s%s", msg,
2236
2237
#ifndef NANO_TINY
		formatstr, backupstr
2238
#else
2239
		"", ""
2240
2241
2242
#endif
		);

2243
2244
	/* 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
2245
	if (i < 0 || *answer == '\n') {
2246
	    statusbar(_("Cancelled"));
2247
2248
	    break;
	} else {
2249
2250
	    functionptrtype func = func_from_key(&i);

2251
2252
2253
2254
	    /* Upon request, abandon the buffer, if user is sure. */
	    if (func == discard_buffer) {
		if (openfile->modified)
		    i = do_yesno_prompt(FALSE,
2255
				_("Save modified buffer anyway? "));
2256
2257
2258
2259
		else
		    i = 0;

		if (i == 0) {
2260
		    free(given);
2261
		    return 2;	/* Yes, discard the buffer. */
2262
2263
		} else
		    continue;	/* Go back to the filename prompt. */
2264
2265
	    }

2266
	    given = mallocstrcpy(given, answer);
Chris Allegretta's avatar
Chris Allegretta committed
2267

2268
#ifndef DISABLE_BROWSER
2269
	    if (func == to_files_void) {
2270
		char *chosen = do_browse_from(answer);
2271

2272
		if (chosen == NULL)
2273
		    continue;
2274

2275
		/* We have a file now.  Indicate this. */
2276
		free(answer);
2277
		answer = chosen;
2278
	    } else
2279
#endif /* !DISABLE_BROWSER */
2280
#ifndef NANO_TINY
2281
	    if (func == dos_format_void) {
2282
2283
		openfile->fmt = (openfile->fmt == DOS_FILE) ? NIX_FILE :
			DOS_FILE;
2284
		continue;
2285
	    } else if (func == mac_format_void) {
2286
2287
		openfile->fmt = (openfile->fmt == MAC_FILE) ? NIX_FILE :
			MAC_FILE;
2288
		continue;
2289
	    } else if (func == backup_file_void) {
2290
2291
2292
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
2293
#endif /* !NANO_TINY */
2294
	    if (func == prepend_void) {
2295
		method = (method == PREPEND) ? OVERWRITE : PREPEND;
2296
		continue;
2297
	    } else if (func == append_void) {
2298
		method = (method == APPEND) ? OVERWRITE : APPEND;
2299
		continue;
2300
	    } else if (func == do_help_void) {
2301
		continue;
2302
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2303

Chris Allegretta's avatar
Chris Allegretta committed
2304
#ifdef DEBUG
2305
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
2306
#endif
2307

2308
#ifndef DISABLE_EXTRA
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2309
2310
2311
2312
2313
2314
2315
	    /* 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) &&
2316
			strcasecmp(answer, "zzy") == 0) {
2317
		do_credits();
2318
		did_credits = TRUE;
2319
2320
		break;
	    }
2321
#endif
2322

2323
	    if (method == OVERWRITE) {
2324
		size_t answer_len = strlen(answer);
2325
		bool name_exists, do_warning;
2326
		char *full_answer, *full_filename;
2327
2328
		struct stat st;

2329
2330
2331
		/* Convert newlines to nulls, just before we get the
		 * full path. */
		sunder(answer);
2332

2333
2334
		full_answer = get_full_path(answer);
		full_filename = get_full_path(openfile->filename);
2335
2336
		name_exists = (stat((full_answer == NULL) ?
				answer : full_answer, &st) != -1);
2337
2338
2339
2340
		if (openfile->filename[0] == '\0')
		    do_warning = name_exists;
		else
		    do_warning = (strcmp((full_answer == NULL) ?
2341
2342
				answer : full_answer, (full_filename == NULL) ?
				openfile->filename : full_filename) != 0);
2343

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2344
2345
2346
2347
		/* Convert nulls to newlines.  answer_len is the
		 * string's real length. */
		unsunder(answer, answer_len);

2348
2349
		free(full_filename);
		free(full_answer);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2350

2351
		if (do_warning) {
2352
2353
2354
		    /* When in restricted mode, we aren't allowed to overwrite
		     * an existing file with the current buffer, nor to change
		     * the name of the current file if it already has one. */
2355
		    if (ISSET(RESTRICTED)) {
2356
			/* TRANSLATORS: Restricted mode forbids overwriting. */
2357
2358
			warn_and_shortly_pause(_("File exists -- "
					"cannot overwrite"));
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2359
			continue;
2360
		    }
2361

2362
		    if (!maychange) {
2363
#ifndef NANO_TINY
2364
2365
2366
			if (exiting || !openfile->mark_set)
#endif
			{
2367
2368
			    if (do_yesno_prompt(FALSE, _("Save file under "
					"DIFFERENT NAME? ")) < 1)
2369
2370
2371
2372
2373
2374
2375
2376
2377
2378
2379
2380
2381
2382
2383
				continue;
			    maychange = TRUE;
			}
		    }

		    if (name_exists) {
			char *question = _("File \"%s\" exists; OVERWRITE? ");
			char *message = charalloc(strlen(question) +
						strlen(answer) + 1);
			sprintf(message, question, answer);

			i = do_yesno_prompt(FALSE, message);
			free(message);

			if (i < 1)
2384
2385
			    continue;
		    }
2386
		}
2387
#ifndef NANO_TINY
2388
2389
2390
		/* Complain if the file exists, the name hasn't changed,
		 * and the stat information we had before does not match
		 * what we have now. */
2391
2392
2393
2394
		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)) {
2395
2396
2397

		    if (do_yesno_prompt(FALSE, _("File was modified since "
				"you opened it; continue saving? ")) < 1)
2398
2399
			continue;
		}
2400
#endif
2401
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2402

2403
	    /* Convert newlines to nulls, just before we save the file. */
2404
2405
	    sunder(answer);

2406
	    /* Here's where we allow the selected text to be written to
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2407
2408
2409
	     * 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. */
2410
#ifndef NANO_TINY
2411
	    if (openfile->mark_set && !exiting && !ISSET(RESTRICTED))
2412
		result = write_marked_file(answer, NULL, FALSE, method);
2413
	    else
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2414
#endif
2415
		result = write_file(answer, NULL, FALSE, method, FALSE);
2416

2417
2418
	    break;
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2419
    }
2420

2421
    free(given);
2422

2423
    return result ? 1 : 0;
Chris Allegretta's avatar
Chris Allegretta committed
2424
2425
}

2426
/* Write the current buffer to disk, or discard it. */
2427
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
2428
{
2429
2430
2431
    /* If the user chose to discard the buffer, close it. */
    if (do_writeout(FALSE) == 2)
	close_and_go();
Chris Allegretta's avatar
Chris Allegretta committed
2432
}
Chris Allegretta's avatar
Chris Allegretta committed
2433

2434
2435
2436
2437
2438
2439
2440
2441
2442
2443
2444
#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

2445
2446
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
2447
char *real_dir_from_tilde(const char *buf)
2448
{
2449
    char *retval;
2450

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2451
    if (*buf == '~') {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2452
	size_t i = 1;
2453
	char *tilde_dir;
2454

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

2459
2460
2461
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
2462
	    tilde_dir = mallocstrcpy(NULL, homedir);
2463
2464
2465
	} else {
	    const struct passwd *userdata;

2466
2467
2468
	    tilde_dir = mallocstrncpy(NULL, buf, i + 1);
	    tilde_dir[i] = '\0';

2469
2470
	    do {
		userdata = getpwent();
2471
2472
	    } while (userdata != NULL &&
			strcmp(userdata->pw_name, tilde_dir + 1) != 0);
2473
	    endpwent();
2474
	    if (userdata != NULL)
2475
		tilde_dir = mallocstrcpy(tilde_dir, userdata->pw_dir);
Chris Allegretta's avatar
Chris Allegretta committed
2476
	}
2477

2478
2479
	retval = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	sprintf(retval, "%s%s", tilde_dir, buf + i);
2480

2481
2482
2483
	free(tilde_dir);
    } else
	retval = mallocstrcpy(NULL, buf);
2484

2485
    return retval;
2486
2487
}

2488
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
2489
/* Our sort routine for file listings.  Sort alphabetically and
2490
 * case-insensitively, and sort directories before filenames. */
2491
2492
2493
2494
2495
2496
2497
2498
2499
2500
2501
2502
2503
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;

2504
2505
2506
    /* 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
2507
     * have to use multibyte strcasecmp() instead. */
2508
    return mbstrcasecmp(a, b);
2509
}
2510
2511
2512
2513
2514

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

2518
2519
    for (; len > 0; len--)
	free(array[len - 1]);
2520

2521
2522
    free(array);
}
2523
2524
#endif

Chris Allegretta's avatar
Chris Allegretta committed
2525
#ifndef DISABLE_TABCOMP
2526
/* Is the given path a directory? */
2527
bool is_dir(const char *buf)
2528
{
2529
    char *dirptr;
2530
    struct stat fileinfo;
2531
2532
2533
2534
    bool retval;

    dirptr = real_dir_from_tilde(buf);

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

2537
    free(dirptr);
2538

2539
    return retval;
2540
}
Chris Allegretta's avatar
Chris Allegretta committed
2541

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2542
2543
2544
/* 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
2545
 * file, with the copyright years updated:
Chris Allegretta's avatar
Chris Allegretta committed
2546
2547
2548
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
2549
 * Copyright (C) 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006, 2007
Chris Allegretta's avatar
Chris Allegretta committed
2550
2551
2552
2553
2554
2555
2556
2557
 *      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.
2558
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
2559

2560
/* We consider the first buf_len characters of buf for ~username tab
2561
2562
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
2563
	size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2564
{
2565
2566
    char **matches = NULL;
    const struct passwd *userdata;
2567

2568
    assert(buf != NULL && num_matches != NULL && buf_len > 0);
2569

2570
    *num_matches = 0;
2571

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

2577
2578
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2579
2580
2581
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
2582
2583
#endif

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2584
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
2585
2586
					sizeof(char *));
	    matches[*num_matches] = charalloc(strlen(userdata->pw_name) + 2);
2587
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
2588
	    ++(*num_matches);
2589
	}
2590
2591
    }
    endpwent();
2592

2593
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2594
2595
}

2596
/* We consider the first buf_len characters of buf for filename tab
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2597
 * completion. */
2598
char **cwd_tab_completion(const char *buf, bool allow_files, size_t
2599
	*num_matches, size_t buf_len)
Chris Allegretta's avatar
Chris Allegretta committed
2600
{
2601
2602
    char *dirname = mallocstrcpy(NULL, buf);
    char *slash, *filename;
2603
2604
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2605
    DIR *dir;
2606
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
2607

2608
    *num_matches = 0;
2609
    null_at(&dirname, buf_len);
2610

2611
    /* If there's a / in the name, split out filename and directory parts. */
2612
2613
2614
    slash = strrchr(dirname, '/');
    if (slash != NULL) {
	char *wasdirname = dirname;
2615

2616
2617
	filename = mallocstrcpy(NULL, ++slash);
	/* Cut off the filename part after the slash. */
2618
	*slash = '\0';
2619
	dirname = real_dir_from_tilde(dirname);
2620
2621
2622
2623
2624
2625
	/* A non-absolute path is relative to the current browser directory. */
	if (dirname[0] != '/') {
	    dirname = charealloc(dirname, strlen(present_path) +
						strlen(wasdirname) + 1);
	    sprintf(dirname, "%s%s", present_path, wasdirname);
	}
2626
	free(wasdirname);
Chris Allegretta's avatar
Chris Allegretta committed
2627
    } else {
2628
	filename = dirname;
2629
	dirname = mallocstrcpy(NULL, present_path);
Chris Allegretta's avatar
Chris Allegretta committed
2630
2631
    }

2632
    assert(dirname[strlen(dirname) - 1] == '/');
2633

Chris Allegretta's avatar
Chris Allegretta committed
2634
    dir = opendir(dirname);
2635

2636
    if (dir == NULL) {
2637
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
2638
	beep();
2639
2640
2641
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2642
    }
2643
2644
2645

    filenamelen = strlen(filename);

2646
    while ((nextdir = readdir(dir)) != NULL) {
2647
2648
	bool skip_match = FALSE;

Chris Allegretta's avatar
Chris Allegretta committed
2649
#ifdef DEBUG
2650
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
2651
#endif
2652
	/* See if this matches. */
2653
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
2654
2655
		(*filename == '.' || (strcmp(nextdir->d_name, ".") != 0 &&
		strcmp(nextdir->d_name, "..") != 0))) {
2656
2657
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2658

2659
	    char *tmp = charalloc(strlen(dirname) + strlen(nextdir->d_name) + 1);
2660
2661
	    sprintf(tmp, "%s%s", dirname, nextdir->d_name);

2662
2663
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2664
2665
2666
2667
2668
2669
2670
	     * 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. */
2671
	    if (!allow_files && !is_dir(tmp))
2672
2673
2674
		skip_match = TRUE;

	    free(tmp);
2675

2676
	    if (skip_match)
2677
		continue;
2678

2679
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
2680
					sizeof(char *));
2681
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
2682
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
2683
2684
	}
    }
2685

2686
2687
    closedir(dir);
    free(dirname);
2688
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
2689

2690
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2691
2692
}

2693
/* Do tab completion.  place refers to how much the statusbar cursor
2694
2695
 * position should be advanced.  refresh_func is the function we will
 * call to refresh the edit window. */
2696
char *input_tab(char *buf, bool allow_files, size_t *place,
2697
	bool *lastwastab, void (*refresh_func)(void), bool *listed)
Chris Allegretta's avatar
Chris Allegretta committed
2698
{
2699
    size_t num_matches = 0, buf_len;
2700
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2701

2702
    assert(buf != NULL && place != NULL && *place <= strlen(buf) &&
2703
2704
2705
		lastwastab != NULL && refresh_func != NULL && listed != NULL);

    *listed = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2706

2707
2708
    /* 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
2709
    if (*place > 0 && *buf == '~') {
2710
	const char *slash = strchr(buf, '/');
Chris Allegretta's avatar
Chris Allegretta committed
2711

2712
	if (slash == NULL || slash >= buf + *place)
2713
	    matches = username_tab_completion(buf, &num_matches, *place);
2714
    }
2715

2716
2717
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
2718
	matches = cwd_tab_completion(buf, allow_files, &num_matches, *place);
2719

2720
2721
2722
    buf_len = strlen(buf);

    if (num_matches == 0 || *place != buf_len)
2723
2724
2725
	beep();
    else {
	size_t match, common_len = 0;
2726
	char *mzero, *glued;
2727
	const char *lastslash = revstrstr(buf, "/", buf + *place);
2728
	size_t lastslash_len = (lastslash == NULL) ? 0 : lastslash - buf + 1;
2729
2730
2731
	char *match1 = charalloc(mb_cur_max());
	char *match2 = charalloc(mb_cur_max());
	int match1_len, match2_len;
2732

2733
	/* Get the number of characters that all matches have in common. */
2734
	while (TRUE) {
2735
	    match1_len = parse_mbchar(matches[0] + common_len, match1, NULL);
2736
2737

	    for (match = 1; match < num_matches; match++) {
2738
2739
2740
2741
		match2_len = parse_mbchar(matches[match] + common_len,
						match2, NULL);
		if (match1_len != match2_len ||
				strncmp(match1, match2, match2_len) != 0)
2742
2743
		    break;
	    }
2744

2745
	    if (match < num_matches || matches[0][common_len] == '\0')
2746
		break;
2747

2748
	    common_len += match1_len;
2749
	}
2750

2751
2752
	free(match1);
	free(match2);
2753

2754
	mzero = charalloc(lastslash_len + common_len + 1);
2755
2756
2757

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

2759
	common_len += lastslash_len;
2760
	mzero[common_len] = '\0';
2761

2762
2763
2764
2765
	/* Cover also the case of the user specifying a relative path. */
	glued = charalloc(strlen(present_path) + strlen(mzero) + 1);
	sprintf(glued, "%s%s", present_path, mzero);

2766
	assert(common_len >= *place);
2767

2768
	if (num_matches == 1 && (is_dir(mzero) || is_dir(glued))) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2769
	    mzero[common_len++] = '/';
2770

2771
2772
	    assert(common_len > *place);
	}
2773

2774
	if (num_matches > 1 && (common_len != *place || !*lastwastab))
2775
	    beep();
2776

2777
	/* If the matches have something in common, show that part. */
2778
	if (common_len != *place) {
2779
	    buf = charealloc(buf, common_len + buf_len - *place + 1);
2780
	    charmove(buf + common_len, buf + *place, buf_len - *place + 1);
2781
	    strncpy(buf, mzero, common_len);
2782
	    *place = common_len;
2783
2784
2785
	}

	if (!*lastwastab)
2786
	    *lastwastab = TRUE;
2787
	else if (num_matches > 1) {
2788
	    int longest_name = 0, ncols, editline = 0;
2789

2790
	    /* Sort the list of available choices. */
2791
2792
	    qsort(matches, num_matches, sizeof(char *), diralphasort);

2793
	    /* Find the length of the longest among the choices. */
2794
	    for (match = 0; match < num_matches; match++) {
2795
		size_t namelen = strlenpt(matches[match]);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2796

2797
2798
		if (namelen > longest_name)
		    longest_name = namelen;
2799
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2800

2801
2802
	    if (longest_name > COLS - 1)
		longest_name = COLS - 1;
Chris Allegretta's avatar
Chris Allegretta committed
2803

2804
	    /* Each column will be (longest_name + 2) columns wide, i.e.
2805
2806
	     * two spaces between columns, except that there will be
	     * only one space after the last column. */
2807
	    ncols = (COLS + 1) / (longest_name + 2);
2808

2809
	    /* Blank the edit window and hide the cursor. */
2810
2811
	    blank_edit();
	    curs_set(0);
2812
	    wmove(edit, 0, 0);
2813

2814
	    /* Now print the list of matches out there. */
2815
2816
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2817

2818
		wmove(edit, editline, (longest_name + 2) * (match % ncols));
2819

2820
		if (match % ncols == 0 && editline == editwinrows - 1 &&
2821
			num_matches - match > ncols) {
2822
2823
2824
		    waddstr(edit, _("(more)"));
		    break;
		}
2825

2826
		disp = display_string(matches[match], 0, longest_name, FALSE);
2827
2828
		waddstr(edit, disp);
		free(disp);
2829

2830
		if ((match + 1) % ncols == 0)
2831
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2832
	    }
2833

2834
	    wnoutrefresh(edit);
2835
	    *listed = TRUE;
2836
2837
	}

2838
	free(glued);
2839
	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2840
2841
    }

2842
    free_chararray(matches, num_matches);
2843

2844
2845
2846
    /* When we didn't list any matches now, refresh the edit window, just
     * in case a previous tab showed a list, so we know where we are. */
    if (!*listed)
2847
2848
	refresh_func();

2849
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2850
}
2851
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2852

2853
2854
/* Return the filename part of the given path. */
const char *tail(const char *path)
2855
{
2856
    const char *slash = strrchr(path, '/');
2857

2858
2859
    if (slash == NULL)
	return path;
2860
    else
2861
	return ++slash;
2862
2863
}

2864
2865
#ifndef DISABLE_HISTORIES
/* Return the constructed dirfile path, or NULL if we can't find the home
2866
 * directory.  The string is dynamically allocated, and should be freed. */
2867
char *construct_filename(const char *str)
2868
{
2869
    char *newstr = NULL;
2870

2871
2872
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2873

2874
2875
2876
	newstr = charalloc(homelen + strlen(str) + 1);
	strcpy(newstr, homedir);
	strcpy(newstr + homelen, str);
Chris Allegretta's avatar
Chris Allegretta committed
2877
    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2878

2879
2880
2881
2882
2883
    return newstr;
}

char *histfilename(void)
{
2884
    return construct_filename("/.nano/search_history");
2885
2886
}

2887
2888
/* Construct the legacy history filename. */
/* (To be removed in 2018.) */
2889
2890
2891
2892
2893
2894
2895
char *legacyhistfilename(void)
{
    return construct_filename("/.nano_history");
}

char *poshistfilename(void)
{
2896
    return construct_filename("/.nano/filepos_history");
2897
2898
2899
2900
}

void history_error(const char *msg, ...)
{
2901
    va_list ap;
2902
2903
2904
2905
2906
2907
2908
2909
2910
2911

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

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

2912
2913
2914
/* 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. */
2915
2916
int check_dotnano(void)
{
2917
    int ret = 1;
2918
2919
2920
2921
2922
    struct stat dirstat;
    char *nanodir = construct_filename("/.nano");

    if (stat(nanodir, &dirstat) == -1) {
	if (mkdir(nanodir, S_IRWXU | S_IRWXG | S_IRWXO) == -1) {
2923
	    history_error(N_("Unable to create directory %s: %s\n"
2924
2925
				"It is required for saving/loading "
				"search history or cursor positions.\n"),
2926
				nanodir, strerror(errno));
2927
	    ret = 0;
2928
2929
	}
    } else if (!S_ISDIR(dirstat.st_mode)) {
2930
	history_error(N_("Path %s is not a directory and needs to be.\n"
2931
2932
				"Nano will be unable to load or save "
				"search history or cursor positions.\n"),
2933
				nanodir);
2934
	ret = 0;
2935
    }
2936
2937
2938

    free(nanodir);
    return ret;
2939
2940
}

2941
/* Load the search and replace histories from ~/.nano/search_history. */
2942
2943
2944
void load_history(void)
{
    char *nanohist = histfilename();
2945
2946
2947
    char *legacyhist = legacyhistfilename();
    struct stat hstat;

2948
2949
    /* If there is an old history file, migrate it. */
    /* (To be removed in 2018.) */
2950
    if (stat(legacyhist, &hstat) != -1 && stat(nanohist, &hstat) == -1) {
2951
	if (rename(legacyhist, nanohist) == -1)
2952
2953
2954
	    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));
2955
	else
2956
2957
2958
	    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);
2959
2960
    }

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

2965
	if (hist == NULL) {
2966
	    if (errno != ENOENT) {
2967
2968
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
2969
		history_error(N_("Error reading %s: %s"), nanohist,
2970
			strerror(errno));
2971
	    }
2972
	} else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2973
2974
2975
	    /* 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. */
2976
	    filestruct **history = &search_history;
2977
	    char *line = NULL;
2978
	    size_t buf_len = 0;
2979
2980
	    ssize_t read;

2981
	    while ((read = getline(&line, &buf_len, hist)) >= 0) {
2982
2983
		if (read > 0 && line[read - 1] == '\n')
		    line[--read] = '\0';
2984
2985
		if (read > 0) {
		    unsunder(line, read);
2986
2987
		    update_history(history, line);
		} else
2988
2989
		    history = &replace_history;
	    }
2990

2991
	    fclose(hist);
2992
	    free(line);
2993
	}
2994
	free(nanohist);
2995
	free(legacyhist);
2996
2997
2998
    }
}

2999
3000
3001
/* 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. */
3002
bool writehist(FILE *hist, filestruct *h)
3003
{
3004
    filestruct *p;
3005

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

3011
	sunder(p->data);
3012

3013
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
3014
3015
		putc('\n', hist) == EOF)
	    return FALSE;
3016
    }
3017

3018
    return TRUE;
3019
3020
}

3021
/* Save the search and replace histories to ~/.nano/search_history. */
3022
3023
void save_history(void)
{
3024
    char *nanohist;
3025

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
3026
    /* Don't save unchanged or empty histories. */
3027
    if (!history_has_changed() || (searchbot->lineno == 1 &&
3028
		replacebot->lineno == 1))
3029
3030
	return;

3031
3032
3033
3034
    nanohist = histfilename();

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

3036
	if (hist == NULL)
3037
3038
	    fprintf(stderr, _("Error writing %s: %s\n"), nanohist,
			strerror(errno));
3039
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
3040
3041
	    /* Make sure no one else can read from or write to the
	     * history file. */
3042
	    chmod(nanohist, S_IRUSR | S_IWUSR);
3043

3044
	    if (!writehist(hist, searchage) || !writehist(hist, replaceage))
3045
		fprintf(stderr, _("Error writing %s: %s\n"), nanohist,
3046
			strerror(errno));
3047

3048
3049
	    fclose(hist);
	}
3050

3051
3052
3053
	free(nanohist);
    }
}
3054

3055
/* Save the recorded last file positions to ~/.nano/filepos_history. */
3056
3057
void save_poshistory(void)
{
3058
    char *poshist = poshistfilename();
3059
3060
    char *statusstr = NULL;
    poshiststruct *posptr;
3061
    FILE *hist;
3062

3063
3064
    if (poshist == NULL)
	return;
3065

3066
    hist = fopen(poshist, "wb");
3067

3068
3069
3070
3071
3072
    if (hist == NULL)
	fprintf(stderr, _("Error writing %s: %s\n"), poshist, strerror(errno));
    else {
	/* Don't allow others to read or write the history file. */
	chmod(poshist, S_IRUSR | S_IWUSR);
3073

3074
	for (posptr = position_history; posptr != NULL; posptr = posptr->next) {
3075
3076
3077
	    /* Assume 20 decimal positions each for line and column number,
	     * plus two spaces, plus the line feed, plus the null byte. */
	    statusstr = charalloc(strlen(posptr->filename) + 44);
3078
	    sprintf(statusstr, "%s %ld %ld\n", posptr->filename, (long)posptr->lineno,
3079
			(long)posptr->xno);
3080
3081
3082
	    if (fwrite(statusstr, sizeof(char), strlen(statusstr), hist) < strlen(statusstr))
		fprintf(stderr, _("Error writing %s: %s\n"), poshist, strerror(errno));
	    free(statusstr);
3083
	}
3084
	fclose(hist);
3085
    }
3086
    free(poshist);
3087
3088
}

3089
3090
/* 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. */
3091
3092
void update_poshistory(char *filename, ssize_t lineno, ssize_t xpos)
{
3093
    poshiststruct *posptr, *theone, *posprev = NULL;
3094
    char *fullpath = get_full_path(filename);
3095

3096
3097
    if (fullpath == NULL || fullpath[strlen(fullpath) - 1] == '/') {
	free(fullpath);
3098
	return;
3099
    }
3100

3101
    /* Look for a matching filename in the list. */
3102
    for (posptr = position_history; posptr != NULL; posptr = posptr->next) {
3103
3104
	if (!strcmp(posptr->filename, fullpath))
	    break;
3105
3106
3107
	posprev = posptr;
    }

3108
3109
3110
3111
3112
3113
3114
3115
3116
3117
3118
3119
3120
3121
    /* Don't record files that have the default cursor position. */
    if (lineno == 1 && xpos == 1) {
	if (posptr != NULL) {
	    if (posprev == NULL)
		position_history = posptr->next;
	    else
		posprev->next = posptr->next;
	    free(posptr->filename);
	    free(posptr);
	}
	free(fullpath);
	return;
    }

3122
    theone = posptr;
3123

3124
3125
3126
3127
3128
3129
3130
3131
3132
3133
    /* If we didn't find it, make a new node; otherwise, if we're
     * not at the end, move the matching one to the end. */
    if (theone == NULL) {
	theone = (poshiststruct *)nmalloc(sizeof(poshiststruct));
	theone->filename = mallocstrcpy(NULL, fullpath);
	if (position_history == NULL)
	    position_history = theone;
	else
	    posprev->next = theone;
    } else if (posptr->next != NULL) {
3134
3135
3136
3137
	if (posprev == NULL)
	    position_history = posptr->next;
	else
	    posprev->next = posptr->next;
3138
3139
3140
3141
3142
3143
3144
3145
3146
	while (posptr->next != NULL)
	    posptr = posptr->next;
	posptr->next = theone;
    }

    /* Store the last cursor position. */
    theone->lineno = lineno;
    theone->xno = xpos;
    theone->next = NULL;
3147
3148
3149
3150

    free(fullpath);
}

3151
3152
3153
/* 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. */
3154
3155
3156
3157
3158
3159
int check_poshistory(const char *file, ssize_t *line, ssize_t *column)
{
    poshiststruct *posptr;
    char *fullpath = get_full_path(file);

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

3162
    for (posptr = position_history; posptr != NULL; posptr = posptr->next) {
3163
3164
3165
	if (!strcmp(posptr->filename, fullpath)) {
	    *line = posptr->lineno;
	    *column = posptr->xno;
3166
	    free(fullpath);
3167
3168
3169
	    return 1;
	}
    }
3170
    free(fullpath);
3171
3172
3173
    return 0;
}

3174
/* Load the recorded file positions from ~/.nano/filepos_history. */
3175
3176
void load_poshistory(void)
{
3177
    char *poshist = poshistfilename();
3178
    FILE *hist;
3179

3180
3181
3182
    /* If the home directory is missing, do_rcfile() will have reported it. */
    if (poshist == NULL)
	return;
3183

3184
    hist = fopen(poshist, "rb");
3185

3186
3187
    if (hist == NULL) {
	if (errno != ENOENT) {
3188
	    /* When reading failed, don't save history when we quit. */
3189
3190
	    UNSET(POS_HISTORY);
	    history_error(N_("Error reading %s: %s"), poshist, strerror(errno));
3191
	}
3192
3193
3194
    } else {
	char *line = NULL, *lineptr, *xptr;
	size_t buf_len = 0;
3195
	ssize_t read, count = 0;
3196
3197
3198
3199
3200
3201
3202
3203
3204
3205
3206
3207
3208
3209
3210
3211
3212
3213
3214
3215
3216
3217
3218
3219
3220
	poshiststruct *record_ptr = NULL, *newrecord;

	/* Read and parse each line, and store the extracted data. */
	while ((read = getline(&line, &buf_len, hist)) > 2) {
	    if (line[read - 1] == '\n')
		line[--read] = '\0';
	    unsunder(line, read);

	    lineptr = parse_next_word(line);
	    xptr = parse_next_word(lineptr);

	    /* Create a new position record. */
	    newrecord = (poshiststruct *)nmalloc(sizeof(poshiststruct));
	    newrecord->filename = mallocstrcpy(NULL, line);
	    newrecord->lineno = atoi(lineptr);
	    newrecord->xno = atoi(xptr);
	    newrecord->next = NULL;

	    /* Add the record to the list. */
	    if (position_history == NULL)
		position_history = newrecord;
	    else
		record_ptr->next = newrecord;

	    record_ptr = newrecord;
3221
3222

	    /* Impose a limit, so the file will not grow indefinitely. */
3223
3224
3225
	    if (++count > 200) {
		poshiststruct *drop_record = position_history;

3226
		position_history = position_history->next;
3227
3228
3229
3230

		free(drop_record->filename);
		free(drop_record);
	    }
3231
3232
3233
	}
	fclose(hist);
	free(line);
3234
    }
3235
    free(poshist);
3236
}
3237
#endif /* !DISABLE_HISTORIES */