/*
* Copyright (C) 2003-2014 The Music Player Daemon Project
* 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.
*/
#include "config.h"
#include "ThreadInputStream.hxx"
#include "thread/Name.hxx"
#include "util/CircularBuffer.hxx"
#include "util/HugeAllocator.hxx"
#include <assert.h>
#include <string.h>
ThreadInputStream::~ThreadInputStream()
{
Lock();
close = true;
wake_cond.signal();
Unlock();
Cancel();
thread.Join();
if (buffer != nullptr) {
buffer->Clear();
HugeFree(buffer->Write().data, buffer_size);
delete buffer;
}
}
InputStream *
ThreadInputStream::Start(Error &error)
{
assert(buffer == nullptr);
void *p = HugeAllocate(buffer_size);
if (p == nullptr) {
error.SetErrno();
return nullptr;
}
buffer = new CircularBuffer<uint8_t>((uint8_t *)p, buffer_size);
if (!thread.Start(ThreadFunc, this, error))
return nullptr;
return this;
}
inline void
ThreadInputStream::ThreadFunc()
{
FormatThreadName("input:%s", plugin);
Lock();
if (!Open(postponed_error)) {
cond.broadcast();
Unlock();
return;
}
/* we're ready, tell it to our client */
SetReady();
while (!close) {
assert(!postponed_error.IsDefined());
auto w = buffer->Write();
if (w.IsEmpty()) {
wake_cond.wait(mutex);
} else {
Unlock();
Error error;
size_t nbytes = Read(w.data, w.size, error);
Lock();
cond.broadcast();
if (nbytes == 0) {
eof = true;
postponed_error = std::move(error);
break;
}
buffer->Append(nbytes);
}
}
Unlock();
Close();
}
void
ThreadInputStream::ThreadFunc(void *ctx)
{
ThreadInputStream &tis = *(ThreadInputStream *)ctx;
tis.ThreadFunc();
}
bool
ThreadInputStream::Check(Error &error)
{
if (postponed_error.IsDefined()) {
error = std::move(postponed_error);
return false;
}
return true;
}
bool
ThreadInputStream::IsAvailable()
{
return !buffer->IsEmpty() || eof || postponed_error.IsDefined();
}
inline size_t
ThreadInputStream::Read(void *ptr, size_t read_size, Error &error)
{
while (true) {
if (postponed_error.IsDefined()) {
error = std::move(postponed_error);
return 0;
}
auto r = buffer->Read();
if (!r.IsEmpty()) {
size_t nbytes = std::min(read_size, r.size);
memcpy(ptr, r.data, nbytes);
buffer->Consume(nbytes);
wake_cond.broadcast();
offset += nbytes;
return nbytes;
}
if (eof)
return 0;
cond.wait(mutex);
}
}
bool
ThreadInputStream::IsEOF()
{
return eof;
}