#include "donar_init.h"

void free_udp_t(void* v) {
  struct udp_target* udp_t = v;
  udp_t->ref_count--;
  if (udp_t->ref_count <= 0) {
    free(udp_t);
  }
}

int on_signal(struct evt_core_ctx* evts, struct evt_core_fdinfo* fdinfo) {
  struct signalfd_siginfo fdsi;

  int sig_rd = read(fdinfo->fd, &fdsi, sizeof(struct signalfd_siginfo));
  if (sig_rd == -1 && errno == EAGAIN) return 1;
  if (sig_rd != sizeof(struct signalfd_siginfo)) {
    fprintf(stderr, "read size: %d / %ld\n", sig_rd, sizeof(struct signalfd_siginfo));
    perror("signal read failed");
    exit(EXIT_FAILURE);
  }
  printf("Signal received: %d\n", fdsi.ssi_signo);

  if (fdsi.ssi_signo == SIGINT || fdsi.ssi_signo == SIGQUIT || fdsi.ssi_signo == SIGTERM) {
    printf("Stop main loop and quit\n");
    evts->loop = 0;
  }
  return 0;
}

void signal_init(struct evt_core_ctx* evts) {
  sigset_t mask = {0};

  struct evt_core_cat signal_read = {
    .name = "signal-read",
    .flags = EPOLLIN | EPOLLET | EPOLLRDHUP,
    .app_ctx = NULL,
    .free_app_ctx = NULL,
    .cb = on_signal,
    .err_cb = NULL
  };
  evt_core_add_cat(evts, &signal_read);

  sigaddset(&mask, SIGINT);
  sigaddset(&mask, SIGQUIT);
  sigaddset(&mask, SIGTERM);
  if (sigprocmask(SIG_BLOCK, &mask, NULL) == -1) {
    perror("sigprocmask");
    exit(EXIT_FAILURE);
  }
  int sigfd = signalfd(-1, &mask, 0);
  if (sigfd == -1) {
    perror("signalfd failed");
    exit(EXIT_FAILURE);
  }

  struct evt_core_fdinfo fdinfo = {0};
  fdinfo.fd = sigfd;
  fdinfo.cat = &signal_read;
  fdinfo.url = "signal:read:int+quit";
  evt_core_add_fd(evts, &fdinfo);
}

void init_udp_remote(char* port, struct evt_core_ctx* evts) {
  int sock1, sock2;
  char url[1024];
  struct evt_core_cat cat = {0};
  struct evt_core_fdinfo fdinfo = {0};

  // 1. Init shared parameters for the fdinfo structure
  struct udp_target* udp_t = malloc(sizeof(struct udp_target));
  if (udp_t == NULL) goto socket_failed;
  memset(udp_t, 0, sizeof(struct udp_target));
  udp_t->ref_count = 2;

  fdinfo.cat = &cat;
  fdinfo.url = url;
  fdinfo.free_other = free_udp_t;
  fdinfo.other = udp_t;

  // 2. Duplicate sockets
  sock1 = create_udp_server ("127.13.3.7", port);
  if (sock1 < 0) goto socket_failed;
  sock2 = dup(sock1);
  if (sock2 < 0) goto socket_failed;

  // 3. Register them
  fdinfo.cat->name = "udp-read";
  fdinfo.fd = sock1;
  sprintf(fdinfo.url, "udp:read:127.0.0.1:%s", port);
  evt_core_add_fd (evts, &fdinfo);

  fdinfo.cat->name = "udp-write";
  fdinfo.fd = sock2;
  sprintf(fdinfo.url, "udp:write:127.0.0.1:%s", port);
  evt_core_add_fd (evts, &fdinfo);
  return;

socket_failed:
  fprintf(stderr, "UDP socket init failed\n");
  exit(EXIT_FAILURE);
}

void init_udp_exposed(char* bound_ip, char* port, struct evt_core_ctx* evts) {
  int sock1, sock2;
  char url[1024];

  struct evt_core_cat cat = {0};
  struct evt_core_fdinfo fdinfo = {0};

  // 1. Init shared parameters for the fdinfo structure
  struct udp_target* udp_t = malloc(sizeof(struct udp_target));
  if (udp_t == NULL) goto socket_failed;
  memset(udp_t, 0, sizeof(struct udp_target));
  udp_t->ref_count = 2;

  fdinfo.cat = &cat;
  fdinfo.url = url;
  fdinfo.free_other = free_udp_t;
  fdinfo.other = udp_t;

  sock1 = create_udp_client (bound_ip, port);
  if (sock1 < 0) goto socket_failed;
  sock2 = dup(sock1);
  if (sock2 < 0) goto socket_failed;

  fdinfo.fd = sock1;
  fdinfo.cat->name = "udp-read";
  sprintf(fdinfo.url, "udp:read:127.0.0.1:%s", port);
  evt_core_add_fd (evts, &fdinfo);

  fdinfo.fd = sock2;
  fdinfo.cat->name = "udp-write";
  sprintf(fdinfo.url, "udp:write:127.0.0.1:%s", port);
  evt_core_add_fd (evts, &fdinfo);

  return;

socket_failed:
  fprintf(stderr, "UDP socket init failed\n");
  exit(EXIT_FAILURE);
}

void free_port (void* ptr) {
  free(ptr);
}

void donar_init_params(struct donar_params* dp) {
  dp->onion_file = NULL;
  dp->algo = NULL;
  dp->capture_file = NULL;
  dp->bound_ip = NULL;
  dp->is_server = 0;
  dp->is_client = 0;
  dp->algo_specific_params = NULL;
  dp->is_waiting_bootstrap = 0;
  dp->errored = 0;
  dp->links = 8;
  dp->fresh_data = 1;
  dp->redundant_data = 0;
  dp->remote_ports = g_ptr_array_new_with_free_func (free_port);
  dp->exposed_ports = g_ptr_array_new_with_free_func (free_port);
}