Return to connmark_plugin.c CVS log

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

Annotation of embedaddon/strongswan/src/libcharon/plugins/connmark/connmark_plugin.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (C) 2014 Martin Willi
                      3:  * Copyright (C) 2014 revosec AG
                      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 "connmark_plugin.h"
                     17: #include "connmark_listener.h"
                     18: 
                     19: #include <daemon.h>
                     20: 
                     21: typedef struct private_connmark_plugin_t private_connmark_plugin_t;
                     22: 
                     23: /**
                     24:  * private data of connmark plugin
                     25:  */
                     26: struct private_connmark_plugin_t {
                     27: 
                     28:        /**
                     29:         * implements plugin interface
                     30:         */
                     31:        connmark_plugin_t public;
                     32: 
                     33:        /**
                     34:         * Listener installing netfilter rules
                     35:         */
                     36:        connmark_listener_t *listener;
                     37: };
                     38: 
                     39: METHOD(plugin_t, get_name, char*,
                     40:        private_connmark_plugin_t *this)
                     41: {
                     42:        return "connmark";
                     43: }
                     44: 
                     45: /**
                     46:  * Register listener
                     47:  */
                     48: static bool plugin_cb(private_connmark_plugin_t *this,
                     49:                                          plugin_feature_t *feature, bool reg, void *cb_data)
                     50: {
                     51:        if (reg)
                     52:        {
                     53:                charon->bus->add_listener(charon->bus, &this->listener->listener);
                     54:        }
                     55:        else
                     56:        {
                     57:                charon->bus->remove_listener(charon->bus, &this->listener->listener);
                     58:        }
                     59:        return TRUE;
                     60: }
                     61: 
                     62: METHOD(plugin_t, get_features, int,
                     63:        private_connmark_plugin_t *this, plugin_feature_t *features[])
                     64: {
                     65:        static plugin_feature_t f[] = {
                     66:                PLUGIN_CALLBACK((plugin_feature_callback_t)plugin_cb, NULL),
                     67:                        PLUGIN_PROVIDE(CUSTOM, "connmark"),
                     68:        };
                     69:        *features = f;
                     70:        return countof(f);
                     71: }
                     72: 
                     73: METHOD(plugin_t, destroy, void,
                     74:        private_connmark_plugin_t *this)
                     75: {
                     76:        this->listener->destroy(this->listener);
                     77:        free(this);
                     78: }
                     79: 
                     80: /**
                     81:  * Plugin constructor
                     82:  */
                     83: plugin_t *connmark_plugin_create()
                     84: {
                     85:        private_connmark_plugin_t *this;
                     86: 
                     87:        if (!lib->caps->keep(lib->caps, CAP_NET_ADMIN))
                     88:        {
                     89:                DBG1(DBG_NET, "connmark plugin requires CAP_NET_ADMIN capability");
                     90:                return NULL;
                     91:        }
                     92: 
                     93:        if (!lib->caps->keep(lib->caps, CAP_NET_RAW))
                     94:        {
                     95:                DBG1(DBG_NET, "connmark plugin requires CAP_NET_RAW capability");
                     96:                return NULL;
                     97:        }
                     98: 
                     99:        INIT(this,
                    100:                .public = {
                    101:                        .plugin = {
                    102:                                .get_name = _get_name,
                    103:                                .get_features = _get_features,
                    104:                                .destroy = _destroy,
                    105:                        },
                    106:                },
                    107:                .listener = connmark_listener_create(),
                    108:        );
                    109: 
                    110:        return &this->public.plugin;
                    111: }