Mirror
/
ardupilot
mirror of https://github.com/ArduPilot/ardupilot
5
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

AP_HAL: added sendto socket method

This commit is contained in:
Andrew Tridgell 2015-05-22 15:53:33 +10:00
parent 4e117bc90f
commit 9336914598
2 changed files with 31 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
libraries/AP_HAL/utility/Socket.cpp
View File

@ -19,15 +19,6 @@
#include <AP_HAL.h>
#if HAL_OS_SOCKETS
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <sys/select.h>
#include "Socket.h"
/*
@ -42,12 +33,8 @@ datagram(_datagram)
setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &one, sizeof(one));
}
/*
connect the socket
*/
bool SocketAPM::connect(const char *address, uint16_t port)
void SocketAPM::make_sockaddr(const char *address, uint16_t port, struct sockaddr_in &sockaddr)
{
struct sockaddr_in sockaddr;
memset(&sockaddr, 0, sizeof(sockaddr));
#ifdef HAVE_SOCK_SIN_LEN
@ -56,6 +43,15 @@ bool SocketAPM::connect(const char *address, uint16_t port)
sockaddr.sin_port = htons(port);
sockaddr.sin_family = AF_INET;
sockaddr.sin_addr.s_addr = inet_addr(address);
}
/*
connect the socket
*/
bool SocketAPM::connect(const char *address, uint16_t port)
{
struct sockaddr_in sockaddr;
make_sockaddr(address, port, sockaddr);
if (::connect(fd, (struct sockaddr *)&sockaddr, sizeof(sockaddr)) != 0) {
return false;
@ -69,14 +65,7 @@ bool SocketAPM::connect(const char *address, uint16_t port)
bool SocketAPM::bind(const char *address, uint16_t port)
{
struct sockaddr_in sockaddr;
memset(&sockaddr, 0, sizeof(sockaddr));
#ifdef HAVE_SOCK_SIN_LEN
sockaddr.sin_len = sizeof(sockaddr);
#endif
sockaddr.sin_port = htons(port);
sockaddr.sin_family = AF_INET;
sockaddr.sin_addr.s_addr = inet_addr(address);
make_sockaddr(address, port, sockaddr);
if (::bind(fd, (struct sockaddr *)&sockaddr, sizeof(sockaddr)) != 0) {
return false;
@ -114,6 +103,16 @@ ssize_t SocketAPM::send(void *buf, size_t size)
return ::send(fd, buf, size, 0);
}
/*
send some data
*/
ssize_t SocketAPM::sendto(void *buf, size_t size, const char *address, uint16_t port)
{
struct sockaddr_in sockaddr;
make_sockaddr(address, port, sockaddr);
return ::sendto(fd, buf, size, 0, (struct sockaddr *)&sockaddr, sizeof(sockaddr));
}
/*
receive some data
*/

10
libraries/AP_HAL/utility/Socket.h
View File

@ -22,7 +22,14 @@
#include <AP_HAL.h>
#if HAL_OS_SOCKETS
#include <fcntl.h>
#include <unistd.h>
#include <sys/ioctl.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <sys/select.h>
class SocketAPM {
public:
@ -33,11 +40,14 @@ public:
void set_blocking(bool blocking);
ssize_t send(void *pkt, size_t size);
ssize_t sendto(void *buf, size_t size, const char *address, uint16_t port);
ssize_t recv(void *pkt, size_t size, uint32_t timeout_ms);
private:
bool datagram;
int fd;
void make_sockaddr(const char *address, uint16_t port, struct sockaddr_in &sockaddr);
};
#endif // HAL_OS_SOCKETS