Initial commit
- basic main.c implementation - event_loop implementation
This commit is contained in:
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
@@ -0,0 +1,3 @@
|
|||||||
|
*.swp
|
||||||
|
*.swo
|
||||||
|
build/
|
||||||
13
build.sh
Executable file
13
build.sh
Executable file
@@ -0,0 +1,13 @@
|
|||||||
|
#!/usr/bin/bash
|
||||||
|
|
||||||
|
# check if build directory exists
|
||||||
|
if [ -d build ]; then
|
||||||
|
echo "[I] build/ directory exists, deleting..."
|
||||||
|
rm -rf build
|
||||||
|
fi
|
||||||
|
# create build/ directory
|
||||||
|
mkdir build
|
||||||
|
|
||||||
|
# build
|
||||||
|
echo "[I] Building ipoim..."
|
||||||
|
gcc *.c $(curl-config --cflags) $(curl-config --libs) -I. -o build/ipoim
|
||||||
63
event_loop.c
Normal file
63
event_loop.c
Normal file
@@ -0,0 +1,63 @@
|
|||||||
|
#include "event_loop.h"
|
||||||
|
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
#include <errno.h>
|
||||||
|
#include <sys/epoll.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Macros
|
||||||
|
*/
|
||||||
|
#define MAX_EPOLL_EVENTS (20)
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Data
|
||||||
|
*/
|
||||||
|
static int _efd = -1;
|
||||||
|
static loop_cb_t _cb = NULL;
|
||||||
|
static struct epoll_event _evs[MAX_EPOLL_EVENTS];
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Public API
|
||||||
|
*/
|
||||||
|
bool loop_init(loop_cb_t callback) {
|
||||||
|
_efd = epoll_create1(0);
|
||||||
|
if (_efd == -1) {
|
||||||
|
printf("[!] epoll_create1 failed: %d\n", errno);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
_cb = callback;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
void loop_deinit() {
|
||||||
|
close(_efd);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool loop_ctl(int op, int fd, struct epoll_event *event) {
|
||||||
|
if (epoll_ctl(_efd, op, fd, event) == -1) {
|
||||||
|
printf("[!] epoll_ctl failed: %d\n", errno);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool loop_wait() {
|
||||||
|
int result = epoll_wait(
|
||||||
|
_efd,
|
||||||
|
_evs,
|
||||||
|
MAX_EPOLL_EVENTS,
|
||||||
|
-1
|
||||||
|
);
|
||||||
|
if (result < 0) {
|
||||||
|
printf("[!] epoll_wait failed: %d\n", errno);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
if (!result) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
for (int i = 0; i < result; ++i) {
|
||||||
|
_cb(_evs[i].data.fd, _evs[i].events);
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
59
event_loop.h
Normal file
59
event_loop.h
Normal file
@@ -0,0 +1,59 @@
|
|||||||
|
#ifndef __EVENT_LOOP_H
|
||||||
|
#define __EVENT_LOOP_H
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
|
||||||
|
#include <sys/epoll.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Type definitions
|
||||||
|
*/
|
||||||
|
typedef void (*loop_cb_t)(int fd, uint32_t events);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Public API
|
||||||
|
*/
|
||||||
|
// Initialize the event loop.
|
||||||
|
// Parameters:
|
||||||
|
// - callback - callback to use in loop_wait
|
||||||
|
// Returns:
|
||||||
|
// - true on success
|
||||||
|
// - false on failure
|
||||||
|
// Remarks:
|
||||||
|
// - must be called before event loop is used
|
||||||
|
bool loop_init(loop_cb_t callback);
|
||||||
|
|
||||||
|
// Finalize the event loop.
|
||||||
|
// - must be called on app termination
|
||||||
|
void loop_deinit();
|
||||||
|
|
||||||
|
// Set callback for file descriptor event.
|
||||||
|
// Remarks:
|
||||||
|
// - this callback gets called in loop_wait
|
||||||
|
void loop_set_callback(loop_cb_t callback);
|
||||||
|
|
||||||
|
// Call epoll_ctl.
|
||||||
|
// Parameters:
|
||||||
|
// - op - op
|
||||||
|
// - fd - fd
|
||||||
|
// - event - event
|
||||||
|
// Returns:
|
||||||
|
// - true on success
|
||||||
|
// - false on failure
|
||||||
|
// Remarks:
|
||||||
|
// - just calls epoll_ctl
|
||||||
|
bool loop_ctl(int op, int fd, struct epoll_event *event);
|
||||||
|
|
||||||
|
// Call epoll_wait and process the events.
|
||||||
|
// Returns:
|
||||||
|
// - true on success
|
||||||
|
// - false on failure
|
||||||
|
// Remarks:
|
||||||
|
// - you should terminate the app if this function
|
||||||
|
// fails
|
||||||
|
bool loop_wait();
|
||||||
|
|
||||||
|
|
||||||
|
#endif
|
||||||
99
main.c
Normal file
99
main.c
Normal file
@@ -0,0 +1,99 @@
|
|||||||
|
#include <stdbool.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
#include <errno.h>
|
||||||
|
#include <signal.h>
|
||||||
|
#include <sys/epoll.h>
|
||||||
|
#include <sys/signalfd.h>
|
||||||
|
|
||||||
|
#include "event_loop.h"
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Data
|
||||||
|
*/
|
||||||
|
static bool _to_work = true;
|
||||||
|
static int _sig_fd = -1;
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Private API
|
||||||
|
*/
|
||||||
|
// Initialize signals.
|
||||||
|
// Returns:
|
||||||
|
// - true on success
|
||||||
|
// Remarks:
|
||||||
|
// - loop_init() must be called before this one
|
||||||
|
static bool signals_init() {
|
||||||
|
sigset_t mask;
|
||||||
|
sigemptyset(&mask);
|
||||||
|
sigaddset(&mask, SIGTERM);
|
||||||
|
sigaddset(&mask, SIGINT);
|
||||||
|
sigaddset(&mask, SIGQUIT);
|
||||||
|
if (sigprocmask(SIG_BLOCK, &mask, NULL) == -1) {
|
||||||
|
printf("[!] sigprocmask failed: %d\n", errno);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
_sig_fd = signalfd(
|
||||||
|
-1,
|
||||||
|
&mask,
|
||||||
|
SFD_NONBLOCK
|
||||||
|
);
|
||||||
|
if (_sig_fd == -1) {
|
||||||
|
printf("[!] signalfd failed: %d\n", errno);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
struct epoll_event ev;
|
||||||
|
ev.data.fd = _sig_fd;
|
||||||
|
ev.events = EPOLLIN;
|
||||||
|
bool loop_res = loop_ctl(
|
||||||
|
EPOLL_CTL_ADD,
|
||||||
|
_sig_fd,
|
||||||
|
&ev
|
||||||
|
);
|
||||||
|
if (!loop_res) {
|
||||||
|
close(_sig_fd);
|
||||||
|
printf("[!] loop_ctl failed\n");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void signals_deinit() {
|
||||||
|
close(_sig_fd);
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Callbacks and main
|
||||||
|
*/
|
||||||
|
// This callback is called by loop_wait when event happens.
|
||||||
|
static void on_fd_event(int fd, uint32_t events) {
|
||||||
|
// Termination
|
||||||
|
if (fd == _sig_fd) {
|
||||||
|
_to_work = false;
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
printf("[!] Event has happened on an unknown file descriptor\n");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char **argv) {
|
||||||
|
if (!loop_init(on_fd_event)) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
if (!signals_init()) {
|
||||||
|
loop_deinit();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
while (_to_work) {
|
||||||
|
if (!loop_wait()) {
|
||||||
|
printf("[!] loop_wait returns false, stopping...\n");
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
printf("--- loop ---\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
signals_deinit();
|
||||||
|
loop_deinit();
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user