mirror of https://github.com/ArduPilot/ardupilot
136 lines
3.8 KiB
C++
136 lines
3.8 KiB
C++
#pragma once
|
|
|
|
#include <AP_HAL/AP_HAL.h>
|
|
#if CONFIG_HAL_BOARD == HAL_BOARD_SITL
|
|
|
|
#include <stdint.h>
|
|
#include <stdarg.h>
|
|
#include "AP_HAL_SITL_Namespace.h"
|
|
#include <AP_HAL/utility/Socket.h>
|
|
#include <AP_HAL/utility/RingBuffer.h>
|
|
|
|
#include <SITL/SIM_SerialDevice.h>
|
|
|
|
class HALSITL::UARTDriver : public AP_HAL::UARTDriver {
|
|
public:
|
|
friend class HALSITL::SITL_State;
|
|
|
|
UARTDriver(const uint8_t portNumber, SITL_State *sitlState) {
|
|
_portNumber = portNumber;
|
|
_sitlState = sitlState;
|
|
|
|
_fd = -1;
|
|
_mc_fd = -1;
|
|
_listen_fd = -1;
|
|
}
|
|
|
|
/* Implementations of UARTDriver virtual methods */
|
|
void begin(uint32_t b) override {
|
|
begin(b, 0, 0);
|
|
}
|
|
void begin(uint32_t b, uint16_t rxS, uint16_t txS) override;
|
|
void end() override;
|
|
void flush() override;
|
|
bool is_initialized() override {
|
|
return true;
|
|
}
|
|
|
|
ssize_t get_system_outqueue_length() const;
|
|
|
|
void set_blocking_writes(bool blocking) override
|
|
{
|
|
_nonblocking_writes = !blocking;
|
|
}
|
|
|
|
bool tx_pending() override {
|
|
return false;
|
|
}
|
|
|
|
/* Implementations of Stream virtual methods */
|
|
uint32_t available() override;
|
|
uint32_t txspace() override;
|
|
int16_t read() override;
|
|
ssize_t read(uint8_t *buffer, uint16_t count) override;
|
|
|
|
bool discard_input() override;
|
|
|
|
/* Implementations of Print virtual methods */
|
|
size_t write(uint8_t c) override;
|
|
size_t write(const uint8_t *buffer, size_t size) override;
|
|
|
|
bool _unbuffered_writes;
|
|
|
|
enum flow_control get_flow_control(void) override { return FLOW_CONTROL_ENABLE; }
|
|
|
|
void configure_parity(uint8_t v) override;
|
|
void set_stop_bits(int n) override;
|
|
bool set_unbuffered_writes(bool on) override;
|
|
|
|
void _timer_tick(void) override;
|
|
|
|
/*
|
|
return timestamp estimate in microseconds for when the start of
|
|
a nbytes packet arrived on the uart. This should be treated as a
|
|
time constraint, not an exact time. It is guaranteed that the
|
|
packet did not start being received after this time, but it
|
|
could have been in a system buffer before the returned time.
|
|
|
|
This takes account of the baudrate of the link. For transports
|
|
that have no baudrate (such as USB) the time estimate may be
|
|
less accurate.
|
|
|
|
A return value of zero means the HAL does not support this API
|
|
*/
|
|
uint64_t receive_time_constraint_us(uint16_t nbytes) override;
|
|
|
|
private:
|
|
|
|
int _fd;
|
|
|
|
// file descriptor for reading multicast packets
|
|
int _mc_fd;
|
|
|
|
uint8_t _portNumber;
|
|
bool _connected = false; // true if a client has connected
|
|
bool _use_send_recv = false;
|
|
int _listen_fd; // socket we are listening on
|
|
struct sockaddr_in _listen_sockaddr;
|
|
int _serial_port;
|
|
static bool _console;
|
|
bool _nonblocking_writes;
|
|
ByteBuffer _readbuffer{16384};
|
|
ByteBuffer _writebuffer{16384};
|
|
|
|
// default multicast IP and port
|
|
const char *mcast_ip_default = "239.255.145.50";
|
|
const uint16_t mcast_port_default = 14550;
|
|
|
|
const char *_uart_path;
|
|
uint32_t _uart_baudrate;
|
|
|
|
// IPv4 address of target for uartC
|
|
const char *_tcp_client_addr;
|
|
|
|
void _tcp_start_connection(uint16_t port, bool wait_for_connection);
|
|
void _uart_start_connection(void);
|
|
void _check_reconnect();
|
|
void _tcp_start_client(const char *address, uint16_t port);
|
|
void _udp_start_client(const char *address, uint16_t port);
|
|
void _udp_start_multicast(const char *address, uint16_t port);
|
|
void _check_connection(void);
|
|
static bool _select_check(int );
|
|
static void _set_nonblocking(int );
|
|
bool set_speed(int speed) const;
|
|
|
|
SITL_State *_sitlState;
|
|
uint64_t _receive_timestamp;
|
|
bool _is_udp;
|
|
bool _packetise;
|
|
uint16_t _mc_myport;
|
|
uint32_t last_tick_us;
|
|
|
|
SITL::SerialDevice *_sim_serial_device;
|
|
};
|
|
|
|
#endif
|