source/network/server.h

   1 #ifndef R2C2_NET_SERVER_H_
   2 #define R2C2_NET_SERVER_H_
   3
   4 #include <msp/io/eventdispatcher.h>
   5 #include <msp/net/communicator.h>
   6 #include <msp/net/streamsocket.h>
   7 #include <msp/net/streamserversocket.h>
   8 #include "libr2c2/layout.h"
   9 #include "libr2c2/trainai.h"
  10 #include "packets.h"
  11 #include "protocol.h"
  12
  13 namespace R2C2 {
  14
  15 class Server
  16 {
  17 private:
  18         struct Connection: Msp::Net::PacketReceiver<DriverStatePacket>,
  19                 Msp::Net::PacketReceiver<TrainControlPacket>,
  20                 Msp::Net::PacketReceiver<TrainFunctionPacket>,
  21                 Msp::Net::PacketReceiver<TrainRoutePacket>
  22         {
  23                 Server &server;
  24                 Msp::Net::StreamSocket *socket;
  25                 Msp::Net::Communicator comm;
  26                 bool stale;
  27
  28                 Connection(Server &, Msp::Net::StreamSocket *);
  29                 ~Connection();
  30
  31                 void handshake_done();
  32                 void end_of_file();
  33                 virtual void receive(const DriverStatePacket &);
  34                 virtual void receive(const TrainControlPacket &);
  35                 virtual void receive(const TrainFunctionPacket &);
  36                 virtual void receive(const TrainRoutePacket &);
  37                 void comm_error(const std::exception &);
  38                 void error(const std::string &);
  39         };
  40
  41         Protocol proto;
  42         Layout &layout;
  43         Msp::Net::StreamServerSocket listen_sock;
  44         Msp::IO::EventDispatcher *event_disp;
  45         std::vector<Connection *> connections;
  46
  47 public:
  48         Server(Layout &);
  49         ~Server();
  50
  51         void use_event_dispatcher(Msp::IO::EventDispatcher &);
  52 private:
  53         void incoming_connection();
  54
  55         void driver_state_changed();
  56         void emergency(Block *, const std::string &);
  57         void train_added(Train &);
  58         void train_control_changed(const Train &, const std::string &, float);
  59         void train_function_changed(const Train &, unsigned, bool);
  60         void train_ai_event(const Train &, TrainAI &, const TrainAI::Message &);
  61
  62         template<typename P>
  63         void send(const P &);
  64 };
  65
  66 } // namespace R2C2
  67
  68 #endif