Console/Server.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2008
   4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
   5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
   6 //     Stefan Bund <g0dil@berlios.de>
   7 //
   8 // This program is free software; you can redistribute it and/or modify
   9 // it under the terms of the GNU General Public License as published by
  10 // the Free Software Foundation; either version 2 of the License, or
  11 // (at your option) any later version.
  12 //
  13 // This program is distributed in the hope that it will be useful,
  14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 // GNU General Public License for more details.
  17 //
  18 // You should have received a copy of the GNU General Public License
  19 // along with this program; if not, write to the
  20 // Free Software Foundation, Inc.,
  21 // 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  22
  23 /** \file
  24     \brief Server non-inline non-template implementation */
  25
  26 #include "Server.hh"
  27 //#include "Server.ih"
  28
  29 // Custom includes
  30 #include <unistd.h>
  31 #include <iostream>
  32 #include <boost/algorithm/string/trim.hpp>
  33 #include <boost/iostreams/device/file_descriptor.hpp>
  34 #include <boost/iostreams/stream.hpp>
  35 #include "../Utils/senfassert.hh"
  36 #include "../Utils/membind.hh"
  37 #include "../Utils/Logger/SenfLog.hh"
  38
  39 //#include "Server.mpp"
  40 #define prefix_
  41 ///////////////////////////////cc.p////////////////////////////////////////
  42
  43 prefix_ senf::console::Server &
  44 senf::console::Server::start(senf::INet4SocketAddress const & address)
  45 {
  46     senf::TCPv4ServerSocketHandle handle (address);
  47     senf::console::Server::start(handle);
  48     SENF_LOG((Server::SENFLogArea)(log::NOTICE)(
  49                  "Console server started at " << address ));
  50     return *instance_;
  51 }
  52
  53 prefix_ senf::console::Server &
  54 senf::console::Server::start(senf::INet6SocketAddress const & address)
  55 {
  56     senf::TCPv6ServerSocketHandle handle (address);
  57     senf::console::Server::start(handle);
  58     SENF_LOG((Server::SENFLogArea)(log::NOTICE)(
  59                  "Console server started at " << address ));
  60     return *instance_;
  61 }
  62
  63 ///////////////////////////////////////////////////////////////////////////
  64 // senf::console::Server
  65
  66 boost::scoped_ptr<senf::console::Server> senf::console::Server::instance_;
  67
  68 prefix_ void senf::console::Server::start(ServerHandle handle)
  69 {
  70     SENF_ASSERT( ! instance_ );
  71     instance_.reset(new Server(handle));
  72 }
  73
  74 prefix_ senf::console::Server::Server(ServerHandle handle)
  75     : handle_ (handle)
  76 {
  77     Scheduler::instance().add( handle_, senf::membind(&Server::newClient, this) );
  78 }
  79
  80 prefix_ senf::console::Server::~Server()
  81 {
  82     Scheduler::instance().remove(handle_);
  83 }
  84
  85 prefix_ void senf::console::Server::newClient(Scheduler::EventId event)
  86 {
  87     ServerHandle::ClientSocketHandle client (handle_.accept());
  88     boost::intrusive_ptr<Client> p (new Client(client, name_));
  89     clients_.insert( p );
  90     SENF_LOG(( "Registered new client " << p.get() ));
  91 }
  92
  93 prefix_ void senf::console::Server::removeClient(Client & client)
  94 {
  95     SENF_LOG(( "Disposing client " << & client ));
  96     // THIS DELETES THE CLIENT INSTANCE !!
  97     clients_.erase(boost::intrusive_ptr<Client>(&client));
  98 }
  99
 100 ///////////////////////////////////////////////////////////////////////////
 101 // senf::console::Client
 102
 103 prefix_ senf::console::Client::Client(ClientHandle handle, std::string const & name)
 104     : handle_ (handle), name_ (name), out_(::dup(handle.fd()))
 105 {
 106     out_ << name_ << "# " << std::flush;
 107     ReadHelper<ClientHandle>::dispatch( handle_, 16384u, ReadUntil("\n"),
 108                                         senf::membind(&Client::clientData, this) );
 109 }
 110
 111 prefix_ senf::console::Client::~Client()
 112 {}
 113
 114 prefix_ void senf::console::Client::stopClient()
 115 {
 116     // THIS COMMITS SUICIDE. THE INSTANCE IS GONE AFTER removeClient RETURNS
 117     Server::instance_->removeClient(*this);
 118 }
 119
 120 prefix_ void senf::console::Client::clientData(ReadHelper<ClientHandle>::ptr helper)
 121 {
 122     if (helper->error() || handle_.eof()) {
 123         // THIS COMMITS SUICIDE. THE INSTANCE IS GONE AFTER stopClient RETURNS
 124         stopClient();
 125         return;
 126     }
 127
 128 #   warning fix Client::clientData implementation
 129     // Remove the 'dup' needed here so we don't close the same fd twice (see Client constructor)
 130     // Make output non-blocking
 131     // Don't register a new ReadHelper every round
 132
 133     std::string data (tail_ + helper->data());
 134     tail_ = helper->tail();
 135     boost::trim(data); // Gets rid of superfluous  \r or \n characters
 136
 137     if (data == "exit") {
 138         // THIS COMMITS SUICIDE. THE INSTANCE IS GONE AFTER stopClient RETURNS
 139         stopClient();
 140         return;
 141     }
 142
 143     ParseCommandInfo command;
 144     if (parser_.parseCommand(data, command))
 145         executor_(command, out_);
 146     else
 147         out_ << "syntax error" << std::endl;
 148
 149     out_ << name_ << "# " << std::flush;
 150     ReadHelper<ClientHandle>::dispatch( handle_, 16384u, ReadUntil("\n"),
 151                                         senf::membind(&Client::clientData, this) );
 152 }
 153
 154 ///////////////////////////////cc.e////////////////////////////////////////
 155 #undef prefix_
 156 //#include "Server.mpp"
 157
 158 \f
 159 // Local Variables:
 160 // mode: c++
 161 // fill-column: 100
 162 // comment-column: 40
 163 // c-file-style: "senf"
 164 // indent-tabs-mode: nil
 165 // ispell-local-dictionary: "american"
 166 // compile-command: "scons -u test"
 167 // End: