Return to load_tester_listener.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / strongswan / src / libcharon / plugins / load_tester

Annotation of embedaddon/strongswan/src/libcharon/plugins/load_tester/load_tester_listener.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (C) 2008 Martin Willi
                      3:  * HSR Hochschule fuer Technik Rapperswil
                      4:  *
                      5:  * This program is free software; you can redistribute it and/or modify it
                      6:  * under the terms of the GNU General Public License as published by the
                      7:  * Free Software Foundation; either version 2 of the License, or (at your
                      8:  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
                      9:  *
                     10:  * This program is distributed in the hope that it will be useful, but
                     11:  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
                     12:  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
                     13:  * for more details.
                     14:  */
                     15: 
                     16: #include "load_tester_listener.h"
                     17: 
                     18: #include <signal.h>
                     19: 
                     20: #include <daemon.h>
                     21: #include <processing/jobs/delete_ike_sa_job.h>
                     22: 
                     23: typedef struct private_load_tester_listener_t private_load_tester_listener_t;
                     24: 
                     25: /**
                     26:  * Private data of an load_tester_listener_t object
                     27:  */
                     28: struct private_load_tester_listener_t {
                     29:        /**
                     30:         * Public part
                     31:         */
                     32:        load_tester_listener_t public;
                     33: 
                     34:        /**
                     35:         * Delete IKE_SA after it has been established
                     36:         */
                     37:        bool delete_after_established;
                     38: 
                     39:        /**
                     40:         * Number of established SAs
                     41:         */
                     42:        u_int established;
                     43: 
                     44:        /**
                     45:         * Number of terminated SAs
                     46:         */
                     47:        u_int terminated;
                     48: 
                     49:        /**
                     50:         * Shutdown the daemon if we have established this SA count
                     51:         */
                     52:        u_int shutdown_on;
                     53: 
                     54:        /**
                     55:         * Configuration backend
                     56:         */
                     57:        load_tester_config_t *config;
                     58: };
                     59: 
                     60: METHOD(listener_t, ike_updown, bool,
                     61:        private_load_tester_listener_t *this, ike_sa_t *ike_sa, bool up)
                     62: {
                     63:        if (up)
                     64:        {
                     65:                ike_sa_id_t *id = ike_sa->get_id(ike_sa);
                     66: 
                     67:                this->established++;
                     68: 
                     69:                if (this->delete_after_established)
                     70:                {
                     71:                        lib->processor->queue_job(lib->processor,
                     72:                                                                        (job_t*)delete_ike_sa_job_create(id, TRUE));
                     73:                }
                     74: 
                     75:                if (id->is_initiator(id))
                     76:                {
                     77:                        if (this->shutdown_on == this->established)
                     78:                        {
                     79:                                DBG1(DBG_CFG, "load-test complete, raising SIGTERM");
                     80:                                kill(0, SIGTERM);
                     81:                        }
                     82:                }
                     83:        }
                     84:        else
                     85:        {
                     86:                this->terminated++;
                     87:        }
                     88:        return TRUE;
                     89: }
                     90: 
                     91: METHOD(listener_t, ike_state_change, bool,
                     92:        private_load_tester_listener_t *this, ike_sa_t *ike_sa, ike_sa_state_t state)
                     93: {
                     94:        if (state == IKE_DESTROYING)
                     95:        {
                     96:                this->config->delete_ip(this->config, ike_sa->get_my_host(ike_sa));
                     97:        }
                     98:        return TRUE;
                     99: }
                    100: 
                    101: METHOD(load_tester_listener_t, get_established, u_int,
                    102:        private_load_tester_listener_t *this)
                    103: {
                    104:        return this->established - this->terminated;
                    105: }
                    106: 
                    107: METHOD(load_tester_listener_t, destroy, void,
                    108:        private_load_tester_listener_t *this)
                    109: {
                    110:        free(this);
                    111: }
                    112: 
                    113: load_tester_listener_t *load_tester_listener_create(u_int shutdown_on,
                    114:                                                                                                        load_tester_config_t *config)
                    115: {
                    116:        private_load_tester_listener_t *this;
                    117: 
                    118:        INIT(this,
                    119:                .public = {
                    120:                        .listener = {
                    121:                                .ike_updown = _ike_updown,
                    122:                                .ike_state_change = _ike_state_change,
                    123:                        },
                    124:                        .get_established = _get_established,
                    125:                        .destroy = _destroy,
                    126:                },
                    127:                .delete_after_established = lib->settings->get_bool(lib->settings,
                    128:                                        "%s.plugins.load-tester.delete_after_established", FALSE,
                    129:                                        lib->ns),
                    130:                .shutdown_on = shutdown_on,
                    131:                .config = config,
                    132:        );
                    133: 
                    134:        return &this->public;
                    135: }