files.c 77.3 KB
Newer Older
Chris Allegretta's avatar
Chris Allegretta committed
1
/* $Id$ */
Chris Allegretta's avatar
Chris Allegretta committed
2
3
4
/**************************************************************************
 *   files.c                                                              *
 *                                                                        *
5
 *   Copyright (C) 1999-2005 Chris Allegretta                             *
Chris Allegretta's avatar
Chris Allegretta committed
6
7
 *   This program is free software; you can redistribute it and/or modify *
 *   it under the terms of the GNU General Public License as published by *
8
 *   the Free Software Foundation; either version 2, or (at your option)  *
Chris Allegretta's avatar
Chris Allegretta committed
9
10
 *   any later version.                                                   *
 *                                                                        *
11
12
13
14
 *   This program is distributed in the hope that it will be useful, but  *
 *   WITHOUT ANY WARRANTY; without even the implied warranty of           *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU    *
 *   General Public License for more details.                             *
Chris Allegretta's avatar
Chris Allegretta committed
15
16
17
 *                                                                        *
 *   You should have received a copy of the GNU General Public License    *
 *   along with this program; if not, write to the Free Software          *
18
19
 *   Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA            *
 *   02110-1301, USA.                                                     *
Chris Allegretta's avatar
Chris Allegretta committed
20
21
22
 *                                                                        *
 **************************************************************************/

23
24
25
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
26

Chris Allegretta's avatar
Chris Allegretta committed
27
28
29
30
#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <unistd.h>
Chris Allegretta's avatar
Chris Allegretta committed
31
#include <sys/wait.h>
32
#include <utime.h>
Chris Allegretta's avatar
Chris Allegretta committed
33
34
#include <fcntl.h>
#include <errno.h>
Chris Allegretta's avatar
Chris Allegretta committed
35
#include <ctype.h>
36
#include <pwd.h>
Chris Allegretta's avatar
Chris Allegretta committed
37
#include <assert.h>
Chris Allegretta's avatar
Chris Allegretta committed
38
#include "proto.h"
Chris Allegretta's avatar
Chris Allegretta committed
39

40
41
static file_format fmt = NIX_FILE;
	/* The format of the current file. */
42

Chris Allegretta's avatar
Chris Allegretta committed
43
44
45
/* What happens when there is no file to open? aiee! */
void new_file(void)
{
46
    fileage = make_new_node(NULL);
47
    fileage->data = mallocstrcpy(NULL, "");
Chris Allegretta's avatar
Chris Allegretta committed
48
49
50
    filebot = fileage;
    edittop = fileage;
    current = fileage;
51
    current_x = 0;
Chris Allegretta's avatar
Chris Allegretta committed
52
    totlines = 1;
53
    totsize = 0;
54

55
56
#ifdef ENABLE_COLOR
    update_color();
57
    if (!ISSET(NO_COLOR_SYNTAX))
58
	edit_refresh();
59
#endif
Chris Allegretta's avatar
Chris Allegretta committed
60
61
}

62
63
/* We make a new line of text from buf.  buf is length len.  If
 * first_line_ins is TRUE, then we put the new line at the top of the
64
65
66
67
 * file.  Otherwise, we assume prevnode is the last line of the file,
 * and put our line after prevnode. */
filestruct *read_line(char *buf, filestruct *prevnode, bool
	*first_line_ins, size_t len)
Chris Allegretta's avatar
Chris Allegretta committed
68
{
69
    filestruct *fileptr = (filestruct *)nmalloc(sizeof(filestruct));
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
70

71
72
    /* Convert nulls to newlines.  len is the string's real length
     * here. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
73
74
    unsunder(buf, len);

75
76
77
    assert(strlen(buf) == len);

    fileptr->data = mallocstrcpy(NULL, buf);
Chris Allegretta's avatar
Chris Allegretta committed
78

79
#ifndef NANO_SMALL
80
81
    /* If it's a DOS file (CR LF), and file conversion isn't disabled,
     * strip out the CR part. */
82
    if (!ISSET(NO_CONVERT) && len > 0 && buf[len - 1] == '\r')
83
	fileptr->data[len - 1] = '\0';
84
85
#endif

86
87
88
    if (*first_line_ins || fileage == NULL) {
	/* Special case: we're inserting with the cursor on the first
	 * line. */
Chris Allegretta's avatar
Chris Allegretta committed
89
90
91
	fileptr->prev = NULL;
	fileptr->next = fileage;
	fileptr->lineno = 1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
92
	if (*first_line_ins == TRUE) {
93
	    *first_line_ins = FALSE;
94
	    /* If we're inserting into the first line of the file, then
95
96
	     * we want to make sure that our edit buffer stays on the
	     * first line and that fileage stays up to date. */
97
98
99
	    edittop = fileptr;
	} else
	    filebot = fileptr;
Chris Allegretta's avatar
Chris Allegretta committed
100
	fileage = fileptr;
101
    } else {
102
	assert(prevnode != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
103

104
	fileptr->prev = prevnode;
Chris Allegretta's avatar
Chris Allegretta committed
105
	fileptr->next = NULL;
106
107
	fileptr->lineno = prevnode->lineno + 1;
	prevnode->next = fileptr;
Chris Allegretta's avatar
Chris Allegretta committed
108
109
110
111
112
    }

    return fileptr;
}

113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/* Load a file into the edit buffer.  This takes data from the file
 * struct. */
void load_file(void)
{
    current = fileage;

#ifdef ENABLE_MULTIBUFFER
    /* Add a new entry to the open_files structure. */
    add_open_file(FALSE);

    /* Reinitialize the shortcut list. */
    shortcut_init(FALSE);
#endif
}

void read_file(FILE *f, const char *filename)
Chris Allegretta's avatar
Chris Allegretta committed
129
{
130
131
    size_t num_lines = 0;
	/* The number of lines in the file. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
132
133
    size_t num_chars;
	/* The number of characters in the file. */
134
135
136
137
    size_t len = 0;
	/* The length of the current line of the file. */
    size_t i = 0;
	/* The position in the current line of the file. */
138
    size_t bufx = MAX_BUF_SIZE;
139
140
141
	/* The size of each chunk of the file that we read. */
    char input = '\0';
	/* The current input character. */
Chris Allegretta's avatar
Chris Allegretta committed
142
    char *buf;
143
144
145
146
147
148
149
150
	/* The buffer where we store chunks of the file. */
    filestruct *fileptr = current;
	/* The current line of the file. */
    bool first_line_ins = FALSE;
	/* Whether we're inserting with the cursor on the first line. */
    int input_int;
	/* The current value we read from the file, whether an input
	 * character or EOF. */
151
#ifndef NANO_SMALL
152
    int format = 0;
153
	/* 0 = *nix, 1 = DOS, 2 = Mac, 3 = both DOS and Mac. */
154
#endif
Chris Allegretta's avatar
Chris Allegretta committed
155

156
    buf = charalloc(bufx);
157
    buf[0] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
158

159
160
    if (current != NULL) {
	if (current == fileage)
161
	    first_line_ins = TRUE;
162
163
	else
	    fileptr = current->prev;
Chris Allegretta's avatar
Chris Allegretta committed
164
    }
165

166
167
    /* For the assertion in read_line(), it must be true that if current
     * is NULL, then so is fileage. */
168
169
    assert(current != NULL || fileage == NULL);

170
#ifndef NANO_SMALL
171
172
173
    /* We don't know which file format we have yet, so assume it's a
     * *nix file for now. */
    fmt = NIX_FILE;
174
175
#endif

176
    /* Read the entire file into the file struct. */
177
    while ((input_int = getc(f)) != EOF) {
178
	input = (char)input_int;
Chris Allegretta's avatar
Chris Allegretta committed
179

180
181
	/* If it's a *nix file (LF) or a DOS file (CR LF), and file
	 * conversion isn't disabled, handle it! */
182
	if (input == '\n') {
183
#ifndef NANO_SMALL
184
185
	    /* If there's a CR before the LF, set format to DOS if we
	     * currently think this is a *nix file, or to both if we
186
187
	     * currently think it's a Mac file. */
	    if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r' &&
188
189
		(format == 0 || format == 2))
		format++;
190
191
192
193
#endif

	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
194

195
196
	    /* Reset the line length in preparation for the next
	     * line. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
197
198
	    len = 0;

199
	    num_lines++;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
200
	    buf[0] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
201
	    i = 0;
Chris Allegretta's avatar
Chris Allegretta committed
202
#ifndef NANO_SMALL
203
204
	/* If it's a Mac file (CR without an LF), and file conversion
	 * isn't disabled, handle it! */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
205
206
	} else if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r') {

207
208
209
210
211
	    /* 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;
212
213
214

	    /* Read in the line properly. */
	    fileptr = read_line(buf, fileptr, &first_line_ins, len);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
215

216
217
218
	    /* Reset the line length in preparation for the next line.
	     * Since we've already read in the next character, reset it
	     * to 1 instead of 0. */
219
	    len = 1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
220

Chris Allegretta's avatar
Chris Allegretta committed
221
	    num_lines++;
222
	    buf[0] = input;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
223
	    buf[1] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
224
225
	    i = 1;
#endif
Chris Allegretta's avatar
Chris Allegretta committed
226
	} else {
227
228
	    /* Calculate the total length of the line.  It might have
	     * nulls in it, so we can't just use strlen() here. */
229
230
	    len++;

231
232
233
234
235
	    /* Now we allocate a bigger buffer MAX_BUF_SIZE characters
	     * at a time.  If we allocate a lot of space for one line,
	     * we may indeed have to use a buffer this big later on, so
	     * we don't decrease it at all.  We do free it at the end,
	     * though. */
Chris Allegretta's avatar
Chris Allegretta committed
236
	    if (i >= bufx - 1) {
237
		bufx += MAX_BUF_SIZE;
David Lawrence Ramsey's avatar
   
David Lawrence Ramsey committed
238
		buf = charealloc(buf, bufx);
Chris Allegretta's avatar
Chris Allegretta committed
239
	    }
240

241
	    buf[i] = input;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
242
	    buf[i + 1] = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
243
244
	    i++;
	}
245
246
    }

247
    /* Perhaps this could use some better handling. */
248
249
250
    if (ferror(f))
	nperror(filename);
    fclose(f);
Chris Allegretta's avatar
Chris Allegretta committed
251

252
#ifndef NANO_SMALL
253
254
255
    /* If file conversion isn't disabled and the last character in this
     * file is a CR, read it in properly as a Mac format line. */
    if (len == 0 && !ISSET(NO_CONVERT) && input == '\r') {
256
257
	len = 1;

258
259
	buf[0] = input;
	buf[1] = '\0';
260
261
    }
#endif
262

263
264
    /* Did we not get a newline and still have stuff to do? */
    if (len > 0) {
265
#ifndef NANO_SMALL
266
	/* If file conversion isn't disabled and the last character in
267
268
	 * this file is a CR, set format to Mac if we currently think
	 * the file is a *nix file, or to both DOS and Mac if we
269
270
	 * currently think the file is a DOS file. */
	if (!ISSET(NO_CONVERT) && buf[len - 1] == '\r' &&
271
272
		(format == 0 || format == 1))
	    format += 2;
273
274
#endif

275
276
277
278
	/* Read in the last line properly. */
	fileptr = read_line(buf, fileptr, &first_line_ins, len);
	num_lines++;
    }
279

280
281
282
283
    free(buf);

    /* If we didn't get a file and we don't already have one, make a new
     * file. */
284
    if (fileptr == NULL)
Chris Allegretta's avatar
Chris Allegretta committed
285
	new_file();
Robert Siemborski's avatar
Robert Siemborski committed
286

287
    /* Did we try to insert a file of 0 bytes? */
288
289
290
291
292
293
294
295
296
297
    if (num_lines != 0) {
	if (current != NULL) {
	    fileptr->next = current;
	    current->prev = fileptr;
	    renumber(current);
	    current_x = 0;
	    placewewant = 0;
	} else if (fileptr->next == NULL) {
	    filebot = fileptr;
	    new_magicline();
298
	    totsize--;
299
	}
Chris Allegretta's avatar
Chris Allegretta committed
300
    }
301

302
303
304
    get_totals(fileage, filebot, NULL, &num_chars);
    totsize += num_chars;

305
#ifndef NANO_SMALL
306
    if (format == 3)
