trunk/src/conf_db.c

/*
 *  ircd-hybrid: an advanced Internet Relay Chat Daemon(ircd).
 *
 *  Copyright (C) 1996-2002 by Andrew Church <achurch@achurch.org>
 *  Copyright (C) 2012 by the Hybrid Development Team.
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 *  USA
 */

/*! \file conf_db.c
 * \brief Includes file utilities for database handling
 * \version $Id: conf_db.c 1569 2012-10-16 18:46:53Z michael $
 */

#include "stdinc.h"
#include "conf_db.h"
#include "memory.h"
#include "client.h"
#include "log.h"
#include "send.h"
#include "irc_string.h"
#include "conf.h"
#include "hostmask.h"
#include "resv.h"


/*! \brief Return the version number on the file.  Return 0 if there is no version
 * number or the number doesn't make sense (i.e. less than 1 or greater
 * than FILE_VERSION).
 *
 * \param f dbFile Struct Member
 * \return int 0 if failure, 1 > is the version number
 */
uint32_t
get_file_version(struct dbFILE *f)
{
  uint32_t version = 0;

  if (read_uint32(&version, f) == -1)
  {
    ilog(LOG_TYPE_IRCD, "Error reading version number on %s: %s",
         f->filename, strerror(errno));
    return 0;
  }

  if (version < 1)
  {
    ilog(LOG_TYPE_IRCD, "Invalid version number (%u) on %s",
         version, f->filename);
    return 0;
  }

  return version;
}

/*! \brief Write the current version number to the file.
 * \param f dbFile Struct Member
 * \return 0 on error, 1 on success.
 */
int
write_file_version(struct dbFILE *f, uint32_t version)
{
  if (write_uint32(version, f) == -1)
  {
    ilog(LOG_TYPE_IRCD, "Error writing version number on %s",
         f->filename);
    return 0;
  }

  return 1;
}

/*! \brief Open the database for reading
 * \param service If error whom to return the error as
 * \param filename File to open as the database
 * \return dbFile struct
 */
static struct dbFILE *
open_db_read(const char *service, const char *filename)
{
  struct dbFILE *f = MyMalloc(sizeof(*f));
  FILE *fp = NULL;

  strlcpy(f->filename, filename, sizeof(f->filename));

  f->mode = 'r';
  fp = fopen(f->filename, "rb");

  if (!fp)
  {
    int errno_save = errno;

    if (errno != ENOENT)
      ilog(LOG_TYPE_IRCD, "Can't read %s database %s", service,
           f->filename);

    MyFree(f);
    errno = errno_save;
    return NULL;
  }

  f->fp = fp;
  f->backupfp = NULL;

  return f;
}

/*! \brief Open the database for writting
 * \param service If error whom to return the error as
 * \param filename File to open as the database
 * \param version Database Version
 * \return dbFile struct
 */
static struct dbFILE *
open_db_write(const char *service, const char *filename,
              uint32_t version)
{
  struct dbFILE *f = MyMalloc(sizeof(*f));
  int fd = 0;

  strlcpy(f->filename, filename, sizeof(f->filename));

  filename = f->filename;
  f->mode = 'w';

  snprintf(f->backupname, sizeof(f->backupname), "%s.save", filename);

  if (!*f->backupname || !strcmp(f->backupname, filename))
  {
    int errno_save = errno;

    ilog(LOG_TYPE_IRCD, "Opening %s database %s for write: Filename too long",
         service, filename);
    MyFree(f);
    errno = errno_save;
    return NULL;
  }

  unlink(filename);

  f->backupfp = fopen(filename, "rb");

  if (rename(filename, f->backupname) < 0 && errno != ENOENT)
  {
    int errno_save = errno;
    static int walloped = 0;

    if (!walloped++)
      sendto_realops_flags(UMODE_ALL, L_ALL, SEND_NOTICE,
                           "Can't back up %s database %s",
                           service, filename);

    errno = errno_save;
    ilog(LOG_TYPE_IRCD, "Can't back up %s database %s", service, filename);

    if (f->backupfp)
      fclose(f->backupfp);

    MyFree(f);
    errno = errno_save;
    return NULL;
  }

  unlink(filename);

  /* Use open() to avoid people sneaking a new file in under us */
  if ((fd = open(filename, O_WRONLY | O_CREAT | O_EXCL, 0666)) >= 0)
    f->fp = fdopen(fd, "wb");

  if (!f->fp || !write_file_version(f, version))
  {
    int errno_save = errno;
    static int walloped = 0;

    if (!walloped++)
      sendto_realops_flags(UMODE_ALL, L_ALL, SEND_NOTICE,
                           "Can't write to %s database %s",
                           service, filename);

    errno = errno_save;
    ilog(LOG_TYPE_IRCD, "Can't write to %s database %s",
         service, filename);

    if (f->fp)
    {
      fclose(f->fp);
      unlink(filename);
    }

    if (f->backupname[0] && rename(f->backupname, filename) < 0)
      ilog(LOG_TYPE_IRCD, "Can't restore backup copy of %s",
           filename);

     MyFree(f);
     errno = errno_save;
     return NULL;
  }

  return f;
}

