#pragma once
#include
#include
#include “bytevec.h”
#include “pool.h”
/// Send a vector of data over a socket.
///
/// @param sd The socket on which to send
/// @param msg The message to send
///
/// @return True if the whole vector was sent, false otherwise
bool send_reliably(int sd, const bytevec &msg);
/// Send a string over a socket.
///
/// @param sd The socket on which to send
/// @param msg The message to send
///
/// @return True if the whole string was sent, false otherwise
bool send_reliably(int sd, const std::string &msg);
/// Perform a reliable read when we have a guess about how many bytes we might
/// get, but it’s OK if the socket EOFs before we get that many bytes.
///
/// @param sd The socket from which to read
/// @param pos The start of the vector where datashould go. It is assumed to
/// be pre-allocated to amnt or more.
/// @param amnt The maximum number of bytes to get
///
/// @return The actual number of bytes read, or -1 on a non-eof error
int reliable_get_to_eof_or_n(int sd, bytevec::iterator pos, int amnt);
/// Perform a reliable read when we are not sure how many bytes we are going
/// to receive.
///
/// @param sd The socket from which to read
///
/// @return A vector with the data that was read, or an empty vector on error
bytevec reliable_get_to_eof(int sd);
/// Connect to a server so that we can have bidirectional communication on the
/// socket (represented by a file descriptor) that this function returns
///
/// @param hostname The name of the server (ip or DNS) to connect to
/// @param port The server’s port that we should use
///
/// @return The socket descriptor for further communication, or -1 on error
int connect_to_server(const std::string &hostname, int port);
/// Create a server socket that we can use to listen for new incoming requests
///
/// @param port The port on which the program should listen for new connections
///
/// @return The new listening socket, or -1 on error
int create_server_socket(size_t port);
/// Given a listening socket, start calling accept() on it to get new
/// connections. Each time a connection comes in, use the provided handler to
/// process the request. Note that this is not multithreaded. Only one client
/// will be served at a time.
///
/// @param sd The socket file descriptor on which to call accept
/// @param handler A function to call when a new connection comes in
///
/// @return true on a graceful shutdown, false on an error
bool accept_client(int sd, std::function
/// Given a listening socket, start calling accept() on it to get new
/// connections. Each time a connection comes in, pass it to the thread pool so
/// that it can be processed.
///
/// @param sd The socket file descriptor on which to call accept
/// @param pool The thread pool that handles new requests
///
/// @return true on a graceful shutdown, false on an error
bool accept_client(int sd, thread_pool &pool);