307
308
309
310
	statusbar(
		P_("Read %lu line (Converted from DOS and Mac format)",
		"Read %lu lines (Converted from DOS and Mac format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
311
312
    else if (format == 2) {
	fmt = MAC_FILE;
313
314
315
	statusbar(P_("Read %lu line (Converted from Mac format)",
		"Read %lu lines (Converted from Mac format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
316
317
    } else if (format == 1) {
	fmt = DOS_FILE;
318
319
320
	statusbar(P_("Read %lu line (Converted from DOS format)",
		"Read %lu lines (Converted from DOS format)",
		(unsigned long)num_lines), (unsigned long)num_lines);
321
    } else
322
#endif
323
	statusbar(P_("Read %lu line", "Read %lu lines",
324
		(unsigned long)num_lines), (unsigned long)num_lines);
325

326
    totlines += num_lines;
Chris Allegretta's avatar
Chris Allegretta committed
327
328
}

329
330
331
332
333
334
335
336
/* Open the file (and decide if it exists).  If newfie is TRUE, display
 * "New File" if the file is missing.  Otherwise, say "[filename] not
 * found".
 *
 * Return -2 if we say "New File".  Otherwise, -1 if the file isn't
 * opened, 0 otherwise.  The file might still have an error while
 * reading with a 0 return value.  *f is set to the opened file. */
int open_file(const char *filename, bool newfie, FILE **f)
Chris Allegretta's avatar
Chris Allegretta committed
337
338
339
340
{
    int fd;
    struct stat fileinfo;

341
    assert(f != NULL);
342

343
344
345
    if (filename == NULL || filename[0] == '\0' ||
	    stat(filename, &fileinfo) == -1) {
	if (newfie) {
Chris Allegretta's avatar
Chris Allegretta committed
346
	    statusbar(_("New File"));
347
	    return -2;
Chris Allegretta's avatar
Chris Allegretta committed
348
	}
349
350
	statusbar(_("\"%s\" not found"), filename);
	return -1;
351
352
353
    } else if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) ||
		S_ISBLK(fileinfo.st_mode)) {
	/* Don't open character or block files.  Sorry, /dev/sndstat! */
354
355
	statusbar(S_ISDIR(fileinfo.st_mode) ? _("\"%s\" is a directory")
		: _("File \"%s\" is a device file"), filename);
356
	return -1;
Chris Allegretta's avatar
Chris Allegretta committed
357
    } else if ((fd = open(filename, O_RDONLY)) == -1) {
358
359
360
	statusbar(_("Error reading %s: %s"), filename, strerror(errno));
 	return -1;
     } else {
361
362
363
	/* File is A-OK.  Open it in binary mode for our own end-of-line
	 * character munging. */
	*f = fdopen(fd, "rb");
364
365

	if (*f == NULL) {
366
367
	    statusbar(_("Error reading %s: %s"), filename,
		strerror(errno));
368
	    close(fd);
369
370
	} else
	    statusbar(_("Reading File"));
Chris Allegretta's avatar
Chris Allegretta committed
371
    }
372
    return 0;
Chris Allegretta's avatar
Chris Allegretta committed
373
374
}

375
/* This function will return the name of the first available extension
376
 * of a filename (starting with [name][suffix], then [name][suffix].1,
377
 * etc.).  Memory is allocated for the return value.  If no writable
378
 * extension exists, we return "". */
379
char *get_next_filename(const char *name, const char *suffix)
380
{
381
    unsigned long i = 0;
382
    char *buf;
383
    size_t namelen, suffixlen;
384

385
386
387
388
389
390
391
    assert(name != NULL && suffix != NULL);

    namelen = strlen(name);
    suffixlen = strlen(suffix);

    buf = charalloc(namelen + suffixlen + digits(ULONG_MAX) + 2);
    sprintf(buf, "%s%s", name, suffix);
392

393
    while (TRUE) {
394
	struct stat fs;
395
396

	if (stat(buf, &fs) == -1)
Chris Allegretta's avatar
Chris Allegretta committed
397
	    return buf;
398
	if (i == ULONG_MAX)
399
400
401
	    break;

	i++;
402
	sprintf(buf + namelen + suffixlen, ".%lu", i);
403
404
    }

Chris Allegretta's avatar
Chris Allegretta committed
405
    /* We get here only if there is no possible save file. */
406
    null_at(&buf, 0);
407

408
409
410
    return buf;
}

411
412
#ifndef NANO_SMALL
void execute_command(const char *command)
Chris Allegretta's avatar
Chris Allegretta committed
413
{
414
415
416
417
418
419
420
#ifdef ENABLE_MULTIBUFFER
    if (ISSET(MULTIBUFFER)) {
	/* Update the current entry in the open_files structure. */
	add_open_file(TRUE);
	new_file();
	UNSET(MODIFIED);
	UNSET(MARK_ISSET);
421
    }
422
423
424
425
426
427
428
429
430
#endif /* ENABLE_MULTIBUFFER */
    open_pipe(command);
#ifdef ENABLE_MULTIBUFFER
    /* Add this new entry to the open_files structure. */
    if (ISSET(MULTIBUFFER))
	load_file();
#endif /* ENABLE_MULTIBUFFER */
}
#endif /* !NANO_SMALL */
Chris Allegretta's avatar
Chris Allegretta committed
431

432
433
434
435
/* name is a file name to open.  We make a new buffer if necessary, then
 * open and read the file. */
void load_buffer(const char *name)
{
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
436
    bool new_buffer = (fileage == NULL
437
438
#ifdef ENABLE_MULTIBUFFER
	 || ISSET(MULTIBUFFER)
439
#endif
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
440
	);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
441
442
443
444
	/* new_buffer says whether we load into this buffer or a new
	 * one.  If new_buffer is TRUE, we display "New File" if the
	 * file is not found, and if it is found we set filename and add
	 * a new open_files entry. */
445
446
447
448
    FILE *f;
    int rc;
	/* rc == -2 means that the statusbar displayed "New File".  -1
	 * means that the open failed.  0 means success. */
449

450
451
#ifndef DISABLE_OPERATINGDIR
    if (check_operating_dir(name, FALSE)) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
452
453
	statusbar(_("Can't insert file from outside of %s"),
		operating_dir);
454
455
	return;
    }
456
457
#endif

458
459
460
#ifdef ENABLE_MULTIBUFFER
    /* Update the current entry in the open_files structure. */
    add_open_file(TRUE);
461
462
#endif

463
464
465
466
467
    rc = open_file(name, new_buffer, &f);

#ifdef ENABLE_MULTIBUFFER
    if (rc != -1 && ISSET(MULTIBUFFER)) {
	UNSET(MODIFIED);
Chris Allegretta's avatar
Chris Allegretta committed
468
#ifndef NANO_SMALL
469
	UNSET(MARK_ISSET);
Chris Allegretta's avatar
Chris Allegretta committed
470
#endif
471
    }
472
#endif
473
474
475
476
477
478
479

    if (rc != -1 && new_buffer) {
	filename = mallocstrcpy(filename, name);
	new_file();
    }

    if (rc == 0) {
480
481
	file_format fmt_save = fmt;

482
	read_file(f, filename);
483
484
485
486
487
488

	/* If we're not loading into a new buffer, preserve the file
	 * format. */
	if (!new_buffer)
	    fmt = fmt_save;

Chris Allegretta's avatar
Chris Allegretta committed
489
#ifndef NANO_SMALL
490
	stat(filename, &originalfilestat);
Chris Allegretta's avatar
Chris Allegretta committed
491
#endif
492
493
494
495
496
497
498
499
    }

    /* Add this new entry to the open_files structure if we have
     * multibuffer support, or to the main filestruct if we don't. */
    if (rc != -1 && new_buffer)
	load_file();
}

500
501
502
503
504
505
506
void do_insertfile(
#ifndef NANO_SMALL
	bool execute
#else
	void
#endif
	)
507
508
509
{
    int i;
    const char *msg;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
510
    char *ans = mallocstrcpy(NULL, "");
511
	/* The last answer the user typed on the statusbar. */
512
    filestruct *edittop_save = edittop;
513
    int current_y_save = current_y;
514
515
    bool at_edittop = FALSE;
	/* Whether we're at the top of the edit window. */
516

517
#ifndef DISABLE_WRAPPING
518
    wrap_reset();
519
#endif
520

521
    while (TRUE) {
522
#ifndef NANO_SMALL
523
	if (execute) {
524
#ifdef ENABLE_MULTIBUFFER
525
526
527
	    if (ISSET(MULTIBUFFER))
		msg = N_("Command to execute in new buffer [from %s] ");
	    else
528
#endif
529
530
		msg = N_("Command to execute [from %s] ");
	} else {
531
532
#endif
#ifdef ENABLE_MULTIBUFFER
533
534
535
	    if (ISSET(MULTIBUFFER)) {
		msg = N_("File to insert into new buffer [from %s] ");
	    } else
536
#endif
537
		msg = N_("File to insert [from %s] ");
538
539
#ifndef NANO_SMALL
	}
540
#endif
541

542
	i = statusq(TRUE,
543
#ifndef NANO_SMALL
544
		execute ? extcmd_list :
545
546
#endif
		insertfile_list, ans,
Chris Allegretta's avatar
Chris Allegretta committed
547
#ifndef NANO_SMALL
548
		NULL,
Chris Allegretta's avatar
Chris Allegretta committed
549
#endif
550
551
552
553
		_(msg),
#ifndef DISABLE_OPERATINGDIR
		operating_dir != NULL && strcmp(operating_dir, ".") != 0 ?
		operating_dir :
554
#endif
555
		"./");
556

557
558
	/* If we're in multibuffer mode and the filename or command is
	 * blank, open a new buffer instead of canceling. */
559
	if (i == -1 || (i == -2
560
#ifdef ENABLE_MULTIBUFFER
561
		&& !ISSET(MULTIBUFFER)
562
#endif
563
		)) {
564
565
566
	    statusbar(_("Cancelled"));
	    break;
	} else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
567
	    size_t pww_save = placewewant;
568

569
	    ans = mallocstrcpy(ans, answer);
Chris Allegretta's avatar
Chris Allegretta committed
570

571
#if !defined(NANO_SMALL) && defined(ENABLE_MULTIBUFFER)
572
573
574
575
576
577
	    if (i == TOGGLE_MULTIBUFFER_KEY) {
		/* Don't allow toggling if we're in view mode. */
		if (!ISSET(VIEW_MODE))
		    TOGGLE(MULTIBUFFER);
		continue;
	    }
578
#endif
579

580
#ifndef DISABLE_BROWSER
581
582
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
583

584
585
		if (tmp == NULL)
		    continue;
586

587
588
		free(answer);
		answer = tmp;
589
590
591
592

		/* We have a file now.  Get out of the statusbar prompt
		 * cleanly. */
		statusq_abort();
593
	    }
Chris Allegretta's avatar
Chris Allegretta committed
594
595
#endif

596
#ifndef NANO_SMALL
597
598
599
600
	    if (i == NANO_TOOTHERINSERT_KEY) {
		execute = !execute;
		continue;
	    }
601
#endif
602

603
604
	    /* If we don't have a file yet, go back to the statusbar
	     * prompt. */
605
606
607
608
609
	    if (i != 0
#ifdef ENABLE_MULTIBUFFER
		&& (i != -2 || !ISSET(MULTIBUFFER))
#endif
		)
610
611
		continue;

612
#ifdef ENABLE_MULTIBUFFER
613
	    if (!ISSET(MULTIBUFFER)) {
614
#endif
615
616
		/* If we're not inserting into a new buffer, partition
		 * the filestruct so that it contains no text and hence
617
618
619
		 * looks like a new buffer, and keep track of whether
		 * the top of the partition is the top of the edit
		 * window. */
620
		filepart = partition_filestruct(current, current_x,
621
			current, current_x);
622
		at_edittop = (fileage == edittop);
623
#ifdef ENABLE_MULTIBUFFER
624
	    }
625
626
#endif

627
628
629
630
631
#ifndef NANO_SMALL
	    if (execute)
		execute_command(answer);
	    else {
#endif
632
633
		answer = mallocstrassn(answer,
			real_dir_from_tilde(answer));
634
		load_buffer(answer);
635
636
637
#ifndef NANO_SMALL
	    }
#endif
638
639

#ifdef ENABLE_MULTIBUFFER
640
	    if (!ISSET(MULTIBUFFER))
641
#endif
642
643
	    {
		filestruct *top_save = fileage;
644

645
646
647
648
649
650
651
652
653
		/* If we didn't insert into a new buffer, and we were at
		 * the top of the edit window before, set the saved
		 * value of edittop to the new top of the edit window,
		 * and update the current y-coordinate to account for
		 * the number of lines inserted. */
		if (at_edittop)
		    edittop_save = fileage;
		current_y += current_y_save;

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
654
		/* If we didn't insert into a new buffer, unpartition
655
656
657
658
		 * 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. */
659
		unpartition_filestruct(&filepart);
660

661
662
663
		/* Renumber starting with the beginning line of the old
		 * partition. */
		renumber(top_save);
664

665
666
		/* Set edittop back to what it was before. */
		edittop = edittop_save;
667
	    }
668

669
#ifdef ENABLE_MULTIBUFFER
670
671
672
	    if (ISSET(MULTIBUFFER)) {
		/* Update the titlebar. */
		titlebar(NULL);
673

674
675
676
		/* Reinitialize the shortcut list. */
		shortcut_init(FALSE);
	    } else {
677
#endif
678
679
		/* Mark the file as modified. */
		set_modified();
Chris Allegretta's avatar
Chris Allegretta committed
680

681
		/* Restore the old place we want. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
682
		placewewant = pww_save;
683
#ifdef ENABLE_MULTIBUFFER
684
	    }
Chris Allegretta's avatar
Chris Allegretta committed
685
686
#endif

687
688
689
690
691
	    /* Refresh the screen. */
	    edit_refresh();

	    break;
	}
692
    }
