native_ethernet_server.cpp

Go to the documentation of this file.

#include <net/ethernet_client.h>
 
#include "native_ethernet_client.h"
#include "native_ethernet_server.h"
 
#include <iostream>
#include <memory>
#include <string>
 
namespace net {
 
bool net::NativeEthernetServerHAL::begin(uint16_t port) {
  running = true;
 
  // Create socket
  server_fd = socket(AF_INET, SOCK_STREAM, 0);
#ifdef _WIN64
  if (server_fd == INVALID_SOCKET) {
#else
  if (server_fd == -1) {
#endif
    std::cerr << "Socket creation failed: " <<
#ifdef _WIN64
        WSAGetLastError()
#else
        strerror(errno)
#endif
              << std::endl;
    return false;
  }
 
  // Set socket options to reuse address and port
  int opt = 1;
#ifdef _WIN64
  if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEADDR, reinterpret_cast<const char *>(&opt), sizeof(opt))) {
#else
  if (setsockopt(server_fd, SOL_SOCKET, SO_REUSEPORT, &opt, sizeof(opt))) {
#endif
#ifdef _WIN64
 
#else
    std::cerr << "setsockopt failed: " << strerror(errno) << std::endl;
#endif
 
    return false;
  }
 
  // Prepare the sockaddr_in structure
  sockaddr_in address{};
  address.sin_family = AF_INET;
  address.sin_addr.s_addr = INADDR_ANY;
  address.sin_port = htons(port);
 
  // Bind the socket
  if (bind(server_fd, reinterpret_cast<sockaddr *>(&address), sizeof(address)) < 0) {
#ifdef _WIN64
#else
 
    std::cerr << "Bind failed: " << strerror(errno) << std::endl;
#endif
 
    return false;
  }
 
  // Listen for connections
  if (listen(server_fd, 8) < 0) {
#ifdef _WIN64
#else
    std::cerr << "Listen failed: " << strerror(errno) << std::endl;
#endif
 
    return false;
  }
 
// Set non-blocking mode
#ifdef _WIN64
  u_long mode = 1;
  ioctlsocket(server_fd, FIONBIO, &mode);
#else
  int flags = fcntl(server_fd, F_GETFL);
  fcntl(server_fd, F_SETFL, flags | O_NONBLOCK);
#endif
 
  std::cout << "Server is listening on port " << port << std::endl;
  return true;
}
 
net::EthernetClient net::NativeEthernetServerHAL::accept() {
  sockaddr_in client_addr{};
#ifdef _WIN64
  int client_len = sizeof(client_addr);
#else
  socklen_t client_len = sizeof(client_addr);
#endif
 
  auto client_socket = ::accept(server_fd, reinterpret_cast<struct sockaddr *>(&client_addr), &client_len);
 
#ifdef _WIN64
  if (client_socket != INVALID_SOCKET) {
#else
  if (client_socket >= 0) {
#endif
    char client_ip[INET_ADDRSTRLEN];
    inet_ntop(AF_INET, &client_addr.sin_addr, client_ip, INET_ADDRSTRLEN);
    std::cout << "New connection from " << client_ip << ":" << ntohs(client_addr.sin_port) << std::endl;
  }
 
  return EthernetClient(std::make_unique<NativeEthernetClientHAL>(client_socket));
}
 
// Add destructor to properly clean up
net::NativeEthernetServerHAL::~NativeEthernetServerHAL() {
  if (running) {
#ifdef _WIN64
    closesocket(server_fd);
#else
    close(server_fd);
#endif
    running = false;
  }
}
 
} // namespace net