/*! \brief Open a database file for reading (*mode == 'r') or writing (*mode == 'w').
 * Return the stream pointer, or NULL on error.  When opening for write, it
 * is an error for rename() to return an error (when backing up the original
 * file) other than ENOENT, if NO_BACKUP_OKAY is not defined; it is an error
 * if the version number cannot be written to the file; and it is a fatal
 * error if opening the file for write fails and the backup was successfully
 * made but cannot be restored.
 *
 * \param service If error whom to return the error as
 * \param filename File to open as the database
 * \param mode Mode for writting or reading
 * \param version Database Version
 * \return dbFile struct
 */
struct dbFILE *
open_db(const char *service, const char *filename,
        const char *mode, uint32_t version)
{
  switch (*mode)
  {
    case 'r':
      return open_db_read(service, filename);
      break;
    case 'w':
      return open_db_write(service, filename, version);
      break;
    default:
      errno = EINVAL;
      return NULL;
  }
}

/*! \brief Restore the database file to its condition before open_db().  This is
 * identical to close_db() for files open for reading; however, for files
 * open for writing, we first attempt to restore any backup file before
 * closing files.
 *
 * \param dbFile struct
 */
void
restore_db(struct dbFILE *f)
{
  int errno_save = errno;

  if (f->mode == 'w')
  {
    int ok = 0;             /* Did we manage to restore the old file? */

    errno = errno_save = 0;

    if (f->backupname[0] && strcmp(f->backupname, f->filename))
      if (rename(f->backupname, f->filename) == 0)
        ok = 1;

    if (!ok && f->backupfp)
    {
      char buf[1024];
      size_t i;

      ok = fseek(f->fp, 0, SEEK_SET) == 0;

      while (ok && (i = fread(buf, 1, sizeof(buf), f->backupfp)) > 0)
        if (fwrite(buf, 1, i, f->fp) != i)
          ok = 0;

      if (ok)
      {
        fflush(f->fp);
        ftruncate(fileno(f->fp), ftell(f->fp));
      }
    }

    if (!ok && errno > 0)
      ilog(LOG_TYPE_IRCD, "Unable to restore backup of %s", f->filename);

    errno_save = errno;

    if (f->backupfp)
      fclose(f->backupfp);
    if (f->backupname[0])
      unlink(f->backupname);
  }

  fclose(f->fp);

  if (!errno_save)
    errno_save = errno;

  MyFree(f);
  errno = errno_save;
}

/*! \brief Close a database file.  If the file was opened for write, remove the
 * backup we (may have) created earlier.
 *
 * \param dbFile struct
 */
void
close_db(struct dbFILE *f)
{
  if (f->mode == 'w' && f->backupname[0] &&
      strcmp(f->backupname, f->filename))
  {
    if (f->backupfp)
      fclose(f->backupfp);

    unlink(f->backupname);
  }

  fclose(f->fp);
  MyFree(f);
}

