developer.intra2net.com Git - libt2n/blob - src/command_server.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2006 by Gerd v. Egidy                                   *
   3  *   gve@intra2net.com                                                     *
   4  *                                                                         *
   5  *   This library is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU Lesser General Public License version   *
   7  *   2.1 as published by the Free Software Foundation.                     *
   8  *                                                                         *
   9  *   This library is distributed in the hope that it will be useful,       *
  10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  12  *   GNU Lesser General Public License for more details.                   *
  13  *                                                                         *
  14  *   You should have received a copy of the GNU Lesser General Public      *
  15  *   License along with this program; if not, write to the                 *
  16  *   Free Software Foundation, Inc.,                                       *
  17  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  18  ***************************************************************************/
  19
  20 #include <string>
  21 #include <sstream>
  22 #include <stdexcept>
  23 #include <iostream>
  24
  25 #include <boost/archive/binary_oarchive.hpp>
  26 #include <boost/archive/binary_iarchive.hpp>
  27 #include <boost/archive/xml_oarchive.hpp>
  28 #include <boost/archive/xml_iarchive.hpp>
  29 #include <boost/serialization/serialization.hpp>
  30
  31 #include <boost/bind.hpp>
  32
  33 #include "command_server.hxx"
  34 #include "container.hxx"
  35 #include "log.hxx"
  36
  37 #ifdef HAVE_CONFIG_H
  38 #include <config.h>
  39 #endif
  40
  41 using namespace std;
  42
  43 namespace libt2n
  44 {
  45
  46 command_server::command_server(server& _s)
  47     : s(_s)
  48 {
  49     // register callback
  50     s.add_callback(new_connection,bind(&command_server::send_hello, boost::ref(*this), _1));
  51 }
  52
  53 void command_server::send_hello(unsigned int conn_id)
  54 {
  55     server_connection* sc=s.get_connection(conn_id);
  56
  57     ostringstream hello;
  58
  59     hello << "T2Nv" << PROTOCOL_VERSION << ';';
  60
  61     int byteordercheck=1;
  62     hello.write((char*)&byteordercheck,sizeof(byteordercheck));
  63
  64     hello << ';';
  65
  66     sc->write(hello.str());
  67 }
  68
  69 /// handle a command including deserialization and answering
  70 void command_server::handle_packet(const std::string& packet, server_connection* conn)
  71 {
  72     OBJLOGSTREAM(s,debug,"handling packet from connection " << conn->get_id());
  73
  74     // deserialize packet
  75     istringstream ifs(packet);
  76     boost::archive::binary_iarchive ia(ifs);
  77     command_container ccont;
  78     result_container res;
  79
  80     try
  81     {
  82         ia >> ccont;
  83     }
  84     catch(boost::archive::archive_exception &e)
  85     {
  86         ostringstream msg;
  87         msg << "archive_exception while deserializing on server-side, "
  88                "code " << e.code << " (" << e.what() << ")";
  89         res.set_exception(new t2n_serialization_error(msg.str()));
  90     }
  91     catch(...)
  92         { throw; }
  93
  94     if (!res.has_exception())
  95     {
  96         std::ostream* ostr;
  97         if ((ostr=s.get_logstream(fulldebug))!=NULL)
  98         {
  99             (*ostr) << "decoded packet data: " << std::endl;
 100             boost::archive::xml_oarchive xo(*ostr);
 101             xo << BOOST_SERIALIZATION_NVP(ccont);
 102         }
 103
 104         // TODO: cast to command subclass (template)
 105         command *cmd=ccont.get_command();
 106
 107         if (cmd)
 108         {
 109             try
 110             {
 111                 res.set_result((*cmd)());
 112             }
 113             catch (t2n_exception &e)
 114                 { res.set_exception(e.clone()); }
 115             catch (...)
 116                 { throw; }
 117         }
 118         else
 119             throw logic_error("uninitialized command called");
 120     }
 121
 122     ostringstream ofs;
 123     boost::archive::binary_oarchive oa(ofs);
 124
 125     try
 126     {
 127         oa << res;
 128     }
 129     catch(boost::archive::archive_exception &e)
 130     {
 131         ostringstream msg;
 132         msg << "archive_exception while serializing on server-side, "
 133                "code " << e.code << " (" << e.what() << ")";
 134         res.set_exception(new t2n_serialization_error(msg.str()));
 135         oa << res;
 136     }
 137     catch(...)
 138         { throw; }
 139
 140     std::ostream* ostr;
 141     if ((ostr=s.get_logstream(fulldebug))!=NULL)
 142     {
 143         (*ostr) << "returning result, decoded data: " << std::endl;
 144         boost::archive::xml_oarchive xo(*ostr);
 145         xo << BOOST_SERIALIZATION_NVP(res);
 146     }
 147
 148     conn->write(ofs.str());
 149 }
 150
 151 /** @brief handle incoming commands
 152     @param[in,out] usec_timeout wait until new data is found, max timeout usecs.
 153             -1: wait endless, 0: instant return
 154 */
 155 void command_server::handle(long long usec_timeout, long long* usec_timeout_remaining)
 156 {
 157     if (s.fill_buffer(usec_timeout,usec_timeout_remaining))
 158     {
 159         string packet;
 160         unsigned int conn_id;
 161
 162         while (s.get_packet(packet,conn_id))
 163             handle_packet(packet,s.get_connection(conn_id));
 164     }
 165     s.cleanup();
 166 }
 167
 168 }