universal_robots_ros_driver/include/ur_modern_driver/tcp_socket.h

#pragma once 
#include <netdb.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <mutex>
#include <atomic>
#include <string>

enum class SocketState
{
  Invalid,
  Connected,
  Disconnected,
  Closed
};

class TCPSocket
{
private:
  std::atomic<int> socket_fd_;
  std::atomic<SocketState> state_;

protected:
  virtual bool open(int socket_fd, struct sockaddr *address, size_t address_len)
  {
    return false;
  }
  virtual void setOptions(int socket_fd);
  
  
  bool setup(std::string &host, int port);
  void close();


public:
  TCPSocket();
  virtual ~TCPSocket();

  SocketState getState() { return state_; }
  
  int getSocketFD() { return socket_fd_; }
  bool setSocketFD(int socket_fd);

  bool read(uint8_t* buf, size_t buf_len, size_t &read);
  bool write(const uint8_t* buf, size_t buf_len, size_t &written);
};