X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=acceptor.h;h=969ea65889419a38efd6683072be36bf60e9dc35;hp=e0a1bd54811df6e4be0a28de9f5e415d48cc53b5;hb=HEAD;hpb=ca9624c43b968a0f29ea44e47851ff686bb64bb6 diff --git a/acceptor.h b/acceptor.h index e0a1bd5..969ea65 100644 --- a/acceptor.h +++ b/acceptor.h @@ -1,32 +1,54 @@ #ifndef _ACCEPTOR_H #define _ACCEPTOR_H +#include + +#include + #include "thread.h" enum SocketType { TCP_SOCKET, UDP_SOCKET, }; -int create_server_socket(int port, SocketType socket_type); +int create_server_socket(const sockaddr_in6 &addr, SocketType socket_type); class AcceptorProto; +sockaddr_in6 create_any_address(int port); +sockaddr_in6 extract_address_from_acceptor_proto(const AcceptorProto &proto); + // A thread that accepts new connections on a given socket, // and hands them off to the server pool. class Acceptor : public Thread { public: - Acceptor(int server_sock, int port); + Acceptor(int server_sock, const sockaddr_in6 &addr, + const std::string &certificate_chain, const std::string &private_key); // Serialization/deserialization. Acceptor(const AcceptorProto &serialized); AcceptorProto serialize() const; + bool is_tls() const { return !certificate_chain.empty(); } + + std::string get_certificate_chain() const { + assert(is_tls()); + return certificate_chain; + } + + std::string get_private_key() const { + assert(is_tls()); + return private_key; + } + void close_socket(); private: virtual void do_work(); - int server_sock, port; + int server_sock; + sockaddr_in6 addr; + std::string certificate_chain, private_key; // Both empty for no TLS. }; #endif // !defined(_ACCEPTOR_H)