in_memory_ethernet.h

Go to the documentation of this file.

 
#pragma once
 
#include <chrono>
#include <condition_variable>
#include <cstdint>
#include <deque>
#include <memory>
#include <mutex>
#include <vector>
 
#include "profinet/IRawEthernetSocket.h"
#include "profinet/util.h"
 
namespace profinet::sim
{
 
class InMemoryEthernetBus;
 
class InMemoryEthernetSocket final : public IRawEthernetSocket
{
 public:
  InMemoryEthernetSocket(InMemoryEthernetBus& bus, MacAddress localMac);
 
  ~InMemoryEthernetSocket() override;
 
  InMemoryEthernetSocket(const InMemoryEthernetSocket&) = delete;
  InMemoryEthernetSocket& operator=(const InMemoryEthernetSocket&) = delete;
 
  void Send(const std::vector<std::uint8_t>& frame) override;
 
  void SetTimeout(std::chrono::milliseconds timeout) override;
 
  [[nodiscard]] std::vector<std::uint8_t> Recv() override;
 
  void SetReceiveEnabled(bool enabled);
 
  [[nodiscard]] bool ReceiveEnabled() const noexcept;
 
  [[nodiscard]] const MacAddress& LocalMac() const noexcept
  {
    return localMac;
  }
 
 private:
  friend class InMemoryEthernetBus;
 
  void Enqueue(const std::vector<std::uint8_t>& frame);
 
  InMemoryEthernetBus& bus;
  MacAddress localMac{};
  std::chrono::milliseconds timeout{1};
  mutable std::mutex mutex;
  std::condition_variable condition;
  std::deque<std::vector<std::uint8_t>> receiveQueue;
  bool connected = true;
  bool receiveEnabled = true;
};
 
class InMemoryEthernetBus
{
 public:
  InMemoryEthernetBus() = default;
  ~InMemoryEthernetBus() = default;
 
  InMemoryEthernetBus(const InMemoryEthernetBus&) = delete;
  InMemoryEthernetBus& operator=(const InMemoryEthernetBus&) = delete;
 
  [[nodiscard]] std::unique_ptr<InMemoryEthernetSocket> CreateSocket(MacAddress localMac);
 
  [[nodiscard]] std::size_t EndpointCount() const;
 
  void ClearQueues();
 
 private:
  friend class InMemoryEthernetSocket;
 
  void Attach(InMemoryEthernetSocket* socket);
 
  void Detach(InMemoryEthernetSocket* socket);
 
  void Transmit(InMemoryEthernetSocket& sender,
                const std::vector<std::uint8_t>& frame);
 
  mutable std::mutex mutex;
  std::vector<InMemoryEthernetSocket*> endpoints;
};
 
}  // namespace profinet::sim