2014-07-30 19:10:28 +02:00
|
|
|
/*
|
2015-01-01 19:48:13 +01:00
|
|
|
* Copyright (C) 2003-2015 The Music Player Daemon Project
|
2014-07-30 19:10:28 +02:00
|
|
|
* http://www.musicpd.org
|
|
|
|
*
|
|
|
|
* 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; either version 2 of the License, or
|
|
|
|
* (at your option) any later version.
|
|
|
|
*
|
|
|
|
* 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.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef MPD_FILE_OUTPUT_STREAM_HXX
|
|
|
|
#define MPD_FILE_OUTPUT_STREAM_HXX
|
|
|
|
|
|
|
|
#include "check.h"
|
|
|
|
#include "OutputStream.hxx"
|
|
|
|
#include "fs/AllocatedPath.hxx"
|
2014-08-30 00:46:52 +02:00
|
|
|
#include "Compiler.h"
|
2014-07-30 19:10:28 +02:00
|
|
|
|
2015-03-03 22:18:38 +01:00
|
|
|
#ifndef WIN32
|
|
|
|
#include "system/FileDescriptor.hxx"
|
|
|
|
#endif
|
|
|
|
|
2014-07-30 19:10:28 +02:00
|
|
|
#include <assert.h>
|
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
#include <windows.h>
|
|
|
|
#endif
|
|
|
|
|
|
|
|
class Path;
|
|
|
|
|
2015-03-23 22:35:43 +01:00
|
|
|
class BaseFileOutputStream : public OutputStream {
|
2015-03-23 22:35:56 +01:00
|
|
|
const AllocatedPath path;
|
2014-07-30 19:10:28 +02:00
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
HANDLE handle;
|
|
|
|
#else
|
2015-03-03 22:18:38 +01:00
|
|
|
FileDescriptor fd;
|
2014-07-30 19:10:28 +02:00
|
|
|
#endif
|
|
|
|
|
2015-03-23 22:35:43 +01:00
|
|
|
protected:
|
|
|
|
#ifdef WIN32
|
|
|
|
template<typename P>
|
|
|
|
BaseFileOutputStream(P &&_path)
|
|
|
|
:path(std::forward<P>(_path)),
|
|
|
|
handle(INVALID_HANDLE_VALUE) {}
|
|
|
|
#else
|
|
|
|
template<typename P>
|
|
|
|
BaseFileOutputStream(P &&_path)
|
|
|
|
:path(std::forward<P>(_path)),
|
|
|
|
fd(FileDescriptor::Undefined()) {}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
~BaseFileOutputStream() {
|
|
|
|
assert(!IsDefined());
|
|
|
|
}
|
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
void SetHandle(HANDLE _handle) {
|
|
|
|
assert(!IsDefined());
|
|
|
|
|
|
|
|
handle = _handle;
|
|
|
|
|
|
|
|
assert(IsDefined());
|
|
|
|
}
|
|
|
|
#else
|
|
|
|
FileDescriptor &SetFD() {
|
|
|
|
assert(!IsDefined());
|
|
|
|
|
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
|
|
|
|
const FileDescriptor &GetFD() const {
|
|
|
|
return fd;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
bool Close() {
|
|
|
|
assert(IsDefined());
|
|
|
|
|
|
|
|
#ifdef WIN32
|
|
|
|
CloseHandle(handle);
|
|
|
|
handle = INVALID_HANDLE_VALUE;
|
|
|
|
return true;
|
|
|
|
#else
|
|
|
|
return fd.Close();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
2015-03-24 21:51:34 +01:00
|
|
|
#ifdef WIN32
|
|
|
|
bool SeekEOF() {
|
2015-03-24 20:38:29 +01:00
|
|
|
return SetFilePointer(handle, 0, nullptr,
|
|
|
|
FILE_END) != 0xffffffff;
|
2015-03-24 21:51:34 +01:00
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
2015-03-23 22:35:43 +01:00
|
|
|
public:
|
|
|
|
bool IsDefined() const {
|
|
|
|
#ifdef WIN32
|
|
|
|
return handle != INVALID_HANDLE_VALUE;
|
|
|
|
#else
|
|
|
|
return fd.IsDefined();
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
|
|
|
|
Path GetPath() const {
|
|
|
|
return path;
|
|
|
|
}
|
|
|
|
|
2015-03-24 21:46:01 +01:00
|
|
|
gcc_pure
|
|
|
|
uint64_t Tell() const;
|
|
|
|
|
2015-03-23 22:35:43 +01:00
|
|
|
/* virtual methods from class OutputStream */
|
|
|
|
bool Write(const void *data, size_t size, Error &error) override;
|
|
|
|
};
|
|
|
|
|
|
|
|
class FileOutputStream final : public BaseFileOutputStream {
|
2015-01-05 20:24:59 +01:00
|
|
|
#ifdef HAVE_LINKAT
|
|
|
|
/**
|
|
|
|
* Was O_TMPFILE used? If yes, then linkat() must be used to
|
|
|
|
* create a link to this file.
|
|
|
|
*/
|
|
|
|
bool is_tmpfile;
|
|
|
|
#endif
|
|
|
|
|
2014-07-30 19:10:28 +02:00
|
|
|
public:
|
|
|
|
FileOutputStream(Path _path, Error &error);
|
|
|
|
|
|
|
|
~FileOutputStream() {
|
|
|
|
if (IsDefined())
|
|
|
|
Cancel();
|
|
|
|
}
|
|
|
|
|
2015-01-14 19:43:19 +01:00
|
|
|
static FileOutputStream *Create(Path path, Error &error);
|
2014-07-30 19:10:28 +02:00
|
|
|
|
|
|
|
bool Commit(Error &error);
|
|
|
|
void Cancel();
|
|
|
|
};
|
|
|
|
|
2015-03-23 22:42:07 +01:00
|
|
|
class AppendFileOutputStream final : public BaseFileOutputStream {
|
|
|
|
public:
|
|
|
|
AppendFileOutputStream(Path _path, Error &error);
|
|
|
|
|
|
|
|
~AppendFileOutputStream() {
|
|
|
|
if (IsDefined())
|
|
|
|
Close();
|
|
|
|
}
|
|
|
|
|
|
|
|
bool Commit(Error &error);
|
|
|
|
};
|
|
|
|
|
2014-07-30 19:10:28 +02:00
|
|
|
#endif
|