* Added support for SCGI rpc calls.

git-svn-id: svn://rakshasa.no/libtorrent/trunk/rtorrent@905 e378c898-3ddf-0310-93e7-cc216c733640

src/rpc/Makefile.am

@@ -3,6 +3,10 @@ noinst_LIBRARIES = libsub_rpc.a
 libsub_rpc_a_SOURCES = \
 	fast_cgi.cc \
 	fast_cgi.h \
+	scgi.cc \
+	scgi.h \
+	scgi_task.cc \
+	scgi_task.h \
 	xmlrpc.h \
 	xmlrpc.cc
 

src/rpc/fast_cgi.cc

@@ -113,6 +113,11 @@ FastCgi::event_read() {
     return;
   }
 
+//   int flags = fcntl(((FCGX_Request*)m_request)->ipcFd, F_GETFL);
+
+//   if (fcntl(((FCGX_Request*)m_request)->ipcFd, F_SETFL, flags & ~O_NONBLOCK) != 0)
+//     throw torrent::internal_error("FastCgi::event_read() could not set socket flags.");
+
   int length;
   char* endPtr;
   char* buffer = NULL;

src/rpc/scgi.cc

@@ -0,0 +1,133 @@
+// rTorrent - BitTorrent client
+// Copyright (C) 2005-2006, Jari Sundell
+//
+// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// In addition, as a special exception, the copyright holders give
+// permission to link the code of portions of this program with the
+// OpenSSL library under certain conditions as described in each
+// individual source file, and distribute linked combinations
+// including the two.
+//
+// You must obey the GNU General Public License in all respects for
+// all of the code used other than OpenSSL.  If you modify file(s)
+// with this exception, you may extend this exception to your version
+// of the file(s), but you are not obligated to do so.  If you do not
+// wish to do so, delete this exception statement from your version.
+// If you delete this exception statement from all source files in the
+// program, then also delete it here.
+//
+// Contact:  Jari Sundell <jaris@ifi.uio.no>
+//
+//           Skomakerveien 33
+//           3185 Skoppum, NORWAY
+
+#include "config.h"
+
+#include <torrent/connection_manager.h>
+#include <torrent/poll.h>
+#include <torrent/torrent.h>
+#include <rak/socket_address.h>
+#include <torrent/exceptions.h>
+
+#include "utils/socket_fd.h"
+
+#include "control.h"
+#include "globals.h"
+#include "scgi.h"
+
+namespace rpc {
+
+SCgi::~SCgi() {
+  if (!get_fd().is_valid())
+    return;
+
+  control->poll()->remove_read(this);
+  control->poll()->remove_error(this);
+  control->poll()->close(this);
+
+  get_fd().close();
+  get_fd().clear();
+}
+
+bool
+SCgi::open(uint16_t port) {
+  if (!get_fd().open_stream())
+    throw torrent::resource_error("Could not allocate socket for listening.");
+
+  try {
+    rak::socket_address sa;
+    sa.sa_inet()->clear();
+    sa.sa_inet()->set_port(port);
+
+    if (!get_fd().set_nonblock() ||
+        !get_fd().set_reuse_address(true) ||
+        !get_fd().bind(sa) ||
+        !get_fd().listen(10))
+      throw torrent::resource_error("Could not allocate socket for listening.");
+
+    torrent::connection_manager()->inc_socket_count();
+
+    control->poll()->open(this);
+    control->poll()->insert_read(this);
+    control->poll()->insert_error(this);
+
+  } catch (torrent::resource_error& e) {
+    get_fd().close();
+    get_fd().clear();
+
+    throw e;
+  }
+
+  return true;
+}
+
+void
+SCgi::event_read() {
+  rak::socket_address sa;
+  utils::SocketFd fd;
+
+  while ((fd = get_fd().accept(&sa)).is_valid()) {
+    SCgiTask* task = std::find_if(m_task, m_task + 10, std::mem_fun_ref(&SCgiTask::is_available));
+
+    if (task == task + 10) {
+      // Ergh... just closing for now.
+      fd.close();
+      continue;
+    }
+    
+    task->open(this, fd.get_fd());
+  }
+}
+
+void
+SCgi::event_write() {
+  throw torrent::internal_error("Listener does not support write().");
+}
+
+void
+SCgi::event_error() {
+  throw torrent::internal_error("SCGI listener port received an error event.");
+}
+
+bool
+SCgi::receive_call(SCgiTask* task, const char* buffer, uint32_t length) {
+  slot_write slotWrite;
+  slotWrite.set(rak::mem_fn(task, &SCgiTask::receive_write));
+
+  return m_slotProcess(buffer, length, slotWrite);
+}
+
+}