/*
 * Read and write 2-, 4- and 8-byte quantities, pointers, and strings.  All
 * multibyte values are stored in big-endian order (most significant byte
 * first).  A pointer is stored as a byte, either 0 if NULL or 1 if not,
 * and read pointers are returned as either (void *)0 or (void *)1.  A
 * string is stored with a 2-byte unsigned length (including the trailing
 * \0) first; a length of 0 indicates that the string pointer is NULL.
 * Written strings are truncated silently at 4294967294 bytes, and are always
 * null-terminated.
 */

/*! \brief Read a unsigned 8bit integer
 *
 * \param ret 8bit integer to read
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
read_uint8(uint8_t*ret, struct dbFILE *f)
{
  int c = fgetc(f->fp);

  if (c == EOF)
    return -1;

  *ret = c;
  return 0;
}

/*! \brief Write a 8bit integer
 *
 * \param ret 8bit integer to write
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
write_uint8(uint8_t val, struct dbFILE *f)
{
  if (fputc(val, f->fp) == EOF)
    return -1;

  return 0;
}

/*! \brief Read a unsigned 8bit integer
 *
 * \param ret 8bit integer to read
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
read_uint16(uint16_t *ret, struct dbFILE *f)
{
  int c1 = fgetc(f->fp);
  int c2 = fgetc(f->fp);

  if (c1 == EOF || c2 == EOF)
    return -1;

  *ret = c1 << 8 | c2;
  return 0;
}

/*! \brief Write a unsigned 16bit integer
 *
 * \param ret 16bit integer to write
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
write_uint16(uint16_t val, struct dbFILE *f)
{
  if (fputc((val >> 8) & 0xFF, f->fp) == EOF ||
      fputc(val        & 0xFF, f->fp) == EOF)
    return -1;

  return 0;
}

/*! \brief Read a unsigned 32bit integer
 *
 * \param ret unsigned 32bit integer to read
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
read_uint32(uint32_t *ret, struct dbFILE *f)
{
  int c1 = fgetc(f->fp);
  int c2 = fgetc(f->fp);
  int c3 = fgetc(f->fp);
  int c4 = fgetc(f->fp);

  if (c1 == EOF || c2 == EOF || c3 == EOF || c4 == EOF)
    return -1;

  *ret = c1 << 24 | c2 << 16 | c3 << 8 | c4;
  return 0;
}


/*! \brief Write a unsigned 32bit integer
 *
 * \param ret unsigned 32bit integer to write
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
write_uint32(uint32_t val, struct dbFILE *f)
{
  if (fputc((val >> 24) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >> 16) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >>  8) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val)       & 0xFF, f->fp) == EOF)
    return -1;
  return 0;
}

/*! \brief Read a unsigned 64bit integer
 *
 * \param ret unsigned 64bit integer to read
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
read_uint64(uint64_t *ret, struct dbFILE *f)
{
  int64_t c1 = fgetc(f->fp);
  int64_t c2 = fgetc(f->fp);
  int64_t c3 = fgetc(f->fp);
  int64_t c4 = fgetc(f->fp);
  int64_t c5 = fgetc(f->fp);
  int64_t c6 = fgetc(f->fp);
  int64_t c7 = fgetc(f->fp);
  int64_t c8 = fgetc(f->fp);

  if (c1 == EOF || c2 == EOF || c3 == EOF || c4 == EOF ||
      c5 == EOF || c6 == EOF || c7 == EOF || c8 == EOF)
    return -1;

  *ret = c1 << 56 | c2 << 48 | c3 << 40 | c4 << 32 |
         c5 << 24 | c6 << 16 | c7 <<  8 | c8;
  return 0;
}


/*! \brief Write a unsigned 64bit integer
 *
 * \param ret unsigned 64bit integer to write
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
write_uint64(uint64_t val, struct dbFILE *f)
{
  if (fputc((val >> 56) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >> 48) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >> 40) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >> 32) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >> 24) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >> 16) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val >>  8) & 0xFF, f->fp) == EOF)
    return -1;
  if (fputc((val)       & 0xFF, f->fp) == EOF)
    return -1;
  return 0;
}

/*! \brief Read Pointer
 *
 * \param ret pointer to read
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
read_ptr(void **ret, struct dbFILE *f)
{
  int c = fgetc(f->fp);

  if (c == EOF)
    return -1;

  *ret = (c ? (void *)1 : (void *)0);
  return 0;
}


/*! \brief Write Pointer
 *
 * \param ret pointer to write
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
write_ptr(const void *ptr, struct dbFILE *f)
{
  if (fputc(ptr ? 1 : 0, f->fp) == EOF)
    return -1;
  return 0;
}

/*! \brief Read String
 *
 * \param ret string
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
read_string(char **ret, struct dbFILE *f)
{
  char *s = NULL;
  uint32_t len = 0;

  if (read_uint32(&len, f) < 0)
    return -1;

  if (len == 0)
  {
    *ret = NULL;
    return 0;
  }

  s = MyMalloc(len);

  if (len != fread(s, 1, len, f->fp))
  {
    MyFree(s);
    return -1;
  }

  *ret = s;
  return 0;
}

/*! \brief Write String
 *
 * \param ret string
 * \param dbFile struct
 * \return -1 on error, 0 otherwise.
 */
int
write_string(const char *s, struct dbFILE *f)
{
  uint32_t len = 0;

  if (!s)
    return write_uint32(0, f);

  len = strlen(s);

  if (len > 4294967294)
    len = 4294967294;
  if (write_uint32(len + 1, f) < 0)
    return -1;
  if (len > 0 && fwrite(s, 1, len, f->fp) != len)
    return -1;
  if (fputc(0, f->fp) == EOF)
    return -1;

  return 0;
}

#define SAFE_READ(x) do {                             \
    if ((x) < 0) {                                    \
        break;                                        \
    }                                                 \
} while (0)

#define SAFE_WRITE(x,db) do {                         \
    if ((x) < 0) {                                    \
        restore_db(f);                                \
        ilog(LOG_TYPE_IRCD, "Write error on %s", db); \
        return;                                       \
    }                                                 \
} while (0)

void
save_kline_database(void)
{
  uint32_t i = 0;
  uint32_t records = 0;
  struct dbFILE *f = NULL;
  dlink_node *ptr = NULL;

  if (!(f = open_db("kline", KPATH, "w", KLINE_DB_VERSION)))
    return;

  for (i = 0; i < ATABLE_SIZE; ++i)
  {
    DLINK_FOREACH(ptr, atable[i].head)
    {
      struct AddressRec *arec = ptr->data;

      if (arec->type == CONF_KLINE && IsConfDatabase(arec->aconf))
        ++records;
    }
  }

  SAFE_WRITE(write_uint32(records, f), KPATH);

  for (i = 0; i < ATABLE_SIZE; ++i)
  {
    DLINK_FOREACH(ptr, atable[i].head)
    {
      struct AddressRec *arec = ptr->data;

      if (arec->type == CONF_KLINE && IsConfDatabase(arec->aconf))
      {
        SAFE_WRITE(write_string(arec->aconf->user, f), KPATH);
        SAFE_WRITE(write_string(arec->aconf->host, f), KPATH);
        SAFE_WRITE(write_string(arec->aconf->reason, f), KPATH);
        SAFE_WRITE(write_uint64(arec->aconf->setat, f), KPATH);
        SAFE_WRITE(write_uint64(arec->aconf->hold, f), KPATH);
      }
    }
  }

  close_db(f);
}

void
load_kline_database(void)
{
  struct dbFILE *f = NULL;
  struct AccessItem *aconf = NULL;
  char *field_1 = NULL;
  char *field_2 = NULL;
  char *field_3 = NULL;
  uint32_t i = 0;
  uint32_t records = 0;
  uint64_t field_4 = 0;
  uint64_t field_5 = 0;

  if (!(f = open_db("kline", KPATH, "r", KLINE_DB_VERSION)))
    return;

  if (get_file_version(f) < 1)
  {
    close_db(f);
    return;
  }

  read_uint32(&records, f);

  for (i = 0; i < records; ++i)
  {
    SAFE_READ(read_string(&field_1, f));
    SAFE_READ(read_string(&field_2, f));
    SAFE_READ(read_string(&field_3, f));
    SAFE_READ(read_uint64(&field_4, f));
    SAFE_READ(read_uint64(&field_5, f));

    aconf = map_to_conf(make_conf_item(KLINE_TYPE));
    aconf->user = field_1;
    aconf->host = field_2;
    aconf->reason = field_3;
    aconf->setat = field_4;
    aconf->hold = field_5;
    SetConfDatabase(aconf);

    add_conf_by_address(CONF_KLINE, aconf);
  }

  close_db(f);
}

void
save_dline_database(void)
{
  uint32_t i = 0;
  uint32_t records = 0;
  struct dbFILE *f = NULL;
  dlink_node *ptr = NULL;

  if (!(f = open_db("dline", DLPATH, "w", KLINE_DB_VERSION)))
    return;

  for (i = 0; i < ATABLE_SIZE; ++i)
  {
    DLINK_FOREACH(ptr, atable[i].head)
    {
      struct AddressRec *arec = ptr->data;

      if (arec->type == CONF_DLINE && IsConfDatabase(arec->aconf))
        ++records;
    }
  }

  SAFE_WRITE(write_uint32(records, f), DLPATH);

  for (i = 0; i < ATABLE_SIZE; ++i)
  {
    DLINK_FOREACH(ptr, atable[i].head)
    {
      struct AddressRec *arec = ptr->data;

      if (arec->type == CONF_DLINE && IsConfDatabase(arec->aconf))
      {
        SAFE_WRITE(write_string(arec->aconf->host, f), DLPATH);
        SAFE_WRITE(write_string(arec->aconf->reason, f), DLPATH);
        SAFE_WRITE(write_uint64(arec->aconf->setat, f), DLPATH);
        SAFE_WRITE(write_uint64(arec->aconf->hold, f), DLPATH);
      }
    }
  }

  close_db(f);
}

void
load_dline_database(void)
{
  struct dbFILE *f = NULL;
  struct AccessItem *aconf = NULL;
  char *field_1 = NULL;
  char *field_2 = NULL;
  uint32_t i = 0;
  uint32_t records = 0;
  uint64_t field_3 = 0;
  uint64_t field_4 = 0;

  if (!(f = open_db("dline", DLPATH, "r", KLINE_DB_VERSION)))
    return;

  if (get_file_version(f) < 1)
  {
    close_db(f);
    return;
  }

  read_uint32(&records, f);

  for (i = 0; i < records; ++i)
  {
    SAFE_READ(read_string(&field_1, f));
    SAFE_READ(read_string(&field_2, f));
    SAFE_READ(read_uint64(&field_3, f));
    SAFE_READ(read_uint64(&field_4, f));

    aconf = map_to_conf(make_conf_item(DLINE_TYPE));
    aconf->host = field_1;
    aconf->reason = field_2;
    aconf->setat = field_3;
    aconf->hold = field_4;
    SetConfDatabase(aconf);

    add_conf_by_address(CONF_DLINE, aconf);
  }

  close_db(f);
}

void
save_gline_database(void)
{
  uint32_t i = 0;
  uint32_t records = 0, version = 0;
  struct dbFILE *f = NULL;
  dlink_node *ptr = NULL;

  if (!(f = open_db("gline", GPATH, "w", KLINE_DB_VERSION)))
    return;

  for (i = 0; i < ATABLE_SIZE; ++i)
  {
    DLINK_FOREACH(ptr, atable[i].head)
    {
      struct AddressRec *arec = ptr->data;

      if (arec->type == CONF_GLINE && IsConfDatabase(arec->aconf))
        ++records;
    }
  }

  SAFE_WRITE(write_uint32(records, f), GPATH);

  for (i = 0; i < ATABLE_SIZE; ++i)
  {
    DLINK_FOREACH(ptr, atable[i].head)
    {
      struct AddressRec *arec = ptr->data;

      if (arec->type == CONF_GLINE && IsConfDatabase(arec->aconf))
      {
        SAFE_WRITE(write_string(arec->aconf->user, f), GPATH);
        SAFE_WRITE(write_string(arec->aconf->host, f), GPATH);
        SAFE_WRITE(write_string(arec->aconf->reason, f), GPATH);
        SAFE_WRITE(write_uint64(arec->aconf->setat, f), GPATH);
        SAFE_WRITE(write_uint64(arec->aconf->hold, f), GPATH);
      }
    }
  }

  close_db(f);
}

