/*

 * This code is in the public domain; do with it what you wish.

 *

 * Copyright (C) 2012 Sami Kerola <kerolasa@iki.fi>

 * Copyright (C) 2012-2024 Karel Zak <kzak@redhat.com>

 */

#include <stdio.h>

#include <stdlib.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <unistd.h>

#include <sys/time.h>

#include <sys/resource.h>

#include <sys/syscall.h>

#include <string.h>

#include <sys/wait.h>

#include <fcntl.h>

#include <errno.h>

#ifdef HAVE_LINUX_OPENAT2_H

# include <linux/openat2.h>

#endif

#include "c.h"

#include "all-io.h"

#include "fileutils.h"

#include "pathnames.h"

int mkstemp_cloexec(char *template)

{

#ifdef HAVE_MKOSTEMP

  return mkostemp(template, O_RDWR|O_CREAT|O_EXCL|O_CLOEXEC);

#else

  int fd, old_flags, errno_save;

  fd = mkstemp(template);

  if (fd < 0)

    return fd;

  old_flags = fcntl(fd, F_GETFD, 0);

  if (old_flags < 0)

    goto unwind;

  if (fcntl(fd, F_SETFD, old_flags | O_CLOEXEC) < 0)

    goto unwind;

  return fd;

unwind:

  errno_save = errno;

  unlink(template);

  close(fd);

  errno = errno_save;

  return -1;

#endif

}

/* Create open temporary file in safe way.  Please notice that the

 * file permissions are -rw------- by default. */

int xmkstemp(char **tmpname, const char *dir, const char *prefix)

{

  char *localtmp;

  const char *tmpenv;

  mode_t old_mode;

  int fd, rc;

  /* Some use cases must be capable of being moved atomically

   * with rename(2), which is the reason why dir is here.  */

  tmpenv = dir ? dir : getenv("TMPDIR");

  if (!tmpenv)

    tmpenv = _PATH_TMP;

  rc = asprintf(&localtmp, "%s/%s.XXXXXX", tmpenv, prefix);

  if (rc < 0)

    return -1;

  old_mode = umask(077);

  fd = mkstemp_cloexec(localtmp);

  umask(old_mode);

  if (fd == -1) {

    free(localtmp);

    localtmp = NULL;

  }

  *tmpname = localtmp;

  return fd;

}

#ifdef F_DUPFD_CLOEXEC

int dup_fd_cloexec(int oldfd, int lowfd)

#else

int dup_fd_cloexec(int oldfd, int lowfd  __attribute__((__unused__)))

#endif

{

  int fd, flags, errno_save;

#ifdef F_DUPFD_CLOEXEC

  fd = fcntl(oldfd, F_DUPFD_CLOEXEC, lowfd);

  if (fd >= 0)

    return fd;

#endif

  fd = dup(oldfd);

  if (fd < 0)

    return fd;

  flags = fcntl(fd, F_GETFD);

  if (flags < 0)

    goto unwind;

  if (fcntl(fd, F_SETFD, flags | FD_CLOEXEC) < 0)

    goto unwind;

  return fd;

unwind:

  errno_save = errno;

  close(fd);

  errno = errno_save;

  return -1;

}

/*

 * portable getdtablesize()

 */

unsigned int get_fd_tabsize(void)

{

  int m;

#if defined(HAVE_GETDTABLESIZE)

  m = getdtablesize();

#elif defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE)

  struct rlimit rl;

  getrlimit(RLIMIT_NOFILE, &rl);

  m = rl.rlim_cur;

#elif defined(HAVE_SYSCONF) && defined(_SC_OPEN_MAX)

  m = sysconf(_SC_OPEN_MAX);

#else

  m = OPEN_MAX;

#endif

  return m;

}

void ul_close_all_fds(unsigned int first, unsigned int last)

{

  struct dirent *d;

  DIR *dir;

  dir = opendir(_PATH_PROC_FDDIR);

  if (dir) {

    while ((d = xreaddir(dir))) {

      char *end;

      unsigned int fd;

      int dfd;

      errno = 0;

      fd = strtoul(d->d_name, &end, 10);

      if (errno || end == d->d_name || !end || *end)

        continue;

      dfd = dirfd(dir);

      if (dfd < 0)

        continue;

      if ((unsigned int)dfd == fd)

        continue;

      if (fd < first || last < fd)

        continue;

      close(fd);

    }

    closedir(dir);

  } else {

    unsigned fd, tbsz = get_fd_tabsize();

    for (fd = 0; fd < tbsz; fd++) {

      if (first <= fd && fd <= last)

        close(fd);

    }

  }

}

/*

 * Fork, drop permissions, and call oper() and return result.

 */

