Annotation of embedaddon/strongswan/src/libimcv/ietf/ietf_attr_assess_result.c, revision 1.1.1.1

1.1       misho       1: /*
                      2:  * Copyright (C) 2012-2014 Andreas Steffen
                      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 "ietf_attr_assess_result.h"
                     17: 
                     18: #include <pa_tnc/pa_tnc_msg.h>
                     19: #include <bio/bio_writer.h>
                     20: #include <bio/bio_reader.h>
                     21: #include <utils/debug.h>
                     22: 
                     23: typedef struct private_ietf_attr_assess_result_t private_ietf_attr_assess_result_t;
                     24: 
                     25: /**
                     26:  * PA-TNC Product Information type  (see section 4.2.2 of RFC 5792)
                     27:  *
                     28:  *                       1                   2                   3
                     29:  *   0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
                     30:  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
                     31:  * |                       Assessment Result                       |
                     32:  * +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
                     33:  */
                     34: 
                     35: #define ASSESS_RESULT_SIZE     4
                     36: 
                     37: /**
                     38:  * Private data of an ietf_attr_assess_result_t object.
                     39:  */
                     40: struct private_ietf_attr_assess_result_t {
                     41: 
                     42:        /**
                     43:         * Public members of ietf_attr_assess_result_t
                     44:         */
                     45:        ietf_attr_assess_result_t public;
                     46: 
                     47:        /**
                     48:         * Vendor-specific attribute type
                     49:         */
                     50:        pen_type_t type;
                     51: 
                     52:        /**
                     53:         * Length of attribute value
                     54:         */
                     55:        size_t length;
                     56: 
                     57:        /**
                     58:         * Attribute value or segment
                     59:         */
                     60:        chunk_t value;
                     61: 
                     62:        /**
                     63:         * Noskip flag
                     64:         */
                     65:        bool noskip_flag;
                     66: 
                     67:        /**
                     68:         * Assessment Result
                     69:         */
                     70:        uint32_t result;
                     71: 
                     72:        /**
                     73:         * Reference count
                     74:         */
                     75:        refcount_t ref;
                     76: };
                     77: 
                     78: METHOD(pa_tnc_attr_t, get_type, pen_type_t,
                     79:        private_ietf_attr_assess_result_t *this)
                     80: {
                     81:        return this->type;
                     82: }
                     83: 
                     84: METHOD(pa_tnc_attr_t, get_value, chunk_t,
                     85:        private_ietf_attr_assess_result_t *this)
                     86: {
                     87:        return this->value;
                     88: }
                     89: 
                     90: METHOD(pa_tnc_attr_t, get_noskip_flag, bool,
                     91:        private_ietf_attr_assess_result_t *this)
                     92: {
                     93:        return this->noskip_flag;
                     94: }
                     95: 
                     96: METHOD(pa_tnc_attr_t, set_noskip_flag,void,
                     97:        private_ietf_attr_assess_result_t *this, bool noskip)
                     98: {
                     99:        this->noskip_flag = noskip;
                    100: }
                    101: 
                    102: METHOD(pa_tnc_attr_t, build, void,
                    103:        private_ietf_attr_assess_result_t *this)
                    104: {
                    105:        bio_writer_t *writer;
                    106: 
                    107:        if (this->value.ptr)
                    108:        {
                    109:                return;
                    110:        }
                    111: 
                    112:        writer = bio_writer_create(ASSESS_RESULT_SIZE);
                    113:        writer->write_uint32(writer, this->result);
                    114:        this->value = writer->extract_buf(writer);
                    115:        this->length = this->value.len;
                    116:        writer->destroy(writer);
                    117: }
                    118: 
                    119: METHOD(pa_tnc_attr_t, process, status_t,
                    120:        private_ietf_attr_assess_result_t *this, uint32_t *offset)
                    121: {
                    122:        bio_reader_t *reader;
                    123: 
                    124:        *offset = 0;
                    125: 
                    126:        if (this->value.len < this->length)
                    127:        {
                    128:                return NEED_MORE;
                    129:        }
                    130:        if (this->value.len < ASSESS_RESULT_SIZE)
                    131:        {
                    132:                DBG1(DBG_TNC, "insufficient data for IETF assessment result");
                    133:                return FAILED;
                    134:        }
                    135:        reader = bio_reader_create(this->value);
                    136:        reader->read_uint32(reader, &this->result);
                    137:        reader->destroy(reader);
                    138: 
                    139:        return SUCCESS;
                    140: }
                    141: 
                    142: METHOD(pa_tnc_attr_t, add_segment, void,
                    143:        private_ietf_attr_assess_result_t *this, chunk_t segment)
                    144: {
                    145:        this->value = chunk_cat("mc", this->value, segment);
                    146: }
                    147: 
                    148: METHOD(pa_tnc_attr_t, get_ref, pa_tnc_attr_t*,
                    149:        private_ietf_attr_assess_result_t *this)
                    150: {
                    151:        ref_get(&this->ref);
                    152:        return &this->public.pa_tnc_attribute;
                    153: }
                    154: 
                    155: METHOD(pa_tnc_attr_t, destroy, void,
                    156:        private_ietf_attr_assess_result_t *this)
                    157: {
                    158:        if (ref_put(&this->ref))
                    159:        {
                    160:                free(this->value.ptr);
                    161:                free(this);
                    162:        }
                    163: }
                    164: 
                    165: METHOD(ietf_attr_assess_result_t, get_result, uint32_t,
                    166:        private_ietf_attr_assess_result_t *this)
                    167: {
                    168:        return this->result;
                    169: }
                    170: 
                    171: /**
                    172:  * Described in header.
                    173:  */
                    174: pa_tnc_attr_t *ietf_attr_assess_result_create(uint32_t result)
                    175: {
                    176:        private_ietf_attr_assess_result_t *this;
                    177: 
                    178:        INIT(this,
                    179:                .public = {
                    180:                        .pa_tnc_attribute = {
                    181:                                .get_type = _get_type,
                    182:                                .get_value = _get_value,
                    183:                                .get_noskip_flag = _get_noskip_flag,
                    184:                                .set_noskip_flag = _set_noskip_flag,
                    185:                                .build = _build,
                    186:                                .process = _process,
                    187:                                .add_segment = _add_segment,
                    188:                                .get_ref = _get_ref,
                    189:                                .destroy = _destroy,
                    190:                        },
                    191:                        .get_result = _get_result,
                    192:                },
                    193:                .type = { PEN_IETF, IETF_ATTR_ASSESSMENT_RESULT },
                    194:                .result = result,
                    195:                .ref = 1,
                    196:        );
                    197: 
                    198:        return &this->public.pa_tnc_attribute;
                    199: }
                    200: 
                    201: /**
                    202:  * Described in header.
                    203:  */
                    204: pa_tnc_attr_t *ietf_attr_assess_result_create_from_data(size_t length,
                    205:                                                                                                                chunk_t data)
                    206: {
                    207:        private_ietf_attr_assess_result_t *this;
                    208: 
                    209:        INIT(this,
                    210:                .public = {
                    211:                        .pa_tnc_attribute = {
                    212:                                .get_type = _get_type,
                    213:                                .get_value = _get_value,
                    214:                                .get_noskip_flag = _get_noskip_flag,
                    215:                                .set_noskip_flag = _set_noskip_flag,
                    216:                                .build = _build,
                    217:                                .process = _process,
                    218:                                .add_segment = _add_segment,
                    219:                                .get_ref = _get_ref,
                    220:                                .destroy = _destroy,
                    221:                        },
                    222:                        .get_result = _get_result,
                    223:                },
                    224:                .type = { PEN_IETF, IETF_ATTR_ASSESSMENT_RESULT },
                    225:                .length = length,
                    226:                .value = chunk_clone(data),
                    227:                .ref = 1,
                    228:        );
                    229: 
                    230:        return &this->public.pa_tnc_attribute;
                    231: }
                    232: 

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