/* $Id$ */ /************************************************************************** * files.c * * * * Copyright (C) 1999-2002 Chris Allegretta * * 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 * * the Free Software Foundation; either version 2, or (at your option) * * any later version. * * * * 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. * * * * You should have received a copy of the GNU General Public License * * along with this program; if not, write to the Free Software * * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. * * * **************************************************************************/ #include "config.h" #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include "proto.h" #include "nano.h" #ifndef NANO_SMALL static int fileformat = 0; /* 0 = *nix, 1 = DOS, 2 = Mac */ #endif /* Load file into edit buffer - takes data from file struct. */ void load_file(int update) { current = fileage; #ifdef ENABLE_MULTIBUFFER /* if update is zero, add a new entry to the open_files structure; otherwise, update the current entry (the latter is needed in the case of the alternate spell checker) */ add_open_file(update); #endif #ifdef ENABLE_COLOR update_color(); edit_refresh(); #endif } /* What happens when there is no file to open? aiee! */ void new_file(void) { fileage = make_new_node(NULL); fileage->data = charalloc(1); fileage->data[0] = '\0'; filebot = fileage; edittop = fileage; editbot = fileage; current = fileage; current_x = 0; totlines = 1; totsize = 0; #ifdef ENABLE_MULTIBUFFER /* if there aren't any entries in open_files, create the entry for this new file; without this, if nano is started without a filename on the command line, a new file will be created, but it will be given no open_files entry */ if (open_files == NULL) { add_open_file(0); /* turn off view mode in this case; this is for consistency whether multibuffers are compiled in or not */ UNSET(VIEW_MODE); } #else /* if multibuffers haven't been compiled in, turn off view mode unconditionally; otherwise, don't turn them off (except in the above case), so that we can view multiple files properly */ UNSET(VIEW_MODE); #endif #ifdef ENABLE_COLOR update_color(); edit_refresh(); #endif } filestruct *read_line(char *buf, filestruct *prev, int *line1ins, int len) { filestruct *fileptr = (filestruct *)nmalloc(sizeof(filestruct)); /* nulls to newlines; len is the string's real length here */ unsunder(buf, len); assert(strlen(buf) == len); fileptr->data = mallocstrcpy(NULL, buf); #ifndef NANO_SMALL /* If it's a DOS file (CRLF), and file conversion isn't disabled, strip out the CR part */ if (!ISSET(NO_CONVERT) && len > 0 && buf[len - 1] == '\r') { fileptr->data[len - 1] = '\0'; totsize--; if (fileformat == 0) fileformat = 1; } #endif if (*line1ins != 0 || fileage == NULL) { /* Special case, insert with cursor on 1st line. */ fileptr->prev = NULL; fileptr->next = fileage; fileptr->lineno = 1; if (*line1ins != 0) { *line1ins = 0; /* If we're inserting into the first line of the file, then we want to make sure that our edit buffer stays on the first line (and that fileage stays up to date!) */ edittop = fileptr; } else filebot = fileptr; fileage = fileptr; } else { assert(prev != NULL); fileptr->prev = prev; fileptr->next = NULL; fileptr->lineno = prev->lineno + 1; prev->next = fileptr; } return fileptr; } int read_file(FILE *f, const char *filename, int quiet) { int num_lines = 0, len = 0; char input = '\0'; /* current input character */ char *buf; long i = 0, bufx = 128; filestruct *fileptr = current, *tmp = NULL; #ifndef NANO_SMALL int old_no_convert = ISSET(NO_CONVERT); #endif int line1ins = 0; int input_int; buf = charalloc(bufx); buf[0] = '\0'; if (current != NULL) { if (current == fileage) line1ins = 1; else fileptr = current->prev; tmp = fileptr; } /* For the assertion in read_line(), it must be true that if current is * NULL then so is fileage. */ assert(current != NULL || fileage == NULL); /* Read the entire file into file struct. */ while ((input_int = getc(f)) != EOF) { input = (char)input_int; #ifndef NANO_SMALL /* If the file has binary chars in it, don't stupidly assume it's a DOS or Mac formatted file! */ if (!ISSET(NO_CONVERT) && is_cntrl_char((int)input) != 0 && input != '\t' && input != '\r' && input != '\n') SET(NO_CONVERT); #endif if (input == '\n') { /* read in the line properly */ fileptr = read_line(buf, fileptr, &line1ins, len); /* reset the line length, in preparation for the next line */ len = 0; num_lines++; buf[0] = '\0'; i = 0; #ifndef NANO_SMALL /* If it's a Mac file (no LF just a CR), and file conversion isn't disabled, handle it! */ } else if (!ISSET(NO_CONVERT) && i > 0 && buf[i - 1] == '\r') { fileformat = 2; /* read in the line properly */ fileptr = read_line(buf, fileptr, &line1ins, len); /* reset the line length, in preparation for the next line; since we've already read in the next character, reset it to 1 instead of 0 */ len = 1; num_lines++; totsize++; buf[0] = input; buf[1] = '\0'; i = 1; #endif } else { /* Calculate the total length of the line; it might have nulls in it, so we can't just use strlen(). */ len++; /* Now we allocate a bigger buffer 128 characters at a time. If we allocate a lot of space for one line, we may indeed have to use a buffer this big later on, so we don't decrease it at all. We do free it at the end, though. */ if (i >= bufx - 1) { bufx += 128; buf = nrealloc(buf, bufx); } buf[i] = input; buf[i + 1] = '\0'; i++; } totsize++; } /* This conditional duplicates previous read_byte() behavior; perhaps this could use some better handling. */ if (ferror(f)) nperror(filename); fclose(f); /* Did we not get a newline but still have stuff to do? */ if (len > 0) { #ifndef NANO_SMALL /* If file conversion isn't disabled, the last character in this file is a CR and fileformat isn't set yet, make sure it's set to Mac format */ if (!ISSET(NO_CONVERT) && buf[len - 1] == '\r' && fileformat == 0) fileformat = 2; #endif /* read in the LAST line properly */ fileptr = read_line(buf, fileptr, &line1ins, len); num_lines++; totsize++; buf[0] = '\0'; } #ifndef NANO_SMALL else if (!ISSET(NO_CONVERT) && input == '\r') { /* 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 */ buf[0] = input; buf[1] = '\0'; len = 1; fileptr = read_line(buf, fileptr, &line1ins, len); num_lines++; totsize++; buf[0] = '\0'; } #endif free(buf); #ifndef NANO_SMALL /* If NO_CONVERT wasn't set before we read the file, but it is now, unset it again. */ if (!old_no_convert && ISSET(NO_CONVERT)) UNSET(NO_CONVERT); #endif /* Did we even GET a file if we don't already have one? */ if (totsize == 0 || fileptr == NULL) new_file(); /* Did we try to insert a file of 0 bytes? */ 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(); totsize--; load_file(quiet); } } #ifndef NANO_SMALL if (fileformat == 2) statusbar(__("Read %d line (Converted from Mac format)", "Read %d lines (Converted from Mac format)", num_lines), num_lines); else if (fileformat == 1) statusbar(__("Read %d line (Converted from DOS format)", "Read %d lines (Converted from DOS format)", num_lines), num_lines); else #endif statusbar(__("Read %d line", "Read %d lines", num_lines), num_lines); totlines += num_lines; return 1; } /* Open the file (and decide if it exists). */ int open_file(const char *filename, int insert, int quiet) { int fd; FILE *f; struct stat fileinfo; if (filename[0] == '\0' || stat(filename, &fileinfo) == -1) { if (insert && !quiet) { statusbar(_("\"%s\" not found"), filename); return -1; } else { /* We have a new file */ statusbar(_("New File")); new_file(); } } else if ((fd = open(filename, O_RDONLY)) == -1) { /* If we're in multibuffer mode, don't be quiet when an error occurs while opening a file */ if (!quiet #ifdef ENABLE_MULTIBUFFER || ISSET(MULTIBUFFER) #endif ) statusbar("%s: %s", strerror(errno), filename); if (!insert) new_file(); return -1; } else { /* File is A-OK */ if (S_ISDIR(fileinfo.st_mode) || S_ISCHR(fileinfo.st_mode) || S_ISBLK(fileinfo.st_mode)) { if (S_ISDIR(fileinfo.st_mode)) statusbar(_("\"%s\" is a directory"), filename); else /* Don't open character or block files. Sorry, /dev/sndstat! */ statusbar(_("File \"%s\" is a device file"), filename); if (!insert) new_file(); return -1; } if (!quiet) statusbar(_("Reading File")); f = fdopen(fd, "rb"); /* Binary for our own line-end munging */ if (f == NULL) { nperror("fdopen"); return -1; } read_file(f, filename, quiet); #ifndef NANO_SMALL stat(filename, &originalfilestat); #endif } return 1; } /* This function will return the name of the first available extension * of a filename (starting with the filename, then filename.1, etc). * Memory is allocated for the return value. If no writable extension * exists, we return "". */ char *get_next_filename(const char *name) { int i = 0; char *buf = NULL; struct stat fs; buf = charalloc(strlen(name) + num_of_digits(INT_MAX) + 2); strcpy(buf, name); while (1) { if (stat(buf, &fs) == -1) return buf; if (i == INT_MAX) break; i++; strcpy(buf, name); sprintf(&buf[strlen(name)], ".%d", i); } /* We get here only if there is no possible save file. */ buf[0] = '\0'; return buf; } int do_insertfile(int loading_file) { int i, old_current_x = current_x; char *realname = NULL; static char *inspath = NULL; if (inspath == NULL) { inspath = charalloc(1); inspath[0] = '\0'; } wrap_reset(); #if !defined(DISABLE_BROWSER) || (!defined(DISABLE_MOUSE) && defined(NCURSES_MOUSE_VERSION)) currshortcut = insertfile_list; #endif #ifndef DISABLE_OPERATINGDIR if (operating_dir != NULL && strcmp(operating_dir, ".") != 0) #ifdef ENABLE_MULTIBUFFER if (ISSET(MULTIBUFFER)) i = statusq(1, insertfile_list, inspath, #ifndef NANO_SMALL 0, #endif _("File to insert into new buffer [from %s] "), operating_dir); else #endif i = statusq(1, insertfile_list, inspath, #ifndef NANO_SMALL 0, #endif _("File to insert [from %s] "), operating_dir); else #endif #ifdef ENABLE_MULTIBUFFER if (ISSET(MULTIBUFFER)) i = statusq(1, insertfile_list, inspath, #ifndef NANO_SMALL 0, #endif _("File to insert into new buffer [from ./] ")); else #endif /* ENABLE_MULTIBUFFER */ i = statusq(1, insertfile_list, inspath, #ifndef NANO_SMALL 0, #endif _("File to insert [from ./] ")); if (i != -1) { inspath = mallocstrcpy(inspath, answer); #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), answer); #endif realname = real_dir_from_tilde(answer); #ifndef DISABLE_BROWSER if (i == NANO_TOFILES_KEY) { char *tmp = do_browse_from(realname); #if !defined(DISABLE_HELP) || (!defined(DISABLE_MOUSE) && defined(NCURSES_MOUSE_VERSION)) currshortcut = insertfile_list; #endif #ifdef DISABLE_TABCOMP realname = NULL; #endif if (tmp != NULL) { free(realname); realname = tmp; } else return do_insertfile(loading_file); } #endif #ifndef DISABLE_OPERATINGDIR if (operating_dir != NULL && check_operating_dir(realname, 0)) { statusbar(_("Can't insert file from outside of %s"), operating_dir); return 0; } #endif #ifndef NANO_SMALL #ifdef ENABLE_MULTIBUFFER if (i == TOGGLE_LOAD_KEY) { /* don't allow toggling if we're in both view mode and multibuffer mode now */ if (!ISSET(VIEW_MODE) || !ISSET(MULTIBUFFER)) TOGGLE(MULTIBUFFER); return do_insertfile(ISSET(MULTIBUFFER)); } #endif /* ENABLE_MULTIBUFFER */ if (i == NANO_EXTCMD_KEY) { int ts; ts = statusq(1, extcmd_list, "", 0, _("Command to execute ")); if (ts == -1 || answer == NULL || answer[0] == '\0') { statusbar(_("Cancelled")); UNSET(KEEP_CUTBUFFER); display_main_list(); return 0; } } #endif /* !NANO_SMALL */ #ifdef ENABLE_MULTIBUFFER if (loading_file) { /* update the current entry in the open_files structure */ add_open_file(1); new_file(); UNSET(MODIFIED); #ifndef NANO_SMALL UNSET(MARK_ISSET); #endif } #endif #ifndef NANO_SMALL if (i == NANO_EXTCMD_KEY) i = open_pipe(answer); else #endif /* NANO_SMALL */ i = open_file(realname, 1, loading_file); #ifdef ENABLE_MULTIBUFFER if (loading_file) { /* if there was an error opening the file, free() realname, free() fileage (which now points to the new buffer we created to hold the file), reload the buffer we had open before, and skip the insertion; otherwise, save realname in filename and continue the insertion */ if (i == -1) { free(realname); free(fileage); load_open_file(); goto skip_insert; } else filename = mallocstrcpy(filename, realname); } #endif free(realname); #ifdef DEBUG dump_buffer(fileage); #endif #ifdef ENABLE_MULTIBUFFER if (loading_file) load_file(0); else #endif set_modified(); /* Here we want to rebuild the edit window */ fix_editbot(); #ifdef ENABLE_MULTIBUFFER /* If we've loaded another file, update the titlebar's contents */ if (loading_file) { clearok(topwin, FALSE); titlebar(NULL); /* And re-init the shortcut list */ shortcut_init(0); } #endif #ifdef ENABLE_MULTIBUFFER if (!loading_file) { #endif /* Restore the old x-coordinate position */ current_x = old_current_x; #ifdef ENABLE_MULTIBUFFER } #endif /* If we've gone off the bottom, recenter; otherwise, just redraw */ if (current->lineno > editbot->lineno) edit_update(current, CENTER); else edit_refresh(); } else { statusbar(_("Cancelled")); i = 0; } #ifdef ENABLE_MULTIBUFFER skip_insert: #endif free(inspath); inspath = NULL; UNSET(KEEP_CUTBUFFER); display_main_list(); return i; } int do_insertfile_void(void) { int result = 0; #ifdef ENABLE_MULTIBUFFER if (ISSET(VIEW_MODE)) { if (ISSET(MULTIBUFFER)) result = do_insertfile(1); else statusbar(_("Key illegal in non-multibuffer mode")); } else result = do_insertfile(ISSET(MULTIBUFFER)); #else result = do_insertfile(0); #endif display_main_list(); return result; } #ifdef ENABLE_MULTIBUFFER /* Create a new openfilestruct node. */ openfilestruct *make_new_opennode(openfilestruct *prevnode) { openfilestruct *newnode = nmalloc(sizeof(openfilestruct)); newnode->filename = NULL; newnode->fileage = NULL; newnode->filebot = NULL; newnode->prev = prevnode; newnode->next = NULL; return newnode; } /* Splice a node into an existing openfilestruct. */ void splice_opennode(openfilestruct *begin, openfilestruct *newnode, openfilestruct *end) { newnode->next = end; newnode->prev = begin; begin->next = newnode; if (end != NULL) end->prev = newnode; } /* Unlink a node from the rest of the openfilestruct. */ void unlink_opennode(const openfilestruct *fileptr) { assert(fileptr != NULL); if (fileptr->prev != NULL) fileptr->prev->next = fileptr->next; if (fileptr->next != NULL) fileptr->next->prev = fileptr->prev; } /* Delete a node from the openfilestruct. */ void delete_opennode(openfilestruct *fileptr) { if (fileptr != NULL) { if (fileptr->filename != NULL) free(fileptr->filename); if (fileptr->fileage != NULL) free_filestruct(fileptr->fileage); free(fileptr); } } /* Deallocate all memory associated with this and later files, * including the lines of text. */ void free_openfilestruct(openfilestruct *src) { if (src != NULL) { while (src->next != NULL) { src = src->next; delete_opennode(src->prev); #ifdef DEBUG fprintf(stderr, _("delete_opennode(): free'd a node, YAY!\n")); #endif } delete_opennode(src); #ifdef DEBUG fprintf(stderr, _("delete_opennode(): free'd last node.\n")); #endif } } /* * Add/update an entry to the open_files openfilestruct. If update is * zero, a new entry is created; otherwise, the current entry is updated. * Return 0 on success or 1 on error. */ int add_open_file(int update) { openfilestruct *tmp; if (fileage == NULL || current == NULL || filename == NULL) return 1; /* if no entries, make the first one */ if (open_files == NULL) open_files = make_new_opennode(NULL); else if (!update) { /* otherwise, if we're not updating, make a new entry for open_files and splice it in after the current one */ #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), open_files->filename); #endif tmp = make_new_opennode(NULL); splice_opennode(open_files, tmp, open_files->next); open_files = open_files->next; } /* save current filename */ open_files->filename = mallocstrcpy(open_files->filename, filename); #ifndef NANO_SMALL /* save the file's stat */ open_files->originalfilestat = originalfilestat; #endif /* save current total number of lines */ open_files->file_totlines = totlines; /* save current total size */ open_files->file_totsize = totsize; /* save current x-coordinate position */ open_files->file_current_x = current_x; /* save current y-coordinate position */ open_files->file_current_y = current_y; /* save current place we want */ open_files->file_placewewant = placewewant; /* save current line number */ open_files->file_lineno = current->lineno; /* if we're updating, save current modification status (and marking status, if available) */ if (update) { #ifndef NANO_SMALL open_files->file_flags = (MODIFIED & ISSET(MODIFIED)) | (MARK_ISSET & ISSET(MARK_ISSET)); if (ISSET(MARK_ISSET)) { open_files->file_mark_beginbuf = mark_beginbuf; open_files->file_mark_beginx = mark_beginx; } #else open_files->file_flags = (MODIFIED & ISSET(MODIFIED)); #endif } /* if we're in view mode and updating, the file contents won't have changed, so we won't bother resaving the filestruct then; otherwise, we will */ if (!(ISSET(VIEW_MODE) || !update)) { /* save current file buffer */ open_files->fileage = fileage; open_files->filebot = filebot; } #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), open_files->filename); #endif return 0; } /* * Read the current entry in the open_files structure and set up the * currently open file using that entry's information. Return 0 on * success or 1 on error. */ int load_open_file(void) { if (open_files == NULL) return 1; /* set up the filename, the file buffer, the total number of lines in the file, and the total file size */ filename = mallocstrcpy(filename, open_files->filename); #ifndef NANO_SMALL originalfilestat = open_files->originalfilestat; #endif fileage = open_files->fileage; current = fileage; filebot = open_files->filebot; totlines = open_files->file_totlines; totsize = open_files->file_totsize; /* restore modification status */ if (open_files->file_flags & MODIFIED) SET(MODIFIED); else UNSET(MODIFIED); #ifndef NANO_SMALL /* restore marking status */ if (open_files->file_flags & MARK_ISSET) { mark_beginbuf = open_files->file_mark_beginbuf; mark_beginx = open_files->file_mark_beginx; SET(MARK_ISSET); } else UNSET(MARK_ISSET); #endif #ifdef ENABLE_COLOR update_color(); #endif /* restore full file position: line number, x-coordinate, y- coordinate, place we want */ do_gotopos(open_files->file_lineno, open_files->file_current_x, open_files->file_current_y, open_files->file_placewewant); /* update the titlebar */ clearok(topwin, FALSE); titlebar(NULL); /* if we're constantly displaying the cursor position and DISABLE_CURPOS isn't set, update it (and do so unconditionally, in the rare case that the character count is the same but the line count isn't) */ if (ISSET(CONSTUPDATE) && !ISSET(DISABLE_CURPOS)) do_cursorpos(0); /* now we're done */ return 0; } /* * Open the previous entry in the open_files structure. If closing_file * is zero, update the current entry before switching from it. * Otherwise, we are about to close that entry, so don't bother doing so. * Return 0 on success and 1 on error. */ int open_prevfile(int closing_file) { if (open_files == NULL) return 1; /* if we're not about to close the current entry, update it before doing anything */ if (!closing_file) add_open_file(1); if (open_files->prev == NULL && open_files->next == NULL) { /* only one file open */ if (!closing_file) statusbar(_("No more open files")); return 1; } if (open_files->prev != NULL) { open_files = open_files->prev; #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), open_files->filename); #endif } else if (open_files->next != NULL) { /* if we're at the beginning, wrap around to the end */ while (open_files->next != NULL) open_files = open_files->next; #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), open_files->filename); #endif } load_open_file(); statusbar(_("Switched to %s"), ((open_files->filename[0] == '\0') ? "New Buffer" : open_files->filename)); #ifdef DEBUG dump_buffer(current); #endif return 0; } /* This function is used by the shortcut list. */ int open_prevfile_void(void) { return open_prevfile(0); } /* * Open the next entry in the open_files structure. If closing_file is * zero, update the current entry before switching from it. Otherwise, we * are about to close that entry, so don't bother doing so. Return 0 on * success and 1 on error. */ int open_nextfile(int closing_file) { if (open_files == NULL) return 1; /* if we're not about to close the current entry, update it before doing anything */ if (!closing_file) add_open_file(1); if (open_files->prev == NULL && open_files->next == NULL) { /* only one file open */ if (!closing_file) statusbar(_("No more open files")); return 1; } if (open_files->next != NULL) { open_files = open_files->next; #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), open_files->filename); #endif } else if (open_files->prev != NULL) { /* if we're at the end, wrap around to the beginning */ while (open_files->prev != NULL) { open_files = open_files->prev; #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), open_files->filename); #endif } } load_open_file(); statusbar(_("Switched to %s"), ((open_files->filename[0] == '\0') ? "New Buffer" : open_files->filename)); #ifdef DEBUG dump_buffer(current); #endif return 0; } /* This function is used by the shortcut list. */ int open_nextfile_void(void) { return open_nextfile(0); } /* * Delete an entry from the open_files filestruct. After deletion of an * entry, the next or previous entry is opened, whichever is found first. * Return 0 on success or 1 on error. */ int close_open_file(void) { openfilestruct *tmp; if (open_files == NULL) return 1; /* make sure open_files->fileage and fileage, and open_files->filebot and filebot, are in sync; they might not be if lines have been cut from the top or bottom of the file */ open_files->fileage = fileage; open_files->filebot = filebot; tmp = open_files; if (open_nextfile(1)) { if (open_prevfile(1)) return 1; } unlink_opennode(tmp); delete_opennode(tmp); shortcut_init(0); display_main_list(); return 0; } #endif /* MULTIBUFFER */ #if !defined(DISABLE_SPELLER) || !defined(DISABLE_OPERATINGDIR) /* * When passed "[relative path]" or "[relative path][filename]" in * origpath, return "[full path]" or "[full path][filename]" on success, * or NULL on error. This is still done if the file doesn't exist but * the relative path does (since the file could exist in memory but not * yet on disk); it is not done if the relative path doesn't exist (since * the first call to chdir() will fail then). */ char *get_full_path(const char *origpath) { char *newpath = NULL, *last_slash, *d_here, *d_there, *d_there_file, tmp; int path_only, last_slash_index; struct stat fileinfo; char *expanded_origpath; /* first, get the current directory, and tack a slash onto the end of it, unless it turns out to be "/", in which case leave it alone */ #ifdef PATH_MAX d_here = getcwd(NULL, PATH_MAX + 1); #else d_here = getcwd(NULL, 0); #endif if (d_here != NULL) { align(&d_here); if (strcmp(d_here, "/")) { d_here = nrealloc(d_here, strlen(d_here) + 2); strcat(d_here, "/"); } /* stat origpath; if stat() fails, assume that origpath refers to a new file that hasn't been saved to disk yet (i. e. set path_only to 0); if stat() succeeds, set path_only to 0 if origpath doesn't refer to a directory, or to 1 if it does */ path_only = !stat(origpath, &fileinfo) && S_ISDIR(fileinfo.st_mode); expanded_origpath = real_dir_from_tilde(origpath); /* save the value of origpath in both d_there and d_there_file */ d_there = mallocstrcpy(NULL, expanded_origpath); d_there_file = mallocstrcpy(NULL, expanded_origpath); free(expanded_origpath); /* if we have a path but no filename, tack slashes onto the ends of both d_there and d_there_file, if they don't end in slashes already */ if (path_only) { tmp = d_there[strlen(d_there) - 1]; if (tmp != '/') { d_there = nrealloc(d_there, strlen(d_there) + 2); strcat(d_there, "/"); d_there_file = nrealloc(d_there_file, strlen(d_there_file) + 2); strcat(d_there_file, "/"); } } /* search for the last slash in d_there */ last_slash = strrchr(d_there, '/'); /* if we didn't find one, copy d_here into d_there; all data is then set up */ if (last_slash == NULL) d_there = mallocstrcpy(d_there, d_here); else { /* otherwise, remove all non-path elements from d_there (i. e. everything after the last slash) */ last_slash_index = strlen(d_there) - strlen(last_slash); null_at(&d_there, last_slash_index + 1); /* and remove all non-file elements from d_there_file (i. e. everything before and including the last slash); if we have a path but no filename, don't do anything */ if (!path_only) { last_slash = strrchr(d_there_file, '/'); last_slash++; strcpy(d_there_file, last_slash); align(&d_there_file); } /* now go to the path specified in d_there */ if (chdir(d_there) != -1) { /* get the full pathname, and save it back in d_there, tacking a slash on the end if we have a path but no filename; if the saving fails, get out */ free(d_there); #ifdef PATH_MAX d_there = getcwd(NULL, PATH_MAX + 1); #else d_there = getcwd(NULL, 0); #endif align(&d_there); if (d_there != NULL) { /* add a slash to d_there, unless it's "/", in which case we don't need it */ if (strcmp(d_there, "/")) { d_there = nrealloc(d_there, strlen(d_there) + 2); strcat(d_there, "/"); } } else return NULL; } /* finally, go back to where we were before, d_here (no error checking is done on this chdir(), because we can do nothing if it fails) */ chdir(d_here); } /* all data is set up; fill in newpath */ /* if we have a path and a filename, newpath = d_there + d_there_file; otherwise, newpath = d_there */ if (!path_only) { newpath = charalloc(strlen(d_there) + strlen(d_there_file) + 1); strcpy(newpath, d_there); strcat(newpath, d_there_file); } else { newpath = charalloc(strlen(d_there) + 1); strcpy(newpath, d_there); } /* finally, clean up */ free(d_there_file); free(d_there); free(d_here); } return newpath; } #endif /* !DISABLE_SPELLER || !DISABLE_OPERATINGDIR */ #ifndef DISABLE_SPELLER /* * This function accepts a path and returns the full path (via * get_full_path()). On error, if the path doesn't reference a * directory, or if the directory isn't writable, it returns NULL. */ char *check_writable_directory(const char *path) { char *full_path = get_full_path(path); int writable; struct stat fileinfo; /* if get_full_path() failed, return NULL */ if (full_path == NULL) return NULL; /* otherwise, stat() the full path to see if it's writable by the user; set writable to 1 if it is, or 0 if it isn't */ writable = !stat(full_path, &fileinfo) && (fileinfo.st_mode & S_IWUSR); /* if the full path doesn't end in a slash (meaning get_full_path() found that it isn't a directory) or isn't writable, free full_path and return NULL */ if (full_path[strlen(full_path) - 1] != '/' || writable == 0) { free(full_path); return NULL; } /* otherwise, return the full path */ return full_path; } /* * This function accepts a directory name and filename prefix the same * way that tempnam() does, determines the location for its temporary * file the same way that tempnam() does, safely creates the temporary * file there via mkstemp(), and returns the name of the temporary file * the same way that tempnam() does. It does not reference the value of * TMP_MAX because the total number of random filenames that it can * generate using one prefix is equal to 256**6, which is a sufficiently * large number to handle most cases. Since the behavior after tempnam() * generates TMP_MAX random filenames is implementation-defined, my * implementation is to go on generating random filenames regardless of * it. */ char *safe_tempnam(const char *dirname, const char *filename_prefix) { char *full_tempdir = NULL; const char *TMPDIR_env; int filedesc; /* 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 */ TMPDIR_env = getenv("TMPDIR"); if (TMPDIR_env != NULL && TMPDIR_env[0] != '\0') full_tempdir = check_writable_directory(TMPDIR_env); /* if $TMPDIR is blank or isn't set, or isn't a writable directory, and dirname isn't NULL, try it; otherwise, leave full_tempdir set to NULL */ if (full_tempdir == NULL && dirname != NULL) full_tempdir = check_writable_directory(dirname); /* if $TMPDIR is blank or isn't set, or if it isn't a writable directory, and dirname is NULL, try P_tmpdir instead */ if (full_tempdir == NULL) full_tempdir = check_writable_directory(P_tmpdir); /* if P_tmpdir didn't work, use /tmp instead */ if (full_tempdir == NULL) { full_tempdir = charalloc(6); strcpy(full_tempdir, "/tmp/"); } full_tempdir = nrealloc(full_tempdir, strlen(full_tempdir) + 12); /* like tempnam(), use only the first 5 characters of the prefix */ strncat(full_tempdir, filename_prefix, 5); strcat(full_tempdir, "XXXXXX"); filedesc = mkstemp(full_tempdir); /* if mkstemp succeeded, close the resulting file; afterwards, it'll be 0 bytes long, so delete it; finally, return the filename (all that's left of it) */ if (filedesc != -1) { close(filedesc); unlink(full_tempdir); return full_tempdir; } free(full_tempdir); return NULL; } #endif /* !DISABLE_SPELLER */ #ifndef DISABLE_OPERATINGDIR /* Initialize full_operating_dir based on operating_dir. */ void init_operating_dir(void) { assert(full_operating_dir == NULL); if (operating_dir == NULL) return; full_operating_dir = get_full_path(operating_dir); /* If get_full_path() failed or the operating directory is inaccessible, unset operating_dir. */ if (full_operating_dir == NULL || chdir(full_operating_dir) == -1) { free(full_operating_dir); full_operating_dir = NULL; free(operating_dir); operating_dir = NULL; } } /* * Check to see if we're inside the operating directory. Return 0 if we * are, or 1 otherwise. If allow_tabcomp is nonzero, allow incomplete * names that would be matches for the operating directory, so that tab * completion will work. */ int check_operating_dir(const char *currpath, int allow_tabcomp) { /* The char *full_operating_dir is global for mem cleanup, and therefore we only need to get it the first time this function is called; also, a relative operating directory path will only be handled properly if this is done */ char *fullpath; int retval = 0; const char *whereami1, *whereami2 = NULL; /* if no operating directory is set, don't bother doing anything */ if (operating_dir == NULL) return 0; fullpath = get_full_path(currpath); if (fullpath == NULL) return 1; whereami1 = strstr(fullpath, full_operating_dir); if (allow_tabcomp) whereami2 = strstr(full_operating_dir, fullpath); /* 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 */ if (whereami1 != fullpath && whereami2 != full_operating_dir) retval = 1; free(fullpath); /* otherwise, we're still inside it */ return retval; } #endif /* * Write a file out. If tmp is nonzero, we set the umask to 0600, * we don't set the global variable filename to its name, and don't * print out how many lines we wrote on the statusbar. * * tmp means we are writing a tmp file in a secure fashion. We use * it when spell checking or dumping the file on an error. * * append == 1 means we are appending instead of overwriting. * append == 2 means we are prepending instead of overwriting. * * nonamechange means don't change the current filename, it is ignored * if tmp is nonzero or if we're appending/prepending. */ int write_file(const char *name, int tmp, int append, int nonamechange) { int retval = -1; /* Instead of returning in this function, you should always * merely set retval then goto cleanup_and_exit. */ long size; int lineswritten = 0; char *buf = NULL; const filestruct *fileptr; FILE *f; int fd; int mask = 0, realexists, anyexists; struct stat st, lst; char *realname = NULL; if (name[0] == '\0') { statusbar(_("Cancelled")); return -1; } titlebar(NULL); fileptr = fileage; realname = real_dir_from_tilde(name); #ifndef DISABLE_OPERATINGDIR /* If we're writing a temporary file, we're probably going outside the operating directory, so skip the operating directory test. */ if (!tmp && operating_dir != NULL && check_operating_dir(realname, 0)) { statusbar(_("Can't write outside of %s"), operating_dir); goto cleanup_and_exit; } #endif /* Save the state of file at the end of the symlink (if there is one). */ realexists = stat(realname, &st); #ifndef NANO_SMALL /* We backup only if the backup toggle is set, the file isn't 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. */ if (ISSET(BACKUP_FILE) && !tmp && realexists == 0 && (append != 0 || ISSET(MARK_ISSET) || originalfilestat.st_mtime == st.st_mtime)) { FILE *backup_file; char *backupname = NULL; char backupbuf[COPYFILEBLOCKSIZE]; size_t bytesread; struct utimbuf filetime; /* save the original file's access and modification times */ filetime.actime = originalfilestat.st_atime; filetime.modtime = originalfilestat.st_mtime; /* open the original file to copy to the backup */ f = fopen(realname, "rb"); if (f == NULL) { statusbar(_("Could not read %s for backup: %s"), realname, strerror(errno)); return -1; } backupname = charalloc(strlen(realname) + 2); sprintf(backupname, "%s~", realname); /* get a file descriptor for the destination backup file */ backup_file = fopen(backupname, "wb"); if (backup_file == NULL) { statusbar(_("Couldn't write backup: %s"), strerror(errno)); free(backupname); return -1; } #ifdef DEBUG fprintf(stderr, _("Backing up %s to %s\n"), realname, backupname); #endif /* copy the file */ while ((bytesread = fread(backupbuf, sizeof(char), COPYFILEBLOCKSIZE, f)) > 0) if (fwrite(backupbuf, sizeof(char), bytesread, backup_file) <= 0) break; fclose(backup_file); fclose(f); if (chmod(backupname, originalfilestat.st_mode) == -1) statusbar(_("Could not set permissions %o on backup %s: %s"), originalfilestat.st_mode, backupname, strerror(errno)); if (chown(backupname, originalfilestat.st_uid, originalfilestat.st_gid) == -1) statusbar(_("Could not set owner %d/group %d on backup %s: %s"), originalfilestat.st_uid, originalfilestat.st_gid, backupname, strerror(errno)); if (utime(backupname, &filetime) == -1) statusbar(_("Could not set access/modification time on backup %s: %s"), backupname, strerror(errno)); free(backupname); } #endif /* Stat the link itself for the check... */ anyexists = lstat(realname, &lst); /* New case: if the file exists, just give up */ if (tmp && anyexists != -1) goto cleanup_and_exit; /* NOTE: If you change this statement, you MUST CHANGE the if statement below (that says: if (realexists == -1 || tmp || (ISSET(NOFOLLOW_SYMLINKS) && S_ISLNK(lst.st_mode))) { to reflect whether or not to link/unlink/rename the file */ else if (append != 2 && (!ISSET(NOFOLLOW_SYMLINKS) || !S_ISLNK(lst.st_mode) || tmp)) { /* Use O_EXCL if tmp is nonzero. This is now copied from joe, because wiggy says so *shrug*. */ if (append != 0) fd = open(realname, O_WRONLY | O_CREAT | O_APPEND, (S_IRUSR | S_IWUSR)); else if (tmp) fd = open(realname, O_WRONLY | O_CREAT | O_EXCL, (S_IRUSR | S_IWUSR)); else fd = open(realname, O_WRONLY | O_CREAT | O_TRUNC, (S_IRUSR | S_IWUSR)); /* First, just give up if we couldn't even open the file */ if (fd == -1) { if (!tmp && ISSET(TEMP_OPT)) { UNSET(TEMP_OPT); retval = do_writeout(filename, 1, 0); } else statusbar(_("Could not open file for writing: %s"), strerror(errno)); goto cleanup_and_exit; } } /* Don't follow symlink. Create new file. */ else { buf = charalloc(strlen(realname) + 8); strcpy(buf, realname); strcat(buf, ".XXXXXX"); if ((fd = mkstemp(buf)) == -1) { if (ISSET(TEMP_OPT)) { UNSET(TEMP_OPT); retval = do_writeout(filename, 1, 0); } else statusbar(_("Could not open file for writing: %s"), strerror(errno)); goto cleanup_and_exit; } } #ifdef DEBUG dump_buffer(fileage); #endif f = fdopen(fd, append == 1 ? "ab" : "wb"); if (f == NULL) { statusbar(_("Could not open file for writing: %s"), strerror(errno)); goto cleanup_and_exit; } while (fileptr != NULL && fileptr->next != NULL) { int data_len; /* Next line is so we discount the "magic line" */ if (filebot == fileptr && fileptr->data[0] == '\0') break; data_len = strlen(fileptr->data); /* newlines to nulls, just before we write to disk */ sunder(fileptr->data); size = fwrite(fileptr->data, 1, data_len, f); /* nulls to newlines; data_len is the string's real length here */ unsunder(fileptr->data, data_len); if (size < data_len) { statusbar(_("Could not open file for writing: %s"), strerror(errno)); fclose(f); goto cleanup_and_exit; } #ifdef DEBUG else fprintf(stderr, _("Wrote >%s\n"), fileptr->data); #endif #ifndef NANO_SMALL if (ISSET(DOS_FILE) || ISSET(MAC_FILE)) putc('\r', f); if (!ISSET(MAC_FILE)) #endif putc('\n', f); fileptr = fileptr->next; lineswritten++; } if (fileptr != NULL) { int data_len = strlen(fileptr->data); /* newlines to nulls, just before we write to disk */ sunder(fileptr->data); size = fwrite(fileptr->data, 1, data_len, f); /* nulls to newlines; data_len is the string's real length here */ unsunder(fileptr->data, data_len); if (size < data_len) { statusbar(_("Could not open file for writing: %s"), strerror(errno)); goto cleanup_and_exit; } else if (data_len > 0) { #ifndef NANO_SMALL if (ISSET(DOS_FILE) || ISSET(MAC_FILE)) { if (putc('\r', f) == EOF) { statusbar(_("Could not open file for writing: %s"), strerror(errno)); fclose(f); goto cleanup_and_exit; } lineswritten++; } if (!ISSET(MAC_FILE)) #endif { if (putc('\n', f) == EOF) { statusbar(_("Could not open file for writing: %s"), strerror(errno)); fclose(f); goto cleanup_and_exit; } lineswritten++; } } } if (fclose(f) != 0) { statusbar(_("Could not close %s: %s"), realname, strerror(errno)); unlink(buf); goto cleanup_and_exit; } /* if we're prepending, open the real file, and append it here */ if (append == 2) { int fd_source, fd_dest; FILE *f_source, *f_dest; int prechar; if ((fd_dest = open(buf, O_WRONLY | O_APPEND, (S_IRUSR | S_IWUSR))) == -1) { statusbar(_("Could not reopen %s: %s"), buf, strerror(errno)); goto cleanup_and_exit; } f_dest = fdopen(fd_dest, "wb"); if (f_dest == NULL) { statusbar(_("Could not reopen %s: %s"), buf, strerror(errno)); close(fd_dest); goto cleanup_and_exit; } if ((fd_source = open(realname, O_RDONLY | O_CREAT)) == -1) { statusbar(_("Could not open %s for prepend: %s"), realname, strerror(errno)); fclose(f_dest); goto cleanup_and_exit; } f_source = fdopen(fd_source, "rb"); if (f_source == NULL) { statusbar(_("Could not open %s for prepend: %s"), realname, strerror(errno)); fclose(f_dest); close(fd_source); goto cleanup_and_exit; } /* Doing this in blocks is an exercise left to some other reader. */ while ((prechar = getc(f_source)) != EOF) { if (putc(prechar, f_dest) == EOF) { statusbar(_("Could not open %s for prepend: %s"), realname, strerror(errno)); fclose(f_source); fclose(f_dest); goto cleanup_and_exit; } } if (ferror(f_source)) { statusbar(_("Could not reopen %s: %s"), buf, strerror(errno)); fclose(f_source); fclose(f_dest); goto cleanup_and_exit; } fclose(f_source); fclose(f_dest); } if (realexists == -1 || tmp || (ISSET(NOFOLLOW_SYMLINKS) && S_ISLNK(lst.st_mode))) { /* Use default umask as file permissions if file is a new file. */ mask = umask(0); umask(mask); if (tmp) /* We don't want anyone reading our temporary file! */ mask = S_IRUSR | S_IWUSR; else mask = (S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH | S_IWOTH) & ~mask; } else /* Use permissions from file we are overwriting. */ mask = st.st_mode; if (append == 2 || (!tmp && (ISSET(NOFOLLOW_SYMLINKS) && S_ISLNK(lst.st_mode)))) { if (unlink(realname) == -1) { if (errno != ENOENT) { statusbar(_("Could not open %s for writing: %s"), realname, strerror(errno)); unlink(buf); goto cleanup_and_exit; } } if (link(buf, realname) != -1) unlink(buf); else if (errno != EPERM) { statusbar(_("Could not open %s for writing: %s"), name, strerror(errno)); unlink(buf); goto cleanup_and_exit; } else if (rename(buf, realname) == -1) { /* Try a rename?? */ statusbar(_("Could not open %s for writing: %s"), realname, strerror(errno)); unlink(buf); goto cleanup_and_exit; } } if (chmod(realname, mask) == -1) statusbar(_("Could not set permissions %o on %s: %s"), mask, realname, strerror(errno)); if (!tmp && append == 0) { if (!nonamechange) { filename = mallocstrcpy(filename, realname); #ifdef ENABLE_COLOR update_color(); edit_refresh(); #endif } #ifndef NANO_SMALL /* Update originalfilestat to reference the file as it is now. */ stat(filename, &originalfilestat); #endif statusbar(__("Wrote %d line", "Wrote %d lines", lineswritten), lineswritten); UNSET(MODIFIED); titlebar(NULL); } retval = 1; cleanup_and_exit: free(realname); free(buf); return retval; } int do_writeout(const char *path, int exiting, int append) { int i = 0; #ifdef NANO_EXTRA static int did_cred = 0; #endif #if !defined(DISABLE_BROWSER) || (!defined(DISABLE_MOUSE) && defined(NCURSES_MOUSE_VERSION)) currshortcut = writefile_list; #endif answer = mallocstrcpy(answer, path); if (exiting && ISSET(TEMP_OPT)) { if (filename[0] != '\0') { i = write_file(answer, 0, 0, 0); display_main_list(); return i; } else { UNSET(TEMP_OPT); do_exit(); /* They cancelled, abort quit */ return -1; } } while (1) { #ifndef NANO_SMALL const char *formatstr, *backupstr; if (ISSET(MAC_FILE)) formatstr = _(" [Mac Format]"); else if (ISSET(DOS_FILE)) formatstr = _(" [DOS Format]"); else formatstr = ""; if (ISSET(BACKUP_FILE)) backupstr = _(" [Backup]"); else backupstr = ""; /* Be nice to the translation folks */ if (ISSET(MARK_ISSET) && !exiting) { if (append == 2) i = statusq(1, writefile_list, "", 0, "%s%s%s", _("Prepend Selection to File"), formatstr, backupstr); else if (append == 1) i = statusq(1, writefile_list, "", 0, "%s%s%s", _("Append Selection to File"), formatstr, backupstr); else i = statusq(1, writefile_list, "", 0, "%s%s%s", _("Write Selection to File"), formatstr, backupstr); } else { if (append == 2) i = statusq(1, writefile_list, answer, 0, "%s%s%s", _("File Name to Prepend to"), formatstr, backupstr); else if (append == 1) i = statusq(1, writefile_list, answer, 0, "%s%s%s", _("File Name to Append to"), formatstr, backupstr); else i = statusq(1, writefile_list, answer, 0, "%s%s%s", _("File Name to Write"), formatstr, backupstr); } #else if (append == 2) i = statusq(1, writefile_list, answer, "%s", _("File Name to Prepend to")); else if (append == 1) i = statusq(1, writefile_list, answer, "%s", _("File Name to Append to")); else i = statusq(1, writefile_list, answer, "%s", _("File Name to Write")); #endif /* !NANO_SMALL */ if (i == -1) { statusbar(_("Cancelled")); display_main_list(); return 0; } #ifndef DISABLE_BROWSER if (i == NANO_TOFILES_KEY) { char *tmp = do_browse_from(answer); currshortcut = writefile_list; if (tmp == NULL) continue; free(answer); answer = tmp; } else #endif /* !DISABLE_BROWSER */ #ifndef NANO_SMALL if (i == TOGGLE_DOS_KEY) { UNSET(MAC_FILE); TOGGLE(DOS_FILE); continue; } else if (i == TOGGLE_MAC_KEY) { UNSET(DOS_FILE); TOGGLE(MAC_FILE); continue; } else if (i == TOGGLE_BACKUP_KEY) { TOGGLE(BACKUP_FILE); continue; } else #endif /* !NANO_SMALL */ if (i == NANO_PREPEND_KEY) { append = append == 2 ? 0 : 2; continue; } else if (i == NANO_APPEND_KEY) { append = append == 1 ? 0 : 1; continue; } #ifdef DEBUG fprintf(stderr, _("filename is %s\n"), answer); #endif #ifdef NANO_EXTRA if (exiting && !ISSET(TEMP_OPT) && !strcasecmp(answer, "zzy") && !did_cred) { do_credits(); did_cred = 1; return -1; } #endif if (append == 0 && strcmp(answer, filename) != 0) { struct stat st; if (!stat(answer, &st)) { i = do_yesno(0, 0, _("File exists, OVERWRITE ?")); if (i == 0 || i == -1) continue; } } #ifndef NANO_SMALL /* Here's where we allow the selected text to be written to a separate file. */ if (ISSET(MARK_ISSET) && !exiting) { filestruct *fileagebak = fileage; filestruct *filebotbak = filebot; filestruct *cutback = cutbuffer; int oldmod = ISSET(MODIFIED); /* write_file() unsets the MODIFIED flag. */ cutbuffer = NULL; /* Put the marked text in the cutbuffer without changing the open file. */ cut_marked_segment(current, current_x, mark_beginbuf, mark_beginx, 0); fileage = cutbuffer; filebot = get_cutbottom(); i = write_file(answer, 0, append, 1); /* Now restore everything */ free_filestruct(cutbuffer); fileage = fileagebak; filebot = filebotbak; cutbuffer = cutback; if (oldmod) set_modified(); } else #endif /* !NANO_SMALL */ i = write_file(answer, 0, append, 0); #ifdef ENABLE_MULTIBUFFER /* If we're not about to exit, update the current entry in the open_files structure. */ if (!exiting) add_open_file(1); #endif display_main_list(); return i; } /* while (1) */ } int do_writeout_void(void) { return do_writeout(filename, 0, 0); } /* Return a malloc()ed string containing the actual directory, used * to convert ~user and ~/ notation... */ char *real_dir_from_tilde(const char *buf) { char *dirtmp = NULL; if (buf[0] == '~') { size_t i; const struct passwd *userdata; /* Figure how how much of the str we need to compare */ for (i = 1; buf[i] != '/' && buf[i] != '\0'; i++) ; if (i == 1) { /* Determine home directory using getpwent(), don't rely on $HOME */ uid_t euid = geteuid(); do { userdata = getpwent(); } while (userdata != NULL && userdata->pw_uid != euid); } else { do { userdata = getpwent(); } while (userdata != NULL && strncmp(userdata->pw_name, buf + 1, i - 1)); } endpwent(); if (userdata != NULL) { /* User found */ dirtmp = charalloc(strlen(userdata->pw_dir) + strlen(buf + i) + 1); sprintf(dirtmp, "%s%s", userdata->pw_dir, &buf[i]); } } if (dirtmp == NULL) dirtmp = mallocstrcpy(dirtmp, buf); return dirtmp; } #ifndef DISABLE_TABCOMP /* Tack a slash onto the string we're completing if it's a directory. We * assume there is room for one more character on the end of buf. The * return value says whether buf is a directory. */ int append_slash_if_dir(char *buf, int *lastwastab, int *place) { char *dirptr = real_dir_from_tilde(buf); struct stat fileinfo; int ret = 0; assert(dirptr != buf); if (stat(dirptr, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode)) { strncat(buf, "/", 1); (*place)++; /* now we start over again with # of tabs so far */ *lastwastab = 0; ret = 1; } free(dirptr); return ret; } /* * These functions (username_tab_completion(), cwd_tab_completion(), and * input_tab()) were taken from busybox 0.46 (cmdedit.c). Here is the * notice from that file: * * Termios command line History and Editting, originally * intended for NetBSD sh (ash) * Copyright (c) 1999 * Main code: Adam Rogoyski * Etc: Dave Cinege * Majorly adjusted/re-written for busybox: * Erik Andersen * * 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. * This code may safely be consumed by a BSD or GPL license. */ char **username_tab_completion(char *buf, int *num_matches) { char **matches = (char **)NULL; char *matchline = NULL; struct passwd *userdata; *num_matches = 0; matches = nmalloc(BUFSIZ * sizeof(char *)); strcat(buf, "*"); while ((userdata = getpwent()) != NULL) { if (check_wildcard_match(userdata->pw_name, &buf[1]) == TRUE) { /* Cool, found a match. Add it to the list * This makes a lot more sense to me (Chris) this way... */ #ifndef DISABLE_OPERATINGDIR /* ...unless the match exists outside the operating directory, in which case just go to the next match */ if (operating_dir != NULL) { if (check_operating_dir(userdata->pw_dir, 1) != 0) continue; } #endif matchline = charalloc(strlen(userdata->pw_name) + 2); sprintf(matchline, "~%s", userdata->pw_name); matches[*num_matches] = matchline; ++*num_matches; /* If there's no more room, bail out */ if (*num_matches == BUFSIZ) break; } } endpwent(); return matches; } /* This was originally called exe_n_cwd_tab_completion, but we're not worried about executables, only filenames :> */ char **cwd_tab_completion(char *buf, int *num_matches) { char *dirname, *dirtmp = NULL, *tmp = NULL, *tmp2 = NULL; char **matches = (char **)NULL; DIR *dir; struct dirent *next; matches = nmalloc(BUFSIZ * sizeof(char *)); /* Stick a wildcard onto the buf, for later use */ strcat(buf, "*"); /* Okie, if there's a / in the buffer, strip out the directory part */ if (buf[0] != '\0' && strstr(buf, "/") != NULL) { dirname = charalloc(strlen(buf) + 1); tmp = buf + strlen(buf); while (*tmp != '/' && tmp != buf) tmp--; tmp++; strncpy(dirname, buf, tmp - buf + 1); dirname[tmp - buf] = '\0'; } else { #ifdef PATH_MAX if ((dirname = getcwd(NULL, PATH_MAX + 1)) == NULL) #else /* The better, but apparently segfault-causing way */ if ((dirname = getcwd(NULL, 0)) == NULL) #endif /* PATH_MAX */ return matches; else tmp = buf; } #ifdef DEBUG fprintf(stderr, "\nDir = %s\n", dirname); fprintf(stderr, "\nbuf = %s\n", buf); fprintf(stderr, "\ntmp = %s\n", tmp); #endif dirtmp = real_dir_from_tilde(dirname); free(dirname); dirname = dirtmp; #ifdef DEBUG fprintf(stderr, "\nDir = %s\n", dirname); fprintf(stderr, "\nbuf = %s\n", buf); fprintf(stderr, "\ntmp = %s\n", tmp); #endif dir = opendir(dirname); if (dir == NULL) { /* Don't print an error, just shut up and return */ *num_matches = 0; beep(); return matches; } while ((next = readdir(dir)) != NULL) { #ifdef DEBUG fprintf(stderr, "Comparing \'%s\'\n", next->d_name); #endif /* See if this matches */ if (check_wildcard_match(next->d_name, tmp) == TRUE) { /* Cool, found a match. Add it to the list * This makes a lot more sense to me (Chris) this way... */ #ifndef DISABLE_OPERATINGDIR /* ...unless the match exists outside the operating 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 */ if (operating_dir != NULL) { tmp2 = charalloc(strlen(dirname) + strlen(next->d_name) + 2); strcpy(tmp2, dirname); strcat(tmp2, "/"); strcat(tmp2, next->d_name); if (check_operating_dir(tmp2, 1)) { free(tmp2); continue; } free(tmp2); } #endif tmp2 = NULL; tmp2 = charalloc(strlen(next->d_name) + 1); strcpy(tmp2, next->d_name); matches[*num_matches] = tmp2; ++*num_matches; /* If there's no more room, bail out */ if (*num_matches == BUFSIZ) break; } } return matches; } /* This function now has an arg which refers to how much the statusbar * (place) should be advanced, i.e. the new cursor pos. */ char *input_tab(char *buf, int place, int *lastwastab, int *newplace, int *list) { /* Do TAB completion */ static int num_matches = 0, match_matches = 0; static char **matches = (char **)NULL; int pos = place, i = 0, col = 0, editline = 0; int longestname = 0, is_dir = 0; char *foo; *list = 0; if (*lastwastab == FALSE) { char *tmp, *copyto, *matchbuf; *lastwastab = 1; /* Make a local copy of the string -- up to the position of the cursor */ matchbuf = (char *)nmalloc((strlen(buf) + 2) * sizeof(char)); memset(matchbuf, '\0', (strlen(buf) + 2)); strncpy(matchbuf, buf, place); tmp = matchbuf; /* skip any leading white space */ while (*tmp && isspace((int)*tmp)) ++tmp; /* Free up any memory already allocated */ if (matches != NULL) { for (i = i; i < num_matches; i++) free(matches[i]); free(matches); matches = (char **)NULL; num_matches = 0; } /* If the word starts with `~' and there is no slash in the word, * then try completing this word as a username. */ /* If the original string begins with a tilde, and the part we're trying to tab-complete doesn't contain a slash, copy the part we're tab-completing into buf, so tab completion will result in buf's containing only the tab-completed username. */ if (buf[0] == '~' && strchr(tmp, '/') == NULL) { buf = mallocstrcpy(buf, tmp); matches = username_tab_completion(tmp, &num_matches); } /* If we're in the middle of the original line, copy the string only up to the cursor position into buf, so tab completion will result in buf's containing only the tab-completed path/filename. */ else if (strlen(buf) > strlen(tmp)) buf = mallocstrcpy(buf, tmp); /* Try to match everything in the current working directory that * matches. */ if (matches == NULL) matches = cwd_tab_completion(tmp, &num_matches); /* Don't leak memory */ free(matchbuf); #ifdef DEBUG fprintf(stderr, "%d matches found...\n", num_matches); #endif /* Did we find exactly one match? */ switch (num_matches) { case 0: blank_edit(); wrefresh(edit); break; case 1: buf = nrealloc(buf, strlen(buf) + strlen(matches[0]) + 1); if (buf[0] != '\0' && strstr(buf, "/") != NULL) { for (tmp = buf + strlen(buf); *tmp != '/' && tmp != buf; tmp--); tmp++; } else tmp = buf; if (!strcmp(tmp, matches[0])) is_dir = append_slash_if_dir(buf, lastwastab, newplace); if (is_dir != 0) break; copyto = tmp; for (pos = 0; *tmp == matches[0][pos] && pos <= strlen(matches[0]); pos++) tmp++; /* write out the matched name */ strncpy(copyto, matches[0], strlen(matches[0]) + 1); *newplace += strlen(matches[0]) - pos; /* if an exact match is typed in and Tab is pressed, *newplace will now be negative; in that case, make it zero, so that the cursor will stay where it is instead of moving backward */ if (*newplace < 0) *newplace = 0; /* Is it a directory? */ append_slash_if_dir(buf, lastwastab, newplace); break; default: /* Check to see if all matches share a beginning, and, if so, tack it onto buf and then beep */ if (buf[0] != '\0' && strstr(buf, "/") != NULL) { for (tmp = buf + strlen(buf); *tmp != '/' && tmp != buf; tmp--); tmp++; } else tmp = buf; for (pos = 0; *tmp == matches[0][pos] && *tmp != '\0' && pos <= strlen(matches[0]); pos++) tmp++; while (1) { match_matches = 0; for (i = 0; i < num_matches; i++) { if (matches[i][pos] == 0) break; else if (matches[i][pos] == matches[0][pos]) match_matches++; } if (match_matches == num_matches && (i == num_matches || matches[i] != 0)) { /* All the matches have the same character at pos+1, so paste it into buf... */ buf = nrealloc(buf, strlen(buf) + 2); strncat(buf, matches[0] + pos, 1); *newplace += 1; pos++; } else { beep(); break; } } break; } } else { /* Ok -- the last char was a TAB. Since they * just hit TAB again, print a list of all the * available choices... */ if (matches != NULL && num_matches > 1) { /* Blank the edit window, and print the matches out there */ blank_edit(); wmove(edit, 0, 0); editline = 0; /* Figure out the length of the longest filename */ for (i = 0; i < num_matches; i++) if (strlen(matches[i]) > longestname) longestname = strlen(matches[i]); if (longestname > COLS - 1) longestname = COLS - 1; foo = charalloc(longestname + 5); /* Print the list of matches */ for (i = 0, col = 0; i < num_matches; i++) { /* make each filename shown be the same length as the longest filename, with two spaces at the end */ snprintf(foo, longestname + 1, matches[i]); while (strlen(foo) < longestname) strcat(foo, " "); strcat(foo, " "); /* Disable el cursor */ curs_set(0); /* now, put the match on the screen */ waddnstr(edit, foo, strlen(foo)); col += strlen(foo); /* And if the next match isn't going to fit on the line, move to the next one */ if (col > (COLS - longestname) && matches[i + 1] != NULL) { editline++; wmove(edit, editline, 0); if (editline == editwinrows - 1) { waddstr(edit, _("(more)")); break; } col = 0; } } free(foo); wrefresh(edit); *list = 1; } else beep(); } /* Only refresh the edit window if we don't have a list of filename matches on it */ if (*list == 0) edit_refresh(); curs_set(1); return buf; } #endif /* !DISABLE_TABCOMP */ #ifndef DISABLE_BROWSER /* Return the stat of the file pointed to by path */ struct stat filestat(const char *path) { struct stat st; stat(path, &st); return st; } /* Our sort routine for file listings - sort directories before * files, and then alphabetically. */ int diralphasort(const void *va, const void *vb) { struct stat fileinfo; const char *a = *(char *const *)va, *b = *(char *const *)vb; int aisdir = stat(a, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode); int bisdir = stat(b, &fileinfo) != -1 && S_ISDIR(fileinfo.st_mode); if (aisdir != 0 && bisdir == 0) return -1; if (aisdir == 0 && bisdir != 0) return 1; return strcasecmp(a, b); } /* Free our malloc()ed memory */ void free_charptrarray(char **array, int len) { for (; len > 0; len--) free(array[len - 1]); free(array); } /* Only print the last part of a path; isn't there a shell * command for this? */ const char *tail(const char *foo) { const char *tmp = foo + strlen(foo); while (*tmp != '/' && tmp != foo) tmp--; if (*tmp == '/') tmp++; return tmp; } /* Strip one dir from the end of a string. */ void striponedir(char *foo) { char *tmp; /* Don't strip the root dir */ if (!strcmp(foo, "/")) return; tmp = foo + strlen(foo); if (*tmp == '/') tmp--; while (*tmp != '/' && tmp != foo) tmp--; if (tmp != foo) *tmp = '\0'; else { /* SPK may need to make a 'default' path here */ if (*tmp != '/') *tmp = '.'; *(tmp + 1) = '\0'; } } /* Initialize the browser code, including the list of files in *path */ char **browser_init(const char *path, int *longest, int *numents) { DIR *dir; struct dirent *next; char **filelist; int i = 0; size_t path_len; dir = opendir(path); if (dir == NULL) return NULL; *numents = 0; while ((next = readdir(dir)) != NULL) { if (!strcmp(next->d_name, ".")) continue; (*numents)++; if (strlen(next->d_name) > *longest) *longest = strlen(next->d_name); } rewinddir(dir); *longest += 10; filelist = nmalloc(*numents * sizeof (char *)); if (!strcmp(path, "/")) path = ""; path_len = strlen(path); while ((next = readdir(dir)) != NULL) { if (!strcmp(next->d_name, ".")) continue; filelist[i] = charalloc(strlen(next->d_name) + path_len + 2); sprintf(filelist[i], "%s/%s", path, next->d_name); i++; } if (*longest > COLS - 1) *longest = COLS - 1; return filelist; } /* Our browser function. inpath is the path to start browsing from */ char *do_browser(const char *inpath) { struct stat st; char *foo, *retval = NULL; static char *path = NULL; int numents = 0, i = 0, j = 0, kbinput = 0, longest = 0, abort = 0; int col = 0, selected = 0, editline = 0, width = 0, filecols = 0; int lineno = 0, kb; char **filelist = (char **)NULL; #if !defined(DISABLE_MOUSE) && defined(NCURSES_MOUSE_VERSION) MEVENT mevent; #endif assert(inpath != NULL); /* If path isn't the same as inpath, we are being passed a new dir as an arg. We free it here so it will be copied from inpath below */ if (path != NULL && strcmp(path, inpath)) { free(path); path = NULL; } /* if path doesn't exist, make it so */ if (path == NULL) path = mallocstrcpy(NULL, inpath); filelist = browser_init(path, &longest, &numents); foo = charalloc(longest + 8); /* Sort the list by directory first, then alphabetically */ qsort(filelist, numents, sizeof(char *), diralphasort); kb = keypad_on(edit, 1); titlebar(path); bottombars(browser_list); curs_set(0); wmove(edit, 0, 0); i = 0; width = 0; filecols = 0; /* Loop invariant: Microsoft sucks. */ do { DIR *test_dir; blank_statusbar_refresh(); #if !defined(DISABLE_HELP) || (!defined(DISABLE_MOUSE) && defined(NCURSES_MOUSE_VERSION)) currshortcut = browser_list; #endif editline = 0; col = 0; /* Compute line number we're on now, so we don't divide by zero later */ lineno = selected; if (width != 0) lineno /= width; switch (kbinput) { #if !defined(DISABLE_MOUSE) && defined(NCURSES_MOUSE_VERSION) case KEY_MOUSE: if (getmouse(&mevent) == ERR) return retval; /* If they clicked in the edit window, they 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. */ selected = (lineno / editwinrows) * editwinrows * 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 where we did last time, select this name! */ if (selected > numents - 1) selected = numents - 1; else if (selectedbackup == selected) ungetch('s'); /* Unget the 'select' key */ } else /* Must be clicking a shortcut */ do_mouse(); break; #endif case NANO_UP_KEY: case KEY_UP: case 'u': if (selected - width >= 0) selected -= width; break; case NANO_BACK_KEY: case KEY_LEFT: case NANO_BACKSPACE_KEY: case 127: case 'l': if (selected > 0) selected--; break; case KEY_DOWN: case NANO_DOWN_KEY: case 'd': if (selected + width <= numents - 1) selected += width; break; case KEY_RIGHT: case NANO_FORWARD_KEY: case 'r': if (selected < numents - 1) selected++; break; case NANO_PREVPAGE_KEY: case NANO_PREVPAGE_FKEY: case KEY_PPAGE: case '-': if (selected >= (editwinrows + lineno % editwinrows) * width) selected -= (editwinrows + lineno % editwinrows) * width; else selected = 0; break; case NANO_NEXTPAGE_KEY: case NANO_NEXTPAGE_FKEY: case KEY_NPAGE: case ' ': selected += (editwinrows - lineno % editwinrows) * width; if (selected >= numents) selected = numents - 1; break; case NANO_HELP_KEY: case NANO_HELP_FKEY: do_help(); break; case KEY_ENTER: case NANO_ENTER_KEY: case 's': /* More Pico compatibility */ case 'S': /* You can't cd up from / */ if (!strcmp(filelist[selected], "/..") && !strcmp(path, "/")) { statusbar(_("Can't move up a directory")); break; } path = mallocstrcpy(path, filelist[selected]); #ifndef DISABLE_OPERATINGDIR /* Note: The case of the user's being completely outside the operating directory is handled elsewhere, before this point */ if (operating_dir != NULL) { if (check_operating_dir(path, 0)) { statusbar(_("Can't visit parent in restricted mode")); beep(); break; } } #endif /* SPK for '.' path, get the current path via getcwd */ if (!strcmp(path, "./..")) { free(path); #ifdef PATH_MAX path = getcwd(NULL, PATH_MAX + 1); #else path = getcwd(NULL, 0); #endif striponedir(path); align(&path); free_charptrarray(filelist, numents); free(foo); return do_browser(path); } st = filestat(path); if (S_ISDIR(st.st_mode)) { if ((test_dir = opendir(path)) == NULL) { /* We can't open this dir for some reason. Complain */ statusbar(_("Can't open \"%s\": %s"), path, strerror(errno)); striponedir(path); align(&path); break; } closedir(test_dir); if (!strcmp("..", tail(path))) { /* They want to go up a level, so strip off .. and the current dir */ striponedir(path); striponedir(path); align(&path); } /* Start over again with the new path value */ free_charptrarray(filelist, numents); free(foo); return do_browser(path); } else { retval = mallocstrcpy(retval, path); abort = 1; } break; /* Goto a specific directory */ case 'g': /* Pico compatibility */ case 'G': case NANO_GOTO_KEY: curs_set(1); j = statusq(0, gotodir_list, "", #ifndef NANO_SMALL 0, #endif _("Goto Directory")); bottombars(browser_list); curs_set(0); #ifndef DISABLE_OPERATINGDIR if (operating_dir != NULL) { if (check_operating_dir(answer, 0)) { statusbar(_("Can't go outside of %s in restricted mode"), operating_dir); break; } } #endif if (j < 0) { statusbar(_("Goto Cancelled")); break; } if (answer[0] != '/') { char *saveanswer = mallocstrcpy(NULL, answer); answer = nrealloc(answer, strlen(path) + strlen(saveanswer) + 2); sprintf(answer, "%s/%s", path, saveanswer); free(saveanswer); } if ((test_dir = opendir(answer)) == NULL) { /* We can't open this dir for some reason. Complain */ statusbar(_("Can't open \"%s\": %s"), answer, strerror(errno)); break; } closedir(test_dir); /* Start over again with the new path value */ path = mallocstrcpy(path, answer); return do_browser(path); /* Stuff we want to abort the browser */ case 'q': case 'Q': case 'e': /* Pico compatibility, yeech */ case 'E': case NANO_CANCEL_KEY: case NANO_EXIT_FKEY: abort = 1; break; } if (abort) break; blank_edit(); if (width != 0) i = width * editwinrows * ((selected / width) / editwinrows); else i = 0; wmove(edit, 0, 0); for (j = i; j < numents && editline <= editwinrows - 1; j++) { filecols++; strncpy(foo, tail(filelist[j]), strlen(tail(filelist[j])) + 1); while (strlen(foo) < longest) strcat(foo, " "); col += strlen(foo); /* Put file info in the string also */ /* We use lstat here to detect links; then, if we find a symlink, we examine it via stat() to see if it is a directory or just a file symlink */ lstat(filelist[j], &st); if (S_ISDIR(st.st_mode)) strcpy(foo + longest - 5, "(dir)"); else { if (S_ISLNK(st.st_mode)) { /* Aha! It's a symlink! Now, is it a dir? If so, mark it as such */ st = filestat(filelist[j]); if (S_ISDIR(st.st_mode)) strcpy(foo + longest - 5, "(dir)"); else strcpy(foo + longest - 2, "--"); } else if (st.st_size < (1 << 10)) /* less than 1 K */ sprintf(foo + longest - 7, "%4d B", (int) st.st_size); else if (st.st_size >= (1 << 30)) /* at least 1 gig */ sprintf(foo + longest - 7, "%4d GB", (int) st.st_size >> 30); else if (st.st_size >= (1 << 20)) /* at least 1 meg */ sprintf(foo + longest - 7, "%4d MB", (int) st.st_size >> 20); else /* It's more than 1 k and less than a meg */ sprintf(foo + longest - 7, "%4d KB", (int) st.st_size >> 10); } /* Highlight the currently selected file/dir */ if (j == selected) wattron(edit, A_REVERSE); waddstr(edit, foo); if (j == selected) wattroff(edit, A_REVERSE); /* And add some space between the cols */ waddstr(edit, " "); col += 2; /* And if the next entry isn't going to fit on the line, move to the next one */ if (col > COLS - longest) { editline++; wmove(edit, editline, 0); col = 0; if (width == 0) width = filecols; } } wrefresh(edit); } while ((kbinput = wgetch(edit)) != NANO_EXIT_KEY); curs_set(1); blank_edit(); titlebar(NULL); edit_refresh(); kb = keypad_on(edit, kb); /* cleanup */ free_charptrarray(filelist, numents); free(foo); return retval; } /* Browser front end, checks to see if inpath has a dir in it and, if so, starts do_browser from there, else from the current dir */ char *do_browse_from(const char *inpath) { struct stat st; char *tmp; char *bob; /* If there's no / in the string, we may as well start from . */ if (inpath == NULL || strchr(inpath, '/') == NULL) { #ifdef PATH_MAX char *from = getcwd(NULL, PATH_MAX + 1); #else char *from = getcwd(NULL, 0); #endif bob = do_browser(from != NULL ? from : "./"); free(from); return bob; } /* If the string is a directory, pass do_browser() that */ st = filestat(inpath); if (S_ISDIR(st.st_mode)) return do_browser(inpath); tmp = mallocstrcpy(NULL, inpath); /* Okay, there's a dir in there, but not at the end of the string... try stripping it off */ striponedir(tmp); align(&tmp); bob = do_browser(tmp); free(tmp); return bob; } #endif /* !DISABLE_BROWSER */