developer.intra2net.com Git - libt2n/blob - src/socket_server.hxx

   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 #ifndef __LIBT2N_SOCKET_SERVER
  20 #define __LIBT2N_SOCKET_SERVER
  21
  22 #include <sys/types.h>
  23 #include <string>
  24
  25 #include "server.hxx"
  26 #include "socket_handler.hxx"
  27 #include "types.hxx"
  28
  29 namespace libt2n
  30 {
  31
  32 class socket_server_connection;
  33
  34 /** @brief Socket based server class
  35
  36     Use this class to instantiate a server listening for client connections.
  37     Call fill_buffer() to read data from the network and get_packet() to retrieve
  38     this data. Don't forget to call cleanup() from time to time to remove closed
  39     connections and close idle ones.
  40 */
  41 class socket_server : public socket_handler, public server
  42 {
  43     friend class socket_server_connection;
  44
  45     private:
  46         fd_set connection_set;
  47         std::string unix_path;
  48
  49         void start_listening();
  50
  51         void new_connection();
  52
  53         bool fill_connection_buffers();
  54         void remove_connection_socket(int sock);
  55
  56     protected:
  57         std::ostream* get_logstream(log_level_values level)
  58             { return server::get_logstream(level); }
  59
  60     public:
  61         socket_server(int port, const std::string& ip="0.0.0.0");
  62         socket_server(const std::string& path, mode_t filemode=00770, const std::string& user="", const std::string& group="");
  63
  64         ~socket_server();
  65
  66         bool fill_buffer(long long usec_timeout=-1,long long* usec_timeout_remaining=NULL);
  67 };
  68
  69 /** @brief Socket based connection
  70
  71     This class is used within a socket_server to represent the connection to each client.
  72 */
  73 class socket_server_connection : public socket_handler, public server_connection
  74 {
  75     friend class socket_server;
  76
  77     private:
  78         socket_server_connection(int _sock, socket_type_value _stype, int _timeout)
  79            : server_connection(_timeout), socket_handler(_sock,_stype)
  80            { }
  81
  82         std::ostream* get_logstream(log_level_values level)
  83             { return server_connection::get_logstream(level); }
  84
  85         void real_write(const std::string& data)
  86             { socket_write(data); }
  87
  88     public:
  89         bool fill_buffer(long long usec_timeout=-1,long long* usec_timeout_remaining=NULL)
  90             { return socket_handler::fill_buffer(buffer,usec_timeout,usec_timeout_remaining); }
  91
  92         void close();
  93 };
  94
  95 }
  96
  97 #endif