Implemented server
This commit is contained in:
45
main.c
45
main.c
@@ -8,12 +8,14 @@
|
|||||||
#include <sys/signalfd.h>
|
#include <sys/signalfd.h>
|
||||||
|
|
||||||
#include "event_loop.h"
|
#include "event_loop.h"
|
||||||
|
#include "server.h"
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Data
|
* Data
|
||||||
*/
|
*/
|
||||||
static bool _to_work = true;
|
static bool _to_work = true;
|
||||||
static int _sig_fd = -1;
|
static int _sig_fd = -1;
|
||||||
|
static int _server_fd = -1;
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Private API
|
* Private API
|
||||||
@@ -58,10 +60,40 @@ static bool signals_init() {
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Finalize signals.
|
||||||
static void signals_deinit() {
|
static void signals_deinit() {
|
||||||
close(_sig_fd);
|
close(_sig_fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Initialize the server.
|
||||||
|
// Returns:
|
||||||
|
// - true on success
|
||||||
|
static bool server_init() {
|
||||||
|
_server_fd = server_open("0.0.0.0", 20070);
|
||||||
|
if (_server_fd == -1) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
struct epoll_event ev;
|
||||||
|
ev.data.fd = _server_fd;
|
||||||
|
ev.events = EPOLLIN;
|
||||||
|
bool loop_res = loop_ctl(
|
||||||
|
EPOLL_CTL_ADD,
|
||||||
|
_server_fd,
|
||||||
|
&ev
|
||||||
|
);
|
||||||
|
if (!loop_res) {
|
||||||
|
server_close();
|
||||||
|
printf("[!] loop_ctl failed\n");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Finalize the server.
|
||||||
|
static void server_deinit() {
|
||||||
|
server_close();
|
||||||
|
}
|
||||||
|
|
||||||
/*
|
/*
|
||||||
* Callbacks and main
|
* Callbacks and main
|
||||||
*/
|
*/
|
||||||
@@ -71,6 +103,13 @@ static void on_fd_event(int fd, uint32_t events) {
|
|||||||
if (fd == _sig_fd) {
|
if (fd == _sig_fd) {
|
||||||
_to_work = false;
|
_to_work = false;
|
||||||
}
|
}
|
||||||
|
// Server event
|
||||||
|
else if (fd == _server_fd) {
|
||||||
|
if (!server_event(_server_fd)) {
|
||||||
|
printf("[!] server_event failed\n");
|
||||||
|
_to_work = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
else {
|
else {
|
||||||
printf("[!] Event has happened on an unknown file descriptor\n");
|
printf("[!] Event has happened on an unknown file descriptor\n");
|
||||||
}
|
}
|
||||||
@@ -84,6 +123,11 @@ int main(int argc, char **argv) {
|
|||||||
loop_deinit();
|
loop_deinit();
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
if (!server_init()) {
|
||||||
|
signals_deinit();
|
||||||
|
loop_deinit();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
while (_to_work) {
|
while (_to_work) {
|
||||||
if (!loop_wait()) {
|
if (!loop_wait()) {
|
||||||
@@ -93,6 +137,7 @@ int main(int argc, char **argv) {
|
|||||||
printf("--- loop ---\n");
|
printf("--- loop ---\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
server_close();
|
||||||
signals_deinit();
|
signals_deinit();
|
||||||
loop_deinit();
|
loop_deinit();
|
||||||
return 0;
|
return 0;
|
||||||
|
|||||||
83
server.c
Normal file
83
server.c
Normal file
@@ -0,0 +1,83 @@
|
|||||||
|
#include "server.h"
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
|
||||||
|
#include <errno.h>
|
||||||
|
#include <netdb.h>
|
||||||
|
#include <netinet/in.h>
|
||||||
|
#include <sys/epoll.h>
|
||||||
|
#include <sys/socket.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Macros
|
||||||
|
*/
|
||||||
|
#define LISTEN_BACKLOG (2048)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Data
|
||||||
|
*/
|
||||||
|
static int _fd = -1;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Private API
|
||||||
|
*/
|
||||||
|
bool _accept() {
|
||||||
|
close(accept(_fd, 0, 0));
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Public API
|
||||||
|
*/
|
||||||
|
int server_open(const char* host, int port) {
|
||||||
|
struct hostent *he = gethostbyname(host);
|
||||||
|
if (!he) {
|
||||||
|
printf("[!] gethostbyname failed: %d (h_errno)\n", h_errno);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
_fd = socket(
|
||||||
|
AF_INET,
|
||||||
|
SOCK_STREAM | SOCK_NONBLOCK,
|
||||||
|
IPPROTO_TCP
|
||||||
|
);
|
||||||
|
if (_fd == -1) {
|
||||||
|
printf("[!] socket failed: %d\n", errno);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
const int opt = 1;
|
||||||
|
if (setsockopt(_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) < 0) {
|
||||||
|
close(_fd);
|
||||||
|
printf("[!] setsockopt failed: %d\n", errno);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
struct sockaddr_in addr;
|
||||||
|
addr.sin_family = AF_INET;
|
||||||
|
addr.sin_port = htobe16(port);
|
||||||
|
memcpy(&addr.sin_addr, he->h_addr, he->h_length);
|
||||||
|
if (bind(_fd, (const void *)&addr, sizeof(addr)) == -1) {
|
||||||
|
close(_fd);
|
||||||
|
printf("[!] bind failed: %d\n", errno);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
if (listen(_fd, LISTEN_BACKLOG) == -1) {
|
||||||
|
close(_fd);
|
||||||
|
printf("[!] listen failed: %d\n", errno);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
return _fd;
|
||||||
|
}
|
||||||
|
|
||||||
|
void server_close() {
|
||||||
|
shutdown(_fd, SHUT_RDWR);
|
||||||
|
close(_fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool server_event(uint32_t events) {
|
||||||
|
if (events & EPOLLIN) {
|
||||||
|
return _accept();
|
||||||
|
}
|
||||||
|
// we assume any event other than EPOLLIN is error
|
||||||
|
return false;
|
||||||
|
}
|
||||||
36
server.h
Normal file
36
server.h
Normal file
@@ -0,0 +1,36 @@
|
|||||||
|
#ifndef __SERVER_H
|
||||||
|
#define __SERVER_H
|
||||||
|
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
// Open the server.
|
||||||
|
// Parameters:
|
||||||
|
// - host - hostname or IP address (domain names are OK)
|
||||||
|
// - port - port to use
|
||||||
|
// Returns:
|
||||||
|
// - -1 on failure
|
||||||
|
// - file descriptor of the server on success
|
||||||
|
// Remarks:
|
||||||
|
// - call only after server_init
|
||||||
|
// - you must add returned fd to epoll with EPOLLIN events
|
||||||
|
int server_open(const char* host, int port);
|
||||||
|
|
||||||
|
// Close the server.
|
||||||
|
// Remarks:
|
||||||
|
// - the socket is automatically removed from epoll
|
||||||
|
void server_close();
|
||||||
|
|
||||||
|
// Notify the server that an event happened on its socket.
|
||||||
|
// Returns:
|
||||||
|
// - true on success
|
||||||
|
// - false on failure
|
||||||
|
// Remarks:
|
||||||
|
// - the server accepts the connection and starts managing
|
||||||
|
// it with connection.h
|
||||||
|
// - you should stop close the server if this function
|
||||||
|
// returns false
|
||||||
|
bool server_event(uint32_t events);
|
||||||
|
|
||||||
|
|
||||||
|
#endif
|
||||||
Reference in New Issue
Block a user