155 lines
4.7 KiB
C
155 lines
4.7 KiB
C
#ifndef FEMTOTCP_H
|
|
#define FEMTOTCP_H
|
|
#include <stdint.h>
|
|
|
|
/* Types */
|
|
struct ipstack;
|
|
typedef uint32_t ip4;
|
|
|
|
/* Macros, compiler specific. */
|
|
#define PACKED __attribute__((packed))
|
|
#define ee16(x) __builtin_bswap16(x)
|
|
#define ee32(x) __builtin_bswap32(x)
|
|
#define DEBUG
|
|
|
|
|
|
#ifdef DEBUG
|
|
#include <stdio.h>
|
|
#define LOG(fmt, ...) printf(fmt, ##__VA_ARGS__)
|
|
#else
|
|
#define LOG(fmt, ...) do{}while(0)
|
|
#endif
|
|
|
|
/* Device driver interface */
|
|
/* Struct to contain a hw device description */
|
|
struct ll {
|
|
uint8_t mac[6];
|
|
char ifname[16];
|
|
/* poll function */
|
|
int (*poll)(struct ll *ll, void *buf, uint32_t len);
|
|
/* send function */
|
|
int (*send)(struct ll *ll, void *buf, uint32_t len);
|
|
};
|
|
|
|
/* Struct to contain an IP device configuration */
|
|
struct ipconf {
|
|
struct ll *ll;
|
|
ip4 ip;
|
|
ip4 mask;
|
|
ip4 gw;
|
|
};
|
|
|
|
/* Socket interface */
|
|
#define MARK_TCP_SOCKET 0x100 /* Mark a socket as TCP */
|
|
#define MARK_UDP_SOCKET 0x200 /* Mark a socket as UDP */
|
|
#if (MARK_TCP_SOCKET >= MARK_UDP_SOCKET)
|
|
#error "MARK_TCP_SOCKET must be less than MARK_UDP_SOCKET"
|
|
#endif
|
|
|
|
|
|
#ifndef FEMTO_POSIX
|
|
#define IPSTACK_SOCK_STREAM 1
|
|
#define IPSTACK_SOCK_DGRAM 2
|
|
|
|
|
|
struct ipstack_sockaddr_in {
|
|
uint16_t sin_family;
|
|
uint16_t sin_port;
|
|
struct sin_addr { uint32_t s_addr; } sin_addr;
|
|
};
|
|
struct ipstack_sockaddr { uint16_t sa_family; };
|
|
typedef uint32_t socklen_t;
|
|
#ifndef AF_INET
|
|
#define AF_INET 2
|
|
#endif
|
|
#else
|
|
#include <sys/socket.h>
|
|
#include <netinet/in.h>
|
|
#include <arpa/inet.h>
|
|
#include <unistd.h>
|
|
#define ipstack_sockaddr_in sockaddr_in
|
|
#define ipstack_sockaddr sockaddr
|
|
#endif
|
|
|
|
int ft_socket(struct ipstack *s, int domain, int type, int protocol);
|
|
int ft_bind(struct ipstack *s, int sockfd, const struct ipstack_sockaddr *addr, socklen_t addrlen);
|
|
int ft_listen(struct ipstack *s, int sockfd, int backlog);
|
|
int ft_accept(struct ipstack *s, int sockfd, struct ipstack_sockaddr *addr, socklen_t *addrlen);
|
|
int ft_connect(struct ipstack *s, int sockfd, const struct ipstack_sockaddr *addr, socklen_t addrlen);
|
|
int ft_sendto(struct ipstack *s, int sockfd, const void *buf, size_t len, int flags, const struct ipstack_sockaddr *dest_addr, socklen_t addrlen);
|
|
int ft_send(struct ipstack *s, int sockfd, const void *buf, size_t len, int flags);
|
|
int ft_write(struct ipstack *s, int sockfd, const void *buf, size_t len);
|
|
int ft_recvfrom(struct ipstack *s, int sockfd, void *buf, size_t len, int flags, struct ipstack_sockaddr *src_addr, socklen_t *addrlen);
|
|
int ft_recv(struct ipstack *s, int sockfd, void *buf, size_t len, int flags);
|
|
int ft_read(struct ipstack *s, int sockfd, void *buf, size_t len);
|
|
int ft_close(struct ipstack *s, int sockfd);
|
|
int ft_getpeername(struct ipstack *s, int sockfd, struct ipstack_sockaddr *addr, socklen_t *addrlen);
|
|
int ft_getsockname(struct ipstack *s, int sockfd, struct ipstack_sockaddr *addr, socklen_t *addrlen);
|
|
|
|
int dhcp_client_init(struct ipstack *s);
|
|
int dhcp_bound(struct ipstack *s);
|
|
|
|
/* DNS client */
|
|
|
|
int nslookup(struct ipstack *s, const char *name, uint16_t *id, void (*lookup_cb)(uint32_t ip));
|
|
|
|
/* IP stack interface */
|
|
void ipstack_init(struct ipstack *s);
|
|
void ipstack_init_static(struct ipstack **s);
|
|
int ipstack_poll(struct ipstack *s, uint64_t now);
|
|
void ipstack_recv(struct ipstack *s, void *buf, uint32_t len);
|
|
void ipstack_ipconfig_set(struct ipstack *s, ip4 ip, ip4 mask, ip4 gw);
|
|
void ipstack_ipconfig_get(struct ipstack *s, ip4 *ip, ip4 *mask, ip4 *gw);
|
|
|
|
struct ll *ipstack_getdev(struct ipstack *s);
|
|
|
|
/* Callback flags */
|
|
#define CB_EVENT_READABLE 0x01 /* Accepted connection or data available */
|
|
#define CB_EVENT_TIMEOUT 0x02 /* Timeout */
|
|
#define CB_EVENT_WRITABLE 0x04 /* Connected or space available to send */
|
|
#define CB_EVENT_CLOSED 0x10 /* Connection closed by peer */
|
|
void ipstack_register_callback(struct ipstack *s, int sock_fd, void (*cb)(int sock_fd, uint16_t events, void *arg), void *arg);
|
|
|
|
/* External requirements */
|
|
uint32_t ipstack_getrandom(void);
|
|
|
|
/* Inline utility functions */
|
|
static inline uint32_t atou(const char *s)
|
|
{
|
|
uint32_t ret = 0;
|
|
while (*s >= '0' && *s <= '9') {
|
|
ret = ret * 10 + (*s - '0');
|
|
s++;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static inline ip4 atoip4(const char *ip)
|
|
{
|
|
ip4 ret = 0;
|
|
int i = 0;
|
|
int j = 0;
|
|
for (i = 0; i < 4; i++) {
|
|
ret |= (atou(ip + j) << (24 - i * 8));
|
|
while (ip[j] != '.' && ip[j] != '\0') j++;
|
|
if (ip[j] == '\0') break;
|
|
j++;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
static inline void iptoa(ip4 ip, char *buf)
|
|
{
|
|
int i, j = 0;
|
|
buf[0] = 0;
|
|
for (i = 0; i < 4; i++) {
|
|
uint8_t x = (ip >> (24 - i * 8)) & 0xFF;
|
|
if (x > 99) buf[j++] = x / 100 + '0';
|
|
if (x > 9) buf[j++] = (x / 10) % 10 + '0';
|
|
buf[j++] = x % 10 + '0';
|
|
if (i < 3) buf[j++] = '.';
|
|
}
|
|
buf[j] = 0;
|
|
}
|
|
|
|
#endif
|