/*****
*
* Copyright (C) 2006 PreludeIDS Technologies. All Rights Reserved.
* Author: Yoann Vandoorselaere <yoann.v@prelude-ids.com>
*
* This file is part of the Prelude-LML program.
*
* 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, 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; see the file COPYING.  If not, write to
* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139, USA.
*
*****/

#include <libprelude/prelude.h>
#include <libprelude/prelude-log.h>
#include <libprelude/prelude-plugin.h>

#include "prelude-correlator.h"
#include "correlation-plugins.h"

#define CORRELATION_PLUGIN_SYMBOL "correlation_plugin_init"


static PRELUDE_LIST(correlation_plugins_instance);



static int subscribe(prelude_plugin_instance_t *pi)
{
        prelude_plugin_generic_t *plugin = prelude_plugin_instance_get_plugin(pi);

        prelude_log_debug(1, "- Subscribing plugin %s[%s]\n",
                          plugin->name, prelude_plugin_instance_get_name(pi));

        prelude_linked_object_add(&correlation_plugins_instance, (prelude_linked_object_t *) pi);

        return 0;
}



static void unsubscribe(prelude_plugin_instance_t *pi)
{
        prelude_plugin_generic_t *plugin = prelude_plugin_instance_get_plugin(pi);

        prelude_log_debug(1, "- Unsubscribing plugin %s[%s]\n",
                          plugin->name, prelude_plugin_instance_get_name(pi));

        prelude_linked_object_del((prelude_linked_object_t *) pi);
}



void correlation_plugins_run(idmef_message_t *idmef)
{
        prelude_list_t *tmp;
        prelude_plugin_instance_t *pi;

        prelude_list_for_each(&correlation_plugins_instance, tmp) {
                pi = prelude_linked_object_get_object(tmp);
                prelude_plugin_run(pi, prelude_correlator_plugin_t, run, pi, idmef);
        }
}


void correlation_plugins_signal(int signo)
{
        prelude_list_t *tmp;
        prelude_plugin_instance_t *pi;

        prelude_list_for_each(&correlation_plugins_instance, tmp) {
                pi = prelude_linked_object_get_object(tmp);

                if ( ((prelude_correlator_plugin_t *) prelude_plugin_instance_get_plugin(pi))->got_signal )
                        prelude_plugin_run(pi, prelude_correlator_plugin_t, got_signal, pi, signo);
        }
}


void correlation_plugins_destroy(void)
{
        prelude_list_t *tmp, *bkp;
        prelude_plugin_instance_t *pi;

        prelude_list_for_each_safe(&correlation_plugins_instance, tmp, bkp) {
                pi = prelude_linked_object_get_object(tmp);
                prelude_plugin_instance_unsubscribe(pi);
        }
}


int correlation_plugins_init(void *data)
{
        int ret;

        ret = prelude_plugin_load_from_dir(NULL, CORRELATION_PLUGIN_DIR,
                                           CORRELATION_PLUGIN_SYMBOL, data, subscribe, unsubscribe);
        if ( ret < 0 ) {
                prelude_perror(ret, "could not load plugin subsystem");
                return -1;
        }

        if ( ret == 0 )
                prelude_log(PRELUDE_LOG_WARN, "* Warning: No correlation plugin loaded from '%s'.\n", CORRELATION_PLUGIN_DIR);

        return ret;
}