Utils/Console/UDPServer.cc

   1 // $Id$
   2 //
   3 // Copyright (C) 2009
   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 UDPServer non-inline non-template implementation */
  25
  26 #include "UDPServer.hh"
  27 //#include "UDPServer.ih"
  28
  29 // Custom includes
  30 #include <boost/algorithm/string/trim.hpp>
  31 #include "../membind.hh"
  32
  33 //#include "UDPServer.mpp"
  34 #define prefix_
  35 ///////////////////////////////cc.p////////////////////////////////////////
  36
  37 prefix_ senf::console::UDPServer::UDPServer(senf::INet4SocketAddress const & address)
  38     : replies_ (true), emptyReplies_ (true), target_ (),
  39       handle_ (senf::UDPv4ClientSocketHandle(address)),
  40       readevent_ ("senf::console::UDPServer::readevent",
  41                   senf::membind(&UDPServer::handleInput, this),
  42                   handle_,
  43                   senf::scheduler::FdEvent::EV_READ),
  44       parser_ (), executor_ ()
  45 {
  46     if (address.address().multicast())
  47         handle_.facet<senf::INet4MulticastSocketProtocol>().mcAddMembership(address.address());
  48     SENF_LOG(("UDP Console server started at " << address ));
  49 }
  50
  51 prefix_ senf::console::UDPServer::UDPServer(senf::INet6SocketAddress const & address)
  52     : replies_ (true), target_ (), handle_ (senf::UDPv6ClientSocketHandle(address)),
  53       readevent_ ("senf::console::UDPServer::readevent",
  54                   senf::membind(&UDPServer::handleInput, this),
  55                   handle_,
  56                   senf::scheduler::FdEvent::EV_READ),
  57       parser_ (), executor_ ()
  58 {
  59     if (address.address().multicast())
  60         handle_.facet<senf::INet6MulticastSocketProtocol>().mcAddMembership(address.address());
  61     SENF_LOG(("UDP Console server started at " << address ));
  62 }
  63
  64 prefix_ senf::console::UDPServer & senf::console::UDPServer::replies(bool enable)
  65 {
  66     replies_ = enable;
  67     return *this;
  68 }
  69
  70 prefix_ senf::console::UDPServer &
  71 senf::console::UDPServer::replies(senf::INet4SocketAddress const & address)
  72 {
  73     SENF_ASSERT( handle_.local().family() == senf::INet4SocketAddress::addressFamily );
  74     target_ = address;
  75     return *this;
  76 }
  77
  78 prefix_ senf::console::UDPServer &
  79 senf::console::UDPServer::replies(senf::INet6SocketAddress const & address)
  80 {
  81     SENF_ASSERT( handle_.local().family() == senf::INet6SocketAddress::addressFamily );
  82     target_ = address;
  83     return *this;
  84 }
  85
  86 prefix_ senf::console::UDPServer & senf::console::UDPServer::emptyReplies(bool enable)
  87 {
  88     emptyReplies_ = enable;
  89     return *this;
  90 }
  91
  92 prefix_ senf::console::DirectoryNode & senf::console::UDPServer::root()
  93     const
  94 {
  95     return executor_.chroot();
  96 }
  97
  98 prefix_ senf::console::UDPServer & senf::console::UDPServer::root(DirectoryNode & root)
  99 {
 100     executor_.chroot(root);
 101     return *this;
 102 }
 103
 104 prefix_ void senf::console::UDPServer::handleInput(int events)
 105 {
 106     if (events != senf::scheduler::FdEvent::EV_READ) {
 107         SENF_LOG((senf::log::IMPORTANT)("Input handle read error. Closing socket."));
 108         readevent_.disable();
 109         handle_.close();
 110         return;
 111     }
 112
 113     std::string data;
 114     senf::GenericBSDSocketAddress address;
 115     handle_.readfrom(data, address, 0u);
 116     boost::trim(data);
 117
 118     executor_.cwd(executor_.chroot());
 119     std::stringstream stream;
 120     try {
 121         parser_.parse(data, boost::bind<void>( boost::ref(executor_), boost::ref(stream), _1));
 122     }
 123     catch (Executor::ExitException &) {
 124         // Ignored
 125     }
 126     catch (ExceptionMixin & ex) {
 127         stream << ex.message() << '\n';
 128         SENF_LOG((senf::log::IMPORTANT)("Error: " << ex.message()));
 129         SENF_LOG((senf::log::NOTICE)(ex.backtrace()));
 130     }
 131     catch (std::exception & ex) {
 132         stream << ex.what() << '\n';
 133         SENF_LOG((senf::log::IMPORTANT)("Error: " << ex.what()));
 134     }
 135     if (replies_ && (emptyReplies_ || ! stream.str().empty())) {
 136         if (target_)
 137             address = target_;
 138         if (stream.str().empty())
 139             stream << '\0';
 140         handle_.writeto(address, stream.str());
 141     }
 142
 143 }
 144
 145 ///////////////////////////////cc.e////////////////////////////////////////
 146 #undef prefix_
 147 //#include "UDPServer.mpp"
 148
 149 \f
 150 // Local Variables:
 151 // mode: c++
 152 // fill-column: 100
 153 // comment-column: 40
 154 // c-file-style: "senf"
 155 // indent-tabs-mode: nil
 156 // ispell-local-dictionary: "american"
 157 // compile-command: "scons -u test"
 158 // End: