capabilities: Some plugins don't actually require capabilities at runtime
[strongswan.git] / src / libcharon / plugins / load_tester / load_tester_plugin.c
index f274022..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;
@@ -148,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);
@@ -171,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);
@@ -202,62 +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,
-                                       "charon.plugins.load-tester.init_limit", 0),
+                                               "%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(),
        );
 
-       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);
-
        if (lib->settings->get_bool(lib->settings,
-                                       "charon.plugins.load-tester.shutdown_when_complete", 0))
-       {
-               shutdown_on = this->iterations * this->initiators;
-       }
-       this->listener = load_tester_listener_create(shutdown_on);
-       charon->bus->add_listener(charon->bus, &this->listener->listener);
-
-       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);
        }
-       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));
-       }
        return &this->public.plugin;
 }