Annotation of embedaddon/strongswan/src/charon-nm/nm/nm_handler.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (C) 2016 Tobias Brunner
                      3:  * Copyright (C) 2009 Martin Willi
                      4:  * HSR Hochschule fuer Technik Rapperswil
                      5:  *
                      6:  * This program is free software; you can redistribute it and/or modify it
                      7:  * under the terms of the GNU General Public License as published by the
                      8:  * Free Software Foundation; either version 2 of the License, or (at your
                      9:  * option) any later version.  See <http://www.fsf.org/copyleft/gpl.txt>.
                     10:  *
                     11:  * This program is distributed in the hope that it will be useful, but
                     12:  * WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
                     13:  * or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
                     14:  * for more details.
                     15:  */
                     16: 
                     17: #include "nm_handler.h"
                     18: 
                     19: #include <daemon.h>
                     20: #include <collections/array.h>
                     21: 
                     22: typedef struct private_nm_handler_t private_nm_handler_t;
                     23: 
                     24: /**
                     25:  * Private data of an nm_handler_t object.
                     26:  */
                     27: struct private_nm_handler_t {
                     28: 
                     29:        /**
                     30:         * Public nm_handler_t interface.
                     31:         */
                     32:        nm_handler_t public;
                     33: 
                     34:        /**
                     35:         * Received DNS server attributes, chunk_t
                     36:         */
                     37:        array_t *dns;
                     38: 
                     39:        /**
                     40:         * Received IPv6 DNS server attributes, chunk_t
                     41:         */
                     42:        array_t *dns6;
                     43: 
                     44:        /**
                     45:         * Received NBNS server attributes, chunk_t
                     46:         */
                     47:        array_t *nbns;
                     48: };
                     49: 
                     50: METHOD(attribute_handler_t, handle, bool,
                     51:        private_nm_handler_t *this, ike_sa_t *ike_sa,
                     52:        configuration_attribute_type_t type, chunk_t data)
                     53: {
                     54:        array_t *list;
                     55: 
                     56:        switch (type)
                     57:        {
                     58:                case INTERNAL_IP4_DNS:
                     59:                        list = this->dns;
                     60:                        break;
                     61:                case INTERNAL_IP6_DNS:
                     62:                        list = this->dns6;
                     63:                        break;
                     64:                case INTERNAL_IP4_NBNS:
                     65:                        list = this->nbns;
                     66:                        break;
                     67:                default:
                     68:                        return FALSE;
                     69:        }
                     70:        data = chunk_clone(data);
                     71:        array_insert(list, ARRAY_TAIL, &data);
                     72:        return TRUE;
                     73: }
                     74: 
                     75: METHOD(enumerator_t, enumerate_dns6, bool,
                     76:        enumerator_t *this, va_list args)
                     77: {
                     78:        configuration_attribute_type_t *type;
                     79:        chunk_t *data;
                     80: 
                     81:        VA_ARGS_VGET(args, type, data);
                     82:        *type = INTERNAL_IP6_DNS;
                     83:        *data = chunk_empty;
                     84:        this->venumerate = (void*)return_false;
                     85:        return TRUE;
                     86: }
                     87: 
                     88: METHOD(enumerator_t, enumerate_nbns, bool,
                     89:        enumerator_t *this, va_list args)
                     90: {
                     91:        configuration_attribute_type_t *type;
                     92:        chunk_t *data;
                     93: 
                     94:        VA_ARGS_VGET(args, type, data);
                     95:        *type = INTERNAL_IP4_NBNS;
                     96:        *data = chunk_empty;
                     97:        /* enumerate IPv6 DNS server as next attribute ... */
                     98:        this->venumerate = _enumerate_dns6;
                     99:        return TRUE;
                    100: }
                    101: 
                    102: /**
                    103:  * Implementation of create_attribute_enumerator().enumerate() for DNS
                    104:  */
                    105: METHOD(enumerator_t, enumerate_dns, bool,
                    106:        enumerator_t *this, va_list args)
                    107: {
                    108:        configuration_attribute_type_t *type;
                    109:        chunk_t *data;
                    110: 
                    111:        VA_ARGS_VGET(args, type, data);
                    112:        *type = INTERNAL_IP4_DNS;
                    113:        *data = chunk_empty;
                    114:        /* enumerate WINS server as next attribute ... */
                    115:        this->venumerate = _enumerate_nbns;
                    116:        return TRUE;
                    117: }
                    118: 
                    119: METHOD(attribute_handler_t, create_attribute_enumerator, enumerator_t*,
                    120:        private_nm_handler_t *this, ike_sa_t *ike_sa, linked_list_t *vips)
                    121: {
                    122:        if (vips->get_count(vips))
                    123:        {
                    124:                enumerator_t *enumerator;
                    125: 
                    126:                INIT(enumerator,
                    127:                        /* enumerate DNS attribute first ... */
                    128:                        .enumerate = enumerator_enumerate_default,
                    129:                        .venumerate = _enumerate_dns,
                    130:                        .destroy = (void*)free,
                    131:                );
                    132:                return enumerator;
                    133:        }
                    134:        return enumerator_create_empty();
                    135: }
                    136: 
                    137: METHOD(nm_handler_t, create_enumerator, enumerator_t*,
                    138:        private_nm_handler_t *this, configuration_attribute_type_t type)
                    139: {
                    140:        array_t *list;
                    141: 
                    142:        switch (type)
                    143:        {
                    144:                case INTERNAL_IP4_DNS:
                    145:                        list = this->dns;
                    146:                        break;
                    147:                case INTERNAL_IP6_DNS:
                    148:                        list = this->dns6;
                    149:                        break;
                    150:                case INTERNAL_IP4_NBNS:
                    151:                        list = this->nbns;
                    152:                        break;
                    153:                default:
                    154:                        return enumerator_create_empty();
                    155:        }
                    156:        return array_create_enumerator(list);
                    157: }
                    158: 
                    159: METHOD(nm_handler_t, reset, void,
                    160:        private_nm_handler_t *this)
                    161: {
                    162:        chunk_t chunk;
                    163: 
                    164:        while (array_remove(this->dns, ARRAY_TAIL, &chunk))
                    165:        {
                    166:                chunk_free(&chunk);
                    167:        }
                    168:        while (array_remove(this->dns6, ARRAY_TAIL, &chunk))
                    169:        {
                    170:                chunk_free(&chunk);
                    171:        }
                    172:        while (array_remove(this->nbns, ARRAY_TAIL, &chunk))
                    173:        {
                    174:                chunk_free(&chunk);
                    175:        }
                    176: }
                    177: 
                    178: METHOD(nm_handler_t, destroy, void,
                    179:        private_nm_handler_t *this)
                    180: {
                    181:        reset(this);
                    182:        array_destroy(this->dns);
                    183:        array_destroy(this->dns6);
                    184:        array_destroy(this->nbns);
                    185:        free(this);
                    186: }
                    187: 
                    188: /**
                    189:  * See header
                    190:  */
                    191: nm_handler_t *nm_handler_create()
                    192: {
                    193:        private_nm_handler_t *this;
                    194: 
                    195:        INIT(this,
                    196:                .public = {
                    197:                        .handler = {
                    198:                                .handle = _handle,
                    199:                                .release = nop,
                    200:                                .create_attribute_enumerator = _create_attribute_enumerator,
                    201:                        },
                    202:                        .create_enumerator = _create_enumerator,
                    203:                        .reset = _reset,
                    204:                        .destroy = _destroy,
                    205:                },
                    206:                .dns = array_create(sizeof(chunk_t), 0),
                    207:                .dns6 = array_create(sizeof(chunk_t), 0),
                    208:                .nbns = array_create(sizeof(chunk_t), 0),
                    209:        );
                    210: 
                    211:        return &this->public;
                    212: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>