120 lines
2.8 KiB
C++
120 lines
2.8 KiB
C++
/*
|
||
* eth_udp_client.hpp
|
||
*
|
||
* Created on: 15 <20><><EFBFBD>. 2023 <20>.
|
||
* Author: sychev
|
||
*/
|
||
|
||
#ifndef FREE_RTOS_ETHERNET_IP_ETH_UDP_CLIENT_HPP_
|
||
#define FREE_RTOS_ETHERNET_IP_ETH_UDP_CLIENT_HPP_
|
||
|
||
#include "free_rtos/semaphore/semaphore.hpp"
|
||
#include "free_rtos/mutex/mutex.hpp"
|
||
#include "free_rtos/ethernet/eth_frame.h"
|
||
#include "free_rtos/ethernet_ip/eth_ip_iface.hpp"
|
||
#include "free_rtos/ethernet/eth_types.h"
|
||
|
||
#include <array>
|
||
#include <vector>
|
||
#include <atomic>
|
||
#include <iterator>
|
||
#include <cstdint>
|
||
|
||
namespace free_rtos {
|
||
|
||
enum UDPToggle : uint32_t {
|
||
UDP_BUF_0 = 0,
|
||
UDP_BUF_1,
|
||
UDP_BUF_NUM,
|
||
UDP_BUSY = UDP_BUF_NUM
|
||
};
|
||
|
||
class EthUpdClient {
|
||
friend class EthUdpServer;
|
||
|
||
public:
|
||
EthUpdClient(EthIpIface& ip_iface, TEthMacPorts port_id,
|
||
uint16_t port_dst_be, uint16_t port_src_be,
|
||
bool use_chksum);
|
||
|
||
bool capture_buffer();
|
||
size_t get_data_size();
|
||
|
||
int32_t read(uint8_t * p_data, uint32_t len);
|
||
int32_t read_async(uint8_t * p_data, uint32_t len);
|
||
|
||
template<class T, class Allocator>
|
||
int32_t read_async(std::vector<T, Allocator>& data, uint32_t len)
|
||
{
|
||
if (len == 0) {
|
||
return 0;
|
||
}
|
||
|
||
if (buff_[toggle_].empty() == true) {
|
||
if(capture_buffer() == true) {
|
||
return 0;
|
||
}
|
||
}
|
||
|
||
return get_data(data, len);
|
||
}
|
||
|
||
bool write(uint32_t ip_dst_be, uint8_t * p_data, uint32_t len);
|
||
|
||
template<class T, class Allocator>
|
||
bool write(uint32_t ip_dst_be, std::vector<T, Allocator>& data)
|
||
{
|
||
return write(ip_dst_be, data.data(), data.size() * sizeof(T));
|
||
}
|
||
|
||
void clear();
|
||
|
||
private:
|
||
void put_data(uint32_t src_ip, uint8_t * p_data, uint32_t len);
|
||
int32_t get_data(uint8_t * p_data, uint32_t len);
|
||
|
||
template<class T, class Allocator>
|
||
int32_t get_data(std::vector<T, Allocator>& data, uint32_t len)
|
||
{
|
||
std::vector<uint8_t>& buff = buff_[toggle_];
|
||
uint32_t size = buff.size();
|
||
|
||
if (size > len) {
|
||
size = len;
|
||
}
|
||
|
||
auto item_begin = buff.begin();
|
||
auto item_end = item_begin + size;
|
||
|
||
data.insert(data.end(), item_begin, item_end);
|
||
buff.erase(item_begin, item_end);
|
||
|
||
return size;
|
||
}
|
||
|
||
private:
|
||
const uint32_t max_buf_size = 0x4000; // 0x4000 - 16384, 32768, 0x10000 - 65536, 0x11CA - 4554, 0x0BDC - 3036, 0x100000 - 1Mb
|
||
|
||
const uint16_t port_dst_be_; /// big endian
|
||
const uint16_t port_src_be_; /// big endian
|
||
const bool use_checksum_;
|
||
|
||
EthIpIface& ip_iface_;
|
||
|
||
const TEthMacPorts port_id_;
|
||
|
||
TEthPkt eth_pkt_;
|
||
|
||
uint32_t src_ip_;
|
||
|
||
std::array<std::vector<uint8_t>, UDP_BUF_NUM> buff_;
|
||
std::atomic<UDPToggle> in_toggle_{UDPToggle::UDP_BUF_1};
|
||
UDPToggle out_toggle_{UDPToggle::UDP_BUF_0};
|
||
|
||
Semaphore rx_sem_;
|
||
};
|
||
|
||
}
|
||
|
||
#endif /* FREE_RTOS_ETHERNET_IP_ETH_UDP_CLIENT_HPP_ */
|