/* Copyright 2016 Alex 'AdUser' Z (ad_user@runbox.com)
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */

#include "common.h"
#include "log.h"
#include "matches.h"
#include "ipaddr.h"
#include "statefile.h"

f2b_statefile_t *
f2b_statefile_create(const char *statedir, const char *jailname) {
  f2b_statefile_t *sf = NULL;
  char path[PATH_MAX];

  assert(statedir != NULL);
  assert(jailname != NULL);

  snprintf(path, sizeof(path), "%s/%s.state", statedir, jailname);

  if (access(path, R_OK | W_OK) < 0) {
    if (errno != ENOENT) {
      f2b_log_msg(log_error, "can't access statefile: %s", strerror(errno));
      return NULL;
    }
    mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP; /* 0640 */
    int fd = open(path, O_CREAT | O_WRONLY, mode);
    if (fd < 0) {
      f2b_log_msg(log_error, "can't create statefile: %s", strerror(errno));
      return NULL;
    }
    close(fd);
  }

  if ((sf = calloc(1, sizeof(f2b_statefile_t))) == NULL) {
    f2b_log_msg(log_error, "can't allocate memory for statefile: %s", strerror(errno));
    return NULL;
  }

  strlcpy(sf->path, path, sizeof(sf->path));

  return sf;
}

void
f2b_statefile_destroy(f2b_statefile_t *sf) {
  if (!sf) return;
  free(sf);
}

f2b_ipaddr_t *
f2b_statefile_load(f2b_statefile_t *sf) {
  const int fields = 3;
  const char *format = "%48s %u %u"; /* 48 == IPADDR_MAX == sizeof(addr) */
  f2b_ipaddr_t *addrlist = NULL, *ipaddr = NULL;
  char buf[256], addr[IPADDR_MAX], *p;
  unsigned int banned_at, release_at;
  FILE *f = NULL;

  assert(sf != NULL);

  if ((f = fopen(sf->path, "r")) == NULL) {
    f2b_log_msg(log_error, "can't open statefile: %s", strerror(errno));
    return NULL;
  }

  while(fgets(buf, sizeof(buf), f)) {
    p = &buf[0];
    while(isspace(*p))
      p++; /* skip leading spaces */
    if (*p == '#')
      continue; /* is comment */
    if (sscanf(p, format, addr, &banned_at, &release_at) != fields) {
      f2b_log_msg(log_warn, "can't parse, ignoring line: %s", buf);
      continue;
    }
    if ((ipaddr = f2b_ipaddr_create(addr)) == NULL) {
      f2b_log_msg(log_warn, "can't parse addr: %s", addr);
      continue;
    }
    ipaddr->banned = true;
    ipaddr->banned_at  = banned_at;
    ipaddr->release_at = release_at;
    ipaddr->next = addrlist;
    addrlist = ipaddr;
  }

  fclose(f);
  return addrlist;
}

bool
f2b_statefile_save(f2b_statefile_t *sf, f2b_ipaddr_t *addrlist) {
  const char *fmt = "%s\t%lu\t%lu\n";
  char path[PATH_MAX];
  FILE *f = NULL;

  assert(sf != NULL);

  if (addrlist == NULL)
    return true;

  strlcpy(path, sf->path, sizeof(path));
  strlcat(path, ".new",   sizeof(path));

  if ((f = fopen(path, "w")) == NULL) {
    f2b_log_msg(log_error, "can't open statefile: %s", strerror(errno));
    return false;
  }

  fprintf(f, "# address\tbanned_at\trelease_at\n");
  for (f2b_ipaddr_t *ipaddr = addrlist; ipaddr != NULL; ipaddr = ipaddr->next) {
    if (!ipaddr->banned)
      continue;
    fprintf(f, fmt, ipaddr->text, ipaddr->banned_at, ipaddr->release_at);
  }
  fclose(f);

  if (rename(path, sf->path) < 0) {
    f2b_log_msg(log_error, "can't replace statefile %s: %s", sf->path, strerror(errno));
    return false;
  }

  sf->need_save = false;
  return true;
}