void
load_gline_database(void)
{
  struct dbFILE *f = NULL;
  struct AccessItem *aconf = NULL;
  char *field_1 = NULL;
  char *field_2 = NULL;
  char *field_3 = NULL;
  uint32_t i = 0;
  uint32_t records = 0;
  uint64_t field_4 = 0;
  uint64_t field_5 = 0;

  if (!(f = open_db("gline", GPATH, "r", KLINE_DB_VERSION)))
    return;

  if (get_file_version(f) < 1)
  {
    close_db(f);
    return;
  }

  read_uint32(&records, f);

  for (i = 0; i < records; ++i)
  {
    SAFE_READ(read_string(&field_1, f));
    SAFE_READ(read_string(&field_2, f));
    SAFE_READ(read_string(&field_3, f));
    SAFE_READ(read_uint64(&field_4, f));
    SAFE_READ(read_uint64(&field_5, f));

    aconf = map_to_conf(make_conf_item(KLINE_TYPE));
    aconf->user = field_1;
    aconf->host = field_2;
    aconf->reason = field_3;
    aconf->setat = field_4;
    aconf->hold = field_5;
    SetConfDatabase(aconf);

    add_conf_by_address(CONF_GLINE, aconf);
  }

  close_db(f);
}

void
save_resv_database(void)
{
  uint32_t records = 0;
  struct dbFILE *f = NULL;
  dlink_node *ptr = NULL;
  struct ConfItem *conf;
  struct ResvChannel *resv_cp;
  struct MatchItem *resv_np;

  if (!(f = open_db("resv", RESVPATH, "w", KLINE_DB_VERSION)))
    return;

  DLINK_FOREACH(ptr, resv_channel_list.head)
  {
    resv_cp = ptr->data;

    if (IsConfDatabase(resv_cp))
      ++records;
  }

  DLINK_FOREACH(ptr, nresv_items.head)
  {
    resv_np = map_to_conf(ptr->data);

    if (IsConfDatabase(resv_np))
      ++records;
  }

  SAFE_WRITE(write_uint32(records, f), RESVPATH);

  DLINK_FOREACH(ptr, resv_channel_list.head)
  {
    resv_cp = ptr->data;

    if (!IsConfDatabase(resv_cp))
      continue;

    SAFE_WRITE(write_string(resv_cp->name, f), RESVPATH);
    SAFE_WRITE(write_string(resv_cp->reason, f), RESVPATH);
    SAFE_WRITE(write_uint64(resv_cp->setat, f), RESVPATH);
    SAFE_WRITE(write_uint64(resv_cp->hold, f), RESVPATH);
  }

  DLINK_FOREACH(ptr, nresv_items.head)
  {
    conf = ptr->data;
    resv_np = map_to_conf(conf);

    if (!IsConfDatabase(resv_np))
      continue;

    SAFE_WRITE(write_string(conf->name, f), RESVPATH);
    SAFE_WRITE(write_string(resv_np->reason, f), RESVPATH);
    SAFE_WRITE(write_uint64(resv_np->setat, f), RESVPATH);
    SAFE_WRITE(write_uint64(resv_np->hold, f), RESVPATH);
  }

  close_db(f);
}