char *ul_restricted_path_oper(const char *path,

      int (*oper)(const char *path, char **result, void *data),

      void *data)

{

  char *result = NULL;

  int errsv = 0;

  int pipes[2];

  ssize_t len;

  pid_t pid;

  if (!path || !*path)

    return NULL;

  if (pipe(pipes) != 0)

    return NULL;

  /*

   * To accurately assume identity of getuid() we must use setuid()

   * but if we do that, we lose ability to reassume euid of 0, so

   * we fork to do the check to keep euid intact.

   */

  pid = fork();

  switch (pid) {

  case -1:

    close(pipes[0]);

    close(pipes[1]);

    return NULL;     /* fork error */

  case 0:

    close(pipes[0]);    /* close unused end */

    pipes[0] = -1;

    errno = 0;

    if (drop_permissions() != 0)

      result = NULL; /* failed */

    else

      oper(path, &result, data);

    len = result ? (ssize_t) strlen(result) :

              errno ? -errno : -EINVAL;

    /* send length or errno */

    write_all(pipes[1], (char *) &len, sizeof(len));

    if (result)

      write_all(pipes[1], result, len);

    _exit(0);

  default:

    break;

  }

  close(pipes[1]);    /* close unused end */

  pipes[1] = -1;

  /* read size or -errno */

  if (read_all(pipes[0], (char *) &len, sizeof(len)) != sizeof(len))

    goto done;

  if (len < 0) {

    errsv = -len;

    goto done;

  }

  result = malloc(len + 1);

  if (!result) {

    errsv = ENOMEM;

    goto done;

  }

  /* read path */

  if (read_all(pipes[0], result, len) != len) {

    errsv = errno;

    goto done;

  }

  result[len] = '\0';

done:

  if (errsv) {

    free(result);

    result = NULL;

  }

  close(pipes[0]);

  /* We make a best effort to reap child */

  ignore_result( waitpid(pid, NULL, 0) );

  errno = errsv;

  return result;

}

#ifdef TEST_PROGRAM_FILEUTILS

int main(int argc, char *argv[])

{

  if (argc < 2)

    errx(EXIT_FAILURE, "Usage %s --{mkstemp,close-fds,copy-file}", argv[0]);

  if (strcmp(argv[1], "--mkstemp") == 0) {

    FILE *f;

    char *tmpname = NULL;

    f = xfmkstemp(&tmpname, NULL, "test");

    unlink(tmpname);

    free(tmpname);

    fclose(f);

  } else if (strcmp(argv[1], "--close-fds") == 0) {

    ignore_result( dup(STDIN_FILENO) );

    ignore_result( dup(STDIN_FILENO) );

    ignore_result( dup(STDIN_FILENO) );

# ifdef HAVE_CLOSE_RANGE

    if (close_range(STDERR_FILENO + 1, ~0U, 0) < 0)

# endif

      ul_close_all_fds(STDERR_FILENO + 1, ~0U);

  } else if (strcmp(argv[1], "--copy-file") == 0) {

    int ret = ul_copy_file(STDIN_FILENO, STDOUT_FILENO);

    if (ret == UL_COPY_READ_ERROR)

      err(EXIT_FAILURE, "read");

    else if (ret == UL_COPY_WRITE_ERROR)

      err(EXIT_FAILURE, "write");

  }

  return EXIT_SUCCESS;

}

#endif

int ul_mkdir_p(const char *path, mode_t mode)

{

  char *p, *dir;

  int rc = 0;

  if (!path || !*path)

    return -EINVAL;

  dir = p = strdup(path);

  if (!dir)

    return -ENOMEM;

  if (*p == '/')

    p++;

  while (p && *p) {

    char *e = strchr(p, '/');

    if (e)

      *e = '\0';

    if (*p) {

      rc = mkdir(dir, mode);

      if (rc && errno != EEXIST)

        break;

      rc = 0;

    }

    if (!e)

      break;

    *e = '/';

    p = e + 1;

  }

  free(dir);

  return rc;

}

/* returns basename and keeps dirname in the @path, if @path is "/" (root)

 * then returns empty string */

char *stripoff_last_component(char *path)

{

  char *p = path ? strrchr(path, '/') : NULL;

  if (!p)

    return NULL;

  *p = '\0';

  return p + 1;

}

static int copy_file_simple(int from, int to)

{

  ssize_t nr;

  char buf[BUFSIZ];

  while ((nr = read_all(from, buf, sizeof(buf))) > 0)

    if (write_all(to, buf, nr) == -1)

      return UL_COPY_WRITE_ERROR;

  if (nr < 0)

    return UL_COPY_READ_ERROR;

#ifdef HAVE_EXPLICIT_BZERO

  explicit_bzero(buf, sizeof(buf));

#endif

  return 0;

}

