initial http parsing
This commit is contained in:
201
src/server/server.c
Normal file
201
src/server/server.c
Normal file
@@ -0,0 +1,201 @@
|
||||
#include "server/server.h"
|
||||
|
||||
#include "http/http.h"
|
||||
#include "utils/colors.h"
|
||||
#include "utils/hashmap.h"
|
||||
#include "utils/utils.h"
|
||||
|
||||
int start_server(const char *hostname, const char *service) {
|
||||
struct addrinfo hints;
|
||||
struct addrinfo *res;
|
||||
|
||||
setup_hints(&hints, sizeof hints, hostname);
|
||||
|
||||
int status = getaddrinfo(hostname, service, &hints, &res);
|
||||
if (status != 0) {
|
||||
fprintf(stderr, RED BOLD "[server] getaddrinfo() error: %s\n" RESET, gai_strerror(status));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
int sockfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
|
||||
fprintf(stdout, YELLOW "[server] sockfd: %d\n" RESET, sockfd);
|
||||
|
||||
int opt = 1;
|
||||
status = setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof opt);
|
||||
if (status != 0) {
|
||||
fprintf(stderr, RED BOLD "[server] setsockopt(): %s\n" RESET, strerror(errno));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
status = bind(sockfd, res->ai_addr, res->ai_addrlen);
|
||||
if (status != 0) {
|
||||
fprintf(stderr, RED BOLD "[server] bind(): %s\n" RESET, strerror(errno));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
status = listen(sockfd, BACKLOG);
|
||||
if (status != 0) {
|
||||
fprintf(stderr, RED BOLD "[server] listen(): %s\n" RESET, gai_strerror(status));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
|
||||
handle_clients(sockfd);
|
||||
|
||||
freeaddrinfo(res);
|
||||
close(sockfd);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
void setup_hints(struct addrinfo *hints, size_t len, const char *hostname) {
|
||||
memset(hints, 0, len);
|
||||
/* IPv4 or IPv6 */
|
||||
hints->ai_family = AF_UNSPEC;
|
||||
/* TCP */
|
||||
hints->ai_socktype = SOCK_STREAM;
|
||||
if (hostname == NULL) {
|
||||
/* Fill in IP for me */
|
||||
hints->ai_flags = AI_PASSIVE;
|
||||
}
|
||||
}
|
||||
|
||||
void handle_clients(int sockfd) {
|
||||
struct sockaddr_storage their_sa;
|
||||
socklen_t theirsa_size = sizeof their_sa;
|
||||
|
||||
bucket_t clients[HASHMAP_MAX_CLIENTS];
|
||||
hm_init(clients);
|
||||
|
||||
int epfd = epoll_create1(0);
|
||||
setnonblocking(sockfd);
|
||||
epoll_ctl_add(epfd, sockfd, EPOLLIN | EPOLLET);
|
||||
|
||||
struct epoll_event *revents = malloc(EPOLL_MAXEVENTS * sizeof(struct epoll_event));
|
||||
int nfds;
|
||||
|
||||
char *msg = "Hello there!";
|
||||
size_t msg_len = strlen(msg);
|
||||
char data[4096];
|
||||
int client_fd;
|
||||
int run = 1;
|
||||
|
||||
while (run) {
|
||||
nfds = epoll_wait(epfd, revents, EPOLL_MAXEVENTS, EPOLL_TIMEOUT);
|
||||
|
||||
for (int i = 0; i < nfds; ++i) {
|
||||
if (revents[i].data.fd == sockfd) {
|
||||
/* New client */
|
||||
char ip[INET_ADDRSTRLEN];
|
||||
client_fd = handle_new_client(sockfd, &their_sa, &theirsa_size);
|
||||
get_client_ip(&their_sa, ip);
|
||||
fprintf(stdout, BLUE "[server] Client (%s) connected\n" RESET, ip);
|
||||
|
||||
setnonblocking(client_fd);
|
||||
epoll_ctl_add(epfd, client_fd, EPOLLIN);
|
||||
hm_push(clients, client_fd, &their_sa);
|
||||
|
||||
int bytes_sent = send(client_fd, msg, msg_len, 0);
|
||||
fprintf(stdout, "[server] Sent %d bytes\n", bytes_sent);
|
||||
} else {
|
||||
/* Incoming data */
|
||||
client_fd = revents[i].data.fd;
|
||||
int bytes_read = recv(client_fd, data, sizeof data, 0);
|
||||
|
||||
if (bytes_read == 0) {
|
||||
/* Client disconnected */
|
||||
char ip[INET_ADDRSTRLEN];
|
||||
bucket_t *client = hm_lookup(clients, client_fd);
|
||||
get_client_ip(&client->sas, ip);
|
||||
|
||||
fprintf(stdout, BLUE "[server] Client (%s) disconnected\n" RESET, ip);
|
||||
epoll_ctl_del(epfd, client_fd);
|
||||
close(client_fd);
|
||||
continue;
|
||||
}
|
||||
|
||||
// fprintf(stdout, "[server] Bytes read (%d):\n%s\n", bytes_read, data);
|
||||
if (strcmp(data, "stop") == 0) {
|
||||
fprintf(stdout, GREEN BOLD "[server] Stopping...\n" RESET);
|
||||
run = 0;
|
||||
break;
|
||||
}
|
||||
|
||||
/* Parse HTTP request */
|
||||
http_t *request = http_parse(data);
|
||||
fprintf(stdout, "[server] request location: %s\n", request->location);
|
||||
http_send_response(request);
|
||||
http_free(request);
|
||||
|
||||
/* Clear str */
|
||||
memset(data, 0, sizeof data);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/* Clean up everything */
|
||||
free(revents);
|
||||
close(epfd);
|
||||
close_fds(clients);
|
||||
hm_free(clients);
|
||||
}
|
||||
|
||||
void epoll_ctl_add(int epfd, int sockfd, uint32_t events) {
|
||||
struct epoll_event event;
|
||||
event.events = events;
|
||||
event.data.fd = sockfd;
|
||||
int status = epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &event);
|
||||
|
||||
if (status != 0) {
|
||||
fprintf(stderr, RED BOLD "[server] epoll_ctl_add(): %s\n" RESET, strerror(errno));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
void epoll_ctl_del(int epfd, int sockfd) {
|
||||
int status = epoll_ctl(epfd, EPOLL_CTL_DEL, sockfd, NULL);
|
||||
|
||||
if (status != 0) {
|
||||
fprintf(stdout, RED BOLD "[server] epoll_ctl_del(): %s\n" RESET, strerror(errno));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
void setnonblocking(int sockfd) {
|
||||
int status = fcntl(sockfd, F_SETFL, O_NONBLOCK);
|
||||
|
||||
if (status == -1) {
|
||||
fprintf(stderr, RED BOLD "[server] fcntl(): %s\n" RESET, gai_strerror(status));
|
||||
exit(EXIT_FAILURE);
|
||||
}
|
||||
}
|
||||
|
||||
int handle_new_client(int sockfd, struct sockaddr_storage *their_sa, socklen_t *theirsa_size) {
|
||||
int client_fd = accept(sockfd, (struct sockaddr *)their_sa, theirsa_size);
|
||||
|
||||
if (client_fd == -1) {
|
||||
if (errno != EWOULDBLOCK) {
|
||||
fprintf(stderr, RED BOLD "[server] accept(): %s\n" RESET, strerror(errno));
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
return client_fd;
|
||||
}
|
||||
|
||||
void close_fds(bucket_t *bucket) {
|
||||
bucket_t *p, *next;
|
||||
|
||||
for (size_t i = 0; i < HASHMAP_MAX_CLIENTS; ++i) {
|
||||
close(bucket[i].sockfd);
|
||||
if (bucket[i].next != NULL) {
|
||||
/* Close the fds */
|
||||
p = bucket[i].next;
|
||||
next = p->next;
|
||||
do {
|
||||
close(p->sockfd);
|
||||
p = next;
|
||||
next = p != NULL ? p->next : NULL;
|
||||
} while (p != NULL);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user