void
load_resv_database(void)
{
  uint32_t i = 0;
  uint32_t records = 0;
  uint64_t tmp64_hold = 0, tmp64_setat = 0;
  struct dbFILE *f = NULL;
  char *name = NULL;
  char *reason = NULL;
  struct ConfItem *conf;
  struct ResvChannel *resv_cp;
  struct MatchItem *resv_np;

  if (!(f = open_db("resv", RESVPATH, "r", KLINE_DB_VERSION)))
    return;

  if (get_file_version(f) < 1)
  {
    close_db(f);
    return;
  }

  read_uint32(&records, f);

  for (i = 0; i < records; ++i)
  {
    SAFE_READ(read_string(&name, f));
    SAFE_READ(read_string(&reason, f));
    SAFE_READ(read_uint64(&tmp64_setat, f));
    SAFE_READ(read_uint64(&tmp64_hold, f));

    if (IsChanPrefix(*name))
    {
      if ((conf = create_channel_resv(name, reason, 0)) == NULL)
        continue;

      resv_cp = map_to_conf(conf);
      resv_cp->setat = tmp64_setat;
      resv_cp->hold = tmp64_hold;
      SetConfDatabase(resv_cp);
    }
    else
    {
      if ((conf = create_nick_resv(name, reason, 0)) == NULL)
        continue;

      resv_np = map_to_conf(conf);
      resv_np->setat = tmp64_setat;
      resv_np->hold = tmp64_hold;
      SetConfDatabase(resv_np);
    }

    MyFree(name);
    MyFree(reason);
  }

  close_db(f);
}

void
save_xline_database(void)
{
  uint32_t records = 0;
  struct dbFILE *f = NULL;
  dlink_node *ptr = NULL;
  struct ConfItem *conf = NULL;
  struct MatchItem *xconf = NULL;

  if (!(f = open_db("xline", XPATH, "w", KLINE_DB_VERSION)))
    return;

  DLINK_FOREACH(ptr, xconf_items.head)
  {
    conf = ptr->data;

    if (IsConfDatabase(xconf))
      ++records;
  }

  SAFE_WRITE(write_uint32(records, f), XPATH);

  DLINK_FOREACH(ptr, xconf_items.head)
  {
    conf = ptr->data;
    xconf = map_to_conf(conf);

    if (!IsConfDatabase(xconf))
      continue;

    SAFE_WRITE(write_string(conf->name, f), XPATH);
    SAFE_WRITE(write_string(xconf->reason, f), XPATH);
    SAFE_WRITE(write_uint64(xconf->setat, f), XPATH);
    SAFE_WRITE(write_uint64(xconf->hold, f), XPATH);
  }

  close_db(f);
}

void
load_xline_database(void)
{
  uint32_t i = 0;
  uint32_t records = 0;
  uint64_t tmp64_hold = 0, tmp64_setat = 0;
  struct dbFILE *f = NULL;
  char *name = NULL;
  char *reason = NULL;
  struct ConfItem *conf = NULL;
  struct MatchItem *xconf = NULL;

  if (!(f = open_db("xline", XPATH, "r", KLINE_DB_VERSION)))
    return;

  if (get_file_version(f) < 1)
  {
    close_db(f);
    return;
  }

  read_uint32(&records, f);

  for (i = 0; i < records; ++i)
  {
    SAFE_READ(read_string(&name, f));
    SAFE_READ(read_string(&reason, f));
    SAFE_READ(read_uint64(&tmp64_setat, f));
    SAFE_READ(read_uint64(&tmp64_hold, f));

    conf = make_conf_item(XLINE_TYPE);
    xconf = map_to_conf(conf);

    SetConfDatabase(xconf);

    conf->name = name;
    xconf->reason = reason;
    xconf->setat = tmp64_setat;
    xconf->hold = tmp64_hold;
  }

  close_db(f);
}

void
save_all_databases(void *unused)
{
  save_kline_database();
  save_dline_database();
  save_gline_database();
  save_xline_database();
  save_resv_database();
}
Revision:	1628
Committed:	Thu Nov 1 21:08:56 2012 UTC (12 years, 9 months ago) by michael
Content type:	text/x-csrc
File size:	24326 byte(s)
Log Message:	- flesh out new *line database handling - simplify temporary bans