Chris Allegretta's avatar
Chris Allegretta committed
693

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
694
    free(ans);
Chris Allegretta's avatar
Chris Allegretta committed
695
696
}

697
void do_insertfile_void(void)
698
{
699
#ifdef ENABLE_MULTIBUFFER
700
701
    if (ISSET(VIEW_MODE) && !ISSET(MULTIBUFFER))
	statusbar(_("Key illegal in non-multibuffer mode"));
702
    else
703
#endif
704
705
706
707
708
	do_insertfile(
#ifndef NANO_SMALL
		FALSE
#endif
		);
709
710
711
712

    display_main_list();
}

713
#ifdef ENABLE_MULTIBUFFER
Chris Allegretta's avatar
Chris Allegretta committed
714
/* Create a new openfilestruct node. */
715
openfilestruct *make_new_opennode(void)
Chris Allegretta's avatar
Chris Allegretta committed
716
{
717
718
    openfilestruct *newnode =
	(openfilestruct *)nmalloc(sizeof(openfilestruct));
Chris Allegretta's avatar
Chris Allegretta committed
719
    newnode->filename = NULL;
720

Chris Allegretta's avatar
Chris Allegretta committed
721
722
723
724
725
    return newnode;
}

/* Splice a node into an existing openfilestruct. */
void splice_opennode(openfilestruct *begin, openfilestruct *newnode,
726
	openfilestruct *end)
Chris Allegretta's avatar
Chris Allegretta committed
727
{
728
    assert(newnode != NULL && begin != NULL);
729

Chris Allegretta's avatar
Chris Allegretta committed
730
731
732
733
734
735
736
    newnode->next = end;
    newnode->prev = begin;
    begin->next = newnode;
    if (end != NULL)
	end->prev = newnode;
}

737
738
/* Unlink a node from the rest of the openfilestruct, and delete it. */
void unlink_opennode(openfilestruct *fileptr)
Chris Allegretta's avatar
Chris Allegretta committed
739
{
740
    assert(fileptr != NULL && fileptr->prev != NULL && fileptr->next != NULL && fileptr != fileptr->prev && fileptr != fileptr->next);
741

742
743
744
    fileptr->prev->next = fileptr->next;
    fileptr->next->prev = fileptr->prev;
    delete_opennode(fileptr);
Chris Allegretta's avatar
Chris Allegretta committed
745
746
747
748
749
}

/* Delete a node from the openfilestruct. */
void delete_opennode(openfilestruct *fileptr)
{
750
    assert(fileptr != NULL && fileptr->filename != NULL && fileptr->fileage != NULL);
751

752
753
754
    free(fileptr->filename);
    free_filestruct(fileptr->fileage);
    free(fileptr);
Chris Allegretta's avatar
Chris Allegretta committed
755
756
}

757
758
759
#ifdef DEBUG
/* Deallocate all memory associated with this and later files, including
 * the lines of text. */
Chris Allegretta's avatar
Chris Allegretta committed
760
761
void free_openfilestruct(openfilestruct *src)
{
762
    assert(src != NULL);
763

764
765
766
    while (src != src->next) {
	src = src->next;
	delete_opennode(src->prev);
Chris Allegretta's avatar
Chris Allegretta committed
767
    }
768
    delete_opennode(src);
Chris Allegretta's avatar
Chris Allegretta committed
769
}
770
#endif
Chris Allegretta's avatar
Chris Allegretta committed
771

772
/* Add/update an entry to the open_files openfilestruct.  If update is
773
 * FALSE, a new entry is created; otherwise, the current entry is
774
 * updated. */
775
void add_open_file(bool update)
776
{
777
    if (update && open_files == NULL)
778
	return;
779

780
781
782
783
784
785
786
787
788
    /* If there are no entries in open_files, make the first one. */
    if (open_files == NULL) {
	open_files = make_new_opennode();
	splice_opennode(open_files, open_files, open_files);
    /* Otherwise, if we're not updating, make a new entry for
     * open_files and splice it in after the current entry. */
    } else if (!update) {
	splice_opennode(open_files, make_new_opennode(),
		open_files->next);
789
790
791
	open_files = open_files->next;
    }

792
    /* Save the current filename. */
793
    open_files->filename = mallocstrcpy(open_files->filename, filename);
794

795
#ifndef NANO_SMALL
796
    /* Save the current file's stat. */
797
798
799
    open_files->originalfilestat = originalfilestat;
#endif

800
801
802
803
804
805
806
807
808
    /* Save the current file buffer. */
    open_files->fileage = fileage;
    open_files->filebot = filebot;

    /* Save the current top of the edit window. */
    open_files->edittop = edittop;

    /* Save the current line. */
    open_files->current = current;
809

810
811
    /* Save the current cursor position. */
    open_files->current_x = current_x;
812

813
814
    /* Save the current place we want. */
    open_files->placewewant = placewewant;
815

816
817
    /* Save the current total number of lines. */
    open_files->totlines = totlines;
818

819
820
    /* Save the current total size. */
    open_files->totsize = totsize;
821

822
823
    /* Start with no flags saved. */
    open_files->flags = 0;
824

825
826
827
    /* Save the current modification status. */
    if (ISSET(MODIFIED))
	open_files->flags |= MODIFIED;
828

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
829
#ifndef NANO_SMALL
830
831
832
833
834
    /* Save the current marking status and mark, if applicable. */
    if (ISSET(MARK_ISSET)) {
	open_files->flags |= MARK_ISSET;
	open_files->mark_beginbuf = mark_beginbuf;
	open_files->mark_beginx = mark_beginx;
Chris Allegretta's avatar
Chris Allegretta committed
835
836
    }

837
838
839
    /* Save the current file format. */
    open_files->fmt = fmt;
#endif
840
841

#ifdef DEBUG
842
    fprintf(stderr, "filename is %s\n", open_files->filename);
843
844
845
#endif
}

846
/* Read the current entry in the open_files structure and set up the
847
 * currently open file buffer using that entry's information. */
848
void load_open_file(void)
849
{
850
    assert(open_files != NULL);
851

852
    /* Restore the current filename. */
853
    filename = mallocstrcpy(filename, open_files->filename);
854

855
#ifndef NANO_SMALL
856
    /* Restore the current file's stat. */
857
858
    originalfilestat = open_files->originalfilestat;
#endif
859
860

    /* Restore the current file buffer. */
861
    fileage = open_files->fileage;
Chris Allegretta's avatar
Chris Allegretta committed
862
    filebot = open_files->filebot;
863

864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
    /* Restore the current top of the edit window. */
    edittop = open_files->edittop;

    /* Restore the current line. */
    current = open_files->current;

    /* Restore the current cursor position. */
    current_x = open_files->current_x;

    /* Restore the current place we want. */
    placewewant = open_files->placewewant;

    /* Restore the current total number of lines. */
    totlines = open_files->totlines;

    /* Restore the current total size. */
    totsize = open_files->totsize;

    /* Restore the current modification status. */
    if (open_files->flags & MODIFIED)
Chris Allegretta's avatar
Chris Allegretta committed
884
885
886
887
888
	SET(MODIFIED);
    else
	UNSET(MODIFIED);

#ifndef NANO_SMALL
889
890
891
892
    /* Restore the current marking status and mark, if applicable. */
    if (open_files->flags & MARK_ISSET) {
	mark_beginbuf = open_files->mark_beginbuf;
	mark_beginx = open_files->mark_beginx;
Chris Allegretta's avatar
Chris Allegretta committed
893
894
895
	SET(MARK_ISSET);
    } else
	UNSET(MARK_ISSET);
896

897
898
    /* Restore the current file format. */
    fmt = open_files->fmt;
Chris Allegretta's avatar
Chris Allegretta committed
899
#endif
900

Chris Allegretta's avatar
Chris Allegretta committed
901
902
903
#ifdef ENABLE_COLOR
    update_color();
#endif
904
    edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
905

906
    /* Update the titlebar. */
907
908
909
    titlebar(NULL);
}

910
/* Open either the next or previous file buffer. */
911
void open_prevnext_file(bool next)
912
{
913
    add_open_file(TRUE);
914

915
    assert(open_files != NULL);
916

917
918
    /* If only one file buffer is open, indicate it on the statusbar and
     * get out. */
919
    if (open_files == open_files->next) {
920
	statusbar(_("No more open file buffers"));
921
	return;
922
923
    }

924
925
926
    /* Switch to the next or previous file, depending on the value of
     * next. */
    open_files = next ? open_files->next : open_files->prev;
927
928

#ifdef DEBUG
929
    fprintf(stderr, "filename is %s\n", open_files->filename);
930
931
#endif

932
    /* Load the file we switched to. */
933
934
    load_open_file();

935
    /* And indicate the switch on the statusbar. */
936
    statusbar(_("Switched to %s"),
937
      ((open_files->filename[0] == '\0') ? _("New Buffer") :
938
	open_files->filename));
939

940
941
942
943
944
#ifdef DEBUG
    dump_buffer(current);
#endif
}

945
946
/* Open the previous entry in the open_files structure.  This function
 * is used by the shortcut list. */
947
void open_prevfile_void(void)
948
{
949
    open_prevnext_file(FALSE);
950
951
}

952
953
/* Open the next entry in the open_files structure.  This function is
 * used by the shortcut list. */
954
void open_nextfile_void(void)
955
{
956
    open_prevnext_file(TRUE);
957
958
}

959
960
/* Delete an entry from the open_files filestruct.  After deletion of an
 * entry, the next entry is opened.  Return TRUE on success or FALSE if
961
 * there are no more open file buffers. */
962
bool close_open_file(void)
963
{
964
    assert(open_files != NULL);
965

966
967
    /* If only one file is open, get out. */
    if (open_files == open_files->next)
968
	return FALSE;
969

970
971
    /* Open the next file. */
    open_nextfile_void();
Chris Allegretta's avatar
Chris Allegretta committed
972

973
974
    /* Close the file we had open before. */
    unlink_opennode(open_files->prev);
975

976
    /* Reinitialize the shortcut list. */
977
    shortcut_init(FALSE);
978
    display_main_list();
979

980
    return TRUE;
981
}
982
#endif /* ENABLE_MULTIBUFFER */
983

984
#if !defined(DISABLE_SPELLER) || !defined(DISABLE_OPERATINGDIR)
985
/* When passed "[relative path]" or "[relative path][filename]" in
986
 * origpath, return "[full path]" or "[full path][filename]" on success,
987
988
989
990
 * 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. */
