native_ethernet_udp.h

Go to the documentation of this file.

#pragma once
 
#ifdef _WIN64
#include <winsock2.h>
#pragma comment(lib, "ws2_32.lib")
#else
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <sys/ioctl.h>
#include <unistd.h>
#endif
 
#include <Arduino.h>
#include <Udp.h>
#include <IPAddress.h>
#include <stdexcept>
 
#include <net/ethernet_client_hal.h>
 
namespace net {
 
class EthernetServer;
 
#ifdef _WIN32
#include <winsock2.h>
#include <ws2tcpip.h>
#pragma comment(lib, "ws2_32.lib")
typedef int socklen_t;
#else
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <unistd.h>
#include <fcntl.h>
#define SOCKET int
#define INVALID_SOCKET -1
#define SOCKET_ERROR -1
#define closesocket close
#endif
 
class NativeEthernetUDP : public VirtualUDP {
private:
  SOCKET _socket;
  struct sockaddr_in _localAddr;
  struct sockaddr_in _remoteAddr;
  uint16_t _localPort;
  uint8_t* _sendBuffer;
  size_t _sendBufferSize;
  size_t _sendBufferPos;
  uint8_t* _recvBuffer;
  size_t _recvBufferSize;
  size_t _recvBufferPos;
  size_t _recvDataSize;
  bool _hasData;
 
  static bool _wsaInitialized;
public:
  NativeEthernetUDP();
  virtual ~NativeEthernetUDP();
 
  uint8_t begin(uint16_t) override;    // initialize, start listening on specified port. Returns 1 if successful, 0 if there are no sockets available to use
  uint8_t beginMulticast(IPAddress, uint16_t) override;
  void stop() override;  // Finish with the UDP socket
 
  // Sending UDP packets
 
  // Start building up a packet to send to the remote host specific in ip and port
  // Returns 1 if successful, 0 if there was a problem with the supplied IP address or port
  int beginPacket(IPAddress ip, uint16_t port) override;
  // Start building up a packet to send to the remote host specific in host and port
  // Returns 1 if successful, 0 if there was a problem resolving the hostname or port
  int beginPacket(const char *host, uint16_t port) override;
  // Finish off this packet and send it
  // Returns 1 if the packet was sent successfully, 0 if there was an error
  int endPacket() override;
  // Write a single byte into the packet
  size_t write(uint8_t) override;
  // Write size bytes from buffer into the packet
  size_t write(const uint8_t *buffer, size_t size) override;
 
  // Start processing the next available incoming packet
  // Returns the size of the packet in bytes, or 0 if no packets are available
  int parsePacket() override;
  // Number of bytes remaining in the current packet
  int available() override;
  // Read a single byte from the current packet
  int read() override;
  // Read up to len bytes from the current packet and place them into buffer
  // Returns the number of bytes read, or 0 if none are available
  int read(unsigned char* buffer, size_t len) override;
  // Read up to len characters from the current packet and place them into buffer
  // Returns the number of characters read, or 0 if none are available
  int read(char* buffer, size_t len) override;
  // Return the next byte from the current packet without moving on to the next byte
  int peek() override;
  void flush() override;   // Finish reading the current packet
 
  // Return the IP address of the host who sent the current incoming packet
  IPAddress remoteIP() override;
  // Return the port of the host who sent the current incoming packet
  uint16_t remotePort() override;
 
  friend class NativeEthernetServerHAL;
};
 
} // namespace net