diff --git a/hello-epoll/Makefile b/hello-epoll/Makefile new file mode 100644 index 0000000..763aa71 --- /dev/null +++ b/hello-epoll/Makefile @@ -0,0 +1,13 @@ +CXX = g++ +CXXFLAGS = -Wall -Wextra -std=c++11 +TARGET = hello-epoll + +all: $(TARGET) + +$(TARGET): hello-epoll.cpp + $(CXX) $(CXXFLAGS) -o $(TARGET) hello-epoll.cpp + +clean: + rm -f $(TARGET) + +.PHONY: all clean diff --git a/hello-epoll/hello-epoll b/hello-epoll/hello-epoll new file mode 100755 index 0000000..5c970f5 Binary files /dev/null and b/hello-epoll/hello-epoll differ diff --git a/hello-epoll/hello-epoll.cpp b/hello-epoll/hello-epoll.cpp new file mode 100644 index 0000000..458f229 --- /dev/null +++ b/hello-epoll/hello-epoll.cpp @@ -0,0 +1,152 @@ +#include +#include +#include +#include + +#include +#include +#include + +#define MAX_EVENTS 10 +#define BUFFER_SIZE 1024 +#define FIFO_PATH "/tmp/hello_epoll_fifo" + +int main() { + int epoll_fd, stdin_fd, fifo_fd; + struct epoll_event ev, events[MAX_EVENTS]; + char buffer[BUFFER_SIZE]; + + // Create a named pipe (FIFO) + unlink(FIFO_PATH); // Remove if it already exists + if (mkfifo(FIFO_PATH, 0666) == -1) { + std::cerr << "Error creating FIFO: " << strerror(errno) << std::endl; + return 1; + } + std::cout << "Created FIFO at: " << FIFO_PATH << std::endl; + std::cout << "From another terminal, run: echo 'Hello from FIFO' > " << FIFO_PATH << std::endl; + std::cout << "----------------------------------------" << std::endl; + + // Create epoll instance + epoll_fd = epoll_create1(0); + if (epoll_fd == -1) { + std::cerr << "Error creating epoll: " << strerror(errno) << std::endl; + unlink(FIFO_PATH); + return 1; + } + + // Set stdin to non-blocking mode + stdin_fd = STDIN_FILENO; + int flags = fcntl(stdin_fd, F_GETFL, 0); + fcntl(stdin_fd, F_SETFL, flags | O_NONBLOCK); + + // Add stdin to epoll + ev.events = EPOLLIN; + ev.data.fd = stdin_fd; + if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, stdin_fd, &ev) == -1) { + std::cerr << "Error adding stdin to epoll: " << strerror(errno) << std::endl; + close(epoll_fd); + unlink(FIFO_PATH); + return 1; + } + std::cout << "Added stdin (fd=" << stdin_fd << ") to epoll" << std::endl; + + // Open FIFO in non-blocking mode for reading + fifo_fd = open(FIFO_PATH, O_RDONLY | O_NONBLOCK); + if (fifo_fd == -1) { + std::cerr << "Error opening FIFO: " << strerror(errno) << std::endl; + close(epoll_fd); + unlink(FIFO_PATH); + return 1; + } + + // Add FIFO to epoll + ev.events = EPOLLIN; + ev.data.fd = fifo_fd; + if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fifo_fd, &ev) == -1) { + std::cerr << "Error adding FIFO to epoll: " << strerror(errno) << std::endl; + close(fifo_fd); + close(epoll_fd); + unlink(FIFO_PATH); + return 1; + } + std::cout << "Added FIFO (fd=" << fifo_fd << ") to epoll" << std::endl; + std::cout << "----------------------------------------" << std::endl; + std::cout << "Waiting for events... (type 'quit' to exit)" << std::endl; + std::cout << std::endl; + + // Event loop + bool running = true; + while (running) { + int nfds = epoll_wait(epoll_fd, events, MAX_EVENTS, -1); + if (nfds == -1) { + std::cerr << "Error in epoll_wait: " << strerror(errno) << std::endl; + break; + } + + // Process all ready file descriptors + for (int i = 0; i < nfds; i++) { + int fd = events[i].data.fd; + + if (events[i].events & EPOLLIN) { + ssize_t bytes_read = read(fd, buffer, BUFFER_SIZE - 1); + + if (bytes_read > 0) { + buffer[bytes_read] = '\0'; + + if (fd == stdin_fd) { + std::cout << "[STDIN] Received: " << buffer; + + // Check for quit command + if (strncmp(buffer, "quit", 4) == 0) { + running = false; + break; + } + } else if (fd == fifo_fd) { + std::cout << "[FIFO] Received: " << buffer; + } + } else if (bytes_read == 0) { + // EOF - for FIFO, this happens when writer closes + if (fd == fifo_fd) { + std::cout << "[FIFO] Writer closed, reopening..." << std::endl; + close(fifo_fd); + + // Reopen FIFO + fifo_fd = open(FIFO_PATH, O_RDONLY | O_NONBLOCK); + if (fifo_fd == -1) { + std::cerr << "Error reopening FIFO: " << strerror(errno) << std::endl; + running = false; + break; + } + + // Re-add to epoll + ev.events = EPOLLIN; + ev.data.fd = fifo_fd; + if (epoll_ctl(epoll_fd, EPOLL_CTL_ADD, fifo_fd, &ev) == -1) { + std::cerr << "Error re-adding FIFO to epoll: " << strerror(errno) << std::endl; + running = false; + break; + } + } + } else { + // Error reading + if (errno != EAGAIN && errno != EWOULDBLOCK) { + std::cerr << "Error reading from fd " << fd << ": " << strerror(errno) << std::endl; + } + } + } + + if (events[i].events & (EPOLLERR | EPOLLHUP)) { + std::cerr << "Error/Hangup on fd " << fd << std::endl; + } + } + } + + // Cleanup + std::cout << std::endl + << "Cleaning up..." << std::endl; + close(fifo_fd); + close(epoll_fd); + unlink(FIFO_PATH); + + return 0; +}