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

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: }