mirrors/aria2
1
0
Fork 0
mirror of https://github.com/aria2/aria2.git synced 2025-04-06 05:57:36 +03:00
Code Issues Projects Releases Packages Wiki Activity

Refactored file open code

Browse source
This commit is contained in:
Tatsuhiro Tsujikawa 2012-11-28 21:46:44 +09:00
parent 16e25ef4e1
commit 506a9358e1
1 changed files with 21 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

46
src/AbstractDiskWriter.cc
View file

@ -104,6 +104,24 @@ void AbstractDiskWriter::closeFile()
}
}
namespace {
int openFileWithFlags(const std::string& filename, int flags,
error_code::Value errCode)
{
int fd;
while((fd = a2open(utf8ToWChar(filename).c_str(), flags, OPEN_MODE)) == -1
&& errno == EINTR);
if(fd < 0) {
int errNum = errno;
throw DL_ABORT_EX3(errNum, fmt(EX_FILE_OPEN,
filename.c_str(),
util::safeStrerror(errNum).c_str()),
errCode);
}
return fd;
}
} // namespace
void AbstractDiskWriter::openExistingFile(int64_t totalLength)
{
int flags = O_BINARY;
@ -112,37 +130,15 @@ void AbstractDiskWriter::openExistingFile(int64_t totalLength)
} else {
flags |= O_RDWR;
}
while((fd_ = a2open(utf8ToWChar(filename_).c_str(),
flags, OPEN_MODE)) == -1 &&
errno == EINTR);
if(fd_ < 0) {
int errNum = errno;
throw DL_ABORT_EX3
(errNum,
fmt(EX_FILE_OPEN,
filename_.c_str(),
util::safeStrerror(errNum).c_str()),
error_code::FILE_OPEN_ERROR);
}
fd_ = openFileWithFlags(filename_, flags, error_code::FILE_OPEN_ERROR);
}
void AbstractDiskWriter::createFile(int addFlags)
{
assert(!filename_.empty());
util::mkdirs(File(filename_).getDirname());
while((fd_ = a2open(utf8ToWChar(filename_).c_str(),
O_CREAT|O_RDWR|O_TRUNC|O_BINARY|addFlags,
OPEN_MODE)) == -1 && errno == EINTR);
if(fd_ < 0) {
int errNum = errno;
throw DL_ABORT_EX3
(errNum,
fmt(EX_FILE_OPEN,
filename_.c_str(),
util::safeStrerror(errNum).c_str()),
error_code::FILE_CREATE_ERROR);
}
fd_ = openFileWithFlags(filename_, O_CREAT|O_RDWR|O_TRUNC|O_BINARY|addFlags,
error_code::FILE_CREATE_ERROR);
}
ssize_t AbstractDiskWriter::writeDataInternal(const unsigned char* data,