X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/a67e67b1f38ddb2a01c1d5f09e32d373344e4d5e..8ef240801bf66e543cdea9df3b32fc09cebcc7e5:/lib/base/filepush.cpp

diff --git a/lib/base/filepush.cpp b/lib/base/filepush.cpp
index 17b767f8..8d116559 100644
--- a/lib/base/filepush.cpp
+++ b/lib/base/filepush.cpp
@@ -2,11 +2,17 @@
 #include <lib/base/eerror.h>
 #include <errno.h>
 #include <fcntl.h>
+#include <sys/ioctl.h>
 
-eFilePushThread::eFilePushThread()
+#define PVR_COMMIT 1
+
+eFilePushThread::eFilePushThread(): m_messagepump(eApp, 0)
 {
 	m_stop = 0;
-	m_buf_start = m_buf_end = 0;
+	m_sg = 0;
+	flush();
+	enablePVRCommit(0);
+	CONNECT(m_messagepump.recv_msg, eFilePushThread::recvEvent);
 }
 
 static void signal_handler(int x)
@@ -15,7 +21,15 @@ static void signal_handler(int x)
 
 void eFilePushThread::thread()
 {
-	off_t dest_pos = 0;
+	off_t dest_pos = 0, source_pos = 0;
+	size_t bytes_read = 0;
+	
+	off_t current_span_offset = 0;
+	size_t current_span_remaining = 0;
+	
+	size_t written_since_last_sync = 0;
+	
+	int already_empty = 0;
 	eDebug("FILEPUSH THREAD START");
 		// this is a race. FIXME.
 	
@@ -26,6 +40,8 @@ void eFilePushThread::thread()
 	sigaction(SIGUSR1, &act, 0);
 	
 	dest_pos = lseek(m_fd_dest, 0, SEEK_CUR);
+	source_pos = m_raw_source.lseek(0, SEEK_CUR);
+	
 		/* m_stop must be evaluated after each syscall. */
 	while (!m_stop)
 	{
@@ -33,44 +49,99 @@ void eFilePushThread::thread()
 		if (m_buf_start != m_buf_end)
 		{
 				// TODO: take care of boundaries.
+			filterRecordData(m_buffer + m_buf_start, m_buf_end - m_buf_start);
 			int w = write(m_fd_dest, m_buffer + m_buf_start, m_buf_end - m_buf_start);
-			eDebug("wrote %d bytes", w);
+//			eDebug("wrote %d bytes", w);
 			if (w <= 0)
 			{
 				if (errno == -EINTR)
 					continue;
-				eDebug("eFilePushThread *write error* - not yet handled");
+				break;
 				// ... we would stop the thread
 			}
 
 				/* this should flush all written pages to disk. */
 			posix_fadvise(m_fd_dest, dest_pos, w, POSIX_FADV_DONTNEED);
-
+			
 			dest_pos += w;
+			written_since_last_sync += w;
+			
+			if (written_since_last_sync >= 512*1024)
+			{
+				fdatasync(m_fd_dest);
+				written_since_last_sync = 0;
+			}
+
 //			printf("FILEPUSH: wrote %d bytes\n", w);
 			m_buf_start += w;
 			continue;
 		}
-			
+
 			/* now fill our buffer. */
+			
+		if (m_sg && !current_span_remaining)
+		{
+			m_sg->getNextSourceSpan(source_pos, bytes_read, current_span_offset, current_span_remaining);
+
+			if (source_pos != current_span_offset)
+				source_pos = m_raw_source.lseek(current_span_offset, SEEK_SET);
+			bytes_read = 0;
+		}
+		
+		size_t maxread = sizeof(m_buffer);
+		
+			/* if we have a source span, don't read past the end */
+		if (m_sg && maxread > current_span_remaining)
+			maxread = current_span_remaining;
+
 		m_buf_start = 0;
-		m_buf_end = read(m_fd_source, m_buffer, sizeof(m_buffer));
+		m_buf_end = 0;
+		
+		if (maxread)
+			m_buf_end = m_raw_source.read(m_buffer, maxread);
+
 		if (m_buf_end < 0)
 		{
 			m_buf_end = 0;
 			if (errno == EINTR)
 				continue;
-			eDebug("eFilePushThread *read error* - not yet handled");
+			if (errno == EOVERFLOW)
+			{
+				eWarning("OVERFLOW while recording");
+				continue;
+			}
+			eDebug("eFilePushThread *read error* (%m) - not yet handled");
 		}
 		if (m_buf_end == 0)
 		{
+				/* on EOF, try COMMITting once. */
+			if (m_send_pvr_commit && !already_empty)
+			{
+				eDebug("sending PVR commit");
+				already_empty = 1;
+				if (::ioctl(m_fd_dest, PVR_COMMIT) == EINTR)
+					continue;
+				eDebug("commit done");
+						/* well check again */
+				continue;
+			}
+			sendEvent(evtEOF);
+#if 0
 			eDebug("FILEPUSH: end-of-file! (currently unhandled)");
-			if (!lseek(m_fd_source, 0, SEEK_SET))
+			if (!m_raw_source.lseek(0, SEEK_SET))
 			{
 				eDebug("(looping)");
 				continue;
 			}
+#endif
 			break;
+		} else
+		{
+			source_pos += m_buf_end;
+			bytes_read += m_buf_end;
+			if (m_sg)
+				current_span_remaining -= m_buf_end;
+			already_empty = 0;
 		}
 //		printf("FILEPUSH: read %d bytes\n", m_buf_end);
 	}
@@ -80,13 +151,24 @@ void eFilePushThread::thread()
 
 void eFilePushThread::start(int fd_source, int fd_dest)
 {
-	m_fd_source = fd_source;
+	m_raw_source.setfd(fd_source);
 	m_fd_dest = fd_dest;
 	resume();
 }
 
+int eFilePushThread::start(const char *filename, int fd_dest)
+{
+	if (m_raw_source.open(filename) < 0)
+		return -1;
+	m_fd_dest = fd_dest;
+	resume();
+	return 0;
+}
+
 void eFilePushThread::stop()
 {
+	if (!thread_running()) /* FIXME: races */
+		return;
 	m_stop = 1;
 	sendSignal(SIGUSR1);
 	kill();
@@ -97,9 +179,9 @@ void eFilePushThread::pause()
 	stop();
 }
 
-void eFilePushThread::seek(off_t where)
+void eFilePushThread::seek(int whence, off_t where)
 {
-	::lseek(m_fd_source, where, SEEK_SET);
+	m_raw_source.lseek(where, whence);
 }
 
 void eFilePushThread::resume()
@@ -107,3 +189,34 @@ void eFilePushThread::resume()
 	m_stop = 0;
 	run();
 }
+
+void eFilePushThread::flush()
+{
+	m_buf_start = m_buf_end = 0;
+}
+
+void eFilePushThread::enablePVRCommit(int s)
+{
+	m_send_pvr_commit = s;
+}
+
+void eFilePushThread::setScatterGather(iFilePushScatterGather *sg)
+{
+	m_sg = sg;
+}
+
+void eFilePushThread::sendEvent(int evt)
+{
+	m_messagepump.send(evt);
+}
+
+void eFilePushThread::recvEvent(const int &evt)
+{
+	m_event(evt);
+}
+
+void eFilePushThread::filterRecordData(const unsigned char *data, int len)
+{
+	/* do nothing */
+}
+