capabilities: Some plugins don't actually require capabilities at runtime
[strongswan.git] / src / libcharon / plugins / load_tester / load_tester_plugin.c
index c4a470f..7f2d425 100644 (file)
@@ -18,6 +18,7 @@
 #include "load_tester_creds.h"
 #include "load_tester_ipsec.h"
 #include "load_tester_listener.h"
+#include "load_tester_control.h"
 #include "load_tester_diffie_hellman.h"
 
 #include <unistd.h>
@@ -28,8 +29,6 @@
 #include <threading/condvar.h>
 #include <threading/mutex.h>
 
-static const char *plugin_name = "load_tester";
-
 typedef struct private_load_tester_plugin_t private_load_tester_plugin_t;
 
 /**
@@ -53,6 +52,11 @@ struct private_load_tester_plugin_t {
        load_tester_creds_t *creds;
 
        /**
+        * Unix control socket to initiate load-tests
+        */
+       load_tester_control_t *control;
+
+       /**
         * event handler, listens on bus
         */
        load_tester_listener_t *listener;
@@ -68,7 +72,7 @@ struct private_load_tester_plugin_t {
        int initiators;
 
        /**
-        * currenly running initiators
+        * currently running initiators
         */
        int running;
 
@@ -78,6 +82,11 @@ struct private_load_tester_plugin_t {
        int delay;
 
        /**
+        * Throttle initiation if half-open IKE_SA count reached
+        */
+       int init_limit;
+
+       /**
         * mutex to lock running field
         */
        mutex_t *mutex;
@@ -96,10 +105,7 @@ static job_requeue_t do_load_test(private_load_tester_plugin_t *this)
        int i, s = 0, ms = 0;
 
        this->mutex->lock(this->mutex);
-       if (!this->running)
-       {
-               this->running = this->initiators;
-       }
+       this->running++;
        this->mutex->unlock(this->mutex);
        if (this->delay)
        {
@@ -113,6 +119,23 @@ static job_requeue_t do_load_test(private_load_tester_plugin_t *this)
                child_cfg_t *child_cfg = NULL;
                enumerator_t *enumerator;
 
+               if (this->init_limit)
+               {
+                       while ((charon->ike_sa_manager->get_count(charon->ike_sa_manager) -
+                                               this->listener->get_established(this->listener)) >
+                                       this->init_limit)
+                       {
+                               if (s)
+                               {
+                                       sleep(s);
+                               }
+                               if (ms)
+                               {
+                                       usleep(ms * 1000);
+                               }
+                       }
+               }
+
                peer_cfg = charon->backends->get_peer_cfg_by_name(charon->backends,
                                                                                                                  "load-test");
                if (!peer_cfg)
@@ -129,7 +152,7 @@ static job_requeue_t do_load_test(private_load_tester_plugin_t *this)
 
                charon->controller->initiate(charon->controller,
                                        peer_cfg, child_cfg->get_ref(child_cfg),
-                                       NULL, NULL);
+                                       NULL, NULL, 0);
                if (s)
                {
                        sleep(s);
@@ -141,8 +164,8 @@ static job_requeue_t do_load_test(private_load_tester_plugin_t *this)
        }
        this->mutex->lock(this->mutex);
        this->running--;
-       this->mutex->unlock(this->mutex);
        this->condvar->signal(this->condvar);
+       this->mutex->unlock(this->mutex);
        return JOB_REQUEUE_NONE;
 }
 
@@ -152,26 +175,81 @@ METHOD(plugin_t, get_name, char*,
        return "load-tester";
 }
 
-METHOD(plugin_t, destroy, void,
-       private_load_tester_plugin_t *this)
+/**
+ * Register load_tester plugin features
+ */
+static bool register_load_tester(private_load_tester_plugin_t *this,
+                                                                plugin_feature_t *feature, bool reg, void *data)
 {
-       this->iterations = -1;
-       this->mutex->lock(this->mutex);
-       while (this->running)
+       if (reg)
        {
-               this->condvar->wait(this->condvar, this->mutex);
+               u_int i, shutdown_on = 0;
+
+               this->config = load_tester_config_create();
+               this->creds = load_tester_creds_create();
+               this->control = load_tester_control_create();
+
+               charon->backends->add_backend(charon->backends, &this->config->backend);
+               lib->credmgr->add_set(lib->credmgr, &this->creds->credential_set);
+
+               if (lib->settings->get_bool(lib->settings,
+                               "%s.plugins.load-tester.shutdown_when_complete", 0, charon->name))
+               {
+                       shutdown_on = this->iterations * this->initiators;
+               }
+               this->listener = load_tester_listener_create(shutdown_on, this->config);
+               charon->bus->add_listener(charon->bus, &this->listener->listener);
+
+               for (i = 0; i < this->initiators; i++)
+               {
+                       lib->processor->queue_job(lib->processor, (job_t*)
+                               callback_job_create_with_prio((callback_job_cb_t)do_load_test,
+                                                                               this, NULL, NULL, JOB_PRIO_CRITICAL));
+               }
        }
-       this->mutex->unlock(this->mutex);
+       else
+       {
+               this->iterations = -1;
+               this->mutex->lock(this->mutex);
+               while (this->running)
+               {
+                       this->condvar->wait(this->condvar, this->mutex);
+               }
+               this->mutex->unlock(this->mutex);
+               charon->backends->remove_backend(charon->backends, &this->config->backend);
+               lib->credmgr->remove_set(lib->credmgr, &this->creds->credential_set);
+               charon->bus->remove_listener(charon->bus, &this->listener->listener);
+               this->config->destroy(this->config);
+               this->creds->destroy(this->creds);
+               this->listener->destroy(this->listener);
+               this->control->destroy(this->control);
+       }
+       return TRUE;
+}
+
+METHOD(plugin_t, get_features, int,
+       private_load_tester_plugin_t *this, plugin_feature_t *features[])
+{
+       static plugin_feature_t f[] = {
+               PLUGIN_REGISTER(DH, load_tester_diffie_hellman_create),
+                       PLUGIN_PROVIDE(DH, MODP_NULL),
+                               PLUGIN_DEPENDS(CUSTOM, "load-tester"),
+               PLUGIN_CALLBACK((plugin_feature_callback_t)register_load_tester, NULL),
+                       PLUGIN_PROVIDE(CUSTOM, "load-tester"),
+                               PLUGIN_DEPENDS(CUSTOM, "kernel-net"),
+                               PLUGIN_SDEPEND(PRIVKEY, KEY_RSA),
+                               PLUGIN_SDEPEND(CERT_DECODE, CERT_ANY),
+                               PLUGIN_SDEPEND(CERT_DECODE, CERT_X509),
+       };
+       *features = f;
+       return countof(f);
+}
+
+METHOD(plugin_t, destroy, void,
+       private_load_tester_plugin_t *this)
+{
        hydra->kernel_interface->remove_ipsec_interface(hydra->kernel_interface,
                                                (kernel_ipsec_constructor_t)load_tester_ipsec_create);
-       charon->backends->remove_backend(charon->backends, &this->config->backend);
-       lib->credmgr->remove_set(lib->credmgr, &this->creds->credential_set);
-       charon->bus->remove_listener(charon->bus, &this->listener->listener);
-       this->config->destroy(this->config);
-       this->creds->destroy(this->creds);
-       this->listener->destroy(this->listener);
-       lib->crypto->remove_dh(lib->crypto,
-                                               (dh_constructor_t)load_tester_diffie_hellman_create);
        this->mutex->destroy(this->mutex);
        this->condvar->destroy(this->condvar);
        free(this);
@@ -183,61 +261,47 @@ METHOD(plugin_t, destroy, void,
 plugin_t *load_tester_plugin_create()
 {
        private_load_tester_plugin_t *this;
-       u_int i, shutdown_on = 0;
 
        if (!lib->settings->get_bool(lib->settings,
-                                                                "charon.plugins.load-tester.enable", FALSE))
+                                               "%s.plugins.load-tester.enable", FALSE, charon->name))
        {
                DBG1(DBG_CFG, "disabling load-tester plugin, not configured");
                return NULL;
        }
 
+       if (!lib->caps->check(lib->caps, CAP_CHOWN))
+       {       /* required to chown(2) control socket */
+               DBG1(DBG_CFG, "load-tester plugin requires CAP_CHOWN capability");
+               return NULL;
+       }
+
        INIT(this,
                .public = {
                        .plugin = {
                                .get_name = _get_name,
+                               .get_features = _get_features,
+                               .reload = (void*)return_false,
                                .destroy = _destroy,
                        },
                },
                .delay = lib->settings->get_int(lib->settings,
-                                       "charon.plugins.load-tester.delay", 0),
+                                               "%s.plugins.load-tester.delay", 0, charon->name),
                .iterations = lib->settings->get_int(lib->settings,
-                                       "charon.plugins.load-tester.iterations", 1),
+                                               "%s.plugins.load-tester.iterations", 1, charon->name),
                .initiators = lib->settings->get_int(lib->settings,
-                                       "charon.plugins.load-tester.initiators", 0),
+                                               "%s.plugins.load-tester.initiators", 0, charon->name),
+               .init_limit = lib->settings->get_int(lib->settings,
+                                               "%s.plugins.load-tester.init_limit", 0, charon->name),
                .mutex = mutex_create(MUTEX_TYPE_DEFAULT),
                .condvar = condvar_create(CONDVAR_TYPE_DEFAULT),
