pgs/pgs.c

// Written by retoor@molodetz.nl

// This source code sets up a simple TCP server that listens for connections and
// handles cleanup on exit. The server is intended to interact with an upstream
// server defined by its host and port.

// Imports: Custom includes 'py.h' and 'sock.h' for additional functionality.

// MIT License

#include "py.h"
#include "sock.h"
#include <Python.h>
#include <arpa/inet.h>
#include <errno.h>
#include <fcntl.h>
#include <netinet/in.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#include <unistd.h>
#define PY_SSIZE_T_CLEAN 1
#define LISTEN_PORT 2222
#define UPSTREAM_HOST "127.0.0.1"
#define UPSTREAM_PORT 9999

void cleanup() {
  close(epoll_fd);
  close(listen_fd);
  py_destruct();
  printf("Graceful exit.\n");
}

void handle_sigint(int sig) {
  printf("\nCtrl+C pressed.\n");
  exit(0);
}

int main() {
  if (signal(SIGINT, handle_sigint) == SIG_ERR) {
    perror("Failed to register signal handler");
    return EXIT_FAILURE;
  }
  atexit(cleanup);

  serve(LISTEN_PORT);

  return EXIT_SUCCESS;
}