Socket.hh

Go to the documentation of this file.

#ifndef COSA_SOCKET_HH
#define COSA_SOCKET_HH

#include "Cosa/Types.h"
#include "Cosa/INET.hh"
#include "Cosa/IOStream.hh"

class Socket : public IOStream::Device {
public:
  static const uint16_t DYNAMIC_PORT = 49152;

  enum Protocol {
    TCP = SOCK_STREAM,
    UDP = SOCK_DGRAM,
    IPRAW = SOCK_RAW,
    MACRAW = SOCK_RDM,
    PPPoE = SOCK_SEQPACKET
  } __attribute__((packed));

  Socket() :
    IOStream::Device(),
    m_src(),
    m_proto(0),
    m_port(0)
  {
  }

  void src(INET::addr_t& addr) const
  {
    addr = m_src;
  }

  Protocol proto() const
  {
    return ((Protocol) m_proto);
  }

  uint16_t port() const
  {
    return (m_port);
  }

  using IOStream::Device::write;

  virtual int write(const void* buf, size_t size)
  {
    return (write(buf, size, false));
  }

  virtual int write_P(const void* buf, size_t size)
  {
    return (write(buf, size, true));
  }

  virtual int getchar()
  {
    char c;
    int res = recv(&c, sizeof(c));
    return (res == sizeof(c) ? (c & 0xff) : res);
  }

  using IOStream::Device::read;

  virtual int read(void* buf, size_t size)
  {
    return (recv(buf, size));
  }

  virtual int open(Protocol proto, uint16_t port, uint8_t flag) = 0;

  virtual int close() = 0;

  virtual int listen() = 0;

  virtual int accept() = 0;

  virtual int connect(uint8_t addr[4], uint16_t port) = 0;

  virtual int connect(const char* hostname, uint16_t port) = 0;

  virtual int is_connected() = 0;

  virtual int disconnect() = 0;

  virtual int datagram(uint8_t addr[4], uint16_t port) = 0;

  int send(const void* buf, size_t len)
  {
    return (send(buf, len, false));
  }

  int send_P(const void* buf, size_t len)
  {
    return (send(buf, len, true));
  }

  virtual int recv(void* buf, size_t len) = 0;

  int send(const void* buf, size_t len, uint8_t dest[4], uint16_t port)
  {
    return (send(buf, len, dest, port, false));
  }

  int send_P(const void* buf, size_t len, uint8_t dest[4], uint16_t port)
  {
    return (send(buf, len, dest, port, true));
  }

  virtual int recv(void* buf, size_t len, uint8_t src[4], uint16_t& port) = 0;

protected:
  INET::addr_t m_src;

  uint8_t m_proto;

  uint16_t m_port;

  virtual int write(const void* buf, size_t size, bool progmem)
  {
    return (send(buf, size, progmem));
  }

  virtual int send(const void* buf, size_t len, bool progmem) = 0;

  virtual int send(const void* buf, size_t len,
                   uint8_t dest[4], uint16_t port,
                   bool progmem) = 0;
};
#endif