tcpserver.h

Go to the documentation of this file.

 
#ifndef TCP_SERVER_H
#define TCP_SERVER_H
 
#include <iostream>
#include <map>
#include <deque>
#include <string.h>
#include "tcpsocket.h"
#include "tcpssl.h"
 
namespace tcp {
 
using namespace std;
 
// Forward declarations:
 
class Server;
class Session;  
 
class Server : public Socket {
  public:
    Server(EPoll &epoll, SSLContext *ctx, const int domain = AF_INET) : Socket(epoll,domain,0,false,EPOLLIN), useSSL_(ctx), ctx_(ctx) {}
    
    virtual ~Server();
 
    void start(in_port_t port, string bindaddress, bool useSSL = false, int backlog = 64);
    void start(in_port_t port, char *bindaddress, bool useSSL = false, int backlog = 64);
    
    void stop();
 
    bool listening() { return state_ == SocketState::LISTENING; } 
    
    static bool printifaddrs();
 
    SSLContext *ctx() { return ctx_; }
 
  protected:
  
    void handleEvents(uint32_t events) override;
    
    virtual Session* createSession(const int socket, const sockaddr_in peer_address) = 0;
 
    bool findifaddr(const string ifname, sockaddr *addr);
 
    std::map<int,tcp::Session*> sessions;
 
  private:
    bool bindToAddress(sockaddr *addr, socklen_t len);
    bool startListening(int backlog);
    bool acceptConnection();
    bool useSSL_ {false};
    SSLContext *ctx_;
    struct sockaddr_storage addr_;
    friend class Session;
};
 
class Session : public DataSocket {
  public:
    
    Server &server() const { return server_; }
 
    in_port_t peer_port() const { return port_;   }
 
    in_addr_t peer_address() const { return addr_;   }
    
    bool connected() const { return state_ == SocketState::CONNECTED; }
 
    void disconnect() override;
 
  protected:
    
    Session(EPoll &epoll, Server& server, const int socket, const struct sockaddr_in peer_addr) 
      : DataSocket(epoll,server.domain(),socket), server_(server), port_(peer_addr.sin_port), addr_(peer_addr.sin_addr.s_addr) { }
    
    virtual ~Session();
 
    virtual void accepted();
    
    void disconnected() override;
 
    friend class SSL;
  private:
    void connectionMessage(string action);
    Server& server_;
    in_port_t port_;
    in_addr_t addr_;
    friend class Server;
};
 
}
 
#endif