src/rpc/scgi.h

@@ -0,0 +1,84 @@
+// rTorrent - BitTorrent client
+// Copyright (C) 2005-2006, Jari Sundell
+//
+// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// In addition, as a special exception, the copyright holders give
+// permission to link the code of portions of this program with the
+// OpenSSL library under certain conditions as described in each
+// individual source file, and distribute linked combinations
+// including the two.
+//
+// You must obey the GNU General Public License in all respects for
+// all of the code used other than OpenSSL.  If you modify file(s)
+// with this exception, you may extend this exception to your version
+// of the file(s), but you are not obligated to do so.  If you do not
+// wish to do so, delete this exception statement from your version.
+// If you delete this exception statement from all source files in the
+// program, then also delete it here.
+//
+// Contact:  Jari Sundell <jaris@ifi.uio.no>
+//
+//           Skomakerveien 33
+//           3185 Skoppum, NORWAY
+
+#ifndef RTORRENT_RPC_SCGI_H
+#define RTORRENT_RPC_SCGI_H
+
+#include <string>
+#include <rak/functional_fun.h>
+#include <torrent/event.h>
+
+#include "scgi_task.h"
+
+namespace utils {
+  class SocketFd;
+}
+
+namespace rpc {
+
+class SCgi : public torrent::Event {
+public:
+  typedef rak::function2<bool, const char*, uint32_t>             slot_write;
+  typedef rak::function3<bool, const char*, uint32_t, slot_write> slot_process;
+
+  SCgi() {}
+  virtual ~SCgi();
+
+  bool                open(uint16_t port);
+
+  const std::string   path() const { return m_path; }
+
+  void                set_slot_process(slot_process::base_type* s) { m_slotProcess.set(s); }
+
+  virtual void        event_read();
+  virtual void        event_write();
+  virtual void        event_error();
+
+  bool                receive_call(SCgiTask* task, const char* buffer, uint32_t length);
+
+  utils::SocketFd&    get_fd()            { return *reinterpret_cast<utils::SocketFd*>(&m_fileDesc); }
+
+private:
+  std::string         m_path;
+
+  slot_process        m_slotProcess;
+
+  SCgiTask            m_task[10];
+};
+
+}
+
+#endif

src/rpc/scgi_task.cc