-               .config = load_tester_config_create(),
-               .creds = load_tester_creds_create(),
-               .listener = load_tester_listener_create(shutdown_on),
        );
 
-       lib->crypto->add_dh(lib->crypto, MODP_NULL, plugin_name,
-                                               (dh_constructor_t)load_tester_diffie_hellman_create);
-       charon->backends->add_backend(charon->backends, &this->config->backend);
-       lib->credmgr->add_set(lib->credmgr, &this->creds->credential_set);
-       charon->bus->add_listener(charon->bus, &this->listener->listener);
-
-       if (lib->settings->get_bool(lib->settings,
-                                       "charon.plugins.load-tester.shutdown_when_complete", 0))
-       {
-               shutdown_on = this->iterations * this->initiators;
-       }
-
-
        if (lib->settings->get_bool(lib->settings,
-                                       "charon.plugins.load-tester.fake_kernel", FALSE))
+                       "%s.plugins.load-tester.fake_kernel", FALSE, charon->name))
        {
                hydra->kernel_interface->add_ipsec_interface(hydra->kernel_interface,
                                                (kernel_ipsec_constructor_t)load_tester_ipsec_create);
        }
-       this->running = 0;
-       for (i = 0; i < this->initiators; i++)
-       {
-               lib->processor->queue_job(lib->processor,
-                                       (job_t*)callback_job_create((callback_job_cb_t)do_load_test,
-                                                                                               this, NULL, NULL));
-       }
        return &this->public.plugin;
 }