LCOV - code coverage report
Current view: top level - contrib/pgcrypto - px-hmac.c (source / functions) Hit Total Coverage
Test: PostgreSQL 13beta1 Lines: 63 73 86.3 %
Date: 2020-06-01 08:06:25 Functions: 6 8 75.0 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : /*
       2             :  * px-hmac.c
       3             :  *      HMAC implementation.
       4             :  *
       5             :  * Copyright (c) 2001 Marko Kreen
       6             :  * All rights reserved.
       7             :  *
       8             :  * Redistribution and use in source and binary forms, with or without
       9             :  * modification, are permitted provided that the following conditions
      10             :  * are met:
      11             :  * 1. Redistributions of source code must retain the above copyright
      12             :  *    notice, this list of conditions and the following disclaimer.
      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             :  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
      18             :  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
      19             :  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
      20             :  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
      21             :  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
      22             :  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
      23             :  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
      24             :  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
      25             :  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
      26             :  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
      27             :  * SUCH DAMAGE.
      28             :  *
      29             :  * contrib/pgcrypto/px-hmac.c
      30             :  */
      31             : 
      32             : #include "postgres.h"
      33             : 
      34             : #include "px.h"
      35             : 
      36             : #define HMAC_IPAD 0x36
      37             : #define HMAC_OPAD 0x5C
      38             : 
      39             : static unsigned
      40          28 : hmac_result_size(PX_HMAC *h)
      41             : {
      42          28 :     return px_md_result_size(h->md);
      43             : }
      44             : 
      45             : static unsigned
      46           0 : hmac_block_size(PX_HMAC *h)
      47             : {
      48           0 :     return px_md_block_size(h->md);
      49             : }
      50             : 
      51             : static void
      52          28 : hmac_init(PX_HMAC *h, const uint8 *key, unsigned klen)
      53             : {
      54             :     unsigned    bs,
      55             :                 i;
      56             :     uint8      *keybuf;
      57          28 :     PX_MD      *md = h->md;
      58             : 
      59          28 :     bs = px_md_block_size(md);
      60          28 :     keybuf = px_alloc(bs);
      61          28 :     memset(keybuf, 0, bs);
      62             : 
      63          28 :     if (klen > bs)
      64             :     {
      65           8 :         px_md_update(md, key, klen);
      66           8 :         px_md_finish(md, keybuf);
      67           8 :         px_md_reset(md);
      68             :     }
      69             :     else
      70          20 :         memcpy(keybuf, key, klen);
      71             : 
      72        1820 :     for (i = 0; i < bs; i++)
      73             :     {
      74        1792 :         h->p.ipad[i] = keybuf[i] ^ HMAC_IPAD;
      75        1792 :         h->p.opad[i] = keybuf[i] ^ HMAC_OPAD;
      76             :     }
      77             : 
      78          28 :     px_memset(keybuf, 0, bs);
      79          28 :     px_free(keybuf);
      80             : 
      81          28 :     px_md_update(md, h->p.ipad, bs);
      82          28 : }
      83             : 
      84             : static void
      85           0 : hmac_reset(PX_HMAC *h)
      86             : {
      87           0 :     PX_MD      *md = h->md;
      88           0 :     unsigned    bs = px_md_block_size(md);
      89             : 
      90           0 :     px_md_reset(md);
      91           0 :     px_md_update(md, h->p.ipad, bs);
      92           0 : }
      93             : 
      94             : static void
      95          28 : hmac_update(PX_HMAC *h, const uint8 *data, unsigned dlen)
      96             : {
      97          28 :     px_md_update(h->md, data, dlen);
      98          28 : }
      99             : 
     100             : static void
     101          28 : hmac_finish(PX_HMAC *h, uint8 *dst)
     102             : {
     103          28 :     PX_MD      *md = h->md;
     104             :     unsigned    bs,
     105             :                 hlen;
     106             :     uint8      *buf;
     107             : 
     108          28 :     bs = px_md_block_size(md);
     109          28 :     hlen = px_md_result_size(md);
     110             : 
     111          28 :     buf = px_alloc(hlen);
     112             : 
     113          28 :     px_md_finish(md, buf);
     114             : 
     115          28 :     px_md_reset(md);
     116          28 :     px_md_update(md, h->p.opad, bs);
     117          28 :     px_md_update(md, buf, hlen);
     118          28 :     px_md_finish(md, dst);
     119             : 
     120          28 :     px_memset(buf, 0, hlen);
     121          28 :     px_free(buf);
     122          28 : }
     123             : 
     124             : static void
     125          28 : hmac_free(PX_HMAC *h)
     126             : {
     127             :     unsigned    bs;
     128             : 
     129          28 :     bs = px_md_block_size(h->md);
     130          28 :     px_md_free(h->md);
     131             : 
     132          28 :     px_memset(h->p.ipad, 0, bs);
     133          28 :     px_memset(h->p.opad, 0, bs);
     134          28 :     px_free(h->p.ipad);
     135          28 :     px_free(h->p.opad);
     136          28 :     px_free(h);
     137          28 : }
     138             : 
     139             : 
     140             : /* PUBLIC FUNCTIONS */
     141             : 
     142             : int
     143          30 : px_find_hmac(const char *name, PX_HMAC **res)
     144             : {
     145             :     int         err;
     146             :     PX_MD      *md;
     147             :     PX_HMAC    *h;
     148             :     unsigned    bs;
     149             : 
     150          30 :     err = px_find_digest(name, &md);
     151          30 :     if (err)
     152           2 :         return err;
     153             : 
     154          28 :     bs = px_md_block_size(md);
     155          28 :     if (bs < 2)
     156             :     {
     157           0 :         px_md_free(md);
     158           0 :         return PXE_HASH_UNUSABLE_FOR_HMAC;
     159             :     }
     160             : 
     161          28 :     h = px_alloc(sizeof(*h));
     162          28 :     h->p.ipad = px_alloc(bs);
     163          28 :     h->p.opad = px_alloc(bs);
     164          28 :     h->md = md;
     165             : 
     166          28 :     h->result_size = hmac_result_size;
     167          28 :     h->block_size = hmac_block_size;
     168          28 :     h->reset = hmac_reset;
     169          28 :     h->update = hmac_update;
     170          28 :     h->finish = hmac_finish;
     171          28 :     h->free = hmac_free;
     172          28 :     h->init = hmac_init;
     173             : 
     174          28 :     *res = h;
     175             : 
     176          28 :     return 0;
     177             : }

Generated by: LCOV version 1.13