cosmopolitan/libc/calls/hefty/copyfile.c

/*-*- mode:c;indent-tabs-mode:nil;c-basic-offset:2;tab-width:8;coding:utf-8 -*-│
│vi: set net ft=c ts=2 sts=2 sw=2 fenc=utf-8                                :vi│
╞══════════════════════════════════════════════════════════════════════════════╡
│ Copyright 2020 Justine Alexandra Roberts Tunney                              │
│                                                                              │
│ 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; version 2 of the License.                      │
│                                                                              │
│ 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., 51 Franklin Street, Fifth Floor, Boston, MA                │
│ 02110-1301 USA                                                               │
╚─────────────────────────────────────────────────────────────────────────────*/
#include "libc/calls/hefty/copyfile.h"
#include "libc/calls/internal.h"
#include "libc/calls/struct/stat.h"
#include "libc/dce.h"
#include "libc/nt/createfile.h"
#include "libc/nt/enum/accessmask.h"
#include "libc/nt/enum/creationdisposition.h"
#include "libc/nt/enum/fileflagandattributes.h"
#include "libc/nt/enum/filesharemode.h"
#include "libc/nt/files.h"
#include "libc/nt/runtime.h"
#include "libc/sysv/consts/at.h"
#include "libc/sysv/consts/madv.h"
#include "libc/sysv/consts/o.h"
#include "libc/time/time.h"

static textwindows int copyfile$nt(const char *src, const char *dst,
                                   int flags) {
  int64_t fhsrc, fhdst;
  struct NtFileTime accessed, modified;
  char16_t src16[PATH_MAX], dst16[PATH_MAX];
  if (mkntpath(src, src16) == -1) return -1;
  if (mkntpath(dst, dst16) == -1) return -1;
  if (CopyFile(src16, dst16, !!(flags & COPYFILE_NOCLOBBER))) {
    if (flags & COPYFILE_PRESERVE_TIMESTAMPS) {
      fhsrc = CreateFile(src16, kNtFileReadAttributes, kNtFileShareRead, NULL,
                         kNtOpenExisting, kNtFileAttributeNormal, 0);
      fhdst = CreateFile(dst16, kNtFileWriteAttributes, kNtFileShareRead, NULL,
                         kNtOpenExisting, kNtFileAttributeNormal, 0);
      if (fhsrc != -1 && fhdst != -1) {
        GetFileTime(fhsrc, NULL, &accessed, &modified);
        SetFileTime(fhdst, NULL, &accessed, &modified);
      }
      CloseHandle(fhsrc);
      CloseHandle(fhdst);
    }
    return 0;
  } else {
    return __winerr();
  }
}

static int copyfile$sysv(const char *src, const char *dst, int flags) {
  struct stat st;
  size_t remaining;
  ssize_t transferred;
  struct timespec amtime[2];
  int64_t inoffset, outoffset;
  int rc, srcfd, dstfd, oflags, omode;
  rc = -1;
  if ((srcfd = openat$sysv(AT_FDCWD, src, O_RDONLY, 0)) != -1) {
    if (fstat$sysv(srcfd, &st) != -1) {
      omode = st.st_mode & 0777;
      oflags = O_WRONLY | O_CREAT;
      if (flags & COPYFILE_NOCLOBBER) oflags |= O_EXCL;
      if ((dstfd = openat$sysv(AT_FDCWD, dst, oflags, omode)) != -1) {
        remaining = st.st_size;
        ftruncate(dstfd, remaining);
        inoffset = 0;
        outoffset = 0;
        while (remaining &&
               (transferred = copy_file_range(
                    srcfd, &inoffset, dstfd, &outoffset, remaining, 0)) != -1) {
          remaining -= transferred;
        }
        if (!remaining) {
          rc = 0;
          if (flags & COPYFILE_PRESERVE_TIMESTAMPS) {
            amtime[0] = st.st_atim;
            amtime[1] = st.st_mtim;
            utimensat$sysv(dstfd, NULL, amtime, 0);
          }
        }
        rc |= close$sysv(dstfd);
      }
    }
    rc |= close$sysv(srcfd);
  }
  return rc;
}

/**
 * Copies file.
 */
int copyfile(const char *src, const char *dst, int flags) {
  if (!IsWindows()) {
    return copyfile$sysv(src, dst, flags);
  } else {
    return copyfile$nt(src, dst, flags);
  }
}