2024-07-08 02:50:28 +00:00
|
|
|
#include <errno.h>
|
2024-07-08 03:05:32 +00:00
|
|
|
#include <string.h>
|
2024-07-08 02:50:28 +00:00
|
|
|
#include <stdlib.h>
|
|
|
|
#include <sys/socket.h>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <arpa/inet.h>
|
|
|
|
#include <sys/wait.h>
|
|
|
|
#include <netdb.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <signal.h>
|
2024-07-09 16:38:31 +00:00
|
|
|
#include <pthread.h>
|
2024-07-08 02:50:28 +00:00
|
|
|
|
|
|
|
#include "lfnetwork.h"
|
|
|
|
|
|
|
|
static void sighandler(int s) {
|
|
|
|
int saved_errno = errno;
|
|
|
|
while (waitpid(-1, NULL, WNOHANG) > 0);
|
|
|
|
errno = saved_errno;
|
|
|
|
}
|
|
|
|
|
|
|
|
void *get_in_addr(struct sockaddr *sa)
|
|
|
|
{
|
|
|
|
if (sa->sa_family == AF_INET) {
|
|
|
|
return &(((struct sockaddr_in*)sa)->sin_addr);
|
|
|
|
}
|
|
|
|
|
|
|
|
return &(((struct sockaddr_in6*)sa)->sin6_addr);
|
|
|
|
}
|
|
|
|
|
|
|
|
Server *new_server(ServerType type, const char *port, void(handler)(Server *s)) {
|
|
|
|
Server *s = (Server *)malloc(sizeof(Server));
|
|
|
|
if (s == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
s->server_type = type;
|
|
|
|
s->handler = handler;
|
|
|
|
|
|
|
|
struct addrinfo *addr = NULL;
|
|
|
|
if (getaddrinfo(NULL, port, NULL, &addr) != 0) {
|
|
|
|
free(s);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
s->port = (int)strtol(port, NULL, 10);
|
|
|
|
|
|
|
|
int socktype = 0;
|
|
|
|
switch (type) {
|
|
|
|
case SERVERTYPE_TCP:
|
|
|
|
socktype = SOCK_STREAM;
|
|
|
|
break;
|
|
|
|
case SERVERTYPE_UDP:
|
|
|
|
socktype = SOCK_DGRAM;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct addrinfo *p;
|
|
|
|
for (p = addr; p != NULL; p = p->ai_next) {
|
|
|
|
s->fd = socket(AF_INET, socktype, 0);
|
|
|
|
if (s->fd == -1) {
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (bind(s->fd, p->ai_addr, p->ai_addrlen) != 0) {
|
|
|
|
close(s->fd);
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (p == NULL) {
|
|
|
|
fprintf(stderr, "Failed to bind\n");
|
|
|
|
free(s);
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
freeaddrinfo(addr);
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
void delete_server(Server *s) {
|
|
|
|
free(s);
|
|
|
|
s = NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
int serve(Server *s, int backlog_size) {
|
|
|
|
if (listen(s->fd, backlog_size) != 0) {
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
struct sigaction sa;
|
|
|
|
sa.sa_handler = sighandler;
|
|
|
|
sa.sa_flags = SA_RESTART;
|
|
|
|
if (sigaction(SIGCHLD, &sa, NULL) == -1) {
|
|
|
|
fprintf(stderr, "Failed to set sigaction\n");
|
|
|
|
return 1;
|
|
|
|
}
|
|
|
|
|
|
|
|
s->handler(s);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2024-07-09 16:38:31 +00:00
|
|
|
static void *tcp_echo_thread(void *vargp) {
|
2024-07-08 02:50:28 +00:00
|
|
|
while (1) {
|
2024-07-09 16:38:31 +00:00
|
|
|
char recv_buf[256];
|
|
|
|
int fd = *(int *) vargp;
|
|
|
|
|
|
|
|
int r = (int)recv(fd, recv_buf, 256, 0);
|
|
|
|
if (r < 1) {
|
2024-07-08 03:05:32 +00:00
|
|
|
if (r == -1) {
|
|
|
|
fprintf(stderr, "Failed to recv. Errno: %d\n", errno);
|
|
|
|
}
|
2024-07-09 16:38:31 +00:00
|
|
|
close(fd);
|
|
|
|
break;
|
|
|
|
}
|
2024-07-08 03:05:32 +00:00
|
|
|
|
2024-07-09 16:38:31 +00:00
|
|
|
if (send(fd, recv_buf, strlen(recv_buf), 0) == -1) {
|
|
|
|
fprintf(stderr, "Failed to send echo. Errno: %d\n", errno);
|
|
|
|
close(fd);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2024-07-08 03:05:32 +00:00
|
|
|
|
2024-07-09 16:38:31 +00:00
|
|
|
void handler_tcp_echo(Server *s) {
|
|
|
|
while (1) {
|
|
|
|
struct sockaddr_storage client_addr;
|
|
|
|
socklen_t client_addr_sz = sizeof(client_addr);
|
|
|
|
int new_fd = accept(s->fd, (struct sockaddr *)&client_addr, &client_addr_sz);
|
|
|
|
if (new_fd == -1) {
|
|
|
|
fprintf(stderr, "failed to accept. Errno: %d\n", errno);
|
|
|
|
continue;
|
|
|
|
}
|
2024-07-08 02:50:28 +00:00
|
|
|
|
2024-07-09 16:38:31 +00:00
|
|
|
pthread_t srv_tid;
|
|
|
|
pthread_attr_t attr;
|
|
|
|
pthread_attr_init(&attr);
|
|
|
|
pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_DETACHED);
|
|
|
|
pthread_create(&srv_tid, &attr, tcp_echo_thread, &new_fd);
|
2024-07-08 02:50:28 +00:00
|
|
|
}
|
|
|
|
}
|