991
char *get_full_path(const char *origpath)
992
{
993
    char *d_here, *d_there = NULL;
994

995
996
    if (origpath == NULL)
    	return NULL;
997

998
999
1000
    /* Get the current directory. */
    d_here = charalloc(PATH_MAX + 1);
    d_here = getcwd(d_here, PATH_MAX + 1);
1001

1002
    if (d_here != NULL) {
1003
1004
1005
1006
	const char *last_slash;
	char *d_there_file = NULL;
	bool path_only;
	struct stat fileinfo;
1007

1008
1009
	align(&d_here);

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

1017
1018
1019
1020
1021
1022
1023
1024
1025
1026
1027
1028
	d_there = real_dir_from_tilde(origpath);

	assert(d_there != NULL);

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

	/* If path_only is TRUE, make sure d_there ends in a slash. */
1029
	if (path_only) {
1030
1031
1032
1033
	    size_t d_there_len = strlen(d_there);

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

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

1041
1042
1043
1044
	/* 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);
1045

1046
1047
1048
1049
1050
1051
1052
1053
1054
1055
1056
1057
1058
1059
1060
	    d_there_file = d_there;
	    d_there = d_here;
	} else {
	    /* If path_only is FALSE, then save the filename portion of
	     * the answer, everything after the last slash, in
	     * d_there_file. */
	    if (!path_only)
		d_there_file = mallocstrcpy(NULL, last_slash + 1);

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

	    /* Go to the path specified in d_there. */
	    if (chdir(d_there) == -1) {
1061
		free(d_there);
1062
1063
1064
1065
		d_there = NULL;
	    } else {
		/* Get the full path and save it in d_there. */
		free(d_there);
1066

1067
1068
1069
		d_there = charalloc(PATH_MAX + 1);
		d_there = getcwd(d_there, PATH_MAX + 1);

1070
1071
1072
1073
1074
1075
1076
1077
1078
1079
		if (d_there != NULL) {
		    align(&d_there);

		    if (strcmp(d_there, "/") != 0) {
			/* Make sure d_there ends in a slash. */
			d_there = charealloc(d_there,
				strlen(d_there) + 2);
			strcat(d_there, "/");
		    }
		} else
1080
1081
1082
1083
1084
1085
1086
1087
		    /* If we couldn't get the full path, set path_only
		     * to TRUE so that we clean up correctly, free all
		     * allocated memory, and return NULL. */
		    path_only = TRUE;

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

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

1094
1095
1096
1097
1098
	/* At this point, if path_only is FALSE and d_there exists,
	 * d_there contains the path portion of the answer and
	 * d_there_file contains the filename portion of the answer.  If
	 * this is the case, tack d_there_file onto the end of
	 * d_there, so that d_there contains the complete answer. */
1099
	if (!path_only && d_there != NULL) {
1100
1101
1102
1103
1104
1105
	    d_there = charealloc(d_there, strlen(d_there) +
		strlen(d_there_file) + 1);
	    strcat(d_there, d_there_file);
 	}

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

1109
    return d_there;
1110
}
1111
#endif /* !DISABLE_SPELLER || !DISABLE_OPERATINGDIR */
1112
1113

#ifndef DISABLE_SPELLER
1114
1115
1116
/* Return the full version of path, as returned by get_full_path().  On
 * error, if path doesn't reference a directory, or if the directory
 * isn't writable, return NULL. */
1117
char *check_writable_directory(const char *path)
1118
{
1119
1120
    char *full_path = get_full_path(path);

1121
    /* If get_full_path() fails, return NULL. */
1122
    if (full_path == NULL)
1123
	return NULL;
1124

1125
1126
1127
    /* If we can't write to path or path isn't a directory, return
     * NULL. */
    if (access(full_path, W_OK) != 0 ||
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1128
	full_path[strlen(full_path) - 1] != '/') {
1129
	free(full_path);
1130
	return NULL;
1131
    }
1132
1133
1134
1135
1136

    /* otherwise, return the full path */
    return full_path;
}

1137
1138
1139
1140
/* This function acts like a call to tempnam(NULL, "nano.").  The
 * difference is that the number of calls is not limited by TMP_MAX.
 * Instead we use mkstemp(). */
char *safe_tempnam(void)
1141
{
1142
1143
    char *full_tempdir = NULL;
    const char *TMPDIR_env;
1144
    int filedesc;
1145

1146
1147
1148
    /* If $TMPDIR is set and non-empty, set tempdir to it, run it
     * through get_full_path(), and save the result in full_tempdir.
     * Otherwise, leave full_tempdir set to NULL. */
1149
    TMPDIR_env = getenv("TMPDIR");
1150
    if (TMPDIR_env != NULL && TMPDIR_env[0] != '\0')
1151
	full_tempdir = check_writable_directory(TMPDIR_env);
1152

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

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

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

1166
1167
    /* If mkstemp() succeeded, close the resulting file, delete it
     * (since it'll be 0 bytes long), and return the filename. */
1168
    if (filedesc != -1) {
1169
	close(filedesc);
1170
1171
	unlink(full_tempdir);
	return full_tempdir;
1172
    }
1173
1174

    free(full_tempdir);
1175

1176
    return NULL;
1177
1178
}
#endif /* !DISABLE_SPELLER */
1179
1180

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

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

1189
1190
1191
    full_operating_dir = get_full_path(operating_dir);

    /* If get_full_path() failed or the operating directory is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1192
     * inaccessible, unset operating_dir. */
1193
    if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) {
1194
1195
1196
1197
1198
1199
1200
	free(full_operating_dir);
	full_operating_dir = NULL;
	free(operating_dir);
	operating_dir = NULL;
    }
}

1201
1202
1203
1204
1205
/* 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)
1206
{
1207
1208
1209
1210
    /* The char *full_operating_dir is global for mem cleanup.  It
     * should have already been initialized by init_operating_dir().
     * Also, a relative operating directory path will only be handled
     * properly if this is done. */
1211

1212
    char *fullpath;
1213
    bool retval = FALSE;
1214
    const char *whereami1, *whereami2 = NULL;
1215

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

1220
    assert(full_operating_dir != NULL);
1221
1222

    fullpath = get_full_path(currpath);
1223
1224

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

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

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

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

1252
1253
1254
1255
1256
1257
1258
1259
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275
#ifndef NANO_SMALL
void init_backup_dir(void)
{
    char *full_backup_dir;

    if (backup_dir == NULL)
	return;

    full_backup_dir = get_full_path(backup_dir);

    /* If get_full_path() failed or the backup directory is
     * inaccessible, unset backup_dir. */
    if (full_backup_dir == NULL ||
	full_backup_dir[strlen(full_backup_dir) - 1] != '/') {
	free(full_backup_dir);
	free(backup_dir);
	backup_dir = NULL;
    } else {
	free(backup_dir);
	backup_dir = full_backup_dir;
    }
}
#endif

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

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

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

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

1304
1305
1306
    return retval;
}

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

1352
    assert(name != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1353

1354
    if (name[0] == '\0')
Chris Allegretta's avatar
Chris Allegretta committed
1355
	return -1;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1356

1357
1358
    if (!tmp)
	titlebar(NULL);
1359

1360
    realname = real_dir_from_tilde(name);
Chris Allegretta's avatar
Chris Allegretta committed
1361

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

1371
    anyexists = (lstat(realname, &lst) != -1);
1372

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1373
    /* If the temp file exists, give up. */
1374
1375
    if (tmp && anyexists)
	goto cleanup_and_exit;
1376

1377
1378
1379
    /* If NOFOLLOW_SYMLINKS is set, it doesn't make sense to prepend or
     * append to a symlink.  Here we warn about the contradiction. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode)) {
1380
1381
	statusbar(
		_("Cannot prepend or append to a symlink with --nofollow set"));
1382
1383
1384
	goto cleanup_and_exit;
    }

1385
    /* Save the state of file at the end of the symlink (if there is
1386
     * one). */
1387
    realexists = (stat(realname, &st) != -1);
1388

1389
1390
#ifndef NANO_SMALL
    /* We backup only if the backup toggle is set, the file isn't
1391
1392
1393
1394
     * 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. */
1395
    if (ISSET(BACKUP_FILE) && !tmp && realexists &&
1396
1397
	(append != 0 || ISSET(MARK_ISSET) ||
	originalfilestat.st_mtime == st.st_mtime)) {
1398
	FILE *backup_file;
1399
	char *backupname;
1400
	struct utimbuf filetime;
1401
	int copy_status;
1402

1403
	/* Save the original file's access and modification times. */
1404
1405
1406
	filetime.actime = originalfilestat.st_atime;
	filetime.modtime = originalfilestat.st_mtime;

1407
	/* Open the original file to copy to the backup. */
1408
	f = fopen(realname, "rb");
1409

1410
	if (f == NULL) {
1411
	    statusbar(_("Error reading %s: %s"), realname,
1412
		strerror(errno));
1413
	    goto cleanup_and_exit;
1414
1415
	}

1416
	/* If backup_dir is set, we set backupname to
1417
1418
1419
1420
	 * 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]. */
1421
	if (backup_dir != NULL) {
1422
	    char *backuptemp = get_full_path(realname);
1423

1424
	    if (backuptemp == NULL)
1425
1426
1427
1428
1429
1430
		/* 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~. */
1431
		backuptemp = mallocstrcpy(NULL, tail(realname));
1432
	    else {
1433
1434
		size_t i = 0;

1435
1436
1437
		for (; backuptemp[i] != '\0'; i++) {
		    if (backuptemp[i] == '/')
			backuptemp[i] = '!';
1438
1439
1440
1441
		}
	    }

	    backupname = charalloc(strlen(backup_dir) +
1442
1443
1444
1445
1446
1447
1448
1449
1450
1451
1452
1453
1454
1455
1456
		strlen(backuptemp) + 1);
	    sprintf(backupname, "%s%s", backup_dir, backuptemp);
	    free(backuptemp);
	    backuptemp = get_next_filename(backupname, "~");
	    if (backuptemp[0] == '\0') {
		statusbar(_("Error writing %s: %s"), backupname,
		    _("Too many backup files?"));
		free(backuptemp);
		free(backupname);
		fclose(f);
		goto cleanup_and_exit;
	    } else {
		free(backupname);
		backupname = backuptemp;
	    }
1457
1458
1459
1460
	} else {
	    backupname = charalloc(strlen(realname) + 2);
	    sprintf(backupname, "%s~", realname);
	}
1461

1462
1463
1464
	/* Open the destination backup file.  Before we write to it, we
	 * set its permissions, so no unauthorized person can read it as
	 * we write. */
1465
	backup_file = fopen(backupname, "wb");
1466

1467
1468
	if (backup_file == NULL ||
		chmod(backupname, originalfilestat.st_mode) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1469
1470
	    statusbar(_("Error writing %s: %s"), backupname,
		strerror(errno));
1471
	    free(backupname);
1472
1473
1474
1475
	    if (backup_file != NULL)
		fclose(backup_file);
	    fclose(f);
	    goto cleanup_and_exit;
1476
1477
1478
	}

#ifdef DEBUG
1479
	fprintf(stderr, "Backing up %s to %s\n", realname, backupname);
1480
1481
#endif

1482
1483
	/* Copy the file. */
	copy_status = copy_file(f, backup_file);
1484

1485
	/* And set metadata. */
1486
1487
1488
1489
	if (copy_status != 0 ||
		chown(backupname, originalfilestat.st_uid,
		originalfilestat.st_gid) == -1 ||
		utime(backupname, &filetime) == -1) {
1490
1491
	    free(backupname);
	    if (copy_status == -1)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1492
1493
		statusbar(_("Error reading %s: %s"), realname,
			strerror(errno));
1494
1495
1496
1497
1498
	    else
		statusbar(_("Error writing %s: %s"), backupname,
			strerror(errno));
	    goto cleanup_and_exit;
	}
1499

1500
1501
	free(backupname);
    }
1502
#endif /* !NANO_SMALL */
1503

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1504
1505
    /* If NOFOLLOW_SYMLINKS is set and the file is a link, we aren't
     * doing prepend or append.  So we delete the link first, and just
1506
1507
1508
1509
     * overwrite. */
    if (ISSET(NOFOLLOW_SYMLINKS) && anyexists && S_ISLNK(lst.st_mode) &&
	unlink(realname) == -1) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1510
	goto cleanup_and_exit;
1511
    }
1512

1513
1514
    original_umask = umask(0);
    umask(original_umask);
1515

1516
    /* If we create a temp file, we don't let anyone else access it.  We
1517
     * create a temp file if tmp is TRUE or if we're prepending. */
1518
1519
1520
    if (tmp || append == 2)
	umask(S_IRWXG | S_IRWXO);

1521
    /* If we're prepending, copy the file to a temp file. */
1522
1523
1524
1525
1526
1527
1528
1529
1530
    if (append == 2) {
	int fd_source;
	FILE *f_source = NULL;

	tempname = charalloc(strlen(realname) + 8);
	strcpy(tempname, realname);
	strcat(tempname, ".XXXXXX");
	fd = mkstemp(tempname);
	f = NULL;
1531

1532
1533
1534
1535
1536
	if (fd != -1) {
	    f = fdopen(fd, "wb");
	    if (f == NULL)
		close(fd);
	}
1537

1538
	if (f == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1539
1540
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
1541
	    unlink(tempname);
1542
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1543
	}
1544

1545
	fd_source = open(realname, O_RDONLY | O_CREAT);
1546

1547
1548
1549
1550
1551
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
	}
1552

1553
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1554
1555
	    statusbar(_("Error reading %s: %s"), realname,
		strerror(errno));
1556
1557
1558
1559
1560
1561
	    fclose(f);
	    unlink(tempname);
	    goto cleanup_and_exit;
	}

	if (copy_file(f_source, f) != 0) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1562
1563
	    statusbar(_("Error writing %s: %s"), tempname,
		strerror(errno));
1564
	    unlink(tempname);
1565
	    goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1566
1567
1568
	}
    }

1569
    /* Now open the file in place.  Use O_EXCL if tmp is TRUE.  This is
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1570
     * copied from joe, because wiggy says so *shrug*. */
1571
    fd = open(realname, O_WRONLY | O_CREAT |
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1572
	((append == 1) ? O_APPEND : (tmp ? O_EXCL : O_TRUNC)),
1573
	S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH);
1574

1575
    /* Set the umask back to the user's original value. */
1576
1577
    umask(original_umask);

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1578
    /* If we couldn't open the file, give up. */
1579
1580
    if (fd == -1) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
1581

1582
1583
1584
	/* tempname has been set only if we're prepending. */
	if (tempname != NULL)
	    unlink(tempname);
1585
1586
	goto cleanup_and_exit;
    }
1587

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1588
    f = fdopen(fd, (append == 1) ? "ab" : "wb");
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1589

1590
    if (f == NULL) {
1591
1592
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
	close(fd);
1593
	goto cleanup_and_exit;
1594
1595
    }

1596
    /* There might not be a magicline.  There won't be when writing out
1597
1598
     * a selection. */
    assert(fileage != NULL && filebot != NULL);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1599

1600
    while (fileptr != filebot) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1601
	size_t data_len = strlen(fileptr->data), size;
1602

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

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

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

1611
	if (size < data_len) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1612
1613
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1614
	    fclose(f);
1615
	    goto cleanup_and_exit;
1616
	}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1617

1618
#ifndef NANO_SMALL
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1619
	if (fmt == DOS_FILE || fmt == MAC_FILE) {
1620
	    if (putc('\r', f) == EOF) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1621
1622
		statusbar(_("Error writing %s: %s"), realname,
			strerror(errno));
1623
1624
1625
		fclose(f);
		goto cleanup_and_exit;
	    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1626
	}
