Annotation of embedaddon/strongswan/src/libimcv/pts/pts_pcr.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Copyright (C) 2012 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 "pts_pcr.h"
        !            17: 
        !            18: #include <utils/debug.h>
        !            19: 
        !            20: #include <stdarg.h>
        !            21: 
        !            22: typedef struct private_pts_pcr_t private_pts_pcr_t;
        !            23: 
        !            24: /**
        !            25:  * Private data of a pts_pcr_t object.
        !            26:  *
        !            27:  */
        !            28: struct private_pts_pcr_t {
        !            29: 
        !            30:        /**
        !            31:         * Public pts_pcr_t interface.
        !            32:         */
        !            33:        pts_pcr_t public;
        !            34: 
        !            35:        /**
        !            36:         * Shadow PCR registers
        !            37:         */
        !            38:        chunk_t pcrs[PTS_PCR_MAX_NUM];
        !            39: 
        !            40:        /**
        !            41:         * Number of extended PCR registers
        !            42:         */
        !            43:        uint32_t pcr_count;
        !            44: 
        !            45:        /**
        !            46:         * Highest extended PCR register
        !            47:         */
        !            48:        uint32_t pcr_max;
        !            49: 
        !            50:        /**
        !            51:         * Bitmap of extended PCR registers
        !            52:         */
        !            53:        uint8_t pcr_select[PTS_PCR_MAX_NUM / 8];
        !            54: 
        !            55:        /**
        !            56:         * Hasher used to extend shadow PCRs
        !            57:         */
        !            58:        hasher_t *hasher;
        !            59: 
        !            60: };
        !            61: 
        !            62: METHOD(pts_pcr_t, get_count, uint32_t,
        !            63:        private_pts_pcr_t *this)
        !            64: {
        !            65:        return this->pcr_count;
        !            66: }
        !            67: 
        !            68: METHOD(pts_pcr_t, select_pcr, bool,
        !            69:        private_pts_pcr_t *this, uint32_t pcr)
        !            70: {
        !            71:        uint32_t i, f;
        !            72: 
        !            73:        if (pcr >= PTS_PCR_MAX_NUM)
        !            74:        {
        !            75:                DBG1(DBG_PTS, "PCR %2u: number is larger than maximum of %u",
        !            76:                                           pcr, PTS_PCR_MAX_NUM-1);
        !            77:                return FALSE;
        !            78:        }
        !            79: 
        !            80:        /* Determine PCR selection flag */
        !            81:        i = pcr / 8;
        !            82:        f = 1 << (pcr - 8*i);
        !            83: 
        !            84:        /* Has this PCR already been selected? */
        !            85:        if (!(this->pcr_select[i] & f))
        !            86:        {
        !            87:                this->pcr_select[i] |= f;
        !            88:                this->pcr_max = max(this->pcr_max, pcr);
        !            89:                this->pcr_count++;
        !            90:        }
        !            91:        return TRUE;
        !            92: }
        !            93: 
        !            94: METHOD(pts_pcr_t, get_selection_size, size_t,
        !            95:        private_pts_pcr_t *this)
        !            96: {
        !            97: 
        !            98:        /**
        !            99:         * A TPM v1.2 has 24 PCR Registers so the bitmask field length
        !           100:         * used by TrouSerS is at least 3 bytes
        !           101:         */
        !           102:        return PTS_PCR_MAX_NUM / 8;
        !           103: }
        !           104: 
        !           105: typedef struct {
        !           106:        /** implements enumerator_t */
        !           107:        enumerator_t public;
        !           108:        /** current PCR */
        !           109:        uint32_t pcr;
        !           110:        /** back reference to parent */
        !           111:        private_pts_pcr_t *pcrs;
        !           112: } pcr_enumerator_t;
        !           113: 
        !           114: METHOD(enumerator_t, pcr_enumerator_enumerate, bool,
        !           115:        pcr_enumerator_t *this, va_list args)
        !           116: {
        !           117:        uint32_t i, f, *pcr;
        !           118: 
        !           119:        VA_ARGS_VGET(args, pcr);
        !           120: 
        !           121:        while (this->pcr <= this->pcrs->pcr_max)
        !           122:        {
        !           123:                /* Determine PCR selection flag */
        !           124:                i = this->pcr / 8;
        !           125:                f = 1 << (this->pcr - 8*i);
        !           126: 
        !           127:                /* Assign current PCR to output argument and increase */
        !           128:                *pcr = this->pcr++;
        !           129: 
        !           130:                /* return if PCR is selected */
        !           131:                if (this->pcrs->pcr_select[i] & f)
        !           132:                {
        !           133:                        return TRUE;
        !           134:                }
        !           135:        }
        !           136:        return FALSE;
        !           137: }
        !           138: 
        !           139: METHOD(pts_pcr_t, create_enumerator, enumerator_t*,
        !           140:        private_pts_pcr_t *this)
        !           141: {
        !           142:        pcr_enumerator_t *enumerator;
        !           143: 
        !           144:        INIT(enumerator,
        !           145:                .public = {
        !           146:                        .enumerate = enumerator_enumerate_default,
        !           147:                        .venumerate = _pcr_enumerator_enumerate,
        !           148:                        .destroy = (void*)free,
        !           149:                },
        !           150:                .pcrs = this,
        !           151:        );
        !           152: 
        !           153:        return (enumerator_t*)enumerator;
        !           154: }
        !           155: 
        !           156: METHOD(pts_pcr_t, get, chunk_t,
        !           157:        private_pts_pcr_t *this, uint32_t pcr)
        !           158: {
        !           159:        return (pcr < PTS_PCR_MAX_NUM) ? this->pcrs[pcr] : chunk_empty;
        !           160: }
        !           161: 
        !           162: METHOD(pts_pcr_t, set, bool,
        !           163:        private_pts_pcr_t *this, uint32_t pcr, chunk_t value)
        !           164: {
        !           165:        if (value.len != PTS_PCR_LEN)
        !           166:        {
        !           167:                DBG1(DBG_PTS, "PCR %2u: value does not fit", pcr);
        !           168:                return FALSE;
        !           169:        }
        !           170:        if (select_pcr(this, pcr))
        !           171:        {
        !           172:                memcpy(this->pcrs[pcr].ptr, value.ptr, PTS_PCR_LEN);
        !           173:                return TRUE;
        !           174:        }
        !           175:        return FALSE;
        !           176: }
        !           177: 
        !           178: METHOD(pts_pcr_t, extend, chunk_t,
        !           179:        private_pts_pcr_t *this, uint32_t pcr, chunk_t measurement)
        !           180: {
        !           181:        if (measurement.len != PTS_PCR_LEN)
        !           182:        {
        !           183:                DBG1(DBG_PTS, "PCR %2u: measurement does not fit", pcr);
        !           184:                return chunk_empty;
        !           185:        }
        !           186:        if (!select_pcr(this, pcr))
        !           187:        {
        !           188:                return chunk_empty;
        !           189:        }
        !           190:        if (!this->hasher->get_hash(this->hasher, this->pcrs[pcr] , NULL) ||
        !           191:                !this->hasher->get_hash(this->hasher, measurement, this->pcrs[pcr].ptr))
        !           192:        {
        !           193:                DBG1(DBG_PTS, "PCR %2u: not extended due to hasher problem", pcr);
        !           194:                return chunk_empty;
        !           195:        }
        !           196:        return this->pcrs[pcr];
        !           197: }
        !           198: 
        !           199: METHOD(pts_pcr_t, get_composite, tpm_tss_pcr_composite_t*,
        !           200:        private_pts_pcr_t *this)
        !           201: {
        !           202:        tpm_tss_pcr_composite_t *pcr_composite;
        !           203:        enumerator_t *enumerator;
        !           204:        uint16_t selection_size;
        !           205:        uint32_t pcr_field_size, pcr;
        !           206:        u_char *pos;
        !           207: 
        !           208:        selection_size = get_selection_size(this);
        !           209:        pcr_field_size = this->pcr_count * PTS_PCR_LEN;
        !           210: 
        !           211:        INIT(pcr_composite,
        !           212:                .pcr_select    = chunk_alloc(selection_size),
        !           213:                .pcr_composite = chunk_alloc(pcr_field_size),
        !           214:        );
        !           215: 
        !           216:        memcpy(pcr_composite->pcr_select.ptr, this->pcr_select, selection_size);
        !           217:        pos = pcr_composite->pcr_composite.ptr;
        !           218: 
        !           219:        enumerator = create_enumerator(this);
        !           220:        while (enumerator->enumerate(enumerator, &pcr))
        !           221:        {
        !           222:                memcpy(pos, this->pcrs[pcr].ptr, PTS_PCR_LEN);
        !           223:                pos += PTS_PCR_LEN;
        !           224:        }
        !           225:        enumerator->destroy(enumerator);
        !           226: 
        !           227:        return pcr_composite;
        !           228: }
        !           229: 
        !           230: METHOD(pts_pcr_t, destroy, void,
        !           231:        private_pts_pcr_t *this)
        !           232: {
        !           233:        uint32_t i;
        !           234: 
        !           235:        for (i = 0; i < PTS_PCR_MAX_NUM; i++)
        !           236:        {
        !           237:                free(this->pcrs[i].ptr);
        !           238:        }
        !           239:        this->hasher->destroy(this->hasher);
        !           240:        free(this);
        !           241: }
        !           242: 
        !           243: /**
        !           244:  * See header
        !           245:  */
        !           246: pts_pcr_t *pts_pcr_create(void)
        !           247: {
        !           248:        private_pts_pcr_t *this;
        !           249:        hasher_t *hasher;
        !           250:        uint32_t i;
        !           251: 
        !           252:        hasher = lib->crypto->create_hasher(lib->crypto, HASH_SHA1);
        !           253:        if (!hasher)
        !           254:        {
        !           255:                DBG1(DBG_PTS, "%N hasher could not be created",
        !           256:                         hash_algorithm_short_names, HASH_SHA1);
        !           257:                return NULL;
        !           258:        }
        !           259: 
        !           260:        INIT(this,
        !           261:                .public = {
        !           262:                        .get_count = _get_count,
        !           263:                        .select_pcr = _select_pcr,
        !           264:                        .get_selection_size = _get_selection_size,
        !           265:                        .create_enumerator = _create_enumerator,
        !           266:                        .get = _get,
        !           267:                        .set = _set,
        !           268:                        .extend = _extend,
        !           269:                        .get_composite = _get_composite,
        !           270:                        .destroy = _destroy,
        !           271:                },
        !           272:                .hasher = hasher,
        !           273:        );
        !           274: 
        !           275:        for (i = 0; i < PTS_PCR_MAX_NUM; i++)
        !           276:        {
        !           277:                this->pcrs[i] = chunk_alloc(PTS_PCR_LEN);
        !           278:                memset(this->pcrs[i].ptr, 0x00, PTS_PCR_LEN);
        !           279:        }
        !           280: 
        !           281:        return &this->public;
        !           282: }
        !           283: 

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