/* Copies the contents of a file. Returns -1 on read error, -2 on write error. */

int ul_copy_file(int from, int to)

{

#ifdef HAVE_SENDFILE

  struct stat st;

  ssize_t nw;

  if (fstat(from, &st) == -1)

    return UL_COPY_READ_ERROR;

  if (!S_ISREG(st.st_mode))

    return copy_file_simple(from, to);

  if (sendfile_all(to, from, NULL, st.st_size) < 0)

    return copy_file_simple(from, to);

  /* ensure we either get an EOF or an error */

  while ((nw = sendfile_all(to, from, NULL, 16*1024*1024)) != 0)

    if (nw < 0)

      return copy_file_simple(from, to);

  return 0;

#else

  return copy_file_simple(from, to);

#endif

}

int ul_reopen(int fd, int flags)

{

  ssize_t ssz;

  char buf[PATH_MAX];

  char fdpath[ sizeof(_PATH_PROC_FDDIR) + sizeof(stringify_value(INT_MAX)) ];

  snprintf(fdpath, sizeof(fdpath), _PATH_PROC_FDDIR "/%d", fd);

  ssz = readlink(fdpath, buf, sizeof(buf) - 1);

  if (ssz < 0)

    return -errno;

  assert(ssz > 0);

  buf[ssz] = '\0';

  return open(buf, flags);

}

/* This is a libc-independent version of basename(), which is necessary to

 * maintain functionality across different libc implementations. It was

 * inspired by the behavior and implementation of glibc.

 */

char *ul_basename(char *path)

{

  char *p;

  if (!path || !*path)

    return (char *) "."; /* ugly, static string */

  p = strrchr(path, '/');

  if (!p)

    return path;   /* no '/', return original */

  if (*(p + 1) != '\0')

    return p + 1;   /* begin of the name */

  while (p > path && *(p - 1) == '/')

    --p;     /* remove trailing '/' */

  if (p > path) {

    *p-- = '\0';

    while (p > path && *(p - 1) != '/')

      --p;   /* move to the beginning of the name */

  } else while (*(p + 1) != '\0')

    ++p;

  return p;

}

#ifdef HAVE_OPENAT

/*

 * fopen_at_no_link() - Open a file stream that is not a symbolic/hard link.

 *

 * This function wraps around openat(2), fstat(2), ftruncate(2) and fdopen(3)

 * to create a file stream that is not a symbolic or hard link in a race-free

 * manner.

 *

 * @dir:  dirfd as passed to openat(2), e.g. AT_FDCWD for the calling process

 *    current working directory

 * @filename: name of the target file

 * @flags:  open(2) file creation/status flags, O_NOFOLLOW is implicitly set

 * @perm: open(2) file mode, can be bitwise ORed, these are only relevant

 *    when O_CREAT is set in @flags, otherwise pass as 0.

 * @mode: fopen(3) mode

 *

 * Return: On success, a valid pointer to a file stream is returned.

 *         On failure, NULL is returned and errno is set to indicate the issue.

 */

FILE *fopen_at_no_link(int dir, const char *filename,

                             int flags, mode_t perm, const char *mode)

{

  FILE *fp;

  int fd;

  struct stat st;

  /* We temporarily clear the O_TRUNC bit because we do not want

   * to accidentally truncate the target file if it is a hard link

   * instead of a symbolic one, where the latter is what we are

   * guarding against here. The test for the hard link is done below

   * with fstat()...

   */

  fd = openat(dir, filename, ((flags & ~O_TRUNC) | O_NOFOLLOW), perm);

  if (fd < 0)

    return NULL;

  if (fstat(fd, &st)) {

    close(fd);

    return NULL;

  }

  if (st.st_nlink > 1) {

    close(fd);

    errno = EMLINK;

    return NULL;

  }

  if ((flags & O_TRUNC) && ftruncate(fd, 0)) {

    close(fd);

    return NULL;

  }

  fp = fdopen(fd, mode);

  if (!fp)

    close(fd);

  return fp;

}

#endif /* HAVE_OPENAT */

int ul_open_no_symlinks(const char *path, int flags, mode_t mode)

{

#if defined(SYS_openat2) && defined(RESOLVE_NO_SYMLINKS)

  struct open_how how = {

    .flags = (__u64) flags,

    .mode = (__u64) mode,

    .resolve = RESOLVE_NO_SYMLINKS,

  };

  int fd = syscall(SYS_openat2, AT_FDCWD, path, &how, sizeof(how));

  /* only fall back to O_NOFOLLOW if the syscall is unavailable */

  if (fd >= 0 || errno != ENOSYS)

    return fd;

#endif

  return open(path, flags | O_NOFOLLOW, mode);

}