Working select wrapper + tests

This commit is contained in:
Daniele Lacamera 2024-11-17 12:59:48 +01:00
parent 579433eddb
commit c5d20305c3
4 changed files with 204 additions and 52 deletions

View file

@ -6,7 +6,8 @@ LDFLAGS+=-pthread
OBJ=build/femtotcp.o \
build/port/posix/linux_tap.o
EXE=build/tcpecho build/test-evloop
EXE=build/tcpecho build/tcp_netcat_poll build/tcp_netcat_select \
build/test-evloop
LIB=libfemtotcp.so
@ -47,6 +48,14 @@ build/tcpecho: $(OBJ) build/port/posix/bsd_socket.o build/test/tcp_echo.o
@echo "[LD] $@"
@$(CC) $(CFLAGS) $(LDFLAGS) -o $@ -Wl,--start-group $(^) -Wl,--end-group
build/tcp_netcat_poll: $(OBJ) build/port/posix/bsd_socket.o build/test/tcp_netcat_poll.o
@echo "[LD] $@"
@$(CC) $(CFLAGS) $(LDFLAGS) -o $@ -Wl,--start-group $(^) -Wl,--end-group
build/tcp_netcat_select: $(OBJ) build/port/posix/bsd_socket.o build/test/tcp_netcat_select.o
@echo "[LD] $@"
@$(CC) $(CFLAGS) $(LDFLAGS) -o $@ -Wl,--start-group $(^) -Wl,--end-group
build/%.o: src/%.c
@mkdir -p `dirname $@` || true
@echo "[CC] $<"

View file

