X-Git-Url: http://g0dil.de/git?a=blobdiff_plain;f=senf%2FUtils%2FConsole%2FServer.cc;h=7ea4bf19089b0605372d5f122c0c841b76fd764b;hb=55b4c5a3afcf821031cef2ca7089fbdfd77d513a;hp=48ac147f697348fbd2a55e1b2f61e8a92d85ebe5;hpb=9cb871b939efe93e35dd96808d25089399acfc46;p=senf.git diff --git a/senf/Utils/Console/Server.cc b/senf/Utils/Console/Server.cc index 48ac147..7ea4bf1 100644 --- a/senf/Utils/Console/Server.cc +++ b/senf/Utils/Console/Server.cc @@ -2,68 +2,85 @@ // // Copyright (C) 2008 // Fraunhofer Institute for Open Communication Systems (FOKUS) -// Competence Center NETwork research (NET), St. Augustin, GERMANY -// Stefan Bund // -// 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. +// The contents of this file are subject to the Fraunhofer FOKUS Public License +// Version 1.0 (the "License"); you may not use this file except in compliance +// with the License. You may obtain a copy of the License at +// http://senf.berlios.de/license.html // -// 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. +// The Fraunhofer FOKUS Public License Version 1.0 is based on, +// but modifies the Mozilla Public License Version 1.1. +// See the full license text for the amendments. // -// 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. +// Software distributed under the License is distributed on an "AS IS" basis, +// WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License +// for the specific language governing rights and limitations under the License. +// +// The Original Code is Fraunhofer FOKUS code. +// +// The Initial Developer of the Original Code is Fraunhofer-Gesellschaft e.V. +// (registered association), Hansastraße 27 c, 80686 Munich, Germany. +// All Rights Reserved. +// +// Contributor(s): +// Stefan Bund /** \file \brief Server non-inline non-template implementation */ #include "Server.hh" -#include "Server.ih" +//#include "Server.ih" // Custom includes -#include -#include #include -#include -#include #include -#include #include #include +#include #include "LineEditor.hh" #include "ScopedDirectory.hh" #include "Sysdir.hh" +#include "SysInfo.hh" #include "ParsedCommand.hh" //#include "Server.mpp" #define prefix_ -///////////////////////////////cc.p//////////////////////////////////////// +//-///////////////////////////////////////////////////////////////////////////////////////////////// + +#ifdef SENF_DEBUG +# define BUILD_TYPE "development" +#else +# define BUILD_TYPE "production" +#endif + +namespace { + senf::console::SysInfo::Proxy addSysInfo ( + "SENF: The Simple and Extensible Network Framework\n" + " (c) 2006-2011 Fraunhofer Institute for Open Communication Systems (FOKUS)\n" + " Contact: senf-dev@lists.berlios.de\n" + " Version: " SENF_LIB_VERSION " Revision number: " SENF_REVISION "\n" + " Build-type: " BUILD_TYPE ", SenfLog compile time limit: " + + senf::str(senf::log::LEVELNAMES[senf::SenfLog::compileLimit::value]), 0); +} -/////////////////////////////////////////////////////////////////////////// -// senf::console::detail::NonBlockingSocketSink +//-///////////////////////////////////////////////////////////////////////////////////////////////// +// senf::console::detail::SocketStreamSink -prefix_ std::streamsize senf::console::detail::NonblockingSocketSink::write(const char * s, +prefix_ std::streamsize senf::console::detail::SocketStreamSink::write(const char * s, std::streamsize n) { +// since handle is now non blocking we done check for writeable try { - if (client_.handle().writeable()) { +// if (client_.handle().writeable()) { std::string data (s, n); client_.write(data); - } - } - catch (SystemException & ex) { - ; +// } } + catch (...) {} return n; } -/////////////////////////////////////////////////////////////////////////// +//-///////////////////////////////////////////////////////////////////////////////////////////////// // senf::console::Server prefix_ senf::console::Server & @@ -116,15 +133,15 @@ prefix_ void senf::console::Server::removeClient(Client & client) try { log << client.handle().peer(); } - catch (senf::SystemException ex) { - log << "(unknown)"; + catch (senf::SystemException & ex) { + log << "(dead socket)"; } })); // THIS DELETES THE CLIENT INSTANCE !! clients_.erase(boost::intrusive_ptr(&client)); } -/////////////////////////////////////////////////////////////////////////// +//-///////////////////////////////////////////////////////////////////////////////////////////////// // senf::console::detail::DumbClientReader prefix_ senf::console::detail::DumbClientReader::DumbClientReader(Client & client) @@ -149,7 +166,7 @@ senf::console::detail::DumbClientReader::clientData(senf::ReadHelperdata()); tail_ = helper->tail(); - boost::trim(data); // Gets rid of superfluous \r or \n characters + boost::trim(data); // Gets rid of superfluous \r or \n characters handleInput(data); showPrompt(); @@ -164,9 +181,9 @@ prefix_ void senf::console::detail::DumbClientReader::showPrompt() prompt += " "; stream() << std::flush; - handle().write(prompt); promptLen_ = prompt.size(); promptActive_ = true; + v_write(prompt); } prefix_ void senf::console::detail::DumbClientReader::v_disablePrompt() @@ -185,7 +202,24 @@ prefix_ void senf::console::detail::DumbClientReader::v_enablePrompt() prefix_ void senf::console::detail::DumbClientReader::v_write(std::string const & data) { - handle().write(data); + try { + handle().write(data); + } + catch (senf::ExceptionMixin & ex) { + SENF_LOG(("unexpected failure writing to socket:" << ex.message())); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + } + catch (std::exception & ex) { + SENF_LOG(("unexpected failure writing to socket:" << ex.what())); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + } + catch (...) { + SENF_LOG(("unexpected failure writing to socket: unknown exception")); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + } } prefix_ unsigned senf::console::detail::DumbClientReader::v_width() @@ -194,15 +228,18 @@ prefix_ unsigned senf::console::detail::DumbClientReader::v_width() return 80; } -/////////////////////////////////////////////////////////////////////////// +//-///////////////////////////////////////////////////////////////////////////////////////////////// // senf::console::detail::NoninteractiveClientReader prefix_ senf::console::detail::NoninteractiveClientReader::NoninteractiveClientReader(Client & client) - : ClientReader (client), + : ClientReader (client), streamBufferMaxSize_( 1024*1024), readevent_ ("senf::console::detail::NoninteractiveClientReader", senf::membind(&NoninteractiveClientReader::newData, this), - handle(), senf::scheduler::FdEvent::EV_READ) + handle(), senf::scheduler::FdEvent::EV_READ), + writeevent_ ("senf::console::detail::NoninteractiveClientReader", + membind(&NoninteractiveClientReader::writeHandler, this), handle(), + scheduler::FdEvent::EV_WRITE, false) {} prefix_ void senf::console::detail::NoninteractiveClientReader::v_disablePrompt() @@ -211,9 +248,27 @@ prefix_ void senf::console::detail::NoninteractiveClientReader::v_disablePrompt( prefix_ void senf::console::detail::NoninteractiveClientReader::v_enablePrompt() {} +prefix_ void senf::console::detail::NoninteractiveClientReader::streamBufferMaxSize(SendQueue::size_type size) +{ + streamBufferMaxSize_ = size; +} + +prefix_ senf::console::detail::NoninteractiveClientReader::SendQueue::size_type +senf::console::detail::NoninteractiveClientReader::streamBufferMaxSize() + const +{ + return streamBufferMaxSize_; +} + prefix_ void senf::console::detail::NoninteractiveClientReader::v_write(std::string const & data) { - handle().write(data); + if (sendQueue_.size() > streamBufferMaxSize_) + return; + sendQueue_.insert( sendQueue_.end(), data.begin(), data.end()); + writeHandler(scheduler::FdEvent::EV_WRITE); + if (! sendQueue_.empty()) + writeevent_.enable(); + } prefix_ unsigned senf::console::detail::NoninteractiveClientReader::v_width() @@ -240,7 +295,41 @@ senf::console::detail::NoninteractiveClientReader::newData(int event) stream() << std::flush; } -/////////////////////////////////////////////////////////////////////////// +prefix_ void +senf::console::detail::NoninteractiveClientReader::writeHandler(int event) +{ + if (event != senf::scheduler::FdEvent::EV_WRITE) { + writeevent_.disable(); + readevent_.disable(); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + return; + } + try { + sendQueue_.erase(sendQueue_.begin(), + handle().write(boost::make_iterator_range(sendQueue_.begin(), sendQueue_.end()))); + } + catch (senf::ExceptionMixin & ex) { + SENF_LOG(("unexpected failure writing to socket:" << ex.message())); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + } + catch (std::exception & ex) { + SENF_LOG(("unexpected failure writing to socket:" << ex.what())); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + } + catch (...) { + SENF_LOG(("unexpected failure writing to socket: unknown exception")); + try { handle().facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} + } + if (sendQueue_.empty()) + writeevent_.disable(); + +} + +//-///////////////////////////////////////////////////////////////////////////////////////////////// // senf::console::Client prefix_ senf::console::Client::Client(Server & server, ClientHandle handle) @@ -256,7 +345,8 @@ prefix_ senf::console::Client::Client(Server & server, ClientHandle handle) false), name_ (server.name()), reader_ (), mode_ (server.mode()) { - handle_.facet().nodelay(); + handle_.facet().nodelay(true); + handle_.blocking(false); executor_.chroot(root()); switch (mode_) { case Server::Interactive : @@ -286,7 +376,14 @@ prefix_ void senf::console::Client::setNoninteractive() readevent_.disable(); timer_.disable(); mode_ = Server::Noninteractive; - reader_.reset(new detail::NoninteractiveClientReader(*this)); + detail::NoninteractiveClientReader * newReader (new detail::NoninteractiveClientReader(*this)); + reader_.reset( newReader); + consoleDir().add( "streamBuffer", senf::console::factory::Command( senf::membind( + SENF_MEMFNP( detail::NoninteractiveClientReader::SendQueue::size_type, detail::NoninteractiveClientReader, streamBufferMaxSize, () const ), + newReader ))); + consoleDir().add( "streamBuffer", senf::console::factory::Command( senf::membind( + SENF_MEMFNP( void, detail::NoninteractiveClientReader, streamBufferMaxSize, (detail::NoninteractiveClientReader::SendQueue::size_type) ), + newReader ))); } prefix_ std::string::size_type senf::console::Client::handleInput(std::string data, @@ -309,7 +406,8 @@ prefix_ std::string::size_type senf::console::Client::handleInput(std::string da // handled gracefully by the ClientReader. We cannot call stop() here, since we are called // from the client reader callback and that will continue executing after stop() has been // called. stop() however will delete *this instance ... BANG ... - handle_.facet().shutdown(senf::TCPSocketProtocol::ShutRD); + try { handle_.facet().shutdown(senf::TCPSocketProtocol::ShutRD); } + catch (...) {} } catch (std::exception & ex) { std::string msg (ex.what()); @@ -317,8 +415,10 @@ prefix_ std::string::size_type senf::console::Client::handleInput(std::string da if (i != std::string::npos) { backtrace_ = msg.substr(0,i); msg = msg.substr(i+4); - } else + } else { backtrace_.clear(); + } + stream() << msg << std::endl; } catch (...) { @@ -349,14 +449,11 @@ prefix_ unsigned senf::console::Client::getWidth(std::ostream & os, unsigned def return rv < minWidth ? defaultWidth : rv; } -/////////////////////////////////////////////////////////////////////////// +//-///////////////////////////////////////////////////////////////////////////////////////////////// // senf::console::Client::SysBacktrace - prefix_ senf::console::Client::SysBacktrace::SysBacktrace() { - namespace fty = senf::console::factory; - - sysdir().add("backtrace", fty::Command(&SysBacktrace::backtrace) + sysdir().add("backtrace", factory::Command(&SysBacktrace::backtrace) .doc("Display the backtrace of the last error / exception in this console") ); } @@ -371,7 +468,7 @@ prefix_ void senf::console::Client::SysBacktrace::backtrace(std::ostream & os) senf::console::Client::SysBacktrace senf::console::Client::SysBacktrace::instance_; -///////////////////////////////cc.e//////////////////////////////////////// +//-///////////////////////////////////////////////////////////////////////////////////////////////// #undef prefix_ //#include "Server.mpp"