Chris Allegretta's avatar
Chris Allegretta committed
1627

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1628
	if (fmt != MAC_FILE) {
1629
#endif
1630
	    if (putc('\n', f) == EOF) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1631
1632
		statusbar(_("Error writing %s: %s"), realname,
			strerror(errno));
1633
1634
1635
		fclose(f);
		goto cleanup_and_exit;
	    }
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1636
1637
1638
#ifndef NANO_SMALL
	}
#endif
Chris Allegretta's avatar
Chris Allegretta committed
1639
1640
1641
1642
1643

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

1644
    /* If we're prepending, open the temp file, and append it to f. */
1645
    if (append == 2) {
1646
1647
1648
1649
	int fd_source;
	FILE *f_source = NULL;

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

1651
1652
1653
1654
	if (fd_source != -1) {
	    f_source = fdopen(fd_source, "rb");
	    if (f_source == NULL)
		close(fd_source);
1655
	}
1656

1657
	if (f_source == NULL) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1658
1659
	    statusbar(_("Error reading %s: %s"), tempname,
		strerror(errno));
1660
	    fclose(f);
1661
	    goto cleanup_and_exit;
1662
1663
	}

1664
	if (copy_file(f_source, f) == -1 || unlink(tempname) == -1) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1665
1666
	    statusbar(_("Error writing %s: %s"), realname,
		strerror(errno));
1667
	    goto cleanup_and_exit;
1668
	}
1669
1670
1671
1672
    } else if (fclose(f) == EOF) {
	statusbar(_("Error writing %s: %s"), realname, strerror(errno));
	unlink(tempname);
	goto cleanup_and_exit;
Chris Allegretta's avatar
Chris Allegretta committed
1673
    }
1674

1675
    if (!tmp && append == 0) {
Chris Allegretta's avatar
Chris Allegretta committed
1676
	if (!nonamechange) {
1677
	    filename = mallocstrcpy(filename, realname);
Chris Allegretta's avatar
Chris Allegretta committed
1678
1679
#ifdef ENABLE_COLOR
	    update_color();
1680
	    if (!ISSET(NO_COLOR_SYNTAX))
1681
		edit_refresh();
Chris Allegretta's avatar
Chris Allegretta committed
1682
1683
#endif
	}
1684

1685
1686
1687
1688
#ifndef NANO_SMALL
	/* Update originalfilestat to reference the file as it is now. */
	stat(filename, &originalfilestat);
#endif
1689
1690
	statusbar(P_("Wrote %lu line", "Wrote %lu lines", lineswritten),
		(unsigned long)lineswritten);
1691
	UNSET(MODIFIED);
Chris Allegretta's avatar
Chris Allegretta committed
1692
	titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
1693
    }
1694
1695
1696
1697
1698

    retval = 1;

  cleanup_and_exit:
    free(realname);
1699
    free(tempname);
1700

1701
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1702
1703
}

1704
1705
1706
#ifndef NANO_SMALL
/* Write a marked selection from a file out.  First, set fileage and
 * filebot as the top and bottom of the mark, respectively.  Then call
1707
1708
1709
1710
 * write_file() with the values of name, temp, and append, and with
 * nonamechange set to TRUE so that we don't change the current
 * filename.  Finally, set fileage and filebot back to their old values
 * and return. */
1711
int write_marked(const char *name, bool tmp, int append)
1712
1713
{
    int retval = -1;
1714
    bool old_modified = ISSET(MODIFIED);
1715
	/* write_file() unsets the MODIFIED flag. */
1716
1717
1718
1719
1720
1721
1722
1723
    bool added_magicline;
	/* Whether we added a magicline after filebot. */
    filestruct *top, *bot;
    size_t top_x, bot_x;

    /* Partition the filestruct so that it contains only the marked
     * text. */
    mark_order((const filestruct **)&top, &top_x,
1724
	(const filestruct **)&bot, &bot_x, NULL);
1725
    filepart = partition_filestruct(top, top_x, bot, bot_x);
1726
1727

    /* If the line at filebot is blank, treat it as the magicline and
1728
1729
1730
1731
1732
     * hence the end of the file.  Otherwise, add a magicline and treat
     * it as the end of the file. */
    added_magicline = (filebot->data[0] != '\0');
    if (added_magicline)
	new_magicline();
1733

1734
    retval = write_file(name, tmp, append, TRUE);
1735

1736
1737
1738
1739
1740
1741
    /* If we added a magicline, remove it now. */
    if (added_magicline)
	remove_magicline();

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

1744
    if (old_modified)
1745
1746
1747
1748
1749
1750
	set_modified();

    return retval;
}
#endif /* !NANO_SMALL */

1751
int do_writeout(bool exiting)
Chris Allegretta's avatar
Chris Allegretta committed
1752
{
1753
    int i;
1754
1755
1756
    int retval = 0, append = 0;
    char *ans;
	/* The last answer the user typed on the statusbar. */
1757
#ifdef NANO_EXTRA
1758
    static bool did_cred = FALSE;
1759
#endif
Chris Allegretta's avatar
Chris Allegretta committed
1760

1761
    currshortcut = writefile_list;
1762

1763
    if (exiting && filename[0] != '\0' && ISSET(TEMP_FILE)) {
1764
1765
1766
1767
1768
	retval = write_file(filename, FALSE, 0, FALSE);

	/* Write succeeded. */
	if (retval == 1)
	    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1769
1770
    }

1771
#ifndef NANO_SMALL
1772
    if (ISSET(MARK_ISSET) && !exiting)
1773
	ans = mallocstrcpy(NULL, "");
1774
1775
    else
#endif
1776
	ans = mallocstrcpy(NULL, filename);
1777
1778
1779
1780

    while (TRUE) {
	const char *msg;
#ifndef NANO_SMALL
1781
1782
	const char *formatstr, *backupstr;

1783
	if (fmt == DOS_FILE)
1784
	   formatstr = N_(" [DOS Format]");
1785
	else if (fmt == MAC_FILE)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1786
	   formatstr = N_(" [Mac Format]");
1787
1788
1789
	else
	   formatstr = "";

1790
	if (ISSET(BACKUP_FILE))
1791
	   backupstr = N_(" [Backup]");
1792
1793
1794
	else
	   backupstr = "";

1795
	/* Be nice to the translation folks. */
1796
	if (ISSET(MARK_ISSET) && !exiting) {
1797
	    if (append == 2)
1798
		msg = N_("Prepend Selection to File");
Chris Allegretta's avatar
Chris Allegretta committed
1799
	    else if (append == 1)
1800
		msg = N_("Append Selection to File");
1801
	    else
1802
		msg = N_("Write Selection to File");
1803
1804
	} else
#endif /* !NANO_SMALL */
1805
	if (append == 2)
1806
	    msg = N_("File Name to Prepend to");
Chris Allegretta's avatar
Chris Allegretta committed
1807
	else if (append == 1)
1808
	    msg = N_("File Name to Append to");
1809
	else
1810
	    msg = N_("File Name to Write");
1811

1812
1813
1814
	/* If we're using restricted mode, the filename isn't blank,
	 * and we're at the "Write File" prompt, disable tab
	 * completion. */
1815
	i = statusq(!ISSET(RESTRICTED) || filename[0] == '\0',
1816
		writefile_list, ans,
1817
#ifndef NANO_SMALL
1818
		NULL, "%s%s%s", _(msg), formatstr, backupstr
1819
#else
1820
		"%s", _(msg)
1821
1822
1823
#endif
		);

1824
	if (i < 0) {
1825
	    statusbar(_("Cancelled"));
1826
1827
1828
1829
	    retval = -1;
	    break;
	} else {
	    ans = mallocstrcpy(ans, answer);
Chris Allegretta's avatar
Chris Allegretta committed
1830

1831
#ifndef DISABLE_BROWSER
1832
1833
	    if (i == NANO_TOFILES_KEY) {
		char *tmp = do_browse_from(answer);
1834

1835
1836
1837
1838
1839
1840
1841
1842
1843
1844
1845
		currshortcut = writefile_list;

		if (tmp == NULL)
		    continue;
		free(answer);
		answer = tmp;

		/* We have a file now.  Get out of the statusbar prompt
		 * cleanly. */
		statusq_abort();
	    } else
1846
#endif /* !DISABLE_BROWSER */
Chris Allegretta's avatar
Chris Allegretta committed
1847
#ifndef NANO_SMALL
1848
	    if (i == TOGGLE_DOS_KEY) {
1849
		fmt = (fmt == DOS_FILE) ? NIX_FILE : DOS_FILE;
1850
1851
		continue;
	    } else if (i == TOGGLE_MAC_KEY) {
1852
		fmt = (fmt == MAC_FILE) ? NIX_FILE : MAC_FILE;
1853
1854
1855
1856
1857
		continue;
	    } else if (i == TOGGLE_BACKUP_KEY) {
		TOGGLE(BACKUP_FILE);
		continue;
	    } else
1858
#endif /* !NANO_SMALL */
1859
1860
1861
1862
1863
1864
1865
	    if (i == NANO_PREPEND_KEY) {
		append = (append == 2) ? 0 : 2;
		continue;
	    } else if (i == NANO_APPEND_KEY) {
		append = (append == 1) ? 0 : 1;
		continue;
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1866

Chris Allegretta's avatar
Chris Allegretta committed
1867
#ifdef DEBUG
1868
	    fprintf(stderr, "filename is %s\n", answer);
Chris Allegretta's avatar
Chris Allegretta committed
1869
#endif
1870
1871

#ifdef NANO_EXTRA
1872
1873
1874
1875
1876
1877
1878
	    if (exiting && !ISSET(TEMP_FILE) &&
		strcasecmp(answer, "zzy") == 0 && !did_cred) {
		do_credits();
		did_cred = TRUE;
		retval = -1;
		break;
	    }
1879
#endif
1880
1881
1882
1883
1884
1885
1886
1887
1888
1889
1890
1891
1892
	    if (append == 0 && strcmp(answer, filename) != 0) {
		struct stat st;

		if (!stat(answer, &st)) {
		    i = do_yesno(FALSE, _("File exists, OVERWRITE ? "));
		    if (i == 0 || i == -1)
			continue;
		/* If we're using restricted mode, we aren't allowed to
		 * change the name of a file once it has one because
		 * that would allow reading from or writing to files not
		 * specified on the command line.  In this case, don't
		 * bother showing the "Different Name" prompt. */
		} else if (!ISSET(RESTRICTED) && filename[0] != '\0'
1893
#ifndef NANO_SMALL
1894
			&& (exiting || !ISSET(MARK_ISSET))
1895
#endif
1896
			) {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1897
1898
		    i = do_yesno(FALSE,
			_("Save file under DIFFERENT NAME ? "));
1899
1900
1901
		    if (i == 0 || i == -1)
			continue;
		}
1902
	    }
Chris Allegretta's avatar
Chris Allegretta committed
1903

1904
#ifndef NANO_SMALL
1905
1906
1907
1908
1909
1910
1911
	    /* Here's where we allow the selected text to be written to
	     * a separate file.  If we're using restricted mode, this is
	     * disabled since it allows reading from or writing to files
	     * not specified on the command line. */
	    if (!ISSET(RESTRICTED) && !exiting && ISSET(MARK_ISSET))
		retval = write_marked(answer, FALSE, append);
	    else
1912
#endif /* !NANO_SMALL */
1913
		retval = write_file(answer, FALSE, append, FALSE);
1914

1915
#ifdef ENABLE_MULTIBUFFER
1916
1917
1918
1919
	    /* If we're not about to exit, update the current entry in
	     * the open_files structure. */
	    if (!exiting)
		add_open_file(TRUE);
1920
#endif
1921
1922
1923

	    break;
	}
1924
    } /* while (TRUE) */
1925
1926

    free(ans);
1927

1928
    return retval;
Chris Allegretta's avatar
Chris Allegretta committed
1929
1930
}

1931
void do_writeout_void(void)
Chris Allegretta's avatar
Chris Allegretta committed
1932
{
1933
    do_writeout(FALSE);
1934
    display_main_list();
Chris Allegretta's avatar
Chris Allegretta committed
1935
}
Chris Allegretta's avatar
Chris Allegretta committed
1936

1937
1938
/* Return a malloc()ed string containing the actual directory, used to
 * convert ~user/ and ~/ notation. */
1939
char *real_dir_from_tilde(const char *buf)
1940
{
Chris Allegretta's avatar
Chris Allegretta committed
1941
    char *dirtmp = NULL;
1942

1943
1944
1945
    if (buf == NULL)
    	return NULL;

1946
    if (buf[0] == '~') {
1947
	size_t i;
1948
	const char *tilde_dir;
1949

1950
	/* Figure out how much of the str we need to compare. */
1951
1952
1953
	for (i = 1; buf[i] != '/' && buf[i] != '\0'; i++)
	    ;

1954
1955
1956
1957
1958
1959
1960
	/* Get the home directory. */
	if (i == 1) {
	    get_homedir();
	    tilde_dir = homedir;
	} else {
	    const struct passwd *userdata;

1961
1962
1963
	    do {
		userdata = getpwent();
	    } while (userdata != NULL &&
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
1964
		strncmp(userdata->pw_name, buf + 1, i - 1) != 0);
1965
1966
	    endpwent();
	    tilde_dir = userdata->pw_dir;
Chris Allegretta's avatar
Chris Allegretta committed
1967
	}
1968

1969
1970
1971
	if (tilde_dir != NULL) {
	    dirtmp = charalloc(strlen(tilde_dir) + strlen(buf + i) + 1);
	    sprintf(dirtmp, "%s%s", tilde_dir, buf + i);
1972
	}
1973
    }
1974

1975
1976
    /* Set a default value for dirtmp, in case the user's home directory
     * isn't found. */
1977
    if (dirtmp == NULL)
1978
	dirtmp = mallocstrcpy(NULL, buf);
1979

1980
    return dirtmp;
1981
1982
}

1983
#if !defined(DISABLE_TABCOMP) || !defined(DISABLE_BROWSER)
1984
1985
/* Our sort routine for file listings.  Sort alphabetically and
 * case-insensitively, and sort directories before filenames. */
1986
1987
1988
1989
1990
1991
1992
1993
1994
1995
1996
1997
1998
1999
2000
2001
2002
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;

    return strcasecmp(a, b);
}
#endif

Chris Allegretta's avatar
Chris Allegretta committed
2003
#ifndef DISABLE_TABCOMP
2004
2005
/* Is the given file a directory? */
int is_dir(const char *buf)
2006
{
2007
    char *dirptr = real_dir_from_tilde(buf);
2008
2009
    struct stat fileinfo;

2010
    int ret = (stat(dirptr, &fileinfo) != -1 &&
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2011
	S_ISDIR(fileinfo.st_mode));
2012

2013
    assert(buf != NULL && dirptr != buf);
2014

2015
    free(dirptr);
2016

2017
    return ret;
2018
}
Chris Allegretta's avatar
Chris Allegretta committed
2019

2020
/* These functions (username_tab_completion(), cwd_tab_completion(), and
Chris Allegretta's avatar
Chris Allegretta committed
2021
 * input_tab()) were taken from busybox 0.46 (cmdedit.c).  Here is the
2022
 * notice from that file:
Chris Allegretta's avatar
Chris Allegretta committed
2023
2024
2025
2026
2027
2028
2029
2030
2031
2032
2033
2034
 *
 * Termios command line History and Editting, originally
 * intended for NetBSD sh (ash)
 * Copyright (c) 1999
 *      Main code:            Adam Rogoyski <rogoyski@cs.utexas.edu>
 *      Etc:                  Dave Cinege <dcinege@psychosis.com>
 *  Majorly adjusted/re-written for busybox:
 *                            Erik Andersen <andersee@debian.org>
 *
 * You may use this code as you wish, so long as the original author(s)
 * are attributed in any redistributions of the source code.
 * This code is 'as is' with no warranty.
2035
 * This code may safely be consumed by a BSD or GPL license. */
Chris Allegretta's avatar
Chris Allegretta committed
2036

2037
2038
2039
2040
/* We consider the first buflen characters of buf for ~username tab
 * completion. */
char **username_tab_completion(const char *buf, size_t *num_matches,
	size_t buflen)
Chris Allegretta's avatar
Chris Allegretta committed
2041
{
2042
2043
    char **matches = NULL;
    const struct passwd *userdata;
2044

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

2047
    *num_matches = 0;
2048

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

2054
2055
#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2056
2057
2058
	     * directory, in which case just go to the next match. */
	    if (check_operating_dir(userdata->pw_dir, TRUE))
		continue;
2059
2060
#endif

2061
2062
2063
2064
2065
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
	    matches[*num_matches] =
		charalloc(strlen(userdata->pw_name) + 2);
	    sprintf(matches[*num_matches], "~%s", userdata->pw_name);
2066
	    ++(*num_matches);
2067
	}
2068
2069
    }
    endpwent();
2070

2071
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2072
2073
}

2074
/* This was originally called exe_n_cwd_tab_completion(), but we're not
2075
2076
2077
 * worried about executables, only filenames :> */
char **cwd_tab_completion(const char *buf, size_t *num_matches, size_t
	buflen)
Chris Allegretta's avatar
Chris Allegretta committed
2078
{
2079
2080
2081
2082
2083
2084
2085
    char *dirname = mallocstrcpy(NULL, buf);
    char *filename;
#ifndef DISABLE_OPERATINGDIR
    size_t dirnamelen;
#endif
    size_t filenamelen;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2086
    DIR *dir;
2087
    const struct dirent *nextdir;
Chris Allegretta's avatar
Chris Allegretta committed
2088

2089
    assert(dirname != NULL && num_matches != NULL && buflen >= 0);
2090

2091
2092
2093
2094
2095
2096
2097
2098
2099
2100
    *num_matches = 0;
    null_at(&dirname, buflen);

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

	filename = mallocstrcpy(NULL, tmpdirname);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2101
	*tmpdirname = '\0';
2102
2103
2104
	tmpdirname = dirname;
	dirname = real_dir_from_tilde(dirname);
	free(tmpdirname);
Chris Allegretta's avatar
Chris Allegretta committed
2105
    } else {
2106
2107
	filename = dirname;
	dirname = mallocstrcpy(NULL, "./");
Chris Allegretta's avatar
Chris Allegretta committed
2108
2109
    }

2110
    assert(dirname[strlen(dirname) - 1] == '/');
2111

Chris Allegretta's avatar
Chris Allegretta committed
2112
    dir = opendir(dirname);
2113

2114
    if (dir == NULL) {
2115
	/* Don't print an error, just shut up and return. */
Chris Allegretta's avatar
Chris Allegretta committed
2116
	beep();
2117
2118
2119
	free(filename);
	free(dirname);
	return NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2120
    }
2121
2122
2123
2124
2125
2126

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

2127
    while ((nextdir = readdir(dir)) != NULL) {
Chris Allegretta's avatar
Chris Allegretta committed
2128
2129

#ifdef DEBUG
2130
	fprintf(stderr, "Comparing \'%s\'\n", nextdir->d_name);
Chris Allegretta's avatar
Chris Allegretta committed
2131
#endif
2132
	/* See if this matches. */
2133
2134
2135
2136
	if (strncmp(nextdir->d_name, filename, filenamelen) == 0 &&
		(*filename == '.' ||
		(strcmp(nextdir->d_name, ".") != 0 &&
		strcmp(nextdir->d_name, "..") != 0))) {
2137
2138
	    /* Cool, found a match.  Add it to the list.  This makes a
	     * lot more sense to me (Chris) this way... */
2139
2140
2141

#ifndef DISABLE_OPERATINGDIR
	    /* ...unless the match exists outside the operating
2142
2143
2144
2145
2146
	     * directory, in which case just go to the next match.  To
	     * properly do operating directory checking, we have to add
	     * the directory name to the beginning of the proposed match
	     * before we check it. */
	    char *tmp2 = charalloc(strlen(dirname) +
2147
		strlen(nextdir->d_name) + 1);
2148

2149
	    sprintf(tmp2, "%s%s", dirname, nextdir->d_name);
2150
2151
2152
	    if (check_operating_dir(tmp2, TRUE)) {
		free(tmp2);
		continue;
2153
	    }
2154
	    free(tmp2);
2155
2156
#endif

2157
2158
	    matches = (char **)nrealloc(matches, (*num_matches + 1) *
		sizeof(char *));
2159
	    matches[*num_matches] = mallocstrcpy(NULL, nextdir->d_name);
2160
	    ++(*num_matches);
Chris Allegretta's avatar
Chris Allegretta committed
2161
2162
	}
    }
2163
2164
    closedir(dir);
    free(dirname);
2165
    free(filename);
Chris Allegretta's avatar
Chris Allegretta committed
2166

2167
    return matches;
Chris Allegretta's avatar
Chris Allegretta committed
2168
2169
}

2170
2171
/* Do tab completion.  place refers to how much the statusbar cursor
 * position should be advanced. */
2172
char *input_tab(char *buf, size_t *place, bool *lastwastab, bool *list)
Chris Allegretta's avatar
Chris Allegretta committed
2173
{
2174
2175
    size_t num_matches = 0;
    char **matches = NULL;
Chris Allegretta's avatar
Chris Allegretta committed
2176

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

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2179
    *list = FALSE;
Chris Allegretta's avatar
Chris Allegretta committed
2180

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

2186
2187
2188
2189
	if (bob == NULL || bob >= buf + *place)
	    matches = username_tab_completion(buf, &num_matches,
		*place);
    }
2190

2191
2192
2193
    /* Match against files relative to the current working directory. */
    if (matches == NULL)
	matches = cwd_tab_completion(buf, &num_matches, *place);
2194

2195
2196
2197
2198
2199
    if (num_matches <= 0)
	beep();
    else {
	size_t match, common_len = 0;
	char *mzero;
2200
2201
2202
	const char *lastslash = revstrstr(buf, "/", buf + *place);
	size_t lastslash_len = (lastslash == NULL) ? 0 :
		lastslash - buf + 1;
2203
2204
2205
	char *match1_mb = charalloc(mb_cur_max() + 1);
	char *match2_mb = charalloc(mb_cur_max() + 1);
	int match1_mb_len, match2_mb_len;
2206
2207
2208

	while (TRUE) {
	    for (match = 1; match < num_matches; match++) {
2209
2210
2211
2212
2213
2214
2215
		match1_mb_len = parse_mbchar(matches[0] + common_len,
			match1_mb, NULL, NULL);
		match2_mb_len = parse_mbchar(matches[match] +
			common_len, match2_mb, NULL, NULL);
		match1_mb[match1_mb_len] = '\0';
		match2_mb[match2_mb_len] = '\0';
		if (strcmp(match1_mb, match2_mb) != 0)
2216
2217
		    break;
	    }
2218

2219
	    if (match < num_matches || matches[0][common_len] == '\0')
2220
		break;
2221

2222
2223
	    common_len += parse_mbchar(buf + common_len, NULL, NULL,
		NULL);
2224
	}
2225

2226
2227
2228
	free(match1_mb);
	free(match2_mb);

2229
2230
	mzero = charalloc(lastslash_len + common_len + 1);
	sprintf(mzero, "%.*s%.*s", lastslash_len, buf, common_len,
2231
		matches[0]);
2232

2233
	common_len += lastslash_len;
2234

2235
	assert(common_len >= *place);
2236

2237
2238
2239
	if (num_matches == 1 && is_dir(mzero)) {
	    mzero[common_len] = '/';
	    common_len++;
2240

2241
2242
	    assert(common_len > *place);
	}
2243

2244
2245
2246
	if (num_matches > 1 && (common_len != *place ||
		*lastwastab == FALSE))
	    beep();
2247

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2248
2249
2250
	/* If there is more of a match to display on the statusbar, show
	 * it.  We reset lastwastab to FALSE: it requires hitting Tab
	 * twice in succession with no statusbar changes to see a match
2251
2252
2253
2254
2255
2256
	 * list. */
	if (common_len != *place) {
	    size_t buflen = strlen(buf);

	    *lastwastab = FALSE;
	    buf = charealloc(buf, common_len + buflen - *place + 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2257
2258
	    charmove(buf + common_len, buf + *place,
		buflen - *place + 1);
2259
	    charcpy(buf, mzero, common_len);
2260
	    *place = common_len;
2261
2262
2263
2264
2265
	} else if (*lastwastab == FALSE || num_matches < 2)
	    *lastwastab = TRUE;
	else {
	    int longest_name = 0, editline = 0;
	    size_t columns;
2266

2267
2268
2269
2270
2271
2272
2273
2274
	    /* Now we show a list of the available choices. */
	    assert(num_matches > 1);

	    /* Sort the list. */
	    qsort(matches, num_matches, sizeof(char *), diralphasort);

	    for (match = 0; match < num_matches; match++) {
		common_len = strnlenpt(matches[match], COLS - 1);
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2275

2276
2277
		if (common_len > COLS - 1) {
		    longest_name = COLS - 1;
2278
2279
		    break;
		}
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2280

2281
2282
		if (common_len > longest_name)
		    longest_name = common_len;
2283
	    }
Chris Allegretta's avatar
Chris Allegretta committed
2284

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

2287
2288
2289
2290
	    /* Each column will be longest_name + 2 characters wide,
	     * i.e, two spaces between columns, except that there will
	     * be only one space after the last column. */
	    columns = (COLS + 1) / (longest_name + 2);
2291

2292
2293
2294
2295
	    /* Blank the edit window, and print the matches out
	     * there. */
	    blank_edit();
	    wmove(edit, 0, 0);
2296

2297
2298
	    /* Disable el cursor. */
	    curs_set(0);
2299

2300
2301
	    for (match = 0; match < num_matches; match++) {
		char *disp;
2302

2303
2304
		wmove(edit, editline, (longest_name + 2) *
			(match % columns));
2305

2306
2307
2308
		if (match % columns == 0 &&
			editline == editwinrows - 1 &&
			num_matches - match > columns) {
2309
2310
2311
		    waddstr(edit, _("(more)"));
		    break;
		}
2312

2313
2314
2315
2316
		disp = display_string(matches[match], 0, longest_name,
			FALSE);
		waddstr(edit, disp);
		free(disp);
2317

2318
		if ((match + 1) % columns == 0)
2319
		    editline++;
Chris Allegretta's avatar
Chris Allegretta committed
2320
2321
	    }
	    wrefresh(edit);
2322
	    *list = TRUE;
2323
2324
2325
	}

	free(mzero);
Chris Allegretta's avatar
Chris Allegretta committed
2326
2327
    }

2328
2329
    free_charptrarray(matches, num_matches);

2330
    /* Only refresh the edit window if we don't have a list of filename
2331
     * matches on it. */
2332
    if (*list == FALSE)
2333
	edit_refresh();
2334
2335

    /* Enable el cursor. */
2336
    curs_set(1);
2337

2338
    return buf;
Chris Allegretta's avatar
Chris Allegretta committed
2339
}
2340
#endif /* !DISABLE_TABCOMP */
Chris Allegretta's avatar
Chris Allegretta committed
2341

2342
2343
/* Only print the last part of a path.  Isn't there a shell command for
 * this? */
2344
2345
const char *tail(const char *foo)
{
2346
    const char *tmp = strrchr(foo, '/');
2347

2348
2349
2350
    if (tmp == NULL)
	tmp = foo;
    else if (*tmp == '/')
2351
2352
2353
2354
2355
	tmp++;

    return tmp;
}

2356
#ifndef DISABLE_BROWSER
2357
/* Free our malloc()ed memory. */
2358
void free_charptrarray(char **array, size_t len)
Chris Allegretta's avatar
Chris Allegretta committed
2359
{
2360
2361
    for (; len > 0; len--)
	free(array[len - 1]);
2362

Chris Allegretta's avatar
Chris Allegretta committed
2363
2364
2365
    free(array);
}

2366
2367
/* Strip one directory from the end of path. */
void striponedir(char *path)
Chris Allegretta's avatar
Chris Allegretta committed
2368
{
2369
    char *tmp;
Chris Allegretta's avatar
Chris Allegretta committed
2370

2371
    assert(path != NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2372

2373
    tmp = strrchr(path, '/');
2374

2375
2376
    if (tmp != NULL)
 	*tmp = '\0';
Chris Allegretta's avatar
Chris Allegretta committed
2377
2378
}

2379
2380
2381
2382
2383
2384
/* Return a list of files contained in the directory path.  *longest is
 * the maximum display length of a file, up to COLS - 1 (but at least
 * 7).  *numents is the number of files.  We assume path exists and is a
 * directory.  If neither is true, we return NULL. */
char **browser_init(const char *path, int *longest, size_t *numents, DIR
	*dir)
2385
{
2386
    const struct dirent *nextdir;
2387
2388
    char **filelist;
    size_t i, path_len;
2389

2390
    assert(dir != NULL);
2391

2392
    *longest = 0;
2393

2394
    i = 0;
2395

2396
    while ((nextdir = readdir(dir)) != NULL) {
2397
2398
2399
	size_t dlen;

	/* Don't show the . entry. */
2400
	if (strcmp(nextdir->d_name, ".") == 0)
2401
	   continue;
2402
2403
	i++;

2404
	dlen = strlenpt(nextdir->d_name);
2405
2406
	if (dlen > *longest)
	    *longest = dlen;
2407
    }
2408
2409

    *numents = i;
2410
2411
2412
    rewinddir(dir);
    *longest += 10;

2413
    filelist = (char **)nmalloc(*numents * sizeof(char *));
2414

2415
2416
    path_len = strlen(path);

2417
2418
    i = 0;

2419
    while ((nextdir = readdir(dir)) != NULL && i < *numents) {
2420
	/* Don't show the "." entry. */
2421
	if (strcmp(nextdir->d_name, ".") == 0)
2422
2423
	   continue;

2424
2425
	filelist[i] = charalloc(path_len + strlen(nextdir->d_name) + 1);
	sprintf(filelist[i], "%s%s", path, nextdir->d_name);
2426
2427
	i++;
    }
2428
2429
2430
2431
2432

    /* Maybe the number of files in the directory changed between the
     * first time we scanned and the second.  i is the actual length of
     * filelist, so record it. */
    *numents = i;
2433
    closedir(dir);
2434
2435
2436

    if (*longest > COLS - 1)
	*longest = COLS - 1;
2437
2438
    if (*longest < 7)
	*longest = 7;
2439
2440
2441
2442

    return filelist;
}

2443
2444
2445
/* Our browser function.  path is the path to start browsing from.
 * Assume path has already been tilde-expanded. */
char *do_browser(char *path, DIR *dir)
Chris Allegretta's avatar
Chris Allegretta committed
2446
{
2447
2448
2449
2450
2451
2452
2453
2454
2455
2456
2457
2458
2459
2460
    int kbinput, longest, selected, width;
    bool meta_key, func_key, old_constupdate = ISSET(CONSTUPDATE);
    size_t numents;
    char **filelist, *retval = NULL;

    curs_set(0);
    blank_statusbar();
    bottombars(browser_list);
    wrefresh(bottomwin);

#if !defined(DISABLE_HELP) || !defined(DISABLE_MOUSE)
    /* Set currshortcut so the user can click in the shortcut area, and
     * so the browser help screen will come up. */
    currshortcut = browser_list;
2461
#endif
Chris Allegretta's avatar
Chris Allegretta committed
2462

2463
    UNSET(CONSTUPDATE);
2464

2465
2466
2467
2468
2469
2470
2471
2472
  change_browser_directory:
	/* We go here after the user selects a new directory. */

    kbinput = ERR;
    selected = 0;
    width = 0;

    path = mallocstrassn(path, get_full_path(path));
Chris Allegretta's avatar
Chris Allegretta committed
2473

2474
2475
    /* Assume that path exists and ends with a slash. */
    assert(path != NULL && path[strlen(path) - 1] == '/');
Chris Allegretta's avatar
Chris Allegretta committed
2476

2477
2478
2479
2480
    /* Get the list of files. */
    filelist = browser_init(path, &longest, &numents, dir);

    assert(filelist != NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2481

2482
    /* Sort the list. */
Chris Allegretta's avatar
Chris Allegretta committed
2483
2484
2485
    qsort(filelist, numents, sizeof(char *), diralphasort);

    titlebar(path);
Chris Allegretta's avatar
Chris Allegretta committed
2486
2487

    /* Loop invariant: Microsoft sucks. */
Chris Allegretta's avatar
Chris Allegretta committed
2488
    do {
2489
	bool abort = FALSE;
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2490
	int j, col = 0, editline = 0, fileline;
2491
2492
2493
2494
	int filecols = 0;
	    /* Used only if width == 0, to calculate the number of files
	     * per row below. */
	struct stat st;
2495
2496
	char *new_path;
	    /* Used by the Go To Directory prompt. */
2497
2498
2499
#ifndef DISABLE_MOUSE
	MEVENT mevent;
#endif
Rocco Corsi's avatar
   
Rocco Corsi committed
2500

2501
	check_statusblank();
Rocco Corsi's avatar
   
Rocco Corsi committed
2502

2503
2504
	/* Compute the line number we're on now, so that we don't divide
	 * by zero later. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2505
	fileline = selected;
2506
	if (width != 0)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2507
	    fileline /= width;
Chris Allegretta's avatar
Chris Allegretta committed
2508
2509

	switch (kbinput) {
2510
#ifndef DISABLE_MOUSE
2511
2512
2513
	    case KEY_MOUSE:
		if (getmouse(&mevent) == ERR)
		    break;
2514

2515
2516
2517
2518
2519
2520
2521
2522
2523
		/* If we clicked in the edit window, we probably clicked
		 * on a file. */
		if (wenclose(edit, mevent.y, mevent.x)) {
		    int selectedbackup = selected;

		    mevent.y -= 2;

		    /* longest is the width of each column.  There are
		     * two spaces between each column. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2524
		    selected = (fileline / editwinrows) * editwinrows *
2525
2526
2527
2528
2529
2530
2531
2532
2533
2534
2535
2536
2537
2538
2539
2540
2541
2542
2543
2544
2545
2546
			width + mevent.y * width + mevent.x /
			(longest + 2);

		    /* If they clicked beyond the end of a row, select
		     * the end of that row. */
		    if (mevent.x > width * (longest + 2))
			selected--;

		    /* If we're off the screen, reset to the last item.
		     * If we clicked the same place as last time, select
		     * this name! */
		    if (selected > numents - 1)
			selected = numents - 1;
		    else if (selectedbackup == selected)
			/* Put back the 'select' key. */
			unget_kbinput('s', FALSE, FALSE);
		} else {
		    /* We must have clicked a shortcut.  Put back the
		     * equivalent shortcut key. */
		    int mouse_x, mouse_y;
		    get_mouseinput(&mouse_x, &mouse_y, TRUE);
		}
2547

2548
		break;
2549
#endif
2550
2551
2552
2553
2554
2555
2556
2557
2558
2559
2560
2561
2562
2563
2564
2565
2566
2567
2568
	    case NANO_PREVLINE_KEY:
		if (selected >= width)
		    selected -= width;
		break;
	    case NANO_BACK_KEY:
		if (selected > 0)
		    selected--;
		break;
	    case NANO_NEXTLINE_KEY:
		if (selected + width <= numents - 1)
		    selected += width;
		break;
	    case NANO_FORWARD_KEY:
		if (selected < numents - 1)
		    selected++;
		break;
	    case NANO_PREVPAGE_KEY:
	    case NANO_PREVPAGE_FKEY:
	    case '-': /* Pico compatibility. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2569
		if (selected >= (editwinrows + fileline % editwinrows) *
2570
			width)
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2571
		    selected -= (editwinrows + fileline % editwinrows) *
2572
2573
2574
2575
2576
2577
2578
			width;
		else
		    selected = 0;
		break;
	    case NANO_NEXTPAGE_KEY:
	    case NANO_NEXTPAGE_FKEY:
	    case ' ': /* Pico compatibility. */
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2579
		selected += (editwinrows - fileline % editwinrows) *
2580
2581
2582
2583
2584
2585
2586
			width;
		if (selected >= numents)
		    selected = numents - 1;
		break;
	    case NANO_HELP_KEY:
	    case NANO_HELP_FKEY:
	    case '?': /* Pico compatibility. */
2587
#ifndef DISABLE_HELP
2588
2589
		do_help();
		curs_set(0);
2590
#else
2591
		nano_disabled_msg();
2592
#endif
Rocco Corsi's avatar
   
Rocco Corsi committed
2593
		break;
2594
2595
2596
2597
2598
2599
2600
2601
2602
	    case NANO_ENTER_KEY:
	    case 'S': /* Pico compatibility. */
	    case 's':
		/* You can't move up from "/". */
		if (strcmp(filelist[selected], "/..") == 0) {
		    statusbar(_("Can't move up a directory"));
		    beep();
		    break;
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2603

2604
#ifndef DISABLE_OPERATINGDIR
2605
2606
2607
2608
2609
2610
		/* Note: the selected file can be outside the operating
		 * directory if it's ".." or if it's a symlink to a
		 * directory outside the operating directory. */
		if (check_operating_dir(filelist[selected], FALSE)) {
		    statusbar(
			_("Can't go outside of %s in restricted mode"),
2611
			operating_dir);
2612
2613
2614
		    beep();
		    break;
		}
2615
2616
#endif

2617
2618
2619
2620
2621
2622
		if (stat(filelist[selected], &st) == -1) {
		    statusbar(_("Error reading %s: %s"),
			filelist[selected], strerror(errno));
		    beep();
		    break;
		}
2623

2624
2625
2626
2627
2628
		if (!S_ISDIR(st.st_mode)) {
		    retval = mallocstrcpy(retval, filelist[selected]);
		    abort = TRUE;
		    break;
		}
2629

2630
2631
2632
2633
2634
2635
2636
		dir = opendir(filelist[selected]);
		if (dir == NULL) {
		    /* We can't open this dir for some reason.
		     * Complain. */
		    statusbar(_("Error reading %s: %s"),
			filelist[selected], strerror(errno));
		    break;
Chris Allegretta's avatar
Chris Allegretta committed
2637
		}
Chris Allegretta's avatar
Chris Allegretta committed
2638

2639
		path = mallocstrcpy(path, filelist[selected]);
2640

2641
2642
2643
2644
		/* Start over again with the new path value. */
		free_charptrarray(filelist, numents);
		goto change_browser_directory;

2645
2646
2647
2648
2649
	    /* Refresh the screen. */
	    case NANO_REFRESH_KEY:
		total_update();
		break;

2650
2651
2652
2653
2654
2655
2656
2657
	    /* Go to a specific directory. */
	    case NANO_GOTOLINE_KEY:
	    case NANO_GOTOLINE_FKEY:
	    case 'G': /* Pico compatibility. */
	    case 'g':
		curs_set(1);

		j = statusq(FALSE, gotodir_list, "",
Chris Allegretta's avatar
Chris Allegretta committed
2658
#ifndef NANO_SMALL
2659
			NULL,
Chris Allegretta's avatar
Chris Allegretta committed
2660
#endif
2661
			_("Go To Directory"));
Rocco Corsi's avatar
   
Rocco Corsi committed
2662

2663
2664
		curs_set(0);
		bottombars(browser_list);
Rocco Corsi's avatar
   
Rocco Corsi committed
2665

2666
2667
2668
2669
		if (j < 0) {
		    statusbar(_("Cancelled"));
		    break;
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2670

2671
2672
2673
2674
2675
2676
2677
		new_path = real_dir_from_tilde(answer);

		if (new_path[0] != '/') {
		    new_path = charealloc(new_path, strlen(path) +
			strlen(answer) + 1);
		    sprintf(new_path, "%s%s", path, answer);
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2678

2679
#ifndef DISABLE_OPERATINGDIR
2680
2681
2682
2683
2684
2685
2686
		if (check_operating_dir(new_path, FALSE)) {
		    statusbar(
			_("Can't go outside of %s in restricted mode"),
			operating_dir);
		    free(new_path);
		    break;
		}
2687
2688
#endif

2689
2690
2691
2692
2693
2694
2695
2696
2697
		dir = opendir(new_path);
		if (dir == NULL) {
		    /* We can't open this dir for some reason.
		     * Complain. */
		    statusbar(_("Error reading %s: %s"), answer,
			strerror(errno));
		    free(new_path);
		    break;
		}
Rocco Corsi's avatar
   
Rocco Corsi committed
2698

2699
2700
2701
2702
2703
2704
2705
2706
2707
2708
2709
2710
2711
		/* Start over again with the new path value. */
		free(path);
		path = new_path;
		free_charptrarray(filelist, numents);
		goto change_browser_directory;

	    /* Abort the browser. */
	    case NANO_EXIT_KEY:
	    case NANO_EXIT_FKEY:
	    case 'E': /* Pico compatibility. */
	    case 'e':
		abort = TRUE;
		break;
Chris Allegretta's avatar
Chris Allegretta committed
2712
	}
2713

Chris Allegretta's avatar
Chris Allegretta committed
2714
2715
2716
	if (abort)
	    break;

Rocco Corsi's avatar
   
Rocco Corsi committed
2717
2718
	blank_edit();

2719
	if (width != 0)
2720
2721
	    j = width * editwinrows *
		((selected / width) / editwinrows);
Chris Allegretta's avatar
Chris Allegretta committed
2722
	else
2723
	    j = 0;
Chris Allegretta's avatar
Chris Allegretta committed
2724
2725
2726

	wmove(edit, 0, 0);

2727
2728
2729
2730
2731
2732
2733
2734
2735
2736
2737
2738
2739
2740
2741
2742
2743
2744
2745
2746
2747
2748
2749
2750
2751
2752
2753
2754
2755
2756
	{
	    int foo_len = mb_cur_max() * 7;
	    char *foo = charalloc(foo_len + 1);

	    for (; j < numents && editline <= editwinrows - 1; j++) {
		char *disp = display_string(tail(filelist[j]), 0,
			longest, FALSE);

		/* Highlight the currently selected file/dir. */
		if (j == selected)
		    wattron(edit, A_REVERSE);

		mvwaddnstr(edit, editline, col, hblank, longest);
		mvwaddstr(edit, editline, col, disp);
		free(disp);

		col += longest;
		filecols++;

		/* Show file info also.  We don't want to report file
		 * sizes for links, so we use lstat().  Also, stat() and
		 * lstat() return an error if, for example, the file is
		 * deleted while the file browser is open.  In that
		 * case, we report "--" as the file info. */
		if (lstat(filelist[j], &st) == -1 ||
			S_ISLNK(st.st_mode)) {
		    /* Aha!  It's a symlink!  Now, is it a dir?  If so,
		     * mark it as such. */
		    if (stat(filelist[j], &st) == 0 &&
			S_ISDIR(st.st_mode)) {
2757
			charcpy(foo, _("(dir)"), foo_len);
2758
2759
2760
			foo[foo_len] = '\0';
		    } else
			strcpy(foo, "--");
2761
		} else if (S_ISDIR(st.st_mode)) {
2762
		    charcpy(foo, _("(dir)"), foo_len);
2763
2764
		    foo[foo_len] = '\0';
		} else if (st.st_size < (1 << 10)) /* less than 1 k. */
2765
2766
2767
2768
2769
2770
2771
2772
2773
2774
2775
2776
2777
2778
2779
2780
2781
2782
2783
2784
2785
2786
2787
2788
2789
2790
2791
2792
2793
		    sprintf(foo, "%4u  B", (unsigned int)st.st_size);
		else if (st.st_size < (1 << 20)) /* less than 1 meg. */
		    sprintf(foo, "%4u KB",
			(unsigned int)(st.st_size >> 10));
		else if (st.st_size < (1 << 30)) /* less than 1 gig. */
		    sprintf(foo, "%4u MB",
			(unsigned int)(st.st_size >> 20));
		else
		    sprintf(foo, "%4u GB",
			(unsigned int)(st.st_size >> 30));

		mvwaddnstr(edit, editline, col - strlen(foo), foo,
			foo_len);

		if (j == selected)
		    wattroff(edit, A_REVERSE);

		/* Add some space between the columns. */
		col += 2;

		/* If the next entry isn't going to fit on the line,
		 * move to the next line. */
		if (col > COLS - longest) {
		    editline++;
		    col = 0;
		    if (width == 0)
			width = filecols;
		}
		wmove(edit, editline, col);
Chris Allegretta's avatar
Chris Allegretta committed
2794
	    }
2795
2796

	    free(foo);
Chris Allegretta's avatar
Chris Allegretta committed
2797
	}
2798

2799
	wrefresh(edit);
2800
2801
    } while ((kbinput = get_kbinput(edit, &meta_key, &func_key)) !=
	NANO_EXIT_KEY && kbinput != NANO_EXIT_FKEY);
2802

Chris Allegretta's avatar
Chris Allegretta committed
2803
    blank_edit();
Chris Allegretta's avatar
Chris Allegretta committed
2804
    titlebar(NULL);
Chris Allegretta's avatar
Chris Allegretta committed
2805
    edit_refresh();
2806
2807
2808
    curs_set(1);
    if (old_constupdate)
	SET(CONSTUPDATE);
Chris Allegretta's avatar
Chris Allegretta committed
2809

2810
    /* Clean up. */
Chris Allegretta's avatar
Chris Allegretta committed
2811
    free_charptrarray(filelist, numents);
2812
2813
    free(path);

Chris Allegretta's avatar
Chris Allegretta committed
2814
2815
    return retval;
}
2816

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2817
2818
2819
/* The file browser front end.  We check to see if inpath has a dir in
 * it.  If it does, we start do_browser() from there.  Otherwise, we
 * start do_browser() from the current directory. */
2820
char *do_browse_from(const char *inpath)
2821
2822
{
    struct stat st;
2823
    char *path;
2824
	/* This holds the tilde-expanded version of inpath. */
2825
    DIR *dir = NULL;
2826

2827
2828
2829
    assert(inpath != NULL);

    path = real_dir_from_tilde(inpath);
2830

2831
2832
2833
2834
2835
2836
    /* Perhaps path is a directory.  If so, we'll pass it to
     * do_browser().  Or perhaps path is a directory / a file.  If so,
     * we'll try stripping off the last path element and passing it to
     * do_browser().  Or perhaps path doesn't have a directory portion
     * at all.  If so, we'll just pass the current directory to
     * do_browser(). */
2837
2838
    if (stat(path, &st) == -1 || !S_ISDIR(st.st_mode)) {
	striponedir(path);
2839
2840
	if (stat(path, &st) == -1 || !S_ISDIR(st.st_mode)) {
	    free(path);
2841

2842
2843
	    path = charalloc(PATH_MAX + 1);
	    path = getcwd(path, PATH_MAX + 1);
2844
2845
2846

	    if (path != NULL)
		align(&path);
2847
	}
2848
    }
2849

2850
#ifndef DISABLE_OPERATINGDIR
2851
2852
2853
2854
2855
2856
2857
    /* If the resulting path isn't in the operating directory, use
     * the operating directory instead. */
    if (check_operating_dir(path, FALSE)) {
	if (path != NULL)
	    free(path);
	path = mallocstrcpy(NULL, operating_dir);
    }
2858
2859
#endif

2860
2861
2862
    if (path != NULL)
	dir = opendir(path);

2863
    if (dir == NULL) {
2864
	beep();
2865
2866
2867
2868
2869
	free(path);
	return NULL;
    }

    return do_browser(path, dir);
2870
}
Chris Allegretta's avatar
Chris Allegretta committed
2871
#endif /* !DISABLE_BROWSER */
2872

2873
#if !defined(NANO_SMALL) && defined(ENABLE_NANORC)
2874
2875
2876
/* Return $HOME/.nano_history, or NULL if we can't find the homedir.
 * The string is dynamically allocated, and should be freed. */
char *histfilename(void)
2877
{
2878
    char *nanohist = NULL;
2879

2880
2881
    if (homedir != NULL) {
	size_t homelen = strlen(homedir);
2882

2883
2884
2885
	nanohist = charalloc(homelen + 15);
	strcpy(nanohist, homedir);
	strcpy(nanohist + homelen, "/.nano_history");
Chris Allegretta's avatar
Chris Allegretta committed
2886
    }
2887
2888
2889
2890
2891
2892
    return nanohist;
}

void load_history(void)
{
    char *nanohist = histfilename();
Chris Allegretta's avatar
Chris Allegretta committed
2893

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

2898
	if (hist == NULL) {
2899
	    if (errno != ENOENT) {
2900
2901
		/* Don't save history when we quit. */
		UNSET(HISTORYLOG);
2902
2903
2904
2905
		rcfile_error(N_("Error reading %s: %s"), nanohist,
			strerror(errno));
		fprintf(stderr,
			_("\nPress Return to continue starting nano\n"));
2906
2907
		while (getchar() != '\n')
		    ;
2908
	    }
2909
	} else {
2910
2911
2912
2913
2914
2915
	    /* Load a history (first the search history, then the
	     * replace history) from oldest to newest.  Assume the last
	     * history entry is a blank line. */
	    filestruct **history = &search_history;
	    filestruct **historyage = &searchage;
	    filestruct **historybot = &searchbot;
2916
2917
2918
2919
2920
2921
2922
2923
2924
2925
2926
	    char *line = NULL;
	    size_t buflen = 0;
	    ssize_t read;

	    while ((read = getline(&line, &buflen, hist)) >= 0) {
		if (read > 0 && line[read - 1] == '\n') {
		    read--;
		    line[read] = '\0';
		}
		if (read > 0) {
		    unsunder(line, read);
2927
2928
2929
		    update_history(history, historyage, historybot,
			line);
		} else {
2930
		    history = &replace_history;
2931
2932
2933
		    historyage = &replaceage;
		    historybot = &replacebot;
		}
2934
	    }
2935

2936
	    fclose(hist);
2937
	    free(line);
2938
	}
2939
2940
2941
2942
	free(nanohist);
    }
}

2943
bool writehist(FILE *hist, filestruct *h)
2944
{
2945
    filestruct *p;
2946

2947
2948
2949
    /* Write history from oldest to newest.  Assume the last history
     * entry is a blank line. */
    for (p = h; p != NULL; p = p->next) {
2950
	size_t p_len = strlen(p->data);
2951

2952
	sunder(p->data);
2953

2954
	if (fwrite(p->data, sizeof(char), p_len, hist) < p_len ||
2955
2956
		putc('\n', hist) == EOF)
	    return FALSE;
2957
    }
2958

2959
    return TRUE;
2960
2961
}

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2962
/* Save histories to ~/.nano_history. */
2963
2964
void save_history(void)
{
2965
    char *nanohist;
2966

David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2967
    /* Don't save unchanged or empty histories. */
2968
2969
    if (!history_has_changed() || (searchbot->lineno == 1 &&
	replacebot->lineno == 1))
2970
2971
	return;

2972
2973
2974
2975
    nanohist = histfilename();

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

2977
	if (hist == NULL)
2978
2979
	    rcfile_error(N_("Error writing %s: %s"), nanohist,
		strerror(errno));
2980
	else {
David Lawrence Ramsey's avatar
David Lawrence Ramsey committed
2981
2982
	    /* Make sure no one else can read from or write to the
	     * history file. */
2983
	    chmod(nanohist, S_IRUSR | S_IWUSR);
2984

2985
2986
	    if (!writehist(hist, searchage) || !writehist(hist,
		replaceage))
2987
2988
		rcfile_error(N_("Error writing %s: %s"), nanohist,
			strerror(errno));
2989

2990
2991
	    fclose(hist);
	}
2992

2993
2994
2995
	free(nanohist);
    }
}
2996
#endif /* !NANO_SMALL && ENABLE_NANORC */