1 files changed, 302 insertions, 0 deletions
diff --git a/src/vdr-plugin/download.c b/src/vdr-plugin/download.c
new file mode 100644
index 0000000..4b7a971
--- /dev/null
+++ b/src/vdr-plugin/download.c
@@ -0,0 +1,302 @@
+/*
+ * download.c: Web video plugin for the Video Disk Recorder
+ *
+ * See the README file for copyright information and how to reach the author.
+ *
+ * $Id$
+ */
+
+#include <errno.h>
+#include <sys/select.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <vdr/tools.h>
+#include "download.h"
+#include "common.h"
+
+static void diff_timeval(struct timeval *a, struct timeval *b,
+                         struct timeval *result) {
+  long usec_diff = a->tv_usec - b->tv_usec;
+  result->tv_sec = a->tv_sec - b->tv_sec;
+
+  while (usec_diff < 0) {
+    usec_diff += 1000000;
+    result->tv_sec -= 1;
+  }
+
+  result->tv_usec = usec_diff;
+}
+
+// --- cWebviThread --------------------------------------------------------
+
+cWebviThread::cWebviThread() {
+  int pipefd[2];
+
+  if (pipe(pipefd) == -1)
+    LOG_ERROR_STR("new request pipe");
+  newreqread = pipefd[0];
+  newreqwrite = pipefd[1];
+  //fcntl(newreqread, F_SETFL, O_NONBLOCK);
+  //fcntl(newreqwrite, F_SETFL, O_NONBLOCK);
+  timerActive = false;
+
+  webvi = webvi_initialize_context();
+  if (webvi != 0) {
+    webvi_set_config(webvi, WEBVI_CONFIG_TIMEOUT_DATA, this);
+    webvi_set_config(webvi, WEBVI_CONFIG_TIMEOUT_CALLBACK, UpdateTimeout);
+  }
+}
+
+cWebviThread::~cWebviThread() {
+  int numactive = activeRequestList.Size();
+  for (int i=0; i<activeRequestList.Size(); i++)
+    delete activeRequestList[i];
+  activeRequestList.Clear();
+
+  for (int i=0; i<finishedRequestList.Size(); i++) {
+    delete finishedRequestList[i];
+  }
+  finishedRequestList.Clear();
+
+  webvi_cleanup_context(webvi);
+
+  if (numactive > 0) {
+    esyslog("%d requests failed to complete", numactive);
+  }
+}
+
+void cWebviThread::UpdateTimeout(long timeout, void *instance) {
+  cWebviThread *self = (cWebviThread *)instance;
+  if (!self)
+    return;
+
+  if (timeout < 0) {
+    self->timerActive = false;
+  } else {
+    struct timeval now;
+    long alrm;
+    gettimeofday(&now, NULL);
+    alrm = timeout + now.tv_usec/1000;
+    self->timer.tv_sec = now.tv_sec + alrm/1000;
+    self->timer.tv_usec = (alrm % 1000) * 1000;
+    self->timerActive = true;
+  }
+}
+
+cWebviThread &cWebviThread::Instance() {
+  static cWebviThread instance;
+  
+  return instance;
+}
+
+void cWebviThread::SetTemplatePath(const char *path) {
+  if (webvi != 0 && path)
+    webvi_set_config(webvi, WEBVI_CONFIG_TEMPLATE_PATH, path);
+}
+
+void cWebviThread::MoveToFinishedList(cMenuRequest *req) {
+  // Move the request from the activeList to finishedList.
+  requestMutex.Lock();
+  for (int i=0; i<activeRequestList.Size(); i++) {
+    if (activeRequestList[i] == req) {
+      activeRequestList.Remove(i);
+      break;
+    }
+  }
+  finishedRequestList.Append(req);
+
+  requestMutex.Unlock();
+}
+
+void cWebviThread::ActivateNewRequest() {
+  // Move requests from newRequestList to activeRequestList and start
+  // them.
+  requestMutex.Lock();
+  for (int i=0; i<newRequestList.Size(); i++) {
+    cMenuRequest *req = newRequestList[i];
+    if (req->IsAborted()) {
+      // The request has been aborted even before we got a chance to
+      // start it.
+      MoveToFinishedList(req);
+    } else {
+      if (!req->Start(webvi)) {
+        error("Request failed to start");
+        req->RequestDone(-1, "Request failed to start");
+        MoveToFinishedList(req);
+      } else {
+        activeRequestList.Append(req);
+      }
+    }
+  }
+
+  newRequestList.Clear();
+  requestMutex.Unlock();
+}
+
+void cWebviThread::StopFinishedRequests() {
+  // Check if some requests have finished, and move them to
+  // finishedRequestList.
+  int msg_remaining;
+  WebviMsg *donemsg;
+  cMenuRequest *req;
+
+  do {
+    donemsg = webvi_get_message(webvi, &msg_remaining);
+
+    if (donemsg && donemsg->msg == WEBVIMSG_DONE) {
+      requestMutex.Lock();
+      req = activeRequestList.FindByHandle(donemsg->handle);
+      if (req) {
+        req->RequestDone(donemsg->status_code, donemsg->data);
+        if (req->IsFinished())
+          MoveToFinishedList(req);
+      }
+      requestMutex.Unlock();
+    }
+  } while (msg_remaining > 0);
+}
+
+void cWebviThread::Stop() {
+  // The thread may be sleeping, wake it up first.
+  TEMP_FAILURE_RETRY(write(newreqwrite, "S", 1));
+  Cancel(5);
+}
+
+void cWebviThread::Action(void) {
+  fd_set readfds, writefds, excfds;
+  int maxfd, s;
+  struct timeval timeout, now;
+  long running_handles;
+  bool check_done = false;
+  bool has_request_files = false;
+
+  if (webvi == 0) {
+    error("Failed to get libwebvi context");
+    return;
+  }
+
+  while (Running()) {
+    FD_ZERO(&readfds);
+    FD_ZERO(&writefds);
+    FD_ZERO(&excfds);
+    webvi_fdset(webvi, &readfds, &writefds, &excfds, &maxfd);
+    FD_SET(newreqread, &readfds);
+    if (newreqread > maxfd)
+      maxfd = newreqread;
+
+    has_request_files = false;
+    requestMutex.Lock();
+    for (int i=0; i<activeRequestList.Size(); i++) {
+      int fd = activeRequestList[i]->File();
+      if (fd != -1) {
+        FD_SET(fd, &readfds);
+        if (fd > maxfd)
+          maxfd = fd;
+        has_request_files = true;
+      }
+    }
+    requestMutex.Unlock();
+
+    if (!timerActive) {
+      timeout.tv_sec = 60;
+      timeout.tv_usec = 0;
+    } else {
+      gettimeofday(&now, NULL);
+      diff_timeval(&timer, &now, &timeout);
+      if (timeout.tv_sec < 0 || timeout.tv_usec < 0) {
+        timeout.tv_sec = 0;
+        timeout.tv_usec = 0;
+      }
+    }
+
+    s = TEMP_FAILURE_RETRY(select(maxfd+1, &readfds, &writefds, NULL,
+                                  &timeout));
+    if (s == -1) {
+      // select error
+      LOG_ERROR_STR("select() error in webvideo downloader thread:");
+      Cancel(-1);
+
+    } else if (s == 0) {
+      // timeout
+      timerActive = false;
+      webvi_perform(webvi, WEBVI_SELECT_TIMEOUT, WEBVI_SELECT_CHECK, &running_handles);
+      check_done = true;
+
+    } else {
+      for (int fd=0; fd<=maxfd; fd++) {
+        if (FD_ISSET(fd, &readfds)) {
+          if (fd == newreqread) {
+            char tmpbuf[8];
+            int n = read(fd, tmpbuf, 8);
+            if (n > 0 && memchr(tmpbuf, 'S', n))
+              Cancel(-1);
+            ActivateNewRequest();
+          } else {
+            cMenuRequest *match = NULL;
+
+            if (has_request_files) {
+              requestMutex.Lock();
+              for (int i=0; i<activeRequestList.Size(); i++) {
+                if (fd == activeRequestList[i]->File()) {
+                  match = activeRequestList[i];
+                  break;
+                }
+              }
+              requestMutex.Unlock();
+
+              // call Read() after releasing the mutex
+              if (match) {
+                match->Read();
+                if (match->IsFinished())
+                  MoveToFinishedList(match);
+              }
+            }
+
+            if (!match) {
+              webvi_perform(webvi, fd, WEBVI_SELECT_READ, &running_handles);
+              check_done = true;
+            }
+          }
+        }
+        if (FD_ISSET(fd, &writefds))
+          webvi_perform(webvi, fd, WEBVI_SELECT_WRITE, &running_handles);
+        if (FD_ISSET(fd, &excfds))
+          webvi_perform(webvi, fd, WEBVI_SELECT_EXCEPTION, &running_handles);
+      }
+    }
+
+    if (check_done) {
+      StopFinishedRequests();
+      check_done = false;
+    }
+  }
+}
+
+void cWebviThread::AddRequest(cMenuRequest *req) {
+  requestMutex.Lock();
+  newRequestList.Append(req);
+  requestMutex.Unlock();
+
+  int s = TEMP_FAILURE_RETRY(write(newreqwrite, "*", 1));
+  if (s == -1)
+    LOG_ERROR_STR("Failed to signal new webvideo request");
+}
+
+cMenuRequest *cWebviThread::GetFinishedRequest() {
+  cMenuRequest *res = NULL;
+  requestMutex.Lock();
+  if (finishedRequestList.Size() > 0) {
+    res = finishedRequestList[finishedRequestList.Size()-1];
+    finishedRequestList.Remove(finishedRequestList.Size()-1);
+  }
+  requestMutex.Unlock();
+
+  return res;
+}
+
+int cWebviThread::GetUnfinishedCount() {
+  if (!Running())
+    return 0;
+  else
+    return activeRequestList.Size();
+}