You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
607 lines
17 KiB
607 lines
17 KiB
/*
|
|
* Copyright 2005 The Android Open Source Project
|
|
*
|
|
* Android "cp" replacement.
|
|
*
|
|
* The GNU/Linux "cp" uses O_LARGEFILE in its open() calls, utimes() instead
|
|
* of utime(), and getxattr()/setxattr() instead of chmod(). These are
|
|
* probably "better", but are non-portable, and not necessary for our
|
|
* purposes.
|
|
*/
|
|
#include <host/CopyFile.h>
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <getopt.h>
|
|
#include <dirent.h>
|
|
#include <fcntl.h>
|
|
#include <utime.h>
|
|
#include <limits.h>
|
|
#include <errno.h>
|
|
#include <assert.h>
|
|
|
|
#if defined(_WIN32)
|
|
#include <direct.h> /* For _mkdir() */
|
|
# define mkdir(path,mode) _mkdir(path)
|
|
# define S_ISLNK(s) 0
|
|
# define lstat stat
|
|
# ifndef EACCESS /* seems to be missing from the Mingw headers */
|
|
# define EACCESS 13
|
|
# endif
|
|
#endif
|
|
|
|
#ifndef O_BINARY
|
|
# define O_BINARY 0
|
|
#endif
|
|
|
|
/*#define DEBUG_MSGS*/
|
|
#ifdef DEBUG_MSGS
|
|
# define DBUG(x) printf x
|
|
#else
|
|
# define DBUG(x) ((void)0)
|
|
#endif
|
|
|
|
#define FSSEP '/' /* filename separator char */
|
|
|
|
static int copyFileRecursive(const char* src, const char* dst, bool isCmdLine, unsigned int options);
|
|
|
|
/*
|
|
* Returns true if the source file is newer than the destination file.
|
|
*
|
|
* The check is based on the modification date, whole seconds only. This
|
|
* also returns true if the file sizes don't match.
|
|
*/
|
|
static bool isSourceNewer(const struct stat* pSrcStat, const struct stat* pDstStat)
|
|
{
|
|
return (pSrcStat->st_mtime > pDstStat->st_mtime) ||
|
|
(pSrcStat->st_size != pDstStat->st_size);
|
|
}
|
|
|
|
/*
|
|
* Returns true if the source file has high resolution modification
|
|
* date. Cygwin/Mingw doesn't support st_mtim and always returns false.
|
|
*/
|
|
static bool isHiresMtime(const struct stat* pSrcStat)
|
|
{
|
|
#if defined(_WIN32)
|
|
return 0;
|
|
#elif defined(__APPLE__)
|
|
return pSrcStat->st_mtimespec.tv_nsec > 0;
|
|
#else
|
|
return pSrcStat->st_mtim.tv_nsec > 0;
|
|
#endif
|
|
}
|
|
|
|
/*
|
|
* Returns true if the source and destination files are actually the
|
|
* same thing. We detect this by checking the inode numbers, which seems
|
|
* to work on Cygwin.
|
|
*/
|
|
static bool isSameFile(const struct stat* pSrcStat, const struct stat* pDstStat)
|
|
{
|
|
#if defined(_WIN32)
|
|
(void)pSrcStat;
|
|
(void)pDstStat;
|
|
/* with MSVCRT.DLL, stat always sets st_ino to 0, and there is no simple way to */
|
|
/* get the equivalent information with Win32 (Cygwin does some weird stuff in */
|
|
/* its winsup/cygwin/fhandler_disk_file.cc to emulate this, too complex for us) */
|
|
return 0;
|
|
#else
|
|
return (pSrcStat->st_ino == pDstStat->st_ino);
|
|
#endif
|
|
}
|
|
|
|
static void printCopyMsg(const char* src, const char* dst, unsigned int options)
|
|
{
|
|
if ((options & COPY_VERBOSE_MASK) > 0)
|
|
printf(" '%s' --> '%s'\n", src, dst);
|
|
}
|
|
|
|
static void printNotNewerMsg(const char* src, const char* dst, unsigned int options)
|
|
{
|
|
(void)src;
|
|
if ((options & COPY_VERBOSE_MASK) > 1)
|
|
printf(" '%s' is up-to-date\n", dst);
|
|
}
|
|
|
|
/*
|
|
* Copy the contents of one file to another.
|
|
*
|
|
* The files are assumed to be seeked to the start.
|
|
*/
|
|
static int copyFileContents(const char* dst, int dstFd, const char* src, int srcFd)
|
|
{
|
|
unsigned char buf[8192];
|
|
ssize_t readCount, writeCount;
|
|
|
|
/*
|
|
* Read a chunk, write it, and repeat.
|
|
*/
|
|
while (1) {
|
|
readCount = read(srcFd, buf, sizeof(buf));
|
|
if (readCount < 0) {
|
|
fprintf(stderr,
|
|
"acp: failed reading '%s': %s\n", src, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
if (readCount > 0) {
|
|
writeCount = write(dstFd, buf, readCount);
|
|
if (writeCount < 0) {
|
|
fprintf(stderr,
|
|
"acp: failed writing '%s': %s\n", dst, strerror(errno));
|
|
return -1;
|
|
}
|
|
if (writeCount != readCount) {
|
|
fprintf(stderr, "acp: partial write to '%s' (%zd of %zd)\n",
|
|
dst, writeCount, readCount);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (readCount < (ssize_t) sizeof(buf))
|
|
break;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Set the permissions, owner, and timestamps on the destination file
|
|
* equal to those of the source file.
|
|
*
|
|
* Failures here are "soft"; they don't produce warning messages and don't
|
|
* cause the cp command to report a failure.
|
|
*/
|
|
static int setPermissions(const char* dst, const struct stat* pSrcStat, unsigned int options)
|
|
{
|
|
struct utimbuf ut;
|
|
|
|
if (options & COPY_TIMESTAMPS) {
|
|
/*
|
|
* Start with timestamps. The access and mod dates are not affected
|
|
* by the next operations.
|
|
*/
|
|
ut.actime = pSrcStat->st_atime;
|
|
ut.modtime = pSrcStat->st_mtime;
|
|
if (isHiresMtime(pSrcStat))
|
|
ut.modtime += 1;
|
|
if (utime(dst, &ut) != 0) {
|
|
DBUG(("--- unable to set timestamps on '%s': %s\n",
|
|
dst, strerror(errno)));
|
|
}
|
|
}
|
|
|
|
if (options & COPY_PERMISSIONS) {
|
|
/*
|
|
* Set the permissions.
|
|
*/
|
|
if (chmod(dst, pSrcStat->st_mode & ~(S_IFMT)) != 0) {
|
|
DBUG(("--- unable to set perms on '%s' to 0%o: %s\n",
|
|
dst, pSrcStat->st_mode & ~(S_IFMT), strerror(errno)));
|
|
}
|
|
#ifndef _WIN32
|
|
/*
|
|
* Set the owner.
|
|
*/
|
|
if (chown(dst, pSrcStat->st_uid, pSrcStat->st_gid) != 0) {
|
|
DBUG(("--- unable to set owner of '%s' to %d/%d: %s\n",
|
|
dst, pSrcStat->st_uid, pSrcStat->st_gid, strerror(errno)));
|
|
}
|
|
#endif
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Copy a regular file. If the destination file exists and is not a
|
|
* regular file, we fail. However, we use stat() rather than lstat(),
|
|
* because it's okay to write through a symlink (the noDereference stuff
|
|
* only applies to the source file).
|
|
*
|
|
* If the file doesn't exist, create it. If it does exist, truncate it.
|
|
*/
|
|
static int copyRegular(const char* src, const char* dst, const struct stat* pSrcStat, unsigned int options)
|
|
{
|
|
struct stat dstStat;
|
|
int srcFd, dstFd, statResult, copyResult;
|
|
|
|
DBUG(("--- copying regular '%s' to '%s'\n", src, dst));
|
|
|
|
statResult = stat(dst, &dstStat);
|
|
if (statResult == 0 && !S_ISREG(dstStat.st_mode)) {
|
|
fprintf(stderr,
|
|
"acp: destination '%s' exists and is not regular file\n",
|
|
dst);
|
|
return -1;
|
|
} else if (statResult != 0 && errno != ENOENT) {
|
|
fprintf(stderr, "acp: unable to stat destination '%s'\n", dst);
|
|
return -1;
|
|
}
|
|
|
|
if (statResult == 0) {
|
|
if (isSameFile(pSrcStat, &dstStat)) {
|
|
fprintf(stderr, "acp: '%s' and '%s' are the same file\n",
|
|
src, dst);
|
|
return -1;
|
|
}
|
|
if (options & COPY_UPDATE_ONLY) {
|
|
if (!isSourceNewer(pSrcStat, &dstStat)) {
|
|
DBUG(("--- source is not newer: '%s'\n", src));
|
|
printNotNewerMsg(src, dst, options);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* open src */
|
|
srcFd = open(src, O_RDONLY | O_BINARY, 0);
|
|
if (srcFd < 0) {
|
|
fprintf(stderr, "acp: unable to open '%s': %s\n", src, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
/* open dest with O_CREAT | O_TRUNC */
|
|
DBUG(("--- opening '%s'\n", dst));
|
|
dstFd = open(dst, O_CREAT | O_TRUNC | O_WRONLY | O_BINARY, 0644);
|
|
|
|
if (dstFd < 0) {
|
|
if (errno == ENOENT) {
|
|
/* this happens if the target directory doesn't exist */
|
|
fprintf(stderr,
|
|
"acp: cannot create '%s': %s\n", dst, strerror(errno));
|
|
(void) close(srcFd);
|
|
return -1;
|
|
}
|
|
|
|
/* if "force" is set, try removing the destination file and retry */
|
|
if (options & COPY_FORCE) {
|
|
if (unlink(dst) != 0) {
|
|
#ifdef _WIN32
|
|
/* MSVCRT.DLL unlink will fail with EACCESS if the file is set read-only */
|
|
/* so try to change its mode, and unlink again */
|
|
if (errno == EACCESS) {
|
|
if (chmod(dst, S_IWRITE|S_IREAD) == 0 && unlink(dst) == 0)
|
|
goto Open_File;
|
|
}
|
|
#endif
|
|
fprintf(stderr, "acp: unable to remove '%s': %s\n",
|
|
dst, strerror(errno));
|
|
(void) close(srcFd);
|
|
return -1;
|
|
}
|
|
#ifdef _WIN32
|
|
Open_File:
|
|
#endif
|
|
dstFd = open(dst, O_CREAT | O_TRUNC | O_WRONLY | O_BINARY, 0644);
|
|
}
|
|
}
|
|
if (dstFd < 0) {
|
|
fprintf(stderr, "acp: unable to open '%s': %s\n",
|
|
dst, strerror(errno));
|
|
(void) close(srcFd);
|
|
return -1;
|
|
}
|
|
|
|
copyResult = copyFileContents(dst, dstFd, src, srcFd);
|
|
|
|
(void) close(srcFd);
|
|
(void) close(dstFd);
|
|
if (copyResult != 0)
|
|
return -1;
|
|
|
|
#if defined(__APPLE__)
|
|
// Copy Mac OS X resource forks too.
|
|
{
|
|
char* srcRsrcName = NULL;
|
|
char* dstRsrcName = NULL;
|
|
struct stat rsrcStat;
|
|
|
|
srcRsrcName = malloc(strlen(src) + 5 + 1);
|
|
strcpy(srcRsrcName, src);
|
|
strcat(srcRsrcName, "/rsrc");
|
|
|
|
dstRsrcName = malloc(strlen(dst) + 5 + 1);
|
|
strcpy(dstRsrcName, dst);
|
|
strcat(dstRsrcName, "/rsrc");
|
|
|
|
if (stat(srcRsrcName, &rsrcStat) == 0 && rsrcStat.st_size > 0) {
|
|
DBUG(("--- RSRC: %s --> %s\n", srcRsrcName, dstRsrcName));
|
|
|
|
srcFd = open(srcRsrcName, O_RDONLY);
|
|
dstFd = open(dstRsrcName, O_TRUNC | O_WRONLY, 0);
|
|
copyResult = -1;
|
|
if (srcFd >= 0 && dstFd >= 0) {
|
|
copyResult = copyFileContents(dstRsrcName, dstFd,
|
|
srcRsrcName, srcFd);
|
|
(void) close(srcFd);
|
|
(void) close(dstFd);
|
|
}
|
|
|
|
if (copyResult != 0) {
|
|
free(srcRsrcName);
|
|
free(dstRsrcName);
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
free(srcRsrcName);
|
|
free(dstRsrcName);
|
|
}
|
|
#endif
|
|
|
|
setPermissions(dst, pSrcStat, options);
|
|
|
|
printCopyMsg(src, dst, options);
|
|
|
|
return 0;
|
|
}
|
|
|
|
|
|
/*
|
|
* Copy a symlink. This only happens if we're in "no derefence" mode,
|
|
* in which we copy the links rather than the files that are pointed at.
|
|
*
|
|
* We always discard the destination file. If it's a symlink already,
|
|
* we want to throw it out and replace it. If it's not a symlink, we
|
|
* need to trash it so we can create one.
|
|
*/
|
|
#if defined(_WIN32)
|
|
extern int copySymlink(const char* src, const char* dst, const struct stat* pSrcStat, unsigned int options)
|
|
#ifdef __clang__
|
|
__attribute__((unavailable("no symlinks on Windows")));
|
|
#else
|
|
__attribute__((error("no symlinks on Windows")));
|
|
#endif
|
|
#else
|
|
static int copySymlink(const char* src, const char* dst, const struct stat* pSrcStat, unsigned int options)
|
|
{
|
|
struct stat dstStat;
|
|
char linkBuf[PATH_MAX+1];
|
|
int statResult, nameLen;
|
|
|
|
assert(options & COPY_NO_DEREFERENCE);
|
|
DBUG(("--- copying symlink '%s' to '%s'\n", src, dst));
|
|
|
|
/* NOTE: we use lstat() here */
|
|
statResult = lstat(dst, &dstStat);
|
|
if (statResult == 0 && !S_ISREG(dstStat.st_mode)
|
|
&& !S_ISLNK(dstStat.st_mode)
|
|
)
|
|
{
|
|
fprintf(stderr,
|
|
"acp: destination '%s' exists and is not regular or symlink\n",
|
|
dst);
|
|
return -1;
|
|
}
|
|
|
|
if (statResult == 0) {
|
|
if (isSameFile(pSrcStat, &dstStat)) {
|
|
fprintf(stderr, "acp: '%s' and '%s' are the same file\n",
|
|
src, dst);
|
|
return -1;
|
|
}
|
|
if (options & COPY_UPDATE_ONLY) {
|
|
if (!isSourceNewer(pSrcStat, &dstStat)) {
|
|
DBUG(("--- source is not newer: '%s'\n", src));
|
|
printNotNewerMsg(src, dst, options);
|
|
return 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
/* extract the symlink contents */
|
|
nameLen = readlink(src, linkBuf, sizeof(linkBuf)-1);
|
|
if (nameLen <= 0) {
|
|
fprintf(stderr, "acp: unable to read symlink '%s': %s\n",
|
|
src, strerror(errno));
|
|
return -1;
|
|
}
|
|
linkBuf[nameLen] = '\0';
|
|
DBUG(("--- creating symlink file '%s' (--> %s)\n", dst, linkBuf));
|
|
|
|
if (statResult == 0) {
|
|
DBUG(("--- removing '%s'\n", dst));
|
|
if (unlink(dst) != 0) {
|
|
fprintf(stderr, "acp: unable to remove '%s': %s\n",
|
|
dst, strerror(errno));
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
if (symlink(linkBuf, dst) != 0) {
|
|
fprintf(stderr, "acp: unable to create symlink '%s' [%s]: %s\n",
|
|
dst, linkBuf, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
/*
|
|
* There's no way to set the file date or access permissions, but
|
|
* it is possible to set the owner.
|
|
*/
|
|
if (options & COPY_PERMISSIONS) {
|
|
if (lchown(dst, pSrcStat->st_uid, pSrcStat->st_gid) != 0)
|
|
DBUG(("--- lchown failed: %s\n", strerror(errno)));
|
|
}
|
|
|
|
printCopyMsg(src, dst, options);
|
|
|
|
return 0;
|
|
}
|
|
#endif
|
|
|
|
/*
|
|
* Copy the contents of one directory to another. Both "src" and "dst"
|
|
* must be directories. We will create "dst" if it does not exist.
|
|
*/
|
|
int copyDirectory(const char* src, const char* dst, const struct stat* pSrcStat, unsigned int options)
|
|
{
|
|
int retVal = 0;
|
|
struct stat dstStat;
|
|
DIR* dir;
|
|
int cc, statResult;
|
|
|
|
DBUG(("--- copy dir '%s' to '%s'\n", src, dst));
|
|
|
|
statResult = stat(dst, &dstStat);
|
|
if (statResult == 0 && !S_ISDIR(dstStat.st_mode)) {
|
|
fprintf(stderr,
|
|
"acp: destination '%s' exists and is not a directory\n", dst);
|
|
return -1;
|
|
} else if (statResult != 0 && errno != ENOENT) {
|
|
fprintf(stderr, "acp: unable to stat destination '%s'\n", dst);
|
|
return -1;
|
|
}
|
|
|
|
if (statResult == 0) {
|
|
if (isSameFile(pSrcStat, &dstStat)) {
|
|
fprintf(stderr,
|
|
"acp: cannot copy directory into itself ('%s' and '%s')\n",
|
|
src, dst);
|
|
return -1;
|
|
}
|
|
} else {
|
|
DBUG(("--- creating dir '%s'\n", dst));
|
|
cc = mkdir(dst, 0755);
|
|
if (cc != 0) {
|
|
fprintf(stderr, "acp: unable to create directory '%s': %s\n",
|
|
dst, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
/* only print on mkdir */
|
|
printCopyMsg(src, dst, options);
|
|
}
|
|
|
|
/*
|
|
* Open the directory, and plow through its contents.
|
|
*/
|
|
dir = opendir(src);
|
|
if (dir == NULL) {
|
|
fprintf(stderr, "acp: unable to open directory '%s': %s\n",
|
|
src, strerror(errno));
|
|
return -1;
|
|
}
|
|
|
|
while (1) {
|
|
struct dirent* ent;
|
|
char* srcFile;
|
|
char* dstFile;
|
|
int srcLen, dstLen, nameLen;
|
|
|
|
ent = readdir(dir);
|
|
if (ent == NULL)
|
|
break;
|
|
|
|
if (strcmp(ent->d_name, ".") == 0 ||
|
|
strcmp(ent->d_name, "..") == 0)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
nameLen = strlen(ent->d_name);
|
|
srcLen = strlen(src);
|
|
dstLen = strlen(dst);
|
|
|
|
srcFile = malloc(srcLen +1 + nameLen +1);
|
|
memcpy(srcFile, src, srcLen);
|
|
srcFile[srcLen] = FSSEP;
|
|
memcpy(srcFile + srcLen+1, ent->d_name, nameLen +1);
|
|
|
|
dstFile = malloc(dstLen +1 + nameLen +1);
|
|
memcpy(dstFile, dst, dstLen);
|
|
dstFile[dstLen] = FSSEP;
|
|
memcpy(dstFile + dstLen+1, ent->d_name, nameLen +1);
|
|
|
|
if (copyFileRecursive(srcFile, dstFile, false, options) != 0)
|
|
retVal = -1; /* note failure and keep going */
|
|
|
|
free(srcFile);
|
|
free(dstFile);
|
|
}
|
|
closedir(dir);
|
|
|
|
setPermissions(dst, pSrcStat, options);
|
|
|
|
return retVal;
|
|
}
|
|
|
|
/*
|
|
* Do the actual copy. This is called recursively from copyDirectory().
|
|
*
|
|
* "dst" should only be a directory if "src" is also a directory.
|
|
*
|
|
* Returns 0 on success.
|
|
*/
|
|
static int copyFileRecursive(const char* src, const char* dst, bool isCmdLine, unsigned int options)
|
|
{
|
|
char* srcExe = NULL;
|
|
char* dstExe = NULL;
|
|
char* dstDir = NULL;
|
|
struct stat srcStat;
|
|
int retVal = 0;
|
|
int statResult, statErrno;
|
|
(void)isCmdLine;
|
|
|
|
/*
|
|
* Stat the source file. If it doesn't exist, fail.
|
|
*/
|
|
if (options & COPY_NO_DEREFERENCE)
|
|
statResult = lstat(src, &srcStat);
|
|
else
|
|
statResult = stat(src, &srcStat);
|
|
statErrno = errno; /* preserve across .exe attempt */
|
|
|
|
if (statResult < 0) {
|
|
if (statErrno == ENOENT)
|
|
fprintf(stderr, "acp: file '%s' does not exist\n", src);
|
|
else
|
|
fprintf(stderr, "acp: unable to stat '%s': %s\n",
|
|
src, strerror(statErrno));
|
|
retVal = -1;
|
|
goto bail;
|
|
}
|
|
|
|
/*
|
|
* If "src" is a directory, ignore it if "recursive" isn't set.
|
|
*
|
|
* We want to create "dst" as a directory (or verify that it already
|
|
* exists as a directory), and then copy its contents.
|
|
*/
|
|
if (S_ISDIR(srcStat.st_mode)) {
|
|
if (!(options & COPY_RECURSIVE)) {
|
|
fprintf(stderr, "acp: omitting directory '%s'\n", src);
|
|
} else {
|
|
retVal = copyDirectory(src, dst, &srcStat, options);
|
|
}
|
|
#if !defined(_WIN32)
|
|
} else if (S_ISLNK(srcStat.st_mode)) {
|
|
retVal = copySymlink(src, dst, &srcStat, options);
|
|
#endif
|
|
} else if (S_ISREG(srcStat.st_mode)) {
|
|
retVal = copyRegular(src, dst, &srcStat, options);
|
|
} else {
|
|
fprintf(stderr, "acp: skipping unusual file '%s' (mode=0%o)\n",
|
|
src, srcStat.st_mode);
|
|
retVal = -1;
|
|
}
|
|
|
|
bail:
|
|
free(srcExe);
|
|
free(dstExe);
|
|
free(dstDir);
|
|
return retVal;
|
|
}
|
|
|
|
int copyFile(const char* src, const char* dst, unsigned int options)
|
|
{
|
|
return copyFileRecursive(src, dst, true, options);
|
|
}
|
|
|
|
|