LCOV - code coverage report
Current view: top level - third_party/heimdal/lib/krb5 - store_mem.c (source / functions) Hit Total Coverage
Test: coverage report for master 98b443d9 Lines: 66 92 71.7 %
Date: 2024-05-31 13:13:24 Functions: 5 9 55.6 %

          Line data    Source code
       1             : /*
       2             :  * Copyright (c) 1997 - 2000, 2002 Kungliga Tekniska Högskolan
       3             :  * (Royal Institute of Technology, Stockholm, Sweden).
       4             :  * All rights reserved.
       5             :  *
       6             :  * Redistribution and use in source and binary forms, with or without
       7             :  * modification, are permitted provided that the following conditions
       8             :  * are met:
       9             :  *
      10             :  * 1. Redistributions of source code must retain the above copyright
      11             :  *    notice, this list of conditions and the following disclaimer.
      12             :  *
      13             :  * 2. Redistributions in binary form must reproduce the above copyright
      14             :  *    notice, this list of conditions and the following disclaimer in the
      15             :  *    documentation and/or other materials provided with the distribution.
      16             :  *
      17             :  * 3. Neither the name of the Institute nor the names of its contributors
      18             :  *    may be used to endorse or promote products derived from this software
      19             :  *    without specific prior written permission.
      20             :  *
      21             :  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
      22             :  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
      23             :  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
      24             :  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
      25             :  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
      26             :  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
      27             :  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
      28             :  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
      29             :  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
      30             :  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
      31             :  * SUCH DAMAGE.
      32             :  */
      33             : 
      34             : #include "krb5_locl.h"
      35             : #include "store-int.h"
      36             : 
      37             : typedef struct mem_storage{
      38             :     unsigned char *base;
      39             :     size_t size;
      40             :     unsigned char *ptr;
      41             : }mem_storage;
      42             : 
      43             : static ssize_t
      44     7580026 : mem_fetch(krb5_storage *sp, void *data, size_t size)
      45             : {
      46     7580026 :     mem_storage *s = (mem_storage*)sp->data;
      47     7580026 :     if(size > (size_t)(s->base + s->size - s->ptr))
      48       23086 :         size = s->base + s->size - s->ptr;
      49     7580026 :     memmove(data, s->ptr, size);
      50     7580026 :     sp->seek(sp, size, SEEK_CUR);
      51     7580026 :     return size;
      52             : }
      53             : 
      54             : static ssize_t
      55        4439 : mem_store(krb5_storage *sp, const void *data, size_t size)
      56             : {
      57        4439 :     mem_storage *s = (mem_storage*)sp->data;
      58        4439 :     if(size > (size_t)(s->base + s->size - s->ptr))
      59           0 :         size = s->base + s->size - s->ptr;
      60        4439 :     memmove(s->ptr, data, size);
      61        4439 :     sp->seek(sp, size, SEEK_CUR);
      62        4439 :     return size;
      63             : }
      64             : 
      65             : static ssize_t
      66           0 : mem_no_store(krb5_storage *sp, const void *data, size_t size)
      67             : {
      68           0 :     return -1;
      69             : }
      70             : 
      71             : static off_t
      72    17864534 : mem_seek(krb5_storage *sp, off_t offset, int whence)
      73             : {
      74    17864534 :     mem_storage *s = (mem_storage*)sp->data;
      75    17864534 :     switch(whence){
      76     9346735 :     case SEEK_SET:
      77     9346735 :         if((size_t)offset > s->size)
      78           0 :             offset = s->size;
      79     9346735 :         if(offset < 0)
      80           0 :             offset = 0;
      81     9346735 :         s->ptr = s->base + offset;
      82     9346735 :         break;
      83     8103331 :     case SEEK_CUR:
      84     8103331 :         return sp->seek(sp, s->ptr - s->base + offset, SEEK_SET);
      85      414468 :     case SEEK_END:
      86      414468 :         return sp->seek(sp, s->size + offset, SEEK_SET);
      87           0 :     default:
      88           0 :         errno = EINVAL;
      89           0 :         return -1;
      90             :     }
      91     9346735 :     return s->ptr - s->base;
      92             : }
      93             : 
      94             : static int
      95           0 : mem_trunc(krb5_storage *sp, off_t offset)
      96             : {
      97           0 :     mem_storage *s = (mem_storage*)sp->data;
      98           0 :     if((size_t)offset > s->size)
      99           0 :         return ERANGE;
     100           0 :     s->size = offset;
     101           0 :     if ((s->ptr - s->base) > offset)
     102           0 :         s->ptr = s->base + offset;
     103           0 :     return 0;
     104             : }
     105             : 
     106             : static int
     107           0 : mem_no_trunc(krb5_storage *sp, off_t offset)
     108             : {
     109           0 :     return EINVAL;
     110             : }
     111             : 
     112             : /**
     113             :  * Create a fixed size memory storage block
     114             :  *
     115             :  * @return A krb5_storage on success, or NULL on out of memory error.
     116             :  *
     117             :  * @ingroup krb5_storage
     118             :  *
     119             :  * @sa krb5_storage_mem()
     120             :  * @sa krb5_storage_from_readonly_mem()
     121             :  * @sa krb5_storage_from_data()
     122             :  * @sa krb5_storage_from_fd()
     123             :  * @sa krb5_storage_from_socket()
     124             :  */
     125             : 
     126             : KRB5_LIB_FUNCTION krb5_storage * KRB5_LIB_CALL
     127      294173 : krb5_storage_from_mem(void *buf, size_t len)
     128             : {
     129      294173 :     krb5_storage *sp = malloc(sizeof(krb5_storage));
     130        6913 :     mem_storage *s;
     131      294173 :     if(sp == NULL)
     132           0 :         return NULL;
     133      294173 :     s = malloc(sizeof(*s));
     134      294173 :     if(s == NULL) {
     135           0 :         free(sp);
     136           0 :         return NULL;
     137             :     }
     138      294173 :     sp->data = s;
     139      294173 :     sp->flags = 0;
     140      294173 :     sp->eof_code = HEIM_ERR_EOF;
     141      294173 :     s->base = buf;
     142      294173 :     s->size = len;
     143      294173 :     s->ptr = buf;
     144      294173 :     sp->fetch = mem_fetch;
     145      294173 :     sp->store = mem_store;
     146      294173 :     sp->seek = mem_seek;
     147      294173 :     sp->trunc = mem_trunc;
     148      294173 :     sp->fsync = NULL;
     149      294173 :     sp->free = NULL;
     150      294173 :     sp->max_alloc = UINT32_MAX/64;
     151      294173 :     return sp;
     152             : }
     153             : 
     154             : /**
     155             :  * Create a fixed size memory storage block
     156             :  *
     157             :  * @return A krb5_storage on success, or NULL on out of memory error.
     158             :  *
     159             :  * @ingroup krb5_storage
     160             :  *
     161             :  * @sa krb5_storage_mem()
     162             :  * @sa krb5_storage_from_mem()
     163             :  * @sa krb5_storage_from_readonly_mem()
     164             :  * @sa krb5_storage_from_fd()
     165             :  */
     166             : 
     167             : KRB5_LIB_FUNCTION krb5_storage * KRB5_LIB_CALL
     168           0 : krb5_storage_from_data(krb5_data *data)
     169             : {
     170           0 :     return krb5_storage_from_mem(data->data, data->length);
     171             : }
     172             : 
     173             : /**
     174             :  * Create a fixed size memory storage block that is read only
     175             :  *
     176             :  * @return A krb5_storage on success, or NULL on out of memory error.
     177             :  *
     178             :  * @ingroup krb5_storage
     179             :  *
     180             :  * @sa krb5_storage_mem()
     181             :  * @sa krb5_storage_from_mem()
     182             :  * @sa krb5_storage_from_data()
     183             :  * @sa krb5_storage_from_fd()
     184             :  */
     185             : 
     186             : KRB5_LIB_FUNCTION krb5_storage * KRB5_LIB_CALL
     187      436979 : krb5_storage_from_readonly_mem(const void *buf, size_t len)
     188             : {
     189      436979 :     krb5_storage *sp = malloc(sizeof(krb5_storage));
     190       10242 :     mem_storage *s;
     191      436979 :     if(sp == NULL)
     192           0 :         return NULL;
     193      436979 :     s = malloc(sizeof(*s));
     194      436979 :     if(s == NULL) {
     195           0 :         free(sp);
     196           0 :         return NULL;
     197             :     }
     198      436979 :     sp->data = s;
     199      436979 :     sp->flags = 0;
     200      436979 :     sp->eof_code = HEIM_ERR_EOF;
     201      436979 :     s->base = rk_UNCONST(buf);
     202      436979 :     s->size = len;
     203      436979 :     s->ptr = rk_UNCONST(buf);
     204      436979 :     sp->fetch = mem_fetch;
     205      436979 :     sp->store = mem_no_store;
     206      436979 :     sp->seek = mem_seek;
     207      436979 :     sp->trunc = mem_no_trunc;
     208      436979 :     sp->fsync = NULL;
     209      436979 :     sp->free = NULL;
     210      436979 :     sp->max_alloc = UINT32_MAX/64;
     211      436979 :     return sp;
     212             : }

Generated by: LCOV version 1.14