src/main/torrentserver.cpp

   1 /***************************************************************************
   2  *                                                                         *
   3  *   Copyright (C) 2006 David Brodsky                                      *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or         *
   6  *   modify it under the terms of the GNU General Public License as        *
   7  *   published by the Free Software Foundation and appearing               *
   8  *   in the file LICENSE.GPL included in the packaging of this file.       *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU      *
  13  *   General Public License for more details.                              *
  14  *                                                                         *
  15  ***************************************************************************/
  16
  17 #include <tairon/core/config.h>
  18 #include <tairon/core/log.h>
  19 #include <tairon/net/server.h>
  20 #include <tairon/net/socket.h>
  21
  22 #include "torrentserver.h"
  23
  24 #include "connection.h"
  25
  26 namespace Tairent
  27 {
  28
  29 namespace Main
  30 {
  31
  32 TorrentServer *TorrentServer::torrentserver = 0;
  33
  34 /* {{{ TorrentServer::TorrentServer() */
  35 TorrentServer::TorrentServer()
  36 {
  37         server = new Tairon::Net::Server(0, atoi((*Tairon::Core::Config::self())["torrent-server-port"]), 2);
  38         server->newConnectionSignal.connect(Tairon::Core::threadMethodDFunctor(Tairon::Core::Thread::current(), this, &TorrentServer::newConnection));
  39
  40         torrentserver = this;
  41 }
  42 /* }}} */
  43
  44 /* {{{ TorrentServer::~TorrentServer */
  45 TorrentServer::~TorrentServer()
  46 {
  47         if (server) {
  48                 server->newConnectionSignal.disable();
  49                 delete server;
  50         }
  51
  52         torrentserver = 0;
  53 }
  54 /* }}} */
  55
  56 /* {{{ TorrentServer::connectionClosed(Connection *) */
  57 void TorrentServer::connectionClosed(Connection *connection)
  58 {
  59         pendingConnections.erase(connection);
  60         delete connection;
  61 }
  62 /* }}} */
  63
  64 /* {{{ TorrentServer::connectionCompleted(Connection *) */
  65 void TorrentServer::connectionCompleted(Connection *connection)
  66 {
  67         pendingConnections.erase(connection);
  68         connection->closedSignal.clear();
  69 }
  70 /* }}} */
  71
  72 /* {{{ TorrentServer::newConnection(Tairon::Net::Server *, int) */
  73 void TorrentServer::newConnection(Tairon::Net::Server *, int fd)
  74 {
  75         DEBUG("new connection");
  76
  77         Connection *connection = new Connection(fd);
  78         connection->closedSignal.connect(Tairon::Core::methodFunctor(this, &TorrentServer::connectionClosed));
  79         pendingConnections.insert(connection);
  80 }
  81 /* }}} */
  82
  83 /* {{{ TorrentServer::stop() */
  84 void TorrentServer::stop()
  85 {
  86         // delete server
  87         server->newConnectionSignal.disable();
  88         delete server;
  89         server = 0;
  90
  91         // close pending connections
  92         for (std::set<Connection *>::const_iterator it = pendingConnections.begin(); it != pendingConnections.end(); ++it)
  93                 (*it)->close();
  94 }
  95 /* }}} */
  96
  97 }; // namespace Main
  98
  99 }; // namespace Tairent
 100
 101 // vim: ai sw=4 ts=4 noet fdm=marker