/*
    connection.c -- connection list management
    Copyright (C) 2000 Guus Sliepen <guus@sliepen.warande.net>,
                  2000 Ivo Timmermans <itimmermans@bigfoot.com>

    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., 675 Mass Ave, Cambridge, MA 02139, USA.

    $Id: connection.c,v 1.1.2.8 2001/01/07 15:25:40 guus Exp $
*/

#include "config.h"

#include <stdio.h>
#include <syslog.h>

#include <avl_tree.h>
#include <list.h>

#include "net.h"	/* Don't ask. */
#include "netutl.h"
#include "config.h"
#include "conf.h"
#include <utils.h>
#include "subnet.h"

#include "xalloc.h"
#include "system.h"

/* Root of the connection list */

avl_tree_t *connection_tree;
avl_tree_t *id_tree;

/* Pointer to connection describing myself */

connection_t *myself = NULL;

/* Initialization and callbacks */

int connection_compare(connection_t *a, connection_t *b)
{
  ipv4_t result;

  result = a->address - b->address;
  if(result)
    return result;
  else
    return a->port - b->port;
}

int id_compare(connection_t *a, connection_t *b)
{
  return strcmp(a->name, b->name);
}

void init_connections(void)
{
  connection_tree = avl_alloc_tree((avl_compare_t)connection_compare, (avl_action_t)free_connection);
  id_tree = avl_alloc_tree((avl_compare_t)id_compare, NULL);
}

/* Creation and deletion of connection elements */

connection_t *new_connection(void)
{
  connection_t *p = (connection_t *)xmalloc_and_zero(sizeof(*p));
cp
  p->subnet_tree = avl_alloc_tree((avl_compare_t)subnet_compare, NULL);
  p->queue = list_alloc((list_action_t)free);
cp
  return p;
}

void free_connection(connection_t *p)
{
cp
  if(p->queue)
    list_delete_list(p->queue);
  if(p->name && p->name!=unknown)
    free(p->name);
  if(p->hostname)
    free(p->hostname);
  if(p->rsa_key)
    RSA_free(p->rsa_key);
  if(p->cipher_pktkey)
    free(p->cipher_pktkey);
  if(p->buffer)
    free(p->buffer);
  if(p->config)
    clear_config(&p->config);
  free(p);
cp
}

/*
  remove all marked connections
*/
void prune_connection_tree(void)
{
  avl_node_t *node, *next;
  connection_t *cl;
cp
  for(node = connection_tree->head; node; node = next)
    {
      next = node->next;
      cl = (connection_t *)node->data;
      if(cl->status.remove)
        connection_del(cl);
    }
cp
}

/*
  free all elements of connection
*/
void destroy_connection_tree(void)
{
cp
  avl_delete_tree(id_tree);
  avl_delete_tree(connection_tree);
cp
}

/* Linked list management */

void connection_add(connection_t *cl)
{
cp
  avl_insert(connection_tree, cl);
cp
}

void id_add(connection_t *cl)
{
cp
  avl_insert(id_tree, cl);
cp
}

void connection_del(connection_t *cl)
{
cp
  avl_delete(id_tree, cl);
  avl_delete(connection_tree, cl);
cp
}

/* Lookup functions */

connection_t *lookup_connection(ipv4_t address, short unsigned int port)
{
  connection_t cl;
cp
  cl.address = address;
  cl.port = port;

  return avl_search(connection_tree, &cl);
}

connection_t *lookup_id(char *name)
{
  connection_t cl, *p;
cp
  cl.name = name;
  p = avl_search(id_tree, &cl);
  if(p && p->status.active)
    return p;
  else
    return NULL;
}

/* Debugging */

void dump_connection_list(void)
{
  avl_node_t *node;
  connection_t *cl;
cp
  syslog(LOG_DEBUG, _("Connection list:"));

  syslog(LOG_DEBUG, _(" %s at %s port %hd flags %d sockets %d, %d status %04x"),
         myself->name, myself->hostname, myself->port, myself->flags,
         myself->socket, myself->meta_socket, myself->status);

  for(node = connection_tree->head; node; node = node->next)
    {
      cl = (connection_t *)node->data;
      syslog(LOG_DEBUG, _(" %s at %s port %hd flags %d sockets %d, %d status %04x"),
             cl->name, cl->hostname, cl->port, cl->flags,
             cl->socket, cl->meta_socket, cl->status);
    }
    
  syslog(LOG_DEBUG, _("End of connection list."));
cp
}

int read_host_config(connection_t *cl)
{
  char *fname;
  int x;
cp
  asprintf(&fname, "%s/hosts/%s", confbase, cl->name);
  x = read_config_file(&cl->config, fname);
  free(fname);
cp
  return x;
}