@@ -0,0 +1,182 @@
+// rTorrent - BitTorrent client
+// Copyright (C) 2005-2006, Jari Sundell
+//
+// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// In addition, as a special exception, the copyright holders give
+// permission to link the code of portions of this program with the
+// OpenSSL library under certain conditions as described in each
+// individual source file, and distribute linked combinations
+// including the two.
+//
+// You must obey the GNU General Public License in all respects for
+// all of the code used other than OpenSSL.  If you modify file(s)
+// with this exception, you may extend this exception to your version
+// of the file(s), but you are not obligated to do so.  If you do not
+// wish to do so, delete this exception statement from your version.
+// If you delete this exception statement from all source files in the
+// program, then also delete it here.
+//
+// Contact:  Jari Sundell <jaris@ifi.uio.no>
+//
+//           Skomakerveien 33
+//           3185 Skoppum, NORWAY
+
+#include "config.h"
+
+#include <rak/error_number.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <torrent/exceptions.h>
+#include <torrent/poll.h>
+
+#include "utils/socket_fd.h"
+
+#include "control.h"
+#include "globals.h"
+#include "scgi.h"
+
+namespace rpc {
+
+void
+SCgiTask::open(SCgi* parent, int fd) {
+  m_parent = parent;
+  m_fileDesc = fd;
+  m_buffer = new char[(m_bufferSize = 2048)];
+  m_position = m_buffer;
+
+  control->poll()->open(this);
+  control->poll()->insert_read(this);
+  control->poll()->insert_error(this);
+}
+
+void
+SCgiTask::close() {
+  if (!get_fd().is_valid())
+    return;
+
+  control->poll()->remove_read(this);
+  control->poll()->remove_write(this);
+  control->poll()->remove_error(this);
+  control->poll()->close(this);
+
+  get_fd().close();
+  get_fd().clear();
+
+  delete [] m_buffer;
+  m_buffer = NULL;
+}
+
+void
+SCgiTask::event_read() {
+  int bytes = ::recv(m_fileDesc, m_position, m_bufferSize - 1 - (m_position - m_buffer), 0);
+
+  if (bytes == -1) {
+    if (!rak::error_number::current().is_blocked_momentary())
+      close();
+
+    return;
+  }
+
+  m_position += bytes;
+  *m_position = '\0';
+
+  // Don't bother caching the parsed values, as we're likely to
+  // receive all the data we need the first time.
+  char* current;
+  int headerSize = strtol(m_buffer, &current, 0);
+
+  if (current == m_buffer || current == m_position)
+    // Need to validate the header size.
+    return;
+
+  if (*current != ':' || headerSize < 17)
+    goto event_read_failed;
+
+  if (std::distance(++current, m_position) < headerSize + 1)
+    return;
+
+  if (std::memcmp(current, "CONTENT_LENGTH", 15) != 0)
+    goto event_read_failed;
+
+  char* contentPos;
+  int contentSize = strtol(current + 15, &contentPos, 0);
+
+  if (*contentPos != '\0' || contentSize <= 0)
+    goto event_read_failed;
+
+  // Start of the data.
+  current += headerSize + 1;
+
+  if (std::distance(current, m_position) < contentSize)
+    return;
+
+  control->poll()->remove_read(this);
+  control->poll()->insert_write(this);
+
+  if (!m_parent->receive_call(this, current, contentSize))
+    close();
+
+  return;
+
+ event_read_failed:
+  throw torrent::internal_error("SCgiTask::event_read() fault not handled.");
+}
+
+void
+SCgiTask::event_write() {
+  int bytes = ::send(m_fileDesc, m_position, m_bufferSize, 0);
+
+  if (bytes == -1) {
+    if (!rak::error_number::current().is_blocked_momentary())
+      close();
+
+    return;
+  }
+
+  m_position += bytes;
+  m_bufferSize -= bytes;
+
+  if (bytes == 0 || m_bufferSize == 0)
+    return close();
+}
+
+void
+SCgiTask::event_error() {
+  close();
+}
+
+bool
+SCgiTask::receive_write(const char* buffer, uint32_t length) {
+  if (length + 44 > m_bufferSize) {
+    delete [] m_buffer;
+    m_buffer = new char[length + 44];
+  }
+
+  // Try writing as much as possible from here, before copying
+  // anything.
+
+  m_position = m_buffer;
+  m_bufferSize = length + 44;
+  
+  std::memcpy(m_buffer, "Status: 200 OK\r\nContent-Type: text/plain\r\n\r\n", 44);
+  std::memcpy(m_buffer + 44, buffer, length);
+
+  event_write();
+
+  return true;
+}
+
+}

src/rpc/scgi_task.h

@@ -0,0 +1,78 @@
+// rTorrent - BitTorrent client
+// Copyright (C) 2005-2006, Jari Sundell
+//
+// 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+//
+// In addition, as a special exception, the copyright holders give
+// permission to link the code of portions of this program with the
+// OpenSSL library under certain conditions as described in each
+// individual source file, and distribute linked combinations
+// including the two.
+//
+// You must obey the GNU General Public License in all respects for
+// all of the code used other than OpenSSL.  If you modify file(s)
+// with this exception, you may extend this exception to your version
+// of the file(s), but you are not obligated to do so.  If you do not
+// wish to do so, delete this exception statement from your version.
+// If you delete this exception statement from all source files in the
+// program, then also delete it here.
+//
+// Contact:  Jari Sundell <jaris@ifi.uio.no>
+//
+//           Skomakerveien 33
+//           3185 Skoppum, NORWAY
+
+#ifndef RTORRENT_RPC_SCGI_TASK_H
+#define RTORRENT_RPC_SCGI_TASK_H
+
+#include <torrent/event.h>
+
+namespace utils {
+  class SocketFd;
+}
+
+namespace rpc {
+
+class SCgi;
+
+class SCgiTask : public torrent::Event {
+public:
+  SCgiTask() { m_fileDesc = -1; }
+
+  bool                is_open() const      { return m_fileDesc != -1; }
+  bool                is_available() const { return m_fileDesc == -1; }
+
+  void                open(SCgi* parent, int fd);
+  void                close();
+
+  virtual void        event_read();
+  virtual void        event_write();
+  virtual void        event_error();
+
+  bool                receive_write(const char* buffer, uint32_t length);
+
+  utils::SocketFd&    get_fd()            { return *reinterpret_cast<utils::SocketFd*>(&m_fileDesc); }
+
+private:
+  SCgi*               m_parent;
+
+  char*               m_buffer;
+  char*               m_position;
+  unsigned int        m_bufferSize;
+};
+
+}
+
+#endif