+++ /dev/null
-/* $OpenBSD: fileio.c,v 1.94 2012/07/10 06:28:12 lum Exp $ */
-
-/* This file is in the public domain. */
-
-/*
- * POSIX fileio.c
- */
-#include "def.h"
-
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <sys/time.h>
-#include <sys/resource.h>
-#include <sys/wait.h>
-
-#include <fcntl.h>
-#include <limits.h>
-#include <dirent.h>
-#include <pwd.h>
-#include <string.h>
-#include <unistd.h>
-
-#include "kbd.h"
-#include "pathnames.h"
-
-static char *bkuplocation(const char *);
-static int bkupleavetmp(const char *);
-char *expandtilde(const char *);
-
-static char *bkupdir;
-static int leavetmp = 0; /* 1 = leave any '~' files in tmp dir */
-
-/*
- * Open a file for reading.
- */
-int
-ffropen(FILE ** ffp, const char *fn, struct buffer *bp)
-{
- if ((*ffp = fopen(fn, "r")) == NULL) {
- if (errno == ENOENT)
- return (FIOFNF);
- return (FIOERR);
- }
-
- /* If 'fn' is a directory open it with dired. */
- if (fisdir(fn) == TRUE)
- return (FIODIR);
-
- ffstat(*ffp, bp);
-
- return (FIOSUC);
-}
-
-/*
- * Update stat/dirty info
- */
-void
-ffstat(FILE *ffp, struct buffer *bp)
-{
- struct stat sb;
-
- if (bp && fstat(fileno(ffp), &sb) == 0) {
- /* set highorder bit to make sure this isn't all zero */
- bp->b_fi.fi_mode = sb.st_mode | 0x8000;
- bp->b_fi.fi_uid = sb.st_uid;
- bp->b_fi.fi_gid = sb.st_gid;
- bp->b_fi.fi_mtime = sb.st_mtimespec;
- /* Clear the ignore flag */
- bp->b_flag &= ~(BFIGNDIRTY | BFDIRTY);
- }
-}
-
-/*
- * Update the status/dirty info. If there is an error,
- * there's not a lot we can do.
- */
-int
-fupdstat(struct buffer *bp)
-{
- FILE *ffp;
-
- if ((ffp = fopen(bp->b_fname, "r")) == NULL) {
- if (errno == ENOENT)
- return (FIOFNF);
- return (FIOERR);
- }
- ffstat(ffp, bp);
- (void)ffclose(ffp, bp);
- return (FIOSUC);
-}
-
-/*
- * Open a file for writing.
- */
-int
-ffwopen(FILE ** ffp, const char *fn, struct buffer *bp)
-{
- int fd;
- mode_t fmode = DEFFILEMODE;
-
- if (bp && bp->b_fi.fi_mode)
- fmode = bp->b_fi.fi_mode & 07777;
-
- fd = open(fn, O_RDWR | O_CREAT | O_TRUNC, fmode);
- if (fd == -1) {
- ffp = NULL;
- ewprintf("Cannot open file for writing : %s", strerror(errno));
- return (FIOERR);
- }
-
- if ((*ffp = fdopen(fd, "w")) == NULL) {
- ewprintf("Cannot open file for writing : %s", strerror(errno));
- close(fd);
- return (FIOERR);
- }
-
- /*
- * If we have file information, use it. We don't bother to check for
- * errors, because there's no a lot we can do about it. Certainly
- * trying to change ownership will fail if we aren't root. That's
- * probably OK. If we don't have info, no need to get it, since any
- * future writes will do the same thing.
- */
- if (bp && bp->b_fi.fi_mode) {
- fchmod(fd, bp->b_fi.fi_mode & 07777);
- fchown(fd, bp->b_fi.fi_uid, bp->b_fi.fi_gid);
- }
- return (FIOSUC);
-}
-
-/*
- * Close a file.
- */
-/* ARGSUSED */
-int
-ffclose(FILE *ffp, struct buffer *bp)
-{
- if (fclose(ffp) == 0)
- return (FIOSUC);
- return (FIOERR);
-}
-
-/*
- * Write a buffer to the already opened file. bp points to the
- * buffer. Return the status.
- */
-int
-ffputbuf(FILE *ffp, struct buffer *bp)
-{
- struct line *lp, *lpend;
-
- lpend = bp->b_headp;
- for (lp = lforw(lpend); lp != lpend; lp = lforw(lp)) {
- if (fwrite(ltext(lp), 1, llength(lp), ffp) != llength(lp)) {
- ewprintf("Write I/O error");
- return (FIOERR);
- }
- if (lforw(lp) != lpend) /* no implied \n on last line */
- putc('\n', ffp);
- }
- /*
- * XXX should be variable controlled (once we have variables)
- */
- if (llength(lback(lpend)) != 0) {
- if (eyorn("No newline at end of file, add one") == TRUE) {
- lnewline_at(lback(lpend), llength(lback(lpend)));
- putc('\n', ffp);
- }
- }
- return (FIOSUC);
-}
-
-/*
- * Read a line from a file, and store the bytes
- * in the supplied buffer. Stop on end of file or end of
- * line. When FIOEOF is returned, there is a valid line
- * of data without the normally implied \n.
- * If the line length exceeds nbuf, FIOLONG is returned.
- */
-int
-ffgetline(FILE *ffp, char *buf, int nbuf, int *nbytes)
-{
- int c, i;
-
- i = 0;
- while ((c = getc(ffp)) != EOF && c != '\n') {
- buf[i++] = c;
- if (i >= nbuf)
- return (FIOLONG);
- }
- if (c == EOF && ferror(ffp) != FALSE) {
- ewprintf("File read error");
- return (FIOERR);
- }
- *nbytes = i;
- return (c == EOF ? FIOEOF : FIOSUC);
-}
-
-/*
- * Make a backup copy of "fname". On Unix the backup has the same
- * name as the original file, with a "~" on the end; this seems to
- * be newest of the new-speak. The error handling is all in "file.c".
- * We do a copy instead of a rename since otherwise another process
- * with an open fd will get the backup, not the new file. This is
- * a problem when using mg with things like crontab and vipw.
- */
-int
-fbackupfile(const char *fn)
-{
- struct stat sb;
- int from, to, serrno;
- ssize_t nread;
- char buf[BUFSIZ];
- char *nname, *tname, *bkpth;
-
- if (stat(fn, &sb) == -1) {
- ewprintf("Can't stat %s : %s", fn, strerror(errno));
- return (FALSE);
- }
-
- if ((bkpth = bkuplocation(fn)) == NULL)
- return (FALSE);
-
- if (asprintf(&nname, "%s~", bkpth) == -1) {
- ewprintf("Can't allocate backup file name : %s", strerror(errno));
- free(bkpth);
- return (ABORT);
- }
- if (asprintf(&tname, "%s.XXXXXXXXXX", bkpth) == -1) {
- ewprintf("Can't allocate temp file name : %s", strerror(errno));
- free(bkpth);
- free(nname);
- return (ABORT);
- }
- free(bkpth);
-
- if ((from = open(fn, O_RDONLY)) == -1) {
- free(nname);
- free(tname);
- return (FALSE);
- }
- to = mkstemp(tname);
- if (to == -1) {
- serrno = errno;
- close(from);
- free(nname);
- free(tname);
- errno = serrno;
- return (FALSE);
- }
- while ((nread = read(from, buf, sizeof(buf))) > 0) {
- if (write(to, buf, (size_t)nread) != nread) {
- nread = -1;
- break;
- }
- }
- serrno = errno;
- (void) fchmod(to, (sb.st_mode & 0777));
- close(from);
- close(to);
- if (nread == -1) {
- if (unlink(tname) == -1)
- ewprintf("Can't unlink temp : %s", strerror(errno));
- } else {
- if (rename(tname, nname) == -1) {
- ewprintf("Can't rename temp : %s", strerror(errno));
- (void) unlink(tname);
- nread = -1;
- }
- }
- free(nname);
- free(tname);
- errno = serrno;
-
- return (nread == -1 ? FALSE : TRUE);
-}
-
-/*
- * Convert "fn" to a canonicalized absolute filename, replacing
- * a leading ~/ with the user's home dir, following symlinks, and
- * and remove all occurrences of /./ and /../
- */
-char *
-adjustname(const char *fn, int slashslash)
-{
- static char fnb[MAXPATHLEN];
- const char *cp, *ep = NULL;
- char *path;
-
- if (slashslash == TRUE) {
- cp = fn + strlen(fn) - 1;
- for (; cp >= fn; cp--) {
- if (ep && (*cp == '/')) {
- fn = ep;
- break;
- }
- if (*cp == '/' || *cp == '~')
- ep = cp;
- else
- ep = NULL;
- }
- }
- if ((path = expandtilde(fn)) == NULL)
- return (NULL);
-
- if (realpath(path, fnb) == NULL)
- (void)strlcpy(fnb, path, sizeof(fnb));
-
- free(path);
- return (fnb);
-}
-
-/*
- * Find a startup file for the user and return its name. As a service
- * to other pieces of code that may want to find a startup file (like
- * the terminal driver in particular), accepts a suffix to be appended
- * to the startup file name.
- */
-char *
-startupfile(char *suffix)
-{
- static char file[NFILEN];
- char *home;
- int ret;
-
- if ((home = getenv("HOME")) == NULL || *home == '\0')
- goto nohome;
-
- if (suffix == NULL) {
- ret = snprintf(file, sizeof(file), _PATH_MG_STARTUP, home);
- if (ret < 0 || ret >= sizeof(file))
- return (NULL);
- } else {
- ret = snprintf(file, sizeof(file), _PATH_MG_TERM, home, suffix);
- if (ret < 0 || ret >= sizeof(file))
- return (NULL);
- }
-
- if (access(file, R_OK) == 0)
- return (file);
-nohome:
-#ifdef STARTUPFILE
- if (suffix == NULL) {
- ret = snprintf(file, sizeof(file), "%s", STARTUPFILE);
- if (ret < 0 || ret >= sizeof(file))
- return (NULL);
- } else {
- ret = snprintf(file, sizeof(file), "%s%s", STARTUPFILE,
- suffix);
- if (ret < 0 || ret >= sizeof(file))
- return (NULL);
- }
-
- if (access(file, R_OK) == 0)
- return (file);
-#endif /* STARTUPFILE */
- return (NULL);
-}
-
-int
-copy(char *frname, char *toname)
-{
- int ifd, ofd;
- char buf[BUFSIZ];
- mode_t fmode = DEFFILEMODE; /* XXX?? */
- struct stat orig;
- ssize_t sr;
-
- if ((ifd = open(frname, O_RDONLY)) == -1)
- return (FALSE);
- if (fstat(ifd, &orig) == -1) {
- ewprintf("fstat: %s", strerror(errno));
- close(ifd);
- return (FALSE);
- }
-
- if ((ofd = open(toname, O_WRONLY|O_CREAT|O_TRUNC, fmode)) == -1) {
- close(ifd);
- return (FALSE);
- }
- while ((sr = read(ifd, buf, sizeof(buf))) > 0) {
- if (write(ofd, buf, (size_t)sr) != sr) {
- ewprintf("write error : %s", strerror(errno));
- break;
- }
- }
- if (fchmod(ofd, orig.st_mode) == -1)
- ewprintf("Cannot set original mode : %s", strerror(errno));
-
- if (sr == -1) {
- ewprintf("Read error : %s", strerror(errno));
- close(ifd);
- close(ofd);
- return (FALSE);
- }
- /*
- * It is "normal" for this to fail since we can't guarantee that
- * we will be running as root.
- */
- if (fchown(ofd, orig.st_uid, orig.st_gid) && errno != EPERM)
- ewprintf("Cannot set owner : %s", strerror(errno));
-
- (void) close(ifd);
- (void) close(ofd);
-
- return (TRUE);
-}
-
-/*
- * return list of file names that match the name in buf.
- */
-struct list *
-make_file_list(char *buf)
-{
- char *dir, *file, *cp;
- size_t len, preflen;
- int ret;
- DIR *dirp;
- struct dirent *dent;
- struct list *last, *current;
- char fl_name[NFILEN + 2];
- char prefixx[NFILEN + 1];
-
- /*
- * We need three different strings:
-
- * dir - the name of the directory containing what the user typed.
- * Must be a real unix file name, e.g. no ~user, etc..
- * Must not end in /.
- * prefix - the portion of what the user typed that is before the
- * names we are going to find in the directory. Must have a
- * trailing / if the user typed it.
- * names from the directory - We open dir, and return prefix
- * concatenated with names.
- */
-
- /* first we get a directory name we can look up */
- /*
- * Names ending in . are potentially odd, because adjustname will
- * treat foo/bar/.. as a foo/, whereas we are
- * interested in names starting with ..
- */
- len = strlen(buf);
- if (len && buf[len - 1] == '.') {
- buf[len - 1] = 'x';
- dir = adjustname(buf, TRUE);
- buf[len - 1] = '.';
- } else
- dir = adjustname(buf, TRUE);
- if (dir == NULL)
- return (NULL);
- /*
- * If the user typed a trailing / or the empty string
- * he wants us to use his file spec as a directory name.
- */
- if (len && buf[len - 1] != '/') {
- file = strrchr(dir, '/');
- if (file) {
- *file = '\0';
- if (*dir == '\0')
- dir = "/";
- } else
- return (NULL);
- }
- /* Now we get the prefix of the name the user typed. */
- if (strlcpy(prefixx, buf, sizeof(prefixx)) >= sizeof(prefixx))
- return (NULL);
- cp = strrchr(prefixx, '/');
- if (cp == NULL)
- prefixx[0] = '\0';
- else
- cp[1] = '\0';
-
- preflen = strlen(prefixx);
- /* cp is the tail of buf that really needs to be compared. */
- cp = buf + preflen;
- len = strlen(cp);
-
- /*
- * Now make sure that file names will fit in the buffers allocated.
- * SV files are fairly short. For BSD, something more general would
- * be required.
- */
- if (preflen > NFILEN - MAXNAMLEN)
- return (NULL);
-
- /* loop over the specified directory, making up the list of files */
-
- /*
- * Note that it is worth our time to filter out names that don't
- * match, even though our caller is going to do so again, and to
- * avoid doing the stat if completion is being done, because stat'ing
- * every file in the directory is relatively expensive.
- */
-
- dirp = opendir(dir);
- if (dirp == NULL)
- return (NULL);
- last = NULL;
-
- while ((dent = readdir(dirp)) != NULL) {
- int isdir;
- if (strncmp(cp, dent->d_name, len) != 0)
- continue;
- isdir = 0;
- if (dent->d_type == DT_DIR) {
- isdir = 1;
- } else if (dent->d_type == DT_LNK ||
- dent->d_type == DT_UNKNOWN) {
- struct stat statbuf;
- char statname[NFILEN + 2];
-
- statbuf.st_mode = 0;
- ret = snprintf(statname, sizeof(statname), "%s/%s",
- dir, dent->d_name);
- if (ret < 0 || ret > sizeof(statname) - 1)
- continue;
- if (stat(statname, &statbuf) < 0)
- continue;
- if (S_ISDIR(statbuf.st_mode))
- isdir = 1;
- }
-
- if ((current = malloc(sizeof(struct list))) == NULL) {
- free_file_list(last);
- closedir(dirp);
- return (NULL);
- }
- ret = snprintf(fl_name, sizeof(fl_name),
- "%s%s%s", prefixx, dent->d_name, isdir ? "/" : "");
- if (ret < 0 || ret >= sizeof(fl_name)) {
- free(current);
- continue;
- }
- current->l_next = last;
- current->l_name = strdup(fl_name);
- last = current;
- }
- closedir(dirp);
-
- return (last);
-}
-
-/*
- * Test if a supplied filename refers to a directory
- * Returns ABORT on error, TRUE if directory. FALSE otherwise
- */
-int
-fisdir(const char *fname)
-{
- struct stat statbuf;
-
- if (stat(fname, &statbuf) != 0)
- return (ABORT);
-
- if (S_ISDIR(statbuf.st_mode))
- return (TRUE);
-
- return (FALSE);
-}
-
-/*
- * Check the mtime of the supplied filename.
- * Return TRUE if last mtime matches, FALSE if not,
- * If the stat fails, return TRUE and try the save anyway
- */
-int
-fchecktime(struct buffer *bp)
-{
- struct stat sb;
-
- if (stat(bp->b_fname, &sb) == -1)
- return (TRUE);
-
- if (bp->b_fi.fi_mtime.tv_sec != sb.st_mtimespec.tv_sec ||
- bp->b_fi.fi_mtime.tv_nsec != sb.st_mtimespec.tv_nsec)
- return (FALSE);
-
- return (TRUE);
-
-}
-
-/*
- * Location of backup file. This function creates the correct path.
- */
-static char *
-bkuplocation(const char *fn)
-{
- struct stat sb;
- char *ret;
-
- if (bkupdir != NULL && (stat(bkupdir, &sb) == 0) &&
- S_ISDIR(sb.st_mode) && !bkupleavetmp(fn)) {
- char fname[NFILEN];
- const char *c;
- int i = 0, len;
-
- c = fn;
- len = strlen(bkupdir);
-
- while (*c != '\0') {
- /* Make sure we don't go over combined:
- * strlen(bkupdir + '/' + fname + '\0')
- */
- if (i >= NFILEN - len - 1)
- return (NULL);
- if (*c == '/') {
- fname[i] = '!';
- } else if (*c == '!') {
- if (i >= NFILEN - len - 2)
- return (NULL);
- fname[i++] = '!';
- fname[i] = '!';
- } else
- fname[i] = *c;
- i++;
- c++;
- }
- fname[i] = '\0';
- if (asprintf(&ret, "%s/%s", bkupdir, fname) == -1)
- return (NULL);
-
- } else if ((ret = strndup(fn, NFILEN)) == NULL)
- return (NULL);
-
- return (ret);
-}
-
-int
-backuptohomedir(int f, int n)
-{
- const char *c = _PATH_MG_DIR;
- char *p;
-
- if (bkupdir == NULL) {
- p = adjustname(c, TRUE);
- bkupdir = strndup(p, NFILEN);
- if (bkupdir == NULL)
- return(FALSE);
-
- if (mkdir(bkupdir, 0700) == -1 && errno != EEXIST) {
- free(bkupdir);
- bkupdir = NULL;
- }
- } else {
- free(bkupdir);
- bkupdir = NULL;
- }
-
- return (TRUE);
-}
-
-/*
- * For applications that use mg as the editor and have a desire to keep
- * '~' files in the TMPDIR, toggle the location: /tmp | ~/.mg.d
- */
-int
-toggleleavetmp(int f, int n)
-{
- leavetmp = !leavetmp;
-
- return (TRUE);
-}
-
-/*
- * Returns TRUE if fn is located in the temp directory and we want to save
- * those backups there.
- */
-int
-bkupleavetmp(const char *fn)
-{
- char *tmpdir, *tmp = NULL;
-
- if (!leavetmp)
- return(FALSE);
-
- if((tmpdir = getenv("TMPDIR")) != NULL && *tmpdir != '\0') {
- tmp = strstr(fn, tmpdir);
- if (tmp == fn)
- return (TRUE);
-
- return (FALSE);
- }
-
- tmp = strstr(fn, "/tmp");
- if (tmp == fn)
- return (TRUE);
-
- return (FALSE);
-}
-
-/*
- * Expand file names beginning with '~' if appropriate:
- * 1, if ./~fn exists, continue without expanding tilde.
- * 2, else, if username 'fn' exists, expand tilde with home directory path.
- * 3, otherwise, continue and create new buffer called ~fn.
- */
-char *
-expandtilde(const char *fn)
-{
- struct passwd *pw;
- struct stat statbuf;
- const char *cp;
- char user[LOGIN_NAME_MAX], path[NFILEN];
- char *un, *ret;
- size_t ulen, plen;
-
- path[0] = '\0';
-
- if (fn[0] != '~' || stat(fn, &statbuf) == 0) {
- if ((ret = strndup(fn, NFILEN)) == NULL)
- return (NULL);
- return(ret);
- }
- cp = strchr(fn, '/');
- if (cp == NULL)
- cp = fn + strlen(fn); /* point to the NUL byte */
- ulen = cp - &fn[1];
- if (ulen >= sizeof(user)) {
- if ((ret = strndup(fn, NFILEN)) == NULL)
- return (NULL);
- return(ret);
- }
- if (ulen == 0) { /* ~/ or ~ */
- if ((un = getlogin()) != NULL)
- (void)strlcpy(user, un, sizeof(user));
- else
- user[0] = '\0';
- } else { /* ~user/ or ~user */
- memcpy(user, &fn[1], ulen);
- user[ulen] = '\0';
- }
- pw = getpwnam(user);
- if (pw != NULL) {
- plen = strlcpy(path, pw->pw_dir, sizeof(path));
- if (plen == 0 || path[plen - 1] != '/') {
- if (strlcat(path, "/", sizeof(path)) >= sizeof(path)) {
- ewprintf("Path too long");
- return (NULL);
- }
- }
- fn = cp;
- if (*fn == '/')
- fn++;
- }
- if (strlcat(path, fn, sizeof(path)) >= sizeof(path)) {
- ewprintf("Path too long");
- return (NULL);
- }
- if ((ret = strndup(path, NFILEN)) == NULL)
- return (NULL);
-
- return (ret);
-}