/* module_init.c - module initialization functions */
/* $OpenLDAP$ */
/* This work is part of OpenLDAP Software <http://www.openldap.org/>.
 *
 * Copyright 1998-2020 The OpenLDAP Foundation.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted only as authorized by the OpenLDAP
 * Public License.
 *
 * A copy of this license is available in the file LICENSE in the
 * top-level directory of the distribution or, alternatively, at
 * <http://www.OpenLDAP.org/license.html>.
 */
/* Portions Copyright (c) 1995 Regents of the University of Michigan.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms are permitted
 * provided that this notice is preserved and that due credit is given
 * to the University of Michigan at Ann Arbor. The name of the University
 * may not be used to endorse or promote products derived from this
 * software without specific prior written permission. This software
 * is provided ``as is'' without express or implied warranty.
 */

#include "portable.h"

#include <stdio.h>

#include <ac/socket.h>
#include <ac/string.h>
#include <ac/time.h>

#include "../servers/slapd/slap.h"

#include "lload.h"
#include "lber_pvt.h"

#include "ldap_rq.h"

ldap_pvt_thread_t lloadd_main_thread;

void *
lload_start_daemon( void *arg )
{
    struct event_base *daemon_base = event_base_new();
    int rc = 0, i;
    if ( !daemon_base ) {
        Debug( LDAP_DEBUG_ANY, "lload_start_daemon: "
                "main event base allocation failed\n" );
        rc = 1;
        return (void *)(uintptr_t)rc;
    }

    rc = lloadd_daemon( daemon_base );
    return (void *)(uintptr_t)rc;
}

/* from init.c */
int
lload_conn_pool_init()
{
    int rc = 0;

    ldap_pvt_thread_mutex_init( &backend_mutex );
    ldap_pvt_thread_mutex_init( &clients_mutex );
    ldap_pvt_thread_mutex_init( &lload_pin_mutex );

    lload_exop_init();
    Debug( LDAP_DEBUG_TRACE, "lload_conn_pool_init: "
            "mutexes initialized.\n" );
    return rc;
}

static int
lload_module_incoming_count( LloadConnection *conn, void *argv )
{
    lload_global_stats_t *tmp_stats = argv;
    tmp_stats->global_incoming++;
    return 0;
}

/* update all global statistics other than rejected and received,
 * these are updated in real time */
void *
lload_module_update_global_stats( void *ctx, void *arg )
{
    struct re_s *rtask = arg;
    lload_global_stats_t tmp_stats = {};
    LloadBackend *b;
    int i;

    Debug( LDAP_DEBUG_TRACE, "lload_module_update_global_stats: "
            "updating stats\n" );
    /* count incoming connections */
    clients_walk( lload_module_incoming_count, &tmp_stats );

    LDAP_CIRCLEQ_FOREACH ( b, &backend, b_next ) {
        LloadConnection *c;

        ldap_pvt_thread_mutex_lock( &b->b_mutex );
        tmp_stats.global_outgoing += b->b_active + b->b_bindavail;

        /* merge completed and failed stats */
        for ( i = 0; i < LLOAD_STATS_OPS_LAST; i++ ) {
            tmp_stats.counters[i].lc_ops_completed +=
                    b->b_counters[i].lc_ops_completed;
            tmp_stats.counters[i].lc_ops_failed +=
                    b->b_counters[i].lc_ops_failed;
        }
        ldap_pvt_thread_mutex_unlock( &b->b_mutex );
    }

    /* update lload_stats */
    lload_stats.global_outgoing = tmp_stats.global_outgoing;
    lload_stats.global_incoming = tmp_stats.global_incoming;
    for ( i = 0; i < LLOAD_STATS_OPS_LAST; i++ ) {
        lload_stats.counters[i].lc_ops_completed =
                tmp_stats.counters[i].lc_ops_completed;
        lload_stats.counters[i].lc_ops_failed =
                tmp_stats.counters[i].lc_ops_failed;
    }

    /* reschedule */
    ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
    ldap_pvt_runqueue_stoptask( &slapd_rq, rtask );
    ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );
    return NULL;
}

void *
lload_module_start_daemon( void *ctx, void *arg )
{
    lload_counters_init();
    lload_monitor_mss_init();

    if ( ldap_pvt_thread_create(
                 &lloadd_main_thread, 0, lload_start_daemon, NULL ) ) {
        return NULL;
    }

    ldap_pvt_thread_mutex_lock( &slapd_rq.rq_mutex );
    ldap_pvt_runqueue_insert( &slapd_rq, 1, lload_module_update_global_stats,
            NULL, "lload_module_update_global_stats", "lloadd" );
    ldap_pvt_thread_mutex_unlock( &slapd_rq.rq_mutex );

    return NULL;
}

int
init_module( int argc, char *argv[] )
{
    if ( argc != 2 ) {
        Debug( LDAP_DEBUG_CONFIG, "lloadd: "
                "incorrect number of arguments to module\n" );
        return -1;
    }

    if ( slapMode & SLAP_TOOL_MODE ) {
        return 0;
    }
    if ( lload_libevent_init() ) {
        return -1;
    }
    global_host = ldap_pvt_get_fqdn( NULL );
#ifdef HAVE_TLS
    if ( ldap_create( &lload_tls_backend_ld ) ) {
        return -1;
    }
#endif /* HAVE_TLS */

    if ( lloadd_daemon_init( argv[1] ) != 0 ) {
        return -1;
    }
    lload_conn_pool_init();

    if ( lload_read_config( argv[0], NULL ) != 0 ) {
        return -1;
    }

    if ( lload_monitor_initialize() != 0 ) {
        return -1;
    }

    return ldap_pvt_thread_pool_submit(
            &connection_pool, lload_module_start_daemon, NULL );
}