X-Git-Url: https://pd.if.org/git/?p=pd_readline;a=blobdiff_plain;f=mg%2Ffileio.c;fp=mg%2Ffileio.c;h=0b63be867fc50b586531fc9854204e07ca3a87a5;hp=0000000000000000000000000000000000000000;hb=a9843085ec916c175bd245a8398f30e6cc03f984;hpb=26fe4e09c6c3c250334fdeed60ce3061febecde2 diff --git a/mg/fileio.c b/mg/fileio.c new file mode 100644 index 0000000..0b63be8 --- /dev/null +++ b/mg/fileio.c @@ -0,0 +1,754 @@ +/* $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 +#include +#include +#include +#include + +#include +#include +#include +#include +#include +#include + +#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); +}