67 lines
1.8 KiB
C++
67 lines
1.8 KiB
C++
/**
|
|
* \brief Socket based communication wrapper functions.
|
|
*/
|
|
|
|
#ifndef __SOCKET_COMM_HPP__
|
|
#define __SOCKET_COMM_HPP__
|
|
|
|
#include <stdio.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <netdb.h>
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <google/protobuf/message.h>
|
|
|
|
namespace fail {
|
|
|
|
class SocketComm {
|
|
public:
|
|
/**
|
|
* This allows us to ignore SIGPIPE.
|
|
*/
|
|
static void init();
|
|
/**
|
|
* Send Protobuf-generated message
|
|
* @param sockfd open socket descriptor to write to
|
|
* @param Msg Reference to Protobuf generated message type
|
|
* \return false if message sending failed
|
|
*/
|
|
static bool sendMsg(int sockfd, google::protobuf::Message& msg);
|
|
/**
|
|
* Receive Protobuf-generated message
|
|
* @param sockfd open socket descriptor to read from
|
|
* @param Msg Reference to Protobuf generated message type
|
|
* \return false if message reception failed
|
|
*/
|
|
static bool rcvMsg(int sockfd, google::protobuf::Message& msg);
|
|
|
|
/**
|
|
* Receive Protobuf-generated message and just drop it
|
|
* @param sockfd open socket descriptor to read from
|
|
* \return false if message reception failed
|
|
*/
|
|
static bool dropMsg(int sockfd);
|
|
|
|
/**
|
|
* An accept() wrapper that times out (using poll(2))
|
|
* @param sockfd listening socket descriptor to accept connections from
|
|
* @param addr same as accept()'s
|
|
* @param addrlen same as accept()'s
|
|
* @param timeout timeout in milliseconds (see poll(2))
|
|
* \return < 0 on failure, > 0 for a new socket connection
|
|
*/
|
|
static int timedAccept(int sockfd, struct sockaddr *addr, socklen_t *addrlen, int timeout);
|
|
|
|
private:
|
|
static char *getBuf(int sockfd, int *size);
|
|
static ssize_t safe_write(int fd, const void *buf, size_t count);
|
|
static ssize_t safe_read(int fd, void *buf, size_t count);
|
|
};
|
|
|
|
} // end-of-namespace: fail
|
|
|
|
#endif // __SOCKET_COMM_HPP__
|