@ -1047,13 +1047,13 @@ static void tcp_input(struct ipstack *S, struct ipstack_tcp_seg *tcp, uint32_t f
t->sock.tcp.state = TCP_CLOSE_WAIT;
t->sock.tcp.ack = ee32(tcp->seq) + 1;
tcp_send_ack(t);
t->events |= CB_EVENT_CLOSED;
t->events |= CB_EVENT_CLOSED | CB_EVENT_READABLE;
}
else if (t->sock.tcp.state == TCP_FIN_WAIT_1) {
t->sock.tcp.state = TCP_CLOSING;
t->sock.tcp.ack = ee32(tcp->seq) + 1;
tcp_send_ack(t);
t->events |= CB_EVENT_CLOSED;
t->events |= CB_EVENT_CLOSED | CB_EVENT_READABLE;
}
}
@ -1105,7 +1105,7 @@ static void tcp_input(struct ipstack *S, struct ipstack_tcp_seg *tcp, uint32_t f
t->sock.tcp.state = TCP_CLOSING;
}
t->sock.tcp.ack = ee32(tcp->seq) + 1;
t->events |= CB_EVENT_CLOSED;
t->events |= CB_EVENT_CLOSED | CB_EVENT_READABLE;
tcp_send_ack(t);
}
if (tcp->flags & 0x10) {

View file

@ -10,12 +10,14 @@
#include <dlfcn.h>
#include <sys/time.h>
#include <poll.h>
#include <string.h>
#define FEMTO_POSIX
#include "config.h"
#include "femtotcp.h"
static __thread int in_the_stack = 1;
static struct ipstack *IPSTACK = NULL;
pthread_mutex_t ipstack_mutex = PTHREAD_MUTEX_INITIALIZER;
/* host_ functions are the original functions from the libc */
static int (*host_socket ) (int domain, int type, int protocol) = NULL;
@ -54,14 +56,18 @@ static int (*host_fcntl) (int fd, int cmd, ...);
if(in_the_stack) { \
return host_##call(fd, ## __VA_ARGS__); \
} else { \
pthread_mutex_lock(&ipstack_mutex); \
if ((fd & (MARK_TCP_SOCKET | MARK_UDP_SOCKET)) != 0) { \
int __femto_retval = ft_##call(IPSTACK, fd, ## __VA_ARGS__); \
if (__femto_retval < 0) { \
errno = __femto_retval; \
pthread_mutex_unlock(&ipstack_mutex); \
return -1; \
} \
pthread_mutex_unlock(&ipstack_mutex); \
return __femto_retval; \
}else { \
pthread_mutex_unlock(&ipstack_mutex); \
return host_##call(fd, ## __VA_ARGS__); \
} \
}
@ -70,6 +76,7 @@ static int (*host_fcntl) (int fd, int cmd, ...);
if(in_the_stack) { \
return host_##call(fd, ## __VA_ARGS__); \
} else { \
pthread_mutex_lock(&ipstack_mutex); \
if ((fd & (MARK_TCP_SOCKET | MARK_UDP_SOCKET)) != 0) { \
int __femto_retval; \
do { \
@ -80,10 +87,13 @@ static int (*host_fcntl) (int fd, int cmd, ...);
} while (__femto_retval == -11); \
if (__femto_retval < 0) { \
errno = __femto_retval; \
pthread_mutex_unlock(&ipstack_mutex); \
return -1; \
} \
pthread_mutex_unlock(&ipstack_mutex); \
return __femto_retval; \
}else { \
pthread_mutex_unlock(&ipstack_mutex); \
return host_##call(fd, ## __VA_ARGS__); \
} \
}
@ -126,10 +136,14 @@ int fcntl(int fd, int cmd, ...) {
va_start(ap, cmd);
arg = va_arg(ap, int);
va_end(ap);
int ret;
if (in_the_stack) {
return host_fcntl(fd, cmd, arg);
} else {
return ft_fcntl(IPSTACK, fd, cmd, arg);
pthread_mutex_lock(&ipstack_mutex);
ret = ft_fcntl(IPSTACK, fd, cmd, arg);
pthread_mutex_unlock(&ipstack_mutex);
return ret;
}
}
@ -171,10 +185,13 @@ void poller_callback(int fd, uint16_t event, void *arg)
int ft_poll(struct ipstack *ipstack, struct pollfd *fds, nfds_t nfds, int timeout) {
nfds_t i;
int fd;
int ret;
printf("Called poll\n");
if (in_the_stack) {
return host_poll(fds, nfds, timeout);
}
memset(tcp_pollers, 0, sizeof(tcp_pollers));
memset(udp_pollers, 0, sizeof(udp_pollers));
for (i = 0; i < nfds; i++) {
fd = fds[i].fd;
struct bsd_poll_helper *poller;
@ -201,7 +218,9 @@ int ft_poll(struct ipstack *ipstack, struct pollfd *fds, nfds_t nfds, int timeou
}
/* Call the original poll */
printf("Calling host_poll\n");
int ret = host_poll(fds, nfds, timeout);
pthread_mutex_unlock(&ipstack_mutex);
ret = host_poll(fds, nfds, timeout);
pthread_mutex_lock(&ipstack_mutex);
for (i = 0; i < nfds; i++) {
struct bsd_poll_helper *poller;
if ((fds[i].fd & MARK_TCP_SOCKET) != 0)
@ -213,7 +232,7 @@ int ft_poll(struct ipstack *ipstack, struct pollfd *fds, nfds_t nfds, int timeou
printf("Replacing events\n");
if (fds[i].revents & POLLIN) {
char c;
read(poller->pipefds[0], &c, 1);
host_read(poller->pipefds[0], &c, 1);
switch(c) {
case 'r':
fds[i].revents |= POLLIN;
@ -231,8 +250,8 @@ int ft_poll(struct ipstack *ipstack, struct pollfd *fds, nfds_t nfds, int timeou
fds[i].revents &= fds[i].events & (POLLHUP | POLLERR);
}
printf("Closing pipes\n");
close(poller->pipefds[0]);
close(poller->pipefds[1]);
host_close(poller->pipefds[0]);
host_close(poller->pipefds[1]);
poller->fd = 0;
fds[i].fd = poller->fd;
fds[i].events = poller->events;
@ -246,74 +265,85 @@ int ft_select(struct ipstack *ipstack, int nfds, fd_set *readfds, fd_set *writef
int i;
int maxfd;
int ret;
printf("Called select\n");
fd_set readfds_local;
/* Assume MARK_TCP_SOCKET < MARK_UDP_SOCKET */
if (nfds < MARK_TCP_SOCKET + 1) {
return host_select(nfds, readfds, writefds, exceptfds, timeout);
}
for (i = 0; i < MARK_TCP_SOCKET && i < nfds; i++) {
if (FD_ISSET(i, readfds) || FD_ISSET(i, writefds) || FD_ISSET(i, exceptfds)) {
memset(tcp_pollers, 0, sizeof(tcp_pollers));
memset(udp_pollers, 0, sizeof(udp_pollers));
for (i = 0; (i < MARK_TCP_SOCKET) && (i < nfds); i++) {
if ((readfds && FD_ISSET(i, readfds)) ||
(writefds && FD_ISSET(i, writefds)) ||
(exceptfds && FD_ISSET(i, exceptfds))) {
maxfd = i;
}
}
for (i = MARK_TCP_SOCKET; i < nfds && i < MAX_TCPSOCKETS; i++) {
if (FD_ISSET(i, readfds) || FD_ISSET(i, writefds) || FD_ISSET(i, exceptfds)) {
/* At this point, we do need a fd_set to read from pipes */
if (!readfds) {
FD_ZERO(&readfds_local);
readfds = &readfds_local;
}
for (i = MARK_TCP_SOCKET; i < nfds && i < (MARK_TCP_SOCKET | MAX_TCPSOCKETS); i++) {
int tcp_pos = i & (~MARK_TCP_SOCKET);
printf("Found TCP %d\n", i);
pipe(tcp_pollers[tcp_pos].pipefds);
if ((readfds && (FD_ISSET(i, readfds))) || (writefds && (FD_ISSET(i, writefds))) || (exceptfds && (FD_ISSET(i, exceptfds)))) {
if (pipe(tcp_pollers[tcp_pos].pipefds) < 0)
return -1;
tcp_pollers[tcp_pos].fd = i;
tcp_pollers[tcp_pos].events = 0;
ipstack_register_callback(ipstack, i, poller_callback, ipstack);
if (FD_ISSET(i, readfds)) {
if (readfds && (FD_ISSET(i, readfds))) {
tcp_pollers[tcp_pos].events |= POLLIN;
FD_CLR(i, readfds);
FD_SET(tcp_pollers[tcp_pos].pipefds[1], readfds);
FD_SET(tcp_pollers[tcp_pos].pipefds[0], readfds);
}
if (FD_ISSET(i, writefds)) {
if (writefds && (FD_ISSET(i, writefds))) {
tcp_pollers[tcp_pos].events |= POLLOUT;
FD_CLR(i, writefds);
FD_SET(tcp_pollers[tcp_pos].pipefds[1], writefds);
FD_SET(tcp_pollers[tcp_pos].pipefds[0], writefds);
}
if (FD_ISSET(i, exceptfds)) {
if (exceptfds && (FD_ISSET(i, exceptfds))) {
tcp_pollers[tcp_pos].events |= POLLERR | POLLHUP;
FD_CLR(i, exceptfds);
FD_SET(tcp_pollers[tcp_pos].pipefds[1], exceptfds);
FD_SET(tcp_pollers[tcp_pos].pipefds[0], exceptfds);
}
if (maxfd < tcp_pollers[tcp_pos].pipefds[1]) {
maxfd = tcp_pollers[tcp_pos].pipefds[1];
if (maxfd < tcp_pollers[tcp_pos].pipefds[0]) {
maxfd = tcp_pollers[tcp_pos].pipefds[0];
}
} else {
}
}
}
for (i = MARK_UDP_SOCKET; i < nfds && i < MAX_UDPSOCKETS; i++) {
if (FD_ISSET(i, readfds) || FD_ISSET(i, writefds) || FD_ISSET(i, exceptfds)) {
for (i = MARK_UDP_SOCKET; i < nfds && i < (MARK_UDP_SOCKET | MAX_UDPSOCKETS); i++) {
int udp_pos = i & (~MARK_UDP_SOCKET);
printf("Found TCP %d\n", i);
if (FD_ISSET(i, readfds) || FD_ISSET(i, writefds) || FD_ISSET(i, exceptfds)) {
pipe(udp_pollers[udp_pos].pipefds);
udp_pollers[udp_pos].fd = i;
udp_pollers[udp_pos].events = 0;
ipstack_register_callback(ipstack, i, poller_callback, ipstack);
if (FD_ISSET(i, readfds)) {
if (readfds && FD_ISSET(i, readfds)) {
udp_pollers[udp_pos].events |= POLLIN;
FD_CLR(i, readfds);
FD_SET(udp_pollers[udp_pos].pipefds[1], readfds);
FD_SET(udp_pollers[udp_pos].pipefds[0], readfds);
}
if (FD_ISSET(i, writefds)) {
if (writefds && FD_ISSET(i, writefds)) {
udp_pollers[udp_pos].events |= POLLOUT;
FD_CLR(i, writefds);
FD_SET(udp_pollers[udp_pos].pipefds[1], writefds);
FD_SET(udp_pollers[udp_pos].pipefds[0], writefds);
}
if (FD_ISSET(i, exceptfds)) {
if (exceptfds && FD_ISSET(i, exceptfds)) {
udp_pollers[udp_pos].events |= POLLERR | POLLHUP;
FD_CLR(i, exceptfds);
FD_SET(udp_pollers[udp_pos].pipefds[1], exceptfds);
FD_SET(udp_pollers[udp_pos].pipefds[0], exceptfds);
}
if (maxfd < udp_pollers[udp_pos].pipefds[1]) {
maxfd = udp_pollers[udp_pos].pipefds[1];
if (maxfd < udp_pollers[udp_pos].pipefds[0]) {
maxfd = udp_pollers[udp_pos].pipefds[0];
}
}
}
/* Call the original select */
pthread_mutex_unlock(&ipstack_mutex);
ret = host_select(maxfd + 1, readfds, writefds, exceptfds, timeout);
pthread_mutex_lock(&ipstack_mutex);
if (ret <= 0) {
return ret;
}
@ -322,39 +352,39 @@ int ft_select(struct ipstack *ipstack, int nfds, fd_set *readfds, fd_set *writef
if (tcp_pollers[i].fd == 0) {
continue;
}
if (FD_ISSET(tcp_pollers[i].pipefds[1], readfds)) {
if (FD_ISSET(tcp_pollers[i].pipefds[0], readfds)) {
char c;
read(tcp_pollers[i].pipefds[1], &c, 1);
if (c == 'r') {
host_read(tcp_pollers[i].pipefds[0], &c, 1);
if (readfds && (c == 'r')) {
FD_SET(tcp_pollers[i].fd, readfds);
} else if (c == 'w') {
} else if (writefds && (c == 'w')) {
FD_SET(tcp_pollers[i].fd, writefds);
} else if (c == 'e') {
} else if (exceptfds && (c == 'e')) {
FD_SET(tcp_pollers[i].fd, exceptfds);
}
}
close(tcp_pollers[i].pipefds[0]);
close(tcp_pollers[i].pipefds[1]);
ipstack_register_callback(ipstack, tcp_pollers[i].fd, NULL, NULL);
host_close(tcp_pollers[i].pipefds[0]);
host_close(tcp_pollers[i].pipefds[1]);
tcp_pollers[i].fd = 0;
}
for (i = 0; i < MAX_UDPSOCKETS; i++) {
if (udp_pollers[i].fd == 0) {
continue;
}
if (FD_ISSET(udp_pollers[i].pipefds[1], readfds)) {
if (FD_ISSET(udp_pollers[i].pipefds[0], readfds)) {
char c;
read(udp_pollers[i].pipefds[1], &c, 1);
if (c == 'r') {
read(udp_pollers[i].pipefds[0], &c, 1);
if (readfds && (c == 'r')) {
FD_SET(udp_pollers[i].fd, readfds);
} else if (c == 'w') {
} else if (writefds && (c == 'w')) {
FD_SET(udp_pollers[i].fd, writefds);
} else if (c == 'e') {
} else if (exceptfds && (c == 'e')) {
FD_SET(udp_pollers[i].fd, exceptfds);
}
}
close(udp_pollers[i].pipefds[0]);
close(udp_pollers[i].pipefds[1]);
host_close(udp_pollers[i].pipefds[0]);
host_close(udp_pollers[i].pipefds[1]);
ipstack_register_callback(ipstack, tcp_pollers[i].fd, NULL, NULL);
udp_pollers[i].fd = 0;
}
@ -362,10 +392,14 @@ int ft_select(struct ipstack *ipstack, int nfds, fd_set *readfds, fd_set *writef
}
int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout) {
int ret;
if (in_the_stack) {
return host_select(nfds, readfds, writefds, exceptfds, timeout);
} else {
return ft_select(IPSTACK, nfds, readfds, writefds, exceptfds, timeout);
pthread_mutex_lock(&ipstack_mutex);
ret = ft_select(IPSTACK, nfds, readfds, writefds, exceptfds, timeout);
pthread_mutex_unlock(&ipstack_mutex);
return ret;
}
}
@ -415,6 +449,7 @@ int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen) {
}
ssize_t recvfrom(int sockfd, void *buf, size_t len, int flags, struct sockaddr *addr, socklen_t *addrlen) {
printf("Called recv(): len=%lu\n", len);
conditional_steal_blocking_call(recvfrom, sockfd, buf, len, flags, addr, addrlen);
}
@ -439,10 +474,14 @@ ssize_t write(int sockfd, const void *buf, size_t len) {
}
int poll(struct pollfd *fds, nfds_t nfds, int timeout) {
int ret;
if (in_the_stack) {
return host_poll(fds, nfds, timeout);
} else {
return ft_poll(IPSTACK, fds, nfds, timeout);
pthread_mutex_lock(&ipstack_mutex);
ret = ft_poll(IPSTACK, fds, nfds, timeout);
pthread_mutex_unlock(&ipstack_mutex);
return ret;
}
}
@ -458,8 +497,10 @@ void *ft_posix_ip_loop(void *arg) {
uint32_t ms_next;
struct timeval tv;
while (1) {
pthread_mutex_lock(&ipstack_mutex);
gettimeofday(&tv, NULL);
ms_next = ipstack_poll(ipstack, tv.tv_sec * 1000 + tv.tv_usec / 1000);
pthread_mutex_unlock(&ipstack_mutex);
usleep(ms_next * 1000);
in_the_stack = 1;
}

View file

@ -0,0 +1,102 @@
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/select.h>
#define PORT 12346
int main() {
int server_fd, new_socket = -1;
struct sockaddr_in server_addr;
fd_set readfds;
int max_fd;
// Create a TCP socket
server_fd = socket(AF_INET, SOCK_STREAM, 0);
if (server_fd == -1) {
perror("Socket creation failed");
exit(EXIT_FAILURE);
}
printf("server socket: %d\n", server_fd);
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = INADDR_ANY;
server_addr.sin_port = htons(PORT);
// Bind the socket to the address
if (bind(server_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) == -1) {
perror("Bind failed");
close(server_fd);
exit(EXIT_FAILURE);
}
// Listen for incoming connections
if (listen(server_fd, 3) == -1) {
perror("Listen failed");
close(server_fd);
exit(EXIT_FAILURE);
}
// Initialize file descriptor sets
FD_ZERO(&readfds);
FD_SET(STDIN_FILENO, &readfds); // Monitor stdin
FD_SET(server_fd, &readfds); // Monitor the server socket
max_fd = server_fd;
while (1) {
fd_set tempfds = readfds;
int activity = select(max_fd + 1, &tempfds, NULL, NULL, NULL);
if (activity == -1) {
perror("Select error");
close(server_fd);
exit(EXIT_FAILURE);
}
if (FD_ISSET(STDIN_FILENO, &tempfds)) {
// Data available on stdin
char buffer[1024];
ssize_t bytes_read = read(STDIN_FILENO, buffer, sizeof(buffer));
if (bytes_read > 0 && new_socket != -1) {
// Write stdin data to the socket
send(new_socket, buffer, bytes_read, 0);
}
}
if ((new_socket == -1) && FD_ISSET(server_fd, &tempfds)) {
printf("Server socket activity\n");
// New connection on the socket
if (new_socket == -1) {
new_socket = accept(server_fd, NULL, NULL);
if (new_socket == -1) {
perror("Accept failed");
continue;
}
printf("New connection established\n");
FD_SET(new_socket, &readfds); // Monitor the new socket
max_fd = (new_socket > max_fd) ? new_socket : max_fd;
continue;
}
}
if ((new_socket != -1) && FD_ISSET(new_socket, &tempfds)) {
// Data available on the socket
char buffer[1024];
ssize_t bytes_received = recv(new_socket, buffer, sizeof(buffer), 0);
if (bytes_received > 0) {
write(STDOUT_FILENO, buffer, bytes_received);
} else if (bytes_received == 0) {
// Connection closed by the client
close(new_socket);
FD_CLR(new_socket, &readfds); // Stop monitoring the socket
new_socket = -1;
printf("Connection closed\n");
}
}
}
close(server_fd);
return 0;
}