Annotation of embedaddon/strongswan/src/libcharon/plugins/load_tester/load_tester_plugin.c, revision 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_plugin.h"
! 17: #include "load_tester_config.h"
! 18: #include "load_tester_creds.h"
! 19: #include "load_tester_ipsec.h"
! 20: #include "load_tester_listener.h"
! 21: #include "load_tester_control.h"
! 22: #include "load_tester_diffie_hellman.h"
! 23:
! 24: #include <unistd.h>
! 25:
! 26: #include <daemon.h>
! 27: #include <processing/jobs/callback_job.h>
! 28: #include <threading/condvar.h>
! 29: #include <threading/mutex.h>
! 30:
! 31: typedef struct private_load_tester_plugin_t private_load_tester_plugin_t;
! 32:
! 33: /**
! 34: * private data of load_tester plugin
! 35: */
! 36: struct private_load_tester_plugin_t {
! 37:
! 38: /**
! 39: * implements plugin interface
! 40: */
! 41: load_tester_plugin_t public;
! 42:
! 43: /**
! 44: * load_tester configuration backend
! 45: */
! 46: load_tester_config_t *config;
! 47:
! 48: /**
! 49: * load_tester credential set implementation
! 50: */
! 51: load_tester_creds_t *creds;
! 52:
! 53: /**
! 54: * Unix control socket to initiate load-tests
! 55: */
! 56: load_tester_control_t *control;
! 57:
! 58: /**
! 59: * event handler, listens on bus
! 60: */
! 61: load_tester_listener_t *listener;
! 62:
! 63: /**
! 64: * number of iterations per thread
! 65: */
! 66: int iterations;
! 67:
! 68: /**
! 69: * number desired initiator threads
! 70: */
! 71: int initiators;
! 72:
! 73: /**
! 74: * currently running initiators
! 75: */
! 76: int running;
! 77:
! 78: /**
! 79: * delay between initiations, in ms
! 80: */
! 81: int delay;
! 82:
! 83: /**
! 84: * Throttle initiation if half-open IKE_SA count reached
! 85: */
! 86: int init_limit;
! 87:
! 88: /**
! 89: * mutex to lock running field
! 90: */
! 91: mutex_t *mutex;
! 92:
! 93: /**
! 94: * condvar to wait for initiators
! 95: */
! 96: condvar_t *condvar;
! 97: };
! 98:
! 99: /**
! 100: * Begin the load test
! 101: */
! 102: static job_requeue_t do_load_test(private_load_tester_plugin_t *this)
! 103: {
! 104: int i, s = 0, ms = 0;
! 105:
! 106: this->mutex->lock(this->mutex);
! 107: this->running++;
! 108: this->mutex->unlock(this->mutex);
! 109: if (this->delay)
! 110: {
! 111: s = this->delay / 1000;
! 112: ms = this->delay % 1000;
! 113: }
! 114:
! 115: for (i = 0; this->iterations == 0 || i < this->iterations; i++)
! 116: {
! 117: peer_cfg_t *peer_cfg;
! 118: child_cfg_t *child_cfg = NULL;
! 119: enumerator_t *enumerator;
! 120:
! 121: if (this->init_limit)
! 122: {
! 123: while ((charon->ike_sa_manager->get_count(charon->ike_sa_manager) -
! 124: this->listener->get_established(this->listener)) >
! 125: this->init_limit)
! 126: {
! 127: if (s)
! 128: {
! 129: sleep(s);
! 130: }
! 131: if (ms)
! 132: {
! 133: usleep(ms * 1000);
! 134: }
! 135: }
! 136: }
! 137:
! 138: peer_cfg = charon->backends->get_peer_cfg_by_name(charon->backends,
! 139: "load-test");
! 140: if (!peer_cfg)
! 141: {
! 142: break;
! 143: }
! 144: enumerator = peer_cfg->create_child_cfg_enumerator(peer_cfg);
! 145: if (!enumerator->enumerate(enumerator, &child_cfg))
! 146: {
! 147: enumerator->destroy(enumerator);
! 148: break;
! 149: }
! 150: enumerator->destroy(enumerator);
! 151:
! 152: charon->controller->initiate(charon->controller,
! 153: peer_cfg, child_cfg->get_ref(child_cfg),
! 154: NULL, NULL, 0, FALSE);
! 155: if (s)
! 156: {
! 157: sleep(s);
! 158: }
! 159: if (ms)
! 160: {
! 161: usleep(ms * 1000);
! 162: }
! 163: }
! 164: this->mutex->lock(this->mutex);
! 165: this->running--;
! 166: this->condvar->signal(this->condvar);
! 167: this->mutex->unlock(this->mutex);
! 168: return JOB_REQUEUE_NONE;
! 169: }
! 170:
! 171: METHOD(plugin_t, get_name, char*,
! 172: private_load_tester_plugin_t *this)
! 173: {
! 174: return "load-tester";
! 175: }
! 176:
! 177: /**
! 178: * Register load_tester plugin features
! 179: */
! 180: static bool register_load_tester(private_load_tester_plugin_t *this,
! 181: plugin_feature_t *feature, bool reg, void *data)
! 182: {
! 183: if (reg)
! 184: {
! 185: u_int i, shutdown_on = 0;
! 186:
! 187: this->config = load_tester_config_create();
! 188: this->creds = load_tester_creds_create();
! 189: this->control = load_tester_control_create();
! 190:
! 191: charon->backends->add_backend(charon->backends, &this->config->backend);
! 192: lib->credmgr->add_set(lib->credmgr, &this->creds->credential_set);
! 193:
! 194: if (lib->settings->get_bool(lib->settings,
! 195: "%s.plugins.load-tester.shutdown_when_complete", 0, lib->ns))
! 196: {
! 197: shutdown_on = this->iterations * this->initiators;
! 198: }
! 199: this->listener = load_tester_listener_create(shutdown_on, this->config);
! 200: charon->bus->add_listener(charon->bus, &this->listener->listener);
! 201:
! 202: for (i = 0; i < this->initiators; i++)
! 203: {
! 204: lib->processor->queue_job(lib->processor, (job_t*)
! 205: callback_job_create_with_prio((callback_job_cb_t)do_load_test,
! 206: this, NULL, NULL, JOB_PRIO_CRITICAL));
! 207: }
! 208: }
! 209: else
! 210: {
! 211: this->iterations = -1;
! 212: this->mutex->lock(this->mutex);
! 213: while (this->running)
! 214: {
! 215: this->condvar->wait(this->condvar, this->mutex);
! 216: }
! 217: this->mutex->unlock(this->mutex);
! 218: charon->backends->remove_backend(charon->backends, &this->config->backend);
! 219: lib->credmgr->remove_set(lib->credmgr, &this->creds->credential_set);
! 220: charon->bus->remove_listener(charon->bus, &this->listener->listener);
! 221: this->config->destroy(this->config);
! 222: this->creds->destroy(this->creds);
! 223: this->listener->destroy(this->listener);
! 224: this->control->destroy(this->control);
! 225: }
! 226: return TRUE;
! 227: }
! 228:
! 229: METHOD(plugin_t, get_features, int,
! 230: private_load_tester_plugin_t *this, plugin_feature_t *features[])
! 231: {
! 232: static plugin_feature_t f[] = {
! 233: PLUGIN_REGISTER(DH, load_tester_diffie_hellman_create),
! 234: PLUGIN_PROVIDE(DH, MODP_NULL),
! 235: PLUGIN_DEPENDS(CUSTOM, "load-tester"),
! 236: PLUGIN_CALLBACK((plugin_feature_callback_t)register_load_tester, NULL),
! 237: PLUGIN_PROVIDE(CUSTOM, "load-tester"),
! 238: PLUGIN_DEPENDS(CUSTOM, "kernel-net"),
! 239: PLUGIN_SDEPEND(PRIVKEY, KEY_RSA),
! 240: PLUGIN_SDEPEND(CERT_DECODE, CERT_ANY),
! 241: PLUGIN_SDEPEND(CERT_DECODE, CERT_X509),
! 242: PLUGIN_CALLBACK(kernel_ipsec_register, load_tester_ipsec_create),
! 243: PLUGIN_PROVIDE(CUSTOM, "kernel-ipsec"),
! 244: };
! 245: int count = countof(f);
! 246:
! 247: *features = f;
! 248:
! 249: if (!lib->settings->get_bool(lib->settings,
! 250: "%s.plugins.load-tester.fake_kernel", FALSE, lib->ns))
! 251: {
! 252: count -= 2;
! 253: }
! 254: return count;
! 255: }
! 256:
! 257: METHOD(plugin_t, destroy, void,
! 258: private_load_tester_plugin_t *this)
! 259: {
! 260: this->mutex->destroy(this->mutex);
! 261: this->condvar->destroy(this->condvar);
! 262: free(this);
! 263: }
! 264:
! 265: /*
! 266: * see header file
! 267: */
! 268: plugin_t *load_tester_plugin_create()
! 269: {
! 270: private_load_tester_plugin_t *this;
! 271:
! 272: if (!lib->settings->get_bool(lib->settings, "%s.plugins.load-tester.enable",
! 273: FALSE, lib->ns))
! 274: {
! 275: DBG1(DBG_CFG, "disabling load-tester plugin, not configured");
! 276: return NULL;
! 277: }
! 278:
! 279: INIT(this,
! 280: .public = {
! 281: .plugin = {
! 282: .get_name = _get_name,
! 283: .get_features = _get_features,
! 284: .reload = (void*)return_false,
! 285: .destroy = _destroy,
! 286: },
! 287: },
! 288: .delay = lib->settings->get_int(lib->settings,
! 289: "%s.plugins.load-tester.delay", 0, lib->ns),
! 290: .iterations = lib->settings->get_int(lib->settings,
! 291: "%s.plugins.load-tester.iterations", 1, lib->ns),
! 292: .initiators = lib->settings->get_int(lib->settings,
! 293: "%s.plugins.load-tester.initiators", 0, lib->ns),
! 294: .init_limit = lib->settings->get_int(lib->settings,
! 295: "%s.plugins.load-tester.init_limit", 0, lib->ns),
! 296: .mutex = mutex_create(MUTEX_TYPE_DEFAULT),
! 297: .condvar = condvar_create(CONDVAR_TYPE_DEFAULT),
! 298: );
! 299: return &this->public.plugin;
! 300: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>