adjust naming convention and initial config support

This commit is contained in:
2025-04-24 17:24:13 +02:00
parent b4f71b0c53
commit dd9e2f557b
19 changed files with 217 additions and 263 deletions

2
.gitignore vendored
View File

@@ -1,6 +1,8 @@
build/ build/
.cache/ .cache/
docs/ docs/
cert.pem
key.pem
# Prerequisites # Prerequisites
*.d *.d

View File

@@ -8,6 +8,8 @@ will it ever be. Use it at your own risk.
- [meson](https://mesonbuild.com/index.html) - [meson](https://mesonbuild.com/index.html)
- [doxygen](https://www.doxygen.nl/) - [doxygen](https://www.doxygen.nl/)
- Optional, just to build the docs. - Optional, just to build the docs.
- libcyaml
- libyaml-0.1
## How to build ## How to build
@@ -17,7 +19,7 @@ $ cd build
$ meson compile $ meson compile
``` ```
And then run `server`! And then run `cws`!
## Docs ## Docs
@@ -39,11 +41,12 @@ And then open the `docs/html/index.html`.
- [ ] Request parser - [ ] Request parser
- [x] Serve static files - [x] Serve static files
- [ ] Implement Keep-Alive - [ ] Implement Keep-Alive
- [ ] Multithreading - [ ] Multithreading to handle concurrent requests
- [ ] Logging - [ ] Logging
- [ ] Advanced - [ ] Advanced Features
- [ ] HTTPS support - [ ] HTTPS support with TLS
- [ ] Caching - [ ] Compression (Gzip)
- [ ] Support for virtual hosting
## Resources ## Resources

6
config.yaml Normal file
View File

@@ -0,0 +1,6 @@
# Default config file of CWS
host: "localhost"
port: 3030
cert: "cert.pem"
key: "key.pem"

View File

@@ -1,17 +0,0 @@
#ifndef CWS_CLIENT_H
#define CWS_CLIENT_H
#include <arpa/inet.h>
#include <errno.h>
#include <netdb.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
int test_client_connection(const char *hostname, const char *service);
#endif

View File

@@ -1,33 +1,35 @@
#ifndef CWS_HTTP_H #ifndef CWS_HTTP_H
#define CWS_HTTP_H #define CWS_HTTP_H
#define WWW "../www" /**< Directory used to get html files */ #define CWS_WWW "../www" /**< Directory used to get html files */
/** In the future I'll move conf stuff under a server struct, I can skip just because I want something that works */ /** In the future I'll move conf stuff under a server struct, I can skip just because I want something that works */
#define LOCATION_LEN 512 #define CWS_HTTP_LOCATION_LEN 512
#define LOCATION_PATH_LEN 1024 #define CWS_HTTP_LOCATION_PATH_LEN 1024
#define HTTP_VERSION_LEN 8 #define CWS_HTTP_VERSION_LEN 8
#define USER_AGENT_LEN 1024 #define CWS_HTTP_USER_AGENT_LEN 1024
#define HOST_LEN 1024 #define CWS_HTTP_HOST_LEN 1024
enum http_method { typedef enum cws_http_method_t {
GET, /**< GET method */ CWS_HTTP_GET, /**< GET method */
POST, /**< POST method */ CWS_HTTP_POST, /**< POST method */
}; CWS_HTTP_PUT,
CWS_HTTP_DELETE,
} cws_http_method;
/* In the future I'll add HEAD, PUT, DELETE */ /* In the future I'll add HEAD, PUT, DELETE */
/** /**
* @brief HTTP request struct * @brief HTTP request struct
* *
*/ */
typedef struct http { typedef struct cws_http_t {
int sockfd; /**< Socket file descriptor */ int sockfd; /**< Socket file descriptor */
enum http_method method; /**< HTTP request method */ cws_http_method method; /**< HTTP request method */
char location[LOCATION_LEN]; /**< Resource requested */ char location[CWS_HTTP_LOCATION_LEN]; /**< Resource requested */
char location_path[LOCATION_PATH_LEN]; /**< Resource path */ char location_path[CWS_HTTP_LOCATION_PATH_LEN]; /**< Full resource path */
char http_version[HTTP_VERSION_LEN]; /**< HTTP version */ char http_version[CWS_HTTP_VERSION_LEN]; /**< HTTP version */
char user_agent[USER_AGENT_LEN]; /**< User-Agent */ char user_agent[CWS_HTTP_USER_AGENT_LEN]; /**< User-Agent */
char host[HOST_LEN]; /**< Host */ char host[CWS_HTTP_HOST_LEN]; /**< Host */
} http_t; } cws_http;
/* Connection */ /* Connection */
/* Accept-Encoding */ /* Accept-Encoding */
/* Accept-Language */ /* Accept-Language */
@@ -35,18 +37,18 @@ typedef struct http {
/** /**
* @brief Parses a HTTP request * @brief Parses a HTTP request
* *
* @param request_str[in] The http request sent to the server * @param[in] request_str The http request sent to the server
* @return Returns a http_t pointer to the request * @return Returns a http_t pointer to the request
*/ */
http_t *http_parse(char *request_str, int sockfd); cws_http *cws_http_parse(char *request_str, int sockfd);
void http_parse_method(http_t *request, const char *method); void cws_http_parse_method(cws_http *request, const char *method);
void http_get_content_type(http_t *request, char *content_type); void cws_http_get_content_type(cws_http *request, char *content_type);
void http_send_response(http_t *request); void cws_http_send_response(cws_http *request);
void http_send_not_found(http_t *request); void cws_http_send_not_found(cws_http *request);
void http_send_not_implemented(http_t *request); void cws_http_send_not_implemented(cws_http *request);
void http_free(http_t *request); void cws_http_free(cws_http *request);
#endif #endif

View File

@@ -7,87 +7,87 @@
#include "utils/hashmap.h" #include "utils/hashmap.h"
/* Clients max queue */ /* Clients max queue */
#define BACKLOG 10 #define CWS_SERVER_BACKLOG 10
/* Size of the epoll_event array */ /* Size of the epoll_event array */
#define EPOLL_MAXEVENTS 10 #define CWS_SERVER_EPOLL_MAXEVENTS 10
/* Wait forever (epoll_wait()) */ /* Wait forever (epoll_wait()) */
#define EPOLL_TIMEOUT -1 #define CWS_SERVER_EPOLL_TIMEOUT -1
/** /**
* @brief Runs the server * @brief Runs the server
* *
* @param hostname[in] The hostname of the server (default localhost, it could be NULL) * @param[in] hostname The hostname of the server (default localhost, it could be NULL)
* @param service[in] The service (found in /etc/services) or the port where to run * @param[in] service The service (found in /etc/services) or the port where to run
* @return 0 on success, -1 on error * @return 0 on success, -1 on error
*/ */
int start_server(const char *hostname, const char *service); int cws_server_start(const char *hostname, const char *service);
/** /**
* @brief Setups hints object * @brief Setups hints object
* *
* @param hints[out] The hints addrinfo * @param[out] hints The hints addrinfo
* @param len[in] The length of hints * @param[in] len The length of hints
* @param hostname[in] The hostname (could be NULL) * @param[in] hostname The hostname (could be NULL)
*/ */
void setup_hints(struct addrinfo *hints, size_t len, const char *hostname); void cws_server_setup_hints(struct addrinfo *hints, size_t len, const char *hostname);
/** /**
* @brief Main server loop * @brief Main server loop
* *
* @param sockfd[in,out] Socket of the commincation endpoint * @param[in,out] sockfd Socket of the commincation endpoint
*/ */
void handle_clients(int sockfd); void cws_server_loop(int sockfd);
/** /**
* @brief Adds a file descriptor to the interest list * @brief Adds a file descriptor to the interest list
* *
* @param epfd[in] epoll file descriptor * @param[in] epfd epoll file descriptor
* @param sockfd[in] The file descriptor to watch * @param[in] sockfd The file descriptor to watch
* @param events[in] The events to follow * @param[in] events The events to follow
*/ */
void epoll_ctl_add(int epfd, int sockfd, uint32_t events); void cws_epoll_add(int epfd, int sockfd, uint32_t events);
/** /**
* @brief Removes a file descriptor from the interest list * @brief Removes a file descriptor from the interest list
* *
* @param epfd[in] epoll file descriptor * @param[in] epfd epoll file descriptor
* @param sockfd[in] The file descriptor to remove * @param[in] sockfd The file descriptor to remove
*/ */
void epoll_ctl_del(int epfd, int sockfd); void cws_epoll_del(int epfd, int sockfd);
/** /**
* @brief Makes a file descriptor non-blocking * @brief Makes a file descriptor non-blocking
* *
* @param sockfd[in] The file descriptor to make non-blocking * @param[in] sockfd The file descriptor to make non-blocking
*/ */
void setnonblocking(int sockfd); void cws_fd_set_nonblocking(int sockfd);
/** /**
* @brief Handles the new client * @brief Handles the new client
* *
* @param sockfd[in] Server's file descriptor * @param[in] sockfd Server's file descriptor
* @param their_sa[out] Populates the struct with client's information * @param[out] their_sa Populates the struct with client's information
* @param theirsa_size[in] Size of the struct * @param[in] theirsa_size Size of the struct
* @return Returns -1 on error or the file descriptor on success * @return Returns -1 on error or the file descriptor on success
*/ */
int handle_new_client(int sockfd, struct sockaddr_storage *their_sa, socklen_t *theirsa_size); int cws_server_accept_client(int sockfd, struct sockaddr_storage *their_sa, socklen_t *theirsa_size);
/** /**
* @brief Closes all the file descriptors opened * @brief Closes all the file descriptors opened
* *
* @param bucket[in] The hash map * @param[in] bucket The hash map
*/ */
void close_fds(bucket_t *bucket); void cws_server_close_all_fds(cws_bucket *bucket);
/** /**
* @brief Disconnect a client * @brief Disconnect a client
* *
* @param epfd[in] Epoll file descriptor * @param[in] epfd Epoll file descriptor
* @param client_fd[in] Client file descriptor * @param[in] client_fd Client file descriptor
* @param bucket[in] Clients hash map * @param[in] bucket Clients hash map
*/ */
void close_client(int epfd, int client_fd, bucket_t *bucket); void cws_server_close_client(int epfd, int client_fd, cws_bucket *bucket);
#endif #endif

7
include/utils/config.h Normal file
View File

@@ -0,0 +1,7 @@
#ifndef CWS_CONFIG_H
#define CWS_CONFIG_H
typedef struct cws_config_t {
} cws_config;
#endif

View File

@@ -2,81 +2,79 @@
#define CWS_HASHMAP_H #define CWS_HASHMAP_H
#include <stdbool.h> #include <stdbool.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h> #include <sys/socket.h>
/* Each process on Linux can have a maximum of 1024 open file descriptors */ /* Each process on Linux can have a maximum of 1024 open file descriptors */
#define HASHMAP_MAX_CLIENTS 1024 #define CWS_HASHMAP_MAX_CLIENTS 1024
/** /**
* @brief Hash map struct * @brief Hash map struct
* *
*/ */
typedef struct bucket { typedef struct cws_bucket_t {
int sockfd; /**< Client socket descriptor */ int sockfd; /**< Client socket descriptor */
struct sockaddr_storage sas; /**< Associated socket address */ struct sockaddr_storage sas; /**< Associated socket address */
struct bucket *next; /**< Next node in case of collision */ struct cws_bucket_t *next; /**< Next node in case of collision */
struct bucket *prev; /**< Previous node in case of collision */ struct cws_bucket_t *prev; /**< Previous node in case of collision */
} bucket_t; } cws_bucket;
/** /**
* @brief Calculates the hash code of a given file descriptor * @brief Calculates the hash code of a given file descriptor
* *
* @param sockfd[in] The file descriptor * @param[in] sockfd The file descriptor
* @return Returns the hash code * @return Returns the hash code
*/ */
int hash(int sockfd); int cws_hm_hash(int sockfd);
/** /**
* @brief Initializes the hash map * @brief Initializes the hash map
* *
* @param bucket[out] The hash map uninitialized * @param[out] bucket The hash map uninitialized
*/ */
void hm_init(bucket_t *bucket); void cws_hm_init(cws_bucket *bucket);
/** /**
* @brief Inserts a key in the hash map * @brief Inserts a key in the hash map
* *
* @param bucket[out] The hash map * @param[out] bucket The hash map
* @param sockfd[in] The file descriptor (value) * @param[in] sockfd The file descriptor (value)
* @param sas[in] The sockaddr (value) * @param[in] sas The sockaddr (value)
*/ */
void hm_push(bucket_t *bucket, int sockfd, struct sockaddr_storage *sas); void cws_hm_push(cws_bucket *bucket, int sockfd, struct sockaddr_storage *sas);
/** /**
* @brief Removes a key from the hash map * @brief Removes a key from the hash map
* *
* @param bucket[out] The hash map * @param[out] bucket The hash map
* @param sockfd[in] The key * @param[in] sockfd The key
*/ */
void hm_remove(bucket_t *bucket, int sockfd); void cws_hm_remove(cws_bucket *bucket, int sockfd);
/** /**
* @brief Searches for a key in the hash map * @brief Searches for a key in the hash map
* *
* @param bucket[in] The hash map * @param[in] bucket The hash map
* @param sockfd[in] The file descriptor (key) * @param[in] sockfd The file descriptor (key)
* @return Returns NULL or the key pointer * @return Returns NULL or the key pointer
*/ */
bucket_t *hm_lookup(bucket_t *bucket, int sockfd); cws_bucket *cws_hm_lookup(cws_bucket *bucket, int sockfd);
/** /**
* @brief Cleans the hash map * @brief Cleans the hash map
* *
* @param bucket[out] The hash map * @param[out] bucket The hash map
*/ */
void hm_free(bucket_t *bucket); void cws_hm_free(cws_bucket *bucket);
/** /**
* @brief Checks if a file descriptor is in the bucket array (not linked list) * @brief Checks if a file descriptor is in the bucket array (not linked list)
* *
* @param bucket[in] * @param[in] bucket
* @param sockfd[in] * @param[in] sockfd
* @return true If the file descriptor is in the bucket array * @return true If the file descriptor is in the bucket array
* @return false If the file descriptor is not in the bucket array (check with hm_lookup()) * @return false If the file descriptor is not in the bucket array (check with hm_lookup())
*/ */
bool hm_is_in_bucket_array(bucket_t *bucket, int sockfd); bool cws_hm_is_in_bucket_array(cws_bucket *bucket, int sockfd);
/** /**
* @brief This function will add a key even if it exists (use hm_push() instead) * @brief This function will add a key even if it exists (use hm_push() instead)
@@ -85,6 +83,6 @@ bool hm_is_in_bucket_array(bucket_t *bucket, int sockfd);
* @param sockfd * @param sockfd
* @param sas * @param sas
*/ */
void hm_insert(bucket_t *bucket, int sockfd, struct sockaddr_storage *sas); void cws_hm_insert(cws_bucket *bucket, int sockfd, struct sockaddr_storage *sas);
#endif #endif

View File

@@ -4,26 +4,23 @@
#include <arpa/inet.h> #include <arpa/inet.h>
#include <netdb.h> #include <netdb.h>
#include <netinet/in.h> #include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <sys/types.h> #include <sys/types.h>
/** /**
* @brief Prints each IP address associated with a host * @brief Prints each IP address associated with a host
* *
* @param hostname[in] Hostname * @param[in] hostname Hostname
* @param port[in] Port * @param[in] port Port
*/ */
void print_ips(const char *hostname, const char *port); void cws_utils_print_ips(const char *hostname, const char *port);
/** /**
* @brief Retrieves the client ip from the sockaddr_storage and put it in the ip str * @brief Retrieves the client ip from the sockaddr_storage and put it in the ip str
* *
* @param sa[in] The sockaddr_storage of the client * @param[in] sa The sockaddr_storage of the client
* @param ip[out] The IP of the client * @param[out] ip The IP of the client
*/ */
void get_client_ip(struct sockaddr_storage *sa, char *ip); void cws_utils_get_client_ip(struct sockaddr_storage *sa, char *ip);
#endif #endif

View File

@@ -4,5 +4,7 @@ subdir('src')
incdir = include_directories('include') incdir = include_directories('include')
executable('server', server, include_directories : incdir) libyaml = dependency('yaml-0.1')
executable('client', client, include_directories : incdir) libcyaml = dependency('libcyaml')
executable('cws', server, include_directories : incdir, dependencies : [libyaml, libcyaml])

View File

@@ -1,40 +0,0 @@
#include "client/client.h"
#include "utils/colors.h"
int test_client_connection(const char *hostname, const char *service) {
struct addrinfo hints;
struct addrinfo *res;
memset(&hints, 0, sizeof hints);
hints.ai_socktype = SOCK_STREAM;
hints.ai_family = AF_INET;
int status = getaddrinfo(hostname, service, &hints, &res);
if (status != 0) {
fprintf(stderr, RED BOLD "[client] getaddrinfo(): %s\n" RESET, gai_strerror(status));
exit(EXIT_FAILURE);
}
int sockfd = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
status = connect(sockfd, res->ai_addr, res->ai_addrlen);
if (status != 0) {
fprintf(stderr, RED BOLD "[client] connect(): %s\n" RESET, strerror(errno));
exit(EXIT_FAILURE);
}
char buf[4096];
int bytes_read = recv(sockfd, buf, sizeof buf, 0);
fprintf(stdout, BLUE "[client] Read %d bytes: %s\n" RESET, bytes_read, buf);
fprintf(stdout, "[client] => ");
fgets(buf, sizeof buf, stdin);
buf[strcspn(buf, "\n")] = '\0';
send(sockfd, buf, strlen(buf), 0);
freeaddrinfo(res);
close(sockfd);
return 0;
}

View File

@@ -1,15 +0,0 @@
#include <stdio.h>
#include "client/client.h"
#include "utils/colors.h"
int main(int argc, char **argv) {
fprintf(stdout, BOLD GREEN "[client] Running client...\n" RESET);
int ret = test_client_connection(argv[1], argv[2]);
if (ret < 0) {
fprintf(stderr, BOLD RED "Unable to start client\n");
}
return 0;
}

View File

@@ -7,8 +7,8 @@
#include "utils/colors.h" #include "utils/colors.h"
http_t *http_parse(char *request_str, int sockfd) { cws_http *cws_http_parse(char *request_str, int sockfd) {
http_t *request = malloc(sizeof(http_t)); cws_http *request = malloc(sizeof(cws_http));
if (request == NULL) { if (request == NULL) {
return NULL; return NULL;
} }
@@ -22,7 +22,7 @@ http_t *http_parse(char *request_str, int sockfd) {
return NULL; return NULL;
} }
printf("[client::http] method: %s\n", pch); printf("[client::http] method: %s\n", pch);
http_parse_method(request, pch); cws_http_parse_method(request, pch);
/* Parse location */ /* Parse location */
pch = strtok(NULL, " "); pch = strtok(NULL, " ");
@@ -30,14 +30,14 @@ http_t *http_parse(char *request_str, int sockfd) {
return NULL; return NULL;
} }
printf("[client::http] location: %s\n", pch); printf("[client::http] location: %s\n", pch);
strncpy(request->location, pch, LOCATION_LEN); strncpy(request->location, pch, CWS_HTTP_LOCATION_LEN);
/* Parse location path */ /* Parse location path */
/* TODO: Prevent Path Traversal */ /* TODO: Prevent Path Traversal */
if (strcmp(request->location, "/") == 0) { if (strcmp(request->location, "/") == 0) {
snprintf(request->location_path, LOCATION_PATH_LEN, "%s/index.html", WWW); snprintf(request->location_path, CWS_HTTP_LOCATION_PATH_LEN, "%s/index.html", CWS_WWW);
} else { } else {
snprintf(request->location_path, LOCATION_PATH_LEN, "%s%s", WWW, request->location); snprintf(request->location_path, CWS_HTTP_LOCATION_PATH_LEN, "%s%s", CWS_WWW, request->location);
} }
fprintf(stdout, "[client::http] location path: %s\n", request->location_path); fprintf(stdout, "[client::http] location path: %s\n", request->location_path);
@@ -47,37 +47,37 @@ http_t *http_parse(char *request_str, int sockfd) {
return NULL; return NULL;
} }
printf("[client::http] version: %s\n", pch); printf("[client::http] version: %s\n", pch);
strncpy(request->http_version, pch, HTTP_VERSION_LEN); strncpy(request->http_version, pch, CWS_HTTP_VERSION_LEN);
/* Parse other stuff... */ /* Parse other stuff... */
return request; return request;
} }
void http_parse_method(http_t *request, const char *method) { void cws_http_parse_method(cws_http *request, const char *method) {
if (strcmp(method, "GET") == 0) { if (strcmp(method, "GET") == 0) {
request->method = GET; request->method = CWS_HTTP_GET;
return; return;
} }
if (strcmp(method, "POST") == 0) { if (strcmp(method, "POST") == 0) {
request->method = POST; request->method = CWS_HTTP_POST;
return; return;
} }
http_send_not_implemented(request); cws_http_send_not_implemented(request);
} }
void http_send_response(http_t *request) { void cws_http_send_response(cws_http *request) {
FILE *file = fopen(request->location_path, "rb"); FILE *file = fopen(request->location_path, "rb");
if (file == NULL) { if (file == NULL) {
/* 404 */ /* 404 */
http_send_not_found(request); cws_http_send_not_found(request);
return; return;
} }
/* Retrieve correct Content-Type */ /* Retrieve correct Content-Type */
char content_type[1024]; char content_type[1024];
http_get_content_type(request, content_type); cws_http_get_content_type(request, content_type);
/* Retrieve file size */ /* Retrieve file size */
fseek(file, 0, SEEK_END); fseek(file, 0, SEEK_END);
@@ -111,10 +111,10 @@ void http_send_response(http_t *request) {
free(file_data); free(file_data);
} }
void http_get_content_type(http_t *request, char *content_type) { void cws_http_get_content_type(cws_http *request, char *content_type) {
char *ptr = strrchr(request->location_path, '.'); char *ptr = strrchr(request->location_path, '.');
if (ptr == NULL) { if (ptr == NULL) {
http_send_not_found(request); cws_http_send_not_found(request);
return; return;
} }
ptr += 1; ptr += 1;
@@ -134,7 +134,7 @@ void http_get_content_type(http_t *request, char *content_type) {
snprintf(content_type, 1024, "%s/%s", ct, ptr); snprintf(content_type, 1024, "%s/%s", ct, ptr);
} }
void http_send_not_implemented(http_t *request) { void cws_http_send_not_implemented(cws_http *request) {
const char response[1024] = const char response[1024] =
"HTTP/1.1 501 Not Implemented\r\n" "HTTP/1.1 501 Not Implemented\r\n"
"Content-Type: text/html\r\n" "Content-Type: text/html\r\n"
@@ -152,7 +152,7 @@ void http_send_not_implemented(http_t *request) {
send(request->sockfd, response, response_len, 0); send(request->sockfd, response, response_len, 0);
} }
void http_send_not_found(http_t *request) { void cws_http_send_not_found(cws_http *request) {
const char response[1024] = const char response[1024] =
"HTTP/1.1 404 Not Found\r\n" "HTTP/1.1 404 Not Found\r\n"
"Content-Type: text/html\r\n" "Content-Type: text/html\r\n"
@@ -170,4 +170,4 @@ void http_send_not_found(http_t *request) {
send(request->sockfd, response, response_len, 0); send(request->sockfd, response, response_len, 0);
} }
void http_free(http_t *request) { free(request); } void cws_http_free(cws_http *request) { free(request); }

View File

@@ -8,7 +8,7 @@ int main(int argc, char **argv) {
fprintf(stdout, BOLD GREEN "[server] Running cws on http://localhost:%s...\n" RESET, default_port); fprintf(stdout, BOLD GREEN "[server] Running cws on http://localhost:%s...\n" RESET, default_port);
int ret = start_server(NULL, default_port); int ret = cws_server_start(NULL, default_port);
if (ret < 0) { if (ret < 0) {
fprintf(stderr, BOLD RED "Unable to start web server\n"); fprintf(stderr, BOLD RED "Unable to start web server\n");
} }

View File

@@ -1,5 +1,3 @@
server = files('main.c', 'server/server.c') server = files('main.c', 'server/server.c')
server += files('utils/utils.c', 'utils/hashmap.c') server += files('utils/utils.c', 'utils/hashmap.c')
server += files('http/http.c') server += files('http/http.c')
client = files('client/main.c', 'client/client.c')

View File

@@ -16,11 +16,11 @@
#include "utils/hashmap.h" #include "utils/hashmap.h"
#include "utils/utils.h" #include "utils/utils.h"
int start_server(const char *hostname, const char *service) { int cws_server_start(const char *hostname, const char *service) {
struct addrinfo hints; struct addrinfo hints;
struct addrinfo *res; struct addrinfo *res;
setup_hints(&hints, sizeof hints, hostname); cws_server_setup_hints(&hints, sizeof hints, hostname);
int status = getaddrinfo(hostname, service, &hints, &res); int status = getaddrinfo(hostname, service, &hints, &res);
if (status != 0) { if (status != 0) {
@@ -43,13 +43,13 @@ int start_server(const char *hostname, const char *service) {
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
status = listen(sockfd, BACKLOG); status = listen(sockfd, CWS_SERVER_BACKLOG);
if (status != 0) { if (status != 0) {
fprintf(stderr, RED BOLD "[server] listen(): %s\n" RESET, gai_strerror(status)); fprintf(stderr, RED BOLD "[server] listen(): %s\n" RESET, gai_strerror(status));
exit(EXIT_FAILURE); exit(EXIT_FAILURE);
} }
handle_clients(sockfd); cws_server_loop(sockfd);
freeaddrinfo(res); freeaddrinfo(res);
close(sockfd); close(sockfd);
@@ -57,7 +57,7 @@ int start_server(const char *hostname, const char *service) {
return 0; return 0;
} }
void setup_hints(struct addrinfo *hints, size_t len, const char *hostname) { void cws_server_setup_hints(struct addrinfo *hints, size_t len, const char *hostname) {
memset(hints, 0, len); memset(hints, 0, len);
/* IPv4 or IPv6 */ /* IPv4 or IPv6 */
hints->ai_family = AF_UNSPEC; hints->ai_family = AF_UNSPEC;
@@ -69,53 +69,53 @@ void setup_hints(struct addrinfo *hints, size_t len, const char *hostname) {
} }
} }
void handle_clients(int sockfd) { void cws_server_loop(int sockfd) {
struct sockaddr_storage their_sa; struct sockaddr_storage their_sa;
socklen_t theirsa_size = sizeof their_sa; socklen_t theirsa_size = sizeof their_sa;
bucket_t clients[HASHMAP_MAX_CLIENTS]; cws_bucket clients[CWS_HASHMAP_MAX_CLIENTS];
hm_init(clients); cws_hm_init(clients);
int epfd = epoll_create1(0); int epfd = epoll_create1(0);
setnonblocking(sockfd); cws_fd_set_nonblocking(sockfd);
epoll_ctl_add(epfd, sockfd, EPOLLIN | EPOLLET); cws_epoll_add(epfd, sockfd, EPOLLIN | EPOLLET);
struct epoll_event *revents = malloc(EPOLL_MAXEVENTS * sizeof(struct epoll_event)); struct epoll_event *revents = malloc(CWS_SERVER_EPOLL_MAXEVENTS * sizeof(struct epoll_event));
int client_fd; int client_fd;
while (1) { while (1) {
int nfds = epoll_wait(epfd, revents, EPOLL_MAXEVENTS, EPOLL_TIMEOUT); int nfds = epoll_wait(epfd, revents, CWS_SERVER_EPOLL_MAXEVENTS, CWS_SERVER_EPOLL_TIMEOUT);
for (int i = 0; i < nfds; ++i) { for (int i = 0; i < nfds; ++i) {
if (revents[i].data.fd == sockfd) { if (revents[i].data.fd == sockfd) {
/* New client */ /* New client */
char ip[INET_ADDRSTRLEN]; char ip[INET_ADDRSTRLEN];
client_fd = handle_new_client(sockfd, &their_sa, &theirsa_size); client_fd = cws_server_accept_client(sockfd, &their_sa, &theirsa_size);
get_client_ip(&their_sa, ip); cws_utils_get_client_ip(&their_sa, ip);
fprintf(stdout, BLUE "[server] Client (%s) connected\n" RESET, ip); fprintf(stdout, BLUE "[server] Client (%s) connected\n" RESET, ip);
setnonblocking(client_fd); cws_fd_set_nonblocking(client_fd);
epoll_ctl_add(epfd, client_fd, EPOLLIN); cws_epoll_add(epfd, client_fd, EPOLLIN);
hm_push(clients, client_fd, &their_sa); cws_hm_push(clients, client_fd, &their_sa);
} else { } else {
char data[4096] = {0}; char data[4096] = {0};
/* Incoming data */ /* Incoming data */
client_fd = revents[i].data.fd; client_fd = revents[i].data.fd;
const ssize_t bytes_read = recv(client_fd, data, sizeof data, 0); const ssize_t bytes_read = recv(client_fd, data, sizeof data, 0);
char ip[INET_ADDRSTRLEN]; char ip[INET_ADDRSTRLEN];
bucket_t *client = hm_lookup(clients, client_fd); cws_bucket *client = cws_hm_lookup(clients, client_fd);
get_client_ip(&client->sas, ip); cws_utils_get_client_ip(&client->sas, ip);
if (bytes_read == 0) { if (bytes_read == 0) {
/* Client disconnected */ /* Client disconnected */
fprintf(stdout, BLUE "[server] Client (%s) disconnected\n" RESET, ip); fprintf(stdout, BLUE "[server] Client (%s) disconnected\n" RESET, ip);
close_client(epfd, client_fd, clients); cws_server_close_client(epfd, client_fd, clients);
continue; continue;
} }
if (bytes_read < 0) { if (bytes_read < 0) {
if (errno != EAGAIN && errno != EWOULDBLOCK) { if (errno != EAGAIN && errno != EWOULDBLOCK) {
/* Error during read, handle it (close client) */ /* Error during read, handle it (close client) */
epoll_ctl_del(epfd, client_fd); cws_epoll_del(epfd, client_fd);
close(client_fd); close(client_fd);
} }
continue; continue;
@@ -124,18 +124,18 @@ void handle_clients(int sockfd) {
data[bytes_read] = '\0'; data[bytes_read] = '\0';
/* Parse HTTP request */ /* Parse HTTP request */
http_t *request = http_parse(data, client_fd); cws_http *request = cws_http_parse(data, client_fd);
if (request == NULL) { if (request == NULL) {
close_client(epfd, client_fd, clients); cws_server_close_client(epfd, client_fd, clients);
http_free(request); cws_http_free(request);
continue; continue;
} }
http_send_response(request); cws_http_send_response(request);
fprintf(stdout, BLUE "[server] Client (%s) disconnected\n" RESET, ip); fprintf(stdout, BLUE "[server] Client (%s) disconnected\n" RESET, ip);
close_client(epfd, client_fd, clients); cws_server_close_client(epfd, client_fd, clients);
http_free(request); cws_http_free(request);
/* Clear str */ /* Clear str */
memset(data, 0, sizeof data); memset(data, 0, sizeof data);
@@ -147,11 +147,11 @@ void handle_clients(int sockfd) {
/* TODO: fix endless loop using cli args */ /* TODO: fix endless loop using cli args */
free(revents); free(revents);
close(epfd); close(epfd);
close_fds(clients); cws_server_close_all_fds(clients);
hm_free(clients); cws_hm_free(clients);
} }
void epoll_ctl_add(int epfd, int sockfd, uint32_t events) { void cws_epoll_add(int epfd, int sockfd, uint32_t events) {
struct epoll_event event; struct epoll_event event;
event.events = events; event.events = events;
event.data.fd = sockfd; event.data.fd = sockfd;
@@ -163,7 +163,7 @@ void epoll_ctl_add(int epfd, int sockfd, uint32_t events) {
} }
} }
void epoll_ctl_del(int epfd, int sockfd) { void cws_epoll_del(int epfd, int sockfd) {
const int status = epoll_ctl(epfd, EPOLL_CTL_DEL, sockfd, NULL); const int status = epoll_ctl(epfd, EPOLL_CTL_DEL, sockfd, NULL);
if (status != 0) { if (status != 0) {
@@ -172,7 +172,7 @@ void epoll_ctl_del(int epfd, int sockfd) {
} }
} }
void setnonblocking(int sockfd) { void cws_fd_set_nonblocking(int sockfd) {
const int status = fcntl(sockfd, F_SETFL, O_NONBLOCK); const int status = fcntl(sockfd, F_SETFL, O_NONBLOCK);
if (status == -1) { if (status == -1) {
@@ -181,7 +181,7 @@ void setnonblocking(int sockfd) {
} }
} }
int handle_new_client(int sockfd, struct sockaddr_storage *their_sa, socklen_t *theirsa_size) { int cws_server_accept_client(int sockfd, struct sockaddr_storage *their_sa, socklen_t *theirsa_size) {
const int client_fd = accept(sockfd, (struct sockaddr *)their_sa, theirsa_size); const int client_fd = accept(sockfd, (struct sockaddr *)their_sa, theirsa_size);
if (client_fd == -1) { if (client_fd == -1) {
@@ -194,13 +194,13 @@ int handle_new_client(int sockfd, struct sockaddr_storage *their_sa, socklen_t *
return client_fd; return client_fd;
} }
void close_fds(bucket_t *bucket) { void cws_server_close_all_fds(cws_bucket *bucket) {
for (size_t i = 0; i < HASHMAP_MAX_CLIENTS; ++i) { for (size_t i = 0; i < CWS_HASHMAP_MAX_CLIENTS; ++i) {
close(bucket[i].sockfd); close(bucket[i].sockfd);
if (bucket[i].next != NULL) { if (bucket[i].next != NULL) {
/* Close the fds */ /* Close the fds */
bucket_t *p = bucket[i].next; cws_bucket *p = bucket[i].next;
bucket_t *next = p->next; cws_bucket *next = p->next;
do { do {
close(p->sockfd); close(p->sockfd);
p = next; p = next;
@@ -210,8 +210,8 @@ void close_fds(bucket_t *bucket) {
} }
} }
void close_client(int epfd, int client_fd, bucket_t *clients) { void cws_server_close_client(int epfd, int client_fd, cws_bucket *clients) {
epoll_ctl_del(epfd, client_fd); cws_epoll_del(epfd, client_fd);
hm_remove(clients, client_fd); cws_hm_remove(clients, client_fd);
close(client_fd); close(client_fd);
} }

5
src/utils/config.c Normal file
View File

@@ -0,0 +1,5 @@
#include "utils/config.h"
int nothing() {
return 0;
}

View File

@@ -1,20 +1,23 @@
#include "utils/hashmap.h" #include "utils/hashmap.h"
int hash(int sockfd) { return sockfd % HASHMAP_MAX_CLIENTS; } #include <stdlib.h>
#include <string.h>
void hm_init(bucket_t *bucket) { int cws_hm_hash(int sockfd) { return sockfd % CWS_HASHMAP_MAX_CLIENTS; }
void cws_hm_init(cws_bucket *bucket) {
/* Initialize everything to 0 for the struct, then -1 for fd and next to NULL */ /* Initialize everything to 0 for the struct, then -1 for fd and next to NULL */
memset(bucket, 0, sizeof(bucket_t) * HASHMAP_MAX_CLIENTS); memset(bucket, 0, sizeof(cws_bucket) * CWS_HASHMAP_MAX_CLIENTS);
for (size_t i = 0; i < HASHMAP_MAX_CLIENTS; ++i) { for (size_t i = 0; i < CWS_HASHMAP_MAX_CLIENTS; ++i) {
bucket[i].sockfd = -1; bucket[i].sockfd = -1;
bucket[i].next = NULL; bucket[i].next = NULL;
bucket[i].prev = NULL; bucket[i].prev = NULL;
} }
} }
void hm_insert(bucket_t *bucket, int sockfd, struct sockaddr_storage *sas) { void cws_hm_insert(cws_bucket *bucket, int sockfd, struct sockaddr_storage *sas) {
int index = hash(sockfd); int index = cws_hm_hash(sockfd);
if (bucket[index].sockfd == -1) { if (bucket[index].sockfd == -1) {
/* Current slot is empty */ /* Current slot is empty */
@@ -23,8 +26,8 @@ void hm_insert(bucket_t *bucket, int sockfd, struct sockaddr_storage *sas) {
} else { } else {
/* Append the new key to the head (not the first element because it belongs to the array) of the linked /* Append the new key to the head (not the first element because it belongs to the array) of the linked
* list */ * list */
bucket_t *p = &bucket[index]; cws_bucket *p = &bucket[index];
bucket_t *new = malloc(sizeof(bucket_t)); cws_bucket *new = malloc(sizeof(cws_bucket));
new->sockfd = sockfd; new->sockfd = sockfd;
new->sas = *sas; new->sas = *sas;
new->next = p->next; new->next = p->next;
@@ -33,13 +36,12 @@ void hm_insert(bucket_t *bucket, int sockfd, struct sockaddr_storage *sas) {
} }
} }
bucket_t *hm_lookup(bucket_t *bucket, int sockfd) { cws_bucket *cws_hm_lookup(cws_bucket *bucket, int sockfd) {
int index = hash(sockfd); int index = cws_hm_hash(sockfd);
if (bucket[index].sockfd != sockfd) { if (bucket[index].sockfd != sockfd) {
bucket_t *p; cws_bucket *p;
for (p = bucket[index].next; p != NULL && p->sockfd != sockfd; p = p->next) for (p = bucket[index].next; p != NULL && p->sockfd != sockfd; p = p->next);
;
return p; return p;
} else { } else {
return &bucket[index]; return &bucket[index];
@@ -48,23 +50,23 @@ bucket_t *hm_lookup(bucket_t *bucket, int sockfd) {
return NULL; return NULL;
} }
void hm_push(bucket_t *bucket, int sockfd, struct sockaddr_storage *sas) { void cws_hm_push(cws_bucket *bucket, int sockfd, struct sockaddr_storage *sas) {
if (hm_lookup(bucket, sockfd) == NULL) { if (cws_hm_lookup(bucket, sockfd) == NULL) {
hm_insert(bucket, sockfd, sas); cws_hm_insert(bucket, sockfd, sas);
} }
} }
void hm_remove(bucket_t *bucket, int sockfd) { void cws_hm_remove(cws_bucket *bucket, int sockfd) {
if (hm_is_in_bucket_array(bucket, sockfd)) { if (cws_hm_is_in_bucket_array(bucket, sockfd)) {
/* Instead of doing this I could copy the memory of the next node into the head */ /* Instead of doing this I could copy the memory of the next node into the head */
int index = hash(sockfd); int index = cws_hm_hash(sockfd);
bucket[index].sockfd = -1; bucket[index].sockfd = -1;
return; return;
} }
/* Key not in the bucket array, let's search in the linked list */ /* Key not in the bucket array, let's search in the linked list */
bucket_t *p = hm_lookup(bucket, sockfd); cws_bucket *p = cws_hm_lookup(bucket, sockfd);
if (p == NULL) return; if (p == NULL) return;
p->prev->next = p->next; p->prev->next = p->next;
if (p->next != NULL) { if (p->next != NULL) {
@@ -74,10 +76,10 @@ void hm_remove(bucket_t *bucket, int sockfd) {
free(p); free(p);
} }
void hm_free(bucket_t *bucket) { void cws_hm_free(cws_bucket *bucket) {
bucket_t *p, *next; cws_bucket *p, *next;
for (size_t i = 0; i < HASHMAP_MAX_CLIENTS; ++i) { for (size_t i = 0; i < CWS_HASHMAP_MAX_CLIENTS; ++i) {
if (bucket[i].next != NULL) { if (bucket[i].next != NULL) {
/* Free the malloc */ /* Free the malloc */
p = bucket[i].next; p = bucket[i].next;
@@ -91,8 +93,8 @@ void hm_free(bucket_t *bucket) {
} }
} }
bool hm_is_in_bucket_array(bucket_t *bucket, int sockfd) { bool cws_hm_is_in_bucket_array(cws_bucket *bucket, int sockfd) {
int index = hash(sockfd); int index = cws_hm_hash(sockfd);
if (bucket[index].sockfd == sockfd) return true; if (bucket[index].sockfd == sockfd) return true;
return false; return false;

View File

@@ -1,8 +1,12 @@
#include "utils/utils.h" #include "utils/utils.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "utils/colors.h" #include "utils/colors.h"
void print_ips(const char *hostname, const char *port) { void cws_utils_print_ips(const char *hostname, const char *port) {
struct addrinfo ai; struct addrinfo ai;
struct addrinfo *res; struct addrinfo *res;
@@ -35,7 +39,7 @@ void print_ips(const char *hostname, const char *port) {
freeaddrinfo(res); freeaddrinfo(res);
} }
void get_client_ip(struct sockaddr_storage *sa, char *ip) { void cws_utils_get_client_ip(struct sockaddr_storage *sa, char *ip) {
struct sockaddr_in *sin = (struct sockaddr_in *)sa; struct sockaddr_in *sin = (struct sockaddr_in *)sa;
inet_ntop(AF_INET, &sin->sin_addr, ip, INET_ADDRSTRLEN); inet_ntop(AF_INET, &sin->sin_addr, ip, INET_ADDRSTRLEN);