diff options
Diffstat (limited to 'code/core/posix/transport.c')
-rw-r--r-- | code/core/posix/transport.c | 114 |
1 files changed, 114 insertions, 0 deletions
diff --git a/code/core/posix/transport.c b/code/core/posix/transport.c new file mode 100644 index 0000000..1787c66 --- /dev/null +++ b/code/core/posix/transport.c @@ -0,0 +1,114 @@ +#include <stdlib.h> +#include <string.h> +#include <unistd.h> +#include <netinet/in.h> +#include <arpa/inet.h> +#include <poll.h> +#include <stdio.h> + +#include <core.h> + +#define ADDR_S_MAX 32 + +static int t_addr_eq(ECPNetAddr *addr1, ECPNetAddr *addr2) { + if (addr1->port != addr2->port) return 0; + if (memcmp(addr1->host, addr2->host, sizeof(addr1->host)) != 0) return 0; + return 1; +} + +static int t_addr_set(ECPNetAddr *addr, void *addr_s) { + int rv; + char addr_c[ADDR_S_MAX]; + char *colon = NULL; + char *endptr = NULL; + + memset(addr_c, 0, sizeof(addr_c)); + strncpy(addr_c, addr_s, ADDR_S_MAX-1); + colon = strchr(addr_c, ':'); + if (colon == NULL) return -1; + *colon = '\0'; + colon++; + if (*colon == '\0') return -1; + rv = inet_pton(AF_INET, addr_c, addr->host); + if (rv != 1) return -1; + addr->port = strtol(colon, &endptr, 10); + if (*endptr != '\0') return -1; + + return 0; +} + +static int t_open(int *sock, void *addr_s) { + struct sockaddr_in _myaddr; + + memset((char *)&_myaddr, 0, sizeof(_myaddr)); + _myaddr.sin_family = AF_INET; + if (addr_s) { + ECPNetAddr addr; + int rv = t_addr_set(&addr, addr_s); + if (rv) return rv; + + memcpy((void *)&_myaddr.sin_addr, addr.host, sizeof(addr.host)); + _myaddr.sin_port = htons(addr.port); + } else { + _myaddr.sin_addr.s_addr = htonl(INADDR_ANY); + _myaddr.sin_port = htons(0); + } + + *sock = socket(PF_INET, SOCK_DGRAM, 0); + if (*sock < 0) return *sock; + + int rv = bind(*sock, (struct sockaddr *)&_myaddr, sizeof(_myaddr)); + if (rv < 0) { + close(*sock); + return rv; + } + return 0; +} + +static void t_close(int *sock) { + close(*sock); +} + +static int t_poll(int *sock, int timeout) { + struct pollfd fds[] = { + {*sock, POLLIN, 0} + }; + return poll(fds, 1, timeout); +} + +static ssize_t t_send(int *sock, void *msg, size_t msg_size, ECPNetAddr *addr) { + struct sockaddr_in servaddr; + + memset((void *)&servaddr, 0, sizeof(servaddr)); + servaddr.sin_family = AF_INET; + servaddr.sin_port = htons(addr->port); + memcpy((void *)&servaddr.sin_addr, addr->host, sizeof(addr->host)); + return sendto(*sock, msg, msg_size, 0, (struct sockaddr *)&servaddr, sizeof(servaddr)); +} + +static ssize_t t_recv(int *sock, void *msg, size_t msg_size, ECPNetAddr *addr) { + struct sockaddr_in servaddr; + socklen_t addrlen = sizeof(servaddr); + memset((void *)&servaddr, 0, sizeof(servaddr)); + + ssize_t recvlen = recvfrom(*sock, msg, msg_size, 0, (struct sockaddr *)&servaddr, &addrlen); + if (recvlen < 0) return recvlen; + + if (addr) { + addr->port = ntohs(servaddr.sin_port); + memcpy(addr->host, (void *)&servaddr.sin_addr, sizeof(addr->host)); + } + return recvlen; +} + +int ecp_transport_init(ECPTransportIface *t) { + t->init = 1; + t->open = t_open; + t->close = t_close; + t->poll = t_poll; + t->send = t_send; + t->recv = t_recv; + t->addr_eq = t_addr_eq; + t->addr_set = t_addr_set; + return 0; +} |