mirror of https://github.com/ArduPilot/ardupilot
90 lines
2.1 KiB
C++
90 lines
2.1 KiB
C++
|
#include "AP_DDS_Client.h"
|
||
|
|
|
||
|
|
#if AP_DDS_UDP_ENABLED
|
||
|
|
|
||
|
|
#include <errno.h>
|
||
|
|
|
||
|
|
/*
|
||
|
|
open connection on UDP
|
||
|
|
*/
|
||
|
|
bool AP_DDS_Client::udp_transport_open(uxrCustomTransport *t)
|
||
|
|
{
|
||
|
|
AP_DDS_Client *dds = (AP_DDS_Client *)t->args;
|
||
|
|
auto *sock = new SocketAPM(true);
|
||
|
|
if (sock == nullptr) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
if (!sock->connect(dds->udp.ip, dds->udp.port.get())) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
dds->udp.socket = sock;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
close UDP connection
|
||
|
|
*/
|
||
|
|
bool AP_DDS_Client::udp_transport_close(uxrCustomTransport *t)
|
||
|
|
{
|
||
|
|
AP_DDS_Client *dds = (AP_DDS_Client *)t->args;
|
||
|
|
delete dds->udp.socket;
|
||
|
|
dds->udp.socket = nullptr;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
write on UDP
|
||
|
|
*/
|
||
|
|
size_t AP_DDS_Client::udp_transport_write(uxrCustomTransport *t, const uint8_t* buf, size_t len, uint8_t* error)
|
||
|
|
{
|
||
|
|
AP_DDS_Client *dds = (AP_DDS_Client *)t->args;
|
||
|
|
if (dds->udp.socket == nullptr) {
|
||
|
|
*error = EINVAL;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const ssize_t ret = dds->udp.socket->send(buf, len);
|
||
|
|
if (ret <= 0) {
|
||
|
|
*error = errno;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
read from UDP
|
||
|
|
*/
|
||
|
|
size_t AP_DDS_Client::udp_transport_read(uxrCustomTransport *t, uint8_t* buf, size_t len, int timeout_ms, uint8_t* error)
|
||
|
|
{
|
||
|
|
AP_DDS_Client *dds = (AP_DDS_Client *)t->args;
|
||
|
|
if (dds->udp.socket == nullptr) {
|
||
|
|
*error = EINVAL;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
const ssize_t ret = dds->udp.socket->recv(buf, len, timeout_ms);
|
||
|
|
if (ret <= 0) {
|
||
|
|
*error = errno;
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
initialise UDP connection
|
||
|
|
*/
|
||
|
|
bool AP_DDS_Client::ddsUdpInit()
|
||
|
|
{
|
||
|
|
// setup a non-framed transport for UDP
|
||
|
|
uxr_set_custom_transport_callbacks(&udp.transport, false,
|
||
|
|
udp_transport_open,
|
||
|
|
udp_transport_close,
|
||
|
|
udp_transport_write,
|
||
|
|
udp_transport_read);
|
||
|
|
|
||
|
|
if (!uxr_init_custom_transport(&udp.transport, (void*)this)) {
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
uxr_init_session(&session, &udp.transport.comm, uniqueClientKey);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
#endif // AP_DDS_UDP_ENABLED
|