//=== is guaranteed to work on *all* UNIX variants.
//===----------------------------------------------------------------------===//
-#include <llvm/Config/config.h>
-#include <llvm/Config/alloca.h>
+#include "llvm/Config/alloca.h"
#include "Unix.h"
+#if HAVE_SYS_STAT_H
#include <sys/stat.h>
+#endif
+#if HAVE_FCNTL_H
#include <fcntl.h>
+#endif
+#if HAVE_UTIME_H
#include <utime.h>
-#include <dirent.h>
+#endif
+#if HAVE_TIME_H
+#include <time.h>
+#endif
+#if HAVE_DIRENT_H
+# include <dirent.h>
+# define NAMLEN(dirent) strlen((dirent)->d_name)
+#else
+# define dirent direct
+# define NAMLEN(dirent) (dirent)->d_namlen
+# if HAVE_SYS_NDIR_H
+# include <sys/ndir.h>
+# endif
+# if HAVE_SYS_DIR_H
+# include <sys/dir.h>
+# endif
+# if HAVE_NDIR_H
+# include <ndir.h>
+# endif
+#endif
+
+// Put in a hack for Cygwin which falsely reports that the mkdtemp function
+// is available when it is not.
+#ifdef __CYGWIN__
+# undef HAVE_MKDTEMP
+#endif
+
+namespace {
+inline bool lastIsSlash(const std::string& path) {
+ return !path.empty() && path[path.length() - 1] == '/';
+}
+
+}
namespace llvm {
using namespace sys;
-Path::Path(std::string unverified_path)
- : path(unverified_path)
-{
- if (unverified_path.empty())
- return;
- if (this->isValid())
- return;
- // oops, not valid.
- path.clear();
- ThrowErrno(unverified_path + ": path is not valid");
+bool
+Path::isValid() const {
+ // Check some obvious things
+ if (path.empty())
+ return false;
+ else if (path.length() >= MAXPATHLEN)
+ return false;
+
+ // Check that the characters are ascii chars
+ size_t len = path.length();
+ unsigned i = 0;
+ while (i < len && isascii(path[i]))
+ ++i;
+ return i >= len;
}
Path
Path::GetRootDirectory() {
Path result;
- result.setDirectory("/");
+ result.set("/");
+ return result;
+}
+
+Path
+Path::GetTemporaryDirectory(std::string* ErrMsg ) {
+#if defined(HAVE_MKDTEMP)
+ // The best way is with mkdtemp but that's not available on many systems,
+ // Linux and FreeBSD have it. Others probably won't.
+ char pathname[MAXPATHLEN];
+ strcpy(pathname,"/tmp/llvm_XXXXXX");
+ if (0 == mkdtemp(pathname)) {
+ MakeErrMsg(ErrMsg,
+ std::string(pathname) + ": can't create temporary directory");
+ return Path();
+ }
+ Path result;
+ result.set(pathname);
+ assert(result.isValid() && "mkdtemp didn't create a valid pathname!");
+ return result;
+#elif defined(HAVE_MKSTEMP)
+ // If no mkdtemp is available, mkstemp can be used to create a temporary file
+ // which is then removed and created as a directory. We prefer this over
+ // mktemp because of mktemp's inherent security and threading risks. We still
+ // have a slight race condition from the time the temporary file is created to
+ // the time it is re-created as a directoy.
+ char pathname[MAXPATHLEN];
+ strcpy(pathname, "/tmp/llvm_XXXXXX");
+ int fd = 0;
+ if (-1 == (fd = mkstemp(pathname))) {
+ MakeErrMsg(ErrMsg,
+ std::string(pathname) + ": can't create temporary directory");
+ return Path();
+ }
+ ::close(fd);
+ ::unlink(pathname); // start race condition, ignore errors
+ if (-1 == ::mkdir(pathname, S_IRWXU)) { // end race condition
+ MakeErrMsg(ErrMsg,
+ std::string(pathname) + ": can't create temporary directory");
+ return Path();
+ }
+ Path result;
+ result.set(pathname);
+ assert(result.isValid() && "mkstemp didn't create a valid pathname!");
+ return result;
+#elif defined(HAVE_MKTEMP)
+ // If a system doesn't have mkdtemp(3) or mkstemp(3) but it does have
+ // mktemp(3) then we'll assume that system (e.g. AIX) has a reasonable
+ // implementation of mktemp(3) and doesn't follow BSD 4.3's lead of replacing
+ // the XXXXXX with the pid of the process and a letter. That leads to only
+ // twenty six temporary files that can be generated.
+ char pathname[MAXPATHLEN];
+ strcpy(pathname, "/tmp/llvm_XXXXXX");
+ char *TmpName = ::mktemp(pathname);
+ if (TmpName == 0) {
+ MakeErrMsg(ErrMsg,
+ std::string(TmpName) + ": can't create unique directory name");
+ return Path();
+ }
+ if (-1 == ::mkdir(TmpName, S_IRWXU)) {
+ MakeErrMsg(ErrMsg,
+ std::string(TmpName) + ": can't create temporary directory");
+ return Path();
+ }
+ Path result;
+ result.set(TmpName);
+ assert(result.isValid() && "mktemp didn't create a valid pathname!");
return result;
+#else
+ // This is the worst case implementation. tempnam(3) leaks memory unless its
+ // on an SVID2 (or later) system. On BSD 4.3 it leaks. tmpnam(3) has thread
+ // issues. The mktemp(3) function doesn't have enough variability in the
+ // temporary name generated. So, we provide our own implementation that
+ // increments an integer from a random number seeded by the current time. This
+ // should be sufficiently unique that we don't have many collisions between
+ // processes. Generally LLVM processes don't run very long and don't use very
+ // many temporary files so this shouldn't be a big issue for LLVM.
+ static time_t num = ::time(0);
+ char pathname[MAXPATHLEN];
+ do {
+ num++;
+ sprintf(pathname, "/tmp/llvm_%010u", unsigned(num));
+ } while ( 0 == access(pathname, F_OK ) );
+ if (-1 == ::mkdir(pathname, S_IRWXU)) {
+ MakeErrMsg(ErrMsg,
+ std::string(pathname) + ": can't create temporary directory");
+ return Path();
+ Path result;
+ result.set(pathname);
+ assert(result.isValid() && "mkstemp didn't create a valid pathname!");
+ return result;
+#endif
}
static void getPathList(const char*path, std::vector<sys::Path>& Paths) {
Path tmpPath;
while( delim != 0 ) {
std::string tmp(at, size_t(delim-at));
- if (tmpPath.setDirectory(tmp))
- if (tmpPath.readable())
+ if (tmpPath.set(tmp))
+ if (tmpPath.canRead())
Paths.push_back(tmpPath);
at = delim + 1;
delim = strchr(at, ':');
}
if (*at != 0)
- if (tmpPath.setDirectory(std::string(at)))
- if (tmpPath.readable())
+ if (tmpPath.set(std::string(at)))
+ if (tmpPath.canRead())
Paths.push_back(tmpPath);
}
+
void
Path::GetSystemLibraryPaths(std::vector<sys::Path>& Paths) {
#ifdef LTDL_SHLIBPATH_VAR
if (env_var != 0) {
getPathList(env_var,Paths);
}
-#ifdef LLVMGCCDIR
- {
- Path tmpPath(std::string(LLVMGCCDIR) + "bytecode-libs/");
- if (tmpPath.readable())
- Paths.push_back(tmpPath);
- }
-#endif
#ifdef LLVM_LIBDIR
{
Path tmpPath;
- if (tmpPath.setDirectory(LLVM_LIBDIR))
- if (tmpPath.readable())
+ if (tmpPath.set(LLVM_LIBDIR))
+ if (tmpPath.canRead())
Paths.push_back(tmpPath);
}
#endif
return Path("/etc/llvm/");
}
-Path
-Path::GetLLVMConfigDir() {
- Path result;
- if (result.setDirectory(LLVM_ETCDIR))
- return result;
- return GetLLVMDefaultConfigDir();
-}
-
Path
Path::GetUserHomeDirectory() {
const char* home = getenv("HOME");
if (home) {
Path result;
- if (result.setDirectory(home))
+ if (result.set(home))
return result;
}
return GetRootDirectory();
}
-bool
-Path::isFile() const {
- return (isValid() && path[path.length()-1] != '/');
-}
-
-bool
-Path::isDirectory() const {
- return (isValid() && path[path.length()-1] == '/');
-}
std::string
Path::getBasename() const {
else
slash++;
- return path.substr(slash, path.rfind('.'));
+ size_t dot = path.rfind('.');
+ if (dot == std::string::npos || dot < slash)
+ return path.substr(slash);
+ else
+ return path.substr(slash, dot - slash);
}
bool Path::hasMagicNumber(const std::string &Magic) const {
size_t len = Magic.size();
assert(len < 1024 && "Request for magic string too long");
char* buf = (char*) alloca(1 + len);
- int fd = ::open(path.c_str(),O_RDONLY);
+ int fd = ::open(path.c_str(), O_RDONLY);
if (fd < 0)
return false;
size_t read_len = ::read(fd, buf, len);
}
bool Path::getMagicNumber(std::string& Magic, unsigned len) const {
- if (!isFile())
- return false;
assert(len < 1024 && "Request for magic string too long");
char* buf = (char*) alloca(1 + len);
- int fd = ::open(path.c_str(),O_RDONLY);
+ int fd = ::open(path.c_str(), O_RDONLY);
if (fd < 0)
return false;
ssize_t bytes_read = ::read(fd, buf, len);
bool
Path::isBytecodeFile() const {
- char buffer[ 4];
+ char buffer[4];
buffer[0] = 0;
- int fd = ::open(path.c_str(),O_RDONLY);
+ int fd = ::open(path.c_str(), O_RDONLY);
if (fd < 0)
return false;
ssize_t bytes_read = ::read(fd, buffer, 4);
return false;
return (buffer[0] == 'l' && buffer[1] == 'l' && buffer[2] == 'v' &&
- (buffer[3] == 'c' || buffer[3] == 'm'));
+ (buffer[3] == 'c' || buffer[3] == 'm'));
}
bool
}
bool
-Path::readable() const {
+Path::canRead() const {
return 0 == access(path.c_str(), F_OK | R_OK );
}
bool
-Path::writable() const {
+Path::canWrite() const {
return 0 == access(path.c_str(), F_OK | W_OK );
}
bool
-Path::executable() const {
- return 0 == access(path.c_str(), R_OK | X_OK );
+Path::canExecute() const {
+ if (0 != access(path.c_str(), R_OK | X_OK ))
+ return false;
+ struct stat st;
+ int r = stat(path.c_str(), &st);
+ if (r != 0 || !S_ISREG(st.st_mode))
+ return false;
+ return true;
}
std::string
return path.substr(pos+1);
}
-void
-Path::getStatusInfo(StatusInfo& info) const {
+bool
+Path::getFileStatus(FileStatus &info, std::string *ErrStr) const {
struct stat buf;
- if (0 != stat(path.c_str(), &buf)) {
- ThrowErrno(std::string("Can't get status: ")+path);
- }
+ if (0 != stat(path.c_str(), &buf))
+ return GetErrno(path + ": can't get status of file '" + path + "'", ErrStr);
info.fileSize = buf.st_size;
info.modTime.fromEpochTime(buf.st_mtime);
info.mode = buf.st_mode;
info.user = buf.st_uid;
info.group = buf.st_gid;
- info.isDir = S_ISDIR(buf.st_mode);
- if (info.isDir && path[path.length()-1] != '/')
- path += '/';
+ info.isDir = S_ISDIR(buf.st_mode);
+ info.isFile = S_ISREG(buf.st_mode);
+ return false;
}
-bool
-Path::getDirectoryContents(std::set<Path>& result) const {
- if (!isDirectory())
+static bool AddPermissionBits(const Path &File, int bits) {
+ // Get the umask value from the operating system. We want to use it
+ // when changing the file's permissions. Since calling umask() sets
+ // the umask and returns its old value, we must call it a second
+ // time to reset it to the user's preference.
+ int mask = umask(0777); // The arg. to umask is arbitrary.
+ umask(mask); // Restore the umask.
+
+ // Get the file's current mode.
+ FileStatus Stat;
+ if (File.getFileStatus(Stat)) return false;
+
+ // Change the file to have whichever permissions bits from 'bits'
+ // that the umask would not disable.
+ if ((chmod(File.c_str(), (Stat.getMode() | (bits & ~mask)))) == -1)
return false;
+
+ return true;
+}
+
+bool Path::makeReadableOnDisk(std::string* ErrMsg) {
+ if (!AddPermissionBits(*this, 0444)) {
+ MakeErrMsg(ErrMsg, path + ": can't make file readable");
+ return true;
+ }
+ return false;
+}
+
+bool Path::makeWriteableOnDisk(std::string* ErrMsg) {
+ if (!AddPermissionBits(*this, 0222)) {
+ MakeErrMsg(ErrMsg, path + ": can't make file writable");
+ return true;
+ }
+ return false;
+}
+
+bool Path::makeExecutableOnDisk(std::string* ErrMsg) {
+ if (!AddPermissionBits(*this, 0111)) {
+ MakeErrMsg(ErrMsg, path + ": can't make file executable");
+ return true;
+ }
+ return false;
+}
+
+bool
+Path::getDirectoryContents(std::set<Path>& result, std::string* ErrMsg) const {
DIR* direntries = ::opendir(path.c_str());
- if (direntries == 0)
- ThrowErrno(path + ": can't open directory");
+ if (direntries == 0) {
+ MakeErrMsg(ErrMsg, path + ": can't open directory");
+ return true;
+ }
+
+ std::string dirPath = path;
+ if (!lastIsSlash(dirPath))
+ dirPath += '/';
result.clear();
struct dirent* de = ::readdir(direntries);
- while (de != 0) {
+ for ( ; de != 0; de = ::readdir(direntries)) {
if (de->d_name[0] != '.') {
- Path aPath(path + (const char*)de->d_name);
- struct stat buf;
- if (0 != stat(aPath.path.c_str(), &buf))
- ThrowErrno(aPath.path + ": can't get status");
- if (S_ISDIR(buf.st_mode))
- aPath.path += "/";
+ Path aPath(dirPath + (const char*)de->d_name);
+ struct stat st;
+ if (0 != lstat(aPath.path.c_str(), &st)) {
+ if (S_ISLNK(st.st_mode))
+ continue; // dangling symlink -- ignore
+ MakeErrMsg(ErrMsg, aPath.path + ": can't determine file object type");
+ return true;
+ }
result.insert(aPath);
}
- de = ::readdir(direntries);
}
closedir(direntries);
- return true;
-}
-
-bool
-Path::setDirectory(const std::string& a_path) {
- if (a_path.size() == 0)
- return false;
- Path save(*this);
- path = a_path;
- size_t last = a_path.size() -1;
- if (last != 0 && a_path[last] != '/')
- path += '/';
- if (!isValid()) {
- path = save.path;
- return false;
- }
- return true;
+ return false;
}
bool
-Path::setFile(const std::string& a_path) {
- if (a_path.size() == 0)
+Path::set(const std::string& a_path) {
+ if (a_path.empty())
return false;
- Path save(*this);
+ std::string save(path);
path = a_path;
- size_t last = a_path.size() - 1;
- while (last > 0 && a_path[last] == '/')
- last--;
- path.erase(last+1);
if (!isValid()) {
- path = save.path;
+ path = save;
return false;
}
return true;
}
bool
-Path::appendDirectory(const std::string& dir) {
- if (isFile())
+Path::appendComponent(const std::string& name) {
+ if (name.empty())
return false;
- Path save(*this);
- path += dir;
- path += "/";
+ std::string save(path);
+ if (!lastIsSlash(path))
+ path += '/';
+ path += name;
if (!isValid()) {
- path = save.path;
+ path = save;
return false;
}
return true;
}
bool
-Path::elideDirectory() {
- if (isFile())
- return false;
+Path::eraseComponent() {
size_t slashpos = path.rfind('/',path.size());
- if (slashpos == 0 || slashpos == std::string::npos)
- return false;
+ if (slashpos == 0 || slashpos == std::string::npos) {
+ path.erase();
+ return true;
+ }
if (slashpos == path.size() - 1)
slashpos = path.rfind('/',slashpos-1);
- if (slashpos == std::string::npos)
- return false;
- path.erase(slashpos);
- return true;
-}
-
-bool
-Path::appendFile(const std::string& file) {
- if (!isDirectory())
- return false;
- Path save(*this);
- path += file;
- if (!isValid()) {
- path = save.path;
- return false;
+ if (slashpos == std::string::npos) {
+ path.erase();
+ return true;
}
- return true;
-}
-
-bool
-Path::elideFile() {
- if (isDirectory())
- return false;
- size_t slashpos = path.rfind('/',path.size());
- if (slashpos == std::string::npos)
- return false;
- path.erase(slashpos+1);
+ path.erase(slashpos);
return true;
}
bool
Path::appendSuffix(const std::string& suffix) {
- if (isDirectory())
- return false;
- Path save(*this);
+ std::string save(path);
path.append(".");
path.append(suffix);
if (!isValid()) {
- path = save.path;
+ path = save;
return false;
}
return true;
}
-bool
-Path::elideSuffix() {
- if (isDirectory()) return false;
+bool
+Path::eraseSuffix() {
+ std::string save = path;
size_t dotpos = path.rfind('.',path.size());
size_t slashpos = path.rfind('/',path.size());
- if (slashpos != std::string::npos && dotpos != std::string::npos &&
- dotpos > slashpos) {
- path.erase(dotpos, path.size()-dotpos);
- return true;
+ if (dotpos != std::string::npos) {
+ if (slashpos == std::string::npos || dotpos > slashpos+1) {
+ path.erase(dotpos, path.size()-dotpos);
+ return true;
+ }
}
+ if (!isValid())
+ path = save;
return false;
}
-
bool
-Path::createDirectory( bool create_parents) {
- // Make sure we're dealing with a directory
- if (!isDirectory()) return false;
-
+Path::createDirectoryOnDisk( bool create_parents, std::string* ErrMsg ) {
// Get a writeable copy of the path name
char pathname[MAXPATHLEN];
path.copy(pathname,MAXPATHLEN);
while ( next != 0 ) {
*next = 0;
if (0 != access(pathname, F_OK | R_OK | W_OK))
- if (0 != mkdir(pathname, S_IRWXU | S_IRWXG))
- ThrowErrno(std::string(pathname) + ": Can't create directory");
+ if (0 != mkdir(pathname, S_IRWXU | S_IRWXG)) {
+ MakeErrMsg(ErrMsg,
+ std::string(pathname) + ": can't create directory");
+ return true;
+ }
char* save = next;
next = strchr(next+1,'/');
*save = '/';
}
if (0 != access(pathname, F_OK | R_OK))
- if (0 != mkdir(pathname, S_IRWXU | S_IRWXG))
- ThrowErrno(std::string(pathname) + ": Can't create directory");
- return true;
+ if (0 != mkdir(pathname, S_IRWXU | S_IRWXG)) {
+ MakeErrMsg(ErrMsg, std::string(pathname) + ": can't create directory");
+ return true;
+ }
+ return false;
}
bool
-Path::createFile() {
- // Make sure we're dealing with a file
- if (!isFile()) return false;
-
+Path::createFileOnDisk(std::string* ErrMsg) {
// Create the file
int fd = ::creat(path.c_str(), S_IRUSR | S_IWUSR);
- if (fd < 0)
- ThrowErrno(path + ": Can't create file");
+ if (fd < 0) {
+ MakeErrMsg(ErrMsg, path + ": can't create file");
+ return true;
+ }
::close(fd);
-
- return true;
+ return false;
}
bool
-Path::createTemporaryFile() {
- // Make sure we're dealing with a file
- if (!isFile()) return false;
+Path::createTemporaryFileOnDisk(bool reuse_current, std::string* ErrMsg) {
+ // Make this into a unique file name
+ makeUnique( reuse_current );
- // Append the filename filler
- char pathname[MAXPATHLEN];
- path.copy(pathname,MAXPATHLEN);
- pathname[path.length()] = 0;
- strcat(pathname,"XXXXXX");
- int fd = ::mkstemp(pathname);
+ // create the file
+ int fd = ::open(path.c_str(), O_WRONLY|O_CREAT|O_TRUNC, 0666);
if (fd < 0) {
- ThrowErrno(path + ": Can't create temporary file");
+ MakeErrMsg(ErrMsg, path + ": can't create temporary file");
+ return true;
}
- path = pathname;
::close(fd);
- return true;
+ return false;
}
bool
-Path::destroyDirectory(bool remove_contents) {
- // Make sure we're dealing with a directory
- if (!isDirectory()) return false;
-
- // If it doesn't exist, we're done.
- if (!exists()) return true;
-
+Path::eraseFromDisk(bool remove_contents, std::string *ErrStr) const {
+ FileStatus Status;
+ if (getFileStatus(Status, ErrStr))
+ return true;
+
+ // Note: this check catches strange situations. In all cases, LLVM should only
+ // be involved in the creation and deletion of regular files. This check
+ // ensures that what we're trying to erase is a regular file. It effectively
+ // prevents LLVM from erasing things like /dev/null, any block special file,
+ // or other things that aren't "regular" files.
+ if (Status.isFile) {
+ if (unlink(path.c_str()) != 0)
+ return GetErrno(path + ": can't destroy file", ErrStr);
+ return false;
+ }
+
+ if (!Status.isDir) {
+ if (ErrStr) *ErrStr = "not a file or directory";
+ return true;
+ }
if (remove_contents) {
- // Recursively descend the directory to remove its content
- std::string cmd("/bin/rm -rf ");
- cmd += path;
+ // Recursively descend the directory to remove its contents.
+ std::string cmd = "/bin/rm -rf " + path;
system(cmd.c_str());
- } else {
- // Otherwise, try to just remove the one directory
- char pathname[MAXPATHLEN];
- path.copy(pathname,MAXPATHLEN);
- int lastchar = path.length() - 1 ;
- if (pathname[lastchar] == '/')
- pathname[lastchar] = 0;
- else
- pathname[lastchar+1] = 0;
- if ( 0 != rmdir(pathname))
- ThrowErrno(std::string(pathname) + ": Can't destroy directory");
+ return false;
}
- return true;
-}
-bool
-Path::destroyFile() {
- if (!isFile()) return false;
- if (0 != unlink(path.c_str()))
- ThrowErrno(path + ": Can't destroy file");
- return true;
+ // Otherwise, try to just remove the one directory.
+ char pathname[MAXPATHLEN];
+ path.copy(pathname, MAXPATHLEN);
+ int lastchar = path.length() - 1 ;
+ if (pathname[lastchar] == '/')
+ pathname[lastchar] = 0;
+ else
+ pathname[lastchar+1] = 0;
+
+ if (rmdir(pathname) != 0)
+ return GetErrno(std::string(pathname) + ": can't destroy directory",
+ ErrStr);
+ return false;
}
bool
-Path::renameFile(const Path& newName) {
- if (!isFile()) return false;
- if (0 != rename(path.c_str(), newName.c_str()))
- ThrowErrno(std::string("can't rename ") + path + " as " +
- newName.toString());
+Path::renamePathOnDisk(const Path& newName) {
+ if (0 != ::rename(path.c_str(), newName.c_str()))
+ ThrowErrno(std::string("can't rename '") + path + "' as '" +
+ newName.toString() + "' ");
return true;
}
bool
-Path::setStatusInfo(const StatusInfo& si) const {
- if (!isFile()) return false;
+Path::setStatusInfoOnDisk(const FileStatus &si, std::string *ErrStr) const {
struct utimbuf utb;
utb.actime = si.modTime.toPosixTime();
utb.modtime = utb.actime;
if (0 != ::utime(path.c_str(),&utb))
- ThrowErrno(path + ": can't set file modification time");
+ return GetErrno(path + ": can't set file modification time", ErrStr);
if (0 != ::chmod(path.c_str(),si.mode))
- ThrowErrno(path + ": can't set mode");
- return true;
+ return GetErrno(path + ": can't set mode", ErrStr);
+ return false;
+}
+
+void
+sys::CopyFile(const sys::Path &Dest, const sys::Path &Src) {
+ int inFile = -1;
+ int outFile = -1;
+ try {
+ inFile = ::open(Src.c_str(), O_RDONLY);
+ if (inFile == -1)
+ ThrowErrno(Src.toString() + ": can't open source file to copy: ");
+
+ outFile = ::open(Dest.c_str(), O_WRONLY|O_CREAT, 0666);
+ if (outFile == -1)
+ ThrowErrno(Dest.toString() +": can't create destination file for copy: ");
+
+ char Buffer[16*1024];
+ while (ssize_t Amt = ::read(inFile, Buffer, 16*1024)) {
+ if (Amt == -1) {
+ if (errno != EINTR && errno != EAGAIN)
+ ThrowErrno(Src.toString()+": can't read source file: ");
+ } else {
+ char *BufPtr = Buffer;
+ while (Amt) {
+ ssize_t AmtWritten = ::write(outFile, BufPtr, Amt);
+ if (AmtWritten == -1) {
+ if (errno != EINTR && errno != EAGAIN)
+ ThrowErrno(Dest.toString() + ": can't write destination file: ");
+ } else {
+ Amt -= AmtWritten;
+ BufPtr += AmtWritten;
+ }
+ }
+ }
+ }
+ ::close(inFile);
+ ::close(outFile);
+ } catch (...) {
+ if (inFile != -1)
+ ::close(inFile);
+ if (outFile != -1)
+ ::close(outFile);
+ throw;
+ }
}
+void
+Path::makeUnique(bool reuse_current) {
+ if (reuse_current && !exists())
+ return; // File doesn't exist already, just use it!
+
+ // Append an XXXXXX pattern to the end of the file for use with mkstemp,
+ // mktemp or our own implementation.
+ char *FNBuffer = (char*) alloca(path.size()+8);
+ path.copy(FNBuffer,path.size());
+ strcpy(FNBuffer+path.size(), "-XXXXXX");
+
+#if defined(HAVE_MKSTEMP)
+ int TempFD;
+ if ((TempFD = mkstemp(FNBuffer)) == -1) {
+ ThrowErrno(path + ": can't make unique filename");
+ }
+
+ // We don't need to hold the temp file descriptor... we will trust that no one
+ // will overwrite/delete the file before we can open it again.
+ close(TempFD);
+
+ // Save the name
+ path = FNBuffer;
+#elif defined(HAVE_MKTEMP)
+ // If we don't have mkstemp, use the old and obsolete mktemp function.
+ if (mktemp(FNBuffer) == 0) {
+ ThrowErrno(path + ": can't make unique filename");
+ }
+
+ // Save the name
+ path = FNBuffer;
+#else
+ // Okay, looks like we have to do it all by our lonesome.
+ static unsigned FCounter = 0;
+ unsigned offset = path.size() + 1;
+ while ( FCounter < 999999 && exists()) {
+ sprintf(FNBuffer+offset,"%06u",++FCounter);
+ path = FNBuffer;
+ }
+ if (FCounter > 999999)
+ throw std::string(path + ": can't make unique filename: too many files");
+#endif
+
+}
}
-// vim: sw=2