Return to mock_nonce_gen.c CVS log

Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / strongswan / src / libcharon / tests / utils

Annotation of embedaddon/strongswan/src/libcharon/tests/utils/mock_nonce_gen.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2016 Tobias Brunner
        !             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 "mock_nonce_gen.h"
        !            17: 
        !            18: typedef struct private_nonce_gen_t private_nonce_gen_t;
        !            19: 
        !            20: struct private_nonce_gen_t {
        !            21: 
        !            22:        /**
        !            23:         * Public interface
        !            24:         */
        !            25:        nonce_gen_t public;
        !            26: 
        !            27:        /**
        !            28:         * Random number generator
        !            29:         */
        !            30:        rng_t* rng;
        !            31: 
        !            32:        /**
        !            33:         * First byte to set to the nonces
        !            34:         */
        !            35:        u_char first;
        !            36: };
        !            37: 
        !            38: METHOD(nonce_gen_t, get_nonce, bool,
        !            39:        private_nonce_gen_t *this, size_t size, uint8_t *buffer)
        !            40: {
        !            41:        if (size > 0)
        !            42:        {
        !            43:                buffer[0] = this->first;
        !            44:                buffer++;
        !            45:                size--;
        !            46:        }
        !            47:        return this->rng->get_bytes(this->rng, size, buffer);
        !            48: }
        !            49: 
        !            50: METHOD(nonce_gen_t, allocate_nonce, bool,
        !            51:        private_nonce_gen_t *this, size_t size, chunk_t *chunk)
        !            52: {
        !            53:        *chunk = chunk_alloc(size);
        !            54:        if (!get_nonce(this, chunk->len, chunk->ptr))
        !            55:        {
        !            56:                chunk_free(chunk);
        !            57:                return FALSE;
        !            58:        }
        !            59:        return TRUE;
        !            60: }
        !            61: 
        !            62: METHOD(nonce_gen_t, destroy, void,
        !            63:        private_nonce_gen_t *this)
        !            64: {
        !            65:        DESTROY_IF(this->rng);
        !            66:        free(this);
        !            67: }
        !            68: 
        !            69: /*
        !            70:  * Described in header
        !            71:  */
        !            72: nonce_gen_t *mock_nonce_gen_create(u_char first)
        !            73: {
        !            74:        private_nonce_gen_t *this;
        !            75: 
        !            76:        INIT(this,
        !            77:                .public = {
        !            78:                        .get_nonce = _get_nonce,
        !            79:                        .allocate_nonce = _allocate_nonce,
        !            80:                        .destroy = _destroy,
        !            81:                },
        !            82:                .rng = lib->crypto->create_rng(lib->crypto, RNG_WEAK),
        !            83:                .first = first,
        !            84:        );
        !            85:        if (!this->rng)
        !            86:        {
        !            87:                destroy(this);
        !            88:                return NULL;
        !            89:        }
        !            90:        return &this->public;
        !            91: }