forked from lix-project/lix
7984cfc7c1
read()/write() to fix this once and for all.
185 lines
3.6 KiB
C++
185 lines
3.6 KiB
C++
#include <iostream>
|
|
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
#include <dirent.h>
|
|
|
|
#include "util.hh"
|
|
|
|
|
|
string thisSystem = SYSTEM;
|
|
|
|
|
|
Error::Error(const format & f)
|
|
{
|
|
err = f.str();
|
|
}
|
|
|
|
|
|
SysError::SysError(const format & f)
|
|
: Error(format("%1%: %2%") % f.str() % strerror(errno))
|
|
{
|
|
}
|
|
|
|
|
|
string absPath(string path, string dir)
|
|
{
|
|
if (path[0] != '/') {
|
|
if (dir == "") {
|
|
char buf[PATH_MAX];
|
|
if (!getcwd(buf, sizeof(buf)))
|
|
throw SysError("cannot get cwd");
|
|
dir = buf;
|
|
}
|
|
path = dir + "/" + path;
|
|
}
|
|
return canonPath(path);
|
|
}
|
|
|
|
|
|
string canonPath(const string & path)
|
|
{
|
|
string s;
|
|
|
|
if (path[0] != '/')
|
|
throw Error(format("not an absolute path: `%1%'") % path);
|
|
|
|
string::const_iterator i = path.begin(), end = path.end();
|
|
|
|
while (1) {
|
|
|
|
/* Skip slashes. */
|
|
while (i != end && *i == '/') i++;
|
|
if (i == end) break;
|
|
|
|
/* Ignore `.'. */
|
|
if (*i == '.' && (i + 1 == end || i[1] == '/'))
|
|
i++;
|
|
|
|
/* If `..', delete the last component. */
|
|
else if (*i == '.' && i + 1 < end && i[1] == '.' &&
|
|
(i + 2 == end || i[2] == '/'))
|
|
{
|
|
if (!s.empty()) s.erase(s.rfind('/'));
|
|
i += 2;
|
|
}
|
|
|
|
/* Normal component; copy it. */
|
|
else {
|
|
s += '/';
|
|
while (i != end && *i != '/') s += *i++;
|
|
}
|
|
}
|
|
|
|
return s.empty() ? "/" : s;
|
|
}
|
|
|
|
|
|
string dirOf(string path)
|
|
{
|
|
unsigned int pos = path.rfind('/');
|
|
if (pos == string::npos)
|
|
throw Error(format("invalid file name: %1%") % path);
|
|
return string(path, 0, pos);
|
|
}
|
|
|
|
|
|
string baseNameOf(string path)
|
|
{
|
|
unsigned int pos = path.rfind('/');
|
|
if (pos == string::npos)
|
|
throw Error(format("invalid file name %1% ") % path);
|
|
return string(path, pos + 1);
|
|
}
|
|
|
|
|
|
bool pathExists(const string & path)
|
|
{
|
|
int res;
|
|
struct stat st;
|
|
res = stat(path.c_str(), &st);
|
|
if (!res) return true;
|
|
if (errno != ENOENT)
|
|
throw SysError(format("getting status of %1%") % path);
|
|
return false;
|
|
}
|
|
|
|
|
|
void deletePath(string path)
|
|
{
|
|
struct stat st;
|
|
if (lstat(path.c_str(), &st))
|
|
throw SysError(format("getting attributes of path %1%") % path);
|
|
|
|
if (S_ISDIR(st.st_mode)) {
|
|
DIR * dir = opendir(path.c_str());
|
|
|
|
struct dirent * dirent;
|
|
while (errno = 0, dirent = readdir(dir)) {
|
|
string name = dirent->d_name;
|
|
if (name == "." || name == "..") continue;
|
|
deletePath(path + "/" + name);
|
|
}
|
|
|
|
closedir(dir); /* !!! close on exception */
|
|
}
|
|
|
|
if (remove(path.c_str()) == -1)
|
|
throw SysError(format("cannot unlink %1%") % path);
|
|
}
|
|
|
|
|
|
static int nestingLevel = 0;
|
|
|
|
|
|
Nest::Nest(bool nest)
|
|
{
|
|
this->nest = nest;
|
|
if (nest) nestingLevel++;
|
|
}
|
|
|
|
|
|
Nest::~Nest()
|
|
{
|
|
if (nest) nestingLevel--;
|
|
}
|
|
|
|
|
|
void msg(const format & f)
|
|
{
|
|
string spaces;
|
|
for (int i = 0; i < nestingLevel; i++)
|
|
spaces += "| ";
|
|
cerr << format("%1%%2%\n") % spaces % f.str();
|
|
}
|
|
|
|
|
|
void debug(const format & f)
|
|
{
|
|
msg(format("debug: %1%") % f.str());
|
|
}
|
|
|
|
|
|
void readFull(int fd, unsigned char * buf, size_t count)
|
|
{
|
|
while (count) {
|
|
ssize_t res = read(fd, (char *) buf, count);
|
|
if (res == -1) throw SysError("reading from file");
|
|
if (res == 0) throw Error("unexpected end-of-file");
|
|
count -= res;
|
|
buf += res;
|
|
}
|
|
}
|
|
|
|
|
|
void writeFull(int fd, const unsigned char * buf, size_t count)
|
|
{
|
|
while (count) {
|
|
ssize_t res = write(fd, (char *) buf, count);
|
|
if (res == -1) throw SysError("writing to file");
|
|
count -= res;
|
|
buf += res;
|
|
}
|
|
}
|