| 1 | /************************************************************************* |
| 2 | * libjson-rpc-cpp |
| 3 | ************************************************************************* |
| 4 | * @file linuxtcpsocketserver.h |
| 5 | * @date 17.07.2015 |
| 6 | * @author Alexandre Poirot <alexandre.poirot@legrand.fr> |
| 7 | * @license See attached LICENSE.txt |
| 8 | ************************************************************************/ |
| 9 | |
| 10 | #ifndef JSONRPC_CPP_LINUXTCPSOCKETSERVERCONNECTOR_H_ |
| 11 | #define JSONRPC_CPP_LINUXTCPSOCKETSERVERCONNECTOR_H_ |
| 12 | |
| 13 | #include <arpa/inet.h> |
| 14 | #include <netinet/in.h> |
| 15 | #include <pthread.h> |
| 16 | #include <stdarg.h> |
| 17 | #include <stdint.h> |
| 18 | #include <sys/socket.h> |
| 19 | #include <sys/time.h> |
| 20 | #include <sys/types.h> |
| 21 | #include <unistd.h> |
| 22 | |
| 23 | #include "../abstractthreadedserver.h" |
| 24 | |
| 25 | namespace jsonrpc { |
| 26 | /** |
| 27 | * This class is the Linux/UNIX implementation of TCPSocketServer. |
| 28 | * It uses the POSIX socket API and POSIX thread API to performs its job. |
| 29 | * Each client request is handled in a new thread. |
| 30 | */ |
| 31 | class LinuxTcpSocketServer : public AbstractThreadedServer { |
| 32 | public: |
| 33 | /** |
| 34 | * @brief LinuxTcpSocketServer, constructor of the Linux/UNIX |
| 35 | * implementation of class TcpSocketServer |
| 36 | * @param ipToBind The ipv4 address on which the server should |
| 37 | * bind and listen |
| 38 | * @param port The port on which the server should bind and listen |
| 39 | */ |
| 40 | LinuxTcpSocketServer(const std::string &ipToBind, const unsigned int &port, size_t threads = 1); |
| 41 | |
| 42 | virtual ~LinuxTcpSocketServer(); |
| 43 | |
| 44 | virtual bool InitializeListener(); |
| 45 | virtual int CheckForConnection(); |
| 46 | virtual void HandleConnection(int connection); |
| 47 | |
| 48 | protected: |
| 49 | std::string ipToBind; |
| 50 | unsigned int port; |
| 51 | int socket_fd; |
| 52 | struct sockaddr_in address; |
| 53 | |
| 54 | /** |
| 55 | * @brief A method that wait for the client to close the tcp session |
| 56 | * |
| 57 | * This method wait for the client to close the tcp session in order to avoid |
| 58 | * the server to enter in TIME_WAIT status. |
| 59 | * Entering in TIME_WAIT status with too many clients may occur in a DOS |
| 60 | * attack |
| 61 | * since server will not be able to use a new socket when a new client |
| 62 | * connects. |
| 63 | * @param fd The file descriptor of the socket that should be closed by the |
| 64 | * client |
| 65 | * @param timeout The maximum time the server will wait for the client to |
| 66 | * close the tcp session in microseconds. |
| 67 | * @returns A boolean indicating the success or the failure of the operation |
| 68 | */ |
| 69 | bool WaitClientClose(const int &fd, const int &timeout = 100000); |
| 70 | /** |
| 71 | * @brief A method that close a socket by reseting it |
| 72 | * |
| 73 | * This method reset the tcp session in order to avoid enter in TIME_WAIT |
| 74 | * state. |
| 75 | * @param fd The file descriptor of the socket that should be reset |
| 76 | * @returns The return value of POSIX close() method |
| 77 | */ |
| 78 | int CloseByReset(const int &fd); |
| 79 | /** |
| 80 | * @brief A method that cleanly close a socket by avoid TIME_WAIT state |
| 81 | * |
| 82 | * This method uses WaitClientClose and ClodeByReset to clenly close a tcp |
| 83 | * session with a client |
| 84 | * (avoiding TIME_WAIT to avoid DOS attacks). |
| 85 | * @param fd The file descriptor of the socket that should be cleanly closed |
| 86 | * @returns The return value of POSIX close() method |
| 87 | */ |
| 88 | int CleanClose(const int &fd); |
| 89 | }; |
| 90 | |
| 91 | } /* namespace jsonrpc */ |
| 92 | #endif /* JSONRPC_CPP_LINUXTCPSOCKETSERVERCONNECTOR_H_ */ |
| 93 | |